]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Library/BootMaintenanceManagerUiLib/Variable.c
MdeModulePkg: Apply uncrustify changes
[mirror_edk2.git] / MdeModulePkg / Library / BootMaintenanceManagerUiLib / Variable.c
index b65d6a5c9ed035258bd1278d90e6fac96e989c75..82a0ed66a714503a4e345ee1621a4e3a4ebfa709 100644 (file)
@@ -1,14 +1,8 @@
 /** @file\r
 Variable operation that will be used by bootmaint\r
 \r
-Copyright (c) 2004 - 2016, 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
-http://opensource.org/licenses/bsd-license.php\r
-\r
-THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
-WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
+Copyright (c) 2004 - 2018, Intel Corporation. All rights reserved.<BR>\r
+SPDX-License-Identifier: BSD-2-Clause-Patent\r
 \r
 **/\r
 \r
@@ -16,13 +10,11 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 \r
 /**\r
   Delete Boot Option that represent a Deleted state in BootOptionMenu.\r
-  After deleting this boot option, call Var_ChangeBootOrder to\r
-  make sure BootOrder is in valid state.\r
 \r
-  @retval EFI_SUCCESS   If all boot load option EFI Variables corresponding to  \r
+  @retval EFI_SUCCESS   If all boot load option EFI Variables corresponding to\r
                         BM_LOAD_CONTEXT marked for deletion is deleted.\r
   @retval EFI_NOT_FOUND If can not find the boot option want to be deleted.\r
-  @return Others        If failed to update the "BootOrder" variable after deletion. \r
+  @return Others        If failed to update the "BootOrder" variable after deletion.\r
 \r
 **/\r
 EFI_STATUS\r
