git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@2196 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			380 lines
		
	
	
		
			7.1 KiB
		
	
	
	
		
			C
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			380 lines
		
	
	
		
			7.1 KiB
		
	
	
	
		
			C
		
	
	
		
			Executable File
		
	
	
	
	
| /*++
 | ||
| 
 | ||
| Copyright (c)  1999-2006 Intel Corporation. All rights reserved
 | ||
| This program and the accompanying materials are licensed and made available 
 | ||
| under the terms and conditions of the BSD License which accompanies this 
 | ||
| distribution.  The full text of the license may be found at
 | ||
| http://opensource.org/licenses/bsd-license.php
 | ||
| 
 | ||
| THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
 | ||
| WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 | ||
| 
 | ||
| 
 | ||
| Module Name:
 | ||
| 
 | ||
|   ModifyInf.c
 | ||
| 
 | ||
| Abstract:
 | ||
| 
 | ||
|   It is a simple tool to modify some fields in a FV inf file 
 | ||
|   and output a new FV inf file.  
 | ||
| 
 | ||
| --*/
 | ||
| 
 | ||
| #include "stdio.h"
 | ||
| #include "string.h"
 | ||
| 
 | ||
| #define UTILITY_NAME "ModifyInf"
 | ||
| #define UTILITY_MAJOR_VERSION 1
 | ||
| #define UTILITY_MINOR_VERSION 1
 | ||
| 
 | ||
| //
 | ||
| // Read a line into buffer including '\r\n'
 | ||
| //
 | ||
| int
 | ||
| ReadLine (
 | ||
|   char *LineBuffer,
 | ||
|   FILE *fp
 | ||
|   )
 | ||
| /*++
 | ||
| 
 | ||
| Routine Description:
 | ||
| 
 | ||
|   GC_TODO: Add function description
 | ||
| 
 | ||
| Arguments:
 | ||
| 
 | ||
|   LineBuffer  - GC_TODO: add argument description
 | ||
|   fp          - GC_TODO: add argument description
 | ||
| 
 | ||
| Returns:
 | ||
| 
 | ||
|   GC_TODO: add return values
 | ||
| 
 | ||
| --*/
 | ||
| {
 | ||
|   int   CharC;
 | ||
|   char  *Line;
 | ||
| 
 | ||
|   Line = LineBuffer;
 | ||
| 
 | ||
|   while ((CharC = fgetc (fp)) != EOF) {
 | ||
|     *Line++ = (char) CharC;
 | ||
|     if (CharC == 0x0a) {
 | ||
|       break;
 | ||
|     }
 | ||
|   }
 | ||
| 
 | ||
|   *Line = 0;
 | ||
| 
 | ||
|   if (CharC == EOF) {
 | ||
|     return 0;
 | ||
|   } else {
 | ||
|     return 1;
 | ||
|   }
 | ||
| 
 | ||
| }
 | ||
| //
 | ||
| // Write a line into output file
 | ||
| //
 | ||
| int
 | ||
| WriteLine (
 | ||
|   char *Line,
 | ||
|   FILE *fp
 | ||
|   )
 | ||
| /*++
 | ||
| 
 | ||
| Routine Description:
 | ||
| 
 | ||
|   GC_TODO: Add function description
 | ||
| 
 | ||
| Arguments:
 | ||
| 
 | ||
|   Line  - GC_TODO: add argument description
 | ||
|   fp    - GC_TODO: add argument description
 | ||
| 
 | ||
| Returns:
 | ||
| 
 | ||
|   GC_TODO: add return values
 | ||
| 
 | ||
| --*/
 | ||
| {
 | ||
|   fwrite (Line, strlen (Line), 1, fp);
 | ||
|   return 0;
 | ||
| }
 | ||
| //
 | ||
| // Apply patterns to a line
 | ||
| // Currently there are 2 patterns to support
 | ||
| // '==' replace a field value with a new value
 | ||
| // '+=' append a string at the end of original line
 | ||
| // '-'  prevent the line from applying any patterns
 | ||
| //      it has the highest priority
 | ||
| //
 | ||
| int
 | ||
| ApplyPattern (
 | ||
|   char *Line,
 | ||
|   char *argv[],
 | ||
|   int  argc
 | ||
|   )
 | ||
| /*++
 | ||
| 
 | ||
| Routine Description:
 | ||
| 
 | ||
|   GC_TODO: Add function description
 | ||
| 
 | ||
| Arguments:
 | ||
| 
 | ||
|   Line  - GC_TODO: add argument description
 | ||
|   ]     - GC_TODO: add argument description
 | ||
|   argc  - GC_TODO: add argument description
 | ||
| 
 | ||
| Returns:
 | ||
| 
 | ||
|   GC_TODO: add return values
 | ||
| 
 | ||
| --*/
 | ||
