[mirror_edk2.git] / ShellPkg / Library / UefiShellNetwork2CommandsLib / Ifconfig6.c

/** @file\r
  The implementation for Shell command IfConfig6.\r
\r
  Copyright (c) 2016, Intel Corporation. All rights reserved.<BR>\r
  (C) Copyright 2017 Hewlett Packard Enterprise Development LP<BR>\r
\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
#include "UefiShellNetwork2CommandsLib.h"\r
\r
enum {\r
  IfConfig6OpList     = 1,\r
  IfConfig6OpSet      = 2,\r
  IfConfig6OpClear    = 3\r
};\r
\r
typedef enum {\r
  VarCheckReserved      = -1,\r
  VarCheckOk            = 0,\r
  VarCheckDuplicate,\r
  VarCheckConflict,\r
  VarCheckUnknown,\r
  VarCheckLackValue,\r
  VarCheckOutOfMem\r
} VAR_CHECK_CODE;\r
\r
typedef enum {\r
  FlagTypeSingle         = 0,\r
  FlagTypeNeedVar,\r
  FlagTypeNeedSet,\r
  FlagTypeSkipUnknown\r
} VAR_CHECK_FLAG_TYPE;\r
\r
#define MACADDRMAXSIZE    32\r
#define PREFIXMAXLEN      16\r
\r
typedef struct _IFCONFIG6_INTERFACE_CB {\r
  EFI_HANDLE                                  NicHandle;\r
  LIST_ENTRY                                  Link;\r
  EFI_IP6_CONFIG_PROTOCOL                     *IfCfg;\r
  EFI_IP6_CONFIG_INTERFACE_INFO               *IfInfo;\r
  EFI_IP6_CONFIG_INTERFACE_ID                 *IfId;\r
  EFI_IP6_CONFIG_POLICY                       Policy;\r
  EFI_IP6_CONFIG_DUP_ADDR_DETECT_TRANSMITS    Xmits;\r
  UINT32                                      DnsCnt;\r
  EFI_IPv6_ADDRESS                            DnsAddr[1];\r
} IFCONFIG6_INTERFACE_CB;\r
\r
typedef struct _ARG_LIST ARG_LIST;\r
\r
struct _ARG_LIST {\r
  ARG_LIST    *Next;\r
  CHAR16      *Arg;\r
};\r
\r
typedef struct _IFCONFIG6_PRIVATE_DATA {\r
  EFI_HANDLE  ImageHandle;\r
  LIST_ENTRY  IfList;\r
\r
  UINT32      OpCode;\r
  CHAR16      *IfName;\r
  ARG_LIST    *VarArg;\r
} IFCONFIG6_PRIVATE_DATA;\r
\r
typedef struct _VAR_CHECK_ITEM{\r
  CHAR16                 *FlagStr;\r
  UINT32                 FlagID;\r
  UINT32                 ConflictMask;\r
  VAR_CHECK_FLAG_TYPE    FlagType;\r
} VAR_CHECK_ITEM;\r
\r
\r
SHELL_PARAM_ITEM    mIfConfig6CheckList[] = {\r
  {\r
    L"-b",\r
    TypeFlag\r
  },\r
  {\r
    L"-s",\r
    TypeMaxValue\r
  },\r
  {\r
    L"-l",\r
    TypeValue\r
  },\r
  {\r
    L"-r",\r
    TypeValue\r
  },\r
  {\r
    L"-?",\r
    TypeFlag\r
  },\r
  {\r
    NULL,\r
    TypeMax\r
  },\r
};\r
\r
VAR_CHECK_ITEM  mIfConfig6SetCheckList[] = {\r
  {\r
   L"auto",\r
    0x00000001,\r
    0x00000001,\r
    FlagTypeSingle\r
  },\r
  {\r
    L"man",\r
    0x00000002,\r
    0x00000001,\r
    FlagTypeSingle\r
  },\r
  {\r
    L"host",\r
    0x00000004,\r
    0x00000002,\r
    FlagTypeSingle\r
  },\r
  {\r
    L"dad",\r
    0x00000008,\r
    0x00000004,\r
    FlagTypeSingle\r
  },\r
  {\r
    L"gw",\r
    0x00000010,\r
    0x00000008,\r
    FlagTypeSingle\r
  },\r
  {\r
    L"dns",\r
    0x00000020,\r
    0x00000010,\r
    FlagTypeSingle\r
  },\r
  {\r
    L"id",\r
    0x00000040,\r
    0x00000020,\r
    FlagTypeSingle\r
  },\r
  {\r
    NULL,\r
    0x0,\r
    0x0,\r
    FlagTypeSkipUnknown\r
  },\r
};\r
\r
/**\r
  Free the ARG_LIST.\r
\r
  @param List Pointer to ARG_LIST to free.\r
**/\r
VOID\r
IfConfig6FreeArgList (\r
  ARG_LIST       *List\r
)\r
{\r
  ARG_LIST       *Next;\r
  while (List->Next != NULL) {\r
    Next = List->Next;\r
    FreePool (List);\r
    List = Next;\r
  }\r
\r
  FreePool (List);\r
}\r
\r
/**\r
  Split a string with specified separator and save the substring to a list.\r
\r
  @param[in]    String       The pointer of the input string.\r
  @param[in]    Separator    The specified separator.\r
\r
  @return The pointer of headnode of ARG_LIST.\r
\r
**/\r
ARG_LIST *\r
IfConfig6SplitStrToList (\r
  IN CONST CHAR16    *String,\r
  IN CHAR16          Separator\r
  )\r
{\r
  CHAR16      *Str;\r
  CHAR16      *ArgStr;\r
  ARG_LIST    *ArgList;\r
  ARG_LIST    *ArgNode;\r
\r
  if (String == NULL || *String == L'\0') {\r
    return NULL;\r
  }\r
\r
  //\r
  // Copy the CONST string to a local copy.\r
  //\r
  Str = AllocateCopyPool (StrSize (String), String);\r
  if (Str == NULL) {\r
    return NULL;\r
  }\r
  ArgStr  = Str;\r
\r
  //\r
  // init a node for the list head.\r
  //\r
  ArgNode = (ARG_LIST *) AllocateZeroPool (sizeof (ARG_LIST));\r
  if (ArgNode == NULL) {\r
    return NULL;\r
  }\r
  ArgList = ArgNode;\r
\r
  //\r
  // Split the local copy and save in the list node.\r
  //\r
  while (*Str != L'\0') {\r
    if (*Str == Separator) {\r
      *Str          = L'\0';\r
      ArgNode->Arg  = ArgStr;\r
      ArgStr        = Str + 1;\r
      ArgNode->Next = (ARG_LIST *) AllocateZeroPool (sizeof (ARG_LIST));\r
      if (ArgNode->Next == NULL) {\r
        //\r
        // Free the local copy of string stored in the first node\r
        //\r
        FreePool (ArgList->Arg);\r
        IfConfig6FreeArgList (ArgList);\r
        return NULL;\r
      }\r
      ArgNode = ArgNode->Next;\r
    }\r
\r
    Str++;\r
  }\r
\r
  ArgNode->Arg  = ArgStr;\r
  ArgNode->Next = NULL;\r
\r
  return ArgList;\r
}\r
\r
/**\r
  Check the correctness of input Args with '-s' option.\r
\r
  @param[in]    CheckList    The pointer of VAR_CHECK_ITEM array.\r
  @param[in]    Name         The pointer of input arg.\r
  @param[in]    Init         The switch to execute the check.\r
\r
  @return The value of VAR_CHECK_CODE.\r
\r
**/\r
VAR_CHECK_CODE\r
IfConfig6RetriveCheckListByName(\r
  IN VAR_CHECK_ITEM    *CheckList,\r
  IN CHAR16            *Name,\r
  IN BOOLEAN           Init\r
)\r
{\r
  STATIC UINT32     CheckDuplicate;\r
  STATIC UINT32     CheckConflict;\r
  VAR_CHECK_CODE    RtCode;\r
  UINT32            Index;\r
  VAR_CHECK_ITEM    Arg;\r
\r
  if (Init) {\r
    CheckDuplicate = 0;\r
    CheckConflict  = 0;\r
    return VarCheckOk;\r
  }\r
\r
  RtCode  = VarCheckOk;\r
  Index   = 0;\r
  Arg     = CheckList[Index];\r
\r
  //\r
  // Check the Duplicated/Conflicted/Unknown input Args.\r
  //\r
  while (Arg.FlagStr != NULL) {\r
    if (StrCmp (Arg.FlagStr, Name) == 0) {\r
\r
      if (CheckDuplicate & Arg.FlagID) {\r
        RtCode = VarCheckDuplicate;\r
        break;\r
      }\r
\r
      if (CheckConflict & Arg.ConflictMask) {\r
        RtCode = VarCheckConflict;\r
        break;\r
      }\r
\r
      CheckDuplicate |= Arg.FlagID;\r
      CheckConflict  |= Arg.ConflictMask;\r
      break;\r
    }\r
\r
    Arg = CheckList[++Index];\r
  }\r
\r
  if (Arg.FlagStr == NULL) {\r
    RtCode = VarCheckUnknown;\r
  }\r
\r
  return RtCode;\r
}\r
\r
/**\r
  The notify function of create event when performing a manual config.\r
\r
  @param[in]    Event        The event this notify function registered to.\r
  @param[in]    Context      Pointer to the context data registered to the event.\r
\r
**/\r
VOID\r
EFIAPI\r
IfConfig6ManualAddressNotify (\r
  IN EFI_EVENT    Event,\r
  IN VOID         *Context\r
  )\r
{\r
  *((BOOLEAN *) Context) = TRUE;\r
}\r
\r
/**\r
  Print MAC address.\r
\r
  @param[in]    Node    The pointer of MAC address buffer.\r
  @param[in]    Size    The size of MAC address buffer.\r
\r
**/\r
VOID\r
IfConfig6PrintMacAddr (\r
  IN UINT8     *Node,\r
  IN UINT32    Size\r
  )\r
{\r
  UINTN    Index;\r
\r
  ASSERT (Size <= MACADDRMAXSIZE);\r
\r
  for (Index = 0; Index < Size; Index++) {\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_MAC_ADDR_BODY), gShellNetwork2HiiHandle, Node[Index]);\r
    if (Index + 1 < Size) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_COLON), gShellNetwork2HiiHandle);\r
    }\r
  }\r