@@ -30,34 +22,29 @@ Var_DelBootOption (
   VOID\r
   )\r
 {\r
-  BM_MENU_ENTRY   *NewMenuEntry;\r
-  BM_LOAD_CONTEXT *NewLoadContext;\r
-  UINT16          BootString[10];\r
-  EFI_STATUS      Status;\r
-  UINTN           Index;\r
-  UINTN           Index2;\r
-\r
-  Status  = EFI_SUCCESS;\r
-  Index2  = 0;\r
+  BM_MENU_ENTRY    *NewMenuEntry;\r
+  BM_LOAD_CONTEXT  *NewLoadContext;\r
+  EFI_STATUS       Status;\r
+  UINTN            Index;\r
+  UINTN            Index2;\r
+\r
+  Index2 = 0;\r
   for (Index = 0; Index < BootOptionMenu.MenuNumber; Index++) {\r
     NewMenuEntry = BOpt_GetMenuEntry (&BootOptionMenu, (Index - Index2));\r
     if (NULL == NewMenuEntry) {\r
       return EFI_NOT_FOUND;\r
     }\r
 \r
-    NewLoadContext = (BM_LOAD_CONTEXT *) NewMenuEntry->VariableContext;\r
+    NewLoadContext = (BM_LOAD_CONTEXT *)NewMenuEntry->VariableContext;\r
     if (!NewLoadContext->Deleted) {\r
       continue;\r
     }\r
 \r
-    UnicodeSPrint (\r
-      BootString,\r
-      sizeof (BootString),\r
-      L"Boot%04x",\r
-      NewMenuEntry->OptionNumber\r
-      );\r
+    Status = EfiBootManagerDeleteLoadOptionVariable (NewMenuEntry->OptionNumber, LoadOptionTypeBoot);\r
+    if (EFI_ERROR (Status)) {\r
+      return Status;\r
+    }\r
 \r
-    EfiLibDeleteVariable (BootString, &gEfiGlobalVariableGuid);\r
     Index2++;\r
     //\r
     // If current Load Option is the same as BootNext,\r
@@ -75,102 +62,11 @@ Var_DelBootOption (
 \r
   BootOptionMenu.MenuNumber -= Index2;\r
 \r
-  Status = Var_ChangeBootOrder ();\r
-  return Status;\r
-}\r
-\r
-/**\r
-  After any operation on Boot####, there will be a discrepancy in BootOrder.\r
-  Since some are missing but in BootOrder, while some are present but are\r
-  not reflected by BootOrder. Then a function rebuild BootOrder from\r
-  scratch by content from BootOptionMenu is needed.\r
-\r
-\r
-  \r
-\r
-  @retval  EFI_SUCCESS  The boot order is updated successfully.\r
-  @return               EFI_STATUS other than EFI_SUCCESS if failed to\r
-                        Set the "BootOrder" EFI Variable.\r
-\r
-**/\r
-EFI_STATUS\r
-Var_ChangeBootOrder (\r
-  VOID\r
-  )\r
-{\r
-\r
-  EFI_STATUS    Status;\r
-  BM_MENU_ENTRY *NewMenuEntry;\r
-  UINT16        *BootOrderList;\r
-  UINT16        *BootOrderListPtr;\r
-  UINTN         BootOrderListSize;\r
-  UINTN         Index;\r
-\r
-  BootOrderList     = NULL;\r
-  BootOrderListSize = 0;\r
-  //\r
-  // First check whether BootOrder is present in current configuration\r
-  //\r
-  GetEfiGlobalVariable2 (L"BootOrder", (VOID **) &BootOrderList, &BootOrderListSize);\r
-\r
-  //\r
-  // If exists, delete it to hold new BootOrder\r
-  //\r
-  if (BootOrderList != NULL) {\r
-    EfiLibDeleteVariable (L"BootOrder", &gEfiGlobalVariableGuid);\r
-    FreePool (BootOrderList);\r
-    BootOrderList = NULL;\r
-  }\r
-  //\r
-  // Maybe here should be some check method to ensure that\r
-  // no new added boot options will be added\r
-  // but the setup engine now will give only one callback\r
-  // that is to say, user are granted only one chance to\r
-  // decide whether the boot option will be added or not\r
-  // there should be no indictor to show whether this\r
-  // is a "new" boot option\r
-  //\r
-  BootOrderListSize = BootOptionMenu.MenuNumber;\r
-\r
-  if (BootOrderListSize > 0) {\r
-    BootOrderList = AllocateZeroPool (BootOrderListSize * sizeof (UINT16));\r
-    ASSERT (BootOrderList != NULL);\r
-    BootOrderListPtr = BootOrderList;\r
-\r
-    //\r
-    // Get all current used Boot#### from BootOptionMenu.\r
-    // OptionNumber in each BM_LOAD_OPTION is really its\r
-    // #### value.\r
-    //\r
-    for (Index = 0; Index < BootOrderListSize; Index++) {\r
-      NewMenuEntry    = BOpt_GetMenuEntry (&BootOptionMenu, Index);\r
-      *BootOrderList  = (UINT16) NewMenuEntry->OptionNumber;\r
-      BootOrderList++;\r
-    }\r
-\r
-    BootOrderList = BootOrderListPtr;\r
-\r
-    //\r
-    // After building the BootOrderList, write it back\r
-    //\r
-    Status = gRT->SetVariable (\r
-                    L"BootOrder",\r
-                    &gEfiGlobalVariableGuid,\r
-                    VAR_FLAG,\r
-                    BootOrderListSize * sizeof (UINT16),\r
-                    BootOrderList\r
-                    );\r
-    if (EFI_ERROR (Status)) {\r
-      return Status;\r
-    }\r
-  }\r
   return EFI_SUCCESS;\r
 }\r
 \r
 /**\r
-  Delete Load Option that represent a Deleted state in BootOptionMenu.\r
-  After deleting this Driver option, call Var_ChangeDriverOrder to\r
-  make sure DriverOrder is in valid state.\r
+  Delete Load Option that represent a Deleted state in DriverOptionMenu.\r
 \r
   @retval EFI_SUCCESS       Load Option is successfully updated.\r
   @retval EFI_NOT_FOUND     Fail to find the driver option want to be deleted.\r
@@ -183,34 +79,29 @@ Var_DelDriverOption (
   VOID\r
   )\r
 {\r
-  BM_MENU_ENTRY   *NewMenuEntry;\r
-  BM_LOAD_CONTEXT *NewLoadContext;\r
-  UINT16          DriverString[12];\r
-  EFI_STATUS      Status;\r
-  UINTN           Index;\r
-  UINTN           Index2;\r
-\r
-  Status  = EFI_SUCCESS;\r
-  Index2  = 0;\r
+  BM_MENU_ENTRY    *NewMenuEntry;\r
+  BM_LOAD_CONTEXT  *NewLoadContext;\r
+  EFI_STATUS       Status;\r
+  UINTN            Index;\r
+  UINTN            Index2;\r
+\r
+  Index2 = 0;\r
   for (Index = 0; Index < DriverOptionMenu.MenuNumber; Index++) {\r
     NewMenuEntry = BOpt_GetMenuEntry (&DriverOptionMenu, (Index - Index2));\r
     if (NULL == NewMenuEntry) {\r
       return EFI_NOT_FOUND;\r
     }\r
 \r
-    NewLoadContext = (BM_LOAD_CONTEXT *) NewMenuEntry->VariableContext;\r
+    NewLoadContext = (BM_LOAD_CONTEXT *)NewMenuEntry->VariableContext;\r
     if (!NewLoadContext->Deleted) {\r
       continue;\r
     }\r
 \r
-    UnicodeSPrint (\r
-      DriverString,\r
-      sizeof (DriverString),\r
-      L"Driver%04x",\r
-      NewMenuEntry->OptionNumber\r
-      );\r
+    Status = EfiBootManagerDeleteLoadOptionVariable (NewMenuEntry->OptionNumber, LoadOptionTypeDriver);\r
+    if (EFI_ERROR (Status)) {\r
+      return Status;\r
+    }\r
 \r
-    EfiLibDeleteVariable (DriverString, &gEfiGlobalVariableGuid);\r
     Index2++;\r
 \r
     RemoveEntryList (&NewMenuEntry->Link);\r
@@ -220,83 +111,6 @@ Var_DelDriverOption (
 \r
   DriverOptionMenu.MenuNumber -= Index2;\r
 \r
-  Status = Var_ChangeDriverOrder ();\r
-  return Status;\r
-}\r
-\r
-/**\r
-  After any operation on Driver####, there will be a discrepancy in\r
-  DriverOrder. Since some are missing but in DriverOrder, while some\r
-  are present but are not reflected by DriverOrder. Then a function\r
-  rebuild DriverOrder from scratch by content from DriverOptionMenu is\r
-  needed.\r
-\r
-  @retval  EFI_SUCCESS  The driver order is updated successfully.\r
-  @return  Other status than EFI_SUCCESS if failed to set the "DriverOrder" EFI Variable.\r
-\r
-**/\r
-EFI_STATUS\r
-Var_ChangeDriverOrder (\r
-  VOID\r
-  )\r
-{\r
-  EFI_STATUS    Status;\r
-  BM_MENU_ENTRY *NewMenuEntry;\r
-  UINT16        *DriverOrderList;\r
-  UINT16        *DriverOrderListPtr;\r
-  UINTN         DriverOrderListSize;\r
-  UINTN         Index;\r
-\r
-  DriverOrderList     = NULL;\r
-  DriverOrderListSize = 0;\r
-\r
-  //\r
-  // First check whether DriverOrder is present in current configuration\r
-  //\r
-  GetEfiGlobalVariable2 (L"DriverOrder", (VOID **) &DriverOrderList, &DriverOrderListSize);\r
-  //\r
-  // If exists, delete it to hold new DriverOrder\r
-  //\r
-  if (DriverOrderList != NULL) {\r
-    EfiLibDeleteVariable (L"DriverOrder", &gEfiGlobalVariableGuid);\r
-    FreePool (DriverOrderList);\r
-    DriverOrderList = NULL;\r
-  }\r
-\r
-  DriverOrderListSize = DriverOptionMenu.MenuNumber;\r
-\r
-  if (DriverOrderListSize > 0) {\r
-    DriverOrderList = AllocateZeroPool (DriverOrderListSize * sizeof (UINT16));\r
-    ASSERT (DriverOrderList != NULL);\r
-    DriverOrderListPtr = DriverOrderList;\r
-\r
-    //\r
-    // Get all current used Driver#### from DriverOptionMenu.\r
-    // OptionNumber in each BM_LOAD_OPTION is really its\r
-    // #### value.\r
-    //\r
-    for (Index = 0; Index < DriverOrderListSize; Index++) {\r
-      NewMenuEntry      = BOpt_GetMenuEntry (&DriverOptionMenu, Index);\r
-      *DriverOrderList  = (UINT16) NewMenuEntry->OptionNumber;\r
-      DriverOrderList++;\r
-    }\r
-\r
-    DriverOrderList = DriverOrderListPtr;\r
-\r
-    //\r
-    // After building the DriverOrderList, write it back\r
-    //\r
-    Status = gRT->SetVariable (\r
-                    L"DriverOrder",\r
-                    &gEfiGlobalVariableGuid,\r
-                    VAR_FLAG,\r
-                    DriverOrderListSize * sizeof (UINT16),\r
-                    DriverOrderList\r
-                    );\r
-    if (EFI_ERROR (Status)) {\r
-      return Status;\r
-    }\r
-  }\r
   return EFI_SUCCESS;\r
 }\r
 \r
@@ -306,7 +120,7 @@ Var_ChangeDriverOrder (
 \r
   This function clear the EFI variable defined by ConsoleName and\r
   gEfiGlobalVariableGuid. It then build the multi-instance device\r
-  path by appending the device path of the Console (In/Out/Err) instance \r
+  path by appending the device path of the Console (In/Out/Err) instance\r
   in ConsoleMenu. Then it scan all corresponding console device by\r
   scanning Terminal (built from device supporting Serial I/O instances)\r
   devices in TerminalMenu. At last, it save a EFI variable specifed\r
@@ -324,9 +138,9 @@ Var_ChangeDriverOrder (
 **/\r
 EFI_STATUS\r
 Var_UpdateConsoleOption (\r
-  IN UINT16                     *ConsoleName,\r
-  IN BM_MENU_OPTION             *ConsoleMenu,\r
-  IN UINT16                     UpdatePageId\r
+  IN UINT16          *ConsoleName,\r
+  IN BM_MENU_OPTION  *ConsoleMenu,\r
+  IN UINT16          UpdatePageId\r
   )\r
 {\r
   EFI_DEVICE_PATH_PROTOCOL  *ConDevicePath;\r
@@ -338,12 +152,12 @@ Var_UpdateConsoleOption (
   EFI_DEVICE_PATH_PROTOCOL  *TerminalDevicePath;\r
   UINTN                     Index;\r
 \r
-  GetEfiGlobalVariable2 (ConsoleName, (VOID**)&ConDevicePath, NULL);\r
+  GetEfiGlobalVariable2 (ConsoleName, (VOID **)&ConDevicePath, NULL);\r
   if (ConDevicePath != NULL) {\r
     EfiLibDeleteVariable (ConsoleName, &gEfiGlobalVariableGuid);\r
     FreePool (ConDevicePath);\r
     ConDevicePath = NULL;\r
-  };\r
+  }\r
 \r
   //\r
   // First add all console input device from console input menu\r
@@ -351,7 +165,7 @@ Var_UpdateConsoleOption (
   for (Index = 0; Index < ConsoleMenu->MenuNumber; Index++) {\r
     NewMenuEntry = BOpt_GetMenuEntry (ConsoleMenu, Index);\r
 \r
-    NewConsoleContext = (BM_CONSOLE_CONTEXT *) NewMenuEntry->VariableContext;\r
+    NewConsoleContext = (BM_CONSOLE_CONTEXT *)NewMenuEntry->VariableContext;\r
     if (NewConsoleContext->IsActive) {\r
       ConDevicePath = AppendDevicePathInstance (\r
                         ConDevicePath,\r
@@ -363,15 +177,16 @@ Var_UpdateConsoleOption (
   for (Index = 0; Index < TerminalMenu.MenuNumber; Index++) {\r
     NewMenuEntry = BOpt_GetMenuEntry (&TerminalMenu, Index);\r
 \r
-    NewTerminalContext = (BM_TERMINAL_CONTEXT *) NewMenuEntry->VariableContext;\r
+    NewTerminalContext = (BM_TERMINAL_CONTEXT *)NewMenuEntry->VariableContext;\r
     if (((NewTerminalContext->IsConIn != 0) && (UpdatePageId == FORM_CON_IN_ID)) ||\r
         ((NewTerminalContext->IsConOut != 0)  && (UpdatePageId == FORM_CON_OUT_ID)) ||\r
         ((NewTerminalContext->IsStdErr  != 0) && (UpdatePageId == FORM_CON_ERR_ID))\r
-        ) {\r
+        )\r
+    {\r
       Vendor.Header.Type    = MESSAGING_DEVICE_PATH;\r
       Vendor.Header.SubType = MSG_VENDOR_DP;\r
-      \r
-      ASSERT (NewTerminalContext->TerminalType < (sizeof (TerminalTypeGuid) / sizeof (TerminalTypeGuid[0])));\r
+\r
+      ASSERT (NewTerminalContext->TerminalType < (ARRAY_SIZE (TerminalTypeGuid)));\r
       CopyMem (\r
         &Vendor.Guid,\r
         &TerminalTypeGuid[NewTerminalContext->TerminalType],\r
@@ -379,9 +194,9 @@ Var_UpdateConsoleOption (
         );\r
       SetDevicePathNodeLength (&Vendor.Header, sizeof (VENDOR_DEVICE_PATH));\r
       TerminalDevicePath = AppendDevicePathNode (\r
-                            NewTerminalContext->DevicePath,\r
-                            (EFI_DEVICE_PATH_PROTOCOL *) &Vendor\r
-                            );\r
+                             NewTerminalContext->DevicePath,\r
+                             (EFI_DEVICE_PATH_PROTOCOL *)&Vendor\r
+                             );\r
       ASSERT (TerminalDevicePath != NULL);\r
       ChangeTerminalDevicePath (TerminalDevicePath, TRUE);\r
       ConDevicePath = AppendDevicePathInstance (\r
@@ -405,7 +220,6 @@ Var_UpdateConsoleOption (
   }\r
 \r
   return EFI_SUCCESS;\r
-\r
 }\r
 \r
 /**\r
@@ -443,7 +257,7 @@ Var_UpdateConsoleOutOption (
   console device.\r
 \r
   @retval EFI_SUCCESS    The function complete successfully.\r
-  @return The EFI variable can not be saved. See gRT->SetVariable for detail return information.  \r
+  @return The EFI variable can not be saved. See gRT->SetVariable for detail return information.\r
 **/\r
 EFI_STATUS\r
 Var_UpdateErrorOutOption (\r
@@ -454,8 +268,8 @@ Var_UpdateErrorOutOption (
 }\r
 \r
 /**\r
-  This function create a currently loaded Drive Option from \r
-  the BMM. It then appends this Driver Option to the end of \r
+  This function create a currently loaded Drive Option from\r
+  the BMM. It then appends this Driver Option to the end of\r
   the "DriverOrder" list. It append this Driver Opotion to the end\r
   of DriverOptionMenu.\r
 \r
@@ -465,34 +279,36 @@ Var_UpdateErrorOutOption (
   @param OptionalData    The optional load option.\r
   @param ForceReconnect  If to force reconnect.\r
 \r
-  @retval EFI_OUT_OF_RESOURCES If not enought memory to complete the operation.\r
+  @retval other                Contain some errors when excuting this function.See function\r
+                               EfiBootManagerInitializeLoadOption/EfiBootManagerAddLoadOptionVariabl\r
+                               for detail return information.\r
   @retval EFI_SUCCESS          If function completes successfully.\r
 \r
 **/\r
 EFI_STATUS\r
 Var_UpdateDriverOption (\r
-  IN  BMM_CALLBACK_DATA         *CallbackData,\r
-  IN  EFI_HII_HANDLE            HiiHandle,\r
-  IN  UINT16                    *DescriptionData,\r
-  IN  UINT16                    *OptionalData,\r
-  IN  UINT8                     ForceReconnect\r
+  IN  BMM_CALLBACK_DATA  *CallbackData,\r
+  IN  EFI_HII_HANDLE     HiiHandle,\r
+  IN  UINT16             *DescriptionData,\r
+  IN  UINT16             *OptionalData,\r
+  IN  UINT8              ForceReconnect\r
   )\r
 {\r
-  UINT16          Index;\r
-  UINT16          DriverString[12];\r
-  BM_MENU_ENTRY   *NewMenuEntry;\r
-  BM_LOAD_CONTEXT *NewLoadContext;\r
-  BOOLEAN         OptionalDataExist;\r
-  EFI_STATUS      Status;\r
+  UINT16                        Index;\r
+  UINT16                        DriverString[12];\r
+  BM_MENU_ENTRY                 *NewMenuEntry;\r
+  BM_LOAD_CONTEXT               *NewLoadContext;\r
+  BOOLEAN                       OptionalDataExist;\r
+  EFI_STATUS                    Status;\r
   EFI_BOOT_MANAGER_LOAD_OPTION  LoadOption;\r
   UINT8                         *OptionalDesData;\r
   UINT32                        OptionalDataSize;\r
 \r
   OptionalDataExist = FALSE;\r
-  OptionalDesData = NULL;\r
-  OptionalDataSize = 0;\r
+  OptionalDesData   = NULL;\r
+  OptionalDataSize  = 0;\r
 \r
-  Index             = BOpt_GetDriverOptionNumber ();\r
+  Index = BOpt_GetDriverOptionNumber ();\r
   UnicodeSPrint (\r
     DriverString,\r
     sizeof (DriverString),\r
@@ -506,8 +322,8 @@ Var_UpdateDriverOption (
 \r
   if (*OptionalData != 0x0000) {\r
     OptionalDataExist = TRUE;\r
-    OptionalDesData = (UINT8 *)OptionalData;\r
-    OptionalDataSize = (UINT32)StrSize (OptionalData);\r
+    OptionalDesData   = (UINT8 *)OptionalData;\r
+    OptionalDataSize  = (UINT32)StrSize (OptionalData);\r
   }\r
 \r
   NewMenuEntry = BOpt_CreateMenuEntry (BM_LOAD_CONTEXT_SELECT);\r
@@ -524,14 +340,20 @@ Var_UpdateDriverOption (
              CallbackData->LoadContext->FilePathList,\r
              OptionalDesData,\r
              OptionalDataSize\r
-           );\r
-  if (!EFI_ERROR (Status)){\r
-    Status = EfiBootManagerAddLoadOptionVariable (&LoadOption,(UINTN) -1 );\r
+             );\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
+  Status = EfiBootManagerAddLoadOptionVariable (&LoadOption, (UINTN)-1);\r
+  if (EFI_ERROR (Status)) {\r
+    EfiBootManagerFreeLoadOption (&LoadOption);\r
+    return Status;\r
   }\r
 \r
-  NewLoadContext                  = (BM_LOAD_CONTEXT *) NewMenuEntry->VariableContext;\r
-  NewLoadContext->Deleted         = FALSE;\r
-  NewLoadContext->Attributes = LoadOption.Attributes;\r
+  NewLoadContext                     = (BM_LOAD_CONTEXT *)NewMenuEntry->VariableContext;\r
+  NewLoadContext->Deleted            = FALSE;\r
+  NewLoadContext->Attributes         = LoadOption.Attributes;\r
   NewLoadContext->FilePathListLength = (UINT16)GetDevicePathSize (LoadOption.FilePath);\r
 \r
   NewLoadContext->Description = AllocateZeroPool (StrSize (DescriptionData));\r
@@ -551,10 +373,10 @@ Var_UpdateDriverOption (
     GetDevicePathSize (CallbackData->LoadContext->FilePathList)\r
     );\r
 \r
-  NewMenuEntry->HelpString    = UiDevicePathToStr (NewLoadContext->FilePathList);\r
-  NewMenuEntry->OptionNumber  = Index;\r
+  NewMenuEntry->HelpString         = UiDevicePathToStr (NewLoadContext->FilePathList);\r
+  NewMenuEntry->OptionNumber       = Index;\r
   NewMenuEntry->DisplayStringToken = HiiSetString (HiiHandle, 0, NewMenuEntry->DisplayString, NULL);\r
-  NewMenuEntry->HelpStringToken = HiiSetString (HiiHandle, 0, NewMenuEntry->HelpString, NULL);\r
+  NewMenuEntry->HelpStringToken    = HiiSetString (HiiHandle, 0, NewMenuEntry->HelpString, NULL);\r
 \r
   if (OptionalDataExist) {\r
     NewLoadContext->OptionalData = AllocateZeroPool (LoadOption.OptionalDataSize);\r
@@ -569,45 +391,47 @@ Var_UpdateDriverOption (
   InsertTailList (&DriverOptionMenu.Head, &NewMenuEntry->Link);\r
   DriverOptionMenu.MenuNumber++;\r
 \r
-  EfiBootManagerFreeLoadOption(&LoadOption);\r
+  EfiBootManagerFreeLoadOption (&LoadOption);\r
 \r
   return EFI_SUCCESS;\r
 }\r
 \r
 /**\r
-  This function create a currently loaded Boot Option from \r
-  the BMM. It then appends this Boot Option to the end of \r
+  This function create a currently loaded Boot Option from\r
+  the BMM. It then appends this Boot Option to the end of\r
   the "BootOrder" list. It also append this Boot Opotion to the end\r
   of BootOptionMenu.\r
 \r
   @param CallbackData    The BMM context data.\r
 \r
-  @retval EFI_OUT_OF_RESOURCES If not enought memory to complete the operation.\r
+  @retval other                Contain some errors when excuting this function. See function\r
+                               EfiBootManagerInitializeLoadOption/EfiBootManagerAddLoadOptionVariabl\r
+                               for detail return information.\r
   @retval EFI_SUCCESS          If function completes successfully.\r
 \r
 **/\r
 EFI_STATUS\r
 Var_UpdateBootOption (\r
-  IN  BMM_CALLBACK_DATA              *CallbackData\r
+  IN  BMM_CALLBACK_DATA  *CallbackData\r
   )\r
 {\r
-  UINT16          BootString[10];\r
-  UINT16          Index;\r
-  BM_MENU_ENTRY   *NewMenuEntry;\r
-  BM_LOAD_CONTEXT *NewLoadContext;\r
-  BOOLEAN         OptionalDataExist;\r
-  EFI_STATUS      Status;\r
-  BMM_FAKE_NV_DATA  *NvRamMap;\r
+  UINT16                        BootString[10];\r
+  UINT16                        Index;\r
+  BM_MENU_ENTRY                 *NewMenuEntry;\r
+  BM_LOAD_CONTEXT               *NewLoadContext;\r
+  BOOLEAN                       OptionalDataExist;\r
+  EFI_STATUS                    Status;\r
+  BMM_FAKE_NV_DATA              *NvRamMap;\r
   EFI_BOOT_MANAGER_LOAD_OPTION  LoadOption;\r
   UINT8                         *OptionalData;\r
   UINT32                        OptionalDataSize;\r
 \r
   OptionalDataExist = FALSE;\r
-  NvRamMap = &CallbackData->BmmFakeNvData;\r
-  OptionalData = NULL;\r
-  OptionalDataSize = 0;\r
+  NvRamMap          = &CallbackData->BmmFakeNvData;\r
+  OptionalData      = NULL;\r
+  OptionalDataSize  = 0;\r
 \r
-  Index = BOpt_GetBootOptionNumber () ;\r
+  Index = BOpt_GetBootOptionNumber ();\r
   UnicodeSPrint (BootString, sizeof (BootString), L"Boot%04x", Index);\r
 \r
   if (NvRamMap->BootDescriptionData[0] == 0x0000) {\r
@@ -616,8 +440,8 @@ Var_UpdateBootOption (
 \r
   if (NvRamMap->BootOptionalData[0] != 0x0000) {\r
     OptionalDataExist = TRUE;\r
-    OptionalData = (UINT8 *)NvRamMap->BootOptionalData;\r
-    OptionalDataSize = (UINT32)StrSize (NvRamMap->BootOptionalData);\r
+    OptionalData      = (UINT8 *)NvRamMap->BootOptionalData;\r
+    OptionalDataSize  = (UINT32)StrSize (NvRamMap->BootOptionalData);\r
   }\r
 \r
   NewMenuEntry = BOpt_CreateMenuEntry (BM_LOAD_CONTEXT_SELECT);\r
@@ -634,15 +458,21 @@ Var_UpdateBootOption (
              CallbackData->LoadContext->FilePathList,\r
              OptionalData,\r
              OptionalDataSize\r
-           );\r
-  if (!EFI_ERROR (Status)){\r
-    Status = EfiBootManagerAddLoadOptionVariable (&LoadOption,(UINTN) -1 );\r
+             );\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
+  Status = EfiBootManagerAddLoadOptionVariable (&LoadOption, (UINTN)-1);\r
+  if (EFI_ERROR (Status)) {\r
+    EfiBootManagerFreeLoadOption (&LoadOption);\r
+    return Status;\r
   }\r
 \r
-  NewLoadContext                  = (BM_LOAD_CONTEXT *) NewMenuEntry->VariableContext;\r
-  NewLoadContext->Deleted         = FALSE;\r
-  NewLoadContext->Attributes = LoadOption.Attributes;\r
-  NewLoadContext->FilePathListLength = (UINT16) GetDevicePathSize (LoadOption.FilePath);\r
+  NewLoadContext                     = (BM_LOAD_CONTEXT *)NewMenuEntry->VariableContext;\r
+  NewLoadContext->Deleted            = FALSE;\r
+  NewLoadContext->Attributes         = LoadOption.Attributes;\r
+  NewLoadContext->FilePathListLength = (UINT16)GetDevicePathSize (LoadOption.FilePath);\r
 \r
   NewLoadContext->Description = AllocateZeroPool (StrSize (NvRamMap->BootDescriptionData));\r
   ASSERT (NewLoadContext->Description != NULL);\r
@@ -663,10 +493,10 @@ Var_UpdateBootOption (
     GetDevicePathSize (CallbackData->LoadContext->FilePathList)\r
     );\r
 \r
-  NewMenuEntry->HelpString    = UiDevicePathToStr (NewLoadContext->FilePathList);\r
-  NewMenuEntry->OptionNumber  = Index;\r
+  NewMenuEntry->HelpString         = UiDevicePathToStr (NewLoadContext->FilePathList);\r
+  NewMenuEntry->OptionNumber       = Index;\r
   NewMenuEntry->DisplayStringToken = HiiSetString (CallbackData->BmmHiiHandle, 0, NewMenuEntry->DisplayString, NULL);\r
-  NewMenuEntry->HelpStringToken = HiiSetString (CallbackData->BmmHiiHandle, 0, NewMenuEntry->HelpString, NULL);\r
+  NewMenuEntry->HelpStringToken    = HiiSetString (CallbackData->BmmHiiHandle, 0, NewMenuEntry->HelpString, NULL);\r
 \r
   if (OptionalDataExist) {\r
     NewLoadContext->OptionalData = AllocateZeroPool (LoadOption.OptionalDataSize);\r
@@ -681,13 +511,13 @@ Var_UpdateBootOption (
   InsertTailList (&BootOptionMenu.Head, &NewMenuEntry->Link);\r
   BootOptionMenu.MenuNumber++;\r
 \r
-  EfiBootManagerFreeLoadOption(&LoadOption);\r
+  EfiBootManagerFreeLoadOption (&LoadOption);\r
 \r
   return EFI_SUCCESS;\r
 }\r
 \r
 /**\r
-  This function update the "BootNext" EFI Variable. If there is \r
+  This function update the "BootNext" EFI Variable. If there is\r
   no "BootNext" specified in BMM, this EFI Variable is deleted.\r
   It also update the BMM context data specified the "BootNext"\r
   vaule.\r
@@ -695,13 +525,13 @@ Var_UpdateBootOption (
   @param CallbackData    The BMM context data.\r
 \r
   @retval EFI_SUCCESS    The function complete successfully.\r
-  @return                The EFI variable can be saved. See gRT->SetVariable \r
+  @return                The EFI variable can be saved. See gRT->SetVariable\r
                          for detail return information.\r
 \r
 **/\r
 EFI_STATUS\r
 Var_UpdateBootNext (\r
-  IN BMM_CALLBACK_DATA            *CallbackData\r
+  IN BMM_CALLBACK_DATA  *CallbackData\r
   )\r
 {\r
   BM_MENU_ENTRY     *NewMenuEntry;\r
@@ -710,14 +540,14 @@ Var_UpdateBootNext (
   UINT16            Index;\r
   EFI_STATUS        Status;\r
 \r
-  Status            = EFI_SUCCESS;\r
-  CurrentFakeNVMap  = &CallbackData->BmmFakeNvData;\r
+  Status           = EFI_SUCCESS;\r
+  CurrentFakeNVMap = &CallbackData->BmmFakeNvData;\r
   for (Index = 0; Index < BootOptionMenu.MenuNumber; Index++) {\r
     NewMenuEntry = BOpt_GetMenuEntry (&BootOptionMenu, Index);\r
     ASSERT (NULL != NewMenuEntry);\r
 \r
-    NewLoadContext              = (BM_LOAD_CONTEXT *) NewMenuEntry->VariableContext;\r
-    NewLoadContext->IsBootNext  = FALSE;\r
+    NewLoadContext             = (BM_LOAD_CONTEXT *)NewMenuEntry->VariableContext;\r
+    NewLoadContext->IsBootNext = FALSE;\r
   }\r
 \r
   if (CurrentFakeNVMap->BootNext == NONE_BOOTNEXT_VALUE) {\r
@@ -726,19 +556,19 @@ Var_UpdateBootNext (
   }\r
 \r
   NewMenuEntry = BOpt_GetMenuEntry (\r
-                  &BootOptionMenu,\r
-                  CurrentFakeNVMap->BootNext\r
-                  );\r
+                   &BootOptionMenu,\r
+                   CurrentFakeNVMap->BootNext\r
+                   );\r
   ASSERT (NewMenuEntry != NULL);\r
 \r
-  NewLoadContext = (BM_LOAD_CONTEXT *) NewMenuEntry->VariableContext;\r
-  Status = gRT->SetVariable (\r
-                  L"BootNext",\r
-                  &gEfiGlobalVariableGuid,\r
-                  VAR_FLAG,\r
-                  sizeof (UINT16),\r
-                  &NewMenuEntry->OptionNumber\r
-                  );\r
+  NewLoadContext = (BM_LOAD_CONTEXT *)NewMenuEntry->VariableContext;\r
+  Status         = gRT->SetVariable (\r
+                          L"BootNext",\r
+                          &gEfiGlobalVariableGuid,\r
+                          VAR_FLAG,\r
+                          sizeof (UINT16),\r
+                          &NewMenuEntry->OptionNumber\r
+                          );\r
   NewLoadContext->IsBootNext              = TRUE;\r
   CallbackData->BmmOldFakeNVData.BootNext = CurrentFakeNVMap->BootNext;\r
   return Status;\r
@@ -758,7 +588,7 @@ Var_UpdateBootNext (
 **/\r
 EFI_STATUS\r
 Var_UpdateBootOrder (\r
-  IN BMM_CALLBACK_DATA            *CallbackData\r
+  IN BMM_CALLBACK_DATA  *CallbackData\r
   )\r
 {\r
   EFI_STATUS  Status;\r
@@ -771,7 +601,7 @@ Var_UpdateBootOrder (
   //\r
   // First check whether BootOrder is present in current configuration\r
   //\r
-  GetEfiGlobalVariable2 (L"BootOrder", (VOID **) &BootOrder, &BootOrderSize);\r
+  GetEfiGlobalVariable2 (L"BootOrder", (VOID **)&BootOrder, &BootOrderSize);\r
   if (BootOrder == NULL) {\r
     return EFI_OUT_OF_RESOURCES;\r
   }\r
@@ -783,7 +613,7 @@ Var_UpdateBootOrder (
   //\r
   for (OrderIndex = 0; (OrderIndex < BootOptionMenu.MenuNumber) && (CallbackData->BmmFakeNvData.BootOptionOrder[OrderIndex] != 0); OrderIndex++) {\r
     for (Index = OrderIndex; Index < BootOrderSize / sizeof (UINT16); Index++) {\r
-      if ((BootOrder[Index] == (UINT16) (CallbackData->BmmFakeNvData.BootOptionOrder[OrderIndex] - 1)) && (OrderIndex != Index)) {\r
+      if ((BootOrder[Index] == (UINT16)(CallbackData->BmmFakeNvData.BootOptionOrder[OrderIndex] - 1)) && (OrderIndex != Index)) {\r
         OptionNumber = BootOrder[Index];\r
         CopyMem (&BootOrder[OrderIndex + 1], &BootOrder[OrderIndex], (Index - OrderIndex) * sizeof (UINT16));\r
         BootOrder[OrderIndex] = OptionNumber;\r
@@ -799,12 +629,11 @@ Var_UpdateBootOrder (
                   BootOrder\r
                   );\r
   FreePool (BootOrder);\r
-  \r
+\r
   BOpt_FreeMenu (&BootOptionMenu);\r
   BOpt_GetBootOptions (CallbackData);\r
 \r
   return Status;\r
-\r
 }\r
 \r
 /**\r
@@ -821,7 +650,7 @@ Var_UpdateBootOrder (
 **/\r
 EFI_STATUS\r
 Var_UpdateDriverOrder (\r
-  IN BMM_CALLBACK_DATA            *CallbackData\r
+  IN BMM_CALLBACK_DATA  *CallbackData\r
   )\r
 {\r
   EFI_STATUS  Status;\r
@@ -836,12 +665,13 @@ Var_UpdateDriverOrder (
   //\r
   // First check whether DriverOrder is present in current configuration\r
   //\r
-  GetEfiGlobalVariable2 (L"DriverOrder", (VOID **) &DriverOrderList, &DriverOrderListSize);\r
+  GetEfiGlobalVariable2 (L"DriverOrder", (VOID **)&DriverOrderList, &DriverOrderListSize);\r
   NewDriverOrderList = AllocateZeroPool (DriverOrderListSize);\r
 \r
   if (NewDriverOrderList == NULL) {\r
     return EFI_OUT_OF_RESOURCES;\r
   }\r
+\r
   //\r
   // If exists, delete it to hold new DriverOrder\r
   //\r
@@ -852,7 +682,7 @@ Var_UpdateDriverOrder (
 \r
   ASSERT (DriverOptionMenu.MenuNumber <= (sizeof (CallbackData->BmmFakeNvData.DriverOptionOrder) / sizeof (CallbackData->BmmFakeNvData.DriverOptionOrder[0])));\r
   for (Index = 0; Index < DriverOptionMenu.MenuNumber; Index++) {\r
-    NewDriverOrderList[Index] = (UINT16) (CallbackData->BmmFakeNvData.DriverOptionOrder[Index] - 1);\r
+    NewDriverOrderList[Index] = (UINT16)(CallbackData->BmmFakeNvData.DriverOptionOrder[Index] - 1);\r
   }\r
 \r
   Status = gRT->SetVariable (\r
@@ -882,7 +712,7 @@ Var_UpdateDriverOrder (
 **/\r
 EFI_STATUS\r
 Var_UpdateConMode (\r
-  IN BMM_CALLBACK_DATA            *CallbackData\r
+  IN BMM_CALLBACK_DATA  *CallbackData\r
   )\r
 {\r
   EFI_STATUS        Status;\r
@@ -892,10 +722,10 @@ Var_UpdateConMode (
   Mode = CallbackData->BmmFakeNvData.ConsoleOutMode;\r
 \r
   Status = gST->ConOut->QueryMode (gST->ConOut, Mode, &(ModeInfo.Column), &(ModeInfo.Row));\r
-  if (!EFI_ERROR(Status)) {\r
-    Status = PcdSet32S (PcdSetupConOutColumn, (UINT32) ModeInfo.Column);\r
+  if (!EFI_ERROR (Status)) {\r
+    Status = PcdSet32S (PcdSetupConOutColumn, (UINT32)ModeInfo.Column);\r
     if (!EFI_ERROR (Status)) {\r
-      Status = PcdSet32S (PcdSetupConOutRow, (UINT32) ModeInfo.Row);\r
+      Status = PcdSet32S (PcdSetupConOutRow, (UINT32)ModeInfo.Row);\r
     }\r
   }\r
 \r