]> git.proxmox.com Git - mirror_edk2.git/blobdiff - BaseTools/Source/C/Common/FirmwareVolumeBuffer.c
Sync BaseTool trunk (version r2670) into EDKII BaseTools.
[mirror_edk2.git] / BaseTools / Source / C / Common / FirmwareVolumeBuffer.c
index 92060e2ca1008510fd64bc5b17ade5eac9282505..fa6ca6bd6d192581bbcd603ec5e5f02272994ddf 100644 (file)
@@ -1,6 +1,6 @@
 /** @file\r
 \r
-Copyright (c) 1999 - 2008, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 1999 - 2014, 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
@@ -32,6 +32,37 @@ Abstract:
         ) \\r
     )\r
 \r
+STATIC\r
+UINT32\r
+FvBufGetSecHdrLen(\r
+   IN EFI_COMMON_SECTION_HEADER *SectionHeader\r
+   )\r
+{\r
+  if (SectionHeader == NULL) {\r
+    return 0;\r
+  }\r
+  if (FvBufExpand3ByteSize(SectionHeader->Size) == 0xffffff) {\r
+    return sizeof(EFI_COMMON_SECTION_HEADER2);\r
+  }\r
+  return sizeof(EFI_COMMON_SECTION_HEADER);\r
+}\r
+\r
+STATIC\r
+UINT32\r
+FvBufGetSecFileLen (\r
+  IN EFI_COMMON_SECTION_HEADER *SectionHeader\r
+  )\r
+{\r
+  UINT32 Length;\r
+  if (SectionHeader == NULL) {\r
+    return 0;\r
+  }\r
+  Length = FvBufExpand3ByteSize(SectionHeader->Size);\r
+  if (Length == 0xffffff) {\r
+    Length = ((EFI_COMMON_SECTION_HEADER2 *)SectionHeader)->ExtendedSize;\r
+  }\r
+  return Length;\r
+}\r
 \r
 //\r
 // Local prototypes\r
@@ -92,7 +123,7 @@ Returns:
     return Status;\r
   }\r
 \r
-  FileToRmLength = FvBufExpand3ByteSize (FileToRm->Size);\r
+  FileToRmLength = FvBufGetFfsFileSize (FileToRm);\r
 \r
   CommonLibBinderSetMem (\r
     FileToRm,\r
@@ -218,7 +249,7 @@ Returns:
   EFI_FFS_FILE_STATE StateBackup;\r
   UINT32 FileSize;\r
 \r
-  FileSize = FvBufExpand3ByteSize (File->Size);\r
+  FileSize = FvBufGetFfsFileSize (File);\r
 \r
   //\r
   // Fill in checksums and state, they must be 0 for checksumming.\r
@@ -231,13 +262,13 @@ Returns:
   File->IntegrityCheck.Checksum.Header =\r
     FvBufCalculateChecksum8 (\r
       (UINT8 *) File,\r
-      sizeof (EFI_FFS_FILE_HEADER)\r
+      FvBufGetFfsHeaderSize (File)\r
       );\r
 \r
   if (File->Attributes & FFS_ATTRIB_CHECKSUM) {\r
     File->IntegrityCheck.Checksum.File = FvBufCalculateChecksum8 (\r
-                                                (VOID*)(File + 1),\r
-                                                FileSize - sizeof (EFI_FFS_FILE_HEADER)\r
+                                                (VOID*)((UINT8 *)File + FvBufGetFfsHeaderSize (File)),\r
+                                                FileSize - FvBufGetFfsHeaderSize (File)\r
                                                 );\r
   } else {\r
     File->IntegrityCheck.Checksum.File = FFS_FIXED_CHECKSUM;\r
@@ -568,7 +599,7 @@ Returns:
   }\r
 \r
   FvbAttributes = hdr->Attributes;\r
-  newSize = FvBufExpand3ByteSize (((EFI_FFS_FILE_HEADER*)File)->Size);\r
+  newSize = FvBufGetFfsFileSize ((EFI_FFS_FILE_HEADER*)File);\r
 \r
   for(\r
       offset = (UINTN)ALIGN_POINTER (hdr->HeaderLength, 8);\r
@@ -587,7 +618,7 @@ Returns:
       // BUGBUG: Need to make sure that the new file does not already\r
       // exist.\r
 \r
-      fsize = FvBufExpand3ByteSize (fhdr->Size);\r
+      fsize = FvBufGetFfsFileSize (fhdr);\r
       if (fsize == 0 || (offset + fsize > fvSize)) {\r
         return EFI_VOLUME_CORRUPTED;\r
       }\r
@@ -725,7 +756,7 @@ Returns:
   }\r
 \r
   erasedUint8 = (UINT8)((hdr->Attributes & EFI_FVB2_ERASE_POLARITY) ? 0xFF : 0);\r
-  NewFileSize = FvBufExpand3ByteSize (((EFI_FFS_FILE_HEADER*)File)->Size);\r
+  NewFileSize = FvBufGetFfsFileSize ((EFI_FFS_FILE_HEADER*)File);\r
 \r
   if (NewFileSize != (UINTN)ALIGN_POINTER (NewFileSize, 8)) {\r
     return EFI_INVALID_PARAMETER;\r
@@ -739,7 +770,7 @@ Returns:
   LastFileSize = 0;\r
   do {\r
     Status = FvBufFindNextFile (Fv, &Key, (VOID **)&LastFile);\r
-    LastFileSize = FvBufExpand3ByteSize (((EFI_FFS_FILE_HEADER*)File)->Size);\r
+    LastFileSize = FvBufGetFfsFileSize ((EFI_FFS_FILE_HEADER*)File);\r
   } while (!EFI_ERROR (Status));\r
 \r
   //\r
@@ -811,6 +842,64 @@ Returns:
   ((UINT8*)SizeDest)[2] = (UINT8)(Size >> 16);\r
 }\r
 \r
+UINT32\r
+FvBufGetFfsFileSize (\r
+  IN EFI_FFS_FILE_HEADER *Ffs\r
+  )\r
+/*++\r
+\r
+Routine Description:\r
+\r
+  Get the FFS file size.\r
+\r
+Arguments:\r
+\r
+  Ffs - Pointer to FFS header\r
+\r
+Returns:\r
+\r
+  UINT32\r
+\r
+--*/\r
+{\r
+  if (Ffs == NULL) {\r
+    return 0;\r
+  }\r
+  if (Ffs->Attributes & FFS_ATTRIB_LARGE_FILE) {\r
+    return ((EFI_FFS_FILE_HEADER2 *)Ffs)->ExtendedSize;\r
+  }\r
+  return FvBufExpand3ByteSize(Ffs->Size);\r
+}\r
+\r
+UINT32\r
+FvBufGetFfsHeaderSize (\r
+  IN EFI_FFS_FILE_HEADER *Ffs\r
+  )\r
+/*++\r
+\r
+Routine Description:\r
+\r
+  Get the FFS header size.\r
+\r
+Arguments:\r
+\r
+  Ffs - Pointer to FFS header\r
+\r
+Returns:\r
+\r
+  UINT32\r
+\r
+--*/\r
+{\r
+  if (Ffs == NULL) {\r
+    return 0;\r
+  }\r
+  if (Ffs->Attributes & FFS_ATTRIB_LARGE_FILE) {\r
+    return sizeof(EFI_FFS_FILE_HEADER2);\r
+  }\r
+  return sizeof(EFI_FFS_FILE_HEADER);\r
+}\r
+\r
 UINT32\r
 FvBufExpand3ByteSize (\r
   IN VOID* Size\r
@@ -897,7 +986,7 @@ Returns:
     ) {\r
 \r
     fhdr = (EFI_FFS_FILE_HEADER*) ((UINT8*)hdr + *Key);\r
-    fsize = FvBufExpand3ByteSize (fhdr->Size);\r
+    fsize = FvBufGetFfsFileSize (fhdr);\r
 \r
     if (!EFI_TEST_FFS_ATTRIBUTES_BIT(\r
           FvbAttributes,\r
@@ -1089,8 +1178,8 @@ Returns:
     //\r
     // Raw filetypes don't have sections, so we just return the raw data\r
     //\r
-    *RawData = (VOID*)(File + 1);\r
-    *RawDataSize = FvBufExpand3ByteSize (File->Size) - sizeof (*File);\r
+    *RawData = (VOID*)((UINT8 *)File + FvBufGetFfsHeaderSize (File));\r
+    *RawDataSize = FvBufGetFfsFileSize (File) - FvBufGetFfsHeaderSize (File);\r
     return EFI_SUCCESS;\r
   }\r
 \r
@@ -1102,9 +1191,9 @@ Returns:
     return Status;\r
   }\r
 \r
-  *RawData = (VOID*)(Section + 1);\r
+  *RawData = (VOID*)((UINT8 *)Section + FvBufGetSecHdrLen(Section));\r
   *RawDataSize =\r
-    FvBufExpand3ByteSize (Section->Size) - sizeof (*Section);\r
+    FvBufGetSecFileLen (Section) - FvBufGetSecHdrLen(Section);\r
 \r
   return EFI_SUCCESS;\r
 \r
@@ -1144,16 +1233,28 @@ Returns:
   UINT32 NewFileSize;\r
   EFI_RAW_SECTION* NewSection;\r
   UINT32 NewSectionSize;\r
+  UINT32 FfsHdrLen;\r
+  UINT32 SecHdrLen;\r
 \r
   //\r
   // The section size is the DataSize + the size of the section header\r
   //\r
   NewSectionSize = (UINT32)sizeof (EFI_RAW_SECTION) + (UINT32)RawDataSize;\r
+  SecHdrLen = sizeof (EFI_RAW_SECTION);\r
+  if (NewSectionSize >= MAX_SECTION_SIZE) {\r
+    NewSectionSize = (UINT32)sizeof (EFI_RAW_SECTION2) + (UINT32)RawDataSize;\r
+    SecHdrLen = sizeof (EFI_RAW_SECTION2);\r
+  }\r
 \r
   //\r
   // The file size is the size of the file header + the section size\r
   //\r
   NewFileSize = sizeof (EFI_FFS_FILE_HEADER) + NewSectionSize;\r
+  FfsHdrLen = sizeof (EFI_FFS_FILE_HEADER);\r
+  if (NewFileSize >= MAX_FFS_SIZE) {\r
+    NewFileSize = sizeof (EFI_FFS_FILE_HEADER2) + NewSectionSize;\r
+    FfsHdrLen = sizeof (EFI_FFS_FILE_HEADER2);\r
+  }\r
 \r
   //\r
   // Try to allocate a buffer to build the new FFS file in\r
@@ -1167,24 +1268,35 @@ Returns:
   //\r
   // The NewSection follow right after the FFS file header\r
   //\r
-  NewSection = (EFI_RAW_SECTION*)(NewFile + 1);\r
-  FvBufCompact3ByteSize (NewSection->Size, NewSectionSize);\r
+  NewSection = (EFI_RAW_SECTION*)((UINT8*)NewFile + FfsHdrLen);\r
+  if (NewSectionSize >= MAX_SECTION_SIZE) {\r
+    FvBufCompact3ByteSize (NewSection->Size, 0xffffff);\r
+    ((EFI_RAW_SECTION2 *)NewSection)->ExtendedSize = NewSectionSize;\r
+  } else {\r
+    FvBufCompact3ByteSize (NewSection->Size, NewSectionSize);\r
+  }\r
   NewSection->Type = EFI_SECTION_RAW;\r
 \r
   //\r
   // Copy the actual file data into the buffer\r
   //\r
-  CommonLibBinderCopyMem (NewSection + 1, RawData, RawDataSize);\r
+  CommonLibBinderCopyMem ((UINT8 *)NewSection + SecHdrLen, RawData, RawDataSize);\r
 \r
   //\r
   // Initialize the FFS file header\r
   //\r
   CommonLibBinderCopyMem (&NewFile->Name, Filename, sizeof (EFI_GUID));\r
-  FvBufCompact3ByteSize (NewFile->Size, NewFileSize);\r
-  NewFile->Type = EFI_FV_FILETYPE_FREEFORM;\r
   NewFile->Attributes = 0;\r
+  if (NewFileSize >= MAX_FFS_SIZE) {\r
+    FvBufCompact3ByteSize (NewFile->Size, 0x0);\r
+    ((EFI_FFS_FILE_HEADER2 *)NewFile)->ExtendedSize = NewFileSize;\r
+    NewFile->Attributes |= FFS_ATTRIB_LARGE_FILE;\r
+  } else {\r
+    FvBufCompact3ByteSize (NewFile->Size, NewFileSize);\r
+  }\r
+  NewFile->Type = EFI_FV_FILETYPE_FREEFORM;\r
   NewFile->IntegrityCheck.Checksum.Header =\r
-    FvBufCalculateChecksum8 ((UINT8*)NewFile, sizeof (*NewFile));\r
+    FvBufCalculateChecksum8 ((UINT8*)NewFile, FfsHdrLen);\r
   NewFile->IntegrityCheck.Checksum.File = FFS_FIXED_CHECKSUM;\r
   NewFile->State = (UINT8)~( EFI_FILE_HEADER_CONSTRUCTION |\r
                              EFI_FILE_HEADER_VALID |\r
@@ -1239,7 +1351,7 @@ Returns:
   }\r
 \r
   sectionHdr = (EFI_COMMON_SECTION_HEADER*)((UINT8*)SectionsStart + *Key);\r
-  sectionSize = FvBufExpand3ByteSize (sectionHdr->Size);\r
+  sectionSize = FvBufGetSecFileLen (sectionHdr);\r
 \r
   if (sectionSize < sizeof (EFI_COMMON_SECTION_HEADER)) {\r
     return EFI_NOT_FOUND;\r
@@ -1287,10 +1399,10 @@ Returns:
   UINTN                      TotalSectionsSize;\r
   EFI_COMMON_SECTION_HEADER* NextSection;\r
 \r
-  SectionStart = (VOID*)((UINTN)FfsFile + sizeof (EFI_FFS_FILE_HEADER));\r
+  SectionStart = (VOID*)((UINTN)FfsFile + FvBufGetFfsHeaderSize(FfsFile));\r
   TotalSectionsSize =\r
-    FvBufExpand3ByteSize (((EFI_FFS_FILE_HEADER*)FfsFile)->Size) -\r
-    sizeof (EFI_FFS_FILE_HEADER);\r
+    FvBufGetFfsFileSize ((EFI_FFS_FILE_HEADER*)FfsFile) -\r
+    FvBufGetFfsHeaderSize(FfsFile);\r
   Key = 0;\r
   *Count = 0;\r
   while (TRUE) {\r
@@ -1352,10 +1464,10 @@ Returns:
   UINTN                      TotalSectionsSize;\r
   EFI_COMMON_SECTION_HEADER* NextSection;\r
 \r
-  SectionStart = (VOID*)((UINTN)FfsFile + sizeof (EFI_FFS_FILE_HEADER));\r
+  SectionStart = (VOID*)((UINTN)FfsFile + FvBufGetFfsHeaderSize(FfsFile));\r
   TotalSectionsSize =\r
-    FvBufExpand3ByteSize (((EFI_FFS_FILE_HEADER*)FfsFile)->Size) -\r
-    sizeof (EFI_FFS_FILE_HEADER);\r
+    FvBufGetFfsFileSize ((EFI_FFS_FILE_HEADER*)FfsFile) -\r
+    FvBufGetFfsHeaderSize(FfsFile);\r
   Key = 0;\r
   while (TRUE) {\r
     Status = FvBufFindNextSection (\r
@@ -1436,7 +1548,7 @@ Returns:
   EndOfLastFile = (UINT8*)FvHdr + FvHdr->FvLength;\r
   while (!EFI_ERROR (FvBufFindNextFile (Fv, &Key, (VOID **)&FileIt))) {\r
     EndOfLastFile =\r
-      (VOID*)((UINT8*)FileIt + FvBufExpand3ByteSize (FileIt->Size));\r
+      (VOID*)((UINT8*)FileIt + FvBufGetFfsFileSize (FileIt));\r
   }\r
 \r
   //\r