\r
  ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_NEWLINE), gShellNetwork2HiiHandle);\r
}\r
\r
/**\r
  Print IPv6 address.\r
\r
  @param[in]    Ip           The pointer of Ip bufffer in EFI_IPv6_ADDRESS format.\r
  @param[in]    PrefixLen    The pointer of PrefixLen that describes the size Prefix.\r
\r
**/\r
VOID\r
IfConfig6PrintIpAddr (\r
  IN EFI_IPv6_ADDRESS    *Ip,\r
  IN UINT8               *PrefixLen\r
  )\r
{\r
  UINTN      Index;\r
  BOOLEAN    Short;\r
\r
  Short = FALSE;\r
\r
  for (Index = 0; Index < PREFIXMAXLEN; Index = Index + 2) {\r
\r
    if (!Short && (Index + 1 < PREFIXMAXLEN) && (Index % 2 == 0) && (Ip->Addr[Index] == 0) && (Ip->Addr[Index + 1] == 0)) {\r
      //\r
      // Deal with the case of ::.\r
      //\r
      if (Index == 0) {\r
        //\r
        // :: is at the beginning of the address.\r
        //\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_COLON), gShellNetwork2HiiHandle);\r
      }\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_COLON), gShellNetwork2HiiHandle);\r
\r
      while ((Ip->Addr[Index] == 0) && (Ip->Addr[Index + 1] == 0) && (Index < PREFIXMAXLEN)) {\r
        Index = Index + 2;\r
        if (Index > PREFIXMAXLEN - 2) {\r
          break;\r
        }\r
      }\r
\r
      Short = TRUE;\r
\r
      if (Index == PREFIXMAXLEN) {\r
        //\r
        // :: is at the end of the address.\r
        //\r
        break;\r
      }\r
    }\r
\r
    if (Index < PREFIXMAXLEN - 1) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_IP_ADDR_BODY), gShellNetwork2HiiHandle, Ip->Addr[Index]);\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_IP_ADDR_BODY), gShellNetwork2HiiHandle, Ip->Addr[Index + 1]);\r
    }\r
\r
    if (Index + 2 < PREFIXMAXLEN) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_COLON), gShellNetwork2HiiHandle);\r
    }\r
  }\r
