diff options
author | Hao Wu <hao.a.wu@intel.com> | 2016-10-08 15:28:21 +0800 |
---|---|---|
committer | Hao Wu <hao.a.wu@intel.com> | 2016-11-08 16:38:15 +0800 |
commit | 5baa399e50854317c8788bc1cbae09c1af3b34c6 (patch) | |
tree | e4a86d7f7a08a7b6403d2acff47c6fc16c5edae8 /BaseTools/Source | |
parent | 2a9afe80c3390d2f3540fbb461bb398e49159838 (diff) | |
download | edk2-platforms-5baa399e50854317c8788bc1cbae09c1af3b34c6.tar.xz |
BaseTools/GenVtf: Provide string width in '%s' specifier in format string
String width is not specified for '%s' specifier in the format string for
scanf functions.
This commit now specifies the string length for '%s' in format strings
according to the size of receiving buffers.
Cc: Liming Gao <liming.gao@intel.com>
Cc: Yonghong Zhu <yonghong.zhu@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Hao Wu <hao.a.wu@intel.com>
Reviewed-by: Liming Gao <liming.gao@intel.com>
Diffstat (limited to 'BaseTools/Source')
-rw-r--r-- | BaseTools/Source/C/GenVtf/GenVtf.c | 82 |
1 files changed, 80 insertions, 2 deletions
diff --git a/BaseTools/Source/C/GenVtf/GenVtf.c b/BaseTools/Source/C/GenVtf/GenVtf.c index c37122c853..acc142a6d1 100644 --- a/BaseTools/Source/C/GenVtf/GenVtf.c +++ b/BaseTools/Source/C/GenVtf/GenVtf.c @@ -1045,6 +1045,7 @@ Arguments: Returns:
EFI_INVALID_PARAMETER - The parameter is invalid
+ EFI_OUT_OF_RESOURCES - Resource can not be allocated
EFI_SUCCESS - The function completed successfully
--*/
@@ -1062,6 +1063,8 @@ Returns: CHAR8 Buff4[10];
CHAR8 Buff5[10];
CHAR8 Token[50];
+ CHAR8 *FormatString;
+ INTN FormatLength;
Fp = fopen (LongFilePath (VtfInfo->CompSymName), "rb");
@@ -1070,10 +1073,47 @@ Returns: return EFI_INVALID_PARAMETER;
}
+ //
+ // Generate the format string for fscanf
+ //
+ FormatLength = snprintf (
+ NULL,
+ 0,
+ "%%%us %%%us %%%us %%%us %%%us %%%us %%%us",
+ (unsigned) sizeof (Buff1) - 1,
+ (unsigned) sizeof (Buff2) - 1,
+ (unsigned) sizeof (OffsetStr) - 1,
+ (unsigned) sizeof (Buff3) - 1,
+ (unsigned) sizeof (Buff4) - 1,
+ (unsigned) sizeof (Buff5) - 1,
+ (unsigned) sizeof (Token) - 1
+ ) + 1;
+
+ FormatString = (CHAR8 *) malloc (FormatLength);
+ if (FormatString == NULL) {
+ fclose (Fp);
+
+ Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");
+ return EFI_OUT_OF_RESOURCES;
+ }
+
+ snprintf (
+ FormatString,
+ FormatLength,
+ "%%%us %%%us %%%us %%%us %%%us %%%us %%%us",
+ (unsigned) sizeof (Buff1) - 1,
+ (unsigned) sizeof (Buff2) - 1,
+ (unsigned) sizeof (OffsetStr) - 1,
+ (unsigned) sizeof (Buff3) - 1,
+ (unsigned) sizeof (Buff4) - 1,
+ (unsigned) sizeof (Buff5) - 1,
+ (unsigned) sizeof (Token) - 1
+ );
+
while (fgets (Buff, sizeof (Buff), Fp) != NULL) {
fscanf (
Fp,
- "%s %s %s %s %s %s %s",
+ FormatString,
Buff1,
Buff2,
OffsetStr,
@@ -1096,6 +1136,10 @@ Returns: memcpy ((VOID *) RelativeAddress, (VOID *) CompStartAddress, sizeof (UINT64));
+ if (FormatString != NULL) {
+ free (FormatString);
+ }
+
if (Fp != NULL) {
fclose (Fp);
}
@@ -2198,6 +2242,8 @@ Returns: CHAR8 Section[MAX_LONG_FILE_PATH];
CHAR8 Token[MAX_LONG_FILE_PATH];
CHAR8 BaseToken[MAX_LONG_FILE_PATH];
+ CHAR8 *FormatString;
+ INTN FormatLength;
UINT64 TokenAddress;
long StartLocation;
@@ -2276,6 +2322,37 @@ Returns: }
//
+ // Generate the format string for fscanf
+ //
+ FormatLength = snprintf (
+ NULL,
+ 0,
+ "%%%us | %%%us | %%%us | %%%us\n",
+ (unsigned) sizeof (Type) - 1,
+ (unsigned) sizeof (Address) - 1,
+ (unsigned) sizeof (Section) - 1,
+ (unsigned) sizeof (Token) - 1
+ ) + 1;
+
+ FormatString = (CHAR8 *) malloc (FormatLength);
+ if (FormatString == NULL) {
+ fclose (SourceFile);
+ fclose (DestFile);
+ Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");
+ return EFI_ABORTED;
+ }
+
+ snprintf (
+ FormatString,
+ FormatLength,
+ "%%%us | %%%us | %%%us | %%%us\n",
+ (unsigned) sizeof (Type) - 1,
+ (unsigned) sizeof (Address) - 1,
+ (unsigned) sizeof (Section) - 1,
+ (unsigned) sizeof (Token) - 1
+ );
+
+ //
// Read in the file
//
while (feof (SourceFile) == 0) {
@@ -2283,7 +2360,7 @@ Returns: //
// Read a line
//
- if (fscanf (SourceFile, "%s | %s | %s | %s\n", Type, Address, Section, Token) == 4) {
+ if (fscanf (SourceFile, FormatString, Type, Address, Section, Token) == 4) {
//
// Get the token address
@@ -2306,6 +2383,7 @@ Returns: }
}
+ free (FormatString);
fclose (SourceFile);
fclose (DestFile);
return EFI_SUCCESS;
|