[mirror_edk2.git] / MdeModulePkg / Universal / DriverSampleDxe / DriverSample.c

/** @file\r
This is an example of how a driver might export data to the HII protocol to be\r
later utilized by the Setup Protocol\r
\r
Copyright (c) 2004 - 2010, Intel Corporation\r
All rights reserved. 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
\r
**/\r
\r
\r
#include "DriverSample.h"\r
\r
#define DISPLAY_ONLY_MY_ITEM  0x0002\r
\r
EFI_GUID   mFormSetGuid = FORMSET_GUID;\r
EFI_GUID   mInventoryGuid = INVENTORY_GUID;\r
\r
CHAR16     VariableName[] = L"MyIfrNVData";\r
EFI_HANDLE                      DriverHandle[2] = {NULL, NULL};\r
DRIVER_SAMPLE_PRIVATE_DATA      *PrivateData = NULL;\r
\r
HII_VENDOR_DEVICE_PATH  mHiiVendorDevicePath0 = {\r
  {\r
    {\r
      HARDWARE_DEVICE_PATH,\r
      HW_VENDOR_DP,\r
      {\r
        (UINT8) (sizeof (VENDOR_DEVICE_PATH)),\r
        (UINT8) ((sizeof (VENDOR_DEVICE_PATH)) >> 8)\r
      }\r
    },\r
    //\r
    // {C153B68D-EBFC-488e-B110-662867745B87}\r
    //\r
    { 0xc153b68d, 0xebfc, 0x488e, { 0xb1, 0x10, 0x66, 0x28, 0x67, 0x74, 0x5b, 0x87 } }\r
  },\r
  {\r
    END_DEVICE_PATH_TYPE,\r
    END_ENTIRE_DEVICE_PATH_SUBTYPE,\r
    {\r
      (UINT8) (END_DEVICE_PATH_LENGTH),\r
      (UINT8) ((END_DEVICE_PATH_LENGTH) >> 8)\r
    }\r
  }\r
};\r
\r
HII_VENDOR_DEVICE_PATH  mHiiVendorDevicePath1 = {\r
  {\r
    {\r
      HARDWARE_DEVICE_PATH,\r
      HW_VENDOR_DP,\r
      {\r
        (UINT8) (sizeof (VENDOR_DEVICE_PATH)),\r
        (UINT8) ((sizeof (VENDOR_DEVICE_PATH)) >> 8)\r
      }\r
    },\r
    //\r
    // {06F37F07-0C48-40e9-8436-0A08A0BB76B0}\r
    //\r
    { 0x6f37f07, 0xc48, 0x40e9, { 0x84, 0x36, 0xa, 0x8, 0xa0, 0xbb, 0x76, 0xb0 } }\r
  },\r
  {\r
    END_DEVICE_PATH_TYPE,\r
    END_ENTIRE_DEVICE_PATH_SUBTYPE,\r
    {\r
      (UINT8) (END_DEVICE_PATH_LENGTH),\r
      (UINT8) ((END_DEVICE_PATH_LENGTH) >> 8)\r
    }\r
  }\r
};\r
\r
/**\r
  Encode the password using a simple algorithm.\r
\r
  @param Password The string to be encoded.\r
  @param MaxSize  The size of the string.\r
\r
**/\r
VOID\r
EncodePassword (\r
  IN  CHAR16                      *Password,\r
  IN  UINTN                       MaxSize\r
  )\r
{\r
  UINTN   Index;\r
  UINTN   Loop;\r
  CHAR16  *Buffer;\r
  CHAR16  *Key;\r
\r
  Key     = L"MAR10648567";\r
  Buffer  = AllocateZeroPool (MaxSize);\r
  ASSERT (Buffer != NULL);\r
\r
  for (Index = 0; Key[Index] != 0; Index++) {\r
    for (Loop = 0; Loop < (UINT8) (MaxSize / 2); Loop++) {\r
      Buffer[Loop] = (CHAR16) (Password[Loop] ^ Key[Index]);\r
    }\r
  }\r
\r
  CopyMem (Password, Buffer, MaxSize);\r
\r
  FreePool (Buffer);\r
  return ;\r
}\r
\r
/**\r
  Validate the user's password.\r
\r
  @param PrivateData This driver's private context data.\r
  @param StringId    The user's input.\r
\r
  @retval EFI_SUCCESS   The user's input matches the password.\r
  @retval EFI_NOT_READY The user's input does not match the password.\r
**/\r
EFI_STATUS\r
ValidatePassword (\r
  IN       DRIVER_SAMPLE_PRIVATE_DATA      *PrivateData,\r
  IN       EFI_STRING_ID                   StringId\r
  )\r
{\r
  EFI_STATUS                      Status;\r
  UINTN                           Index;\r
  UINTN                           BufferSize;\r
  UINTN                           PasswordMaxSize;\r
  CHAR16                          *Password;\r
  CHAR16                          *EncodedPassword;\r
  BOOLEAN                         OldPassword;\r
\r
  //\r
  // Get encoded password first\r
  //\r
  BufferSize = sizeof (DRIVER_SAMPLE_CONFIGURATION);\r
  Status = gRT->GetVariable (\r
                  VariableName,\r
                  &mFormSetGuid,\r
                  NULL,\r
                  &BufferSize,\r
                  &PrivateData->Configuration\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    //\r
    // Old password not exist, prompt for new password\r
    //\r
    return EFI_SUCCESS;\r
  }\r
\r
  OldPassword = FALSE;\r
  PasswordMaxSize = sizeof (PrivateData->Configuration.WhatIsThePassword2);\r
  //\r
  // Check whether we have any old password set\r
  //\r
  for (Index = 0; Index < PasswordMaxSize / sizeof (UINT16); Index++) {\r
    if (PrivateData->Configuration.WhatIsThePassword2[Index] != 0) {\r
      OldPassword = TRUE;\r
      break;\r
    }\r
  }\r
  if (!OldPassword) {\r
    //\r
    // Old password not exist, return EFI_SUCCESS to prompt for new password\r
    //\r
    return EFI_SUCCESS;\r
  }\r
\r
  //\r
  // Get user input password\r
  //\r
  Password = HiiGetString (PrivateData->HiiHandle[0], StringId, NULL);\r
  if (Password == NULL) {\r
    return EFI_NOT_READY;\r
  }\r
  if (StrSize (Password) > PasswordMaxSize) {\r
    FreePool (Password);\r
    return EFI_NOT_READY;\r
  }\r
\r
  //\r
  // Validate old password\r
  //\r
  EncodedPassword = AllocateZeroPool (PasswordMaxSize);\r
  ASSERT (EncodedPassword != NULL);\r
  StrnCpy (EncodedPassword, Password, StrLen (Password));\r
  EncodePassword (EncodedPassword, StrLen (EncodedPassword) * sizeof (CHAR16));\r
  if (CompareMem (EncodedPassword, PrivateData->Configuration.WhatIsThePassword2, StrLen (EncodedPassword) * sizeof (CHAR16)) != 0) {\r
    //\r
    // Old password mismatch, return EFI_NOT_READY to prompt for error message\r
    //\r
    Status = EFI_NOT_READY;\r
  } else {\r
    Status = EFI_SUCCESS;\r
  }\r
\r
  FreePool (Password);\r
  FreePool (EncodedPassword);\r
\r
  return Status;\r
}\r
\r
/**\r
  Encode the password using a simple algorithm.\r
\r
  @param PrivateData This driver's private context data.\r
  @param StringId    The password from User.\r
\r
  @retval  EFI_SUCESS The operation is successful.\r
  @return  Other value if gRT->SetVariable () fails.\r
\r
**/\r
EFI_STATUS\r
SetPassword (\r
  IN DRIVER_SAMPLE_PRIVATE_DATA      *PrivateData,\r
  IN EFI_STRING_ID                   StringId\r
  )\r
{\r
  EFI_STATUS                      Status;\r
  CHAR16                          *Password;\r
  CHAR16                          *TempPassword;\r
  UINTN                           PasswordSize;\r
  DRIVER_SAMPLE_CONFIGURATION     *Configuration;\r
  UINTN                           BufferSize;\r
\r
  //\r
  // Get Buffer Storage data from EFI variable\r
  //\r
  BufferSize = sizeof (DRIVER_SAMPLE_CONFIGURATION);\r
  Status = gRT->GetVariable (\r
                  VariableName,\r
                  &mFormSetGuid,\r
                  NULL,\r
                  &BufferSize,\r
                  &PrivateData->Configuration\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Get user input password\r
  //\r
  Password = &PrivateData->Configuration.WhatIsThePassword2[0];\r
  PasswordSize = sizeof (PrivateData->Configuration.WhatIsThePassword2);\r
  ZeroMem (Password, PasswordSize);\r
\r
  TempPassword = HiiGetString (PrivateData->HiiHandle[0], StringId, NULL);\r
  if (TempPassword == NULL) {\r
    return EFI_NOT_READY;\r
  }\r
  if (StrSize (TempPassword) > PasswordSize) {\r
    FreePool (TempPassword);\r
    return EFI_NOT_READY;\r
  }\r
  StrnCpy (Password, TempPassword, StrLen (TempPassword));\r
  FreePool (TempPassword);\r
\r
  //\r
  // Retrive uncommitted data from Browser\r
  //\r
  Configuration = AllocateZeroPool (sizeof (DRIVER_SAMPLE_CONFIGURATION));\r
  ASSERT (Configuration != NULL);\r
  if (HiiGetBrowserData (&mFormSetGuid, VariableName, sizeof (DRIVER_SAMPLE_CONFIGURATION), (UINT8 *) Configuration)) {\r
    //\r
    // Update password's clear text in the screen\r
    //\r
    CopyMem (Configuration->PasswordClearText, Password, StrSize (Password));\r
\r
    //\r
    // Update uncommitted data of Browser\r
    //\r
    HiiSetBrowserData (\r
       &mFormSetGuid,\r
       VariableName,\r
       sizeof (DRIVER_SAMPLE_CONFIGURATION),\r
       (UINT8 *) Configuration,\r
       NULL\r
       );\r
  }\r
\r
  //\r
  // Free Configuration Buffer\r
  //\r
  FreePool (Configuration);\r
\r
\r
  //\r
  // Set password\r
  //\r
  EncodePassword (Password, StrLen (Password) * 2);\r
  Status = gRT->SetVariable(\r
                  VariableName,\r
                  &mFormSetGuid,\r
                  EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS,\r
                  sizeof (DRIVER_SAMPLE_CONFIGURATION),\r
                  &PrivateData->Configuration\r
                  );\r
  return Status;\r
}\r
\r
/**\r
 Update names of Name/Value storage to current language.\r
\r
 @param PrivateData   Points to the driver private data.\r
\r
 @retval EFI_SUCCESS   All names are successfully updated.\r
 @retval EFI_NOT_FOUND Failed to get Name from HII database.\r
\r
**/\r
EFI_STATUS\r
LoadNameValueNames (\r
  IN DRIVER_SAMPLE_PRIVATE_DATA      *PrivateData\r
  )\r
{\r
  UINTN      Index;\r
\r
  //\r
  // Get Name/Value name string of current language\r
  //\r
  for (Index = 0; Index < NAME_VALUE_NAME_NUMBER; Index++) {\r
    PrivateData->NameValueName[Index] = HiiGetString (\r
                                         PrivateData->HiiHandle[0],\r
                                         PrivateData->NameStringId[Index],\r
                                         NULL\r
                                         );\r
    if (PrivateData->NameValueName[Index] == NULL) {\r
      return EFI_NOT_FOUND;\r
    }\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  This function allows a caller to extract the current configuration for one\r
  or more named elements from the target driver.\r
\r
  @param  This                   Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL.\r
  @param  Request                A null-terminated Unicode string in\r
                                 <ConfigRequest> format.\r
  @param  Progress               On return, points to a character in the Request\r
                                 string. Points to the string's null terminator if\r
                                 request was successful. Points to the most recent\r
                                 '&' before the first failing name/value pair (or\r
                                 the beginning of the string if the failure is in\r
                                 the first name/value pair) if the request was not\r
                                 successful.\r
  @param  Results                A null-terminated Unicode string in\r
                                 <ConfigAltResp> format which has all values filled\r
                                 in for the names in the Request string. String to\r
                                 be allocated by the called function.\r
\r
  @retval EFI_SUCCESS            The Results is filled with the requested values.\r
  @retval EFI_OUT_OF_RESOURCES   Not enough memory to store the results.\r
  @retval EFI_INVALID_PARAMETER  Request is NULL, illegal syntax, or unknown name.\r
  @retval EFI_NOT_FOUND          Routing data doesn't match any storage in this\r
                                 driver.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
ExtractConfig (\r
  IN  CONST EFI_HII_CONFIG_ACCESS_PROTOCOL   *This,\r
  IN  CONST EFI_STRING                       Request,\r
  OUT EFI_STRING                             *Progress,\r
  OUT EFI_STRING                             *Results\r
  )\r
{\r
  EFI_STATUS                       Status;\r
  UINTN                            BufferSize;\r
  DRIVER_SAMPLE_PRIVATE_DATA       *PrivateData;\r
  EFI_HII_CONFIG_ROUTING_PROTOCOL  *HiiConfigRouting;\r
  EFI_STRING                       ConfigRequest;\r
  EFI_STRING                       ConfigRequestHdr;\r
  UINTN                            Size;\r
  EFI_STRING                       Value;\r
  UINTN                            ValueStrLen;\r
  CHAR16                           BackupChar;\r
  CHAR16                           *StrPointer;\r
  BOOLEAN                          AllocatedRequest;\r
\r
  if (Progress == NULL || Results == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  //\r
  // Initialize the local variables.\r
  //\r
  ConfigRequestHdr  = NULL;\r
  ConfigRequest     = NULL;\r
  Size              = 0;\r
  *Progress         = Request;\r
  AllocatedRequest  = FALSE;\r
\r
  PrivateData = DRIVER_SAMPLE_PRIVATE_FROM_THIS (This);\r
  HiiConfigRouting = PrivateData->HiiConfigRouting;\r
\r
  //\r
  // Get Buffer Storage data from EFI variable.\r
  // Try to get the current setting from variable.\r
  //\r
  BufferSize = sizeof (DRIVER_SAMPLE_CONFIGURATION);\r
  Status = gRT->GetVariable (\r
            VariableName,\r
            &mFormSetGuid,\r
            NULL,\r
            &BufferSize,\r
            &PrivateData->Configuration\r
            );\r
  if (EFI_ERROR (Status)) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  if (Request == NULL) {\r
    //\r
    // Request is set to NULL, construct full request string.\r
    //\r
\r
    //\r
    // Allocate and fill a buffer large enough to hold the <ConfigHdr> template\r
    // followed by "&OFFSET=0&WIDTH=WWWWWWWWWWWWWWWW" followed by a Null-terminator\r
    //\r
    ConfigRequestHdr = HiiConstructConfigHdr (&mFormSetGuid, VariableName, PrivateData->DriverHandle[0]);\r
    Size = (StrLen (ConfigRequestHdr) + 32 + 1) * sizeof (CHAR16);\r
    ConfigRequest = AllocateZeroPool (Size);\r
    AllocatedRequest = TRUE;\r
    UnicodeSPrint (ConfigRequest, Size, L"%s&OFFSET=0&WIDTH=%016LX", ConfigRequestHdr, (UINT64)BufferSize);\r
    FreePool (ConfigRequestHdr);\r
  } else {\r
    //\r
    // Check routing data in <ConfigHdr>.\r
    // Note: if only one Storage is used, then this checking could be skipped.\r
    //\r
    if (!HiiIsConfigHdrMatch (Request, &mFormSetGuid, NULL)) {\r
      return EFI_NOT_FOUND;\r
    }\r
    //\r
    // Set Request to the unified request string.\r
    //\r
    ConfigRequest = Request;\r
    //\r
    // Check whether Request includes Request Element.\r
    //\r
    if (StrStr (Request, L"OFFSET") == NULL) {\r
      //\r
      // Check Request Element does exist in Reques String\r
      //\r
      StrPointer = StrStr (Request, L"PATH");\r
      if (StrPointer == NULL) {\r
        return EFI_INVALID_PARAMETER;\r
      }\r
      if (StrStr (StrPointer, L"&") == NULL) {\r
        Size = (StrLen (Request) + 32 + 1) * sizeof (CHAR16);\r
        ConfigRequest    = AllocateZeroPool (Size);\r
        AllocatedRequest = TRUE;\r
        UnicodeSPrint (ConfigRequest, Size, L"%s&OFFSET=0&WIDTH=%016LX", Request, (UINT64)BufferSize);\r
      }\r
    }\r
  }\r
\r
  //\r
  // Check if requesting Name/Value storage\r
  //\r
  if (StrStr (ConfigRequest, L"OFFSET") == NULL) {\r
    //\r
    // Update Name/Value storage Names\r
    //\r
    Status = LoadNameValueNames (PrivateData);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
\r
    //\r
    // Allocate memory for <ConfigResp>, e.g. Name0=0x11, Name1=0x1234, Name2="ABCD"\r
    // <Request>   ::=<ConfigHdr>&Name0&Name1&Name2\r
    // <ConfigResp>::=<ConfigHdr>&Name0=11&Name1=1234&Name2=0041004200430044\r
    //\r
    BufferSize = (StrLen (ConfigRequest) +\r
      1 + sizeof (PrivateData->Configuration.NameValueVar0) * 2 +\r
      1 + sizeof (PrivateData->Configuration.NameValueVar1) * 2 +\r
      1 + sizeof (PrivateData->Configuration.NameValueVar2) * 2 + 1) * sizeof (CHAR16);\r
    *Results = AllocateZeroPool (BufferSize);\r
    ASSERT (*Results != NULL);\r
    StrCpy (*Results, ConfigRequest);\r
    Value = *Results;\r
\r
    //\r
    // Append value of NameValueVar0, type is UINT8\r
    //\r
    if ((Value = StrStr (*Results, PrivateData->NameValueName[0])) != NULL) {\r
      Value += StrLen (PrivateData->NameValueName[0]);\r
      ValueStrLen = ((sizeof (PrivateData->Configuration.NameValueVar0) * 2) + 1);\r
      CopyMem (Value + ValueStrLen, Value, StrSize (Value));\r
\r
      BackupChar = Value[ValueStrLen];\r
      *Value++   = L'=';\r
      Value += UnicodeValueToString (\r
                 Value, \r
                 PREFIX_ZERO | RADIX_HEX, \r
                 PrivateData->Configuration.NameValueVar0, \r
                 sizeof (PrivateData->Configuration.NameValueVar0) * 2\r
                 );\r
      *Value = BackupChar;\r
    }\r
\r
    //\r
    // Append value of NameValueVar1, type is UINT16\r
    //\r
    if ((Value = StrStr (*Results, PrivateData->NameValueName[1])) != NULL) {\r
      Value += StrLen (PrivateData->NameValueName[1]);\r
      ValueStrLen = ((sizeof (PrivateData->Configuration.NameValueVar1) * 2) + 1);\r
      CopyMem (Value + ValueStrLen, Value, StrSize (Value));\r
\r
      BackupChar = Value[ValueStrLen];\r
      *Value++   = L'=';\r
      Value += UnicodeValueToString (\r
                Value, \r
                PREFIX_ZERO | RADIX_HEX, \r
                PrivateData->Configuration.NameValueVar1, \r
                sizeof (PrivateData->Configuration.NameValueVar1) * 2\r
                );\r
      *Value = BackupChar;\r
    }\r
\r
    //\r
    // Append value of NameValueVar2, type is CHAR16 *\r
    //\r
    if ((Value = StrStr (*Results, PrivateData->NameValueName[2])) != NULL) {\r
      Value += StrLen (PrivateData->NameValueName[2]);\r
      ValueStrLen = StrLen (PrivateData->Configuration.NameValueVar2) * 4 + 1;\r
      CopyMem (Value + ValueStrLen, Value, StrSize (Value));\r
\r
      *Value++ = L'=';\r
      //\r
      // Convert Unicode String to Config String, e.g. "ABCD" => "0041004200430044"\r
      //\r
      StrPointer = (CHAR16 *) PrivateData->Configuration.NameValueVar2;\r
      for (; *StrPointer != L'\0'; StrPointer++) {\r
        Value += UnicodeValueToString (Value, PREFIX_ZERO | RADIX_HEX, *StrPointer, 4);\r
      }\r
    }\r
    \r
    Status = EFI_SUCCESS;\r
  } else {\r
    //\r
    // Convert buffer data to <ConfigResp> by helper function BlockToConfig()\r
    //\r
    Status = HiiConfigRouting->BlockToConfig (\r
                                  HiiConfigRouting,\r
                                  ConfigRequest,\r
                                  (UINT8 *) &PrivateData->Configuration,\r
                                  BufferSize,\r
                                  Results,\r
                                  Progress\r
                                  );\r
  }\r
\r
  //\r
  // Free the allocated config request string.\r
  //\r
  if (AllocatedRequest) {\r
    FreePool (ConfigRequest);\r
  }\r
  //\r
  // Set Progress string to the original request string.\r
  //\r
  if (Request == NULL) {\r
    *Progress = NULL;\r
  } else if (StrStr (Request, L"OFFSET") == NULL) {\r
    *Progress = Request + StrLen (Request);\r
  }\r
\r
  return Status;\r
}\r
\r
\r
/**\r
  This function processes the results of changes in configuration.\r
\r
  @param  This                   Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL.\r
  @param  Configuration          A null-terminated Unicode string in <ConfigResp>\r
                                 format.\r
  @param  Progress               A pointer to a string filled in with the offset of\r
                                 the most recent '&' before the first failing\r
                                 name/value pair (or the beginning of the string if\r
                                 the failure is in the first name/value pair) or\r
                                 the terminating NULL if all was successful.\r
\r
  @retval EFI_SUCCESS            The Results is processed successfully.\r
  @retval EFI_INVALID_PARAMETER  Configuration is NULL.\r
  @retval EFI_NOT_FOUND          Routing data doesn't match any storage in this\r
                                 driver.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
RouteConfig (\r
  IN  CONST EFI_HII_CONFIG_ACCESS_PROTOCOL   *This,\r
  IN  CONST EFI_STRING                       Configuration,\r
  OUT EFI_STRING                             *Progress\r
  )\r
{\r
  EFI_STATUS                       Status;\r
  UINTN                            BufferSize;\r
  DRIVER_SAMPLE_PRIVATE_DATA       *PrivateData;\r
  EFI_HII_CONFIG_ROUTING_PROTOCOL  *HiiConfigRouting;\r
  CHAR16                           *Value;\r
  CHAR16                           *StrPtr;\r
  CHAR16                           TemStr[5];\r
  UINT8                            *DataBuffer;\r
  UINT8                            DigitUint8;\r
  UINTN                            Index;\r
  CHAR16                           *StrBuffer;\r
\r
  if (Configuration == NULL || Progress == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  PrivateData = DRIVER_SAMPLE_PRIVATE_FROM_THIS (This);\r
  HiiConfigRouting = PrivateData->HiiConfigRouting;\r
  *Progress = Configuration;\r
\r
  //\r
  // Check routing data in <ConfigHdr>.\r
  // Note: if only one Storage is used, then this checking could be skipped.\r
  //\r
  if (!HiiIsConfigHdrMatch (Configuration, &mFormSetGuid, NULL)) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  //\r
  // Get Buffer Storage data from EFI variable\r
  //\r
  BufferSize = sizeof (DRIVER_SAMPLE_CONFIGURATION);\r
  Status = gRT->GetVariable (\r
            VariableName,\r
            &mFormSetGuid,\r
            NULL,\r
            &BufferSize,\r
            &PrivateData->Configuration\r
            );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Check if configuring Name/Value storage\r
  //\r
  if (StrStr (Configuration, L"OFFSET") == NULL) {\r
    //\r
    // Update Name/Value storage Names\r
    //\r
    Status = LoadNameValueNames (PrivateData);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
\r
    //\r
    // Convert value for NameValueVar0\r
    //\r
    if ((Value = StrStr (Configuration, PrivateData->NameValueName[0])) != NULL) {\r
      //\r
      // Skip "Name="\r
      //\r
      Value += StrLen (PrivateData->NameValueName[0]);\r
      Value++;\r
      //\r
      // Get Value String\r
      //\r
      StrPtr = StrStr (Value, L"&");\r
      if (StrPtr == NULL) {\r
        StrPtr = Value + StrLen (Value);\r
      }\r
      //\r
      // Convert Value to Buffer data\r
      //\r
      DataBuffer = (UINT8 *) &PrivateData->Configuration.NameValueVar0;\r
      ZeroMem (TemStr, sizeof (TemStr));\r
      for (Index = 0, StrPtr --; StrPtr >= Value; StrPtr --, Index ++) {\r
        TemStr[0] = *StrPtr;\r
        DigitUint8 = (UINT8) StrHexToUint64 (TemStr);\r
        if ((Index & 1) == 0) {\r
          DataBuffer [Index/2] = DigitUint8;\r
        } else {\r
          DataBuffer [Index/2] = (UINT8) ((UINT8) (DigitUint8 << 4) + DataBuffer [Index/2]);\r
        }\r
      }\r
    }\r
\r
    //\r
    // Convert value for NameValueVar1\r
    //\r
    if ((Value = StrStr (Configuration, PrivateData->NameValueName[1])) != NULL) {\r
      //\r
      // Skip "Name="\r
      //\r
      Value += StrLen (PrivateData->NameValueName[1]);\r
      Value++;\r
      //\r
      // Get Value String\r
      //\r
      StrPtr = StrStr (Value, L"&");\r
      if (StrPtr == NULL) {\r
        StrPtr = Value + StrLen (Value);\r
      }\r
      //\r
      // Convert Value to Buffer data\r
      //\r
      DataBuffer = (UINT8 *) &PrivateData->Configuration.NameValueVar1;\r
      ZeroMem (TemStr, sizeof (TemStr));\r
      for (Index = 0, StrPtr --; StrPtr >= Value; StrPtr --, Index ++) {\r
        TemStr[0] = *StrPtr;\r
        DigitUint8 = (UINT8) StrHexToUint64 (TemStr);\r
        if ((Index & 1) == 0) {\r
          DataBuffer [Index/2] = DigitUint8;\r
        } else {\r
          DataBuffer [Index/2] = (UINT8) ((UINT8) (DigitUint8 << 4) + DataBuffer [Index/2]);\r
        }\r
      }\r
    }\r
\r
    //\r
    // Convert value for NameValueVar2\r
    //\r
    if ((Value = StrStr (Configuration, PrivateData->NameValueName[2])) != NULL) {\r
      //\r
      // Skip "Name="\r
      //\r
      Value += StrLen (PrivateData->NameValueName[2]);\r
      Value++;\r
      //\r
      // Get Value String\r
      //\r
      StrPtr = StrStr (Value, L"&");\r
      if (StrPtr == NULL) {\r
        StrPtr = Value + StrLen (Value);\r
      }\r
      //\r
      // Convert Config String to Unicode String, e.g "0041004200430044" => "ABCD"\r
      //\r
      StrBuffer = (CHAR16 *) PrivateData->Configuration.NameValueVar2;\r
      ZeroMem (TemStr, sizeof (TemStr));\r
      while (Value < StrPtr) {\r
        StrnCpy (TemStr, Value, 4);\r
        *(StrBuffer++) = (CHAR16) StrHexToUint64 (TemStr);\r
        Value += 4;\r
      }\r
      *StrBuffer = L'\0';\r
    }\r
\r
    //\r
    // Store Buffer Storage back to EFI variable\r
    //\r
    Status = gRT->SetVariable(\r
      VariableName,\r
      &mFormSetGuid,\r
      EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS,\r
      sizeof (DRIVER_SAMPLE_CONFIGURATION),\r
      &PrivateData->Configuration\r
      );\r
\r
    return Status;\r
  }\r
\r
  //\r
  // Convert <ConfigResp> to buffer data by helper function ConfigToBlock()\r
  //\r
  BufferSize = sizeof (DRIVER_SAMPLE_CONFIGURATION);\r
  Status = HiiConfigRouting->ConfigToBlock (\r
                               HiiConfigRouting,\r
                               Configuration,\r
                               (UINT8 *) &PrivateData->Configuration,\r
                               &BufferSize,\r
                               Progress\r
                               );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Store Buffer Storage back to EFI variable\r
  //\r
  Status = gRT->SetVariable(\r
                  VariableName,\r
                  &mFormSetGuid,\r
                  EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS,\r
                  sizeof (DRIVER_SAMPLE_CONFIGURATION),\r
                  &PrivateData->Configuration\r
                  );\r
\r
  return Status;\r
}\r
\r
\r
/**\r
  This function processes the results of changes in configuration.\r
\r
  @param  This                   Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL.\r
  @param  Action                 Specifies the type of action taken by the browser.\r
  @param  QuestionId             A unique value which is sent to the original\r
                                 exporting driver so that it can identify the type\r
                                 of data to expect.\r
  @param  Type                   The type of value for the question.\r
  @param  Value                  A pointer to the data being sent to the original\r
                                 exporting driver.\r
  @param  ActionRequest          On return, points to the action requested by the\r
                                 callback function.\r
\r
  @retval EFI_SUCCESS            The callback successfully handled the action.\r
  @retval EFI_OUT_OF_RESOURCES   Not enough storage is available to hold the\r
                                 variable and its data.\r
  @retval EFI_DEVICE_ERROR       The variable could not be saved.\r
  @retval EFI_UNSUPPORTED        The specified Action is not supported by the\r
                                 callback.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
DriverCallback (\r
  IN  CONST EFI_HII_CONFIG_ACCESS_PROTOCOL   *This,\r
  IN  EFI_BROWSER_ACTION                     Action,\r
  IN  EFI_QUESTION_ID                        QuestionId,\r
  IN  UINT8                                  Type,\r
  IN  EFI_IFR_TYPE_VALUE                     *Value,\r
  OUT EFI_BROWSER_ACTION_REQUEST             *ActionRequest\r
  )\r
{\r
  DRIVER_SAMPLE_PRIVATE_DATA      *PrivateData;\r
  EFI_STATUS                      Status;\r
  UINT8                           MyVar;\r
  VOID                            *StartOpCodeHandle;\r
  VOID                            *OptionsOpCodeHandle;\r
  EFI_IFR_GUID_LABEL              *StartLabel;\r
  VOID                            *EndOpCodeHandle;\r
  EFI_IFR_GUID_LABEL              *EndLabel;\r
  EFI_INPUT_KEY                   Key;\r
  DRIVER_SAMPLE_CONFIGURATION     *Configuration;\r
  UINTN                           MyVarSize;\r
\r
  if (Action == EFI_BROWSER_ACTION_FORM_OPEN) {\r
    //\r
    // On FORM_OPEN event, update the form on-the-fly\r
    //\r
    PrivateData = DRIVER_SAMPLE_PRIVATE_FROM_THIS (This);\r
\r
    //\r
    // Initialize the container for dynamic opcodes\r
    //\r
    StartOpCodeHandle = HiiAllocateOpCodeHandle ();\r
    ASSERT (StartOpCodeHandle != NULL);\r
\r
    //\r
    // Create Hii Extend Label OpCode as the start opcode\r
    //\r
    StartLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (StartOpCodeHandle, &gEfiIfrTianoGuid, NULL, sizeof (EFI_IFR_GUID_LABEL));\r
    StartLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL;\r
    StartLabel->Number       = LABEL_UPDATE2;\r
\r
    HiiCreateActionOpCode (\r
      StartOpCodeHandle,                // Container for dynamic created opcodes\r
      0x1238,                           // Question ID\r
      STRING_TOKEN(STR_SAVE_TEXT),      // Prompt text\r
      STRING_TOKEN(STR_SAVE_TEXT),      // Help text\r
      EFI_IFR_FLAG_CALLBACK,            // Question flag\r
      0                                 // Action String ID\r
    );\r
\r
    HiiUpdateForm (\r
      PrivateData->HiiHandle[0],  // HII handle\r
      &mFormSetGuid,              // Formset GUID\r
      0x3,                        // Form ID\r
      StartOpCodeHandle,          // Label for where to insert opcodes\r
      NULL                        // Insert data\r
      );\r
\r
    HiiFreeOpCodeHandle (StartOpCodeHandle);\r
    return EFI_SUCCESS;\r
  }\r
\r
  if (Action == EFI_BROWSER_ACTION_FORM_CLOSE) {\r
    //\r
    // On FORM_CLOSE event, show up a pop-up\r
    //\r
    do {\r
      CreatePopUp (\r
        EFI_LIGHTGRAY | EFI_BACKGROUND_BLUE,\r
        &Key,\r
        L"",\r
        L"You are going to leave the Form!",\r
        L"Press ESC or ENTER to continue ...",\r
        L"",\r
        NULL\r
        );\r
    } while ((Key.ScanCode != SCAN_ESC) && (Key.UnicodeChar != CHAR_CARRIAGE_RETURN));\r
\r
    return EFI_SUCCESS;\r
  }\r
\r
  if ((Value == NULL) || (ActionRequest == NULL)) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if ((Type == EFI_IFR_TYPE_STRING) && (Value->string == 0)) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Status = EFI_SUCCESS;\r
  PrivateData = DRIVER_SAMPLE_PRIVATE_FROM_THIS (This);\r
\r
  switch (QuestionId) {\r
  case 0x1234:\r
    //\r
    // Initialize the container for dynamic opcodes\r
    //\r
    StartOpCodeHandle = HiiAllocateOpCodeHandle ();\r
    ASSERT (StartOpCodeHandle != NULL);\r
\r
    EndOpCodeHandle = HiiAllocateOpCodeHandle ();\r
    ASSERT (EndOpCodeHandle != NULL);\r
\r
    //\r
    // Create Hii Extend Label OpCode as the start opcode\r
    //\r
    StartLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (StartOpCodeHandle, &gEfiIfrTianoGuid, NULL, sizeof (EFI_IFR_GUID_LABEL));\r
    StartLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL;\r
    StartLabel->Number       = LABEL_UPDATE1;\r
\r
    //\r
    // Create Hii Extend Label OpCode as the end opcode\r
    //\r
    EndLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (EndOpCodeHandle, &gEfiIfrTianoGuid, NULL, sizeof (EFI_IFR_GUID_LABEL));\r
    EndLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL;\r
    EndLabel->Number       = LABEL_END;\r
\r
    HiiCreateActionOpCode (\r
      StartOpCodeHandle,                // Container for dynamic created opcodes\r
      0x1237,                           // Question ID\r
      STRING_TOKEN(STR_EXIT_TEXT),      // Prompt text\r
      STRING_TOKEN(STR_EXIT_TEXT),      // Help text\r
      EFI_IFR_FLAG_CALLBACK,            // Question flag\r
      0                                 // Action String ID\r
    );\r
\r
    //\r
    // Create Option OpCode\r
    //\r
    OptionsOpCodeHandle = HiiAllocateOpCodeHandle ();\r
    ASSERT (OptionsOpCodeHandle != NULL);\r
\r
    HiiCreateOneOfOptionOpCode (\r
      OptionsOpCodeHandle,\r
      STRING_TOKEN (STR_BOOT_OPTION1),\r
      0,\r
      EFI_IFR_NUMERIC_SIZE_1,\r
      1\r
      );\r
\r
    HiiCreateOneOfOptionOpCode (\r
      OptionsOpCodeHandle,\r
      STRING_TOKEN (STR_BOOT_OPTION2),\r
      0,\r
      EFI_IFR_NUMERIC_SIZE_1,\r
      2\r
      );\r
\r
    //\r
    // Prepare initial value for the dynamic created oneof Question\r
    //\r
    PrivateData->Configuration.DynamicOneof = 2;\r
    Status = gRT->SetVariable(\r
                    VariableName,\r
                    &mFormSetGuid,\r
                    EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS,\r
                    sizeof (DRIVER_SAMPLE_CONFIGURATION),\r
                    &PrivateData->Configuration\r
                    );\r
\r
    //\r
    // Set initial vlaue of dynamic created oneof Question in Form Browser\r
    //\r
    Configuration = AllocateZeroPool (sizeof (DRIVER_SAMPLE_CONFIGURATION));\r
    ASSERT (Configuration != NULL);\r
    if (HiiGetBrowserData (&mFormSetGuid, VariableName, sizeof (DRIVER_SAMPLE_CONFIGURATION), (UINT8 *) Configuration)) {\r
      Configuration->DynamicOneof = 2;\r
\r
      //\r
      // Update uncommitted data of Browser\r
      //\r
      HiiSetBrowserData (\r
        &mFormSetGuid,\r
        VariableName,\r
        sizeof (DRIVER_SAMPLE_CONFIGURATION),\r
        (UINT8 *) Configuration,\r
        NULL\r
        );\r
    }\r
    FreePool (Configuration);\r
\r
    HiiCreateOneOfOpCode (\r
      StartOpCodeHandle,                         // Container for dynamic created opcodes\r
      0x8001,                                    // Question ID (or call it "key")\r
      CONFIGURATION_VARSTORE_ID,                 // VarStore ID\r
      (UINT16) DYNAMIC_ONE_OF_VAR_OFFSET,        // Offset in Buffer Storage\r
      STRING_TOKEN (STR_ONE_OF_PROMPT),          // Question prompt text\r
      STRING_TOKEN (STR_ONE_OF_HELP),            // Question help text\r
      EFI_IFR_FLAG_CALLBACK,                     // Question flag\r
      EFI_IFR_NUMERIC_SIZE_1,                    // Data type of Question Value\r
      OptionsOpCodeHandle,                       // Option Opcode list\r
      NULL                                       // Default Opcode is NULl\r
      );\r
\r
    HiiCreateOrderedListOpCode (\r
      StartOpCodeHandle,                         // Container for dynamic created opcodes\r
      0x8002,                                    // Question ID\r
      CONFIGURATION_VARSTORE_ID,                 // VarStore ID\r
      (UINT16) DYNAMIC_ORDERED_LIST_VAR_OFFSET,  // Offset in Buffer Storage\r
      STRING_TOKEN (STR_BOOT_OPTIONS),           // Question prompt text\r
      STRING_TOKEN (STR_BOOT_OPTIONS),           // Question help text\r
      EFI_IFR_FLAG_RESET_REQUIRED,               // Question flag\r
      0,                                         // Ordered list flag, e.g. EFI_IFR_UNIQUE_SET\r
      EFI_IFR_NUMERIC_SIZE_1,                    // Data type of Question value\r
      5,                                         // Maximum container\r
      OptionsOpCodeHandle,                       // Option Opcode list\r
      NULL                                       // Default Opcode is NULl\r
      );\r
\r
    HiiCreateGotoOpCode (\r
      StartOpCodeHandle,                // Container for dynamic created opcodes\r
      1,                                // Target Form ID\r
      STRING_TOKEN (STR_GOTO_FORM1),    // Prompt text\r
      STRING_TOKEN (STR_GOTO_HELP),     // Help text\r
      0,                                // Question flag\r
      0x8003                            // Question ID\r
      );\r
\r
    HiiUpdateForm (\r
      PrivateData->HiiHandle[0],  // HII handle\r
      &mFormSetGuid,              // Formset GUID\r
      0x1234,                     // Form ID\r
      StartOpCodeHandle,          // Label for where to insert opcodes\r
      EndOpCodeHandle             // Replace data\r
      );\r
\r
    HiiFreeOpCodeHandle (StartOpCodeHandle);\r
    HiiFreeOpCodeHandle (OptionsOpCodeHandle);\r
    HiiFreeOpCodeHandle (EndOpCodeHandle);\r
    break;\r
\r
  case 0x5678:\r
    //\r
    // We will reach here once the Question is refreshed\r
    //\r
\r
    //\r
    // Initialize the container for dynamic opcodes\r
    //\r
    StartOpCodeHandle = HiiAllocateOpCodeHandle ();\r
    ASSERT (StartOpCodeHandle != NULL);\r
\r
    //\r
    // Create Hii Extend Label OpCode as the start opcode\r
    //\r
    StartLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (StartOpCodeHandle, &gEfiIfrTianoGuid, NULL, sizeof (EFI_IFR_GUID_LABEL));\r
    StartLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL;\r
    StartLabel->Number       = LABEL_UPDATE2;\r
\r
    HiiCreateActionOpCode (\r
      StartOpCodeHandle,                // Container for dynamic created opcodes\r
      0x1237,                           // Question ID\r
      STRING_TOKEN(STR_EXIT_TEXT),      // Prompt text\r
      STRING_TOKEN(STR_EXIT_TEXT),      // Help text\r
      EFI_IFR_FLAG_CALLBACK,            // Question flag\r
      0                                 // Action String ID\r
    );\r
\r
    HiiUpdateForm (\r
      PrivateData->HiiHandle[0],  // HII handle\r
      &mFormSetGuid,              // Formset GUID\r
      0x3,                        // Form ID\r
      StartOpCodeHandle,          // Label for where to insert opcodes\r
      NULL                        // Insert data\r
      );\r
\r
    HiiFreeOpCodeHandle (StartOpCodeHandle);\r
\r
    //\r
    // Refresh the Question value\r
    //\r
    PrivateData->Configuration.DynamicRefresh++;\r
    Status = gRT->SetVariable(\r
                    VariableName,\r
                    &mFormSetGuid,\r
                    EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS,\r
                    sizeof (DRIVER_SAMPLE_CONFIGURATION),\r
                    &PrivateData->Configuration\r
                    );\r
\r
    //\r
    // Change an EFI Variable storage (MyEfiVar) asynchronous, this will cause\r
    // the first statement in Form 3 be suppressed\r
    //\r
    MyVarSize = 1;\r
    MyVar = 111;\r
    Status = gRT->SetVariable(\r
                    L"MyVar",\r
                    &mFormSetGuid,\r
                    EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS,\r
                    MyVarSize,\r
                    &MyVar\r
                    );\r
    break;\r
\r
  case 0x1237:\r
    //\r
    // User press "Exit now", request Browser to exit\r
    //\r
    *ActionRequest = EFI_BROWSER_ACTION_REQUEST_EXIT;\r
    break;\r
\r
  case 0x1238:\r
    //\r
    // User press "Save now", request Browser to save the uncommitted data.\r
    //\r
    *ActionRequest = EFI_BROWSER_ACTION_REQUEST_SUBMIT;\r
    break;\r
\r
  case 0x2000:\r
    //\r
    // When try to set a new password, user will be chanlleged with old password.\r
    // The Callback is responsible for validating old password input by user,\r
    // If Callback return EFI_SUCCESS, it indicates validation pass.\r
    //\r
    switch (PrivateData->PasswordState) {\r
    case BROWSER_STATE_VALIDATE_PASSWORD:\r
      Status = ValidatePassword (PrivateData, Value->string);\r
      if (Status == EFI_SUCCESS) {\r
        PrivateData->PasswordState = BROWSER_STATE_SET_PASSWORD;\r
      }\r
      break;\r
\r
    case BROWSER_STATE_SET_PASSWORD:\r
      Status = SetPassword (PrivateData, Value->string);\r
      PrivateData->PasswordState = BROWSER_STATE_VALIDATE_PASSWORD;\r
      break;\r
\r
    default:\r
      break;\r
    }\r
\r
    break;\r
\r
  case 0x1111:\r
    //\r
    // EfiVarstore question takes sample action (print value as debug information) \r
    // after read/write question.\r
    //\r
    MyVarSize = 1;\r
    Status = gRT->GetVariable(\r
                    L"MyVar",\r
                    &mFormSetGuid,\r
                    NULL,\r
                    &MyVarSize,\r
                    &MyVar\r
                    );\r
    ASSERT_EFI_ERROR (Status);\r
    DEBUG ((DEBUG_INFO, "EfiVarstore question: Tall value is %d with value width %d\n", MyVar, MyVarSize));\r
  default:\r
    break;\r
  }\r
\r
  return Status;\r
}\r
\r
/**\r
  Main entry for this driver.\r
\r
  @param ImageHandle     Image handle this driver.\r
  @param SystemTable     Pointer to SystemTable.\r
\r
  @retval EFI_SUCESS     This function always complete successfully.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
DriverSampleInit (\r
  IN EFI_HANDLE                   ImageHandle,\r
  IN EFI_SYSTEM_TABLE             *SystemTable\r
  )\r
{\r
  EFI_STATUS                      Status;\r
  EFI_HII_HANDLE                  HiiHandle[2];\r
  EFI_SCREEN_DESCRIPTOR           Screen;\r
  EFI_HII_DATABASE_PROTOCOL       *HiiDatabase;\r
  EFI_HII_STRING_PROTOCOL         *HiiString;\r
  EFI_FORM_BROWSER2_PROTOCOL      *FormBrowser2;\r
  EFI_HII_CONFIG_ROUTING_PROTOCOL *HiiConfigRouting;\r
  CHAR16                          *NewString;\r
  UINTN                           BufferSize;\r
  DRIVER_SAMPLE_CONFIGURATION     *Configuration;\r
  BOOLEAN                         ActionFlag;\r
  EFI_STRING                      ConfigRequestHdr;\r
\r
  //\r
  // Initialize the local variables.\r
  //\r
  ConfigRequestHdr = NULL;\r
  //\r
  // Initialize screen dimensions for SendForm().\r
  // Remove 3 characters from top and bottom\r
  //\r
  ZeroMem (&Screen, sizeof (EFI_SCREEN_DESCRIPTOR));\r
  gST->ConOut->QueryMode (gST->ConOut, gST->ConOut->Mode->Mode, &Screen.RightColumn, &Screen.BottomRow);\r
\r
  Screen.TopRow     = 3;\r
  Screen.BottomRow  = Screen.BottomRow - 3;\r
\r
  //\r
  // Initialize driver private data\r
  //\r
  PrivateData = AllocateZeroPool (sizeof (DRIVER_SAMPLE_PRIVATE_DATA));\r
  if (PrivateData == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  PrivateData->Signature = DRIVER_SAMPLE_PRIVATE_SIGNATURE;\r
\r
  PrivateData->ConfigAccess.ExtractConfig = ExtractConfig;\r
  PrivateData->ConfigAccess.RouteConfig = RouteConfig;\r
  PrivateData->ConfigAccess.Callback = DriverCallback;\r
  PrivateData->PasswordState = BROWSER_STATE_VALIDATE_PASSWORD;\r
\r
  //\r
  // Locate Hii Database protocol\r
  //\r
  Status = gBS->LocateProtocol (&gEfiHiiDatabaseProtocolGuid, NULL, (VOID **) &HiiDatabase);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
  PrivateData->HiiDatabase = HiiDatabase;\r
\r
  //\r
  // Locate HiiString protocol\r
  //\r
  Status = gBS->LocateProtocol (&gEfiHiiStringProtocolGuid, NULL, (VOID **) &HiiString);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
  PrivateData->HiiString = HiiString;\r
\r
  //\r
  // Locate Formbrowser2 protocol\r
  //\r
  Status = gBS->LocateProtocol (&gEfiFormBrowser2ProtocolGuid, NULL, (VOID **) &FormBrowser2);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
  PrivateData->FormBrowser2 = FormBrowser2;\r
\r
  //\r
  // Locate ConfigRouting protocol\r
  //\r
  Status = gBS->LocateProtocol (&gEfiHiiConfigRoutingProtocolGuid, NULL, (VOID **) &HiiConfigRouting);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
  PrivateData->HiiConfigRouting = HiiConfigRouting;\r
\r
  Status = gBS->InstallMultipleProtocolInterfaces (\r
                  &DriverHandle[0],\r
                  &gEfiDevicePathProtocolGuid,\r
                  &mHiiVendorDevicePath0,\r
                  &gEfiHiiConfigAccessProtocolGuid,\r
                  &PrivateData->ConfigAccess,\r
                  NULL\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  PrivateData->DriverHandle[0] = DriverHandle[0];\r
\r
  //\r
  // Publish our HII data\r
  //\r
  HiiHandle[0] = HiiAddPackages (\r
                   &mFormSetGuid,\r
                   DriverHandle[0],\r
                   DriverSampleStrings,\r
                   VfrBin,\r
                   NULL\r
                   );\r
  if (HiiHandle[0] == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  PrivateData->HiiHandle[0] = HiiHandle[0];\r
\r
  //\r
  // Publish another Fromset\r
  //\r
  Status = gBS->InstallMultipleProtocolInterfaces (\r
                  &DriverHandle[1],\r
                  &gEfiDevicePathProtocolGuid,\r
                  &mHiiVendorDevicePath1,\r
                  NULL\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  PrivateData->DriverHandle[1] = DriverHandle[1];\r
\r
  HiiHandle[1] = HiiAddPackages (\r
                   &mInventoryGuid,\r
                   DriverHandle[1],\r
                   DriverSampleStrings,\r
                   InventoryBin,\r
                   NULL\r
                   );\r
  if (HiiHandle[1] == NULL) {\r
    DriverSampleUnload (ImageHandle);\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  PrivateData->HiiHandle[1] = HiiHandle[1];\r
\r
  //\r
  // Very simple example of how one would update a string that is already\r
  // in the HII database\r
  //\r
  NewString = L"700 Mhz";\r
\r
  if (HiiSetString (HiiHandle[0], STRING_TOKEN (STR_CPU_STRING2), NewString, NULL) == 0) {\r
    DriverSampleUnload (ImageHandle);\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  HiiSetString (HiiHandle[0], 0, NewString, NULL);\r
\r
  //\r
  // Initialize Name/Value name String ID\r
  //\r
  PrivateData->NameStringId[0] = STR_NAME_VALUE_VAR_NAME0;\r
  PrivateData->NameStringId[1] = STR_NAME_VALUE_VAR_NAME1;\r
  PrivateData->NameStringId[2] = STR_NAME_VALUE_VAR_NAME2;\r
\r
  //\r
  // Initialize configuration data\r
  //\r
  Configuration = &PrivateData->Configuration;\r
  ZeroMem (Configuration, sizeof (DRIVER_SAMPLE_CONFIGURATION));\r
\r
  //\r
  // Try to read NV config EFI variable first\r
  //\r
  ConfigRequestHdr = HiiConstructConfigHdr (&mFormSetGuid, VariableName, DriverHandle[0]);\r
  ASSERT (ConfigRequestHdr != NULL);\r
\r
  BufferSize = sizeof (DRIVER_SAMPLE_CONFIGURATION);\r
  Status = gRT->GetVariable (VariableName, &mFormSetGuid, NULL, &BufferSize, Configuration);\r
  if (EFI_ERROR (Status)) {\r
    //\r
    // Store zero data Buffer Storage to EFI variable\r
    //\r
    Status = gRT->SetVariable(\r
                    VariableName,\r
                    &mFormSetGuid,\r
                    EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS,\r
                    sizeof (DRIVER_SAMPLE_CONFIGURATION),\r
                    Configuration\r
                    );\r
    ASSERT (Status == EFI_SUCCESS);\r
    //\r
    // EFI variable for NV config doesn't exit, we should build this variable\r
    // based on default values stored in IFR\r
    //\r
    ActionFlag = HiiSetToDefaults (ConfigRequestHdr, EFI_HII_DEFAULT_CLASS_STANDARD);\r
    ASSERT (ActionFlag);\r
  } else {\r
    //\r
    // EFI variable does exist and Validate Current Setting\r
    //\r
    ActionFlag = HiiValidateSettings (ConfigRequestHdr);\r
    ASSERT (ActionFlag);\r
  }\r
\r
  FreePool (ConfigRequestHdr);\r
\r
\r
  //\r
  // In default, this driver is built into Flash device image,\r
  // the following code doesn't run.\r
  //\r
\r
  //\r
  // Example of how to display only the item we sent to HII\r
  // When this driver is not built into Flash device image,\r
  // it need to call SendForm to show front page by itself.\r
  //\r
  if (DISPLAY_ONLY_MY_ITEM <= 1) {\r
    //\r
    // Have the browser pull out our copy of the data, and only display our data\r
    //\r
    Status = FormBrowser2->SendForm (\r
                             FormBrowser2,\r
                             &(HiiHandle[DISPLAY_ONLY_MY_ITEM]),\r
                             1,\r
                             NULL,\r
                             0,\r
                             NULL,\r
                             NULL\r
                             );\r
\r
    HiiRemovePackages (HiiHandle[0]);\r
\r
    HiiRemovePackages (HiiHandle[1]);\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Unloads the application and its installed protocol.\r
\r
  @param[in]  ImageHandle       Handle that identifies the image to be unloaded.\r
\r
  @retval EFI_SUCCESS           The image has been unloaded.\r
**/\r
EFI_STATUS\r
EFIAPI\r
DriverSampleUnload (\r
  IN EFI_HANDLE  ImageHandle\r
  )\r
{\r
  UINTN Index;\r
  if (DriverHandle[0] != NULL) {\r
    gBS->UninstallMultipleProtocolInterfaces (\r
            DriverHandle[0],\r
            &gEfiDevicePathProtocolGuid,\r
            &mHiiVendorDevicePath0,\r
            &gEfiHiiConfigAccessProtocolGuid,\r
            &PrivateData->ConfigAccess,\r
            NULL\r
           );\r
    DriverHandle[0] = NULL;\r
  }\r
\r
  if (DriverHandle[1] != NULL) {\r
    gBS->UninstallMultipleProtocolInterfaces (\r
            DriverHandle[1],\r
            &gEfiDevicePathProtocolGuid,\r
            &mHiiVendorDevicePath1,\r
            NULL\r
           );\r
    DriverHandle[1] = NULL;\r
  }\r
\r
  if (PrivateData->HiiHandle[0] != NULL) {\r
    HiiRemovePackages (PrivateData->HiiHandle[0]);\r
  }\r
\r
  if (PrivateData->HiiHandle[1] != NULL) {\r
    HiiRemovePackages (PrivateData->HiiHandle[1]);\r
  }\r
\r
  if (PrivateData != NULL) {\r
    for (Index = 0; Index < NAME_VALUE_NAME_NUMBER; Index++) {\r
      if (PrivateData->NameValueName[Index] != NULL) {\r
        FreePool (PrivateData->NameValueName[Index]);\r
      }\r
    }\r
    FreePool (PrivateData);\r
    PrivateData = NULL;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r