ShellPkg/Library/UefiShellNetwork1CommandsLib/Ifconfig.c

   1 /** @file
   2   The implementation for Shell command ifconfig based on IP4Config2 protocol.
   3
   4   (C) Copyright 2013-2015 Hewlett-Packard Development Company, L.P.<BR>
   5   Copyright (c) 2006 - 2015, Intel Corporation. All rights reserved.<BR>
   6
   7   This program and the accompanying materials
   8   are licensed and made available under the terms and conditions of the BSD License
   9   which accompanies this distribution.  The full text of the license may be found at
  10   http://opensource.org/licenses/bsd-license.php.
  11
  12   THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
  13   WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
  14
  15 **/
  16
  17 #include "UefiShellNetwork1CommandsLib.h"
  18
  19 typedef enum {
  20   IfConfigOpList     = 1,
  21   IfConfigOpSet      = 2,
  22   IfConfigOpClear    = 3
  23 } IFCONFIG_OPCODE;
  24
  25 typedef enum {
  26   VarCheckReserved      = -1,
  27   VarCheckOk            = 0,
  28   VarCheckDuplicate,
  29   VarCheckConflict,
  30   VarCheckUnknown,
  31   VarCheckLackValue,
  32   VarCheckOutOfMem
  33 } VAR_CHECK_CODE;
  34
  35 typedef enum {
  36   FlagTypeSingle         = 0,
  37   FlagTypeNeedVar,
  38   FlagTypeNeedSet,
  39   FlagTypeSkipUnknown
  40 } VAR_CHECK_FLAG_TYPE;
  41
  42 #define MACADDRMAXSIZE    32
  43
  44 typedef struct _IFCONFIG_INTERFACE_CB {
  45   EFI_HANDLE                                  NicHandle;
  46   LIST_ENTRY                                  Link;
  47   EFI_IP4_CONFIG2_PROTOCOL                    *IfCfg;
  48   EFI_IP4_CONFIG2_INTERFACE_INFO              *IfInfo;
  49   EFI_IP4_CONFIG2_POLICY                      Policy;
  50   UINT32                                      DnsCnt;
  51   EFI_IPv4_ADDRESS                            DnsAddr[1];
  52 } IFCONFIG_INTERFACE_CB;
  53
  54 typedef struct _ARG_LIST ARG_LIST;
  55
  56 struct _ARG_LIST {
  57   ARG_LIST    *Next;
  58   CHAR16      *Arg;
  59 };
  60
  61 typedef struct _IFCONFIG4_PRIVATE_DATA {
  62   LIST_ENTRY  IfList;
  63
  64   UINT32      OpCode;
  65   CHAR16      *IfName;
  66   ARG_LIST    *VarArg;
  67 } IFCONFIG_PRIVATE_DATA;
  68
  69 typedef struct _VAR_CHECK_ITEM{
  70   CHAR16                 *FlagStr;
  71   UINT32                 FlagID;
  72   UINT32                 ConflictMask;
  73   VAR_CHECK_FLAG_TYPE    FlagType;
  74 } VAR_CHECK_ITEM;
  75
  76 SHELL_PARAM_ITEM    mIfConfigCheckList[] = {
  77   {
  78     L"-b",
  79     TypeFlag
  80   },
  81   {
  82     L"-l",
  83     TypeValue
  84   },
  85   {
  86     L"-r",
  87     TypeValue
  88   },
  89   {
  90     L"-c",
  91     TypeValue
  92   },
  93   {
  94     L"-s",
  95     TypeMaxValue
  96   },
  97   {
  98     NULL,
  99     TypeMax
 100   },
 101 };
 102
 103 VAR_CHECK_ITEM  mSetCheckList[] = {
 104   {
 105    L"static",
 106     0x00000001,
 107     0x00000001,
 108     FlagTypeSingle
 109   },
 110   {
 111     L"dhcp",
 112     0x00000002,
 113     0x00000001,
 114     FlagTypeSingle
 115   },
 116   {
 117     L"dns",
 118     0x00000008,
 119     0x00000004,
 120     FlagTypeSingle
 121   },
 122   {
 123     NULL,
 124     0x0,
 125     0x0,
 126     FlagTypeSkipUnknown
 127   },
 128 };
 129
 130 STATIC CONST CHAR16 PermanentString[10] = L"PERMANENT";
 131
 132 /**
 133   Split a string with specified separator and save the substring to a list.
 134
 135   @param[in]    String       The pointer of the input string.
 136   @param[in]    Separator    The specified separator.
 137
 138   @return The pointer of headnode of ARG_LIST.
 139
 140 **/
 141 ARG_LIST *
 142 SplitStrToList (
 143   IN CONST CHAR16    *String,
 144   IN CHAR16          Separator
 145   )
 146 {
 147   CHAR16      *Str;
 148   CHAR16      *ArgStr;
 149   ARG_LIST    *ArgList;
 150   ARG_LIST    *ArgNode;
 151
 152   if (*String == L'\0') {
 153     return NULL;
 154   }
 155
 156   //
 157   // Copy the CONST string to a local copy.
 158   //
 159   Str = AllocateCopyPool (StrSize (String), String);
 160   ASSERT (Str != NULL);
 161   ArgStr  = Str;
 162
 163   //
 164   // init a node for the list head.
 165   //
 166   ArgNode = (ARG_LIST *) AllocateZeroPool (sizeof (ARG_LIST));
 167   ASSERT (ArgNode != NULL);
 168   ArgList = ArgNode;
 169
 170   //
 171   // Split the local copy and save in the list node.
 172   //
 173   while (*Str != L'\0') {
 174     if (*Str == Separator) {
 175       *Str          = L'\0';
 176       ArgNode->Arg  = ArgStr;
 177       ArgStr        = Str + 1;
 178       ArgNode->Next = (ARG_LIST *) AllocateZeroPool (sizeof (ARG_LIST));
 179       ASSERT (ArgNode->Next != NULL);
 180       ArgNode = ArgNode->Next;
 181     }
 182
 183     Str++;
 184   }
 185
 186   ArgNode->Arg  = ArgStr;
 187   ArgNode->Next = NULL;
 188
 189   return ArgList;
 190 }
 191
 192 /**
 193   Check the correctness of input Args with '-s' option.
 194
 195   @param[in]    CheckList    The pointer of VAR_CHECK_ITEM array.
 196   @param[in]    Name         The pointer of input arg.
 197   @param[in]    Init         The switch to execute the check.
 198
 199   @return   VarCheckOk          Valid parameter or Initialize check successfully.
 200   @return   VarCheckDuplicate   Duplicated parameter happened.
 201   @return   VarCheckConflict    Conflicted parameter happened
 202   @return   VarCheckUnknown     Unknown parameter.
 203
 204 **/
 205 VAR_CHECK_CODE
 206 IfConfigRetriveCheckListByName(
 207   IN VAR_CHECK_ITEM    *CheckList,
 208   IN CHAR16            *Name,
 209   IN BOOLEAN           Init
 210 )
 211 {
 212   STATIC UINT32     CheckDuplicate;
 213   STATIC UINT32     CheckConflict;
 214   VAR_CHECK_CODE    RtCode;
 215   UINT32            Index;
 216   VAR_CHECK_ITEM    Arg;
 217
 218   if (Init) {
 219     CheckDuplicate = 0;
 220     CheckConflict  = 0;
 221     return VarCheckOk;
 222   }
 223
 224   RtCode  = VarCheckOk;
 225   Index   = 0;
 226   Arg     = CheckList[Index];
 227
 228   //
 229   // Check the Duplicated/Conflicted/Unknown input Args.
 230   //
 231   while (Arg.FlagStr != NULL) {
 232     if (StrCmp (Arg.FlagStr, Name) == 0) {
 233
 234       if (CheckDuplicate & Arg.FlagID) {
 235         RtCode = VarCheckDuplicate;
 236         break;
 237       }
 238
 239       if (CheckConflict & Arg.ConflictMask) {
 240         RtCode = VarCheckConflict;
 241         break;
 242       }
 243
 244       CheckDuplicate |= Arg.FlagID;
 245       CheckConflict  |= Arg.ConflictMask;
 246       break;
 247     }
 248
 249     Arg = CheckList[++Index];
 250   }
 251
 252   if (Arg.FlagStr == NULL) {
 253     RtCode = VarCheckUnknown;
 254   }
 255
 256   return RtCode;
 257 }
 258
 259 /**
 260   The notify function of create event when performing a manual config.
 261
 262   @param[in]    Event        The event this notify function registered to.
 263   @param[in]    Context      Pointer to the context data registered to the event.
 264
 265 **/
 266 VOID
 267 EFIAPI
 268 IfConfigManualAddressNotify (
 269   IN EFI_EVENT    Event,
 270   IN VOID         *Context
 271   )
 272 {
 273   *((BOOLEAN *) Context) = TRUE;
 274 }
 275
 276
 277 /**
 278   Create an IP child, use it to start the auto configuration, then destroy it.
 279
 280   @param[in] Controller       The controller which has the service installed.
 281   @param[in] Image            The image handle used to open service.
 282
 283   @retval EFI_SUCCESS         The configuration is done.
 284 **/
 285 EFI_STATUS
 286 EFIAPI
 287 IfConfigStartIp4(
 288   IN  EFI_HANDLE            Controller,
 289   IN  EFI_HANDLE            Image
 290   )
 291 {
 292   EFI_IP4_PROTOCOL              *Ip4;
 293   EFI_HANDLE                    Ip4Handle;
 294   EFI_IP4_CONFIG_DATA           Ip4ConfigData;
 295   EFI_STATUS                    Status;
 296
 297   //
 298   // Get the Ip4ServiceBinding Protocol
 299   //
 300   Ip4Handle     = NULL;
 301   Ip4           = NULL;
 302
 303   Status = NetLibCreateServiceChild (
 304              Controller,
 305              Image,
 306              &gEfiIp4ServiceBindingProtocolGuid,
 307              &Ip4Handle
 308              );
 309
 310   if (EFI_ERROR (Status)) {
 311     return Status;
 312   }
 313
 314   Status = gBS->OpenProtocol (
 315                  Ip4Handle,
 316                  &gEfiIp4ProtocolGuid,
 317                  (VOID **) &Ip4,
 318                  Controller,
 319                  Image,
 320                  EFI_OPEN_PROTOCOL_GET_PROTOCOL
 321                  );
 322
 323   if (EFI_ERROR (Status)) {
 324     goto ON_EXIT;
 325   }
 326
 327   Ip4ConfigData.DefaultProtocol          = EFI_IP_PROTO_ICMP;
 328   Ip4ConfigData.AcceptAnyProtocol        = FALSE;
 329   Ip4ConfigData.AcceptIcmpErrors         = FALSE;
 330   Ip4ConfigData.AcceptBroadcast          = FALSE;
 331   Ip4ConfigData.AcceptPromiscuous        = FALSE;
 332   Ip4ConfigData.UseDefaultAddress        = TRUE;
 333   ZeroMem (&Ip4ConfigData.StationAddress, sizeof (EFI_IPv4_ADDRESS));
 334   ZeroMem (&Ip4ConfigData.SubnetMask, sizeof (EFI_IPv4_ADDRESS));
 335   Ip4ConfigData.TypeOfService            = 0;
 336   Ip4ConfigData.TimeToLive               = 1;
 337   Ip4ConfigData.DoNotFragment            = FALSE;
 338   Ip4ConfigData.RawData                  = FALSE;
 339   Ip4ConfigData.ReceiveTimeout           = 0;
 340   Ip4ConfigData.TransmitTimeout          = 0;
 341
 342   Ip4->Configure (Ip4, &Ip4ConfigData);
 343
 344 ON_EXIT:
 345   NetLibDestroyServiceChild (
 346     Controller,
 347     Image,
 348     &gEfiIp4ServiceBindingProtocolGuid,
 349     Ip4Handle
 350     );
 351
 352   return Status;
 353 }
 354
 355
 356 /**
 357   Print MAC address.
 358
 359   @param[in]    Node    The pointer of MAC address buffer.
 360   @param[in]    Size    The size of MAC address buffer.
 361
 362 **/
 363 VOID
 364 IfConfigPrintMacAddr (
 365   IN UINT8     *Node,
 366   IN UINT32    Size
 367   )
 368 {
 369   UINTN    Index;
 370
 371   ASSERT (Size <= MACADDRMAXSIZE);
 372
 373   for (Index = 0; Index < Size; Index++) {
 374     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_MAC_ADDR_BODY), gShellNetwork1HiiHandle, Node[Index]);
 375     if (Index + 1 < Size) {
 376       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_COLON), gShellNetwork1HiiHandle);
 377     }
 378   }
 379
 380   ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_NEWLINE), gShellNetwork1HiiHandle);
 381 }
 382
 383
 384 /**
 385   The get current status of all handles.
 386
 387   @param[in]   IfName         The pointer of IfName(interface name).
 388   @param[in]   IfList         The pointer of IfList(interface list).
 389
 390   @retval EFI_SUCCESS    The get status processed successfully.
 391   @retval others         The get status process failed.
 392
 393 **/
 394 EFI_STATUS
 395 IfConfigGetInterfaceInfo (
 396   IN CHAR16        *IfName,
 397   IN LIST_ENTRY    *IfList
 398   )
 399 {
 400   EFI_STATUS                       Status;
 401   UINTN                            HandleIndex;
 402   UINTN                            HandleNum;
 403   EFI_HANDLE                       *HandleBuffer;
 404   EFI_IP4_CONFIG2_PROTOCOL         *Ip4Cfg2;
 405   EFI_IP4_CONFIG2_INTERFACE_INFO   *IfInfo;
 406   IFCONFIG_INTERFACE_CB            *IfCb;
 407   UINTN                            DataSize;
 408
 409   HandleBuffer = NULL;
 410   HandleNum    = 0;
 411
 412   IfInfo       = NULL;
 413   IfCb         = NULL;
 414
 415   //
 416   // Locate all the handles with ip4 service binding protocol.
 417   //
 418   Status = gBS->LocateHandleBuffer (
 419                   ByProtocol,
 420                   &gEfiIp4ServiceBindingProtocolGuid,
 421                   NULL,
 422                   &HandleNum,
 423                   &HandleBuffer
 424                  );
 425   if (EFI_ERROR (Status) || (HandleNum == 0)) {
 426     return Status;
 427   }
 428
 429   //
 430   // Enumerate all handles that installed with ip4 service binding protocol.
 431   //
 432   for (HandleIndex = 0; HandleIndex < HandleNum; HandleIndex++) {
 433     IfCb      = NULL;
 434     IfInfo    = NULL;
 435     DataSize  = 0;
 436
 437     //
 438     // Ip4config protocol and ip4 service binding protocol are installed
 439     // on the same handle.
 440     //
 441     ASSERT (HandleBuffer != NULL);
 442     Status = gBS->HandleProtocol (
 443                     HandleBuffer[HandleIndex],
 444                     &gEfiIp4Config2ProtocolGuid,
 445                     (VOID **) &Ip4Cfg2
 446                     );
 447
 448     if (EFI_ERROR (Status)) {
 449       goto ON_ERROR;
 450     }
 451
 452     //
 453     // Get the interface information size.
 454     //
 455     Status = Ip4Cfg2->GetData (
 456                        Ip4Cfg2,
 457                        Ip4Config2DataTypeInterfaceInfo,
 458                        &DataSize,
 459                        NULL
 460                        );
 461
 462     if (Status != EFI_BUFFER_TOO_SMALL) {
 463       goto ON_ERROR;
 464     }
 465
 466     IfInfo = AllocateZeroPool (DataSize);
 467
 468     if (IfInfo == NULL) {
 469       Status = EFI_OUT_OF_RESOURCES;
 470       goto ON_ERROR;
 471     }
 472
 473     //
 474     // Get the interface info.
 475     //
 476     Status = Ip4Cfg2->GetData (
 477                        Ip4Cfg2,
 478                        Ip4Config2DataTypeInterfaceInfo,
 479                        &DataSize,
 480                        IfInfo
 481                        );
 482
 483     if (EFI_ERROR (Status)) {
 484       goto ON_ERROR;
 485     }
 486
 487     //
 488     // Check the interface name if required.
 489     //
 490     if ((IfName != NULL) && (StrCmp (IfName, IfInfo->Name) != 0)) {
 491       FreePool (IfInfo);
 492       continue;
 493     }
 494
 495     DataSize = 0;
 496
 497     //
 498     // Get the size of dns server list.
 499     //
 500     Status = Ip4Cfg2->GetData (
 501                        Ip4Cfg2,
 502                        Ip4Config2DataTypeDnsServer,
 503                        &DataSize,
 504                        NULL
 505                        );
 506
 507     if ((Status != EFI_BUFFER_TOO_SMALL) && (Status != EFI_NOT_FOUND)) {
 508       goto ON_ERROR;
 509     }
 510
 511     IfCb = AllocateZeroPool (sizeof (IFCONFIG_INTERFACE_CB) + DataSize);
 512
 513     if (IfCb == NULL) {
 514       Status = EFI_OUT_OF_RESOURCES;
 515       goto ON_ERROR;
 516     }
 517
 518     IfCb->NicHandle = HandleBuffer[HandleIndex];
 519     IfCb->IfInfo    = IfInfo;
 520     IfCb->IfCfg     = Ip4Cfg2;
 521     IfCb->DnsCnt    = (UINT32) (DataSize / sizeof (EFI_IPv4_ADDRESS));
 522
 523     //
 524     // Get the dns server list if has.
 525     //
 526     if (DataSize > 0) {
 527       Status = Ip4Cfg2->GetData (
 528                          Ip4Cfg2,
 529                          Ip4Config2DataTypeDnsServer,
 530                          &DataSize,
 531                          IfCb->DnsAddr
 532                          );
 533
 534       if (EFI_ERROR (Status)) {
 535         goto ON_ERROR;
 536       }
 537     }
 538
 539     //
 540     // Get the config policy.
 541     //
 542     DataSize = sizeof (EFI_IP4_CONFIG2_POLICY);
 543     Status   = Ip4Cfg2->GetData (
 544                          Ip4Cfg2,
 545                          Ip4Config2DataTypePolicy,
 546                          &DataSize,
 547                          &IfCb->Policy
 548                          );
 549
 550     if (EFI_ERROR (Status)) {
 551       goto ON_ERROR;
 552     }
 553
 554     InsertTailList (IfList, &IfCb->Link);
 555
 556     if ((IfName != NULL) && (StrCmp (IfName, IfInfo->Name) == 0)) {
 557       //
 558       // Only need the appointed interface, keep the allocated buffer.
 559       //
 560       IfCb   = NULL;
 561       IfInfo = NULL;
 562       break;
 563     }
 564   }
 565
 566   if (HandleBuffer != NULL) {
 567     FreePool (HandleBuffer);
 568   }
 569
 570   return EFI_SUCCESS;
 571
 572 ON_ERROR:
 573
 574   if (IfInfo != NULL) {
 575     FreePool (IfInfo);
 576   }
 577
 578   if (IfCb != NULL) {
 579     FreePool (IfCb);
 580   }
 581
 582   return Status;
 583 }
 584
 585 /**
 586   The list process of the ifconfig command.
 587
 588   @param[in]   IfList    The pointer of IfList(interface list).
 589
 590   @retval SHELL_SUCCESS  The ifconfig command list processed successfully.
 591   @retval others         The ifconfig command list process failed.
 592
 593 **/
 594 SHELL_STATUS
 595 IfConfigShowInterfaceInfo (
 596   IN LIST_ENTRY    *IfList
 597   )
 598 {
 599   LIST_ENTRY                   *Entry;
 600   LIST_ENTRY                   *Next;
 601   IFCONFIG_INTERFACE_CB        *IfCb;
 602   BOOLEAN                       MediaPresent;
 603   EFI_IPv4_ADDRESS              Gateway;
 604   UINT32                        Index;
 605
 606   MediaPresent = TRUE;
 607
 608   if (IsListEmpty (IfList)) {
 609     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INVALID_INTERFACE), gShellNetwork1HiiHandle);
 610   }
 611
 612   //
 613   // Go through the interface list.
 614   //
 615   NET_LIST_FOR_EACH_SAFE (Entry, Next, IfList) {
 616     IfCb = NET_LIST_USER_STRUCT (Entry, IFCONFIG_INTERFACE_CB, Link);
 617
 618     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_BREAK), gShellNetwork1HiiHandle);
 619
 620     //
 621     // Print interface name.
 622     //
 623     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_IF_NAME), gShellNetwork1HiiHandle, IfCb->IfInfo->Name);
 624
 625     //
 626     // Get Media State.
 627     //
 628     NetLibDetectMedia (IfCb->NicHandle, &MediaPresent);
 629     if (!MediaPresent) {
 630       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_MEDIA_STATE), gShellNetwork1HiiHandle, L"Media disconnected");
 631     } else {
 632       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_MEDIA_STATE), gShellNetwork1HiiHandle, L"Media present");
 633     }
 634
 635     //
 636     // Print interface config policy.
 637     //
 638     if (IfCb->Policy == Ip4Config2PolicyDhcp) {
 639       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_POLICY_DHCP), gShellNetwork1HiiHandle);
 640     } else {
 641       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_POLICY_MAN), gShellNetwork1HiiHandle);
 642     }
 643
 644     //
 645     // Print mac address of the interface.
 646     //
 647     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_MAC_ADDR_HEAD), gShellNetwork1HiiHandle);
 648
 649     IfConfigPrintMacAddr (
 650       IfCb->IfInfo->HwAddress.Addr,
 651       IfCb->IfInfo->HwAddressSize
 652       );
 653
 654     //
 655     // Print IPv4 address list of the interface.
 656     //
 657     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_IP_ADDR_HEAD), gShellNetwork1HiiHandle);
 658
 659     ShellPrintHiiEx(
 660       -1,
 661       -1,
 662       NULL,
 663       STRING_TOKEN (STR_IFCONFIG_INFO_IP_ADDR_BODY),
 664       gShellNetwork1HiiHandle,
 665       (UINTN)IfCb->IfInfo->StationAddress.Addr[0],
 666       (UINTN)IfCb->IfInfo->StationAddress.Addr[1],
 667       (UINTN)IfCb->IfInfo->StationAddress.Addr[2],
 668       (UINTN)IfCb->IfInfo->StationAddress.Addr[3]
 669       );
 670
 671     //
 672     // Print subnet mask list of the interface.
 673     //
 674     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_SUBNET_MASK_HEAD), gShellNetwork1HiiHandle);
 675
 676     ShellPrintHiiEx(
 677       -1,
 678       -1,
 679       NULL,
 680       STRING_TOKEN (STR_IFCONFIG_INFO_IP_ADDR_BODY),
 681       gShellNetwork1HiiHandle,
 682       (UINTN)IfCb->IfInfo->SubnetMask.Addr[0],
 683       (UINTN)IfCb->IfInfo->SubnetMask.Addr[1],
 684       (UINTN)IfCb->IfInfo->SubnetMask.Addr[2],
 685       (UINTN)IfCb->IfInfo->SubnetMask.Addr[3]
 686       );
 687
 688     //
 689     // Print default gateway of the interface.
 690     //
 691     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_GATEWAY_HEAD), gShellNetwork1HiiHandle);
 692
 693     ZeroMem (&Gateway, sizeof (EFI_IPv4_ADDRESS));
 694
 695     for (Index = 0; Index < IfCb->IfInfo->RouteTableSize; Index++) {
 696       if ((CompareMem (&IfCb->IfInfo->RouteTable[Index].SubnetAddress, &mZeroIp4Addr, sizeof (EFI_IPv4_ADDRESS)) == 0) &&
 697           (CompareMem (&IfCb->IfInfo->RouteTable[Index].SubnetMask   , &mZeroIp4Addr, sizeof (EFI_IPv4_ADDRESS)) == 0) ){
 698         CopyMem (&Gateway, &IfCb->IfInfo->RouteTable[Index].GatewayAddress, sizeof (EFI_IPv4_ADDRESS));
 699       }
 700     }
 701
 702     ShellPrintHiiEx(
 703       -1,
 704       -1,
 705       NULL,
 706       STRING_TOKEN (STR_IFCONFIG_INFO_IP_ADDR_BODY),
 707       gShellNetwork1HiiHandle,
 708       (UINTN)Gateway.Addr[0],
 709       (UINTN)Gateway.Addr[1],
 710       (UINTN)Gateway.Addr[2],
 711       (UINTN)Gateway.Addr[3]
 712       );
 713
 714     //
 715     // Print route table entry.
 716     //
 717     ShellPrintHiiEx(-1, -1, NULL,STRING_TOKEN (STR_IFCONFIG_ROUTES_SIZE), gShellNetwork1HiiHandle, IfCb->IfInfo->RouteTableSize);
 718
 719     for (Index = 0; Index < IfCb->IfInfo->RouteTableSize; Index++) {
 720       ShellPrintHiiEx(-1, -1, NULL,STRING_TOKEN (STR_IFCONFIG_ROUTES_ENTRY_INDEX), gShellNetwork1HiiHandle, Index);
 721
 722       ShellPrintHiiEx(
 723         -1,
 724         -1,
 725         NULL,
 726         STRING_TOKEN (STR_IFCONFIG_SHOW_IP_ADDR),
 727         gShellNetwork1HiiHandle,
 728         L"Subnet ",
 729         (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetAddress.Addr[0],
 730         (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetAddress.Addr[1],
 731         (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetAddress.Addr[2],
 732         (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetAddress.Addr[3]
 733         );
 734
 735       ShellPrintHiiEx(
 736         -1,
 737         -1,
 738         NULL,
 739         STRING_TOKEN (STR_IFCONFIG_SHOW_IP_ADDR),
 740         gShellNetwork1HiiHandle,
 741         L"Netmask",
 742         (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetMask.Addr[0],
 743         (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetMask.Addr[1],
 744         (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetMask.Addr[2],
 745         (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetMask.Addr[3]
 746         );
 747
 748       ShellPrintHiiEx(
 749         -1,
 750         -1,
 751         NULL,
 752         STRING_TOKEN (STR_IFCONFIG_SHOW_IP_ADDR),
 753         gShellNetwork1HiiHandle,
 754         L"Gateway",
 755         (UINTN)IfCb->IfInfo->RouteTable[Index].GatewayAddress.Addr[0],
 756         (UINTN)IfCb->IfInfo->RouteTable[Index].GatewayAddress.Addr[1],
 757         (UINTN)IfCb->IfInfo->RouteTable[Index].GatewayAddress.Addr[2],
 758         (UINTN)IfCb->IfInfo->RouteTable[Index].GatewayAddress.Addr[3]
 759         );
 760     }
 761
 762     //
 763     // Print dns server addresses list of the interface if has.
 764     //
 765     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_DNS_ADDR_HEAD), gShellNetwork1HiiHandle);
 766
 767     for (Index = 0; Index < IfCb->DnsCnt; Index++) {
 768       ShellPrintHiiEx(
 769         -1,
 770         -1,
 771         NULL,
 772         STRING_TOKEN (STR_IFCONFIG_INFO_DNS_ADDR_BODY),
 773         gShellNetwork1HiiHandle,
 774         (UINTN) IfCb->DnsAddr[Index].Addr[0],
 775         (UINTN) IfCb->DnsAddr[Index].Addr[1],
 776         (UINTN) IfCb->DnsAddr[Index].Addr[2],
 777         (UINTN) IfCb->DnsAddr[Index].Addr[3]
 778         );
 779
 780       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_NEWLINE), gShellNetwork1HiiHandle);
 781     }
 782   }
 783
 784   ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_BREAK), gShellNetwork1HiiHandle);
 785
 786   return SHELL_SUCCESS;
 787 }
 788
 789 /**
 790   The clean process of the ifconfig command to clear interface info.
 791
 792   @param[in]   IfList    The pointer of IfList(interface list).
 793
 794   @retval SHELL_SUCCESS  The ifconfig command clean processed successfully.
 795   @retval others         The ifconfig command clean process failed.
 796
 797 **/
 798 SHELL_STATUS
 799 IfConfigClearInterfaceInfo (
 800   IN LIST_ENTRY    *IfList
 801   )
 802 {
 803   EFI_STATUS                Status;
 804   SHELL_STATUS              ShellStatus;
 805   LIST_ENTRY                *Entry;
 806   LIST_ENTRY                *Next;
 807   IFCONFIG_INTERFACE_CB     *IfCb;
 808   EFI_IP4_CONFIG2_POLICY    Policy;
 809
 810   Policy = Ip4Config2PolicyDhcp;
 811   Status = EFI_SUCCESS;
 812   ShellStatus = SHELL_SUCCESS;
 813
 814   if (IsListEmpty (IfList)) {
 815     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INVALID_INTERFACE), gShellNetwork1HiiHandle);
 816   }
 817
 818   //
 819   // Go through the interface list.
 820   //
 821   NET_LIST_FOR_EACH_SAFE (Entry, Next, IfList) {
 822     IfCb = NET_LIST_USER_STRUCT (Entry, IFCONFIG_INTERFACE_CB, Link);
 823
 824     Status = IfCb->IfCfg->SetData (
 825                             IfCb->IfCfg,
 826                             Ip4Config2DataTypePolicy,
 827                             sizeof (EFI_IP4_CONFIG2_POLICY),
 828                             &Policy
 829                             );
 830     if (EFI_ERROR (Status)) {
 831       ShellStatus = SHELL_ACCESS_DENIED;
 832       break;
 833     }
 834   }
 835
 836   return ShellStatus;
 837 }
 838
 839 /**
 840   The set process of the ifconfig command.
 841
 842   @param[in]   IfList    The pointer of IfList(interface list).
 843   @param[in]   VarArg    The pointer of ARG_LIST(Args with "-s" option).
 844
 845   @retval SHELL_SUCCESS  The ifconfig command set processed successfully.
 846   @retval others         The ifconfig command set process failed.
 847
 848 **/
 849 SHELL_STATUS
 850 IfConfigSetInterfaceInfo (
 851   IN LIST_ENTRY    *IfList,
 852   IN ARG_LIST      *VarArg
 853   )
 854 {
 855   EFI_STATUS                       Status;
 856   SHELL_STATUS                     ShellStatus;
 857   IFCONFIG_INTERFACE_CB            *IfCb;
 858   VAR_CHECK_CODE                   CheckCode;
 859   EFI_EVENT                        TimeOutEvt;
 860   EFI_EVENT                        MappedEvt;
 861   BOOLEAN                          IsAddressOk;
 862
 863   EFI_IP4_CONFIG2_POLICY           Policy;
 864   EFI_IP4_CONFIG2_MANUAL_ADDRESS   ManualAddress;
 865   UINTN                            DataSize;
 866   EFI_IPv4_ADDRESS                 Gateway;
 867   EFI_IPv4_ADDRESS                 *Dns;
 868   ARG_LIST                         *Tmp;
 869   UINTN                            Index;
 870
 871   CONST CHAR16* TempString;
 872
 873   Dns = NULL;
 874
 875   if (IsListEmpty (IfList)) {
 876     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INVALID_INTERFACE), gShellNetwork1HiiHandle);
 877     return SHELL_INVALID_PARAMETER;
 878   }
 879
 880   //
 881   // Make sure to set only one interface each time.
 882   //
 883   IfCb   = NET_LIST_USER_STRUCT (IfList->ForwardLink, IFCONFIG_INTERFACE_CB, Link);
 884   Status = EFI_SUCCESS;
 885   ShellStatus = SHELL_SUCCESS;
 886
 887   //
 888   // Initialize check list mechanism.
 889   //
 890   CheckCode = IfConfigRetriveCheckListByName(
 891                 NULL,
 892                 NULL,
 893                 TRUE
 894                 );
 895
 896   //
 897   // Create events & timers for asynchronous settings.
 898   //
 899   Status = gBS->CreateEvent (
 900                   EVT_TIMER,
 901                   TPL_CALLBACK,
 902                   NULL,
 903                   NULL,
 904                   &TimeOutEvt
 905                   );
 906   if (EFI_ERROR (Status)) {
 907     ShellStatus = SHELL_ACCESS_DENIED;
 908     goto ON_EXIT;
 909   }
 910
 911   Status = gBS->CreateEvent (
 912                   EVT_NOTIFY_SIGNAL,
 913                   TPL_NOTIFY,
 914                   IfConfigManualAddressNotify,
 915                   &IsAddressOk,
 916                   &MappedEvt
 917                   );
 918   if (EFI_ERROR (Status)) {
 919     ShellStatus = SHELL_ACCESS_DENIED;
 920     goto ON_EXIT;
 921   }
 922
 923   //
 924   // Parse the setting variables.
 925   //
 926   while (VarArg != NULL) {
 927     //
 928     // Check invalid parameters (duplication & unknown & conflict).
 929     //
 930     CheckCode = IfConfigRetriveCheckListByName(
 931                   mSetCheckList,
 932                   VarArg->Arg,
 933                   FALSE
 934                   );
 935
 936     if (VarCheckOk != CheckCode) {
 937       switch (CheckCode) {
 938         case VarCheckDuplicate:
 939           ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_DUPLICATE_COMMAND), gShellNetwork1HiiHandle, VarArg->Arg);
 940           break;
 941
 942         case VarCheckConflict:
 943           ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_CONFLICT_COMMAND), gShellNetwork1HiiHandle, VarArg->Arg);
 944           break;
 945
 946         case VarCheckUnknown:
 947           //
 948           // To handle unsupported option.
 949           //
 950           TempString = PermanentString;
 951           if (StringNoCaseCompare(&VarArg->Arg, &TempString) == 0) {
 952             ShellPrintHiiEx(-1, -1, NULL,STRING_TOKEN (STR_IFCONFIG_UNSUPPORTED_OPTION), gShellNetwork1HiiHandle, PermanentString);
 953             goto ON_EXIT;
 954           }
 955
 956           //
 957           // To handle unknown option.
 958           //
 959           ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_UNKNOWN_COMMAND), gShellNetwork1HiiHandle, VarArg->Arg);
 960           break;
 961
 962         default:
 963           break;
 964       }
 965
 966       VarArg = VarArg->Next;
 967       continue;
 968     }
 969
 970     //
 971     // Process valid variables.
 972     //
 973     if (StrCmp(VarArg->Arg, L"dhcp") == 0) {
 974       if (IfCb->Policy == Ip4Config2PolicyDhcp) {
 975         Status = IfConfigStartIp4 (IfCb->NicHandle, gImageHandle);
 976         if (EFI_ERROR(Status)) {
 977           ShellStatus = SHELL_ACCESS_DENIED;
 978           goto ON_EXIT;
 979         }
 980       } else {
 981         //
 982         // Set dhcp config policy
 983         //
 984         Policy = Ip4Config2PolicyDhcp;
 985         Status = IfCb->IfCfg->SetData (
 986                                 IfCb->IfCfg,
 987                                 Ip4Config2DataTypePolicy,
 988                                 sizeof (EFI_IP4_CONFIG2_POLICY),
 989                                 &Policy
 990                                 );
 991         if (EFI_ERROR(Status)) {
 992           ShellStatus = SHELL_ACCESS_DENIED;
 993           goto ON_EXIT;
 994         }
 995       }
 996
 997       VarArg= VarArg->Next;
 998
 999     } else if (StrCmp (VarArg->Arg, L"static") == 0) {
1000       //
1001       // Set manual config policy.
1002       //
1003       Policy = Ip4Config2PolicyStatic;
1004       Status = IfCb->IfCfg->SetData (
1005                               IfCb->IfCfg,
1006                               Ip4Config2DataTypePolicy,
1007                               sizeof (EFI_IP4_CONFIG2_POLICY),
1008                               &Policy
1009                               );
1010       if (EFI_ERROR(Status)) {
1011         ShellStatus = SHELL_ACCESS_DENIED;
1012         goto ON_EXIT;
1013       }
1014
1015       VarArg= VarArg->Next;
1016
1017       ZeroMem (&ManualAddress, sizeof (ManualAddress));
1018
1019       //
1020       // Get manual IP address.
1021       //
1022       Status = NetLibStrToIp4 (VarArg->Arg, &ManualAddress.Address);
1023       if (EFI_ERROR(Status)) {
1024         ShellStatus = SHELL_INVALID_PARAMETER;
1025         goto ON_EXIT;
1026       }
1027
1028       //
1029       // Get subnetmask.
1030       //
1031       VarArg = VarArg->Next;
1032       Status = NetLibStrToIp4 (VarArg->Arg, &ManualAddress.SubnetMask);
1033       if (EFI_ERROR(Status)) {
1034         ShellStatus = SHELL_INVALID_PARAMETER;
1035         goto ON_EXIT;
1036       }
1037
1038       //
1039       // Get gateway.
1040       //
1041       VarArg = VarArg->Next;
1042       Status = NetLibStrToIp4 (VarArg->Arg, &Gateway);
1043       if (EFI_ERROR(Status)) {
1044         ShellStatus = SHELL_INVALID_PARAMETER;
1045         goto ON_EXIT;
1046       }
1047
1048       IsAddressOk = FALSE;
1049
1050       Status = IfCb->IfCfg->RegisterDataNotify (
1051                               IfCb->IfCfg,
1052                               Ip4Config2DataTypeManualAddress,
1053                               MappedEvt
1054                               );
1055       if (EFI_ERROR (Status)) {
1056         ShellStatus = SHELL_ACCESS_DENIED;
1057         goto ON_EXIT;
1058       }
1059
1060       DataSize = sizeof (EFI_IP4_CONFIG2_MANUAL_ADDRESS);
1061
1062       Status = IfCb->IfCfg->SetData (
1063                               IfCb->IfCfg,
1064                               Ip4Config2DataTypeManualAddress,
1065                               DataSize,
1066                               &ManualAddress
1067                               );
1068
1069       if (Status == EFI_NOT_READY) {
1070         gBS->SetTimer (TimeOutEvt, TimerRelative, 50000000);
1071
1072         while (EFI_ERROR (gBS->CheckEvent (TimeOutEvt))) {
1073           if (IsAddressOk) {
1074             Status = EFI_SUCCESS;
1075             break;
1076           }
1077         }
1078       }
1079
1080       IfCb->IfCfg->UnregisterDataNotify (
1081                      IfCb->IfCfg,
1082                      Ip4Config2DataTypeManualAddress,
1083                      MappedEvt
1084                      );
1085
1086       if (EFI_ERROR (Status)) {
1087         ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_SET_ADDR_FAILED), gShellNetwork1HiiHandle, Status);
1088         ShellStatus = SHELL_ACCESS_DENIED;
1089         goto ON_EXIT;
1090       }
1091
1092       //
1093       // Set gateway.
1094       //
1095       DataSize = sizeof (EFI_IPv4_ADDRESS);
1096
1097       Status = IfCb->IfCfg->SetData (
1098                               IfCb->IfCfg,
1099                               Ip4Config2DataTypeGateway,
1100                               DataSize,
1101                               &Gateway
1102                               );
1103       if (EFI_ERROR (Status)) {
1104         ShellStatus = SHELL_ACCESS_DENIED;
1105         goto ON_EXIT;
1106       }
1107
1108       VarArg = VarArg->Next;
1109
1110     } else if (StrCmp (VarArg->Arg, L"dns") == 0) {
1111       //
1112       // Get DNS addresses.
1113       //
1114       VarArg = VarArg->Next;
1115       Tmp    = VarArg;
1116       Index  = 0;
1117       while (Tmp != NULL) {
1118         Index ++;
1119         Tmp = Tmp->Next;
1120       }
1121
1122       Dns   = AllocatePool (Index * sizeof (EFI_IPv4_ADDRESS));
1123       ASSERT(Dns != NULL);
1124       Tmp   = VarArg;
1125       Index = 0;
1126       while (Tmp != NULL) {
1127         Status = NetLibStrToIp4 (Tmp->Arg, Dns + Index);
1128         if (EFI_ERROR(Status)) {
1129           ShellStatus = SHELL_INVALID_PARAMETER;
1130           goto ON_EXIT;
1131         }
1132         Index ++;
1133         Tmp = Tmp->Next;
1134       }
1135
1136       VarArg = Tmp;
1137
1138       //
1139       // Set DNS addresses.
1140       //
1141       DataSize = Index * sizeof (EFI_IPv4_ADDRESS);
1142
1143       Status = IfCb->IfCfg->SetData (
1144                               IfCb->IfCfg,
1145                               Ip4Config2DataTypeDnsServer,
1146                               DataSize,
1147                               Dns
1148                               );
1149       if (EFI_ERROR (Status)) {
1150         ShellStatus = SHELL_ACCESS_DENIED;
1151         goto ON_EXIT;
1152       }
1153     }
1154   }
1155
1156 ON_EXIT:
1157   if (Dns != NULL) {
1158     FreePool (Dns);
1159   }
1160
1161   return ShellStatus;
1162
1163 }
1164
1165 /**
1166   The ifconfig command main process.
1167
1168   @param[in]   Private    The pointer of IFCONFIG_PRIVATE_DATA.
1169
1170   @retval SHELL_SUCCESS  ifconfig command processed successfully.
1171   @retval others         The ifconfig command process failed.
1172
1173 **/
1174 SHELL_STATUS
1175 IfConfig (
1176   IN IFCONFIG_PRIVATE_DATA    *Private
1177   )
1178 {
1179   EFI_STATUS    Status;
1180   SHELL_STATUS  ShellStatus;
1181
1182   ShellStatus = SHELL_SUCCESS;
1183
1184   //
1185   // Get configure information of all interfaces.
1186   //
1187   Status = IfConfigGetInterfaceInfo (
1188              Private->IfName,
1189              &Private->IfList
1190              );
1191   if (EFI_ERROR (Status)) {
1192     ShellStatus = SHELL_NOT_FOUND;
1193     goto ON_EXIT;
1194   }
1195
1196   switch (Private->OpCode) {
1197   case IfConfigOpList:
1198     ShellStatus = IfConfigShowInterfaceInfo (&Private->IfList);
1199     break;
1200
1201   case IfConfigOpClear:
1202     ShellStatus = IfConfigClearInterfaceInfo (&Private->IfList);
1203     break;
1204
1205   case IfConfigOpSet:
1206     ShellStatus = IfConfigSetInterfaceInfo (&Private->IfList, Private->VarArg);
1207     break;
1208
1209   default:
1210     ShellStatus = SHELL_UNSUPPORTED;
1211   }
1212
1213 ON_EXIT:
1214   return ShellStatus;
1215 }
1216
1217 /**
1218   The ifconfig command cleanup process, free the allocated memory.
1219
1220   @param[in]   Private    The pointer of  IFCONFIG_PRIVATE_DATA.
1221
1222 **/
1223 VOID
1224 IfConfigCleanup (
1225   IN IFCONFIG_PRIVATE_DATA  *Private
1226   )
1227 {
1228   LIST_ENTRY                *Entry;
1229   LIST_ENTRY                *NextEntry;
1230   IFCONFIG_INTERFACE_CB     *IfCb;
1231   ARG_LIST                  *ArgNode;
1232   ARG_LIST                  *ArgHead;
1233
1234   ASSERT (Private != NULL);
1235
1236   //
1237   // Clean the list which save the set config Args.
1238   //
1239   if (Private->VarArg != NULL) {
1240     ArgHead = Private->VarArg;
1241
1242     while (ArgHead->Next != NULL) {
1243       ArgNode = ArgHead->Next;
1244       FreePool (ArgHead);
1245       ArgHead = ArgNode;
1246     }
1247
1248     FreePool (ArgHead);
1249   }
1250
1251   if (Private->IfName != NULL) {
1252     FreePool (Private->IfName);
1253   }
1254
1255   //
1256   // Clean the IFCONFIG_INTERFACE_CB list.
1257   //
1258   NET_LIST_FOR_EACH_SAFE (Entry, NextEntry, &Private->IfList) {
1259     IfCb = NET_LIST_USER_STRUCT (Entry, IFCONFIG_INTERFACE_CB, Link);
1260
1261     RemoveEntryList (&IfCb->Link);
1262
1263     if (IfCb->IfInfo != NULL) {
1264
1265       FreePool (IfCb->IfInfo);
1266     }
1267
1268     FreePool (IfCb);
1269   }
1270
1271   FreePool (Private);
1272 }
1273
1274 /**
1275   Function for 'ifconfig' command.
1276
1277   @param[in] ImageHandle  Handle to the Image (NULL if Internal).
1278   @param[in] SystemTable  Pointer to the System Table (NULL if Internal).
1279
1280   @retval EFI_SUCCESS    ifconfig command processed successfully.
1281   @retval others         The ifconfig command process failed.
1282
1283 **/
1284 SHELL_STATUS
1285 EFIAPI
1286 ShellCommandRunIfconfig (
1287   IN EFI_HANDLE        ImageHandle,
1288   IN EFI_SYSTEM_TABLE  *SystemTable
1289   )
1290 {
1291   EFI_STATUS                Status;
1292   IFCONFIG_PRIVATE_DATA     *Private;
1293   LIST_ENTRY                *ParamPackage;
1294   SHELL_STATUS              ShellStatus;
1295   CONST CHAR16              *ValueStr;
1296   ARG_LIST                  *ArgList;
1297   CHAR16                    *ProblemParam;
1298   CHAR16                    *Str;
1299
1300   Status = EFI_INVALID_PARAMETER;
1301   Private = NULL;
1302   ShellStatus = SHELL_SUCCESS;
1303
1304   Status = ShellCommandLineParseEx (mIfConfigCheckList, &ParamPackage, &ProblemParam, TRUE, FALSE);
1305   if (EFI_ERROR (Status)) {
1306     if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {
1307       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellNetwork1HiiHandle, L"ifconfig", ProblemParam);
1308       FreePool(ProblemParam);
1309       ShellStatus = SHELL_INVALID_PARAMETER;
1310     } else {
1311       ASSERT(FALSE);
1312     }
1313
1314     goto ON_EXIT;
1315   }
1316
1317   //
1318   // To handle unsupported option.
1319   //
1320   if (ShellCommandLineGetFlag (ParamPackage, L"-c")) {
1321     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_UNSUPPORTED_OPTION), gShellNetwork1HiiHandle,L"-c");
1322     ShellStatus = SHELL_INVALID_PARAMETER;
1323     goto ON_EXIT;
1324   }
1325
1326   //
1327   // To handle no option.
1328   //
1329   if (!ShellCommandLineGetFlag (ParamPackage, L"-r") && !ShellCommandLineGetFlag (ParamPackage, L"-s") &&
1330       !ShellCommandLineGetFlag (ParamPackage, L"-l")) {
1331     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_LACK_OPTION), gShellNetwork1HiiHandle);
1332     ShellStatus = SHELL_INVALID_PARAMETER;
1333     goto ON_EXIT;
1334   }
1335
1336   //
1337   // To handle conflict options.
1338   //
1339   if (((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-s"))) ||
1340       ((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-l"))) ||
1341       ((ShellCommandLineGetFlag (ParamPackage, L"-s")) && (ShellCommandLineGetFlag (ParamPackage, L"-l")))) {
1342     ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CON), gShellNetwork1HiiHandle, L"ifconfig");
1343     ShellStatus = SHELL_INVALID_PARAMETER;
1344     goto ON_EXIT;
1345   }
1346
1347   Private = AllocateZeroPool (sizeof (IFCONFIG_PRIVATE_DATA));
1348   if (Private == NULL) {
1349     ShellStatus = SHELL_OUT_OF_RESOURCES;
1350     goto ON_EXIT;
1351   }
1352
1353   InitializeListHead (&Private->IfList);
1354
1355   //
1356   // To get interface name for the list option.
1357   //
1358   if (ShellCommandLineGetFlag (ParamPackage, L"-l")) {
1359     Private->OpCode = IfConfigOpList;
1360     ValueStr = ShellCommandLineGetValue (ParamPackage, L"-l");
1361     if (ValueStr != NULL) {
1362       Str = AllocateCopyPool (StrSize (ValueStr), ValueStr);
1363       ASSERT (Str != NULL);
1364       Private->IfName = Str;
1365     }
1366   }
1367
1368   //
1369   // To get interface name for the clear option.
1370   //
1371   if (ShellCommandLineGetFlag (ParamPackage, L"-r")) {
1372     Private->OpCode = IfConfigOpClear;
1373     ValueStr = ShellCommandLineGetValue (ParamPackage, L"-r");
1374     if (ValueStr != NULL) {
1375       Str = AllocateCopyPool (StrSize (ValueStr), ValueStr);
1376       ASSERT (Str != NULL);
1377       Private->IfName = Str;
1378     }
1379   }
1380
1381   //
1382   // To get interface name and corresponding Args for the set option.
1383   //
1384   if (ShellCommandLineGetFlag (ParamPackage, L"-s")) {
1385     ValueStr = ShellCommandLineGetValue (ParamPackage, L"-s");
1386     if (ValueStr == NULL) {
1387       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_LACK_INTERFACE), gShellNetwork1HiiHandle);
1388       ShellStatus = SHELL_INVALID_PARAMETER;
1389       goto ON_EXIT;
1390     }
1391
1392     //
1393     // To split the configuration into multi-section.
1394     //
1395     ArgList         = SplitStrToList (ValueStr, L' ');
1396     ASSERT (ArgList != NULL);
1397
1398     Private->OpCode = IfConfigOpSet;
1399     Private->IfName = ArgList->Arg;
1400
1401     Private->VarArg = ArgList->Next;
1402
1403     if (Private->IfName == NULL || Private->VarArg == NULL) {
1404       ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_LACK_COMMAND), gShellNetwork1HiiHandle);
1405       ShellStatus = SHELL_INVALID_PARAMETER;
1406       goto ON_EXIT;
1407     }
1408   }
1409
1410   //
1411   // Main process of ifconfig.
1412   //
1413   ShellStatus = IfConfig (Private);
1414
1415 ON_EXIT:
1416
1417   ShellCommandLineFreeVarList (ParamPackage);
1418
1419   if (Private != NULL) {
1420     IfConfigCleanup (Private);
1421   }
1422
1423   return ShellStatus;
1424 }