From 6cd7b51e48b2387444105f60fd0f5c90e973c297 Mon Sep 17 00:00:00 2001
From: lgao4 <lgao4@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Tue, 4 Mar 2008 09:32:41 +0000
Subject: Correct the PeImage Source file to be measured by TCG.

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

(limited to 'MdeModulePkg/Core/Dxe')

diff --git a/MdeModulePkg/Core/Dxe/Image/Image.c b/MdeModulePkg/Core/Dxe/Image/Image.c
index 125134d..3587460 100644
--- a/MdeModulePkg/Core/Dxe/Image/Image.c
+++ b/MdeModulePkg/Core/Dxe/Image/Image.c
@@ -208,7 +208,9 @@ Returns:
   UINTN                     Size;
   UINTN                     LinkTimeBase;
   EFI_TCG_PLATFORM_PROTOCOL *TcgPlatformProtocol;
+  IMAGE_FILE_HANDLE         *FHandle;
 
+  FHandle = NULL;
   ZeroMem (&Image->ImageContext, sizeof (Image->ImageContext));
 
   Image->ImageContext.Handle    = Pe32Handle;
@@ -362,17 +364,18 @@ Returns:
              NULL,
              (VOID **) &TcgPlatformProtocol
              );
-  if (!EFI_ERROR (Status)) {    
+  if (!EFI_ERROR (Status)) {
+    FHandle = (IMAGE_FILE_HANDLE *) Image->ImageContext.Handle;
     Status = TcgPlatformProtocol->MeasurePeImage (
                                     BootPolicy,
-                                    Image->ImageContext.ImageAddress,
-                                    (UINTN) Image->ImageContext.ImageSize,
+                                    (EFI_PHYSICAL_ADDRESS) (UINTN) FHandle->Source,
+                                    FHandle->SourceSize,
                                     LinkTimeBase,
                                     Image->ImageContext.ImageType,
                                     Image->Info.DeviceHandle,
                                     Image->Info.FilePath
                                     );
-    
+
     ASSERT_EFI_ERROR (Status);
   }
 
-- 
cgit v1.1