DestPtr may point to the lower aligned address in the free memory region (FreeMemBase...

Signed-off-by: Star Zeng <star.zeng@intel.com>
Reviewed-by: Liming Gao <liming.gao@intel.com>
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@13303 6f19259b-4bc3-4df7-8a09-765794883524

diff --git a/MdeModulePkg/Universal/CapsulePei/Common/CapsuleCoalesce.c b/MdeModulePkg/Universal/CapsulePei/Common/CapsuleCoalesce.c
index 042811167579e10f72562273ea0986b8310533e9..a95b4722c542733d1bceedb9e178f24c703ca75f 100644 (file)
--- a/MdeModulePkg/Universal/CapsulePei/Common/CapsuleCoalesce.c
+++ b/MdeModulePkg/Universal/CapsulePei/Common/CapsuleCoalesce.c
@@ -945,7 +945,7 @@ CapsuleDataCoalesce (
   DestPtr         = FreeMemBase + FreeMemSize - CapsuleSize;\r
   DestPtr         = (UINT8 *) ((UINTN) DestPtr &~ (UINTN) (sizeof (UINTN) - 1));\r
   FreeMemBase     = (UINT8 *) BlockList + DescriptorsSize;\r
-  FreeMemSize     = FreeMemSize - DescriptorsSize - CapsuleSize;\r
+  FreeMemSize     = (UINTN) DestPtr - (UINTN) FreeMemBase;\r
   NewCapsuleBase  = (VOID *) DestPtr;\r
 \r
   //\r