MdeModulePkg: Move to new location

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Guo Mang <mang.guo@intel.com>

4 files changed, 0 insertions, 1168 deletions

diff --git a/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/MemoryAllocationLib.c b/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/MemoryAllocationLib.c
deleted file mode 100644
index 5e13a3eda2..0000000000
--- a/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/MemoryAllocationLib.c
+++ /dev/null
@@ -1,929 +0,0 @@
-/** @file

-  Support routines for memory allocation routines based on SMM Core internal functions.

-  

-  The PI System Management Mode Core Interface Specification only allows the use

-  of EfiRuntimeServicesCode and EfiRuntimeServicesData memory types for memory 

-  allocations as the SMRAM space should be reserved after BDS phase.  The functions 

-  in the Memory Allocation Library use EfiBootServicesData as the default memory 

-  allocation type.  For this SMM specific instance of the Memory Allocation Library, 

-  EfiRuntimeServicesData is used as the default memory type for all allocations. 

-  In addition, allocation for the Reserved memory types are not supported and will 

-  always return NULL.

-

-  Copyright (c) 2006 - 2015, Intel Corporation. All rights reserved.<BR>

-  This program and the accompanying materials                          

-  are licensed and made available under the terms and conditions of the BSD License         

-  which accompanies this distribution.  The full text of the license may be found at        

-  http://opensource.org/licenses/bsd-license.php                                            

-

-  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,                     

-  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.             

-

-**/

-

-#include <PiSmm.h>

-

-#include <Protocol/SmmAccess2.h>

-#include <Library/MemoryAllocationLib.h>

-#include <Library/UefiBootServicesTableLib.h>

-#include <Library/BaseMemoryLib.h>

-#include <Library/DebugLib.h>

-#include "PiSmmCoreMemoryAllocationServices.h"

-

-EFI_SMRAM_DESCRIPTOR  *mSmmCoreMemoryAllocLibSmramRanges    = NULL;

-UINTN                 mSmmCoreMemoryAllocLibSmramRangeCount = 0;

-

-/**

-  Check whether the start address of buffer is within any of the SMRAM ranges.

-

-  @param[in]  Buffer   The pointer to the buffer to be checked.

-

-  @retval     TURE     The buffer is in SMRAM ranges.

-  @retval     FALSE    The buffer is out of SMRAM ranges.

-**/

-BOOLEAN

-EFIAPI

-BufferInSmram (

-  IN VOID *Buffer

-  )

-{

-  UINTN  Index;

-

-  for (Index = 0; Index < mSmmCoreMemoryAllocLibSmramRangeCount; Index ++) {

-    if (((EFI_PHYSICAL_ADDRESS) (UINTN) Buffer >= mSmmCoreMemoryAllocLibSmramRanges[Index].CpuStart) && 

-        ((EFI_PHYSICAL_ADDRESS) (UINTN) Buffer < (mSmmCoreMemoryAllocLibSmramRanges[Index].CpuStart + mSmmCoreMemoryAllocLibSmramRanges[Index].PhysicalSize))) {

-      return TRUE;

-    }

-  }

-

-  return FALSE;

-}

-

-/**

-  Allocates one or more 4KB pages of a certain memory type.

-

-  Allocates the number of 4KB pages of a certain memory type and returns a pointer to the allocated

-  buffer.  The buffer returned is aligned on a 4KB boundary.  If Pages is 0, then NULL is returned.

-  If there is not enough memory remaining to satisfy the request, then NULL is returned.

-

-  @param  MemoryType            The type of memory to allocate.

-  @param  Pages                 The number of 4 KB pages to allocate.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-InternalAllocatePages (

-  IN EFI_MEMORY_TYPE  MemoryType,  

-  IN UINTN            Pages

-  )

-{

-  EFI_STATUS            Status;

-  EFI_PHYSICAL_ADDRESS  Memory; 

-

-  if (Pages == 0) {

-    return NULL;

-  }

-

-  Status = SmmAllocatePages (AllocateAnyPages, MemoryType, Pages, &Memory);

-  if (EFI_ERROR (Status)) {

-    return NULL;

-  }

-  return (VOID *) (UINTN) Memory;

-}

-

-/**

-  Allocates one or more 4KB pages of type EfiRuntimeServicesData.

-

-  Allocates the number of 4KB pages of type EfiRuntimeServicesData and returns a pointer to the

-  allocated buffer.  The buffer returned is aligned on a 4KB boundary.  If Pages is 0, then NULL

-  is returned.  If there is not enough memory remaining to satisfy the request, then NULL is

-  returned.

-

-  @param  Pages                 The number of 4 KB pages to allocate.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocatePages (

-  IN UINTN  Pages

-  )

-{

-  return InternalAllocatePages (EfiRuntimeServicesData, Pages);

-}

-

-/**

-  Allocates one or more 4KB pages of type EfiRuntimeServicesData.

-

-  Allocates the number of 4KB pages of type EfiRuntimeServicesData and returns a pointer to the

-  allocated buffer.  The buffer returned is aligned on a 4KB boundary.  If Pages is 0, then NULL

-  is returned.  If there is not enough memory remaining to satisfy the request, then NULL is

-  returned.

-

-  @param  Pages                 The number of 4 KB pages to allocate.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocateRuntimePages (

-  IN UINTN  Pages

-  )

-{

-  return InternalAllocatePages (EfiRuntimeServicesData, Pages);

-}

-

-/**

-  Allocates one or more 4KB pages of type EfiReservedMemoryType.

-

-  Allocates the number of 4KB pages of type EfiReservedMemoryType and returns a pointer to the

-  allocated buffer.  The buffer returned is aligned on a 4KB boundary.  If Pages is 0, then NULL

-  is returned.  If there is not enough memory remaining to satisfy the request, then NULL is

-  returned.

-

-  @param  Pages                 The number of 4 KB pages to allocate.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocateReservedPages (

-  IN UINTN  Pages

-  )

-{

-  return NULL;

-}

-

-/**

-  Frees one or more 4KB pages that were previously allocated with one of the page allocation

-  functions in the Memory Allocation Library.

-

-  Frees the number of 4KB pages specified by Pages from the buffer specified by Buffer.  Buffer

-  must have been allocated on a previous call to the page allocation services of the Memory

-  Allocation Library.  If it is not possible to free allocated pages, then this function will

-  perform no actions.

-  

-  If Buffer was not allocated with a page allocation function in the Memory Allocation Library,

-  then ASSERT().

-  If Pages is zero, then ASSERT().

- 

-  @param  Buffer                Pointer to the buffer of pages to free.

-  @param  Pages                 The number of 4 KB pages to free.

-

-**/

