[mirror_edk2.git] / NetworkPkg / WifiConnectionManagerDxe / WifiConnectionMgrHiiConfigAccess.c

/** @file\r
  The Hii functions for WiFi Connection Manager.\r
\r
  Copyright (c) 2019, Intel Corporation. All rights reserved.<BR>\r
\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
\r
**/\r
\r
#include "WifiConnectionMgrDxe.h"\r
\r
CHAR16  mVendorStorageName[] = L"WIFI_MANAGER_IFR_NVDATA";\r
\r
HII_VENDOR_DEVICE_PATH  mWifiMgrDxeHiiVendorDevicePath = {\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
    WIFI_CONNECTION_MANAGER_CONFIG_GUID\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
// HII Config Access Protocol instance\r
//\r
GLOBAL_REMOVE_IF_UNREFERENCED\r
EFI_HII_CONFIG_ACCESS_PROTOCOL gWifiMgrDxeHiiConfigAccess = {\r
  WifiMgrDxeHiiConfigAccessExtractConfig,\r
  WifiMgrDxeHiiConfigAccessRouteConfig,\r
  WifiMgrDxeHiiConfigAccessCallback\r
};\r
\r
CHAR16*   mSecurityType[] = {\r
  L"OPEN           ",\r
  L"WPA-Enterprise ",\r
  L"WPA2-Enterprise",\r
  L"WPA-Personal   ",\r
  L"WPA2-Personal  ",\r
  L"WEP            ",\r
  L"UnKnown        "\r
};\r
\r
CHAR16*  mSignalStrengthBar[] = {\r
  L"[-----]",\r
  L"[*----]",\r
  L"[**---]",\r
  L"[***--]",\r
  L"[****-]",\r
  L"[*****]"\r
};\r
\r
#define  RSSI_TO_SIGNAL_STRENGTH_BAR(Rssi)  mSignalStrengthBar[((Rssi + 19)/20)]\r
\r
#define  NET_LIST_FOR_EACH_FROM_NODE(Entry, Node, ListHead) \\r
  for(Entry = Node->ForwardLink; Entry != (ListHead); Entry = Entry->ForwardLink)\r
\r
extern EFI_GUID    gWifiConfigFormSetGuid;\r
\r
/**\r
  Create Hii Extend Label OpCode as the start opcode and end opcode.\r
  The caller is responsible for freeing the OpCode with HiiFreeOpCodeHandle().\r
\r
  @param[in]  StartLabelNumber   The number of start label.\r
  @param[out] StartOpCodeHandle  Points to the start opcode handle.\r
  @param[out] EndOpCodeHandle    Points to the end opcode handle.\r
\r
  @retval EFI_OUT_OF_RESOURCES   Do not have sufficient resource to finish this\r
                                 operation.\r
  @retval EFI_INVALID_PARAMETER  Any input parameter is invalid.\r
  @retval EFI_SUCCESS            The operation is completed successfully.\r
  @retval Other Errors           Returned errors when updating the HII form.\r
\r
**/\r
EFI_STATUS\r
WifiMgrCreateOpCode (\r
  IN  UINT16    StartLabelNumber,\r
  OUT VOID      **StartOpCodeHandle,\r
  OUT VOID      **EndOpCodeHandle\r
  )\r
{\r
  EFI_STATUS            Status;\r
  EFI_IFR_GUID_LABEL    *InternalStartLabel;\r
  EFI_IFR_GUID_LABEL    *InternalEndLabel;\r
\r
  if (StartOpCodeHandle == NULL || EndOpCodeHandle == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Status             = EFI_OUT_OF_RESOURCES;\r
  *StartOpCodeHandle = NULL;\r
  *EndOpCodeHandle   = NULL;\r
\r
  //\r
  // Initialize the container for dynamic opcodes.\r
  //\r
  *StartOpCodeHandle = HiiAllocateOpCodeHandle ();\r
  if (*StartOpCodeHandle == NULL) {\r
    goto Exit;\r
  }\r
  *EndOpCodeHandle = HiiAllocateOpCodeHandle ();\r
  if (*EndOpCodeHandle == NULL) {\r
    goto Exit;\r
  }\r
\r
  //\r
  // Create Hii Extend Label OpCode as the start opcode.\r
  //\r
  InternalStartLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (\r
                                                *StartOpCodeHandle,\r
                                                &gEfiIfrTianoGuid,\r
                                                NULL,\r
                                                sizeof (EFI_IFR_GUID_LABEL)\r
                                                );\r
  if (InternalStartLabel == NULL) {\r
    goto Exit;\r
  }\r
  InternalStartLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL;\r
  InternalStartLabel->Number       = StartLabelNumber;\r
\r
  //\r
  // Create Hii Extend Label OpCode as the end opcode.\r
  //\r
  InternalEndLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (\r
                                              *EndOpCodeHandle,\r
                                              &gEfiIfrTianoGuid,\r
                                              NULL,\r
                                              sizeof (EFI_IFR_GUID_LABEL)\r
                                              );\r
  if (InternalEndLabel == NULL) {\r
    goto Exit;\r
  }\r
  InternalEndLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL;\r
  InternalEndLabel->Number       = LABEL_END;\r
\r
  return EFI_SUCCESS;\r
\r
Exit:\r
\r
  if (*StartOpCodeHandle != NULL) {\r
    HiiFreeOpCodeHandle (*StartOpCodeHandle);\r
  }\r
  if (*EndOpCodeHandle != NULL) {\r
    HiiFreeOpCodeHandle (*EndOpCodeHandle);\r
  }\r
  return Status;\r
}\r
\r
/**\r
  Display the Nic list contains all available Nics.\r
\r
  @param[in]  Private            The pointer to the global private data structure.\r
\r
  @retval EFI_INVALID_PARAMETER  Any input parameter is invalid.\r
  @retval EFI_SUCCESS            The operation is completed successfully.\r
\r
**/\r
EFI_STATUS\r
WifiMgrShowNicList (\r
  IN  WIFI_MGR_PRIVATE_DATA    *Private\r
)\r
{\r
  EFI_STATUS              Status;\r
  CHAR16                  MacString[WIFI_MGR_MAX_MAC_STRING_LEN];\r
  CHAR16                  PortString[WIFI_STR_MAX_SIZE];\r
  EFI_STRING_ID           PortTitleToken;\r
  EFI_STRING_ID           PortTitleHelpToken;\r
  WIFI_MGR_DEVICE_DATA    *Nic;\r
  LIST_ENTRY              *Entry;\r
  VOID                    *StartOpCodeHandle;\r
  VOID                    *EndOpCodeHandle;\r
\r
  if (Private == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Status = WifiMgrCreateOpCode (\r
             LABEL_MAC_ENTRY,\r
             &StartOpCodeHandle,\r
             &EndOpCodeHandle\r
             );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  NET_LIST_FOR_EACH (Entry, &Private->NicList) {\r
    Nic = NET_LIST_USER_STRUCT_S (Entry, WIFI_MGR_DEVICE_DATA, Link, WIFI_MGR_DEVICE_DATA_SIGNATURE);\r
    WifiMgrMacAddrToStr (&Nic->MacAddress, sizeof (MacString), MacString);\r
    UnicodeSPrint (PortString, sizeof (PortString), L"MAC %s", MacString);\r
    PortTitleToken = HiiSetString (\r
                       Private->RegisteredHandle,\r
                       0,\r
                       PortString,\r
                       NULL\r
                       );\r
    if (PortTitleToken == 0) {\r
      Status = EFI_INVALID_PARAMETER;\r
      goto Exit;\r
    }\r
\r
    UnicodeSPrint (PortString, sizeof (PortString), L"MAC Address");\r
    PortTitleHelpToken = HiiSetString (\r
                           Private->RegisteredHandle,\r
                           0,\r
                           PortString,\r
                           NULL\r
                           );\r
    if (PortTitleHelpToken == 0) {\r
      Status = EFI_INVALID_PARAMETER;\r
      goto Exit;\r
    }\r
\r
    HiiCreateGotoOpCode (\r
      StartOpCodeHandle,\r
      FORMID_WIFI_MAINPAGE,\r
      PortTitleToken,\r
      PortTitleHelpToken,\r
      EFI_IFR_FLAG_CALLBACK,\r
      (UINT16) (KEY_MAC_ENTRY_BASE + Nic->NicIndex)\r
      );\r
  }\r
\r
  Status = HiiUpdateForm (\r
             Private->RegisteredHandle,       // HII handle\r
             &gWifiConfigFormSetGuid,         // Formset GUID\r
             FORMID_MAC_SELECTION,            // Form ID\r
             StartOpCodeHandle,               // Label for where to insert opcodes\r
             EndOpCodeHandle                  // Replace data\r
             );\r
\r
Exit:\r
\r
  HiiFreeOpCodeHandle (StartOpCodeHandle);\r
  HiiFreeOpCodeHandle (EndOpCodeHandle);\r
  return Status;\r
}\r
\r
/**\r
  Retreive the unicode string of the AKM Suite list of a profile.\r
  The caller is responsible for freeing the string with FreePool().\r
\r
  @param[in]  Profile           The network profile contains a AKM suite list.\r
\r
  @return the unicode string of AKM suite list or "None".\r
\r
**/\r
CHAR16*\r
WifiMgrGetStrAKMList (\r
  IN  WIFI_MGR_NETWORK_PROFILE         *Profile\r
)\r
{\r
  UINT8     Index;\r
  UINT16    AKMSuiteCount;\r
  CHAR16    *AKMListDisplay;\r
\r
  AKMListDisplay = NULL;\r
  if (Profile == NULL || Profile->Network.AKMSuite == NULL) {\r
    goto Exit;\r
  }\r
\r
  AKMSuiteCount = Profile->Network.AKMSuite->AKMSuiteCount;\r
  if (AKMSuiteCount != 0) {\r
\r
    //\r
    // Current AKM Suite is between 1-9\r
    //\r
    AKMListDisplay = (CHAR16 *) AllocateZeroPool(sizeof (CHAR16) * AKMSuiteCount * 2);\r
    if (AKMListDisplay != NULL) {\r
      for (Index = 0; Index < AKMSuiteCount; Index ++) {\r
        UnicodeSPrint (\r
          AKMListDisplay + (Index * 2),\r
          sizeof (CHAR16) * 2,\r
          L"%d ",\r
          Profile->Network.AKMSuite->AKMSuiteList[Index].SuiteType\r
          );\r
        if (Index == AKMSuiteCount - 1) {\r
          *(AKMListDisplay + (Index * 2 + 1)) = L'\0';\r
        }\r
      }\r
    }\r
  }\r
\r
Exit:\r
\r
  if (AKMListDisplay == NULL) {\r
    AKMListDisplay = AllocateCopyPool (sizeof (L"None"), L"None");\r
  }\r
  return AKMListDisplay;\r
}\r
\r
/**\r
  Retreive the unicode string of the Cipher Suite list of a profile.\r
  The caller is responsible for freeing the string with FreePool().\r
\r
  @param[in]  Profile           The network profile contains a Cipher suite list.\r
\r
  @return the unicode string of Cipher suite list or "None".\r
\r
**/\r
CHAR16*\r
WifiMgrGetStrCipherList (\r
  IN  WIFI_MGR_NETWORK_PROFILE          *Profile\r
)\r
{\r
  UINT8     Index;\r
  UINT16    CipherSuiteCount;\r
  CHAR16    *CipherListDisplay;\r
\r
  CipherListDisplay = NULL;\r
  if (Profile == NULL || Profile->Network.CipherSuite == NULL) {\r
    goto Exit;\r
  }\r
\r
  CipherSuiteCount   = Profile->Network.CipherSuite->CipherSuiteCount;\r
  if (CipherSuiteCount != 0) {\r
\r
    //\r
    // Current Cipher Suite is between 1-9\r
    //\r
    CipherListDisplay = (CHAR16 *) AllocateZeroPool(sizeof (CHAR16) * CipherSuiteCount * 2);\r
    if (CipherListDisplay != NULL) {\r
      for (Index = 0; Index < CipherSuiteCount; Index ++) {\r
        UnicodeSPrint (\r
          CipherListDisplay + (Index * 2),\r
          sizeof (CHAR16) * 2,\r
          L"%d ",\r
          Profile->Network.CipherSuite->CipherSuiteList[Index].SuiteType\r
          );\r
        if (Index == CipherSuiteCount - 1) {\r
          *(CipherListDisplay + (Index * 2 + 1)) = L'\0';\r
        }\r
      }\r
    }\r
  }\r
\r
Exit:\r
\r
  if (CipherListDisplay == NULL) {\r
    CipherListDisplay = AllocateCopyPool (sizeof (L"None"), L"None");\r
  }\r
  return CipherListDisplay;\r
}\r
\r
/**\r
  Refresh the network list display of the current Nic.\r
\r
  @param[in]   Private           The pointer to the global private data structure.\r
  @param[out]  IfrNvData         The IFR NV data.\r
\r
  @retval EFI_SUCCESS            The operation is completed successfully.\r
  @retval EFI_OUT_OF_RESOURCES   Failed to allocate memory.\r
  @retval Other Errors           Returned errors when creating Opcodes or updating the\r
                                 Hii form.\r
\r
**/\r
EFI_STATUS\r
WifiMgrRefreshNetworkList (\r
  IN    WIFI_MGR_PRIVATE_DATA      *Private,\r
  OUT   WIFI_MANAGER_IFR_NVDATA    *IfrNvData\r
  )\r
{\r
  EFI_STATUS                         Status;\r
  EFI_TPL                            OldTpl;\r
  UINT32                             AvailableCount;\r
  EFI_STRING_ID                      PortPromptToken;\r
  EFI_STRING_ID                      PortTextToken;\r
  EFI_STRING_ID                      PortHelpToken;\r
  WIFI_MGR_NETWORK_PROFILE           *Profile;\r
  LIST_ENTRY                         *Entry;\r
  VOID                               *StartOpCodeHandle;\r
  VOID                               *EndOpCodeHandle;\r
  CHAR16                             *AKMListDisplay;\r
  CHAR16                             *CipherListDisplay;\r
  CHAR16                             PortString[WIFI_STR_MAX_SIZE];\r
  UINTN                              PortStringSize;\r
  WIFI_MGR_NETWORK_PROFILE           *ConnectedProfile;\r
\r
  if (Private->CurrentNic == NULL) {\r
    return EFI_SUCCESS;\r
  }\r
\r
  Status = WifiMgrCreateOpCode (\r
             LABEL_NETWORK_LIST_ENTRY,\r
             &StartOpCodeHandle,\r
             &EndOpCodeHandle\r
             );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  OldTpl = gBS->RaiseTPL (TPL_CALLBACK);\r
  AvailableCount    = 0;\r
  PortStringSize    = sizeof (PortString);\r
  ConnectedProfile  = NULL;\r
  AKMListDisplay    = NULL;\r
  CipherListDisplay = NULL;\r
\r
  if (Private->CurrentNic->ConnectState == WifiMgrConnectedToAp) {\r
\r
    //\r
    // Display the current connected network.\r
    // Find the current operate network under connected status.\r
    //\r
    if (Private->CurrentNic->CurrentOperateNetwork != NULL &&\r
      Private->CurrentNic->CurrentOperateNetwork->IsAvailable) {\r
\r
      Profile = Private->CurrentNic->CurrentOperateNetwork;\r
      AvailableCount ++;\r
\r
      AKMListDisplay = WifiMgrGetStrAKMList (Profile);\r
      if (AKMListDisplay == NULL) {\r
        Status = EFI_OUT_OF_RESOURCES;\r
        goto Exit;\r
      }\r
      CipherListDisplay = WifiMgrGetStrCipherList(Profile);\r
      if (CipherListDisplay == NULL) {\r
        Status = EFI_OUT_OF_RESOURCES;\r
        goto Exit;\r
      }\r
\r
      UnicodeSPrint (PortString, PortStringSize, L"%s (Connected)", Profile->SSId);\r
      PortPromptToken = HiiSetString (Private->RegisteredHandle, 0, PortString, NULL);\r
\r
      if (Profile->SecurityType == SECURITY_TYPE_NONE) {\r
        PortHelpToken = 0;\r
      } else {\r
        UnicodeSPrint (PortString, PortStringSize, L"AKMSuite: %s CipherSuite: %s", AKMListDisplay, CipherListDisplay);\r
        PortHelpToken = HiiSetString (Private->RegisteredHandle, 0, PortString, NULL);\r
      }\r
      FreePool (AKMListDisplay);\r
      FreePool (CipherListDisplay);\r
      AKMListDisplay    = NULL;\r
      CipherListDisplay = NULL;\r
\r
      HiiCreateGotoOpCode (\r
        StartOpCodeHandle,\r
        FORMID_CONNECT_NETWORK,\r
        PortPromptToken,\r
        PortHelpToken,\r
        EFI_IFR_FLAG_CALLBACK,\r
        (UINT16) (KEY_AVAILABLE_NETWORK_ENTRY_BASE + Profile->ProfileIndex)\r
        );\r
\r
      UnicodeSPrint (\r
        PortString,\r
        PortStringSize,\r
        L"%s       %s %s",\r
        (Profile->SecurityType != SECURITY_TYPE_NONE ? L"Secured" : L"Open   "),\r
        mSecurityType[Profile->SecurityType],\r
        RSSI_TO_SIGNAL_STRENGTH_BAR(Profile->NetworkQuality)\r
        );\r
      PortTextToken = HiiSetString (Private->RegisteredHandle, 0, PortString, NULL);\r
\r
      HiiCreateTextOpCode (\r
        StartOpCodeHandle,\r
        PortTextToken,\r
        0,\r
        0\r
      );\r
\r
      ConnectedProfile = Profile;\r
    } else {\r
      Private->CurrentNic->HasDisconnectPendingNetwork = TRUE;\r
    }\r
  }\r
\r
  //\r
  // Display all supported available networks.\r
  //\r
  NET_LIST_FOR_EACH (Entry, &Private->CurrentNic->ProfileList) {\r
\r
    Profile = NET_LIST_USER_STRUCT_S (\r
                Entry,\r
                WIFI_MGR_NETWORK_PROFILE,\r
                Link,\r
                WIFI_MGR_PROFILE_SIGNATURE\r
                );\r
    if (ConnectedProfile == Profile) {\r
      continue;\r
    }\r
    if (Profile->IsAvailable && Profile->CipherSuiteSupported) {\r
\r
      AvailableCount ++;\r
\r
      AKMListDisplay = WifiMgrGetStrAKMList (Profile);\r
      if (AKMListDisplay == NULL) {\r
        Status = EFI_OUT_OF_RESOURCES;\r
        goto Exit;\r
      }\r
      CipherListDisplay = WifiMgrGetStrCipherList(Profile);\r
      if (CipherListDisplay == NULL) {\r
        Status = EFI_OUT_OF_RESOURCES;\r
        goto Exit;\r
      }\r
\r
      PortPromptToken = HiiSetString (Private->RegisteredHandle, 0, Profile->SSId, NULL);\r
      if (PortPromptToken == 0) {\r
        Status = EFI_OUT_OF_RESOURCES;\r
        goto Exit;\r
      }\r
\r
      if (Profile->SecurityType == SECURITY_TYPE_NONE) {\r
        PortHelpToken = 0;\r
      } else {\r
        UnicodeSPrint (\r
          PortString,\r
          PortStringSize,\r
          L"AKMSuite: %s CipherSuite: %s",\r
          AKMListDisplay, CipherListDisplay\r
          );\r
        PortHelpToken = HiiSetString (Private->RegisteredHandle, 0, PortString, NULL);\r
        if (PortHelpToken == 0) {\r
          Status = EFI_OUT_OF_RESOURCES;\r
          goto Exit;\r
        }\r
      }\r
      FreePool (AKMListDisplay);\r
      FreePool (CipherListDisplay);\r
      AKMListDisplay    = NULL;\r
      CipherListDisplay = NULL;\r
\r
      HiiCreateGotoOpCode (\r
        StartOpCodeHandle,\r
        FORMID_CONNECT_NETWORK,\r
        PortPromptToken,\r
        PortHelpToken,\r
        EFI_IFR_FLAG_CALLBACK,\r
        (UINT16) (KEY_AVAILABLE_NETWORK_ENTRY_BASE + Profile->ProfileIndex)\r
        );\r
\r
      UnicodeSPrint (\r
        PortString,\r
        PortStringSize,\r
        L"%s       %s %s",\r
        (Profile->SecurityType != SECURITY_TYPE_NONE ? L"Secured" : L"Open   "),\r
        mSecurityType[Profile->SecurityType],\r
        RSSI_TO_SIGNAL_STRENGTH_BAR(Profile->NetworkQuality)\r
        );\r
      PortTextToken = HiiSetString (Private->RegisteredHandle, 0, PortString, NULL);\r
      if (PortTextToken == 0) {\r
        Status = EFI_OUT_OF_RESOURCES;\r
        goto Exit;\r
      }\r
      HiiCreateTextOpCode (\r
        StartOpCodeHandle,\r
        PortTextToken,\r
        0,\r
        0\r
        );\r
    }\r
  }\r
\r
  //\r
  // Display all Unsupported available networks.\r
  //\r
  NET_LIST_FOR_EACH (Entry, &Private->CurrentNic->ProfileList) {\r
\r
    Profile = NET_LIST_USER_STRUCT_S (\r
                Entry,\r
                WIFI_MGR_NETWORK_PROFILE,\r
                Link,\r
                WIFI_MGR_PROFILE_SIGNATURE\r
                );\r
    if (ConnectedProfile == Profile) {\r
      continue;\r
    }\r
    if (Profile->IsAvailable && !Profile->CipherSuiteSupported) {\r
\r
      AvailableCount ++;\r
\r
      AKMListDisplay = WifiMgrGetStrAKMList (Profile);\r
      if (AKMListDisplay == NULL) {\r
        Status = EFI_OUT_OF_RESOURCES;\r
        goto Exit;\r
      }\r
      CipherListDisplay = WifiMgrGetStrCipherList(Profile);\r
      if (CipherListDisplay == NULL) {\r
        Status = EFI_OUT_OF_RESOURCES;\r
        goto Exit;\r
      }\r
\r
      PortPromptToken = HiiSetString (Private->RegisteredHandle, 0, Profile->SSId, NULL);\r
\r
      if (Profile->AKMSuiteSupported) {\r
        UnicodeSPrint (\r
          PortString,\r
          PortStringSize,\r
          L"AKMSuite: %s CipherSuite(UnSupported): %s",\r
          AKMListDisplay, CipherListDisplay\r
          );\r
      } else {\r
        UnicodeSPrint (\r
          PortString,\r
          PortStringSize,\r
          L"AKMSuite(UnSupported): %s CipherSuite(UnSupported): %s",\r
          AKMListDisplay, CipherListDisplay\r
          );\r
      }\r
      FreePool (AKMListDisplay);\r
      FreePool (CipherListDisplay);\r
      AKMListDisplay    = NULL;\r
      CipherListDisplay = NULL;\r
\r
      PortHelpToken = HiiSetString (Private->RegisteredHandle, 0, PortString, NULL);\r
\r
      HiiCreateGotoOpCode (\r
        StartOpCodeHandle,\r
        FORMID_CONNECT_NETWORK,\r
        PortPromptToken,\r
        PortHelpToken,\r
        EFI_IFR_FLAG_CALLBACK,\r
        (UINT16) (KEY_AVAILABLE_NETWORK_ENTRY_BASE + Profile->ProfileIndex)\r
        );\r
\r
      UnicodeSPrint (\r
        PortString,\r
        PortStringSize,\r
        L"%s       %s %s",\r
        L"UnSupported",\r
        mSecurityType[Profile->SecurityType],\r
        RSSI_TO_SIGNAL_STRENGTH_BAR(Profile->NetworkQuality)\r
        );\r
      PortTextToken = HiiSetString (Private->RegisteredHandle, 0, PortString, NULL);\r
\r
      HiiCreateTextOpCode (\r
        StartOpCodeHandle,\r
        PortTextToken,\r
        0,\r
        0\r
        );\r
    }\r
  }\r
\r
  Status = HiiUpdateForm (\r
             Private->RegisteredHandle,       // HII handle\r
             &gWifiConfigFormSetGuid,         // Formset GUID\r
             FORMID_NETWORK_LIST,             // Form ID\r
             StartOpCodeHandle,               // Label for where to insert opcodes\r
             EndOpCodeHandle                  // Replace data\r
             );\r
\r
Exit:\r
\r
  gBS->RestoreTPL (OldTpl);\r
\r
  if (AKMListDisplay != NULL) {\r
    FreePool (AKMListDisplay);\r
  }\r
  if (CipherListDisplay != NULL) {\r
    FreePool (CipherListDisplay);\r
  }\r
\r
  HiiFreeOpCodeHandle (StartOpCodeHandle);\r
  HiiFreeOpCodeHandle (EndOpCodeHandle);\r
\r
  DEBUG ((DEBUG_INFO, "[WiFi Connection Manager] Network List is Refreshed!\n"));\r
  return Status;\r
}\r
\r
/**\r
  Refresh the hidden network list configured by user.\r
\r
  @param[in]   Private           The pointer to the global private data structure.\r
\r
  @retval EFI_SUCCESS            The operation is completed successfully.\r
  @retval Other Errors           Returned errors when creating Opcodes or updating the\r
                                 Hii form.\r
**/\r
EFI_STATUS\r
WifiMgrRefreshHiddenList (\r
  IN    WIFI_MGR_PRIVATE_DATA      *Private\r
  )\r
{\r
  EFI_STATUS                       Status;\r
  EFI_TPL                          OldTpl;\r
  UINTN                            Index;\r
  EFI_STRING_ID                    StringId;\r
  VOID                             *StartOpCodeHandle;\r
  VOID                             *EndOpCodeHandle;\r
  WIFI_HIDDEN_NETWORK_DATA         *HiddenNetwork;\r
  LIST_ENTRY                       *Entry;\r
\r
  if (Private == NULL) {\r
    return EFI_SUCCESS;\r
  }\r
\r
  Status = WifiMgrCreateOpCode (\r
             LABEL_HIDDEN_NETWORK_ENTRY,\r
             &StartOpCodeHandle,\r
             &EndOpCodeHandle\r
             );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  OldTpl = gBS->RaiseTPL (TPL_CALLBACK);\r
  Index  = 0;\r
\r
  NET_LIST_FOR_EACH (Entry, &Private->HiddenNetworkList) {\r
\r
    HiddenNetwork = NET_LIST_USER_STRUCT_S (\r
                      Entry,\r
                      WIFI_HIDDEN_NETWORK_DATA,\r
                      Link,\r
                      WIFI_MGR_HIDDEN_NETWORK_SIGNATURE\r
                      );\r
    StringId = HiiSetString (Private->RegisteredHandle, 0, HiddenNetwork->SSId, NULL);\r
\r
    HiiCreateCheckBoxOpCode (\r
      StartOpCodeHandle,\r
      (EFI_QUESTION_ID) (KEY_HIDDEN_NETWORK_ENTRY_BASE + Index),\r
      MANAGER_VARSTORE_ID,\r
      (UINT16) (HIDDEN_NETWORK_LIST_VAR_OFFSET + Index),\r
      StringId,\r
      0,\r
      0,\r
      0,\r
      NULL\r
      );\r
    Index ++;\r
  }\r
\r
  Status = HiiUpdateForm (\r
             Private->RegisteredHandle,       // HII handle\r
             &gWifiConfigFormSetGuid,         // Formset GUID\r
             FORMID_HIDDEN_NETWORK_LIST,      // Form ID\r
             StartOpCodeHandle,               // Label for where to insert opcodes\r
             EndOpCodeHandle                  // Replace data\r
             );\r
\r
  gBS->RestoreTPL (OldTpl);\r
  HiiFreeOpCodeHandle (StartOpCodeHandle);\r
  HiiFreeOpCodeHandle (EndOpCodeHandle);\r
  return Status;\r
}\r
\r
\r
/**\r
  Callback function for user to select a Nic.\r
\r
  @param[in]  Private            The pointer to the global private data structure.\r
  @param[in]  KeyValue           The key value received from HII input.\r
\r
  @retval EFI_NOT_FOUND          The corresponding Nic is not found.\r
  @retval EFI_SUCCESS            The operation is completed successfully.\r
\r
**/\r
EFI_STATUS\r
WifiMgrSelectNic (\r
  IN     WIFI_MGR_PRIVATE_DATA         *Private,\r
  IN     EFI_QUESTION_ID               KeyValue\r
  )\r
{\r
  WIFI_MGR_DEVICE_DATA    *Nic;\r
  UINT32                  NicIndex;\r
  CHAR16                  MacString[WIFI_MGR_MAX_MAC_STRING_LEN];\r
\r
  NicIndex = KeyValue - KEY_MAC_ENTRY_BASE;\r
  Nic      = WifiMgrGetNicByIndex (Private, NicIndex);\r
  if (Nic == NULL) {\r
    return EFI_NOT_FOUND;\r
  }\r
  Private->CurrentNic = Nic;\r
\r
  WifiMgrMacAddrToStr (&Nic->MacAddress, sizeof (MacString), MacString);\r
  HiiSetString (Private->RegisteredHandle, STRING_TOKEN(STR_MAC_ADDRESS), MacString, NULL);\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Restore the NV data to be default.\r
\r
  @param[in]  Private             The pointer to the global private data structure.\r
  @param[out] IfrNvData           The IFR NV data.\r
\r
**/\r
VOID\r
WifiMgrCleanUserInput (\r
  IN  WIFI_MGR_PRIVATE_DATA      *Private\r
  )\r
{\r
  Private->SecurityType        = SECURITY_TYPE_NONE;\r
  Private->EapAuthMethod       = EAP_AUTH_METHOD_TTLS;\r
  Private->EapSecondAuthMethod = EAP_SEAUTH_METHOD_MSCHAPV2;\r
  Private->FileType            = FileTypeMax;\r
}\r
\r
/**\r
  UI handle function when user select a network to connect.\r
\r
  @param[in]  Private             The pointer to the global private data structure.\r
  @param[in]  ProfileIndex        The profile index user selected to connect.\r
\r
  @retval EFI_INVALID_PARAMETER   Nic is null.\r
  @retval EFI_NOT_FOUND           Profile could not be found.\r
  @retval EFI_SUCCESS             The operation is completed successfully.\r
\r
**/\r
EFI_STATUS\r
WifiMgrUserSelectProfileToConnect(\r
  IN     WIFI_MGR_PRIVATE_DATA         *Private,\r
  IN     UINT32                        ProfileIndex\r
  )\r
{\r
  WIFI_MGR_NETWORK_PROFILE         *Profile;\r
  WIFI_MGR_DEVICE_DATA             *Nic;\r
\r
  Nic = Private->CurrentNic;\r
  if (Nic == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  //Initialize the connection page\r
  //\r
  WifiMgrCleanUserInput(Private);\r
\r
  Profile = WifiMgrGetProfileByProfileIndex (ProfileIndex, &Nic->ProfileList);\r
  if (Profile == NULL) {\r
    return EFI_NOT_FOUND;\r
  }\r
  Private->CurrentNic->UserSelectedProfile = Profile;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Record password from a HII input string.\r
\r
  @param[in]  Private             The pointer to the global private data structure.\r
  @param[in]  StringId            The QuestionId received from HII input.\r
  @param[in]  StringBuffer        The unicode string buffer to store password.\r
  @param[in]  StringBufferLen     The len of unicode string buffer.\r
\r
  @retval EFI_INVALID_PARAMETER   Any input parameter is invalid.\r
  @retval EFI_NOT_FOUND           The password string is not found or invalid.\r
  @retval EFI_SUCCESS             The operation is completed successfully.\r
\r
**/\r
EFI_STATUS\r
WifiMgrRecordPassword (\r
  IN   WIFI_MGR_PRIVATE_DATA      *Private,\r
  IN   EFI_STRING_ID              StringId,\r
  IN   CHAR16                     *StringBuffer,\r
  IN   UINTN                      StringBufferLen\r
  )\r
{\r
  CHAR16                          *Password;\r
\r
  if (StringId == 0 || StringBuffer == NULL || StringBufferLen <= 0) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Password = HiiGetString (Private->RegisteredHandle, StringId, NULL);\r
  if (Password == NULL) {\r
    return EFI_NOT_FOUND;\r
  }\r
  if (StrLen (Password) > StringBufferLen) {\r
    FreePool (Password);\r
    return EFI_NOT_FOUND;\r
  }\r
  StrnCpyS (StringBuffer, StringBufferLen, Password, StrLen (Password));\r
  ZeroMem (Password, (StrLen (Password) + 1) * sizeof (CHAR16));\r
  FreePool (Password);\r
\r
  //\r
  // Clean password in string package\r
  //\r
  HiiSetString (Private->RegisteredHandle, StringId, L"", NULL);\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Update connection message on connect configuration page, and trigger related form refresh.\r
\r
  @param[in]   Nic                        The related Nic for updating message.\r
  @param[in]   ConnectStateChanged        The tag to tell if the connection state has been changed, only\r
                                          when the connection changes from "Connected" or "Disconnecting"\r
                                          to "Disconnected", or from "Disconnected" or "Connecting" to\r
                                          "Connected", this tag can be set as TRUE.\r
  @param[in]   ConnectStatusMessage       The message to show on connected status bar, if NULL, will\r
                                          use default message.\r
\r
**/\r
VOID\r
WifiMgrUpdateConnectMessage (\r
  IN  WIFI_MGR_DEVICE_DATA      *Nic,\r
  IN  BOOLEAN                   ConnectStateChanged,\r
  IN  EFI_STRING                ConnectStatusMessage\r
  )\r
{\r
  CHAR16                   ConnectStatusStr[WIFI_STR_MAX_SIZE];\r
  WIFI_MGR_PRIVATE_DATA    *Private;\r
\r
  Private = Nic->Private;\r
  if (Private == NULL || Private->CurrentNic != Nic) {\r
    return;\r
  }\r
\r
  //\r
  // Update Connection Status Bar\r
  //\r
  if (ConnectStatusMessage != NULL) {\r
    HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_CONNECT_STATUS), ConnectStatusMessage, NULL);\r
  } else {\r
    if (Nic->ConnectState == WifiMgrConnectedToAp) {\r
\r
      UnicodeSPrint (ConnectStatusStr, sizeof (ConnectStatusStr), L"Connected to %s",\r
        Nic->CurrentOperateNetwork->SSId);\r
      HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_CONNECT_STATUS), ConnectStatusStr, NULL);\r
    } else if (Nic->ConnectState == WifiMgrDisconnected) {\r
\r
      HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_CONNECT_STATUS), L"Disconnected", NULL);\r
    } else if (Nic->ConnectState == WifiMgrConnectingToAp) {\r
\r
      UnicodeSPrint (ConnectStatusStr, sizeof (ConnectStatusStr), L"Connecting to %s ...",\r
        Nic->CurrentOperateNetwork->SSId);\r
      HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_CONNECT_STATUS), ConnectStatusStr, NULL);\r
    } else if (Nic->ConnectState == WifiMgrDisconnectingToAp) {\r
\r
      UnicodeSPrint (ConnectStatusStr, sizeof (ConnectStatusStr), L"Disconnecting from %s ...",\r
        Nic->CurrentOperateNetwork->SSId);\r
      HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_CONNECT_STATUS), ConnectStatusStr, NULL);\r
    } else {\r
      return;\r
    }\r
  }\r
