NetworkPkg/Application/IfConfig6/IfConfig6.c

   1 /** @file
   2   The implementation for Shell application IfConfig6.
   3
   4   Copyright (c) 2009 - 2011, Intel Corporation. All rights reserved.<BR>
   5
   6   This program and the accompanying materials
   7   are licensed and made available under the terms and conditions of the BSD License
   8   which accompanies this distribution.  The full text of the license may be found at
   9   http://opensource.org/licenses/bsd-license.php.
  10
  11   THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
  12   WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
  13
  14 **/
  15
  16 #include <Library/ShellLib.h>
  17 #include <Library/BaseMemoryLib.h>
  18 #include <Library/BaseLib.h>
  19 #include <Library/MemoryAllocationLib.h>
  20 #include <Library/DebugLib.h>
  21 #include <Library/UefiBootServicesTableLib.h>
  22 #include <Library/HiiLib.h>
  23 #include <Library/NetLib.h>
  24
  25 #include <Protocol/Ip6.h>
  26 #include <Protocol/Ip6Config.h>
  27
  28 #include "IfConfig6.h"
  29
  30 EFI_HII_HANDLE      mHiiHandle;
  31
  32 SHELL_PARAM_ITEM    mIfConfig6CheckList[] = {
  33   {
  34     L"-b",
  35     TypeFlag
  36   },
  37   {
  38     L"-s",
  39     TypeMaxValue
  40   },
  41   {
  42     L"-l",
  43     TypeValue
  44   },
  45   {
  46     L"-r",
  47     TypeValue
  48   },
  49   {
  50     L"-?",
  51     TypeFlag
  52   },
  53   {
  54     NULL,
  55     TypeMax
  56   },
  57 };
  58
  59 VAR_CHECK_ITEM  mSetCheckList[] = {
  60   {
  61    L"auto",
  62     0x00000001,
  63     0x00000001,
  64     FlagTypeSingle
  65   },
  66   {
  67     L"man",
  68     0x00000002,
  69     0x00000001,
  70     FlagTypeSingle
  71   },
  72   {
  73     L"host",
  74     0x00000004,
  75     0x00000002,
  76     FlagTypeSingle
  77   },
  78   {
  79     L"dad",
  80     0x00000008,
  81     0x00000004,
  82     FlagTypeSingle
  83   },
  84   {
  85     L"gw",
  86     0x00000010,
  87     0x00000008,
  88     FlagTypeSingle
  89   },
  90   {
  91     L"dns",
  92     0x00000020,
  93     0x00000010,
  94     FlagTypeSingle
  95   },
  96   {
  97     L"id",
  98     0x00000040,
  99     0x00000020,
 100     FlagTypeSingle
 101   },
 102   {
 103     NULL,
 104     0x0,
 105     0x0,
 106     FlagTypeSkipUnknown
 107   },
 108 };
 109
 110 /**
 111   Split a string with specified separator and save the substring to a list.
 112
 113   @param[in]    String       The pointer of the input string.
 114   @param[in]    Separator    The specified separator.
 115
 116   @return The pointer of headnode of ARG_LIST.
 117
 118 **/
 119 ARG_LIST *
 120 SplitStrToList (
 121   IN CONST CHAR16    *String,
 122   IN CHAR16          Separator
 123   )
 124 {
 125   CHAR16      *Str;
 126   CHAR16      *ArgStr;
 127   ARG_LIST    *ArgList;
 128   ARG_LIST    *ArgNode;
 129
 130   if (*String == L'\0') {
 131     return NULL;
 132   }
 133
 134   //
 135   // Copy the CONST string to a local copy.
 136   //
 137   Str     = (CHAR16 *) AllocateZeroPool (StrSize (String));
 138   ASSERT (Str != NULL);
 139   Str     = StrCpy (Str, String);
 140   ArgStr  = Str;
 141
 142   //
 143   // init a node for the list head.
 144   //
 145   ArgNode = (ARG_LIST *) AllocateZeroPool (sizeof (ARG_LIST));
 146   ASSERT (ArgNode != NULL);
 147   ArgList = ArgNode;
 148
 149   //
 150   // Split the local copy and save in the list node.
 151   //
 152   while (*Str != L'\0') {
 153     if (*Str == Separator) {
 154       *Str          = L'\0';
 155       ArgNode->Arg  = ArgStr;
 156       ArgStr        = Str + 1;
 157       ArgNode->Next = (ARG_LIST *) AllocateZeroPool (sizeof (ARG_LIST));
 158       ASSERT (ArgNode->Next != NULL);
 159       ArgNode = ArgNode->Next;
 160     }
 161
 162     Str++;
 163   }
 164
 165   ArgNode->Arg  = ArgStr;
 166   ArgNode->Next = NULL;
 167
 168   return ArgList;
 169 }
 170
 171 /**
 172   Check the correctness of input Args with '-s' option.
 173
 174   @param[in]    CheckList    The pointer of VAR_CHECK_ITEM array.
 175   @param[in]    Name         The pointer of input arg.
 176   @param[in]    Init         The switch to execute the check.
 177
 178   @return The value of VAR_CHECK_CODE.
 179
 180 **/
 181 VAR_CHECK_CODE
 182 IfConfig6RetriveCheckListByName(
 183   IN VAR_CHECK_ITEM    *CheckList,
 184   IN CHAR16            *Name,
 185   IN BOOLEAN           Init
 186 )
 187 {
 188   STATIC UINT32     CheckDuplicate;
 189   STATIC UINT32     CheckConflict;
 190   VAR_CHECK_CODE    RtCode;
 191   UINT32            Index;
 192   VAR_CHECK_ITEM    Arg;
 193
 194   if (Init) {
 195     CheckDuplicate = 0;
 196     CheckConflict  = 0;
 197     return VarCheckOk;
 198   }
 199
 200   RtCode  = VarCheckOk;
 201   Index   = 0;
 202   Arg     = CheckList[Index];
 203
 204   //
 205   // Check the Duplicated/Conflicted/Unknown input Args.
 206   //
 207   while (Arg.FlagStr != NULL) {
 208     if (StrCmp (Arg.FlagStr, Name) == 0) {
 209
 210       if (CheckDuplicate & Arg.FlagID) {
 211         RtCode = VarCheckDuplicate;
 212         break;
 213       }
 214
 215       if (CheckConflict & Arg.ConflictMask) {
 216         RtCode = VarCheckConflict;
 217         break;
 218       }
 219
 220       CheckDuplicate |= Arg.FlagID;
 221       CheckConflict  |= Arg.ConflictMask;
 222       break;
 223     }
 224
 225     Arg = CheckList[++Index];
 226   }
 227
 228   if (Arg.FlagStr == NULL) {
 229     RtCode = VarCheckUnknown;
 230   }
 231
 232   return RtCode;
 233 }
 234
 235 /**
 236   The notify function of create event when performing a manual config.
 237
 238   @param[in]    Event        The event this notify function registered to.
 239   @param[in]    Context      Pointer to the context data registered to the event.
 240
 241 **/
 242 VOID
 243 EFIAPI
 244 IfConfig6ManualAddressNotify (
 245   IN EFI_EVENT    Event,
 246   IN VOID         *Context
 247   )
 248 {
 249   *((BOOLEAN *) Context) = TRUE;
 250 }
 251
 252 /**
 253   Print MAC address.
 254
 255   @param[in]    Node    The pointer of MAC address buffer.
 256   @param[in]    Size    The size of MAC address buffer.
 257
 258 **/
 259 VOID
 260 IfConfig6PrintMacAddr (
 261   IN UINT8     *Node,
 262   IN UINT32    Size
 263   )
 264 {
 265   UINTN    Index;
 266
 267   ASSERT (Size <= MACADDRMAXSIZE);
 268
 269   for (Index = 0; Index < Size; Index++) {
 270     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_MAC_ADDR_BODY), mHiiHandle, Node[Index]);
 271     if (Index + 1 < Size) {
 272       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_COLON), mHiiHandle);
 273     }
 274   }
 275
 276   ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_NEWLINE), mHiiHandle);
 277 }
 278
 279 /**
 280   Print IPv6 address.
 281
 282   @param[in]    Ip           The pointer of Ip bufffer in EFI_IPv6_ADDRESS format.
 283   @param[in]    PrefixLen    The pointer of PrefixLen that describes the size Prefix.
 284
 285 **/
 286 VOID
 287 IfConfig6PrintIpAddr (
 288   IN EFI_IPv6_ADDRESS    *Ip,
 289   IN UINT8               *PrefixLen
 290   )
 291 {
 292   UINTN      Index;
 293   BOOLEAN    Short;
 294
 295   Short = FALSE;
 296
 297   for (Index = 0; Index < PREFIXMAXLEN; Index = Index + 2) {
 298
 299     if (!Short && (Index + 1 < PREFIXMAXLEN) && (Index % 2 == 0) && (Ip->Addr[Index] == 0) && (Ip->Addr[Index + 1] == 0)) {
 300       //
 301       // Deal with the case of ::.
 302       //
 303       if (Index == 0) {
 304         //
 305         // :: is at the beginning of the address.
 306         //
 307         ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_COLON), mHiiHandle);
 308       }
 309       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_COLON), mHiiHandle);
 310
 311       while ((Ip->Addr[Index] == 0) && (Ip->Addr[Index + 1] == 0) && (Index < PREFIXMAXLEN)) {
 312         Index = Index + 2;
 313         if (Index > PREFIXMAXLEN - 2) {
 314           break;
 315         }
 316       }
 317
 318       Short = TRUE;
 319
 320       if (Index == PREFIXMAXLEN) {
 321         //
 322         // :: is at the end of the address.
 323         //
 324         break;
 325       }
 326     }
 327
 328     if (Index < PREFIXMAXLEN - 1) {
 329       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_IP_ADDR_BODY), mHiiHandle, Ip->Addr[Index]);
 330       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_IP_ADDR_BODY), mHiiHandle, Ip->Addr[Index + 1]);
 331     }
 332
 333     if (Index + 2 < PREFIXMAXLEN) {
 334       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_COLON), mHiiHandle);
 335     }
 336   }
 337
 338   if (PrefixLen != NULL) {
 339     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_PREFIX_LEN), mHiiHandle, *PrefixLen);
 340   }
 341 }
 342
 343 /**
 344   Pick up host IPv6 address in string format from Args with "-s" option and convert it to EFI_IP6_CONFIG_MANUAL_ADDRESS format.
 345
 346   @param[in, out]    Arg        The pointer of the address of ARG_LIST which save Args with the "-s" option.
 347   @param[out]        Buf        The pointer of the address of EFI_IP6_CONFIG_MANUAL_ADDRESS.
 348   @param[out]        BufSize    The pointer of BufSize that describes the size of Buf in bytes.
 349
 350   @retval EFI_SUCCESS    The convertion is successful.
 351   @retval Others         Does't find the host address, or it is an invalid IPv6 address in string format.
 352
 353 **/
 354 EFI_STATUS
 355 IfConfig6ParseManualAddressList (
 356   IN OUT ARG_LIST                         **Arg,
 357      OUT EFI_IP6_CONFIG_MANUAL_ADDRESS    **Buf,
 358      OUT UINTN                            *BufSize
 359   )
 360 {
 361   EFI_STATUS                       Status;
 362   EFI_IP6_CONFIG_MANUAL_ADDRESS    *AddrBuf;
 363   ARG_LIST                         *VarArg;
 364   EFI_IPv6_ADDRESS                 Address;
 365   UINT8                            Prefix;
 366   UINT8                            AddrCnt;
 367
 368   AddrCnt  = 0;
 369   *BufSize = 0;
 370   *Buf     = NULL;
 371   VarArg   = *Arg;
 372   Status   = EFI_SUCCESS;
 373
 374   //
 375   // Go through the list to check the correctness of input host ip6 address.
 376   //
 377   while ((!EFI_ERROR (Status)) && (VarArg != NULL)) {
 378
 379     Status = NetLibStrToIp6andPrefix (VarArg->Arg, &Address, &Prefix);
 380
 381     if (EFI_ERROR (Status)) {
 382       //
 383       // host ip ip ... gw
 384       //
 385       break;
 386     }
 387
 388     VarArg = VarArg->Next;
 389     AddrCnt++;
 390   }
 391
 392   if (AddrCnt == 0) {
 393     return EFI_INVALID_PARAMETER;
 394   }
 395
 396   AddrBuf = AllocateZeroPool (AddrCnt * sizeof (EFI_IP6_CONFIG_MANUAL_ADDRESS));
 397   ASSERT (AddrBuf != NULL);
 398
 399   AddrCnt = 0;
 400   VarArg  = *Arg;
 401   Status  = EFI_SUCCESS;
 402
 403   //
 404   // Go through the list to fill in the EFI_IP6_CONFIG_MANUAL_ADDRESS structure.
 405   //
 406   while ((!EFI_ERROR (Status)) && (VarArg != NULL)) {
 407
 408     Status = NetLibStrToIp6andPrefix (VarArg->Arg, &Address, &Prefix);
 409
 410     if (EFI_ERROR (Status)) {
 411       break;
 412     }
 413
 414     //
 415     // If prefix length is not set, set it as Zero here. In the IfConfigSetInterfaceInfo()
 416     // Zero prefix, length will be transfered to default prefix length.
 417     //
 418     if (Prefix == 0xFF) {
 419       Prefix = 0;
 420     }
 421     AddrBuf[AddrCnt].IsAnycast    = FALSE;
 422     AddrBuf[AddrCnt].PrefixLength = Prefix;
 423     IP6_COPY_ADDRESS (&AddrBuf[AddrCnt].Address, &Address);
 424     VarArg = VarArg->Next;
 425     AddrCnt++;
 426   }
 427
 428   *Arg = VarArg;
 429
 430   if (EFI_ERROR (Status) && (Status != EFI_INVALID_PARAMETER)) {
 431     goto ON_ERROR;
 432   }
 433
 434   *Buf     = AddrBuf;
 435   *BufSize = AddrCnt * sizeof (EFI_IP6_CONFIG_MANUAL_ADDRESS);
 436
 437   return EFI_SUCCESS;
 438
 439 ON_ERROR:
 440
 441   FreePool (AddrBuf);
 442   return Status;
 443 }
 444
 445 /**
 446   Pick up gw/dns IPv6 address in string format from Args with "-s" option and convert it to EFI_IPv6_ADDRESS format.
 447
 448   @param[in, out]    Arg        The pointer of the address of ARG_LIST that save Args with the "-s" option.
 449   @param[out]        Buf        The pointer of the address of EFI_IPv6_ADDRESS.
 450   @param[out]        BufSize    The pointer of BufSize that describes the size of Buf in bytes.
 451
 452   @retval EFI_SUCCESS    The conversion is successful.
 453   @retval Others         Doesn't find the host address, or it is an invalid IPv6 address in string format.
 454
 455 **/
 456 EFI_STATUS
 457 IfConfig6ParseGwDnsAddressList (
 458   IN OUT ARG_LIST            **Arg,
 459      OUT EFI_IPv6_ADDRESS    **Buf,
 460      OUT UINTN               *BufSize
 461   )
 462 {
 463   EFI_STATUS          Status;
 464   EFI_IPv6_ADDRESS    *AddrBuf;
 465   ARG_LIST            *VarArg;
 466   EFI_IPv6_ADDRESS    Address;
 467   UINT8               Prefix;
 468   UINT8               AddrCnt;
 469
 470   AddrCnt  = 0;
 471   *BufSize = 0;
 472   *Buf     = NULL;
 473   VarArg   = *Arg;
 474   Status   = EFI_SUCCESS;
 475
 476   //
 477   // Go through the list to check the correctness of input gw/dns address.
 478   //
 479   while ((!EFI_ERROR (Status)) && (VarArg != NULL)) {
 480
 481     Status = NetLibStrToIp6andPrefix (VarArg->Arg, &Address, &Prefix);
 482
 483     if (EFI_ERROR (Status)) {
 484       //
 485       // gw ip ip ... host
 486       //
 487       break;
 488     }
 489
 490     VarArg = VarArg->Next;
 491     AddrCnt++;
 492   }
 493
 494   if (AddrCnt == 0) {
 495     return EFI_INVALID_PARAMETER;
 496   }
 497
 498   AddrBuf = AllocateZeroPool (AddrCnt * sizeof (EFI_IPv6_ADDRESS));
 499   ASSERT (AddrBuf != NULL);
 500
 501   AddrCnt = 0;
 502   VarArg  = *Arg;
 503   Status  = EFI_SUCCESS;
 504
 505   //
 506   // Go through the list to fill in the EFI_IPv6_ADDRESS structure.
 507   //
 508   while ((!EFI_ERROR (Status)) && (VarArg != NULL)) {
 509
 510     Status = NetLibStrToIp6andPrefix (VarArg->Arg, &Address, &Prefix);
 511
 512     if (EFI_ERROR (Status)) {
 513       break;
 514     }
 515
 516     IP6_COPY_ADDRESS (&AddrBuf[AddrCnt], &Address);
 517
 518     VarArg = VarArg->Next;
 519     AddrCnt++;
 520   }
 521
 522   *Arg = VarArg;
 523
 524   if (EFI_ERROR (Status) && (Status != EFI_INVALID_PARAMETER)) {
 525    goto ON_ERROR;
 526   }
 527
 528   *Buf     = AddrBuf;
 529   *BufSize = AddrCnt * sizeof (EFI_IPv6_ADDRESS);
 530
 531   return EFI_SUCCESS;
 532
 533 ON_ERROR:
 534
 535   FreePool (AddrBuf);
 536   return Status;
 537 }
 538
 539 /**
 540   Parse InterfaceId in string format from Args with the "-s" option and convert it to EFI_IP6_CONFIG_INTERFACE_ID format.
 541
 542   @param[in, out]   Arg     The pointer of the address of ARG_LIST that saves Args with the "-s" option.
 543   @param[out]       IfId    The pointer of EFI_IP6_CONFIG_INTERFACE_ID.
 544
 545   @retval EFI_SUCCESS              The get status processed successfullly.
 546   @retval EFI_INVALID_PARAMETER    The get status process failed.
 547
 548 **/
 549 EFI_STATUS
 550 IfConfig6ParseInterfaceId (
 551   IN OUT ARG_LIST                       **Arg,
 552      OUT EFI_IP6_CONFIG_INTERFACE_ID    **IfId
 553   )
 554 {
 555   UINT8     Index;
 556   UINT8     NodeVal;
 557   CHAR16    *IdStr;
 558
 559   if (*Arg == NULL) {
 560     return EFI_INVALID_PARAMETER;
 561   }
 562
 563   Index = 0;
 564   IdStr = (*Arg)->Arg;
 565   ASSERT (IfId != NULL);
 566   *IfId = AllocateZeroPool (sizeof (EFI_IP6_CONFIG_INTERFACE_ID));
 567   ASSERT (*IfId != NULL);
 568
 569   while ((*IdStr != L'\0') && (Index < 8)) {
 570
 571     NodeVal = 0;
 572     while ((*IdStr != L':') && (*IdStr != L'\0')) {
 573
 574       if ((*IdStr <= L'F') && (*IdStr >= L'A')) {
 575         NodeVal = (UINT8)((NodeVal << 4) + *IdStr - L'A' + 10);
 576       } else if ((*IdStr <= L'f') && (*IdStr >= L'a')) {
 577         NodeVal = (UINT8)((NodeVal << 4) + *IdStr - L'a' + 10);
 578       } else if ((*IdStr <= L'9') && (*IdStr >= L'0')) {
 579         NodeVal = (UINT8)((NodeVal << 4) + *IdStr - L'0');
 580       } else {
 581         FreePool (*IfId);
 582         return EFI_INVALID_PARAMETER;
 583       }
 584
 585       IdStr++;
 586     }
 587
 588     (*IfId)->Id[Index++] = NodeVal;
 589
 590     if (*IdStr == L':') {
 591       IdStr++;
 592     }
 593   }
 594
 595   *Arg = (*Arg)->Next;
 596   return EFI_SUCCESS;
 597 }
 598
 599 /**
 600   Parse dad in string format from Args with the "-s" option and convert it to UINT32 format.
 601
 602   @param[in, out]   Arg      The pointer of the address of ARG_LIST that saves Args with the "-s" option.
 603   @param[out]       Xmits    The pointer of Xmits.
 604
 605   @retval EFI_SUCCESS    The get status processed successfully.
 606   @retval others         The get status process failed.
 607
 608 **/
 609 EFI_STATUS
 610 IfConfig6ParseDadXmits (
 611   IN OUT ARG_LIST    **Arg,
 612      OUT UINT32      *Xmits
 613   )
 614 {
 615   CHAR16    *ValStr;
 616
 617   if (*Arg == NULL) {
 618     return EFI_INVALID_PARAMETER;
 619   }
 620
 621   ValStr = (*Arg)->Arg;
 622   *Xmits = 0;
 623
 624   while (*ValStr != L'\0') {
 625
 626     if ((*ValStr <= L'9') && (*ValStr >= L'0')) {
 627
 628       *Xmits = (*Xmits * 10) + (*ValStr - L'0');
 629
 630     } else {
 631
 632       return EFI_INVALID_PARAMETER;
 633     }
 634
 635     ValStr++;
 636   }
 637
 638   *Arg = (*Arg)->Next;
 639   return EFI_SUCCESS;
 640 }
 641
 642 /**
 643   The get current status of all handles.
 644
 645   @param[in]   ImageHandle    The handle of  ImageHandle.
 646   @param[in]   IfName         The pointer of  IfName(interface name).
 647   @param[in]   IfList         The pointer of  IfList(interface list).
 648
 649   @retval EFI_SUCCESS    The get status processed successfully.
 650   @retval others         The get status process failed.
 651
 652 **/
 653 EFI_STATUS
 654 IfConfig6GetInterfaceInfo (
 655   IN EFI_HANDLE    ImageHandle,
 656   IN CHAR16        *IfName,
 657   IN LIST_ENTRY    *IfList
 658   )
 659 {
 660   EFI_STATUS                       Status;
 661   UINTN                            HandleIndex;
 662   UINTN                            HandleNum;
 663   EFI_HANDLE                       *HandleBuffer;
 664   EFI_IP6_CONFIG_PROTOCOL          *Ip6Cfg;
 665   EFI_IP6_CONFIG_INTERFACE_INFO    *IfInfo;
 666   IFCONFIG6_INTERFACE_CB           *IfCb;
 667   UINTN                            DataSize;
 668
 669   HandleBuffer = NULL;
 670   HandleNum    = 0;
 671
 672   IfInfo       = NULL;
 673   IfCb         = NULL;
 674
 675   //
 676   // Locate all the handles with ip6 service binding protocol.
 677   //
 678   Status = gBS->LocateHandleBuffer (
 679                   ByProtocol,
 680                   &gEfiIp6ServiceBindingProtocolGuid,
 681                   NULL,
 682                   &HandleNum,
 683                   &HandleBuffer
 684                  );
 685   if (EFI_ERROR (Status) || (HandleNum == 0)) {
 686     return EFI_ABORTED;
 687   }
 688
 689   //
 690   // Enumerate all handles that installed with ip6 service binding protocol.
 691   //
 692   for (HandleIndex = 0; HandleIndex < HandleNum; HandleIndex++) {
 693     IfCb      = NULL;
 694     IfInfo    = NULL;
 695     DataSize  = 0;
 696
 697     //
 698     // Ip6config protocol and ip6 service binding protocol are installed
 699     // on the same handle.
 700     //
 701     ASSERT (HandleBuffer != NULL);
 702     Status = gBS->HandleProtocol (
 703                     HandleBuffer[HandleIndex],
 704                     &gEfiIp6ConfigProtocolGuid,
 705                     (VOID **) &Ip6Cfg
 706                     );
 707
 708     if (EFI_ERROR (Status)) {
 709       goto ON_ERROR;
 710     }
 711     //
 712     // Get the interface information size.
 713     //
 714     Status = Ip6Cfg->GetData (
 715                        Ip6Cfg,
 716                        Ip6ConfigDataTypeInterfaceInfo,
 717                        &DataSize,
 718                        NULL
 719                        );
 720
 721     if (Status != EFI_BUFFER_TOO_SMALL) {
 722       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
 723       goto ON_ERROR;
 724     }
 725
 726     IfInfo = AllocateZeroPool (DataSize);
 727
 728     if (IfInfo == NULL) {
 729       Status = EFI_OUT_OF_RESOURCES;
 730       goto ON_ERROR;
 731     }
 732     //
 733     // Get the interface info.
 734     //
 735     Status = Ip6Cfg->GetData (
 736                        Ip6Cfg,
 737                        Ip6ConfigDataTypeInterfaceInfo,
 738                        &DataSize,
 739                        IfInfo
 740                        );
 741
 742     if (EFI_ERROR (Status)) {
 743       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
 744       goto ON_ERROR;
 745     }
 746     //
 747     // Check the interface name if required.
 748     //
 749     if ((IfName != NULL) && (StrCmp (IfName, IfInfo->Name) != 0)) {
 750       FreePool (IfInfo);
 751       continue;
 752     }
 753
 754     DataSize = 0;
 755     //
 756     // Get the size of dns server list.
 757     //
 758     Status = Ip6Cfg->GetData (
 759                        Ip6Cfg,
 760                        Ip6ConfigDataTypeDnsServer,
 761                        &DataSize,
 762                        NULL
 763                        );
 764
 765     if ((Status != EFI_BUFFER_TOO_SMALL) && (Status != EFI_NOT_FOUND)) {
 766       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
 767       goto ON_ERROR;
 768     }
 769
 770     IfCb = AllocateZeroPool (sizeof (IFCONFIG6_INTERFACE_CB) + DataSize);
 771
 772     if (IfCb == NULL) {
 773       Status = EFI_OUT_OF_RESOURCES;
 774       goto ON_ERROR;
 775     }
 776
 777     IfCb->NicHandle = HandleBuffer[HandleIndex];
 778     IfCb->IfInfo    = IfInfo;
 779     IfCb->IfCfg     = Ip6Cfg;
 780     IfCb->DnsCnt    = (UINT32) (DataSize / sizeof (EFI_IPv6_ADDRESS));
 781
 782     //
 783     // Get the dns server list if has.
 784     //
 785     if (DataSize > 0) {
 786
 787       Status = Ip6Cfg->GetData (
 788                          Ip6Cfg,
 789                          Ip6ConfigDataTypeDnsServer,
 790                          &DataSize,
 791                          IfCb->DnsAddr
 792                          );
 793
 794       if (EFI_ERROR (Status)) {
 795         ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
 796         goto ON_ERROR;
 797       }
 798     }
 799     //
 800     // Get the interface id if has.
 801     //
 802     DataSize   = sizeof (EFI_IP6_CONFIG_INTERFACE_ID);
 803     IfCb->IfId = AllocateZeroPool (DataSize);
 804
 805     if (IfCb->IfId == NULL) {
 806       goto ON_ERROR;
 807     }
 808
 809     Status = Ip6Cfg->GetData (
 810                        Ip6Cfg,
 811                        Ip6ConfigDataTypeAltInterfaceId,
 812                        &DataSize,
 813                        IfCb->IfId
 814                        );
 815
 816     if (EFI_ERROR (Status) && (Status != EFI_NOT_FOUND)) {
 817       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
 818       goto ON_ERROR;
 819     }
 820
 821     if (Status == EFI_NOT_FOUND) {
 822       FreePool (IfCb->IfId);
 823       IfCb->IfId = NULL;
 824     }
 825     //
 826     // Get the config policy.
 827     //
 828     DataSize = sizeof (EFI_IP6_CONFIG_POLICY);
 829     Status   = Ip6Cfg->GetData (
 830                          Ip6Cfg,
 831                          Ip6ConfigDataTypePolicy,
 832                          &DataSize,
 833                          &IfCb->Policy
 834                          );
 835
 836     if (EFI_ERROR (Status)) {
 837       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
 838       goto ON_ERROR;
 839     }
 840     //
 841     // Get the dad transmits.
 842     //
 843     DataSize = sizeof (EFI_IP6_CONFIG_DUP_ADDR_DETECT_TRANSMITS);
 844     Status   = Ip6Cfg->GetData (
 845                          Ip6Cfg,
 846                          Ip6ConfigDataTypeDupAddrDetectTransmits,
 847                          &DataSize,
 848                          &IfCb->Xmits
 849                          );
 850
 851     if (EFI_ERROR (Status)) {
 852       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
 853       goto ON_ERROR;
 854     }
 855
 856     InsertTailList (IfList, &IfCb->Link);
 857
 858     if ((IfName != NULL) && (StrCmp (IfName, IfInfo->Name) == 0)) {
 859       //
 860       // Only need the appointed interface, keep the allocated buffer.
 861       //
 862       IfCb   = NULL;
 863       IfInfo = NULL;
 864       break;
 865     }
 866   }
 867
 868   if (HandleBuffer != NULL) {
 869     FreePool (HandleBuffer);
 870   }
 871
 872   return EFI_SUCCESS;
 873
 874 ON_ERROR:
 875
 876   if (IfInfo != NULL) {
 877     FreePool (IfInfo);
 878   }
 879
 880   if (IfCb != NULL) {
 881     if (IfCb->IfId != NULL) {
 882       FreePool (IfCb->IfId);
 883     }
 884
 885     FreePool (IfCb);
 886   }
 887
 888   return Status;
 889 }
 890
 891 /**
 892   The list process of the IfConfig6 application.
 893
 894   @param[in]   IfList    The pointer of IfList(interface list).
 895
 896   @retval EFI_SUCCESS    The IfConfig6 list processed successfully.
 897   @retval others         The IfConfig6 list process failed.
 898
 899 **/
 900 EFI_STATUS
 901 IfConfig6ShowInterfaceInfo (
 902   IN LIST_ENTRY    *IfList
 903   )
 904 {
 905   EFI_STATUS                Status;
 906   LIST_ENTRY                *Entry;
 907   IFCONFIG6_INTERFACE_CB    *IfCb;
 908   UINTN                     Index;
 909
 910   Entry  = IfList->ForwardLink;
 911   Status = EFI_SUCCESS;
 912
 913   if (IsListEmpty (IfList)) {
 914     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_INTERFACE), mHiiHandle);
 915   }
 916
 917   //
 918   // Go through the interface list.
 919   //
 920   while (Entry != IfList) {
 921
 922     IfCb = BASE_CR (Entry, IFCONFIG6_INTERFACE_CB, Link);
 923
 924     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_BREAK), mHiiHandle);
 925
 926     //
 927     // Print interface name.
 928     //
 929     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_IF_NAME), mHiiHandle, IfCb->IfInfo->Name);
 930
 931     //
 932     // Print interface config policy.
 933     //
 934     if (IfCb->Policy == Ip6ConfigPolicyAutomatic) {
 935       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_POLICY_AUTO), mHiiHandle);
 936     } else {
 937       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_POLICY_MAN), mHiiHandle);
 938     }
 939
 940     //
 941     // Print dad transmit.
 942     //
 943     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_DAD_TRANSMITS), mHiiHandle, IfCb->Xmits);
 944
 945     //
 946     // Print interface id if has.
 947     //
 948     if (IfCb->IfId != NULL) {
 949       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_INTERFACE_ID_HEAD), mHiiHandle);
 950
 951       IfConfig6PrintMacAddr (
 952         IfCb->IfId->Id,
 953         8
 954         );
 955     }
 956     //
 957     // Print mac address of the interface.
 958     //
 959     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_MAC_ADDR_HEAD), mHiiHandle);
 960
 961     IfConfig6PrintMacAddr (
 962       IfCb->IfInfo->HwAddress.Addr,
 963       IfCb->IfInfo->HwAddressSize
 964       );
 965
 966     //
 967     // Print ip addresses list of the interface.
 968     //
 969     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_IP_ADDR_HEAD), mHiiHandle);
 970
 971     for (Index = 0; Index < IfCb->IfInfo->AddressInfoCount; Index++) {
 972       IfConfig6PrintIpAddr (
 973         &IfCb->IfInfo->AddressInfo[Index].Address,
 974         &IfCb->IfInfo->AddressInfo[Index].PrefixLength
 975         );
 976       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_NEWLINE), mHiiHandle);
 977     }
 978
 979     //
 980     // Print dns server addresses list of the interface if has.
 981     //
 982     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_DNS_ADDR_HEAD), mHiiHandle);
 983
 984     for (Index = 0; Index < IfCb->DnsCnt; Index++) {
 985       IfConfig6PrintIpAddr (
 986         &IfCb->DnsAddr[Index],
 987         NULL
 988         );
 989       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_NEWLINE), mHiiHandle);
 990     }
 991
 992     //
 993     // Print route table of the interface if has.
 994     //
 995     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_ROUTE_HEAD), mHiiHandle);
 996
 997     for (Index = 0; Index < IfCb->IfInfo->RouteCount; Index++) {
 998       IfConfig6PrintIpAddr (
 999         &IfCb->IfInfo->RouteTable[Index].Destination,
1000         &IfCb->IfInfo->RouteTable[Index].PrefixLength
1001         );
1002       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_JOINT), mHiiHandle);
1003
1004       IfConfig6PrintIpAddr (
1005         &IfCb->IfInfo->RouteTable[Index].Gateway,
1006         NULL
1007         );
1008       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_NEWLINE), mHiiHandle);
1009     }
1010
1011     Entry = Entry->ForwardLink;
1012   }
1013
1014   ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_BREAK), mHiiHandle);
1015
1016   return Status;
1017 }
1018
1019 /**
1020   The clean process of the IfConfig6 application.
1021
1022   @param[in]   IfList    The pointer of IfList(interface list).
1023
1024   @retval EFI_SUCCESS    The IfConfig6 clean processed successfully.
1025   @retval others         The IfConfig6 clean process failed.
1026
1027 **/
1028 EFI_STATUS
1029 IfConfig6ClearInterfaceInfo (
1030   IN LIST_ENTRY    *IfList
1031   )
1032 {
1033   EFI_STATUS                Status;
1034   LIST_ENTRY                *Entry;
1035   IFCONFIG6_INTERFACE_CB    *IfCb;
1036   EFI_IP6_CONFIG_POLICY     Policy;
1037
1038   Policy = Ip6ConfigPolicyAutomatic;
1039   Entry  = IfList->ForwardLink;
1040   Status = EFI_SUCCESS;
1041
1042   if (IsListEmpty (IfList)) {
1043     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_INTERFACE), mHiiHandle);
1044   }
1045
1046   //
1047   // Go through the interface list.
1048   //
1049   while (Entry != IfList) {
1050
1051     IfCb = BASE_CR (Entry, IFCONFIG6_INTERFACE_CB, Link);
1052
1053     Status = IfCb->IfCfg->SetData (
1054                             IfCb->IfCfg,
1055                             Ip6ConfigDataTypePolicy,
1056                             sizeof (EFI_IP6_CONFIG_POLICY),
1057                             &Policy
1058                             );
1059
1060     if (EFI_ERROR (Status)) {
1061       break;
1062     }
1063
1064     Entry  = Entry->ForwardLink;
1065   }
1066
1067   return Status;
1068 }
1069
1070 /**
1071   The set process of the IfConfig6 application.
1072
1073   @param[in]   IfList    The pointer of IfList(interface list).
1074   @param[in]   VarArg    The pointer of ARG_LIST(Args with "-s" option).
1075
1076   @retval EFI_SUCCESS    The IfConfig6 set processed successfully.
1077   @retval others         The IfConfig6 set process failed.
1078
1079 **/
1080 EFI_STATUS
1081 IfConfig6SetInterfaceInfo (
1082   IN LIST_ENTRY    *IfList,
1083   IN ARG_LIST      *VarArg
1084   )
1085 {
1086   EFI_STATUS                       Status;
1087   IFCONFIG6_INTERFACE_CB           *IfCb;
1088   EFI_IP6_CONFIG_MANUAL_ADDRESS    *CfgManAddr;
1089   EFI_IPv6_ADDRESS                 *CfgAddr;
1090   UINTN                            AddrSize;
1091   EFI_IP6_CONFIG_INTERFACE_ID      *InterfaceId;
1092   UINT32                           DadXmits;
1093   UINT32                           CurDadXmits;
1094   UINTN                            CurDadXmitsLen;
1095   EFI_IP6_CONFIG_POLICY            Policy;
1096
1097   VAR_CHECK_CODE                   CheckCode;
1098   EFI_EVENT                        TimeOutEvt;
1099   EFI_EVENT                        MappedEvt;
1100   BOOLEAN                          IsAddressOk;
1101
1102   UINTN                            DataSize;
1103   UINT32                           Index;
1104   UINT32                           Index2;
1105   BOOLEAN                          IsAddressSet;
1106   EFI_IP6_CONFIG_INTERFACE_INFO    *IfInfo;
1107
1108   CfgManAddr  = NULL;
1109   CfgAddr     = NULL;
1110   TimeOutEvt  = NULL;
1111   MappedEvt   = NULL;
1112   IfInfo      = NULL;
1113   InterfaceId = NULL;
1114   CurDadXmits = 0;
1115
1116   if (IsListEmpty (IfList)) {
1117     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_INTERFACE), mHiiHandle);
1118     return EFI_INVALID_PARAMETER;
1119   }
1120   //
1121   // Make sure to set only one interface each time.
1122   //
1123   IfCb   = BASE_CR (IfList->ForwardLink, IFCONFIG6_INTERFACE_CB, Link);
1124   Status = EFI_SUCCESS;
1125
1126   //
1127   // Initialize check list mechanism.
1128   //
1129   CheckCode = IfConfig6RetriveCheckListByName(
1130                 NULL,
1131                 NULL,
1132                 TRUE
1133                 );
1134
1135   //
1136   // Create events & timers for asynchronous settings.
1137   //
1138   Status = gBS->CreateEvent (
1139                   EVT_TIMER,
1140                   TPL_CALLBACK,
1141                   NULL,
1142                   NULL,
1143                   &TimeOutEvt
1144                   );
1145   if (EFI_ERROR (Status)) {
1146     goto ON_EXIT;
1147   }
1148
1149   Status = gBS->CreateEvent (
1150                   EVT_NOTIFY_SIGNAL,
1151                   TPL_NOTIFY,
1152                   IfConfig6ManualAddressNotify,
1153                   &IsAddressOk,
1154                   &MappedEvt
1155                   );
1156   if (EFI_ERROR (Status)) {
1157     goto ON_EXIT;
1158   }
1159   //
1160   // Parse the setting variables.
1161   //
1162   while (VarArg != NULL) {
1163      //
1164      // Check invalid parameters (duplication & unknown & conflict).
1165      //
1166     CheckCode = IfConfig6RetriveCheckListByName(
1167                   mSetCheckList,
1168                   VarArg->Arg,
1169                   FALSE
1170                   );
1171
1172     if (VarCheckOk != CheckCode) {
1173       switch (CheckCode) {
1174         case VarCheckDuplicate:
1175           ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_DUPLICATE_COMMAND), mHiiHandle, VarArg->Arg);
1176           break;
1177
1178         case VarCheckConflict:
1179           ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_CONFLICT_COMMAND), mHiiHandle, VarArg->Arg);
1180           break;
1181
1182         case VarCheckUnknown:
1183           ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_UNKNOWN_COMMAND), mHiiHandle, VarArg->Arg);
1184           break;
1185
1186         default:
1187           break;
1188       }
1189
1190       VarArg = VarArg->Next;
1191       continue;
1192     }
1193     //
1194     // Process valid variables.
1195     //
1196     if (StrCmp(VarArg->Arg, L"auto") == 0) {
1197       //
1198       // Set automaic config policy
1199       //
1200       Policy = Ip6ConfigPolicyAutomatic;
1201       Status = IfCb->IfCfg->SetData (
1202                               IfCb->IfCfg,
1203                               Ip6ConfigDataTypePolicy,
1204                               sizeof (EFI_IP6_CONFIG_POLICY),
1205                               &Policy
1206                               );
1207
1208       if (EFI_ERROR(Status)) {
1209         goto ON_EXIT;
1210       }
1211
1212       VarArg= VarArg->Next;
1213
1214     } else if (StrCmp (VarArg->Arg, L"man") == 0) {
1215       //
1216       // Set manual config policy.
1217       //
1218       Policy = Ip6ConfigPolicyManual;
1219       Status = IfCb->IfCfg->SetData (
1220                               IfCb->IfCfg,
1221                               Ip6ConfigDataTypePolicy,
1222                               sizeof (EFI_IP6_CONFIG_POLICY),
1223                               &Policy
1224                               );
1225
1226       if (EFI_ERROR(Status)) {
1227         goto ON_EXIT;
1228       }
1229
1230       VarArg= VarArg->Next;
1231
1232     } else if (StrCmp (VarArg->Arg, L"host") == 0) {
1233       //
1234       // Parse till the next tag or the end of command line.
1235       //
1236       VarArg = VarArg->Next;
1237       Status = IfConfig6ParseManualAddressList (
1238                  &VarArg,
1239                  &CfgManAddr,
1240                  &AddrSize
1241                  );
1242
1243       if (EFI_ERROR (Status)) {
1244         if (Status == EFI_INVALID_PARAMETER) {
1245           ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_ARGUMENTS), mHiiHandle, L"host");
1246           continue;
1247         } else {
1248           goto ON_EXIT;
1249         }
1250       }
1251       //
1252       // Set static host ip6 address list.
1253       //   This is a asynchronous process.
1254       //
1255       IsAddressOk = FALSE;
1256
1257       Status = IfCb->IfCfg->RegisterDataNotify (
1258                               IfCb->IfCfg,
1259                               Ip6ConfigDataTypeManualAddress,
1260                               MappedEvt
1261                               );
1262       if (EFI_ERROR (Status)) {
1263         goto ON_EXIT;
1264       }
1265
1266       Status = IfCb->IfCfg->SetData (
1267                               IfCb->IfCfg,
1268                               Ip6ConfigDataTypeManualAddress,
1269                               AddrSize,
1270                               CfgManAddr
1271                               );
1272
1273       if (Status == EFI_NOT_READY) {
1274         //
1275         // Get current dad transmits count.
1276         //
1277         CurDadXmitsLen = sizeof (EFI_IP6_CONFIG_DUP_ADDR_DETECT_TRANSMITS);
1278         IfCb->IfCfg->GetData (
1279                        IfCb->IfCfg,
1280                        Ip6ConfigDataTypeDupAddrDetectTransmits,
1281                        &CurDadXmitsLen,
1282                        &CurDadXmits
1283                        );
1284
1285         gBS->SetTimer (TimeOutEvt, TimerRelative, 50000000 + 10000000 * CurDadXmits);
1286
1287         while (EFI_ERROR (gBS->CheckEvent (TimeOutEvt))) {
1288           if (IsAddressOk) {
1289             Status = EFI_SUCCESS;
1290             break;
1291           }
1292         }
1293       }
1294
1295       IfCb->IfCfg->UnregisterDataNotify (
1296                      IfCb->IfCfg,
1297                      Ip6ConfigDataTypeManualAddress,
1298                      MappedEvt
1299                      );
1300
1301       if (EFI_ERROR (Status)) {
1302         ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_MAN_HOST), mHiiHandle, Status);
1303         goto ON_EXIT;
1304       }
1305
1306       //
1307       // Check whether the address is set successfully.
1308       //
1309       DataSize = 0;
1310
1311       Status = IfCb->IfCfg->GetData (
1312                               IfCb->IfCfg,
1313                               Ip6ConfigDataTypeInterfaceInfo,
1314                               &DataSize,
1315                               NULL
1316                               );
1317
1318       if (Status != EFI_BUFFER_TOO_SMALL) {
1319         ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
1320         goto ON_EXIT;
1321       }
1322
1323       IfInfo = AllocateZeroPool (DataSize);
1324
1325       if (IfInfo == NULL) {
1326         Status = EFI_OUT_OF_RESOURCES;
1327         goto ON_EXIT;
1328       }
1329
1330       Status = IfCb->IfCfg->GetData (
1331                               IfCb->IfCfg,
1332                               Ip6ConfigDataTypeInterfaceInfo,
1333                               &DataSize,
1334                               IfInfo
1335                               );
1336
1337       if (EFI_ERROR (Status)) {
1338         ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
1339         goto ON_EXIT;
1340       }
1341
1342       for ( Index = 0; Index < (UINTN) (AddrSize / sizeof (EFI_IP6_CONFIG_MANUAL_ADDRESS)); Index++) {
1343         IsAddressSet = FALSE;
1344         //
1345         // By default, the prefix length 0 is regarded as 64.
1346         //
1347         if (CfgManAddr[Index].PrefixLength == 0) {
1348           CfgManAddr[Index].PrefixLength = 64;
1349         }
1350
1351         for (Index2 = 0; Index2 < IfInfo->AddressInfoCount; Index2++) {
1352           if (EFI_IP6_EQUAL (&IfInfo->AddressInfo[Index2].Address, &CfgManAddr[Index].Address) &&
1353               (IfInfo->AddressInfo[Index2].PrefixLength == CfgManAddr[Index].PrefixLength)) {
1354             IsAddressSet = TRUE;
1355             break;
1356           }
1357         }
1358
1359         if (!IsAddressSet) {
1360           ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_ADDRESS_FAILED), mHiiHandle);
1361           IfConfig6PrintIpAddr (
1362             &CfgManAddr[Index].Address,
1363             &CfgManAddr[Index].PrefixLength
1364             );
1365         }
1366       }
1367
1368     } else if (StrCmp (VarArg->Arg, L"gw") == 0) {
1369       //
1370       // Parse till the next tag or the end of command line.
1371       //
1372       VarArg = VarArg->Next;
1373       Status = IfConfig6ParseGwDnsAddressList (
1374                  &VarArg,
1375                  &CfgAddr,
1376                  &AddrSize
1377                  );
1378
1379       if (EFI_ERROR (Status)) {
1380         if (Status == EFI_INVALID_PARAMETER) {
1381           ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_ARGUMENTS), mHiiHandle, L"gw");
1382           continue;
1383         } else {
1384           goto ON_EXIT;
1385         }
1386       }
1387       //
1388       // Set static gateway ip6 address list.
1389       //
1390       Status = IfCb->IfCfg->SetData (
1391                               IfCb->IfCfg,
1392                               Ip6ConfigDataTypeGateway,
1393                               AddrSize,
1394                               CfgAddr
1395                               );
1396
1397       if (EFI_ERROR (Status)) {
1398         goto ON_EXIT;
1399       }
1400
1401     } else if (StrCmp (VarArg->Arg, L"dns") == 0) {
1402       //
1403       // Parse till the next tag or the end of command line.
1404       //
1405       VarArg = VarArg->Next;
1406       Status = IfConfig6ParseGwDnsAddressList (
1407                  &VarArg,
1408                  &CfgAddr,
1409                  &AddrSize
1410                  );
1411
1412       if (EFI_ERROR (Status)) {
1413         if (Status == EFI_INVALID_PARAMETER) {
1414           ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_ARGUMENTS), mHiiHandle, L"dns");
1415           continue;
1416         } else {
1417           goto ON_EXIT;
1418         }
1419       }
1420       //
1421       // Set static dhs server ip6 address list.
1422       //
1423       Status = IfCb->IfCfg->SetData (
1424                               IfCb->IfCfg,
1425                               Ip6ConfigDataTypeDnsServer,
1426                               AddrSize,
1427                               CfgAddr
1428                               );
1429
1430       if (EFI_ERROR (Status)) {
1431         goto ON_EXIT;
1432       }
1433
1434     } else if (StrCmp (VarArg->Arg, L"id") == 0) {
1435       //
1436       // Parse till the next tag or the end of command line.
1437       //
1438       VarArg = VarArg->Next;
1439       Status = IfConfig6ParseInterfaceId (&VarArg, &InterfaceId);
1440
1441       if (EFI_ERROR (Status)) {
1442         goto ON_EXIT;
1443       }
1444       //
1445       // Set alternative interface id.
1446       //
1447       Status = IfCb->IfCfg->SetData (
1448                               IfCb->IfCfg,
1449                               Ip6ConfigDataTypeAltInterfaceId,
1450                               sizeof (EFI_IP6_CONFIG_INTERFACE_ID),
1451                               InterfaceId
1452                               );
1453
1454       if (EFI_ERROR (Status)) {
1455         goto ON_EXIT;
1456       }
1457
1458     } else if (StrCmp (VarArg->Arg, L"dad") == 0) {
1459       //
1460       // Parse till the next tag or the end of command line.
1461       //
1462       VarArg = VarArg->Next;
1463       Status = IfConfig6ParseDadXmits (&VarArg, &DadXmits);
1464
1465       if (EFI_ERROR (Status)) {
1466         goto ON_EXIT;
1467       }
1468       //
1469       // Set dad transmits count.
1470       //
1471       Status = IfCb->IfCfg->SetData (
1472                               IfCb->IfCfg,
1473                               Ip6ConfigDataTypeDupAddrDetectTransmits,
1474                               sizeof (EFI_IP6_CONFIG_DUP_ADDR_DETECT_TRANSMITS),
1475                               &DadXmits
1476                               );
1477
1478       if (EFI_ERROR(Status)) {
1479         goto ON_EXIT;
1480       }
1481     }
1482   }
1483
1484 ON_EXIT:
1485
1486   if (CfgManAddr != NULL) {
1487     FreePool (CfgManAddr);
1488   }
1489
1490   if (CfgAddr != NULL) {
1491     FreePool (CfgAddr);
1492   }
1493
1494   if (MappedEvt != NULL) {
1495     gBS->CloseEvent (MappedEvt);
1496   }
1497
1498   if (TimeOutEvt != NULL) {
1499     gBS->CloseEvent (TimeOutEvt);
1500   }
1501
1502   if (IfInfo != NULL) {
1503     FreePool (IfInfo);
1504   }
1505
1506   return Status;
1507
1508 }
1509
1510 /**
1511   The IfConfig6 main process.
1512
1513   @param[in]   Private    The pointer of IFCONFIG6_PRIVATE_DATA.
1514
1515   @retval EFI_SUCCESS    IfConfig6 processed successfully.
1516   @retval others         The IfConfig6 process failed.
1517
1518 **/
1519 EFI_STATUS
1520 IfConfig6 (
1521   IN IFCONFIG6_PRIVATE_DATA    *Private
1522   )
1523 {
1524   EFI_STATUS    Status;
1525
1526   //
1527   // Get configure information of all interfaces.
1528   //
1529   Status = IfConfig6GetInterfaceInfo (
1530              Private->ImageHandle,
1531              Private->IfName,
1532              &Private->IfList
1533              );
1534
1535   if (EFI_ERROR (Status)) {
1536     goto ON_EXIT;
1537   }
1538
1539   switch (Private->OpCode) {
1540   case IfConfig6OpList:
1541     Status = IfConfig6ShowInterfaceInfo (&Private->IfList);
1542     break;
1543
1544   case IfConfig6OpClear:
1545     Status = IfConfig6ClearInterfaceInfo (&Private->IfList);
1546     break;
1547
1548   case IfConfig6OpSet:
1549     Status = IfConfig6SetInterfaceInfo (&Private->IfList, Private->VarArg);
1550     break;
1551
1552   default:
1553     Status = EFI_ABORTED;
1554   }
1555
1556 ON_EXIT:
1557
1558   return Status;
1559 }
1560
1561 /**
1562   The IfConfig6 cleanup process, free the allocated memory.
1563
1564   @param[in]   Private    The pointer of  IFCONFIG6_PRIVATE_DATA.
1565
1566 **/
1567 VOID
1568 IfConfig6Cleanup (
1569   IN IFCONFIG6_PRIVATE_DATA    *Private
1570   )
1571 {
1572   LIST_ENTRY                *Entry;
1573   LIST_ENTRY                *NextEntry;
1574   IFCONFIG6_INTERFACE_CB    *IfCb;
1575   ARG_LIST                  *ArgNode;
1576   ARG_LIST                  *ArgHead;
1577
1578   ASSERT (Private != NULL);
1579
1580   //
1581   // Clean the list which save the set config Args.
1582   //
1583   if (Private->VarArg != NULL) {
1584     ArgHead = Private->VarArg;
1585
1586     while (ArgHead->Next != NULL) {
1587       ArgNode = ArgHead->Next;
1588       FreePool (ArgHead);
1589       ArgHead = ArgNode;
1590     }
1591
1592     FreePool (ArgHead);
1593   }
1594
1595   if (Private->IfName != NULL)
1596     FreePool (Private->IfName);
1597
1598
1599   //
1600   // Clean the IFCONFIG6_INTERFACE_CB list.
1601   //
1602   Entry     = Private->IfList.ForwardLink;
1603   NextEntry = Entry->ForwardLink;
1604
1605   while (Entry != &Private->IfList) {
1606
1607     IfCb = BASE_CR (Entry, IFCONFIG6_INTERFACE_CB, Link);
1608
1609     RemoveEntryList (&IfCb->Link);
1610
1611     if (IfCb->IfId != NULL) {
1612
1613       FreePool (IfCb->IfId);
1614     }
1615
1616     if (IfCb->IfInfo != NULL) {
1617
1618       FreePool (IfCb->IfInfo);
1619     }
1620
1621     FreePool (IfCb);
1622
1623     Entry     = NextEntry;
1624     NextEntry = Entry->ForwardLink;
1625   }
1626
1627   FreePool (Private);
1628 }
1629
1630 /**
1631   This is the declaration of an EFI image entry point. This entry point is
1632   the same for UEFI Applications, UEFI OS Loaders, and UEFI Drivers, including
1633   both device drivers and bus drivers.
1634
1635   The entry point for the IfConfig6 application which parses the command line input and calls the IfConfig6 process.
1636
1637   @param[in] ImageHandle    The image handle of this application.
1638   @param[in] SystemTable    The pointer to the EFI System Table.
1639
1640   @retval EFI_SUCCESS    The operation completed successfully.
1641   @retval Others         Some errors occur.
1642
1643 **/
1644 EFI_STATUS
1645 EFIAPI
1646 IfConfig6Initialize (
1647   IN  EFI_HANDLE         ImageHandle,
1648   IN  EFI_SYSTEM_TABLE    *SystemTable
1649   )
1650 {
1651   EFI_STATUS                Status;
1652   IFCONFIG6_PRIVATE_DATA    *Private;
1653   LIST_ENTRY                *ParamPackage;
1654   CONST CHAR16              *ValueStr;
1655   ARG_LIST                  *ArgList;
1656   CHAR16                    *ProblemParam;
1657   CHAR16                    *Str;
1658
1659   Private = NULL;
1660
1661   //
1662   // Register our string package with HII and return the handle to it.
1663   //
1664   mHiiHandle = HiiAddPackages (&gEfiCallerIdGuid, ImageHandle, IfConfig6Strings, NULL);
1665   ASSERT (mHiiHandle != NULL);
1666
1667   Status = ShellCommandLineParseEx (mIfConfig6CheckList, &ParamPackage, &ProblemParam, TRUE, FALSE);
1668   if (EFI_ERROR (Status)) {
1669     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_COMMAND), mHiiHandle, ProblemParam);
1670     goto ON_EXIT;
1671   }
1672
1673   //
1674   // To handle no option.
1675   //
1676   if (!ShellCommandLineGetFlag (ParamPackage, L"-r") && !ShellCommandLineGetFlag (ParamPackage, L"-s") &&
1677       !ShellCommandLineGetFlag (ParamPackage, L"-?") && !ShellCommandLineGetFlag (ParamPackage, L"-l")) {
1678     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_LACK_OPTION), mHiiHandle);
1679     goto ON_EXIT;
1680   }
1681   //
1682   // To handle conflict options.
1683   //
1684   if (((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-s"))) ||
1685       ((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-l"))) ||
1686       ((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-?"))) ||
1687       ((ShellCommandLineGetFlag (ParamPackage, L"-s")) && (ShellCommandLineGetFlag (ParamPackage, L"-l"))) ||
1688       ((ShellCommandLineGetFlag (ParamPackage, L"-s")) && (ShellCommandLineGetFlag (ParamPackage, L"-?"))) ||
1689       ((ShellCommandLineGetFlag (ParamPackage, L"-l")) && (ShellCommandLineGetFlag (ParamPackage, L"-?")))) {
1690     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_CONFLICT_OPTIONS), mHiiHandle);
1691     goto ON_EXIT;
1692   }
1693   //
1694   // To show the help information of ifconfig6 command.
1695   //
1696   if (ShellCommandLineGetFlag (ParamPackage, L"-?")) {
1697     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_HELP), mHiiHandle);
1698     goto ON_EXIT;
1699   }
1700
1701   Status = EFI_INVALID_PARAMETER;
1702
1703   Private = AllocateZeroPool (sizeof (IFCONFIG6_PRIVATE_DATA));
1704
1705   if (Private == NULL) {
1706     Status = EFI_OUT_OF_RESOURCES;
1707     goto ON_EXIT;
1708   }
1709
1710   InitializeListHead (&Private->IfList);
1711
1712   //
1713   // To get interface name for the list option.
1714   //
1715   if (ShellCommandLineGetFlag (ParamPackage, L"-l")) {
1716     Private->OpCode = IfConfig6OpList;
1717     ValueStr = ShellCommandLineGetValue (ParamPackage, L"-l");
1718     if (ValueStr != NULL) {
1719       Str             = (CHAR16 *) AllocateZeroPool (StrSize (ValueStr));
1720       ASSERT (Str != NULL);
1721
1722       Str             = StrCpy (Str, ValueStr);
1723       Private->IfName = Str;
1724     }
1725   }
1726   //
1727   // To get interface name for the clear option.
1728   //
1729   if (ShellCommandLineGetFlag (ParamPackage, L"-r")) {
1730     Private->OpCode = IfConfig6OpClear;
1731     ValueStr = ShellCommandLineGetValue (ParamPackage, L"-r");
1732     if (ValueStr != NULL) {
1733       Str             = (CHAR16 *) AllocateZeroPool (StrSize (ValueStr));
1734       ASSERT (Str != NULL);
1735
1736       Str             = StrCpy (Str, ValueStr);
1737       Private->IfName = Str;
1738     }
1739   }
1740   //
1741   // To get interface name and corresponding Args for the set option.
1742   //
1743   if (ShellCommandLineGetFlag (ParamPackage, L"-s")) {
1744
1745     ValueStr = ShellCommandLineGetValue (ParamPackage, L"-s");
1746     if (ValueStr == NULL) {
1747       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_INTERFACE), mHiiHandle);
1748       goto ON_EXIT;
1749     }
1750     //
1751     // To split the configuration into multi-section.
1752     //
1753     ArgList         = SplitStrToList (ValueStr, L' ');
1754     ASSERT (ArgList != NULL);
1755
1756     Private->OpCode = IfConfig6OpSet;
1757     Private->IfName = ArgList->Arg;
1758
1759     Private->VarArg = ArgList->Next;
1760
1761     if (Private->IfName == NULL || Private->VarArg == NULL) {
1762       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_COMMAND), mHiiHandle);
1763       goto ON_EXIT;
1764     }
1765   }
1766   //
1767   // Main process of ifconfig6.
1768   //
1769   Status = IfConfig6 (Private);
1770
1771 ON_EXIT:
1772
1773   ShellCommandLineFreeVarList (ParamPackage);
1774   HiiRemovePackages (mHiiHandle);
1775   if (Private != NULL)
1776     IfConfig6Cleanup (Private);
1777
1778   return Status;
1779 }
1780