Use SmmMemLib to check communication buffer.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: "Yao, Jiewen" <jiewen.yao@intel.com>
Reviewed-by: "Gao, Liming" <liming.gao@intel.com>
Reviewed-by: "Fan, Jeff" <jeff.fan@intel.com>


git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@16694 6f19259b-4bc3-4df7-8a09-765794883524

3 files changed, 63 insertions, 4 deletions

diff --git a/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/MemoryAllocationLib.c b/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/MemoryAllocationLib.c
index f0f48438f4..e4f7756f72 100644
--- a/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/MemoryAllocationLib.c
+++ b/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/MemoryAllocationLib.c
@@ -1,7 +1,7 @@
 /** @file

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

 

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

+  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        

@@ -936,3 +936,28 @@ FreePool (
   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;

+

+  SmmCorePrivate = (SMM_CORE_PRIVATE_DATA *)ImageHandle;

+  //

+  // Initialize memory service using free SMRAM

+  //

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

+  return EFI_SUCCESS;

+}

diff --git a/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationLib.inf b/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationLib.inf
index cfd2f4a56d..e8f7081149 100644
--- a/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationLib.inf
+++ b/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationLib.inf
@@ -4,7 +4,7 @@
 # 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 - 2014, Intel Corporation. All rights reserved.<BR>

+# 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

@@ -24,7 +24,8 @@
   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.

 #

diff --git a/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationServices.h b/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationServices.h
index 84893af85c..5c74045039 100644
--- a/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationServices.h
+++ b/MdeModulePkg/Library/PiSmmCoreMemoryAllocationLib/PiSmmCoreMemoryAllocationServices.h
@@ -4,7 +4,7 @@
   This header file borrows the PiSmmCore Memory Allocation services as the primitive

   for memory allocation. 

 

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

+  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        

@@ -18,6 +18,39 @@
 #ifndef _PI_SMM_CORE_MEMORY_ALLOCATION_SERVICES_H_

 #define _PI_SMM_CORE_MEMORY_ALLOCATION_SERVICES_H_

 

+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;

+} 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.