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

/** @file\r
  The Mac Connection2 Protocol adapter functions for WiFi Connection Manager.\r
\r
  Copyright (c) 2019, Intel Corporation. All rights reserved.<BR>\r
\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "WifiConnectionMgrDxe.h"\r
\r
EFI_EAP_TYPE mEapAuthMethod[] = {\r
  EFI_EAP_TYPE_TTLS,\r
  EFI_EAP_TYPE_PEAP,\r
  EFI_EAP_TYPE_EAPTLS\r
};\r
\r
EFI_EAP_TYPE mEapSecondAuthMethod[] = {\r
  EFI_EAP_TYPE_MSCHAPV2\r
};\r
\r
/**\r
  The callback function for scan operation. This function updates networks\r
  according to the latest scan result, and trigger UI refresh.\r
\r
  ASSERT when errors occur in config token.\r
\r
  @param[in]  Event                 The GetNetworks token receive event.\r
  @param[in]  Context               The context of the GetNetworks token.\r
\r
**/\r
VOID\r
EFIAPI\r
WifiMgrOnScanFinished (\r
  IN  EFI_EVENT                     Event,\r
  IN  VOID                          *Context\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  WIFI_MGR_MAC_CONFIG_TOKEN         *ConfigToken;\r
  WIFI_MGR_DEVICE_DATA              *Nic;\r
  WIFI_MGR_NETWORK_PROFILE          *Profile;\r
  EFI_80211_NETWORK                 *Network;\r
  UINTN                             DataSize;\r
  EFI_80211_NETWORK_DESCRIPTION     *NetworkDescription;\r
  EFI_80211_GET_NETWORKS_RESULT     *Result;\r
  LIST_ENTRY                        *Entry;\r
  UINT8                             SecurityType;\r
  BOOLEAN                           AKMSuiteSupported;\r
  BOOLEAN                           CipherSuiteSupported;\r
  CHAR8                             *AsciiSSId;\r
  UINTN                             Index;\r
\r
  ASSERT (Context != NULL);\r
\r
  ConfigToken = (WIFI_MGR_MAC_CONFIG_TOKEN *) Context;\r
  ASSERT (ConfigToken->Nic != NULL);\r
  ASSERT (ConfigToken->Type == TokenTypeGetNetworksToken);\r
\r
  //\r
  // It is the GetNetworks token, set scan state to "ScanFinished"\r
  //\r
  ConfigToken->Nic->ScanState = WifiMgrScanFinished;\r
\r
  ASSERT (ConfigToken->Token.GetNetworksToken != NULL);\r
  Result = ConfigToken->Token.GetNetworksToken->Result;\r
  Nic    = ConfigToken->Nic;\r
\r
  //\r
  // Clean previous result, and update network list according to the scan result\r
  //\r
  Nic->AvailableCount    = 0;\r
\r
  NET_LIST_FOR_EACH (Entry, &Nic->ProfileList) {\r
    Profile = NET_LIST_USER_STRUCT_S (Entry, WIFI_MGR_NETWORK_PROFILE,\r
                Link, WIFI_MGR_PROFILE_SIGNATURE);\r
    Profile->IsAvailable = FALSE;\r
  }\r
\r
  if (Result == NULL) {\r
    gBS->SignalEvent (Nic->Private->NetworkListRefreshEvent);\r
    WifiMgrFreeToken(ConfigToken);\r
    return;\r
  }\r
\r
  for (Index = 0; Index < Result->NumOfNetworkDesc; Index ++) {\r
\r
    NetworkDescription = Result->NetworkDesc + Index;\r
    if (NetworkDescription == NULL) {\r
      continue;\r
    }\r
\r
    Network = &NetworkDescription->Network;\r
    if (Network == NULL || Network->SSId.SSIdLen == 0) {\r
      continue;\r
    }\r
\r
    Status = WifiMgrCheckRSN (\r
               Network->AKMSuite,\r
               Network->CipherSuite,\r
               Nic,\r
               &SecurityType,\r
               &AKMSuiteSupported,\r
               &CipherSuiteSupported\r
               );\r
    if (EFI_ERROR (Status)) {\r
\r
      SecurityType          = SECURITY_TYPE_UNKNOWN;\r
      AKMSuiteSupported     = FALSE;\r
      CipherSuiteSupported  = FALSE;\r
    }\r
\r
    AsciiSSId = (CHAR8*) AllocateZeroPool(sizeof (CHAR8) * (Network->SSId.SSIdLen + 1));\r
    if (AsciiSSId == NULL) {\r
      continue;\r
    }\r
    CopyMem(AsciiSSId, (CHAR8 *) Network->SSId.SSId, sizeof (CHAR8) * Network->SSId.SSIdLen);\r
    *(AsciiSSId + Network->SSId.SSIdLen) = '\0';\r
\r
    Profile = WifiMgrGetProfileByAsciiSSId (AsciiSSId, SecurityType, &Nic->ProfileList);\r
    if (Profile == NULL) {\r
\r
      if (Nic->MaxProfileIndex >= NETWORK_LIST_COUNT_MAX) {\r
        FreePool (AsciiSSId);\r
        continue;\r
      }\r
\r
      //\r
      // Create a new profile\r
      //\r
      Profile = AllocateZeroPool (sizeof (WIFI_MGR_NETWORK_PROFILE));\r
      if (Profile == NULL) {\r
        FreePool (AsciiSSId);\r
        continue;\r
      }\r
      Profile->Signature    = WIFI_MGR_PROFILE_SIGNATURE;\r
      Profile->NicIndex     = Nic->NicIndex;\r
      Profile->ProfileIndex = Nic->MaxProfileIndex + 1;\r
      AsciiStrToUnicodeStrS (AsciiSSId, Profile->SSId, SSID_STORAGE_SIZE);\r
      InsertTailList (&Nic->ProfileList, &Profile->Link);\r
      Nic->MaxProfileIndex ++;\r
    }\r
    FreePool (AsciiSSId);\r
\r
    //\r
    //May receive duplicate networks in scan results, check if it has already\r
    //been processed.\r
    //\r
    if (!Profile->IsAvailable) {\r
\r
      Profile->IsAvailable          = TRUE;\r
      Profile->SecurityType         = SecurityType;\r
      Profile->AKMSuiteSupported    = AKMSuiteSupported;\r
      Profile->CipherSuiteSupported = CipherSuiteSupported;\r
      Profile->NetworkQuality       = NetworkDescription->NetworkQuality;\r
      Nic->AvailableCount ++;\r
\r
      //\r
      //Copy BSSType and SSId\r
      //\r
      CopyMem(&Profile->Network, Network, sizeof (EFI_80211_NETWORK));\r
\r
      //\r
      //Copy AKMSuite list\r
      //\r
      if (Network->AKMSuite != NULL) {\r
\r
        if (Network->AKMSuite->AKMSuiteCount == 0) {\r
          DataSize = sizeof (EFI_80211_AKM_SUITE_SELECTOR);\r
        } else {\r
          DataSize = sizeof (EFI_80211_AKM_SUITE_SELECTOR) + sizeof (EFI_80211_SUITE_SELECTOR)\r
                       * (Network->AKMSuite->AKMSuiteCount - 1);\r
        }\r
        Profile->Network.AKMSuite = (EFI_80211_AKM_SUITE_SELECTOR *) AllocateZeroPool (DataSize);\r
        if (Profile->Network.AKMSuite == NULL) {\r
          continue;\r
        }\r
        CopyMem (Profile->Network.AKMSuite, Network->AKMSuite, DataSize);\r
      }\r
\r
      //\r
      //Copy CipherSuite list\r
      //\r
      if (Network->CipherSuite != NULL) {\r
\r
        if (Network->CipherSuite->CipherSuiteCount == 0) {\r
          DataSize = sizeof (EFI_80211_CIPHER_SUITE_SELECTOR);\r
        } else {\r
          DataSize = sizeof (EFI_80211_CIPHER_SUITE_SELECTOR) + sizeof (EFI_80211_SUITE_SELECTOR)\r
                       * (Network->CipherSuite->CipherSuiteCount - 1);\r
        }\r
        Profile->Network.CipherSuite = (EFI_80211_CIPHER_SUITE_SELECTOR *) AllocateZeroPool (DataSize);\r
        if (Profile->Network.CipherSuite == NULL) {\r
          continue;\r
        }\r
        CopyMem (Profile->Network.CipherSuite, Network->CipherSuite, DataSize);\r
      }\r
    } else {\r
      //\r
      // A duplicate network, update signal quality\r
      //\r
      if (Profile->NetworkQuality < NetworkDescription->NetworkQuality) {\r
        Profile->NetworkQuality = NetworkDescription->NetworkQuality;\r
      }\r
      continue;\r
    }\r
  }\r
\r
  gBS->SignalEvent (Nic->Private->NetworkListRefreshEvent);\r
\r
  //\r
  // The current connected network should always be available until disconnection\r
  // happens in Wifi FW layer, even when it is not in this time's scan result.\r
  //\r
  if (Nic->ConnectState == WifiMgrConnectedToAp && Nic->CurrentOperateNetwork != NULL) {\r
    if (!Nic->CurrentOperateNetwork->IsAvailable) {\r
      Nic->CurrentOperateNetwork->IsAvailable = TRUE;\r
      Nic->AvailableCount ++;\r
    }\r
  }\r
\r
  WifiMgrFreeToken(ConfigToken);\r
}\r
\r
/**\r
  Start scan operation, and send out a token to collect available networks.\r
\r
  @param[in]  Nic                 Pointer to the device data of the selected NIC.\r
\r
  @retval EFI_SUCCESS             The operation is completed.\r
  @retval EFI_ALREADY_STARTED     A former scan operation is already ongoing.\r
  @retval EFI_INVALID_PARAMETER   One or more parameters are invalid.\r
  @retval EFI_OUT_OF_RESOURCES    Failed to allocate memory.\r
  @retval Other Errors            Return errors when getting networks from low layer.\r
\r
**/\r
EFI_STATUS\r
WifiMgrStartScan (\r
  IN      WIFI_MGR_DEVICE_DATA        *Nic\r
  )\r
{\r
  EFI_STATUS                          Status;\r
  EFI_TPL                             OldTpl;\r
  WIFI_MGR_MAC_CONFIG_TOKEN           *ConfigToken;\r
  EFI_80211_GET_NETWORKS_TOKEN        *GetNetworksToken;\r
  UINT32                              HiddenSSIdIndex;\r
  UINT32                              HiddenSSIdCount;\r
  EFI_80211_SSID                      *HiddenSSIdList;\r
  WIFI_HIDDEN_NETWORK_DATA            *HiddenNetwork;\r
  LIST_ENTRY                          *Entry;\r
\r
  if (Nic == NULL || Nic->Wmp == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if (Nic->ScanState == WifiMgrScanning) {\r
    return EFI_ALREADY_STARTED;\r
  }\r
\r
  Nic->ScanState  = WifiMgrScanning;\r
  OldTpl          = gBS->RaiseTPL (TPL_CALLBACK);\r
  Status          = EFI_SUCCESS;\r
  HiddenSSIdList  = NULL;\r
  HiddenSSIdCount = Nic->Private->HiddenNetworkCount;\r
  HiddenSSIdIndex = 0;\r
\r
  //\r
  //create a new get network token\r
  //\r
  ConfigToken     = AllocateZeroPool (sizeof (WIFI_MGR_MAC_CONFIG_TOKEN));\r
  if (ConfigToken == NULL) {\r
    gBS->RestoreTPL (OldTpl);\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  ConfigToken->Type      = TokenTypeGetNetworksToken;\r
  ConfigToken->Nic       = Nic;\r
  ConfigToken->Token.GetNetworksToken = AllocateZeroPool (sizeof (EFI_80211_GET_NETWORKS_TOKEN));\r
  if (ConfigToken->Token.GetNetworksToken == NULL) {\r
    WifiMgrFreeToken(ConfigToken);\r
    gBS->RestoreTPL (OldTpl);\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
  GetNetworksToken = ConfigToken->Token.GetNetworksToken;\r
\r
  //\r
  // There are some hidden networks to scan, add them into scan list\r
  //\r
  if (HiddenSSIdCount > 0) {\r
    HiddenSSIdList = AllocateZeroPool(HiddenSSIdCount * sizeof (EFI_80211_SSID));\r
    if (HiddenSSIdList == NULL) {\r
      WifiMgrFreeToken(ConfigToken);\r
      gBS->RestoreTPL (OldTpl);\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
\r
    HiddenSSIdIndex = 0;\r
    NET_LIST_FOR_EACH (Entry, &Nic->Private->HiddenNetworkList) {\r
\r
      HiddenNetwork = NET_LIST_USER_STRUCT_S (Entry, WIFI_HIDDEN_NETWORK_DATA,\r
                        Link, WIFI_MGR_HIDDEN_NETWORK_SIGNATURE);\r
      HiddenSSIdList[HiddenSSIdIndex].SSIdLen = (UINT8) StrLen (HiddenNetwork->SSId);\r
      UnicodeStrToAsciiStrS(HiddenNetwork->SSId,\r
        (CHAR8 *) HiddenSSIdList[HiddenSSIdIndex].SSId, SSID_STORAGE_SIZE);\r
      HiddenSSIdIndex ++;\r
    }\r
    GetNetworksToken->Data = AllocateZeroPool (sizeof (EFI_80211_GET_NETWORKS_DATA) +\r
                               (HiddenSSIdCount - 1) * sizeof (EFI_80211_SSID));\r
    if (GetNetworksToken->Data == NULL) {\r
      FreePool (HiddenSSIdList);\r
      WifiMgrFreeToken(ConfigToken);\r
      gBS->RestoreTPL (OldTpl);\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
    GetNetworksToken->Data->NumOfSSID = HiddenSSIdCount;\r
    CopyMem(GetNetworksToken->Data->SSIDList, HiddenSSIdList, HiddenSSIdCount * sizeof (EFI_80211_SSID));\r
    FreePool(HiddenSSIdList);\r
  } else {\r
\r
    GetNetworksToken->Data = AllocateZeroPool (sizeof (EFI_80211_GET_NETWORKS_DATA));\r
    if (GetNetworksToken->Data == NULL) {\r
      WifiMgrFreeToken(ConfigToken);\r
      gBS->RestoreTPL (OldTpl);\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
\r
    GetNetworksToken->Data->NumOfSSID = 0;\r
  }\r
\r
  //\r
  //Create a handle when scan process ends\r
  //\r
  Status = gBS->CreateEvent (\r
                  EVT_NOTIFY_SIGNAL,\r
                  TPL_CALLBACK,\r
                  WifiMgrOnScanFinished,\r
                  ConfigToken,\r
                  &GetNetworksToken->Event\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    WifiMgrFreeToken(ConfigToken);\r
    gBS->RestoreTPL (OldTpl);\r
    return Status;\r
  }\r
\r
  //\r
  //Start scan ...\r
  //\r
  Status = Nic->Wmp->GetNetworks (Nic->Wmp, GetNetworksToken);\r
  if (EFI_ERROR (Status)) {\r
    Nic->ScanState  = WifiMgrScanFinished;\r
    WifiMgrFreeToken(ConfigToken);\r
    gBS->RestoreTPL (OldTpl);\r
    return Status;\r
  }\r
\r
  gBS->RestoreTPL (OldTpl);\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Configure password to supplicant before connecting to a secured network.\r
\r
  @param[in]  Nic                 Pointer to the device data of the selected NIC.\r
  @param[in]  Profile             The target network to be connected.\r
\r
  @retval EFI_SUCCESS             The operation is completed.\r
  @retval EFI_INVALID_PARAMETER   One or more parameters are invalid.\r
  @retval EFI_OUT_OF_RESOURCES    Failed to allocate memory.\r
  @retval EFI_NOT_FOUND           No valid password is found to configure.\r
  @retval Other Errors            Returned errors when setting data to supplicant.\r
\r
**/\r
EFI_STATUS\r
WifiMgrConfigPassword (\r
  IN    WIFI_MGR_DEVICE_DATA              *Nic,\r
  IN    WIFI_MGR_NETWORK_PROFILE          *Profile\r
  )\r
{\r
  EFI_STATUS                 Status;\r
  EFI_SUPPLICANT_PROTOCOL    *Supplicant;\r
  EFI_80211_SSID             SSId;\r
  UINT8                      *AsciiPassword;\r
\r
  if (Nic == NULL || Nic->Supplicant == NULL || Profile == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  Supplicant = Nic->Supplicant;\r
  //\r
  //Set SSId to supplicant\r
  //\r
  SSId.SSIdLen = Profile->Network.SSId.SSIdLen;\r
  CopyMem(SSId.SSId, Profile->Network.SSId.SSId, sizeof (Profile->Network.SSId.SSId));\r
  Status = Supplicant->SetData(Supplicant,EfiSupplicant80211TargetSSIDName,\r
                         (VOID *)&SSId, sizeof(EFI_80211_SSID));\r
  if (EFI_ERROR(Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  //Set password to supplicant\r
  //\r
  if (StrLen (Profile->Password) < PASSWORD_MIN_LEN) {\r
    return EFI_NOT_FOUND;\r
  }\r
  AsciiPassword = AllocateZeroPool ((StrLen(Profile->Password) + 1) * sizeof (UINT8));\r
  if (AsciiPassword == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
  UnicodeStrToAsciiStrS (Profile->Password, (CHAR8 *) AsciiPassword, PASSWORD_STORAGE_SIZE);\r
  Status = Supplicant->SetData (Supplicant, EfiSupplicant80211PskPassword,\r
                         AsciiPassword, (StrLen(Profile->Password) + 1) * sizeof (UINT8));\r
  ZeroMem (AsciiPassword, AsciiStrLen ((CHAR8 *) AsciiPassword) + 1);\r
  FreePool(AsciiPassword);\r
\r
  return Status;\r
}\r
\r
/**\r
  Conduct EAP configuration to supplicant before connecting to a EAP network.\r
  Current WiFi Connection Manager only supports three kinds of EAP networks:\r
  1). EAP-TLS (Two-Way Authentication is required in our implementation)\r
  2). EAP-TTLS/MSCHAPv2 (One-Way Authentication is required in our implementation)\r
  3). PEAPv0/MSCHAPv2 (One-Way Authentication is required in our implementation)\r
\r
  @param[in]  Nic                 Pointer to the device data of the selected NIC.\r
  @param[in]  Profile             The target network to be connected.\r
\r
  @retval EFI_SUCCESS             The operation is completed.\r
  @retval EFI_INVALID_PARAMETER   One or more parameters are invalid.\r
  @retval EFI_UNSUPPORTED         The expected EAP method is not supported.\r
  @retval EFI_OUT_OF_RESOURCES    Failed to allocate memory.\r
  @retval Other Errors            Returned errors when setting data to supplicant.\r
\r
**/\r
EFI_STATUS\r
WifiMgrConfigEap (\r
  IN    WIFI_MGR_DEVICE_DATA              *Nic,\r
  IN    WIFI_MGR_NETWORK_PROFILE          *Profile\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  EFI_EAP_CONFIGURATION_PROTOCOL    *EapConfig;\r
  EFI_EAP_TYPE                      EapAuthMethod;\r
  EFI_EAP_TYPE                      EapSecondAuthMethod;\r
  EFI_EAP_TYPE                      *AuthMethodList;\r
  CHAR8                             *Identity;\r
  UINTN                             IdentitySize;\r
  CHAR16                            *Password;\r
  UINTN                             PasswordSize;\r
  UINTN                             EncryptPasswordLen;\r
  CHAR8                             *AsciiEncryptPassword;\r
  UINTN                             AuthMethodListSize;\r
  UINTN                             Index;\r
\r
  if (Nic == NULL || Nic->EapConfig == NULL || Profile == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  EapConfig = Nic->EapConfig;\r
\r
  if (Profile->EapAuthMethod >= EAP_AUTH_METHOD_MAX) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  EapAuthMethod = mEapAuthMethod[Profile->EapAuthMethod];\r
\r
  if (EapAuthMethod != EFI_EAP_TYPE_EAPTLS) {\r
    if (Profile->EapSecondAuthMethod >= EAP_SEAUTH_METHOD_MAX) {\r
      return EFI_INVALID_PARAMETER;\r
    }\r
    EapSecondAuthMethod = mEapSecondAuthMethod[Profile->EapSecondAuthMethod];\r
  }\r
\r
  //\r
  //The first time to get Supported Auth Method list, return the size.\r
  //\r
  AuthMethodListSize  = 0;\r
  AuthMethodList      = NULL;\r
  Status = EapConfig->GetData (EapConfig, EFI_EAP_TYPE_ATTRIBUTE, EfiEapConfigEapSupportedAuthMethod,\r
                        (VOID *) AuthMethodList, &AuthMethodListSize);\r
  if (Status == EFI_SUCCESS) {\r
    //\r
    //No Supported Eap Auth Method\r
    //\r
    return EFI_UNSUPPORTED;\r
  } else if (Status != EFI_BUFFER_TOO_SMALL) {\r
    return Status;\r
  }\r
\r
  //\r
  // The second time to get Supported Auth Method list, return the list.\r
  // In current design, only EAPTLS, TTLS and PEAP are supported\r
  //\r
  AuthMethodList = (EFI_EAP_TYPE *) AllocateZeroPool(AuthMethodListSize);\r
  if (AuthMethodList == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
  Status = EapConfig->GetData (EapConfig, EFI_EAP_TYPE_ATTRIBUTE, EfiEapConfigEapSupportedAuthMethod,\r
                        (VOID *) AuthMethodList, &AuthMethodListSize);\r
  if (EFI_ERROR (Status)) {\r
    FreePool (AuthMethodList);\r
    return Status;\r
  }\r
\r
  //\r
  //Check if EapAuthMethod is in supported Auth Method list, if found, skip the loop.\r
  //\r
  for (Index = 0; Index < AuthMethodListSize / sizeof (EFI_EAP_TYPE); Index ++) {\r
    if (EapAuthMethod == AuthMethodList[Index]) {\r
      break;\r
    }\r
  }\r
  if (Index == AuthMethodListSize / sizeof (EFI_EAP_TYPE)) {\r
    FreePool (AuthMethodList);\r
    return EFI_UNSUPPORTED;\r
  }\r
  FreePool (AuthMethodList);\r
\r
  //\r
  // Set Identity to Eap peer, Mandatory field for PEAP and TTLS\r
  //\r
  if (StrLen (Profile->EapIdentity) > 0) {\r
\r
    IdentitySize = sizeof(CHAR8) * (StrLen(Profile->EapIdentity) + 1);\r
    Identity = AllocateZeroPool (IdentitySize);\r
    if (Identity == NULL) {\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
    UnicodeStrToAsciiStrS(Profile->EapIdentity, Identity, IdentitySize);\r
    Status = EapConfig->SetData (EapConfig, EFI_EAP_TYPE_IDENTITY, EfiEapConfigIdentityString,\r
                          (VOID *) Identity, IdentitySize - 1);\r
    if (EFI_ERROR(Status)) {\r
      FreePool (Identity);\r
      return Status;\r
    }\r
    FreePool (Identity);\r
  } else {\r
    if (EapAuthMethod != EFI_EAP_TYPE_EAPTLS) {\r
      return EFI_INVALID_PARAMETER;\r
    }\r
  }\r
\r
  //\r
  //Set Auth Method to Eap peer, Mandatory field\r
  //\r
  Status = EapConfig->SetData (EapConfig, EFI_EAP_TYPE_ATTRIBUTE, EfiEapConfigEapAuthMethod,\r
                        (VOID *) &EapAuthMethod, sizeof (EapAuthMethod));\r
  if (EFI_ERROR(Status)) {\r
    return Status;\r
  }\r
\r
  if (EapAuthMethod == EFI_EAP_TYPE_TTLS || EapAuthMethod == EFI_EAP_TYPE_PEAP) {\r
\r
    Status = EapConfig->SetData (EapConfig, EapAuthMethod, EfiEapConfigEap2ndAuthMethod,\r
                          (VOID *) &EapSecondAuthMethod, sizeof (EapSecondAuthMethod));\r
    if (EFI_ERROR(Status)) {\r
      return Status;\r
    }\r
\r
    //\r
    // Set Password to Eap peer\r
    //\r
    if (StrLen (Profile->EapPassword) < PASSWORD_MIN_LEN) {\r
\r
      DEBUG ((DEBUG_ERROR, "[WiFi Connection Manager] Error: No Eap Password for Network: %s.\n", Profile->SSId));\r
      return EFI_INVALID_PARAMETER;\r
    }\r
\r
    PasswordSize = sizeof (CHAR16) * (StrLen (Profile->EapPassword) + 1);\r
    Password = AllocateZeroPool (PasswordSize);\r
    if (Password == NULL) {\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
    StrCpyS (Password, PasswordSize, Profile->EapPassword);;\r
    Status = EapConfig->SetData (EapConfig, EFI_EAP_TYPE_MSCHAPV2, EfiEapConfigEapMSChapV2Password,\r
               (VOID *) Password, PasswordSize);\r
    ZeroMem (Password, PasswordSize);\r
    FreePool (Password);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
\r
    //\r
    //If CA cert is required, set it to Eap peer\r
    //\r
    if (Profile->CACertData != NULL) {\r
\r
      Status = EapConfig->SetData (EapConfig, EapAuthMethod, EfiEapConfigEapTlsCACert,\r
                 Profile->CACertData, Profile->CACertSize);\r
      if (EFI_ERROR(Status)) {\r
        return Status;\r
      }\r
    } else {\r
      return EFI_INVALID_PARAMETER;\r
    }\r
  } else if (EapAuthMethod == EFI_EAP_TYPE_EAPTLS) {\r
\r
    //\r
    //Set CA cert to Eap peer\r
    //\r
    if (Profile->CACertData == NULL) {\r
      return EFI_INVALID_PARAMETER;\r
    }\r
    Status = EapConfig->SetData (EapConfig, EFI_EAP_TYPE_EAPTLS, EfiEapConfigEapTlsCACert,\r
               Profile->CACertData, Profile->CACertSize);\r
    if (EFI_ERROR(Status)) {\r
      return Status;\r
    }\r
\r
    //\r
    //Set Client cert to Eap peer\r
    //\r
    if (Profile->ClientCertData == NULL) {\r
      return EFI_INVALID_PARAMETER;\r
    }\r
    Status = EapConfig->SetData (EapConfig, EFI_EAP_TYPE_EAPTLS, EfiEapConfigEapTlsClientCert,\r
               Profile->ClientCertData, Profile->ClientCertSize);\r
    if (EFI_ERROR(Status)) {\r
      return Status;\r
    }\r
\r
    //\r
    //Set Private key to Eap peer\r
    //\r
    if (Profile->PrivateKeyData == NULL) {\r
\r
      DEBUG ((DEBUG_ERROR, "[WiFi Connection Manager]  Error: No Private Key for Network: %s.\n", Profile->SSId));\r
      return EFI_INVALID_PARAMETER;\r
    }\r
\r
    Status = EapConfig->SetData (EapConfig, EFI_EAP_TYPE_EAPTLS, EfiEapConfigEapTlsClientPrivateKeyFile,\r
               Profile->PrivateKeyData, Profile->PrivateKeyDataSize);\r
    if (EFI_ERROR(Status)) {\r
      return Status;\r
    }\r
\r
    if (StrLen (Profile->PrivateKeyPassword) > 0) {\r
\r
      EncryptPasswordLen = StrLen (Profile->PrivateKeyPassword);\r
      AsciiEncryptPassword = AllocateZeroPool(EncryptPasswordLen + 1);\r
      if (AsciiEncryptPassword == NULL) {\r
        return EFI_OUT_OF_RESOURCES;\r
      }\r
      UnicodeStrToAsciiStrS(Profile->PrivateKeyPassword, AsciiEncryptPassword, EncryptPasswordLen + 1);\r
      Status = EapConfig->SetData(EapConfig, EFI_EAP_TYPE_EAPTLS,\r
                                    EfiEapConfigEapTlsClientPrivateKeyFilePassword,\r
                                    (VOID *) AsciiEncryptPassword, EncryptPasswordLen + 1);\r
      if (EFI_ERROR(Status)) {\r
\r
        ZeroMem (AsciiEncryptPassword, EncryptPasswordLen + 1);\r
        FreePool (AsciiEncryptPassword);\r
        return Status;\r
      }\r
\r
      ZeroMem (AsciiEncryptPassword, EncryptPasswordLen + 1);\r
      FreePool (AsciiEncryptPassword);\r
    }\r
  } else {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Get current link state from low layer.\r
\r
  @param[in]   Nic                Pointer to the device data of the selected NIC.\r
  @param[out]  LinkState          The pointer to buffer to retrieve link state.\r
\r
  @retval EFI_SUCCESS             The operation is completed.\r
  @retval EFI_INVALID_PARAMETER   One or more parameters are invalid.\r
  @retval EFI_UNSUPPORTED         Adapter information protocol is not supported.\r
  @retval Other Errors            Returned errors when retrieving link state from low layer.\r
\r
**/\r
EFI_STATUS\r
WifiMgrGetLinkState (\r
  IN   WIFI_MGR_DEVICE_DATA            *Nic,\r
  OUT  EFI_ADAPTER_INFO_MEDIA_STATE    *LinkState\r
  )\r
{\r
  EFI_STATUS                           Status;\r
  EFI_TPL                              OldTpl;\r
  UINTN                                DataSize;\r
  EFI_ADAPTER_INFO_MEDIA_STATE         *UndiState;\r
  EFI_ADAPTER_INFORMATION_PROTOCOL     *Aip;\r
\r
  if (Nic == NULL || LinkState == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  OldTpl = gBS->RaiseTPL (TPL_CALLBACK);\r
  Status = gBS->OpenProtocol (\r
                  Nic->ControllerHandle,\r
                  &gEfiAdapterInformationProtocolGuid,\r
                  (VOID**) &Aip,\r
                  Nic->DriverHandle,\r
                  Nic->ControllerHandle,\r
                  EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    gBS->RestoreTPL (OldTpl);\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  Status = Aip->GetInformation(\r
                  Aip,\r
                  &gEfiAdapterInfoMediaStateGuid,\r
                  (VOID **) &UndiState,\r
                  &DataSize\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    gBS->RestoreTPL (OldTpl);\r
    return Status;\r
  }\r
  gBS->RestoreTPL (OldTpl);\r
\r
  CopyMem (LinkState, UndiState, sizeof (EFI_ADAPTER_INFO_MEDIA_STATE));\r
  FreePool (UndiState);\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Prepare configuration work before connecting to the target network.\r
  For WPA2 Personal networks, password should be checked; and for EAP networks, parameters\r
  are different for different networks.\r
\r
  @param[in]  Nic                 Pointer to the device data of the selected NIC.\r
  @param[in]  Profile             The target network to be connected.\r
\r
  @retval EFI_SUCCESS             The operation is completed.\r
  @retval EFI_UNSUPPORTED         This network is not supported.\r
  @retval EFI_INVALID_PARAMETER   One or more parameters are invalid.\r
\r
**/\r
EFI_STATUS\r
WifiMgrPrepareConnection (\r
  IN    WIFI_MGR_DEVICE_DATA              *Nic,\r
  IN    WIFI_MGR_NETWORK_PROFILE          *Profile\r
  )\r
{\r
  EFI_STATUS           Status;\r
  UINT8                SecurityType;\r
  BOOLEAN              AKMSuiteSupported;\r
  BOOLEAN              CipherSuiteSupported;\r
\r
  if (Profile == NULL || Nic == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Status = WifiMgrCheckRSN (Profile->Network.AKMSuite, Profile->Network.CipherSuite,\r
             Nic, &SecurityType, &AKMSuiteSupported, &CipherSuiteSupported);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  if (AKMSuiteSupported && CipherSuiteSupported) {\r
    switch (SecurityType) {\r
      case SECURITY_TYPE_WPA2_PERSONAL:\r
\r
        Status = WifiMgrConfigPassword (Nic, Profile);\r
        if (EFI_ERROR (Status)) {\r
          if (Status == EFI_NOT_FOUND) {\r
            if (Nic->OneTimeConnectRequest) {\r
              WifiMgrUpdateConnectMessage (Nic, FALSE, L"Connect Failed: Invalid Password!");\r
            }\r
          }\r
          return Status;\r
        }\r
        break;\r
\r
      case SECURITY_TYPE_WPA2_ENTERPRISE:\r
\r
        Status = WifiMgrConfigEap (Nic, Profile);\r
        if (EFI_ERROR (Status)) {\r
          if (Status == EFI_INVALID_PARAMETER) {\r
            if (Nic->OneTimeConnectRequest) {\r
              WifiMgrUpdateConnectMessage (Nic, FALSE, L"Connect Failed: Invalid Configuration!");\r
            }\r
          }\r
          return Status;\r
        }\r
        break;\r
\r
      case SECURITY_TYPE_NONE:\r
        break;\r
\r
      default:\r
        return EFI_UNSUPPORTED;\r
    }\r
  } else {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  The callback function for connect operation.\r
\r
  ASSERT when errors occur in config token.\r
\r
  @param[in]  Event                 The Connect token receive event.\r
  @param[in]  Context               The context of the connect token.\r
\r
**/\r
VOID\r
EFIAPI\r
WifiMgrOnConnectFinished (\r
  IN  EFI_EVENT              Event,\r
  IN  VOID                   *Context\r
  )\r
{\r
  EFI_STATUS                         Status;\r
  WIFI_MGR_MAC_CONFIG_TOKEN          *ConfigToken;\r
  WIFI_MGR_NETWORK_PROFILE           *ConnectedProfile;\r
  UINT8                              SecurityType;\r
  UINT8                              SSIdLen;\r
  CHAR8                              *AsciiSSId;\r
\r
  ASSERT (Context != NULL);\r
\r
  ConnectedProfile = NULL;\r
  ConfigToken = (WIFI_MGR_MAC_CONFIG_TOKEN*) Context;\r
  ASSERT (ConfigToken->Nic != NULL);\r
\r
  ConfigToken->Nic->ConnectState = WifiMgrDisconnected;\r
  ASSERT (ConfigToken->Type == TokenTypeConnectNetworkToken);\r
\r
  ASSERT (ConfigToken->Token.ConnectNetworkToken != NULL);\r
  if (ConfigToken->Token.ConnectNetworkToken->Status != EFI_SUCCESS) {\r
\r
    if (ConfigToken->Nic->OneTimeConnectRequest) {\r
      //\r
      // Only update message for user triggered connection\r
      //\r
      if (ConfigToken->Token.ConnectNetworkToken->Status == EFI_ACCESS_DENIED) {\r
\r
        WifiMgrUpdateConnectMessage (ConfigToken->Nic, FALSE, L"Connect Failed: Permission Denied!");\r
      } else {\r
        WifiMgrUpdateConnectMessage (ConfigToken->Nic, FALSE, L"Connect Failed!");\r
      }\r
      ConfigToken->Nic->OneTimeConnectRequest = FALSE;\r
    }\r
    ConfigToken->Nic->CurrentOperateNetwork = NULL;\r
    return;\r
  }\r
\r
  if (ConfigToken->Token.ConnectNetworkToken->ResultCode != ConnectSuccess) {\r
\r
    if (ConfigToken->Nic->OneTimeConnectRequest) {\r
\r
      if (ConfigToken->Token.ConnectNetworkToken->ResultCode == ConnectFailedReasonUnspecified) {\r
        WifiMgrUpdateConnectMessage (ConfigToken->Nic, FALSE, L"Connect Failed: Wrong Password or Unexpected Error!");\r
      } else {\r
        WifiMgrUpdateConnectMessage (ConfigToken->Nic, FALSE, L"Connect Failed!");\r
      }\r
    }\r
    goto Exit;\r
  }\r
\r
  if (ConfigToken->Token.ConnectNetworkToken->Data == NULL ||\r
    ConfigToken->Token.ConnectNetworkToken->Data->Network == NULL) {\r
\r
    //\r
    // An unexpected error occurs, tell low layer to perform a disconnect\r
    //\r
    ConfigToken->Nic->HasDisconnectPendingNetwork = TRUE;\r
    WifiMgrUpdateConnectMessage (ConfigToken->Nic, FALSE, NULL);\r
    goto Exit;\r
  }\r
\r
  //\r
  // A correct connect token received, terminate the connection process\r
  //\r
  Status = WifiMgrCheckRSN(ConfigToken->Token.ConnectNetworkToken->Data->Network->AKMSuite,\r
             ConfigToken->Token.ConnectNetworkToken->Data->Network->CipherSuite,\r
             ConfigToken->Nic, &SecurityType, NULL, NULL);\r
  if (EFI_ERROR(Status)) {\r
    SecurityType = SECURITY_TYPE_UNKNOWN;\r
  }\r
\r
  SSIdLen   = ConfigToken->Token.ConnectNetworkToken->Data->Network->SSId.SSIdLen;\r
  AsciiSSId = (CHAR8*) AllocateZeroPool(sizeof (CHAR8) * (SSIdLen + 1));\r
  if (AsciiSSId == NULL) {\r
    ConfigToken->Nic->HasDisconnectPendingNetwork = TRUE;\r
    WifiMgrUpdateConnectMessage (ConfigToken->Nic, FALSE, NULL);\r
    goto Exit;\r
  }\r
\r
  CopyMem(AsciiSSId, ConfigToken->Token.ConnectNetworkToken->Data->Network->SSId.SSId, SSIdLen);\r
  *(AsciiSSId + SSIdLen) = '\0';\r
\r
  ConnectedProfile = WifiMgrGetProfileByAsciiSSId(AsciiSSId, SecurityType, &ConfigToken->Nic->ProfileList);\r
  FreePool(AsciiSSId);\r
  if (ConnectedProfile == NULL) {\r
    ConfigToken->Nic->HasDisconnectPendingNetwork = TRUE;\r
    WifiMgrUpdateConnectMessage (ConfigToken->Nic, FALSE, NULL);\r
    goto Exit;\r
  }\r
\r
  ConfigToken->Nic->ConnectState = WifiMgrConnectedToAp;\r
  WifiMgrUpdateConnectMessage (ConfigToken->Nic, TRUE, NULL);\r
\r
Exit:\r
\r
  if (ConfigToken->Nic->ConnectState == WifiMgrDisconnected) {\r
    ConfigToken->Nic->CurrentOperateNetwork = NULL;\r
  }\r
  ConfigToken->Nic->OneTimeConnectRequest = FALSE;\r
  WifiMgrFreeToken(ConfigToken);\r
}\r
\r
/**\r
  Start connect operation, and send out a token to connect to a target network.\r
\r
  @param[in]  Nic                 Pointer to the device data of the selected NIC.\r
  @param[in]  Profile             The target network to be connected.\r
\r
  @retval EFI_SUCCESS             The operation is completed.\r
  @retval EFI_ALREADY_STARTED     Already in "connected" state, need to perform a disconnect\r
                                  operation first.\r
  @retval EFI_INVALID_PARAMETER   One or more parameters are invalid.\r
  @retval EFI_OUT_OF_RESOURCES    Failed to allocate memory.\r
  @retval Other Errors            Return errors when connecting network on low layer.\r
\r
**/\r
EFI_STATUS\r
WifiMgrConnectToNetwork (\r
  IN    WIFI_MGR_DEVICE_DATA              *Nic,\r
  IN    WIFI_MGR_NETWORK_PROFILE          *Profile\r
  )\r
{\r
  EFI_STATUS                             Status;\r
  EFI_TPL                                OldTpl;\r
  EFI_ADAPTER_INFO_MEDIA_STATE           LinkState;\r
  WIFI_MGR_MAC_CONFIG_TOKEN              *ConfigToken;\r
  EFI_80211_CONNECT_NETWORK_TOKEN        *ConnectToken;\r
\r
  if (Nic == NULL || Nic->Wmp == NULL || Profile == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Status = WifiMgrGetLinkState (Nic, &LinkState);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
  if (LinkState.MediaState == EFI_SUCCESS) {\r
    return EFI_ALREADY_STARTED;\r
  }\r
\r
  OldTpl = gBS->RaiseTPL (TPL_CALLBACK);\r
  Status = WifiMgrPrepareConnection (Nic, Profile);\r
  if (EFI_ERROR (Status)) {\r
    gBS->RestoreTPL (OldTpl);\r
    return Status;\r
  }\r
\r
  //\r
  // Create a new connect token\r
  //\r
  ConfigToken = AllocateZeroPool (sizeof (WIFI_MGR_MAC_CONFIG_TOKEN));\r
  if (ConfigToken == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto Exit;\r
  }\r
\r
  ConfigToken->Type      = TokenTypeConnectNetworkToken;\r
  ConfigToken->Nic       = Nic;\r
  ConfigToken->Token.ConnectNetworkToken  = AllocateZeroPool (sizeof (EFI_80211_CONNECT_NETWORK_TOKEN));\r
  if (ConfigToken->Token.ConnectNetworkToken == NULL) {\r
    goto Exit;\r
  }\r
\r
  ConnectToken           = ConfigToken->Token.ConnectNetworkToken;\r
  ConnectToken->Data     = AllocateZeroPool (sizeof (EFI_80211_CONNECT_NETWORK_DATA));\r
  if (ConnectToken->Data == NULL) {\r
    goto Exit;\r
  }\r
\r
  ConnectToken->Data->Network = AllocateZeroPool (sizeof (EFI_80211_NETWORK));\r
  if (ConnectToken->Data->Network == NULL) {\r
    goto Exit;\r
  }\r
  CopyMem(ConnectToken->Data->Network, &Profile->Network, sizeof (EFI_80211_NETWORK));\r
\r
  //\r
  // Add event handle and start to connect\r
  //\r
  Status = gBS->CreateEvent (\r
                  EVT_NOTIFY_SIGNAL,\r
                  TPL_CALLBACK,\r
                  WifiMgrOnConnectFinished,\r
                  ConfigToken,\r
                  &ConnectToken->Event\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    goto Exit;\r
  }\r
\r
  Nic->ConnectState = WifiMgrConnectingToAp;\r
  Nic->CurrentOperateNetwork = Profile;\r
  WifiMgrUpdateConnectMessage (Nic, FALSE, NULL);\r
\r
  //\r
  //Start Connecting ...\r
  //\r
  Status = Nic->Wmp->ConnectNetwork (Nic->Wmp, ConnectToken);\r
\r
  //\r
  // Erase secrets after connection is triggered\r
  //\r
  WifiMgrCleanProfileSecrets (Profile);\r
\r
  if (EFI_ERROR (Status)) {\r
    if (Status == EFI_ALREADY_STARTED) {\r
      Nic->ConnectState = WifiMgrConnectedToAp;\r
      WifiMgrUpdateConnectMessage (Nic, TRUE, NULL);\r
    } else {\r
\r
      Nic->ConnectState          = WifiMgrDisconnected;\r
      Nic->CurrentOperateNetwork = NULL;\r
\r
      if (Nic->OneTimeConnectRequest) {\r
        if (Status == EFI_NOT_FOUND) {\r
          WifiMgrUpdateConnectMessage (Nic, FALSE, L"Connect Failed: Not Available!");\r
        } else {\r
          WifiMgrUpdateConnectMessage (Nic, FALSE, L"Connect Failed: Unexpected Error!");\r
        }\r
      }\r
    }\r
    goto Exit;\r
  }\r
\r
Exit:\r
\r
  if (EFI_ERROR (Status)) {\r
    WifiMgrFreeToken (ConfigToken);\r
  }\r
  gBS->RestoreTPL (OldTpl);\r
\r
  DEBUG ((DEBUG_INFO, "[WiFi Connection Manager] WifiMgrConnectToNetwork: %r\n", Status));\r
  return Status;\r
}\r
\r
/**\r
  The callback function for disconnect operation.\r
\r
  ASSERT when errors occur in config token.\r
\r
  @param[in]  Event                 The Disconnect token receive event.\r
  @param[in]  Context               The context of the Disconnect token.\r
\r
**/\r
VOID\r
EFIAPI\r
WifiMgrOnDisconnectFinished (\r
  IN EFI_EVENT              Event,\r
  IN VOID                   *Context\r
  )\r
{\r
  WIFI_MGR_MAC_CONFIG_TOKEN         *ConfigToken;\r
\r
  ASSERT (Context != NULL);\r
\r
  ConfigToken = (WIFI_MGR_MAC_CONFIG_TOKEN*) Context;\r
  ASSERT (ConfigToken->Nic != NULL);\r
  ASSERT (ConfigToken->Type == TokenTypeDisconnectNetworkToken);\r
\r
  ASSERT (ConfigToken->Token.DisconnectNetworkToken != NULL);\r
  if (ConfigToken->Token.DisconnectNetworkToken->Status != EFI_SUCCESS) {\r
    ConfigToken->Nic->ConnectState          = WifiMgrConnectedToAp;\r
    WifiMgrUpdateConnectMessage (ConfigToken->Nic, FALSE, NULL);\r
    ConfigToken->Nic->OneTimeDisconnectRequest = FALSE;\r
    goto Exit;\r
  }\r
\r
  ConfigToken->Nic->ConnectState          = WifiMgrDisconnected;\r
  ConfigToken->Nic->CurrentOperateNetwork = NULL;\r
  WifiMgrUpdateConnectMessage (ConfigToken->Nic, TRUE, NULL);\r
  ConfigToken->Nic->OneTimeDisconnectRequest = FALSE;\r
\r
  //\r
  // Disconnected network may not be in network list now, trigger a scan again!\r
  //\r
  ConfigToken->Nic->OneTimeScanRequest       = TRUE;\r
\r
  Exit:\r
    WifiMgrFreeToken(ConfigToken);\r
    return;\r
}\r
\r
/**\r
  Start disconnect operation, and send out a token to disconnect from current connected\r
  network.\r
\r
  @param[in]  Nic                 Pointer to the device data of the selected NIC.\r
\r
  @retval EFI_SUCCESS             The operation is completed.\r
  @retval EFI_OUT_OF_RESOURCES    Failed to allocate memory.\r
  @retval EFI_INVALID_PARAMETER   One or more parameters are invalid.\r
  @retval Other Errors            Return errors when disconnecting a network on low layer.\r
\r
**/\r
EFI_STATUS\r
WifiMgrDisconnectToNetwork (\r
  IN    WIFI_MGR_DEVICE_DATA             *Nic\r
  )\r
{\r
  EFI_STATUS                             Status;\r
  EFI_TPL                                OldTpl;\r
  WIFI_MGR_MAC_CONFIG_TOKEN              *ConfigToken;\r
  EFI_80211_DISCONNECT_NETWORK_TOKEN     *DisconnectToken;\r
\r
  if (Nic == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  OldTpl      = gBS->RaiseTPL (TPL_CALLBACK);\r
  Status      = EFI_SUCCESS;\r
  ConfigToken = AllocateZeroPool (sizeof (WIFI_MGR_MAC_CONFIG_TOKEN));\r
  if (ConfigToken == NULL) {\r
    gBS->RestoreTPL (OldTpl);\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  ConfigToken->Type      = TokenTypeDisconnectNetworkToken;\r
  ConfigToken->Nic       = Nic;\r
  ConfigToken->Token.DisconnectNetworkToken = AllocateZeroPool (sizeof (EFI_80211_DISCONNECT_NETWORK_TOKEN));\r
  if (ConfigToken->Token.DisconnectNetworkToken == NULL) {\r
    WifiMgrFreeToken(ConfigToken);\r
    gBS->RestoreTPL (OldTpl);\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  DisconnectToken = ConfigToken->Token.DisconnectNetworkToken;\r
\r
  Status = gBS->CreateEvent (\r
                  EVT_NOTIFY_SIGNAL,\r
                  TPL_CALLBACK,\r
                  WifiMgrOnDisconnectFinished,\r
                  ConfigToken,\r
                  &DisconnectToken->Event\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    WifiMgrFreeToken(ConfigToken);\r
    gBS->RestoreTPL (OldTpl);\r
    return Status;\r
  }\r
\r
  Nic->ConnectState = WifiMgrDisconnectingToAp;\r
  WifiMgrUpdateConnectMessage (ConfigToken->Nic, FALSE, NULL);\r
\r
  Status = Nic->Wmp->DisconnectNetwork (Nic->Wmp, DisconnectToken);\r
  if (EFI_ERROR (Status)) {\r
    if (Status == EFI_NOT_FOUND) {\r
\r
      Nic->ConnectState          = WifiMgrDisconnected;\r
      Nic->CurrentOperateNetwork = NULL;\r
\r
      //\r
      // This network is not in network list now, trigger a scan again!\r
      //\r
      Nic->OneTimeScanRequest    = TRUE;\r
\r
      //\r
      // State has been changed from Connected to Disconnected\r
      //\r
      WifiMgrUpdateConnectMessage (ConfigToken->Nic, TRUE, NULL);\r
      Status                     = EFI_SUCCESS;\r
    } else {\r
      if (Nic->OneTimeDisconnectRequest) {\r
\r
        WifiMgrUpdateConnectMessage (ConfigToken->Nic, FALSE, L"Disconnect Failed: Unexpected Error!");\r
      }\r
\r
      Nic->ConnectState     = WifiMgrConnectedToAp;\r
      WifiMgrUpdateConnectMessage (ConfigToken->Nic, FALSE, NULL);\r
    }\r
    WifiMgrFreeToken(ConfigToken);\r
  }\r
\r
  gBS->RestoreTPL (OldTpl);\r
  return Status;\r
}\r
\r
/**\r
  The state machine of the connection manager, periodically check the state and\r
  perform a corresponding operation.\r
\r
  @param[in]  Event                   The timer event to be triggered.\r
  @param[in]  Context                 The context of the Nic device data.\r
\r
**/\r
VOID\r
EFIAPI\r
WifiMgrOnTimerTick (\r
  IN EFI_EVENT                        Event,\r
  IN VOID                             *Context\r
  )\r
{\r
  WIFI_MGR_DEVICE_DATA                *Nic;\r
  EFI_STATUS                          Status;\r
  EFI_ADAPTER_INFO_MEDIA_STATE        LinkState;\r
  WIFI_MGR_NETWORK_PROFILE            *Profile;\r
\r
  if (Context == NULL) {\r
    return;\r
  }\r
\r
  Nic = (WIFI_MGR_DEVICE_DATA*) Context;\r
  NET_CHECK_SIGNATURE (Nic, WIFI_MGR_DEVICE_DATA_SIGNATURE);\r
\r
  Status = WifiMgrGetLinkState (Nic, &LinkState);\r
  if (EFI_ERROR (Status)) {\r
    DEBUG ((DEBUG_INFO, "[WiFi Connection Manager] Error: Failed to get link state!\n"));\r
    return;\r
  }\r
\r
  if (Nic->LastLinkState.MediaState != LinkState.MediaState) {\r
    if (Nic->LastLinkState.MediaState == EFI_SUCCESS && LinkState.MediaState == EFI_NO_MEDIA) {\r
      Nic->HasDisconnectPendingNetwork = TRUE;\r
    }\r
    Nic->LastLinkState.MediaState = LinkState.MediaState;\r
  }\r
\r
  Nic->ScanTickTime ++;\r
  if ((Nic->ScanTickTime > WIFI_SCAN_FREQUENCY || Nic->OneTimeScanRequest) &&\r
    Nic->ScanState == WifiMgrScanFinished) {\r
\r
    Nic->OneTimeScanRequest = FALSE;\r
    Nic->ScanTickTime = 0;\r
\r
    DEBUG ((DEBUG_INFO, "[WiFi Connection Manager] Scan is triggered.\n"));\r
    WifiMgrStartScan (Nic);\r
  }\r
\r
  if (Nic->AvailableCount > 0 && Nic->ScanState == WifiMgrScanFinished) {\r
\r
    switch (Nic->ConnectState) {\r
    case WifiMgrDisconnected:\r
\r
      if (Nic->HasDisconnectPendingNetwork) {\r
        Nic->HasDisconnectPendingNetwork = FALSE;\r
      }\r
\r
      if (Nic->ConnectPendingNetwork != NULL) {\r
\r
        Profile   = Nic->ConnectPendingNetwork;\r
        Status    = WifiMgrConnectToNetwork(Nic, Profile);\r
        Nic->ConnectPendingNetwork = NULL;\r
        if (EFI_ERROR (Status)) {\r
          //\r
          // Some error happened, don't wait for a return connect token!\r
          //\r
          Nic->OneTimeConnectRequest = FALSE;\r
        }\r
      }\r
      break;\r
\r
    case WifiMgrConnectingToAp:\r
      break;\r
\r
    case WifiMgrDisconnectingToAp:\r
      break;\r
\r
    case WifiMgrConnectedToAp:\r
\r
      if (Nic->ConnectPendingNetwork != NULL || Nic->HasDisconnectPendingNetwork) {\r
\r
        Status    = WifiMgrDisconnectToNetwork(Nic);\r
        if (EFI_ERROR (Status)) {\r
          //\r
          // Some error happened, don't wait for a return disconnect token!\r
          //\r
          Nic->OneTimeDisconnectRequest = FALSE;\r
        }\r
      }\r
      break;\r
\r
    default:\r
      break;\r
    }\r
  }\r
}\r