From f942f107607e18944bd44f1d223670d190156f2b Mon Sep 17 00:00:00 2001
From: qhuang8 <qhuang8@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Thu, 8 Nov 2007 03:52:35 +0000
Subject: [PATCH] Merge EDK899:  fixed bug to support
 EFI_HOB_TYPE_MEMORY_ALLOCATION to allocate resource in PEI

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@4289 6f19259b-4bc3-4df7-8a09-765794883524
---
 MdeModulePkg/Core/Dxe/Gcd/gcd.c | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/MdeModulePkg/Core/Dxe/Gcd/gcd.c b/MdeModulePkg/Core/Dxe/Gcd/gcd.c
index accc329a4a..1d023548d3 100644
--- a/MdeModulePkg/Core/Dxe/Gcd/gcd.c
+++ b/MdeModulePkg/Core/Dxe/Gcd/gcd.c
@@ -2415,18 +2415,18 @@ Returns:
     if (GET_HOB_TYPE (Hob) == EFI_HOB_TYPE_MEMORY_ALLOCATION) {
       MemoryHob = Hob.MemoryAllocation;
       BaseAddress = MemoryHob->AllocDescriptor.MemoryBaseAddress;
-      Status = CoreAllocateMemorySpace (
-                 EfiGcdAllocateAddress,
-                 EfiGcdMemoryTypeSystemMemory, 
-                 0,
-                 MemoryHob->AllocDescriptor.MemoryLength,
-                 &BaseAddress,
-                 gDxeCoreImageHandle,
-                 NULL
-                 );
+      Status = CoreGetMemorySpaceDescriptor  (BaseAddress, &Descriptor);
       if (!EFI_ERROR (Status)) {
-        Status = CoreGetMemorySpaceDescriptor (MemoryHob->AllocDescriptor.MemoryBaseAddress, &Descriptor);
-        if (!EFI_ERROR (Status)) {
+        Status = CoreAllocateMemorySpace (
+                   EfiGcdAllocateAddress,
+                   Descriptor.GcdMemoryType, 
+                   0,
+                   MemoryHob->AllocDescriptor.MemoryLength,
+                   &BaseAddress,
+                   gDxeCoreImageHandle,
+                   NULL
+                   );
+        if (!EFI_ERROR (Status) && Descriptor.GcdMemoryType == EfiGcdMemoryTypeSystemMemory) {
           CoreAddMemoryDescriptor (
             MemoryHob->AllocDescriptor.MemoryType,
             MemoryHob->AllocDescriptor.MemoryBaseAddress,