]> git.proxmox.com Git - mirror_edk2.git/blame_incremental - ShellPkg/Library/UefiHandleParsingLib/UefiHandleParsingLib.c
ShellPkg: Fixes and updates for the 'devices' command
[mirror_edk2.git] / ShellPkg / Library / UefiHandleParsingLib / UefiHandleParsingLib.c
... / ...
CommitLineData
1/** @file\r
2 Provides interface to advanced shell functionality for parsing both handle and protocol database.\r
3\r
4 Copyright (c) 2013 - 2014, Hewlett-Packard Development Company, L.P.\r
5 Copyright (c) 2010 - 2014, Intel Corporation. All rights reserved.<BR>\r
6 This program and the accompanying materials\r
7 are licensed and made available under the terms and conditions of the BSD License\r
8 which accompanies this distribution. The full text of the license may be found at\r
9 http://opensource.org/licenses/bsd-license.php\r
10\r
11 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
12 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
13\r
14**/\r
15\r
16#include "UefiHandleParsingLib.h"\r
17#include "IndustryStandard/Acpi10.h"\r
18\r
19EFI_HANDLE mHandleParsingHiiHandle = NULL;\r
20HANDLE_INDEX_LIST mHandleList = {{{NULL,NULL},0,0},0};\r
21GUID_INFO_BLOCK *GuidList;\r
22UINTN GuidListCount;\r
23/**\r
24 Function to translate the EFI_MEMORY_TYPE into a string.\r
25\r
26 @param[in] Memory The memory type.\r
27\r
28 @retval A string representation of the type allocated from BS Pool.\r
29**/\r
30CHAR16*\r
31EFIAPI\r
32ConvertMemoryType (\r
33 IN CONST EFI_MEMORY_TYPE Memory\r
34 )\r
35{\r
36 CHAR16 *RetVal;\r
37 RetVal = NULL;\r
38\r
39 switch (Memory) {\r
40 case EfiReservedMemoryType: StrnCatGrow(&RetVal, NULL, L"EfiReservedMemoryType", 0); break;\r
41 case EfiLoaderCode: StrnCatGrow(&RetVal, NULL, L"EfiLoaderCode", 0); break;\r
42 case EfiLoaderData: StrnCatGrow(&RetVal, NULL, L"EfiLoaderData", 0); break;\r
43 case EfiBootServicesCode: StrnCatGrow(&RetVal, NULL, L"EfiBootServicesCode", 0); break;\r
44 case EfiBootServicesData: StrnCatGrow(&RetVal, NULL, L"EfiBootServicesData", 0); break;\r
45 case EfiRuntimeServicesCode: StrnCatGrow(&RetVal, NULL, L"EfiRuntimeServicesCode", 0); break;\r
46 case EfiRuntimeServicesData: StrnCatGrow(&RetVal, NULL, L"EfiRuntimeServicesData", 0); break;\r
47 case EfiConventionalMemory: StrnCatGrow(&RetVal, NULL, L"EfiConventionalMemory", 0); break;\r
48 case EfiUnusableMemory: StrnCatGrow(&RetVal, NULL, L"EfiUnusableMemory", 0); break;\r
49 case EfiACPIReclaimMemory: StrnCatGrow(&RetVal, NULL, L"EfiACPIReclaimMemory", 0); break;\r
50 case EfiACPIMemoryNVS: StrnCatGrow(&RetVal, NULL, L"EfiACPIMemoryNVS", 0); break;\r
51 case EfiMemoryMappedIO: StrnCatGrow(&RetVal, NULL, L"EfiMemoryMappedIO", 0); break;\r
52 case EfiMemoryMappedIOPortSpace: StrnCatGrow(&RetVal, NULL, L"EfiMemoryMappedIOPortSpace", 0); break;\r
53 case EfiPalCode: StrnCatGrow(&RetVal, NULL, L"EfiPalCode", 0); break;\r
54 case EfiMaxMemoryType: StrnCatGrow(&RetVal, NULL, L"EfiMaxMemoryType", 0); break;\r
55 default: ASSERT(FALSE);\r
56 }\r
57 return (RetVal);\r
58}\r
59\r
60/**\r
61 Function to translate the EFI_GRAPHICS_PIXEL_FORMAT into a string.\r
62\r
63 @param[in] Fmt The format type.\r
64\r
65 @retval A string representation of the type allocated from BS Pool.\r
66**/\r
67CHAR16*\r
68EFIAPI\r
69ConvertPixelFormat (\r
70 IN CONST EFI_GRAPHICS_PIXEL_FORMAT Fmt\r
71 )\r
72{\r
73 CHAR16 *RetVal;\r
74 RetVal = NULL;\r
75\r
76 switch (Fmt) {\r
77 case PixelRedGreenBlueReserved8BitPerColor: StrnCatGrow(&RetVal, NULL, L"PixelRedGreenBlueReserved8BitPerColor", 0); break;\r
78 case PixelBlueGreenRedReserved8BitPerColor: StrnCatGrow(&RetVal, NULL, L"PixelBlueGreenRedReserved8BitPerColor", 0); break;\r
79 case PixelBitMask: StrnCatGrow(&RetVal, NULL, L"PixelBitMask", 0); break;\r
80 case PixelBltOnly: StrnCatGrow(&RetVal, NULL, L"PixelBltOnly", 0); break;\r
81 case PixelFormatMax: StrnCatGrow(&RetVal, NULL, L"PixelFormatMax", 0); break;\r
82 default: ASSERT(FALSE);\r
83 }\r
84 return (RetVal);\r
85}\r
86\r
87/**\r
88 Constructor for the library.\r
89\r
90 @param[in] ImageHandle Ignored.\r
91 @param[in] SystemTable Ignored.\r
92\r
93 @retval EFI_SUCCESS The operation was successful.\r
94**/\r
95EFI_STATUS\r
96EFIAPI\r
97HandleParsingLibConstructor (\r
98 IN EFI_HANDLE ImageHandle,\r
99 IN EFI_SYSTEM_TABLE *SystemTable\r
100 )\r
101{\r
102 GuidListCount = 0;\r
103 GuidList = NULL;\r
104\r
105 //\r
106 // Do nothing with mHandleParsingHiiHandle. Initialize HII as needed.\r
107 //\r
108 return (EFI_SUCCESS);\r
109}\r
110\r
111/** \r
112 Initialization function for HII packages.\r
113 \r
114**/\r
115VOID\r
116HandleParsingHiiInit (VOID)\r
117{\r
118 if (mHandleParsingHiiHandle == NULL) {\r
119 mHandleParsingHiiHandle = HiiAddPackages (&gHandleParsingHiiGuid, gImageHandle, UefiHandleParsingLibStrings, NULL);\r
120 ASSERT (mHandleParsingHiiHandle != NULL);\r
121 }\r
122}\r
123\r
124/**\r
125 Destructor for the library. free any resources.\r
126\r
127 @param[in] ImageHandle Ignored.\r
128 @param[in] SystemTable Ignored.\r
129\r
130 @retval EFI_SUCCESS The operation was successful.\r
131**/\r
132EFI_STATUS\r
133EFIAPI\r
134HandleParsingLibDestructor (\r
135 IN EFI_HANDLE ImageHandle,\r
136 IN EFI_SYSTEM_TABLE *SystemTable\r
137 )\r
138{\r
139 UINTN LoopCount;\r
140\r
141 for (LoopCount = 0; GuidList != NULL && LoopCount < GuidListCount; LoopCount++) {\r
142 SHELL_FREE_NON_NULL(GuidList[LoopCount].GuidId);\r
143 }\r
144\r
145 SHELL_FREE_NON_NULL(GuidList);\r
146 if (mHandleParsingHiiHandle != NULL) {\r
147 HiiRemovePackages(mHandleParsingHiiHandle);\r
148 }\r
149 return (EFI_SUCCESS);\r
150}\r
151\r
152/**\r
153 Function to dump information about LoadedImage.\r
154\r
155 This will allocate the return buffer from boot services pool.\r
156\r
157 @param[in] TheHandle The handle that has LoadedImage installed.\r
158 @param[in] Verbose TRUE for additional information, FALSE otherwise.\r
159\r
160 @retval A poitner to a string containing the information.\r
161**/\r
162CHAR16*\r
163EFIAPI\r
164LoadedImageProtocolDumpInformation(\r
165 IN CONST EFI_HANDLE TheHandle,\r
166 IN CONST BOOLEAN Verbose\r
167 )\r
168{\r
169 EFI_LOADED_IMAGE_PROTOCOL *LoadedImage;\r
170 EFI_STATUS Status;\r
171 CHAR16 *RetVal;\r
172 CHAR16 *Temp;\r
173 CHAR16 *CodeType;\r
174 CHAR16 *DataType;\r
175\r
176 if (!Verbose) {\r
177 return (CatSPrint(NULL, L"LoadedImage"));\r
178 }\r
179\r
180 HandleParsingHiiInit();\r
181\r
182 Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_LI_DUMP_MAIN), NULL);\r
183 RetVal = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));\r
184 if (Temp == NULL || RetVal == NULL) {\r
185 SHELL_FREE_NON_NULL(Temp);\r
186 SHELL_FREE_NON_NULL(RetVal);\r
187 return NULL;\r
188 }\r
189\r
190 Status = gBS->OpenProtocol (\r
191 TheHandle,\r
192 &gEfiLoadedImageProtocolGuid,\r
193 (VOID**)&LoadedImage,\r
194 gImageHandle,\r
195 NULL,\r
196 EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
197 );\r
198\r
199 if (EFI_ERROR (Status)) {\r
200 SHELL_FREE_NON_NULL (Temp);\r
201 SHELL_FREE_NON_NULL (RetVal);\r
202 return NULL;\r
203 }\r
204\r
205 DataType = ConvertMemoryType(LoadedImage->ImageDataType);\r
206 CodeType = ConvertMemoryType(LoadedImage->ImageCodeType);\r
207\r
208 RetVal = CatSPrint(RetVal,\r
209 Temp,\r
210 LoadedImage->Revision,\r
211 LoadedImage->ParentHandle,\r
212 LoadedImage->SystemTable,\r
213 LoadedImage->DeviceHandle,\r
214 LoadedImage->FilePath,\r
215 LoadedImage->LoadOptionsSize,\r
216 LoadedImage->LoadOptions,\r
217 LoadedImage->ImageBase,\r
218 LoadedImage->ImageSize,\r
219 CodeType,\r
220 DataType,\r
221 LoadedImage->Unload);\r
222\r
223 \r
224 SHELL_FREE_NON_NULL(Temp);\r
225 SHELL_FREE_NON_NULL(CodeType);\r
226 SHELL_FREE_NON_NULL(DataType);\r
227\r
228 return RetVal;\r
229}\r
230\r
231/**\r
232 Function to dump information about GOP.\r
233\r
234 This will allocate the return buffer from boot services pool.\r
235\r
236 @param[in] TheHandle The handle that has LoadedImage installed.\r
237 @param[in] Verbose TRUE for additional information, FALSE otherwise.\r
238\r
239 @retval A poitner to a string containing the information.\r
240**/\r
241CHAR16*\r
242EFIAPI\r
243GraphicsOutputProtocolDumpInformation(\r
244 IN CONST EFI_HANDLE TheHandle,\r
245 IN CONST BOOLEAN Verbose\r
246 )\r
247{\r
248 EFI_GRAPHICS_OUTPUT_PROTOCOL *GraphicsOutput;\r
249 EFI_STATUS Status;\r
250 CHAR16 *RetVal;\r
251 CHAR16 *Temp;\r
252 CHAR16 *Fmt;\r
253\r
254 if (!Verbose) {\r
255 return (CatSPrint(NULL, L"GraphicsOutput"));\r
256 }\r
257\r
258 HandleParsingHiiInit();\r
259\r
260 Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_GOP_DUMP_MAIN), NULL);\r
261 RetVal = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));\r
262 if (Temp == NULL || RetVal == NULL) {\r
263 SHELL_FREE_NON_NULL(Temp);\r
264 SHELL_FREE_NON_NULL(RetVal);\r
265 return NULL;\r
266 }\r
267\r
268 Status = gBS->OpenProtocol (\r
269 TheHandle,\r
270 &gEfiGraphicsOutputProtocolGuid,\r
271 (VOID**)&GraphicsOutput,\r
272 gImageHandle,\r
273 NULL,\r
274 EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
275 );\r
276\r
277 if (EFI_ERROR (Status)) {\r
278 SHELL_FREE_NON_NULL (Temp);\r
279 SHELL_FREE_NON_NULL (RetVal);\r
280 return NULL;\r
281 }\r
282\r
283 Fmt = ConvertPixelFormat(GraphicsOutput->Mode->Info->PixelFormat);\r
284\r
285 RetVal = CatSPrint(RetVal,\r
286 Temp,\r
287 GraphicsOutput->Mode->MaxMode,\r
288 GraphicsOutput->Mode->Mode,\r
289 GraphicsOutput->Mode->FrameBufferBase,\r
290 (UINT64)GraphicsOutput->Mode->FrameBufferSize,\r
291 (UINT64)GraphicsOutput->Mode->SizeOfInfo,\r
292 GraphicsOutput->Mode->Info->Version,\r
293 GraphicsOutput->Mode->Info->HorizontalResolution,\r
294 GraphicsOutput->Mode->Info->VerticalResolution,\r
295 Fmt,\r
296 GraphicsOutput->Mode->Info->PixelsPerScanLine,\r
297 GraphicsOutput->Mode->Info->PixelFormat!=PixelBitMask?0:GraphicsOutput->Mode->Info->PixelInformation.RedMask,\r
298 GraphicsOutput->Mode->Info->PixelFormat!=PixelBitMask?0:GraphicsOutput->Mode->Info->PixelInformation.GreenMask,\r
299 GraphicsOutput->Mode->Info->PixelFormat!=PixelBitMask?0:GraphicsOutput->Mode->Info->PixelInformation.BlueMask\r
300 );\r
301 \r
302 SHELL_FREE_NON_NULL(Temp);\r
303 SHELL_FREE_NON_NULL(Fmt);\r
304\r
305 return RetVal;\r
306}\r
307\r
308/**\r
309 Function to dump information about PciRootBridgeIo.\r
310\r
311 This will allocate the return buffer from boot services pool.\r
312\r
313 @param[in] TheHandle The handle that has PciRootBridgeIo installed.\r
314 @param[in] Verbose TRUE for additional information, FALSE otherwise.\r
315\r
316 @retval A poitner to a string containing the information.\r
317**/\r
318CHAR16*\r
319EFIAPI\r
320PciRootBridgeIoDumpInformation(\r
321 IN CONST EFI_HANDLE TheHandle,\r
322 IN CONST BOOLEAN Verbose\r
323 )\r
324{\r
325 EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL *PciRootBridgeIo;\r
326 EFI_ACPI_ADDRESS_SPACE_DESCRIPTOR *Configuration;\r
327 UINT64 Supports;\r
328 UINT64 Attributes;\r
329 CHAR16 *Temp;\r
330 CHAR16 *Temp2;\r
331 CHAR16 *RetVal;\r
332 EFI_STATUS Status;\r
333\r
334 RetVal = NULL;\r
335\r
336 if (!Verbose) {\r
337 return (CatSPrint(NULL, L"PciRootBridgeIo"));\r
338 }\r
339\r
340 HandleParsingHiiInit();\r
341\r
342 Status = gBS->HandleProtocol(\r
343 TheHandle,\r
344 &gEfiPciRootBridgeIoProtocolGuid,\r
345 (VOID**)&PciRootBridgeIo);\r
346\r
347 if (EFI_ERROR(Status)) {\r
348 return NULL;\r
349 }\r
350\r
351 Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_PH), NULL);\r
352 ASSERT (Temp != NULL);\r
353 Temp2 = CatSPrint(L"\r\n", Temp, PciRootBridgeIo->ParentHandle);\r
354 FreePool(Temp);\r
355 RetVal = Temp2;\r
356 Temp2 = NULL;\r
357 \r
358 Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_SEG), NULL);\r
359 ASSERT (Temp != NULL);\r
360 Temp2 = CatSPrint(RetVal, Temp, PciRootBridgeIo->SegmentNumber);\r
361 FreePool(Temp);\r
362 FreePool(RetVal);\r
363 RetVal = Temp2;\r
364 Temp2 = NULL;\r
365\r
366 Supports = 0;\r
367 Attributes = 0;\r
368 Status = PciRootBridgeIo->GetAttributes (PciRootBridgeIo, &Supports, &Attributes);\r
369 if (!EFI_ERROR(Status)) {\r
370 Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_ATT), NULL);\r
371 ASSERT (Temp != NULL); \r
372 Temp2 = CatSPrint(RetVal, Temp, Attributes);\r
373 FreePool(Temp);\r
374 FreePool(RetVal);\r
375 RetVal = Temp2;\r
376 Temp2 = NULL;\r
377 \r
378 Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_SUPPORTS), NULL);\r
379 ASSERT (Temp != NULL);\r
380 Temp2 = CatSPrint(RetVal, Temp, Supports);\r
381 FreePool(Temp);\r
382 FreePool(RetVal);\r
383 RetVal = Temp2;\r
384 Temp2 = NULL;\r
385 }\r
386\r
387 Configuration = NULL;\r
388 Status = PciRootBridgeIo->Configuration (PciRootBridgeIo, (VOID **) &Configuration);\r
389 if (!EFI_ERROR(Status) && Configuration != NULL) {\r
390 Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_TITLE), NULL);\r
391 ASSERT (Temp != NULL);\r
392 Temp2 = CatSPrint(RetVal, Temp, Supports);\r
393 FreePool(Temp);\r
394 FreePool(RetVal);\r
395 RetVal = Temp2;\r
396 Temp2 = NULL;\r
397 while (Configuration->Desc == ACPI_ADDRESS_SPACE_DESCRIPTOR) {\r
398 Temp = NULL;\r
399 switch (Configuration->ResType) {\r
400 case ACPI_ADDRESS_SPACE_TYPE_MEM:\r
401 Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_MEM), NULL);\r
402 break;\r
403 case ACPI_ADDRESS_SPACE_TYPE_IO:\r
404 Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_IO), NULL);\r
405 break;\r
406 case ACPI_ADDRESS_SPACE_TYPE_BUS:\r
407 Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_BUS), NULL);\r
408 break;\r
409 }\r
410 if (Temp != NULL) {\r
411 Temp2 = CatSPrint(RetVal, L"%s", Temp);\r
412 FreePool(Temp);\r
413 FreePool(RetVal);\r
414 RetVal = Temp2;\r
415 Temp2 = NULL;\r
416 }\r
417\r
418 Temp2 = CatSPrint(RetVal, \r
419 L"%H%02x %016lx %016lx %02x%N\r\n",\r
420 Configuration->SpecificFlag,\r
421 Configuration->AddrRangeMin,\r
422 Configuration->AddrRangeMax,\r
423 Configuration->AddrSpaceGranularity\r
424 );\r
425 FreePool(RetVal);\r
426 RetVal = Temp2;\r
427 Temp2 = NULL;\r
428 Configuration++;\r
429 }\r
430 }\r
431 return (RetVal);\r
432}\r
433\r
434/**\r
435 Function to dump information about SimpleTextOut.\r
436\r
437 This will allocate the return buffer from boot services pool.\r
438\r
439 @param[in] TheHandle The handle that has SimpleTextOut installed.\r
440 @param[in] Verbose TRUE for additional information, FALSE otherwise.\r
441\r
442 @retval A poitner to a string containing the information.\r
443**/\r
444CHAR16*\r
445EFIAPI\r
446TxtOutProtocolDumpInformation(\r
447 IN CONST EFI_HANDLE TheHandle,\r
448 IN CONST BOOLEAN Verbose\r
449 )\r
450{\r
451 EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *Dev;\r
452 INTN Index;\r
453 UINTN Col;\r
454 UINTN Row;\r
455 EFI_STATUS Status;\r
456 CHAR16 *RetVal;\r
457 UINTN Size;\r
458 CHAR16 *Temp;\r
459 UINTN NewSize;\r
460\r
461 if (!Verbose) {\r
462 return (NULL);\r
463 }\r
464\r
465 HandleParsingHiiInit();\r
466\r
467 RetVal = NULL;\r
468 Size = 0;\r
469\r
470 Status = gBS->HandleProtocol(\r
471 TheHandle,\r
472 &gEfiSimpleTextOutProtocolGuid,\r
473 (VOID**)&Dev);\r
474\r
475 ASSERT_EFI_ERROR(Status);\r
476 ASSERT (Dev != NULL && Dev->Mode != NULL);\r
477\r
478 Size = (Dev->Mode->MaxMode + 1) * 80;\r
479 RetVal = AllocateZeroPool(Size);\r
480\r
481 Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_TXT_OUT_DUMP_HEADER), NULL);\r
482 if (Temp != NULL) {\r
483 UnicodeSPrint(RetVal, Size, Temp, Dev, Dev->Mode->Attribute);\r
484 FreePool(Temp);\r
485 }\r
486\r
487 //\r
488 // Dump TextOut Info\r
489 //\r
490 Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_TXT_OUT_DUMP_LINE), NULL);\r
491 for (Index = 0; Index < Dev->Mode->MaxMode; Index++) {\r
492 Status = Dev->QueryMode (Dev, Index, &Col, &Row);\r
493 NewSize = Size - StrSize(RetVal);\r
494 UnicodeSPrint(\r
495 RetVal + StrLen(RetVal),\r
496 NewSize,\r
497 Temp == NULL?L"":Temp,\r
498 Index == Dev->Mode->Mode ? L'*' : L' ',\r
499 Index,\r
500 !EFI_ERROR(Status)?(INTN)Col:-1,\r
501 !EFI_ERROR(Status)?(INTN)Row:-1\r
502 );\r
503 }\r
504 FreePool(Temp);\r
505 return (RetVal);\r
506}\r
507\r
508STATIC CONST UINTN VersionStringSize = 60;\r
509\r
510/**\r
511 Function to dump information about EfiDriverSupportedEfiVersion protocol.\r
512\r
513 This will allocate the return buffer from boot services pool.\r
514\r
515 @param[in] TheHandle The handle that has the protocol installed.\r
516 @param[in] Verbose TRUE for additional information, FALSE otherwise.\r
517\r
518 @retval A poitner to a string containing the information.\r
519**/\r
520CHAR16*\r
521EFIAPI\r
522DriverEfiVersionProtocolDumpInformation(\r
523 IN CONST EFI_HANDLE TheHandle,\r
524 IN CONST BOOLEAN Verbose\r
525 )\r
526{\r
527 EFI_DRIVER_SUPPORTED_EFI_VERSION_PROTOCOL *DriverEfiVersion;\r
528 EFI_STATUS Status;\r
529 CHAR16 *RetVal;\r
530\r
531 Status = gBS->HandleProtocol(\r
532 TheHandle,\r
533 &gEfiDriverSupportedEfiVersionProtocolGuid,\r
534 (VOID**)&DriverEfiVersion);\r
535\r
536 ASSERT_EFI_ERROR(Status);\r
537\r
538 RetVal = AllocateZeroPool(VersionStringSize);\r
539 ASSERT(RetVal != NULL);\r
540 UnicodeSPrint(RetVal, VersionStringSize, L"0x%08x", DriverEfiVersion->FirmwareVersion);\r
541 return (RetVal);\r
542}\r
543\r
544/**\r
545 Function to dump information about DevicePath protocol.\r
546\r
547 This will allocate the return buffer from boot services pool.\r
548\r
549 @param[in] TheHandle The handle that has the protocol installed.\r
550 @param[in] Verbose TRUE for additional information, FALSE otherwise.\r
551\r
552 @retval A poitner to a string containing the information.\r
553**/\r
554CHAR16*\r
555EFIAPI\r
556DevicePathProtocolDumpInformation(\r
557 IN CONST EFI_HANDLE TheHandle,\r
558 IN CONST BOOLEAN Verbose\r
559 )\r
560{\r
561 EFI_DEVICE_PATH_PROTOCOL *DevPath;\r
562 CHAR16 *Temp;\r
563 CHAR16 *Temp2;\r
564 EFI_STATUS Status;\r
565 Temp = NULL;\r
566\r
567 Status = gBS->OpenProtocol(TheHandle, &gEfiDevicePathProtocolGuid, (VOID**)&DevPath, gImageHandle, NULL, EFI_OPEN_PROTOCOL_GET_PROTOCOL);\r
568 if (!EFI_ERROR(Status)) {\r
569 //\r
570 // I cannot decide whether to allow shortcuts here (the second BOOLEAN on the next line)\r
571 //\r
572 Temp = ConvertDevicePathToText(DevPath, TRUE, TRUE);\r
573 gBS->CloseProtocol(TheHandle, &gEfiDevicePathProtocolGuid, gImageHandle, NULL);\r
574 }\r
575 if (!Verbose && Temp != NULL && StrLen(Temp) > 30) {\r
576 Temp2 = NULL;\r
577 Temp2 = StrnCatGrow(&Temp2, NULL, Temp+(StrLen(Temp) - 30), 30);\r
578 FreePool(Temp);\r
579 Temp = Temp2;\r
580 }\r
581 return (Temp);\r
582}\r
583\r
584//\r
585// Put the information on the NT32 protocol GUIDs here so we are not dependant on the Nt32Pkg\r
586//\r
587#define LOCAL_EFI_WIN_NT_THUNK_PROTOCOL_GUID \\r
588 { \\r
589 0x58c518b1, 0x76f3, 0x11d4, { 0xbc, 0xea, 0x0, 0x80, 0xc7, 0x3c, 0x88, 0x81 } \\r
590 }\r
591\r
592#define LOCAL_EFI_WIN_NT_BUS_DRIVER_IO_PROTOCOL_GUID \\r
593 { \\r
594 0x96eb4ad6, 0xa32a, 0x11d4, { 0xbc, 0xfd, 0x0, 0x80, 0xc7, 0x3c, 0x88, 0x81 } \\r
595 }\r
596\r
597#define LOCAL_EFI_WIN_NT_SERIAL_PORT_GUID \\r
598 { \\r
599 0xc95a93d, 0xa006, 0x11d4, { 0xbc, 0xfa, 0x0, 0x80, 0xc7, 0x3c, 0x88, 0x81 } \\r
600 }\r
601STATIC CONST EFI_GUID WinNtThunkProtocolGuid = LOCAL_EFI_WIN_NT_THUNK_PROTOCOL_GUID;\r
602STATIC CONST EFI_GUID WinNtIoProtocolGuid = LOCAL_EFI_WIN_NT_BUS_DRIVER_IO_PROTOCOL_GUID;\r
603STATIC CONST EFI_GUID WinNtSerialPortGuid = LOCAL_EFI_WIN_NT_SERIAL_PORT_GUID;\r
604\r
605STATIC CONST GUID_INFO_BLOCK mGuidStringListNT[] = {\r
606 {STRING_TOKEN(STR_WINNT_THUNK), (EFI_GUID*)&WinNtThunkProtocolGuid, NULL},\r
607 {STRING_TOKEN(STR_WINNT_DRIVER_IO), (EFI_GUID*)&WinNtIoProtocolGuid, NULL},\r
608 {STRING_TOKEN(STR_WINNT_SERIAL_PORT), (EFI_GUID*)&WinNtSerialPortGuid, NULL},\r
609 {STRING_TOKEN(STR_UNKNOWN_DEVICE), NULL, NULL},\r
610};\r
611\r
612STATIC CONST GUID_INFO_BLOCK mGuidStringList[] = {\r
613 {STRING_TOKEN(STR_LOADED_IMAGE), &gEfiLoadedImageProtocolGuid, LoadedImageProtocolDumpInformation},\r
614 {STRING_TOKEN(STR_DEVICE_PATH), &gEfiDevicePathProtocolGuid, DevicePathProtocolDumpInformation},\r
615 {STRING_TOKEN(STR_IMAGE_PATH), &gEfiLoadedImageDevicePathProtocolGuid, DevicePathProtocolDumpInformation},\r
616 {STRING_TOKEN(STR_DEVICE_PATH_UTIL), &gEfiDevicePathUtilitiesProtocolGuid, NULL},\r
617 {STRING_TOKEN(STR_DEVICE_PATH_TXT), &gEfiDevicePathToTextProtocolGuid, NULL},\r
618 {STRING_TOKEN(STR_DEVICE_PATH_FTXT), &gEfiDevicePathFromTextProtocolGuid, NULL},\r
619 {STRING_TOKEN(STR_DEVICE_PATH_PC), &gEfiPcAnsiGuid, NULL},\r
620 {STRING_TOKEN(STR_DEVICE_PATH_VT100), &gEfiVT100Guid, NULL},\r
621 {STRING_TOKEN(STR_DEVICE_PATH_VT100P), &gEfiVT100PlusGuid, NULL},\r
622 {STRING_TOKEN(STR_DEVICE_PATH_VTUTF8), &gEfiVTUTF8Guid, NULL},\r
623 {STRING_TOKEN(STR_DRIVER_BINDING), &gEfiDriverBindingProtocolGuid, NULL},\r
624 {STRING_TOKEN(STR_PLATFORM_OVERRIDE), &gEfiPlatformDriverOverrideProtocolGuid, NULL},\r
625 {STRING_TOKEN(STR_BUS_OVERRIDE), &gEfiBusSpecificDriverOverrideProtocolGuid, NULL},\r
626 {STRING_TOKEN(STR_DRIVER_DIAG), &gEfiDriverDiagnosticsProtocolGuid, NULL},\r
627 {STRING_TOKEN(STR_DRIVER_DIAG2), &gEfiDriverDiagnostics2ProtocolGuid, NULL},\r
628 {STRING_TOKEN(STR_DRIVER_CN), &gEfiComponentNameProtocolGuid, NULL},\r
629 {STRING_TOKEN(STR_DRIVER_CN2), &gEfiComponentName2ProtocolGuid, NULL},\r
630 {STRING_TOKEN(STR_PLAT_DRV_CFG), &gEfiPlatformToDriverConfigurationProtocolGuid, NULL},\r
631 {STRING_TOKEN(STR_DRIVER_VERSION), &gEfiDriverSupportedEfiVersionProtocolGuid, DriverEfiVersionProtocolDumpInformation},\r
632 {STRING_TOKEN(STR_TXT_IN), &gEfiSimpleTextInProtocolGuid, NULL},\r
633 {STRING_TOKEN(STR_TXT_IN_EX), &gEfiSimpleTextInputExProtocolGuid, NULL},\r
634 {STRING_TOKEN(STR_TXT_OUT), &gEfiSimpleTextOutProtocolGuid, TxtOutProtocolDumpInformation},\r
635 {STRING_TOKEN(STR_SIM_POINTER), &gEfiSimplePointerProtocolGuid, NULL},\r
636 {STRING_TOKEN(STR_ABS_POINTER), &gEfiAbsolutePointerProtocolGuid, NULL},\r
637 {STRING_TOKEN(STR_SERIAL_IO), &gEfiSerialIoProtocolGuid, NULL},\r
638 {STRING_TOKEN(STR_GRAPHICS_OUTPUT), &gEfiGraphicsOutputProtocolGuid, GraphicsOutputProtocolDumpInformation},\r
639 {STRING_TOKEN(STR_EDID_DISCOVERED), &gEfiEdidDiscoveredProtocolGuid, NULL},\r
640 {STRING_TOKEN(STR_EDID_ACTIVE), &gEfiEdidActiveProtocolGuid, NULL},\r
641 {STRING_TOKEN(STR_EDID_OVERRIDE), &gEfiEdidOverrideProtocolGuid, NULL},\r
642 {STRING_TOKEN(STR_CON_IN), &gEfiConsoleInDeviceGuid, NULL},\r
643 {STRING_TOKEN(STR_CON_OUT), &gEfiConsoleOutDeviceGuid, NULL},\r
644 {STRING_TOKEN(STR_STD_ERR), &gEfiStandardErrorDeviceGuid, NULL},\r
645 {STRING_TOKEN(STR_LOAD_FILE), &gEfiLoadFileProtocolGuid, NULL},\r
646 {STRING_TOKEN(STR_LOAD_FILE2), &gEfiLoadFile2ProtocolGuid, NULL},\r
647 {STRING_TOKEN(STR_SIMPLE_FILE_SYS), &gEfiSimpleFileSystemProtocolGuid, NULL},\r
648 {STRING_TOKEN(STR_TAPE_IO), &gEfiTapeIoProtocolGuid, NULL},\r
649 {STRING_TOKEN(STR_DISK_IO), &gEfiDiskIoProtocolGuid, NULL},\r
650 {STRING_TOKEN(STR_BLK_IO), &gEfiBlockIoProtocolGuid, NULL},\r
651 {STRING_TOKEN(STR_UC), &gEfiUnicodeCollationProtocolGuid, NULL},\r
652 {STRING_TOKEN(STR_UC2), &gEfiUnicodeCollation2ProtocolGuid, NULL},\r
653 {STRING_TOKEN(STR_PCIRB_IO), &gEfiPciRootBridgeIoProtocolGuid, PciRootBridgeIoDumpInformation},\r
654 {STRING_TOKEN(STR_PCI_IO), &gEfiPciIoProtocolGuid, NULL},\r
655 {STRING_TOKEN(STR_SCSI_PT), &gEfiScsiPassThruProtocolGuid, NULL},\r
656 {STRING_TOKEN(STR_SCSI_IO), &gEfiScsiIoProtocolGuid, NULL},\r
657 {STRING_TOKEN(STR_SCSI_PT_EXT), &gEfiExtScsiPassThruProtocolGuid, NULL},\r
658 {STRING_TOKEN(STR_ISCSI), &gEfiIScsiInitiatorNameProtocolGuid, NULL},\r
659 {STRING_TOKEN(STR_USB_IO), &gEfiUsbIoProtocolGuid, NULL},\r
660 {STRING_TOKEN(STR_USB_HC), &gEfiUsbHcProtocolGuid, NULL},\r
661 {STRING_TOKEN(STR_USB_HC2), &gEfiUsb2HcProtocolGuid, NULL},\r
662 {STRING_TOKEN(STR_DEBUG_SUPPORT), &gEfiDebugSupportProtocolGuid, NULL},\r
663 {STRING_TOKEN(STR_DEBUG_PORT), &gEfiDebugPortProtocolGuid, NULL},\r
664 {STRING_TOKEN(STR_DECOMPRESS), &gEfiDecompressProtocolGuid, NULL},\r
665 {STRING_TOKEN(STR_ACPI_TABLE), &gEfiAcpiTableProtocolGuid, NULL},\r
666 {STRING_TOKEN(STR_EBC_INTERPRETER), &gEfiEbcProtocolGuid, NULL},\r
667 {STRING_TOKEN(STR_SNP), &gEfiSimpleNetworkProtocolGuid, NULL},\r
668 {STRING_TOKEN(STR_NII), &gEfiNetworkInterfaceIdentifierProtocolGuid, NULL},\r
669 {STRING_TOKEN(STR_NII_31), &gEfiNetworkInterfaceIdentifierProtocolGuid_31, NULL},\r
670 {STRING_TOKEN(STR_PXE_BC), &gEfiPxeBaseCodeProtocolGuid, NULL},\r
671 {STRING_TOKEN(STR_PXE_CB), &gEfiPxeBaseCodeCallbackProtocolGuid, NULL},\r
672 {STRING_TOKEN(STR_BIS), &gEfiBisProtocolGuid, NULL},\r
673 {STRING_TOKEN(STR_MNP_SB), &gEfiManagedNetworkServiceBindingProtocolGuid, NULL},\r
674 {STRING_TOKEN(STR_MNP), &gEfiManagedNetworkProtocolGuid, NULL},\r
675 {STRING_TOKEN(STR_ARP_SB), &gEfiArpServiceBindingProtocolGuid, NULL},\r
676 {STRING_TOKEN(STR_ARP), &gEfiArpProtocolGuid, NULL},\r
677 {STRING_TOKEN(STR_DHCPV4_SB), &gEfiDhcp4ServiceBindingProtocolGuid, NULL},\r
678 {STRING_TOKEN(STR_DHCPV4), &gEfiDhcp4ProtocolGuid, NULL},\r
679 {STRING_TOKEN(STR_TCPV4_SB), &gEfiTcp4ServiceBindingProtocolGuid, NULL},\r
680 {STRING_TOKEN(STR_TCPV4), &gEfiTcp4ProtocolGuid, NULL},\r
681 {STRING_TOKEN(STR_IPV4_SB), &gEfiIp4ServiceBindingProtocolGuid, NULL},\r
682 {STRING_TOKEN(STR_IPV4), &gEfiIp4ProtocolGuid, NULL},\r
683 {STRING_TOKEN(STR_IPV4_CFG), &gEfiIp4ConfigProtocolGuid, NULL},\r
684 {STRING_TOKEN(STR_UDPV4_SB), &gEfiUdp4ServiceBindingProtocolGuid, NULL},\r
685 {STRING_TOKEN(STR_UDPV4), &gEfiUdp4ProtocolGuid, NULL},\r
686 {STRING_TOKEN(STR_MTFTPV4_SB), &gEfiMtftp4ServiceBindingProtocolGuid, NULL},\r
687 {STRING_TOKEN(STR_MTFTPV4), &gEfiMtftp4ProtocolGuid, NULL},\r
688 {STRING_TOKEN(STR_AUTH_INFO), &gEfiAuthenticationInfoProtocolGuid, NULL},\r
689 {STRING_TOKEN(STR_HASH_SB), &gEfiHashServiceBindingProtocolGuid, NULL},\r
690 {STRING_TOKEN(STR_HASH), &gEfiHashProtocolGuid, NULL},\r
691 {STRING_TOKEN(STR_HII_FONT), &gEfiHiiFontProtocolGuid, NULL},\r
692 {STRING_TOKEN(STR_HII_STRING), &gEfiHiiStringProtocolGuid, NULL},\r
693 {STRING_TOKEN(STR_HII_IMAGE), &gEfiHiiImageProtocolGuid, NULL},\r
694 {STRING_TOKEN(STR_HII_DATABASE), &gEfiHiiDatabaseProtocolGuid, NULL},\r
695 {STRING_TOKEN(STR_HII_CONFIG_ROUT), &gEfiHiiConfigRoutingProtocolGuid, NULL},\r
696 {STRING_TOKEN(STR_HII_CONFIG_ACC), &gEfiHiiConfigAccessProtocolGuid, NULL},\r
697 {STRING_TOKEN(STR_HII_FORM_BROWSER2), &gEfiFormBrowser2ProtocolGuid, NULL},\r
698 {STRING_TOKEN(STR_DRIVER_FAM_OVERRIDE), &gEfiDriverFamilyOverrideProtocolGuid, NULL},\r
699 {STRING_TOKEN(STR_PCD), &gPcdProtocolGuid, NULL},\r
700 {STRING_TOKEN(STR_TCG), &gEfiTcgProtocolGuid, NULL},\r
701 {STRING_TOKEN(STR_HII_PACKAGE_LIST), &gEfiHiiPackageListProtocolGuid, NULL},\r
702\r
703//\r
704// the ones under this are deprecated by the current UEFI Spec, but may be found anyways...\r
705//\r
706 {STRING_TOKEN(STR_SHELL_INTERFACE), &gEfiShellInterfaceGuid, NULL},\r
707 {STRING_TOKEN(STR_SHELL_ENV2), &gEfiShellEnvironment2Guid, NULL},\r
708 {STRING_TOKEN(STR_SHELL_ENV), &gEfiShellEnvironment2Guid, NULL},\r
709 {STRING_TOKEN(STR_DEVICE_IO), &gEfiDeviceIoProtocolGuid, NULL},\r
710 {STRING_TOKEN(STR_UGA_DRAW), &gEfiUgaDrawProtocolGuid, NULL},\r
711 {STRING_TOKEN(STR_UGA_IO), &gEfiUgaIoProtocolGuid, NULL},\r
712 {STRING_TOKEN(STR_ESP), &gEfiPartTypeSystemPartGuid, NULL},\r
713 {STRING_TOKEN(STR_GPT_NBR), &gEfiPartTypeLegacyMbrGuid, NULL},\r
714 {STRING_TOKEN(STR_DRIVER_CONFIG), &gEfiDriverConfigurationProtocolGuid, NULL},\r
715 {STRING_TOKEN(STR_DRIVER_CONFIG2), &gEfiDriverConfiguration2ProtocolGuid, NULL},\r
716\r
717//\r
718// the ones under this are GUID identified structs, not protocols\r
719//\r
720 {STRING_TOKEN(STR_FILE_INFO), &gEfiFileInfoGuid, NULL},\r
721 {STRING_TOKEN(STR_FILE_SYS_INFO), &gEfiFileSystemInfoGuid, NULL},\r
722\r
723//\r
724// the ones under this are misc GUIDS.\r
725//\r
726 {STRING_TOKEN(STR_EFI_GLOBAL_VARIABLE), &gEfiGlobalVariableGuid, NULL},\r
727\r
728//\r
729// UEFI 2.2\r
730//\r
731 {STRING_TOKEN(STR_IP6_SB), &gEfiIp6ServiceBindingProtocolGuid, NULL},\r
732 {STRING_TOKEN(STR_IP6), &gEfiIp6ProtocolGuid, NULL},\r
733 {STRING_TOKEN(STR_IP6_CONFIG), &gEfiIp6ConfigProtocolGuid, NULL},\r
734 {STRING_TOKEN(STR_MTFTP6_SB), &gEfiMtftp6ServiceBindingProtocolGuid, NULL},\r
735 {STRING_TOKEN(STR_MTFTP6), &gEfiMtftp6ProtocolGuid, NULL},\r
736 {STRING_TOKEN(STR_DHCP6_SB), &gEfiDhcp6ServiceBindingProtocolGuid, NULL},\r
737 {STRING_TOKEN(STR_DHCP6), &gEfiDhcp6ProtocolGuid, NULL},\r
738 {STRING_TOKEN(STR_UDP6_SB), &gEfiUdp6ServiceBindingProtocolGuid, NULL},\r
739 {STRING_TOKEN(STR_UDP6), &gEfiUdp6ProtocolGuid, NULL},\r
740 {STRING_TOKEN(STR_TCP6_SB), &gEfiTcp6ServiceBindingProtocolGuid, NULL},\r
741 {STRING_TOKEN(STR_TCP6), &gEfiTcp6ProtocolGuid, NULL},\r
742 {STRING_TOKEN(STR_VLAN_CONFIG), &gEfiVlanConfigProtocolGuid, NULL},\r
743 {STRING_TOKEN(STR_EAP), &gEfiEapProtocolGuid, NULL},\r
744 {STRING_TOKEN(STR_EAP_MGMT), &gEfiEapManagementProtocolGuid, NULL},\r
745 {STRING_TOKEN(STR_FTP4_SB), &gEfiFtp4ServiceBindingProtocolGuid, NULL},\r
746 {STRING_TOKEN(STR_FTP4), &gEfiFtp4ProtocolGuid, NULL},\r
747 {STRING_TOKEN(STR_IP_SEC_CONFIG), &gEfiIpSecConfigProtocolGuid, NULL},\r
748 {STRING_TOKEN(STR_DH), &gEfiDriverHealthProtocolGuid, NULL},\r
749 {STRING_TOKEN(STR_DEF_IMG_LOAD), &gEfiDeferredImageLoadProtocolGuid, NULL},\r
750 {STRING_TOKEN(STR_USER_CRED), &gEfiUserCredentialProtocolGuid, NULL},\r
751 {STRING_TOKEN(STR_USER_MNGR), &gEfiUserManagerProtocolGuid, NULL},\r
752 {STRING_TOKEN(STR_ATA_PASS_THRU), &gEfiAtaPassThruProtocolGuid, NULL},\r
753\r
754//\r
755// UEFI 2.3\r
756//\r
757 {STRING_TOKEN(STR_FW_MGMT), &gEfiFirmwareManagementProtocolGuid, NULL},\r
758 {STRING_TOKEN(STR_IP_SEC), &gEfiIpSecProtocolGuid, NULL},\r
759 {STRING_TOKEN(STR_IP_SEC2), &gEfiIpSec2ProtocolGuid, NULL},\r
760\r
761//\r
762// UEFI 2.3.1\r
763//\r
764 {STRING_TOKEN(STR_KMS), &gEfiKmsProtocolGuid, NULL},\r
765 {STRING_TOKEN(STR_BLK_IO2), &gEfiBlockIo2ProtocolGuid, NULL},\r
766 {STRING_TOKEN(STR_SSC), &gEfiStorageSecurityCommandProtocolGuid, NULL},\r
767 {STRING_TOKEN(STR_UCRED2), &gEfiUserCredential2ProtocolGuid, NULL},\r
768\r
769//\r
770// UEFI 2.4\r
771//\r
772 {STRING_TOKEN(STR_DISK_IO2), &gEfiDiskIo2ProtocolGuid, NULL},\r
773 {STRING_TOKEN(STR_ADAPTER_INFO), &gEfiAdapterInformationProtocolGuid, NULL},\r
774\r
775//\r
776// PI Spec ones\r
777//\r
778 {STRING_TOKEN(STR_IDE_CONT_INIT), &gEfiIdeControllerInitProtocolGuid, NULL},\r
779\r
780//\r
781// UEFI Shell Spec 2.0\r
782//\r
783 {STRING_TOKEN(STR_SHELL_PARAMETERS), &gEfiShellParametersProtocolGuid, NULL},\r
784 {STRING_TOKEN(STR_SHELL), &gEfiShellProtocolGuid, NULL},\r
785\r
786//\r
787// UEFI Shell Spec 2.1\r
788//\r
789 {STRING_TOKEN(STR_SHELL_DYNAMIC), &gEfiShellDynamicCommandProtocolGuid, NULL},\r
790\r
791//\r
792// terminator\r
793//\r
794 {STRING_TOKEN(STR_UNKNOWN_DEVICE), NULL, NULL},\r
795};\r
796\r
797/**\r
798 Function to get the node for a protocol or struct from it's GUID.\r
799\r
800 if Guid is NULL, then ASSERT.\r
801\r
802 @param[in] Guid The GUID to look for the name of.\r
803\r
804 @return The node.\r
805**/\r
806CONST GUID_INFO_BLOCK *\r
807EFIAPI\r
808InternalShellGetNodeFromGuid(\r
809 IN CONST EFI_GUID* Guid\r
810 )\r
811{\r
812 CONST GUID_INFO_BLOCK *ListWalker;\r
813 UINTN LoopCount;\r
814\r
815 ASSERT(Guid != NULL);\r
816\r
817 for (LoopCount = 0, ListWalker = GuidList; GuidList != NULL && LoopCount < GuidListCount; LoopCount++, ListWalker++) {\r
818 if (CompareGuid(ListWalker->GuidId, Guid)) {\r
819 return (ListWalker);\r
820 }\r
821 }\r
822\r
823 if (PcdGetBool(PcdShellIncludeNtGuids)) {\r
824 for (ListWalker = mGuidStringListNT ; ListWalker != NULL && ListWalker->GuidId != NULL ; ListWalker++) {\r
825 if (CompareGuid(ListWalker->GuidId, Guid)) {\r
826 return (ListWalker);\r
827 }\r
828 }\r
829 }\r
830 for (ListWalker = mGuidStringList ; ListWalker != NULL && ListWalker->GuidId != NULL ; ListWalker++) {\r
831 if (CompareGuid(ListWalker->GuidId, Guid)) {\r
832 return (ListWalker);\r
833 }\r
834 }\r
835 return (NULL);\r
836}\r
837\r
838/**\r
839Function to add a new GUID/Name mapping.\r
840\r
841@param[in] Guid The Guid\r
842@param[in] NameID The STRING id of the HII string to use\r
843@param[in] DumpFunc The pointer to the dump function\r
844\r
845\r
846@retval EFI_SUCCESS The operation was sucessful\r
847@retval EFI_OUT_OF_RESOURCES A memory allocation failed\r
848@retval EFI_INVALID_PARAMETER Guid NameId was invalid\r
849**/\r
850STATIC\r
851EFI_STATUS\r
852EFIAPI\r
853InsertNewGuidNameMapping(\r
854 IN CONST EFI_GUID *Guid,\r
855 IN CONST EFI_STRING_ID NameID,\r
856 IN CONST DUMP_PROTOCOL_INFO DumpFunc OPTIONAL\r
857 )\r
858{\r
859 ASSERT(Guid != NULL);\r
860 ASSERT(NameID != 0);\r
861\r
862 GuidList = ReallocatePool(GuidListCount * sizeof(GUID_INFO_BLOCK), GuidListCount+1 * sizeof(GUID_INFO_BLOCK), GuidList);\r
863 if (GuidList == NULL) {\r
864 GuidListCount = 0;\r
865 return (EFI_OUT_OF_RESOURCES);\r
866 }\r
867 GuidListCount++;\r
868\r
869 GuidList[GuidListCount - 1].GuidId = AllocateCopyPool(sizeof(EFI_GUID), Guid);\r
870 GuidList[GuidListCount - 1].StringId = NameID;\r
871 GuidList[GuidListCount - 1].DumpInfo = DumpFunc;\r
872\r
873 if (GuidList[GuidListCount - 1].GuidId == NULL) {\r
874 return (EFI_OUT_OF_RESOURCES);\r
875 }\r
876\r
877 return (EFI_SUCCESS);\r
878}\r
879\r
880/**\r
881 Function to add a new GUID/Name mapping.\r
882\r
883 This cannot overwrite an existing mapping.\r
884\r
885 @param[in] Guid The Guid\r
886 @param[in] TheName The Guid's name\r
887 @param[in] Lang RFC4646 language code list or NULL\r
888\r
889 @retval EFI_SUCCESS The operation was sucessful\r
890 @retval EFI_ACCESS_DENIED There was a duplicate\r
891 @retval EFI_OUT_OF_RESOURCES A memory allocation failed\r
892 @retval EFI_INVALID_PARAMETER Guid or TheName was NULL\r
893**/\r
894EFI_STATUS\r
895EFIAPI\r
896AddNewGuidNameMapping(\r
897 IN CONST EFI_GUID *Guid,\r
898 IN CONST CHAR16 *TheName,\r
899 IN CONST CHAR8 *Lang OPTIONAL\r
900 )\r
901{\r
902 CONST GUID_INFO_BLOCK *Temp;\r
903 EFI_STRING_ID NameID;\r
904\r
905 HandleParsingHiiInit();\r
906\r
907 if (Guid == NULL || TheName == NULL){\r
908 return (EFI_INVALID_PARAMETER);\r
909 }\r
910\r
911 if ((Temp = InternalShellGetNodeFromGuid(Guid)) != NULL) {\r
912 return (EFI_ACCESS_DENIED);\r
913 }\r
914\r
915 NameID = HiiSetString(mHandleParsingHiiHandle, 0, (CHAR16*)TheName, Lang);\r
916 if (NameID == 0) {\r
917 return (EFI_OUT_OF_RESOURCES);\r
918 }\r
919\r
920 return (InsertNewGuidNameMapping(Guid, NameID, NULL));\r
921}\r
922\r
923/**\r
924 Function to get the name of a protocol or struct from it's GUID.\r
925\r
926 if Guid is NULL, then ASSERT.\r
927\r
928 @param[in] Guid The GUID to look for the name of.\r
929 @param[in] Lang The language to use.\r
930\r
931 @return pointer to string of the name. The caller\r
932 is responsible to free this memory.\r
933**/\r
934CHAR16*\r
935EFIAPI\r
936GetStringNameFromGuid(\r
937 IN CONST EFI_GUID *Guid,\r
938 IN CONST CHAR8 *Lang OPTIONAL\r
939 )\r
940{\r
941 CONST GUID_INFO_BLOCK *Id;\r
942\r
943 HandleParsingHiiInit();\r
944\r
945 Id = InternalShellGetNodeFromGuid(Guid);\r
946 return (HiiGetString(mHandleParsingHiiHandle, Id==NULL?STRING_TOKEN(STR_UNKNOWN_DEVICE):Id->StringId, Lang));\r
947}\r
948\r
949/**\r
950 Function to dump protocol information from a handle.\r
951\r
952 This function will return a allocated string buffer containing the\r
953 information. The caller is responsible for freeing the memory.\r
954\r
955 If Guid is NULL, ASSERT().\r
956 If TheHandle is NULL, ASSERT().\r
957\r
958 @param[in] TheHandle The handle to dump information from.\r
959 @param[in] Guid The GUID of the protocol to dump.\r
960 @param[in] Verbose TRUE for extra info. FALSE otherwise.\r
961\r
962 @return The pointer to string.\r
963 @retval NULL An error was encountered.\r
964**/\r
965CHAR16*\r
966EFIAPI\r
967GetProtocolInformationDump(\r
968 IN CONST EFI_HANDLE TheHandle,\r
969 IN CONST EFI_GUID *Guid,\r
970 IN CONST BOOLEAN Verbose\r
971 )\r
972{\r
973 CONST GUID_INFO_BLOCK *Id;\r
974\r
975 ASSERT(TheHandle != NULL);\r
976 ASSERT(Guid != NULL);\r
977\r
978 if (TheHandle == NULL || Guid == NULL) {\r
979 return (NULL);\r
980 }\r
981\r
982 Id = InternalShellGetNodeFromGuid(Guid);\r
983 if (Id != NULL && Id->DumpInfo != NULL) {\r
984 return (Id->DumpInfo(TheHandle, Verbose));\r
985 }\r
986 return (NULL);\r
987}\r
988\r
989/**\r
990 Function to get the Guid for a protocol or struct based on it's string name.\r
991\r
992 do not modify the returned Guid.\r
993\r
994 @param[in] Name The pointer to the string name.\r
995 @param[in] Lang The pointer to the language code.\r
996 @param[out] Guid The pointer to the Guid.\r
997\r
998 @retval EFI_SUCCESS The operation was sucessful.\r
999**/\r
1000EFI_STATUS\r
1001EFIAPI\r
1002GetGuidFromStringName(\r
1003 IN CONST CHAR16 *Name,\r
1004 IN CONST CHAR8 *Lang OPTIONAL,\r
1005 OUT EFI_GUID **Guid\r
1006 )\r
1007{\r
1008 CONST GUID_INFO_BLOCK *ListWalker;\r
1009 CHAR16 *String;\r
1010 UINTN LoopCount;\r
1011\r
1012 HandleParsingHiiInit();\r
1013\r
1014 ASSERT(Guid != NULL);\r
1015 if (Guid == NULL) {\r
1016 return (EFI_INVALID_PARAMETER);\r
1017 }\r
1018 *Guid = NULL;\r
1019\r
1020 if (PcdGetBool(PcdShellIncludeNtGuids)) {\r
1021 for (ListWalker = mGuidStringListNT ; ListWalker != NULL && ListWalker->GuidId != NULL ; ListWalker++) {\r
1022 String = HiiGetString(mHandleParsingHiiHandle, ListWalker->StringId, Lang);\r
1023 if (Name != NULL && String != NULL && StringNoCaseCompare (&Name, &String) == 0) {\r
1024 *Guid = ListWalker->GuidId;\r
1025 }\r
1026 SHELL_FREE_NON_NULL(String);\r
1027 if (*Guid != NULL) {\r
1028 return (EFI_SUCCESS);\r
1029 }\r
1030 }\r
1031 }\r
1032 for (ListWalker = mGuidStringList ; ListWalker != NULL && ListWalker->GuidId != NULL ; ListWalker++) {\r
1033 String = HiiGetString(mHandleParsingHiiHandle, ListWalker->StringId, Lang);\r
1034 if (Name != NULL && String != NULL && StringNoCaseCompare (&Name, &String) == 0) {\r
1035 *Guid = ListWalker->GuidId;\r
1036 }\r
1037 SHELL_FREE_NON_NULL(String);\r
1038 if (*Guid != NULL) {\r
1039 return (EFI_SUCCESS);\r
1040 }\r
1041 }\r
1042\r
1043 for (LoopCount = 0, ListWalker = GuidList; GuidList != NULL && LoopCount < GuidListCount; LoopCount++, ListWalker++) {\r
1044 String = HiiGetString(mHandleParsingHiiHandle, ListWalker->StringId, Lang);\r
1045 if (Name != NULL && String != NULL && StringNoCaseCompare (&Name, &String) == 0) {\r
1046 *Guid = ListWalker->GuidId;\r
1047 }\r
1048 SHELL_FREE_NON_NULL(String);\r
1049 if (*Guid != NULL) {\r
1050 return (EFI_SUCCESS);\r
1051 }\r
1052 }\r
1053\r
1054 return (EFI_NOT_FOUND);\r
1055}\r
1056\r
1057/**\r
1058 Get best support language for this driver.\r
1059 \r
1060 First base on the user input language to search, second base on the current \r
1061 platform used language to search, third get the first language from the \r
1062 support language list. The caller need to free the buffer of the best language.\r
1063\r
1064 @param[in] SupportedLanguages The support languages for this driver.\r
1065 @param[in] InputLanguage The user input language.\r
1066 @param[in] Iso639Language Whether get language for ISO639.\r
1067\r
1068 @return The best support language for this driver.\r
1069**/\r
1070CHAR8 *\r
1071EFIAPI\r
1072GetBestLanguageForDriver (\r
1073 IN CONST CHAR8 *SupportedLanguages,\r
1074 IN CONST CHAR8 *InputLanguage,\r
1075 IN BOOLEAN Iso639Language\r
1076 )\r
1077{\r
1078 CHAR8 *LanguageVariable;\r
1079 CHAR8 *BestLanguage;\r
1080\r
1081 LanguageVariable = GetVariable (Iso639Language ? L"Lang" : L"PlatformLang", &gEfiGlobalVariableGuid);\r
1082\r
1083 BestLanguage = GetBestLanguage(\r
1084 SupportedLanguages,\r
1085 Iso639Language,\r
1086 (InputLanguage != NULL) ? InputLanguage : "",\r
1087 (LanguageVariable != NULL) ? LanguageVariable : "",\r
1088 SupportedLanguages,\r
1089 NULL\r
1090 );\r
1091\r
1092 if (LanguageVariable != NULL) {\r
1093 FreePool (LanguageVariable);\r
1094 }\r
1095\r
1096 return BestLanguage;\r
1097}\r
1098\r
1099/**\r
1100 Function to retrieve the driver name (if possible) from the ComponentName or\r
1101 ComponentName2 protocol\r
1102\r
1103 @param[in] TheHandle The driver handle to get the name of.\r
1104 @param[in] Language The language to use.\r
1105\r
1106 @retval NULL The name could not be found.\r
1107 @return A pointer to the string name. Do not de-allocate the memory.\r
1108**/\r
1109CONST CHAR16*\r
1110EFIAPI\r
1111GetStringNameFromHandle(\r
1112 IN CONST EFI_HANDLE TheHandle,\r
1113 IN CONST CHAR8 *Language\r
1114 )\r
1115{\r
1116 EFI_COMPONENT_NAME2_PROTOCOL *CompNameStruct;\r
1117 EFI_STATUS Status;\r
1118 CHAR16 *RetVal;\r
1119 CHAR8 *BestLang;\r
1120\r
1121 BestLang = NULL;\r
1122\r
1123 Status = gBS->OpenProtocol(\r
1124 TheHandle,\r
1125 &gEfiComponentName2ProtocolGuid,\r
1126 (VOID**)&CompNameStruct,\r
1127 gImageHandle,\r
1128 NULL,\r
1129 EFI_OPEN_PROTOCOL_GET_PROTOCOL);\r
1130 if (!EFI_ERROR(Status)) {\r
1131 BestLang = GetBestLanguageForDriver (CompNameStruct->SupportedLanguages, Language, FALSE);\r
1132 Status = CompNameStruct->GetDriverName(CompNameStruct, BestLang, &RetVal);\r
1133 if (BestLang != NULL) {\r
1134 FreePool (BestLang);\r
1135 BestLang = NULL;\r
1136 }\r
1137 if (!EFI_ERROR(Status)) {\r
1138 return (RetVal);\r
1139 }\r
1140 }\r
1141 Status = gBS->OpenProtocol(\r
1142 TheHandle,\r
1143 &gEfiComponentNameProtocolGuid,\r
1144 (VOID**)&CompNameStruct,\r
1145 gImageHandle,\r
1146 NULL,\r
1147 EFI_OPEN_PROTOCOL_GET_PROTOCOL);\r
1148 if (!EFI_ERROR(Status)) {\r
1149 BestLang = GetBestLanguageForDriver (CompNameStruct->SupportedLanguages, Language, FALSE);\r
1150 Status = CompNameStruct->GetDriverName(CompNameStruct, BestLang, &RetVal);\r
1151 if (BestLang != NULL) {\r
1152 FreePool (BestLang);\r
1153 }\r
1154 if (!EFI_ERROR(Status)) {\r
1155 return (RetVal);\r
1156 }\r
1157 }\r
1158 return (NULL);\r
1159}\r
1160\r
1161/**\r
1162 Function to initialize the file global mHandleList object for use in\r
1163 vonverting handles to index and index to handle.\r
1164\r
1165 @retval EFI_SUCCESS The operation was successful.\r
1166**/\r
1167EFI_STATUS\r
1168EFIAPI\r
1169InternalShellInitHandleList(\r
1170 VOID\r
1171 )\r
1172{\r
1173 EFI_STATUS Status;\r
1174 EFI_HANDLE *HandleBuffer;\r
1175 UINTN HandleCount;\r
1176 HANDLE_LIST *ListWalker;\r
1177\r
1178 if (mHandleList.NextIndex != 0) {\r
1179 return EFI_SUCCESS;\r
1180 }\r
1181 InitializeListHead(&mHandleList.List.Link);\r
1182 mHandleList.NextIndex = 1;\r
1183 Status = gBS->LocateHandleBuffer (\r
1184 AllHandles,\r
1185 NULL,\r
1186 NULL,\r
1187 &HandleCount,\r
1188 &HandleBuffer\r
1189 );\r
1190 ASSERT_EFI_ERROR(Status);\r
1191 if (EFI_ERROR(Status)) {\r
1192 return (Status);\r
1193 }\r
1194 for (mHandleList.NextIndex = 1 ; mHandleList.NextIndex <= HandleCount ; mHandleList.NextIndex++){\r
1195 ListWalker = AllocateZeroPool(sizeof(HANDLE_LIST));\r
1196 ASSERT(ListWalker != NULL);\r
1197 ListWalker->TheHandle = HandleBuffer[mHandleList.NextIndex-1];\r
1198 ListWalker->TheIndex = mHandleList.NextIndex;\r
1199 InsertTailList(&mHandleList.List.Link,&ListWalker->Link);\r
1200 }\r
1201 FreePool(HandleBuffer);\r
1202 return (EFI_SUCCESS);\r
1203}\r
1204\r
1205/**\r
1206 Function to retrieve the human-friendly index of a given handle. If the handle\r
1207 does not have a index one will be automatically assigned. The index value is valid\r
1208 until the termination of the shell application.\r
1209\r
1210 @param[in] TheHandle The handle to retrieve an index for.\r
1211\r
1212 @retval 0 A memory allocation failed.\r
1213 @return The index of the handle.\r
1214\r
1215**/\r
1216UINTN\r
1217EFIAPI\r
1218ConvertHandleToHandleIndex(\r
1219 IN CONST EFI_HANDLE TheHandle\r
1220 )\r
1221{\r
1222 EFI_STATUS Status;\r
1223 EFI_GUID **ProtocolBuffer;\r
1224 UINTN ProtocolCount;\r
1225 HANDLE_LIST *ListWalker;\r
1226\r
1227 if (TheHandle == NULL) {\r
1228 return 0;\r
1229 }\r
1230\r
1231 InternalShellInitHandleList();\r
1232\r
1233 for (ListWalker = (HANDLE_LIST*)GetFirstNode(&mHandleList.List.Link)\r
1234 ; !IsNull(&mHandleList.List.Link,&ListWalker->Link)\r
1235 ; ListWalker = (HANDLE_LIST*)GetNextNode(&mHandleList.List.Link,&ListWalker->Link)\r
1236 ){\r
1237 if (ListWalker->TheHandle == TheHandle) {\r
1238 //\r
1239 // Verify that TheHandle is still present in the Handle Database\r
1240 //\r
1241 Status = gBS->ProtocolsPerHandle(TheHandle, &ProtocolBuffer, &ProtocolCount);\r
1242 if (EFI_ERROR (Status)) {\r
1243 //\r
1244 // TheHandle is not present in the Handle Database, so delete from the handle list\r
1245 //\r
1246 RemoveEntryList (&ListWalker->Link);\r
1247 return 0;\r
1248 }\r
1249 FreePool (ProtocolBuffer);\r
1250 return (ListWalker->TheIndex);\r
1251 }\r
1252 }\r
1253\r
1254 //\r
1255 // Verify that TheHandle is valid handle\r
1256 //\r
1257 Status = gBS->ProtocolsPerHandle(TheHandle, &ProtocolBuffer, &ProtocolCount);\r
1258 if (EFI_ERROR (Status)) {\r
1259 //\r
1260 // TheHandle is not valid, so do not add to handle list\r
1261 //\r
1262 return 0;\r
1263 }\r
1264 FreePool (ProtocolBuffer);\r
1265\r
1266 ListWalker = AllocateZeroPool(sizeof(HANDLE_LIST));\r
1267 ASSERT(ListWalker != NULL);\r
1268 ListWalker->TheHandle = TheHandle;\r
1269 ListWalker->TheIndex = mHandleList.NextIndex++;\r
1270 InsertTailList(&mHandleList.List.Link,&ListWalker->Link);\r
1271 return (ListWalker->TheIndex);\r
1272}\r
1273\r
1274\r
1275\r
1276/**\r
1277 Function to retrieve the EFI_HANDLE from the human-friendly index.\r
1278\r
1279 @param[in] TheIndex The index to retrieve the EFI_HANDLE for.\r
1280\r
1281 @retval NULL The index was invalid.\r
1282 @return The EFI_HANDLE that index represents.\r
1283\r
1284**/\r
1285EFI_HANDLE\r
1286EFIAPI\r
1287ConvertHandleIndexToHandle(\r
1288 IN CONST UINTN TheIndex\r
1289 )\r
1290{\r
1291 EFI_STATUS Status;\r
1292 EFI_GUID **ProtocolBuffer;\r
1293 UINTN ProtocolCount;\r
1294 HANDLE_LIST *ListWalker;\r
1295\r
1296 InternalShellInitHandleList();\r
1297\r
1298 if (TheIndex >= mHandleList.NextIndex) {\r
1299 return NULL;\r
1300 }\r
1301\r
1302 for (ListWalker = (HANDLE_LIST*)GetFirstNode(&mHandleList.List.Link)\r
1303 ; !IsNull(&mHandleList.List.Link,&ListWalker->Link)\r
1304 ; ListWalker = (HANDLE_LIST*)GetNextNode(&mHandleList.List.Link,&ListWalker->Link)\r
1305 ){\r
1306 if (ListWalker->TheIndex == TheIndex && ListWalker->TheHandle != NULL) {\r
1307 //\r
1308 // Verify that LinkWalker->TheHandle is valid handle\r
1309 //\r
1310 Status = gBS->ProtocolsPerHandle(ListWalker->TheHandle, &ProtocolBuffer, &ProtocolCount);\r
1311 if (EFI_ERROR (Status)) {\r
1312 //\r
1313 // TheHandle is not valid, so do not add to handle list\r
1314 //\r
1315 ListWalker->TheHandle = NULL;\r
1316 }\r
1317 return (ListWalker->TheHandle);\r
1318 }\r
1319 }\r
1320 return NULL;\r
1321}\r
1322\r
1323/**\r
1324 Gets all the related EFI_HANDLEs based on the mask supplied.\r
1325\r
1326 This function scans all EFI_HANDLES in the UEFI environment's handle database\r
1327 and returns the ones with the specified relationship (Mask) to the specified\r
1328 controller handle.\r
1329\r
1330 If both DriverBindingHandle and ControllerHandle are NULL, then ASSERT.\r
1331 If MatchingHandleCount is NULL, then ASSERT.\r
1332\r
1333 If MatchingHandleBuffer is not NULL upon a successful return the memory must be\r
1334 caller freed.\r
1335\r
1336 @param[in] DriverBindingHandle The handle with Driver Binding protocol on it.\r
1337 @param[in] ControllerHandle The handle with Device Path protocol on it.\r
1338 @param[in] MatchingHandleCount The pointer to UINTN that specifies the number of HANDLES in\r
1339 MatchingHandleBuffer.\r
1340 @param[out] MatchingHandleBuffer On a successful return, a buffer of MatchingHandleCount\r
1341 EFI_HANDLEs with a terminating NULL EFI_HANDLE.\r
1342 @param[out] HandleType An array of type information.\r
1343\r
1344 @retval EFI_SUCCESS The operation was successful, and any related handles\r
1345 are in MatchingHandleBuffer.\r
1346 @retval EFI_NOT_FOUND No matching handles were found.\r
1347 @retval EFI_INVALID_PARAMETER A parameter was invalid or out of range.\r
1348**/\r
1349EFI_STATUS\r
1350EFIAPI\r
1351ParseHandleDatabaseByRelationshipWithType (\r
1352 IN CONST EFI_HANDLE DriverBindingHandle OPTIONAL,\r
1353 IN CONST EFI_HANDLE ControllerHandle OPTIONAL,\r
1354 IN UINTN *HandleCount,\r
1355 OUT EFI_HANDLE **HandleBuffer,\r
1356 OUT UINTN **HandleType\r
1357 )\r
1358{\r
1359 EFI_STATUS Status;\r
1360 UINTN HandleIndex;\r
1361 EFI_GUID **ProtocolGuidArray;\r
1362 UINTN ArrayCount;\r
1363 UINTN ProtocolIndex;\r
1364 EFI_OPEN_PROTOCOL_INFORMATION_ENTRY *OpenInfo;\r
1365 UINTN OpenInfoCount;\r
1366 UINTN OpenInfoIndex;\r
1367 UINTN ChildIndex;\r
1368 INTN DriverBindingHandleIndex;\r
1369\r
1370 ASSERT(HandleCount != NULL);\r
1371 ASSERT(HandleBuffer != NULL);\r
1372 ASSERT(HandleType != NULL);\r
1373 ASSERT(DriverBindingHandle != NULL || ControllerHandle != NULL);\r
1374\r
1375 *HandleCount = 0;\r
1376 *HandleBuffer = NULL;\r
1377 *HandleType = NULL;\r
1378\r
1379 //\r
1380 // Retrieve the list of all handles from the handle database\r
1381 //\r
1382 Status = gBS->LocateHandleBuffer (\r
1383 AllHandles,\r
1384 NULL,\r
1385 NULL,\r
1386 HandleCount,\r
1387 HandleBuffer\r
1388 );\r
1389 if (EFI_ERROR (Status)) {\r
1390 return (Status);\r
1391 }\r
1392\r
1393 *HandleType = AllocateZeroPool (*HandleCount * sizeof (UINTN));\r
1394 ASSERT(*HandleType != NULL);\r
1395\r
1396 DriverBindingHandleIndex = -1;\r
1397 for (HandleIndex = 0; HandleIndex < *HandleCount; HandleIndex++) {\r
1398 if (DriverBindingHandle != NULL && (*HandleBuffer)[HandleIndex] == DriverBindingHandle) {\r
1399 DriverBindingHandleIndex = (INTN)HandleIndex;\r
1400 }\r
1401 }\r
1402\r
1403 for (HandleIndex = 0; HandleIndex < *HandleCount; HandleIndex++) {\r
1404 //\r
1405 // Retrieve the list of all the protocols on each handle\r
1406 //\r
1407 Status = gBS->ProtocolsPerHandle (\r
1408 (*HandleBuffer)[HandleIndex],\r
1409 &ProtocolGuidArray,\r
1410 &ArrayCount\r
1411 );\r
1412 if (EFI_ERROR (Status)) {\r
1413 continue;\r
1414 }\r
1415\r
1416 for (ProtocolIndex = 0; ProtocolIndex < ArrayCount; ProtocolIndex++) {\r
1417\r
1418 //\r
1419 // Set the bit describing what this handle has\r
1420 //\r
1421 if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiLoadedImageProtocolGuid) ) {\r
1422 (*HandleType)[HandleIndex] |= (UINTN)HR_IMAGE_HANDLE;\r
1423 } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiDriverBindingProtocolGuid) ) {\r
1424 (*HandleType)[HandleIndex] |= (UINTN)HR_DRIVER_BINDING_HANDLE;\r
1425 } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiDriverConfiguration2ProtocolGuid)) {\r
1426 (*HandleType)[HandleIndex] |= (UINTN)HR_DRIVER_CONFIGURATION_HANDLE;\r
1427 } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiDriverConfigurationProtocolGuid) ) {\r
1428 (*HandleType)[HandleIndex] |= (UINTN)HR_DRIVER_CONFIGURATION_HANDLE;\r
1429 } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiDriverDiagnostics2ProtocolGuid) ) {\r
1430 (*HandleType)[HandleIndex] |= (UINTN)HR_DRIVER_DIAGNOSTICS_HANDLE;\r
1431 } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiDriverDiagnosticsProtocolGuid) ) {\r
1432 (*HandleType)[HandleIndex] |= (UINTN)HR_DRIVER_DIAGNOSTICS_HANDLE;\r
1433 } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiComponentName2ProtocolGuid) ) {\r
1434 (*HandleType)[HandleIndex] |= (UINTN)HR_COMPONENT_NAME_HANDLE;\r
1435 } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiComponentNameProtocolGuid) ) {\r
1436 (*HandleType)[HandleIndex] |= (UINTN)HR_COMPONENT_NAME_HANDLE;\r
1437 } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiDevicePathProtocolGuid) ) {\r
1438 (*HandleType)[HandleIndex] |= (UINTN)HR_DEVICE_HANDLE;\r
1439 } else {\r
1440 DEBUG_CODE_BEGIN();\r
1441 ASSERT((*HandleType)[HandleIndex] == (*HandleType)[HandleIndex]);\r
1442 DEBUG_CODE_END();\r
1443 }\r
1444 //\r
1445 // Retrieve the list of agents that have opened each protocol\r
1446 //\r
1447 Status = gBS->OpenProtocolInformation (\r
1448 (*HandleBuffer)[HandleIndex],\r
1449 ProtocolGuidArray[ProtocolIndex],\r
1450 &OpenInfo,\r
1451 &OpenInfoCount\r
1452 );\r
1453 if (EFI_ERROR (Status)) {\r
1454 continue;\r
1455 }\r
1456\r
1457 if (ControllerHandle == NULL) {\r
1458 //\r
1459 // ControllerHandle == NULL and DriverBindingHandle != NULL. \r
1460 // Return information on all the controller handles that the driver specified by DriverBindingHandle is managing\r
1461 //\r
1462 for (OpenInfoIndex = 0; OpenInfoIndex < OpenInfoCount; OpenInfoIndex++) {\r
1463 if (OpenInfo[OpenInfoIndex].AgentHandle == DriverBindingHandle && (OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_DRIVER) != 0) {\r
1464 (*HandleType)[HandleIndex] |= (UINTN)(HR_DEVICE_HANDLE | HR_CONTROLLER_HANDLE);\r
1465 if (DriverBindingHandleIndex != -1) {\r
1466 (*HandleType)[DriverBindingHandleIndex] |= (UINTN)HR_DEVICE_DRIVER;\r
1467 }\r
1468 }\r
1469 if (OpenInfo[OpenInfoIndex].AgentHandle == DriverBindingHandle && (OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER) != 0) {\r
1470 (*HandleType)[HandleIndex] |= (UINTN)(HR_DEVICE_HANDLE | HR_CONTROLLER_HANDLE);\r
1471 if (DriverBindingHandleIndex != -1) {\r
1472 (*HandleType)[DriverBindingHandleIndex] |= (UINTN)(HR_BUS_DRIVER | HR_DEVICE_DRIVER);\r
1473 }\r
1474 for (ChildIndex = 0; ChildIndex < *HandleCount; ChildIndex++) {\r
1475 if (OpenInfo[OpenInfoIndex].ControllerHandle == (*HandleBuffer)[ChildIndex]) {\r
1476 (*HandleType)[ChildIndex] |= (UINTN)(HR_DEVICE_HANDLE | HR_CHILD_HANDLE);\r
1477 }\r
1478 }\r
1479 }\r
1480 }\r
1481 }\r
1482 if (DriverBindingHandle == NULL && ControllerHandle != NULL) {\r
1483 if (ControllerHandle == (*HandleBuffer)[HandleIndex]) {\r
1484 (*HandleType)[HandleIndex] |= (UINTN)(HR_DEVICE_HANDLE | HR_CONTROLLER_HANDLE);\r
1485 for (OpenInfoIndex = 0; OpenInfoIndex < OpenInfoCount; OpenInfoIndex++) {\r
1486 if ((OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_DRIVER) != 0) {\r
1487 for (ChildIndex = 0; ChildIndex < *HandleCount; ChildIndex++) {\r
1488 if (OpenInfo[OpenInfoIndex].AgentHandle == (*HandleBuffer)[ChildIndex]) {\r
1489 (*HandleType)[ChildIndex] |= (UINTN)HR_DEVICE_DRIVER;\r
1490 }\r
1491 }\r
1492 }\r
1493 if ((OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER) != 0) {\r
1494 for (ChildIndex = 0; ChildIndex < *HandleCount; ChildIndex++) {\r
1495 if (OpenInfo[OpenInfoIndex].AgentHandle == (*HandleBuffer)[ChildIndex]) {\r
1496 (*HandleType)[ChildIndex] |= (UINTN)(HR_BUS_DRIVER | HR_DEVICE_DRIVER);\r
1497 }\r
1498 if (OpenInfo[OpenInfoIndex].ControllerHandle == (*HandleBuffer)[ChildIndex]) {\r
1499 (*HandleType)[ChildIndex] |= (UINTN)(HR_DEVICE_HANDLE | HR_CHILD_HANDLE);\r
1500 }\r
1501 }\r
1502 }\r
1503 }\r
1504 } else {\r
1505 for (OpenInfoIndex = 0; OpenInfoIndex < OpenInfoCount; OpenInfoIndex++) {\r
1506 if ((OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER) != 0) {\r
1507 if (OpenInfo[OpenInfoIndex].ControllerHandle == ControllerHandle) {\r
1508 (*HandleType)[HandleIndex] |= (UINTN)(HR_DEVICE_HANDLE | HR_PARENT_HANDLE);\r
1509 }\r
1510 }\r
1511 }\r
1512 }\r
1513 }\r
1514 if (DriverBindingHandle != NULL && ControllerHandle != NULL) {\r
1515 if (ControllerHandle == (*HandleBuffer)[HandleIndex]) {\r
1516 (*HandleType)[HandleIndex] |= (UINTN)(HR_DEVICE_HANDLE | HR_CONTROLLER_HANDLE);\r
1517 for (OpenInfoIndex = 0; OpenInfoIndex < OpenInfoCount; OpenInfoIndex++) {\r
1518 if ((OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_DRIVER) != 0) {\r
1519 if (OpenInfo[OpenInfoIndex].AgentHandle == DriverBindingHandle) {\r
1520 if (DriverBindingHandleIndex != -1) {\r
1521 (*HandleType)[DriverBindingHandleIndex] |= (UINTN)HR_DEVICE_DRIVER;\r
1522 }\r
1523 }\r
1524 }\r
1525 if ((OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER) != 0) {\r
1526 if (OpenInfo[OpenInfoIndex].AgentHandle == DriverBindingHandle) {\r
1527 for (ChildIndex = 0; ChildIndex < *HandleCount; ChildIndex++) {\r
1528 if (OpenInfo[OpenInfoIndex].ControllerHandle == (*HandleBuffer)[ChildIndex]) {\r
1529 (*HandleType)[ChildIndex] |= (UINTN)(HR_DEVICE_HANDLE | HR_CHILD_HANDLE);\r
1530 }\r
1531 }\r
1532 }\r
1533\r
1534 for (ChildIndex = 0; ChildIndex < *HandleCount; ChildIndex++) {\r
1535 if (OpenInfo[OpenInfoIndex].AgentHandle == (*HandleBuffer)[ChildIndex]) {\r
1536 (*HandleType)[ChildIndex] |= (UINTN)(HR_BUS_DRIVER | HR_DEVICE_DRIVER);\r
1537 }\r
1538 }\r
1539 }\r
1540 }\r
1541 } else {\r
1542 for (OpenInfoIndex = 0; OpenInfoIndex < OpenInfoCount; OpenInfoIndex++) {\r
1543 if ((OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER) != 0) {\r
1544 if (OpenInfo[OpenInfoIndex].ControllerHandle == ControllerHandle) {\r
1545 (*HandleType)[HandleIndex] |= (UINTN)(HR_DEVICE_HANDLE | HR_PARENT_HANDLE);\r
1546 }\r
1547 }\r
1548 }\r
1549 }\r
1550 }\r
1551 FreePool (OpenInfo);\r
1552 }\r
1553 FreePool (ProtocolGuidArray);\r
1554 }\r
1555 return EFI_SUCCESS;\r
1556}\r
1557\r
1558/**\r
1559 Gets all the related EFI_HANDLEs based on the single EFI_HANDLE and the mask\r
1560 supplied.\r
1561\r
1562 This function will scan all EFI_HANDLES in the UEFI environment's handle database\r
1563 and return all the ones with the specified relationship (Mask) to the specified\r
1564 controller handle.\r
1565\r
1566 If both DriverBindingHandle and ControllerHandle are NULL, then ASSERT.\r
1567 If MatchingHandleCount is NULL, then ASSERT.\r
1568\r
1569 If MatchingHandleBuffer is not NULL upon a sucessful return the memory must be\r
1570 caller freed.\r
1571\r
1572 @param[in] DriverBindingHandle Handle to a object with Driver Binding protocol\r
1573 on it.\r
1574 @param[in] ControllerHandle Handle to a device with Device Path protocol on it.\r
1575 @param[in] Mask Mask of what relationship(s) is desired.\r
1576 @param[in] MatchingHandleCount Poitner to UINTN specifying number of HANDLES in\r
1577 MatchingHandleBuffer.\r
1578 @param[out] MatchingHandleBuffer On a sucessful return a buffer of MatchingHandleCount\r
1579 EFI_HANDLEs and a terminating NULL EFI_HANDLE.\r
1580\r
1581 @retval EFI_SUCCESS The operation was sucessful and any related handles\r
1582 are in MatchingHandleBuffer;\r
1583 @retval EFI_NOT_FOUND No matching handles were found.\r
1584 @retval EFI_INVALID_PARAMETER A parameter was invalid or out of range.\r
1585**/\r
1586EFI_STATUS\r
1587EFIAPI\r
1588ParseHandleDatabaseByRelationship (\r
1589 IN CONST EFI_HANDLE DriverBindingHandle OPTIONAL,\r
1590 IN CONST EFI_HANDLE ControllerHandle OPTIONAL,\r
1591 IN CONST UINTN Mask,\r
1592 IN UINTN *MatchingHandleCount,\r
1593 OUT EFI_HANDLE **MatchingHandleBuffer OPTIONAL\r
1594 )\r
1595{\r
1596 EFI_STATUS Status;\r
1597 UINTN HandleCount;\r
1598 EFI_HANDLE *HandleBuffer;\r
1599 UINTN *HandleType;\r
1600 UINTN HandleIndex;\r
1601\r
1602 ASSERT(MatchingHandleCount != NULL);\r
1603 ASSERT(DriverBindingHandle != NULL || ControllerHandle != NULL);\r
1604\r
1605 if ((Mask & HR_VALID_MASK) != Mask) {\r
1606 return (EFI_INVALID_PARAMETER);\r
1607 }\r
1608\r
1609 if ((Mask & HR_CHILD_HANDLE) != 0 && DriverBindingHandle == NULL) {\r
1610 return (EFI_INVALID_PARAMETER);\r
1611 }\r
1612\r
1613 *MatchingHandleCount = 0;\r
1614 if (MatchingHandleBuffer != NULL) {\r
1615 *MatchingHandleBuffer = NULL;\r
1616 }\r
1617\r
1618 HandleBuffer = NULL;\r
1619 HandleType = NULL;\r
1620\r
1621 Status = ParseHandleDatabaseByRelationshipWithType (\r
1622 DriverBindingHandle,\r
1623 ControllerHandle,\r
1624 &HandleCount,\r
1625 &HandleBuffer,\r
1626 &HandleType\r
1627 );\r
1628 if (!EFI_ERROR (Status)) {\r
1629 //\r
1630 // Count the number of handles that match the attributes in Mask\r
1631 //\r
1632 for (HandleIndex = 0; HandleIndex < HandleCount; HandleIndex++) {\r
1633 if ((HandleType[HandleIndex] & Mask) == Mask) {\r
1634 (*MatchingHandleCount)++;\r
1635 }\r
1636 }\r
1637 //\r
1638 // If no handles match the attributes in Mask then return EFI_NOT_FOUND\r
1639 //\r
1640 if (*MatchingHandleCount == 0) {\r
1641 Status = EFI_NOT_FOUND;\r
1642 } else {\r
1643\r
1644 if (MatchingHandleBuffer == NULL) {\r
1645 //\r
1646 // Someone just wanted the count...\r
1647 //\r
1648 Status = EFI_SUCCESS;\r
1649 } else {\r
1650 //\r
1651 // Allocate a handle buffer for the number of handles that matched the attributes in Mask\r
1652 //\r
1653 *MatchingHandleBuffer = AllocateZeroPool ((*MatchingHandleCount +1)* sizeof (EFI_HANDLE));\r
1654 ASSERT(*MatchingHandleBuffer != NULL);\r
1655\r
1656 for (HandleIndex = 0,*MatchingHandleCount = 0\r
1657 ; HandleIndex < HandleCount\r
1658 ; HandleIndex++\r
1659 ){\r
1660 //\r
1661 // Fill the allocated buffer with the handles that matched the attributes in Mask\r
1662 //\r
1663 if ((HandleType[HandleIndex] & Mask) == Mask) {\r
1664 (*MatchingHandleBuffer)[(*MatchingHandleCount)++] = HandleBuffer[HandleIndex];\r
1665 }\r
1666 }\r
1667\r
1668 //\r
1669 // Make the last one NULL\r
1670 //\r
1671 (*MatchingHandleBuffer)[*MatchingHandleCount] = NULL;\r
1672\r
1673 Status = EFI_SUCCESS;\r
1674 } // MacthingHandleBuffer == NULL (ELSE)\r
1675 } // *MatchingHandleCount == 0 (ELSE)\r
1676 } // no error on ParseHandleDatabaseByRelationshipWithType\r
1677\r
1678 if (HandleBuffer != NULL) {\r
1679 FreePool (HandleBuffer);\r
1680 }\r
1681\r
1682 if (HandleType != NULL) {\r
1683 FreePool (HandleType);\r
1684 }\r
1685\r
1686 return Status;\r
1687}\r
1688\r
1689/**\r
1690 Gets handles for any child controllers of the passed in controller.\r
1691\r
1692 @param[in] ControllerHandle The handle of the "parent controller"\r
1693 @param[in] MatchingHandleCount Pointer to the number of handles in\r
1694 MatchingHandleBuffer on return.\r
1695 @param[out] MatchingHandleBuffer Buffer containing handles on a successful\r
1696 return.\r
1697\r
1698\r
1699 @retval EFI_SUCCESS The operation was sucessful.\r
1700**/\r
1701EFI_STATUS\r
1702EFIAPI\r
1703ParseHandleDatabaseForChildControllers(\r
1704 IN CONST EFI_HANDLE ControllerHandle,\r
1705 IN UINTN *MatchingHandleCount,\r
1706 OUT EFI_HANDLE **MatchingHandleBuffer OPTIONAL\r
1707 )\r
1708{\r
1709 EFI_STATUS Status;\r
1710 UINTN HandleIndex;\r
1711 UINTN DriverBindingHandleCount;\r
1712 EFI_HANDLE *DriverBindingHandleBuffer;\r
1713 UINTN DriverBindingHandleIndex;\r
1714 UINTN ChildControllerHandleCount;\r
1715 EFI_HANDLE *ChildControllerHandleBuffer;\r
1716 UINTN ChildControllerHandleIndex;\r
1717 EFI_HANDLE *HandleBufferForReturn;\r
1718\r
1719 if (MatchingHandleCount == NULL) {\r
1720 return (EFI_INVALID_PARAMETER);\r
1721 }\r
1722 *MatchingHandleCount = 0;\r
1723\r
1724 Status = PARSE_HANDLE_DATABASE_UEFI_DRIVERS (\r
1725 ControllerHandle,\r
1726 &DriverBindingHandleCount,\r
1727 &DriverBindingHandleBuffer\r
1728 );\r
1729 if (EFI_ERROR (Status)) {\r
1730 return Status;\r
1731 }\r
1732\r
1733 //\r
1734 // Get a buffer big enough for all the controllers.\r
1735 //\r
1736 HandleBufferForReturn = GetHandleListByProtocol(NULL);\r
1737 if (HandleBufferForReturn == NULL) {\r
1738 FreePool (DriverBindingHandleBuffer);\r
1739 return (EFI_NOT_FOUND);\r
1740 }\r
1741\r
1742 for (DriverBindingHandleIndex = 0; DriverBindingHandleIndex < DriverBindingHandleCount; DriverBindingHandleIndex++) {\r
1743 Status = PARSE_HANDLE_DATABASE_MANAGED_CHILDREN (\r
1744 DriverBindingHandleBuffer[DriverBindingHandleIndex],\r
1745 ControllerHandle,\r
1746 &ChildControllerHandleCount,\r
1747 &ChildControllerHandleBuffer\r
1748 );\r
1749 if (EFI_ERROR (Status)) {\r
1750 continue;\r
1751 }\r
1752\r
1753 for (ChildControllerHandleIndex = 0;\r
1754 ChildControllerHandleIndex < ChildControllerHandleCount;\r
1755 ChildControllerHandleIndex++\r
1756 ) {\r
1757 for (HandleIndex = 0; HandleIndex < *MatchingHandleCount; HandleIndex++) {\r
1758 if (HandleBufferForReturn[HandleIndex] == ChildControllerHandleBuffer[ChildControllerHandleIndex]) {\r
1759 break;\r
1760 }\r
1761 }\r
1762 if (HandleIndex >= *MatchingHandleCount) {\r
1763 HandleBufferForReturn[(*MatchingHandleCount)++] = ChildControllerHandleBuffer[ChildControllerHandleIndex];\r
1764 }\r
1765 }\r
1766\r
1767 FreePool (ChildControllerHandleBuffer);\r
1768 }\r
1769\r
1770 FreePool (DriverBindingHandleBuffer);\r
1771\r
1772 if (MatchingHandleBuffer != NULL) {\r
1773 *MatchingHandleBuffer = HandleBufferForReturn;\r
1774 } else {\r
1775 FreePool(HandleBufferForReturn);\r
1776 }\r
1777\r
1778 return (EFI_SUCCESS);\r
1779}\r
1780\r
1781/**\r
1782 Appends 1 buffer to another buffer. This will re-allocate the destination buffer\r
1783 if necessary to fit all of the data.\r
1784\r
1785 If DestinationBuffer is NULL, then ASSERT().\r
1786\r
1787 @param[in, out] DestinationBuffer The pointer to the pointer to the buffer to append onto.\r
1788 @param[in, out] DestinationSize The pointer to the size of DestinationBuffer.\r
1789 @param[in] SourceBuffer The pointer to the buffer to append onto DestinationBuffer.\r
1790 @param[in] SourceSize The number of bytes of SourceBuffer to append.\r
1791\r
1792 @retval NULL A memory allocation failed.\r
1793 @retval NULL A parameter was invalid.\r
1794 @return A pointer to (*DestinationBuffer).\r
1795**/\r
1796VOID*\r
1797EFIAPI\r
1798BuffernCatGrow (\r
1799 IN OUT VOID **DestinationBuffer,\r
1800 IN OUT UINTN *DestinationSize,\r
1801 IN VOID *SourceBuffer,\r
1802 IN UINTN SourceSize\r
1803 )\r
1804{\r
1805 UINTN LocalDestinationSize;\r
1806 UINTN LocalDestinationFinalSize;\r
1807\r
1808 ASSERT(DestinationBuffer != NULL);\r
1809\r
1810 if (SourceSize == 0 || SourceBuffer == NULL) {\r
1811 return (*DestinationBuffer);\r
1812 }\r
1813\r
1814 if (DestinationSize == NULL) {\r
1815 LocalDestinationSize = 0;\r
1816 } else {\r
1817 LocalDestinationSize = *DestinationSize;\r
1818 }\r
1819\r
1820 LocalDestinationFinalSize = LocalDestinationSize + SourceSize;\r
1821\r
1822 if (DestinationSize != NULL) {\r
1823 *DestinationSize = LocalDestinationSize;\r
1824 }\r
1825\r
1826 if (LocalDestinationSize == 0) {\r
1827 // allcoate\r
1828 *DestinationBuffer = AllocateZeroPool(LocalDestinationFinalSize);\r
1829 } else {\r
1830 // reallocate\r
1831 *DestinationBuffer = ReallocatePool(LocalDestinationSize, LocalDestinationFinalSize, *DestinationBuffer);\r
1832 }\r
1833\r
1834 ASSERT(*DestinationBuffer != NULL);\r
1835\r
1836 // copy\r
1837 return (CopyMem(((UINT8*)(*DestinationBuffer)) + LocalDestinationSize, SourceBuffer, SourceSize));\r
1838}\r
1839\r
1840/**\r
1841 Gets handles for any child devices produced by the passed in driver.\r
1842\r
1843 @param[in] DriverHandle The handle of the driver.\r
1844 @param[in] MatchingHandleCount Pointer to the number of handles in\r
1845 MatchingHandleBuffer on return.\r
1846 @param[out] MatchingHandleBuffer Buffer containing handles on a successful\r
1847 return.\r
1848 @retval EFI_SUCCESS The operation was sucessful.\r
1849 @sa ParseHandleDatabaseByRelationship\r
1850**/\r
1851EFI_STATUS\r
1852EFIAPI\r
1853ParseHandleDatabaseForChildDevices(\r
1854 IN CONST EFI_HANDLE DriverHandle,\r
1855 IN UINTN *MatchingHandleCount,\r
1856 OUT EFI_HANDLE **MatchingHandleBuffer OPTIONAL\r
1857 )\r
1858{\r
1859 EFI_HANDLE *Buffer;\r
1860 EFI_HANDLE *Buffer2;\r
1861 UINTN Count1;\r
1862 UINTN Count2;\r
1863 UINTN HandleIndex;\r
1864 EFI_STATUS Status;\r
1865 UINTN HandleBufferSize;\r
1866\r
1867 ASSERT(MatchingHandleCount != NULL);\r
1868\r
1869 HandleBufferSize = 0;\r
1870 Buffer = NULL;\r
1871 Buffer2 = NULL;\r
1872 *MatchingHandleCount = 0;\r
1873\r
1874 Status = PARSE_HANDLE_DATABASE_DEVICES (\r
1875 DriverHandle,\r
1876 &Count1,\r
1877 &Buffer\r
1878 );\r
1879 if (!EFI_ERROR (Status)) {\r
1880 for (HandleIndex = 0; HandleIndex < Count1; HandleIndex++) {\r
1881 //\r
1882 // now find the children\r
1883 //\r
1884 Status = PARSE_HANDLE_DATABASE_MANAGED_CHILDREN (\r
1885 DriverHandle,\r
1886 Buffer[HandleIndex],\r
1887 &Count2,\r
1888 &Buffer2\r
1889 );\r
1890 if (EFI_ERROR(Status)) {\r
1891 break;\r
1892 }\r
1893 //\r
1894 // save out required and optional data elements\r
1895 //\r
1896 *MatchingHandleCount += Count2;\r
1897 if (MatchingHandleBuffer != NULL) {\r
1898 *MatchingHandleBuffer = BuffernCatGrow((VOID**)MatchingHandleBuffer, &HandleBufferSize, Buffer2, Count2 * sizeof(Buffer2[0]));\r
1899 }\r
1900\r
1901 //\r
1902 // free the memory\r
1903 //\r
1904 if (Buffer2 != NULL) {\r
1905 FreePool(Buffer2);\r
1906 }\r
1907 }\r
1908 }\r
1909\r
1910 if (Buffer != NULL) {\r
1911 FreePool(Buffer);\r
1912 }\r
1913 return (Status);\r
1914}\r
1915\r
1916/**\r
1917 Function to get all handles that support a given protocol or all handles.\r
1918\r
1919 @param[in] ProtocolGuid The guid of the protocol to get handles for. If NULL\r
1920 then the function will return all handles.\r
1921\r
1922 @retval NULL A memory allocation failed.\r
1923 @return A NULL terminated list of handles.\r
1924**/\r
1925EFI_HANDLE*\r
1926EFIAPI\r
1927GetHandleListByProtocol (\r
1928 IN CONST EFI_GUID *ProtocolGuid OPTIONAL\r
1929 )\r
1930{\r
1931 EFI_HANDLE *HandleList;\r
1932 UINTN Size;\r
1933 EFI_STATUS Status;\r
1934\r
1935 Size = 0;\r
1936 HandleList = NULL;\r
1937\r
1938 //\r
1939 // We cannot use LocateHandleBuffer since we need that NULL item on the ends of the list!\r
1940 //\r
1941 if (ProtocolGuid == NULL) {\r
1942 Status = gBS->LocateHandle(AllHandles, NULL, NULL, &Size, HandleList);\r
1943 if (Status == EFI_BUFFER_TOO_SMALL) {\r
1944 HandleList = AllocateZeroPool(Size + sizeof(EFI_HANDLE));\r
1945 if (HandleList == NULL) {\r
1946 return (NULL);\r
1947 }\r
1948 Status = gBS->LocateHandle(AllHandles, NULL, NULL, &Size, HandleList);\r
1949 HandleList[Size/sizeof(EFI_HANDLE)] = NULL;\r
1950 }\r
1951 } else {\r
1952 Status = gBS->LocateHandle(ByProtocol, (EFI_GUID*)ProtocolGuid, NULL, &Size, HandleList);\r
1953 if (Status == EFI_BUFFER_TOO_SMALL) {\r
1954 HandleList = AllocateZeroPool(Size + sizeof(EFI_HANDLE));\r
1955 if (HandleList == NULL) {\r
1956 return (NULL);\r
1957 }\r
1958 Status = gBS->LocateHandle(ByProtocol, (EFI_GUID*)ProtocolGuid, NULL, &Size, HandleList);\r
1959 HandleList[Size/sizeof(EFI_HANDLE)] = NULL;\r
1960 }\r
1961 }\r
1962 if (EFI_ERROR(Status)) {\r
1963 if (HandleList != NULL) {\r
1964 FreePool(HandleList);\r
1965 }\r
1966 return (NULL);\r
1967 }\r
1968 return (HandleList);\r
1969}\r
1970\r
1971/**\r
1972 Function to get all handles that support some protocols.\r
1973\r
1974 @param[in] ProtocolGuids A NULL terminated list of protocol GUIDs.\r
1975\r
1976 @retval NULL A memory allocation failed.\r
1977 @retval NULL ProtocolGuids was NULL.\r
1978 @return A NULL terminated list of EFI_HANDLEs.\r
1979**/\r
1980EFI_HANDLE*\r
1981EFIAPI\r
1982GetHandleListByProtocolList (\r
1983 IN CONST EFI_GUID **ProtocolGuids\r
1984 )\r
1985{\r
1986 EFI_HANDLE *HandleList;\r
1987 UINTN Size;\r
1988 UINTN TotalSize;\r
1989 UINTN TempSize;\r
1990 EFI_STATUS Status;\r
1991 CONST EFI_GUID **GuidWalker;\r
1992 EFI_HANDLE *HandleWalker1;\r
1993 EFI_HANDLE *HandleWalker2;\r
1994\r
1995 Size = 0;\r
1996 HandleList = NULL;\r
1997 TotalSize = sizeof(EFI_HANDLE);\r
1998\r
1999 for (GuidWalker = ProtocolGuids ; GuidWalker != NULL && *GuidWalker != NULL ; GuidWalker++,Size = 0){\r
2000 Status = gBS->LocateHandle(ByProtocol, (EFI_GUID*)(*GuidWalker), NULL, &Size, NULL);\r
2001 if (Status == EFI_BUFFER_TOO_SMALL) {\r
2002 TotalSize += Size;\r
2003 }\r
2004 }\r
2005\r
2006 //\r
2007 // No handles were found... \r
2008 //\r
2009 if (TotalSize == sizeof(EFI_HANDLE)) {\r
2010 return (NULL);\r
2011 }\r
2012\r
2013 HandleList = AllocateZeroPool(TotalSize);\r
2014 if (HandleList == NULL) {\r
2015 return (NULL);\r
2016 }\r
2017\r
2018 Size = 0;\r
2019 for (GuidWalker = ProtocolGuids ; GuidWalker != NULL && *GuidWalker != NULL ; GuidWalker++){\r
2020 TempSize = TotalSize - Size;\r
2021 Status = gBS->LocateHandle(ByProtocol, (EFI_GUID*)(*GuidWalker), NULL, &TempSize, HandleList+(Size/sizeof(EFI_HANDLE)));\r
2022\r
2023 //\r
2024 // Allow for missing protocols... Only update the 'used' size upon success.\r
2025 //\r
2026 if (!EFI_ERROR(Status)) {\r
2027 Size += TempSize;\r
2028 }\r
2029 }\r
2030 ASSERT(HandleList[(TotalSize/sizeof(EFI_HANDLE))-1] == NULL);\r
2031\r
2032 for (HandleWalker1 = HandleList ; HandleWalker1 != NULL && *HandleWalker1 != NULL ; HandleWalker1++) {\r
2033 for (HandleWalker2 = HandleWalker1 + 1; HandleWalker2 != NULL && *HandleWalker2 != NULL ; HandleWalker2++) {\r
2034 if (*HandleWalker1 == *HandleWalker2) {\r
2035 //\r
2036 // copy memory back 1 handle width.\r
2037 //\r
2038 CopyMem(HandleWalker2, HandleWalker2 + 1, TotalSize - ((HandleWalker2-HandleList+1)*sizeof(EFI_HANDLE)));\r
2039 }\r
2040 }\r
2041 }\r
2042\r
2043 return (HandleList);\r
2044}\r
2045\r
2046\r
2047\r
2048\r
2049\r
2050\r
2051\r
2052\r
2053\r
2054\r