\r
  if (PrefixLen != NULL) {\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_PREFIX_LEN), gShellNetwork2HiiHandle, *PrefixLen);\r
  }\r
}\r
\r
/**\r
  Pick up host IPv6 address in string format from Args with "-s" option and convert it to EFI_IP6_CONFIG_MANUAL_ADDRESS format.\r
\r
  @param[in, out]    Arg        The pointer of the address of ARG_LIST which save Args with the "-s" option.\r
  @param[out]        Buf        The pointer of the address of EFI_IP6_CONFIG_MANUAL_ADDRESS.\r
  @param[out]        BufSize    The pointer of BufSize that describes the size of Buf in bytes.\r
\r
  @retval EFI_SUCCESS    The convertion is successful.\r
  @retval Others         Does't find the host address, or it is an invalid IPv6 address in string format.\r
\r
**/\r
EFI_STATUS\r
IfConfig6ParseManualAddressList (\r
  IN OUT ARG_LIST                         **Arg,\r
     OUT EFI_IP6_CONFIG_MANUAL_ADDRESS    **Buf,\r
     OUT UINTN                            *BufSize\r
  )\r
{\r
  EFI_STATUS                       Status;\r
  EFI_IP6_CONFIG_MANUAL_ADDRESS    *AddrBuf;\r
  ARG_LIST                         *VarArg;\r
  EFI_IPv6_ADDRESS                 Address;\r
  UINT8                            Prefix;\r
  UINT8                            AddrCnt;\r
\r
  Prefix   = 0;\r
  AddrCnt  = 0;\r
  *BufSize = 0;\r
  *Buf     = NULL;\r
  VarArg   = *Arg;\r
  Status   = EFI_SUCCESS;\r
\r
  //\r
  // Go through the list to check the correctness of input host ip6 address.\r
  //\r
  while ((!EFI_ERROR (Status)) && (VarArg != NULL)) {\r
\r
    Status = NetLibStrToIp6andPrefix (VarArg->Arg, &Address, &Prefix);\r
\r
    if (EFI_ERROR (Status)) {\r
      //\r
      // host ip ip ... gw\r
      //\r
      break;\r
    }\r
\r
    VarArg = VarArg->Next;\r
    AddrCnt++;\r
  }\r
\r
  if (AddrCnt == 0) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  AddrBuf = AllocateZeroPool (AddrCnt * sizeof (EFI_IP6_CONFIG_MANUAL_ADDRESS));\r
  ASSERT (AddrBuf != NULL);\r
\r
  AddrCnt = 0;\r
  VarArg  = *Arg;\r
  Status  = EFI_SUCCESS;\r
\r
  //\r
  // Go through the list to fill in the EFI_IP6_CONFIG_MANUAL_ADDRESS structure.\r
  //\r
  while ((!EFI_ERROR (Status)) && (VarArg != NULL)) {\r
\r
    Status = NetLibStrToIp6andPrefix (VarArg->Arg, &Address, &Prefix);\r
\r
    if (EFI_ERROR (Status)) {\r
      break;\r
    }\r
\r
    //\r
    // If prefix length is not set, set it as Zero here. In the IfConfigSetInterfaceInfo()\r
    // Zero prefix, length will be transfered to default prefix length.\r
    //\r
    if (Prefix == 0xFF) {\r
      Prefix = 0;\r
    }\r
    AddrBuf[AddrCnt].IsAnycast    = FALSE;\r
    AddrBuf[AddrCnt].PrefixLength = Prefix;\r
    IP6_COPY_ADDRESS (&AddrBuf[AddrCnt].Address, &Address);\r
    VarArg = VarArg->Next;\r
    AddrCnt++;\r
  }\r
\r
  *Arg = VarArg;\r
\r
  if (EFI_ERROR (Status) && (Status != EFI_INVALID_PARAMETER)) {\r
    goto ON_ERROR;\r
  }\r
\r
  *Buf     = AddrBuf;\r
  *BufSize = AddrCnt * sizeof (EFI_IP6_CONFIG_MANUAL_ADDRESS);\r
\r
  return EFI_SUCCESS;\r
\r
ON_ERROR:\r
\r
  FreePool (AddrBuf);\r
  return Status;\r
}\r
\r
/**\r
  Pick up gw/dns IPv6 address in string format from Args with "-s" option and convert it to EFI_IPv6_ADDRESS format.\r
\r
  @param[in, out]    Arg        The pointer of the address of ARG_LIST that save Args with the "-s" option.\r
  @param[out]        Buf        The pointer of the address of EFI_IPv6_ADDRESS.\r
  @param[out]        BufSize    The pointer of BufSize that describes the size of Buf in bytes.\r
\r
  @retval EFI_SUCCESS    The conversion is successful.\r
  @retval Others         Doesn't find the host address, or it is an invalid IPv6 address in string format.\r
\r
**/\r
EFI_STATUS\r
IfConfig6ParseGwDnsAddressList (\r
  IN OUT ARG_LIST            **Arg,\r
     OUT EFI_IPv6_ADDRESS    **Buf,\r
     OUT UINTN               *BufSize\r
  )\r
{\r
  EFI_STATUS          Status;\r
  EFI_IPv6_ADDRESS    *AddrBuf;\r
  ARG_LIST            *VarArg;\r
  EFI_IPv6_ADDRESS    Address;\r
  UINT8               Prefix;\r
  UINT8               AddrCnt;\r
\r
  AddrCnt  = 0;\r
  *BufSize = 0;\r
  *Buf     = NULL;\r
  VarArg   = *Arg;\r
  Status   = EFI_SUCCESS;\r
\r
  //\r
  // Go through the list to check the correctness of input gw/dns address.\r
  //\r
  while ((!EFI_ERROR (Status)) && (VarArg != NULL)) {\r
\r
    Status = NetLibStrToIp6andPrefix (VarArg->Arg, &Address, &Prefix);\r
\r
    if (EFI_ERROR (Status)) {\r
      //\r
      // gw ip ip ... host\r
      //\r
      break;\r
    }\r
\r
    VarArg = VarArg->Next;\r
    AddrCnt++;\r
  }\r
\r
  if (AddrCnt == 0) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  AddrBuf = AllocateZeroPool (AddrCnt * sizeof (EFI_IPv6_ADDRESS));\r
  ASSERT (AddrBuf != NULL);\r
\r
  AddrCnt = 0;\r
  VarArg  = *Arg;\r
  Status  = EFI_SUCCESS;\r
\r
  //\r
  // Go through the list to fill in the EFI_IPv6_ADDRESS structure.\r
  //\r
  while ((!EFI_ERROR (Status)) && (VarArg != NULL)) {\r
\r
    Status = NetLibStrToIp6andPrefix (VarArg->Arg, &Address, &Prefix);\r
\r
    if (EFI_ERROR (Status)) {\r
      break;\r
    }\r
\r
    IP6_COPY_ADDRESS (&AddrBuf[AddrCnt], &Address);\r
\r
    VarArg = VarArg->Next;\r
    AddrCnt++;\r
  }\r
\r
  *Arg = VarArg;\r
\r
  if (EFI_ERROR (Status) && (Status != EFI_INVALID_PARAMETER)) {\r
   goto ON_ERROR;\r
  }\r
\r
  *Buf     = AddrBuf;\r
  *BufSize = AddrCnt * sizeof (EFI_IPv6_ADDRESS);\r
\r
  return EFI_SUCCESS;\r
\r
ON_ERROR:\r
\r
  FreePool (AddrBuf);\r
  return Status;\r
}\r
\r
/**\r
  Parse InterfaceId in string format from Args with the "-s" option and convert it to EFI_IP6_CONFIG_INTERFACE_ID format.\r
\r
  @param[in, out]   Arg     The pointer of the address of ARG_LIST that saves Args with the "-s" option.\r
  @param[out]       IfId    The pointer of EFI_IP6_CONFIG_INTERFACE_ID.\r
\r
  @retval EFI_SUCCESS              The get status processed successfullly.\r
  @retval EFI_INVALID_PARAMETER    The get status process failed.\r
\r
**/\r
EFI_STATUS\r
IfConfig6ParseInterfaceId (\r
  IN OUT ARG_LIST                       **Arg,\r
     OUT EFI_IP6_CONFIG_INTERFACE_ID    **IfId\r
  )\r
{\r
  UINT8     Index;\r
  UINT8     NodeVal;\r
  CHAR16    *IdStr;\r
\r
  if (*Arg == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Index = 0;\r
  IdStr = (*Arg)->Arg;\r
  ASSERT (IfId != NULL);\r
  *IfId = AllocateZeroPool (sizeof (EFI_IP6_CONFIG_INTERFACE_ID));\r
  ASSERT (*IfId != NULL);\r
\r
  while ((*IdStr != L'\0') && (Index < 8)) {\r
\r
    NodeVal = 0;\r
    while ((*IdStr != L':') && (*IdStr != L'\0')) {\r
\r
      if ((*IdStr <= L'F') && (*IdStr >= L'A')) {\r
        NodeVal = (UINT8)((NodeVal << 4) + *IdStr - L'A' + 10);\r
      } else if ((*IdStr <= L'f') && (*IdStr >= L'a')) {\r
        NodeVal = (UINT8)((NodeVal << 4) + *IdStr - L'a' + 10);\r
      } else if ((*IdStr <= L'9') && (*IdStr >= L'0')) {\r
        NodeVal = (UINT8)((NodeVal << 4) + *IdStr - L'0');\r
      } else {\r
        FreePool (*IfId);\r
        return EFI_INVALID_PARAMETER;\r
      }\r
\r
      IdStr++;\r
    }\r
\r
    (*IfId)->Id[Index++] = NodeVal;\r
\r
    if (*IdStr == L':') {\r
      IdStr++;\r
    }\r
  }\r
\r
  *Arg = (*Arg)->Next;\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Parse dad in string format from Args with the "-s" option and convert it to UINT32 format.\r
\r
  @param[in, out]   Arg      The pointer of the address of ARG_LIST that saves Args with the "-s" option.\r
  @param[out]       Xmits    The pointer of Xmits.\r
\r
  @retval EFI_SUCCESS    The get status processed successfully.\r
  @retval others         The get status process failed.\r
\r
**/\r
EFI_STATUS\r
IfConfig6ParseDadXmits (\r
  IN OUT ARG_LIST    **Arg,\r
     OUT UINT32      *Xmits\r
  )\r
{\r
  CHAR16    *ValStr;\r
\r
  if (*Arg == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  ValStr = (*Arg)->Arg;\r
  *Xmits = 0;\r
\r
  while (*ValStr != L'\0') {\r
\r
    if ((*ValStr <= L'9') && (*ValStr >= L'0')) {\r
\r
      *Xmits = (*Xmits * 10) + (*ValStr - L'0');\r
\r
    } else {\r
\r
      return EFI_INVALID_PARAMETER;\r
    }\r
\r
    ValStr++;\r
  }\r
\r
  *Arg = (*Arg)->Next;\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  The get current status of all handles.\r
\r
  @param[in]   ImageHandle    The handle of  ImageHandle.\r
  @param[in]   IfName         The pointer of  IfName(interface name).\r
  @param[in]   IfList         The pointer of  IfList(interface list).\r
\r
  @retval EFI_SUCCESS    The get status processed successfully.\r
  @retval others         The get status process failed.\r
\r
**/\r
EFI_STATUS\r
IfConfig6GetInterfaceInfo (\r
  IN EFI_HANDLE    ImageHandle,\r
  IN CHAR16        *IfName,\r
  IN LIST_ENTRY    *IfList\r
  )\r
{\r
  EFI_STATUS                       Status;\r
  UINTN                            HandleIndex;\r
  UINTN                            HandleNum;\r
  EFI_HANDLE                       *HandleBuffer;\r
  EFI_IP6_CONFIG_PROTOCOL          *Ip6Cfg;\r
  EFI_IP6_CONFIG_INTERFACE_INFO    *IfInfo;\r
  IFCONFIG6_INTERFACE_CB           *IfCb;\r
  UINTN                            DataSize;\r
\r
  HandleBuffer = NULL;\r
  HandleNum    = 0;\r
\r
  IfInfo       = NULL;\r
  IfCb         = NULL;\r
\r
  //\r
  // Locate all the handles with ip6 service binding protocol.\r
  //\r
  Status = gBS->LocateHandleBuffer (\r
                  ByProtocol,\r
                  &gEfiIp6ServiceBindingProtocolGuid,\r
                  NULL,\r
                  &HandleNum,\r
                  &HandleBuffer\r
                 );\r
  if (EFI_ERROR (Status) || (HandleNum == 0)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Enumerate all handles that installed with ip6 service binding protocol.\r
  //\r
  for (HandleIndex = 0; HandleIndex < HandleNum; HandleIndex++) {\r
    IfCb      = NULL;\r
    IfInfo    = NULL;\r
    DataSize  = 0;\r
\r
    //\r
    // Ip6config protocol and ip6 service binding protocol are installed\r
    // on the same handle.\r
    //\r
    ASSERT (HandleBuffer != NULL);\r
    Status = gBS->HandleProtocol (\r
                    HandleBuffer[HandleIndex],\r
                    &gEfiIp6ConfigProtocolGuid,\r
                    (VOID **) &Ip6Cfg\r
                    );\r
\r
    if (EFI_ERROR (Status)) {\r
      goto ON_ERROR;\r
    }\r
    //\r
    // Get the interface information size.\r
    //\r
    Status = Ip6Cfg->GetData (\r
                       Ip6Cfg,\r
                       Ip6ConfigDataTypeInterfaceInfo,\r
                       &DataSize,\r
                       NULL\r
                       );\r
\r
    if (Status != EFI_BUFFER_TOO_SMALL) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), gShellNetwork2HiiHandle, Status);\r
      goto ON_ERROR;\r
    }\r
\r
    IfInfo = AllocateZeroPool (DataSize);\r
\r
    if (IfInfo == NULL) {\r
      Status = EFI_OUT_OF_RESOURCES;\r
      goto ON_ERROR;\r
    }\r
    //\r
    // Get the interface info.\r
    //\r
    Status = Ip6Cfg->GetData (\r
                       Ip6Cfg,\r
                       Ip6ConfigDataTypeInterfaceInfo,\r
                       &DataSize,\r
                       IfInfo\r
                       );\r
\r
    if (EFI_ERROR (Status)) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), gShellNetwork2HiiHandle, Status);\r
      goto ON_ERROR;\r
    }\r
    //\r
    // Check the interface name if required.\r
    //\r
    if ((IfName != NULL) && (StrCmp (IfName, IfInfo->Name) != 0)) {\r
      FreePool (IfInfo);\r
      continue;\r
    }\r
\r
    DataSize = 0;\r
    //\r
    // Get the size of dns server list.\r
    //\r
    Status = Ip6Cfg->GetData (\r
                       Ip6Cfg,\r
                       Ip6ConfigDataTypeDnsServer,\r
                       &DataSize,\r
                       NULL\r
                       );\r
\r
    if ((Status != EFI_BUFFER_TOO_SMALL) && (Status != EFI_NOT_FOUND)) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), gShellNetwork2HiiHandle, Status);\r
      goto ON_ERROR;\r
    }\r
