| | 1 | /*++\r |
| | 2 | \r |
| | 3 | Copyright (c) 2006, Intel Corporation\r |
| | 4 | All rights reserved. This program and the accompanying materials\r |
| | 5 | are licensed and made available under the terms and conditions of the BSD License\r |
| | 6 | which accompanies this distribution. The full text of the license may be found at\r |
| | 7 | http://opensource.org/licenses/bsd-license.php\r |
| | 8 | \r |
| | 9 | THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r |
| | 10 | WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r |
| | 11 | \r |
| | 12 | Module Name:\r |
| | 13 | \r |
| | 14 | SecMain.c\r |
| | 15 | \r |
| | 16 | Abstract:\r |
| | 17 | WinNt emulator of SEC phase. It's really a Win32 application, but this is\r |
| | 18 | Ok since all the other modules for NT32 are NOT Win32 applications.\r |
| | 19 | \r |
| | 20 | This program processes Windows environment variables and figures out\r |
| | 21 | what the memory layout will be, how may FD's will be loaded and also\r |
| | 22 | what the boot mode is.\r |
| | 23 | \r |
| | 24 | The SEC registers a set of services with the SEC core. gPrivateDispatchTable\r |
| | 25 | is a list of PPI's produced by the SEC that are availble for usage in PEI.\r |
| | 26 | \r |
| | 27 | This code produces 128 K of temporary memory for the PEI stack by opening a\r |
| | 28 | Windows file and mapping it directly to memory addresses.\r |
| | 29 | \r |
| | 30 | The system.cmd script is used to set windows environment variables that drive\r |
| | 31 | the configuration opitons of the SEC.\r |
| | 32 | \r |
| | 33 | --*/\r |
| | 34 | \r |
| | 35 | #include "SecMain.h"\r |
| | 36 | \r |
| | 37 | \r |
| | 38 | //\r |
| | 39 | // Globals\r |
| | 40 | //\r |
| | 41 | EFI_PEI_PE_COFF_LOADER_PROTOCOL_INSTANCE mPeiEfiPeiPeCoffLoaderInstance = {\r |
| | 42 | {\r |
| | 43 | SecNt32PeCoffGetImageInfo,\r |
| | 44 | SecNt32PeCoffLoadImage,\r |
| | 45 | SecNt32PeCoffRelocateImage,\r |
| | 46 | SecNt32PeCoffUnloadimage\r |
| | 47 | },\r |
| | 48 | NULL\r |
| | 49 | };\r |
| | 50 | \r |
| | 51 | \r |
| | 52 | \r |
| | 53 | EFI_PEI_PE_COFF_LOADER_PROTOCOL *gPeiEfiPeiPeCoffLoader = &mPeiEfiPeiPeCoffLoaderInstance.PeCoff;\r |
| | 54 | \r |
| | 55 | NT_PEI_LOAD_FILE_PPI mSecNtLoadFilePpi = { SecWinNtPeiLoadFile };\r |
| | 56 | \r |
| | 57 | PEI_NT_AUTOSCAN_PPI mSecNtAutoScanPpi = { SecWinNtPeiAutoScan };\r |
| | 58 | \r |
| | 59 | PEI_NT_THUNK_PPI mSecWinNtThunkPpi = { SecWinNtWinNtThunkAddress };\r |
| | 60 | \r |
| | 61 | EFI_PEI_PROGRESS_CODE_PPI mSecStatusCodePpi = { SecPeiReportStatusCode };\r |
| | 62 | \r |
| | 63 | NT_FWH_PPI mSecFwhInformationPpi = { SecWinNtFdAddress };\r |
| | 64 | \r |
| | 65 | \r |
| | 66 | EFI_PEI_PPI_DESCRIPTOR gPrivateDispatchTable[] = {\r |
| | 67 | {\r |
| | 68 | EFI_PEI_PPI_DESCRIPTOR_PPI,\r |
| | 69 | &gEfiPeiPeCoffLoaderGuid,\r |
| | 70 | NULL\r |
| | 71 | },\r |
| | 72 | {\r |
| | 73 | EFI_PEI_PPI_DESCRIPTOR_PPI,\r |
| | 74 | &gNtPeiLoadFilePpiGuid,\r |
| | 75 | &mSecNtLoadFilePpi\r |
| | 76 | },\r |
| | 77 | {\r |
| | 78 | EFI_PEI_PPI_DESCRIPTOR_PPI,\r |
| | 79 | &gPeiNtAutoScanPpiGuid,\r |
| | 80 | &mSecNtAutoScanPpi\r |
| | 81 | },\r |
| | 82 | {\r |
| | 83 | EFI_PEI_PPI_DESCRIPTOR_PPI,\r |
| | 84 | &gPeiNtThunkPpiGuid,\r |
| | 85 | &mSecWinNtThunkPpi\r |
| | 86 | },\r |
| | 87 | {\r |
| | 88 | EFI_PEI_PPI_DESCRIPTOR_PPI,\r |
| | 89 | &gEfiPeiStatusCodePpiGuid,\r |
| | 90 | &mSecStatusCodePpi\r |
| | 91 | },\r |
| | 92 | {\r |
| | 93 | EFI_PEI_PPI_DESCRIPTOR_PPI | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST,\r |
| | 94 | &gNtFwhPpiGuid,\r |
| | 95 | &mSecFwhInformationPpi\r |
| | 96 | }\r |
| | 97 | };\r |
| | 98 | \r |
| | 99 | \r |
| | 100 | //\r |
| | 101 | // Default information about where the FD is located.\r |
| | 102 | // This array gets filled in with information from EFI_FIRMWARE_VOLUMES\r |
| | 103 | // EFI_FIRMWARE_VOLUMES is a Windows environment variable set by system.cmd.\r |
| | 104 | // The number of array elements is allocated base on parsing\r |
| | 105 | // EFI_FIRMWARE_VOLUMES and the memory is never freed.\r |
| | 106 | //\r |
| | 107 | UINTN gFdInfoCount = 0;\r |
| | 108 | NT_FD_INFO *gFdInfo;\r |
| | 109 | \r |
| | 110 | //\r |
| | 111 | // Array that supports seperate memory rantes.\r |
| | 112 | // The memory ranges are set in system.cmd via the EFI_MEMORY_SIZE variable.\r |
| | 113 | // The number of array elements is allocated base on parsing\r |
| | 114 | // EFI_MEMORY_SIZE and the memory is never freed.\r |
| | 115 | //\r |
| | 116 | UINTN gSystemMemoryCount = 0;\r |
| | 117 | NT_SYSTEM_MEMORY *gSystemMemory;\r |
| | 118 | \r |
| | 119 | \r |
| | 120 | UINTN mPdbNameModHandleArraySize = 0;\r |
| | 121 | PDB_NAME_TO_MOD_HANDLE *mPdbNameModHandleArray = NULL;\r |
| | 122 | \r |
| | 123 | \r |
| | 124 | \r |
| | 125 | \r |
| | 126 | INTN\r |
| | 127 | EFIAPI\r |
| | 128 | main (\r |
| | 129 | IN INTN Argc,\r |
| | 130 | IN CHAR8 **Argv,\r |
| | 131 | IN CHAR8 **Envp\r |
| | 132 | )\r |
| | 133 | /*++\r |
| | 134 | \r |
| | 135 | Routine Description:\r |
| | 136 | Main entry point to SEC for WinNt. This is a Windows program\r |
| | 137 | \r |
| | 138 | Arguments:\r |
| | 139 | Argc - Number of command line arguments\r |
| | 140 | Argv - Array of command line argument strings\r |
| | 141 | Envp - Array of environmemt variable strings\r |
| | 142 | \r |
| | 143 | Returns:\r |
| | 144 | 0 - Normal exit\r |
| | 145 | 1 - Abnormal exit\r |
| | 146 | \r |
| | 147 | --*/\r |
| | 148 | {\r |
| | 149 | EFI_STATUS Status;\r |
| | 150 | EFI_PHYSICAL_ADDRESS InitialStackMemory;\r |
| | 151 | UINT64 InitialStackMemorySize;\r |
| | 152 | UINTN Index;\r |
| | 153 | UINTN Index1;\r |
| | 154 | UINTN Index2;\r |
| | 155 | UINTN PeiIndex;\r |
| | 156 | CHAR16 *FileName;\r |
| | 157 | CHAR16 *FileNamePtr;\r |
| | 158 | BOOLEAN Done;\r |
| | 159 | VOID *PeiCoreFile;\r |
| | 160 | CHAR16 *MemorySizeStr;\r |
| | 161 | CHAR16 *FirmwareVolumesStr;\r |
| | 162 | \r |
| | 163 | MemorySizeStr = (CHAR16 *)L"64!64";\r |
| | 164 | FirmwareVolumesStr = (CHAR16 *)L"..\\Fv\\Fv_Recovery.fd";\r |
| | 165 | \r |
| | 166 | printf ("\nEDK SEC Main NT Emulation Environment from www.TianoCore.org\n");\r |
| | 167 | \r |
| | 168 | //\r |
| | 169 | // Make some Windows calls to Set the process to the highest priority in the\r |
| | 170 | // idle class. We need this to have good performance.\r |
| | 171 | //\r |
| | 172 | SetPriorityClass (GetCurrentProcess (), IDLE_PRIORITY_CLASS);\r |
| | 173 | SetThreadPriority (GetCurrentThread (), THREAD_PRIORITY_HIGHEST);\r |
| | 174 | \r |
| | 175 | //\r |
| | 176 | // Allocate space for gSystemMemory Array\r |
| | 177 | //\r |
| | 178 | gSystemMemoryCount = CountSeperatorsInString (MemorySizeStr, '!') + 1;\r |
| | 179 | gSystemMemory = calloc (gSystemMemoryCount, sizeof (NT_SYSTEM_MEMORY));\r |
| | 180 | if (gSystemMemory == NULL) {\r |
| | 181 | printf ("ERROR : Can not allocate memory for %s. Exiting.\n", MemorySizeStr);\r |
| | 182 | exit (1);\r |
| | 183 | }\r |
| | 184 | //\r |
| | 185 | // Allocate space for gSystemMemory Array\r |
| | 186 | //\r |
| | 187 | gFdInfoCount = CountSeperatorsInString (FirmwareVolumesStr, '!') + 1;\r |
| | 188 | gFdInfo = calloc (gFdInfoCount, sizeof (NT_FD_INFO));\r |
| | 189 | if (gFdInfo == NULL) {\r |
| | 190 | printf ("ERROR : Can not allocate memory for %s. Exiting.\n", FirmwareVolumesStr);\r |
| | 191 | exit (1);\r |
| | 192 | }\r |
| | 193 | //\r |
| | 194 | // Setup Boot Mode. If BootModeStr == "" then BootMode = 0 (BOOT_WITH_FULL_CONFIGURATION)\r |
| | 195 | //\r |
| | 196 | printf (" BootMode 0x%02x\n", FixedPcdGet32 (PcdWinNtBootMode));\r |
| | 197 | \r |
| | 198 | //\r |
| | 199 | // Open up a 128K file to emulate temp memory for PEI.\r |
| | 200 | // on a real platform this would be SRAM, or using the cache as RAM.\r |
| | 201 | // Set InitialStackMemory to zero so WinNtOpenFile will allocate a new mapping\r |
| | 202 | //\r |
| | 203 | InitialStackMemory = 0;\r |
| | 204 | InitialStackMemorySize = 0x20000;\r |
| | 205 | Status = WinNtOpenFile (\r |
| | 206 | L"SecStack",\r |
| | 207 | (UINT32) InitialStackMemorySize,\r |
| | 208 | OPEN_ALWAYS,\r |
| | 209 | &InitialStackMemory,\r |
| | 210 | &InitialStackMemorySize\r |
| | 211 | );\r |
| | 212 | if (EFI_ERROR (Status)) {\r |
| | 213 | printf ("ERROR : Can not open SecStack Exiting\n");\r |
| | 214 | exit (1);\r |
| | 215 | }\r |
| | 216 | \r |
| | 217 | printf (" SEC passing in %d bytes of temp RAM to PEI\n", InitialStackMemorySize);\r |
| | 218 | \r |
| | 219 | //\r |
| | 220 | // Open All the firmware volumes and remember the info in the gFdInfo global\r |
| | 221 | //\r |
| | 222 | FileNamePtr = (CHAR16 *)malloc (StrLen ((CHAR16 *)FirmwareVolumesStr) * sizeof(CHAR16));\r |
| | 223 | if (FileNamePtr == NULL) {\r |
| | 224 | printf ("ERROR : Can not allocate memory for firmware volume string\n");\r |
| | 225 | exit (1);\r |
| | 226 | }\r |
| | 227 | \r |
| | 228 | StrCpy (FileNamePtr, (CHAR16*)FirmwareVolumesStr);\r |
| | 229 | \r |
| | 230 | for (Done = FALSE, Index = 0, PeiIndex = 0, PeiCoreFile = NULL; !Done; Index++) {\r |
| | 231 | FileName = FileNamePtr;\r |
| | 232 | for (Index1 = 0; (FileNamePtr[Index1] != '!') && (FileNamePtr[Index1] != 0); Index1++)\r |
| | 233 | ;\r |
| | 234 | if (FileNamePtr[Index1] == 0) {\r |
| | 235 | Done = TRUE;\r |
| | 236 | } else {\r |
| | 237 | FileNamePtr[Index1] = '\0';\r |
| | 238 | FileNamePtr = FileNamePtr + Index1 + 1;\r |
| | 239 | }\r |
| | 240 | \r |
| | 241 | //\r |
| | 242 | // Open the FD and remmeber where it got mapped into our processes address space\r |
| | 243 | //\r |
| | 244 | Status = WinNtOpenFile (\r |
| | 245 | FileName,\r |
| | 246 | 0,\r |
| | 247 | OPEN_EXISTING,\r |
| | 248 | &gFdInfo[Index].Address,\r |
| | 249 | &gFdInfo[Index].Size\r |
| | 250 | );\r |
| | 251 | if (EFI_ERROR (Status)) {\r |
| | 252 | printf ("ERROR : Can not open Firmware Device File %S (%r). Exiting.\n", FileName, Status);\r |
| | 253 | exit (1);\r |
| | 254 | }\r |
| | 255 | \r |
| | 256 | printf (" FD loaded from");\r |
| | 257 | //\r |
| | 258 | // printf can't print filenames directly as the \ gets interperted as an\r |
| | 259 | // escape character.\r |
| | 260 | //\r |
| | 261 | for (Index2 = 0; FileName[Index2] != '\0'; Index2++) {\r |
| | 262 | printf ("%c", FileName[Index2]);\r |
| | 263 | }\r |
| | 264 | \r |
| | 265 | if (PeiCoreFile == NULL) {\r |
| | 266 | //\r |
| | 267 | // Assume the beginning of the FD is an FV and look for the PEI Core.\r |
| | 268 | // Load the first one we find.\r |
| | 269 | //\r |
| | 270 | Status = SecFfsFindPeiCore ((EFI_FIRMWARE_VOLUME_HEADER *) (UINTN) gFdInfo[Index].Address, &PeiCoreFile);\r |
| | 271 | if (!EFI_ERROR (Status)) {\r |
| | 272 | PeiIndex = Index;\r |
| | 273 | printf (" contains SEC Core");\r |
| | 274 | }\r |
| | 275 | }\r |
| | 276 | \r |
| | 277 | printf ("\n");\r |
| | 278 | }\r |
| | 279 | //\r |
| | 280 | // Calculate memory regions and store the information in the gSystemMemory\r |
| | 281 | // global for later use. The autosizing code will use this data to\r |
| | 282 | // map this memory into the SEC process memory space.\r |
| | 283 | //\r |
| | 284 | for (Index = 0, Done = FALSE; !Done; Index++) {\r |
| | 285 | //\r |
| | 286 | // Save the size of the memory and make a Unicode filename SystemMemory00, ...\r |
| | 287 | //\r |
| | 288 | gSystemMemory[Index].Size = _wtoi (MemorySizeStr) * 0x100000;\r |
| | 289 | _snwprintf (gSystemMemory[Index].FileName, NT_SYSTEM_MEMORY_FILENAME_SIZE, L"SystemMemory%02d", Index);\r |
| | 290 | \r |
| | 291 | //\r |
| | 292 | // Find the next region\r |
| | 293 | //\r |
| | 294 | for (Index1 = 0; MemorySizeStr[Index1] != '!' && MemorySizeStr[Index1] != 0; Index1++)\r |
| | 295 | ;\r |
| | 296 | if (MemorySizeStr[Index1] == 0) {\r |
| | 297 | Done = TRUE;\r |
| | 298 | }\r |
| | 299 | \r |
| | 300 | MemorySizeStr = MemorySizeStr + Index1 + 1;\r |
| | 301 | }\r |
| | 302 | \r |
| | 303 | printf ("\n");\r |
| | 304 | \r |
| | 305 | //\r |
| | 306 | // Hand off to PEI Core\r |
| | 307 | //\r |
| | 308 | SecLoadFromCore ((UINTN) InitialStackMemory, (UINTN) InitialStackMemorySize, (UINTN) gFdInfo[0].Address, PeiCoreFile);\r |
| | 309 | \r |
| | 310 | //\r |
| | 311 | // If we get here, then the PEI Core returned. This is an error as PEI should\r |
| | 312 | // always hand off to DXE.\r |
| | 313 | //\r |
| | 314 | printf ("ERROR : PEI Core returned\n");\r |
| | 315 | exit (1);\r |
| | 316 | }\r |
| | 317 | \r |
| | 318 | EFI_STATUS\r |
| | 319 | WinNtOpenFile (\r |
| | 320 | IN CHAR16 *FileName,\r |
| | 321 | IN UINT32 MapSize,\r |
| | 322 | IN DWORD CreationDisposition,\r |
| | 323 | IN OUT EFI_PHYSICAL_ADDRESS *BaseAddress,\r |
| | 324 | OUT UINT64 *Length\r |
| | 325 | )\r |
| | 326 | /*++\r |
| | 327 | \r |
| | 328 | Routine Description:\r |
| | 329 | Opens and memory maps a file using WinNt services. If BaseAddress is non zero\r |
| | 330 | the process will try and allocate the memory starting at BaseAddress.\r |
| | 331 | \r |
| | 332 | Arguments:\r |
| | 333 | FileName - The name of the file to open and map\r |
| | 334 | MapSize - The amount of the file to map in bytes\r |
| | 335 | CreationDisposition - The flags to pass to CreateFile(). Use to create new files for\r |
| | 336 | memory emulation, and exiting files for firmware volume emulation\r |
| | 337 | BaseAddress - The base address of the mapped file in the user address space.\r |
| | 338 | If passed in as NULL the a new memory region is used.\r |
| | 339 | If passed in as non NULL the request memory region is used for\r |
| | 340 | the mapping of the file into the process space.\r |
| | 341 | Length - The size of the mapped region in bytes\r |
| | 342 | \r |
| | 343 | Returns:\r |
| | 344 | EFI_SUCCESS - The file was opened and mapped.\r |
| | 345 | EFI_NOT_FOUND - FileName was not found in the current directory\r |
| | 346 | EFI_DEVICE_ERROR - An error occured attempting to map the opened file\r |
| | 347 | \r |
| | 348 | --*/\r |
| | 349 | {\r |
| | 350 | HANDLE NtFileHandle;\r |
| | 351 | HANDLE NtMapHandle;\r |
| | 352 | VOID *VirtualAddress;\r |
| | 353 | UINTN FileSize;\r |
| | 354 | \r |
| | 355 | //\r |
| | 356 | // Use Win API to open/create a file\r |
| | 357 | //\r |
| | 358 | NtFileHandle = CreateFile (\r |
| | 359 | FileName,\r |
| | 360 | GENERIC_READ | GENERIC_WRITE,\r |
| | 361 | FILE_SHARE_READ,\r |
| | 362 | NULL,\r |
| | 363 | CreationDisposition,\r |
| | 364 | FILE_ATTRIBUTE_NORMAL,\r |
| | 365 | NULL\r |
| | 366 | );\r |
| | 367 | if (NtFileHandle == INVALID_HANDLE_VALUE) {\r |
| | 368 | return EFI_NOT_FOUND;\r |
| | 369 | }\r |
| | 370 | //\r |
| | 371 | // Map the open file into a memory range\r |
| | 372 | //\r |
| | 373 | NtMapHandle = CreateFileMapping (\r |
| | 374 | NtFileHandle,\r |
| | 375 | NULL,\r |
| | 376 | PAGE_READWRITE,\r |
| | 377 | 0,\r |
| | 378 | MapSize,\r |
| | 379 | NULL\r |
| | 380 | );\r |
| | 381 | if (NtMapHandle == NULL) {\r |
| | 382 | return EFI_DEVICE_ERROR;\r |
| | 383 | }\r |
| | 384 | //\r |
| | 385 | // Get the virtual address (address in the emulator) of the mapped file\r |
| | 386 | //\r |
| | 387 | VirtualAddress = MapViewOfFileEx (\r |
| | 388 | NtMapHandle,\r |
| | 389 | FILE_MAP_ALL_ACCESS,\r |
| | 390 | 0,\r |
| | 391 | 0,\r |
| | 392 | MapSize,\r |
| | 393 | (LPVOID) (UINTN) *BaseAddress\r |
| | 394 | );\r |
| | 395 | if (VirtualAddress == NULL) {\r |
| | 396 | return EFI_DEVICE_ERROR;\r |
| | 397 | }\r |
| | 398 | \r |
| | 399 | if (MapSize == 0) {\r |
| | 400 | //\r |
| | 401 | // Seek to the end of the file to figure out the true file size.\r |
| | 402 | //\r |
| | 403 | FileSize = SetFilePointer (\r |
| | 404 | NtFileHandle,\r |
| | 405 | 0,\r |
| | 406 | NULL,\r |
| | 407 | FILE_END\r |
| | 408 | );\r |
| | 409 | if (FileSize == -1) {\r |
| | 410 | return EFI_DEVICE_ERROR;\r |
| | 411 | }\r |
| | 412 | \r |
| | 413 | *Length = (UINT64) FileSize;\r |
| | 414 | } else {\r |
| | 415 | *Length = (UINT64) MapSize;\r |
| | 416 | }\r |
| | 417 | \r |
| | 418 | *BaseAddress = (EFI_PHYSICAL_ADDRESS) (UINTN) VirtualAddress;\r |
| | 419 | \r |
| | 420 | return EFI_SUCCESS;\r |
| | 421 | }\r |
| | 422 | \r |
| | 423 | \r |
| | 424 | #define BYTES_PER_RECORD 512\r |
| | 425 | \r |
| | 426 | EFI_STATUS\r |
| | 427 | EFIAPI\r |
| | 428 | SecPeiReportStatusCode (\r |
| | 429 | IN CONST EFI_PEI_SERVICES **PeiServices,\r |
| | 430 | IN EFI_STATUS_CODE_TYPE CodeType,\r |
| | 431 | IN EFI_STATUS_CODE_VALUE Value,\r |
| | 432 | IN UINT32 Instance,\r |
| | 433 | IN CONST EFI_GUID *CallerId,\r |
| | 434 | IN CONST EFI_STATUS_CODE_DATA *Data OPTIONAL\r |
| | 435 | )\r |
| | 436 | /*++\r |
| | 437 | \r |
| | 438 | Routine Description:\r |
| | 439 | \r |
| | 440 | This routine produces the ReportStatusCode PEI service. It's passed\r |
| | 441 | up to the PEI Core via a PPI. T\r |
| | 442 | \r |
| | 443 | This code currently uses the NT clib printf. This does not work the same way\r |
| | 444 | as the EFI Print (), as %t, %g, %s as Unicode are not supported.\r |
| | 445 | \r |
| | 446 | Arguments:\r |
| | 447 | (see EFI_PEI_REPORT_STATUS_CODE)\r |
| | 448 | \r |
| | 449 | Returns:\r |
| | 450 | EFI_SUCCESS - Always return success\r |
| | 451 | \r |
| | 452 | --*/\r |
| | 453 | // TODO: PeiServices - add argument and description to function comment\r |
| | 454 | // TODO: CodeType - add argument and description to function comment\r |
| | 455 | // TODO: Value - add argument and description to function comment\r |
| | 456 | // TODO: Instance - add argument and description to function comment\r |
| | 457 | // TODO: CallerId - add argument and description to function comment\r |
| | 458 | // TODO: Data - add argument and description to function comment\r |
| | 459 | {\r |
| | 460 | CHAR8 *Format;\r |
| | 461 | VA_LIST Marker;\r |
| | 462 | CHAR8 PrintBuffer[BYTES_PER_RECORD * 2];\r |
| | 463 | CHAR8 *Filename;\r |
| | 464 | CHAR8 *Description;\r |
| | 465 | UINT32 LineNumber;\r |
| | 466 | UINT32 ErrorLevel;\r |
| | 467 | \r |
| | 468 | \r |
| | 469 | if (Data == NULL) {\r |
| | 470 | } else if (ReportStatusCodeExtractAssertInfo (CodeType, Value, Data, &Filename, &Description, &LineNumber)) {\r |
| | 471 | //\r |
| | 472 | // Processes ASSERT ()\r |
| | 473 | //\r |
| | 474 | printf ("ASSERT %s(%d): %s\n", Filename, LineNumber, Description);\r |
| | 475 | \r |
| | 476 | } else if (ReportStatusCodeExtractDebugInfo (Data, &ErrorLevel, &Marker, &Format)) {\r |
| | 477 | //\r |
| | 478 | // Process DEBUG () macro \r |
| | 479 | //\r |
| | 480 | AsciiVSPrint (PrintBuffer, BYTES_PER_RECORD, Format, Marker);\r |
| | 481 | printf (PrintBuffer);\r |
| | 482 | }\r |
| | 483 | \r |
| | 484 | return EFI_SUCCESS;\r |
| | 485 | }\r |
| | 486 | \r |
| | 487 | /**\r |
| | 488 | Transfers control to a function starting with a new stack.\r |
| | 489 | \r |
| | 490 | Transfers control to the function specified by EntryPoint using the new stack\r |
| | 491 | specified by NewStack and passing in the parameters specified by Context1 and\r |
| | 492 | Context2. Context1 and Context2 are optional and may be NULL. The function\r |
| | 493 | EntryPoint must never return.\r |
| | 494 | \r |
| | 495 | If EntryPoint is NULL, then ASSERT().\r |
| | 496 | If NewStack is NULL, then ASSERT().\r |
| | 497 | \r |
| | 498 | @param EntryPoint A pointer to function to call with the new stack.\r |
| | 499 | @param Context1 A pointer to the context to pass into the EntryPoint\r |
| | 500 | function.\r |
| | 501 | @param Context2 A pointer to the context to pass into the EntryPoint\r |
| | 502 | function.\r |
| | 503 | @param NewStack A pointer to the new stack to use for the EntryPoint\r |
| | 504 | function.\r |
| | 505 | @param NewBsp A pointer to the new BSP for the EntryPoint on IPF. It's\r |
| | 506 | Reserved on other architectures.\r |
| | 507 | \r |
| | 508 | **/\r |
| | 509 | VOID\r |
| | 510 | EFIAPI\r |
| | 511 | PeiSwitchStacks (\r |
| | 512 | IN SWITCH_STACK_ENTRY_POINT EntryPoint,\r |
| | 513 | IN VOID *Context1, OPTIONAL\r |
| | 514 | IN VOID *Context2, OPTIONAL\r |
| | 515 | IN VOID *Context3, OPTIONAL\r |
| | 516 | IN VOID *NewStack\r |
| | 517 | )\r |
| | 518 | {\r |
| | 519 | BASE_LIBRARY_JUMP_BUFFER JumpBuffer;\r |
| | 520 | \r |
| | 521 | ASSERT (EntryPoint != NULL);\r |
| | 522 | ASSERT (NewStack != NULL);\r |
| | 523 | \r |
| | 524 | //\r |
| | 525 | // Stack should be aligned with CPU_STACK_ALIGNMENT\r |
| | 526 | //\r |
| | 527 | ASSERT (((UINTN)NewStack & (CPU_STACK_ALIGNMENT - 1)) == 0);\r |
| | 528 | \r |
| | 529 | JumpBuffer.Eip = (UINTN)EntryPoint;\r |
| | 530 | JumpBuffer.Esp = (UINTN)NewStack - sizeof (VOID*);\r |
| | 531 | JumpBuffer.Esp -= sizeof (Context1) + sizeof (Context2) + sizeof(Context3);\r |
| | 532 | ((VOID**)JumpBuffer.Esp)[1] = Context1;\r |
| | 533 | ((VOID**)JumpBuffer.Esp)[2] = Context2;\r |
| | 534 | ((VOID**)JumpBuffer.Esp)[3] = Context3;\r |
| | 535 | \r |
| | 536 | LongJump (&JumpBuffer, (UINTN)-1);\r |
| | 537 | \r |
| | 538 | \r |
| | 539 | //\r |
| | 540 | // InternalSwitchStack () will never return\r |
| | 541 | //\r |
| | 542 | ASSERT (FALSE); \r |
| | 543 | }\r |
| | 544 | \r |
| | 545 | VOID\r |
| | 546 | SecLoadFromCore (\r |
| | 547 | IN UINTN LargestRegion,\r |
| | 548 | IN UINTN LargestRegionSize,\r |
| | 549 | IN UINTN BootFirmwareVolumeBase,\r |
| | 550 | IN VOID *PeiCorePe32File\r |
| | 551 | )\r |
| | 552 | /*++\r |
| | 553 | \r |
| | 554 | Routine Description:\r |
| | 555 | This is the service to load the PEI Core from the Firmware Volume\r |
| | 556 | \r |
| | 557 | Arguments:\r |
| | 558 | LargestRegion - Memory to use for PEI.\r |
| | 559 | LargestRegionSize - Size of Memory to use for PEI\r |
| | 560 | BootFirmwareVolumeBase - Start of the Boot FV\r |
| | 561 | PeiCorePe32File - PEI Core PE32\r |
| | 562 | \r |
| | 563 | Returns:\r |
| | 564 | Success means control is transfered and thus we should never return\r |
| | 565 | \r |
| | 566 | --*/\r |
| | 567 | {\r |
| | 568 | EFI_STATUS Status;\r |
| | 569 | EFI_PHYSICAL_ADDRESS TopOfMemory;\r |
| | 570 | VOID *TopOfStack;\r |
| | 571 | UINT64 PeiCoreSize;\r |
| | 572 | EFI_PHYSICAL_ADDRESS PeiCoreEntryPoint;\r |
| | 573 | EFI_PHYSICAL_ADDRESS PeiImageAddress;\r |
| | 574 | EFI_SEC_PEI_HAND_OFF *SecCoreData;\r |
| | 575 | \r |
| | 576 | //\r |
| | 577 | // Compute Top Of Memory for Stack and PEI Core Allocations\r |
| | 578 | //\r |
| | 579 | TopOfMemory = LargestRegion + LargestRegionSize;\r |
| | 580 | \r |
| | 581 | //\r |
| | 582 | // Allocate 128KB for the Stack\r |
| | 583 | //\r |
| | 584 | TopOfStack = (VOID *)((UINTN)TopOfMemory - sizeof (EFI_SEC_PEI_HAND_OFF) - CPU_STACK_ALIGNMENT);\r |
| | 585 | TopOfStack = ALIGN_POINTER (TopOfStack, CPU_STACK_ALIGNMENT);\r |
| | 586 | TopOfMemory = TopOfMemory - STACK_SIZE;\r |
| | 587 | \r |
| | 588 | //\r |
| | 589 | // Patch value in dispatch table values\r |
| | 590 | //\r |
| | 591 | gPrivateDispatchTable[0].Ppi = gPeiEfiPeiPeCoffLoader;\r |
| | 592 | \r |
| | 593 | //\r |
| | 594 | // Bind this information into the SEC hand-off state\r |
| | 595 | //\r |
| | 596 | SecCoreData = (EFI_SEC_PEI_HAND_OFF*)(UINTN) TopOfStack;\r |
| | 597 | SecCoreData->DataSize = sizeof(EFI_SEC_PEI_HAND_OFF);\r |
| | 598 | SecCoreData->BootFirmwareVolumeBase = (VOID*)BootFirmwareVolumeBase;\r |
| | 599 | SecCoreData->BootFirmwareVolumeSize = FixedPcdGet32(PcdWinNtFirmwareFdSize);\r |
| | 600 | SecCoreData->TemporaryRamBase = (VOID*)(UINTN)TopOfMemory; \r |
| | 601 | SecCoreData->TemporaryRamSize = STACK_SIZE;\r |
| | 602 | SecCoreData->PeiTemporaryRamBase = SecCoreData->TemporaryRamBase;\r |
| | 603 | SecCoreData->PeiTemporaryRamSize = (UINTN)RShiftU64((UINT64)STACK_SIZE,1);\r |
| | 604 | SecCoreData->StackBase = (VOID*)((UINTN)SecCoreData->TemporaryRamBase + (UINTN)SecCoreData->TemporaryRamSize);\r |
| | 605 | SecCoreData->StackSize = (UINTN)RShiftU64((UINT64)STACK_SIZE,1);\r |
| | 606 | \r |
| | 607 | //\r |
| | 608 | // Load the PEI Core from a Firmware Volume\r |
| | 609 | //\r |
| | 610 | Status = SecWinNtPeiLoadFile (\r |
| | 611 | PeiCorePe32File,\r |
| | 612 | &PeiImageAddress,\r |
| | 613 | &PeiCoreSize,\r |
| | 614 | &PeiCoreEntryPoint\r |
| | 615 | );\r |
| | 616 | if (EFI_ERROR (Status)) {\r |
| | 617 | return ;\r |
| | 618 | }\r |
| | 619 | \r |
| | 620 | //\r |
| | 621 | // Transfer control to the PEI Core\r |
| | 622 | //\r |
| | 623 | PeiSwitchStacks (\r |
| | 624 | (SWITCH_STACK_ENTRY_POINT) (UINTN) PeiCoreEntryPoint,\r |
| | 625 | SecCoreData,\r |
| | 626 | (VOID *) (UINTN) ((EFI_PEI_PPI_DESCRIPTOR *) &gPrivateDispatchTable),\r |
| | 627 | NULL,\r |
| | 628 | TopOfStack\r |
| | 629 | );\r |
| | 630 | //\r |
| | 631 | // If we get here, then the PEI Core returned. This is an error\r |
| | 632 | //\r |
| | 633 | return ;\r |
| | 634 | }\r |
| | 635 | \r |
| | 636 | EFI_STATUS\r |
| | 637 | EFIAPI\r |
| | 638 | SecWinNtPeiAutoScan (\r |
| | 639 | IN UINTN Index,\r |
| | 640 | OUT EFI_PHYSICAL_ADDRESS *MemoryBase,\r |
| | 641 | OUT UINT64 *MemorySize\r |
| | 642 | )\r |
| | 643 | /*++\r |
| | 644 | \r |
| | 645 | Routine Description:\r |
| | 646 | This service is called from Index == 0 until it returns EFI_UNSUPPORTED.\r |
| | 647 | It allows discontiguous memory regions to be supported by the emulator.\r |
| | 648 | It uses gSystemMemory[] and gSystemMemoryCount that were created by\r |
| | 649 | parsing the Windows environment variable EFI_MEMORY_SIZE.\r |
| | 650 | The size comes from the varaible and the address comes from the call to\r |
| | 651 | WinNtOpenFile.\r |
| | 652 | \r |
| | 653 | Arguments:\r |
| | 654 | Index - Which memory region to use\r |
| | 655 | MemoryBase - Return Base address of memory region\r |
| | 656 | MemorySize - Return size in bytes of the memory region\r |
| | 657 | \r |
| | 658 | Returns:\r |
| | 659 | EFI_SUCCESS - If memory region was mapped\r |
| | 660 | EFI_UNSUPPORTED - If Index is not supported\r |
| | 661 | \r |
| | 662 | --*/\r |
| | 663 | {\r |
| | 664 | EFI_STATUS Status;\r |
| | 665 | \r |
| | 666 | if (Index >= gSystemMemoryCount) {\r |
| | 667 | return EFI_UNSUPPORTED;\r |
| | 668 | }\r |
| | 669 | \r |
| | 670 | *MemoryBase = 0;\r |
| | 671 | Status = WinNtOpenFile (\r |
| | 672 | gSystemMemory[Index].FileName,\r |
| | 673 | (UINT32) gSystemMemory[Index].Size,\r |
| | 674 | OPEN_ALWAYS,\r |
| | 675 | MemoryBase,\r |
| | 676 | MemorySize\r |
| | 677 | );\r |
| | 678 | \r |
| | 679 | gSystemMemory[Index].Memory = *MemoryBase;\r |
| | 680 | \r |
| | 681 | return Status;\r |
| | 682 | }\r |
| | 683 | \r |
| | 684 | VOID *\r |
| | 685 | EFIAPI\r |
| | 686 | SecWinNtWinNtThunkAddress (\r |
| | 687 | VOID\r |
| | 688 | )\r |
| | 689 | /*++\r |
| | 690 | \r |
| | 691 | Routine Description:\r |
| | 692 | Since the SEC is the only Windows program in stack it must export\r |
| | 693 | an interface to do Win API calls. That's what the WinNtThunk address\r |
| | 694 | is for. gWinNt is initailized in WinNtThunk.c.\r |
| | 695 | \r |
| | 696 | Arguments:\r |
| | 697 | InterfaceSize - sizeof (EFI_WIN_NT_THUNK_PROTOCOL);\r |
| | 698 | InterfaceBase - Address of the gWinNt global\r |
| | 699 | \r |
| | 700 | Returns:\r |
| | 701 | EFI_SUCCESS - Data returned\r |
| | 702 | \r |
| | 703 | --*/\r |
| | 704 | {\r |
| | 705 | return gWinNt;\r |
| | 706 | }\r |
| | 707 | \r |
| | 708 | \r |
| | 709 | EFI_STATUS\r |
| | 710 | EFIAPI\r |
| | 711 | SecWinNtPeiLoadFile (\r |
| | 712 | IN VOID *Pe32Data,\r |
| | 713 | IN EFI_PHYSICAL_ADDRESS *ImageAddress,\r |
| | 714 | IN UINT64 *ImageSize,\r |
| | 715 | IN EFI_PHYSICAL_ADDRESS *EntryPoint\r |
| | 716 | )\r |
| | 717 | /*++\r |
| | 718 | \r |
| | 719 | Routine Description:\r |
| | 720 | Loads and relocates a PE/COFF image into memory.\r |
| | 721 | \r |
| | 722 | Arguments:\r |
| | 723 | Pe32Data - The base address of the PE/COFF file that is to be loaded and relocated\r |
| | 724 | ImageAddress - The base address of the relocated PE/COFF image\r |
| | 725 | ImageSize - The size of the relocated PE/COFF image\r |
| | 726 | EntryPoint - The entry point of the relocated PE/COFF image\r |
| | 727 | \r |
| | 728 | Returns:\r |
| | 729 | EFI_SUCCESS - The file was loaded and relocated\r |
| | 730 | EFI_OUT_OF_RESOURCES - There was not enough memory to load and relocate the PE/COFF file\r |
| | 731 | \r |
| | 732 | --*/\r |
| | 733 | {\r |
| | 734 | EFI_STATUS Status;\r |
| | 735 | PE_COFF_LOADER_IMAGE_CONTEXT ImageContext;\r |
| | 736 | \r |
| | 737 | ZeroMem (&ImageContext, sizeof (ImageContext));\r |
| | 738 | ImageContext.Handle = Pe32Data;\r |
| | 739 | \r |
| | 740 | ImageContext.ImageRead = (PE_COFF_LOADER_READ_FILE) SecImageRead;\r |
| | 741 | \r |
| | 742 | Status = gPeiEfiPeiPeCoffLoader->GetImageInfo (gPeiEfiPeiPeCoffLoader, &ImageContext);\r |
| | 743 | if (EFI_ERROR (Status)) {\r |
| | 744 | return Status;\r |
| | 745 | }\r |
| | 746 | //\r |
| | 747 | // Allocate space in NT (not emulator) memory. Extra space is for alignment\r |
| | 748 | //\r |
| | 749 | ImageContext.ImageAddress = (EFI_PHYSICAL_ADDRESS) (UINTN) malloc ((UINTN) (ImageContext.ImageSize + (ImageContext.SectionAlignment * 2)));\r |
| | 750 | if (ImageContext.ImageAddress == 0) {\r |
| | 751 | return EFI_OUT_OF_RESOURCES;\r |
| | 752 | }\r |
| | 753 | //\r |
| | 754 | // Align buffer on section boundry\r |
| | 755 | //\r |
| | 756 | ImageContext.ImageAddress += ImageContext.SectionAlignment;\r |
| | 757 | ImageContext.ImageAddress &= ~(ImageContext.SectionAlignment - 1);\r |
| | 758 | \r |
| | 759 | Status = gPeiEfiPeiPeCoffLoader->LoadImage (gPeiEfiPeiPeCoffLoader, &ImageContext);\r |
| | 760 | if (EFI_ERROR (Status)) {\r |
| | 761 | return Status;\r |
| | 762 | }\r |
| | 763 | \r |
| | 764 | Status = gPeiEfiPeiPeCoffLoader->RelocateImage (gPeiEfiPeiPeCoffLoader, &ImageContext);\r |
| | 765 | if (EFI_ERROR (Status)) {\r |
| | 766 | return Status;\r |
| | 767 | }\r |
| | 768 | \r |
| | 769 | //\r |
| | 770 | // BugBug: Flush Instruction Cache Here when CPU Lib is ready\r |
| | 771 | //\r |
| | 772 | \r |
| | 773 | *ImageAddress = ImageContext.ImageAddress;\r |
| | 774 | *ImageSize = ImageContext.ImageSize;\r |
| | 775 | *EntryPoint = ImageContext.EntryPoint;\r |
| | 776 | \r |
| | 777 | return EFI_SUCCESS;\r |
| | 778 | }\r |
| | 779 | \r |
| | 780 | EFI_STATUS\r |
| | 781 | EFIAPI\r |
| | 782 | SecWinNtFdAddress (\r |
| | 783 | IN UINTN Index,\r |
| | 784 | IN OUT EFI_PHYSICAL_ADDRESS *FdBase,\r |
| | 785 | IN OUT UINT64 *FdSize\r |
| | 786 | )\r |
| | 787 | /*++\r |
| | 788 | \r |
| | 789 | Routine Description:\r |
| | 790 | Return the FD Size and base address. Since the FD is loaded from a\r |
| | 791 | file into Windows memory only the SEC will know it's address.\r |
| | 792 | \r |
| | 793 | Arguments:\r |
| | 794 | Index - Which FD, starts at zero.\r |
| | 795 | FdSize - Size of the FD in bytes\r |
| | 796 | FdBase - Start address of the FD. Assume it points to an FV Header\r |
| | 797 | \r |
| | 798 | Returns:\r |
| | 799 | EFI_SUCCESS - Return the Base address and size of the FV\r |
| | 800 | EFI_UNSUPPORTED - Index does nto map to an FD in the system\r |
| | 801 | \r |
| | 802 | --*/\r |
| | 803 | {\r |
| | 804 | if (Index >= gFdInfoCount) {\r |
| | 805 | return EFI_UNSUPPORTED;\r |
| | 806 | }\r |
| | 807 | \r |
| | 808 | *FdBase = gFdInfo[Index].Address;\r |
| | 809 | *FdSize = gFdInfo[Index].Size;\r |
| | 810 | \r |
| | 811 | if (*FdBase == 0 && *FdSize == 0) {\r |
| | 812 | return EFI_UNSUPPORTED;\r |
| | 813 | }\r |
| | 814 | \r |
| | 815 | return EFI_SUCCESS;\r |
| | 816 | }\r |
| | 817 | \r |
| | 818 | EFI_STATUS\r |
| | 819 | EFIAPI\r |
| | 820 | SecImageRead (\r |
| | 821 | IN VOID *FileHandle,\r |
| | 822 | IN UINTN FileOffset,\r |
| | 823 | IN OUT UINTN *ReadSize,\r |
| | 824 | OUT VOID *Buffer\r |
| | 825 | )\r |
| | 826 | /*++\r |
| | 827 | \r |
| | 828 | Routine Description:\r |
| | 829 | Support routine for the PE/COFF Loader that reads a buffer from a PE/COFF file\r |
| | 830 | \r |
| | 831 | Arguments:\r |
| | 832 | FileHandle - The handle to the PE/COFF file\r |
| | 833 | FileOffset - The offset, in bytes, into the file to read\r |
| | 834 | ReadSize - The number of bytes to read from the file starting at FileOffset\r |
| | 835 | Buffer - A pointer to the buffer to read the data into.\r |
| | 836 | \r |
| | 837 | Returns:\r |
| | 838 | EFI_SUCCESS - ReadSize bytes of data were read into Buffer from the PE/COFF file starting at FileOffset\r |
| | 839 | \r |
| | 840 | --*/\r |
| | 841 | {\r |
| | 842 | CHAR8 *Destination8;\r |
| | 843 | CHAR8 *Source8;\r |
| | 844 | UINTN Length;\r |
| | 845 | \r |
| | 846 | Destination8 = Buffer;\r |
| | 847 | Source8 = (CHAR8 *) ((UINTN) FileHandle + FileOffset);\r |
| | 848 | Length = *ReadSize;\r |
| | 849 | while (Length--) {\r |
| | 850 | *(Destination8++) = *(Source8++);\r |
| | 851 | }\r |
| | 852 | \r |
| | 853 | return EFI_SUCCESS;\r |
| | 854 | }\r |
| | 855 | \r |
| | 856 | CHAR16 *\r |
| | 857 | AsciiToUnicode (\r |
| | 858 | IN CHAR8 *Ascii,\r |
| | 859 | IN UINTN *StrLen OPTIONAL\r |
| | 860 | )\r |
| | 861 | /*++\r |
| | 862 | \r |
| | 863 | Routine Description:\r |
| | 864 | Convert the passed in Ascii string to Unicode.\r |
| | 865 | Optionally return the length of the strings.\r |
| | 866 | \r |
| | 867 | Arguments:\r |
| | 868 | Ascii - Ascii string to convert\r |
| | 869 | StrLen - Length of string\r |
| | 870 | \r |
| | 871 | Returns:\r |
| | 872 | Pointer to malloc'ed Unicode version of Ascii\r |
| | 873 | \r |
| | 874 | --*/\r |
| | 875 | {\r |
| | 876 | UINTN Index;\r |
| | 877 | CHAR16 *Unicode;\r |
| | 878 | \r |
| | 879 | //\r |
| | 880 | // Allocate a buffer for unicode string\r |
| | 881 | //\r |
| | 882 | for (Index = 0; Ascii[Index] != '\0'; Index++)\r |
| | 883 | ;\r |
| | 884 | Unicode = malloc ((Index + 1) * sizeof (CHAR16));\r |
| | 885 | if (Unicode == NULL) {\r |
| | 886 | return NULL;\r |
| | 887 | }\r |
| | 888 | \r |
| | 889 | for (Index = 0; Ascii[Index] != '\0'; Index++) {\r |
| | 890 | Unicode[Index] = (CHAR16) Ascii[Index];\r |
| | 891 | }\r |
| | 892 | \r |
| | 893 | Unicode[Index] = '\0';\r |
| | 894 | \r |
| | 895 | if (StrLen != NULL) {\r |
| | 896 | *StrLen = Index;\r |
| | 897 | }\r |
| | 898 | \r |
| | 899 | return Unicode;\r |
| | 900 | }\r |
| | 901 | \r |
| | 902 | UINTN\r |
| | 903 | CountSeperatorsInString (\r |
| | 904 | IN const CHAR16 *String,\r |
| | 905 | IN CHAR16 Seperator\r |
| | 906 | )\r |
| | 907 | /*++\r |
| | 908 | \r |
| | 909 | Routine Description:\r |
| | 910 | Count the number of seperators in String\r |
| | 911 | \r |
| | 912 | Arguments:\r |
| | 913 | String - String to process\r |
| | 914 | Seperator - Item to count\r |
| | 915 | \r |
| | 916 | Returns:\r |
| | 917 | Number of Seperator in String\r |
| | 918 | \r |
| | 919 | --*/\r |
| | 920 | {\r |
| | 921 | UINTN Count;\r |
| | 922 | \r |
| | 923 | for (Count = 0; *String != '\0'; String++) {\r |
| | 924 | if (*String == Seperator) {\r |
| | 925 | Count++;\r |
| | 926 | }\r |
| | 927 | }\r |
| | 928 | \r |
| | 929 | return Count;\r |
| | 930 | }\r |
| | 931 | \r |
| | 932 | \r |
| | 933 | EFI_STATUS\r |
| | 934 | AddModHandle (\r |
| | 935 | IN PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext,\r |
| | 936 | IN VOID *ModHandle\r |
| | 937 | )\r |
| | 938 | /*++\r |
| | 939 | \r |
| | 940 | Routine Description:\r |
| | 941 | Store the ModHandle in an array indexed by the Pdb File name.\r |
| | 942 | The ModHandle is needed to unload the image. \r |
| | 943 | \r |
| | 944 | Arguments:\r |
| | 945 | ImageContext - Input data returned from PE Laoder Library. Used to find the \r |
| | 946 | .PDB file name of the PE Image.\r |
| | 947 | ModHandle - Returned from LoadLibraryEx() and stored for call to \r |
| | 948 | FreeLibrary().\r |
| | 949 | \r |
| | 950 | Returns:\r |
| | 951 | EFI_SUCCESS - ModHandle was stored. \r |
| | 952 | \r |
| | 953 | --*/\r |
| | 954 | {\r |
| | 955 | UINTN Index;\r |
| | 956 | PDB_NAME_TO_MOD_HANDLE *Array;\r |
| | 957 | UINTN PreviousSize;\r |
| | 958 | \r |
| | 959 | \r |
| | 960 | Array = mPdbNameModHandleArray;\r |
| | 961 | for (Index = 0; Index < mPdbNameModHandleArraySize; Index++, Array++) {\r |
| | 962 | if (Array->PdbPointer == NULL) {\r |
| | 963 | //\r |
| | 964 | // Make a copy of the stirng and store the ModHandle\r |
| | 965 | //\r |
| | 966 | Array->PdbPointer = malloc (strlen (ImageContext->PdbPointer) + 1);\r |
| | 967 | ASSERT (Array->PdbPointer != NULL);\r |
| | 968 | \r |
| | 969 | strcpy (Array->PdbPointer, ImageContext->PdbPointer);\r |
| | 970 | Array->ModHandle = ModHandle;\r |
| | 971 | return EFI_SUCCESS;\r |
| | 972 | }\r |
| | 973 | }\r |
| | 974 | \r |
| | 975 | //\r |
| | 976 | // No free space in mPdbNameModHandleArray so grow it by \r |
| | 977 | // MAX_PDB_NAME_TO_MOD_HANDLE_ARRAY_SIZE entires. realloc will\r |
| | 978 | // copy the old values to the new locaiton. But it does\r |
| | 979 | // not zero the new memory area.\r |
| | 980 | //\r |
| | 981 | PreviousSize = mPdbNameModHandleArraySize * sizeof (PDB_NAME_TO_MOD_HANDLE);\r |
| | 982 | mPdbNameModHandleArraySize += MAX_PDB_NAME_TO_MOD_HANDLE_ARRAY_SIZE;\r |
| | 983 | \r |
| | 984 | mPdbNameModHandleArray = realloc (mPdbNameModHandleArray, mPdbNameModHandleArraySize * sizeof (PDB_NAME_TO_MOD_HANDLE));\r |
| | 985 | if (mPdbNameModHandleArray == NULL) {\r |
| | 986 | ASSERT (FALSE);\r |
| | 987 | return EFI_OUT_OF_RESOURCES;\r |
| | 988 | }\r |
| | 989 | \r |
| | 990 | memset (mPdbNameModHandleArray + PreviousSize, 0, MAX_PDB_NAME_TO_MOD_HANDLE_ARRAY_SIZE * sizeof (PDB_NAME_TO_MOD_HANDLE));\r |
| | 991 | \r |
| | 992 | return AddModHandle (ImageContext, ModHandle);\r |
| | 993 | }\r |
| | 994 | \r |
| | 995 | \r |
| | 996 | VOID *\r |
| | 997 | RemoveModeHandle (\r |
| | 998 | IN PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext\r |
| | 999 | )\r |
| | 1000 | /*++\r |
| | 1001 | \r |
| | 1002 | Routine Description:\r |
| | 1003 | Return the ModHandle and delete the entry in the array.\r |
| | 1004 | \r |
| | 1005 | Arguments:\r |
| | 1006 | ImageContext - Input data returned from PE Laoder Library. Used to find the \r |
| | 1007 | .PDB file name of the PE Image.\r |
| | 1008 | \r |
| | 1009 | Returns:\r |
| | 1010 | ModHandle - ModHandle assoicated with ImageContext is returned\r |
| | 1011 | NULL - No ModHandle associated with ImageContext\r |
| | 1012 | \r |
| | 1013 | --*/\r |
| | 1014 | {\r |
| | 1015 | UINTN Index;\r |
| | 1016 | PDB_NAME_TO_MOD_HANDLE *Array;\r |
| | 1017 | \r |
| | 1018 | if (ImageContext->PdbPointer == NULL) {\r |
| | 1019 | //\r |
| | 1020 | // If no PDB pointer there is no ModHandle so return NULL\r |
| | 1021 | //\r |
| | 1022 | return NULL;\r |
| | 1023 | }\r |
| | 1024 | \r |
| | 1025 | Array = mPdbNameModHandleArray;\r |
| | 1026 | for (Index = 0; Index < mPdbNameModHandleArraySize; Index++, Array++) {\r |
| | 1027 | if ((Array->PdbPointer != NULL) && (strcmp(Array->PdbPointer, ImageContext->PdbPointer) == 0)) {\r |
| | 1028 | //\r |
| | 1029 | // If you find a match return it and delete the entry\r |
| | 1030 | //\r |
| | 1031 | free (Array->PdbPointer);\r |
| | 1032 | Array->PdbPointer = NULL;\r |
| | 1033 | return Array->ModHandle;\r |
| | 1034 | }\r |
| | 1035 | }\r |
| | 1036 | \r |
| | 1037 | return NULL;\r |
| | 1038 | }\r |
| | 1039 | \r |
| | 1040 | \r |
| | 1041 | \r |
| | 1042 | EFI_STATUS\r |
| | 1043 | EFIAPI\r |
| | 1044 | SecNt32PeCoffGetImageInfo (\r |
| | 1045 | IN EFI_PEI_PE_COFF_LOADER_PROTOCOL *This,\r |
| | 1046 | IN OUT PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext\r |
| | 1047 | )\r |
| | 1048 | {\r |
| | 1049 | EFI_STATUS Status;\r |
| | 1050 | \r |
| | 1051 | Status = PeCoffLoaderGetImageInfo (ImageContext);\r |
| | 1052 | if (EFI_ERROR (Status)) {\r |
| | 1053 | return Status;\r |
| | 1054 | }\r |
| | 1055 | \r |
| | 1056 | switch (ImageContext->ImageType) {\r |
| | 1057 | \r |
| | 1058 | case EFI_IMAGE_SUBSYSTEM_EFI_APPLICATION:\r |
| | 1059 | ImageContext->ImageCodeMemoryType = EfiLoaderCode;\r |
| | 1060 | ImageContext->ImageDataMemoryType = EfiLoaderData;\r |
| | 1061 | break;\r |
| | 1062 | \r |
| | 1063 | case EFI_IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVER:\r |
| | 1064 | ImageContext->ImageCodeMemoryType = EfiBootServicesCode;\r |
| | 1065 | ImageContext->ImageDataMemoryType = EfiBootServicesData;\r |
| | 1066 | break;\r |
| | 1067 | \r |
| | 1068 | case EFI_IMAGE_SUBSYSTEM_EFI_RUNTIME_DRIVER:\r |
| | 1069 | case EFI_IMAGE_SUBSYSTEM_SAL_RUNTIME_DRIVER:\r |
| | 1070 | ImageContext->ImageCodeMemoryType = EfiRuntimeServicesCode;\r |
| | 1071 | ImageContext->ImageDataMemoryType = EfiRuntimeServicesData;\r |
| | 1072 | break;\r |
| | 1073 | \r |
| | 1074 | default:\r |
| | 1075 | ImageContext->ImageError = IMAGE_ERROR_INVALID_SUBSYSTEM;\r |
| | 1076 | return RETURN_UNSUPPORTED;\r |
| | 1077 | }\r |
| | 1078 | \r |
| | 1079 | return Status;\r |
| | 1080 | }\r |
| | 1081 | \r |
| | 1082 | EFI_STATUS\r |
| | 1083 | EFIAPI\r |
| | 1084 | SecNt32PeCoffLoadImage (\r |
| | 1085 | IN EFI_PEI_PE_COFF_LOADER_PROTOCOL *This,\r |
| | 1086 | IN OUT PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext\r |
| | 1087 | )\r |
| | 1088 | {\r |
| | 1089 | EFI_STATUS Status;\r |
| | 1090 | \r |
| | 1091 | Status = PeCoffLoaderLoadImage (ImageContext);\r |
| | 1092 | return Status;\r |
| | 1093 | }\r |
| | 1094 | \r |
| | 1095 | EFI_STATUS\r |
| | 1096 | EFIAPI\r |
| | 1097 | SecNt32PeCoffRelocateImage (\r |
| | 1098 | IN EFI_PEI_PE_COFF_LOADER_PROTOCOL *This,\r |
| | 1099 | IN OUT PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext\r |
| | 1100 | )\r |
| | 1101 | {\r |
| | 1102 | EFI_STATUS Status;\r |
| | 1103 | VOID *DllEntryPoint;\r |
| | 1104 | CHAR16 *DllFileName;\r |
| | 1105 | HMODULE Library;\r |
| | 1106 | UINTN Index;\r |
| | 1107 | \r |
| | 1108 | \r |
| | 1109 | Status = PeCoffLoaderRelocateImage (ImageContext);\r |
| | 1110 | if (EFI_ERROR (Status)) {\r |
| | 1111 | //\r |
| | 1112 | // We could not relocated the image in memory properly\r |
| | 1113 | //\r |
| | 1114 | return Status;\r |
| | 1115 | }\r |
| | 1116 | \r |
| | 1117 | //\r |
| | 1118 | // If we load our own PE COFF images the Windows debugger can not source\r |
| | 1119 | // level debug our code. If a valid PDB pointer exists usw it to load\r |
| | 1120 | // the *.dll file as a library using Windows* APIs. This allows \r |
| | 1121 | // source level debug. The image is still loaded and reloaced\r |
| | 1122 | // in the Framework memory space like on a real system (by the code above),\r |
| | 1123 | // but the entry point points into the DLL loaded by the code bellow. \r |
| | 1124 | //\r |
| | 1125 | \r |
| | 1126 | DllEntryPoint = NULL;\r |
| | 1127 | \r |
| | 1128 | //\r |
| | 1129 | // Load the DLL if it's not an EBC image.\r |
| | 1130 | //\r |
| | 1131 | if ((ImageContext->PdbPointer != NULL) &&\r |
| | 1132 | (ImageContext->Machine != EFI_IMAGE_MACHINE_EBC)) {\r |
| | 1133 | //\r |
| | 1134 | // Convert filename from ASCII to Unicode\r |
| | 1135 | //\r |
| | 1136 | DllFileName = AsciiToUnicode (ImageContext->PdbPointer, &Index);\r |
| | 1137 | \r |
| | 1138 | //\r |
| | 1139 | // Check that we have a valid filename\r |
| | 1140 | //\r |
| | 1141 | if (Index < 5 || DllFileName[Index - 4] != '.') {\r |
| | 1142 | free (DllFileName);\r |
| | 1143 | \r |
| | 1144 | //\r |
| | 1145 | // Never return an error if PeCoffLoaderRelocateImage() succeeded.\r |
| | 1146 | // The image will run, but we just can't source level debug. If we\r |
| | 1147 | // return an error the image will not run.\r |
| | 1148 | //\r |
| | 1149 | return EFI_SUCCESS;\r |
| | 1150 | }\r |
| | 1151 | //\r |
| | 1152 | // Replace .PDB with .DLL on the filename\r |
| | 1153 | //\r |
| | 1154 | DllFileName[Index - 3] = 'D';\r |
| | 1155 | DllFileName[Index - 2] = 'L';\r |
| | 1156 | DllFileName[Index - 1] = 'L';\r |
| | 1157 | \r |
| | 1158 | //\r |
| | 1159 | // Load the .DLL file into the user process's address space for source \r |
| | 1160 | // level debug\r |
| | 1161 | //\r |
| | 1162 | Library = LoadLibraryEx (DllFileName, NULL, DONT_RESOLVE_DLL_REFERENCES);\r |
| | 1163 | if (Library != NULL) {\r |
| | 1164 | //\r |
| | 1165 | // InitializeDriver is the entry point we put in all our EFI DLL's. The\r |
| | 1166 | // DONT_RESOLVE_DLL_REFERENCES argument to LoadLIbraryEx() supresses the \r |
| | 1167 | // normal DLL entry point of DllMain, and prevents other modules that are\r |
| | 1168 | // referenced in side the DllFileName from being loaded. There is no error \r |
| | 1169 | // checking as the we can point to the PE32 image loaded by Tiano. This \r |
| | 1170 | // step is only needed for source level debuging\r |
| | 1171 | //\r |
| | 1172 | DllEntryPoint = (VOID *) (UINTN) GetProcAddress (Library, "InitializeDriver");\r |
| | 1173 | \r |
| | 1174 | }\r |
| | 1175 | \r |
| | 1176 | if ((Library != NULL) && (DllEntryPoint != NULL)) {\r |
| | 1177 | AddModHandle (ImageContext, Library);\r |
| | 1178 | ImageContext->EntryPoint = (EFI_PHYSICAL_ADDRESS) (UINTN) DllEntryPoint;\r |
| | 1179 | wprintf (L"LoadLibraryEx (%s,\n NULL, DONT_RESOLVE_DLL_REFERENCES)\n", DllFileName);\r |
| | 1180 | } else {\r |
| | 1181 | wprintf (L"WARNING: No source level debug %s. \n", DllFileName);\r |
| | 1182 | }\r |
| | 1183 | \r |
| | 1184 | free (DllFileName);\r |
| | 1185 | }\r |
| | 1186 | \r |
| | 1187 | //\r |
| | 1188 | // Never return an error if PeCoffLoaderRelocateImage() succeeded.\r |
| | 1189 | // The image will run, but we just can't source level debug. If we\r |
| | 1190 | // return an error the image will not run.\r |
| | 1191 | //\r |
| | 1192 | return EFI_SUCCESS;\r |
| | 1193 | }\r |
| | 1194 | \r |
| | 1195 | \r |
| | 1196 | EFI_STATUS\r |
| | 1197 | EFIAPI\r |
| | 1198 | SecNt32PeCoffUnloadimage (\r |
| | 1199 | IN EFI_PEI_PE_COFF_LOADER_PROTOCOL *This,\r |
| | 1200 | IN PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext\r |
| | 1201 | )\r |
| | 1202 | {\r |
| | 1203 | VOID *ModHandle;\r |
| | 1204 | \r |
| | 1205 | ModHandle = RemoveModeHandle (ImageContext);\r |
| | 1206 | if (ModHandle != NULL) {\r |
| | 1207 | FreeLibrary (ModHandle);\r |
| | 1208 | }\r |
| | 1209 | return EFI_SUCCESS;\r |
| | 1210 | }\r |
| | 1211 | \r |
| | 1212 | VOID\r |
| | 1213 | _ModuleEntryPoint (\r |
| | 1214 | VOID\r |
| | 1215 | )\r |
| | 1216 | {\r |
| | 1217 | }\r |
| | 1218 | \r |
projects / mirror_edk2.git / blame_incremental