]>
Commit | Line | Data |
---|---|---|
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 | 19 | typedef enum {\r |
20 | IfConfigOpList = 1,\r | |
21 | IfConfigOpSet = 2,\r | |
22 | IfConfigOpClear = 3\r | |
23 | } IFCONFIG_OPCODE;\r | |
24 | \r | |
25 | typedef 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 | |
35 | typedef 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 | |
44 | typedef 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 | |
54 | typedef struct _ARG_LIST ARG_LIST;\r | |
55 | \r | |
56 | struct _ARG_LIST {\r | |
57 | ARG_LIST *Next;\r | |
58 | CHAR16 *Arg;\r | |
59 | };\r | |
60 | \r | |
61 | typedef 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 | |
69 | typedef 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 | |
76 | SHELL_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 | |
103 | VAR_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 | 130 | STATIC 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 | 141 | ARG_LIST *\r |
142 | SplitStrToList (\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 | 205 | VAR_CHECK_CODE\r |
206 | IfConfigRetriveCheckListByName(\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 |
266 | VOID\r | |
68fb0527 | 267 | EFIAPI\r |
7c25b7ea | 268 | IfConfigManualAddressNotify (\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 | |
285 | EFI_STATUS\r | |
286 | EFIAPI\r | |
287 | IfConfigStartIp4(\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 | |
344 | ON_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 | 363 | VOID\r |
364 | IfConfigPrintMacAddr (\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 |
394 | EFI_STATUS\r | |
7c25b7ea | 395 | IfConfigGetInterfaceInfo (\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 | 572 | ON_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 | 594 | EFI_STATUS\r |
595 | IfConfigShowInterfaceInfo (\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 | 785 | EFI_STATUS\r |
786 | IfConfigClearInterfaceInfo (\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 |
834 | EFI_STATUS\r | |
7c25b7ea | 835 | IfConfigSetInterfaceInfo (\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 | 1121 | ON_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 |
1139 | EFI_STATUS\r | |
1140 | IfConfig (\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 | 1175 | ON_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 |
1186 | VOID\r | |
7c25b7ea | 1187 | IfConfigCleanup (\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 |
1247 | SHELL_STATUS\r | |
1248 | EFIAPI\r | |
1249 | ShellCommandRunIfconfig (\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 | 1366 | ON_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 |