\r
    IfCb = AllocateZeroPool (sizeof (IFCONFIG6_INTERFACE_CB) + DataSize);\r
\r
    if (IfCb == NULL) {\r
      Status = EFI_OUT_OF_RESOURCES;\r
      goto ON_ERROR;\r
    }\r
\r
    IfCb->NicHandle = HandleBuffer[HandleIndex];\r
    IfCb->IfInfo    = IfInfo;\r
    IfCb->IfCfg     = Ip6Cfg;\r
    IfCb->DnsCnt    = (UINT32) (DataSize / sizeof (EFI_IPv6_ADDRESS));\r
\r
    //\r
    // Get the dns server list if has.\r
    //\r
    if (DataSize > 0) {\r
\r
      Status = Ip6Cfg->GetData (\r
                         Ip6Cfg,\r
                         Ip6ConfigDataTypeDnsServer,\r
                         &DataSize,\r
                         IfCb->DnsAddr\r
                         );\r
\r
      if (EFI_ERROR (Status)) {\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), gShellNetwork2HiiHandle, Status);\r
        goto ON_ERROR;\r
      }\r
    }\r
    //\r
    // Get the interface id if has.\r
    //\r
    DataSize   = sizeof (EFI_IP6_CONFIG_INTERFACE_ID);\r
    IfCb->IfId = AllocateZeroPool (DataSize);\r
\r
    if (IfCb->IfId == NULL) {\r
      goto ON_ERROR;\r
    }\r
\r
    Status = Ip6Cfg->GetData (\r
                       Ip6Cfg,\r
                       Ip6ConfigDataTypeAltInterfaceId,\r
                       &DataSize,\r
                       IfCb->IfId\r
                       );\r
\r
    if (EFI_ERROR (Status) && (Status != EFI_NOT_FOUND)) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), gShellNetwork2HiiHandle, Status);\r
      goto ON_ERROR;\r
    }\r
\r
    if (Status == EFI_NOT_FOUND) {\r
      FreePool (IfCb->IfId);\r
      IfCb->IfId = NULL;\r
    }\r
    //\r
    // Get the config policy.\r
    //\r
    DataSize = sizeof (EFI_IP6_CONFIG_POLICY);\r
    Status   = Ip6Cfg->GetData (\r
                         Ip6Cfg,\r
                         Ip6ConfigDataTypePolicy,\r
                         &DataSize,\r
                         &IfCb->Policy\r
                         );\r
\r
    if (EFI_ERROR (Status)) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), gShellNetwork2HiiHandle, Status);\r
      goto ON_ERROR;\r
    }\r
    //\r
    // Get the dad transmits.\r
    //\r
    DataSize = sizeof (EFI_IP6_CONFIG_DUP_ADDR_DETECT_TRANSMITS);\r
    Status   = Ip6Cfg->GetData (\r
                         Ip6Cfg,\r
                         Ip6ConfigDataTypeDupAddrDetectTransmits,\r
                         &DataSize,\r
                         &IfCb->Xmits\r
                         );\r
\r
    if (EFI_ERROR (Status)) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), gShellNetwork2HiiHandle, Status);\r
      goto ON_ERROR;\r
    }\r
\r
    InsertTailList (IfList, &IfCb->Link);\r
\r
    if ((IfName != NULL) && (StrCmp (IfName, IfInfo->Name) == 0)) {\r
      //\r
      // Only need the appointed interface, keep the allocated buffer.\r
      //\r
      IfCb   = NULL;\r
      IfInfo = NULL;\r
      break;\r
    }\r
  }\r
\r
  if (HandleBuffer != NULL) {\r
    FreePool (HandleBuffer);\r
  }\r
\r
  return EFI_SUCCESS;\r
\r
ON_ERROR:\r
\r
  if (IfInfo != NULL) {\r
    FreePool (IfInfo);\r
  }\r
\r
  if (IfCb != NULL) {\r
    if (IfCb->IfId != NULL) {\r
      FreePool (IfCb->IfId);\r
    }\r
\r
    FreePool (IfCb);\r
  }\r
\r
  return Status;\r
}\r
\r
/**\r
  The list process of the IfConfig6 application.\r
\r
  @param[in]   IfList    The pointer of IfList(interface list).\r
\r
  @retval SHELL_SUCCESS  The IfConfig6 list processed successfully.\r
  @retval others         The IfConfig6 list process failed.\r
\r
**/\r
SHELL_STATUS\r
IfConfig6ShowInterfaceInfo (\r
  IN LIST_ENTRY    *IfList\r
  )\r
{\r
  LIST_ENTRY                *Entry;\r
  IFCONFIG6_INTERFACE_CB    *IfCb;\r
  UINTN                     Index;\r
\r
  Entry  = IfList->ForwardLink;\r
\r
  if (IsListEmpty (IfList)) {\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_INTERFACE), gShellNetwork2HiiHandle);\r
  }\r
