[mirror_edk2.git] / MdeModulePkg / Universal / Network / IScsiDxe / IScsiConfig.c

/*++\r
\r
Copyright (c) 2004 - 2008, Intel Corporation\r
All rights reserved. This program and the accompanying materials\r
are licensed and made available under the terms and conditions of the BSD License\r
which accompanies this distribution.  The full text of the license may be found at\r
http://opensource.org/licenses/bsd-license.php\r
\r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
Module Name:\r
\r
  IScsiConfig.c\r
\r
Abstract:\r
\r
--*/\r
\r
#include "IScsiImpl.h"\r
\r
EFI_GUID        mVendorGuid              = ISCSI_CONFIG_GUID;\r
BOOLEAN         mIScsiDeviceListUpdated  = FALSE;\r
UINTN           mNumberOfIScsiDevices    = 0;\r
ISCSI_FORM_CALLBACK_INFO  *mCallbackInfo = NULL;\r
\r
LIST_ENTRY      mIScsiConfigFormList = {\r
  &mIScsiConfigFormList,\r
  &mIScsiConfigFormList\r
};\r
\r
STATIC\r
VOID\r
IScsiIpToStr (\r
  IN  EFI_IPv4_ADDRESS  *Ip,\r
  OUT CHAR16            *Str\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Convert the IPv4 address into a dotted string.\r
\r
Arguments:\r
\r
  Ip  - The IPv4 address.\r
  Str - The dotted IP string.\r
\r
Returns:\r
\r
  None.\r
\r
--*/\r
{\r
  UnicodeSPrint ( Str, 2 * IP4_STR_MAX_SIZE, L"%d.%d.%d.%d", Ip->Addr[0], Ip->Addr[1], Ip->Addr[2], Ip->Addr[3]);\r
}\r
\r
VOID\r
PopUpInvalidNotify (\r
  IN CHAR16 *Warning\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Pop up an invalid notify which displays the message in Warning.\r
\r
Arguments:\r
\r
  Warning - The warning message.\r
\r
Returns:\r
\r
  None.\r
\r
--*/\r
{\r
  EFI_INPUT_KEY             Key;\r
\r
  IfrLibCreatePopUp (1, &Key, Warning);\r
}\r
\r
EFI_STATUS\r
IScsiUpdateDeviceList (\r
  VOID\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Update the list of iSCSI devices the iSCSI driver is controlling.\r
\r
Arguments:\r
\r
  None.\r
\r
Returns:\r
\r
  None.\r
\r
--*/\r
{\r
  EFI_STATUS                  Status;\r
  ISCSI_DEVICE_LIST           *DeviceList;\r
  UINTN                       DataSize;\r
  UINTN                       NumHandles;\r
  EFI_HANDLE                  *Handles;\r
  UINTN                       HandleIndex;\r
  UINTN                       Index;\r
  UINTN                       LastDeviceIndex;\r
  EFI_SIMPLE_NETWORK_PROTOCOL *Snp;\r
  EFI_SIMPLE_NETWORK_MODE     *Mode;\r
  ISCSI_MAC_INFO              *CurMacInfo;\r
  ISCSI_MAC_INFO              TempMacInfo;\r
  CHAR16                      MacString[65];\r
  UINTN                       DeviceListSize;\r
\r
  //\r
  // Dump all the handles the Simple Network Protocol is installed on.\r
  //\r
  Status = gBS->LocateHandleBuffer (\r
                  ByProtocol,\r
                  &gEfiSimpleNetworkProtocolGuid,\r
                  NULL,\r
                  &NumHandles,\r
                  &Handles\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  DataSize = 0;\r
  Status = gRT->GetVariable (\r
                  L"iSCSIDeviceList",\r
                  &mVendorGuid,\r
                  NULL,\r
                  &DataSize,\r
                  NULL\r
                  );\r
  if (Status == EFI_BUFFER_TOO_SMALL) {\r
    DeviceList = (ISCSI_DEVICE_LIST *) AllocatePool (DataSize);\r
\r
    gRT->GetVariable (\r
          L"iSCSIDeviceList",\r
          &mVendorGuid,\r
          NULL,\r
          &DataSize,\r
          DeviceList\r
          );\r
\r
    LastDeviceIndex = 0;\r
\r
    for (HandleIndex = 0; HandleIndex < NumHandles; HandleIndex++) {\r
      gBS->HandleProtocol (Handles[HandleIndex], &gEfiSimpleNetworkProtocolGuid, (VOID **)&Snp);\r
\r
      Mode = Snp->Mode;\r
\r
      for (Index = LastDeviceIndex; Index < DeviceList->NumDevice; Index++) {\r
        CurMacInfo = &DeviceList->MacInfo[Index];\r
        if ((CurMacInfo->Len == Mode->HwAddressSize) &&\r
            (NET_MAC_EQUAL (&CurMacInfo->Mac, &Mode->PermanentAddress, Mode->HwAddressSize))\r
            ) {\r
          //\r
          // The previous configured NIC is still here.\r
          //\r
          if (Index != LastDeviceIndex) {\r
            //\r
            // Swap the current MAC address entry with the one indexed by\r
            // LastDeviceIndex.\r
            //\r
            CopyMem (&TempMacInfo, CurMacInfo, sizeof (ISCSI_MAC_INFO));\r
            CopyMem (CurMacInfo, &DeviceList->MacInfo[LastDeviceIndex], sizeof (ISCSI_MAC_INFO));\r
            CopyMem (&DeviceList->MacInfo[LastDeviceIndex], &TempMacInfo, sizeof (ISCSI_MAC_INFO));\r
          }\r
\r
          LastDeviceIndex++;\r
        }\r
      }\r
\r
      if (LastDeviceIndex == DeviceList->NumDevice) {\r
        break;\r
      }\r
    }\r
\r
    for (Index = LastDeviceIndex; Index < DeviceList->NumDevice; Index++) {\r
      //\r
      // delete the variables\r
      //\r
      CurMacInfo = &DeviceList->MacInfo[Index];\r
      IScsiMacAddrToStr (&CurMacInfo->Mac, CurMacInfo->Len, MacString);\r
      gRT->SetVariable (MacString, &gEfiIScsiInitiatorNameProtocolGuid, 0, 0, NULL);\r
      gRT->SetVariable (MacString, &mIScsiCHAPAuthInfoGuid, 0, 0, NULL);\r
    }\r
\r
    gBS->FreePool (DeviceList);\r
  } else if (Status != EFI_NOT_FOUND) {\r
    gBS->FreePool (Handles);\r
    return Status;\r
  }\r
  //\r
  // Construct the new iSCSI device list.\r
  //\r
  DeviceListSize        = sizeof (ISCSI_DEVICE_LIST) + (NumHandles - 1) * sizeof (ISCSI_MAC_INFO);\r
  DeviceList            = (ISCSI_DEVICE_LIST *) AllocatePool (DeviceListSize);\r
  DeviceList->NumDevice = (UINT8) NumHandles;\r
\r
  for (Index = 0; Index < NumHandles; Index++) {\r
    gBS->HandleProtocol (Handles[Index], &gEfiSimpleNetworkProtocolGuid, (VOID **)&Snp);\r
    Mode        = Snp->Mode;\r
\r
    CurMacInfo  = &DeviceList->MacInfo[Index];\r
    CopyMem (&CurMacInfo->Mac, &Mode->PermanentAddress, Mode->HwAddressSize);\r
    CurMacInfo->Len = (UINT8) Mode->HwAddressSize;\r
  }\r
\r
  gRT->SetVariable (\r
        L"iSCSIDeviceList",\r
        &mVendorGuid,\r
        ISCSI_CONFIG_VAR_ATTR,\r
        DeviceListSize,\r
        DeviceList\r
        );\r
\r
  gBS->FreePool (DeviceList);\r
\r
  return Status;\r
}\r
\r
STATIC\r
ISCSI_CONFIG_FORM_ENTRY *\r
IScsiGetConfigFormEntryByIndex (\r
  IN UINT32 Index\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Get the iSCSI configuration form entry by the index of the goto opcode actived.\r
\r
Arguments:\r
\r
  Index - The 0-based index of the goto opcode actived.\r
\r
Returns:\r
\r
  The iSCSI configuration form entry found.\r
\r
--*/\r
{\r
  UINT32                  CurrentIndex;\r
  LIST_ENTRY              *Entry;\r
  ISCSI_CONFIG_FORM_ENTRY *ConfigFormEntry;\r
\r
  CurrentIndex    = 0;\r
  ConfigFormEntry = NULL;\r
\r
  NET_LIST_FOR_EACH (Entry, &mIScsiConfigFormList) {\r
    if (CurrentIndex == Index) {\r
      ConfigFormEntry = NET_LIST_USER_STRUCT (Entry, ISCSI_CONFIG_FORM_ENTRY, Link);\r
      break;\r
    }\r
\r
    CurrentIndex++;\r
  }\r
\r
  return ConfigFormEntry;\r
}\r
\r
STATIC\r
VOID\r
IScsiConvertDeviceConfigDataToIfrNvData (\r
  IN ISCSI_CONFIG_FORM_ENTRY  *ConfigFormEntry,\r
  IN ISCSI_CONFIG_IFR_NVDATA  *IfrNvData\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Convert the iSCSI configuration data into the IFR data.\r
\r
Arguments:\r
\r
  ConfigFormEntry - The iSCSI configuration form entry.\r
  IfrNvData       - The IFR nv data.\r
\r
Returns:\r
\r
  None.\r
\r
--*/\r
{\r
  ISCSI_SESSION_CONFIG_NVDATA   *SessionConfigData;\r
  ISCSI_CHAP_AUTH_CONFIG_NVDATA *AuthConfigData;\r
\r
  //\r
  // Normal session configuration parameters.\r
  //\r
  SessionConfigData                 = &ConfigFormEntry->SessionConfigData;\r
  IfrNvData->Enabled                = SessionConfigData->Enabled;\r
\r
  IfrNvData->InitiatorInfoFromDhcp  = SessionConfigData->InitiatorInfoFromDhcp;\r
  IfrNvData->TargetInfoFromDhcp     = SessionConfigData->TargetInfoFromDhcp;\r
  IfrNvData->TargetPort             = SessionConfigData->TargetPort;\r
\r
  IScsiIpToStr (&SessionConfigData->LocalIp, IfrNvData->LocalIp);\r
  IScsiIpToStr (&SessionConfigData->SubnetMask, IfrNvData->SubnetMask);\r
  IScsiIpToStr (&SessionConfigData->Gateway, IfrNvData->Gateway);\r
  IScsiIpToStr (&SessionConfigData->TargetIp, IfrNvData->TargetIp);\r
\r
  IScsiAsciiStrToUnicodeStr (SessionConfigData->TargetName, IfrNvData->TargetName);\r
\r
  IScsiLunToUnicodeStr (SessionConfigData->BootLun, IfrNvData->BootLun);\r
\r
  //\r
  // CHAP authentication parameters.\r
  //\r
  AuthConfigData      = &ConfigFormEntry->AuthConfigData;\r
\r
  IfrNvData->CHAPType = AuthConfigData->CHAPType;\r
\r
  IScsiAsciiStrToUnicodeStr (AuthConfigData->CHAPName, IfrNvData->CHAPName);\r
  IScsiAsciiStrToUnicodeStr (AuthConfigData->CHAPSecret, IfrNvData->CHAPSecret);\r
  IScsiAsciiStrToUnicodeStr (AuthConfigData->ReverseCHAPName, IfrNvData->ReverseCHAPName);\r
  IScsiAsciiStrToUnicodeStr (AuthConfigData->ReverseCHAPSecret, IfrNvData->ReverseCHAPSecret);\r
}\r
\r
\r
EFI_STATUS\r
EFIAPI\r
IScsiFormExtractConfig (\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
\r
  Routine Description:\r
    This function allows a caller to extract the current configuration for one\r
    or more named elements from the target driver.\r
\r
  Arguments:\r
    This       - Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL.\r
    Request    - A null-terminated Unicode string in <ConfigRequest> format.\r
    Progress   - On return, points to a character in the Request string.\r
                 Points to the string's null terminator if request was successful.\r
                 Points to the most recent '&' before the first failing name/value\r
                 pair (or the beginning of the string if the failure is in the\r
                 first name/value pair) if the request was not successful.\r
    Results    - A null-terminated Unicode string in <ConfigAltResp> format which\r
                 has all values filled in for the names in the Request string.\r
                 String to be allocated by the called function.\r
\r
  Returns:\r
    EFI_SUCCESS           - The Results is filled with the requested values.\r
    EFI_OUT_OF_RESOURCES  - Not enough memory to store the results.\r
    EFI_INVALID_PARAMETER - Request is NULL, illegal syntax, or unknown name.\r
    EFI_NOT_FOUND         - Routing data doesn't match any storage in this driver.\r
\r
--*/\r
{\r
  EFI_STATUS                       Status;\r
  CHAR8                            InitiatorName[ISCSI_NAME_IFR_MAX_SIZE];\r
  UINTN                            BufferSize;\r
  ISCSI_CONFIG_IFR_NVDATA          *IfrNvData;\r
  ISCSI_FORM_CALLBACK_INFO         *Private;\r
  EFI_HII_CONFIG_ROUTING_PROTOCOL  *HiiConfigRouting;\r
\r
  if (!mIScsiDeviceListUpdated) {\r
    //\r
    // Update the device list.\r
    //\r
    IScsiUpdateDeviceList ();\r
    mIScsiDeviceListUpdated = TRUE;\r
  }\r
\r
  Private = ISCSI_FORM_CALLBACK_INFO_FROM_FORM_CALLBACK (This);\r
  IfrNvData = AllocateZeroPool (sizeof (ISCSI_CONFIG_IFR_NVDATA));\r
  ASSERT (IfrNvData != NULL);\r
  if (Private->Current != NULL) {\r
    IScsiConvertDeviceConfigDataToIfrNvData (Private->Current, IfrNvData);\r
  }\r
\r
  BufferSize  = ISCSI_NAME_IFR_MAX_SIZE;\r
  Status      = gIScsiInitiatorName.Get (&gIScsiInitiatorName, &BufferSize, InitiatorName);\r
  if (EFI_ERROR (Status)) {\r
    IfrNvData->InitiatorName[0] = L'\0';\r
  } else {\r
    IScsiAsciiStrToUnicodeStr (InitiatorName, IfrNvData->InitiatorName);\r
  }\r
\r
  //\r
  // Convert buffer data to <ConfigResp> by helper function BlockToConfig()\r
  //\r
  HiiConfigRouting = Private->ConfigRouting;\r
  BufferSize = sizeof (ISCSI_CONFIG_IFR_NVDATA);\r
  Status = HiiConfigRouting->BlockToConfig (\r
                               HiiConfigRouting,\r
                               Request,\r
                               (UINT8 *) IfrNvData,\r
                               BufferSize,\r
                               Results,\r
                               Progress\r
                               );\r
  gBS->FreePool (IfrNvData);\r
  return Status;\r
}\r
\r
EFI_STATUS\r
EFIAPI\r
IScsiFormRouteConfig (\r
  IN  CONST EFI_HII_CONFIG_ACCESS_PROTOCOL   *This,\r
  IN  CONST EFI_STRING                       Configuration,\r
  OUT EFI_STRING                             *Progress\r
  )\r
/*++\r
\r
  Routine Description:\r
    This function processes the results of changes in configuration.\r
\r
  Arguments:\r
    This          - Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL.\r
    Configuration - A null-terminated Unicode string in <ConfigResp> format.\r
    Progress      - A pointer to a string filled in with the offset of the most\r
                    recent '&' before the first failing name/value pair (or the\r
                    beginning of the string if the failure is in the first\r
                    name/value pair) or the terminating NULL if all was successful.\r
\r
  Returns:\r
    EFI_SUCCESS           - The Results is processed successfully.\r
    EFI_INVALID_PARAMETER - Configuration is NULL.\r
    EFI_NOT_FOUND         - Routing data doesn't match any storage in this driver.\r
\r
--*/\r
{\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
EFIAPI\r
IScsiFormCallback (\r
  IN  CONST EFI_HII_CONFIG_ACCESS_PROTOCOL   *This,\r
  IN  EFI_BROWSER_ACTION                     Action,\r
  IN  EFI_QUESTION_ID                        KeyValue,\r
  IN  UINT8                                  Type,\r
  IN  EFI_IFR_TYPE_VALUE                     *Value,\r
  OUT EFI_BROWSER_ACTION_REQUEST             *ActionRequest\r
  )\r
/*++\r
\r
  Routine Description:\r
    This function processes the results of changes in configuration.\r
\r
  Arguments:\r
    This          - Points to the EFI_HII_CONFIG_ACCESS_PROTOCOL.\r
    Action        - Specifies the type of action taken by the browser.\r
    QuestionId    - A unique value which is sent to the original exporting driver\r
                    so that it can identify the type of data to expect.\r
    Type          - The type of value for the question.\r
    Value         - A pointer to the data being sent to the original exporting driver.\r
    ActionRequest - On return, points to the action requested by the callback function.\r
\r
  Returns:\r
    EFI_SUCCESS          - The callback successfully handled the action.\r
    EFI_OUT_OF_RESOURCES - Not enough storage is available to hold the variable and its data.\r
    EFI_DEVICE_ERROR     - The variable could not be saved.\r
    EFI_UNSUPPORTED      - The specified Action is not supported by the callback.\r
\r
--*/\r
{\r
  ISCSI_FORM_CALLBACK_INFO  *Private;\r
  UINTN                     BufferSize;\r
  CHAR8                     IScsiName[ISCSI_NAME_IFR_MAX_SIZE];\r
  CHAR16                    PortString[128];\r
  CHAR8                     Ip4String[IP4_STR_MAX_SIZE];\r
  CHAR8                     LunString[ISCSI_LUN_STR_MAX_LEN];\r
  UINT64                    Lun;\r
  STRING_REF                DeviceFormTitleToken;\r
  ISCSI_CONFIG_IFR_NVDATA   *IfrNvData;\r
  ISCSI_CONFIG_FORM_ENTRY   *ConfigFormEntry;\r
  EFI_IP_ADDRESS            HostIp;\r
  EFI_IP_ADDRESS            SubnetMask;\r
  EFI_IP_ADDRESS            Gateway;\r
  EFI_STATUS                Status;\r
\r
  Private   = ISCSI_FORM_CALLBACK_INFO_FROM_FORM_CALLBACK (This);\r
\r
  //\r
  // Retrive uncommitted data from Browser\r
  //\r
  BufferSize = sizeof (ISCSI_CONFIG_IFR_NVDATA);\r
  IfrNvData = AllocateZeroPool (BufferSize);\r
  ASSERT (IfrNvData != NULL);\r
  Status = GetBrowserData (NULL, NULL, &BufferSize, (UINT8 *) IfrNvData);\r
  if (EFI_ERROR (Status)) {\r
    gBS->FreePool (IfrNvData);\r
    return Status;\r
  }\r
\r
  switch (KeyValue) {\r
  case KEY_INITIATOR_NAME:\r
    IScsiUnicodeStrToAsciiStr (IfrNvData->InitiatorName, IScsiName);\r
    BufferSize  = AsciiStrLen (IScsiName) + 1;\r
\r
    Status      = gIScsiInitiatorName.Set (&gIScsiInitiatorName, &BufferSize, IScsiName);\r
    if (EFI_ERROR (Status)) {\r
      PopUpInvalidNotify (L"Invalid iSCSI Name!");\r
    }\r
\r
    break;\r
\r
  case KEY_LOCAL_IP:\r
    IScsiUnicodeStrToAsciiStr (IfrNvData->LocalIp, Ip4String);\r
    Status = IScsiAsciiStrToIp (Ip4String, &HostIp.v4);\r
    if (EFI_ERROR (Status) || !Ip4IsUnicast (NTOHL (HostIp.Addr[0]), 0)) {\r
      PopUpInvalidNotify (L"Invalid IP address!");\r
      Status = EFI_INVALID_PARAMETER;\r
    } else {\r
      CopyMem (&Private->Current->SessionConfigData.LocalIp, &HostIp.v4, sizeof (HostIp.v4));\r
    }\r
\r
    break;\r
\r
  case KEY_SUBNET_MASK:\r
    IScsiUnicodeStrToAsciiStr (IfrNvData->SubnetMask, Ip4String);\r
    Status = IScsiAsciiStrToIp (Ip4String, &SubnetMask.v4);\r
    if (EFI_ERROR (Status) || ((SubnetMask.Addr[0] != 0) && (IScsiGetSubnetMaskPrefixLength (&SubnetMask.v4) == 0))) {\r
      PopUpInvalidNotify (L"Invalid Subnet Mask!");\r
      Status = EFI_INVALID_PARAMETER;\r
    } else {\r
      CopyMem (&Private->Current->SessionConfigData.SubnetMask, &SubnetMask.v4, sizeof (SubnetMask.v4));\r
    }\r
\r
    break;\r
\r
  case KEY_GATE_WAY:\r
    IScsiUnicodeStrToAsciiStr (IfrNvData->Gateway, Ip4String);\r
    Status = IScsiAsciiStrToIp (Ip4String, &Gateway.v4);\r
    if (EFI_ERROR (Status) || ((Gateway.Addr[0] != 0) && !Ip4IsUnicast (NTOHL (Gateway.Addr[0]), 0))) {\r
      PopUpInvalidNotify (L"Invalid Gateway!");\r
      Status = EFI_INVALID_PARAMETER;\r
    } else {\r
      CopyMem (&Private->Current->SessionConfigData.Gateway, &Gateway.v4, sizeof (Gateway.v4));\r
    }\r
\r
    break;\r
\r
  case KEY_TARGET_IP:\r
    IScsiUnicodeStrToAsciiStr (IfrNvData->TargetIp, Ip4String);\r
    Status = IScsiAsciiStrToIp (Ip4String, &HostIp.v4);\r
    if (EFI_ERROR (Status) || !Ip4IsUnicast (NTOHL (HostIp.Addr[0]), 0)) {\r
      PopUpInvalidNotify (L"Invalid IP address!");\r
      Status = EFI_INVALID_PARAMETER;\r
    } else {\r
      CopyMem (&Private->Current->SessionConfigData.TargetIp, &HostIp.v4, sizeof (HostIp.v4));\r
    }\r
\r
    break;\r
\r
  case KEY_TARGET_NAME:\r
    IScsiUnicodeStrToAsciiStr (IfrNvData->TargetName, IScsiName);\r
    Status = IScsiNormalizeName (IScsiName, AsciiStrLen (IScsiName));\r
    if (EFI_ERROR (Status)) {\r
      PopUpInvalidNotify (L"Invalid iSCSI Name!");\r
    } else {\r
      AsciiStrCpy (Private->Current->SessionConfigData.TargetName, IScsiName);\r
    }\r
\r
    break;\r
\r
  case KEY_DHCP_ENABLE:\r
    if (IfrNvData->InitiatorInfoFromDhcp == 0) {\r
      IfrNvData->TargetInfoFromDhcp = 0;\r
    }\r
\r
    break;\r
\r
  case KEY_BOOT_LUN:\r
    IScsiUnicodeStrToAsciiStr (IfrNvData->BootLun, LunString);\r
    Status = IScsiAsciiStrToLun (LunString, (UINT8 *) &Lun);\r
    if (EFI_ERROR (Status)) {\r
      PopUpInvalidNotify (L"Invalid LUN string!");\r
    } else {\r
      CopyMem (Private->Current->SessionConfigData.BootLun, &Lun, sizeof (Lun));\r
    }\r
\r
    break;\r
\r
  case KEY_CHAP_NAME:\r
    IScsiUnicodeStrToAsciiStr (IfrNvData->CHAPName, Private->Current->AuthConfigData.CHAPName);\r
    break;\r
\r
  case KEY_CHAP_SECRET:\r
    IScsiUnicodeStrToAsciiStr (IfrNvData->CHAPSecret, Private->Current->AuthConfigData.CHAPSecret);\r
    break;\r
\r
  case KEY_REVERSE_CHAP_NAME:\r
    IScsiUnicodeStrToAsciiStr (IfrNvData->ReverseCHAPName, Private->Current->AuthConfigData.ReverseCHAPName);\r
    break;\r
\r
  case KEY_REVERSE_CHAP_SECRET:\r
    IScsiUnicodeStrToAsciiStr (IfrNvData->ReverseCHAPSecret, Private->Current->AuthConfigData.ReverseCHAPSecret);\r
    break;\r
\r
  case KEY_SAVE_CHANGES:\r
    //\r
    // First, update those fields which don't have INTERACTIVE set.\r
    //\r
    Private->Current->SessionConfigData.Enabled               = IfrNvData->Enabled;\r
    Private->Current->SessionConfigData.InitiatorInfoFromDhcp = IfrNvData->InitiatorInfoFromDhcp;\r
    Private->Current->SessionConfigData.TargetPort            = IfrNvData->TargetPort;\r
    if (Private->Current->SessionConfigData.TargetPort == 0) {\r
      Private->Current->SessionConfigData.TargetPort = ISCSI_WELL_KNOWN_PORT;\r
    }\r
\r
    Private->Current->SessionConfigData.TargetInfoFromDhcp  = IfrNvData->TargetInfoFromDhcp;\r
    Private->Current->AuthConfigData.CHAPType               = IfrNvData->CHAPType;\r
\r
    //\r
    // Only do full parameter validation if iSCSI is enabled on this device.\r
    //\r
    if (Private->Current->SessionConfigData.Enabled) {\r
      //\r
      // Validate the address configuration of the Initiator if DHCP isn't\r
      // deployed.\r
      //\r
      if (!Private->Current->SessionConfigData.InitiatorInfoFromDhcp) {\r
        CopyMem (&HostIp.v4, &Private->Current->SessionConfigData.LocalIp, sizeof (HostIp.v4));\r
        CopyMem (&SubnetMask.v4, &Private->Current->SessionConfigData.SubnetMask, sizeof (SubnetMask.v4));\r
        CopyMem (&Gateway.v4, &Private->Current->SessionConfigData.Gateway, sizeof (Gateway.v4));\r
\r
        if ((Gateway.Addr[0] != 0)) {\r
          if (SubnetMask.Addr[0] == 0) {\r
            PopUpInvalidNotify (L"Gateway address is set but subnet mask is zero.");\r
            Status = EFI_INVALID_PARAMETER;\r
            break;\r
          } else if (!IP4_NET_EQUAL (HostIp.Addr[0], Gateway.Addr[0], SubnetMask.Addr[0])) {\r
            PopUpInvalidNotify (L"Local IP and Gateway are not in the same subnet.");\r
            Status = EFI_INVALID_PARAMETER;\r
            break;\r
          }\r
        }\r
      }\r
      //\r
      // Validate target configuration if DHCP isn't deployed.\r
      //\r
      if (!Private->Current->SessionConfigData.TargetInfoFromDhcp) {\r
        CopyMem (&HostIp.v4, &Private->Current->SessionConfigData.TargetIp, sizeof (HostIp.v4));\r
        if (!Ip4IsUnicast (NTOHL (HostIp.Addr[0]), 0)) {\r
          PopUpInvalidNotify (L"Target IP is invalid!");\r
          Status = EFI_INVALID_PARAMETER;\r
          break;\r
        }\r
      }\r
\r
      if (IfrNvData->CHAPType != ISCSI_CHAP_NONE) {\r
        if ((IfrNvData->CHAPName[0] == '\0') || (IfrNvData->CHAPSecret[0] == '\0')) {\r
          PopUpInvalidNotify (L"CHAP Name or CHAP Secret is invalid!");\r
          Status = EFI_INVALID_PARAMETER;\r
          break;\r
        }\r
\r
        if ((IfrNvData->CHAPType == ISCSI_CHAP_MUTUAL) &&\r
            ((IfrNvData->ReverseCHAPName[0] == '\0') || (IfrNvData->ReverseCHAPSecret[0] == '\0'))\r
            ) {\r
          PopUpInvalidNotify (L"Reverse CHAP Name or Reverse CHAP Secret is invalid!");\r
          Status = EFI_INVALID_PARAMETER;\r
          break;\r
        }\r
      }\r
    }\r
\r
    BufferSize = sizeof (Private->Current->SessionConfigData);\r
    gRT->SetVariable (\r
          Private->Current->MacString,\r
          &gEfiIScsiInitiatorNameProtocolGuid,\r
          ISCSI_CONFIG_VAR_ATTR,\r
          BufferSize,\r
          &Private->Current->SessionConfigData\r
          );\r
\r
    BufferSize = sizeof (Private->Current->AuthConfigData);\r
    gRT->SetVariable (\r
          Private->Current->MacString,\r
          &mIScsiCHAPAuthInfoGuid,\r
          ISCSI_CONFIG_VAR_ATTR,\r
          BufferSize,\r
          &Private->Current->AuthConfigData\r
          );\r
    *ActionRequest = EFI_BROWSER_ACTION_REQUEST_SUBMIT;\r
    break;\r
\r
  default:\r
    if ((KeyValue >= KEY_DEVICE_ENTRY_BASE) && (KeyValue < (mNumberOfIScsiDevices + KEY_DEVICE_ENTRY_BASE))) {\r
      //\r
      // In case goto the device configuration form, update the device form title.\r
      //\r
      ConfigFormEntry = IScsiGetConfigFormEntryByIndex ((UINT32) (KeyValue - KEY_DEVICE_ENTRY_BASE));\r
      ASSERT (ConfigFormEntry != NULL);\r
\r
      UnicodeSPrint (PortString, (UINTN) 128, L"Port %s", ConfigFormEntry->MacString);\r
      DeviceFormTitleToken = (STRING_REF) STR_ISCSI_DEVICE_FORM_TITLE;\r
      HiiLibSetString (Private->RegisteredHandle, DeviceFormTitleToken, PortString);\r
\r
      IScsiConvertDeviceConfigDataToIfrNvData (ConfigFormEntry, IfrNvData);\r
\r
      Private->Current = ConfigFormEntry;\r
    }\r
\r
    break;\r
  }\r
\r
  if (!EFI_ERROR (Status)) {\r
    //\r
    // Pass changed uncommitted data back to Form Browser\r
    //\r
    BufferSize = sizeof (ISCSI_CONFIG_IFR_NVDATA);\r
    Status = SetBrowserData (NULL, NULL, BufferSize, (UINT8 *) IfrNvData, NULL);\r
  }\r
\r
  gBS->FreePool (IfrNvData);\r
  return Status;\r
}\r
\r
\r
EFI_STATUS\r
IScsiConfigUpdateForm (\r
  IN EFI_HANDLE  DriverBindingHandle,\r
  IN EFI_HANDLE  Controller,\r
  IN BOOLEAN     AddForm\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Updates the iSCSI configuration form to add/delete an entry for the iSCSI\r
  device specified by the Controller.\r
\r
Arguments:\r
\r
  DriverBindingHandle - The driverbinding handle.\r
  Controller          - The controller handle of the iSCSI device.\r
  AddForm             - Whether to add or delete a form entry.\r
\r
Returns:\r
\r
  EFI_SUCCESS          - The iSCSI configuration form is updated.\r
  EFI_OUT_OF_RESOURCES - Failed to allocate memory.\r
\r
--*/\r
{\r
  LIST_ENTRY                  *Entry;\r
  ISCSI_CONFIG_FORM_ENTRY     *ConfigFormEntry;\r
  BOOLEAN                     EntryExisted;\r
  EFI_STATUS                  Status;\r
  EFI_HII_UPDATE_DATA         UpdateData;\r
  EFI_SIMPLE_NETWORK_PROTOCOL *Snp;\r
  CHAR16                      PortString[128];\r
  UINT16                      FormIndex;\r
  UINTN                       BufferSize;\r
\r
\r
  ConfigFormEntry = NULL;\r
  EntryExisted    = FALSE;\r
\r
  NET_LIST_FOR_EACH (Entry, &mIScsiConfigFormList) {\r
    ConfigFormEntry = NET_LIST_USER_STRUCT (Entry, ISCSI_CONFIG_FORM_ENTRY, Link);\r
\r
    if (ConfigFormEntry->Controller == Controller) {\r
      EntryExisted = TRUE;\r
      break;\r
    }\r
  }\r
\r
  if (AddForm) {\r
    if (EntryExisted) {\r
      return EFI_SUCCESS;\r
    } else {\r
      //\r
      // Add a new form.\r
      //\r
      ConfigFormEntry = (ISCSI_CONFIG_FORM_ENTRY *) AllocateZeroPool (sizeof (ISCSI_CONFIG_FORM_ENTRY));\r
      if (ConfigFormEntry == NULL) {\r
        return EFI_OUT_OF_RESOURCES;\r
      }\r
\r
      InitializeListHead (&ConfigFormEntry->Link);\r
      ConfigFormEntry->Controller = Controller;\r
\r
      //\r
      // Get the simple network protocol and convert the MAC address into\r
      // the formatted string.\r
      //\r
      Status = gBS->HandleProtocol (\r
                      Controller,\r
                      &gEfiSimpleNetworkProtocolGuid,\r
                      (VOID **)&Snp\r
                      );\r
      ASSERT (Status == EFI_SUCCESS);\r
\r
      IScsiMacAddrToStr (&Snp->Mode->PermanentAddress, Snp->Mode->HwAddressSize, ConfigFormEntry->MacString);\r
\r
      //\r
      // Get the normal session configuration data.\r
      //\r
      BufferSize = sizeof (ConfigFormEntry->SessionConfigData);\r
      Status = gRT->GetVariable (\r
                      ConfigFormEntry->MacString,\r
                      &gEfiIScsiInitiatorNameProtocolGuid,\r
                      NULL,\r
                      &BufferSize,\r
                      &ConfigFormEntry->SessionConfigData\r
                      );\r
      if (EFI_ERROR (Status)) {\r
        ZeroMem (&ConfigFormEntry->SessionConfigData, sizeof (ConfigFormEntry->SessionConfigData));\r
      }\r
      //\r
      // Get the CHAP authentication configuration data.\r
      //\r
      BufferSize = sizeof (ConfigFormEntry->AuthConfigData);\r
      Status = gRT->GetVariable (\r
                      ConfigFormEntry->MacString,\r
                      &mIScsiCHAPAuthInfoGuid,\r
                      NULL,\r
                      &BufferSize,\r
                      &ConfigFormEntry->AuthConfigData\r
                      );\r
      if (EFI_ERROR (Status)) {\r
        ZeroMem (&ConfigFormEntry->AuthConfigData, sizeof (ConfigFormEntry->AuthConfigData));\r
      }\r
      //\r
      // Compose the Port string and create a new STRING_REF.\r
      //\r
      UnicodeSPrint (PortString, 128, L"Port %s", ConfigFormEntry->MacString);\r
      HiiLibNewString (mCallbackInfo->RegisteredHandle, &ConfigFormEntry->PortTitleToken, PortString);\r
\r
      //\r
      // Compose the help string of this port and create a new STRING_REF.\r
      //\r
      UnicodeSPrint (PortString, 128, L"Set the iSCSI parameters on port %s", ConfigFormEntry->MacString);\r
      HiiLibNewString (mCallbackInfo->RegisteredHandle, &ConfigFormEntry->PortTitleHelpToken, PortString);\r
\r
      InsertTailList (&mIScsiConfigFormList, &ConfigFormEntry->Link);\r
      mNumberOfIScsiDevices++;\r
    }\r
  } else {\r
    ASSERT (EntryExisted);\r
\r
    mNumberOfIScsiDevices--;\r
    RemoveEntryList (&ConfigFormEntry->Link);\r
    gBS->FreePool (ConfigFormEntry);\r
  }\r
  //\r
  // Allocate space for creation of Buffer\r
  //\r
  UpdateData.BufferSize = 0x1000;\r
  UpdateData.Data = AllocateZeroPool (0x1000);\r
  UpdateData.Offset = 0;\r
\r
  FormIndex = 0;\r
  NET_LIST_FOR_EACH (Entry, &mIScsiConfigFormList) {\r
    ConfigFormEntry = NET_LIST_USER_STRUCT (Entry, ISCSI_CONFIG_FORM_ENTRY, Link);\r
\r
    CreateGotoOpCode (\r
      FORMID_DEVICE_FORM,\r
      ConfigFormEntry->PortTitleToken,\r
      ConfigFormEntry->PortTitleHelpToken,\r
      EFI_IFR_FLAG_CALLBACK,\r
      (UINT16)(KEY_DEVICE_ENTRY_BASE + FormIndex),\r
      &UpdateData\r
      );\r
\r
    FormIndex++;\r
  }\r
\r
  IfrLibUpdateForm (\r
    mCallbackInfo->RegisteredHandle,\r
    &mVendorGuid,\r
    FORMID_MAIN_FORM,\r
    DEVICE_ENTRY_LABEL,\r
    FALSE,\r
    &UpdateData\r
    );\r
\r
  gBS->FreePool (UpdateData.Data);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
IScsiConfigFormInit (\r
  IN EFI_HANDLE  DriverBindingHandle\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Initialize the iSCSI configuration form.\r
\r
Arguments:\r
\r
  DriverBindingHandle - The iSCSI driverbinding handle.\r
\r
Returns:\r
\r
  EFI_SUCCESS          - The iSCSI configuration form is initialized.\r
  EFI_OUT_OF_RESOURCES - Failed to allocate memory.\r
\r
--*/\r
{\r
  EFI_STATUS                  Status;\r
  EFI_HII_DATABASE_PROTOCOL   *HiiDatabase;\r
  EFI_HII_PACKAGE_LIST_HEADER *PackageList;\r
  ISCSI_FORM_CALLBACK_INFO    *CallbackInfo;\r
\r
  Status = gBS->LocateProtocol (&gEfiHiiDatabaseProtocolGuid, NULL, (VOID **)&HiiDatabase);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  CallbackInfo = (ISCSI_FORM_CALLBACK_INFO *) AllocatePool (sizeof (ISCSI_FORM_CALLBACK_INFO));\r
  if (CallbackInfo == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  CallbackInfo->Signature   = ISCSI_FORM_CALLBACK_INFO_SIGNATURE;\r
  CallbackInfo->HiiDatabase = HiiDatabase;\r
  CallbackInfo->Current     = NULL;\r
\r
  CallbackInfo->ConfigAccess.ExtractConfig = IScsiFormExtractConfig;\r
  CallbackInfo->ConfigAccess.RouteConfig = IScsiFormRouteConfig;\r
  CallbackInfo->ConfigAccess.Callback = IScsiFormCallback;\r
\r
  Status = gBS->LocateProtocol (&gEfiHiiConfigRoutingProtocolGuid, NULL, (VOID **)&CallbackInfo->ConfigRouting);\r
  if (EFI_ERROR (Status)) {\r
    FreePool(CallbackInfo);\r
    return Status;\r
  }\r
\r
  //\r
  // Create driver handle used by HII database\r
  //\r
  Status = HiiLibCreateHiiDriverHandle (&CallbackInfo->DriverHandle);\r
  if (EFI_ERROR (Status)) {\r
    FreePool(CallbackInfo);\r
    return Status;\r
  }\r
  \r
  //\r
  // Install Config Access protocol to driver handle\r
  //\r
  Status = gBS->InstallProtocolInterface (\r
                  &CallbackInfo->DriverHandle,\r
                  &gEfiHiiConfigAccessProtocolGuid,\r
                  EFI_NATIVE_INTERFACE,\r
                  &CallbackInfo->ConfigAccess\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
  \r
  //\r
  // Publish our HII data\r
  //\r
  PackageList = HiiLibPreparePackageList (2, &mVendorGuid, IScsiDxeStrings, IScsiConfigDxeBin);\r
  ASSERT (PackageList != NULL);\r
  \r
  Status = HiiDatabase->NewPackageList (\r
                           HiiDatabase,\r
                           PackageList,\r
                           CallbackInfo->DriverHandle,\r
                           &CallbackInfo->RegisteredHandle\r
                           );\r
  FreePool (PackageList);\r
  if (EFI_ERROR (Status)) {\r
    FreePool(CallbackInfo);\r
    return Status;\r
  }\r
\r
  mCallbackInfo = CallbackInfo;\r
\r
  return Status;\r
}\r
\r
EFI_STATUS\r
IScsiConfigFormUnload (\r
  IN EFI_HANDLE  DriverBindingHandle\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Unload the iSCSI configuration form, this includes: delete all the iSCSI\r
  device configuration entries, uninstall the form callback protocol and\r
  free the resources used.\r
\r
Arguments:\r
\r
  DriverBindingHandle - The iSCSI driverbinding handle.\r
\r
Returns:\r
\r
  EFI_SUCCESS          - The iSCSI configuration form is unloaded.\r
  EFI_OUT_OF_RESOURCES - Failed to allocate memory.\r
\r
--*/\r
{\r
  ISCSI_CONFIG_FORM_ENTRY     *ConfigFormEntry;\r
\r
  while (!IsListEmpty (&mIScsiConfigFormList)) {\r
    //\r
    // Uninstall the device forms as the iSCSI driver instance may fail to\r
    // control the controller but still install the device configuration form.\r
    // In such case, upon driver unloading, the driver instance's driverbinding.\r
    // stop () won't be called, so we have to take this chance here to uninstall\r
    // the device form.\r
    //\r
    ConfigFormEntry = NET_LIST_USER_STRUCT (mIScsiConfigFormList.ForwardLink, ISCSI_CONFIG_FORM_ENTRY, Link);\r
    IScsiConfigUpdateForm (DriverBindingHandle, ConfigFormEntry->Controller, FALSE);\r
  }\r
\r
  //\r
  // Remove HII package list\r
  //\r
  mCallbackInfo->HiiDatabase->RemovePackageList (\r
                                mCallbackInfo->HiiDatabase,\r
                                mCallbackInfo->RegisteredHandle\r
                                );\r
\r
  //\r
  // Uninstall EFI_HII_CONFIG_ACCESS_PROTOCOL\r
  //\r
  gBS->UninstallProtocolInterface (\r
        mCallbackInfo->DriverHandle,\r
        &gEfiHiiConfigAccessProtocolGuid,\r
        &mCallbackInfo->ConfigAccess\r
        );\r
  HiiLibDestroyHiiDriverHandle (mCallbackInfo->DriverHandle);\r
\r
  gBS->FreePool (mCallbackInfo);\r
\r
  return EFI_SUCCESS;\r
}\r