]> git.proxmox.com Git - mirror_edk2.git/blame - ArmPlatformPkg/Bds/BootMenu.c
EmbeddedPkg/Ebl: Fix EBL copy file command
[mirror_edk2.git] / ArmPlatformPkg / Bds / BootMenu.c
CommitLineData
ea46ebbe 1/** @file\r
2*\r
3* Copyright (c) 2011, ARM Limited. All rights reserved.\r
4*\r
5* This program and the accompanying materials\r
6* are licensed and made available under the terms and conditions of the BSD License\r
7* which accompanies this distribution. The full text of the license may be found at\r
8* http://opensource.org/licenses/bsd-license.php\r
9*\r
10* THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
11* WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
12*\r
13**/\r
14\r
15#include "BdsInternal.h"\r
16\r
17extern EFI_HANDLE mImageHandle;\r
18extern BDS_LOAD_OPTION_SUPPORT *BdsLoadOptionSupportList;\r
19\r
20EFI_STATUS\r
656416bc 21SelectBootDevice (\r
22 OUT BDS_SUPPORTED_DEVICE** SupportedBootDevice\r
ea46ebbe 23 )\r
24{\r
656416bc 25 EFI_STATUS Status;\r
ea46ebbe 26 LIST_ENTRY SupportedDeviceList;\r
27 UINTN SupportedDeviceCount;\r
ea46ebbe 28 LIST_ENTRY* Entry;\r
29 UINTN SupportedDeviceSelected;\r
ea46ebbe 30 UINTN Index;\r
ea46ebbe 31\r
32 //\r
33 // List the Boot Devices supported\r
34 //\r
35\r
36 // Start all the drivers first\r
37 BdsConnectAllDrivers ();\r
38\r
39 // List the supported devices\r
40 Status = BootDeviceListSupportedInit (&SupportedDeviceList);\r
41 ASSERT_EFI_ERROR(Status);\r
42\r
43 SupportedDeviceCount = 0;\r
44 for (Entry = GetFirstNode (&SupportedDeviceList);\r
45 !IsNull (&SupportedDeviceList,Entry);\r
46 Entry = GetNextNode (&SupportedDeviceList,Entry)\r
47 )\r
48 {\r
656416bc 49 *SupportedBootDevice = SUPPORTED_BOOT_DEVICE_FROM_LINK(Entry);\r
50 Print(L"[%d] %s\n",SupportedDeviceCount+1,(*SupportedBootDevice)->Description);\r
ea46ebbe 51\r
52 DEBUG_CODE_BEGIN();\r
53 CHAR16* DevicePathTxt;\r
54 EFI_DEVICE_PATH_TO_TEXT_PROTOCOL* DevicePathToTextProtocol;\r
55\r
ff7666c5 56 Status = gBS->LocateProtocol (&gEfiDevicePathToTextProtocolGuid, NULL, (VOID **)&DevicePathToTextProtocol);\r
ea46ebbe 57 ASSERT_EFI_ERROR(Status);\r
ff7666c5 58 DevicePathTxt = DevicePathToTextProtocol->ConvertDevicePathToText ((*SupportedBootDevice)->DevicePathProtocol,TRUE,TRUE);\r
ea46ebbe 59\r
60 Print(L"\t- %s\n",DevicePathTxt);\r
61\r
62 FreePool(DevicePathTxt);\r
63 DEBUG_CODE_END();\r
64\r
65 SupportedDeviceCount++;\r
66 }\r
67\r
68 if (SupportedDeviceCount == 0) {\r
69 Print(L"There is no supported device.\n");\r
70 Status = EFI_ABORTED;\r
71 goto EXIT;\r
72 }\r
73\r
74 //\r
75 // Select the Boot Device\r
76 //\r
77 SupportedDeviceSelected = 0;\r
78 while (SupportedDeviceSelected == 0) {\r
79 Print(L"Select the Boot Device: ");\r
80 Status = GetHIInputInteger (&SupportedDeviceSelected);\r
81 if (EFI_ERROR(Status)) {\r
82 Status = EFI_ABORTED;\r
83 goto EXIT;\r
84 } else if ((SupportedDeviceSelected == 0) || (SupportedDeviceSelected > SupportedDeviceCount)) {\r
ff7666c5 85 Print(L"Invalid input (max %d)\n",SupportedDeviceCount);\r
ea46ebbe 86 SupportedDeviceSelected = 0;\r
87 }\r
88 }\r
89\r
90 //\r
91 // Get the Device Path for the selected boot device\r
92 //\r
93 Index = 1;\r
94 for (Entry = GetFirstNode (&SupportedDeviceList);\r
95 !IsNull (&SupportedDeviceList,Entry);\r
96 Entry = GetNextNode (&SupportedDeviceList,Entry)\r
97 )\r
98 {\r
99 if (Index == SupportedDeviceSelected) {\r
656416bc 100 *SupportedBootDevice = SUPPORTED_BOOT_DEVICE_FROM_LINK(Entry);\r
ea46ebbe 101 break;\r
102 }\r
103 Index++;\r
104 }\r
656416bc 105 \r
106EXIT:\r
107 BootDeviceListSupportedFree (&SupportedDeviceList, *SupportedBootDevice);\r
108 return Status;\r
109}\r
110\r
111EFI_STATUS\r
112BootMenuAddBootOption (\r
113 IN LIST_ENTRY *BootOptionsList\r
114 )\r
115{\r
2ccfb71e 116 EFI_STATUS Status;\r
117 BDS_SUPPORTED_DEVICE* SupportedBootDevice;\r
118 ARM_BDS_LOADER_ARGUMENTS* BootArguments;\r
74b96132 119 CHAR16 BootDescription[BOOT_DEVICE_DESCRIPTION_MAX];\r
2ccfb71e 120 CHAR8 CmdLine[BOOT_DEVICE_OPTION_MAX];\r
121 UINT32 Attributes;\r
122 ARM_BDS_LOADER_TYPE BootType;\r
a6e97d28 123 BDS_LOAD_OPTION_ENTRY *BdsLoadOptionEntry;\r
2ccfb71e 124 EFI_DEVICE_PATH *DevicePath;\r
125 EFI_DEVICE_PATH_PROTOCOL *DevicePathNode;\r
126 EFI_DEVICE_PATH_PROTOCOL *InitrdPathNode;\r
127 EFI_DEVICE_PATH_PROTOCOL *InitrdPath;\r
128 UINTN CmdLineSize;\r
129 UINTN InitrdSize;\r
656416bc 130\r
131 Attributes = 0;\r
132 SupportedBootDevice = NULL;\r
133\r
134 // List the Boot Devices supported\r
2ccfb71e 135 Status = SelectBootDevice (&SupportedBootDevice);\r
656416bc 136 if (EFI_ERROR(Status)) {\r
137 Status = EFI_ABORTED;\r
138 goto EXIT;\r
139 }\r
ea46ebbe 140\r
141 // Create the specific device path node\r
656416bc 142 Print(L"File path of the EFI Application or the kernel: ");\r
ea46ebbe 143 Status = SupportedBootDevice->Support->CreateDevicePathNode (SupportedBootDevice, &DevicePathNode, &BootType, &Attributes);\r
144 if (EFI_ERROR(Status)) {\r
145 Status = EFI_ABORTED;\r
146 goto EXIT;\r
147 }\r
148 // Append the Device Path node to the select device path\r
149 DevicePath = AppendDevicePathNode (SupportedBootDevice->DevicePathProtocol, (CONST EFI_DEVICE_PATH_PROTOCOL *)DevicePathNode);\r
150\r
2ccfb71e 151 if ((BootType == BDS_LOADER_KERNEL_LINUX_ATAG) || (BootType == BDS_LOADER_KERNEL_LINUX_FDT)) {\r
656416bc 152 // Create the specific device path node\r
153 Print(L"File path of the initrd: ");\r
154 Status = SupportedBootDevice->Support->CreateDevicePathNode (SupportedBootDevice, &InitrdPathNode, NULL, NULL);\r
155 if (EFI_ERROR(Status) && Status != EFI_NOT_FOUND) { // EFI_NOT_FOUND is returned on empty input string, but we can boot without an initrd\r
156 Status = EFI_ABORTED;\r
157 goto EXIT;\r
158 }\r
159\r
160 if (InitrdPathNode != NULL) {\r
161 // Append the Device Path node to the select device path\r
2ccfb71e 162 InitrdPath = AppendDevicePathNode (SupportedBootDevice->DevicePathProtocol, (CONST EFI_DEVICE_PATH_PROTOCOL *)InitrdPathNode);\r
656416bc 163 } else {\r
2ccfb71e 164 InitrdPath = NULL;\r
656416bc 165 }\r
166\r
167 Print(L"Arguments to pass to the binary: ");\r
2ccfb71e 168 Status = GetHIInputAscii (CmdLine,BOOT_DEVICE_OPTION_MAX);\r
656416bc 169 if (EFI_ERROR(Status)) {\r
170 Status = EFI_ABORTED;\r
171 goto FREE_DEVICE_PATH;\r
172 }\r
2ccfb71e 173\r
174 CmdLineSize = AsciiStrSize (CmdLine);\r
175 InitrdSize = GetDevicePathSize (InitrdPath);\r
176\r
177 BootArguments = (ARM_BDS_LOADER_ARGUMENTS*)AllocatePool (sizeof(ARM_BDS_LOADER_ARGUMENTS) + CmdLineSize + InitrdSize);\r
178 \r
179 BootArguments->LinuxArguments.CmdLineSize = CmdLineSize;\r
180 BootArguments->LinuxArguments.InitrdSize = InitrdSize;\r
181 CopyMem ((VOID*)(&BootArguments->LinuxArguments + 1), CmdLine, CmdLineSize);\r
182 CopyMem ((VOID*)((UINTN)(&BootArguments->LinuxArguments + 1) + CmdLineSize), InitrdPath, InitrdSize);\r
183 } else {\r
184 BootArguments = NULL;\r
ea46ebbe 185 }\r
186\r
187 Print(L"Description for this new Entry: ");\r
74b96132 188 Status = GetHIInputStr (BootDescription, BOOT_DEVICE_DESCRIPTION_MAX);\r
ea46ebbe 189 if (EFI_ERROR(Status)) {\r
190 Status = EFI_ABORTED;\r
191 goto FREE_DEVICE_PATH;\r
192 }\r
193\r
ea46ebbe 194 // Create new entry\r
a6e97d28 195 BdsLoadOptionEntry = (BDS_LOAD_OPTION_ENTRY*)AllocatePool (sizeof(BDS_LOAD_OPTION_ENTRY));\r
196 Status = BootOptionCreate (Attributes, BootDescription, DevicePath, BootType, BootArguments, &BdsLoadOptionEntry->BdsLoadOption);\r
ea46ebbe 197 if (!EFI_ERROR(Status)) {\r
a6e97d28 198 InsertTailList (BootOptionsList, &BdsLoadOptionEntry->Link);\r
ea46ebbe 199 }\r
200\r
ea46ebbe 201FREE_DEVICE_PATH:\r
202 FreePool (DevicePath);\r
203\r
656416bc 204 \r
ea46ebbe 205EXIT:\r
656416bc 206 if (Status == EFI_ABORTED) {\r
207 Print(L"\n");\r
208 }\r
209 FreePool(SupportedBootDevice);\r
ea46ebbe 210 return Status;\r
211}\r
212\r
213STATIC\r
214EFI_STATUS\r
215BootMenuSelectBootOption (\r
a6e97d28 216 IN LIST_ENTRY* BootOptionsList,\r
217 IN CONST CHAR16* InputStatement,\r
218 IN BOOLEAN OnlyArmBdsBootEntry,\r
219 OUT BDS_LOAD_OPTION_ENTRY** BdsLoadOptionEntry\r
ea46ebbe 220 )\r
221{\r
a6e97d28 222 EFI_STATUS Status;\r
223 LIST_ENTRY* Entry;\r
224 BDS_LOAD_OPTION* BdsLoadOption;\r
225 UINTN BootOptionSelected;\r
226 UINTN BootOptionCount;\r
227 UINTN Index;\r
ea46ebbe 228\r
229 // Display the list of supported boot devices\r
c93ab96c 230 BootOptionCount = 0;\r
ea46ebbe 231 for (Entry = GetFirstNode (BootOptionsList);\r
232 !IsNull (BootOptionsList,Entry);\r
a6e97d28 233 Entry = GetNextNode (BootOptionsList, Entry)\r
ea46ebbe 234 )\r
235 {\r
a6e97d28 236 BdsLoadOption = LOAD_OPTION_FROM_LINK(Entry);\r
237\r
238 if (OnlyArmBdsBootEntry && !IS_ARM_BDS_BOOTENTRY (BdsLoadOption)) {\r
239 continue;\r
240 }\r
241\r
c93ab96c 242 Print (L"[%d] %s\n", (BootOptionCount + 1), BdsLoadOption->Description);\r
ea46ebbe 243\r
244 DEBUG_CODE_BEGIN();\r
245 CHAR16* DevicePathTxt;\r
246 EFI_DEVICE_PATH_TO_TEXT_PROTOCOL* DevicePathToTextProtocol;\r
2ccfb71e 247 ARM_BDS_LOADER_TYPE LoaderType;\r
a6e97d28 248 ARM_BDS_LOADER_OPTIONAL_DATA* OptionalData;\r
ea46ebbe 249\r
250 Status = gBS->LocateProtocol(&gEfiDevicePathToTextProtocolGuid, NULL, (VOID **)&DevicePathToTextProtocol);\r
251 ASSERT_EFI_ERROR(Status);\r
a6e97d28 252 DevicePathTxt = DevicePathToTextProtocol->ConvertDevicePathToText(BdsLoadOption->FilePathList,TRUE,TRUE);\r
ea46ebbe 253\r
254 Print(L"\t- %s\n",DevicePathTxt);\r
2ccfb71e 255 OptionalData = BdsLoadOption->OptionalData;\r
256 LoaderType = (ARM_BDS_LOADER_TYPE)ReadUnaligned32 ((CONST UINT32*)&OptionalData->Header.LoaderType);\r
257 if ((LoaderType == BDS_LOADER_KERNEL_LINUX_ATAG) || (LoaderType == BDS_LOADER_KERNEL_LINUX_FDT)) {\r
258 Print (L"\t- Arguments: %a\n",&OptionalData->Arguments.LinuxArguments + 1);\r
ea46ebbe 259 }\r
260\r
261 FreePool(DevicePathTxt);\r
262 DEBUG_CODE_END();\r
263\r
264 BootOptionCount++;\r
265 }\r
266\r
c93ab96c 267 // Check if a valid boot option(s) is found\r
a6e97d28 268 if (BootOptionCount == 0) {\r
c93ab96c 269 if (StrCmp (InputStatement, DELETE_BOOT_ENTRY) == 0) {\r
270 Print (L"Nothing to remove!\n");\r
271 }else if (StrCmp (InputStatement, UPDATE_BOOT_ENTRY) == 0) {\r
272 Print (L"Couldn't find valid boot entries\n");\r
273 } else{\r
274 Print (L"No supported Boot Entry.\n");\r
275 }\r
276\r
a6e97d28 277 return EFI_NOT_FOUND;\r
278 }\r
279\r
ea46ebbe 280 // Get the index of the boot device to delete\r
281 BootOptionSelected = 0;\r
282 while (BootOptionSelected == 0) {\r
283 Print(InputStatement);\r
284 Status = GetHIInputInteger (&BootOptionSelected);\r
285 if (EFI_ERROR(Status)) {\r
286 return Status;\r
c93ab96c 287 } else if ((BootOptionSelected == 0) || (BootOptionSelected > BootOptionCount)) {\r
288 Print(L"Invalid input (max %d)\n",BootOptionCount);\r
ea46ebbe 289 BootOptionSelected = 0;\r
290 }\r
291 }\r
292\r
293 // Get the structure of the Boot device to delete\r
294 Index = 1;\r
295 for (Entry = GetFirstNode (BootOptionsList);\r
ff7666c5 296 !IsNull (BootOptionsList, Entry);\r
ea46ebbe 297 Entry = GetNextNode (BootOptionsList,Entry)\r
298 )\r
299 {\r
300 if (Index == BootOptionSelected) {\r
a6e97d28 301 *BdsLoadOptionEntry = LOAD_OPTION_ENTRY_FROM_LINK(Entry);\r
ea46ebbe 302 break;\r
303 }\r
304 Index++;\r
305 }\r
306\r
307 return EFI_SUCCESS;\r
308}\r
309\r
310EFI_STATUS\r
311BootMenuRemoveBootOption (\r
312 IN LIST_ENTRY *BootOptionsList\r
313 )\r
314{\r
a6e97d28 315 EFI_STATUS Status;\r
316 BDS_LOAD_OPTION_ENTRY* BootOptionEntry;\r
ea46ebbe 317\r
a6e97d28 318 Status = BootMenuSelectBootOption (BootOptionsList, L"Delete entry: ", FALSE, &BootOptionEntry);\r
ea46ebbe 319 if (EFI_ERROR(Status)) {\r
320 return Status;\r
321 }\r
322\r
a6e97d28 323 // If the Boot Option was attached to a list remove it\r
324 if (!IsListEmpty (&BootOptionEntry->Link)) {\r
325 // Remove the entry from the list\r
326 RemoveEntryList (&BootOptionEntry->Link);\r
327 }\r
328\r
ea46ebbe 329 // Delete the BDS Load option structures\r
a6e97d28 330 BootOptionDelete (BootOptionEntry->BdsLoadOption);\r
ea46ebbe 331\r
332 return EFI_SUCCESS;\r
333}\r
334\r
335EFI_STATUS\r
336BootMenuUpdateBootOption (\r
337 IN LIST_ENTRY *BootOptionsList\r
338 )\r
339{\r
2ccfb71e 340 EFI_STATUS Status;\r
a6e97d28 341 BDS_LOAD_OPTION_ENTRY *BootOptionEntry;\r
2ccfb71e 342 BDS_LOAD_OPTION *BootOption;\r
343 BDS_LOAD_OPTION_SUPPORT* DeviceSupport;\r
344 ARM_BDS_LOADER_ARGUMENTS* BootArguments;\r
74b96132 345 CHAR16 BootDescription[BOOT_DEVICE_DESCRIPTION_MAX];\r
2ccfb71e 346 CHAR8 CmdLine[BOOT_DEVICE_OPTION_MAX];\r
347 EFI_DEVICE_PATH* DevicePath;\r
348 ARM_BDS_LOADER_TYPE BootType;\r
349 ARM_BDS_LOADER_OPTIONAL_DATA* OptionalData;\r
350 ARM_BDS_LINUX_ARGUMENTS* LinuxArguments;\r
351 EFI_DEVICE_PATH* InitrdPathList;\r
352 UINTN InitrdSize;\r
353 UINTN CmdLineSize;\r
ea46ebbe 354\r
a6e97d28 355 Status = BootMenuSelectBootOption (BootOptionsList, L"Update entry: ", TRUE, &BootOptionEntry);\r
ea46ebbe 356 if (EFI_ERROR(Status)) {\r
357 return Status;\r
358 }\r
a6e97d28 359 BootOption = BootOptionEntry->BdsLoadOption;\r
ea46ebbe 360\r
361 // Get the device support for this Boot Option\r
ff7666c5 362 Status = BootDeviceGetDeviceSupport (BootOption, &DeviceSupport);\r
ea46ebbe 363 if (EFI_ERROR(Status)) {\r
ff7666c5 364 Print(L"Not possible to retrieve the supported device for the update\n");\r
ea46ebbe 365 return EFI_UNSUPPORTED;\r
366 }\r
367\r
656416bc 368 Print(L"File path of the EFI Application or the kernel: ");\r
369 Status = DeviceSupport->UpdateDevicePathNode (BootOption->FilePathList, &DevicePath, NULL, NULL);\r
ea46ebbe 370 if (EFI_ERROR(Status)) {\r
371 Status = EFI_ABORTED;\r
372 goto EXIT;\r
373 }\r
374\r
2ccfb71e 375 OptionalData = BootOption->OptionalData;\r
376 BootType = (ARM_BDS_LOADER_TYPE)ReadUnaligned32 ((UINT32 *)(&OptionalData->Header.LoaderType));\r
656416bc 377\r
378 // TODO: Allow adding an initrd to a boot entry without one\r
2ccfb71e 379 if ((BootType == BDS_LOADER_KERNEL_LINUX_ATAG) || (BootType == BDS_LOADER_KERNEL_LINUX_FDT)) {\r
380 LinuxArguments = &OptionalData->Arguments.LinuxArguments;\r
381\r
382 CmdLineSize = ReadUnaligned16 ((CONST UINT16*)&LinuxArguments->CmdLineSize);\r
2ccfb71e 383\r
3e710183 384 InitrdSize = ReadUnaligned16 ((CONST UINT16*)&LinuxArguments->InitrdSize);\r
385 if (InitrdSize > 0) {\r
386 Print(L"File path of the initrd: ");\r
387 Status = DeviceSupport->UpdateDevicePathNode ((EFI_DEVICE_PATH*)((LinuxArguments + 1) + CmdLineSize), &InitrdPathList, NULL, NULL);\r
388 if (EFI_ERROR(Status) && Status != EFI_NOT_FOUND) {// EFI_NOT_FOUND is returned on empty input string, but we can boot without an initrd\r
389 Status = EFI_ABORTED;\r
390 goto EXIT;\r
391 }\r
392 InitrdSize = GetDevicePathSize (InitrdPathList);\r
656416bc 393 }\r
394\r
2ccfb71e 395 Print(L"Arguments to pass to the binary: "); \r
396 if (CmdLineSize > 0) {\r
397 AsciiStrnCpy(CmdLine, (CONST CHAR8*)(LinuxArguments + 1), CmdLineSize);\r
656416bc 398 } else {\r
2ccfb71e 399 CmdLine[0] = '\0';\r
656416bc 400 }\r
2ccfb71e 401 Status = EditHIInputAscii (CmdLine, BOOT_DEVICE_OPTION_MAX);\r
656416bc 402 if (EFI_ERROR(Status)) {\r
403 Status = EFI_ABORTED;\r
404 goto FREE_DEVICE_PATH;\r
405 }\r
2ccfb71e 406\r
407 CmdLineSize = AsciiStrSize (CmdLine);\r
2ccfb71e 408\r
409 BootArguments = (ARM_BDS_LOADER_ARGUMENTS*)AllocatePool(sizeof(ARM_BDS_LOADER_ARGUMENTS) + CmdLineSize + InitrdSize);\r
410 BootArguments->LinuxArguments.CmdLineSize = CmdLineSize;\r
411 BootArguments->LinuxArguments.InitrdSize = InitrdSize;\r
412 CopyMem (&BootArguments->LinuxArguments + 1, CmdLine, CmdLineSize);\r
413 CopyMem ((UINTN)(&BootArguments->LinuxArguments + 1) + CmdLine, InitrdPathList, InitrdSize);\r
414 } else {\r
415 BootArguments = NULL;\r
ea46ebbe 416 }\r
417\r
418 Print(L"Description for this new Entry: ");\r
3e710183 419 StrnCpy (BootDescription, BootOption->Description, BOOT_DEVICE_DESCRIPTION_MAX);\r
74b96132 420 Status = EditHIInputStr (BootDescription, BOOT_DEVICE_DESCRIPTION_MAX);\r
ea46ebbe 421 if (EFI_ERROR(Status)) {\r
422 Status = EFI_ABORTED;\r
423 goto FREE_DEVICE_PATH;\r
424 }\r
425\r
ea46ebbe 426 // Update the entry\r
2ccfb71e 427 Status = BootOptionUpdate (BootOption, BootOption->Attributes, BootDescription, DevicePath, BootType, BootArguments);\r
ea46ebbe 428\r
ea46ebbe 429FREE_DEVICE_PATH:\r
430 FreePool (DevicePath);\r
431\r
432EXIT:\r
326d1df9 433 if (Status == EFI_ABORTED) {\r
434 Print(L"\n");\r
435 }\r
ea46ebbe 436 return Status;\r
437}\r
438\r
439struct BOOT_MANAGER_ENTRY {\r
440 CONST CHAR16* Description;\r
441 EFI_STATUS (*Callback) (IN LIST_ENTRY *BootOptionsList);\r
442} BootManagerEntries[] = {\r
443 { L"Add Boot Device Entry", BootMenuAddBootOption },\r
444 { L"Update Boot Device Entry", BootMenuUpdateBootOption },\r
445 { L"Remove Boot Device Entry", BootMenuRemoveBootOption },\r
446};\r
447\r
448EFI_STATUS\r
449BootMenuManager (\r
450 IN LIST_ENTRY *BootOptionsList\r
451 )\r
452{\r
453 UINTN Index;\r
454 UINTN OptionSelected;\r
455 UINTN BootManagerEntryCount;\r
456 EFI_STATUS Status;\r
457\r
458 BootManagerEntryCount = sizeof(BootManagerEntries) / sizeof(struct BOOT_MANAGER_ENTRY);\r
459\r
460 while (TRUE) {\r
461 // Display Boot Manager menu\r
462 for (Index = 0; Index < BootManagerEntryCount; Index++) {\r
463 Print(L"[%d] %s\n",Index+1,BootManagerEntries[Index]);\r
464 }\r
465 Print(L"[%d] Return to main menu\n",Index+1);\r
466\r
467 // Select which entry to call\r
468 Print(L"Choice: ");\r
469 Status = GetHIInputInteger (&OptionSelected);\r
470 if (EFI_ERROR(Status) || (OptionSelected == (BootManagerEntryCount+1))) {\r
326d1df9 471 if (EFI_ERROR(Status)) {\r
472 Print(L"\n");\r
473 }\r
ea46ebbe 474 return EFI_SUCCESS;\r
475 } else if ((OptionSelected > 0) && (OptionSelected <= BootManagerEntryCount)) {\r
ff7666c5 476 BootManagerEntries[OptionSelected-1].Callback (BootOptionsList);\r
ea46ebbe 477 }\r
478 }\r
479\r
480 return EFI_SUCCESS;\r
481}\r
482\r
483EFI_STATUS\r
484BootEBL (\r
485 IN LIST_ENTRY *BootOptionsList\r
486 )\r
487{\r
488 EFI_STATUS Status;\r
489\r
490 // Start EFI Shell\r
2755d844 491 Status = BdsLoadApplication (mImageHandle, L"Ebl", 0, NULL);\r
ea46ebbe 492 if (Status == EFI_NOT_FOUND) {\r
493 Print (L"Error: EFI Application not found.\n");\r
494 } else if (EFI_ERROR(Status)) {\r
495 Print (L"Error: Status Code: 0x%X\n",(UINT32)Status);\r
496 }\r
497\r
498 return Status;\r
499}\r
500\r
501struct BOOT_MAIN_ENTRY {\r
502 CONST CHAR16* Description;\r
503 EFI_STATUS (*Callback) (IN LIST_ENTRY *BootOptionsList);\r
504} BootMainEntries[] = {\r
505 { L"EBL", BootEBL },\r
506 { L"Boot Manager", BootMenuManager },\r
507};\r
508\r
509\r
510EFI_STATUS\r
511BootMenuMain (\r
512 VOID\r
513 )\r
514{\r
2ccfb71e 515 LIST_ENTRY BootOptionsList;\r
516 UINTN OptionCount;\r
517 UINTN BootOptionCount;\r
518 EFI_STATUS Status;\r
519 LIST_ENTRY* Entry;\r
520 BDS_LOAD_OPTION* BootOption;\r
521 UINTN BootOptionSelected;\r
522 UINTN Index;\r
523 UINTN BootMainEntryCount;\r
ea46ebbe 524\r
e862cd50 525 BootOption = NULL;\r
ea46ebbe 526 BootMainEntryCount = sizeof(BootMainEntries) / sizeof(struct BOOT_MAIN_ENTRY);\r
a6caee65 527\r
ea46ebbe 528 while (TRUE) {\r
a6e97d28 529 // Get Boot#### list\r
530 BootOptionList (&BootOptionsList);\r
531\r
ea46ebbe 532 OptionCount = 1;\r
533\r
534 // Display the Boot options\r
535 for (Entry = GetFirstNode (&BootOptionsList);\r
536 !IsNull (&BootOptionsList,Entry);\r
537 Entry = GetNextNode (&BootOptionsList,Entry)\r
538 )\r
539 {\r
540 BootOption = LOAD_OPTION_FROM_LINK(Entry);\r
541\r
ff7666c5 542 Print(L"[%d] %s\n", OptionCount, BootOption->Description);\r
ea46ebbe 543\r
544 DEBUG_CODE_BEGIN();\r
545 CHAR16* DevicePathTxt;\r
546 EFI_DEVICE_PATH_TO_TEXT_PROTOCOL* DevicePathToTextProtocol;\r
2ccfb71e 547 ARM_BDS_LOADER_OPTIONAL_DATA* OptionalData;\r
548 UINTN CmdLineSize;\r
3e710183 549 ARM_BDS_LOADER_TYPE LoaderType;\r
ea46ebbe 550\r
ff7666c5 551 Status = gBS->LocateProtocol (&gEfiDevicePathToTextProtocolGuid, NULL, (VOID **)&DevicePathToTextProtocol);\r
aa95e2f7 552 if (EFI_ERROR(Status)) {\r
553 // You must provide an implementation of DevicePathToTextProtocol in your firmware (eg: DevicePathDxe)\r
554 DEBUG((EFI_D_ERROR,"Error: Bds requires DevicePathToTextProtocol\n"));\r
555 return Status;\r
556 }\r
ff7666c5 557 DevicePathTxt = DevicePathToTextProtocol->ConvertDevicePathToText (BootOption->FilePathList, TRUE, TRUE);\r
ea46ebbe 558\r
559 Print(L"\t- %s\n",DevicePathTxt);\r
2ccfb71e 560\r
561 // If it is a supported BootEntry then print its details\r
562 if (IS_ARM_BDS_BOOTENTRY (BootOption)) {\r
563 OptionalData = BootOption->OptionalData;\r
564 LoaderType = (ARM_BDS_LOADER_TYPE)ReadUnaligned32 ((CONST UINT32*)&OptionalData->Header.LoaderType);\r
565 if ((LoaderType == BDS_LOADER_KERNEL_LINUX_ATAG) || (LoaderType == BDS_LOADER_KERNEL_LINUX_FDT)) {\r
566 if (ReadUnaligned16 (&OptionalData->Arguments.LinuxArguments.InitrdSize) > 0) {\r
567 CmdLineSize = ReadUnaligned16 (&OptionalData->Arguments.LinuxArguments.CmdLineSize);\r
568 DevicePathTxt = DevicePathToTextProtocol->ConvertDevicePathToText (\r
569 GetAlignedDevicePath ((EFI_DEVICE_PATH*)((UINTN)(&OptionalData->Arguments.LinuxArguments + 1) + CmdLineSize)), TRUE, TRUE);\r
570 Print(L"\t- Initrd: %s\n", DevicePathTxt);\r
571 }\r
572 Print(L"\t- Arguments: %a\n", (&OptionalData->Arguments.LinuxArguments + 1));\r
ea46ebbe 573 }\r
2ccfb71e 574 Print(L"\t- LoaderType: %d\n", LoaderType);\r
ea46ebbe 575 }\r
ea46ebbe 576 FreePool(DevicePathTxt);\r
577 DEBUG_CODE_END();\r
578\r
579 OptionCount++;\r
580 }\r
581 BootOptionCount = OptionCount-1;\r
582\r
583 // Display the hardcoded Boot entries\r
584 for (Index = 0; Index < BootMainEntryCount; Index++) {\r
585 Print(L"[%d] %s\n",OptionCount,BootMainEntries[Index]);\r
586 OptionCount++;\r
587 }\r
588\r
589 // Request the boot entry from the user\r
590 BootOptionSelected = 0;\r
591 while (BootOptionSelected == 0) {\r
592 Print(L"Start: ");\r
593 Status = GetHIInputInteger (&BootOptionSelected);\r
594 if (EFI_ERROR(Status) || (BootOptionSelected == 0) || (BootOptionSelected > OptionCount)) {\r
595 Print(L"Invalid input (max %d)\n",(OptionCount-1));\r
596 BootOptionSelected = 0;\r
597 }\r
598 }\r
599\r
600 // Start the selected entry\r
601 if (BootOptionSelected > BootOptionCount) {\r
602 // Start the hardcoded entry\r
603 Status = BootMainEntries[BootOptionSelected - BootOptionCount - 1].Callback (&BootOptionsList);\r
604 } else {\r
605 // Find the selected entry from the Boot#### list\r
606 Index = 1;\r
607 for (Entry = GetFirstNode (&BootOptionsList);\r
608 !IsNull (&BootOptionsList,Entry);\r
609 Entry = GetNextNode (&BootOptionsList,Entry)\r
610 )\r
611 {\r
612 if (Index == BootOptionSelected) {\r
613 BootOption = LOAD_OPTION_FROM_LINK(Entry);\r
614 break;\r
615 }\r
616 Index++;\r
617 }\r
618\r
619 Status = BootOptionStart (BootOption);\r
620 }\r
621 }\r
622\r
623 return Status;\r
624}\r