\r
  //\r
  // Go through the interface list.\r
  //\r
  while (Entry != IfList) {\r
\r
    IfCb = BASE_CR (Entry, IFCONFIG6_INTERFACE_CB, Link);\r
\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_BREAK), gShellNetwork2HiiHandle);\r
\r
    //\r
    // Print interface name.\r
    //\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_IF_NAME), gShellNetwork2HiiHandle, IfCb->IfInfo->Name);\r
\r
    //\r
    // Print interface config policy.\r
    //\r
    if (IfCb->Policy == Ip6ConfigPolicyAutomatic) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_POLICY_AUTO), gShellNetwork2HiiHandle);\r
    } else {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_POLICY_MAN), gShellNetwork2HiiHandle);\r
    }\r
\r
    //\r
    // Print dad transmit.\r
    //\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_DAD_TRANSMITS), gShellNetwork2HiiHandle, IfCb->Xmits);\r
\r
    //\r
    // Print interface id if has.\r
    //\r
    if (IfCb->IfId != NULL) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_INTERFACE_ID_HEAD), gShellNetwork2HiiHandle);\r
\r
      IfConfig6PrintMacAddr (\r
        IfCb->IfId->Id,\r
        8\r
        );\r
    }\r
    //\r
    // Print mac address of the interface.\r
    //\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_MAC_ADDR_HEAD), gShellNetwork2HiiHandle);\r
\r
    IfConfig6PrintMacAddr (\r
      IfCb->IfInfo->HwAddress.Addr,\r
      IfCb->IfInfo->HwAddressSize\r
      );\r
\r
    //\r
    // Print ip addresses list of the interface.\r
    //\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_IP_ADDR_HEAD), gShellNetwork2HiiHandle);\r
\r
    for (Index = 0; Index < IfCb->IfInfo->AddressInfoCount; Index++) {\r
      IfConfig6PrintIpAddr (\r
        &IfCb->IfInfo->AddressInfo[Index].Address,\r
        &IfCb->IfInfo->AddressInfo[Index].PrefixLength\r
        );\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_NEWLINE), gShellNetwork2HiiHandle);\r
    }\r
\r
    //\r
    // Print dns server addresses list of the interface if has.\r
    //\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_DNS_ADDR_HEAD), gShellNetwork2HiiHandle);\r
\r
    for (Index = 0; Index < IfCb->DnsCnt; Index++) {\r
      IfConfig6PrintIpAddr (\r
        &IfCb->DnsAddr[Index],\r
        NULL\r
        );\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_NEWLINE), gShellNetwork2HiiHandle);\r
    }\r
\r
    //\r
    // Print route table of the interface if has.\r
    //\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_ROUTE_HEAD), gShellNetwork2HiiHandle);\r
\r
    for (Index = 0; Index < IfCb->IfInfo->RouteCount; Index++) {\r
      IfConfig6PrintIpAddr (\r
        &IfCb->IfInfo->RouteTable[Index].Destination,\r
        &IfCb->IfInfo->RouteTable[Index].PrefixLength\r
        );\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_JOINT), gShellNetwork2HiiHandle);\r
\r
      IfConfig6PrintIpAddr (\r
        &IfCb->IfInfo->RouteTable[Index].Gateway,\r
        NULL\r
        );\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_NEWLINE), gShellNetwork2HiiHandle);\r
    }\r
\r
    Entry = Entry->ForwardLink;\r
  }\r
\r
  ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_BREAK), gShellNetwork2HiiHandle);\r
  ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_NEWLINE), gShellNetwork2HiiHandle);\r
\r
  return SHELL_SUCCESS;\r
}\r
\r
/**\r
  The clean process of the IfConfig6 application.\r
\r
  @param[in]   IfList    The pointer of IfList(interface list).\r
  @param[in]   IfName    The pointer of interface name.\r
\r
  @retval SHELL_SUCCESS  The IfConfig6 clean processed successfully.\r
  @retval others         The IfConfig6 clean process failed.\r
\r
**/\r
SHELL_STATUS\r
IfConfig6ClearInterfaceInfo (\r
  IN LIST_ENTRY    *IfList,\r
  IN CHAR16        *IfName\r
  )\r
{\r
  EFI_STATUS                Status;\r
  SHELL_STATUS              ShellStatus;\r
  LIST_ENTRY                *Entry;\r
  IFCONFIG6_INTERFACE_CB    *IfCb;\r
  EFI_IP6_CONFIG_POLICY     Policy;\r
\r
  Entry  = IfList->ForwardLink;\r
  Status = EFI_SUCCESS;\r
  ShellStatus = SHELL_SUCCESS;\r
\r
  if (IsListEmpty (IfList)) {\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_INTERFACE), gShellNetwork2HiiHandle);\r
  }\r
\r
  //\r
  // Go through the interface list.If the interface name is specified, then\r
  // need to refresh the configuration.\r
  //\r
  while (Entry != IfList) {\r
\r
    IfCb = BASE_CR (Entry, IFCONFIG6_INTERFACE_CB, Link);\r
\r
    if ((IfName != NULL) && (StrCmp (IfName, IfCb->IfInfo->Name) == 0)) {\r
      Policy = Ip6ConfigPolicyManual;\r
\r
      Status = IfCb->IfCfg->SetData (\r
                              IfCb->IfCfg,\r
                              Ip6ConfigDataTypePolicy,\r
                              sizeof (EFI_IP6_CONFIG_POLICY),\r
                              &Policy\r
                              );\r
      if (EFI_ERROR (Status)) {\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_AD), gShellNetwork2HiiHandle, L"ifconfig6");\r
        ShellStatus = SHELL_ACCESS_DENIED;\r
        break;\r
      }\r
    }\r
\r
    Policy = Ip6ConfigPolicyAutomatic;\r
\r
    Status = IfCb->IfCfg->SetData (\r
                            IfCb->IfCfg,\r
                            Ip6ConfigDataTypePolicy,\r
                            sizeof (EFI_IP6_CONFIG_POLICY),\r
                            &Policy\r
                            );\r
\r
    if (EFI_ERROR (Status)) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_AD), gShellNetwork2HiiHandle, L"ifconfig6");\r
      ShellStatus = SHELL_ACCESS_DENIED;\r
      break;\r
    }\r
\r
    Entry  = Entry->ForwardLink;\r
  }\r
\r
  return ShellStatus;\r
}\r
\r
/**\r
  The set process of the IfConfig6 application.\r
\r
  @param[in]   IfList    The pointer of IfList(interface list).\r
  @param[in]   VarArg    The pointer of ARG_LIST(Args with "-s" option).\r
\r
  @retval SHELL_SUCCESS  The IfConfig6 set processed successfully.\r
  @retval others         The IfConfig6 set process failed.\r
\r
**/\r
SHELL_STATUS\r
IfConfig6SetInterfaceInfo (\r
  IN LIST_ENTRY    *IfList,\r
  IN ARG_LIST      *VarArg\r
  )\r
{\r
  EFI_STATUS                       Status;\r
  SHELL_STATUS                     ShellStatus;\r
  IFCONFIG6_INTERFACE_CB           *IfCb;\r
  EFI_IP6_CONFIG_MANUAL_ADDRESS    *CfgManAddr;\r
  EFI_IPv6_ADDRESS                 *CfgAddr;\r
  UINTN                            AddrSize;\r
  EFI_IP6_CONFIG_INTERFACE_ID      *InterfaceId;\r
  UINT32                           DadXmits;\r
  UINT32                           CurDadXmits;\r
  UINTN                            CurDadXmitsLen;\r
  EFI_IP6_CONFIG_POLICY            Policy;\r
\r
  VAR_CHECK_CODE                   CheckCode;\r
  EFI_EVENT                        TimeOutEvt;\r
  EFI_EVENT                        MappedEvt;\r
  BOOLEAN                          IsAddressOk;\r
\r
  UINTN                            DataSize;\r
  UINT32                           Index;\r
  UINT32                           Index2;\r
  BOOLEAN                          IsAddressSet;\r
  EFI_IP6_CONFIG_INTERFACE_INFO    *IfInfo;\r
\r
  CfgManAddr  = NULL;\r
  CfgAddr     = NULL;\r
  TimeOutEvt  = NULL;\r
  MappedEvt   = NULL;\r
  IfInfo      = NULL;\r
  InterfaceId = NULL;\r
  CurDadXmits = 0;\r
\r
  if (IsListEmpty (IfList)) {\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_INTERFACE), gShellNetwork2HiiHandle);\r
    return SHELL_INVALID_PARAMETER;\r
  }\r
  //\r
  // Make sure to set only one interface each time.\r
  //\r
  IfCb   = BASE_CR (IfList->ForwardLink, IFCONFIG6_INTERFACE_CB, Link);\r
  Status = EFI_SUCCESS;\r
  ShellStatus = SHELL_SUCCESS;\r
