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 |
17 | extern EFI_HANDLE mImageHandle;\r |
18 | extern BDS_LOAD_OPTION_SUPPORT *BdsLoadOptionSupportList;\r |
19 | \r |
20 | EFI_STATUS\r |
656416bc |
21 | SelectBootDevice (\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 |
106 | EXIT:\r |
107 | BootDeviceListSupportedFree (&SupportedDeviceList, *SupportedBootDevice);\r |
108 | return Status;\r |
109 | }\r |
110 | \r |
111 | EFI_STATUS\r |
112 | BootMenuAddBootOption (\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 |
201 | FREE_DEVICE_PATH:\r |
202 | FreePool (DevicePath);\r |
203 | \r |
656416bc |
204 | \r |
ea46ebbe |
205 | EXIT:\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 |
213 | STATIC\r |
214 | EFI_STATUS\r |
215 | BootMenuSelectBootOption (\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 |
310 | EFI_STATUS\r |
311 | BootMenuRemoveBootOption (\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 |
11c20f4e |
318 | Status = BootMenuSelectBootOption (BootOptionsList, DELETE_BOOT_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 |
335 | EFI_STATUS\r |
336 | BootMenuUpdateBootOption (\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 |
11c20f4e |
355 | Status = BootMenuSelectBootOption (BootOptionsList, UPDATE_BOOT_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 |
429 | FREE_DEVICE_PATH:\r |
430 | FreePool (DevicePath);\r |
431 | \r |
432 | EXIT:\r |
326d1df9 |
433 | if (Status == EFI_ABORTED) {\r |
434 | Print(L"\n");\r |
435 | }\r |
ea46ebbe |
436 | return Status;\r |
437 | }\r |
438 | \r |
439 | struct 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 |
448 | EFI_STATUS\r |
449 | BootMenuManager (\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 |
11c20f4e |
479 | // Should never go here\r |
ea46ebbe |
480 | }\r |
481 | \r |
482 | EFI_STATUS\r |
483 | BootEBL (\r |
484 | IN LIST_ENTRY *BootOptionsList\r |
485 | )\r |
486 | {\r |
487 | EFI_STATUS Status;\r |
488 | \r |
489 | // Start EFI Shell\r |
2755d844 |
490 | Status = BdsLoadApplication (mImageHandle, L"Ebl", 0, NULL);\r |
ea46ebbe |
491 | if (Status == EFI_NOT_FOUND) {\r |
492 | Print (L"Error: EFI Application not found.\n");\r |
493 | } else if (EFI_ERROR(Status)) {\r |
494 | Print (L"Error: Status Code: 0x%X\n",(UINT32)Status);\r |
495 | }\r |
496 | \r |
497 | return Status;\r |
498 | }\r |
499 | \r |
500 | struct BOOT_MAIN_ENTRY {\r |
501 | CONST CHAR16* Description;\r |
502 | EFI_STATUS (*Callback) (IN LIST_ENTRY *BootOptionsList);\r |
503 | } BootMainEntries[] = {\r |
504 | { L"EBL", BootEBL },\r |
505 | { L"Boot Manager", BootMenuManager },\r |
506 | };\r |
507 | \r |
508 | \r |
509 | EFI_STATUS\r |
510 | BootMenuMain (\r |
511 | VOID\r |
512 | )\r |
513 | {\r |
2ccfb71e |
514 | LIST_ENTRY BootOptionsList;\r |
515 | UINTN OptionCount;\r |
516 | UINTN BootOptionCount;\r |
517 | EFI_STATUS Status;\r |
518 | LIST_ENTRY* Entry;\r |
519 | BDS_LOAD_OPTION* BootOption;\r |
520 | UINTN BootOptionSelected;\r |
521 | UINTN Index;\r |
522 | UINTN BootMainEntryCount;\r |
ea46ebbe |
523 | \r |
e862cd50 |
524 | BootOption = NULL;\r |
ea46ebbe |
525 | BootMainEntryCount = sizeof(BootMainEntries) / sizeof(struct BOOT_MAIN_ENTRY);\r |
a6caee65 |
526 | \r |
ea46ebbe |
527 | while (TRUE) {\r |
a6e97d28 |
528 | // Get Boot#### list\r |
529 | BootOptionList (&BootOptionsList);\r |
530 | \r |
ea46ebbe |
531 | OptionCount = 1;\r |
532 | \r |
533 | // Display the Boot options\r |
534 | for (Entry = GetFirstNode (&BootOptionsList);\r |
535 | !IsNull (&BootOptionsList,Entry);\r |
536 | Entry = GetNextNode (&BootOptionsList,Entry)\r |
537 | )\r |
538 | {\r |
539 | BootOption = LOAD_OPTION_FROM_LINK(Entry);\r |
540 | \r |
ff7666c5 |
541 | Print(L"[%d] %s\n", OptionCount, BootOption->Description);\r |
ea46ebbe |
542 | \r |
543 | DEBUG_CODE_BEGIN();\r |
544 | CHAR16* DevicePathTxt;\r |
545 | EFI_DEVICE_PATH_TO_TEXT_PROTOCOL* DevicePathToTextProtocol;\r |
2ccfb71e |
546 | ARM_BDS_LOADER_OPTIONAL_DATA* OptionalData;\r |
547 | UINTN CmdLineSize;\r |
3e710183 |
548 | ARM_BDS_LOADER_TYPE LoaderType;\r |
ea46ebbe |
549 | \r |
ff7666c5 |
550 | Status = gBS->LocateProtocol (&gEfiDevicePathToTextProtocolGuid, NULL, (VOID **)&DevicePathToTextProtocol);\r |
aa95e2f7 |
551 | if (EFI_ERROR(Status)) {\r |
552 | // You must provide an implementation of DevicePathToTextProtocol in your firmware (eg: DevicePathDxe)\r |
553 | DEBUG((EFI_D_ERROR,"Error: Bds requires DevicePathToTextProtocol\n"));\r |
554 | return Status;\r |
555 | }\r |
ff7666c5 |
556 | DevicePathTxt = DevicePathToTextProtocol->ConvertDevicePathToText (BootOption->FilePathList, TRUE, TRUE);\r |
ea46ebbe |
557 | \r |
558 | Print(L"\t- %s\n",DevicePathTxt);\r |
2ccfb71e |
559 | \r |
560 | // If it is a supported BootEntry then print its details\r |
561 | if (IS_ARM_BDS_BOOTENTRY (BootOption)) {\r |
562 | OptionalData = BootOption->OptionalData;\r |
563 | LoaderType = (ARM_BDS_LOADER_TYPE)ReadUnaligned32 ((CONST UINT32*)&OptionalData->Header.LoaderType);\r |
564 | if ((LoaderType == BDS_LOADER_KERNEL_LINUX_ATAG) || (LoaderType == BDS_LOADER_KERNEL_LINUX_FDT)) {\r |
565 | if (ReadUnaligned16 (&OptionalData->Arguments.LinuxArguments.InitrdSize) > 0) {\r |
566 | CmdLineSize = ReadUnaligned16 (&OptionalData->Arguments.LinuxArguments.CmdLineSize);\r |
567 | DevicePathTxt = DevicePathToTextProtocol->ConvertDevicePathToText (\r |
568 | GetAlignedDevicePath ((EFI_DEVICE_PATH*)((UINTN)(&OptionalData->Arguments.LinuxArguments + 1) + CmdLineSize)), TRUE, TRUE);\r |
569 | Print(L"\t- Initrd: %s\n", DevicePathTxt);\r |
570 | }\r |
571 | Print(L"\t- Arguments: %a\n", (&OptionalData->Arguments.LinuxArguments + 1));\r |
ea46ebbe |
572 | }\r |
2ccfb71e |
573 | Print(L"\t- LoaderType: %d\n", LoaderType);\r |
ea46ebbe |
574 | }\r |
ea46ebbe |
575 | FreePool(DevicePathTxt);\r |
576 | DEBUG_CODE_END();\r |
577 | \r |
578 | OptionCount++;\r |
579 | }\r |
580 | BootOptionCount = OptionCount-1;\r |
581 | \r |
582 | // Display the hardcoded Boot entries\r |
583 | for (Index = 0; Index < BootMainEntryCount; Index++) {\r |
584 | Print(L"[%d] %s\n",OptionCount,BootMainEntries[Index]);\r |
585 | OptionCount++;\r |
586 | }\r |
587 | \r |
588 | // Request the boot entry from the user\r |
589 | BootOptionSelected = 0;\r |
590 | while (BootOptionSelected == 0) {\r |
591 | Print(L"Start: ");\r |
592 | Status = GetHIInputInteger (&BootOptionSelected);\r |
593 | if (EFI_ERROR(Status) || (BootOptionSelected == 0) || (BootOptionSelected > OptionCount)) {\r |
594 | Print(L"Invalid input (max %d)\n",(OptionCount-1));\r |
595 | BootOptionSelected = 0;\r |
596 | }\r |
597 | }\r |
598 | \r |
599 | // Start the selected entry\r |
600 | if (BootOptionSelected > BootOptionCount) {\r |
601 | // Start the hardcoded entry\r |
602 | Status = BootMainEntries[BootOptionSelected - BootOptionCount - 1].Callback (&BootOptionsList);\r |
603 | } else {\r |
604 | // Find the selected entry from the Boot#### list\r |
605 | Index = 1;\r |
606 | for (Entry = GetFirstNode (&BootOptionsList);\r |
607 | !IsNull (&BootOptionsList,Entry);\r |
608 | Entry = GetNextNode (&BootOptionsList,Entry)\r |
609 | )\r |
610 | {\r |
611 | if (Index == BootOptionSelected) {\r |
612 | BootOption = LOAD_OPTION_FROM_LINK(Entry);\r |
613 | break;\r |
614 | }\r |
615 | Index++;\r |
616 | }\r |
617 | \r |
618 | Status = BootOptionStart (BootOption);\r |
619 | }\r |
620 | }\r |
11c20f4e |
621 | // Should never go here\r |
ea46ebbe |
622 | }\r |