git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@8274 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			138 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			138 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /**@file
 | |
| 
 | |
| Copyright (c) 2006, Intel Corporation
 | |
| All rights reserved. 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.
 | |
| 
 | |
| Module Name:
 | |
| 
 | |
|   PeiUnixPeCoffExtraActionLib.c
 | |
| 
 | |
| Abstract:
 | |
| 
 | |
|   Provides services to perform additional actions to relocate and unload
 | |
|   PE/Coff image for UNIX environment specific purpose such as souce level debug.
 | |
|   This version only works for PEI phase
 | |
| 
 | |
| 
 | |
| **/
 | |
| #include <PiPei.h>
 | |
| #include <Ppi/UnixThunk.h>
 | |
| #include <FrameworkModuleBase.h>
 | |
| 
 | |
| #include <Library/PeCoffLib.h>
 | |
| #include <Library/PeiServicesLib.h>
 | |
| #include <Library/DebugLib.h>
 | |
| #include <Library/BaseLib.h>
 | |
| #include <Library/PeCoffExtraActionLib.h>
 | |
| 
 | |
| //
 | |
| // Cache of UnixThunk protocol 
 | |
| //
 | |
| EFI_UNIX_THUNK_PROTOCOL   *mUnix = NULL;
 | |
| 
 | |
| /**
 | |
|   The function caches the pointer of the WinNT thunk functions
 | |
|   It will ASSERT() if Unix thunk ppi is not installed.
 | |
| 
 | |
|   @retval EFI_SUCCESS   WinNT thunk protocol is found and cached.
 | |
| 
 | |
| **/
 | |
| EFI_STATUS
 | |
| EFIAPI
 | |
| UnixPeCoffGetUnixThunkStucture (
 | |
|   )
 | |
| {
 | |
| 	PEI_UNIX_THUNK_PPI  *UnixThunkPpi;
 | |
|   EFI_STATUS        Status;
 | |
| 
 | |
|   
 | |
|   //
 | |
|   // Locate Unix ThunkPpi for retrieving standard output handle
 | |
|   //
 | |
|   Status = PeiServicesLocatePpi (
 | |
|               &gPeiUnixThunkPpiGuid,
 | |
|               0,
 | |
|               NULL,
 | |
|               (VOID **) &UnixThunkPpi
 | |
|               );
 | |
| 
 | |
|   ASSERT_EFI_ERROR (Status);
 | |
| 
 | |
|   mUnix  = (EFI_UNIX_THUNK_PROTOCOL *) UnixThunkPpi->UnixThunk ();
 | |
| 
 | |
|   return EFI_SUCCESS;
 | |
| }
 | |
| 
 | |
| /**
 | |
|   Performs additional actions after a PE/COFF image has been loaded and relocated.
 | |
| 
 | |
|   If ImageContext is NULL, then ASSERT().
 | |
| 
 | |
|   @param  ImageContext  Pointer to the image context structure that describes the
 | |
|                         PE/COFF image that has already been loaded and relocated.
 | |
| 
 | |
| **/
 | |
| VOID
 | |
| EFIAPI
 | |
| PeCoffLoaderRelocateImageExtraAction (
 | |
|   IN OUT PE_COFF_LOADER_IMAGE_CONTEXT  *ImageContext
 | |
|   )
 | |
| {
 | |
|   VOID * Handle;
 | |
|   VOID * Entry;
 | |
| 
 | |
|   ASSERT (ImageContext != NULL);
 | |
| 
 | |
|   Handle = NULL;
 | |
|   Entry  = NULL;
 | |
|   
 | |
|   if (mUnix == NULL) {
 | |
|     UnixPeCoffGetUnixThunkStucture ();
 | |
|   }
 | |
|   DEBUG ((EFI_D_ERROR, "Loading %s 0x%08lx - entry point 0x%08lx\n",
 | |
|           ImageContext->PdbPointer,
 | |
|           (UINTN)ImageContext->ImageAddress,
 | |
|           (UINTN)ImageContext->EntryPoint));
 | |
| 
 | |
|   Handle = mUnix->Dlopen (ImageContext->PdbPointer, RTLD_NOW);
 | |
|   
 | |
|   if (Handle) {
 | |
|     Entry = mUnix->Dlsym(Handle, "_ModuleEntryPoint");
 | |
|   } else {
 | |
|   	DEBUG ((EFI_D_ERROR, "%s\n", mUnix->Dlerror()));
 | |
|   }
 | |
|   
 | |
|   if (Entry != NULL) {
 | |
|     ImageContext->EntryPoint = Entry;
 | |
|     DEBUG ((EFI_D_ERROR, "Change %s Entrypoint to :0x%08lx\n", ImageContext->PdbPointer, Entry));
 | |
|   }
 | |
| 
 | |
| 
 | |
|   return;
 | |
|  }  
 | |
| 
 | |
| /**
 | |
|   Performs additional actions just before a PE/COFF image is unloaded.  Any resources
 | |
|   that were allocated by PeCoffLoaderRelocateImageExtraAction() must be freed.
 | |
|   
 | |
|   If ImageContext is NULL, then ASSERT().
 | |
|   
 | |
|   @param  ImageContext  Pointer to the image context structure that describes the
 | |
|                         PE/COFF image that is being unloaded.
 | |
| 
 | |
| **/
 | |
| VOID
 | |
| EFIAPI
 | |
| PeCoffLoaderUnloadImageExtraAction (
 | |
|   IN OUT PE_COFF_LOADER_IMAGE_CONTEXT  *ImageContext
 | |
|   )
 | |
| {
 | |
|   ASSERT (ImageContext != NULL);
 | |
| }
 |