-VOID

-EFIAPI

-FreePages (

-  IN VOID   *Buffer,

-  IN UINTN  Pages

-  )

-{

-  EFI_STATUS  Status;

-

-  ASSERT (Pages != 0);

-  if (BufferInSmram (Buffer)) {

-    //

-    // When Buffer is in SMRAM range, it should be allocated by SmmAllocatePages() service.

-    // So, SmmFreePages() service is used to free it.

-    //

-    Status = SmmFreePages ((EFI_PHYSICAL_ADDRESS) (UINTN) Buffer, Pages);

-  } else {

-    //

-    // When Buffer is out of SMRAM range, it should be allocated by gBS->AllocatePages() service.

-    // So, gBS->FreePages() service is used to free it.

-    //

-    Status = gBS->FreePages ((EFI_PHYSICAL_ADDRESS) (UINTN) Buffer, Pages);

-  }

-  ASSERT_EFI_ERROR (Status);

-}

-

-/**

-  Allocates one or more 4KB pages of a certain memory type at a specified alignment.

-

-  Allocates the number of 4KB pages specified by Pages of a certain memory type with an alignment

-  specified by Alignment.  The allocated buffer is returned.  If Pages is 0, then NULL is returned.

-  If there is not enough memory at the specified alignment remaining to satisfy the request, then

-  NULL is returned.

-  If Alignment is not a power of two and Alignment is not zero, then ASSERT().

-  If Pages plus EFI_SIZE_TO_PAGES (Alignment) overflows, then ASSERT().

-

-  @param  MemoryType            The type of memory to allocate.

-  @param  Pages                 The number of 4 KB pages to allocate.

-  @param  Alignment             The requested alignment of the allocation.  Must be a power of two.

-                                If Alignment is zero, then byte alignment is used.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-InternalAllocateAlignedPages (

-  IN EFI_MEMORY_TYPE  MemoryType,  

-  IN UINTN            Pages,

-  IN UINTN            Alignment

-  )

-{

-  EFI_STATUS            Status;

-  EFI_PHYSICAL_ADDRESS  Memory;

-  UINTN                 AlignedMemory;

-  UINTN                 AlignmentMask;

-  UINTN                 UnalignedPages;

-  UINTN                 RealPages;

-

-  //

-  // Alignment must be a power of two or zero.

-  //

-  ASSERT ((Alignment & (Alignment - 1)) == 0);

- 

-  if (Pages == 0) {

-    return NULL;

-  }

-  if (Alignment > EFI_PAGE_SIZE) {

-    //

-    // Calculate the total number of pages since alignment is larger than page size.

-    //

-    AlignmentMask  = Alignment - 1;

-    RealPages      = Pages + EFI_SIZE_TO_PAGES (Alignment);

-    //

-    // Make sure that Pages plus EFI_SIZE_TO_PAGES (Alignment) does not overflow.

-    //

-    ASSERT (RealPages > Pages);

- 

-    Status         = SmmAllocatePages (AllocateAnyPages, MemoryType, RealPages, &Memory);

-    if (EFI_ERROR (Status)) {

-      return NULL;

-    }

-    AlignedMemory  = ((UINTN) Memory + AlignmentMask) & ~AlignmentMask;

-    UnalignedPages = EFI_SIZE_TO_PAGES (AlignedMemory - (UINTN) Memory);

-    if (UnalignedPages > 0) {

-      //

-      // Free first unaligned page(s).

-      //

-      Status = SmmFreePages (Memory, UnalignedPages);

-      ASSERT_EFI_ERROR (Status);

-    }

-    Memory         = (EFI_PHYSICAL_ADDRESS) (AlignedMemory + EFI_PAGES_TO_SIZE (Pages));

-    UnalignedPages = RealPages - Pages - UnalignedPages;

-    if (UnalignedPages > 0) {

-      //

-      // Free last unaligned page(s).

-      //

-      Status = SmmFreePages (Memory, UnalignedPages);

-      ASSERT_EFI_ERROR (Status);

-    }

-  } else {

-    //

-    // Do not over-allocate pages in this case.

-    //

-    Status = SmmAllocatePages (AllocateAnyPages, MemoryType, Pages, &Memory);

-    if (EFI_ERROR (Status)) {

-      return NULL;

-    }

-    AlignedMemory  = (UINTN) Memory;

-  }

-  return (VOID *) AlignedMemory;

-}

-

-/**

-  Allocates one or more 4KB pages of type EfiRuntimeServicesData at a specified alignment.

-

-  Allocates the number of 4KB pages specified by Pages of type EfiRuntimeServicesData with an

-  alignment specified by Alignment.  The allocated buffer is returned.  If Pages is 0, then NULL is

-  returned.  If there is not enough memory at the specified alignment remaining to satisfy the

-  request, then NULL is returned.

-  

-  If Alignment is not a power of two and Alignment is not zero, then ASSERT().

-  If Pages plus EFI_SIZE_TO_PAGES (Alignment) overflows, then ASSERT().

-

-  @param  Pages                 The number of 4 KB pages to allocate.

-  @param  Alignment             The requested alignment of the allocation.  Must be a power of two.

-                                If Alignment is zero, then byte alignment is used.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocateAlignedPages (

-  IN UINTN  Pages,

-  IN UINTN  Alignment

-  )

-{

-  return InternalAllocateAlignedPages (EfiRuntimeServicesData, Pages, Alignment);

-}

-

-/**

-  Allocates one or more 4KB pages of type EfiRuntimeServicesData at a specified alignment.

-

-  Allocates the number of 4KB pages specified by Pages of type EfiRuntimeServicesData with an

-  alignment specified by Alignment.  The allocated buffer is returned.  If Pages is 0, then NULL is

-  returned.  If there is not enough memory at the specified alignment remaining to satisfy the

-  request, then NULL is returned.

-  

-  If Alignment is not a power of two and Alignment is not zero, then ASSERT().

-  If Pages plus EFI_SIZE_TO_PAGES (Alignment) overflows, then ASSERT().

-

-  @param  Pages                 The number of 4 KB pages to allocate.

-  @param  Alignment             The requested alignment of the allocation.  Must be a power of two.

-                                If Alignment is zero, then byte alignment is used.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocateAlignedRuntimePages (

-  IN UINTN  Pages,

-  IN UINTN  Alignment

-  )

-{

-  return InternalAllocateAlignedPages (EfiRuntimeServicesData, Pages, Alignment);

-}

-

-/**

-  Allocates one or more 4KB pages of type EfiReservedMemoryType at a specified alignment.

-

-  Allocates the number of 4KB pages specified by Pages of type EfiReservedMemoryType with an

-  alignment specified by Alignment.  The allocated buffer is returned.  If Pages is 0, then NULL is

-  returned.  If there is not enough memory at the specified alignment remaining to satisfy the

-  request, then NULL is returned.

-  

-  If Alignment is not a power of two and Alignment is not zero, then ASSERT().

-  If Pages plus EFI_SIZE_TO_PAGES (Alignment) overflows, then ASSERT().

-

-  @param  Pages                 The number of 4 KB pages to allocate.

-  @param  Alignment             The requested alignment of the allocation.  Must be a power of two.

-                                If Alignment is zero, then byte alignment is used.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocateAlignedReservedPages (

-  IN UINTN  Pages,

-  IN UINTN  Alignment

-  )

-{

-  return NULL;

-}

-

-/**

-  Frees one or more 4KB pages that were previously allocated with one of the aligned page

-  allocation functions in the Memory Allocation Library.

-

-  Frees the number of 4KB pages specified by Pages from the buffer specified by Buffer.  Buffer

-  must have been allocated on a previous call to the aligned page allocation services of the Memory

-  Allocation Library.  If it is not possible to free allocated pages, then this function will 

-  perform no actions.

-  

-  If Buffer was not allocated with an aligned page allocation function in the Memory Allocation

-  Library, then ASSERT().

-  If Pages is zero, then ASSERT().

-  

-  @param  Buffer                Pointer to the buffer of pages to free.

-  @param  Pages                 The number of 4 KB pages to free.

-

-**/