| {
 | ||
|   static char Section[256];
 | ||
|   char        PatternBuffer[256];
 | ||
|   char        *Pattern;
 | ||
|   char        *Pattern1;
 | ||
|   char        *Pattern2;
 | ||
|   int         PatternNum;
 | ||
|   char        *Ptr;
 | ||
| 
 | ||
|   Pattern     = PatternBuffer;
 | ||
| 
 | ||
|   PatternNum  = argc;
 | ||
| 
 | ||
|   //
 | ||
|   // For section field
 | ||
|   // record current scope section into static buffer
 | ||
|   //
 | ||
|   Ptr = Line;
 | ||
|   if (*Ptr == '[') {
 | ||
|     while (*Ptr != ']') {
 | ||
|       if (!(*Ptr++)) {
 | ||
|         return -1;
 | ||
|       }
 | ||
|     }
 | ||
| 
 | ||
|     strcpy (Section, Line);
 | ||
|     Section[Ptr - Line + 1] = 0;
 | ||
|   }
 | ||
|   //
 | ||
|   // Apply each pattern on the line
 | ||
|   //
 | ||
|   while (PatternNum-- > 3) {
 | ||
| 
 | ||
|     strcpy (Pattern, argv[PatternNum]);
 | ||
| 
 | ||
|     //
 | ||
|     // For pattern '-'
 | ||
|     // keep it unmodified by other patterns
 | ||
|     //
 | ||
|     if (*Pattern == '-') {
 | ||
|       if (strstr (Line, Pattern + 1)) {
 | ||
|         return 0;
 | ||
|       } else {
 | ||
|         continue;
 | ||
|       }
 | ||
|     }
 | ||
|     //
 | ||
|     // For other patterns
 | ||
|     // get its section at first if it has
 | ||
|     //
 | ||
|     if (*Pattern == '[') {
 | ||
|       if (strncmp (Section, Pattern, strlen (Section))) {
 | ||
|         //
 | ||
|         // This pattern can't be appied for current section
 | ||
|         //
 | ||
|         continue;
 | ||
|       }
 | ||
|       //
 | ||
|       // Strip the section field
 | ||
|       //
 | ||
|       while (*Pattern != ']') {
 | ||
|         if (!(*Pattern++)) {
 | ||
|           return -1;
 | ||
|         }
 | ||
|       }
 | ||
| 
 | ||
|       Pattern++;
 | ||
|     }
 | ||
|     //
 | ||
|     // Apply patterns
 | ||
|     //
 | ||
|     Pattern1  = strstr (Pattern, "==");
 | ||
|     Pattern2  = strstr (Pattern, "+=");
 | ||
|     if (Pattern1) {
 | ||
|       //
 | ||
|       // For pattern '=='
 | ||
|       // replace the field value with a new string
 | ||
|       //
 | ||
|       if (!strncmp (Line, Pattern, Pattern1 - Pattern)) {
 | ||
|         Pattern1 += 2;
 | ||
|         Ptr = strstr (Line, "=");
 | ||
|         if (!Ptr) {
 | ||
|           return -1;
 | ||
|         }
 | ||
| 
 | ||
|         while (*(++Ptr) == ' ')
 | ||
|           ;
 | ||
|         *Ptr = 0;
 | ||
|         strcat (Line, Pattern1);
 | ||
|         strcat (Line, "\r\n");
 | ||
|       }
 | ||
|     } else if (Pattern2) {
 | ||
|       //
 | ||
|       // For pattern '+='
 | ||
|       // append a string at end of the original string
 | ||
|       //
 | ||
|       if (!strncmp (Line, Pattern, Pattern2 - Pattern)) {
 | ||
|         Pattern2 += 2;
 | ||
|         Ptr = Line;
 | ||
|         while (*Ptr != 0x0D && *Ptr != 0x0A) {
 | ||
|           Ptr++;
 | ||
|         }
 | ||
| 
 | ||
|         *Ptr = 0;
 | ||
|         strcat (Line, Pattern2);
 | ||
|         strcat (Line, "\r\n");
 | ||
|       }
 | ||
|     }
 | ||
|   }
 | ||
| 
 | ||
|   return 0;
 | ||
| }
 | ||
| 
 | ||
| void 
 | ||
| Version(
 | ||
|   void
 | ||
|   )
 | ||
| /*++
 | ||
| 
 | ||
| Routine Description:
 | ||
| 
 | ||
|   Print out version information for Strip.
 | ||
| 
 | ||
| Arguments:
 | ||
| 
 | ||
|   None
 | ||
|   
 | ||
| Returns:
 | ||
| 
 | ||
|   None
 | ||
|   
 | ||
| --*/ 
 | ||
