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;
|
||
}
|