-VOID

-EFIAPI

-FreeAlignedPages (

-  IN VOID   *Buffer,

-  IN UINTN  Pages

-  )

-{

-  EFI_STATUS  Status;

-

-  ASSERT (Pages != 0);

-  if (BufferInSmram (Buffer)) {

-    //

-    // When Buffer is in SMRAM range, it should be allocated by SmmAllocatePages() service.

-    // So, SmmFreePages() service is used to free it.

-    //

-    Status = SmmFreePages ((EFI_PHYSICAL_ADDRESS) (UINTN) Buffer, Pages);

-  } else {

-    //

-    // When Buffer is out of SMRAM range, it should be allocated by gBS->AllocatePages() service.

-    // So, gBS->FreePages() service is used to free it.

-    //

-    Status = gBS->FreePages ((EFI_PHYSICAL_ADDRESS) (UINTN) Buffer, Pages);

-  }

-  ASSERT_EFI_ERROR (Status);

-}

-

-/**

-  Allocates a buffer of a certain pool type.

-

-  Allocates the number bytes specified by AllocationSize of a certain pool type and returns a

-  pointer to the allocated buffer.  If AllocationSize is 0, then a valid buffer of 0 size is

-  returned.  If there is not enough memory remaining to satisfy the request, then NULL is returned.

-

-  @param  MemoryType            The type of memory to allocate.

-  @param  AllocationSize        The number of bytes to allocate.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-InternalAllocatePool (

-  IN EFI_MEMORY_TYPE  MemoryType,  

-  IN UINTN            AllocationSize

-  )

-{

-  EFI_STATUS  Status;

-  VOID        *Memory;

-

-  Memory = NULL;

-

-  Status = SmmAllocatePool (MemoryType, AllocationSize, &Memory);

-  if (EFI_ERROR (Status)) {

-    Memory = NULL;

-  }

-  return Memory;

-}

-

-/**

-  Allocates a buffer of type EfiRuntimeServicesData.

-

-  Allocates the number bytes specified by AllocationSize of type EfiRuntimeServicesData and returns a

-  pointer to the allocated buffer.  If AllocationSize is 0, then a valid buffer of 0 size is

-  returned.  If there is not enough memory remaining to satisfy the request, then NULL is returned.

-

-  @param  AllocationSize        The number of bytes to allocate.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocatePool (

-  IN UINTN  AllocationSize

-  )

-{

-  return InternalAllocatePool (EfiRuntimeServicesData, AllocationSize);

-}

-

-/**

-  Allocates a buffer of type EfiRuntimeServicesData.

-

-  Allocates the number bytes specified by AllocationSize of type EfiRuntimeServicesData and returns

-  a pointer to the allocated buffer.  If AllocationSize is 0, then a valid buffer of 0 size is

-  returned.  If there is not enough memory remaining to satisfy the request, then NULL is returned.

-

-  @param  AllocationSize        The number of bytes to allocate.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocateRuntimePool (

-  IN UINTN  AllocationSize

-  )

-{

-  return InternalAllocatePool (EfiRuntimeServicesData, AllocationSize);

-}

-

-/**

-  Allocates a buffer of type EfiReservedMemoryType.

-

-  Allocates the number bytes specified by AllocationSize of type EfiReservedMemoryType and returns

-  a pointer to the allocated buffer.  If AllocationSize is 0, then a valid buffer of 0 size is

-  returned.  If there is not enough memory remaining to satisfy the request, then NULL is returned.

-

-  @param  AllocationSize        The number of bytes to allocate.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocateReservedPool (

-  IN UINTN  AllocationSize

-  )

-{

-  return NULL;

-}

-

-/**

-  Allocates and zeros a buffer of a certain pool type.

-

-  Allocates the number bytes specified by AllocationSize of a certain pool type, clears the buffer

-  with zeros, and returns a pointer to the allocated buffer.  If AllocationSize is 0, then a valid

-  buffer of 0 size is returned.  If there is not enough memory remaining to satisfy the request,

-  then NULL is returned.

-

-  @param  PoolType              The type of memory to allocate.

-  @param  AllocationSize        The number of bytes to allocate and zero.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-InternalAllocateZeroPool (

-  IN EFI_MEMORY_TYPE  PoolType,  

-  IN UINTN            AllocationSize

-  ) 

-{

-  VOID  *Memory;

-

-  Memory = InternalAllocatePool (PoolType, AllocationSize);

-  if (Memory != NULL) {

-    Memory = ZeroMem (Memory, AllocationSize);

-  }

-  return Memory;

-}

-

-/**

-  Allocates and zeros a buffer of type EfiRuntimeServicesData.

-

-  Allocates the number bytes specified by AllocationSize of type EfiRuntimeServicesData, clears the

-  buffer with zeros, and returns a pointer to the allocated buffer.  If AllocationSize is 0, then a

-  valid buffer of 0 size is returned.  If there is not enough memory remaining to satisfy the

-  request, then NULL is returned.

-

-  @param  AllocationSize        The number of bytes to allocate and zero.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocateZeroPool (

-  IN UINTN  AllocationSize

-  )

-{

-  return InternalAllocateZeroPool (EfiRuntimeServicesData, AllocationSize);

-}

-

-/**

-  Allocates and zeros a buffer of type EfiRuntimeServicesData.

-

-  Allocates the number bytes specified by AllocationSize of type EfiRuntimeServicesData, clears the

-  buffer with zeros, and returns a pointer to the allocated buffer.  If AllocationSize is 0, then a

-  valid buffer of 0 size is returned.  If there is not enough memory remaining to satisfy the

-  request, then NULL is returned.

-

-  @param  AllocationSize        The number of bytes to allocate and zero.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocateRuntimeZeroPool (

-  IN UINTN  AllocationSize

-  )

-{

-  return InternalAllocateZeroPool (EfiRuntimeServicesData, AllocationSize);

-}

-

-/**

-  Allocates and zeros a buffer of type EfiReservedMemoryType.

-

-  Allocates the number bytes specified by AllocationSize of type EfiReservedMemoryType, clears the

-  buffer with zeros, and returns a pointer to the allocated buffer.  If AllocationSize is 0, then a

-  valid buffer of 0 size is returned.  If there is not enough memory remaining to satisfy the

-  request, then NULL is returned.

-

-  @param  AllocationSize        The number of bytes to allocate and zero.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocateReservedZeroPool (

-  IN UINTN  AllocationSize

-  )

-{

-  return NULL;

-}

-

-/**

-  Copies a buffer to an allocated buffer of a certain pool type.

-

-  Allocates the number bytes specified by AllocationSize of a certain pool type, copies

-  AllocationSize bytes from Buffer to the newly allocated buffer, and returns a pointer to the

-  allocated buffer.  If AllocationSize is 0, then a valid buffer of 0 size is returned.  If there

-  is not enough memory remaining to satisfy the request, then NULL is returned.

-  If Buffer is NULL, then ASSERT().

-  If AllocationSize is greater than (MAX_ADDRESS - Buffer + 1), then ASSERT(). 

-

-  @param  PoolType              The type of pool to allocate.

-  @param  AllocationSize        The number of bytes to allocate and zero.

-  @param  Buffer                The buffer to copy to the allocated buffer.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-InternalAllocateCopyPool (

-  IN EFI_MEMORY_TYPE  PoolType,  

-  IN UINTN            AllocationSize,

-  IN CONST VOID       *Buffer

-  ) 

-{

-  VOID  *Memory;

-

-  ASSERT (Buffer != NULL);

-  ASSERT (AllocationSize <= (MAX_ADDRESS - (UINTN) Buffer + 1));

-

-  Memory = InternalAllocatePool (PoolType, AllocationSize);

-  if (Memory != NULL) {

-     Memory = CopyMem (Memory, Buffer, AllocationSize);

-  }

-  return Memory;

-} 

-

-/**

-  Copies a buffer to an allocated buffer of type EfiRuntimeServicesData.

-

-  Allocates the number bytes specified by AllocationSize of type EfiRuntimeServicesData, copies

-  AllocationSize bytes from Buffer to the newly allocated buffer, and returns a pointer to the

-  allocated buffer.  If AllocationSize is 0, then a valid buffer of 0 size is returned.  If there

-  is not enough memory remaining to satisfy the request, then NULL is returned.

-  

-  If Buffer is NULL, then ASSERT().

-  If AllocationSize is greater than (MAX_ADDRESS - Buffer + 1), then ASSERT(). 

-

-  @param  AllocationSize        The number of bytes to allocate and zero.

-  @param  Buffer                The buffer to copy to the allocated buffer.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocateCopyPool (

-  IN UINTN       AllocationSize,

-  IN CONST VOID  *Buffer

-  )

-{

-  return InternalAllocateCopyPool (EfiRuntimeServicesData, AllocationSize, Buffer);

-}

-

-/**

-  Copies a buffer to an allocated buffer of type EfiRuntimeServicesData.

-

-  Allocates the number bytes specified by AllocationSize of type EfiRuntimeServicesData, copies

-  AllocationSize bytes from Buffer to the newly allocated buffer, and returns a pointer to the

-  allocated buffer.  If AllocationSize is 0, then a valid buffer of 0 size is returned.  If there

-  is not enough memory remaining to satisfy the request, then NULL is returned.

-  

-  If Buffer is NULL, then ASSERT().

-  If AllocationSize is greater than (MAX_ADDRESS - Buffer + 1), then ASSERT(). 

-

-  @param  AllocationSize        The number of bytes to allocate and zero.

-  @param  Buffer                The buffer to copy to the allocated buffer.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocateRuntimeCopyPool (

-  IN UINTN       AllocationSize,

-  IN CONST VOID  *Buffer

-  )

-{

-  return InternalAllocateCopyPool (EfiRuntimeServicesData, AllocationSize, Buffer);

-}

-

-/**

-  Copies a buffer to an allocated buffer of type EfiReservedMemoryType.

-

-  Allocates the number bytes specified by AllocationSize of type EfiReservedMemoryType, copies

-  AllocationSize bytes from Buffer to the newly allocated buffer, and returns a pointer to the

-  allocated buffer.  If AllocationSize is 0, then a valid buffer of 0 size is returned.  If there

-  is not enough memory remaining to satisfy the request, then NULL is returned.

-  

-  If Buffer is NULL, then ASSERT().

-  If AllocationSize is greater than (MAX_ADDRESS - Buffer + 1), then ASSERT(). 

-

-  @param  AllocationSize        The number of bytes to allocate and zero.

-  @param  Buffer                The buffer to copy to the allocated buffer.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-AllocateReservedCopyPool (

-  IN UINTN       AllocationSize,

-  IN CONST VOID  *Buffer

-  )

-{

-  return NULL;

-}

-

-/**

-  Reallocates a buffer of a specified memory type.

-

-  Allocates and zeros the number bytes specified by NewSize from memory of the type

-  specified by PoolType.  If OldBuffer is not NULL, then the smaller of OldSize and 

-  NewSize bytes are copied from OldBuffer to the newly allocated buffer, and 

-  OldBuffer is freed.  A pointer to the newly allocated buffer is returned.  

-  If NewSize is 0, then a valid buffer of 0 size is  returned.  If there is not 

-  enough memory remaining to satisfy the request, then NULL is returned.

-  

-  If the allocation of the new buffer is successful and the smaller of NewSize and OldSize

-  is greater than (MAX_ADDRESS - OldBuffer + 1), then ASSERT().

-

-  @param  PoolType       The type of pool to allocate.

-  @param  OldSize        The size, in bytes, of OldBuffer.

-  @param  NewSize        The size, in bytes, of the buffer to reallocate.

-  @param  OldBuffer      The buffer to copy to the allocated buffer.  This is an optional 

-                         parameter that may be NULL.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-InternalReallocatePool (

-  IN EFI_MEMORY_TYPE  PoolType,  

-  IN UINTN            OldSize,

-  IN UINTN            NewSize,

-  IN VOID             *OldBuffer  OPTIONAL

-  )

-{

-  VOID  *NewBuffer;

-

-  NewBuffer = InternalAllocateZeroPool (PoolType, NewSize);

-  if (NewBuffer != NULL && OldBuffer != NULL) {

-    CopyMem (NewBuffer, OldBuffer, MIN (OldSize, NewSize));

-    FreePool (OldBuffer);

-  }

-  return NewBuffer;

-}

-

-/**

-  Reallocates a buffer of type EfiRuntimeServicesData.

-

-  Allocates and zeros the number bytes specified by NewSize from memory of type

-  EfiRuntimeServicesData.  If OldBuffer is not NULL, then the smaller of OldSize and 

-  NewSize bytes are copied from OldBuffer to the newly allocated buffer, and 

-  OldBuffer is freed.  A pointer to the newly allocated buffer is returned.  

-  If NewSize is 0, then a valid buffer of 0 size is  returned.  If there is not 

-  enough memory remaining to satisfy the request, then NULL is returned.

-  

-  If the allocation of the new buffer is successful and the smaller of NewSize and OldSize

-  is greater than (MAX_ADDRESS - OldBuffer + 1), then ASSERT().

-

-  @param  OldSize        The size, in bytes, of OldBuffer.

-  @param  NewSize        The size, in bytes, of the buffer to reallocate.

-  @param  OldBuffer      The buffer to copy to the allocated buffer.  This is an optional 

-                         parameter that may be NULL.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-ReallocatePool (

-  IN UINTN  OldSize,

-  IN UINTN  NewSize,

-  IN VOID   *OldBuffer  OPTIONAL

-  )

-{

-  return InternalReallocatePool (EfiRuntimeServicesData, OldSize, NewSize, OldBuffer);

-}

-

-/**

-  Reallocates a buffer of type EfiRuntimeServicesData.

-

-  Allocates and zeros the number bytes specified by NewSize from memory of type

-  EfiRuntimeServicesData.  If OldBuffer is not NULL, then the smaller of OldSize and 

-  NewSize bytes are copied from OldBuffer to the newly allocated buffer, and 

-  OldBuffer is freed.  A pointer to the newly allocated buffer is returned.  

-  If NewSize is 0, then a valid buffer of 0 size is  returned.  If there is not 

-  enough memory remaining to satisfy the request, then NULL is returned.

-

-  If the allocation of the new buffer is successful and the smaller of NewSize and OldSize

-  is greater than (MAX_ADDRESS - OldBuffer + 1), then ASSERT().

-

-  @param  OldSize        The size, in bytes, of OldBuffer.

-  @param  NewSize        The size, in bytes, of the buffer to reallocate.

-  @param  OldBuffer      The buffer to copy to the allocated buffer.  This is an optional 

-                         parameter that may be NULL.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-ReallocateRuntimePool (

-  IN UINTN  OldSize,

-  IN UINTN  NewSize,

-  IN VOID   *OldBuffer  OPTIONAL

-  )

-{

-  return InternalReallocatePool (EfiRuntimeServicesData, OldSize, NewSize, OldBuffer);

-}

-

-/**

-  Reallocates a buffer of type EfiReservedMemoryType.

-

-  Allocates and zeros the number bytes specified by NewSize from memory of type

-  EfiReservedMemoryType.  If OldBuffer is not NULL, then the smaller of OldSize and 

-  NewSize bytes are copied from OldBuffer to the newly allocated buffer, and 

-  OldBuffer is freed.  A pointer to the newly allocated buffer is returned.  

-  If NewSize is 0, then a valid buffer of 0 size is  returned.  If there is not 

-  enough memory remaining to satisfy the request, then NULL is returned.

-

-  If the allocation of the new buffer is successful and the smaller of NewSize and OldSize

-  is greater than (MAX_ADDRESS - OldBuffer + 1), then ASSERT().

-

-  @param  OldSize        The size, in bytes, of OldBuffer.

-  @param  NewSize        The size, in bytes, of the buffer to reallocate.

-  @param  OldBuffer      The buffer to copy to the allocated buffer.  This is an optional 

-                         parameter that may be NULL.

-

-  @return A pointer to the allocated buffer or NULL if allocation fails.

-

-**/

