2 Main file for map shell level 2 command.
4 Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>
5 This program and the accompanying materials
6 are licensed and made available under the terms and conditions of the BSD License
7 which accompanies this distribution. The full text of the license may be found at
8 http://opensource.org/licenses/bsd-license.php
10 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
15 #include "UefiShellLevel2CommandsLib.h"
16 #include <Protocol/SimpleFileSystem.h>
17 #include <Protocol/BlockIo.h>
18 #include <Library/DevicePathLib.h>
19 #include <Library/HandleParsingLib.h>
20 #include <Library/SortLib.h>
25 IN CONST CHAR16
*List
,
26 IN CONST CHAR16
*MetaTarget
,
27 OUT CHAR16
**FullName OPTIONAL
,
28 IN CONST BOOLEAN Meta
,
29 IN CONST BOOLEAN SkipTrailingNumbers
,
30 IN CONST CHAR16
*Target
35 CONST CHAR16
*ListWalker
;
39 for (ListWalker
= List
, TempList
= NULL
40 ; ListWalker
!= NULL
&& *ListWalker
!= CHAR_NULL
43 TempList
= StrnCatGrow(&TempList
, NULL
, ListWalker
, 0);
44 ASSERT(TempList
!= NULL
);
45 TempSpot
= StrStr(TempList
, Target
);
46 if (TempSpot
!= NULL
) {
47 *TempSpot
= CHAR_NULL
;
50 while (SkipTrailingNumbers
&& (ShellIsDecimalDigitCharacter(TempList
[StrLen(TempList
)-1]) || TempList
[StrLen(TempList
)-1] == L
':')) {
51 TempList
[StrLen(TempList
)-1] = CHAR_NULL
;
54 ListWalker
= StrStr(ListWalker
, Target
);
55 while(ListWalker
!= NULL
&& *ListWalker
== *Target
) {
59 Result
= gUnicodeCollation
->MetaiMatch(gUnicodeCollation
, (CHAR16
*)TempList
, (CHAR16
*)MetaTarget
);
61 Result
= (BOOLEAN
)(StrCmp(TempList
, MetaTarget
)==0);
64 if (FullName
!= NULL
) {
85 EFI_HANDLE
*HandleList
;
87 EFI_DEVICE_PATH_PROTOCOL
**DevicePathList
;
88 CHAR16
*NewDefaultName
;
89 CHAR16
*NewConsistName
;
90 EFI_DEVICE_PATH_PROTOCOL
**ConsistMappingTable
;
96 // remove mappings that represent removed devices.
100 // Find each handle with Simple File System
102 HandleList
= GetHandleListByProtocol(&gEfiSimpleFileSystemProtocolGuid
);
103 if (HandleList
!= NULL
) {
105 // Do a count of the handles
107 for (Count
= 0 ; HandleList
[Count
] != NULL
; Count
++);
110 // Get all Device Paths
112 DevicePathList
= AllocatePool(sizeof(EFI_DEVICE_PATH_PROTOCOL
*) * Count
);
113 ASSERT(DevicePathList
!= NULL
);
115 for (Count
= 0 ; HandleList
[Count
] != NULL
; Count
++) {
116 DevicePathList
[Count
] = DevicePathFromHandle(HandleList
[Count
]);
120 // Sort all DevicePaths
122 PerformQuickSort(DevicePathList
, Count
, sizeof(EFI_DEVICE_PATH_PROTOCOL
*), DevicePathCompare
);
124 ShellCommandConsistMappingInitialize(&ConsistMappingTable
);
127 // Assign new Mappings to remainders
129 for (Count
= 0 ; HandleList
[Count
] != NULL
&& !EFI_ERROR(Status
); Count
++) {
131 // Skip ones that already have
133 if (gEfiShellProtocol
->GetMapFromDevicePath(&DevicePathList
[Count
]) != NULL
) {
139 NewDefaultName
= ShellCommandCreateNewMappingName(MappingTypeFileSystem
);
140 ASSERT(NewDefaultName
!= NULL
);
143 // Call shell protocol SetMap function now...
145 Status
= gEfiShellProtocol
->SetMap(DevicePathList
[Count
], NewDefaultName
);
147 if (!EFI_ERROR(Status
)) {
149 // Now do consistent name
151 NewConsistName
= ShellCommandConsistMappingGenMappingName(DevicePathList
[Count
], ConsistMappingTable
);
152 if (NewConsistName
!= NULL
) {
153 Status
= gEfiShellProtocol
->SetMap(DevicePathList
[Count
], NewConsistName
);
154 FreePool(NewConsistName
);
158 FreePool(NewDefaultName
);
160 ShellCommandConsistMappingUnInitialize(ConsistMappingTable
);
161 SHELL_FREE_NON_NULL(HandleList
);
162 SHELL_FREE_NON_NULL(DevicePathList
);
169 // Do it all over again for gEfiBlockIoProtocolGuid
178 IN EFI_DEVICE_PATH_PROTOCOL
*DevicePath
181 ACPI_HID_DEVICE_PATH
*Acpi
;
184 // Parse the device path:
185 // Devicepath sub type mediatype
186 // MEDIA_HANRDDRIVE_DP -> Hard Disk
187 // MEDIA_CDROM_DP -> CD Rom
188 // Acpi.HID = 0X0604 -> Floppy
190 if (NULL
== DevicePath
) {
191 return HiiGetString(gShellLevel2HiiHandle
, STRING_TOKEN(STR_MAP_MEDIA_UNKNOWN
), NULL
);
194 for (;!IsDevicePathEndType (DevicePath
) ;DevicePath
= NextDevicePathNode (DevicePath
)) {
195 if (DevicePathType (DevicePath
) == MEDIA_DEVICE_PATH
) {
196 switch (DevicePathSubType (DevicePath
)) {
197 case MEDIA_HARDDRIVE_DP
:
198 return HiiGetString(gShellLevel2HiiHandle
, STRING_TOKEN(STR_MAP_MEDIA_HARDDISK
), NULL
);
200 return HiiGetString(gShellLevel2HiiHandle
, STRING_TOKEN(STR_MAP_MEDIA_CDROM
), NULL
);
202 } else if (DevicePathType (DevicePath
) == ACPI_DEVICE_PATH
) {
203 Acpi
= (ACPI_HID_DEVICE_PATH
*) DevicePath
;
204 if (EISA_ID_TO_NUM (Acpi
->HID
) == 0x0604) {
205 return HiiGetString(gShellLevel2HiiHandle
, STRING_TOKEN(STR_MAP_MEDIA_FLOPPY
), NULL
);
210 return HiiGetString(gShellLevel2HiiHandle
, STRING_TOKEN(STR_MAP_MEDIA_UNKNOWN
), NULL
);
214 Function to detemine if a handle has removable storage.
216 @param[in] DevicePath DevicePath to test.
218 @retval TRUE The handle has removable storage.
219 @retval FALSE The handle does not have removable storage.
224 IN EFI_DEVICE_PATH_PROTOCOL
*DevicePath
227 if (NULL
== DevicePath
) {
231 while (!IsDevicePathEndType (DevicePath
)) {
232 if (DevicePathType (DevicePath
) == MESSAGING_DEVICE_PATH
) {
233 switch (DevicePathSubType (DevicePath
)) {
241 DevicePath
= NextDevicePathNode (DevicePath
);
247 Function to detemine if a something on the map list matches.
249 @param[in] MapList The pointer to the list to test.
250 @param[in] Specific The pointer to a specific name to test for.
251 @param[in] TypeString The pointer to the list of types.
252 @param[in] Normal Always show normal mappings.
253 @param[in] Consist Always show consistent mappings.
255 @retval TRUE The map should be displayed.
256 @retval FALSE The map should not be displayed.
261 IN CONST CHAR16
*MapList
,
262 IN CONST CHAR16
*Specific
,
263 IN CONST CHAR16
*TypeString
,
264 IN CONST BOOLEAN Normal
,
265 IN CONST BOOLEAN Consist
269 // specific has priority
271 if ( Specific
!= NULL
272 && SearchList(MapList
, Specific
, NULL
, TRUE
, FALSE
, L
";")) {
277 && (SearchList(MapList
, L
"HD*", NULL
, TRUE
, TRUE
, L
";")
278 ||SearchList(MapList
, L
"CD*", NULL
, TRUE
, TRUE
, L
";")
279 ||SearchList(MapList
, L
"F*", NULL
, TRUE
, TRUE
, L
";")
280 ||SearchList(MapList
, L
"FP*", NULL
, TRUE
, TRUE
, L
";"))){
285 && (SearchList(MapList
, L
"FS", NULL
, FALSE
, TRUE
, L
";")
286 ||SearchList(MapList
, L
"BLK", NULL
, FALSE
, TRUE
, L
";"))){
290 if (TypeString
!= NULL
&& SearchList(MapList
, TypeString
, NULL
, TRUE
, TRUE
, L
";")) {
299 PerformSingleMappingDisplay(
300 IN CONST BOOLEAN Verbose
,
301 IN CONST BOOLEAN Consist
,
302 IN CONST BOOLEAN Normal
,
303 IN CONST CHAR16
*TypeString
,
304 IN CONST BOOLEAN SFO
,
305 IN CONST CHAR16
*Specific OPTIONAL
,
306 IN CONST EFI_HANDLE Handle
309 EFI_DEVICE_PATH_PROTOCOL
*DevPath
;
310 EFI_DEVICE_PATH_PROTOCOL
*DevPathCopy
;
311 CONST CHAR16
*MapList
;
314 CHAR16
*DevPathString
;
318 CONST CHAR16
*TempSpot2
;
320 DevPath
= DevicePathFromHandle(Handle
);
321 DevPathCopy
= DevPath
;
322 MapList
= gEfiShellProtocol
->GetMapFromDevicePath(&DevPathCopy
);
323 if (MapList
== NULL
) {
327 if (!MappingListHasType(MapList
, Specific
, TypeString
, Normal
, Consist
)){
332 CurrentName
= StrnCatGrow(&CurrentName
, 0, MapList
, 0);
333 TempSpot
= StrStr(CurrentName
, L
";");
334 if (TempSpot
!= NULL
) {
335 *TempSpot
= CHAR_NULL
;
337 DevPathString
= gDevPathToText
->ConvertDevicePathToText(DevPath
, TRUE
, FALSE
);
339 TempLen
= StrLen(CurrentName
);
344 STRING_TOKEN (STR_MAP_ENTRY
),
345 gShellLevel2HiiHandle
,
347 TempLen
< StrLen(MapList
)?MapList
+ TempLen
+1:L
"",
352 // also print handle, media type, removable (y/n), and current directory
354 MediaType
= GetDeviceMediaType(DevPath
);
355 if ((TypeString
!= NULL
&&MediaType
!= NULL
&& StrStr(TypeString
, MediaType
) != NULL
) || TypeString
== NULL
) {
356 Removable
= IsRemoveableDevice(DevPath
);
357 TempSpot2
= ShellGetCurrentDir(CurrentName
);
362 STRING_TOKEN (STR_MAP_ENTRY_VERBOSE
),
363 gShellLevel2HiiHandle
,
364 ConvertHandleToHandleIndex(Handle
),
366 Removable
?L
"Yes":L
"No",
373 TempLen
= StrLen(CurrentName
);
378 STRING_TOKEN (STR_MAP_SFO_MAPPINGS
),
379 gShellLevel2HiiHandle
,
382 TempLen
< StrLen(MapList
)?MapList
+ TempLen
+1:L
""
385 FreePool(DevPathString
);
386 FreePool(CurrentName
);
392 PerformSingleMappingDelete(
393 IN CONST CHAR16
*Specific
,
394 IN CONST EFI_HANDLE Handle
397 EFI_DEVICE_PATH_PROTOCOL
*DevPath
;
398 EFI_DEVICE_PATH_PROTOCOL
*DevPathCopy
;
399 CONST CHAR16
*MapList
;
402 DevPath
= DevicePathFromHandle(Handle
);
403 DevPathCopy
= DevPath
;
404 MapList
= gEfiShellProtocol
->GetMapFromDevicePath(&DevPathCopy
);
407 if (MapList
== NULL
) {
408 return (EFI_NOT_FOUND
);
411 // if there is a specific and its not on the list...
413 if (!SearchList(MapList
, Specific
, &CurrentName
, TRUE
, FALSE
, L
";")) {
414 return (EFI_NOT_FOUND
);
416 return (gEfiShellProtocol
->SetMap(NULL
, CurrentName
));
420 Function to display mapping information to the user.
422 if Specific is specified then Consist and Normal will be ignored since information will
423 be printed for the specific item only.
425 @param[in] Verbose TRUE to display (extra) verbose information
426 @param[in] Consist TRUE to display consistent mappings
427 @param[in] Normal TRUE to display normal (not consist) mappings
428 @param[in] TypeString pointer to string of filter types
429 @param[in] SFO TRUE to display output in Standard Output Format
430 @param[in] Specific pointer to string for specific map to display
432 @retval SHELL_SUCCESS the display was printed
433 @retval SHELL_INVALID_PARAMETER one of Consist or Normal must be TRUE if no Specific
436 CONST CHAR16 Cd
[] = L
"cd*";
437 CONST CHAR16 Hd
[] = L
"hd*";
438 CONST CHAR16 Fp
[] = L
"fp*";
439 CONST CHAR16 F
[] = L
"F*";
442 PerformMappingDisplay(
443 IN CONST BOOLEAN Verbose
,
444 IN CONST BOOLEAN Consist
,
445 IN CONST BOOLEAN Normal
,
446 IN CONST CHAR16
*TypeString
,
447 IN CONST BOOLEAN SFO
,
448 IN CONST CHAR16
*Specific OPTIONAL
,
449 IN CONST BOOLEAN Header
453 EFI_HANDLE
*HandleBuffer
;
458 if (!Consist
&& !Normal
&& Specific
== NULL
&& TypeString
== NULL
) {
459 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_TOO_FEW
), gShellLevel2HiiHandle
);
460 return (SHELL_INVALID_PARAMETER
);
463 if (TypeString
!= NULL
) {
465 if (StrnCmp(TypeString
, Test
, StrLen(Test
)-1) != 0) {
467 if (StrnCmp(TypeString
, Test
, StrLen(Test
)-1) != 0) {
469 if (StrnCmp(TypeString
, Test
, StrLen(Test
)-1) != 0) {
470 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PROBLEM
), gShellLevel2HiiHandle
, TypeString
);
471 return (SHELL_INVALID_PARAMETER
);
473 } else if (Test
== NULL
) {
486 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_MAP_HEADER
), gShellLevel2HiiHandle
);
488 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_SFO_HEADER
), gShellLevel2HiiHandle
, L
"map");
496 // Look up all SimpleFileSystems in the platform
498 Status
= gBS
->LocateHandle(
500 &gEfiDevicePathProtocolGuid
,
504 if (Status
== EFI_BUFFER_TOO_SMALL
) {
505 HandleBuffer
= AllocatePool(BufferSize
);
506 if (HandleBuffer
== NULL
) {
507 return (SHELL_OUT_OF_RESOURCES
);
509 Status
= gBS
->LocateHandle(
511 &gEfiDevicePathProtocolGuid
,
516 ASSERT_EFI_ERROR(Status
);
517 ASSERT(HandleBuffer
!= NULL
);
520 // Get the map name(s) for each one.
523 ; LoopVar
< (BufferSize
/ sizeof(EFI_HANDLE
))
526 PerformSingleMappingDisplay(
533 HandleBuffer
[LoopVar
]);
537 // Look up all BlockIo in the platform
539 Status
= gBS
->LocateHandle(
541 &gEfiBlockIoProtocolGuid
,
545 if (Status
== EFI_BUFFER_TOO_SMALL
) {
546 FreePool(HandleBuffer
);
547 HandleBuffer
= AllocatePool(BufferSize
);
548 if (HandleBuffer
== NULL
) {
549 return (SHELL_OUT_OF_RESOURCES
);
551 Status
= gBS
->LocateHandle(
553 &gEfiBlockIoProtocolGuid
,
558 if (!EFI_ERROR(Status
)) {
560 // Get the map name(s) for each one.
563 ; LoopVar
< BufferSize
/ sizeof(EFI_HANDLE
)
567 // Skip any that were already done...
569 if (gBS
->OpenProtocol(
570 HandleBuffer
[LoopVar
],
571 &gEfiDevicePathProtocolGuid
,
575 EFI_OPEN_PROTOCOL_TEST_PROTOCOL
) == EFI_SUCCESS
) {
578 PerformSingleMappingDisplay(
585 HandleBuffer
[LoopVar
]);
587 FreePool(HandleBuffer
);
589 return (SHELL_SUCCESS
);
594 PerformMappingDisplay2(
595 IN CONST BOOLEAN Verbose
,
596 IN CONST BOOLEAN Consist
,
597 IN CONST BOOLEAN Normal
,
598 IN CONST CHAR16
*TypeString
,
599 IN CONST BOOLEAN SFO
,
600 IN CONST CHAR16
*Specific OPTIONAL
603 CONST CHAR16
*TypeWalker
;
604 SHELL_STATUS ShellStatus
;
608 if (TypeString
== NULL
) {
609 return (PerformMappingDisplay(Verbose
, Consist
, Normal
, NULL
, SFO
, Specific
, TRUE
));
611 ShellStatus
= SHELL_SUCCESS
;
612 for (TypeWalker
= TypeString
; TypeWalker
!= NULL
&& *TypeWalker
!= CHAR_NULL
;) {
613 Comma
= StrStr(TypeWalker
, L
",");
615 if (ShellStatus
== SHELL_SUCCESS
) {
616 ShellStatus
= PerformMappingDisplay(Verbose
, Consist
, Normal
, TypeWalker
, SFO
, Specific
, (BOOLEAN
)(TypeWalker
== TypeString
));
618 PerformMappingDisplay(Verbose
, Consist
, Normal
, TypeWalker
, SFO
, Specific
, (BOOLEAN
)(TypeWalker
== TypeString
));
623 if (ShellStatus
== SHELL_SUCCESS
) {
624 ShellStatus
= PerformMappingDisplay(Verbose
, Consist
, Normal
, TypeWalker
, SFO
, Specific
, (BOOLEAN
)(TypeWalker
== TypeString
));
626 PerformMappingDisplay(Verbose
, Consist
, Normal
, TypeWalker
, SFO
, Specific
, (BOOLEAN
)(TypeWalker
== TypeString
));
629 TypeWalker
= Comma
+ 1;
633 return (ShellStatus
);
638 PerformMappingDelete(
639 IN CONST CHAR16
*Specific
643 EFI_HANDLE
*HandleBuffer
;
648 if (Specific
== NULL
) {
649 return (EFI_INVALID_PARAMETER
);
657 // Look up all SimpleFileSystems in the platform
659 Status
= gBS
->LocateHandle(
661 &gEfiDevicePathProtocolGuid
,
665 if (Status
== EFI_BUFFER_TOO_SMALL
) {
666 HandleBuffer
= AllocatePool(BufferSize
);
667 if (HandleBuffer
== NULL
) {
668 return (EFI_OUT_OF_RESOURCES
);
670 Status
= gBS
->LocateHandle(
672 &gEfiDevicePathProtocolGuid
,
677 if (EFI_ERROR(Status
)) {
678 SHELL_FREE_NON_NULL(HandleBuffer
);
682 if (HandleBuffer
!= NULL
) {
684 // Get the map name(s) for each one.
687 ; LoopVar
< BufferSize
/ sizeof(EFI_HANDLE
)
690 if (PerformSingleMappingDelete(Specific
,HandleBuffer
[LoopVar
]) == SHELL_SUCCESS
) {
696 // Look up all BlockIo in the platform
698 Status
= gBS
->LocateHandle(
700 &gEfiBlockIoProtocolGuid
,
704 if (Status
== EFI_BUFFER_TOO_SMALL
) {
705 FreePool(HandleBuffer
);
706 HandleBuffer
= AllocatePool(BufferSize
);
707 if (HandleBuffer
== NULL
) {
708 return (EFI_OUT_OF_RESOURCES
);
710 Status
= gBS
->LocateHandle(
712 &gEfiBlockIoProtocolGuid
,
717 if (EFI_ERROR(Status
)) {
718 SHELL_FREE_NON_NULL(HandleBuffer
);
722 if (HandleBuffer
!= NULL
) {
724 // Get the map name(s) for each one.
727 ; LoopVar
< BufferSize
/ sizeof(EFI_HANDLE
)
731 // Skip any that were already done...
733 if (gBS
->OpenProtocol(
734 HandleBuffer
[LoopVar
],
735 &gEfiDevicePathProtocolGuid
,
739 EFI_OPEN_PROTOCOL_TEST_PROTOCOL
) == EFI_SUCCESS
) {
742 if (PerformSingleMappingDelete(Specific
,HandleBuffer
[LoopVar
]) == SHELL_SUCCESS
) {
747 SHELL_FREE_NON_NULL(HandleBuffer
);
749 return (EFI_NOT_FOUND
);
751 return (EFI_SUCCESS
);
755 function to add a mapping from mapping.
757 This function will get the device path associated with the mapping and call SetMap.
759 @param[in] Map The Map to add a mapping for
760 @param[in] SName The name of the new mapping
762 @retval SHELL_SUCCESS the mapping was added
763 @retval SHELL_INVALID_PARAMETER the device path for Map could not be retrieved.
764 @return Shell version of a return value from EfiShellProtocol->SetMap
769 AddMappingFromMapping(
770 IN CONST CHAR16
*Map
,
771 IN CONST CHAR16
*SName
774 CONST EFI_DEVICE_PATH_PROTOCOL
*DevPath
;
777 if (StrStr(SName
, L
"*") != NULL
778 ||StrStr(SName
, L
"?") != NULL
779 ||StrStr(SName
, L
"[") != NULL
780 ||StrStr(SName
, L
"]") != NULL
) {
781 return (SHELL_INVALID_PARAMETER
);
784 DevPath
= gEfiShellProtocol
->GetDevicePathFromMap(Map
);
785 if (DevPath
== NULL
) {
786 return (SHELL_INVALID_PARAMETER
);
789 Status
= gEfiShellProtocol
->SetMap(DevPath
, SName
);
790 if (EFI_ERROR(Status
)) {
791 return (SHELL_DEVICE_ERROR
);
793 return (SHELL_SUCCESS
);
797 function to determine if a string has only numbers and letters
799 This is useful for such things as Map names which can only be letters and numbers
801 @param[in] String pointer to the string to analyze
803 @retval TRUE String has only numbers and letters
804 @retval FALSE String has at least one other character.
809 IN CONST CHAR16
*String
812 while(String
!= NULL
&& *String
!= CHAR_NULL
) {
813 if (! ((*String
>= L
'a' && *String
<= L
'z') ||
814 (*String
>= L
'A' && *String
<= L
'Z') ||
815 (*String
>= L
'0' && *String
<= L
'9'))
825 function to add a mapping from an EFI_HANDLE.
827 This function will get the device path associated with the Handle and call SetMap.
829 @param[in] Handle The handle to add a mapping for
830 @param[in] SName The name of the new mapping
832 @retval SHELL_SUCCESS the mapping was added
833 @retval SHELL_INVALID_PARAMETER SName was not valid for a map name.
834 @return Shell version of a return value from either
835 gBS->OpenProtocol or EfiShellProtocol->SetMap
840 AddMappingFromHandle(
841 IN CONST EFI_HANDLE Handle
,
842 IN CONST CHAR16
*SName
845 EFI_DEVICE_PATH_PROTOCOL
*DevPath
;
848 if (!IsNumberLetterOnly(SName
)) {
849 return (SHELL_INVALID_PARAMETER
);
852 Status
= gBS
->OpenProtocol(
854 &gEfiDevicePathProtocolGuid
,
858 EFI_OPEN_PROTOCOL_GET_PROTOCOL
860 if (EFI_ERROR(Status
)) {
861 return (SHELL_DEVICE_ERROR
);
863 Status
= gEfiShellProtocol
->SetMap(DevPath
, SName
);
864 if (EFI_ERROR(Status
)) {
865 return (SHELL_DEVICE_ERROR
);
867 return (SHELL_SUCCESS
);
870 STATIC CONST SHELL_PARAM_ITEM MapParamList
[] = {
878 {L
"-sfo", TypeValue
},
883 Function for 'map' command.
885 @param[in] ImageHandle Handle to the Image (NULL if Internal).
886 @param[in] SystemTable Pointer to the System Table (NULL if Internal).
891 IN EFI_HANDLE ImageHandle
,
892 IN EFI_SYSTEM_TABLE
*SystemTable
897 CHAR16
*ProblemParam
;
899 CONST CHAR16
*Mapping
;
900 EFI_HANDLE MappingAsHandle
;
901 CONST EFI_DEVICE_PATH_PROTOCOL
*DevPath
;
902 SHELL_STATUS ShellStatus
;
906 CONST CHAR16
*Param1
;
907 CONST CHAR16
*TypeString
;
913 ShellStatus
= SHELL_SUCCESS
;
914 MappingAsHandle
= NULL
;
917 // initialize the shell lib (we must be in non-auto-init...)
919 Status
= ShellInitialize();
920 ASSERT_EFI_ERROR(Status
);
922 Status
= CommandInit();
923 ASSERT_EFI_ERROR(Status
);
926 // parse the command line
928 Status
= ShellCommandLineParse (MapParamList
, &Package
, &ProblemParam
, TRUE
);
929 if (EFI_ERROR(Status
)) {
930 if (Status
== EFI_VOLUME_CORRUPTED
&& ProblemParam
!= NULL
) {
931 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PROBLEM
), gShellLevel2HiiHandle
, ProblemParam
);
932 FreePool(ProblemParam
);
933 ShellStatus
= SHELL_INVALID_PARAMETER
;
941 SfoMode
= ShellCommandLineGetFlag(Package
, L
"-sfo");
942 ConstMode
= ShellCommandLineGetFlag(Package
, L
"-c");
943 NormlMode
= ShellCommandLineGetFlag(Package
, L
"-f");
944 if (ShellCommandLineGetFlag(Package
, L
"-?")) {
946 } else if (ShellCommandLineGetRawValue(Package
, 3) != NULL
) {
947 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_TOO_MANY
), gShellLevel2HiiHandle
);
948 ShellStatus
= SHELL_INVALID_PARAMETER
;
951 // Deleting a map name...
953 if (ShellCommandLineGetFlag(Package
, L
"-d")) {
954 if ( ShellCommandLineGetFlag(Package
, L
"-r")
955 || ShellCommandLineGetFlag(Package
, L
"-v")
958 || ShellCommandLineGetFlag(Package
, L
"-u")
959 || ShellCommandLineGetFlag(Package
, L
"-t")
961 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PARAM_CON
), gShellLevel2HiiHandle
);
962 ShellStatus
= SHELL_INVALID_PARAMETER
;
964 SName
= ShellCommandLineGetValue(Package
, L
"-d");
966 Status
= PerformMappingDelete(SName
);
967 if (EFI_ERROR(Status
)) {
969 case EFI_ACCESS_DENIED
:
970 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_ERR_AD
), gShellLevel2HiiHandle
);
971 ShellStatus
= SHELL_ACCESS_DENIED
;
974 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_MAP_NF
), gShellLevel2HiiHandle
, SName
);
975 ShellStatus
= SHELL_INVALID_PARAMETER
;
978 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_ERR_UK
), gShellLevel2HiiHandle
, Status
);
979 ShellStatus
= SHELL_UNSUPPORTED
;
983 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_TOO_FEW
), gShellLevel2HiiHandle
);
984 ShellStatus
= SHELL_INVALID_PARAMETER
;
987 } else if ( ShellCommandLineGetFlag(Package
, L
"-r")
988 // || ShellCommandLineGetFlag(Package, L"-v")
991 || ShellCommandLineGetFlag(Package
, L
"-u")
992 || ShellCommandLineGetFlag(Package
, L
"-t")
994 if ( ShellCommandLineGetFlag(Package
, L
"-r")) {
998 Status
= ShellCommandCreateInitialMappingsAndPaths();
999 if (EFI_ERROR(Status
)) {
1000 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_ERR_UK
), gShellLevel2HiiHandle
, Status
);
1001 ShellStatus
= SHELL_UNSUPPORTED
;
1004 if ( ShellStatus
== SHELL_SUCCESS
&& ShellCommandLineGetFlag(Package
, L
"-u")) {
1008 Status
= UpdateMapping();
1009 if (EFI_ERROR(Status
)) {
1010 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_ERR_UK
), gShellLevel2HiiHandle
, Status
);
1011 ShellStatus
= SHELL_UNSUPPORTED
;
1014 if (ShellStatus
== SHELL_SUCCESS
) {
1015 Param1
= ShellCommandLineGetRawValue(Package
, 1);
1016 TypeString
= ShellCommandLineGetValue(Package
, L
"-t");
1019 &&TypeString
== NULL
1022 // now do the display...
1024 ShellStatus
= PerformMappingDisplay(
1025 ShellCommandLineGetFlag(Package
, L
"-v"),
1035 // now do the display...
1037 ShellStatus
= PerformMappingDisplay2(
1038 ShellCommandLineGetFlag(Package
, L
"-v"),
1049 // adding or displaying (there were no flags)
1051 SName
= ShellCommandLineGetRawValue(Package
, 1);
1052 Mapping
= ShellCommandLineGetRawValue(Package
, 2);
1057 // display only since no flags
1059 ShellStatus
= PerformMappingDisplay(
1060 ShellCommandLineGetFlag(Package
, L
"-v"),
1068 } else if ( SName
== NULL
1072 // Display only the one specified
1074 ShellStatus
= PerformMappingDisplay(
1080 SName
, // note the variable here...
1084 if (ShellIsHexOrDecimalNumber(Mapping
, TRUE
, FALSE
)) {
1085 MappingAsHandle
= ConvertHandleIndexToHandle(StrHexToUintn(Mapping
));
1087 MappingAsHandle
= NULL
;
1089 if (MappingAsHandle
== NULL
&& Mapping
[StrLen(Mapping
)-1] != L
':') {
1090 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PROBLEM
), gShellLevel2HiiHandle
, Mapping
);
1091 ShellStatus
= SHELL_INVALID_PARAMETER
;
1093 if (MappingAsHandle
!= NULL
) {
1094 ShellStatus
= AddMappingFromHandle(MappingAsHandle
, SName
);
1096 ShellStatus
= AddMappingFromMapping(Mapping
, SName
);
1098 if (ShellStatus
!= SHELL_SUCCESS
) {
1099 switch (ShellStatus
) {
1100 case SHELL_ACCESS_DENIED
:
1101 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_ERR_AD
), gShellLevel2HiiHandle
);
1103 case SHELL_INVALID_PARAMETER
:
1104 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PARAM_INV
), gShellLevel2HiiHandle
);
1107 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_ERR_UK
), gShellLevel2HiiHandle
, ShellStatus
|MAX_BIT
);
1111 // now do the display...
1113 ShellStatus
= PerformMappingDisplay(
1122 } // we were sucessful so do an output
1123 } // got a valid map target
1124 } // got 2 variables
1125 } // we are adding a mapping
1126 } // got valid parameters
1130 // free the command line package
1132 ShellCommandLineFreeVarList (Package
);
1134 return (ShellStatus
);