| {
 | ||
|   printf ("%s v%d.%d -EDK Modify fields in FV inf files.\n", UTILITY_NAME, UTILITY_MAJOR_VERSION, UTILITY_MINOR_VERSION);
 | ||
|   printf ("Copyright (c) 2005-2006 Intel Corporation. All rights reserved.\n");
 | ||
| }
 | ||
| 
 | ||
| void 
 | ||
| Usage(
 | ||
|   void
 | ||
|   )
 | ||
| /*++
 | ||
| 
 | ||
| Routine Description:
 | ||
| 
 | ||
|   Print out usage information for Strip.
 | ||
| 
 | ||
| Arguments:
 | ||
| 
 | ||
|   None
 | ||
|   
 | ||
| Returns:
 | ||
| 
 | ||
|   None
 | ||
|   
 | ||
| --*/ 
 | ||
| {
 | ||
|   Version();
 | ||
|   printf ("\nUsage: %s InputFile OutputFile Pattern_String [Pattern_String <20><>]\n\
 | ||
|    Where: \n\
 | ||
|      Pattern_String is of the format (note that the section name must be \n\
 | ||
|      enclosed within square brackets):\n\
 | ||
|          [section]FieldKey<op>Value [(FieldKey<op>Value) <20><>] \n\
 | ||
|      The operator, <op>, must be one of the following: \n\
 | ||
|          '==' replace a field value with a new value \n\
 | ||
|          '+=' append a string at the end of original line \n\
 | ||
|          '-'  prevent the line from applying any patterns \n\
 | ||
|      Example: \n\
 | ||
|          ModifyInf BuildRootFvFvMain.inf BuildRootFvFvMainEXP.inf \\ \n\
 | ||
|                [files]EFI_FILE_NAME+=.Org EFI_NUM_BLOCKS==0x20 \\ \n\
 | ||
|                [options]EFI_FILENAME==FcMainCompact.fv -DpsdSignature.dxe \n", UTILITY_NAME);
 | ||
| }
 | ||
| 
 | ||
| 
 | ||
| int
 | ||
| main (
 | ||
|   int argc,
 | ||
|   char*argv[]
 | ||
|   )
 | ||
| /*++
 | ||
| 
 | ||
| Routine Description:
 | ||
| 
 | ||
|   GC_TODO: Add function description
 | ||
| 
 | ||
| Arguments:
 | ||
| 
 | ||
|   argc  - GC_TODO: add argument description
 | ||
|   ]     - GC_TODO: add argument description
 | ||
| 
 | ||
| Returns:
 | ||
| 
 | ||
|   GC_TODO: add return values
 | ||
| 
 | ||
| --*/
 | ||
| {
 | ||
|   char  LineBuffer[256];
 | ||
|   FILE  *fpin;
 | ||
|   FILE  *fpout;
 | ||
| 
 | ||
|   if (argc == 1) {
 | ||
|     Usage();
 | ||
|     return -1;
 | ||
|   }
 | ||
|   
 | ||
|   if ((strcmp(argv[1], "-h") == 0) || (strcmp(argv[1], "--help") == 0) ||
 | ||
|       (strcmp(argv[1], "-?") == 0) || (strcmp(argv[1], "/?") == 0)) {
 | ||
|     Usage();
 | ||
|     return 0;
 | ||
|   }
 | ||
|   
 | ||
|   if ((strcmp(argv[1], "-V") == 0) || (strcmp(argv[1], "--version") == 0)) {
 | ||
|     Version();
 | ||
|     return 0;
 | ||
|   }
 | ||
|   
 | ||
|   if (argc < 3) {
 | ||
|     Usage();
 | ||
|     return -1;
 | ||
|   }
 | ||
| 
 | ||
|   fpin = fopen (argv[1], "rb");
 | ||
|   if (!fpin) {
 | ||
|     printf ("Can't open input file!\r\n");
 | ||
|     return -1;
 | ||
|   }
 | ||
| 
 | ||
|   fpout = fopen (argv[2], "wb");
 | ||
|   if (!fpout) {
 | ||
|     fclose (fpin);
 | ||
|     printf ("Can't create output file!\r\n");
 | ||
|     return -1;
 | ||
|   }
 | ||
| 
 | ||
|   while (ReadLine (LineBuffer, fpin)) {
 | ||
|     ApplyPattern (LineBuffer, argv, argc);
 | ||
|     WriteLine (LineBuffer, fpout);
 | ||
|   }
 | ||
| 
 | ||
|   fclose (fpin);
 | ||
|   fclose (fpout);
 | ||
| 
 | ||
|   return 0;
 | ||
| }
 |