]> git.proxmox.com Git - mirror_edk2.git/blobdiff - SecurityPkg/VariableAuthenticated/SecureBootConfigDxe/SecureBootConfigFileExplorer.c
UefiCpuPkg: Move AsmRelocateApLoopStart from Mpfuncs.nasm to AmdSev.nasm
[mirror_edk2.git] / SecurityPkg / VariableAuthenticated / SecureBootConfigDxe / SecureBootConfigFileExplorer.c
index 5ca013c92df21f91a0e635a1549349c632c5a766..5a23dc517aa75a8f399203e3695c2b8e7c313bea 100644 (file)
 /** @file\r
   Internal file explorer functions for SecureBoot configuration module.\r
 \r
-Copyright (c) 2012 - 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
-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) 2012 - 2016, Intel Corporation. All rights reserved.<BR>\r
+SPDX-License-Identifier: BSD-2-Clause-Patent\r
 \r
 **/\r
 \r
 #include "SecureBootConfigImpl.h"\r
 \r
-///\r
-/// File system selection menu\r
-///\r
-SECUREBOOT_MENU_OPTION      FsOptionMenu = {\r
-  SECUREBOOT_MENU_OPTION_SIGNATURE,\r
-  {NULL},\r
-  0\r
-};\r
-\r
-///\r
-/// Files and sub-directories in current directory menu\r
-///\r
-SECUREBOOT_MENU_OPTION      DirectoryMenu = {\r
-  SECUREBOOT_MENU_OPTION_SIGNATURE,\r
-  {NULL},\r
-  0\r
-};\r
-\r
-VOID                  *mStartOpCodeHandle = NULL;\r
-VOID                  *mEndOpCodeHandle = NULL;\r
-EFI_IFR_GUID_LABEL    *mStartLabel = NULL;\r
-EFI_IFR_GUID_LABEL    *mEndLabel = NULL;\r
-\r
-/**\r
-  Duplicate a string.\r
-\r
-  @param[in]    Src             The source string.\r
-\r
-  @return      A new string which is duplicated copy of the source,\r
-               or NULL if there is not enough memory.\r
-\r
-**/\r
-CHAR16 *\r
-StrDuplicate (\r
-  IN CHAR16   *Src\r
-  )\r
-{\r
-  CHAR16  *Dest;\r
-  UINTN   Size;\r
-\r
-  Size  = StrSize (Src);\r
-  Dest  = AllocateZeroPool (Size);\r
-  ASSERT (Dest != NULL);\r
-  if (Dest != NULL) {\r
-    CopyMem (Dest, Src, Size);\r
-  }\r
-\r
-  return Dest;\r
-}\r
+VOID                *mStartOpCodeHandle = NULL;\r
+VOID                *mEndOpCodeHandle   = NULL;\r
+EFI_IFR_GUID_LABEL  *mStartLabel        = NULL;\r
+EFI_IFR_GUID_LABEL  *mEndLabel          = NULL;\r
 \r
 /**\r
-  Helper function called as part of the code needed to allocate\r
-  the proper sized buffer for various EFI interfaces.\r
-\r
-  @param[in, out]   Status          Current status\r
-  @param[in, out]   Buffer          Current allocated buffer, or NULL\r
-  @param[in]        BufferSize      Current buffer size needed\r
-\r
-  @retval  TRUE         If the buffer was reallocated and the caller\r
-                        should try the API again.\r
-  @retval  FALSE        The caller should not call this function again.\r
+  Refresh the global UpdateData structure.\r
 \r
 **/\r