-VOID *

-EFIAPI

-ReallocateReservedPool (

-  IN UINTN  OldSize,

-  IN UINTN  NewSize,

-  IN VOID   *OldBuffer  OPTIONAL

-  )

-{

-  return NULL;

-}

-

-/**

-  Frees a buffer that was previously allocated with one of the pool allocation functions in the

-  Memory Allocation Library.

-

-  Frees the buffer specified by Buffer.  Buffer must have been allocated on a previous call to the

-  pool allocation services of the Memory Allocation Library.  If it is not possible to free pool

-  resources, then this function will perform no actions.

-  

-  If Buffer was not allocated with a pool allocation function in the Memory Allocation Library,

-  then ASSERT().

-

-  @param  Buffer                Pointer to the buffer to free.

-

-**/

-VOID

-EFIAPI

-FreePool (

-  IN VOID   *Buffer

-  )

-{

-  EFI_STATUS    Status;

-

-  if (BufferInSmram (Buffer)) {

-    //

-    // When Buffer is in SMRAM range, it should be allocated by SmmAllocatePool() service.

-    // So, SmmFreePool() service is used to free it.

-    //

-    Status = SmmFreePool (Buffer);

-  } else {

-    //

-    // When Buffer is out of SMRAM range, it should be allocated by gBS->AllocatePool() service.

-    // So, gBS->FreePool() service is used to free it.

-    //

-    Status = gBS->FreePool (Buffer);

-  }

-  ASSERT_EFI_ERROR (Status);

-}

