]> git.proxmox.com Git - mirror_edk2.git/blobdiff - IntelFrameworkModulePkg/Universal/BdsDxe/FrontPage.c
Report the setting variable failure to platform through the status code when core...
[mirror_edk2.git] / IntelFrameworkModulePkg / Universal / BdsDxe / FrontPage.c
index 3dad6bd65bfe94247c70f91904d66425951f1b34..3bbe71a8fe7c0082d80648022f30f4615f3a6c18 100644 (file)
@@ -1,7 +1,7 @@
 /** @file\r
   FrontPage routines to handle the callbacks and browser calls\r
 \r
-Copyright (c) 2004 - 2012, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2004 - 2014, Intel Corporation. All rights reserved.<BR>\r
 This program and the accompanying materials\r
 are licensed and made available under the terms and conditions of the BSD License\r
 which accompanies this distribution.  The full text of the license may be found at\r
@@ -21,6 +21,7 @@ BOOLEAN   mModeInitialized = FALSE;
 \r
 BOOLEAN   gConnectAllHappened = FALSE;\r
 UINTN     gCallbackKey;\r
+CHAR8     *mLanguageString;\r
 \r
 //\r
 // Boot video resolution and text mode.\r
@@ -178,13 +179,9 @@ FrontPageCallback (
   OUT EFI_BROWSER_ACTION_REQUEST             *ActionRequest\r
   )\r
 {\r
-  CHAR8                         *LanguageString;\r
   CHAR8                         *LangCode;\r
   CHAR8                         *Lang;\r
   UINTN                         Index;\r
-  EFI_STATUS                    Status;\r
-  CHAR8                         *PlatformSupportedLanguages;\r
-  CHAR8                         *BestLanguage;\r
 \r
   if (Action != EFI_BROWSER_ACTION_CHANGING && Action != EFI_BROWSER_ACTION_CHANGED) {\r
     //\r
@@ -209,19 +206,14 @@ FrontPageCallback (
       break;\r
 \r
     case FRONT_PAGE_KEY_LANGUAGE:\r
-      //\r
-      // Collect the languages from what our current Language support is based on our VFR\r
-      //\r
-      LanguageString = HiiGetSupportedLanguages (gFrontPagePrivate.HiiHandle);\r
-      ASSERT (LanguageString != NULL);\r
       //\r
       // Allocate working buffer for RFC 4646 language in supported LanguageString.\r
       //\r
-      Lang = AllocatePool (AsciiStrSize (LanguageString));\r
-      ASSERT (Lang != NULL);\r
+      Lang = AllocatePool (AsciiStrSize (mLanguageString));\r
+      ASSERT (Lang != NULL);  \r
 \r
       Index = 0;\r
-      LangCode = LanguageString;\r
+      LangCode = mLanguageString;\r
       while (*LangCode != 0) {\r
         GetNextLanguage (&LangCode, Lang);\r
 \r
@@ -232,43 +224,21 @@ FrontPageCallback (
         Index++;\r
       }\r
 \r
-      GetEfiGlobalVariable2 (L"PlatformLangCodes", (VOID**)&PlatformSupportedLanguages, NULL);\r
-      if (PlatformSupportedLanguages == NULL) {\r
-        PlatformSupportedLanguages = AllocateCopyPool (\r
-                                       AsciiStrSize ((CHAR8 *) PcdGetPtr (PcdUefiVariableDefaultPlatformLangCodes)),\r
-                                       (CHAR8 *) PcdGetPtr (PcdUefiVariableDefaultPlatformLangCodes)\r
-                                       );\r
-        ASSERT (PlatformSupportedLanguages != NULL);\r
-      }\r
-\r
-      //\r
-      // Select the best language in platform supported Language.\r
-      //\r
-      BestLanguage = GetBestLanguage (\r
-                       PlatformSupportedLanguages,\r
-                       FALSE,\r
-                       Lang,\r
-                       NULL\r
-                       );\r
-      if (BestLanguage != NULL) {\r
-        Status = gRT->SetVariable (\r
+      if (Index == Value->u8) {\r
+        BdsDxeSetVariableAndReportStatusCodeOnError (\r
                         L"PlatformLang",\r
                         &gEfiGlobalVariableGuid,\r
                         EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS,\r
-                        AsciiStrSize (BestLanguage),\r
+                        AsciiStrSize (Lang),\r
                         Lang\r
                         );\r
-        ASSERT_EFI_ERROR(Status);\r
-        FreePool (BestLanguage);\r
       } else {\r
         ASSERT (FALSE);\r
       }\r
 \r
       *ActionRequest = EFI_BROWSER_ACTION_REQUEST_EXIT;\r
 \r
-      FreePool (PlatformSupportedLanguages);\r
       FreePool (Lang);\r
-      FreePool (LanguageString);\r
       break;\r
 \r
     default:\r
@@ -328,11 +298,9 @@ InitializeFrontPage (
   )\r
 {\r
   EFI_STATUS                  Status;\r
-  CHAR8                       *LanguageString;\r
   CHAR8                       *LangCode;\r
   CHAR8                       *Lang;\r
   CHAR8                       *CurrentLang;\r
-  CHAR8                       *BestLanguage;\r
   UINTN                       OptionCount;\r
   CHAR16                      *StringBuffer;\r
   EFI_HII_HANDLE              HiiHandle;\r
@@ -341,7 +309,11 @@ InitializeFrontPage (
   VOID                        *EndOpCodeHandle;\r
   EFI_IFR_GUID_LABEL          *StartLabel;\r
   EFI_IFR_GUID_LABEL          *EndLabel;\r
-  BOOLEAN                     FirstFlag;\r
+  EFI_HII_STRING_PROTOCOL     *HiiString;\r
+  UINTN                       StringSize;\r
+\r
+  Lang         = NULL;\r
+  StringBuffer = NULL;\r
 \r
   if (InitializeHiiData) {\r
     //\r
@@ -422,62 +394,84 @@ InitializeFrontPage (
   // Collect the languages from what our current Language support is based on our VFR\r
   //\r
   HiiHandle = gFrontPagePrivate.HiiHandle;\r
-  LanguageString = HiiGetSupportedLanguages (HiiHandle);\r
-  ASSERT (LanguageString != NULL);\r
-  //\r
-  // Allocate working buffer for RFC 4646 language in supported LanguageString.\r
-  //\r
-  Lang = AllocatePool (AsciiStrSize (LanguageString));\r
-  ASSERT (Lang != NULL);\r
 \r
-  GetEfiGlobalVariable2 (L"PlatformLang", (VOID**)&CurrentLang, NULL);\r
-  //\r
-  // Select the best language in LanguageString as the default one.\r
-  //\r
-  BestLanguage = GetBestLanguage (\r
-                   LanguageString,\r
-                   FALSE,\r
-                   (CurrentLang != NULL) ? CurrentLang : "",\r
-                   (CHAR8 *) PcdGetPtr (PcdUefiVariableDefaultPlatformLang),\r
-                   LanguageString,\r
-                   NULL\r
-                   );\r
+  CurrentLang = GetEfiGlobalVariable (L"PlatformLang");\r
+\r
   //\r
-  // BestLanguage must be selected as it is the first language in LanguageString by default\r
+  // Get Support language list from variable.\r
   //\r
-  ASSERT (BestLanguage != NULL);\r
-\r
-  OptionCount = 0;\r
-  LangCode    = LanguageString;\r
-  FirstFlag   = FALSE;\r
+  if (mLanguageString == NULL){\r
+    mLanguageString = GetEfiGlobalVariable (L"PlatformLangCodes");\r
+    if (mLanguageString == NULL) {\r
+      mLanguageString = AllocateCopyPool (\r
+                                 AsciiStrSize ((CHAR8 *) PcdGetPtr (PcdUefiVariableDefaultPlatformLangCodes)),\r
+                                 (CHAR8 *) PcdGetPtr (PcdUefiVariableDefaultPlatformLangCodes)\r
+                                 );\r
+      ASSERT (mLanguageString != NULL);\r
+    }\r
+  }\r
 \r
   if (gFrontPagePrivate.LanguageToken == NULL) {\r
+    //\r
+    // Count the language list number.\r
+    //  \r
+    LangCode      = mLanguageString;\r
+    Lang          = AllocatePool (AsciiStrSize (mLanguageString));\r
+    ASSERT (Lang != NULL);\r
+    OptionCount = 0;\r
     while (*LangCode != 0) {\r
       GetNextLanguage (&LangCode, Lang);\r
       OptionCount ++;\r
     }\r
-    gFrontPagePrivate.LanguageToken = AllocatePool (OptionCount * sizeof (EFI_STRING_ID));\r
+\r
+    //\r
+    // Allocate extra 1 as the end tag.\r
+    //\r
+    gFrontPagePrivate.LanguageToken = AllocateZeroPool ((OptionCount + 1) * sizeof (EFI_STRING_ID));\r
     ASSERT (gFrontPagePrivate.LanguageToken != NULL);\r
-    FirstFlag = TRUE;\r
-  }\r
 \r
-  OptionCount = 0;\r
-  LangCode = LanguageString;\r
-  while (*LangCode != 0) {\r
-    GetNextLanguage (&LangCode, Lang);\r
+    Status = gBS->LocateProtocol (&gEfiHiiStringProtocolGuid, NULL, (VOID **) &HiiString);\r
+    ASSERT_EFI_ERROR (Status);\r
 \r
-    if (FirstFlag) {\r
-      StringBuffer = HiiGetString (HiiHandle, PRINTABLE_LANGUAGE_NAME_STRING_ID, Lang);\r
-      ASSERT (StringBuffer != NULL);\r
+    LangCode     = mLanguageString;\r
+    OptionCount  = 0;\r
+    while (*LangCode != 0) {\r
+      GetNextLanguage (&LangCode, Lang);\r
 \r
-      //\r
-      // Save the string Id for each language\r
-      //\r
+      StringSize = 0;\r
+      Status = HiiString->GetString (HiiString, Lang, HiiHandle, PRINTABLE_LANGUAGE_NAME_STRING_ID, StringBuffer, &StringSize, NULL);\r
+      if (Status == EFI_BUFFER_TOO_SMALL) {\r
+        StringBuffer = AllocateZeroPool (StringSize);\r
+        ASSERT (StringBuffer != NULL);\r
+        Status = HiiString->GetString (HiiString, Lang, HiiHandle, PRINTABLE_LANGUAGE_NAME_STRING_ID, StringBuffer, &StringSize, NULL);\r
+        ASSERT_EFI_ERROR (Status);\r
+      }\r
+\r
+      if (EFI_ERROR (Status)) {\r
+        StringBuffer = AllocatePool (AsciiStrSize (Lang) * sizeof (CHAR16));\r
+        ASSERT (StringBuffer != NULL);\r
+        AsciiStrToUnicodeStr (Lang, StringBuffer);\r
+      }\r
+\r
+      ASSERT (StringBuffer != NULL);\r
       gFrontPagePrivate.LanguageToken[OptionCount] = HiiSetString (HiiHandle, 0, StringBuffer, NULL);\r
       FreePool (StringBuffer);\r
+\r
+      OptionCount++;\r
     }\r
+  }\r
 \r
-    if (AsciiStrCmp (Lang, BestLanguage) == 0) {\r
+  ASSERT (gFrontPagePrivate.LanguageToken != NULL);\r
+  LangCode     = mLanguageString;\r
+  OptionCount  = 0;\r
+  if (Lang == NULL) {\r
+    Lang = AllocatePool (AsciiStrSize (mLanguageString));\r
+    ASSERT (Lang != NULL);\r
+  }\r
+  while (*LangCode != 0) {\r
+    GetNextLanguage (&LangCode, Lang);\r
+\r
+    if (CurrentLang != NULL && AsciiStrCmp (Lang, CurrentLang) == 0) {\r
       HiiCreateOneOfOptionOpCode (\r
         OptionsOpCodeHandle,\r
         gFrontPagePrivate.LanguageToken[OptionCount],\r
@@ -501,9 +495,7 @@ InitializeFrontPage (
   if (CurrentLang != NULL) {\r
     FreePool (CurrentLang);\r
   }\r
-  FreePool (BestLanguage);\r
   FreePool (Lang);\r
-  FreePool (LanguageString);\r
 \r
   HiiCreateOneOfOpCode (\r
     StartOpCodeHandle,\r
@@ -1101,6 +1093,9 @@ PlatformBdsEnterFrontPage (
                     sizeof(UINT64),\r
                     &OsIndication\r
                     );\r
+    //\r
+    // Changing the content without increasing its size with current variable implementation shouldn't fail.\r
+    //\r
     ASSERT_EFI_ERROR (Status);\r
 \r
     //\r
@@ -1229,6 +1224,10 @@ PlatformBdsEnterFrontPage (
 \r
   } while ((Status == EFI_SUCCESS) && (gCallbackKey != FRONT_PAGE_KEY_CONTINUE));\r
 \r
+  if (mLanguageString != NULL) {\r
+    FreePool (mLanguageString);\r
+    mLanguageString = NULL;\r
+  }\r
   //\r
   //Will leave browser, check any reset required change is applied? if yes, reset system\r
   //\r