BaseTools/PiFirmwareFile: fix undefined behavior in SECTION_SIZE

[mirror_edk2.git] / BaseTools / Source / C / Include / Common / PiFirmwareFile.h

diff --git a/BaseTools/Source/C/Include/Common/PiFirmwareFile.h b/BaseTools/Source/C/Include/Common/PiFirmwareFile.h
index 5bc871df4855a801db9dbd4eb31068f82af972ef..7d8acb669b69a379f06a78b500bf70306575f040 100644 (file)
--- a/BaseTools/Source/C/Include/Common/PiFirmwareFile.h
+++ b/BaseTools/Source/C/Include/Common/PiFirmwareFile.h
@@ -300,8 +300,15 @@ typedef struct {
   CHAR16                      VersionString[1];\r
 } EFI_VERSION_SECTION2;\r
 \r
-#define SECTION_SIZE(SectionHeaderPtr) \\r
-    ((UINT32) (*((UINT32 *) ((EFI_COMMON_SECTION_HEADER *) SectionHeaderPtr)->Size) & 0x00ffffff))\r
+//\r
+// The argument passed as the SectionHeaderPtr parameter to the SECTION_SIZE()\r
+// function-like macro below must not have side effects: SectionHeaderPtr is\r
+// evaluated multiple times.\r
+//\r
+#define SECTION_SIZE(SectionHeaderPtr) ((UINT32) ( \\r
+    (((EFI_COMMON_SECTION_HEADER *) (SectionHeaderPtr))->Size[0]      ) | \\r
+    (((EFI_COMMON_SECTION_HEADER *) (SectionHeaderPtr))->Size[1] <<  8) | \\r
+    (((EFI_COMMON_SECTION_HEADER *) (SectionHeaderPtr))->Size[2] << 16)))\r
 \r
 #pragma pack()\r
 \r