-

-/**

-  The constructor function calls SmmInitializeMemoryServices to initialize memory in SMRAM.

-

-  @param  ImageHandle   The firmware allocated handle for the EFI image.

-  @param  SystemTable   A pointer to the EFI System Table.

-

-  @retval EFI_SUCCESS   The constructor always returns EFI_SUCCESS.

-

-**/

-EFI_STATUS

-EFIAPI

-PiSmmCoreMemoryAllocationLibConstructor (

-  IN EFI_HANDLE        ImageHandle,

-  IN EFI_SYSTEM_TABLE  *SystemTable

-  )

-{

-  SMM_CORE_PRIVATE_DATA  *SmmCorePrivate;

-  UINTN                  Size;

-

-  SmmCorePrivate = (SMM_CORE_PRIVATE_DATA *)ImageHandle;

-  //

-  // Initialize memory service using free SMRAM

-  //

-  SmmInitializeMemoryServices (SmmCorePrivate->SmramRangeCount, SmmCorePrivate->SmramRanges);

-

-  mSmmCoreMemoryAllocLibSmramRangeCount = SmmCorePrivate->SmramRangeCount;

-  Size = mSmmCoreMemoryAllocLibSmramRangeCount * sizeof (EFI_SMRAM_DESCRIPTOR);

-  mSmmCoreMemoryAllocLibSmramRanges = (EFI_SMRAM_DESCRIPTOR *) AllocatePool (Size);

-  ASSERT (mSmmCoreMemoryAllocLibSmramRanges != NULL);

-  CopyMem (mSmmCoreMemoryAllocLibSmramRanges, SmmCorePrivate->SmramRanges, Size);

-

-  return EFI_SUCCESS;

-}

