]> git.proxmox.com Git - mirror_edk2.git/blame - ShellPkg/Library/UefiShellNetwork1CommandsLib/Ifconfig.c
ShellPkg: Fix 'ifconfig' can't get the address from dhcp in some case
[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
7c25b7ea 276\r
d6cf1af9
JW
277/**\r
278 Create an IP child, use it to start the auto configuration, then destroy it.\r
279\r
280 @param[in] Controller The controller which has the service installed.\r
281 @param[in] Image The image handle used to open service.\r
282\r
283 @retval EFI_SUCCESS The configuration is done.\r
284**/\r
285EFI_STATUS\r
286EFIAPI\r
287IfConfigStartIp4(\r
288 IN EFI_HANDLE Controller,\r
289 IN EFI_HANDLE Image\r
290 )\r
291{\r
292 EFI_IP4_PROTOCOL *Ip4;\r
293 EFI_HANDLE Ip4Handle;\r
294 EFI_IP4_CONFIG_DATA Ip4ConfigData;\r
295 EFI_STATUS Status;\r
296\r
297 //\r
298 // Get the Ip4ServiceBinding Protocol\r
299 //\r
300 Ip4Handle = NULL;\r
301 Ip4 = NULL;\r
302\r
303 Status = NetLibCreateServiceChild (\r
304 Controller,\r
305 Image,\r
306 &gEfiIp4ServiceBindingProtocolGuid,\r
307 &Ip4Handle\r
308 );\r
309\r
310 if (EFI_ERROR (Status)) {\r
311 return Status;\r
312 }\r
313\r
314 Status = gBS->OpenProtocol (\r
315 Ip4Handle,\r
316 &gEfiIp4ProtocolGuid,\r
317 (VOID **) &Ip4,\r
318 Controller,\r
319 Image,\r
320 EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
321 );\r
322\r
323 if (EFI_ERROR (Status)) {\r
324 goto ON_EXIT;\r
325 }\r
326\r
327 Ip4ConfigData.DefaultProtocol = EFI_IP_PROTO_ICMP;\r
328 Ip4ConfigData.AcceptAnyProtocol = FALSE;\r
329 Ip4ConfigData.AcceptIcmpErrors = FALSE;\r
330 Ip4ConfigData.AcceptBroadcast = FALSE;\r
331 Ip4ConfigData.AcceptPromiscuous = FALSE;\r
332 Ip4ConfigData.UseDefaultAddress = TRUE;\r
333 ZeroMem (&Ip4ConfigData.StationAddress, sizeof (EFI_IPv4_ADDRESS));\r
334 ZeroMem (&Ip4ConfigData.SubnetMask, sizeof (EFI_IPv4_ADDRESS));\r
335 Ip4ConfigData.TypeOfService = 0;\r
336 Ip4ConfigData.TimeToLive = 1;\r
337 Ip4ConfigData.DoNotFragment = FALSE;\r
338 Ip4ConfigData.RawData = FALSE;\r
339 Ip4ConfigData.ReceiveTimeout = 0;\r
340 Ip4ConfigData.TransmitTimeout = 0;\r
341\r
342 Ip4->Configure (Ip4, &Ip4ConfigData);\r
343 \r
344ON_EXIT: \r
345 NetLibDestroyServiceChild (\r
346 Controller,\r
347 Image,\r
348 &gEfiIp4ServiceBindingProtocolGuid,\r
349 Ip4Handle\r
350 );\r
351 \r
352 return Status;\r
353}\r
354\r
355\r
68fb0527 356/**\r
7c25b7ea 357 Print MAC address.\r
68fb0527 358\r
7c25b7ea 359 @param[in] Node The pointer of MAC address buffer.\r
360 @param[in] Size The size of MAC address buffer.\r
68fb0527 361\r
68fb0527 362**/\r
7c25b7ea 363VOID\r
364IfConfigPrintMacAddr (\r
365 IN UINT8 *Node,\r
366 IN UINT32 Size\r
367 )\r
68fb0527 368{\r
7c25b7ea 369 UINTN Index;\r
68fb0527 370\r
7c25b7ea 371 ASSERT (Size <= MACADDRMAXSIZE);\r
68fb0527 372\r
7c25b7ea 373 for (Index = 0; Index < Size; Index++) {\r
374 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_MAC_ADDR_BODY), gShellNetwork1HiiHandle, Node[Index]);\r
375 if (Index + 1 < Size) {\r
376 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_COLON), gShellNetwork1HiiHandle);\r
377 }\r
68fb0527 378 }\r
68fb0527 379\r
7c25b7ea 380 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_NEWLINE), gShellNetwork1HiiHandle);\r
381}\r
68fb0527 382\r
68fb0527 383\r
7c25b7ea 384/**\r
385 The get current status of all handles.\r
68fb0527 386\r
7c25b7ea 387 @param[in] IfName The pointer of IfName(interface name).\r
388 @param[in] IfList The pointer of IfList(interface list).\r
68fb0527 389\r
7c25b7ea 390 @retval EFI_SUCCESS The get status processed successfully.\r
391 @retval others The get status process failed.\r
68fb0527 392\r
68fb0527 393**/\r
394EFI_STATUS\r
7c25b7ea 395IfConfigGetInterfaceInfo (\r
396 IN CHAR16 *IfName,\r
397 IN LIST_ENTRY *IfList\r
68fb0527 398 )\r
399{\r
7c25b7ea 400 EFI_STATUS Status;\r
401 UINTN HandleIndex;\r
402 UINTN HandleNum;\r
403 EFI_HANDLE *HandleBuffer;\r
404 EFI_IP4_CONFIG2_PROTOCOL *Ip4Cfg2;\r
405 EFI_IP4_CONFIG2_INTERFACE_INFO *IfInfo;\r
406 IFCONFIG_INTERFACE_CB *IfCb;\r
407 UINTN DataSize;\r
68fb0527 408\r
7c25b7ea 409 HandleBuffer = NULL;\r
410 HandleNum = 0;\r
411\r
412 IfInfo = NULL;\r
413 IfCb = NULL;\r
68fb0527 414\r
415 //\r
7c25b7ea 416 // Locate all the handles with ip4 service binding protocol.\r
68fb0527 417 //\r
418 Status = gBS->LocateHandleBuffer (\r
7c25b7ea 419 ByProtocol,\r
420 &gEfiIp4ServiceBindingProtocolGuid,\r
421 NULL,\r
422 &HandleNum,\r
423 &HandleBuffer\r
68fb0527 424 );\r
7c25b7ea 425 if (EFI_ERROR (Status) || (HandleNum == 0)) {\r
426 return EFI_ABORTED;\r
68fb0527 427 }\r
428\r
7c25b7ea 429 //\r
430 // Enumerate all handles that installed with ip4 service binding protocol.\r
431 //\r
432 for (HandleIndex = 0; HandleIndex < HandleNum; HandleIndex++) {\r
433 IfCb = NULL;\r
434 IfInfo = NULL;\r
435 DataSize = 0;\r
436\r
437 //\r
438 // Ip4config protocol and ip4 service binding protocol are installed\r
439 // on the same handle.\r
440 //\r
441 ASSERT (HandleBuffer != NULL);\r
442 Status = gBS->HandleProtocol (\r
443 HandleBuffer[HandleIndex],\r
444 &gEfiIp4Config2ProtocolGuid,\r
445 (VOID **) &Ip4Cfg2\r
446 );\r
447\r
68fb0527 448 if (EFI_ERROR (Status)) {\r
7c25b7ea 449 goto ON_ERROR;\r
68fb0527 450 }\r
7c25b7ea 451 \r
68fb0527 452 //\r
7c25b7ea 453 // Get the interface information size.\r
68fb0527 454 //\r
7c25b7ea 455 Status = Ip4Cfg2->GetData (\r
456 Ip4Cfg2,\r
457 Ip4Config2DataTypeInterfaceInfo,\r
458 &DataSize,\r
459 NULL\r
460 );\r
461\r
462 if (Status != EFI_BUFFER_TOO_SMALL) {\r
463 goto ON_ERROR;\r
33c031ee 464 }\r
7c25b7ea 465\r
466 IfInfo = AllocateZeroPool (DataSize);\r
467\r
468 if (IfInfo == NULL) {\r
68fb0527 469 Status = EFI_OUT_OF_RESOURCES;\r
470 goto ON_ERROR;\r
471 }\r
7c25b7ea 472 \r
68fb0527 473 //\r
7c25b7ea 474 // Get the interface info.\r
68fb0527 475 //\r
7c25b7ea 476 Status = Ip4Cfg2->GetData (\r
477 Ip4Cfg2,\r
478 Ip4Config2DataTypeInterfaceInfo,\r
479 &DataSize,\r
480 IfInfo\r
481 );\r
68fb0527 482\r
7c25b7ea 483 if (EFI_ERROR (Status)) {\r
68fb0527 484 goto ON_ERROR;\r
485 }\r
7c25b7ea 486 \r
68fb0527 487 //\r
7c25b7ea 488 // Check the interface name if required.\r
68fb0527 489 //\r
7c25b7ea 490 if ((IfName != NULL) && (StrCmp (IfName, IfInfo->Name) != 0)) {\r
491 FreePool (IfInfo);\r
492 continue;\r
68fb0527 493 }\r
494\r
7c25b7ea 495 DataSize = 0;\r
496 \r
68fb0527 497 //\r
7c25b7ea 498 // Get the size of dns server list.\r
68fb0527 499 //\r
7c25b7ea 500 Status = Ip4Cfg2->GetData (\r
501 Ip4Cfg2,\r
502 Ip4Config2DataTypeDnsServer,\r
503 &DataSize,\r
504 NULL\r
505 );\r
506\r
507 if ((Status != EFI_BUFFER_TOO_SMALL) && (Status != EFI_NOT_FOUND)) {\r
508 goto ON_ERROR;\r
509 }\r
510\r
511 IfCb = AllocateZeroPool (sizeof (IFCONFIG_INTERFACE_CB) + DataSize);\r
68fb0527 512\r
7c25b7ea 513 if (IfCb == NULL) {\r
68fb0527 514 Status = EFI_OUT_OF_RESOURCES;\r
515 goto ON_ERROR;\r
516 }\r
517\r
7c25b7ea 518 IfCb->NicHandle = HandleBuffer[HandleIndex];\r
519 IfCb->IfInfo = IfInfo;\r
520 IfCb->IfCfg = Ip4Cfg2;\r
521 IfCb->DnsCnt = (UINT32) (DataSize / sizeof (EFI_IPv4_ADDRESS));\r
522\r
68fb0527 523 //\r
7c25b7ea 524 // Get the dns server list if has.\r
68fb0527 525 //\r
7c25b7ea 526 if (DataSize > 0) {\r
527 Status = Ip4Cfg2->GetData (\r
528 Ip4Cfg2,\r
529 Ip4Config2DataTypeDnsServer,\r
530 &DataSize,\r
531 IfCb->DnsAddr\r
532 );\r
533\r
534 if (EFI_ERROR (Status)) {\r
535 goto ON_ERROR;\r
68fb0527 536 }\r
68fb0527 537 }\r
538\r
539 //\r
7c25b7ea 540 // Get the config policy.\r
68fb0527 541 //\r
7c25b7ea 542 DataSize = sizeof (EFI_IP4_CONFIG2_POLICY);\r
543 Status = Ip4Cfg2->GetData (\r
544 Ip4Cfg2,\r
545 Ip4Config2DataTypePolicy,\r
546 &DataSize,\r
547 &IfCb->Policy\r
548 );\r
549\r
550 if (EFI_ERROR (Status)) {\r
551 goto ON_ERROR;\r
552 }\r
553\r
554 InsertTailList (IfList, &IfCb->Link);\r
68fb0527 555\r
7c25b7ea 556 if ((IfName != NULL) && (StrCmp (IfName, IfInfo->Name) == 0)) {\r
557 //\r
558 // Only need the appointed interface, keep the allocated buffer.\r
559 //\r
560 IfCb = NULL;\r
561 IfInfo = NULL;\r
562 break;\r
563 }\r
68fb0527 564 }\r
565\r
7c25b7ea 566 if (HandleBuffer != NULL) {\r
567 FreePool (HandleBuffer);\r
568 }\r
68fb0527 569\r
570 return EFI_SUCCESS;\r
7c25b7ea 571\r
68fb0527 572ON_ERROR:\r
7c25b7ea 573\r
574 if (IfInfo != NULL) {\r
575 FreePool (IfInfo);\r
68fb0527 576 }\r
577\r
7c25b7ea 578 if (IfCb != NULL) {\r
579 FreePool (IfCb);\r
580 }\r
68fb0527 581\r
582 return Status;\r
583}\r
584\r
585/**\r
7c25b7ea 586 The list process of the ifconfig command.\r
68fb0527 587\r
7c25b7ea 588 @param[in] IfList The pointer of IfList(interface list).\r
589\r
590 @retval EFI_SUCCESS The ifconfig command list processed successfully.\r
591 @retval others The ifconfig command list process failed.\r
68fb0527 592\r
68fb0527 593**/\r
7c25b7ea 594EFI_STATUS\r
595IfConfigShowInterfaceInfo (\r
596 IN LIST_ENTRY *IfList\r
68fb0527 597 )\r
598{\r
7c25b7ea 599 LIST_ENTRY *Entry;\r
600 LIST_ENTRY *Next;\r
601 IFCONFIG_INTERFACE_CB *IfCb;\r
602 EFI_IPv4_ADDRESS Gateway;\r
603 UINT32 Index;\r
68fb0527 604\r
7c25b7ea 605 if (IsListEmpty (IfList)) {\r
606 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INVALID_INTERFACE), gShellNetwork1HiiHandle);\r
68fb0527 607 }\r
608\r
609 //\r
7c25b7ea 610 // Go through the interface list.\r
68fb0527 611 //\r
7c25b7ea 612 NET_LIST_FOR_EACH_SAFE (Entry, Next, IfList) {\r
613 IfCb = NET_LIST_USER_STRUCT (Entry, IFCONFIG_INTERFACE_CB, Link);\r
68fb0527 614\r
7c25b7ea 615 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_BREAK), gShellNetwork1HiiHandle);\r
68fb0527 616\r
7c25b7ea 617 //\r
618 // Print interface name.\r
619 //\r
620 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_IF_NAME), gShellNetwork1HiiHandle, IfCb->IfInfo->Name);\r
68fb0527 621\r
7c25b7ea 622 //\r
623 // Print interface config policy.\r
624 //\r
625 if (IfCb->Policy == Ip4Config2PolicyDhcp) {\r
626 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_POLICY_DHCP), gShellNetwork1HiiHandle);\r
627 } else {\r
628 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_POLICY_MAN), gShellNetwork1HiiHandle);\r
629 }\r
68fb0527 630\r
7c25b7ea 631 //\r
632 // Print mac address of the interface.\r
633 //\r
634 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_MAC_ADDR_HEAD), gShellNetwork1HiiHandle);\r
68fb0527 635\r
7c25b7ea 636 IfConfigPrintMacAddr (\r
637 IfCb->IfInfo->HwAddress.Addr,\r
638 IfCb->IfInfo->HwAddressSize\r
639 );\r
68fb0527 640\r
7c25b7ea 641 //\r
642 // Print IPv4 address list of the interface.\r
643 //\r
644 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_IP_ADDR_HEAD), gShellNetwork1HiiHandle);\r
68fb0527 645\r
7c25b7ea 646 ShellPrintHiiEx(\r
647 -1, \r
648 -1, \r
649 NULL,\r
650 STRING_TOKEN (STR_IFCONFIG_INFO_IP_ADDR_BODY), \r
651 gShellNetwork1HiiHandle,\r
652 (UINTN)IfCb->IfInfo->StationAddress.Addr[0],\r
653 (UINTN)IfCb->IfInfo->StationAddress.Addr[1],\r
654 (UINTN)IfCb->IfInfo->StationAddress.Addr[2],\r
655 (UINTN)IfCb->IfInfo->StationAddress.Addr[3]\r
656 );\r
657\r
658 //\r
659 // Print subnet mask list of the interface.\r
660 //\r
661 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_SUBNET_MASK_HEAD), gShellNetwork1HiiHandle);\r
68fb0527 662\r
68fb0527 663 ShellPrintHiiEx(\r
664 -1, \r
665 -1, \r
666 NULL,\r
7c25b7ea 667 STRING_TOKEN (STR_IFCONFIG_INFO_IP_ADDR_BODY), \r
668 gShellNetwork1HiiHandle,\r
669 (UINTN)IfCb->IfInfo->SubnetMask.Addr[0],\r
670 (UINTN)IfCb->IfInfo->SubnetMask.Addr[1],\r
671 (UINTN)IfCb->IfInfo->SubnetMask.Addr[2],\r
672 (UINTN)IfCb->IfInfo->SubnetMask.Addr[3]\r
68fb0527 673 );\r
68fb0527 674\r
7c25b7ea 675 //\r
676 // Print default gateway of the interface.\r
677 //\r
678 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_GATEWAY_HEAD), gShellNetwork1HiiHandle);\r
679\r
680 ZeroMem (&Gateway, sizeof (EFI_IPv4_ADDRESS));\r
681 \r
682 for (Index = 0; Index < IfCb->IfInfo->RouteTableSize; Index++) {\r
683 if ((CompareMem (&IfCb->IfInfo->RouteTable[Index].SubnetAddress, &mZeroIp4Addr, sizeof (EFI_IPv4_ADDRESS)) == 0) &&\r
684 (CompareMem (&IfCb->IfInfo->RouteTable[Index].SubnetMask , &mZeroIp4Addr, sizeof (EFI_IPv4_ADDRESS)) == 0) ){\r
685 CopyMem (&Gateway, &IfCb->IfInfo->RouteTable[Index].GatewayAddress, sizeof (EFI_IPv4_ADDRESS));\r
686 }\r
687 } \r
688\r
68fb0527 689 ShellPrintHiiEx(\r
7c25b7ea 690 -1, \r
691 -1, \r
692 NULL,\r
693 STRING_TOKEN (STR_IFCONFIG_INFO_IP_ADDR_BODY), \r
694 gShellNetwork1HiiHandle,\r
695 (UINTN)Gateway.Addr[0],\r
696 (UINTN)Gateway.Addr[1],\r
697 (UINTN)Gateway.Addr[2],\r
698 (UINTN)Gateway.Addr[3]\r
699 );\r
700 \r
701 //\r
702 // Print route table entry.\r
703 //\r
704 ShellPrintHiiEx(-1, -1, NULL,STRING_TOKEN (STR_IFCONFIG_ROUTES_SIZE), gShellNetwork1HiiHandle, IfCb->IfInfo->RouteTableSize);\r
705\r
706 for (Index = 0; Index < IfCb->IfInfo->RouteTableSize; Index++) {\r
707 ShellPrintHiiEx(-1, -1, NULL,STRING_TOKEN (STR_IFCONFIG_ROUTES_ENTRY_INDEX), gShellNetwork1HiiHandle, Index);\r
708\r
709 ShellPrintHiiEx(\r
710 -1, \r
711 -1, \r
712 NULL,\r
713 STRING_TOKEN (STR_IFCONFIG_SHOW_IP_ADDR), \r
714 gShellNetwork1HiiHandle, \r
715 L"Subnet ",\r
716 (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetAddress.Addr[0],\r
717 (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetAddress.Addr[1],\r
718 (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetAddress.Addr[2],\r
719 (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetAddress.Addr[3]\r
720 );\r
721\r
722 ShellPrintHiiEx(\r
723 -1, \r
724 -1, \r
725 NULL,\r
726 STRING_TOKEN (STR_IFCONFIG_SHOW_IP_ADDR), \r
727 gShellNetwork1HiiHandle, \r
728 L"Netmask",\r
729 (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetMask.Addr[0],\r
730 (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetMask.Addr[1],\r
731 (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetMask.Addr[2],\r
732 (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetMask.Addr[3]\r
733 );\r
734\r
735 ShellPrintHiiEx(\r
736 -1, \r
737 -1, \r
738 NULL,\r
739 STRING_TOKEN (STR_IFCONFIG_SHOW_IP_ADDR), \r
740 gShellNetwork1HiiHandle, \r
741 L"Gateway",\r
742 (UINTN)IfCb->IfInfo->RouteTable[Index].GatewayAddress.Addr[0],\r
743 (UINTN)IfCb->IfInfo->RouteTable[Index].GatewayAddress.Addr[1],\r
744 (UINTN)IfCb->IfInfo->RouteTable[Index].GatewayAddress.Addr[2],\r
745 (UINTN)IfCb->IfInfo->RouteTable[Index].GatewayAddress.Addr[3]\r
746 );\r
747 }\r
748\r
749 //\r
750 // Print dns server addresses list of the interface if has.\r
751 //\r
752 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_DNS_ADDR_HEAD), gShellNetwork1HiiHandle);\r
753\r
754 for (Index = 0; Index < IfCb->DnsCnt; Index++) {\r
755 ShellPrintHiiEx(\r
756 -1, \r
757 -1, \r
758 NULL,\r
759 STRING_TOKEN (STR_IFCONFIG_INFO_DNS_ADDR_BODY), \r
760 gShellNetwork1HiiHandle,\r
761 (UINTN) IfCb->DnsAddr[Index].Addr[0],\r
762 (UINTN) IfCb->DnsAddr[Index].Addr[1],\r
763 (UINTN) IfCb->DnsAddr[Index].Addr[2],\r
764 (UINTN) IfCb->DnsAddr[Index].Addr[3]\r
765 );\r
766\r
767 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_NEWLINE), gShellNetwork1HiiHandle);\r
768 }\r
68fb0527 769 }\r
7c25b7ea 770 \r
771 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_BREAK), gShellNetwork1HiiHandle);\r
68fb0527 772\r
7c25b7ea 773 return EFI_SUCCESS;\r
68fb0527 774}\r
775\r
776/**\r
7c25b7ea 777 The clean process of the ifconfig command to clear interface info.\r
778\r
779 @param[in] IfList The pointer of IfList(interface list).\r
68fb0527 780\r
7c25b7ea 781 @retval EFI_SUCCESS The ifconfig command clean processed successfully.\r
782 @retval others The ifconfig command clean process failed.\r
68fb0527 783\r
68fb0527 784**/\r
7c25b7ea 785EFI_STATUS\r
786IfConfigClearInterfaceInfo (\r
787 IN LIST_ENTRY *IfList\r
68fb0527 788 )\r
789{\r
7c25b7ea 790 EFI_STATUS Status;\r
791 LIST_ENTRY *Entry;\r
792 LIST_ENTRY *Next;\r
793 IFCONFIG_INTERFACE_CB *IfCb;\r
794 EFI_IP4_CONFIG2_POLICY Policy;\r
68fb0527 795\r
7c25b7ea 796 Policy = Ip4Config2PolicyDhcp;\r
797 Status = EFI_SUCCESS;\r
68fb0527 798\r
7c25b7ea 799 if (IsListEmpty (IfList)) {\r
800 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INVALID_INTERFACE), gShellNetwork1HiiHandle);\r
68fb0527 801 }\r
802\r
803 //\r
7c25b7ea 804 // Go through the interface list.\r
68fb0527 805 //\r
7c25b7ea 806 NET_LIST_FOR_EACH_SAFE (Entry, Next, IfList) {\r
807 IfCb = NET_LIST_USER_STRUCT (Entry, IFCONFIG_INTERFACE_CB, Link);\r
68fb0527 808 \r
7c25b7ea 809 Status = IfCb->IfCfg->SetData (\r
810 IfCb->IfCfg,\r
811 Ip4Config2DataTypePolicy,\r
812 sizeof (EFI_IP4_CONFIG2_POLICY),\r
813 &Policy\r
814 );\r
68fb0527 815\r
7c25b7ea 816 if (EFI_ERROR (Status)) {\r
817 break;\r
818 }\r
68fb0527 819 }\r
820\r
7c25b7ea 821 return Status;\r
68fb0527 822}\r
823\r
824/**\r
7c25b7ea 825 The set process of the ifconfig command.\r
68fb0527 826\r
7c25b7ea 827 @param[in] IfList The pointer of IfList(interface list).\r
828 @param[in] VarArg The pointer of ARG_LIST(Args with "-s" option).\r
68fb0527 829\r
7c25b7ea 830 @retval EFI_SUCCESS The ifconfig command set processed successfully.\r
831 @retval others The ifconfig command set process failed.\r
68fb0527 832\r
68fb0527 833**/\r
834EFI_STATUS\r
7c25b7ea 835IfConfigSetInterfaceInfo (\r
836 IN LIST_ENTRY *IfList,\r
837 IN ARG_LIST *VarArg\r
68fb0527 838 )\r
839{\r
68fb0527 840\r
7c25b7ea 841 EFI_STATUS Status;\r
842 IFCONFIG_INTERFACE_CB *IfCb;\r
843 VAR_CHECK_CODE CheckCode;\r
844 EFI_EVENT TimeOutEvt;\r
845 EFI_EVENT MappedEvt;\r
846 BOOLEAN IsAddressOk;\r
68fb0527 847\r
7c25b7ea 848 EFI_IP4_CONFIG2_POLICY Policy;\r
849 EFI_IP4_CONFIG2_MANUAL_ADDRESS ManualAddress;\r
850 UINTN DataSize;\r
851 EFI_IPv4_ADDRESS Gateway;\r
852 EFI_IPv4_ADDRESS *Dns;\r
853 ARG_LIST *Tmp;\r
854 UINTN Index;\r
68fb0527 855\r
7c25b7ea 856 CONST CHAR16* TempString;\r
68fb0527 857\r
7c25b7ea 858 Dns = NULL;\r
68fb0527 859\r
7c25b7ea 860 if (IsListEmpty (IfList)) {\r
861 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INVALID_INTERFACE), gShellNetwork1HiiHandle);\r
862 return EFI_INVALID_PARAMETER;\r
68fb0527 863 }\r
68fb0527 864 \r
7c25b7ea 865 //\r
866 // Make sure to set only one interface each time.\r
867 //\r
868 IfCb = NET_LIST_USER_STRUCT (IfList->ForwardLink, IFCONFIG_INTERFACE_CB, Link);\r
869 Status = EFI_SUCCESS;\r
68fb0527 870\r
7c25b7ea 871 //\r
872 // Initialize check list mechanism.\r
873 //\r
874 CheckCode = IfConfigRetriveCheckListByName(\r
875 NULL,\r
876 NULL,\r
877 TRUE\r
878 );\r
68fb0527 879\r
7c25b7ea 880 //\r
881 // Create events & timers for asynchronous settings.\r
882 //\r
883 Status = gBS->CreateEvent (\r
884 EVT_TIMER,\r
885 TPL_CALLBACK,\r
886 NULL,\r
887 NULL,\r
888 &TimeOutEvt\r
889 );\r
68fb0527 890 if (EFI_ERROR (Status)) {\r
7c25b7ea 891 goto ON_EXIT;\r
68fb0527 892 }\r
893\r
7c25b7ea 894 Status = gBS->CreateEvent (\r
895 EVT_NOTIFY_SIGNAL,\r
896 TPL_NOTIFY,\r
897 IfConfigManualAddressNotify,\r
898 &IsAddressOk,\r
899 &MappedEvt\r
900 );\r
901 if (EFI_ERROR (Status)) {\r
902 goto ON_EXIT;\r
68fb0527 903 }\r
904\r
7c25b7ea 905 //\r
906 // Parse the setting variables.\r
907 //\r
908 while (VarArg != NULL) {\r
909 //\r
910 // Check invalid parameters (duplication & unknown & conflict).\r
911 //\r
912 CheckCode = IfConfigRetriveCheckListByName(\r
913 mSetCheckList,\r
914 VarArg->Arg,\r
915 FALSE\r
916 );\r
68fb0527 917\r
7c25b7ea 918 if (VarCheckOk != CheckCode) {\r
919 switch (CheckCode) {\r
920 case VarCheckDuplicate:\r
921 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_DUPLICATE_COMMAND), gShellNetwork1HiiHandle, VarArg->Arg);\r
922 break;\r
68fb0527 923\r
7c25b7ea 924 case VarCheckConflict:\r
925 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_CONFLICT_COMMAND), gShellNetwork1HiiHandle, VarArg->Arg);\r
926 break;\r
68fb0527 927\r
7c25b7ea 928 case VarCheckUnknown:\r
929 //\r
930 // To handle unsupported option.\r
931 //\r
932 TempString = PermanentString;\r
933 if (StringNoCaseCompare(&VarArg->Arg, &TempString) == 0) {\r
934 ShellPrintHiiEx(-1, -1, NULL,STRING_TOKEN (STR_IFCONFIG_UNSUPPORTED_OPTION), gShellNetwork1HiiHandle, PermanentString);\r
935 goto ON_EXIT;\r
936 }\r
68fb0527 937\r
7c25b7ea 938 //\r
939 // To handle unknown option.\r
940 //\r
941 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_UNKNOWN_COMMAND), gShellNetwork1HiiHandle, VarArg->Arg);\r
942 break;\r
68fb0527 943\r
7c25b7ea 944 default:\r
945 break;\r
946 }\r
68fb0527 947\r
7c25b7ea 948 VarArg = VarArg->Next;\r
949 continue; \r
950 }\r
68fb0527 951\r
68fb0527 952 //\r
7c25b7ea 953 // Process valid variables.\r
68fb0527 954 //\r
7c25b7ea 955 if (StrCmp(VarArg->Arg, L"dhcp") == 0) {\r
d6cf1af9
JW
956 if (IfCb->Policy == Ip4Config2PolicyDhcp) {\r
957 Status = IfConfigStartIp4 (IfCb->NicHandle, gImageHandle);\r
958 if (EFI_ERROR(Status)) {\r
959 goto ON_EXIT;\r
960 }\r
961 } else {\r
962 //\r
963 // Set dhcp config policy\r
964 //\r
965 Policy = Ip4Config2PolicyDhcp;\r
966 Status = IfCb->IfCfg->SetData (\r
967 IfCb->IfCfg,\r
968 Ip4Config2DataTypePolicy,\r
969 sizeof (EFI_IP4_CONFIG2_POLICY),\r
970 &Policy\r
971 );\r
972 if (EFI_ERROR(Status)) {\r
973 goto ON_EXIT;\r
974 }\r
7c25b7ea 975 }\r
d6cf1af9 976 \r
7c25b7ea 977 VarArg= VarArg->Next; \r
68fb0527 978\r
7c25b7ea 979 } else if (StrCmp (VarArg->Arg, L"static") == 0) {\r
980 //\r
981 // Set manual config policy.\r
982 //\r
983 Policy = Ip4Config2PolicyStatic;\r
984 Status = IfCb->IfCfg->SetData (\r
985 IfCb->IfCfg,\r
986 Ip4Config2DataTypePolicy,\r
987 sizeof (EFI_IP4_CONFIG2_POLICY),\r
988 &Policy\r
989 );\r
990\r
991 if (EFI_ERROR(Status)) {\r
68fb0527 992 goto ON_EXIT;\r
993 }\r
994\r
7c25b7ea 995 VarArg= VarArg->Next; \r
68fb0527 996\r
7c25b7ea 997 ZeroMem (&ManualAddress, sizeof (ManualAddress));\r
998 \r
999 //\r
1000 // Get manual IP address.\r
1001 //\r
1002 Status = NetLibStrToIp4 (VarArg->Arg, &ManualAddress.Address);\r
1003 if (EFI_ERROR(Status)) {\r
1004 goto ON_EXIT;\r
1005 }\r
68fb0527 1006\r
7c25b7ea 1007 //\r
1008 // Get subnetmask.\r
1009 // \r
1010 VarArg = VarArg->Next;\r
1011 Status = NetLibStrToIp4 (VarArg->Arg, &ManualAddress.SubnetMask);\r
1012 if (EFI_ERROR(Status)) {\r
1013 goto ON_EXIT;\r
1014 }\r
68fb0527 1015\r
7c25b7ea 1016 //\r
1017 // Get gateway.\r
1018 //\r
1019 VarArg = VarArg->Next;\r
1020 Status = NetLibStrToIp4 (VarArg->Arg, &Gateway);\r
1021 if (EFI_ERROR(Status)) {\r
1022 goto ON_EXIT;\r
1023 }\r
1024 \r
1025 IsAddressOk = FALSE;\r
1026\r
1027 Status = IfCb->IfCfg->RegisterDataNotify (\r
1028 IfCb->IfCfg,\r
1029 Ip4Config2DataTypeManualAddress,\r
1030 MappedEvt\r
1031 );\r
1032 if (EFI_ERROR (Status)) {\r
1033 goto ON_EXIT;\r
1034 }\r
68fb0527 1035\r
7c25b7ea 1036 DataSize = sizeof (EFI_IP4_CONFIG2_MANUAL_ADDRESS);\r
68fb0527 1037\r
7c25b7ea 1038 Status = IfCb->IfCfg->SetData (\r
1039 IfCb->IfCfg,\r
1040 Ip4Config2DataTypeManualAddress,\r
1041 DataSize,\r
1042 &ManualAddress\r
1043 );\r
68fb0527 1044\r
7c25b7ea 1045 if (Status == EFI_NOT_READY) {\r
1046 gBS->SetTimer (TimeOutEvt, TimerRelative, 50000000);\r
68fb0527 1047\r
7c25b7ea 1048 while (EFI_ERROR (gBS->CheckEvent (TimeOutEvt))) {\r
1049 if (IsAddressOk) {\r
1050 Status = EFI_SUCCESS;\r
1051 break;\r
1052 }\r
1053 }\r
1054 }\r
68fb0527 1055\r
7c25b7ea 1056 IfCb->IfCfg->UnregisterDataNotify (\r
1057 IfCb->IfCfg,\r
1058 Ip4Config2DataTypeManualAddress,\r
1059 MappedEvt\r
1060 );\r
68fb0527 1061\r
7c25b7ea 1062 if (EFI_ERROR (Status)) {\r
1063 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_SET_ADDR_FAILED), gShellNetwork1HiiHandle, Status);\r
68fb0527 1064 goto ON_EXIT;\r
1065 }\r
1066\r
7c25b7ea 1067 //\r
1068 // Set gateway.\r
1069 //\r
1070 DataSize = sizeof (EFI_IPv4_ADDRESS);\r
1071\r
1072 Status = IfCb->IfCfg->SetData (\r
1073 IfCb->IfCfg,\r
1074 Ip4Config2DataTypeGateway,\r
1075 DataSize,\r
1076 &Gateway\r
1077 );\r
1078 VarArg = VarArg->Next;\r
1079 \r
1080 } else if (StrCmp (VarArg->Arg, L"dns") == 0) {\r
1081 //\r
1082 // Get DNS addresses.\r
1083 //\r
1084 VarArg = VarArg->Next;\r
1085 Tmp = VarArg;\r
1086 Index = 0;\r
1087 while (Tmp != NULL) {\r
1088 Index ++;\r
1089 Tmp = Tmp->Next;\r
1090 }\r
68fb0527 1091\r
7c25b7ea 1092 Dns = AllocatePool (Index * sizeof (EFI_IPv4_ADDRESS));\r
1093 ASSERT(Dns != NULL);\r
1094 Tmp = VarArg;\r
1095 Index = 0;\r
1096 while (Tmp != NULL) {\r
1097 Status = NetLibStrToIp4 (Tmp->Arg, Dns + Index);\r
1098 if (EFI_ERROR(Status)) {\r
1099 goto ON_EXIT;\r
1100 }\r
1101 Index ++;\r
1102 Tmp = Tmp->Next;\r
1103 }\r
1104 \r
1105 VarArg = Tmp;\r
68fb0527 1106\r
7c25b7ea 1107 //\r
1108 // Set DNS addresses.\r
1109 //\r
1110 DataSize = Index * sizeof (EFI_IPv4_ADDRESS);\r
1111\r
1112 Status = IfCb->IfCfg->SetData (\r
1113 IfCb->IfCfg,\r
1114 Ip4Config2DataTypeDnsServer,\r
1115 DataSize,\r
1116 Dns\r
1117 );\r
68fb0527 1118 }\r
7c25b7ea 1119 }\r
68fb0527 1120\r
7c25b7ea 1121ON_EXIT:\r
1122 if (Dns != NULL) {\r
1123 FreePool (Dns);\r
1124 }\r
1125 \r
d6cf1af9 1126 return Status;\r
68fb0527 1127\r
7c25b7ea 1128}\r
68fb0527 1129\r
7c25b7ea 1130/**\r
1131 The ifconfig command main process.\r
68fb0527 1132\r
7c25b7ea 1133 @param[in] Private The pointer of IFCONFIG_PRIVATE_DATA.\r
68fb0527 1134\r
7c25b7ea 1135 @retval EFI_SUCCESS ifconfig command processed successfully.\r
1136 @retval others The ifconfig command process failed.\r
68fb0527 1137\r
7c25b7ea 1138**/\r
1139EFI_STATUS\r
1140IfConfig (\r
1141 IN IFCONFIG_PRIVATE_DATA *Private\r
1142 )\r
1143{\r
1144 EFI_STATUS Status;\r
68fb0527 1145\r
1146 //\r
7c25b7ea 1147 // Get configure information of all interfaces.\r
68fb0527 1148 //\r
7c25b7ea 1149 Status = IfConfigGetInterfaceInfo (\r
1150 Private->IfName,\r
1151 &Private->IfList\r
1152 );\r
68fb0527 1153\r
7c25b7ea 1154 if (EFI_ERROR (Status)) {\r
1155 goto ON_EXIT;\r
68fb0527 1156 }\r
1157\r
7c25b7ea 1158 switch (Private->OpCode) {\r
1159 case IfConfigOpList:\r
1160 Status = IfConfigShowInterfaceInfo (&Private->IfList);\r
1161 break;\r
1162\r
1163 case IfConfigOpClear:\r
1164 Status = IfConfigClearInterfaceInfo (&Private->IfList);\r
1165 break;\r
1166\r
1167 case IfConfigOpSet:\r
1168 Status = IfConfigSetInterfaceInfo (&Private->IfList, Private->VarArg);\r
1169 break;\r
1170\r
1171 default:\r
1172 Status = EFI_ABORTED;\r
68fb0527 1173 }\r
7c25b7ea 1174\r
68fb0527 1175ON_EXIT:\r
7c25b7ea 1176\r
1177 return Status;\r
68fb0527 1178}\r
1179\r
1180/**\r
7c25b7ea 1181 The ifconfig command cleanup process, free the allocated memory.\r
1182\r
1183 @param[in] Private The pointer of IFCONFIG_PRIVATE_DATA.\r
68fb0527 1184\r
68fb0527 1185**/\r
1186VOID\r
7c25b7ea 1187IfConfigCleanup (\r
1188 IN IFCONFIG_PRIVATE_DATA *Private\r
68fb0527 1189 )\r
1190{\r
1191 LIST_ENTRY *Entry;\r
1192 LIST_ENTRY *NextEntry;\r
7c25b7ea 1193 IFCONFIG_INTERFACE_CB *IfCb;\r
1194 ARG_LIST *ArgNode;\r
1195 ARG_LIST *ArgHead;\r
68fb0527 1196\r
7c25b7ea 1197 ASSERT (Private != NULL);\r
68fb0527 1198\r
7c25b7ea 1199 //\r
1200 // Clean the list which save the set config Args.\r
1201 //\r
1202 if (Private->VarArg != NULL) {\r
1203 ArgHead = Private->VarArg;\r
68fb0527 1204\r
7c25b7ea 1205 while (ArgHead->Next != NULL) {\r
1206 ArgNode = ArgHead->Next;\r
1207 FreePool (ArgHead);\r
1208 ArgHead = ArgNode;\r
68fb0527 1209 }\r
1210\r
7c25b7ea 1211 FreePool (ArgHead);\r
68fb0527 1212 }\r
1213\r
7c25b7ea 1214 if (Private->IfName != NULL) {\r
1215 FreePool (Private->IfName);\r
1216 }\r
68fb0527 1217\r
7c25b7ea 1218 //\r
1219 // Clean the IFCONFIG_INTERFACE_CB list.\r
1220 //\r
1221 NET_LIST_FOR_EACH_SAFE (Entry, NextEntry, &Private->IfList) {\r
1222 IfCb = NET_LIST_USER_STRUCT (Entry, IFCONFIG_INTERFACE_CB, Link);\r
68fb0527 1223\r
7c25b7ea 1224 RemoveEntryList (&IfCb->Link);\r
68fb0527 1225\r
7c25b7ea 1226 if (IfCb->IfInfo != NULL) {\r
68fb0527 1227\r
7c25b7ea 1228 FreePool (IfCb->IfInfo);\r
68fb0527 1229 }\r
1230\r
7c25b7ea 1231 FreePool (IfCb);\r
68fb0527 1232 }\r
1233\r
7c25b7ea 1234 FreePool (Private);\r
68fb0527 1235}\r
1236\r
1237/**\r
1238 Function for 'ifconfig' command.\r
1239\r
1240 @param[in] ImageHandle Handle to the Image (NULL if Internal).\r
1241 @param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
7c25b7ea 1242\r
1243 @retval EFI_SUCCESS ifconfig command processed successfully.\r
1244 @retval others The ifconfig command process failed.\r
1245 \r
68fb0527 1246**/\r
1247SHELL_STATUS\r
1248EFIAPI\r
1249ShellCommandRunIfconfig (\r
1250 IN EFI_HANDLE ImageHandle,\r
1251 IN EFI_SYSTEM_TABLE *SystemTable\r
1252 )\r
1253{\r
7c25b7ea 1254 EFI_STATUS Status;\r
1255 IFCONFIG_PRIVATE_DATA *Private;\r
1256 LIST_ENTRY *ParamPackage;\r
1257 CONST CHAR16 *ValueStr;\r
1258 ARG_LIST *ArgList;\r
1259 CHAR16 *ProblemParam;\r
1260 CHAR16 *Str;\r
1261\r
1262 Private = NULL;\r
1263\r
1264 Status = ShellCommandLineParseEx (mIfConfigCheckList, &ParamPackage, &ProblemParam, TRUE, FALSE);\r
1265 if (EFI_ERROR (Status)) {\r
1266 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellNetwork1HiiHandle, L"ifconfig", ProblemParam);\r
1267 goto ON_EXIT;\r
1268 }\r
68fb0527 1269\r
1270 //\r
7c25b7ea 1271 // To handle unsupported option.\r
68fb0527 1272 //\r
7c25b7ea 1273 if (ShellCommandLineGetFlag (ParamPackage, L"-c")) {\r
1274 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_UNSUPPORTED_OPTION), gShellNetwork1HiiHandle,L"-c");\r
1275 goto ON_EXIT;\r
1276 }\r
68fb0527 1277\r
1278 //\r
7c25b7ea 1279 // To handle no option.\r
68fb0527 1280 //\r
7c25b7ea 1281 if (!ShellCommandLineGetFlag (ParamPackage, L"-r") && !ShellCommandLineGetFlag (ParamPackage, L"-s") &&\r
1282 !ShellCommandLineGetFlag (ParamPackage, L"-l")) {\r
1283 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_LACK_OPTION), gShellNetwork1HiiHandle);\r
1284 goto ON_EXIT;\r
68fb0527 1285 }\r
1286\r
7c25b7ea 1287 //\r
1288 // To handle conflict options.\r
1289 //\r
1290 if (((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-s"))) ||\r
1291 ((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-l"))) ||\r
1292 ((ShellCommandLineGetFlag (ParamPackage, L"-s")) && (ShellCommandLineGetFlag (ParamPackage, L"-l")))) {\r
1293 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CON), gShellNetwork1HiiHandle, L"ifconfig");\r
1294 goto ON_EXIT;\r
68fb0527 1295 }\r
68fb0527 1296\r
7c25b7ea 1297 Status = EFI_INVALID_PARAMETER;\r
1298\r
1299 Private = AllocateZeroPool (sizeof (IFCONFIG_PRIVATE_DATA));\r
1300\r
1301 if (Private == NULL) {\r
1302 Status = EFI_OUT_OF_RESOURCES;\r
1303 goto ON_EXIT;\r
68fb0527 1304 }\r
1305\r
7c25b7ea 1306 InitializeListHead (&Private->IfList);\r
68fb0527 1307\r
7c25b7ea 1308 //\r
1309 // To get interface name for the list option.\r
1310 //\r
1311 if (ShellCommandLineGetFlag (ParamPackage, L"-l")) {\r
1312 Private->OpCode = IfConfigOpList;\r
1313 ValueStr = ShellCommandLineGetValue (ParamPackage, L"-l");\r
1314 if (ValueStr != NULL) {\r
1315 Str = AllocateCopyPool (StrSize (ValueStr), ValueStr);\r
1316 ASSERT (Str != NULL);\r
1317 Private->IfName = Str;\r
1318 }\r
1319 }\r
1320 \r
1321 //\r
1322 // To get interface name for the clear option.\r
1323 //\r
1324 if (ShellCommandLineGetFlag (ParamPackage, L"-r")) {\r
1325 Private->OpCode = IfConfigOpClear;\r
1326 ValueStr = ShellCommandLineGetValue (ParamPackage, L"-r");\r
1327 if (ValueStr != NULL) {\r
1328 Str = AllocateCopyPool (StrSize (ValueStr), ValueStr);\r
1329 ASSERT (Str != NULL);\r
1330 Private->IfName = Str;\r
1331 }\r
1332 }\r
1333 \r
1334 //\r
1335 // To get interface name and corresponding Args for the set option.\r
1336 //\r
1337 if (ShellCommandLineGetFlag (ParamPackage, L"-s")) {\r
1338 ValueStr = ShellCommandLineGetValue (ParamPackage, L"-s");\r
1339 if (ValueStr == NULL) {\r
1340 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_LACK_INTERFACE), gShellNetwork1HiiHandle);\r
1341 goto ON_EXIT;\r
1342 }\r
68fb0527 1343 \r
1344 //\r
7c25b7ea 1345 // To split the configuration into multi-section.\r
68fb0527 1346 //\r
7c25b7ea 1347 ArgList = SplitStrToList (ValueStr, L' ');\r
1348 ASSERT (ArgList != NULL);\r
68fb0527 1349\r
7c25b7ea 1350 Private->OpCode = IfConfigOpSet;\r
1351 Private->IfName = ArgList->Arg;\r
68fb0527 1352\r
7c25b7ea 1353 Private->VarArg = ArgList->Next;\r
68fb0527 1354\r
7c25b7ea 1355 if (Private->IfName == NULL || Private->VarArg == NULL) {\r
1356 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_LACK_COMMAND), gShellNetwork1HiiHandle);\r
1357 goto ON_EXIT;\r
68fb0527 1358 }\r
68fb0527 1359 }\r
7c25b7ea 1360 \r
1361 //\r
1362 // Main process of ifconfig.\r
1363 //\r
1364 Status = IfConfig (Private);\r
68fb0527 1365\r
7c25b7ea 1366ON_EXIT:\r
68fb0527 1367\r
7c25b7ea 1368 ShellCommandLineFreeVarList (ParamPackage);\r
1369 \r
1370 if (Private != NULL) {\r
1371 IfConfigCleanup (Private);\r
68fb0527 1372 }\r
1373\r
7c25b7ea 1374 return Status;\r
68fb0527 1375}\r