BaseTools: Add DevicePath support for PCD values

[mirror_edk2.git] / BaseTools / Source / C / GenFv / GenFvInternalLib.c

diff --git a/BaseTools/Source/C/GenFv/GenFvInternalLib.c b/BaseTools/Source/C/GenFv/GenFvInternalLib.c
index 01c862e89a393d74b8e004b4cce661fe22f2b9fa..fc1a7602ab996f9b6fc64a1928998b798999ef30 100644 (file)
--- a/BaseTools/Source/C/GenFv/GenFvInternalLib.c
+++ b/BaseTools/Source/C/GenFv/GenFvInternalLib.c
@@ -44,6 +44,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 \r
 BOOLEAN mArm = FALSE;\r
 STATIC UINT32   MaxFfsAlignment = 0;\r
+BOOLEAN VtfFileFlag = FALSE;\r
 \r
 EFI_GUID  mEfiFirmwareVolumeTopFileGuid       = EFI_FFS_VOLUME_TOP_FILE_GUID;\r
 EFI_GUID  mFileGuidArray [MAX_NUMBER_OF_FILES_IN_FV];\r
@@ -600,7 +601,9 @@ Returns:
   UINTN               PadFileSize;\r
   UINT32              NextFfsHeaderSize;\r
   UINT32              CurFfsHeaderSize;\r
+  UINT32              Index;\r
 \r
+  Index = 0;\r
   CurFfsHeaderSize = sizeof (EFI_FFS_FILE_HEADER);\r
   //\r
   // Verify input parameters.\r
@@ -705,6 +708,19 @@ Returns:
     //\r
     // Copy Fv Extension Header and Set Fv Extension header offset\r
     //\r
+    if (ExtHeader->ExtHeaderSize > sizeof (EFI_FIRMWARE_VOLUME_EXT_HEADER)) {\r
+      for (Index = sizeof (EFI_FIRMWARE_VOLUME_EXT_HEADER); Index < ExtHeader->ExtHeaderSize;) {\r
+        if (((EFI_FIRMWARE_VOLUME_EXT_ENTRY *)((UINT8 *)ExtHeader + Index))-> ExtEntryType == EFI_FV_EXT_TYPE_USED_SIZE_TYPE) {\r
+          if (VtfFileFlag) {\r
+            ((EFI_FIRMWARE_VOLUME_EXT_ENTRY_USED_SIZE_TYPE *)((UINT8 *)ExtHeader + Index))->UsedSize = mFvTotalSize;\r
+          } else {\r
+            ((EFI_FIRMWARE_VOLUME_EXT_ENTRY_USED_SIZE_TYPE *)((UINT8 *)ExtHeader + Index))->UsedSize = mFvTakenSize;\r
+          }\r
+          break;\r
+        }\r
+        Index += ((EFI_FIRMWARE_VOLUME_EXT_ENTRY *)((UINT8 *)ExtHeader + Index))-> ExtEntrySize;\r
+      }\r
+    }\r
     memcpy ((UINT8 *)PadFile + CurFfsHeaderSize, ExtHeader, ExtHeader->ExtHeaderSize);\r
     ((EFI_FIRMWARE_VOLUME_HEADER *) FvImage->FileImage)->ExtHeaderOffset = (UINT16) ((UINTN) ((UINT8 *)PadFile + CurFfsHeaderSize) - (UINTN) FvImage->FileImage);\r
          //\r
@@ -808,7 +824,11 @@ Returns:
   //\r
   // Construct Map file Name \r
   //\r
-  strcpy (PeMapFileName, FileName);\r
+  if (strlen (FileName) >= MAX_LONG_FILE_PATH) {\r
+    return EFI_ABORTED;\r
+  }\r
+  strncpy (PeMapFileName, FileName, MAX_LONG_FILE_PATH - 1);\r
+  PeMapFileName[MAX_LONG_FILE_PATH - 1] = 0;\r
   \r
   //\r
   // Change '\\' to '/', unified path format.\r
@@ -845,7 +865,11 @@ Returns:
     Cptr --;\r
   }\r
        *Cptr2 = '\0';\r
-       strcpy (KeyWord, Cptr + 1);\r
+  if (strlen (Cptr + 1) >= MAX_LINE_LEN) {\r
+    return EFI_ABORTED;\r
+  }\r
+  strncpy (KeyWord, Cptr + 1, MAX_LINE_LEN - 1);\r
+  KeyWord[MAX_LINE_LEN - 1] = 0;\r
        *Cptr2 = '.';\r
 \r
   //\r
@@ -3059,12 +3083,10 @@ Returns:
   UINT32              FfsAlignment;\r
   UINT32              FfsHeaderSize;\r
   EFI_FFS_FILE_HEADER FfsHeader;\r
-  BOOLEAN             VtfFileFlag;\r
   UINTN               VtfFileSize;\r
   \r
   FvExtendHeaderSize = 0;\r
   VtfFileSize = 0;\r
-  VtfFileFlag = FALSE;\r
   fpin  = NULL;\r
   Index = 0;\r
 \r
@@ -3520,7 +3542,12 @@ Returns:
           //\r
           // Construct the original efi file Name \r
           //\r
-          strcpy (PeFileName, FileName);\r
+          if (strlen (FileName) >= MAX_LONG_FILE_PATH) {\r
+            Error (NULL, 0, 2000, "Invalid", "The file name %s is too long.", FileName);\r
+            return EFI_ABORTED;\r
+          }\r
+          strncpy (PeFileName, FileName, MAX_LONG_FILE_PATH - 1);\r
+          PeFileName[MAX_LONG_FILE_PATH - 1] = 0;\r
           Cptr = PeFileName + strlen (PeFileName);\r
           while (*Cptr != '.') {\r
             Cptr --;\r
@@ -3775,7 +3802,12 @@ Returns:
       //\r
       // Construct the original efi file name \r
       //\r
-      strcpy (PeFileName, FileName);\r
+      if (strlen (FileName) >= MAX_LONG_FILE_PATH) {\r
+        Error (NULL, 0, 2000, "Invalid", "The file name %s is too long.", FileName);\r
+        return EFI_ABORTED;\r
+      }\r
+      strncpy (PeFileName, FileName, MAX_LONG_FILE_PATH - 1);\r
+      PeFileName[MAX_LONG_FILE_PATH - 1] = 0;\r
       Cptr = PeFileName + strlen (PeFileName);\r
       while (*Cptr != '.') {\r
         Cptr --;\r