-BOOLEAN\r
-GrowBuffer (\r
-  IN OUT EFI_STATUS   *Status,\r
-  IN OUT VOID         **Buffer,\r
-  IN UINTN            BufferSize\r
+VOID\r
+RefreshUpdateData (\r
+  VOID\r
   )\r
 {\r
-  BOOLEAN TryAgain;\r
-\r
   //\r
-  // If this is an initial request, buffer will be null with a new buffer size\r
+  // Free current updated date\r
   //\r
-  if ((*Buffer == NULL) && (BufferSize != 0)) {\r
-    *Status = EFI_BUFFER_TOO_SMALL;\r
+  if (mStartOpCodeHandle != NULL) {\r
+    HiiFreeOpCodeHandle (mStartOpCodeHandle);\r
   }\r
-  //\r
-  // If the status code is "buffer too small", resize the buffer\r
-  //\r
-  TryAgain = FALSE;\r
-  if (*Status == EFI_BUFFER_TOO_SMALL) {\r
-\r
-    if (*Buffer != NULL) {\r
-      FreePool (*Buffer);\r
-    }\r
-\r
-    *Buffer = AllocateZeroPool (BufferSize);\r
 \r
-    if (*Buffer != NULL) {\r
-      TryAgain = TRUE;\r
-    } else {\r
-      *Status = EFI_OUT_OF_RESOURCES;\r
-    }\r
-  }\r
   //\r
-  // If there's an error, free the buffer\r
+  // Create new OpCode Handle\r
   //\r
-  if (!TryAgain && EFI_ERROR (*Status) && (*Buffer != NULL)) {\r
-    FreePool (*Buffer);\r
-    *Buffer = NULL;\r
-  }\r
-\r
-  return TryAgain;\r
-}\r
-\r
-/**\r
-  Append file name to existing file name, and allocate a new buffer\r
-  to hold the appended result.\r
-\r
-  @param[in]  Str1  The existing file name\r
-  @param[in]  Str2  The file name to be appended\r
-\r
-  @return      A new string with appended result.\r
-\r
-**/\r
-CHAR16 *\r
-AppendFileName (\r
-  IN  CHAR16  *Str1,\r
-  IN  CHAR16  *Str2\r
-  )\r
-{\r
-  UINTN   Size1;\r
-  UINTN   Size2;\r
-  CHAR16  *Str;\r
-  CHAR16  *TmpStr;\r
-  CHAR16  *Ptr;\r
-  CHAR16  *LastSlash;\r
-\r
-  Size1 = StrSize (Str1);\r
-  Size2 = StrSize (Str2);\r
-  Str   = AllocateZeroPool (Size1 + Size2 + sizeof (CHAR16));\r
-  ASSERT (Str != NULL);\r
-\r
-  TmpStr = AllocateZeroPool (Size1 + Size2 + sizeof (CHAR16));\r
-  ASSERT (TmpStr != NULL);\r
-\r
-  StrCat (Str, Str1);\r
-  if (!((*Str == '\\') && (*(Str + 1) == 0))) {\r
-    StrCat (Str, L"\\");\r
-  }\r
-\r
-  StrCat (Str, Str2);\r
-\r
-  Ptr       = Str;\r
-  LastSlash = Str;\r
-  while (*Ptr != 0) {\r
-    if (*Ptr == '\\' && *(Ptr + 1) == '.' && *(Ptr + 2) == '.' && *(Ptr + 3) == L'\\') {\r
-      //\r
-      // Convert "\Name\..\" to "\"\r
-      // DO NOT convert the .. if it is at the end of the string. This will\r
-      // break the .. behavior in changing directories.\r
-      //\r
-\r
-      //\r
-      // Use TmpStr as a backup, as StrCpy in BaseLib does not handle copy of two strings\r
-      // that overlap.\r
-      //\r
-      StrCpy (TmpStr, Ptr + 3);\r
-      StrCpy (LastSlash, TmpStr);\r
-      Ptr = LastSlash;\r
-    } else if (*Ptr == '\\' && *(Ptr + 1) == '.' && *(Ptr + 2) == '\\') {\r
-      //\r
-      // Convert a "\.\" to a "\"\r
-      //\r
-\r
-      //\r
-      // Use TmpStr as a backup, as StrCpy in BaseLib does not handle copy of two strings\r
-      // that overlap.\r
-      //\r
-      StrCpy (TmpStr, Ptr + 2);\r
-      StrCpy (Ptr, TmpStr);\r
-      Ptr = LastSlash;\r
-    } else if (*Ptr == '\\') {\r
-      LastSlash = Ptr;\r
-    }\r
-\r
-    Ptr++;\r
-  }\r
-\r
-  FreePool (TmpStr);\r
-\r
-  return Str;\r
-}\r
-\r
-/**\r
-  Create a SECUREBOOT_MENU_ENTRY, and stores it in a buffer allocated from the pool.\r
-\r
-  @return           The new menu entry or NULL of error happens.\r
-\r
-**/\r
-SECUREBOOT_MENU_ENTRY *\r
-CreateMenuEntry (\r
-  VOID\r
-  )\r
-{\r
-  SECUREBOOT_MENU_ENTRY *MenuEntry;\r
-  UINTN                 ContextSize;\r
+  mStartOpCodeHandle = HiiAllocateOpCodeHandle ();\r
 \r
   //\r
-  // Create new menu entry\r
+  // Create Hii Extend Label OpCode as the start opcode\r
   //\r
-  MenuEntry = AllocateZeroPool (sizeof (SECUREBOOT_MENU_ENTRY));\r
-  if (MenuEntry == NULL) {\r
-    return NULL;\r
-  }\r
-\r
-  ContextSize = sizeof (SECUREBOOT_FILE_CONTEXT);\r
-  MenuEntry->FileContext = AllocateZeroPool (ContextSize);\r
-  if (MenuEntry->FileContext == NULL) {\r
-    FreePool (MenuEntry);\r
-    return NULL;\r
-  }\r
-\r
-  MenuEntry->Signature = SECUREBOOT_MENU_ENTRY_SIGNATURE;\r
-\r
-  return MenuEntry;\r
-}\r
-\r
-/**\r
-  Get Menu Entry from the Menu Entry List by MenuNumber.\r
-\r
-  If MenuNumber is great or equal to the number of Menu\r
-  Entry in the list, then ASSERT.\r
-\r
-  @param[in]  MenuOption      The Menu Entry List to read the menu entry.\r
-  @param[in]  MenuNumber      The index of Menu Entry.\r
-\r
-  @return     The Menu Entry.\r
-\r
-**/\r
-SECUREBOOT_MENU_ENTRY *\r
-GetMenuEntry (\r
-  IN  SECUREBOOT_MENU_OPTION      *MenuOption,\r
-  IN  UINTN                       MenuNumber\r
-  )\r
-{\r
-  SECUREBOOT_MENU_ENTRY       *NewMenuEntry;\r
-  UINTN                       Index;\r
-  LIST_ENTRY                  *List;\r
-\r
-  ASSERT (MenuNumber < MenuOption->MenuNumber);\r
-\r
-  List = MenuOption->Head.ForwardLink;\r
-  for (Index = 0; Index < MenuNumber; Index++) {\r
-    List = List->ForwardLink;\r
-  }\r
-\r
-  NewMenuEntry = CR (List, SECUREBOOT_MENU_ENTRY, Link, SECUREBOOT_MENU_ENTRY_SIGNATURE);\r
-\r
-  return NewMenuEntry;\r
-}\r
-\r
-/**\r
-  Create string tokens for a menu from its help strings and display strings.\r
-\r
-  @param[in] HiiHandle          Hii Handle of the package to be updated.\r
-  @param[in] MenuOption         The Menu whose string tokens need to be created.\r
-\r
-**/\r
-VOID\r
-CreateMenuStringToken (\r
-  IN EFI_HII_HANDLE                          HiiHandle,\r
-  IN SECUREBOOT_MENU_OPTION                  *MenuOption\r
-  )\r
-{\r
-  SECUREBOOT_MENU_ENTRY *NewMenuEntry;\r
-  UINTN         Index;\r
-\r
-  for (Index = 0; Index < MenuOption->MenuNumber; Index++) {\r
-    NewMenuEntry = GetMenuEntry (MenuOption, Index);\r
-\r
-    NewMenuEntry->DisplayStringToken = HiiSetString (\r
-                                         HiiHandle,\r
-                                         0,\r
-                                         NewMenuEntry->DisplayString,\r
-                                         NULL\r
-                                         );\r
-\r
-    if (NewMenuEntry->HelpString == NULL) {\r
-      NewMenuEntry->HelpStringToken = NewMenuEntry->DisplayStringToken;\r
-    } else {\r
-      NewMenuEntry->HelpStringToken = HiiSetString (\r
-                                        HiiHandle,\r
-                                        0,\r
-                                        NewMenuEntry->HelpString,\r
-                                        NULL\r
+  mStartLabel = (EFI_IFR_GUID_LABEL *)HiiCreateGuidOpCode (\r
+                                        mStartOpCodeHandle,\r
+                                        &gEfiIfrTianoGuid,\r
+                                        NULL,\r
+                                        sizeof (EFI_IFR_GUID_LABEL)\r
                                         );\r
-    }\r
-  }\r
-}\r
-\r
-/**\r
-  Free up all resources allocated for a SECUREBOOT_MENU_ENTRY.\r
-\r
-  @param[in, out]  MenuEntry   A pointer to SECUREBOOT_MENU_ENTRY.\r
-\r
-**/\r
-VOID\r
-DestroyMenuEntry (\r
-  IN OUT SECUREBOOT_MENU_ENTRY         *MenuEntry\r
-  )\r
-{\r
-  SECUREBOOT_FILE_CONTEXT           *FileContext;\r
-\r
-\r
-  FileContext = (SECUREBOOT_FILE_CONTEXT *) MenuEntry->FileContext;\r
-\r
-  if (!FileContext->IsRoot && FileContext->DevicePath != NULL) {\r
-    FreePool (FileContext->DevicePath);\r
-  } else {\r
-    if (FileContext->FHandle != NULL) {\r
-      FileContext->FHandle->Close (FileContext->FHandle);\r
-    }\r
-  }\r
-\r
-  if (FileContext->FileName != NULL) {\r
-    FreePool (FileContext->FileName);\r
-  }\r
-  if (FileContext->Info != NULL) {\r
-    FreePool (FileContext->Info);\r
-  }\r
-\r
-  FreePool (FileContext);\r
-\r
-  if (MenuEntry->DisplayString != NULL) {\r
-    FreePool (MenuEntry->DisplayString);\r
-  }\r
-  if (MenuEntry->HelpString != NULL) {\r
-    FreePool (MenuEntry->HelpString);\r
-  }\r
-\r
-  FreePool (MenuEntry);\r
+  mStartLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL;\r
 }\r
 \r
 /**\r
-  Free resources allocated in Allocate Rountine.\r
+  Clean up the dynamic opcode at label and form specified by both LabelId.\r
 \r
-  @param[in, out]  MenuOption        Menu to be freed\r
+  @param[in] LabelId         It is both the Form ID and Label ID for opcode deletion.\r
+  @param[in] PrivateData     Module private data.\r
 \r
 **/\r
 VOID\r
-FreeMenu (\r
-  IN OUT SECUREBOOT_MENU_OPTION        *MenuOption\r
-  )\r
-{\r
-  SECUREBOOT_MENU_ENTRY      *MenuEntry;\r
-  while (!IsListEmpty (&MenuOption->Head)) {\r
-    MenuEntry = CR (\r
-                  MenuOption->Head.ForwardLink,\r
-                  SECUREBOOT_MENU_ENTRY,\r
-                  Link,\r
-                  SECUREBOOT_MENU_ENTRY_SIGNATURE\r
-                  );\r
-    RemoveEntryList (&MenuEntry->Link);\r
-    DestroyMenuEntry (MenuEntry);\r
-  }\r
-  MenuOption->MenuNumber = 0;\r
-}\r
-\r
-/**\r
-  This function gets the file information from an open file descriptor, and stores it\r
-  in a buffer allocated from pool.\r
-\r
-  @param[in]  FHand           File Handle.\r
-\r
-  @return    A pointer to a buffer with file information or NULL is returned\r
-\r
-**/\r
-EFI_FILE_INFO *\r
-FileInfo (\r
-  IN EFI_FILE_HANDLE      FHand\r
+CleanUpPage (\r
+  IN UINT16                          LabelId,\r
+  IN SECUREBOOT_CONFIG_PRIVATE_DATA  *PrivateData\r
   )\r
 {\r
-  EFI_STATUS    Status;\r
-  EFI_FILE_INFO *Buffer;\r
-  UINTN         BufferSize;\r
-\r
-  //\r
-  // Initialize for GrowBuffer loop\r
-  //\r
-  Buffer      = NULL;\r
-  BufferSize  = SIZE_OF_EFI_FILE_INFO + 200;\r
+  RefreshUpdateData ();\r
 \r
   //\r
-  // Call the real function\r
+  // Remove all op-codes from dynamic page\r
   //\r
-  while (GrowBuffer (&Status, (VOID **) &Buffer, BufferSize)) {\r
-    Status = FHand->GetInfo (\r
-                      FHand,\r
-                      &gEfiFileInfoGuid,\r
-                      &BufferSize,\r
-                      Buffer\r
-                      );\r
-  }\r
-\r
-  return Buffer;\r
+  mStartLabel->Number = LabelId;\r
+  HiiUpdateForm (\r
+    PrivateData->HiiHandle,\r
+    &gSecureBootConfigFormSetGuid,\r
+    LabelId,\r
+    mStartOpCodeHandle, // Label LabelId\r
+    mEndOpCodeHandle    // LABEL_END\r
+    );\r
 }\r
 \r
 /**\r
-  This function gets the file system information from an open file descriptor,\r
-  and stores it in a buffer allocated from pool.\r
+  Extract filename from device path. The returned buffer is allocated using AllocateCopyPool.\r
+  The caller is responsible for freeing the allocated buffer using FreePool(). If return NULL\r
+  means not enough memory resource.\r
 \r
-  @param[in] FHand       The file handle.\r
+  @param DevicePath       Device path.\r
 \r
-  @return                A pointer to a buffer with file information.\r
-  @retval                NULL is returned if failed to get Vaolume Label Info.\r
+  @retval NULL            Not enough memory resource for AllocateCopyPool.\r
+  @retval Other           A new allocated string that represents the file name.\r
 \r
 **/\r
-EFI_FILE_SYSTEM_VOLUME_LABEL *\r
-FileSystemVolumeLabelInfo (\r
-  IN EFI_FILE_HANDLE      FHand\r
+CHAR16 *\r
+ExtractFileNameFromDevicePath (\r
+  IN   EFI_DEVICE_PATH_PROTOCOL  *DevicePath\r
   )\r
 {\r
-  EFI_STATUS                        Status;\r
-  EFI_FILE_SYSTEM_VOLUME_LABEL      *Buffer;\r
-  UINTN                             BufferSize;\r
-  //\r
-  // Initialize for GrowBuffer loop\r
-  //\r
-  Buffer      = NULL;\r
-  BufferSize  = SIZE_OF_EFI_FILE_SYSTEM_VOLUME_LABEL + 200;\r
-\r
-  //\r
-  // Call the real function\r
-  //\r
-  while (GrowBuffer (&Status, (VOID **) &Buffer, BufferSize)) {\r
-    Status = FHand->GetInfo (\r
-                      FHand,\r
-                      &gEfiFileSystemVolumeLabelInfoIdGuid,\r
-                      &BufferSize,\r
-                      Buffer\r
-                      );\r
-  }\r
-\r
-  return Buffer;\r
-}\r
-\r
-/**\r
-  This function will open a file or directory referenced by DevicePath.\r
-\r
-  This function opens a file with the open mode according to the file path. The\r
-  Attributes is valid only for EFI_FILE_MODE_CREATE.\r
-\r
-  @param[in, out]  FilePath        On input, the device path to the file.\r
-                                   On output, the remaining device path.\r
-  @param[out]      FileHandle      Pointer to the file handle.\r
-  @param[in]       OpenMode        The mode to open the file with.\r
-  @param[in]       Attributes      The file's file attributes.\r
+  CHAR16  *String;\r
+  CHAR16  *MatchString;\r
+  CHAR16  *LastMatch;\r
+  CHAR16  *FileName;\r
+  UINTN   Length;\r
 \r
-  @retval EFI_SUCCESS              The information was set.\r
-  @retval EFI_INVALID_PARAMETER    One of the parameters has an invalid value.\r
-  @retval EFI_UNSUPPORTED          Could not open the file path.\r
-  @retval EFI_NOT_FOUND            The specified file could not be found on the\r
-                                   device or the file system could not be found on\r
-                                   the device.\r
-  @retval EFI_NO_MEDIA             The device has no medium.\r
-  @retval EFI_MEDIA_CHANGED        The device has a different medium in it or the\r
-                                   medium is no longer supported.\r
-  @retval EFI_DEVICE_ERROR         The device reported an error.\r
-  @retval EFI_VOLUME_CORRUPTED     The file system structures are corrupted.\r
-  @retval EFI_WRITE_PROTECTED      The file or medium is write protected.\r
-  @retval EFI_ACCESS_DENIED        The file was opened read only.\r
-  @retval EFI_OUT_OF_RESOURCES     Not enough resources were available to open the\r
-                                   file.\r
-  @retval EFI_VOLUME_FULL          The volume is full.\r
-**/\r
-EFI_STATUS\r
-EFIAPI\r
-OpenFileByDevicePath(\r
-  IN OUT EFI_DEVICE_PATH_PROTOCOL     **FilePath,\r
-  OUT EFI_FILE_HANDLE                 *FileHandle,\r
-  IN UINT64                           OpenMode,\r
-  IN UINT64                           Attributes\r
-  )\r
-{\r
-  EFI_STATUS                      Status;\r
-  EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *EfiSimpleFileSystemProtocol;\r
-  EFI_FILE_PROTOCOL               *Handle1;\r
-  EFI_FILE_PROTOCOL               *Handle2;\r
-  EFI_HANDLE                      DeviceHandle;\r
+  ASSERT (DevicePath != NULL);\r
 \r
-  if ((FilePath == NULL || FileHandle == NULL)) {\r
-    return EFI_INVALID_PARAMETER;\r
-  }\r
+  String      = DevicePathToStr (DevicePath);\r
+  MatchString = String;\r
+  LastMatch   = String;\r
+  FileName    = NULL;\r
 \r
-  Status = gBS->LocateDevicePath (\r
-                  &gEfiSimpleFileSystemProtocolGuid,\r
-                  FilePath,\r
-                  &DeviceHandle\r
-                  );\r
-  if (EFI_ERROR (Status)) {\r
-    return Status;\r
+  while (MatchString != NULL) {\r
+    LastMatch   = MatchString + 1;\r
+    MatchString = StrStr (LastMatch, L"\\");\r
   }\r
 \r
-  Status = gBS->OpenProtocol(\r
-                  DeviceHandle,\r
-                  &gEfiSimpleFileSystemProtocolGuid,\r
-                  (VOID**)&EfiSimpleFileSystemProtocol,\r
-                  gImageHandle,\r
-                  NULL,\r
-                  EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
-                  );\r
-  if (EFI_ERROR (Status)) {\r
-    return Status;\r
+  Length   = StrLen (LastMatch);\r
+  FileName = AllocateCopyPool ((Length + 1) * sizeof (CHAR16), LastMatch);\r
+  if (FileName != NULL) {\r
+    *(FileName + Length) = 0;\r
   }\r
 \r
-  Status = EfiSimpleFileSystemProtocol->OpenVolume(EfiSimpleFileSystemProtocol, &Handle1);\r
-  if (EFI_ERROR (Status)) {\r
-    FileHandle = NULL;\r
-    return Status;\r
-  }\r
+  FreePool (String);\r
 \r
-  //\r
-  // go down directories one node at a time.\r
-  //\r
-  while (!IsDevicePathEnd (*FilePath)) {\r
-    //\r
-    // For file system access each node should be a file path component\r
-    //\r
-    if (DevicePathType    (*FilePath) != MEDIA_DEVICE_PATH ||\r
-        DevicePathSubType (*FilePath) != MEDIA_FILEPATH_DP\r
-       ) {\r
-      FileHandle = NULL;\r
-      return (EFI_INVALID_PARAMETER);\r
-    }\r
-    //\r
-    // Open this file path node\r
-    //\r
-    Handle2  = Handle1;\r
-    Handle1 = NULL;\r
-\r
-    //\r
-    // Try to test opening an existing file\r
-    //\r
-    Status = Handle2->Open (\r
-                          Handle2,\r
-                          &Handle1,\r
-                          ((FILEPATH_DEVICE_PATH*)*FilePath)->PathName,\r
-                          OpenMode &~EFI_FILE_MODE_CREATE,\r
-                          0\r
-                         );\r
-\r
-    //\r
-    // see if the error was that it needs to be created\r
-    //\r
-    if ((EFI_ERROR (Status)) && (OpenMode != (OpenMode &~EFI_FILE_MODE_CREATE))) {\r
-      Status = Handle2->Open (\r
-                            Handle2,\r
-                            &Handle1,\r
-                            ((FILEPATH_DEVICE_PATH*)*FilePath)->PathName,\r
-                            OpenMode,\r
-                            Attributes\r
-                           );\r
-    }\r
-    //\r
-    // Close the last node\r
-    //\r
-    Handle2->Close (Handle2);\r
-\r
-    if (EFI_ERROR(Status)) {\r
-      return (Status);\r
-    }\r
-\r
-    //\r
-    // Get the next node\r
-    //\r
-    *FilePath = NextDevicePathNode (*FilePath);\r
-  }\r
-\r
-  //\r
-  // This is a weak spot since if the undefined SHELL_FILE_HANDLE format changes this must change also!\r
-  //\r
-  *FileHandle = (VOID*)Handle1;\r
-  return EFI_SUCCESS;\r
+  return FileName;\r
 }\r
 \r
 /**\r
-  Function opens and returns a file handle to the root directory of a volume.\r
-\r
-  @param[in]   DeviceHandle    A handle for a device\r
-\r
-  @return      A valid file handle or NULL if error happens.\r
-\r
-**/\r
-EFI_FILE_HANDLE\r
-OpenRoot (\r
-  IN EFI_HANDLE                   DeviceHandle\r
-  )\r
-{\r
-  EFI_STATUS                      Status;\r
-  EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *Volume;\r
-  EFI_FILE_HANDLE                 File;\r
-\r
-  File = NULL;\r
-\r
-  //\r
-  // File the file system interface to the device\r
-  //\r
-  Status = gBS->HandleProtocol (\r
-                  DeviceHandle,\r
-                  &gEfiSimpleFileSystemProtocolGuid,\r
-                  (VOID *) &Volume\r
-                  );\r
-\r
-  //\r
-  // Open the root directory of the volume\r
-  //\r
-  if (!EFI_ERROR (Status)) {\r
-    Status = Volume->OpenVolume (\r
-                      Volume,\r
-                      &File\r
-                      );\r
-  }\r
-  //\r
-  // Done\r
-  //\r
-  return EFI_ERROR (Status) ? NULL : File;\r
-}\r
+  Update  the form base on the selected file.\r
 \r
-/**\r
-  This function builds the FsOptionMenu list which records all\r
-  available file system in the system. They include all instances\r
-  of EFI_SIMPLE_FILE_SYSTEM_PROTOCOL, all instances of EFI_LOAD_FILE_SYSTEM\r
-  and all type of legacy boot device.\r
+  @param FilePath   Point to the file path.\r
+  @param FormId     The form need to display.\r
 \r
-  @retval  EFI_SUCCESS             Success find the file system\r
-  @retval  EFI_OUT_OF_RESOURCES    Can not create menu entry\r
+  @retval TRUE   Exit caller function.\r
+  @retval FALSE  Not exit caller function.\r
 \r
 **/\r
-EFI_STATUS\r
-FindFileSystem (\r
-  VOID\r
+BOOLEAN\r
+UpdatePage (\r
+  IN  EFI_DEVICE_PATH_PROTOCOL  *FilePath,\r
+  IN  EFI_FORM_ID               FormId\r
   )\r
 {\r
-  UINTN                     NoBlkIoHandles;\r
-  UINTN                     NoSimpleFsHandles;\r
-  EFI_HANDLE                *BlkIoHandle;\r
-  EFI_HANDLE                *SimpleFsHandle;\r
-  UINT16                    *VolumeLabel;\r
-  EFI_BLOCK_IO_PROTOCOL     *BlkIo;\r
-  UINTN                     Index;\r
-  EFI_STATUS                Status;\r
-  SECUREBOOT_MENU_ENTRY     *MenuEntry;\r
-  SECUREBOOT_FILE_CONTEXT   *FileContext;\r
-  UINT16                    *TempStr;\r
-  UINTN                     OptionNumber;\r
-  VOID                      *Buffer;\r
-\r
-  BOOLEAN                   RemovableMedia;\r
-\r
-\r
-  NoSimpleFsHandles = 0;\r
-  OptionNumber      = 0;\r
-  InitializeListHead (&FsOptionMenu.Head);\r
-\r
-  //\r
-  // Locate Handles that support BlockIo protocol\r
-  //\r
-  Status = gBS->LocateHandleBuffer (\r
-                  ByProtocol,\r
-                  &gEfiBlockIoProtocolGuid,\r
-                  NULL,\r
-                  &NoBlkIoHandles,\r
-                  &BlkIoHandle\r
-                  );\r
-  if (!EFI_ERROR (Status)) {\r
-\r
-    for (Index = 0; Index < NoBlkIoHandles; Index++) {\r
-      Status = gBS->HandleProtocol (\r
-                      BlkIoHandle[Index],\r
-                      &gEfiBlockIoProtocolGuid,\r
-                      (VOID **) &BlkIo\r
-                      );\r
+  CHAR16         *FileName;\r
+  EFI_STRING_ID  StringToken;\r
 \r
-      if (EFI_ERROR (Status)) {\r
-        continue;\r
-      }\r
+  FileName = NULL;\r
 \r
-      //\r
-      // Issue a dummy read to trigger reinstall of BlockIo protocol for removable media\r
-      //\r
-      if (BlkIo->Media->RemovableMedia) {\r
-        Buffer = AllocateZeroPool (BlkIo->Media->BlockSize);\r
-        if (NULL == Buffer) {\r
-          FreePool (BlkIoHandle);\r
-          return EFI_OUT_OF_RESOURCES;\r
-        }\r
-\r
-        BlkIo->ReadBlocks (\r
-                BlkIo,\r
-                BlkIo->Media->MediaId,\r
-                0,\r
-                BlkIo->Media->BlockSize,\r
-                Buffer\r
-                );\r
-        FreePool (Buffer);\r
-      }\r
-    }\r
-    FreePool (BlkIoHandle);\r
+  if (FilePath != NULL) {\r
+    FileName = ExtractFileNameFromDevicePath (FilePath);\r
   }\r
 \r
-  //\r
-  // Locate Handles that support Simple File System protocol\r
-  //\r
-  Status = gBS->LocateHandleBuffer (\r
-                  ByProtocol,\r
-                  &gEfiSimpleFileSystemProtocolGuid,\r
-                  NULL,\r
-                  &NoSimpleFsHandles,\r
-                  &SimpleFsHandle\r
-                  );\r
-  if (!EFI_ERROR (Status)) {\r
+  if (FileName == NULL) {\r
     //\r
-    // Find all the instances of the File System prototocol\r
+    // FileName = NULL has two case:\r
+    // 1. FilePath == NULL, not select file.\r
+    // 2. FilePath != NULL, but ExtractFileNameFromDevicePath return NULL not enough memory resource.\r
+    // In these two case, no need to update the form, and exit the caller function.\r
     //\r
-    for (Index = 0; Index < NoSimpleFsHandles; Index++) {\r
-      Status = gBS->HandleProtocol (\r
-                      SimpleFsHandle[Index],\r
-                      &gEfiBlockIoProtocolGuid,\r
-                      (VOID **) &BlkIo\r
-                      );\r
-      if (EFI_ERROR (Status)) {\r
-        //\r
-        // If no block IO exists assume it's NOT a removable media\r
-        //\r
-        RemovableMedia = FALSE;\r
-      } else {\r
-        //\r
-        // If block IO exists check to see if it's remobable media\r
-        //\r
-        RemovableMedia = BlkIo->Media->RemovableMedia;\r
-      }\r
-\r
-      //\r
-      // Allocate pool for this instance.\r
-      //\r
-      MenuEntry = CreateMenuEntry ();\r
-      if (NULL == MenuEntry) {\r
-        FreePool (SimpleFsHandle);\r
-        return EFI_OUT_OF_RESOURCES;\r
-      }\r
-\r
-      FileContext = (SECUREBOOT_FILE_CONTEXT *) MenuEntry->FileContext;\r
-\r
-      FileContext->Handle     = SimpleFsHandle[Index];\r
-      MenuEntry->OptionNumber = Index;\r
-      FileContext->FHandle    = OpenRoot (FileContext->Handle);\r
-      if (FileContext->FHandle == NULL) {\r
-        DestroyMenuEntry (MenuEntry);\r
-        continue;\r
-      }\r
-\r
-      MenuEntry->HelpString = DevicePathToStr (DevicePathFromHandle (FileContext->Handle));\r
-      FileContext->Info = FileSystemVolumeLabelInfo (FileContext->FHandle);\r
-      FileContext->FileName = StrDuplicate (L"\\");\r
-      FileContext->DevicePath = FileDevicePath (\r
-                                  FileContext->Handle,\r
-                                  FileContext->FileName\r
-                                  );\r
-      FileContext->IsDir            = TRUE;\r
-      FileContext->IsRoot           = TRUE;\r
-      FileContext->IsRemovableMedia = RemovableMedia;\r
-      FileContext->IsLoadFile       = FALSE;\r
-\r
-      //\r
-      // Get current file system's Volume Label\r
-      //\r
-      if (FileContext->Info == NULL) {\r
-        VolumeLabel = L"NO FILE SYSTEM INFO";\r
-      } else {\r
-        if (FileContext->Info->VolumeLabel == NULL) {\r
-          VolumeLabel = L"NULL VOLUME LABEL";\r
-        } else {\r
-          VolumeLabel = FileContext->Info->VolumeLabel;\r
-          if (*VolumeLabel == 0x0000) {\r
-            VolumeLabel = L"NO VOLUME LABEL";\r
-          }\r
-        }\r
-      }\r
-\r
-      TempStr                   = MenuEntry->HelpString;\r
-      MenuEntry->DisplayString  = AllocateZeroPool (MAX_CHAR);\r
-      ASSERT (MenuEntry->DisplayString != NULL);\r
-      UnicodeSPrint (\r
-        MenuEntry->DisplayString,\r
-        MAX_CHAR,\r
-        L"%s, [%s]",\r
-        VolumeLabel,\r
-        TempStr\r
-        );\r
-      OptionNumber++;\r
-      InsertTailList (&FsOptionMenu.Head, &MenuEntry->Link);\r
-    }\r
+    return TRUE;\r
   }\r
 \r
-  if (NoSimpleFsHandles != 0) {\r
-    FreePool (SimpleFsHandle);\r
-  }\r
+  StringToken =  HiiSetString (gSecureBootPrivateData->HiiHandle, 0, FileName, NULL);\r
+\r
+  gSecureBootPrivateData->FileContext->FileName = FileName;\r
 \r
+  EfiOpenFileByDevicePath (\r
+    &FilePath,\r
+    &gSecureBootPrivateData->FileContext->FHandle,\r
+    EFI_FILE_MODE_READ,\r
+    0\r
+    );\r
   //\r
-  // Remember how many file system options are here\r
+  // Create Subtitle op-code for the display string of the option.\r
   //\r
-  FsOptionMenu.MenuNumber = OptionNumber;\r
-  return EFI_SUCCESS;\r
-}\r
+  RefreshUpdateData ();\r
+  mStartLabel->Number = FormId;\r
+\r
+  HiiCreateSubTitleOpCode (\r
+    mStartOpCodeHandle,\r
+    StringToken,\r
+    0,\r
+    0,\r
+    0\r
+    );\r
 \r
+  HiiUpdateForm (\r
+    gSecureBootPrivateData->HiiHandle,\r
+    &gSecureBootConfigFormSetGuid,\r
+    FormId,\r
+    mStartOpCodeHandle, // Label FormId\r
+    mEndOpCodeHandle    // LABEL_END\r
+    );\r
 \r
-/**\r
-  Find files under the current directory. All files and sub-directories\r
-  in current directory will be stored in DirectoryMenu for future use.\r
+  return TRUE;\r
+}\r
 \r
-  @param[in] MenuEntry     The Menu Entry.\r
+/**\r
+  Update the PK form base on the input file path info.\r
 \r
-  @retval EFI_SUCCESS      Get files from current dir successfully.\r
-  @return Other            Can't get files from current dir.\r
+  @param FilePath    Point to the file path.\r
 \r
+  @retval TRUE   Exit caller function.\r
+  @retval FALSE  Not exit caller function.\r
 **/\r
-EFI_STATUS\r
-FindFiles (\r
-  IN SECUREBOOT_MENU_ENTRY              *MenuEntry\r
+BOOLEAN\r
+EFIAPI\r
+UpdatePKFromFile (\r
+  IN EFI_DEVICE_PATH_PROTOCOL  *FilePath\r
   )\r
 {\r
-  EFI_FILE_HANDLE           NewDir;\r
-  EFI_FILE_HANDLE           Dir;\r
-  EFI_FILE_INFO             *DirInfo;\r
-  UINTN                     BufferSize;\r
-  UINTN                     DirBufferSize;\r
-  SECUREBOOT_MENU_ENTRY     *NewMenuEntry;\r
-  SECUREBOOT_FILE_CONTEXT   *FileContext;\r
-  SECUREBOOT_FILE_CONTEXT   *NewFileContext;\r
-  UINTN                     Pass;\r
-  EFI_STATUS                Status;\r
-  UINTN                     OptionNumber;\r
-\r
-  FileContext   = (SECUREBOOT_FILE_CONTEXT *) MenuEntry->FileContext;\r
-  Dir           = FileContext->FHandle;\r
-  OptionNumber  = 0;\r
-  //\r
-  // Open current directory to get files from it\r
-  //\r
-  Status = Dir->Open (\r
-                  Dir,\r
-                  &NewDir,\r
-                  FileContext->FileName,\r
-                  EFI_FILE_READ_ONLY,\r
-                  0\r
-                  );\r
-  if (!FileContext->IsRoot) {\r
-    Dir->Close (Dir);\r
-  }\r
-\r
-  if (EFI_ERROR (Status)) {\r
-    return Status;\r
-  }\r
-\r
-  DirInfo = FileInfo (NewDir);\r
-  if (DirInfo == NULL) {\r
-    return EFI_NOT_FOUND;\r
-  }\r
-\r
-  if ((DirInfo->Attribute & EFI_FILE_DIRECTORY) == 0) {\r
-    return EFI_INVALID_PARAMETER;\r
-  }\r
-\r
-  FileContext->DevicePath = FileDevicePath (\r
-                              FileContext->Handle,\r
-                              FileContext->FileName\r
-                              );\r
-\r
-  DirBufferSize = sizeof (EFI_FILE_INFO) + 1024;\r
-  DirInfo       = AllocateZeroPool (DirBufferSize);\r
-  if (DirInfo == NULL) {\r
-    return EFI_OUT_OF_RESOURCES;\r
-  }\r
-\r
-  //\r
-  // Get all files in current directory\r
-  // Pass 1 to get Directories\r
-  // Pass 2 to get files that are EFI images\r
-  //\r
-  for (Pass = 1; Pass <= 2; Pass++) {\r
-    NewDir->SetPosition (NewDir, 0);\r
-    for (;;) {\r
-      BufferSize  = DirBufferSize;\r
-      Status      = NewDir->Read (NewDir, &BufferSize, DirInfo);\r
-      if (EFI_ERROR (Status) || BufferSize == 0) {\r
-        break;\r
-      }\r
-\r
-      if (((DirInfo->Attribute & EFI_FILE_DIRECTORY) != 0 && Pass == 2) ||\r
-          ((DirInfo->Attribute & EFI_FILE_DIRECTORY) == 0 && Pass == 1)\r
-          ) {\r
-        //\r
-        // Pass 1 is for Directories\r
-        // Pass 2 is for file names\r
-        //\r
-        continue;\r
-      }\r
-\r
-      NewMenuEntry = CreateMenuEntry ();\r
-      if (NULL == NewMenuEntry) {\r
-        return EFI_OUT_OF_RESOURCES;\r
-      }\r
-\r
-      NewFileContext          = (SECUREBOOT_FILE_CONTEXT *) NewMenuEntry->FileContext;\r
-      NewFileContext->Handle  = FileContext->Handle;\r
-      NewFileContext->FileName = AppendFileName (\r
-                                  FileContext->FileName,\r
-                                  DirInfo->FileName\r
-                                  );\r
-      NewFileContext->FHandle = NewDir;\r
-      NewFileContext->DevicePath = FileDevicePath (\r
-                                    NewFileContext->Handle,\r
-                                    NewFileContext->FileName\r
-                                    );\r
-      NewMenuEntry->HelpString = NULL;\r
-\r
-      NewFileContext->IsDir = (BOOLEAN) ((DirInfo->Attribute & EFI_FILE_DIRECTORY) == EFI_FILE_DIRECTORY);\r
-      if (NewFileContext->IsDir) {\r
-        BufferSize = StrLen (DirInfo->FileName) * 2 + 6;\r
-        NewMenuEntry->DisplayString = AllocateZeroPool (BufferSize);\r
-\r
-        UnicodeSPrint (\r
-          NewMenuEntry->DisplayString,\r
-          BufferSize,\r
-          L"<%s>",\r
-          DirInfo->FileName\r
-          );\r
-\r
-      } else {\r
-        NewMenuEntry->DisplayString = StrDuplicate (DirInfo->FileName);\r
-      }\r
-\r
-      NewFileContext->IsRoot            = FALSE;\r
-      NewFileContext->IsLoadFile        = FALSE;\r
-      NewFileContext->IsRemovableMedia  = FALSE;\r
-\r
-      NewMenuEntry->OptionNumber        = OptionNumber;\r
-      OptionNumber++;\r
-      InsertTailList (&DirectoryMenu.Head, &NewMenuEntry->Link);\r
-    }\r
-  }\r
-\r
-  DirectoryMenu.MenuNumber = OptionNumber;\r
-  FreePool (DirInfo);\r
-  return EFI_SUCCESS;\r
+  return UpdatePage (FilePath, FORMID_ENROLL_PK_FORM);\r
 }\r
 \r
 /**\r
-  Refresh the global UpdateData structure.\r
+  Update the KEK form base on the input file path info.\r
+\r
+  @param FilePath    Point to the file path.\r
 \r
+  @retval TRUE   Exit caller function.\r
+  @retval FALSE  Not exit caller function.\r
 **/\r
-VOID\r
-RefreshUpdateData (\r
-  VOID\r
+BOOLEAN\r
+EFIAPI\r
+UpdateKEKFromFile (\r
+  IN EFI_DEVICE_PATH_PROTOCOL  *FilePath\r
   )\r
 {\r
-  //\r
-  // Free current updated date\r
-  //\r
-  if (mStartOpCodeHandle != NULL) {\r
-    HiiFreeOpCodeHandle (mStartOpCodeHandle);\r
-  }\r
-\r
-  //\r
-  // Create new OpCode Handle\r
-  //\r
-  mStartOpCodeHandle = HiiAllocateOpCodeHandle ();\r
-\r
-  //\r
-  // Create Hii Extend Label OpCode as the start opcode\r
-  //\r
-  mStartLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (\r
-                                         mStartOpCodeHandle,\r
-                                         &gEfiIfrTianoGuid,\r
-                                         NULL,\r
-                                         sizeof (EFI_IFR_GUID_LABEL)\r
-                                         );\r
-  mStartLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL;\r
+  return UpdatePage (FilePath, FORMID_ENROLL_KEK_FORM);\r
 }\r
 \r
 /**\r
-  Update the File Explore page.\r
+  Update the DB form base on the input file path info.\r
 \r
-  @param[in] HiiHandle          Hii Handle of the package to be updated.\r
-  @param[in] MenuOption         The Menu whose string tokens need to be updated.\r
-  @param[in] FeCurrentState     Current file explorer state.\r
+  @param FilePath    Point to the file path.\r
 \r
+  @retval TRUE   Exit caller function.\r
+  @retval FALSE  Not exit caller function.\r
 **/\r
-VOID\r
-UpdateFileExplorePage (\r
-  IN EFI_HII_HANDLE               HiiHandle,\r
-  IN SECUREBOOT_MENU_OPTION       *MenuOption,\r
-  IN FILE_EXPLORER_STATE          FeCurrentState\r
+BOOLEAN\r
+EFIAPI\r
+UpdateDBFromFile (\r
+  IN EFI_DEVICE_PATH_PROTOCOL  *FilePath\r
   )\r
 {\r
-  UINTN                   Index;\r
-  SECUREBOOT_MENU_ENTRY   *NewMenuEntry;\r
-  SECUREBOOT_FILE_CONTEXT *NewFileContext;\r
-  EFI_FORM_ID             FormId;\r
-  EFI_FORM_ID             FileFormId;\r
-\r
-  if (FeCurrentState == FileExplorerStateEnrollPkFile) {\r
-    FormId     = SECUREBOOT_ADD_PK_FILE_FORM_ID;\r
-    FileFormId = FORM_FILE_EXPLORER_ID_PK;\r
-  } else if (FeCurrentState == FileExplorerStateEnrollKekFile) {\r
-    FormId     = FORMID_ENROLL_KEK_FORM;\r
-    FileFormId = FORM_FILE_EXPLORER_ID_KEK;\r
-  } else if (FeCurrentState == FileExplorerStateEnrollSignatureFileToDb) {\r
-    FormId     = SECUREBOOT_ENROLL_SIGNATURE_TO_DB;\r
-    FileFormId = FORM_FILE_EXPLORER_ID_DB;\r
-  } else if (FeCurrentState == FileExplorerStateEnrollSignatureFileToDbx) {\r
-    FormId     = SECUREBOOT_ENROLL_SIGNATURE_TO_DBX;\r
-    FileFormId = FORM_FILE_EXPLORER_ID_DBX;\r
-  } else if (FeCurrentState == FileExplorerStateEnrollSignatureFileToDbt) {\r
-    FormId     = SECUREBOOT_ENROLL_SIGNATURE_TO_DBT;\r
-    FileFormId = FORM_FILE_EXPLORER_ID_DBT;\r
-  } else {\r
-    return;\r
-  }\r
-\r
-  NewMenuEntry    = NULL;\r
-  NewFileContext  = NULL;\r
-\r
-  RefreshUpdateData ();\r
-  mStartLabel->Number = FORM_FILE_EXPLORER_ID;\r
-\r
-  for (Index = 0; Index < MenuOption->MenuNumber; Index++) {\r
-    NewMenuEntry    = GetMenuEntry (MenuOption, Index);\r
-    NewFileContext  = (SECUREBOOT_FILE_CONTEXT *) NewMenuEntry->FileContext;\r
-\r
-    if (NewFileContext->IsDir) {\r
-      //\r
-      // Create Text opcode for directory.\r
-      //\r
-      HiiCreateActionOpCode (\r
-        mStartOpCodeHandle,\r
-        (UINT16) (FILE_OPTION_OFFSET + Index),\r
-        NewMenuEntry->DisplayStringToken,\r
-        STRING_TOKEN (STR_NULL),\r
-        EFI_IFR_FLAG_CALLBACK,\r
-        0\r
-        );\r
-    } else {\r
-\r
-      //\r
-      // Create Goto opcode for file.\r
-      //\r
-      HiiCreateGotoOpCode (\r
-        mStartOpCodeHandle,\r
-        FormId,\r
-        NewMenuEntry->DisplayStringToken,\r
-        STRING_TOKEN (STR_NULL),\r
-        EFI_IFR_FLAG_CALLBACK,\r
-        (UINT16) (FILE_OPTION_GOTO_OFFSET + Index)\r
-        );\r
-    }\r
-  }\r
-\r
-  HiiUpdateForm (\r
-    HiiHandle,\r
-    &gSecureBootConfigFormSetGuid,\r
-    FileFormId,\r
-    mStartOpCodeHandle, // Label FORM_FILE_EXPLORER_ID\r
-    mEndOpCodeHandle    // LABEL_END\r
-    );\r
+  return UpdatePage (FilePath, SECUREBOOT_ENROLL_SIGNATURE_TO_DB);\r
 }\r
 \r
 /**\r
-  Update the file explorer page with the refreshed file system.\r
-\r
-  @param[in] PrivateData     Module private data.\r
-  @param[in] KeyValue        Key value to identify the type of data to expect.\r
+  Update the DBX form base on the input file path info.\r
 \r
-  @retval  TRUE           Inform the caller to create a callback packet to exit file explorer.\r
-  @retval  FALSE          Indicate that there is no need to exit file explorer.\r
+  @param FilePath    Point to the file path.\r
 \r
+  @retval TRUE   Exit caller function.\r
+  @retval FALSE  Not exit caller function.\r
 **/\r
 BOOLEAN\r
-UpdateFileExplorer (\r
-  IN SECUREBOOT_CONFIG_PRIVATE_DATA   *PrivateData,\r
-  IN UINT16                           KeyValue\r
+EFIAPI\r
+UpdateDBXFromFile (\r
+  IN EFI_DEVICE_PATH_PROTOCOL  *FilePath\r
   )\r
 {\r
-  UINT16                              FileOptionMask;\r
-  SECUREBOOT_MENU_ENTRY               *NewMenuEntry;\r
-  SECUREBOOT_FILE_CONTEXT             *NewFileContext;\r
-  EFI_FORM_ID                         FormId;\r
-  BOOLEAN                             ExitFileExplorer;\r
-  EFI_STATUS                          Status;\r
-  EFI_DEVICE_PATH_PROTOCOL            *TmpDevicePath;\r
-\r
-  NewMenuEntry      = NULL;\r
-  NewFileContext    = NULL;\r
-  ExitFileExplorer  = FALSE;\r
-  FileOptionMask    = (UINT16) (FILE_OPTION_MASK & KeyValue);\r
-\r
-  if (PrivateData->FeDisplayContext == FileExplorerDisplayUnknown) {\r
-    //\r
-    // First in, display file system.\r
-    //\r
-    FreeMenu (&FsOptionMenu);\r
-    FindFileSystem ();\r
-\r
-    CreateMenuStringToken (PrivateData->HiiHandle, &FsOptionMenu);\r
-    UpdateFileExplorePage (PrivateData->HiiHandle, &FsOptionMenu, PrivateData->FeCurrentState);\r
-\r
-    PrivateData->FeDisplayContext = FileExplorerDisplayFileSystem;\r
-  } else {\r
-    if (PrivateData->FeDisplayContext == FileExplorerDisplayFileSystem) {\r
-      NewMenuEntry = GetMenuEntry (&FsOptionMenu, FileOptionMask);\r
-    } else if (PrivateData->FeDisplayContext == FileExplorerDisplayDirectory) {\r
-      NewMenuEntry = GetMenuEntry (&DirectoryMenu, FileOptionMask);\r
-    }\r
-\r
-    NewFileContext = (SECUREBOOT_FILE_CONTEXT *) NewMenuEntry->FileContext;\r
-\r
-    if (NewFileContext->IsDir ) {\r
-      PrivateData->FeDisplayContext = FileExplorerDisplayDirectory;\r
-\r
-      RemoveEntryList (&NewMenuEntry->Link);\r
-      FreeMenu (&DirectoryMenu);\r
-      Status = FindFiles (NewMenuEntry);\r
-       if (EFI_ERROR (Status)) {\r
-         ExitFileExplorer = TRUE;\r
-         goto OnExit;\r
-       }\r
-      CreateMenuStringToken (PrivateData->HiiHandle, &DirectoryMenu);\r
-      DestroyMenuEntry (NewMenuEntry);\r
-\r
-      UpdateFileExplorePage (PrivateData->HiiHandle, &DirectoryMenu, PrivateData->FeCurrentState);\r
-\r
-    } else {\r
-      if (PrivateData->FeCurrentState == FileExplorerStateEnrollPkFile) {\r
-        FormId = SECUREBOOT_ADD_PK_FILE_FORM_ID;\r
-      } else if (PrivateData->FeCurrentState == FileExplorerStateEnrollKekFile) {\r
-        FormId = FORMID_ENROLL_KEK_FORM;\r
-      } else if (PrivateData->FeCurrentState == FileExplorerStateEnrollSignatureFileToDb) {\r
-        FormId = SECUREBOOT_ENROLL_SIGNATURE_TO_DB;\r
-      } else if (PrivateData->FeCurrentState == FileExplorerStateEnrollSignatureFileToDbx) {\r
-        FormId = SECUREBOOT_ENROLL_SIGNATURE_TO_DBX;\r
-      } else if (PrivateData->FeCurrentState == FileExplorerStateEnrollSignatureFileToDbt) {\r
-        FormId = SECUREBOOT_ENROLL_SIGNATURE_TO_DBT;\r
-      } else {\r
-        return FALSE;\r
-      }\r
-\r
-      PrivateData->MenuEntry = NewMenuEntry;\r
-      PrivateData->FileContext->FileName = NewFileContext->FileName;\r
-\r
-      TmpDevicePath = NewFileContext->DevicePath;\r
-      OpenFileByDevicePath (\r
-        &TmpDevicePath,\r
-        &PrivateData->FileContext->FHandle,\r
-        EFI_FILE_MODE_READ,\r
-        0\r
-        );\r
-\r
-      //\r
-      // Create Subtitle op-code for the display string of the option.\r
-      //\r
-      RefreshUpdateData ();\r
-      mStartLabel->Number = FormId;\r
-\r
-      HiiCreateSubTitleOpCode (\r
-        mStartOpCodeHandle,\r
-        NewMenuEntry->DisplayStringToken,\r
-        0,\r
-        0,\r
-        0\r
-        );\r
-\r
-      HiiUpdateForm (\r
-        PrivateData->HiiHandle,\r
-        &gSecureBootConfigFormSetGuid,\r
-        FormId,\r
-        mStartOpCodeHandle, // Label FormId\r
-        mEndOpCodeHandle    // LABEL_END\r
-        );\r
-    }\r
-  }\r
-\r
-OnExit:\r
-  return ExitFileExplorer;\r
+  return UpdatePage (FilePath, SECUREBOOT_ENROLL_SIGNATURE_TO_DBX);\r
 }\r
 \r
 /**\r
-  Clean up the dynamic opcode at label and form specified by both LabelId.\r
+  Update the DBT form base on the input file path info.\r
 \r
-  @param[in] LabelId         It is both the Form ID and Label ID for opcode deletion.\r
-  @param[in] PrivateData     Module private data.\r
+  @param FilePath    Point to the file path.\r
 \r
+  @retval TRUE   Exit caller function.\r
+  @retval FALSE  Not exit caller function.\r
 **/\r
-VOID\r
-CleanUpPage (\r
-  IN UINT16                           LabelId,\r
-  IN SECUREBOOT_CONFIG_PRIVATE_DATA   *PrivateData\r
+BOOLEAN\r
+EFIAPI\r
+UpdateDBTFromFile (\r
+  IN EFI_DEVICE_PATH_PROTOCOL  *FilePath\r
   )\r
 {\r
-  RefreshUpdateData ();\r
-\r
-  //\r
-  // Remove all op-codes from dynamic page\r
-  //\r
-  mStartLabel->Number = LabelId;\r
-  HiiUpdateForm (\r
-    PrivateData->HiiHandle,\r
-    &gSecureBootConfigFormSetGuid,\r
-    LabelId,\r
-    mStartOpCodeHandle, // Label LabelId\r
-    mEndOpCodeHandle    // LABEL_END\r
-    );\r
+  return UpdatePage (FilePath, SECUREBOOT_ENROLL_SIGNATURE_TO_DBT);\r
 }\r