sravan/system76-edk2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

EdkCompatibilityPkg: Add UEFI memory and SMRAM profile support.

Browse Source 
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Star Zeng <star.zeng@intel.com>
Reviewed-by: Jiewen Yao <jiewen.yao@intel.com>

git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@16336 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
Star Zeng
2014-11-12 03:30:11 +00:00
committed by lzeng14
parent 84edd20bd0
commit 09fc7dbb94
6 changed files with 281 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
EdkCompatibilityPkg/Compatibility/SmmBaseHelper/SmmBaseHelper.c
View File

@ -34,6 +34,7 @@
#include <Library/MemoryAllocationLib.h>
#include <Library/SynchronizationLib.h>
#include <Library/CpuLib.h>
#include <Library/PcdLib.h>
#include <Guid/SmmBaseThunkCommunication.h>
#include <Protocol/SmmBaseHelperReady.h>
#include <Protocol/SmmCpu.h>
@ -44,6 +45,42 @@
#include <Protocol/SmmReadyToLock.h>
#include <Protocol/SmmAccess2.h>
/**
Register SMM image to SMRAM profile.
@param[in] FilePath File path of the image.
@param[in] ImageBuffer Image base address.
@param[in] NumberOfPage Number of page.
@retval TRUE Register success.
@retval FALSE Register fail.
**/
BOOLEAN
RegisterSmramProfileImage (
IN EFI_DEVICE_PATH_PROTOCOL *FilePath,
IN PHYSICAL_ADDRESS ImageBuffer,
IN UINTN NumberOfPage
);
/**
Unregister SMM image from SMRAM profile.
@param[in] FilePath File path of the image.
@param[in] ImageBuffer Image base address.
@param[in] NumberOfPage Number of page.
@retval TRUE Unregister success.
@retval FALSE Unregister fail.
**/
BOOLEAN
UnregisterSmramProfileImage (
IN EFI_DEVICE_PATH_PROTOCOL *FilePath,
IN PHYSICAL_ADDRESS ImageBuffer,
IN UINTN NumberOfPage
);
///
/// Structure for tracking paired information of registered Framework SMI handler
/// and correpsonding dispatch handle for SMI handler thunk.
@ -694,11 +731,13 @@ LoadImage (
mFrameworkSmst->NumberOfCpus = mNumberOfProcessors;
mFrameworkSmst->CurrentlyExecutingCpu = gSmst->CurrentlyExecutingCpu;
RegisterSmramProfileImage (FilePath, DstBuffer, PageCount);
Status = gBS->StartImage (*ImageHandle, NULL, NULL);
if (EFI_ERROR (Status)) {
mLoadPe32Image->UnLoadPeImage (mLoadPe32Image, *ImageHandle);
*ImageHandle = NULL;
FreePages ((VOID *)(UINTN)DstBuffer, PageCount);
UnregisterSmramProfileImage (FilePath, DstBuffer, PageCount);
}
}