]>
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 | |
34297cef JW |
599 | LIST_ENTRY *Entry;\r |
600 | LIST_ENTRY *Next;\r | |
601 | IFCONFIG_INTERFACE_CB *IfCb;\r | |
602 | BOOLEAN MediaPresent;\r | |
603 | EFI_IPv4_ADDRESS Gateway;\r | |
604 | UINT32 Index;\r | |
605 | \r | |
606 | MediaPresent = TRUE;\r | |
68fb0527 | 607 | \r |
7c25b7ea | 608 | if (IsListEmpty (IfList)) {\r |
609 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INVALID_INTERFACE), gShellNetwork1HiiHandle);\r | |
68fb0527 | 610 | }\r |
611 | \r | |
612 | //\r | |
7c25b7ea | 613 | // Go through the interface list.\r |
68fb0527 | 614 | //\r |
7c25b7ea | 615 | NET_LIST_FOR_EACH_SAFE (Entry, Next, IfList) {\r |
616 | IfCb = NET_LIST_USER_STRUCT (Entry, IFCONFIG_INTERFACE_CB, Link);\r | |
68fb0527 | 617 | \r |
7c25b7ea | 618 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_BREAK), gShellNetwork1HiiHandle);\r |
68fb0527 | 619 | \r |
7c25b7ea | 620 | //\r |
621 | // Print interface name.\r | |
622 | //\r | |
34297cef JW |
623 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_IF_NAME), gShellNetwork1HiiHandle, IfCb->IfInfo->Name); \r |
624 | \r | |
625 | //\r | |
626 | // Get Media State.\r | |
627 | //\r | |
628 | NetLibDetectMedia (IfCb->NicHandle, &MediaPresent);\r | |
629 | if (!MediaPresent) {\r | |
630 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_MEDIA_STATE), gShellNetwork1HiiHandle, L"Media disconnected");\r | |
631 | } else {\r | |
632 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_MEDIA_STATE), gShellNetwork1HiiHandle, L"Media present");\r | |
633 | }\r | |
68fb0527 | 634 | \r |
7c25b7ea | 635 | //\r |
636 | // Print interface config policy.\r | |
637 | //\r | |
638 | if (IfCb->Policy == Ip4Config2PolicyDhcp) {\r | |
639 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_POLICY_DHCP), gShellNetwork1HiiHandle);\r | |
640 | } else {\r | |
641 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_POLICY_MAN), gShellNetwork1HiiHandle);\r | |
642 | }\r | |
68fb0527 | 643 | \r |
7c25b7ea | 644 | //\r |
645 | // Print mac address of the interface.\r | |
646 | //\r | |
647 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_MAC_ADDR_HEAD), gShellNetwork1HiiHandle);\r | |
68fb0527 | 648 | \r |
7c25b7ea | 649 | IfConfigPrintMacAddr (\r |
650 | IfCb->IfInfo->HwAddress.Addr,\r | |
651 | IfCb->IfInfo->HwAddressSize\r | |
652 | );\r | |
68fb0527 | 653 | \r |
7c25b7ea | 654 | //\r |
655 | // Print IPv4 address list of the interface.\r | |
656 | //\r | |
657 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_IP_ADDR_HEAD), gShellNetwork1HiiHandle);\r | |
68fb0527 | 658 | \r |
7c25b7ea | 659 | ShellPrintHiiEx(\r |
660 | -1, \r | |
661 | -1, \r | |
662 | NULL,\r | |
663 | STRING_TOKEN (STR_IFCONFIG_INFO_IP_ADDR_BODY), \r | |
664 | gShellNetwork1HiiHandle,\r | |
665 | (UINTN)IfCb->IfInfo->StationAddress.Addr[0],\r | |
666 | (UINTN)IfCb->IfInfo->StationAddress.Addr[1],\r | |
667 | (UINTN)IfCb->IfInfo->StationAddress.Addr[2],\r | |
668 | (UINTN)IfCb->IfInfo->StationAddress.Addr[3]\r | |
669 | );\r | |
670 | \r | |
671 | //\r | |
672 | // Print subnet mask list of the interface.\r | |
673 | //\r | |
674 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_SUBNET_MASK_HEAD), gShellNetwork1HiiHandle);\r | |
68fb0527 | 675 | \r |
68fb0527 | 676 | ShellPrintHiiEx(\r |
677 | -1, \r | |
678 | -1, \r | |
679 | NULL,\r | |
7c25b7ea | 680 | STRING_TOKEN (STR_IFCONFIG_INFO_IP_ADDR_BODY), \r |
681 | gShellNetwork1HiiHandle,\r | |
682 | (UINTN)IfCb->IfInfo->SubnetMask.Addr[0],\r | |
683 | (UINTN)IfCb->IfInfo->SubnetMask.Addr[1],\r | |
684 | (UINTN)IfCb->IfInfo->SubnetMask.Addr[2],\r | |
685 | (UINTN)IfCb->IfInfo->SubnetMask.Addr[3]\r | |
68fb0527 | 686 | );\r |
68fb0527 | 687 | \r |
7c25b7ea | 688 | //\r |
689 | // Print default gateway of the interface.\r | |
690 | //\r | |
691 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_GATEWAY_HEAD), gShellNetwork1HiiHandle);\r | |
692 | \r | |
693 | ZeroMem (&Gateway, sizeof (EFI_IPv4_ADDRESS));\r | |
694 | \r | |
695 | for (Index = 0; Index < IfCb->IfInfo->RouteTableSize; Index++) {\r | |
696 | if ((CompareMem (&IfCb->IfInfo->RouteTable[Index].SubnetAddress, &mZeroIp4Addr, sizeof (EFI_IPv4_ADDRESS)) == 0) &&\r | |
697 | (CompareMem (&IfCb->IfInfo->RouteTable[Index].SubnetMask , &mZeroIp4Addr, sizeof (EFI_IPv4_ADDRESS)) == 0) ){\r | |
698 | CopyMem (&Gateway, &IfCb->IfInfo->RouteTable[Index].GatewayAddress, sizeof (EFI_IPv4_ADDRESS));\r | |
699 | }\r | |
700 | } \r | |
701 | \r | |
68fb0527 | 702 | ShellPrintHiiEx(\r |
7c25b7ea | 703 | -1, \r |
704 | -1, \r | |
705 | NULL,\r | |
706 | STRING_TOKEN (STR_IFCONFIG_INFO_IP_ADDR_BODY), \r | |
707 | gShellNetwork1HiiHandle,\r | |
708 | (UINTN)Gateway.Addr[0],\r | |
709 | (UINTN)Gateway.Addr[1],\r | |
710 | (UINTN)Gateway.Addr[2],\r | |
711 | (UINTN)Gateway.Addr[3]\r | |
712 | );\r | |
713 | \r | |
714 | //\r | |
715 | // Print route table entry.\r | |
716 | //\r | |
717 | ShellPrintHiiEx(-1, -1, NULL,STRING_TOKEN (STR_IFCONFIG_ROUTES_SIZE), gShellNetwork1HiiHandle, IfCb->IfInfo->RouteTableSize);\r | |
718 | \r | |
719 | for (Index = 0; Index < IfCb->IfInfo->RouteTableSize; Index++) {\r | |
720 | ShellPrintHiiEx(-1, -1, NULL,STRING_TOKEN (STR_IFCONFIG_ROUTES_ENTRY_INDEX), gShellNetwork1HiiHandle, Index);\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"Subnet ",\r | |
729 | (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetAddress.Addr[0],\r | |
730 | (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetAddress.Addr[1],\r | |
731 | (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetAddress.Addr[2],\r | |
732 | (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetAddress.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"Netmask",\r | |
742 | (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetMask.Addr[0],\r | |
743 | (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetMask.Addr[1],\r | |
744 | (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetMask.Addr[2],\r | |
745 | (UINTN)IfCb->IfInfo->RouteTable[Index].SubnetMask.Addr[3]\r | |
746 | );\r | |
747 | \r | |
748 | ShellPrintHiiEx(\r | |
749 | -1, \r | |
750 | -1, \r | |
751 | NULL,\r | |
752 | STRING_TOKEN (STR_IFCONFIG_SHOW_IP_ADDR), \r | |
753 | gShellNetwork1HiiHandle, \r | |
754 | L"Gateway",\r | |
755 | (UINTN)IfCb->IfInfo->RouteTable[Index].GatewayAddress.Addr[0],\r | |
756 | (UINTN)IfCb->IfInfo->RouteTable[Index].GatewayAddress.Addr[1],\r | |
757 | (UINTN)IfCb->IfInfo->RouteTable[Index].GatewayAddress.Addr[2],\r | |
758 | (UINTN)IfCb->IfInfo->RouteTable[Index].GatewayAddress.Addr[3]\r | |
759 | );\r | |
760 | }\r | |
761 | \r | |
762 | //\r | |
763 | // Print dns server addresses list of the interface if has.\r | |
764 | //\r | |
765 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_DNS_ADDR_HEAD), gShellNetwork1HiiHandle);\r | |
766 | \r | |
767 | for (Index = 0; Index < IfCb->DnsCnt; Index++) {\r | |
768 | ShellPrintHiiEx(\r | |
769 | -1, \r | |
770 | -1, \r | |
771 | NULL,\r | |
772 | STRING_TOKEN (STR_IFCONFIG_INFO_DNS_ADDR_BODY), \r | |
773 | gShellNetwork1HiiHandle,\r | |
774 | (UINTN) IfCb->DnsAddr[Index].Addr[0],\r | |
775 | (UINTN) IfCb->DnsAddr[Index].Addr[1],\r | |
776 | (UINTN) IfCb->DnsAddr[Index].Addr[2],\r | |
777 | (UINTN) IfCb->DnsAddr[Index].Addr[3]\r | |
778 | );\r | |
779 | \r | |
780 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_NEWLINE), gShellNetwork1HiiHandle);\r | |
781 | }\r | |
68fb0527 | 782 | }\r |
7c25b7ea | 783 | \r |
784 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INFO_BREAK), gShellNetwork1HiiHandle);\r | |
68fb0527 | 785 | \r |
7c25b7ea | 786 | return EFI_SUCCESS;\r |
68fb0527 | 787 | }\r |
788 | \r | |
789 | /**\r | |
7c25b7ea | 790 | The clean process of the ifconfig command to clear interface info.\r |
791 | \r | |
792 | @param[in] IfList The pointer of IfList(interface list).\r | |
68fb0527 | 793 | \r |
7c25b7ea | 794 | @retval EFI_SUCCESS The ifconfig command clean processed successfully.\r |
795 | @retval others The ifconfig command clean process failed.\r | |
68fb0527 | 796 | \r |
68fb0527 | 797 | **/\r |
7c25b7ea | 798 | EFI_STATUS\r |
799 | IfConfigClearInterfaceInfo (\r | |
800 | IN LIST_ENTRY *IfList\r | |
68fb0527 | 801 | )\r |
802 | {\r | |
7c25b7ea | 803 | EFI_STATUS Status;\r |
804 | LIST_ENTRY *Entry;\r | |
805 | LIST_ENTRY *Next;\r | |
806 | IFCONFIG_INTERFACE_CB *IfCb;\r | |
807 | EFI_IP4_CONFIG2_POLICY Policy;\r | |
68fb0527 | 808 | \r |
7c25b7ea | 809 | Policy = Ip4Config2PolicyDhcp;\r |
810 | Status = EFI_SUCCESS;\r | |
68fb0527 | 811 | \r |
7c25b7ea | 812 | if (IsListEmpty (IfList)) {\r |
813 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INVALID_INTERFACE), gShellNetwork1HiiHandle);\r | |
68fb0527 | 814 | }\r |
815 | \r | |
816 | //\r | |
7c25b7ea | 817 | // Go through the interface list.\r |
68fb0527 | 818 | //\r |
7c25b7ea | 819 | NET_LIST_FOR_EACH_SAFE (Entry, Next, IfList) {\r |
820 | IfCb = NET_LIST_USER_STRUCT (Entry, IFCONFIG_INTERFACE_CB, Link);\r | |
68fb0527 | 821 | \r |
7c25b7ea | 822 | Status = IfCb->IfCfg->SetData (\r |
823 | IfCb->IfCfg,\r | |
824 | Ip4Config2DataTypePolicy,\r | |
825 | sizeof (EFI_IP4_CONFIG2_POLICY),\r | |
826 | &Policy\r | |
827 | );\r | |
68fb0527 | 828 | \r |
7c25b7ea | 829 | if (EFI_ERROR (Status)) {\r |
830 | break;\r | |
831 | }\r | |
68fb0527 | 832 | }\r |
833 | \r | |
7c25b7ea | 834 | return Status;\r |
68fb0527 | 835 | }\r |
836 | \r | |
837 | /**\r | |
7c25b7ea | 838 | The set process of the ifconfig command.\r |
68fb0527 | 839 | \r |
7c25b7ea | 840 | @param[in] IfList The pointer of IfList(interface list).\r |
841 | @param[in] VarArg The pointer of ARG_LIST(Args with "-s" option).\r | |
68fb0527 | 842 | \r |
7c25b7ea | 843 | @retval EFI_SUCCESS The ifconfig command set processed successfully.\r |
844 | @retval others The ifconfig command set process failed.\r | |
68fb0527 | 845 | \r |
68fb0527 | 846 | **/\r |
847 | EFI_STATUS\r | |
7c25b7ea | 848 | IfConfigSetInterfaceInfo (\r |
849 | IN LIST_ENTRY *IfList,\r | |
850 | IN ARG_LIST *VarArg\r | |
68fb0527 | 851 | )\r |
852 | {\r | |
68fb0527 | 853 | \r |
7c25b7ea | 854 | EFI_STATUS Status;\r |
855 | IFCONFIG_INTERFACE_CB *IfCb;\r | |
856 | VAR_CHECK_CODE CheckCode;\r | |
857 | EFI_EVENT TimeOutEvt;\r | |
858 | EFI_EVENT MappedEvt;\r | |
859 | BOOLEAN IsAddressOk;\r | |
68fb0527 | 860 | \r |
7c25b7ea | 861 | EFI_IP4_CONFIG2_POLICY Policy;\r |
862 | EFI_IP4_CONFIG2_MANUAL_ADDRESS ManualAddress;\r | |
863 | UINTN DataSize;\r | |
864 | EFI_IPv4_ADDRESS Gateway;\r | |
865 | EFI_IPv4_ADDRESS *Dns;\r | |
866 | ARG_LIST *Tmp;\r | |
867 | UINTN Index;\r | |
68fb0527 | 868 | \r |
7c25b7ea | 869 | CONST CHAR16* TempString;\r |
68fb0527 | 870 | \r |
7c25b7ea | 871 | Dns = NULL;\r |
68fb0527 | 872 | \r |
7c25b7ea | 873 | if (IsListEmpty (IfList)) {\r |
874 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_INVALID_INTERFACE), gShellNetwork1HiiHandle);\r | |
875 | return EFI_INVALID_PARAMETER;\r | |
68fb0527 | 876 | }\r |
68fb0527 | 877 | \r |
7c25b7ea | 878 | //\r |
879 | // Make sure to set only one interface each time.\r | |
880 | //\r | |
881 | IfCb = NET_LIST_USER_STRUCT (IfList->ForwardLink, IFCONFIG_INTERFACE_CB, Link);\r | |
882 | Status = EFI_SUCCESS;\r | |
68fb0527 | 883 | \r |
7c25b7ea | 884 | //\r |
885 | // Initialize check list mechanism.\r | |
886 | //\r | |
887 | CheckCode = IfConfigRetriveCheckListByName(\r | |
888 | NULL,\r | |
889 | NULL,\r | |
890 | TRUE\r | |
891 | );\r | |
68fb0527 | 892 | \r |
7c25b7ea | 893 | //\r |
894 | // Create events & timers for asynchronous settings.\r | |
895 | //\r | |
896 | Status = gBS->CreateEvent (\r | |
897 | EVT_TIMER,\r | |
898 | TPL_CALLBACK,\r | |
899 | NULL,\r | |
900 | NULL,\r | |
901 | &TimeOutEvt\r | |
902 | );\r | |
68fb0527 | 903 | if (EFI_ERROR (Status)) {\r |
7c25b7ea | 904 | goto ON_EXIT;\r |
68fb0527 | 905 | }\r |
906 | \r | |
7c25b7ea | 907 | Status = gBS->CreateEvent (\r |
908 | EVT_NOTIFY_SIGNAL,\r | |
909 | TPL_NOTIFY,\r | |
910 | IfConfigManualAddressNotify,\r | |
911 | &IsAddressOk,\r | |
912 | &MappedEvt\r | |
913 | );\r | |
914 | if (EFI_ERROR (Status)) {\r | |
915 | goto ON_EXIT;\r | |
68fb0527 | 916 | }\r |
917 | \r | |
7c25b7ea | 918 | //\r |
919 | // Parse the setting variables.\r | |
920 | //\r | |
921 | while (VarArg != NULL) {\r | |
922 | //\r | |
923 | // Check invalid parameters (duplication & unknown & conflict).\r | |
924 | //\r | |
925 | CheckCode = IfConfigRetriveCheckListByName(\r | |
926 | mSetCheckList,\r | |
927 | VarArg->Arg,\r | |
928 | FALSE\r | |
929 | );\r | |
68fb0527 | 930 | \r |
7c25b7ea | 931 | if (VarCheckOk != CheckCode) {\r |
932 | switch (CheckCode) {\r | |
933 | case VarCheckDuplicate:\r | |
934 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_DUPLICATE_COMMAND), gShellNetwork1HiiHandle, VarArg->Arg);\r | |
935 | break;\r | |
68fb0527 | 936 | \r |
7c25b7ea | 937 | case VarCheckConflict:\r |
938 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_CONFLICT_COMMAND), gShellNetwork1HiiHandle, VarArg->Arg);\r | |
939 | break;\r | |
68fb0527 | 940 | \r |
7c25b7ea | 941 | case VarCheckUnknown:\r |
942 | //\r | |
943 | // To handle unsupported option.\r | |
944 | //\r | |
945 | TempString = PermanentString;\r | |
946 | if (StringNoCaseCompare(&VarArg->Arg, &TempString) == 0) {\r | |
947 | ShellPrintHiiEx(-1, -1, NULL,STRING_TOKEN (STR_IFCONFIG_UNSUPPORTED_OPTION), gShellNetwork1HiiHandle, PermanentString);\r | |
948 | goto ON_EXIT;\r | |
949 | }\r | |
68fb0527 | 950 | \r |
7c25b7ea | 951 | //\r |
952 | // To handle unknown option.\r | |
953 | //\r | |
954 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_UNKNOWN_COMMAND), gShellNetwork1HiiHandle, VarArg->Arg);\r | |
955 | break;\r | |
68fb0527 | 956 | \r |
7c25b7ea | 957 | default:\r |
958 | break;\r | |
959 | }\r | |
68fb0527 | 960 | \r |
7c25b7ea | 961 | VarArg = VarArg->Next;\r |
962 | continue; \r | |
963 | }\r | |
68fb0527 | 964 | \r |
68fb0527 | 965 | //\r |
7c25b7ea | 966 | // Process valid variables.\r |
68fb0527 | 967 | //\r |
7c25b7ea | 968 | if (StrCmp(VarArg->Arg, L"dhcp") == 0) {\r |
d6cf1af9 JW |
969 | if (IfCb->Policy == Ip4Config2PolicyDhcp) {\r |
970 | Status = IfConfigStartIp4 (IfCb->NicHandle, gImageHandle);\r | |
971 | if (EFI_ERROR(Status)) {\r | |
972 | goto ON_EXIT;\r | |
973 | }\r | |
974 | } else {\r | |
975 | //\r | |
976 | // Set dhcp config policy\r | |
977 | //\r | |
978 | Policy = Ip4Config2PolicyDhcp;\r | |
979 | Status = IfCb->IfCfg->SetData (\r | |
980 | IfCb->IfCfg,\r | |
981 | Ip4Config2DataTypePolicy,\r | |
982 | sizeof (EFI_IP4_CONFIG2_POLICY),\r | |
983 | &Policy\r | |
984 | );\r | |
985 | if (EFI_ERROR(Status)) {\r | |
986 | goto ON_EXIT;\r | |
987 | }\r | |
7c25b7ea | 988 | }\r |
d6cf1af9 | 989 | \r |
7c25b7ea | 990 | VarArg= VarArg->Next; \r |
68fb0527 | 991 | \r |
7c25b7ea | 992 | } else if (StrCmp (VarArg->Arg, L"static") == 0) {\r |
993 | //\r | |
994 | // Set manual config policy.\r | |
995 | //\r | |
996 | Policy = Ip4Config2PolicyStatic;\r | |
997 | Status = IfCb->IfCfg->SetData (\r | |
998 | IfCb->IfCfg,\r | |
999 | Ip4Config2DataTypePolicy,\r | |
1000 | sizeof (EFI_IP4_CONFIG2_POLICY),\r | |
1001 | &Policy\r | |
1002 | );\r | |
1003 | \r | |
1004 | if (EFI_ERROR(Status)) {\r | |
68fb0527 | 1005 | goto ON_EXIT;\r |
1006 | }\r | |
1007 | \r | |
7c25b7ea | 1008 | VarArg= VarArg->Next; \r |
68fb0527 | 1009 | \r |
7c25b7ea | 1010 | ZeroMem (&ManualAddress, sizeof (ManualAddress));\r |
1011 | \r | |
1012 | //\r | |
1013 | // Get manual IP address.\r | |
1014 | //\r | |
1015 | Status = NetLibStrToIp4 (VarArg->Arg, &ManualAddress.Address);\r | |
1016 | if (EFI_ERROR(Status)) {\r | |
1017 | goto ON_EXIT;\r | |
1018 | }\r | |
68fb0527 | 1019 | \r |
7c25b7ea | 1020 | //\r |
1021 | // Get subnetmask.\r | |
1022 | // \r | |
1023 | VarArg = VarArg->Next;\r | |
1024 | Status = NetLibStrToIp4 (VarArg->Arg, &ManualAddress.SubnetMask);\r | |
1025 | if (EFI_ERROR(Status)) {\r | |
1026 | goto ON_EXIT;\r | |
1027 | }\r | |
68fb0527 | 1028 | \r |
7c25b7ea | 1029 | //\r |
1030 | // Get gateway.\r | |
1031 | //\r | |
1032 | VarArg = VarArg->Next;\r | |
1033 | Status = NetLibStrToIp4 (VarArg->Arg, &Gateway);\r | |
1034 | if (EFI_ERROR(Status)) {\r | |
1035 | goto ON_EXIT;\r | |
1036 | }\r | |
1037 | \r | |
1038 | IsAddressOk = FALSE;\r | |
1039 | \r | |
1040 | Status = IfCb->IfCfg->RegisterDataNotify (\r | |
1041 | IfCb->IfCfg,\r | |
1042 | Ip4Config2DataTypeManualAddress,\r | |
1043 | MappedEvt\r | |
1044 | );\r | |
1045 | if (EFI_ERROR (Status)) {\r | |
1046 | goto ON_EXIT;\r | |
1047 | }\r | |
68fb0527 | 1048 | \r |
7c25b7ea | 1049 | DataSize = sizeof (EFI_IP4_CONFIG2_MANUAL_ADDRESS);\r |
68fb0527 | 1050 | \r |
7c25b7ea | 1051 | Status = IfCb->IfCfg->SetData (\r |
1052 | IfCb->IfCfg,\r | |
1053 | Ip4Config2DataTypeManualAddress,\r | |
1054 | DataSize,\r | |
1055 | &ManualAddress\r | |
1056 | );\r | |
68fb0527 | 1057 | \r |
7c25b7ea | 1058 | if (Status == EFI_NOT_READY) {\r |
1059 | gBS->SetTimer (TimeOutEvt, TimerRelative, 50000000);\r | |
68fb0527 | 1060 | \r |
7c25b7ea | 1061 | while (EFI_ERROR (gBS->CheckEvent (TimeOutEvt))) {\r |
1062 | if (IsAddressOk) {\r | |
1063 | Status = EFI_SUCCESS;\r | |
1064 | break;\r | |
1065 | }\r | |
1066 | }\r | |
1067 | }\r | |
68fb0527 | 1068 | \r |
7c25b7ea | 1069 | IfCb->IfCfg->UnregisterDataNotify (\r |
1070 | IfCb->IfCfg,\r | |
1071 | Ip4Config2DataTypeManualAddress,\r | |
1072 | MappedEvt\r | |
1073 | );\r | |
68fb0527 | 1074 | \r |
7c25b7ea | 1075 | if (EFI_ERROR (Status)) {\r |
1076 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_SET_ADDR_FAILED), gShellNetwork1HiiHandle, Status);\r | |
68fb0527 | 1077 | goto ON_EXIT;\r |
1078 | }\r | |
1079 | \r | |
7c25b7ea | 1080 | //\r |
1081 | // Set gateway.\r | |
1082 | //\r | |
1083 | DataSize = sizeof (EFI_IPv4_ADDRESS);\r | |
1084 | \r | |
1085 | Status = IfCb->IfCfg->SetData (\r | |
1086 | IfCb->IfCfg,\r | |
1087 | Ip4Config2DataTypeGateway,\r | |
1088 | DataSize,\r | |
1089 | &Gateway\r | |
1090 | );\r | |
1091 | VarArg = VarArg->Next;\r | |
1092 | \r | |
1093 | } else if (StrCmp (VarArg->Arg, L"dns") == 0) {\r | |
1094 | //\r | |
1095 | // Get DNS addresses.\r | |
1096 | //\r | |
1097 | VarArg = VarArg->Next;\r | |
1098 | Tmp = VarArg;\r | |
1099 | Index = 0;\r | |
1100 | while (Tmp != NULL) {\r | |
1101 | Index ++;\r | |
1102 | Tmp = Tmp->Next;\r | |
1103 | }\r | |
68fb0527 | 1104 | \r |
7c25b7ea | 1105 | Dns = AllocatePool (Index * sizeof (EFI_IPv4_ADDRESS));\r |
1106 | ASSERT(Dns != NULL);\r | |
1107 | Tmp = VarArg;\r | |
1108 | Index = 0;\r | |
1109 | while (Tmp != NULL) {\r | |
1110 | Status = NetLibStrToIp4 (Tmp->Arg, Dns + Index);\r | |
1111 | if (EFI_ERROR(Status)) {\r | |
1112 | goto ON_EXIT;\r | |
1113 | }\r | |
1114 | Index ++;\r | |
1115 | Tmp = Tmp->Next;\r | |
1116 | }\r | |
1117 | \r | |
1118 | VarArg = Tmp;\r | |
68fb0527 | 1119 | \r |
7c25b7ea | 1120 | //\r |
1121 | // Set DNS addresses.\r | |
1122 | //\r | |
1123 | DataSize = Index * sizeof (EFI_IPv4_ADDRESS);\r | |
1124 | \r | |
1125 | Status = IfCb->IfCfg->SetData (\r | |
1126 | IfCb->IfCfg,\r | |
1127 | Ip4Config2DataTypeDnsServer,\r | |
1128 | DataSize,\r | |
1129 | Dns\r | |
1130 | );\r | |
68fb0527 | 1131 | }\r |
7c25b7ea | 1132 | }\r |
68fb0527 | 1133 | \r |
7c25b7ea | 1134 | ON_EXIT:\r |
1135 | if (Dns != NULL) {\r | |
1136 | FreePool (Dns);\r | |
1137 | }\r | |
1138 | \r | |
d6cf1af9 | 1139 | return Status;\r |
68fb0527 | 1140 | \r |
7c25b7ea | 1141 | }\r |
68fb0527 | 1142 | \r |
7c25b7ea | 1143 | /**\r |
1144 | The ifconfig command main process.\r | |
68fb0527 | 1145 | \r |
7c25b7ea | 1146 | @param[in] Private The pointer of IFCONFIG_PRIVATE_DATA.\r |
68fb0527 | 1147 | \r |
7c25b7ea | 1148 | @retval EFI_SUCCESS ifconfig command processed successfully.\r |
1149 | @retval others The ifconfig command process failed.\r | |
68fb0527 | 1150 | \r |
7c25b7ea | 1151 | **/\r |
1152 | EFI_STATUS\r | |
1153 | IfConfig (\r | |
1154 | IN IFCONFIG_PRIVATE_DATA *Private\r | |
1155 | )\r | |
1156 | {\r | |
1157 | EFI_STATUS Status;\r | |
68fb0527 | 1158 | \r |
1159 | //\r | |
7c25b7ea | 1160 | // Get configure information of all interfaces.\r |
68fb0527 | 1161 | //\r |
7c25b7ea | 1162 | Status = IfConfigGetInterfaceInfo (\r |
1163 | Private->IfName,\r | |
1164 | &Private->IfList\r | |
1165 | );\r | |
68fb0527 | 1166 | \r |
7c25b7ea | 1167 | if (EFI_ERROR (Status)) {\r |
1168 | goto ON_EXIT;\r | |
68fb0527 | 1169 | }\r |
1170 | \r | |
7c25b7ea | 1171 | switch (Private->OpCode) {\r |
1172 | case IfConfigOpList:\r | |
1173 | Status = IfConfigShowInterfaceInfo (&Private->IfList);\r | |
1174 | break;\r | |
1175 | \r | |
1176 | case IfConfigOpClear:\r | |
1177 | Status = IfConfigClearInterfaceInfo (&Private->IfList);\r | |
1178 | break;\r | |
1179 | \r | |
1180 | case IfConfigOpSet:\r | |
1181 | Status = IfConfigSetInterfaceInfo (&Private->IfList, Private->VarArg);\r | |
1182 | break;\r | |
1183 | \r | |
1184 | default:\r | |
1185 | Status = EFI_ABORTED;\r | |
68fb0527 | 1186 | }\r |
7c25b7ea | 1187 | \r |
68fb0527 | 1188 | ON_EXIT:\r |
7c25b7ea | 1189 | \r |
1190 | return Status;\r | |
68fb0527 | 1191 | }\r |
1192 | \r | |
1193 | /**\r | |
7c25b7ea | 1194 | The ifconfig command cleanup process, free the allocated memory.\r |
1195 | \r | |
1196 | @param[in] Private The pointer of IFCONFIG_PRIVATE_DATA.\r | |
68fb0527 | 1197 | \r |
68fb0527 | 1198 | **/\r |
1199 | VOID\r | |
7c25b7ea | 1200 | IfConfigCleanup (\r |
1201 | IN IFCONFIG_PRIVATE_DATA *Private\r | |
68fb0527 | 1202 | )\r |
1203 | {\r | |
1204 | LIST_ENTRY *Entry;\r | |
1205 | LIST_ENTRY *NextEntry;\r | |
7c25b7ea | 1206 | IFCONFIG_INTERFACE_CB *IfCb;\r |
1207 | ARG_LIST *ArgNode;\r | |
1208 | ARG_LIST *ArgHead;\r | |
68fb0527 | 1209 | \r |
7c25b7ea | 1210 | ASSERT (Private != NULL);\r |
68fb0527 | 1211 | \r |
7c25b7ea | 1212 | //\r |
1213 | // Clean the list which save the set config Args.\r | |
1214 | //\r | |
1215 | if (Private->VarArg != NULL) {\r | |
1216 | ArgHead = Private->VarArg;\r | |
68fb0527 | 1217 | \r |
7c25b7ea | 1218 | while (ArgHead->Next != NULL) {\r |
1219 | ArgNode = ArgHead->Next;\r | |
1220 | FreePool (ArgHead);\r | |
1221 | ArgHead = ArgNode;\r | |
68fb0527 | 1222 | }\r |
1223 | \r | |
7c25b7ea | 1224 | FreePool (ArgHead);\r |
68fb0527 | 1225 | }\r |
1226 | \r | |
7c25b7ea | 1227 | if (Private->IfName != NULL) {\r |
1228 | FreePool (Private->IfName);\r | |
1229 | }\r | |
68fb0527 | 1230 | \r |
7c25b7ea | 1231 | //\r |
1232 | // Clean the IFCONFIG_INTERFACE_CB list.\r | |
1233 | //\r | |
1234 | NET_LIST_FOR_EACH_SAFE (Entry, NextEntry, &Private->IfList) {\r | |
1235 | IfCb = NET_LIST_USER_STRUCT (Entry, IFCONFIG_INTERFACE_CB, Link);\r | |
68fb0527 | 1236 | \r |
7c25b7ea | 1237 | RemoveEntryList (&IfCb->Link);\r |
68fb0527 | 1238 | \r |
7c25b7ea | 1239 | if (IfCb->IfInfo != NULL) {\r |
68fb0527 | 1240 | \r |
7c25b7ea | 1241 | FreePool (IfCb->IfInfo);\r |
68fb0527 | 1242 | }\r |
1243 | \r | |
7c25b7ea | 1244 | FreePool (IfCb);\r |
68fb0527 | 1245 | }\r |
1246 | \r | |
7c25b7ea | 1247 | FreePool (Private);\r |
68fb0527 | 1248 | }\r |
1249 | \r | |
1250 | /**\r | |
1251 | Function for 'ifconfig' command.\r | |
1252 | \r | |
1253 | @param[in] ImageHandle Handle to the Image (NULL if Internal).\r | |
1254 | @param[in] SystemTable Pointer to the System Table (NULL if Internal).\r | |
7c25b7ea | 1255 | \r |
1256 | @retval EFI_SUCCESS ifconfig command processed successfully.\r | |
1257 | @retval others The ifconfig command process failed.\r | |
1258 | \r | |
68fb0527 | 1259 | **/\r |
1260 | SHELL_STATUS\r | |
1261 | EFIAPI\r | |
1262 | ShellCommandRunIfconfig (\r | |
1263 | IN EFI_HANDLE ImageHandle,\r | |
1264 | IN EFI_SYSTEM_TABLE *SystemTable\r | |
1265 | )\r | |
1266 | {\r | |
7c25b7ea | 1267 | EFI_STATUS Status;\r |
1268 | IFCONFIG_PRIVATE_DATA *Private;\r | |
1269 | LIST_ENTRY *ParamPackage;\r | |
1270 | CONST CHAR16 *ValueStr;\r | |
1271 | ARG_LIST *ArgList;\r | |
1272 | CHAR16 *ProblemParam;\r | |
1273 | CHAR16 *Str;\r | |
1274 | \r | |
1275 | Private = NULL;\r | |
1276 | \r | |
1277 | Status = ShellCommandLineParseEx (mIfConfigCheckList, &ParamPackage, &ProblemParam, TRUE, FALSE);\r | |
1278 | if (EFI_ERROR (Status)) {\r | |
1279 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellNetwork1HiiHandle, L"ifconfig", ProblemParam);\r | |
1280 | goto ON_EXIT;\r | |
1281 | }\r | |
68fb0527 | 1282 | \r |
1283 | //\r | |
7c25b7ea | 1284 | // To handle unsupported option.\r |
68fb0527 | 1285 | //\r |
7c25b7ea | 1286 | if (ShellCommandLineGetFlag (ParamPackage, L"-c")) {\r |
1287 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_UNSUPPORTED_OPTION), gShellNetwork1HiiHandle,L"-c");\r | |
1288 | goto ON_EXIT;\r | |
1289 | }\r | |
68fb0527 | 1290 | \r |
1291 | //\r | |
7c25b7ea | 1292 | // To handle no option.\r |
68fb0527 | 1293 | //\r |
7c25b7ea | 1294 | if (!ShellCommandLineGetFlag (ParamPackage, L"-r") && !ShellCommandLineGetFlag (ParamPackage, L"-s") &&\r |
1295 | !ShellCommandLineGetFlag (ParamPackage, L"-l")) {\r | |
1296 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_LACK_OPTION), gShellNetwork1HiiHandle);\r | |
1297 | goto ON_EXIT;\r | |
68fb0527 | 1298 | }\r |
1299 | \r | |
7c25b7ea | 1300 | //\r |
1301 | // To handle conflict options.\r | |
1302 | //\r | |
1303 | if (((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-s"))) ||\r | |
1304 | ((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-l"))) ||\r | |
1305 | ((ShellCommandLineGetFlag (ParamPackage, L"-s")) && (ShellCommandLineGetFlag (ParamPackage, L"-l")))) {\r | |
1306 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CON), gShellNetwork1HiiHandle, L"ifconfig");\r | |
1307 | goto ON_EXIT;\r | |
68fb0527 | 1308 | }\r |
68fb0527 | 1309 | \r |
7c25b7ea | 1310 | Status = EFI_INVALID_PARAMETER;\r |
1311 | \r | |
1312 | Private = AllocateZeroPool (sizeof (IFCONFIG_PRIVATE_DATA));\r | |
1313 | \r | |
1314 | if (Private == NULL) {\r | |
1315 | Status = EFI_OUT_OF_RESOURCES;\r | |
1316 | goto ON_EXIT;\r | |
68fb0527 | 1317 | }\r |
1318 | \r | |
7c25b7ea | 1319 | InitializeListHead (&Private->IfList);\r |
68fb0527 | 1320 | \r |
7c25b7ea | 1321 | //\r |
1322 | // To get interface name for the list option.\r | |
1323 | //\r | |
1324 | if (ShellCommandLineGetFlag (ParamPackage, L"-l")) {\r | |
1325 | Private->OpCode = IfConfigOpList;\r | |
1326 | ValueStr = ShellCommandLineGetValue (ParamPackage, L"-l");\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 for the clear option.\r | |
1336 | //\r | |
1337 | if (ShellCommandLineGetFlag (ParamPackage, L"-r")) {\r | |
1338 | Private->OpCode = IfConfigOpClear;\r | |
1339 | ValueStr = ShellCommandLineGetValue (ParamPackage, L"-r");\r | |
1340 | if (ValueStr != NULL) {\r | |
1341 | Str = AllocateCopyPool (StrSize (ValueStr), ValueStr);\r | |
1342 | ASSERT (Str != NULL);\r | |
1343 | Private->IfName = Str;\r | |
1344 | }\r | |
1345 | }\r | |
1346 | \r | |
1347 | //\r | |
1348 | // To get interface name and corresponding Args for the set option.\r | |
1349 | //\r | |
1350 | if (ShellCommandLineGetFlag (ParamPackage, L"-s")) {\r | |
1351 | ValueStr = ShellCommandLineGetValue (ParamPackage, L"-s");\r | |
1352 | if (ValueStr == NULL) {\r | |
1353 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_LACK_INTERFACE), gShellNetwork1HiiHandle);\r | |
1354 | goto ON_EXIT;\r | |
1355 | }\r | |
68fb0527 | 1356 | \r |
1357 | //\r | |
7c25b7ea | 1358 | // To split the configuration into multi-section.\r |
68fb0527 | 1359 | //\r |
7c25b7ea | 1360 | ArgList = SplitStrToList (ValueStr, L' ');\r |
1361 | ASSERT (ArgList != NULL);\r | |
68fb0527 | 1362 | \r |
7c25b7ea | 1363 | Private->OpCode = IfConfigOpSet;\r |
1364 | Private->IfName = ArgList->Arg;\r | |
68fb0527 | 1365 | \r |
7c25b7ea | 1366 | Private->VarArg = ArgList->Next;\r |
68fb0527 | 1367 | \r |
7c25b7ea | 1368 | if (Private->IfName == NULL || Private->VarArg == NULL) {\r |
1369 | ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG_LACK_COMMAND), gShellNetwork1HiiHandle);\r | |
1370 | goto ON_EXIT;\r | |
68fb0527 | 1371 | }\r |
68fb0527 | 1372 | }\r |
7c25b7ea | 1373 | \r |
1374 | //\r | |
1375 | // Main process of ifconfig.\r | |
1376 | //\r | |
1377 | Status = IfConfig (Private);\r | |
68fb0527 | 1378 | \r |
7c25b7ea | 1379 | ON_EXIT:\r |
68fb0527 | 1380 | \r |
7c25b7ea | 1381 | ShellCommandLineFreeVarList (ParamPackage);\r |
1382 | \r | |
1383 | if (Private != NULL) {\r | |
1384 | IfConfigCleanup (Private);\r | |
68fb0527 | 1385 | }\r |
1386 | \r | |
7c25b7ea | 1387 | return Status;\r |
68fb0527 | 1388 | }\r |