X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=MdeModulePkg%2FCore%2FPiSmmCore%2FPiSmmCore.h;h=d494519d2c2047bab9a92d4e845bc00f7c1ac155;hp=5cd123705600b0e5ad67add6a3c058ae7ec42f52;hb=84edd20bd0756ef5719835498d4283435d6b5e77;hpb=fa542a1e770ac651b7a2f9e6650210465c824f7c

diff --git a/MdeModulePkg/Core/PiSmmCore/PiSmmCore.h b/MdeModulePkg/Core/PiSmmCore/PiSmmCore.h
index 5cd1237056..d494519d2c 100644
--- a/MdeModulePkg/Core/PiSmmCore/PiSmmCore.h
+++ b/MdeModulePkg/Core/PiSmmCore/PiSmmCore.h
@@ -2,7 +2,7 @@
   The internal header file includes the common header files, defines
   internal structure and functions used by SmmCore module.
 
-  Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>
+  Copyright (c) 2009 - 2014, 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        
@@ -20,6 +20,7 @@
 
 #include <Protocol/DxeSmmReadyToLock.h>
 #include <Protocol/SmmReadyToLock.h>
+#include <Protocol/SmmEndOfDxe.h>
 #include <Protocol/CpuIo2.h>
 #include <Protocol/SmmCommunication.h>
 #include <Protocol/SmmAccess2.h>
@@ -27,10 +28,13 @@
 #include <Protocol/LoadedImage.h>       
 #include <Protocol/DevicePath.h>        
 #include <Protocol/Security.h>          
+#include <Protocol/Security2.h>
 
 #include <Guid/Apriori.h>
 #include <Guid/EventGroup.h>
 #include <Guid/EventLegacyBios.h>
+#include <Guid/ZeroGuid.h>
+#include <Guid/MemoryProfile.h>
 
 #include <Library/BaseLib.h>
 #include <Library/BaseMemoryLib.h>
@@ -43,6 +47,9 @@
 #include <Library/UefiLib.h>                   
 #include <Library/UefiBootServicesTableLib.h>
 #include <Library/PcdLib.h>
+#include <Library/SmmCorePlatformHookLib.h>
+#include <Library/PerformanceLib.h>
+#include <Library/TimerLib.h>
 
 #include "PiSmmCorePrivateData.h"
 
@@ -266,6 +273,31 @@ SmmAllocatePages (
   OUT     EFI_PHYSICAL_ADDRESS      *Memory
   );
 
+/**
+  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
+SmmInternalAllocatePages (
+  IN      EFI_ALLOCATE_TYPE         Type,
+  IN      EFI_MEMORY_TYPE           MemoryType,
+  IN      UINTN                     NumberOfPages,
+  OUT     EFI_PHYSICAL_ADDRESS      *Memory
+  );
+
 /**
   Frees previous allocated pages.
 
@@ -284,6 +316,24 @@ SmmFreePages (
   IN      UINTN                     NumberOfPages
   );
 
+/**
+  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
+SmmInternalFreePages (
+  IN      EFI_PHYSICAL_ADDRESS      Memory,
+  IN      UINTN                     NumberOfPages
+  );
+
 /**
   Allocate pool of a particular type.
 
@@ -305,6 +355,27 @@ SmmAllocatePool (
   OUT     VOID                      **Buffer
   );
 
+/**
+  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
+SmmInternalAllocatePool (
+  IN      EFI_MEMORY_TYPE           PoolType,
+  IN      UINTN                     Size,
+  OUT     VOID                      **Buffer
+  );
+
 /**
   Frees pool.
 
@@ -320,6 +391,21 @@ SmmFreePool (
   IN      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
+SmmInternalFreePool (
+  IN      VOID                      *Buffer
+  );
+
 /**
   Installs a protocol interface into the boot services environment.
 
@@ -582,6 +668,28 @@ SmmReadyToLockHandler (
   IN OUT UINTN                    *CommBufferSize  OPTIONAL
   );
 
+/**
+  This function is the main entry point for an SMM handler dispatch
+  or communicate-based callback.
+
+  @param  DispatchHandle  The unique handle assigned to this handler by SmiHandlerRegister().
+  @param  Context         Points to an optional handler context which was specified when the handler was registered.
+  @param  CommBuffer      A pointer to a collection of data in memory that will
+                          be conveyed from a non-SMM environment into an SMM environment.
+  @param  CommBufferSize  The size of the CommBuffer.
+
+  @return Status Code
+
+**/
+EFI_STATUS
+EFIAPI
+SmmEndOfDxeHandler (
+  IN     EFI_HANDLE               DispatchHandle,
+  IN     CONST VOID               *Context,        OPTIONAL
+  IN OUT VOID                     *CommBuffer,     OPTIONAL
+  IN OUT UINTN                    *CommBufferSize  OPTIONAL
+  );
+
 /**
   Place holder function until all the SMM System Table Service are available.
 
@@ -714,4 +822,101 @@ SmmIsSchedulable (
   IN  EFI_SMM_DRIVER_ENTRY   *DriverEntry
   );
 
+//
+// SmramProfile
+//
+
+/**
+  Initialize SMRAM profile.
+
+**/
+VOID
+SmramProfileInit (
+  VOID
+  );
+
+/**
+  Register SMM image to SMRAM profile.
+
+  @param DriverEntry    SMM image info.
+  @param RegisterToDxe  Register image to DXE.
+
+  @retval TRUE          Register success.
+  @retval FALSE         Register fail.
+
+**/
+BOOLEAN
+RegisterSmramProfileImage (
+  IN EFI_SMM_DRIVER_ENTRY   *DriverEntry,
+  IN BOOLEAN                RegisterToDxe
+  );
+
+/**
+  Unregister image from SMRAM profile.
+
+  @param DriverEntry        SMM image info.
+  @param UnregisterToDxe    Unregister image from DXE.
+
+  @retval TRUE              Unregister success.
+  @retval FALSE             Unregister fail.
+
+**/
+BOOLEAN
+UnregisterSmramProfileImage (
+  IN EFI_SMM_DRIVER_ENTRY   *DriverEntry,
+  IN BOOLEAN                UnregisterToDxe
+  );
+
+/**
+  Update SMRAM profile information.
+
+  @param CallerAddress  Address of caller who call Allocate or Free.
+  @param Action         This Allocate or Free action.
+  @param MemoryType     Memory type.
+  @param Size           Buffer size.
+  @param Buffer         Buffer address.
+
+  @retval TRUE          Profile udpate success.
+  @retval FALSE         Profile update fail.
+
+**/
+BOOLEAN
+SmmCoreUpdateProfile (
+  IN EFI_PHYSICAL_ADDRESS CallerAddress,
+  IN MEMORY_PROFILE_ACTION Action,
+  IN EFI_MEMORY_TYPE      MemoryType, // Valid for AllocatePages/AllocatePool
+  IN UINTN                Size,       // Valid for AllocatePages/FreePages/AllocatePool
+  IN VOID                 *Buffer
+  );
+
+/**
+  Register SMRAM profile handler.
+
+**/
+VOID
+RegisterSmramProfileHandler (
+  VOID
+  );
+
+/**
+  SMRAM profile ready to lock callback function.
+
+**/
+VOID
+SmramProfileReadyToLock (
+  VOID
+  );
+
+/**
+  Dump SMRAM infromation.
+
+**/
+VOID
+DumpSmramInfo (
+  VOID
+  );
+
+extern UINTN                    mFullSmramRangeCount;
+extern EFI_SMRAM_DESCRIPTOR     *mFullSmramRanges;
+
 #endif