mirror of https://github.com/acidanthera/audk.git
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);
|
|
}
|