diff --git a/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationLib.inf b/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationLib.inf
deleted file mode 100644
index e8f7081149..0000000000
--- a/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationLib.inf
+++ /dev/null
@@ -1,48 +0,0 @@
-## @file

-# Memory Allocation Library instance dedicated to SMM Core.

-# The implementation borrows the SMM Core Memory Allocation services as the primitive

-# for memory allocation instead of using SMM System Table servces in an indirect way.

-# It is assumed that this library instance must be linked with SMM Cre in this package. 

-#

-# Copyright (c) 2010 - 2015, Intel Corporation. All rights reserved.<BR>

-#

-#  This program and the accompanying materials

-#  are licensed and made available under the terms and conditions of the BSD License

-#  which accompanies this distribution. The full text of the license may be found at

-#  http://opensource.org/licenses/bsd-license.php

-#  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,

-#  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.

-#

-##

-

-[Defines]

-  INF_VERSION                    = 0x00010005

-  BASE_NAME                      = PiSmmCoreMemoryAllocationLib

-  MODULE_UNI_FILE                = PiSmmCoreMemoryAllocationLib.uni

-  FILE_GUID                      = B618E089-9ABA-4d97-AE80-57B5BCCDA51D

-  MODULE_TYPE                    = SMM_CORE

-  VERSION_STRING                 = 1.0

