8d464b8e616be5cbdb22c9858315186bc564fefc
[mirror_edk2.git] / NetworkPkg / Application / IfConfig6 / IfConfig6.c
1 /** @file
2 The implementation for Shell application IfConfig6.
3
4 Copyright (c) 2009 - 2015, Intel Corporation. All rights reserved.<BR>
5
6 This program and the accompanying materials
7 are licensed and made available under the terms and conditions of the BSD License
8 which accompanies this distribution. The full text of the license may be found at
9 http://opensource.org/licenses/bsd-license.php.
10
11 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
12 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
13
14 **/
15
16 #include <Library/ShellLib.h>
17 #include <Library/BaseMemoryLib.h>
18 #include <Library/BaseLib.h>
19 #include <Library/MemoryAllocationLib.h>
20 #include <Library/DebugLib.h>
21 #include <Library/UefiBootServicesTableLib.h>
22 #include <Library/HiiLib.h>
23 #include <Library/NetLib.h>
24
25 #include <Protocol/Ip6.h>
26 #include <Protocol/Ip6Config.h>
27
28 #include "IfConfig6.h"
29
30 EFI_HII_HANDLE mHiiHandle;
31
32 SHELL_PARAM_ITEM mIfConfig6CheckList[] = {
33 {
34 L"-b",
35 TypeFlag
36 },
37 {
38 L"-s",
39 TypeMaxValue
40 },
41 {
42 L"-l",
43 TypeValue
44 },
45 {
46 L"-r",
47 TypeValue
48 },
49 {
50 L"-?",
51 TypeFlag
52 },
53 {
54 NULL,
55 TypeMax
56 },
57 };
58
59 VAR_CHECK_ITEM mSetCheckList[] = {
60 {
61 L"auto",
62 0x00000001,
63 0x00000001,
64 FlagTypeSingle
65 },
66 {
67 L"man",
68 0x00000002,
69 0x00000001,
70 FlagTypeSingle
71 },
72 {
73 L"host",
74 0x00000004,
75 0x00000002,
76 FlagTypeSingle
77 },
78 {
79 L"dad",
80 0x00000008,
81 0x00000004,
82 FlagTypeSingle
83 },
84 {
85 L"gw",
86 0x00000010,
87 0x00000008,
88 FlagTypeSingle
89 },
90 {
91 L"dns",
92 0x00000020,
93 0x00000010,
94 FlagTypeSingle
95 },
96 {
97 L"id",
98 0x00000040,
99 0x00000020,
100 FlagTypeSingle
101 },
102 {
103 NULL,
104 0x0,
105 0x0,
106 FlagTypeSkipUnknown
107 },
108 };
109
110 /**
111 Split a string with specified separator and save the substring to a list.
112
113 @param[in] String The pointer of the input string.
114 @param[in] Separator The specified separator.
115
116 @return The pointer of headnode of ARG_LIST.
117
118 **/
119 ARG_LIST *
120 SplitStrToList (
121 IN CONST CHAR16 *String,
122 IN CHAR16 Separator
123 )
124 {
125 CHAR16 *Str;
126 CHAR16 *ArgStr;
127 ARG_LIST *ArgList;
128 ARG_LIST *ArgNode;
129
130 if (String == NULL || *String == L'\0') {
131 return NULL;
132 }
133
134 //
135 // Copy the CONST string to a local copy.
136 //
137 Str = AllocateCopyPool (StrSize (String), String);
138 ASSERT (Str != NULL);
139 ArgStr = Str;
140
141 //
142 // init a node for the list head.
143 //
144 ArgNode = (ARG_LIST *) AllocateZeroPool (sizeof (ARG_LIST));
145 ASSERT (ArgNode != NULL);
146 ArgList = ArgNode;
147
148 //
149 // Split the local copy and save in the list node.
150 //
151 while (*Str != L'\0') {
152 if (*Str == Separator) {
153 *Str = L'\0';
154 ArgNode->Arg = ArgStr;
155 ArgStr = Str + 1;
156 ArgNode->Next = (ARG_LIST *) AllocateZeroPool (sizeof (ARG_LIST));
157 ASSERT (ArgNode->Next != NULL);
158 ArgNode = ArgNode->Next;
159 }
160
161 Str++;
162 }
163
164 ArgNode->Arg = ArgStr;
165 ArgNode->Next = NULL;
166
167 return ArgList;
168 }
169
170 /**
171 Check the correctness of input Args with '-s' option.
172
173 @param[in] CheckList The pointer of VAR_CHECK_ITEM array.
174 @param[in] Name The pointer of input arg.
175 @param[in] Init The switch to execute the check.
176
177 @return The value of VAR_CHECK_CODE.
178
179 **/
180 VAR_CHECK_CODE
181 IfConfig6RetriveCheckListByName(
182 IN VAR_CHECK_ITEM *CheckList,
183 IN CHAR16 *Name,
184 IN BOOLEAN Init
185 )
186 {
187 STATIC UINT32 CheckDuplicate;
188 STATIC UINT32 CheckConflict;
189 VAR_CHECK_CODE RtCode;
190 UINT32 Index;
191 VAR_CHECK_ITEM Arg;
192
193 if (Init) {
194 CheckDuplicate = 0;
195 CheckConflict = 0;
196 return VarCheckOk;
197 }
198
199 RtCode = VarCheckOk;
200 Index = 0;
201 Arg = CheckList[Index];
202
203 //
204 // Check the Duplicated/Conflicted/Unknown input Args.
205 //
206 while (Arg.FlagStr != NULL) {
207 if (StrCmp (Arg.FlagStr, Name) == 0) {
208
209 if (CheckDuplicate & Arg.FlagID) {
210 RtCode = VarCheckDuplicate;
211 break;
212 }
213
214 if (CheckConflict & Arg.ConflictMask) {
215 RtCode = VarCheckConflict;
216 break;
217 }
218
219 CheckDuplicate |= Arg.FlagID;
220 CheckConflict |= Arg.ConflictMask;
221 break;
222 }
223
224 Arg = CheckList[++Index];
225 }
226
227 if (Arg.FlagStr == NULL) {
228 RtCode = VarCheckUnknown;
229 }
230
231 return RtCode;
232 }
233
234 /**
235 The notify function of create event when performing a manual config.
236
237 @param[in] Event The event this notify function registered to.
238 @param[in] Context Pointer to the context data registered to the event.
239
240 **/
241 VOID
242 EFIAPI
243 IfConfig6ManualAddressNotify (
244 IN EFI_EVENT Event,
245 IN VOID *Context
246 )
247 {
248 *((BOOLEAN *) Context) = TRUE;
249 }
250
251 /**
252 Print MAC address.
253
254 @param[in] Node The pointer of MAC address buffer.
255 @param[in] Size The size of MAC address buffer.
256
257 **/
258 VOID
259 IfConfig6PrintMacAddr (
260 IN UINT8 *Node,
261 IN UINT32 Size
262 )
263 {
264 UINTN Index;
265
266 ASSERT (Size <= MACADDRMAXSIZE);
267
268 for (Index = 0; Index < Size; Index++) {
269 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_MAC_ADDR_BODY), mHiiHandle, Node[Index]);
270 if (Index + 1 < Size) {
271 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_COLON), mHiiHandle);
272 }
273 }
274
275 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_NEWLINE), mHiiHandle);
276 }
277
278 /**
279 Print IPv6 address.
280
281 @param[in] Ip The pointer of Ip bufffer in EFI_IPv6_ADDRESS format.
282 @param[in] PrefixLen The pointer of PrefixLen that describes the size Prefix.
283
284 **/
285 VOID
286 IfConfig6PrintIpAddr (
287 IN EFI_IPv6_ADDRESS *Ip,
288 IN UINT8 *PrefixLen
289 )
290 {
291 UINTN Index;
292 BOOLEAN Short;
293
294 Short = FALSE;
295
296 for (Index = 0; Index < PREFIXMAXLEN; Index = Index + 2) {
297
298 if (!Short && (Index + 1 < PREFIXMAXLEN) && (Index % 2 == 0) && (Ip->Addr[Index] == 0) && (Ip->Addr[Index + 1] == 0)) {
299 //
300 // Deal with the case of ::.
301 //
302 if (Index == 0) {
303 //
304 // :: is at the beginning of the address.
305 //
306 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_COLON), mHiiHandle);
307 }
308 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_COLON), mHiiHandle);
309
310 while ((Ip->Addr[Index] == 0) && (Ip->Addr[Index + 1] == 0) && (Index < PREFIXMAXLEN)) {
311 Index = Index + 2;
312 if (Index > PREFIXMAXLEN - 2) {
313 break;
314 }
315 }
316
317 Short = TRUE;
318
319 if (Index == PREFIXMAXLEN) {
320 //
321 // :: is at the end of the address.
322 //
323 break;
324 }
325 }
326
327 if (Index < PREFIXMAXLEN - 1) {
328 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_IP_ADDR_BODY), mHiiHandle, Ip->Addr[Index]);
329 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_IP_ADDR_BODY), mHiiHandle, Ip->Addr[Index + 1]);
330 }
331
332 if (Index + 2 < PREFIXMAXLEN) {
333 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_COLON), mHiiHandle);
334 }
335 }
336
337 if (PrefixLen != NULL) {
338 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_PREFIX_LEN), mHiiHandle, *PrefixLen);
339 }
340 }
341
342 /**
343 Pick up host IPv6 address in string format from Args with "-s" option and convert it to EFI_IP6_CONFIG_MANUAL_ADDRESS format.
344
345 @param[in, out] Arg The pointer of the address of ARG_LIST which save Args with the "-s" option.
346 @param[out] Buf The pointer of the address of EFI_IP6_CONFIG_MANUAL_ADDRESS.
347 @param[out] BufSize The pointer of BufSize that describes the size of Buf in bytes.
348
349 @retval EFI_SUCCESS The convertion is successful.
350 @retval Others Does't find the host address, or it is an invalid IPv6 address in string format.
351
352 **/
353 EFI_STATUS
354 IfConfig6ParseManualAddressList (
355 IN OUT ARG_LIST **Arg,
356 OUT EFI_IP6_CONFIG_MANUAL_ADDRESS **Buf,
357 OUT UINTN *BufSize
358 )
359 {
360 EFI_STATUS Status;
361 EFI_IP6_CONFIG_MANUAL_ADDRESS *AddrBuf;
362 ARG_LIST *VarArg;
363 EFI_IPv6_ADDRESS Address;
364 UINT8 Prefix;
365 UINT8 AddrCnt;
366
367 Prefix = 0;
368 AddrCnt = 0;
369 *BufSize = 0;
370 *Buf = NULL;
371 VarArg = *Arg;
372 Status = EFI_SUCCESS;
373
374 //
375 // Go through the list to check the correctness of input host ip6 address.
376 //
377 while ((!EFI_ERROR (Status)) && (VarArg != NULL)) {
378
379 Status = NetLibStrToIp6andPrefix (VarArg->Arg, &Address, &Prefix);
380
381 if (EFI_ERROR (Status)) {
382 //
383 // host ip ip ... gw
384 //
385 break;
386 }
387
388 VarArg = VarArg->Next;
389 AddrCnt++;
390 }
391
392 if (AddrCnt == 0) {
393 return EFI_INVALID_PARAMETER;
394 }
395
396 AddrBuf = AllocateZeroPool (AddrCnt * sizeof (EFI_IP6_CONFIG_MANUAL_ADDRESS));
397 ASSERT (AddrBuf != NULL);
398
399 AddrCnt = 0;
400 VarArg = *Arg;
401 Status = EFI_SUCCESS;
402
403 //
404 // Go through the list to fill in the EFI_IP6_CONFIG_MANUAL_ADDRESS structure.
405 //
406 while ((!EFI_ERROR (Status)) && (VarArg != NULL)) {
407
408 Status = NetLibStrToIp6andPrefix (VarArg->Arg, &Address, &Prefix);
409
410 if (EFI_ERROR (Status)) {
411 break;
412 }
413
414 //
415 // If prefix length is not set, set it as Zero here. In the IfConfigSetInterfaceInfo()
416 // Zero prefix, length will be transfered to default prefix length.
417 //
418 if (Prefix == 0xFF) {
419 Prefix = 0;
420 }
421 AddrBuf[AddrCnt].IsAnycast = FALSE;
422 AddrBuf[AddrCnt].PrefixLength = Prefix;
423 IP6_COPY_ADDRESS (&AddrBuf[AddrCnt].Address, &Address);
424 VarArg = VarArg->Next;
425 AddrCnt++;
426 }
427
428 *Arg = VarArg;
429
430 if (EFI_ERROR (Status) && (Status != EFI_INVALID_PARAMETER)) {
431 goto ON_ERROR;
432 }
433
434 *Buf = AddrBuf;
435 *BufSize = AddrCnt * sizeof (EFI_IP6_CONFIG_MANUAL_ADDRESS);
436
437 return EFI_SUCCESS;
438
439 ON_ERROR:
440
441 FreePool (AddrBuf);
442 return Status;
443 }
444
445 /**
446 Pick up gw/dns IPv6 address in string format from Args with "-s" option and convert it to EFI_IPv6_ADDRESS format.
447
448 @param[in, out] Arg The pointer of the address of ARG_LIST that save Args with the "-s" option.
449 @param[out] Buf The pointer of the address of EFI_IPv6_ADDRESS.
450 @param[out] BufSize The pointer of BufSize that describes the size of Buf in bytes.
451
452 @retval EFI_SUCCESS The conversion is successful.
453 @retval Others Doesn't find the host address, or it is an invalid IPv6 address in string format.
454
455 **/
456 EFI_STATUS
457 IfConfig6ParseGwDnsAddressList (
458 IN OUT ARG_LIST **Arg,
459 OUT EFI_IPv6_ADDRESS **Buf,
460 OUT UINTN *BufSize
461 )
462 {
463 EFI_STATUS Status;
464 EFI_IPv6_ADDRESS *AddrBuf;
465 ARG_LIST *VarArg;
466 EFI_IPv6_ADDRESS Address;
467 UINT8 Prefix;
468 UINT8 AddrCnt;
469
470 AddrCnt = 0;
471 *BufSize = 0;
472 *Buf = NULL;
473 VarArg = *Arg;
474 Status = EFI_SUCCESS;
475
476 //
477 // Go through the list to check the correctness of input gw/dns address.
478 //
479 while ((!EFI_ERROR (Status)) && (VarArg != NULL)) {
480
481 Status = NetLibStrToIp6andPrefix (VarArg->Arg, &Address, &Prefix);
482
483 if (EFI_ERROR (Status)) {
484 //
485 // gw ip ip ... host
486 //
487 break;
488 }
489
490 VarArg = VarArg->Next;
491 AddrCnt++;
492 }
493
494 if (AddrCnt == 0) {
495 return EFI_INVALID_PARAMETER;
496 }
497
498 AddrBuf = AllocateZeroPool (AddrCnt * sizeof (EFI_IPv6_ADDRESS));
499 ASSERT (AddrBuf != NULL);
500
501 AddrCnt = 0;
502 VarArg = *Arg;
503 Status = EFI_SUCCESS;
504
505 //
506 // Go through the list to fill in the EFI_IPv6_ADDRESS structure.
507 //
508 while ((!EFI_ERROR (Status)) && (VarArg != NULL)) {
509
510 Status = NetLibStrToIp6andPrefix (VarArg->Arg, &Address, &Prefix);
511
512 if (EFI_ERROR (Status)) {
513 break;
514 }
515
516 IP6_COPY_ADDRESS (&AddrBuf[AddrCnt], &Address);
517
518 VarArg = VarArg->Next;
519 AddrCnt++;
520 }
521
522 *Arg = VarArg;
523
524 if (EFI_ERROR (Status) && (Status != EFI_INVALID_PARAMETER)) {
525 goto ON_ERROR;
526 }
527
528 *Buf = AddrBuf;
529 *BufSize = AddrCnt * sizeof (EFI_IPv6_ADDRESS);
530
531 return EFI_SUCCESS;
532
533 ON_ERROR:
534
535 FreePool (AddrBuf);
536 return Status;
537 }
538
539 /**
540 Parse InterfaceId in string format from Args with the "-s" option and convert it to EFI_IP6_CONFIG_INTERFACE_ID format.
541
542 @param[in, out] Arg The pointer of the address of ARG_LIST that saves Args with the "-s" option.
543 @param[out] IfId The pointer of EFI_IP6_CONFIG_INTERFACE_ID.
544
545 @retval EFI_SUCCESS The get status processed successfullly.
546 @retval EFI_INVALID_PARAMETER The get status process failed.
547
548 **/
549 EFI_STATUS
550 IfConfig6ParseInterfaceId (
551 IN OUT ARG_LIST **Arg,
552 OUT EFI_IP6_CONFIG_INTERFACE_ID **IfId
553 )
554 {
555 UINT8 Index;
556 UINT8 NodeVal;
557 CHAR16 *IdStr;
558
559 if (*Arg == NULL) {
560 return EFI_INVALID_PARAMETER;
561 }
562
563 Index = 0;
564 IdStr = (*Arg)->Arg;
565 ASSERT (IfId != NULL);
566 *IfId = AllocateZeroPool (sizeof (EFI_IP6_CONFIG_INTERFACE_ID));
567 ASSERT (*IfId != NULL);
568
569 while ((*IdStr != L'\0') && (Index < 8)) {
570
571 NodeVal = 0;
572 while ((*IdStr != L':') && (*IdStr != L'\0')) {
573
574 if ((*IdStr <= L'F') && (*IdStr >= L'A')) {
575 NodeVal = (UINT8)((NodeVal << 4) + *IdStr - L'A' + 10);
576 } else if ((*IdStr <= L'f') && (*IdStr >= L'a')) {
577 NodeVal = (UINT8)((NodeVal << 4) + *IdStr - L'a' + 10);
578 } else if ((*IdStr <= L'9') && (*IdStr >= L'0')) {
579 NodeVal = (UINT8)((NodeVal << 4) + *IdStr - L'0');
580 } else {
581 FreePool (*IfId);
582 return EFI_INVALID_PARAMETER;
583 }
584
585 IdStr++;
586 }
587
588 (*IfId)->Id[Index++] = NodeVal;
589
590 if (*IdStr == L':') {
591 IdStr++;
592 }
593 }
594
595 *Arg = (*Arg)->Next;
596 return EFI_SUCCESS;
597 }
598
599 /**
600 Parse dad in string format from Args with the "-s" option and convert it to UINT32 format.
601
602 @param[in, out] Arg The pointer of the address of ARG_LIST that saves Args with the "-s" option.
603 @param[out] Xmits The pointer of Xmits.
604
605 @retval EFI_SUCCESS The get status processed successfully.
606 @retval others The get status process failed.
607
608 **/
609 EFI_STATUS
610 IfConfig6ParseDadXmits (
611 IN OUT ARG_LIST **Arg,
612 OUT UINT32 *Xmits
613 )
614 {
615 CHAR16 *ValStr;
616
617 if (*Arg == NULL) {
618 return EFI_INVALID_PARAMETER;
619 }
620
621 ValStr = (*Arg)->Arg;
622 *Xmits = 0;
623
624 while (*ValStr != L'\0') {
625
626 if ((*ValStr <= L'9') && (*ValStr >= L'0')) {
627
628 *Xmits = (*Xmits * 10) + (*ValStr - L'0');
629
630 } else {
631
632 return EFI_INVALID_PARAMETER;
633 }
634
635 ValStr++;
636 }
637
638 *Arg = (*Arg)->Next;
639 return EFI_SUCCESS;
640 }
641
642 /**
643 The get current status of all handles.
644
645 @param[in] ImageHandle The handle of ImageHandle.
646 @param[in] IfName The pointer of IfName(interface name).
647 @param[in] IfList The pointer of IfList(interface list).
648
649 @retval EFI_SUCCESS The get status processed successfully.
650 @retval others The get status process failed.
651
652 **/
653 EFI_STATUS
654 IfConfig6GetInterfaceInfo (
655 IN EFI_HANDLE ImageHandle,
656 IN CHAR16 *IfName,
657 IN LIST_ENTRY *IfList
658 )
659 {
660 EFI_STATUS Status;
661 UINTN HandleIndex;
662 UINTN HandleNum;
663 EFI_HANDLE *HandleBuffer;
664 EFI_IP6_CONFIG_PROTOCOL *Ip6Cfg;
665 EFI_IP6_CONFIG_INTERFACE_INFO *IfInfo;
666 IFCONFIG6_INTERFACE_CB *IfCb;
667 UINTN DataSize;
668
669 HandleBuffer = NULL;
670 HandleNum = 0;
671
672 IfInfo = NULL;
673 IfCb = NULL;
674
675 //
676 // Locate all the handles with ip6 service binding protocol.
677 //
678 Status = gBS->LocateHandleBuffer (
679 ByProtocol,
680 &gEfiIp6ServiceBindingProtocolGuid,
681 NULL,
682 &HandleNum,
683 &HandleBuffer
684 );
685 if (EFI_ERROR (Status) || (HandleNum == 0)) {
686 return EFI_ABORTED;
687 }
688
689 //
690 // Enumerate all handles that installed with ip6 service binding protocol.
691 //
692 for (HandleIndex = 0; HandleIndex < HandleNum; HandleIndex++) {
693 IfCb = NULL;
694 IfInfo = NULL;
695 DataSize = 0;
696
697 //
698 // Ip6config protocol and ip6 service binding protocol are installed
699 // on the same handle.
700 //
701 ASSERT (HandleBuffer != NULL);
702 Status = gBS->HandleProtocol (
703 HandleBuffer[HandleIndex],
704 &gEfiIp6ConfigProtocolGuid,
705 (VOID **) &Ip6Cfg
706 );
707
708 if (EFI_ERROR (Status)) {
709 goto ON_ERROR;
710 }
711 //
712 // Get the interface information size.
713 //
714 Status = Ip6Cfg->GetData (
715 Ip6Cfg,
716 Ip6ConfigDataTypeInterfaceInfo,
717 &DataSize,
718 NULL
719 );
720
721 if (Status != EFI_BUFFER_TOO_SMALL) {
722 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
723 goto ON_ERROR;
724 }
725
726 IfInfo = AllocateZeroPool (DataSize);
727
728 if (IfInfo == NULL) {
729 Status = EFI_OUT_OF_RESOURCES;
730 goto ON_ERROR;
731 }
732 //
733 // Get the interface info.
734 //
735 Status = Ip6Cfg->GetData (
736 Ip6Cfg,
737 Ip6ConfigDataTypeInterfaceInfo,
738 &DataSize,
739 IfInfo
740 );
741
742 if (EFI_ERROR (Status)) {
743 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
744 goto ON_ERROR;
745 }
746 //
747 // Check the interface name if required.
748 //
749 if ((IfName != NULL) && (StrCmp (IfName, IfInfo->Name) != 0)) {
750 FreePool (IfInfo);
751 continue;
752 }
753
754 DataSize = 0;
755 //
756 // Get the size of dns server list.
757 //
758 Status = Ip6Cfg->GetData (
759 Ip6Cfg,
760 Ip6ConfigDataTypeDnsServer,
761 &DataSize,
762 NULL
763 );
764
765 if ((Status != EFI_BUFFER_TOO_SMALL) && (Status != EFI_NOT_FOUND)) {
766 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
767 goto ON_ERROR;
768 }
769
770 IfCb = AllocateZeroPool (sizeof (IFCONFIG6_INTERFACE_CB) + DataSize);
771
772 if (IfCb == NULL) {
773 Status = EFI_OUT_OF_RESOURCES;
774 goto ON_ERROR;
775 }
776
777 IfCb->NicHandle = HandleBuffer[HandleIndex];
778 IfCb->IfInfo = IfInfo;
779 IfCb->IfCfg = Ip6Cfg;
780 IfCb->DnsCnt = (UINT32) (DataSize / sizeof (EFI_IPv6_ADDRESS));
781
782 //
783 // Get the dns server list if has.
784 //
785 if (DataSize > 0) {
786
787 Status = Ip6Cfg->GetData (
788 Ip6Cfg,
789 Ip6ConfigDataTypeDnsServer,
790 &DataSize,
791 IfCb->DnsAddr
792 );
793
794 if (EFI_ERROR (Status)) {
795 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
796 goto ON_ERROR;
797 }
798 }
799 //
800 // Get the interface id if has.
801 //
802 DataSize = sizeof (EFI_IP6_CONFIG_INTERFACE_ID);
803 IfCb->IfId = AllocateZeroPool (DataSize);
804
805 if (IfCb->IfId == NULL) {
806 goto ON_ERROR;
807 }
808
809 Status = Ip6Cfg->GetData (
810 Ip6Cfg,
811 Ip6ConfigDataTypeAltInterfaceId,
812 &DataSize,
813 IfCb->IfId
814 );
815
816 if (EFI_ERROR (Status) && (Status != EFI_NOT_FOUND)) {
817 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
818 goto ON_ERROR;
819 }
820
821 if (Status == EFI_NOT_FOUND) {
822 FreePool (IfCb->IfId);
823 IfCb->IfId = NULL;
824 }
825 //
826 // Get the config policy.
827 //
828 DataSize = sizeof (EFI_IP6_CONFIG_POLICY);
829 Status = Ip6Cfg->GetData (
830 Ip6Cfg,
831 Ip6ConfigDataTypePolicy,
832 &DataSize,
833 &IfCb->Policy
834 );
835
836 if (EFI_ERROR (Status)) {
837 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
838 goto ON_ERROR;
839 }
840 //
841 // Get the dad transmits.
842 //
843 DataSize = sizeof (EFI_IP6_CONFIG_DUP_ADDR_DETECT_TRANSMITS);
844 Status = Ip6Cfg->GetData (
845 Ip6Cfg,
846 Ip6ConfigDataTypeDupAddrDetectTransmits,
847 &DataSize,
848 &IfCb->Xmits
849 );
850
851 if (EFI_ERROR (Status)) {
852 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
853 goto ON_ERROR;
854 }
855
856 InsertTailList (IfList, &IfCb->Link);
857
858 if ((IfName != NULL) && (StrCmp (IfName, IfInfo->Name) == 0)) {
859 //
860 // Only need the appointed interface, keep the allocated buffer.
861 //
862 IfCb = NULL;
863 IfInfo = NULL;
864 break;
865 }
866 }
867
868 if (HandleBuffer != NULL) {
869 FreePool (HandleBuffer);
870 }
871
872 return EFI_SUCCESS;
873
874 ON_ERROR:
875
876 if (IfInfo != NULL) {
877 FreePool (IfInfo);
878 }
879
880 if (IfCb != NULL) {
881 if (IfCb->IfId != NULL) {
882 FreePool (IfCb->IfId);
883 }
884
885 FreePool (IfCb);
886 }
887
888 return Status;
889 }
890
891 /**
892 The list process of the IfConfig6 application.
893
894 @param[in] IfList The pointer of IfList(interface list).
895
896 @retval EFI_SUCCESS The IfConfig6 list processed successfully.
897 @retval others The IfConfig6 list process failed.
898
899 **/
900 EFI_STATUS
901 IfConfig6ShowInterfaceInfo (
902 IN LIST_ENTRY *IfList
903 )
904 {
905 EFI_STATUS Status;
906 LIST_ENTRY *Entry;
907 IFCONFIG6_INTERFACE_CB *IfCb;
908 UINTN Index;
909
910 Entry = IfList->ForwardLink;
911 Status = EFI_SUCCESS;
912
913 if (IsListEmpty (IfList)) {
914 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_INTERFACE), mHiiHandle);
915 }
916
917 //
918 // Go through the interface list.
919 //
920 while (Entry != IfList) {
921
922 IfCb = BASE_CR (Entry, IFCONFIG6_INTERFACE_CB, Link);
923
924 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_BREAK), mHiiHandle);
925
926 //
927 // Print interface name.
928 //
929 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_IF_NAME), mHiiHandle, IfCb->IfInfo->Name);
930
931 //
932 // Print interface config policy.
933 //
934 if (IfCb->Policy == Ip6ConfigPolicyAutomatic) {
935 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_POLICY_AUTO), mHiiHandle);
936 } else {
937 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_POLICY_MAN), mHiiHandle);
938 }
939
940 //
941 // Print dad transmit.
942 //
943 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_DAD_TRANSMITS), mHiiHandle, IfCb->Xmits);
944
945 //
946 // Print interface id if has.
947 //
948 if (IfCb->IfId != NULL) {
949 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_INTERFACE_ID_HEAD), mHiiHandle);
950
951 IfConfig6PrintMacAddr (
952 IfCb->IfId->Id,
953 8
954 );
955 }
956 //
957 // Print mac address of the interface.
958 //
959 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_MAC_ADDR_HEAD), mHiiHandle);
960
961 IfConfig6PrintMacAddr (
962 IfCb->IfInfo->HwAddress.Addr,
963 IfCb->IfInfo->HwAddressSize
964 );
965
966 //
967 // Print ip addresses list of the interface.
968 //
969 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_IP_ADDR_HEAD), mHiiHandle);
970
971 for (Index = 0; Index < IfCb->IfInfo->AddressInfoCount; Index++) {
972 IfConfig6PrintIpAddr (
973 &IfCb->IfInfo->AddressInfo[Index].Address,
974 &IfCb->IfInfo->AddressInfo[Index].PrefixLength
975 );
976 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_NEWLINE), mHiiHandle);
977 }
978
979 //
980 // Print dns server addresses list of the interface if has.
981 //
982 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_DNS_ADDR_HEAD), mHiiHandle);
983
984 for (Index = 0; Index < IfCb->DnsCnt; Index++) {
985 IfConfig6PrintIpAddr (
986 &IfCb->DnsAddr[Index],
987 NULL
988 );
989 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_NEWLINE), mHiiHandle);
990 }
991
992 //
993 // Print route table of the interface if has.
994 //
995 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_ROUTE_HEAD), mHiiHandle);
996
997 for (Index = 0; Index < IfCb->IfInfo->RouteCount; Index++) {
998 IfConfig6PrintIpAddr (
999 &IfCb->IfInfo->RouteTable[Index].Destination,
1000 &IfCb->IfInfo->RouteTable[Index].PrefixLength
1001 );
1002 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_JOINT), mHiiHandle);
1003
1004 IfConfig6PrintIpAddr (
1005 &IfCb->IfInfo->RouteTable[Index].Gateway,
1006 NULL
1007 );
1008 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_NEWLINE), mHiiHandle);
1009 }
1010
1011 Entry = Entry->ForwardLink;
1012 }
1013
1014 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_INFO_BREAK), mHiiHandle);
1015
1016 return Status;
1017 }
1018
1019 /**
1020 The clean process of the IfConfig6 application.
1021
1022 @param[in] IfList The pointer of IfList(interface list).
1023
1024 @retval EFI_SUCCESS The IfConfig6 clean processed successfully.
1025 @retval others The IfConfig6 clean process failed.
1026
1027 **/
1028 EFI_STATUS
1029 IfConfig6ClearInterfaceInfo (
1030 IN LIST_ENTRY *IfList
1031 )
1032 {
1033 EFI_STATUS Status;
1034 LIST_ENTRY *Entry;
1035 IFCONFIG6_INTERFACE_CB *IfCb;
1036 EFI_IP6_CONFIG_POLICY Policy;
1037
1038 Policy = Ip6ConfigPolicyAutomatic;
1039 Entry = IfList->ForwardLink;
1040 Status = EFI_SUCCESS;
1041
1042 if (IsListEmpty (IfList)) {
1043 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_INTERFACE), mHiiHandle);
1044 }
1045
1046 //
1047 // Go through the interface list.
1048 //
1049 while (Entry != IfList) {
1050
1051 IfCb = BASE_CR (Entry, IFCONFIG6_INTERFACE_CB, Link);
1052
1053 Status = IfCb->IfCfg->SetData (
1054 IfCb->IfCfg,
1055 Ip6ConfigDataTypePolicy,
1056 sizeof (EFI_IP6_CONFIG_POLICY),
1057 &Policy
1058 );
1059
1060 if (EFI_ERROR (Status)) {
1061 break;
1062 }
1063
1064 Entry = Entry->ForwardLink;
1065 }
1066
1067 return Status;
1068 }
1069
1070 /**
1071 The set process of the IfConfig6 application.
1072
1073 @param[in] IfList The pointer of IfList(interface list).
1074 @param[in] VarArg The pointer of ARG_LIST(Args with "-s" option).
1075
1076 @retval EFI_SUCCESS The IfConfig6 set processed successfully.
1077 @retval others The IfConfig6 set process failed.
1078
1079 **/
1080 EFI_STATUS
1081 IfConfig6SetInterfaceInfo (
1082 IN LIST_ENTRY *IfList,
1083 IN ARG_LIST *VarArg
1084 )
1085 {
1086 EFI_STATUS Status;
1087 IFCONFIG6_INTERFACE_CB *IfCb;
1088 EFI_IP6_CONFIG_MANUAL_ADDRESS *CfgManAddr;
1089 EFI_IPv6_ADDRESS *CfgAddr;
1090 UINTN AddrSize;
1091 EFI_IP6_CONFIG_INTERFACE_ID *InterfaceId;
1092 UINT32 DadXmits;
1093 UINT32 CurDadXmits;
1094 UINTN CurDadXmitsLen;
1095 EFI_IP6_CONFIG_POLICY Policy;
1096
1097 VAR_CHECK_CODE CheckCode;
1098 EFI_EVENT TimeOutEvt;
1099 EFI_EVENT MappedEvt;
1100 BOOLEAN IsAddressOk;
1101
1102 UINTN DataSize;
1103 UINT32 Index;
1104 UINT32 Index2;
1105 BOOLEAN IsAddressSet;
1106 EFI_IP6_CONFIG_INTERFACE_INFO *IfInfo;
1107
1108 CfgManAddr = NULL;
1109 CfgAddr = NULL;
1110 TimeOutEvt = NULL;
1111 MappedEvt = NULL;
1112 IfInfo = NULL;
1113 InterfaceId = NULL;
1114 CurDadXmits = 0;
1115
1116 if (IsListEmpty (IfList)) {
1117 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_INTERFACE), mHiiHandle);
1118 return EFI_INVALID_PARAMETER;
1119 }
1120 //
1121 // Make sure to set only one interface each time.
1122 //
1123 IfCb = BASE_CR (IfList->ForwardLink, IFCONFIG6_INTERFACE_CB, Link);
1124 Status = EFI_SUCCESS;
1125
1126 //
1127 // Initialize check list mechanism.
1128 //
1129 CheckCode = IfConfig6RetriveCheckListByName(
1130 NULL,
1131 NULL,
1132 TRUE
1133 );
1134
1135 //
1136 // Create events & timers for asynchronous settings.
1137 //
1138 Status = gBS->CreateEvent (
1139 EVT_TIMER,
1140 TPL_CALLBACK,
1141 NULL,
1142 NULL,
1143 &TimeOutEvt
1144 );
1145 if (EFI_ERROR (Status)) {
1146 goto ON_EXIT;
1147 }
1148
1149 Status = gBS->CreateEvent (
1150 EVT_NOTIFY_SIGNAL,
1151 TPL_NOTIFY,
1152 IfConfig6ManualAddressNotify,
1153 &IsAddressOk,
1154 &MappedEvt
1155 );
1156 if (EFI_ERROR (Status)) {
1157 goto ON_EXIT;
1158 }
1159 //
1160 // Parse the setting variables.
1161 //
1162 while (VarArg != NULL) {
1163 //
1164 // Check invalid parameters (duplication & unknown & conflict).
1165 //
1166 CheckCode = IfConfig6RetriveCheckListByName(
1167 mSetCheckList,
1168 VarArg->Arg,
1169 FALSE
1170 );
1171
1172 if (VarCheckOk != CheckCode) {
1173 switch (CheckCode) {
1174 case VarCheckDuplicate:
1175 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_DUPLICATE_COMMAND), mHiiHandle, VarArg->Arg);
1176 break;
1177
1178 case VarCheckConflict:
1179 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_CONFLICT_COMMAND), mHiiHandle, VarArg->Arg);
1180 break;
1181
1182 case VarCheckUnknown:
1183 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_UNKNOWN_COMMAND), mHiiHandle, VarArg->Arg);
1184 break;
1185
1186 default:
1187 break;
1188 }
1189
1190 VarArg = VarArg->Next;
1191 continue;
1192 }
1193 //
1194 // Process valid variables.
1195 //
1196 if (StrCmp(VarArg->Arg, L"auto") == 0) {
1197 //
1198 // Set automaic config policy
1199 //
1200 Policy = Ip6ConfigPolicyAutomatic;
1201 Status = IfCb->IfCfg->SetData (
1202 IfCb->IfCfg,
1203 Ip6ConfigDataTypePolicy,
1204 sizeof (EFI_IP6_CONFIG_POLICY),
1205 &Policy
1206 );
1207
1208 if (EFI_ERROR(Status)) {
1209 goto ON_EXIT;
1210 }
1211
1212 VarArg= VarArg->Next;
1213
1214 } else if (StrCmp (VarArg->Arg, L"man") == 0) {
1215 //
1216 // Set manual config policy.
1217 //
1218 Policy = Ip6ConfigPolicyManual;
1219 Status = IfCb->IfCfg->SetData (
1220 IfCb->IfCfg,
1221 Ip6ConfigDataTypePolicy,
1222 sizeof (EFI_IP6_CONFIG_POLICY),
1223 &Policy
1224 );
1225
1226 if (EFI_ERROR(Status)) {
1227 goto ON_EXIT;
1228 }
1229
1230 VarArg= VarArg->Next;
1231
1232 } else if (StrCmp (VarArg->Arg, L"host") == 0) {
1233 //
1234 // Parse till the next tag or the end of command line.
1235 //
1236 VarArg = VarArg->Next;
1237 Status = IfConfig6ParseManualAddressList (
1238 &VarArg,
1239 &CfgManAddr,
1240 &AddrSize
1241 );
1242
1243 if (EFI_ERROR (Status)) {
1244 if (Status == EFI_INVALID_PARAMETER) {
1245 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_ARGUMENTS), mHiiHandle, L"host");
1246 continue;
1247 } else {
1248 goto ON_EXIT;
1249 }
1250 }
1251 //
1252 // Set static host ip6 address list.
1253 // This is a asynchronous process.
1254 //
1255 IsAddressOk = FALSE;
1256
1257 Status = IfCb->IfCfg->RegisterDataNotify (
1258 IfCb->IfCfg,
1259 Ip6ConfigDataTypeManualAddress,
1260 MappedEvt
1261 );
1262 if (EFI_ERROR (Status)) {
1263 goto ON_EXIT;
1264 }
1265
1266 Status = IfCb->IfCfg->SetData (
1267 IfCb->IfCfg,
1268 Ip6ConfigDataTypeManualAddress,
1269 AddrSize,
1270 CfgManAddr
1271 );
1272
1273 if (Status == EFI_NOT_READY) {
1274 //
1275 // Get current dad transmits count.
1276 //
1277 CurDadXmitsLen = sizeof (EFI_IP6_CONFIG_DUP_ADDR_DETECT_TRANSMITS);
1278 IfCb->IfCfg->GetData (
1279 IfCb->IfCfg,
1280 Ip6ConfigDataTypeDupAddrDetectTransmits,
1281 &CurDadXmitsLen,
1282 &CurDadXmits
1283 );
1284
1285 gBS->SetTimer (TimeOutEvt, TimerRelative, 50000000 + 10000000 * CurDadXmits);
1286
1287 while (EFI_ERROR (gBS->CheckEvent (TimeOutEvt))) {
1288 if (IsAddressOk) {
1289 Status = EFI_SUCCESS;
1290 break;
1291 }
1292 }
1293 }
1294
1295 IfCb->IfCfg->UnregisterDataNotify (
1296 IfCb->IfCfg,
1297 Ip6ConfigDataTypeManualAddress,
1298 MappedEvt
1299 );
1300
1301 if (EFI_ERROR (Status)) {
1302 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_MAN_HOST), mHiiHandle, Status);
1303 goto ON_EXIT;
1304 }
1305
1306 //
1307 // Check whether the address is set successfully.
1308 //
1309 DataSize = 0;
1310
1311 Status = IfCb->IfCfg->GetData (
1312 IfCb->IfCfg,
1313 Ip6ConfigDataTypeInterfaceInfo,
1314 &DataSize,
1315 NULL
1316 );
1317
1318 if (Status != EFI_BUFFER_TOO_SMALL) {
1319 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
1320 goto ON_EXIT;
1321 }
1322
1323 IfInfo = AllocateZeroPool (DataSize);
1324
1325 if (IfInfo == NULL) {
1326 Status = EFI_OUT_OF_RESOURCES;
1327 goto ON_EXIT;
1328 }
1329
1330 Status = IfCb->IfCfg->GetData (
1331 IfCb->IfCfg,
1332 Ip6ConfigDataTypeInterfaceInfo,
1333 &DataSize,
1334 IfInfo
1335 );
1336
1337 if (EFI_ERROR (Status)) {
1338 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_IP6CFG_GETDATA), mHiiHandle, Status);
1339 goto ON_EXIT;
1340 }
1341
1342 for ( Index = 0; Index < (UINTN) (AddrSize / sizeof (EFI_IP6_CONFIG_MANUAL_ADDRESS)); Index++) {
1343 IsAddressSet = FALSE;
1344 //
1345 // By default, the prefix length 0 is regarded as 64.
1346 //
1347 if (CfgManAddr[Index].PrefixLength == 0) {
1348 CfgManAddr[Index].PrefixLength = 64;
1349 }
1350
1351 for (Index2 = 0; Index2 < IfInfo->AddressInfoCount; Index2++) {
1352 if (EFI_IP6_EQUAL (&IfInfo->AddressInfo[Index2].Address, &CfgManAddr[Index].Address) &&
1353 (IfInfo->AddressInfo[Index2].PrefixLength == CfgManAddr[Index].PrefixLength)) {
1354 IsAddressSet = TRUE;
1355 break;
1356 }
1357 }
1358
1359 if (!IsAddressSet) {
1360 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_ADDRESS_FAILED), mHiiHandle);
1361 IfConfig6PrintIpAddr (
1362 &CfgManAddr[Index].Address,
1363 &CfgManAddr[Index].PrefixLength
1364 );
1365 }
1366 }
1367
1368 } else if (StrCmp (VarArg->Arg, L"gw") == 0) {
1369 //
1370 // Parse till the next tag or the end of command line.
1371 //
1372 VarArg = VarArg->Next;
1373 Status = IfConfig6ParseGwDnsAddressList (
1374 &VarArg,
1375 &CfgAddr,
1376 &AddrSize
1377 );
1378
1379 if (EFI_ERROR (Status)) {
1380 if (Status == EFI_INVALID_PARAMETER) {
1381 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_ARGUMENTS), mHiiHandle, L"gw");
1382 continue;
1383 } else {
1384 goto ON_EXIT;
1385 }
1386 }
1387 //
1388 // Set static gateway ip6 address list.
1389 //
1390 Status = IfCb->IfCfg->SetData (
1391 IfCb->IfCfg,
1392 Ip6ConfigDataTypeGateway,
1393 AddrSize,
1394 CfgAddr
1395 );
1396
1397 if (EFI_ERROR (Status)) {
1398 goto ON_EXIT;
1399 }
1400
1401 } else if (StrCmp (VarArg->Arg, L"dns") == 0) {
1402 //
1403 // Parse till the next tag or the end of command line.
1404 //
1405 VarArg = VarArg->Next;
1406 Status = IfConfig6ParseGwDnsAddressList (
1407 &VarArg,
1408 &CfgAddr,
1409 &AddrSize
1410 );
1411
1412 if (EFI_ERROR (Status)) {
1413 if (Status == EFI_INVALID_PARAMETER) {
1414 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_ARGUMENTS), mHiiHandle, L"dns");
1415 continue;
1416 } else {
1417 goto ON_EXIT;
1418 }
1419 }
1420 //
1421 // Set static dhs server ip6 address list.
1422 //
1423 Status = IfCb->IfCfg->SetData (
1424 IfCb->IfCfg,
1425 Ip6ConfigDataTypeDnsServer,
1426 AddrSize,
1427 CfgAddr
1428 );
1429
1430 if (EFI_ERROR (Status)) {
1431 goto ON_EXIT;
1432 }
1433
1434 } else if (StrCmp (VarArg->Arg, L"id") == 0) {
1435 //
1436 // Parse till the next tag or the end of command line.
1437 //
1438 VarArg = VarArg->Next;
1439 Status = IfConfig6ParseInterfaceId (&VarArg, &InterfaceId);
1440
1441 if (EFI_ERROR (Status)) {
1442 goto ON_EXIT;
1443 }
1444 //
1445 // Set alternative interface id.
1446 //
1447 Status = IfCb->IfCfg->SetData (
1448 IfCb->IfCfg,
1449 Ip6ConfigDataTypeAltInterfaceId,
1450 sizeof (EFI_IP6_CONFIG_INTERFACE_ID),
1451 InterfaceId
1452 );
1453
1454 if (EFI_ERROR (Status)) {
1455 goto ON_EXIT;
1456 }
1457
1458 } else if (StrCmp (VarArg->Arg, L"dad") == 0) {
1459 //
1460 // Parse till the next tag or the end of command line.
1461 //
1462 VarArg = VarArg->Next;
1463 Status = IfConfig6ParseDadXmits (&VarArg, &DadXmits);
1464
1465 if (EFI_ERROR (Status)) {
1466 goto ON_EXIT;
1467 }
1468 //
1469 // Set dad transmits count.
1470 //
1471 Status = IfCb->IfCfg->SetData (
1472 IfCb->IfCfg,
1473 Ip6ConfigDataTypeDupAddrDetectTransmits,
1474 sizeof (EFI_IP6_CONFIG_DUP_ADDR_DETECT_TRANSMITS),
1475 &DadXmits
1476 );
1477
1478 if (EFI_ERROR(Status)) {
1479 goto ON_EXIT;
1480 }
1481 }
1482 }
1483
1484 ON_EXIT:
1485
1486 if (CfgManAddr != NULL) {
1487 FreePool (CfgManAddr);
1488 }
1489
1490 if (CfgAddr != NULL) {
1491 FreePool (CfgAddr);
1492 }
1493
1494 if (MappedEvt != NULL) {
1495 gBS->CloseEvent (MappedEvt);
1496 }
1497
1498 if (TimeOutEvt != NULL) {
1499 gBS->CloseEvent (TimeOutEvt);
1500 }
1501
1502 if (IfInfo != NULL) {
1503 FreePool (IfInfo);
1504 }
1505
1506 return Status;
1507
1508 }
1509
1510 /**
1511 The IfConfig6 main process.
1512
1513 @param[in] Private The pointer of IFCONFIG6_PRIVATE_DATA.
1514
1515 @retval EFI_SUCCESS IfConfig6 processed successfully.
1516 @retval others The IfConfig6 process failed.
1517
1518 **/
1519 EFI_STATUS
1520 IfConfig6 (
1521 IN IFCONFIG6_PRIVATE_DATA *Private
1522 )
1523 {
1524 EFI_STATUS Status;
1525
1526 //
1527 // Get configure information of all interfaces.
1528 //
1529 Status = IfConfig6GetInterfaceInfo (
1530 Private->ImageHandle,
1531 Private->IfName,
1532 &Private->IfList
1533 );
1534
1535 if (EFI_ERROR (Status)) {
1536 goto ON_EXIT;
1537 }
1538
1539 switch (Private->OpCode) {
1540 case IfConfig6OpList:
1541 Status = IfConfig6ShowInterfaceInfo (&Private->IfList);
1542 break;
1543
1544 case IfConfig6OpClear:
1545 Status = IfConfig6ClearInterfaceInfo (&Private->IfList);
1546 break;
1547
1548 case IfConfig6OpSet:
1549 Status = IfConfig6SetInterfaceInfo (&Private->IfList, Private->VarArg);
1550 break;
1551
1552 default:
1553 Status = EFI_ABORTED;
1554 }
1555
1556 ON_EXIT:
1557
1558 return Status;
1559 }
1560
1561 /**
1562 The IfConfig6 cleanup process, free the allocated memory.
1563
1564 @param[in] Private The pointer of IFCONFIG6_PRIVATE_DATA.
1565
1566 **/
1567 VOID
1568 IfConfig6Cleanup (
1569 IN IFCONFIG6_PRIVATE_DATA *Private
1570 )
1571 {
1572 LIST_ENTRY *Entry;
1573 LIST_ENTRY *NextEntry;
1574 IFCONFIG6_INTERFACE_CB *IfCb;
1575 ARG_LIST *ArgNode;
1576 ARG_LIST *ArgHead;
1577
1578 ASSERT (Private != NULL);
1579
1580 //
1581 // Clean the list which save the set config Args.
1582 //
1583 if (Private->VarArg != NULL) {
1584 ArgHead = Private->VarArg;
1585
1586 while (ArgHead->Next != NULL) {
1587 ArgNode = ArgHead->Next;
1588 FreePool (ArgHead);
1589 ArgHead = ArgNode;
1590 }
1591
1592 FreePool (ArgHead);
1593 }
1594
1595 if (Private->IfName != NULL)
1596 FreePool (Private->IfName);
1597
1598
1599 //
1600 // Clean the IFCONFIG6_INTERFACE_CB list.
1601 //
1602 Entry = Private->IfList.ForwardLink;
1603 NextEntry = Entry->ForwardLink;
1604
1605 while (Entry != &Private->IfList) {
1606
1607 IfCb = BASE_CR (Entry, IFCONFIG6_INTERFACE_CB, Link);
1608
1609 RemoveEntryList (&IfCb->Link);
1610
1611 if (IfCb->IfId != NULL) {
1612
1613 FreePool (IfCb->IfId);
1614 }
1615
1616 if (IfCb->IfInfo != NULL) {
1617
1618 FreePool (IfCb->IfInfo);
1619 }
1620
1621 FreePool (IfCb);
1622
1623 Entry = NextEntry;
1624 NextEntry = Entry->ForwardLink;
1625 }
1626
1627 FreePool (Private);
1628 }
1629
1630 /**
1631 This is the declaration of an EFI image entry point. This entry point is
1632 the same for UEFI Applications, UEFI OS Loaders, and UEFI Drivers, including
1633 both device drivers and bus drivers.
1634
1635 The entry point for the IfConfig6 application which parses the command line input and calls the IfConfig6 process.
1636
1637 @param[in] ImageHandle The image handle of this application.
1638 @param[in] SystemTable The pointer to the EFI System Table.
1639
1640 @retval EFI_SUCCESS The operation completed successfully.
1641 @retval Others Some errors occur.
1642
1643 **/
1644 EFI_STATUS
1645 EFIAPI
1646 IfConfig6Initialize (
1647 IN EFI_HANDLE ImageHandle,
1648 IN EFI_SYSTEM_TABLE *SystemTable
1649 )
1650 {
1651 EFI_STATUS Status;
1652 IFCONFIG6_PRIVATE_DATA *Private;
1653 LIST_ENTRY *ParamPackage;
1654 CONST CHAR16 *ValueStr;
1655 ARG_LIST *ArgList;
1656 CHAR16 *ProblemParam;
1657 CHAR16 *Str;
1658
1659 Private = NULL;
1660
1661 //
1662 // Register our string package with HII and return the handle to it.
1663 //
1664 mHiiHandle = HiiAddPackages (&gEfiCallerIdGuid, ImageHandle, IfConfig6Strings, NULL);
1665 ASSERT (mHiiHandle != NULL);
1666
1667 Status = ShellCommandLineParseEx (mIfConfig6CheckList, &ParamPackage, &ProblemParam, TRUE, FALSE);
1668 if (EFI_ERROR (Status)) {
1669 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_INVALID_COMMAND), mHiiHandle, ProblemParam);
1670 goto ON_EXIT;
1671 }
1672
1673 //
1674 // To handle no option.
1675 //
1676 if (!ShellCommandLineGetFlag (ParamPackage, L"-r") && !ShellCommandLineGetFlag (ParamPackage, L"-s") &&
1677 !ShellCommandLineGetFlag (ParamPackage, L"-?") && !ShellCommandLineGetFlag (ParamPackage, L"-l")) {
1678 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_LACK_OPTION), mHiiHandle);
1679 goto ON_EXIT;
1680 }
1681 //
1682 // To handle conflict options.
1683 //
1684 if (((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-s"))) ||
1685 ((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-l"))) ||
1686 ((ShellCommandLineGetFlag (ParamPackage, L"-r")) && (ShellCommandLineGetFlag (ParamPackage, L"-?"))) ||
1687 ((ShellCommandLineGetFlag (ParamPackage, L"-s")) && (ShellCommandLineGetFlag (ParamPackage, L"-l"))) ||
1688 ((ShellCommandLineGetFlag (ParamPackage, L"-s")) && (ShellCommandLineGetFlag (ParamPackage, L"-?"))) ||
1689 ((ShellCommandLineGetFlag (ParamPackage, L"-l")) && (ShellCommandLineGetFlag (ParamPackage, L"-?")))) {
1690 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_CONFLICT_OPTIONS), mHiiHandle);
1691 goto ON_EXIT;
1692 }
1693 //
1694 // To show the help information of ifconfig6 command.
1695 //
1696 if (ShellCommandLineGetFlag (ParamPackage, L"-?")) {
1697 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_HELP), mHiiHandle);
1698 goto ON_EXIT;
1699 }
1700
1701 Status = EFI_INVALID_PARAMETER;
1702
1703 Private = AllocateZeroPool (sizeof (IFCONFIG6_PRIVATE_DATA));
1704
1705 if (Private == NULL) {
1706 Status = EFI_OUT_OF_RESOURCES;
1707 goto ON_EXIT;
1708 }
1709
1710 InitializeListHead (&Private->IfList);
1711
1712 //
1713 // To get interface name for the list option.
1714 //
1715 if (ShellCommandLineGetFlag (ParamPackage, L"-l")) {
1716 Private->OpCode = IfConfig6OpList;
1717 ValueStr = ShellCommandLineGetValue (ParamPackage, L"-l");
1718 if (ValueStr != NULL) {
1719 Str = AllocateCopyPool (StrSize (ValueStr), ValueStr);
1720 ASSERT (Str != NULL);
1721 Private->IfName = Str;
1722 }
1723 }
1724 //
1725 // To get interface name for the clear option.
1726 //
1727 if (ShellCommandLineGetFlag (ParamPackage, L"-r")) {
1728 Private->OpCode = IfConfig6OpClear;
1729 ValueStr = ShellCommandLineGetValue (ParamPackage, L"-r");
1730 if (ValueStr != NULL) {
1731 Str = AllocateCopyPool (StrSize (ValueStr), ValueStr);
1732 ASSERT (Str != NULL);
1733 Private->IfName = Str;
1734 }
1735 }
1736 //
1737 // To get interface name and corresponding Args for the set option.
1738 //
1739 if (ShellCommandLineGetFlag (ParamPackage, L"-s")) {
1740
1741 ValueStr = ShellCommandLineGetValue (ParamPackage, L"-s");
1742 if (ValueStr == NULL) {
1743 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_INTERFACE), mHiiHandle);
1744 goto ON_EXIT;
1745 }
1746 //
1747 // To split the configuration into multi-section.
1748 //
1749 ArgList = SplitStrToList (ValueStr, L' ');
1750 ASSERT (ArgList != NULL);
1751
1752 Private->OpCode = IfConfig6OpSet;
1753 Private->IfName = ArgList->Arg;
1754
1755 Private->VarArg = ArgList->Next;
1756
1757 if (Private->IfName == NULL || Private->VarArg == NULL) {
1758 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IFCONFIG6_ERR_LACK_COMMAND), mHiiHandle);
1759 goto ON_EXIT;
1760 }
1761 }
1762 //
1763 // Main process of ifconfig6.
1764 //
1765 Status = IfConfig6 (Private);
1766
1767 ON_EXIT:
1768
1769 ShellCommandLineFreeVarList (ParamPackage);
1770 HiiRemovePackages (mHiiHandle);
1771 if (Private != NULL)
1772 IfConfig6Cleanup (Private);
1773
1774 return Status;
1775 }
1776