\r
  //\r
  // Initialize check list mechanism.\r
  //\r
  CheckCode = IfConfig6RetriveCheckListByName(\r
                NULL,\r
                NULL,\r
                TRUE\r
                );\r
\r
  //\r
  // Create events & timers for asynchronous settings.\r
  //\r
  Status = gBS->CreateEvent (\r
                  EVT_TIMER,\r
                  TPL_CALLBACK,\r
                  NULL,\r
                  NULL,\r
                  &TimeOutEvt\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    ShellStatus = SHELL_ACCESS_DENIED;\r
    goto ON_EXIT;\r
  }\r
\r
  Status = gBS->CreateEvent (\r
                  EVT_NOTIFY_SIGNAL,\r
                  TPL_NOTIFY,\r
                  IfConfig6ManualAddressNotify,\r
                  &IsAddressOk,\r
                  &MappedEvt\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    ShellStatus = SHELL_ACCESS_DENIED;\r
    goto ON_EXIT;\r
  }\r
  //\r
  // Parse the setting variables.\r
  //\r
  while (VarArg != NULL) {\r
     //\r
     // Check invalid parameters (duplication & unknown & conflict).\r
     //\r
    CheckCode = IfConfig6RetriveCheckListByName(\r
                  mIfConfig6SetCheckList,\r
                  VarArg->Arg,\r
                  FALSE\r
                  );\r
\r
    if (VarCheckOk != CheckCode) {\r
      switch (CheckCode) {\r
        case VarCheckDuplicate:\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_DUPLICATE_COMMAND), gShellNetwork2HiiHandle, VarArg->Arg);\r
          break;\r
\r
        case VarCheckConflict:\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_CONFLICT_COMMAND), gShellNetwork2HiiHandle, VarArg->Arg);\r
          break;\r
\r
        case VarCheckUnknown:\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_UNKNOWN_COMMAND), gShellNetwork2HiiHandle, VarArg->Arg);\r
          break;\r
\r
        default:\r
          break;\r
      }\r
\r
      VarArg = VarArg->Next;\r
      continue;\r
    }\r
    //\r
    // Process valid variables.\r
    //\r
    if (StrCmp(VarArg->Arg, L"auto") == 0) {\r
      //\r
      // Set automaic config policy\r
      //\r
      Policy = Ip6ConfigPolicyAutomatic;\r
      Status = IfCb->IfCfg->SetData (\r
                              IfCb->IfCfg,\r
                              Ip6ConfigDataTypePolicy,\r
                              sizeof (EFI_IP6_CONFIG_POLICY),\r
                              &Policy\r
                              );\r
\r
      if (EFI_ERROR(Status)) {\r
        ShellStatus = SHELL_ACCESS_DENIED;\r
        goto ON_EXIT;\r
      }\r
\r
      VarArg= VarArg->Next;\r
\r
      if (VarArg != NULL) {\r
        if (StrCmp (VarArg->Arg, L"host") == 0) {\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_IP_CONFIG), gShellNetwork2HiiHandle, Status);\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
          goto ON_EXIT;\r
        } else if (StrCmp (VarArg->Arg, L"gw") == 0) {\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_GW_CONFIG), gShellNetwork2HiiHandle, Status);\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
          goto ON_EXIT;\r
        } else if (StrCmp (VarArg->Arg, L"dns") == 0) {\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_DNS_CONFIG), gShellNetwork2HiiHandle, Status);\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
          goto ON_EXIT;\r
        }\r
      }\r
    } else if (StrCmp (VarArg->Arg, L"man") == 0) {\r
      //\r
      // Set manual config policy.\r
      //\r
      Policy = Ip6ConfigPolicyManual;\r
      Status = IfCb->IfCfg->SetData (\r
                              IfCb->IfCfg,\r
                              Ip6ConfigDataTypePolicy,\r
                              sizeof (EFI_IP6_CONFIG_POLICY),\r
                              &Policy\r
                              );\r
\r
      if (EFI_ERROR(Status)) {\r
        ShellStatus = SHELL_ACCESS_DENIED;\r
        goto ON_EXIT;\r
      }\r
\r
      VarArg= VarArg->Next;\r
\r
    } else if (StrCmp (VarArg->Arg, L"host") == 0) {\r
      //\r
      // Parse till the next tag or the end of command line.\r
      //\r
      VarArg = VarArg->Next;\r
      Status = IfConfig6ParseManualAddressList (\r
                 &VarArg,\r
                 &CfgManAddr,\r
                 &AddrSize\r
                 );\r
\r
      if (EFI_ERROR (Status)) {\r
        if (Status == EFI_INVALID_PARAMETER) {\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_ARGUMENTS), gShellNetwork2HiiHandle, L"host");\r
          continue;\r
        } else {\r
          ShellStatus = SHELL_ACCESS_DENIED;\r
          goto ON_EXIT;\r
        }\r
      }\r
      //\r
      // Set static host ip6 address list.\r
      //   This is a asynchronous process.\r
      //\r
      IsAddressOk = FALSE;\r
\r
      Status = IfCb->IfCfg->RegisterDataNotify (\r
                              IfCb->IfCfg,\r
                              Ip6ConfigDataTypeManualAddress,\r
                              MappedEvt\r
                              );\r
      if (EFI_ERROR (Status)) {\r
        ShellStatus = SHELL_ACCESS_DENIED;\r
        goto ON_EXIT;\r
      }\r
\r
      Status = IfCb->IfCfg->SetData (\r
                              IfCb->IfCfg,\r
                              Ip6ConfigDataTypeManualAddress,\r
                              AddrSize,\r
                              CfgManAddr\r
                              );\r
\r
      if (Status == EFI_NOT_READY) {\r
        //\r
        // Get current dad transmits count.\r
        //\r
        CurDadXmitsLen = sizeof (EFI_IP6_CONFIG_DUP_ADDR_DETECT_TRANSMITS);\r
        IfCb->IfCfg->GetData (\r
                       IfCb->IfCfg,\r
                       Ip6ConfigDataTypeDupAddrDetectTransmits,\r
                       &CurDadXmitsLen,\r
                       &CurDadXmits\r
                       );\r
\r
        gBS->SetTimer (TimeOutEvt, TimerRelative, 50000000 + 10000000 * CurDadXmits);\r
\r
        while (EFI_ERROR (gBS->CheckEvent (TimeOutEvt))) {\r
          if (IsAddressOk) {\r
            Status = EFI_SUCCESS;\r
            break;\r
          }\r
        }\r
      }\r
\r
      IfCb->IfCfg->UnregisterDataNotify (\r
                     IfCb->IfCfg,\r
                     Ip6ConfigDataTypeManualAddress,\r
                     MappedEvt\r
                     );\r
\r
      if (EFI_ERROR (Status)) {\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_MAN_HOST), gShellNetwork2HiiHandle, Status);\r
        ShellStatus = SHELL_ACCESS_DENIED;\r
        goto ON_EXIT;\r
      }\r
\r
      //\r
      // Check whether the address is set successfully.\r
      //\r
      DataSize = 0;\r
\r
      Status = IfCb->IfCfg->GetData (\r
                              IfCb->IfCfg,\r
                              Ip6ConfigDataTypeInterfaceInfo,\r
                              &DataSize,\r
                              NULL\r
                              );\r
\r
      if (Status != EFI_BUFFER_TOO_SMALL) {\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), gShellNetwork2HiiHandle, Status);\r
        ShellStatus = SHELL_ACCESS_DENIED;\r
        goto ON_EXIT;\r
      }\r
\r
      IfInfo = AllocateZeroPool (DataSize);\r
\r
      if (IfInfo == NULL) {\r
        ShellStatus = SHELL_OUT_OF_RESOURCES;\r
        goto ON_EXIT;\r
      }\r
