Refine soma code to make code run safely.

[mirror_edk2.git] / MdeModulePkg / Core / Pei / Hob / Hob.c

diff --git a/MdeModulePkg/Core/Pei/Hob/Hob.c b/MdeModulePkg/Core/Pei/Hob/Hob.c
index fa9b150dcc6e7cdea1105d29f0e5c43e2db65f8b..682fa781cdcb17c7ce0420cc322b205736ce9d2c 100644 (file)
--- a/MdeModulePkg/Core/Pei/Hob/Hob.c
+++ b/MdeModulePkg/Core/Pei/Hob/Hob.c
@@ -110,7 +110,7 @@ PeiCreateHob (
   HandOffHob->EfiEndOfHobList = (EFI_PHYSICAL_ADDRESS) (UINTN) HobEnd;\r
  \r
   HobEnd->HobType   = EFI_HOB_TYPE_END_OF_HOB_LIST;\r
-  HobEnd->HobLength = sizeof(EFI_HOB_GENERIC_HEADER);\r
+  HobEnd->HobLength = (UINT16) sizeof (EFI_HOB_GENERIC_HEADER);\r
   HobEnd->Reserved  = 0;\r
   HobEnd++;\r
   HandOffHob->EfiFreeMemoryBottom = (EFI_PHYSICAL_ADDRESS) (UINTN) HobEnd;\r
@@ -142,11 +142,11 @@ PeiCoreBuildHobHandoffInfoTable (
   Hob                      = (VOID *)(UINTN)MemoryBegin;\r
   HobEnd                   = (EFI_HOB_GENERIC_HEADER*) (Hob+1);\r
   Hob->Header.HobType      = EFI_HOB_TYPE_HANDOFF;\r
-  Hob->Header.HobLength    = sizeof(EFI_HOB_HANDOFF_INFO_TABLE);\r
+  Hob->Header.HobLength    = (UINT16) sizeof (EFI_HOB_HANDOFF_INFO_TABLE);\r
   Hob->Header.Reserved     = 0;\r
   \r
   HobEnd->HobType          = EFI_HOB_TYPE_END_OF_HOB_LIST;\r
-  HobEnd->HobLength        = sizeof(EFI_HOB_GENERIC_HEADER);\r
+  HobEnd->HobLength        = (UINT16) sizeof (EFI_HOB_GENERIC_HEADER);\r
   HobEnd->Reserved         = 0;\r
 \r
   Hob->Version             = EFI_HOB_HANDOFF_TABLE_VERSION;\r