-  PI_SPECIFICATION_VERSION       = 0x0001000A

-  LIBRARY_CLASS                  = MemoryAllocationLib|SMM_CORE

-  CONSTRUCTOR                    = PiSmmCoreMemoryAllocationLibConstructor

-

-#

-# The following information is for reference only and not required by the build tools.

-#

-#  VALID_ARCHITECTURES           = IA32 X64

-#

-

-[Sources]

-  MemoryAllocationLib.c

-  PiSmmCoreMemoryAllocationServices.h

-

-[Packages]

-  MdePkg/MdePkg.dec

-

-[LibraryClasses]

-  DebugLib

-  BaseMemoryLib

-  UefiBootServicesTableLib

-

-[Protocols]

-  gEfiSmmAccess2ProtocolGuid    ## CONSUMES  

diff --git a/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationLib.uni b/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationLib.uni
deleted file mode 100644
index adfc0381c1..0000000000
--- a/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationLib.uni
+++ /dev/null
diff --git a/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationServices.h b/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationServices.h
deleted file mode 100644
index a2b89acf5d..0000000000
--- a/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationServices.h
+++ /dev/null
@@ -1,191 +0,0 @@
-/** @file

-  Contains function prototypes for Memory Services in the SMM Core.

-

-  This header file borrows the PiSmmCore Memory Allocation services as the primitive

-  for memory allocation. 

-

-  Copyright (c) 2008 - 2015, Intel Corporation. All rights reserved.<BR>

-  This program and the accompanying materials                          

-  are licensed and made available under the terms and conditions of the BSD License         

-  which accompanies this distribution.  The full text of the license may be found at        

-  http://opensource.org/licenses/bsd-license.php                                            

-

-  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,                     

-  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.             

-

-**/

-

-#ifndef _PI_SMM_CORE_MEMORY_ALLOCATION_SERVICES_H_

-#define _PI_SMM_CORE_MEMORY_ALLOCATION_SERVICES_H_

-

-//

-// It should be aligned with the definition in PiSmmCore.

-//