\r
      Status = IfCb->IfCfg->GetData (\r
                              IfCb->IfCfg,\r
                              Ip6ConfigDataTypeInterfaceInfo,\r
                              &DataSize,\r
                              IfInfo\r
                              );\r
\r
      if (EFI_ERROR (Status)) {\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), gShellNetwork2HiiHandle, Status);\r
        ShellStatus = SHELL_ACCESS_DENIED;\r
        goto ON_EXIT;\r
      }\r
\r
      for ( Index = 0; Index < (UINTN) (AddrSize / sizeof (EFI_IP6_CONFIG_MANUAL_ADDRESS)); Index++) {\r
        IsAddressSet = FALSE;\r
        //\r
        // By default, the prefix length 0 is regarded as 64.\r
        //\r
        if (CfgManAddr[Index].PrefixLength == 0) {\r
          CfgManAddr[Index].PrefixLength = 64;\r
        }\r
\r
        for (Index2 = 0; Index2 < IfInfo->AddressInfoCount; Index2++) {\r
          if (EFI_IP6_EQUAL (&IfInfo->AddressInfo[Index2].Address, &CfgManAddr[Index].Address) &&\r
              (IfInfo->AddressInfo[Index2].PrefixLength == CfgManAddr[Index].PrefixLength)) {\r
            IsAddressSet = TRUE;\r
            break;\r
          }\r
        }\r
\r
        if (!IsAddressSet) {\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_ADDRESS_FAILED), gShellNetwork2HiiHandle);\r
          IfConfig6PrintIpAddr (\r
            &CfgManAddr[Index].Address,\r
            &CfgManAddr[Index].PrefixLength\r
            );\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_NEWLINE), gShellNetwork2HiiHandle);\r
        }\r
      }\r
\r
    } else if (StrCmp (VarArg->Arg, L"gw") == 0) {\r
      //\r
      // Parse till the next tag or the end of command line.\r
      //\r
      VarArg = VarArg->Next;\r
      Status = IfConfig6ParseGwDnsAddressList (\r
                 &VarArg,\r
                 &CfgAddr,\r
                 &AddrSize\r
                 );\r
\r
      if (EFI_ERROR (Status)) {\r
        if (Status == EFI_INVALID_PARAMETER) {\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_ARGUMENTS), gShellNetwork2HiiHandle, L"gw");\r
          continue;\r
        } else {\r
          ShellStatus = SHELL_ACCESS_DENIED;\r
          goto ON_EXIT;\r
        }\r
      }\r
      //\r
      // Set static gateway ip6 address list.\r
      //\r
      Status = IfCb->IfCfg->SetData (\r
                              IfCb->IfCfg,\r
                              Ip6ConfigDataTypeGateway,\r
                              AddrSize,\r
                              CfgAddr\r
                              );\r
\r
      if (EFI_ERROR (Status)) {\r
        ShellStatus = SHELL_ACCESS_DENIED;\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_MAN_GW), gShellNetwork2HiiHandle, Status);\r
        goto ON_EXIT;\r
      }\r
\r
    } else if (StrCmp (VarArg->Arg, L"dns") == 0) {\r
      //\r
      // Parse till the next tag or the end of command line.\r
      //\r
      VarArg = VarArg->Next;\r
      Status = IfConfig6ParseGwDnsAddressList (\r
                 &VarArg,\r
                 &CfgAddr,\r
                 &AddrSize\r
                 );\r
\r
      if (EFI_ERROR (Status)) {\r
        if (Status == EFI_INVALID_PARAMETER) {\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_ARGUMENTS), gShellNetwork2HiiHandle, L"dns");\r
          continue;\r
        } else {\r
          ShellStatus = SHELL_ACCESS_DENIED;\r
          goto ON_EXIT;\r
        }\r
      }\r
      //\r
      // Set static DNS server ip6 address list.\r
      //\r
      Status = IfCb->IfCfg->SetData (\r
                              IfCb->IfCfg,\r
                              Ip6ConfigDataTypeDnsServer,\r
                              AddrSize,\r
                              CfgAddr\r
                              );\r
\r
      if (EFI_ERROR (Status)) {\r
        ShellStatus = SHELL_ACCESS_DENIED;\r
        goto ON_EXIT;\r
      }\r
\r
    } else if (StrCmp (VarArg->Arg, L"id") == 0) {\r
      //\r
      // Parse till the next tag or the end of command line.\r
      //\r
      VarArg = VarArg->Next;\r
      Status = IfConfig6ParseInterfaceId (&VarArg, &InterfaceId);\r
\r
      if (EFI_ERROR (Status)) {\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
        goto ON_EXIT;\r
      }\r
      //\r
      // Set alternative interface id.\r
      //\r
      Status = IfCb->IfCfg->SetData (\r
                              IfCb->IfCfg,\r
                              Ip6ConfigDataTypeAltInterfaceId,\r
                              sizeof (EFI_IP6_CONFIG_INTERFACE_ID),\r
                              InterfaceId\r
                              );\r
\r
      if (EFI_ERROR (Status)) {\r
        ShellStatus = SHELL_ACCESS_DENIED;\r
        goto ON_EXIT;\r
      }\r
\r
    } else if (StrCmp (VarArg->Arg, L"dad") == 0) {\r
      //\r
      // Parse till the next tag or the end of command line.\r
      //\r
      VarArg = VarArg->Next;\r
      Status = IfConfig6ParseDadXmits (&VarArg, &DadXmits);\r
\r
      if (EFI_ERROR (Status)) {\r
        ShellStatus = SHELL_ACCESS_DENIED;\r
        goto ON_EXIT;\r
      }\r
      //\r
      // Set dad transmits count.\r
      //\r
      Status = IfCb->IfCfg->SetData (\r
                              IfCb->IfCfg,\r
                              Ip6ConfigDataTypeDupAddrDetectTransmits,\r
                              sizeof (EFI_IP6_CONFIG_DUP_ADDR_DETECT_TRANSMITS),\r
                              &DadXmits\r
                              );\r
\r
      if (EFI_ERROR(Status)) {\r
        ShellStatus = SHELL_ACCESS_DENIED;\r
        goto ON_EXIT;\r
      }\r
    }\r
  }\r
\r
ON_EXIT:\r
\r
  if (CfgManAddr != NULL) {\r
    FreePool (CfgManAddr);\r
  }\r
\r
  if (CfgAddr != NULL) {\r
    FreePool (CfgAddr);\r
  }\r
\r
  if (MappedEvt != NULL) {\r
    gBS->CloseEvent (MappedEvt);\r
  }\r
\r
  if (TimeOutEvt != NULL) {\r
    gBS->CloseEvent (TimeOutEvt);\r
  }\r
\r
  if (IfInfo != NULL) {\r
    FreePool (IfInfo);\r
  }\r
\r
  return ShellStatus;\r
