[mirror_edk2.git] / EdkCompatibilityPkg / Sample / Tools / Source / ModifyInf / ModifyInf.c

/*++\r
\r
Copyright (c) 1999 - 2010, Intel Corporation. All rights reserved.<BR>\r
This program and the accompanying materials                          \r
are licensed and made available under the terms and conditions of the BSD License         \r
which accompanies this distribution.  The full text of the license may be found at        \r
http://opensource.org/licenses/bsd-license.php                                            \r
                                                                                          \r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,                     \r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.             \r
\r
Module Name:\r
\r
  ModifyInf.c\r
\r
Abstract:\r
\r
  It is a simple tool to modify some fields in a FV inf file \r
  and output a new FV inf file.  \r
\r
--*/\r
\r
#include "stdio.h"\r
#include "string.h"\r
\r
#define UTILITY_NAME    "ModifyInf"\r
#define UTILITY_VERSION "v1.0"\r
//\r
// Read a line into buffer including '\r\n'\r
//\r
int\r
ReadLine (\r
  char *LineBuffer,\r
  FILE *fp\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  LineBuffer  - GC_TODO: add argument description\r
  fp          - GC_TODO: add argument description\r
\r
Returns:\r
\r
  GC_TODO: add return values\r
\r
--*/\r
{\r
  int   CharC;\r
  char  *Line;\r
\r
  Line = LineBuffer;\r
\r
  while ((CharC = fgetc (fp)) != EOF) {\r
    *Line++ = (char) CharC;\r
    if (CharC == 0x0a) {\r
      break;\r
    }\r
  }\r
\r
  *Line = 0;\r
\r
  if (CharC == EOF) {\r
    return 0;\r
  } else {\r
    return 1;\r
  }\r
\r
}\r
//\r
// Write a line into output file\r
//\r
int\r
WriteLine (\r
  char *Line,\r
  FILE *fp\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  Line  - GC_TODO: add argument description\r
  fp    - GC_TODO: add argument description\r
\r
Returns:\r
\r
  GC_TODO: add return values\r
\r
--*/\r
{\r
  fwrite (Line, strlen (Line), 1, fp);\r
  return 0;\r
}\r
//\r
// Apply patterns to a line\r
// Currently there are 2 patterns to support\r
// '==' replace a field value with a new value\r
// '+=' append a string at the end of original line\r
// '-'  prevent the line from applying any patterns\r
//      it has the highest priority\r
//\r
int\r
ApplyPattern (\r
  char *Line,\r
  char *argv[],\r
  int  argc\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  Line  - GC_TODO: add argument description\r
  ]     - GC_TODO: add argument description\r
  argc  - GC_TODO: add argument description\r
\r
Returns:\r
\r
  GC_TODO: add return values\r
\r
--*/\r
{\r
  static char Section[256];\r
  int         SectionLength;\r
  char        PatternBuffer[256];\r
  char        *Pattern;\r
  char        *Pattern1;\r
  char        *Pattern2;\r
  int         PatternNum;\r
  char        *Ptr;\r
\r
  Pattern     = PatternBuffer;\r
\r
  PatternNum  = argc;\r
\r
  //\r
  // For section field\r
  // record current scope section into static buffer\r
  //\r
  Ptr = Line;\r
  if (*Ptr == '[') {\r
    while (*Ptr != ']') {\r
      if (!(*Ptr++)) {\r
        return -1;\r
      }\r
    }\r
    SectionLength = Ptr - Line + 1;\r
    SectionLength = SectionLength > 255 ? 255 : SectionLength;\r
    strncpy (Section, Line, SectionLength);\r
    Section[SectionLength] = 0;\r
  }\r
  //\r
  // Apply each pattern on the line\r
  //\r
  while (PatternNum-- > 3) {\r
\r
    strcpy (Pattern, argv[PatternNum]);\r
\r
    //\r
    // For pattern '-'\r
    // keep it unmodified by other patterns\r
    //\r
    if (*Pattern == '-') {\r
      if (strstr (Line, Pattern + 1)) {\r
        return 0;\r
      } else {\r
        continue;\r
      }\r
    }\r
    //\r
    // For other patterns\r
    // get its section at first if it has\r
    //\r
    if (*Pattern == '[') {\r
      if (strncmp (Section, Pattern, strlen (Section))) {\r
        //\r
        // This pattern can't be appied for current section\r
        //\r
        continue;\r
      }\r
      //\r
      // Strip the section field\r
      //\r
      while (*Pattern != ']') {\r
        if (!(*Pattern++)) {\r
          return -1;\r
        }\r
      }\r
\r
      Pattern++;\r
    }\r
    //\r
    // Apply patterns\r
    //\r
    Pattern1  = strstr (Pattern, "==");\r
    Pattern2  = strstr (Pattern, "+=");\r
    if (Pattern1) {\r
      //\r
      // For pattern '=='\r
      // replace the field value with a new string\r
      //\r
      if (!strncmp (Line, Pattern, Pattern1 - Pattern)) {\r
        Pattern1 += 2;\r
        Ptr = strstr (Line, "=");\r
        if (!Ptr) {\r
          return -1;\r
        }\r
\r
        while (*(++Ptr) == ' ')\r
          ;\r
        *Ptr = 0;\r
        strcat (Line, Pattern1);\r
        strcat (Line, "\r\n");\r
      }\r
    } else if (Pattern2) {\r
      //\r
      // For pattern '+='\r
      // append a string at end of the original string\r
      //\r
      if (!strncmp (Line, Pattern, Pattern2 - Pattern)) {\r
        Pattern2 += 2;\r
        Ptr = Line;\r
        while (*Ptr != 0x0D && *Ptr != 0x0A) {\r
          Ptr++;\r
        }\r
\r
        *Ptr = 0;\r
        strcat (Line, Pattern2);\r
        strcat (Line, "\r\n");\r
      }\r
    }\r
  }\r
\r
  return 0;\r
}\r
\r
void\r
Usage (\r
  void\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  None\r
\r
Returns:\r
\r
  GC_TODO: add return values\r
\r
--*/\r
{\r
  int         Index;\r
  const char  *Str[] = {\r
    UTILITY_NAME" "UTILITY_VERSION" - Intel Modify INF File Utility",\r
    "  Copyright (C), 1999 - 2008 Intel Corporation",\r
    \r
#if ( defined(UTILITY_BUILD) && defined(UTILITY_VENDOR) )\r
    "  Built from "UTILITY_BUILD", project of "UTILITY_VENDOR,\r
#endif\r
    "",\r
    "Usage:",\r
    "  "UTILITY_NAME" SOURCE DEST [PATTERN]",\r
    NULL\r
  };\r
  for (Index = 0; Str[Index] != NULL; Index++) {\r
    fprintf (stdout, "%s\n", Str[Index]);\r
  }  \r
}\r
\r
int\r
main (\r
  int argc,\r
  char*argv[]\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  argc  - GC_TODO: add argument description\r
  ]     - GC_TODO: add argument description\r
\r
Returns:\r
\r
  GC_TODO: add return values\r
\r
--*/\r
{\r
  char  LineBuffer[256];\r
  FILE  *fpin;\r
  FILE  *fpout;\r
\r
  if (argc < 3) {\r
    Usage ();\r
    return -1;\r
  }\r
\r
  fpin = fopen (argv[1], "rb");\r
  if (!fpin) {\r
    printf ("Can't open input file!\r\n");\r
    return -1;\r
  }\r
\r
  fpout = fopen (argv[2], "wb");\r
  if (!fpout) {\r
    fclose (fpin);\r
    printf ("Can't create output file!\r\n");\r
    return -1;\r
  }\r
\r
  while (ReadLine (LineBuffer, fpin)) {\r
    ApplyPattern (LineBuffer, argv, argc);\r
    WriteLine (LineBuffer, fpout);\r
  }\r
\r
  fclose (fpin);\r
  fclose (fpout);\r
\r
  return 0;\r
}\r
Commit	Line	Data
3eb9473e	1	/*++\r
3eb9473e	2	\r
3e99020d	3	Copyright (c) 1999 - 2010, Intel Corporation. All rights reserved.<BR>\r
4b1e1121	4	This program and the accompanying materials \r
3eb9473e	5	are licensed and made available under the terms and conditions of the BSD License \r
	6	which accompanies this distribution. The full text of the license may be found at \r
	7	http://opensource.org/licenses/bsd-license.php \r
	8	\r
	9	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS, \r
	10	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. \r
	11	\r
	12	Module Name:\r
	13	\r
	14	ModifyInf.c\r
	15	\r
	16	Abstract:\r
	17	\r
	18	It is a simple tool to modify some fields in a FV inf file \r
	19	and output a new FV inf file. \r
	20	\r
	21	--*/\r
	22	\r
	23	#include "stdio.h"\r
	24	#include "string.h"\r
	25	\r
3e99020d LG	26	#define UTILITY_NAME "ModifyInf"\r
3e99020d LG	27	#define UTILITY_VERSION "v1.0"\r
3eb9473e	28	//\r
	29	// Read a line into buffer including '\r\n'\r
	30	//\r
	31	int\r
	32	ReadLine (\r
	33	char *LineBuffer,\r
	34	FILE *fp\r
	35	)\r
	36	/*++\r
	37	\r
	38	Routine Description:\r
	39	\r
	40	GC_TODO: Add function description\r
	41	\r
	42	Arguments:\r
	43	\r
	44	LineBuffer - GC_TODO: add argument description\r
	45	fp - GC_TODO: add argument description\r
	46	\r
	47	Returns:\r
	48	\r
	49	GC_TODO: add return values\r
	50	\r
	51	--*/\r
	52	{\r
	53	int CharC;\r
	54	char *Line;\r
	55	\r
	56	Line = LineBuffer;\r
	57	\r
	58	while ((CharC = fgetc (fp)) != EOF) {\r
	59	*Line++ = (char) CharC;\r
	60	if (CharC == 0x0a) {\r
	61	break;\r
	62	}\r
	63	}\r
	64	\r
	65	*Line = 0;\r
	66	\r
	67	if (CharC == EOF) {\r
	68	return 0;\r
	69	} else {\r
	70	return 1;\r
	71	}\r
	72	\r
	73	}\r
	74	//\r
	75	// Write a line into output file\r
	76	//\r
	77	int\r
	78	WriteLine (\r
	79	char *Line,\r
	80	FILE *fp\r
	81	)\r
	82	/*++\r
	83	\r
	84	Routine Description:\r
	85	\r
	86	GC_TODO: Add function description\r
	87	\r
	88	Arguments:\r
	89	\r
	90	Line - GC_TODO: add argument description\r
	91	fp - GC_TODO: add argument description\r
92	\r
93	Returns:\r
94	\r
95	GC_TODO: add return values\r
96	\r
97	--*/\r
98	{\r
99	fwrite (Line, strlen (Line), 1, fp);\r
100	return 0;\r
101	}\r
102	//\r
103	// Apply patterns to a line\r
104	// Currently there are 2 patterns to support\r
105	// '==' replace a field value with a new value\r
106	// '+=' append a string at the end of original line\r
107	// '-' prevent the line from applying any patterns\r
108	// it has the highest priority\r
109	//\r
110	int\r
111	ApplyPattern (\r
112	char *Line,\r
113	char *argv[],\r
114	int argc\r
115	)\r
116	/*++\r
117	\r
118	Routine Description:\r
119	\r
120	GC_TODO: Add function description\r
121	\r
122	Arguments:\r
123	\r
124	Line - GC_TODO: add argument description\r
125	] - GC_TODO: add argument description\r
126	argc - GC_TODO: add argument description\r
127	\r
128	Returns:\r
129	\r
130	GC_TODO: add return values\r
131	\r
132	--*/\r
133	{\r
134	static char Section[256];\r
135	int SectionLength;\r
136	char PatternBuffer[256];\r
137	char *Pattern;\r
138	char *Pattern1;\r
139	char *Pattern2;\r
140	int PatternNum;\r
141	char *Ptr;\r
142	\r
143	Pattern = PatternBuffer;\r
144	\r
145	PatternNum = argc;\r
146	\r
147	//\r
148	// For section field\r
149	// record current scope section into static buffer\r
150	//\r
151	Ptr = Line;\r
152	if (*Ptr == '[') {\r
153	while (*Ptr != ']') {\r
154	if (!(*Ptr++)) {\r
155	return -1;\r
156	}\r
157	}\r
158	SectionLength = Ptr - Line + 1;\r
159	SectionLength = SectionLength > 255 ? 255 : SectionLength;\r
160	strncpy (Section, Line, SectionLength);\r
161	Section[SectionLength] = 0;\r
162	}\r
163	//\r
164	// Apply each pattern on the line\r
165	//\r
166	while (PatternNum-- > 3) {\r
167	\r
168	strcpy (Pattern, argv[PatternNum]);\r
169	\r
170	//\r
171	// For pattern '-'\r
172	// keep it unmodified by other patterns\r
173	//\r
174	if (*Pattern == '-') {\r
175	if (strstr (Line, Pattern + 1)) {\r
176	return 0;\r
177	} else {\r
178	continue;\r
179	}\r
180	}\r
181	//\r
182	// For other patterns\r
183	// get its section at first if it has\r
184	//\r
185	if (*Pattern == '[') {\r
186	if (strncmp (Section, Pattern, strlen (Section))) {\r
187	//\r
188	// This pattern can't be appied for current section\r
189	//\r
190	continue;\r
191	}\r
192	//\r
193	// Strip the section field\r
194	//\r
195	while (*Pattern != ']') {\r
196	if (!(*Pattern++)) {\r
197	return -1;\r
198	}\r
199	}\r
200	\r
201	Pattern++;\r
202	}\r
203	//\r
204	// Apply patterns\r
205	//\r
206	Pattern1 = strstr (Pattern, "==");\r
207	Pattern2 = strstr (Pattern, "+=");\r
208	if (Pattern1) {\r
209	//\r
210	// For pattern '=='\r
211	// replace the field value with a new string\r
212	//\r
213	if (!strncmp (Line, Pattern, Pattern1 - Pattern)) {\r
214	Pattern1 += 2;\r
215	Ptr = strstr (Line, "=");\r
216	if (!Ptr) {\r
217	return -1;\r
218	}\r
219	\r
220	while (*(++Ptr) == ' ')\r
221	;\r
222	*Ptr = 0;\r
223	strcat (Line, Pattern1);\r
224	strcat (Line, "\r\n");\r
225	}\r
226	} else if (Pattern2) {\r
227	//\r
228	// For pattern '+='\r
229	// append a string at end of the original string\r
230	//\r
231	if (!strncmp (Line, Pattern, Pattern2 - Pattern)) {\r
232	Pattern2 += 2;\r
233	Ptr = Line;\r
234	while (Ptr != 0x0D && Ptr != 0x0A) {\r
235	Ptr++;\r
236	}\r
237	\r
238	*Ptr = 0;\r
239	strcat (Line, Pattern2);\r
240	strcat (Line, "\r\n");\r
241	}\r
242	}\r
243	}\r
244	\r
245	return 0;\r
246	}\r
247	\r
248	void\r
249	Usage (\r
250	void\r
251	)\r
252	/*++\r
253	\r
254	Routine Description:\r
255	\r
256	GC_TODO: Add function description\r
257	\r
258	Arguments:\r
259	\r
260	None\r
261	\r
262	Returns:\r
263	\r
264	GC_TODO: add return values\r
265	\r
266	--*/\r
267	{\r
3e99020d LG	268	int Index;\r
	269	const char *Str[] = {\r
	270	UTILITY_NAME" "UTILITY_VERSION" - Intel Modify INF File Utility",\r
	271	" Copyright (C), 1999 - 2008 Intel Corporation",\r
	272	\r
	273	#if ( defined(UTILITY_BUILD) && defined(UTILITY_VENDOR) )\r
	274	" Built from "UTILITY_BUILD", project of "UTILITY_VENDOR,\r
	275	#endif\r
	276	"",\r
	277	"Usage:",\r
	278	" "UTILITY_NAME" SOURCE DEST [PATTERN]",\r
	279	NULL\r
	280	};\r
	281	for (Index = 0; Str[Index] != NULL; Index++) {\r
	282	fprintf (stdout, "%s\n", Str[Index]);\r
	283	} \r
3eb9473e	284	}\r
	285	\r
	286	int\r
	287	main (\r
	288	int argc,\r
	289	char*argv[]\r
	290	)\r
	291	/*++\r
	292	\r
	293	Routine Description:\r
	294	\r
	295	GC_TODO: Add function description\r
	296	\r
	297	Arguments:\r
	298	\r
	299	argc - GC_TODO: add argument description\r
	300	] - GC_TODO: add argument description\r
	301	\r
	302	Returns:\r
	303	\r
	304	GC_TODO: add return values\r
	305	\r
	306	--*/\r
	307	{\r
	308	char LineBuffer[256];\r
	309	FILE *fpin;\r
	310	FILE *fpout;\r
	311	\r
	312	if (argc < 3) {\r
	313	Usage ();\r
	314	return -1;\r
	315	}\r
	316	\r
	317	fpin = fopen (argv[1], "rb");\r
	318	if (!fpin) {\r
	319	printf ("Can't open input file!\r\n");\r
	320	return -1;\r
	321	}\r
	322	\r
	323	fpout = fopen (argv[2], "wb");\r
	324	if (!fpout) {\r
	325	fclose (fpin);\r
	326	printf ("Can't create output file!\r\n");\r
	327	return -1;\r
	328	}\r
	329	\r
	330	while (ReadLine (LineBuffer, fpin)) {\r
	331	ApplyPattern (LineBuffer, argv, argc);\r
	332	WriteLine (LineBuffer, fpout);\r
	333	}\r
	334	\r
	335	fclose (fpin);\r
	336	fclose (fpout);\r
	337	\r
	338	return 0;\r
	339	}\r