-typedef struct {

-  UINTN                           Signature;

-

-  ///

-  /// The ImageHandle passed into the entry point of the SMM IPL.  This ImageHandle

-  /// is used by the SMM Core to fill in the ParentImageHandle field of the Loaded

-  /// Image Protocol for each SMM Driver that is dispatched by the SMM Core.

-  ///

-  EFI_HANDLE                      SmmIplImageHandle;

-

-  ///

-  /// The number of SMRAM ranges passed from the SMM IPL to the SMM Core.  The SMM

-  /// Core uses these ranges of SMRAM to initialize the SMM Core memory manager.

-  ///

-  UINTN                           SmramRangeCount;

-

-  ///

-  /// A table of SMRAM ranges passed from the SMM IPL to the SMM Core.  The SMM

-  /// Core uses these ranges of SMRAM to initialize the SMM Core memory manager.

-  ///

-  EFI_SMRAM_DESCRIPTOR            *SmramRanges;

-

-  ///

-  /// The SMM Foundation Entry Point.  The SMM Core fills in this field when the 

-  /// SMM Core is initialized.  The SMM IPL is responsbile for registering this entry 

-  /// point with the SMM Configuration Protocol.  The SMM Configuration Protocol may 

-  /// not be available at the time the SMM IPL and SMM Core are started, so the SMM IPL

-  /// sets up a protocol notification on the SMM Configuration Protocol and registers 

-  /// the SMM Foundation Entry Point as soon as the SMM Configuration Protocol is 

-  /// available.

-  ///

-  EFI_SMM_ENTRY_POINT             SmmEntryPoint;

-  

-  ///

-  /// Boolean flag set to TRUE while an SMI is being processed by the SMM Core.

-  /// 

-  BOOLEAN                         SmmEntryPointRegistered;

-

-  ///

-  /// Boolean flag set to TRUE while an SMI is being processed by the SMM Core.

-  /// 

-  BOOLEAN                         InSmm;

-

-  ///

-  /// This field is set by the SMM Core then the SMM Core is initialized.  This field is

-  /// used by the SMM Base 2 Protocol and SMM Communication Protocol implementations in

-  /// the SMM IPL.  

-  ///

-  EFI_SMM_SYSTEM_TABLE2           *Smst;

-

-  ///

-  /// This field is used by the SMM Communicatioon Protocol to pass a buffer into 

-  /// a software SMI handler and for the software SMI handler to pass a buffer back to

-  /// the caller of the SMM Communication Protocol.  

-  ///

-  VOID                            *CommunicationBuffer;

-

-  ///

-  /// This field is used by the SMM Communicatioon Protocol to pass the size of a buffer,

-  /// in bytes, into a software SMI handler and for the software SMI handler to pass the 

-  /// size, in bytes, of a buffer back to the caller of the SMM Communication Protocol.

-  ///

-  UINTN                           BufferSize;

-

-  ///

-  /// This field is used by the SMM Communication Protocol to pass the return status from

-  /// a software SMI handler back to the caller of the SMM Communication Protocol.

-  ///

-  EFI_STATUS                      ReturnStatus;

-

-  EFI_PHYSICAL_ADDRESS            PiSmmCoreImageBase;

-  UINT64                          PiSmmCoreImageSize;

-  EFI_PHYSICAL_ADDRESS            PiSmmCoreEntryPoint;

-} SMM_CORE_PRIVATE_DATA;

-

-/**

-  Called to initialize the memory service.

-

-  @param   SmramRangeCount       Number of SMRAM Regions

-  @param   SmramRanges           Pointer to SMRAM Descriptors

-

-**/

-VOID

-SmmInitializeMemoryServices (

-  IN UINTN                 SmramRangeCount,

-  IN EFI_SMRAM_DESCRIPTOR  *SmramRanges

-  );

-

-/**

-  Allocates pages from the memory map.

-

-  @param  Type                   The type of allocation to perform

-  @param  MemoryType             The type of memory to turn the allocated pages

-                                 into

-  @param  NumberOfPages          The number of pages to allocate

-  @param  Memory                 A pointer to receive the base allocated memory

-                                 address

-

-  @retval EFI_INVALID_PARAMETER  Parameters violate checking rules defined in spec.

-  @retval EFI_NOT_FOUND          Could not allocate pages match the requirement.

-  @retval EFI_OUT_OF_RESOURCES   No enough pages to allocate.

-  @retval EFI_SUCCESS            Pages successfully allocated.

-

-**/

-EFI_STATUS

-EFIAPI

-SmmAllocatePages (

-  IN      EFI_ALLOCATE_TYPE         Type,

-  IN      EFI_MEMORY_TYPE           MemoryType,

-  IN      UINTN                     NumberOfPages,

-  OUT     EFI_PHYSICAL_ADDRESS      *Memory

-  );

-

-/**

-  Frees previous allocated pages.

-

-  @param  Memory                 Base address of memory being freed

-  @param  NumberOfPages          The number of pages to free

-

-  @retval EFI_NOT_FOUND          Could not find the entry that covers the range

-  @retval EFI_INVALID_PARAMETER  Address not aligned

-  @return EFI_SUCCESS            Pages successfully freed.

-

-**/

-EFI_STATUS

-EFIAPI

-SmmFreePages (

-  IN      EFI_PHYSICAL_ADDRESS      Memory,

-  IN      UINTN                     NumberOfPages

-  );

-

-/**

-  Allocate pool of a particular type.

-

-  @param  PoolType               Type of pool to allocate

-  @param  Size                   The amount of pool to allocate

-  @param  Buffer                 The address to return a pointer to the allocated

-                                 pool

-

-  @retval EFI_INVALID_PARAMETER  PoolType not valid

-  @retval EFI_OUT_OF_RESOURCES   Size exceeds max pool size or allocation failed.

-  @retval EFI_SUCCESS            Pool successfully allocated.

-

-**/

-EFI_STATUS

-EFIAPI

-SmmAllocatePool (

-  IN      EFI_MEMORY_TYPE           PoolType,

-  IN      UINTN                     Size,

-  OUT     VOID                      **Buffer

-  );

-

-/**

-  Frees pool.

-

-  @param  Buffer                 The allocated pool entry to free

-

-  @retval EFI_INVALID_PARAMETER  Buffer is not a valid value.

-  @retval EFI_SUCCESS            Pool successfully freed.

-

-**/

-EFI_STATUS

-EFIAPI

-SmmFreePool (

-  IN      VOID                      *Buffer

-  );

-

-#endif