\r
}\r
\r
/**\r
  The IfConfig6 main process.\r
\r
  @param[in]   Private    The pointer of IFCONFIG6_PRIVATE_DATA.\r
\r
  @retval SHELL_SUCCESS   IfConfig6 processed successfully.\r
  @retval others          The IfConfig6 process failed.\r
\r
**/\r
SHELL_STATUS\r
IfConfig6 (\r
  IN IFCONFIG6_PRIVATE_DATA    *Private\r
  )\r
{\r
  EFI_STATUS    Status;\r
  SHELL_STATUS  ShellStatus;\r
\r
  ShellStatus = SHELL_SUCCESS;\r
\r
  //\r
  // Get configure information of all interfaces.\r
  //\r
  Status = IfConfig6GetInterfaceInfo (\r
             Private->ImageHandle,\r
             Private->IfName,\r
             &Private->IfList\r
             );\r
\r
  if (EFI_ERROR (Status)) {\r
    ShellStatus = SHELL_NOT_FOUND;\r
    goto ON_EXIT;\r
  }\r
\r
  switch (Private->OpCode) {\r
  case IfConfig6OpList:\r
    ShellStatus = IfConfig6ShowInterfaceInfo (&Private->IfList);\r
    break;\r
\r
  case IfConfig6OpClear:\r
    ShellStatus = IfConfig6ClearInterfaceInfo (&Private->IfList, Private->IfName);\r
    break;\r
\r
  case IfConfig6OpSet:\r
    ShellStatus = IfConfig6SetInterfaceInfo (&Private->IfList, Private->VarArg);\r
    break;\r
\r
  default:\r
    ShellStatus = SHELL_UNSUPPORTED;\r
  }\r
\r
ON_EXIT:\r
\r
  return ShellStatus;\r
}\r
\r
/**\r
  The IfConfig6 cleanup process, free the allocated memory.\r
\r
  @param[in]   Private    The pointer of  IFCONFIG6_PRIVATE_DATA.\r
\r
**/\r
VOID\r
IfConfig6Cleanup (\r
  IN IFCONFIG6_PRIVATE_DATA    *Private\r
  )\r
{\r
  LIST_ENTRY                *Entry;\r
  LIST_ENTRY                *NextEntry;\r
  IFCONFIG6_INTERFACE_CB    *IfCb;\r
\r
  ASSERT (Private != NULL);\r
\r
  //\r
  // Clean the list which save the set config Args.\r
  //\r
  if (Private->VarArg != NULL) {\r
    IfConfig6FreeArgList (Private->VarArg);\r
  }\r
\r
  if (Private->IfName != NULL)\r
    FreePool (Private->IfName);\r
\r
\r
  //\r
  // Clean the IFCONFIG6_INTERFACE_CB list.\r
  //\r
  Entry     = Private->IfList.ForwardLink;\r
  NextEntry = Entry->ForwardLink;\r
\r
  while (Entry != &Private->IfList) {\r
\r
    IfCb = BASE_CR (Entry, IFCONFIG6_INTERFACE_CB, Link);\r
\r
    RemoveEntryList (&IfCb->Link);\r
\r
    if (IfCb->IfId != NULL) {\r
\r
      FreePool (IfCb->IfId);\r
    }\r
\r
    if (IfCb->IfInfo != NULL) {\r
\r
      FreePool (IfCb->IfInfo);\r
    }\r
\r
    FreePool (IfCb);\r
\r
    Entry     = NextEntry;\r
    NextEntry = Entry->ForwardLink;\r
  }\r
\r
  FreePool (Private);\r
}\r
\r
/**\r
  Function for 'ifconfig6' command.\r
\r
  @param[in] ImageHandle  Handle to the Image (NULL if Internal).\r
  @param[in] SystemTable  Pointer to the System Table (NULL if Internal).\r
\r
  @retval SHELL_SUCCESS   ifconfig6 command processed successfully.\r
  @retval others          The ifconfig6 command process failed.\r
\r
**/\r
SHELL_STATUS\r
EFIAPI\r
ShellCommandRunIfconfig6 (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  EFI_STATUS                Status;\r
  SHELL_STATUS              ShellStatus;\r
  IFCONFIG6_PRIVATE_DATA    *Private;\r
  LIST_ENTRY                *ParamPackage;\r
  CONST CHAR16              *ValueStr;\r
  ARG_LIST                  *ArgList;\r
  CHAR16                    *ProblemParam;\r
  CHAR16                    *Str;\r
\r
  Private = NULL;\r
  Status = EFI_INVALID_PARAMETER;\r
  ShellStatus = SHELL_SUCCESS;\r
\r
  Status = ShellCommandLineParseEx (mIfConfig6CheckList, &ParamPackage, &ProblemParam, TRUE, FALSE);\r
  if (EFI_ERROR (Status)) {\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_COMMAND), gShellNetwork2HiiHandle, L"ifconfig6", ProblemParam);\r
    ShellStatus = SHELL_INVALID_PARAMETER;\r
    goto ON_EXIT;\r
  }\r
\r
  //\r
  // To handle no option.\r
  //\r
  if (!ShellCommandLineGetFlag (ParamPackage, L"-r") && !ShellCommandLineGetFlag (ParamPackage, L"-s") &&\r
      !ShellCommandLineGetFlag (ParamPackage, L"-?") && !ShellCommandLineGetFlag (ParamPackage, L"-l")) {\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_LACK_OPTION), gShellNetwork2HiiHandle);\r
    ShellStatus = SHELL_INVALID_PARAMETER;\r
    goto ON_EXIT;\r
  }\r
  //\r
  // To handle conflict options.\r
  //\r
  if (((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-s"))) ||\r
      ((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-l"))) ||\r
      ((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-?"))) ||\r
      ((ShellCommandLineGetFlag (ParamPackage, L"-s")) && (ShellCommandLineGetFlag (ParamPackage, L"-l"))) ||\r
      ((ShellCommandLineGetFlag (ParamPackage, L"-s")) && (ShellCommandLineGetFlag (ParamPackage, L"-?"))) ||\r
      ((ShellCommandLineGetFlag (ParamPackage, L"-l")) && (ShellCommandLineGetFlag (ParamPackage, L"-?")))) {\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_CONFLICT_OPTIONS), gShellNetwork2HiiHandle);\r
    ShellStatus = SHELL_INVALID_PARAMETER;\r
    goto ON_EXIT;\r
  }\r
\r
  Private = AllocateZeroPool (sizeof (IFCONFIG6_PRIVATE_DATA));\r
\r
  if (Private == NULL) {\r
    ShellStatus = SHELL_OUT_OF_RESOURCES;\r
    goto ON_EXIT;\r
  }\r
\r
  InitializeListHead (&Private->IfList);\r
\r
  //\r
  // To get interface name for the list option.\r
  //\r
  if (ShellCommandLineGetFlag (ParamPackage, L"-l")) {\r
    Private->OpCode = IfConfig6OpList;\r
    ValueStr = ShellCommandLineGetValue (ParamPackage, L"-l");\r
    if (ValueStr != NULL) {\r
      Str = AllocateCopyPool (StrSize (ValueStr), ValueStr);\r
      if (Str == NULL) {\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellNetwork2HiiHandle, L"ifconfig6");\r
        ShellStatus = SHELL_OUT_OF_RESOURCES;\r
        goto ON_EXIT;\r
      }\r
      Private->IfName = Str;\r
    }\r
  }\r
  //\r
  // To get interface name for the clear option.\r
  //\r
  if (ShellCommandLineGetFlag (ParamPackage, L"-r")) {\r
    Private->OpCode = IfConfig6OpClear;\r
    ValueStr = ShellCommandLineGetValue (ParamPackage, L"-r");\r
    if (ValueStr != NULL) {\r
      Str = AllocateCopyPool (StrSize (ValueStr), ValueStr);\r
      if (Str == NULL) {\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellNetwork2HiiHandle, L"ifconfig6");\r
        ShellStatus = SHELL_OUT_OF_RESOURCES;\r
        goto ON_EXIT;\r
      }\r
      Private->IfName = Str;\r
    }\r
  }\r
  //\r
  // To get interface name and corresponding Args for the set option.\r
  //\r
  if (ShellCommandLineGetFlag (ParamPackage, L"-s")) {\r
\r
    ValueStr = ShellCommandLineGetValue (ParamPackage, L"-s");\r
    if (ValueStr == NULL) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_INTERFACE), gShellNetwork2HiiHandle);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
      goto ON_EXIT;\r
    }\r
    //\r
    // To split the configuration into multi-section.\r
    //\r
    ArgList = IfConfig6SplitStrToList (ValueStr, L' ');\r
    if (ArgList == NULL) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellNetwork2HiiHandle, L"ifconfig6");\r
      ShellStatus = SHELL_OUT_OF_RESOURCES;\r
      goto ON_EXIT;\r
    }\r
\r
    Private->OpCode = IfConfig6OpSet;\r
    Private->IfName = ArgList->Arg;\r
\r
    Private->VarArg = ArgList->Next;\r
\r
    if (Private->IfName == NULL || Private->VarArg == NULL) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_COMMAND), gShellNetwork2HiiHandle);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
      goto ON_EXIT;\r
    }\r
  }\r
  //\r
  // Main process of ifconfig6.\r
  //\r
  ShellStatus = IfConfig6 (Private);\r
\r
ON_EXIT:\r
\r
  ShellCommandLineFreeVarList (ParamPackage);\r
  if (Private != NULL) {\r
    IfConfig6Cleanup (Private);\r
  }\r
  return ShellStatus;\r
\r
}\r
\r