\r
  //\r
  // Update Connect Button\r
  //\r
  if (Nic->ConnectState == WifiMgrConnectedToAp && Nic->UserSelectedProfile == Nic->CurrentOperateNetwork) {\r
\r
    HiiSetString (Private->RegisteredHandle,\r
      STRING_TOKEN (STR_CONNECT_NOW), L"Disconnect from this Network", NULL);\r
  } else {\r
    HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_CONNECT_NOW), L"Connect to this Network", NULL);\r
  }\r
  gBS->SignalEvent (Private->ConnectFormRefreshEvent);\r
\r
  //\r
  // Update Main Page and Network List\r
  //\r
  if (ConnectStateChanged) {\r
\r
    if (Nic->ConnectState == WifiMgrConnectedToAp) {\r
\r
      HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_CONNECTION_INFO), L"Connected to", NULL);\r
      HiiSetString (Private->RegisteredHandle,\r
        STRING_TOKEN (STR_CONNECTED_SSID), Nic->CurrentOperateNetwork->SSId, NULL);\r
    } else {\r
      HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_CONNECTION_INFO), L"Disconnected", NULL);\r
      HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_CONNECTED_SSID), L"", NULL);\r
    }\r
\r
    gBS->SignalEvent (Private->NetworkListRefreshEvent);\r
    gBS->SignalEvent (Private->MainPageRefreshEvent);\r
  }\r
}\r
\r
/**\r
  Convert the driver configuration data into the IFR data.\r
\r
  @param[in]   Private            The pointer to the global private data structure.\r
  @param[out]  IfrNvData          The IFR NV data.\r
\r
  @retval EFI_SUCCESS             The operation is completed successfully.\r
\r
**/\r
EFI_STATUS\r
WifiMgrConvertConfigDataToIfrNvData (\r
  IN   WIFI_MGR_PRIVATE_DATA      *Private,\r
  OUT  WIFI_MANAGER_IFR_NVDATA    *IfrNvData\r
  )\r
{\r
  //\r
  // Private shouldn't be NULL here, assert if Private is NULL.\r
  //\r
  ASSERT (Private != NULL);\r
\r
  if (Private->CurrentNic != NULL) {\r
    IfrNvData->ProfileCount = Private->CurrentNic->AvailableCount;\r
  } else {\r
    IfrNvData->ProfileCount = 0;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Convert the IFR data into the driver configuration data.\r
\r
  @param[in]       Private             The pointer to the global private data structure.\r
  @param[in, out]  IfrNvData           The IFR NV data.\r
\r
  @retval EFI_SUCCESS                  The operation is completed successfully.\r
\r
**/\r
EFI_STATUS\r
WifiMgrConvertIfrNvDataToConfigData (\r
  IN     WIFI_MGR_PRIVATE_DATA         *Private,\r
  IN OUT WIFI_MANAGER_IFR_NVDATA       *IfrNvData\r
  )\r
{\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  This function allows the caller to request the current\r
  configuration for one or more named elements. The resulting\r
  string is in <ConfigAltResp> format. Any and all alternative\r
  configuration strings shall also be appended to the end of the\r
  current configuration string. If they are, they must appear\r
  after the current configuration. They must contain the same\r
  routing (GUID, NAME, PATH) as the current configuration string.\r
  They must have an additional description indicating the type of\r
  alternative configuration the string represents,\r
  "ALTCFG=<StringToken>". That <StringToken> (when\r
  converted from Hex UNICODE to binary) is a reference to a\r
  string in the associated string pack.\r
\r
  @param This       Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL.\r
\r
  @param Request    A null-terminated Unicode string in\r
                    <ConfigRequest> format. Note that this\r
                    includes the routing information as well as\r
                    the configurable name / value pairs. It is\r
                    invalid for this string to be in\r
                    <MultiConfigRequest> format.\r
                    If a NULL is passed in for the Request field,\r
                    all of the settings being abstracted by this function\r
                    will be returned in the Results field.  In addition,\r
                    if a ConfigHdr is passed in with no request elements,\r
                    all of the settings being abstracted for that particular\r
                    ConfigHdr reference will be returned in the Results Field.\r
\r
  @param Progress   On return, points to a character in the\r
                    Request string. Points to the string's null\r
                    terminator if request was successful. Points\r
                    to the most recent "&" before the first\r
                    failing name / value pair (or the beginning\r
                    of the string if the failure is in the first\r
                    name / value pair) if the request was not\r
                    successful.\r
\r
  @param Results    A null-terminated Unicode string in\r
                    <MultiConfigAltResp> format which has all values\r
                    filled in for the names in the Request string.\r
                    String to be allocated by the called function.\r
\r
  @retval EFI_SUCCESS             The Results string is filled with the\r
                                  values corresponding to all requested\r
                                  names.\r
\r
  @retval EFI_OUT_OF_RESOURCES    Not enough memory to store the\r
                                  parts of the results that must be\r
                                  stored awaiting possible future\r
                                  protocols.\r
\r
  @retval EFI_NOT_FOUND           Routing data doesn't match any\r
                                  known driver. Progress set to the\r
                                  first character in the routing header.\r
                                  Note: There is no requirement that the\r
                                  driver validate the routing data. It\r
                                  must skip the <ConfigHdr> in order to\r
                                  process the names.\r
\r
  @retval EFI_INVALID_PARAMETER   Illegal syntax. Progress set\r
                                  to most recent "&" before the\r
                                  error or the beginning of the\r
                                  string.\r
\r
  @retval EFI_INVALID_PARAMETER   Unknown name. Progress points\r
                                  to the & before the name in\r
                                  question.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
WifiMgrDxeHiiConfigAccessExtractConfig (\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
  WIFI_MGR_PRIVATE_DATA             *Private;\r
  WIFI_MANAGER_IFR_NVDATA           *IfrNvData;\r
  EFI_STRING                        ConfigRequestHdr;\r
  EFI_STRING                        ConfigRequest;\r
  UINTN                             Size;\r
  BOOLEAN                           AllocatedRequest;\r
  UINTN                             BufferSize;\r
  EFI_STATUS                        Status;\r
\r
  if (This == NULL || Progress == NULL || Results == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  *Progress = Request;\r
  if ((Request != NULL) &&\r
      !HiiIsConfigHdrMatch (Request, &gWifiConfigFormSetGuid, mVendorStorageName)) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  ConfigRequestHdr = NULL;\r
  ConfigRequest    = NULL;\r
  AllocatedRequest = FALSE;\r
  Size             = 0;\r
\r
  Private   = WIFI_MGR_PRIVATE_DATA_FROM_CONFIG_ACCESS (This);\r
\r
  BufferSize = sizeof (WIFI_MANAGER_IFR_NVDATA);\r
  IfrNvData = AllocateZeroPool (BufferSize);\r
  if (IfrNvData == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  WifiMgrConvertConfigDataToIfrNvData (Private, IfrNvData);\r
\r
  ConfigRequest = Request;\r
  if ((Request == NULL) || (StrStr (Request, L"OFFSET") == NULL)) {\r
    //\r
    // Request has no request element, construct full request string.\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 (\r
                         &gWifiConfigFormSetGuid,\r
                         mVendorStorageName,\r
                         Private->DriverHandle);\r
    if (ConfigRequestHdr == NULL) {\r
      FreePool (IfrNvData);\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
\r
    Size = (StrLen (ConfigRequestHdr) + 32 + 1) * sizeof (CHAR16);\r
    ConfigRequest = AllocateZeroPool (Size);\r
    if (ConfigRequest == NULL) {\r
\r
      FreePool (IfrNvData);\r
      FreePool (ConfigRequestHdr);\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
\r
    AllocatedRequest = TRUE;\r
    UnicodeSPrint (\r
      ConfigRequest,\r
      Size,\r
      L"%s&OFFSET=0&WIDTH=%016LX",\r
      ConfigRequestHdr,\r
      (UINT64) BufferSize\r
      );\r
    FreePool (ConfigRequestHdr);\r
  }\r
\r
  //\r
  // Convert buffer data to <ConfigResp> by helper function BlockToConfig()\r
  //\r
  Status = gHiiConfigRouting->BlockToConfig (\r
                                gHiiConfigRouting,\r
                                ConfigRequest,\r
                                (UINT8 *) IfrNvData,\r
                                BufferSize,\r
                                Results,\r
                                Progress\r
                                );\r
\r
  FreePool (IfrNvData);\r
  //\r
  // Free the allocated config request string.\r
  //\r
  if (AllocatedRequest) {\r
    FreePool (ConfigRequest);\r
    ConfigRequest = NULL;\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
  This function applies changes in a driver's configuration.\r
  Input is a Configuration, which has the routing data for this\r
  driver followed by name / value configuration pairs. The driver\r
  must apply those pairs to its configurable storage. If the\r
  driver's configuration is stored in a linear block of data\r
  and the driver's name / value pairs are in <BlockConfig>\r
  format, it may use the ConfigToBlock helper function (above) to\r
  simplify the job.\r
\r
  @param This           Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL.\r
\r
  @param Configuration  A null-terminated Unicode string in\r
                        <ConfigString> format.\r
\r
  @param Progress       A pointer to a string filled in with the\r
                        offset of the most recent '&' before the\r
                        first failing name / value pair (or the\r
                        beginn ing of the string if the failure\r
                        is in the first name / value pair) or\r
                        the terminating NULL if all was\r
                        successful.\r
\r
  @retval EFI_SUCCESS             The results have been distributed or are\r
                                  awaiting distribution.\r
\r
  @retval EFI_OUT_OF_RESOURCES    Not enough memory to store the\r
                                  parts of the results that must be\r
                                  stored awaiting possible future\r
                                  protocols.\r
\r
  @retval EFI_INVALID_PARAMETERS  Passing in a NULL for the\r
                                  Results parameter would result\r
                                  in this type of error.\r
\r
  @retval EFI_NOT_FOUND           Target for the specified routing data\r
                                  was not found\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
WifiMgrDxeHiiConfigAccessRouteConfig (\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
  WIFI_MGR_PRIVATE_DATA          *Private;\r
  WIFI_MANAGER_IFR_NVDATA        *IfrNvData;\r
\r
  if (Configuration == NULL || Progress == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  IfrNvData  = NULL;\r
  *Progress  = Configuration;\r
  BufferSize = sizeof (WIFI_MANAGER_IFR_NVDATA);\r
  Private    = WIFI_MGR_PRIVATE_DATA_FROM_CONFIG_ACCESS (This);\r
\r
  if (!HiiIsConfigHdrMatch (Configuration, &gWifiConfigFormSetGuid, mVendorStorageName)) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  IfrNvData = AllocateZeroPool (BufferSize);\r
  if (IfrNvData == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  WifiMgrConvertConfigDataToIfrNvData (Private, IfrNvData);\r
\r
  Status = gHiiConfigRouting->ConfigToBlock (\r
                                gHiiConfigRouting,\r
                                Configuration,\r
                                (UINT8*) IfrNvData,\r
                                &BufferSize,\r
                                Progress\r
                                );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = WifiMgrConvertIfrNvDataToConfigData (Private, IfrNvData);\r
  ZeroMem (IfrNvData, sizeof (WIFI_MANAGER_IFR_NVDATA));\r
  FreePool (IfrNvData);\r
\r
  return Status;\r
}\r
\r
/**\r
  This function is called to provide results data to the driver.\r
  This data consists of a unique key that is used to identify\r
  which data is either being passed back or being asked for.\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. The format of the data tends to\r
                                 vary based on the opcode that generated the callback.\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
WifiMgrDxeHiiConfigAccessCallback (\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 OUT EFI_IFR_TYPE_VALUE                      *Value,\r
  OUT    EFI_BROWSER_ACTION_REQUEST              *ActionRequest\r
  )\r
{\r
  EFI_STATUS                         Status;\r
  EFI_INPUT_KEY                      Key;\r
  UINTN                              BufferSize;\r
  WIFI_MGR_PRIVATE_DATA              *Private;\r
  WIFI_MANAGER_IFR_NVDATA            *IfrNvData;\r
  EFI_DEVICE_PATH_PROTOCOL           *FilePath;\r
  WIFI_MGR_NETWORK_PROFILE           *Profile;\r
  WIFI_MGR_NETWORK_PROFILE           *ProfileToConnect;\r
  WIFI_HIDDEN_NETWORK_DATA           *HiddenNetwork;\r
  UINTN                              TempDataSize;\r
  VOID                               *TempData;\r
  LIST_ENTRY                         *Entry;\r
  UINT32                             Index;\r
  UINT32                             RemoveCount;\r
  CHAR16                             *TempPassword;\r
  CHAR16                             *ErrorMessage;\r
\r
  if (Action != EFI_BROWSER_ACTION_FORM_OPEN &&\r
      Action != EFI_BROWSER_ACTION_FORM_CLOSE &&\r
      Action != EFI_BROWSER_ACTION_CHANGING &&\r
      Action != EFI_BROWSER_ACTION_CHANGED &&\r
      Action != EFI_BROWSER_ACTION_RETRIEVE) {\r
\r
    return EFI_UNSUPPORTED;\r
  }\r
  if ((Value == NULL) || (ActionRequest == NULL)) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Status  = EFI_SUCCESS;\r
  Private = WIFI_MGR_PRIVATE_DATA_FROM_CONFIG_ACCESS (This);\r
  if (Private->CurrentNic == NULL) {\r
    return EFI_DEVICE_ERROR;\r
  }\r
\r
  //\r
  // Retrieve uncommitted data from Browser\r
  //\r
  BufferSize = sizeof (WIFI_MANAGER_IFR_NVDATA);\r
  IfrNvData = AllocateZeroPool (BufferSize);\r
  if (IfrNvData == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
  HiiGetBrowserData (&gWifiConfigFormSetGuid, mVendorStorageName, BufferSize, (UINT8 *) IfrNvData);\r
\r
  if (Action == EFI_BROWSER_ACTION_FORM_OPEN) {\r
    switch (QuestionId) {\r
\r
    case KEY_MAC_LIST:\r
\r
      Status = WifiMgrShowNicList (Private);\r
      break;\r
\r
    case KEY_REFRESH_NETWORK_LIST:\r
\r
      if (Private->CurrentNic->UserSelectedProfile != NULL) {\r
\r
        Profile = Private->CurrentNic->UserSelectedProfile;\r
\r
        //\r
        // Erase secrets since user has left Connection Page\r
        // Connection Page may direct to Network List Page or Eap Configuration Page,\r
        // secrets only need to be erased when head to Network List Page\r
        //\r
        WifiMgrCleanProfileSecrets (Profile);\r
\r
        Private->CurrentNic->UserSelectedProfile = NULL;\r
      }\r
\r
      break;\r
\r
    case KEY_CONNECT_ACTION:\r
\r
      if (Private->CurrentNic->UserSelectedProfile == NULL) {\r
        break;\r
      }\r
      Profile = Private->CurrentNic->UserSelectedProfile;\r
\r
      //\r
      //Enter the network connection configuration page\r
      //Recovery from restored data\r
      //\r
      if (HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_SSID), Profile->SSId, NULL) == 0) {\r
        return EFI_OUT_OF_RESOURCES;\r
      }\r
      IfrNvData->SecurityType = Profile->SecurityType;\r
      if (HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_SECURITY_TYPE),\r
            mSecurityType[IfrNvData->SecurityType], NULL) == 0) {\r
        return EFI_OUT_OF_RESOURCES;\r
      }\r
\r
      if (IfrNvData->SecurityType == SECURITY_TYPE_WPA2_ENTERPRISE) {\r
\r
        IfrNvData->EapAuthMethod        = Profile->EapAuthMethod;\r
        IfrNvData->EapSecondAuthMethod  = Profile->EapSecondAuthMethod;\r
        StrCpyS (IfrNvData->EapIdentity, EAP_IDENTITY_SIZE, Profile->EapIdentity);\r
      }\r
\r
      break;\r
\r
    case KEY_ENROLLED_CERT_NAME:\r
\r
      if (Private->CurrentNic->UserSelectedProfile == NULL) {\r
        break;\r
      }\r
      Profile = Private->CurrentNic->UserSelectedProfile;\r
\r
      //\r
      //Enter the key enrollment page\r
      //For TTLS and PEAP, only CA cert needs to be cared\r
      //\r
      if (Private->FileType == FileTypeCACert) {\r
\r
        if (Profile->CACertData != NULL) {\r
          HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_EAP_ENROLLED_CERT_NAME), Profile->CACertName, NULL);\r
        } else {\r
          HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_EAP_ENROLLED_CERT_NAME), L"", NULL);\r
        }\r
      } else if (Private->FileType == FileTypeClientCert) {\r
\r
        if (Profile->ClientCertData != NULL) {\r
          HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_EAP_ENROLLED_CERT_NAME), Profile->ClientCertName, NULL);\r
        } else {\r
          HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_EAP_ENROLLED_CERT_NAME), L"", NULL);\r
        }\r
      }\r
      break;\r
\r
    case KEY_ENROLLED_PRIVATE_KEY_NAME:\r
\r
      if (Private->CurrentNic->UserSelectedProfile == NULL) {\r
        break;\r
      }\r
      Profile = Private->CurrentNic->UserSelectedProfile;\r
\r
      if (Profile->PrivateKeyData != NULL) {\r
        HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_EAP_ENROLLED_PRIVATE_KEY_NAME), Profile->PrivateKeyName, NULL);\r
      } else {\r
        HiiSetString (Private->RegisteredHandle, STRING_TOKEN (STR_EAP_ENROLLED_PRIVATE_KEY_NAME), L"", NULL);\r
      }\r
      break;\r
\r
    default:\r
      break;\r
    }\r
  } else if (Action == EFI_BROWSER_ACTION_FORM_CLOSE) {\r
    switch (QuestionId) {\r
\r
    case KEY_CONNECT_ACTION:\r
\r
      if (Private->CurrentNic->UserSelectedProfile == NULL) {\r
        break;\r
      }\r
      Profile = Private->CurrentNic->UserSelectedProfile;\r
\r
      //\r
      //Restore User Config Data for Page recovery\r
      //\r
      if (IfrNvData->SecurityType == SECURITY_TYPE_WPA2_ENTERPRISE) {\r
\r
        Profile->EapAuthMethod        = IfrNvData->EapAuthMethod;\r
        Profile->EapSecondAuthMethod  = IfrNvData->EapSecondAuthMethod;\r
        StrCpyS (Profile->EapIdentity, EAP_IDENTITY_SIZE, IfrNvData->EapIdentity);\r
      }\r
      break;\r
\r
    default:\r
      break;\r
    }\r
  } else if (Action == EFI_BROWSER_ACTION_CHANGING) {\r
    switch (QuestionId) {\r
\r
    case KEY_NETWORK_LIST:\r
\r
      //\r
      //User triggered a scan process.\r
      //\r
      Private->CurrentNic->OneTimeScanRequest = TRUE;\r
      break;\r
\r
    case KEY_PASSWORD_CONNECT_NETWORK:\r
    case KEY_EAP_PASSWORD_CONNECT_NETWORK:\r
    case KEY_PRIVATE_KEY_PASSWORD:\r
\r
      if (Private->CurrentNic->UserSelectedProfile == NULL) {\r
        break;\r
      }\r
      Profile = Private->CurrentNic->UserSelectedProfile;\r
\r
      if (QuestionId == KEY_PASSWORD_CONNECT_NETWORK) {\r
        TempPassword = Profile->Password;\r
      } else if (QuestionId == KEY_EAP_PASSWORD_CONNECT_NETWORK) {\r
        TempPassword = Profile->EapPassword;\r
      } else {\r
        TempPassword = Profile->PrivateKeyPassword;\r
      }\r
\r
      Status = WifiMgrRecordPassword (Private, Value->string, TempPassword, PASSWORD_STORAGE_SIZE);\r
      if (EFI_ERROR (Status)) {\r
        DEBUG ((DEBUG_ERROR, "[WiFi Connection Manager] Error: Failed to input password!"));\r
        break;\r
      }\r
\r
      //\r
      // This password is not a new created password, so no need to confirm.\r
      //\r
      Status = EFI_NOT_FOUND;\r
      break;\r
\r
    case KEY_CONNECT_ACTION:\r
\r
      ErrorMessage     = NULL;\r
      ProfileToConnect = NULL;\r
\r
      if (Private->CurrentNic->UserSelectedProfile == NULL) {\r
        break;\r
      }\r
      Profile = Private->CurrentNic->UserSelectedProfile;\r
\r
      if (Private->CurrentNic->ConnectState == WifiMgrDisconnected ||\r
        Profile != Private->CurrentNic->CurrentOperateNetwork) {\r
\r
        //\r
        // When this network is not currently connected, pend it to connect.\r
        //\r
        if (Profile->AKMSuiteSupported && Profile->CipherSuiteSupported) {\r
\r
          if (Profile->SecurityType == SECURITY_TYPE_NONE || Profile->SecurityType == SECURITY_TYPE_WPA2_PERSONAL) {\r
\r
            //\r
            // For Open network, connect directly.\r
            //\r
            ProfileToConnect = Profile;\r
\r
          } else if (Profile->SecurityType == SECURITY_TYPE_WPA2_ENTERPRISE) {\r
\r
            //\r
            // For WPA/WPA2-Enterprise network, conduct eap configuration first.\r
            // Only EAP-TLS, TTLS and PEAP is supported now!\r
            //\r
            Profile->EapAuthMethod = IfrNvData->EapAuthMethod;\r
            StrCpyS (Profile->EapIdentity, EAP_IDENTITY_SIZE, IfrNvData->EapIdentity);\r
\r
            if (IfrNvData->EapAuthMethod == EAP_AUTH_METHOD_TTLS || IfrNvData->EapAuthMethod == EAP_AUTH_METHOD_PEAP) {\r
\r
              Profile->EapSecondAuthMethod = IfrNvData->EapSecondAuthMethod;\r
              ProfileToConnect = Profile;\r
            } else if (IfrNvData->EapAuthMethod == EAP_AUTH_METHOD_TLS) {\r
              ProfileToConnect = Profile;\r
            } else {\r
              ErrorMessage = L"ERROR: Only EAP-TLS, TTLS or PEAP is supported now!";\r
            }\r
          } else {\r
            ErrorMessage = L"ERROR: Can't connect to this network!";\r
          }\r
        } else {\r
          ErrorMessage = L"ERROR: This network is not supported!";\r
        }\r
\r
        if (ErrorMessage != NULL) {\r
          CreatePopUp (\r
            EFI_LIGHTGRAY | EFI_BACKGROUND_BLUE,\r
            &Key,\r
            ErrorMessage,\r
            NULL\r
            );\r
        }\r
\r
        if (ProfileToConnect != NULL) {\r
\r
          Private->CurrentNic->OneTimeConnectRequest = TRUE;\r
          Private->CurrentNic->ConnectPendingNetwork = ProfileToConnect;\r
        }\r
      } else if (Private->CurrentNic->ConnectState == WifiMgrConnectedToAp) {\r
\r
        //\r
        // This network is currently connected, just disconnect from it.\r
        //\r
        Private->CurrentNic->OneTimeDisconnectRequest    = TRUE;\r
        Private->CurrentNic->HasDisconnectPendingNetwork = TRUE;\r
      }\r
      break;\r
\r
    case KEY_ENROLL_CA_CERT_CONNECT_NETWORK:\r
\r
      Private->FileType = FileTypeCACert;\r
      break;\r
\r
    case KEY_ENROLL_CLIENT_CERT_CONNECT_NETWORK:\r
\r
      Private->FileType = FileTypeClientCert;\r
      break;\r
\r
    case KEY_EAP_ENROLL_PRIVATE_KEY_FROM_FILE:\r
\r
      FilePath = NULL;\r
      ChooseFile (NULL, NULL, NULL, &FilePath);\r
\r
      if (FilePath != NULL) {\r
\r
        UpdatePrivateKeyFromFile(Private, FilePath);\r
        FreePool (FilePath);\r
      }\r
      break;\r
\r
    case KEY_EAP_ENROLL_CERT_FROM_FILE:\r
\r
      //\r
      //User will select a cert file from File Explore\r
      //\r
      FilePath = NULL;\r
      ChooseFile( NULL, NULL, NULL, &FilePath);\r
\r
      if (FilePath != NULL) {\r
\r
        UpdateCAFromFile(Private, FilePath);\r
        FreePool (FilePath);\r
      }\r
      break;\r
\r
    case KEY_SAVE_PRIVATE_KEY_TO_MEM:\r
\r
      if (Private->FileContext != NULL && Private->FileContext->FHandle != NULL &&\r
        Private->CurrentNic->UserSelectedProfile != NULL) {\r
\r
        //\r
        // Read Private Key file to Buffer\r
        //\r
        Profile = Private->CurrentNic->UserSelectedProfile;\r
        if (Profile->PrivateKeyData != NULL) {\r
\r
          ZeroMem (Profile->PrivateKeyData, Profile->PrivateKeyDataSize);\r
          FreePool (Profile->PrivateKeyData);\r
          Profile->PrivateKeyData = NULL;\r
        }\r
\r
        Status = WifiMgrReadFileToBuffer (\r
                   Private->FileContext,\r
                   &TempData,\r
                   &TempDataSize\r
                   );\r
        if (EFI_ERROR (Status)) {\r
          CreatePopUp (\r
            EFI_LIGHTGRAY | EFI_BACKGROUND_BLUE,\r
            &Key,\r
            L"ERROR: Can't read this private key file!",\r
            NULL\r
            );\r
        } else {\r
\r
          ASSERT (Private->FileContext->FileName != NULL);\r
\r
          Profile->PrivateKeyData = TempData;\r
          Profile->PrivateKeyDataSize = TempDataSize;\r
          StrCpyS(Profile->PrivateKeyName, WIFI_FILENAME_STR_MAX_SIZE, Private->FileContext->FileName);\r
\r
          DEBUG ((DEBUG_INFO, "[WiFi Connection Manager] Private Key: %s has been enrolled! Size: %d\n",\r
            Profile->PrivateKeyName, Profile->PrivateKeyDataSize));\r
        }\r
      }\r
      break;\r
\r
    case KEY_SAVE_CERT_TO_MEM:\r
\r
      if (Private->FileContext != NULL && Private->FileContext->FHandle != NULL &&\r
        Private->CurrentNic->UserSelectedProfile != NULL) {\r
\r
        //\r
        // Read Cert file to Buffer\r
        //\r
        Profile = Private->CurrentNic->UserSelectedProfile;\r
\r
        if (Private->FileType == FileTypeCACert) {\r
          if (Profile->CACertData != NULL) {\r
\r
            ZeroMem (Profile->CACertData, Profile->CACertSize);\r
            FreePool (Profile->CACertData);\r
            Profile->CACertData = NULL;\r
          }\r
        } else if (Private->FileType == FileTypeClientCert) {\r
          if (Profile->ClientCertData != NULL) {\r
\r
            ZeroMem (Profile->ClientCertData, Profile->ClientCertSize);\r
            FreePool (Profile->ClientCertData);\r
            Profile->ClientCertData = NULL;\r
          }\r
        } else {\r
          break;\r
        }\r
\r
        Status = WifiMgrReadFileToBuffer (\r
                   Private->FileContext,\r
                   &TempData,\r
                   &TempDataSize\r
                   );\r
        if (EFI_ERROR (Status)) {\r
          CreatePopUp (\r
            EFI_LIGHTGRAY | EFI_BACKGROUND_BLUE,\r
            &Key,\r
            L"ERROR: Can't read this certificate file!",\r
            NULL\r
            );\r
        } else {\r
\r
          ASSERT (Private->FileContext->FileName != NULL);\r
          if (Private->FileType == FileTypeCACert) {\r
\r
            Profile->CACertData = TempData;\r
            Profile->CACertSize = TempDataSize;\r
            StrCpyS(Profile->CACertName, WIFI_FILENAME_STR_MAX_SIZE, Private->FileContext->FileName);\r
            DEBUG ((DEBUG_INFO, "[WiFi Connection Manager] CA Cert: %s has been enrolled! Size: %d\n",\r
              Profile->CACertName, Profile->CACertSize));\r
          } else {\r
\r
            Profile->ClientCertData = TempData;\r
            Profile->ClientCertSize = TempDataSize;\r
            StrCpyS(Profile->ClientCertName, WIFI_FILENAME_STR_MAX_SIZE, Private->FileContext->FileName);\r
            DEBUG ((DEBUG_INFO, "[WiFi Connection Manager] Client Cert: %s has been enrolled! Size: %d\n",\r
              Profile->ClientCertName, Profile->ClientCertSize));\r
          }\r
        }\r
      }\r
      break;\r
\r
    case KEY_ADD_HIDDEN_NETWORK:\r
\r
      //\r
      // Add a Hidden Network\r
      //\r
      if (StrLen (IfrNvData->SSId) < SSID_MIN_LEN ||\r
        Private->HiddenNetworkCount >= HIDDEN_NETWORK_LIST_COUNT_MAX) {\r
\r
        Status = EFI_ABORTED;\r
        break;\r
      } else {\r
\r
        //\r
        // Check if this SSId is already in Hidden Network List\r
        //\r
        NET_LIST_FOR_EACH (Entry, &Private->HiddenNetworkList) {\r
\r
          HiddenNetwork = NET_LIST_USER_STRUCT_S (Entry, WIFI_HIDDEN_NETWORK_DATA,\r
                            Link, WIFI_MGR_HIDDEN_NETWORK_SIGNATURE);\r
          if (StrCmp (HiddenNetwork->SSId, IfrNvData->SSId) == 0) {\r
\r
            Status = EFI_ABORTED;\r
            break;\r
          }\r
        }\r
      }\r
\r
      HiddenNetwork = (WIFI_HIDDEN_NETWORK_DATA *) AllocateZeroPool (sizeof (WIFI_HIDDEN_NETWORK_DATA));\r
      if (HiddenNetwork == NULL) {\r
\r
        Status = EFI_OUT_OF_RESOURCES;\r
        break;\r
      }\r
      HiddenNetwork->Signature = WIFI_MGR_HIDDEN_NETWORK_SIGNATURE;\r
      StrCpyS (HiddenNetwork->SSId, SSID_STORAGE_SIZE, IfrNvData->SSId);\r
\r
      InsertTailList (&Private->HiddenNetworkList, &HiddenNetwork->Link);\r
      Private->HiddenNetworkCount ++;\r
\r
      WifiMgrRefreshHiddenList (Private);\r
      break;\r
\r
    case KEY_REMOVE_HIDDEN_NETWORK:\r
\r
      //\r
      // Remove Hidden Networks\r
      //\r
      Entry = GetFirstNode (&Private->HiddenNetworkList);\r
      RemoveCount = 0;\r
      for (Index = 0; Index < Private->HiddenNetworkCount; Index ++) {\r
        if (IfrNvData->HiddenNetworkList[Index] != 0) {\r
\r
          HiddenNetwork = NET_LIST_USER_STRUCT_S (Entry, WIFI_HIDDEN_NETWORK_DATA, Link, WIFI_MGR_HIDDEN_NETWORK_SIGNATURE);\r
          Entry = RemoveEntryList (Entry);\r
          RemoveCount ++;\r
\r
          FreePool (HiddenNetwork);\r
        } else {\r
          Entry = GetNextNode (&Private->HiddenNetworkList, Entry);\r
        }\r
      }\r
\r
      Private->HiddenNetworkCount -= RemoveCount;\r
      WifiMgrRefreshHiddenList (Private);\r
      break;\r
\r
    default:\r
\r
      if (QuestionId >= KEY_MAC_ENTRY_BASE && QuestionId < KEY_MAC_ENTRY_BASE + Private->NicCount) {\r
        //\r
        // User selects a wireless NIC.\r
        //\r
        Status = WifiMgrSelectNic (Private, QuestionId);\r
        if (EFI_ERROR (Status)) {\r
          CreatePopUp (\r
            EFI_LIGHTGRAY | EFI_BACKGROUND_BLUE,\r
            &Key,\r
            L"ERROR: Fail to operate the wireless NIC!",\r
            NULL\r
          );\r
        }\r
      } else if (Private->CurrentNic != NULL) {\r
        if (QuestionId >= KEY_AVAILABLE_NETWORK_ENTRY_BASE &&\r
          QuestionId <= KEY_AVAILABLE_NETWORK_ENTRY_BASE + Private->CurrentNic->MaxProfileIndex) {\r
\r
          Status = WifiMgrUserSelectProfileToConnect (Private, QuestionId - KEY_AVAILABLE_NETWORK_ENTRY_BASE);\r
          if (!EFI_ERROR (Status)) {\r
            WifiMgrUpdateConnectMessage(Private->CurrentNic, FALSE, NULL);\r
          }\r
        }\r
\r
        if (EFI_ERROR (Status)) {\r
          CreatePopUp (\r
            EFI_LIGHTGRAY | EFI_BACKGROUND_BLUE,\r
            &Key,\r
            L"ERROR: Fail to operate this profile!",\r
            NULL\r
          );\r
        }\r
      }\r
\r
      break;\r
    }\r
  } else if (Action == EFI_BROWSER_ACTION_CHANGED) {\r
    switch (QuestionId) {\r
\r
    case KEY_SAVE_CERT_TO_MEM:\r
    case KEY_SAVE_PRIVATE_KEY_TO_MEM:\r
\r
      *ActionRequest = EFI_BROWSER_ACTION_REQUEST_FORM_SUBMIT_EXIT;\r
      break;\r
\r
    case KEY_NO_SAVE_CERT_TO_MEM:\r
    case KEY_NO_SAVE_PRIVATE_KEY_TO_MEM:\r
\r
      *ActionRequest = EFI_BROWSER_ACTION_REQUEST_FORM_DISCARD_EXIT;\r
      break;\r
\r
    default:\r
\r
      *ActionRequest = EFI_BROWSER_ACTION_REQUEST_FORM_APPLY;\r
      break;\r
    }\r
  } else if (Action == EFI_BROWSER_ACTION_RETRIEVE) {\r
\r
    switch (QuestionId) {\r
\r
    case KEY_REFRESH_NETWORK_LIST:\r
\r
      WifiMgrRefreshNetworkList (Private, IfrNvData);\r
      break;\r
\r
    default:\r
      break;\r
    }\r
  }\r
\r
  if (!EFI_ERROR (Status)) {\r
    //\r
    // Pass changed uncommitted data back to Form Browser.\r
    //\r
    BufferSize = sizeof (WIFI_MANAGER_IFR_NVDATA);\r
    HiiSetBrowserData (&gWifiConfigFormSetGuid, mVendorStorageName, BufferSize, (UINT8 *) IfrNvData, NULL);\r
  }\r
\r
  ZeroMem (IfrNvData, sizeof (WIFI_MANAGER_IFR_NVDATA));\r
  FreePool (IfrNvData);\r
  return Status;\r
}\r
\r
/**\r
  Initialize the WiFi configuration form.\r
\r
  @param[in]  Private             The pointer to the global private data structure.\r
\r
  @retval EFI_SUCCESS             The configuration form is initialized.\r
  @retval EFI_OUT_OF_RESOURCES    Failed to allocate memory.\r
  @retval EFI_INVALID_PARAMETER   Any input parameter is invalid.\r
  @retval Other Erros             Returned Errors when installing protocols.\r
\r
**/\r
EFI_STATUS\r
WifiMgrDxeConfigFormInit (\r
  WIFI_MGR_PRIVATE_DATA    *Private\r
)\r
{\r
  EFI_STATUS                      Status;\r
\r
  if (Private == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Private->ConfigAccess.ExtractConfig = WifiMgrDxeHiiConfigAccessExtractConfig;\r
  Private->ConfigAccess.RouteConfig   = WifiMgrDxeHiiConfigAccessRouteConfig;\r
  Private->ConfigAccess.Callback      = WifiMgrDxeHiiConfigAccessCallback;\r
\r
  //\r
  // Install Device Path Protocol and Config Access protocol to driver handle.\r
  //\r
  Status = gBS->InstallMultipleProtocolInterfaces (\r
                  &Private->DriverHandle,\r
                  &gEfiDevicePathProtocolGuid,\r
                  &mWifiMgrDxeHiiVendorDevicePath,\r
                  &gEfiHiiConfigAccessProtocolGuid,\r
                  &Private->ConfigAccess,\r
                  NULL\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Publish our HII data.\r
  //\r
  Private->RegisteredHandle = HiiAddPackages (\r
                                &gWifiConfigFormSetGuid,\r
                                Private->DriverHandle,\r
                                WifiConnectionManagerDxeStrings,\r
                                WifiConnectionManagerDxeBin,\r
                                NULL\r
                                );\r
  if (Private->RegisteredHandle == NULL) {\r
    gBS->UninstallMultipleProtocolInterfaces (\r
           Private->DriverHandle,\r
           &gEfiDevicePathProtocolGuid,\r
           &mWifiMgrDxeHiiVendorDevicePath,\r
           &gEfiHiiConfigAccessProtocolGuid,\r
           &Private->ConfigAccess,\r
           NULL\r
           );\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  Private->FileContext = AllocateZeroPool (sizeof (WIFI_MGR_FILE_CONTEXT));\r
  if (Private->FileContext == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Unload the WiFi configuration form.\r
\r
  @param[in]  Private             The pointer to the global private data structure.\r
\r
  @retval EFI_SUCCESS             The configuration form is unloaded successfully.\r
  @retval EFI_INVALID_PARAMETER   Any input parameter is invalid.\r
  @retval Other Errors            Returned Erros when uninstalling protocols.\r
\r
**/\r
EFI_STATUS\r
WifiMgrDxeConfigFormUnload (\r
  WIFI_MGR_PRIVATE_DATA    *Private\r
)\r
{\r
  EFI_STATUS    Status;\r
\r
  if (Private == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if (Private->FileContext != NULL) {\r
\r
    if (Private->FileContext->FHandle != NULL) {\r
      Private->FileContext->FHandle->Close (Private->FileContext->FHandle);\r
    }\r
\r
    if (Private->FileContext->FileName != NULL) {\r
      FreePool (Private->FileContext->FileName);\r
    }\r
    FreePool (Private->FileContext);\r
  }\r
\r
  HiiRemovePackages(Private->RegisteredHandle);\r
\r
  Status = gBS->UninstallMultipleProtocolInterfaces (\r
             Private->DriverHandle,\r
             &gEfiDevicePathProtocolGuid,\r
             &mWifiMgrDxeHiiVendorDevicePath,\r
             &gEfiHiiConfigAccessProtocolGuid,\r
             &Private->ConfigAccess,\r
             NULL\r
             );\r
\r
  return Status;\r
}\r