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