]>
Commit | Line | Data |
---|---|---|
398b646f | 1 | /**@file |
2 | ||
3 | Copyright (c) 2006, Intel Corporation | |
4 | All rights reserved. This program and the accompanying materials | |
5 | are licensed and made available under the terms and conditions of the BSD License | |
6 | which accompanies this distribution. The full text of the license may be found at | |
7 | http://opensource.org/licenses/bsd-license.php | |
8 | ||
9 | THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS, | |
10 | WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. | |
11 | ||
12 | Module Name: | |
13 | ||
14 | PeiUnixPeCoffExtraActionLib.c | |
15 | ||
16 | Abstract: | |
17 | ||
18 | Provides services to perform additional actions to relocate and unload | |
19 | PE/Coff image for UNIX environment specific purpose such as souce level debug. | |
20 | This version only works for PEI phase | |
21 | ||
22 | ||
23 | **/ | |
24 | #include <PiPei.h> | |
25 | #include <Ppi/UnixThunk.h> | |
26 | #include <FrameworkModuleBase.h> | |
27 | ||
28 | #include <Library/PeCoffLib.h> | |
29 | #include <Library/PeiServicesLib.h> | |
30 | #include <Library/DebugLib.h> | |
31 | #include <Library/BaseLib.h> | |
32 | #include <Library/PeCoffExtraActionLib.h> | |
33 | ||
34 | // | |
35 | // Cache of UnixThunk protocol | |
36 | // | |
37 | EFI_UNIX_THUNK_PROTOCOL *mUnix = NULL; | |
38 | ||
39 | /** | |
40 | The function caches the pointer of the WinNT thunk functions | |
41 | It will ASSERT() if Unix thunk ppi is not installed. | |
42 | ||
43 | @retval EFI_SUCCESS WinNT thunk protocol is found and cached. | |
44 | ||
45 | **/ | |
46 | EFI_STATUS | |
47 | EFIAPI | |
48 | UnixPeCoffGetUnixThunkStucture ( | |
49 | ) | |
50 | { | |
51 | PEI_UNIX_THUNK_PPI *UnixThunkPpi; | |
52 | EFI_STATUS Status; | |
53 | ||
54 | ||
55 | // | |
56 | // Locate Unix ThunkPpi for retrieving standard output handle | |
57 | // | |
58 | Status = PeiServicesLocatePpi ( | |
59 | &gPeiUnixThunkPpiGuid, | |
60 | 0, | |
61 | NULL, | |
62 | (VOID **) &UnixThunkPpi | |
63 | ); | |
64 | ||
65 | ASSERT_EFI_ERROR (Status); | |
66 | ||
67 | mUnix = (EFI_UNIX_THUNK_PROTOCOL *) UnixThunkPpi->UnixThunk (); | |
68 | ||
69 | return EFI_SUCCESS; | |
70 | } | |
71 | ||
72 | /** | |
6cb6f078 | 73 | Performs additional actions after a PE/COFF image has been loaded and relocated.\r |
74 | \r | |
75 | If ImageContext is NULL, then ASSERT().\r | |
76 | \r | |
77 | @param ImageContext Pointer to the image context structure that describes the\r | |
78 | PE/COFF image that has already been loaded and relocated.\r | |
398b646f | 79 | |
80 | **/ | |
81 | VOID | |
82 | EFIAPI | |
83 | PeCoffLoaderRelocateImageExtraAction ( | |
84 | IN OUT PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext | |
85 | ) | |
86 | { | |
87 | VOID * Handle;\r | |
88 | VOID * Entry;\r | |
89 | \r | |
6cb6f078 | 90 | ASSERT (ImageContext != NULL);\r |
91 | \r | |
398b646f | 92 | Handle = NULL;\r |
93 | Entry = NULL; | |
94 | ||
95 | if (mUnix == NULL) { | |
96 | UnixPeCoffGetUnixThunkStucture (); | |
97 | } | |
98 | DEBUG ((EFI_D_ERROR, "Loading %s 0x%08lx - entry point 0x%08lx\n",\r | |
99 | ImageContext->PdbPointer,\r | |
100 | (UINTN)ImageContext->ImageAddress,\r | |
101 | (UINTN)ImageContext->EntryPoint));\r | |
102 | \r | |
103 | Handle = mUnix->Dlopen (ImageContext->PdbPointer, RTLD_NOW);\r | |
104 | \r | |
105 | if (Handle) {\r | |
106 | Entry = mUnix->Dlsym(Handle, "_ModuleEntryPoint");\r | |
107 | } else { | |
108 | DEBUG ((EFI_D_ERROR, "%s\n", mUnix->Dlerror()));\r | |
109 | }\r | |
110 | \r | |
111 | if (Entry != NULL) {\r | |
112 | ImageContext->EntryPoint = Entry; | |
113 | DEBUG ((EFI_D_ERROR, "Change %s Entrypoint to :0x%08lx\n", ImageContext->PdbPointer, Entry));\r | |
114 | }\r | |
115 | \r | |
116 | \r | |
117 | return; | |
118 | } | |
119 | ||
120 | /** | |
6cb6f078 | 121 | Performs additional actions just before a PE/COFF image is unloaded. Any resources\r |
122 | that were allocated by PeCoffLoaderRelocateImageExtraAction() must be freed.\r | |
123 | \r | |
124 | If ImageContext is NULL, then ASSERT().\r | |
125 | \r | |
126 | @param ImageContext Pointer to the image context structure that describes the\r | |
127 | PE/COFF image that is being unloaded.\r | |
398b646f | 128 | |
129 | **/ | |
130 | VOID | |
131 | EFIAPI | |
132 | PeCoffLoaderUnloadImageExtraAction ( | |
133 | IN OUT PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext | |
134 | ) | |
135 | { | |
6cb6f078 | 136 | ASSERT (ImageContext != NULL);\r |
1664e0af | 137 | } |