]> git.proxmox.com Git - mirror_edk2.git/blame - IntelFrameworkModulePkg/Universal/BdsDxe/FrontPage.c
Add array index check to avoid potential buffer overflow.
[mirror_edk2.git] / IntelFrameworkModulePkg / Universal / BdsDxe / FrontPage.c
CommitLineData
5c08e117 1/** @file\r
2 FrontPage routines to handle the callbacks and browser calls\r
3\r
5d7c1609 4Copyright (c) 2004 - 2009, Intel Corporation. <BR>\r
5c08e117 5All rights reserved. This program and the accompanying materials\r
6are licensed and made available under the terms and conditions of the BSD License\r
7which accompanies this distribution. The full text of the license may be found at\r
8http://opensource.org/licenses/bsd-license.php\r
9\r
10THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
11WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
12\r
13**/\r
14\r
15#include "Bds.h"\r
16#include "FrontPage.h"\r
ad5a96ab 17#include "Language.h"\r
5c08e117 18\r
19EFI_GUID mFrontPageGuid = FRONT_PAGE_FORMSET_GUID;\r
20\r
21BOOLEAN gConnectAllHappened = FALSE;\r
22UINTN gCallbackKey;\r
23\r
5c08e117 24EFI_FORM_BROWSER2_PROTOCOL *gFormBrowser2;\r
5c08e117 25\r
26FRONT_PAGE_CALLBACK_DATA gFrontPagePrivate = {\r
27 FRONT_PAGE_CALLBACK_DATA_SIGNATURE,\r
28 NULL,\r
29 NULL,\r
30 NULL,\r
31 {\r
32 FakeExtractConfig,\r
33 FakeRouteConfig,\r
34 FrontPageCallback\r
35 }\r
36};\r
37\r
f6f910dd 38HII_VENDOR_DEVICE_PATH mFrontPageHiiVendorDevicePath = {\r
39 {\r
40 {\r
41 HARDWARE_DEVICE_PATH,\r
42 HW_VENDOR_DP,\r
43 {\r
44 (UINT8) (sizeof (VENDOR_DEVICE_PATH)),\r
45 (UINT8) ((sizeof (VENDOR_DEVICE_PATH)) >> 8)\r
46 }\r
47 },\r
48 //\r
49 // {8E6D99EE-7531-48f8-8745-7F6144468FF2}\r
50 //\r
51 { 0x8e6d99ee, 0x7531, 0x48f8, { 0x87, 0x45, 0x7f, 0x61, 0x44, 0x46, 0x8f, 0xf2 } }\r
52 },\r
53 {\r
54 END_DEVICE_PATH_TYPE,\r
55 END_ENTIRE_DEVICE_PATH_SUBTYPE,\r
128efbbc 56 {\r
f6f910dd 57 (UINT8) (END_DEVICE_PATH_LENGTH),\r
58 (UINT8) ((END_DEVICE_PATH_LENGTH) >> 8)\r
59 }\r
60 }\r
61};\r
62\r
5c08e117 63/**\r
64 This function allows a caller to extract the current configuration for one\r
65 or more named elements from the target driver.\r
66\r
67\r
68 @param This Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL.\r
69 @param Request A null-terminated Unicode string in <ConfigRequest> format.\r
70 @param Progress On return, points to a character in the Request string.\r
71 Points to the string's null terminator if request was successful.\r
72 Points to the most recent '&' before the first failing name/value\r
73 pair (or the beginning of the string if the failure is in the\r
74 first name/value pair) if the request was not successful.\r
75 @param Results A null-terminated Unicode string in <ConfigAltResp> format which\r
76 has all values filled in for the names in the Request string.\r
77 String to be allocated by the called function.\r
78\r
79 @retval EFI_SUCCESS The Results is filled with the requested values.\r
80 @retval EFI_OUT_OF_RESOURCES Not enough memory to store the results.\r
81 @retval EFI_INVALID_PARAMETER Request is NULL, illegal syntax, or unknown name.\r
82 @retval EFI_NOT_FOUND Routing data doesn't match any storage in this driver.\r
83\r
84**/\r
85EFI_STATUS\r
86EFIAPI\r
87FakeExtractConfig (\r
88 IN CONST EFI_HII_CONFIG_ACCESS_PROTOCOL *This,\r
89 IN CONST EFI_STRING Request,\r
90 OUT EFI_STRING *Progress,\r
91 OUT EFI_STRING *Results\r
92 )\r
93{\r
dedfc3bb
LG
94 if (Request == NULL || Progress == NULL || Results == NULL) {\r
95 return EFI_INVALID_PARAMETER;\r
96 }\r
cdcc36ae 97 *Progress = Request;\r
5c08e117 98 return EFI_NOT_FOUND;\r
99}\r
100\r
101/**\r
102 This function processes the results of changes in configuration.\r
103\r
104\r
105 @param This Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL.\r
106 @param Configuration A null-terminated Unicode string in <ConfigResp> format.\r
107 @param Progress A pointer to a string filled in with the offset of the most\r
108 recent '&' before the first failing name/value pair (or the\r
109 beginning of the string if the failure is in the first\r
110 name/value pair) or the terminating NULL if all was successful.\r
111\r
112 @retval EFI_SUCCESS The Results is processed successfully.\r
113 @retval EFI_INVALID_PARAMETER Configuration is NULL.\r
114 @retval EFI_NOT_FOUND Routing data doesn't match any storage in this driver.\r
115\r
116**/\r
117EFI_STATUS\r
118EFIAPI\r
119FakeRouteConfig (\r
120 IN CONST EFI_HII_CONFIG_ACCESS_PROTOCOL *This,\r
121 IN CONST EFI_STRING Configuration,\r
122 OUT EFI_STRING *Progress\r
123 )\r
124{\r
dedfc3bb 125 if (Configuration == NULL || Progress == NULL) {\r
cdcc36ae
LG
126 return EFI_INVALID_PARAMETER;\r
127 }\r
128\r
dedfc3bb
LG
129 *Progress = Configuration;\r
130 if (!HiiIsConfigHdrMatch (Configuration, &mBootMaintGuid, mBootMaintStorageName)\r
131 && !HiiIsConfigHdrMatch (Configuration, &mFileExplorerGuid, mFileExplorerStorageName)) {\r
132 return EFI_NOT_FOUND;\r
cdcc36ae
LG
133 }\r
134\r
dedfc3bb
LG
135 *Progress = Configuration + StrLen (Configuration);\r
136 return EFI_SUCCESS;\r
5c08e117 137}\r
138\r
139/**\r
140 This function processes the results of changes in configuration.\r
141\r
142\r
143 @param This Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL.\r
144 @param Action Specifies the type of action taken by the browser.\r
145 @param QuestionId A unique value which is sent to the original exporting driver\r
146 so that it can identify the type of data to expect.\r
147 @param Type The type of value for the question.\r
148 @param Value A pointer to the data being sent to the original exporting driver.\r
149 @param ActionRequest On return, points to the action requested by the callback function.\r
150\r
151 @retval EFI_SUCCESS The callback successfully handled the action.\r
152 @retval EFI_OUT_OF_RESOURCES Not enough storage is available to hold the variable and its data.\r
153 @retval EFI_DEVICE_ERROR The variable could not be saved.\r
154 @retval EFI_UNSUPPORTED The specified Action is not supported by the callback.\r
155\r
156**/\r
157EFI_STATUS\r
158EFIAPI\r
159FrontPageCallback (\r
160 IN CONST EFI_HII_CONFIG_ACCESS_PROTOCOL *This,\r
161 IN EFI_BROWSER_ACTION Action,\r
162 IN EFI_QUESTION_ID QuestionId,\r
163 IN UINT8 Type,\r
164 IN EFI_IFR_TYPE_VALUE *Value,\r
165 OUT EFI_BROWSER_ACTION_REQUEST *ActionRequest\r
166 )\r
167{\r
168 CHAR8 *LanguageString;\r
169 CHAR8 *LangCode;\r
a3a795af 170 CHAR8 *Lang;\r
5c08e117 171 UINTN Index;\r
172 EFI_STATUS Status;\r
a3a795af 173 CHAR8 *PlatformSupportedLanguages;\r
174 CHAR8 *BestLanguage;\r
5c08e117 175\r
176 if ((Value == NULL) || (ActionRequest == NULL)) {\r
177 return EFI_INVALID_PARAMETER;\r
178 }\r
179\r
180 gCallbackKey = QuestionId;\r
181\r
182 //\r
183 // The first 4 entries in the Front Page are to be GUARANTEED to remain constant so IHV's can\r
184 // describe to their customers in documentation how to find their setup information (namely\r
185 // under the device manager and specific buckets)\r
186 //\r
187 switch (QuestionId) {\r
188 case FRONT_PAGE_KEY_CONTINUE:\r
189 //\r
190 // This is the continue - clear the screen and return an error to get out of FrontPage loop\r
191 //\r
192 break;\r
193\r
194 case FRONT_PAGE_KEY_LANGUAGE:\r
195 //\r
196 // Collect the languages from what our current Language support is based on our VFR\r
197 //\r
cb7d01c0 198 LanguageString = HiiGetSupportedLanguages (gFrontPagePrivate.HiiHandle);\r
5c08e117 199 ASSERT (LanguageString != NULL);\r
a3a795af 200 //\r
201 // Allocate working buffer for RFC 4646 language in supported LanguageString.\r
202 //\r
203 Lang = AllocatePool (AsciiStrSize (LanguageString));\r
204 ASSERT (Lang != NULL);\r
128efbbc 205\r
5c08e117 206 Index = 0;\r
207 LangCode = LanguageString;\r
208 while (*LangCode != 0) {\r
0f268521 209 GetNextLanguage (&LangCode, Lang);\r
5c08e117 210\r
211 if (Index == Value->u8) {\r
212 break;\r
213 }\r
214\r
215 Index++;\r
216 }\r
217\r
a3a795af 218 PlatformSupportedLanguages = GetEfiGlobalVariable (L"PlatformLangCodes");\r
219 if (PlatformSupportedLanguages == NULL) {\r
220 PlatformSupportedLanguages = AllocateCopyPool (\r
221 AsciiStrSize ((CHAR8 *) PcdGetPtr (PcdUefiVariableDefaultPlatformLangCodes)),\r
222 (CHAR8 *) PcdGetPtr (PcdUefiVariableDefaultPlatformLangCodes)\r
223 );\r
224 ASSERT (PlatformSupportedLanguages != NULL);\r
225 }\r
128efbbc 226\r
a3a795af 227 //\r
228 // Select the best language in platform supported Language.\r
229 //\r
230 BestLanguage = GetBestLanguage (\r
231 PlatformSupportedLanguages,\r
232 FALSE,\r
233 Lang,\r
234 NULL\r
235 );\r
236 if (BestLanguage != NULL) {\r
237 Status = gRT->SetVariable (\r
238 L"PlatformLang",\r
239 &gEfiGlobalVariableGuid,\r
240 EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS,\r
241 AsciiStrSize (BestLanguage),\r
242 Lang\r
243 );\r
b96c2106 244 ASSERT_EFI_ERROR(Status);\r
a3a795af 245 FreePool (BestLanguage);\r
d357145f 246 } else {\r
247 ASSERT (FALSE);\r
5c08e117 248 }\r
d357145f 249\r
a3a795af 250 FreePool (PlatformSupportedLanguages);\r
251 FreePool (Lang);\r
5c08e117 252 FreePool (LanguageString);\r
253 break;\r
254\r
255 case FRONT_PAGE_KEY_BOOT_MANAGER:\r
256 //\r
257 // Boot Manager\r
258 //\r
259 break;\r
260\r
261 case FRONT_PAGE_KEY_DEVICE_MANAGER:\r
262 //\r
263 // Device Manager\r
264 //\r
265 break;\r
266\r
267 case FRONT_PAGE_KEY_BOOT_MAINTAIN:\r
268 //\r
269 // Boot Maintenance Manager\r
270 //\r
271 break;\r
272\r
273 default:\r
274 gCallbackKey = 0;\r
275 break;\r
276 }\r
277\r
278 *ActionRequest = EFI_BROWSER_ACTION_REQUEST_EXIT;\r
279\r
280 return EFI_SUCCESS;\r
281}\r
282\r
283/**\r
284 Initialize HII information for the FrontPage\r
285\r
286\r
287 @param InitializeHiiData TRUE if HII elements need to be initialized.\r
288\r
289 @retval EFI_SUCCESS The operation is successful.\r
290 @retval EFI_DEVICE_ERROR If the dynamic opcode creation failed.\r
291\r
292**/\r
293EFI_STATUS\r
294InitializeFrontPage (\r
295 IN BOOLEAN InitializeHiiData\r
296 )\r
297{\r
298 EFI_STATUS Status;\r
5c08e117 299 CHAR8 *LanguageString;\r
300 CHAR8 *LangCode;\r
a3a795af 301 CHAR8 *Lang;\r
302 CHAR8 *CurrentLang;\r
303 CHAR8 *BestLanguage;\r
5c08e117 304 UINTN OptionCount;\r
5c08e117 305 CHAR16 *StringBuffer;\r
5c08e117 306 EFI_HII_HANDLE HiiHandle;\r
75bf9d0e
LG
307 VOID *OptionsOpCodeHandle;\r
308 VOID *StartOpCodeHandle;\r
309 VOID *EndOpCodeHandle;\r
310 EFI_IFR_GUID_LABEL *StartLabel;\r
311 EFI_IFR_GUID_LABEL *EndLabel;\r
312 BOOLEAN FirstFlag;\r
5c08e117 313\r
314 if (InitializeHiiData) {\r
315 //\r
316 // Initialize the Device Manager\r
317 //\r
318 InitializeDeviceManager ();\r
319\r
320 //\r
321 // Initialize the Device Manager\r
322 //\r
323 InitializeBootManager ();\r
324\r
325 gCallbackKey = 0;\r
326\r
327 //\r
328 // Locate Hii relative protocols\r
329 //\r
5c08e117 330 Status = gBS->LocateProtocol (&gEfiFormBrowser2ProtocolGuid, NULL, (VOID **) &gFormBrowser2);\r
331 if (EFI_ERROR (Status)) {\r
332 return Status;\r
333 }\r
334\r
5c08e117 335 //\r
f6f910dd 336 // Install Device Path Protocol and Config Access protocol to driver handle\r
5c08e117 337 //\r
f6f910dd 338 Status = gBS->InstallMultipleProtocolInterfaces (\r
5c08e117 339 &gFrontPagePrivate.DriverHandle,\r
f6f910dd 340 &gEfiDevicePathProtocolGuid,\r
341 &mFrontPageHiiVendorDevicePath,\r
5c08e117 342 &gEfiHiiConfigAccessProtocolGuid,\r
f6f910dd 343 &gFrontPagePrivate.ConfigAccess,\r
344 NULL\r
5c08e117 345 );\r
346 ASSERT_EFI_ERROR (Status);\r
347\r
348 //\r
349 // Publish our HII data\r
350 //\r
cb7d01c0 351 gFrontPagePrivate.HiiHandle = HiiAddPackages (\r
352 &mFrontPageGuid,\r
353 gFrontPagePrivate.DriverHandle,\r
354 FrontPageVfrBin,\r
355 BdsDxeStrings,\r
356 NULL\r
357 );\r
358 if (gFrontPagePrivate.HiiHandle == NULL) {\r
359 return EFI_OUT_OF_RESOURCES;\r
5c08e117 360 }\r
361 }\r
362\r
5c08e117 363\r
364 //\r
75bf9d0e 365 // Init OpCode Handle and Allocate space for creation of UpdateData Buffer\r
5c08e117 366 //\r
75bf9d0e
LG
367 StartOpCodeHandle = HiiAllocateOpCodeHandle ();\r
368 ASSERT (StartOpCodeHandle != NULL);\r
5c08e117 369\r
75bf9d0e
LG
370 EndOpCodeHandle = HiiAllocateOpCodeHandle ();\r
371 ASSERT (EndOpCodeHandle != NULL);\r
372\r
373 OptionsOpCodeHandle = HiiAllocateOpCodeHandle ();\r
374 ASSERT (OptionsOpCodeHandle != NULL);\r
375 //\r
376 // Create Hii Extend Label OpCode as the start opcode\r
377 //\r
378 StartLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (StartOpCodeHandle, &gEfiIfrTianoGuid, NULL, sizeof (EFI_IFR_GUID_LABEL));\r
379 StartLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL;\r
380 StartLabel->Number = LABEL_SELECT_LANGUAGE;\r
381\r
382 //\r
383 // Create Hii Extend Label OpCode as the end opcode\r
384 //\r
385 EndLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (EndOpCodeHandle, &gEfiIfrTianoGuid, NULL, sizeof (EFI_IFR_GUID_LABEL));\r
386 EndLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL;\r
387 EndLabel->Number = LABEL_END;\r
5c08e117 388\r
389 //\r
390 // Collect the languages from what our current Language support is based on our VFR\r
391 //\r
392 HiiHandle = gFrontPagePrivate.HiiHandle;\r
cb7d01c0 393 LanguageString = HiiGetSupportedLanguages (HiiHandle);\r
5c08e117 394 ASSERT (LanguageString != NULL);\r
a3a795af 395 //\r
396 // Allocate working buffer for RFC 4646 language in supported LanguageString.\r
397 //\r
398 Lang = AllocatePool (AsciiStrSize (LanguageString));\r
399 ASSERT (Lang != NULL);\r
400\r
401 CurrentLang = GetEfiGlobalVariable (L"PlatformLang");\r
402 //\r
403 // Select the best language in LanguageString as the default one.\r
404 //\r
405 BestLanguage = GetBestLanguage (\r
406 LanguageString,\r
407 FALSE,\r
408 (CurrentLang != NULL) ? CurrentLang : "",\r
409 (CHAR8 *) PcdGetPtr (PcdUefiVariableDefaultPlatformLang),\r
410 LanguageString,\r
411 NULL\r
412 );\r
413 //\r
414 // BestLanguage must be selected as it is the first language in LanguageString by default\r
415 //\r
416 ASSERT (BestLanguage != NULL);\r
5c08e117 417\r
75bf9d0e
LG
418 OptionCount = 0;\r
419 LangCode = LanguageString;\r
420 FirstFlag = FALSE;\r
128efbbc 421\r
75bf9d0e
LG
422 if (gFrontPagePrivate.LanguageToken == NULL) {\r
423 while (*LangCode != 0) {\r
0b3f5a4d 424 GetNextLanguage (&LangCode, Lang);\r
75bf9d0e
LG
425 OptionCount ++;\r
426 }\r
427 gFrontPagePrivate.LanguageToken = AllocatePool (OptionCount * sizeof (EFI_STRING_ID));\r
428 ASSERT (gFrontPagePrivate.LanguageToken != NULL);\r
429 FirstFlag = TRUE;\r
430 }\r
431\r
5c08e117 432 OptionCount = 0;\r
433 LangCode = LanguageString;\r
434 while (*LangCode != 0) {\r
0f268521 435 GetNextLanguage (&LangCode, Lang);\r
5c08e117 436\r
75bf9d0e 437 if (FirstFlag) {\r
33bb6326 438 StringBuffer = HiiGetString (HiiHandle, PRINTABLE_LANGUAGE_NAME_STRING_ID, Lang);\r
439 ASSERT_EFI_ERROR (StringBuffer != NULL);\r
5c08e117 440\r
75bf9d0e
LG
441 //\r
442 // Save the string Id for each language\r
443 //\r
444 gFrontPagePrivate.LanguageToken[OptionCount] = HiiSetString (HiiHandle, 0, StringBuffer, NULL);\r
5c08e117 445 FreePool (StringBuffer);\r
5c08e117 446 }\r
447\r
a3a795af 448 if (AsciiStrCmp (Lang, BestLanguage) == 0) {\r
75bf9d0e
LG
449 HiiCreateOneOfOptionOpCode (\r
450 OptionsOpCodeHandle,\r
451 gFrontPagePrivate.LanguageToken[OptionCount],\r
452 EFI_IFR_OPTION_DEFAULT,\r
453 EFI_IFR_NUMERIC_SIZE_1,\r
454 (UINT8) OptionCount\r
455 );\r
5c08e117 456 } else {\r
75bf9d0e
LG
457 HiiCreateOneOfOptionOpCode (\r
458 OptionsOpCodeHandle,\r
459 gFrontPagePrivate.LanguageToken[OptionCount],\r
460 0,\r
461 EFI_IFR_NUMERIC_SIZE_1,\r
462 (UINT8) OptionCount\r
463 );\r
5c08e117 464 }\r
5c08e117 465\r
466 OptionCount++;\r
467 }\r
468\r
a3a795af 469 if (CurrentLang != NULL) {\r
470 FreePool (CurrentLang);\r
471 }\r
472 FreePool (BestLanguage);\r
473 FreePool (Lang);\r
5c08e117 474 FreePool (LanguageString);\r
475\r
75bf9d0e
LG
476 HiiCreateOneOfOpCode (\r
477 StartOpCodeHandle,\r
5c08e117 478 FRONT_PAGE_KEY_LANGUAGE,\r
479 0,\r
480 0,\r
481 STRING_TOKEN (STR_LANGUAGE_SELECT),\r
482 STRING_TOKEN (STR_LANGUAGE_SELECT_HELP),\r
483 EFI_IFR_FLAG_CALLBACK,\r
484 EFI_IFR_NUMERIC_SIZE_1,\r
75bf9d0e
LG
485 OptionsOpCodeHandle,\r
486 NULL\r
5c08e117 487 );\r
488\r
75bf9d0e 489 Status = HiiUpdateForm (\r
5c08e117 490 HiiHandle,\r
491 &mFrontPageGuid,\r
492 FRONT_PAGE_FORM_ID,\r
75bf9d0e
LG
493 StartOpCodeHandle, // LABEL_SELECT_LANGUAGE\r
494 EndOpCodeHandle // LABEL_END\r
5c08e117 495 );\r
496\r
75bf9d0e
LG
497 HiiFreeOpCodeHandle (StartOpCodeHandle);\r
498 HiiFreeOpCodeHandle (EndOpCodeHandle);\r
499 HiiFreeOpCodeHandle (OptionsOpCodeHandle);\r
5c08e117 500 return Status;\r
501}\r
502\r
503/**\r
504 Call the browser and display the front page\r
505\r
506 @return Status code that will be returned by\r
507 EFI_FORM_BROWSER2_PROTOCOL.SendForm ().\r
508\r
509**/\r
510EFI_STATUS\r
511CallFrontPage (\r
512 VOID\r
513 )\r
514{\r
515 EFI_STATUS Status;\r
516 EFI_BROWSER_ACTION_REQUEST ActionRequest;\r
517\r
518 //\r
519 // Begin waiting for USER INPUT\r
520 //\r
521 REPORT_STATUS_CODE (\r
522 EFI_PROGRESS_CODE,\r
523 (EFI_SOFTWARE_DXE_BS_DRIVER | EFI_SW_PC_INPUT_WAIT)\r
524 );\r
525\r
526 ActionRequest = EFI_BROWSER_ACTION_REQUEST_NONE;\r
527 Status = gFormBrowser2->SendForm (\r
528 gFormBrowser2,\r
529 &gFrontPagePrivate.HiiHandle,\r
530 1,\r
f0da9ffb 531 &mFrontPageGuid,\r
5c08e117 532 0,\r
533 NULL,\r
534 &ActionRequest\r
535 );\r
536 //\r
baf46e70 537 // Check whether user change any option setting which needs a reset to be effective\r
5c08e117 538 //\r
539 if (ActionRequest == EFI_BROWSER_ACTION_REQUEST_RESET) {\r
540 EnableResetRequired ();\r
541 }\r
542\r
543 return Status;\r
544}\r
545\r
546/**\r
547 Acquire the string associated with the ProducerGuid and return it.\r
548\r
549\r
550 @param ProducerGuid The Guid to search the HII database for\r
551 @param Token The token value of the string to extract\r
552 @param String The string that is extracted\r
553\r
554 @retval EFI_SUCCESS The function returns EFI_SUCCESS always.\r
555\r
556**/\r
557EFI_STATUS\r
558GetProducerString (\r
559 IN EFI_GUID *ProducerGuid,\r
560 IN EFI_STRING_ID Token,\r
561 OUT CHAR16 **String\r
562 )\r
563{\r
cb7d01c0 564 EFI_STRING TmpString;\r
5c08e117 565\r
cb7d01c0 566 TmpString = HiiGetPackageString (ProducerGuid, Token, NULL);\r
567 if (TmpString == NULL) {\r
5c08e117 568 *String = GetStringById (STRING_TOKEN (STR_MISSING_STRING));\r
cb7d01c0 569 } else {\r
570 *String = TmpString;\r
5c08e117 571 }\r
572\r
573 return EFI_SUCCESS;\r
574}\r
575\r
576/**\r
13078b3f 577 Convert Processor Frequency Data to a string.\r
5c08e117 578\r
579 @param ProcessorFrequency The frequency data to process\r
6cfbf7ad 580 @param Base10Exponent The exponent based on 10\r
5c08e117 581 @param String The string that is created\r
582\r
583**/\r
584VOID\r
585ConvertProcessorToString (\r
6cfbf7ad 586 IN UINT16 ProcessorFrequency,\r
587 IN UINT16 Base10Exponent,\r
588 OUT CHAR16 **String\r
5c08e117 589 )\r
590{\r
591 CHAR16 *StringBuffer;\r
592 UINTN Index;\r
593 UINT32 FreqMhz;\r
594\r
6cfbf7ad 595 if (Base10Exponent >= 6) {\r
596 FreqMhz = ProcessorFrequency;\r
597 for (Index = 0; Index < (UINTN) (Base10Exponent - 6); Index++) {\r
5c08e117 598 FreqMhz *= 10;\r
599 }\r
600 } else {\r
601 FreqMhz = 0;\r
602 }\r
603\r
604 StringBuffer = AllocateZeroPool (0x20);\r
605 ASSERT (StringBuffer != NULL);\r
606 Index = UnicodeValueToString (StringBuffer, LEFT_JUSTIFY, FreqMhz / 1000, 3);\r
607 StrCat (StringBuffer, L".");\r
608 UnicodeValueToString (StringBuffer + Index + 1, PREFIX_ZERO, (FreqMhz % 1000) / 10, 2);\r
609 StrCat (StringBuffer, L" GHz");\r
5c08e117 610 *String = (CHAR16 *) StringBuffer;\r
5c08e117 611 return ;\r
612}\r
613\r
6cfbf7ad 614\r
5c08e117 615/**\r
616 Convert Memory Size to a string.\r
617\r
618 @param MemorySize The size of the memory to process\r
619 @param String The string that is created\r
620\r
621**/\r
622VOID\r
623ConvertMemorySizeToString (\r
624 IN UINT32 MemorySize,\r
625 OUT CHAR16 **String\r
626 )\r
627{\r
628 CHAR16 *StringBuffer;\r
629\r
630 StringBuffer = AllocateZeroPool (0x20);\r
631 ASSERT (StringBuffer != NULL);\r
632 UnicodeValueToString (StringBuffer, LEFT_JUSTIFY, MemorySize, 6);\r
633 StrCat (StringBuffer, L" MB RAM");\r
634\r
635 *String = (CHAR16 *) StringBuffer;\r
636\r
637 return ;\r
638}\r
639\r
6cfbf7ad 640/**\r
641\r
642 Acquire the string associated with the Index from smbios structure and return it.\r
643 The caller is responsible for free the string buffer.\r
644\r
645 @param OptionalStrStart The start position to search the string\r
646 @param Index The index of the string to extract\r
647 @param String The string that is extracted\r
648\r
649 @retval EFI_SUCCESS The function returns EFI_SUCCESS always.\r
650\r
651**/\r
652EFI_STATUS\r
653GetOptionalStringByIndex (\r
654 IN CHAR8 *OptionalStrStart,\r
655 IN UINT8 Index,\r
656 OUT CHAR16 **String\r
657 )\r
658{\r
659 UINT8 StrNum;\r
660 UINTN CurrentStrLen;\r
661 CHAR8* CharInStr;\r
662 EFI_STATUS Status;\r
663\r
664 StrNum = 0;\r
665 Status = EFI_NOT_FOUND;\r
666 CharInStr = OptionalStrStart;\r
667\r
668 if (Index != 1) {\r
669 CurrentStrLen = 0;\r
670 //\r
671 // look for the two consecutive zeros, check the string limit by the way.\r
672 //\r
673 while (*CharInStr != 0 || *(CharInStr+1) != 0) { \r
674 if (*CharInStr == 0) {\r
675 StrNum += 1;\r
676 CharInStr++;\r
677 }\r
678 \r
679 if (StrNum == Index) {\r
680 Status = EFI_SUCCESS;\r
681 break;\r
682 }\r
683 \r
684 CurrentStrLen = AsciiStrLen(CharInStr);\r
685 \r
686 //\r
687 // forward the pointer\r
688 //\r
689 OptionalStrStart = CharInStr;\r
690 CharInStr += CurrentStrLen;\r
691 }\r
692 \r
693 if (EFI_ERROR (Status)) {\r
694 *String = GetStringById (STRING_TOKEN (STR_MISSING_STRING));\r
695 return Status;\r
696 }\r
697 } else {\r
698 CurrentStrLen = AsciiStrLen(CharInStr);\r
699 }\r
700\r
701 *String = AllocatePool((CurrentStrLen + 1)*sizeof(CHAR16));\r
702 AsciiStrToUnicodeStr(OptionalStrStart, *String);\r
703\r
704 return EFI_SUCCESS;\r
705}\r
706\r
707\r
5c08e117 708/**\r
709 Update the banner information for the Front Page based on DataHub information.\r
710\r
711**/\r
712VOID\r
713UpdateFrontPageStrings (\r
714 VOID\r
715 )\r
716{\r
6cfbf7ad 717 UINT8 StrIndex;\r
5c08e117 718 CHAR16 *NewString;\r
5c08e117 719 BOOLEAN Find[5];\r
6cfbf7ad 720 EFI_STATUS Status;\r
721 EFI_STRING_ID TokenToUpdate;\r
722 EFI_SMBIOS_HANDLE SmbiosHandle;\r
723 EFI_SMBIOS_PROTOCOL *Smbios;\r
724 SMBIOS_TABLE_TYPE0 *Type0Record;\r
725 SMBIOS_TABLE_TYPE1 *Type1Record;\r
726 SMBIOS_TABLE_TYPE4 *Type4Record;\r
727 SMBIOS_TABLE_TYPE19 *Type19Record;\r
728 EFI_SMBIOS_TABLE_HEADER *Record;\r
5c08e117 729\r
730 ZeroMem (Find, sizeof (Find));\r
731\r
732 //\r
733 // Update Front Page strings\r
734 //\r
735 Status = gBS->LocateProtocol (\r
6cfbf7ad 736 &gEfiSmbiosProtocolGuid,\r
5c08e117 737 NULL,\r
6cfbf7ad 738 (VOID **) &Smbios\r
5c08e117 739 );\r
740 ASSERT_EFI_ERROR (Status);\r
741\r
6cfbf7ad 742 SmbiosHandle = 0;\r
5c08e117 743 do {\r
6cfbf7ad 744 Status = Smbios->GetNext (Smbios, &SmbiosHandle, NULL, &Record, NULL);\r
745 if (EFI_ERROR(Status)) {\r
b55f8b01
LG
746 break;\r
747 }\r
5c08e117 748\r
6cfbf7ad 749 if (Record->Type == EFI_SMBIOS_TYPE_BIOS_INFORMATION) {\r
750 Type0Record = (SMBIOS_TABLE_TYPE0 *) Record;\r
751 StrIndex = Type0Record->BiosVersion;\r
ed79462b 752 GetOptionalStringByIndex ((CHAR8*)((UINT8*)Type0Record + Type0Record->Hdr.Length), StrIndex, &NewString);\r
6cfbf7ad 753 TokenToUpdate = STRING_TOKEN (STR_FRONT_PAGE_BIOS_VERSION);\r
754 HiiSetString (gFrontPagePrivate.HiiHandle, TokenToUpdate, NewString, NULL);\r
755 FreePool (NewString);\r
756 Find[0] = TRUE;\r
757 } \r
758\r
759 if (Record->Type == EFI_SMBIOS_TYPE_SYSTEM_INFORMATION) {\r
760 Type1Record = (SMBIOS_TABLE_TYPE1 *) Record;\r
761 StrIndex = Type1Record->ProductName;\r
ed79462b 762 GetOptionalStringByIndex ((CHAR8*)((UINT8*)Type1Record + Type1Record->Hdr.Length), StrIndex, &NewString);\r
6cfbf7ad 763 TokenToUpdate = STRING_TOKEN (STR_FRONT_PAGE_COMPUTER_MODEL);\r
764 HiiSetString (gFrontPagePrivate.HiiHandle, TokenToUpdate, NewString, NULL);\r
765 FreePool (NewString);\r
766 Find[1] = TRUE;\r
5c08e117 767 }\r
6cfbf7ad 768 \r
769 if (Record->Type == EFI_SMBIOS_TYPE_PROCESSOR_INFORMATION) {\r
770 Type4Record = (SMBIOS_TABLE_TYPE4 *) Record;\r
771 StrIndex = Type4Record->ProcessorVersion;\r
ed79462b 772 GetOptionalStringByIndex ((CHAR8*)((UINT8*)Type4Record + Type4Record->Hdr.Length), StrIndex, &NewString);\r
6cfbf7ad 773 TokenToUpdate = STRING_TOKEN (STR_FRONT_PAGE_CPU_MODEL);\r
774 HiiSetString (gFrontPagePrivate.HiiHandle, TokenToUpdate, NewString, NULL);\r
775 FreePool (NewString);\r
776 Find[2] = TRUE;\r
777 } \r
778\r
779 if (Record->Type == EFI_SMBIOS_TYPE_PROCESSOR_INFORMATION) {\r
780 Type4Record = (SMBIOS_TABLE_TYPE4 *) Record;\r
781 ConvertProcessorToString(Type4Record->CurrentSpeed, 6, &NewString);\r
782 TokenToUpdate = STRING_TOKEN (STR_FRONT_PAGE_CPU_SPEED);\r
783 HiiSetString (gFrontPagePrivate.HiiHandle, TokenToUpdate, NewString, NULL);\r
784 FreePool (NewString);\r
785 Find[3] = TRUE;\r
786 } \r
787\r
788 if ( Record->Type == EFI_SMBIOS_TYPE_MEMORY_ARRAY_MAPPED_ADDRESS ) {\r
789 Type19Record = (SMBIOS_TABLE_TYPE19 *) Record;\r
790 ConvertMemorySizeToString (\r
791 (UINT32)(RShiftU64((Type19Record->EndingAddress - Type19Record->StartingAddress + 1), 10)),\r
792 &NewString\r
793 );\r
794 TokenToUpdate = STRING_TOKEN (STR_FRONT_PAGE_MEMORY_SIZE);\r
795 HiiSetString (gFrontPagePrivate.HiiHandle, TokenToUpdate, NewString, NULL);\r
796 FreePool (NewString);\r
797 Find[4] = TRUE; \r
798 }\r
799 } while ( !(Find[0] && Find[1] && Find[2] && Find[3] && Find[4]));\r
5c08e117 800 return ;\r
801}\r
802\r
6cfbf7ad 803\r
5c08e117 804/**\r
805 Function waits for a given event to fire, or for an optional timeout to expire.\r
806\r
6cfbf7ad 807 @param Event The event to wait for\r
808 @param Timeout An optional timeout value in 100 ns units.\r
5c08e117 809\r
810 @retval EFI_SUCCESS Event fired before Timeout expired.\r
811 @retval EFI_TIME_OUT Timout expired before Event fired..\r
812\r
813**/\r
814EFI_STATUS\r
815WaitForSingleEvent (\r
816 IN EFI_EVENT Event,\r
817 IN UINT64 Timeout OPTIONAL\r
818 )\r
819{\r
5c08e117 820 UINTN Index;\r
6cfbf7ad 821 EFI_STATUS Status;\r
5c08e117 822 EFI_EVENT TimerEvent;\r
823 EFI_EVENT WaitList[2];\r
824\r
825 if (Timeout != 0) {\r
826 //\r
827 // Create a timer event\r
828 //\r
829 Status = gBS->CreateEvent (EVT_TIMER, 0, NULL, NULL, &TimerEvent);\r
830 if (!EFI_ERROR (Status)) {\r
831 //\r
832 // Set the timer event\r
833 //\r
834 gBS->SetTimer (\r
6cfbf7ad 835 TimerEvent,\r
836 TimerRelative,\r
837 Timeout\r
838 );\r
5c08e117 839\r
840 //\r
841 // Wait for the original event or the timer\r
842 //\r
843 WaitList[0] = Event;\r
844 WaitList[1] = TimerEvent;\r
845 Status = gBS->WaitForEvent (2, WaitList, &Index);\r
846 gBS->CloseEvent (TimerEvent);\r
847\r
848 //\r
849 // If the timer expired, change the return to timed out\r
850 //\r
851 if (!EFI_ERROR (Status) && Index == 1) {\r
852 Status = EFI_TIMEOUT;\r
853 }\r
854 }\r
855 } else {\r
856 //\r
857 // No timeout... just wait on the event\r
858 //\r
859 Status = gBS->WaitForEvent (1, &Event, &Index);\r
860 ASSERT (!EFI_ERROR (Status));\r
861 ASSERT (Index == 0);\r
862 }\r
863\r
864 return Status;\r
865}\r
866\r
867/**\r
868 Function show progress bar to wait for user input.\r
869\r
870\r
6cfbf7ad 871 @param TimeoutDefault The fault time out value before the system continue to boot.\r
5c08e117 872\r
873 @retval EFI_SUCCESS User pressed some key except "Enter"\r
baf46e70 874 @retval EFI_TIME_OUT Timeout expired or user press "Enter"\r
5c08e117 875\r
876**/\r
877EFI_STATUS\r
878ShowProgress (\r
879 IN UINT16 TimeoutDefault\r
880 )\r
881{\r
5c08e117 882 CHAR16 *TmpStr;\r
6cfbf7ad 883 UINT16 TimeoutRemain;\r
884 EFI_STATUS Status;\r
885 EFI_INPUT_KEY Key;\r
5c08e117 886 EFI_GRAPHICS_OUTPUT_BLT_PIXEL Foreground;\r
887 EFI_GRAPHICS_OUTPUT_BLT_PIXEL Background;\r
888 EFI_GRAPHICS_OUTPUT_BLT_PIXEL Color;\r
5c08e117 889\r
890 if (TimeoutDefault == 0) {\r
891 return EFI_TIMEOUT;\r
892 }\r
893\r
894 DEBUG ((EFI_D_INFO, "\n\nStart showing progress bar... Press any key to stop it! ...Zzz....\n"));\r
128efbbc 895\r
5c08e117 896 SetMem (&Foreground, sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL), 0xff);\r
897 SetMem (&Background, sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL), 0x0);\r
898 SetMem (&Color, sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL), 0xff);\r
899\r
900 //\r
901 // Clear the progress status bar first\r
902 //\r
903 TmpStr = GetStringById (STRING_TOKEN (STR_START_BOOT_OPTION));\r
904 if (TmpStr != NULL) {\r
905 PlatformBdsShowProgress (Foreground, Background, TmpStr, Color, 0, 0);\r
906 }\r
907\r
908 TimeoutRemain = TimeoutDefault;\r
909 while (TimeoutRemain != 0) {\r
910 DEBUG ((EFI_D_INFO, "Showing progress bar...Remaining %d second!\n", TimeoutRemain));\r
128efbbc 911\r
5c08e117 912 Status = WaitForSingleEvent (gST->ConIn->WaitForKey, ONE_SECOND);\r
913 if (Status != EFI_TIMEOUT) {\r
914 break;\r
915 }\r
916 TimeoutRemain--;\r
917\r
918 //\r
919 // Show progress\r
920 //\r
921 if (TmpStr != NULL) {\r
922 PlatformBdsShowProgress (\r
923 Foreground,\r
924 Background,\r
925 TmpStr,\r
926 Color,\r
927 ((TimeoutDefault - TimeoutRemain) * 100 / TimeoutDefault),\r
928 0\r
929 );\r
930 }\r
931 }\r
932 gBS->FreePool (TmpStr);\r
933\r
934 //\r
935 // Timeout expired\r
936 //\r
937 if (TimeoutRemain == 0) {\r
938 return EFI_TIMEOUT;\r
939 }\r
940\r
941 //\r
942 // User pressed some key\r
943 //\r
944 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);\r
945 if (EFI_ERROR (Status)) {\r
946 return Status;\r
947 }\r
948\r
949 if (Key.UnicodeChar == CHAR_CARRIAGE_RETURN) {\r
950 //\r
951 // User pressed enter, equivalent to select "continue"\r
952 //\r
953 return EFI_TIMEOUT;\r
954 }\r
955\r
956 return EFI_SUCCESS;\r
957}\r
958\r
959/**\r
960 This function is the main entry of the platform setup entry.\r
961 The function will present the main menu of the system setup,\r
962 this is the platform reference part and can be customize.\r
963\r
964\r
965 @param TimeoutDefault The fault time out value before the system\r
966 continue to boot.\r
967 @param ConnectAllHappened The indicater to check if the connect all have\r
baf46e70 968 already happened.\r
5c08e117 969\r
970**/\r
971VOID\r
972PlatformBdsEnterFrontPage (\r
973 IN UINT16 TimeoutDefault,\r
974 IN BOOLEAN ConnectAllHappened\r
975 )\r
976{\r
977 EFI_STATUS Status;\r
5c08e117 978\r
128efbbc 979 PERF_START (NULL, "BdsTimeOut", "BDS", 0);\r
5c08e117 980 //\r
981 // Indicate if we need connect all in the platform setup\r
982 //\r
983 if (ConnectAllHappened) {\r
984 gConnectAllHappened = TRUE;\r
985 }\r
986\r
987 if (TimeoutDefault != 0xffff) {\r
988 Status = ShowProgress (TimeoutDefault);\r
989\r
990 //\r
991 // Ensure screen is clear when switch Console from Graphics mode to Text mode\r
992 //\r
993 gST->ConOut->EnableCursor (gST->ConOut, TRUE);\r
994 gST->ConOut->ClearScreen (gST->ConOut);\r
995\r
996 if (EFI_ERROR (Status)) {\r
997 //\r
998 // Timeout or user press enter to continue\r
999 //\r
1000 goto Exit;\r
1001 }\r
1002 }\r
1003\r
1004 do {\r
1005\r
1006 InitializeFrontPage (FALSE);\r
1007\r
1008 //\r
1009 // Update Front Page strings\r
1010 //\r
1011 UpdateFrontPageStrings ();\r
1012\r
1013 gCallbackKey = 0;\r
1014 Status = CallFrontPage ();\r
1015\r
1016 //\r
1017 // If gCallbackKey is greater than 1 and less or equal to 5,\r
baf46e70 1018 // it will launch configuration utilities.\r
5c08e117 1019 // 2 = set language\r
1020 // 3 = boot manager\r
1021 // 4 = device manager\r
baf46e70 1022 // 5 = boot maintenance manager\r
5c08e117 1023 //\r
1024 if (gCallbackKey != 0) {\r
1025 REPORT_STATUS_CODE (\r
1026 EFI_PROGRESS_CODE,\r
1027 (EFI_SOFTWARE_DXE_BS_DRIVER | EFI_SW_PC_USER_SETUP)\r
1028 );\r
1029 }\r
1030 //\r
1031 // Based on the key that was set, we can determine what to do\r
1032 //\r
1033 switch (gCallbackKey) {\r
1034 //\r
1035 // The first 4 entries in the Front Page are to be GUARANTEED to remain constant so IHV's can\r
1036 // describe to their customers in documentation how to find their setup information (namely\r
1037 // under the device manager and specific buckets)\r
1038 //\r
1039 // These entries consist of the Continue, Select language, Boot Manager, and Device Manager\r
1040 //\r
1041 case FRONT_PAGE_KEY_CONTINUE:\r
1042 //\r
1043 // User hit continue\r
1044 //\r
1045 break;\r
1046\r
1047 case FRONT_PAGE_KEY_LANGUAGE:\r
1048 //\r
1049 // User made a language setting change - display front page again\r
1050 //\r
1051 break;\r
1052\r
1053 case FRONT_PAGE_KEY_BOOT_MANAGER:\r
1054 //\r
1055 // User chose to run the Boot Manager\r
1056 //\r
bd2057a8 1057 CallBootManager ();\r
5c08e117 1058 break;\r
1059\r
1060 case FRONT_PAGE_KEY_DEVICE_MANAGER:\r
1061 //\r
1062 // Display the Device Manager\r
1063 //\r
1064 do {\r
bd2057a8 1065 CallDeviceManager ();\r
5c08e117 1066 } while (gCallbackKey == FRONT_PAGE_KEY_DEVICE_MANAGER);\r
1067 break;\r
1068\r
1069 case FRONT_PAGE_KEY_BOOT_MAINTAIN:\r
1070 //\r
1071 // Display the Boot Maintenance Manager\r
1072 //\r
1073 BdsStartBootMaint ();\r
1074 break;\r
1075 }\r
1076\r
1077 } while ((Status == EFI_SUCCESS) && (gCallbackKey != FRONT_PAGE_KEY_CONTINUE));\r
1078\r
1079 //\r
1080 //Will leave browser, check any reset required change is applied? if yes, reset system\r
1081 //\r
1082 SetupResetReminder ();\r
1083\r
1084Exit:\r
1085 //\r
1086 // Automatically load current entry\r
1087 // Note: The following lines of code only execute when Auto boot\r
1088 // takes affect\r
1089 //\r
128efbbc 1090 PERF_END (NULL, "BdsTimeOut", "BDS", 0);\r
5c08e117 1091}\r