ShellPkg/map: Fix out-of-bound read when "map fsn"
[mirror_edk2.git] / ShellPkg / Library / UefiShellLevel2CommandsLib / Map.c
index d74a47b..9166ca2 100644 (file)
@@ -1,7 +1,10 @@
 /** @file\r
   Main file for map shell level 2 command.\r
 \r
-  Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
+  Copyright (c) 2009 - 2018, Intel Corporation. All rights reserved.<BR>\r
+  (C) Copyright 2013-2015 Hewlett-Packard Development Company, L.P.<BR>\r
+  (C) Copyright 2016 Hewlett Packard Enterprise Development LP<BR>\r
+  \r
   This program and the accompanying materials\r
   are licensed and made available under the terms and conditions of the BSD License\r
   which accompanies this distribution.  The full text of the license may be found at\r
 #include <Library/HandleParsingLib.h>\r
 #include <Library/SortLib.h>\r
 \r
+/**\r
+  Determine if a string has only numbers and letters.\r
+\r
+  This is useful for such things as Map names which can only be letters and numbers.\r
+\r
+  @param[in] String       pointer to the string to analyze,\r
+  @param[in] Len          Number of characters to analyze.\r
+\r
+  @retval TRUE            String has only numbers and letters\r
+  @retval FALSE           String has at least one other character.\r
+**/\r
+BOOLEAN\r
+IsNumberLetterOnly(\r
+  IN CONST CHAR16 *String,\r
+  IN CONST UINTN  Len\r
+  )\r
+{\r
+  UINTN Count;\r
+  for (Count = 0 ; Count < Len && String != NULL && *String != CHAR_NULL ; String++,Count++) {\r
+    if (! ((*String >= L'a' && *String <= L'z') ||\r
+           (*String >= L'A' && *String <= L'Z') ||\r
+           (*String >= L'0' && *String <= L'9'))\r
+        ){\r
+      return (FALSE);\r
+    }\r
+  }\r
+  return (TRUE);\r
+}\r
+\r
+/**\r
+  Do a search in the Target delimited list.\r
+\r
+  @param[in] List         The list to seatch in.\r
+  @param[in] MetaTarget   The item to search for. MetaMatching supported.\r
+  @param[out] FullName    Optional pointer to an allocated buffer containing \r
+                          the match.\r
+  @param[in] Meta         TRUE to use MetaMatching.\r
+  @param[in] SkipTrailingNumbers  TRUE to allow for numbers after the MetaTarget.\r
+  @param[in] Target       The single character that delimits list \r
+                          items (";" normally). \r
+**/\r
 BOOLEAN\r
-EFIAPI\r
 SearchList(\r
   IN CONST CHAR16   *List,\r
   IN CONST CHAR16   *MetaTarget,\r
@@ -75,105 +118,19 @@ SearchList(
   return (FALSE);\r
 }\r
 \r
-EFI_STATUS\r
-EFIAPI\r
-UpdateMapping (\r
-  VOID\r
-  )\r
-{\r
-  EFI_STATUS                Status;\r
-  EFI_HANDLE                *HandleList;\r
-  UINTN                     Count;\r
-  EFI_DEVICE_PATH_PROTOCOL  **DevicePathList;\r
-  CHAR16                    *NewDefaultName;\r
-  CHAR16                    *NewConsistName;\r
-  EFI_DEVICE_PATH_PROTOCOL  **ConsistMappingTable;\r
-\r
-  HandleList  = NULL;\r
-  Status      = EFI_SUCCESS;\r
-\r
-  //\r
-  // remove mappings that represent removed devices.\r
-  //\r
-\r
-  //\r
-  // Find each handle with Simple File System\r
-  //\r
-  HandleList = GetHandleListByProtocol(&gEfiSimpleFileSystemProtocolGuid);\r
-  if (HandleList != NULL) {\r
-    //\r
-    // Do a count of the handles\r
-    //\r
-    for (Count = 0 ; HandleList[Count] != NULL ; Count++);\r
-\r
-    //\r
-    // Get all Device Paths\r
-    //\r
-    DevicePathList = AllocatePool(sizeof(EFI_DEVICE_PATH_PROTOCOL*) * Count);\r
-    ASSERT(DevicePathList != NULL);\r
-\r
-    for (Count = 0 ; HandleList[Count] != NULL ; Count++) {\r
-      DevicePathList[Count] = DevicePathFromHandle(HandleList[Count]);\r
-    }\r
-\r
-    //\r
-    // Sort all DevicePaths\r
-    //\r
-    PerformQuickSort(DevicePathList, Count, sizeof(EFI_DEVICE_PATH_PROTOCOL*), DevicePathCompare);\r
-\r
-    ShellCommandConsistMappingInitialize(&ConsistMappingTable);\r
-\r
-    //\r
-    // Assign new Mappings to remainders\r
-    //\r
-    for (Count = 0 ; HandleList[Count] != NULL && !EFI_ERROR(Status); Count++) {\r
-      //\r
-      // Skip ones that already have\r
-      //\r
-      if (gEfiShellProtocol->GetMapFromDevicePath(&DevicePathList[Count]) != NULL) {\r
-        continue;\r
-      }\r
-      //\r
-      // Get default name\r
-      //\r
-      NewDefaultName = ShellCommandCreateNewMappingName(MappingTypeFileSystem);\r
-      ASSERT(NewDefaultName != NULL);\r
-\r
-      //\r
-      // Call shell protocol SetMap function now...\r
-      //\r
-      Status = gEfiShellProtocol->SetMap(DevicePathList[Count], NewDefaultName);\r
-\r
-      if (!EFI_ERROR(Status)) {\r
-        //\r
-        // Now do consistent name\r
-        //\r
-        NewConsistName = ShellCommandConsistMappingGenMappingName(DevicePathList[Count], ConsistMappingTable);\r
-        if (NewConsistName != NULL) {\r
-          Status = gEfiShellProtocol->SetMap(DevicePathList[Count], NewConsistName);\r
-          FreePool(NewConsistName);\r
-        }\r
-      }\r
-\r
-      FreePool(NewDefaultName);\r
-    }\r
-    ShellCommandConsistMappingUnInitialize(ConsistMappingTable);\r
-    SHELL_FREE_NON_NULL(HandleList);\r
-    SHELL_FREE_NON_NULL(DevicePathList);\r
-\r
-    HandleList = NULL;\r
-  } else {\r
-    Count = (UINTN)-1;\r
-  }\r
-  //\r
-  // Do it all over again for gEfiBlockIoProtocolGuid\r
-  //\r
+/**\r
+  Determine what type of device is represented and return it's string.  The \r
+  string is in allocated memory and must be callee freed.  The HII is is listed below.\r
+  The actual string cannot be determined.\r
 \r
-  return (Status);\r
-}\r
+  @param[in] DevicePath     The device to analyze.\r
 \r
+  @retval STR_MAP_MEDIA_UNKNOWN   The media type is unknown.\r
+  @retval STR_MAP_MEDIA_HARDDISK  The media is a hard drive.\r
+  @retval STR_MAP_MEDIA_CDROM     The media is a CD ROM.\r
+  @retval STR_MAP_MEDIA_FLOPPY    The media is a floppy drive.\r
+**/\r
 CHAR16*\r
-EFIAPI\r
 GetDeviceMediaType (\r
   IN  EFI_DEVICE_PATH_PROTOCOL     *DevicePath\r
   )\r
@@ -219,7 +176,6 @@ GetDeviceMediaType (
   @retval FALSE                     The handle does not have removable storage.\r
 **/\r
 BOOLEAN\r
-EFIAPI\r
 IsRemoveableDevice (\r
   IN EFI_DEVICE_PATH_PROTOCOL      *DevicePath\r
   )\r
@@ -256,7 +212,6 @@ IsRemoveableDevice (
   @retval FALSE               The map should not be displayed.\r
 **/\r
 BOOLEAN\r
-EFIAPI\r
 MappingListHasType(\r
   IN CONST CHAR16     *MapList,\r
   IN CONST CHAR16     *Specific,\r
@@ -265,15 +220,39 @@ MappingListHasType(
   IN CONST BOOLEAN    Consist\r
   )\r
 {\r
+  CHAR16              *NewSpecific;\r
+  RETURN_STATUS       Status;\r
+  UINTN               Length;\r
+  \r
   //\r
   // specific has priority\r
   //\r
-  if ( Specific != NULL\r
-    && SearchList(MapList, Specific, NULL, TRUE, FALSE, L";")) {\r
-    return (TRUE);\r
-  }\r
+  if (Specific != NULL) {\r
+    Length      = StrLen (Specific);\r
+    //\r
+    // Allocate enough buffer for Specific and potential ":"\r
+    //\r
+    NewSpecific = AllocatePool ((Length + 2) * sizeof(CHAR16));\r
+    if (NewSpecific == NULL){\r
+      return FALSE;\r
+    }\r
+    StrCpyS (NewSpecific, Length + 2, Specific);\r
+    if (Specific[Length - 1] != L':') {\r
+      Status = StrnCatS(NewSpecific, Length + 2, L":", StrLen(L":"));\r
+      if (EFI_ERROR (Status)) {\r
+        FreePool(NewSpecific);\r
+        return FALSE;\r
+      }\r
+    }\r
 \r
+    if (SearchList(MapList, NewSpecific, NULL, TRUE, FALSE, L";")) {\r
+      FreePool(NewSpecific);\r
+      return (TRUE);\r
+    }\r
+    FreePool(NewSpecific);\r
+  }\r
   if (  Consist\r
+    && Specific == NULL\r
     && (SearchList(MapList, L"HD*",  NULL, TRUE, TRUE, L";")\r
       ||SearchList(MapList, L"CD*",  NULL, TRUE, TRUE, L";")\r
       ||SearchList(MapList, L"F*",   NULL, TRUE, TRUE, L";")\r
@@ -282,6 +261,7 @@ MappingListHasType(
   }\r
 \r
   if (  Normal\r
+    && Specific == NULL\r
     && (SearchList(MapList, L"FS",  NULL, FALSE, TRUE, L";")\r
       ||SearchList(MapList, L"BLK", NULL, FALSE, TRUE, L";"))){\r
     return (TRUE);\r
@@ -294,8 +274,20 @@ MappingListHasType(
 }\r
 \r
 \r
-VOID\r
-EFIAPI\r
+/**\r
+  Display a single map line for device Handle if conditions are met.\r
+\r
+  @param[in] Verbose                TRUE to display (extra) verbose information.\r
+  @param[in] Consist                TRUE to display consistent mappings.\r
+  @param[in] Normal                 TRUE to display normal (not consist) mappings.\r
+  @param[in] TypeString             pointer to string of filter types.\r
+  @param[in] SFO                    TRUE to display output in Standard Output Format.\r
+  @param[in] Specific               pointer to string for specific map to display.\r
+  @param[in] Handle                 The handle to display from.\r
+\r
+  @retval EFI_SUCCESS               The mapping was displayed.\r
+**/\r
+EFI_STATUS\r
 PerformSingleMappingDisplay(\r
   IN CONST BOOLEAN    Verbose,\r
   IN CONST BOOLEAN    Consist,\r
@@ -313,30 +305,102 @@ PerformSingleMappingDisplay(
   CHAR16                    *MediaType;\r
   CHAR16                    *DevPathString;\r
   CHAR16                    *TempSpot;\r
+  CHAR16                    *Alias;\r
   UINTN                     TempLen;\r
   BOOLEAN                   Removable;\r
   CONST CHAR16              *TempSpot2;\r
 \r
+  Alias       = NULL;\r
+  TempSpot2   = NULL;\r
+  CurrentName = NULL;\r
   DevPath = DevicePathFromHandle(Handle);\r
   DevPathCopy = DevPath;\r
   MapList = gEfiShellProtocol->GetMapFromDevicePath(&DevPathCopy);\r
   if (MapList == NULL) {\r
-    return;\r
+    return EFI_NOT_FOUND;\r
   }\r
 \r
   if (!MappingListHasType(MapList, Specific, TypeString, Normal, Consist)){\r
-    return;\r
+    return EFI_NOT_FOUND;\r
   }\r
 \r
-  CurrentName = NULL;\r
-  CurrentName = StrnCatGrow(&CurrentName, 0, MapList, 0);\r
-  TempSpot = StrStr(CurrentName, L";");\r
-  if (TempSpot != NULL) {\r
-    *TempSpot = CHAR_NULL;\r
+  if (Normal || !Consist) {\r
+    //\r
+    // need the Normal here since people can use both on command line.  otherwise unused.\r
+    //\r
+\r
+    //\r
+    // Allocate a name\r
+    //\r
+    CurrentName = NULL;\r
+    CurrentName = StrnCatGrow(&CurrentName, 0, MapList, 0);\r
+    if (CurrentName == NULL) {\r
+      return (EFI_OUT_OF_RESOURCES);\r
+    }\r
+\r
+    //\r
+    // Chop off the other names that become "Alias(s)"\r
+    // leaving just the normal name\r
+    //\r
+    TempSpot = StrStr(CurrentName, L";");\r
+    if (TempSpot != NULL) {\r
+      *TempSpot = CHAR_NULL;\r
+    }\r
+  } else {\r
+    CurrentName = NULL;\r
+\r
+    //\r
+    // Skip the first name.  This is the standard name.\r
+    //\r
+    TempSpot = StrStr(MapList, L";");\r
+    if (TempSpot != NULL) {\r
+      TempSpot++;\r
+    }\r
+    SearchList(TempSpot, L"HD*", &CurrentName, TRUE, FALSE, L";");\r
+    if (CurrentName == NULL) {\r
+      SearchList(TempSpot, L"CD*", &CurrentName, TRUE, FALSE, L";");\r
+    }\r
+    if (CurrentName == NULL) {\r
+      SearchList(TempSpot, L"FP*", &CurrentName, TRUE, FALSE, L";");\r
+    }\r
+    if (CurrentName == NULL) {\r
+      SearchList(TempSpot, L"F*",  &CurrentName, TRUE, FALSE, L";");\r
+    }\r
+    if (CurrentName == NULL) {\r
+      //\r
+      // We didnt find anything, so just the first one in the list...\r
+      //\r
+      CurrentName = StrnCatGrow(&CurrentName, 0, MapList, 0);\r
+      if (CurrentName == NULL) {\r
+        return (EFI_OUT_OF_RESOURCES);\r
+      }\r
+      TempSpot = StrStr(CurrentName, L";");\r
+      if (TempSpot != NULL) {\r
+        *TempSpot = CHAR_NULL;\r
+      }\r
+    } else {\r
+      Alias = StrnCatGrow(&Alias, 0, MapList, 0);\r
+      if (Alias == NULL) {\r
+        return EFI_OUT_OF_RESOURCES;\r
+      }\r
+      TempSpot = StrStr(Alias, CurrentName);\r
+      if (TempSpot != NULL) {\r
+        TempSpot2 = StrStr(TempSpot, L";");\r
+        if (TempSpot2 != NULL) {\r
+          TempSpot2++; // Move past ";" from CurrentName\r
+          CopyMem(TempSpot, TempSpot2, StrSize(TempSpot2));\r
+        } else {\r
+          *TempSpot = CHAR_NULL;\r
+        }\r
+      }\r
+      if (Alias[StrLen(Alias)-1] == L';') {\r
+        Alias[StrLen(Alias)-1] = CHAR_NULL;\r
+      }\r
+    }\r
   }\r
-  DevPathString = gDevPathToText->ConvertDevicePathToText(DevPath, TRUE, FALSE);\r
+  DevPathString = ConvertDevicePathToText(DevPath, TRUE, FALSE);\r
+  TempLen = StrLen(CurrentName);\r
   if (!SFO) {\r
-    TempLen = StrLen(CurrentName);\r
     ShellPrintHiiEx (\r
       -1,\r
       -1,\r
@@ -344,7 +408,7 @@ PerformSingleMappingDisplay(
       STRING_TOKEN (STR_MAP_ENTRY),\r
       gShellLevel2HiiHandle,\r
       CurrentName,\r
-      TempLen < StrLen(MapList)?MapList + TempLen+1:L"",\r
+      Alias!=NULL?Alias:(TempLen < StrLen(MapList)?MapList + TempLen+1:L""),\r
       DevPathString\r
      );\r
     if (Verbose) {\r
@@ -367,10 +431,9 @@ PerformSingleMappingDisplay(
           TempSpot2\r
          );\r
       }\r
-      FreePool(MediaType);\r
+      SHELL_FREE_NON_NULL(MediaType);\r
     }\r
   } else {\r
-    TempLen = StrLen(CurrentName);\r
     ShellPrintHiiEx (\r
       -1,\r
       -1,\r
@@ -379,16 +442,25 @@ PerformSingleMappingDisplay(
       gShellLevel2HiiHandle,\r
       CurrentName,\r
       DevPathString,\r
-      TempLen < StrLen(MapList)?MapList + TempLen+1:L""\r
+      Consist?L"":(TempLen < StrLen(MapList)?MapList + TempLen+1:L"")\r
      );\r
   }\r
-  FreePool(DevPathString);\r
-  FreePool(CurrentName);\r
-  return;\r
+  SHELL_FREE_NON_NULL(DevPathString);\r
+  SHELL_FREE_NON_NULL(CurrentName);\r
+  SHELL_FREE_NON_NULL(Alias);\r
+  return EFI_SUCCESS;\r
 }\r
 \r
+/**\r
+  Delete Specific from the list of maps for device Handle.\r
+\r
+  @param[in] Specific   The name to delete.\r
+  @param[in] Handle     The device to look on.\r
+\r
+  @retval EFI_SUCCESS     The delete was successful.\r
+  @retval EFI_NOT_FOUND   Name was not a map on Handle.\r
+**/\r
 EFI_STATUS\r
-EFIAPI\r
 PerformSingleMappingDelete(\r
   IN CONST CHAR16     *Specific,\r
   IN CONST EFI_HANDLE Handle\r
@@ -416,29 +488,29 @@ PerformSingleMappingDelete(
   return (gEfiShellProtocol->SetMap(NULL, CurrentName));\r
 }\r
 \r
+CONST CHAR16 Cd[] = L"cd*";\r
+CONST CHAR16 Hd[] = L"hd*";\r
+CONST CHAR16 Fp[] = L"fp*";\r
+CONST CHAR16 AnyF[] = L"F*";\r
 /**\r
   Function to display mapping information to the user.\r
 \r
-  if Specific is specified then Consist and Normal will be ignored since information will\r
+  If Specific is specified then Consist and Normal will be ignored since information will\r
   be printed for the specific item only.\r
 \r
-  @param[in] Verbose                TRUE to display (extra) verbose information\r
-  @param[in] Consist                TRUE to display consistent mappings\r
-  @param[in] Normal                 TRUE to display normal (not consist) mappings\r
-  @param[in] TypeString             pointer to string of filter types\r
-  @param[in] SFO                    TRUE to display output in Standard Output Format\r
-  @param[in] Specific               pointer to string for specific map to display\r
+  @param[in] Verbose                TRUE to display (extra) verbose information.\r
+  @param[in] Consist                TRUE to display consistent mappings.\r
+  @param[in] Normal                 TRUE to display normal (not consist) mappings.\r
+  @param[in] TypeString             Pointer to string of filter types.\r
+  @param[in] SFO                    TRUE to display output in Standard Output Format.\r
+  @param[in] Specific               Pointer to string for specific map to display.\r
+  @param[in] Header                 TRUE to print the header block.\r
 \r
-  @retval SHELL_SUCCESS               the display was printed\r
-  @retval SHELL_INVALID_PARAMETER     one of Consist or Normal must be TRUE if no Specific\r
+  @retval SHELL_SUCCESS               The display was printed.\r
+  @retval SHELL_INVALID_PARAMETER     One of Consist or Normal must be TRUE if no Specific.\r
 \r
 **/\r
-CONST CHAR16 Cd[] = L"cd*";\r
-CONST CHAR16 Hd[] = L"hd*";\r
-CONST CHAR16 Fp[] = L"fp*";\r
-CONST CHAR16 F[] = L"F*";\r
 SHELL_STATUS\r
-EFIAPI\r
 PerformMappingDisplay(\r
   IN CONST BOOLEAN Verbose,\r
   IN CONST BOOLEAN Consist,\r
@@ -454,9 +526,10 @@ PerformMappingDisplay(
   UINTN                     BufferSize;\r
   UINTN                     LoopVar;\r
   CHAR16                    *Test;\r
+  BOOLEAN                   Found;\r
 \r
   if (!Consist && !Normal && Specific == NULL && TypeString == NULL) {\r
-    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel2HiiHandle);\r
+    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel2HiiHandle, L"map");  \r
     return (SHELL_INVALID_PARAMETER);\r
   }\r
 \r
@@ -467,11 +540,11 @@ PerformMappingDisplay(
       if (StrnCmp(TypeString, Test, StrLen(Test)-1) != 0) {\r
         Test = (CHAR16*)Fp;\r
         if (StrnCmp(TypeString, Test, StrLen(Test)-1) != 0) {\r
-          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, TypeString);\r
+          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellLevel2HiiHandle, L"map", TypeString);  \r
           return (SHELL_INVALID_PARAMETER);\r
         }\r
       } else if (Test == NULL) {\r
-        Test = (CHAR16*)F;\r
+        Test = (CHAR16*)AnyF;\r
       }\r
     }\r
   } else {\r
@@ -497,33 +570,31 @@ PerformMappingDisplay(
   //\r
   Status = gBS->LocateHandle(\r
     ByProtocol,\r
-    &gEfiDevicePathProtocolGuid,\r
+    &gEfiSimpleFileSystemProtocolGuid,\r
     NULL,\r
     &BufferSize,\r
     HandleBuffer);\r
   if (Status == EFI_BUFFER_TOO_SMALL) {\r
-    HandleBuffer = AllocatePool(BufferSize);\r
+    HandleBuffer = AllocateZeroPool(BufferSize);\r
     if (HandleBuffer == NULL) {\r
       return (SHELL_OUT_OF_RESOURCES);\r
     }\r
     Status = gBS->LocateHandle(\r
       ByProtocol,\r
-      &gEfiDevicePathProtocolGuid,\r
+      &gEfiSimpleFileSystemProtocolGuid,\r
       NULL,\r
       &BufferSize,\r
       HandleBuffer);\r
   }\r
-  ASSERT_EFI_ERROR(Status);\r
-  ASSERT(HandleBuffer != NULL);\r
 \r
   //\r
   // Get the map name(s) for each one.\r
   //\r
-  for ( LoopVar = 0\r
-      ; LoopVar < (BufferSize / sizeof(EFI_HANDLE))\r
+  for ( LoopVar = 0, Found = FALSE\r
+      ; LoopVar < (BufferSize / sizeof(EFI_HANDLE)) && HandleBuffer != NULL\r
       ; LoopVar ++\r
      ){\r
-    PerformSingleMappingDisplay(\r
+    Status = PerformSingleMappingDisplay(\r
       Verbose,\r
       Consist,\r
       Normal,\r
@@ -531,6 +602,9 @@ PerformMappingDisplay(
       SFO,\r
       Specific,\r
       HandleBuffer[LoopVar]);\r
+    if (!EFI_ERROR(Status)) {\r
+      Found = TRUE;\r
+    }\r
   }\r
 \r
   //\r
@@ -543,8 +617,8 @@ PerformMappingDisplay(
     &BufferSize,\r
     HandleBuffer);\r
   if (Status == EFI_BUFFER_TOO_SMALL) {\r
-    FreePool(HandleBuffer);\r
-    HandleBuffer = AllocatePool(BufferSize);\r
+    SHELL_FREE_NON_NULL(HandleBuffer);\r
+    HandleBuffer = AllocateZeroPool(BufferSize);\r
     if (HandleBuffer == NULL) {\r
       return (SHELL_OUT_OF_RESOURCES);\r
     }\r
@@ -555,7 +629,7 @@ PerformMappingDisplay(
       &BufferSize,\r
       HandleBuffer);\r
   }\r
-  if (!EFI_ERROR(Status)) {\r
+  if (!EFI_ERROR(Status) && HandleBuffer != NULL) {\r
     //\r
     // Get the map name(s) for each one.\r
     //\r
@@ -568,14 +642,14 @@ PerformMappingDisplay(
       //\r
       if (gBS->OpenProtocol(\r
         HandleBuffer[LoopVar],\r
-        &gEfiDevicePathProtocolGuid,\r
+        &gEfiSimpleFileSystemProtocolGuid,\r
         NULL,\r
         gImageHandle,\r
         NULL,\r
         EFI_OPEN_PROTOCOL_TEST_PROTOCOL) == EFI_SUCCESS) {\r
           continue;\r
       }\r
-      PerformSingleMappingDisplay(\r
+      Status = PerformSingleMappingDisplay(\r
         Verbose,\r
         Consist,\r
         Normal,\r
@@ -583,14 +657,37 @@ PerformMappingDisplay(
         SFO,\r
         Specific,\r
         HandleBuffer[LoopVar]);\r
+      if (!EFI_ERROR(Status)) {\r
+        Found = TRUE;\r
+      }\r
     }\r
     FreePool(HandleBuffer);\r
   }\r
+  if (!Found) {\r
+    if (Specific != NULL) {\r
+      ShellPrintHiiEx(gST->ConOut->Mode->CursorColumn, gST->ConOut->Mode->CursorRow-1, NULL, STRING_TOKEN (STR_MAP_NF), gShellLevel2HiiHandle, L"map", Specific);  \r
+    } else {\r
+      ShellPrintHiiEx(gST->ConOut->Mode->CursorColumn, gST->ConOut->Mode->CursorRow-1, NULL, STRING_TOKEN (STR_CD_NF), gShellLevel2HiiHandle, L"map");  \r
+    }\r
+  }\r
   return (SHELL_SUCCESS);\r
 }\r
 \r
+/**\r
+  Perform a mapping display and parse for multiple types in the TypeString.\r
+\r
+  @param[in] Verbose      TRUE to use verbose output.\r
+  @param[in] Consist      TRUE to display consistent names.\r
+  @param[in] Normal       TRUE to display normal names.\r
+  @param[in] TypeString   An optional comma-delimited list of types.\r
+  @param[in] SFO          TRUE to display in SFO format.  See Spec.\r
+  @param[in] Specific     An optional specific map name to display alone.\r
+\r
+  @retval SHELL_INVALID_PARAMETER   A parameter was invalid.\r
+  @retval SHELL_SUCCESS             The display was successful.\r
+  @sa PerformMappingDisplay\r
+**/\r
 SHELL_STATUS\r
-EFIAPI\r
 PerformMappingDisplay2(\r
   IN CONST BOOLEAN Verbose,\r
   IN CONST BOOLEAN Consist,\r
@@ -633,8 +730,16 @@ PerformMappingDisplay2(
   return (ShellStatus);\r
 }\r
 \r
+/**\r
+  Delete a specific map.\r
+\r
+  @param[in] Specific  The pointer to the name of the map to delete.\r
+\r
+  @retval EFI_INVALID_PARAMETER     Specific was NULL.\r
+  @retval EFI_SUCCESS               The operation was successful.\r
+  @retval EFI_NOT_FOUND             Specific could not be found.\r
+**/\r
 EFI_STATUS\r
-EFIAPI\r
 PerformMappingDelete(\r
   IN CONST CHAR16  *Specific\r
   )\r
@@ -663,7 +768,7 @@ PerformMappingDelete(
     &BufferSize,\r
     HandleBuffer);\r
   if (Status == EFI_BUFFER_TOO_SMALL) {\r
-    HandleBuffer = AllocatePool(BufferSize);\r
+    HandleBuffer = AllocateZeroPool(BufferSize);\r
     if (HandleBuffer == NULL) {\r
       return (EFI_OUT_OF_RESOURCES);\r
     }\r
@@ -703,7 +808,7 @@ PerformMappingDelete(
     HandleBuffer);\r
   if (Status == EFI_BUFFER_TOO_SMALL) {\r
     FreePool(HandleBuffer);\r
-    HandleBuffer = AllocatePool(BufferSize);\r
+    HandleBuffer = AllocateZeroPool(BufferSize);\r
     if (HandleBuffer == NULL) {\r
       return (EFI_OUT_OF_RESOURCES);\r
     }\r
@@ -765,7 +870,6 @@ PerformMappingDelete(
 \r
 **/\r
 SHELL_STATUS\r
-EFIAPI\r
 AddMappingFromMapping(\r
   IN CONST CHAR16     *Map,\r
   IN CONST CHAR16     *SName\r
@@ -773,54 +877,40 @@ AddMappingFromMapping(
 {\r
   CONST EFI_DEVICE_PATH_PROTOCOL  *DevPath;\r
   EFI_STATUS                      Status;\r
+  CHAR16                          *NewSName;\r
+  RETURN_STATUS                   StrRetStatus;\r
+  \r
+  NewSName = AllocateCopyPool(StrSize(SName) + sizeof(CHAR16), SName);\r
+  if (NewSName == NULL) {\r
+    return (SHELL_OUT_OF_RESOURCES);\r
+  }\r
+  if (NewSName[StrLen(NewSName)-1] != L':') {\r
+    StrRetStatus = StrnCatS(NewSName, (StrSize(SName) + sizeof(CHAR16))/sizeof(CHAR16), L":", StrLen(L":"));\r
+    if (EFI_ERROR(StrRetStatus)) {\r
+      FreePool(NewSName);\r
+      return ((SHELL_STATUS) (StrRetStatus & (~MAX_BIT)));\r
+    }\r
+  }\r
 \r
-  if (StrStr(SName, L"*") != NULL\r
-    ||StrStr(SName, L"?") != NULL\r
-    ||StrStr(SName, L"[") != NULL\r
-    ||StrStr(SName, L"]") != NULL) {\r
+  if (!IsNumberLetterOnly(NewSName, StrLen(NewSName)-1)) {\r
+    FreePool(NewSName);\r
     return (SHELL_INVALID_PARAMETER);\r
   }\r
 \r
   DevPath = gEfiShellProtocol->GetDevicePathFromMap(Map);\r
   if (DevPath == NULL) {\r
+    FreePool(NewSName);\r
     return (SHELL_INVALID_PARAMETER);\r
   }\r
 \r
-  Status = gEfiShellProtocol->SetMap(DevPath, SName);\r
+  Status = gEfiShellProtocol->SetMap(DevPath, NewSName);\r
+  FreePool(NewSName);\r
   if (EFI_ERROR(Status)) {\r
     return (SHELL_DEVICE_ERROR);\r
   }\r
   return (SHELL_SUCCESS);\r
 }\r
 \r
-/**\r
-  function to determine if a string has only numbers and letters\r
-\r
-  This is useful for such things as Map names which can only be letters and numbers\r
-\r
-  @param[in] String       pointer to the string to analyze\r
-\r
-  @retval TRUE            String has only numbers and letters\r
-  @retval FALSE           String has at least one other character.\r
-**/\r
-BOOLEAN\r
-EFIAPI\r
-IsNumberLetterOnly(\r
-  IN CONST CHAR16 *String\r
-  )\r
-{\r
-  while(String != NULL && *String != CHAR_NULL) {\r
-    if (! ((*String >= L'a' && *String <= L'z') ||\r
-           (*String >= L'A' && *String <= L'Z') ||\r
-           (*String >= L'0' && *String <= L'9'))\r
-        ){\r
-      return (FALSE);\r
-    }\r
-    String++;\r
-  }\r
-  return (TRUE);\r
-}\r
-\r
 /**\r
   function to add a mapping from an EFI_HANDLE.\r
 \r
@@ -836,7 +926,6 @@ IsNumberLetterOnly(
 \r
 **/\r
 SHELL_STATUS\r
-EFIAPI\r
 AddMappingFromHandle(\r
   IN CONST EFI_HANDLE Handle,\r
   IN CONST CHAR16     *SName\r
@@ -844,8 +933,23 @@ AddMappingFromHandle(
 {\r
   EFI_DEVICE_PATH_PROTOCOL  *DevPath;\r
   EFI_STATUS                Status;\r
+  CHAR16                    *NewSName;\r
+  RETURN_STATUS             StrRetStatus;\r
+  \r
+  NewSName = AllocateCopyPool(StrSize(SName) + sizeof(CHAR16), SName);\r
+  if (NewSName == NULL) {\r
+    return (SHELL_OUT_OF_RESOURCES);\r
+  }\r
+  if (NewSName[StrLen(NewSName)-1] != L':') {\r
+    StrRetStatus = StrnCatS(NewSName, (StrSize(SName) + sizeof(CHAR16))/sizeof(CHAR16), L":", StrLen(L":"));\r
+    if (EFI_ERROR(StrRetStatus)) {\r
+      FreePool(NewSName);\r
+      return ((SHELL_STATUS) (StrRetStatus & (~MAX_BIT)));\r
+    }\r
+  }\r
 \r
-  if (!IsNumberLetterOnly(SName)) {\r
+  if (!IsNumberLetterOnly(NewSName, StrLen(NewSName)-1)) {\r
+    FreePool(NewSName);\r
     return (SHELL_INVALID_PARAMETER);\r
   }\r
 \r
@@ -858,9 +962,11 @@ AddMappingFromHandle(
     EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
    );\r
   if (EFI_ERROR(Status)) {\r
+    FreePool(NewSName);\r
     return (SHELL_DEVICE_ERROR);\r
   }\r
-  Status = gEfiShellProtocol->SetMap(DevPath, SName);\r
+  Status = gEfiShellProtocol->SetMap(DevPath, NewSName);\r
+  FreePool(NewSName);\r
   if (EFI_ERROR(Status)) {\r
     return (SHELL_DEVICE_ERROR);\r
   }\r
@@ -879,6 +985,57 @@ STATIC CONST SHELL_PARAM_ITEM MapParamList[] = {
   {NULL, TypeMax}\r
   };\r
 \r
+/**\r
+  The routine issues dummy read for every physical block device to cause\r
+  the BlockIo re-installed if media change happened.\r
+**/\r
+VOID\r
+ProbeForMediaChange (\r
+  VOID\r
+  )\r
+{\r
+  EFI_STATUS                            Status;\r
+  UINTN                                 HandleCount;\r
+  EFI_HANDLE                            *Handles;\r
+  EFI_BLOCK_IO_PROTOCOL                 *BlockIo;\r
+  UINTN                                 Index;\r
+\r
+  gBS->LocateHandleBuffer (\r
+         ByProtocol,\r
+         &gEfiBlockIoProtocolGuid,\r
+         NULL,\r
+         &HandleCount,\r
+         &Handles\r
+         );\r
+  //\r
+  // Probe for media change for every physical block io\r
+  //\r
+  for (Index = 0; Index < HandleCount; Index++) {\r
+    Status = gBS->HandleProtocol (\r
+                    Handles[Index],\r
+                    &gEfiBlockIoProtocolGuid,\r
+                    (VOID **) &BlockIo\r
+                    );\r
+    if (!EFI_ERROR (Status)) {\r
+      if (!BlockIo->Media->LogicalPartition) {\r
+        //\r
+        // Per spec:\r
+        //   The function (ReadBlocks) must return EFI_NO_MEDIA or\r
+        //   EFI_MEDIA_CHANGED even if LBA, BufferSize, or Buffer are invalid so the caller can probe\r
+        //   for changes in media state.\r
+        //\r
+        BlockIo->ReadBlocks (\r
+                   BlockIo,\r
+                   BlockIo->Media->MediaId,\r
+                   0,\r
+                   0,\r
+                   NULL\r
+                   );\r
+      }\r
+    }\r
+  }\r
+}\r
+\r
 /**\r
   Function for 'map' command.\r
 \r
@@ -897,21 +1054,20 @@ ShellCommandRunMap (
   CHAR16        *ProblemParam;\r
   CONST CHAR16  *SName;\r
   CONST CHAR16  *Mapping;\r
-  EFI_HANDLE    MappingAsHandle;\r
-  CONST EFI_DEVICE_PATH_PROTOCOL *DevPath;\r
+  EFI_HANDLE    MapAsHandle;\r
   SHELL_STATUS  ShellStatus;\r
   BOOLEAN       SfoMode;\r
   BOOLEAN       ConstMode;\r
   BOOLEAN       NormlMode;\r
   CONST CHAR16  *Param1;\r
   CONST CHAR16  *TypeString;\r
+  UINTN         TempStringLength;\r
 \r
   ProblemParam  = NULL;\r
   Mapping       = NULL;\r
   SName         = NULL;\r
-  DevPath       = NULL;\r
   ShellStatus   = SHELL_SUCCESS;\r
-  MappingAsHandle = NULL;\r
+  MapAsHandle = NULL;\r
 \r
   //\r
   // initialize the shell lib (we must be in non-auto-init...)\r
@@ -928,7 +1084,7 @@ ShellCommandRunMap (
   Status = ShellCommandLineParse (MapParamList, &Package, &ProblemParam, TRUE);\r
   if (EFI_ERROR(Status)) {\r
     if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
-      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, ProblemParam);\r
+      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, L"map", ProblemParam);  \r
       FreePool(ProblemParam);\r
       ShellStatus = SHELL_INVALID_PARAMETER;\r
     } else {\r
@@ -944,7 +1100,7 @@ ShellCommandRunMap (
     if (ShellCommandLineGetFlag(Package, L"-?")) {\r
       ASSERT(FALSE);\r
     } else if (ShellCommandLineGetRawValue(Package, 3) != NULL) {\r
-      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel2HiiHandle);\r
+      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel2HiiHandle, L"map");  \r
       ShellStatus = SHELL_INVALID_PARAMETER;\r
     } else {\r
       //\r
@@ -958,29 +1114,26 @@ ShellCommandRunMap (
           || ShellCommandLineGetFlag(Package, L"-u")\r
           || ShellCommandLineGetFlag(Package, L"-t")\r
          ){\r
-          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CON), gShellLevel2HiiHandle);\r
+          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CON), gShellLevel2HiiHandle, L"map");  \r
           ShellStatus = SHELL_INVALID_PARAMETER;\r
         } else {\r
           SName = ShellCommandLineGetValue(Package, L"-d");\r
           if (SName != NULL) {\r
             Status = PerformMappingDelete(SName);\r
             if (EFI_ERROR(Status)) {\r
-              switch (Status) {\r
-                case EFI_ACCESS_DENIED:\r
-                  ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_AD), gShellLevel2HiiHandle);\r
-                  ShellStatus = SHELL_ACCESS_DENIED;\r
-                  break;\r
-                case EFI_NOT_FOUND:\r
-                  ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_MAP_NF), gShellLevel2HiiHandle, SName);\r
-                  ShellStatus = SHELL_INVALID_PARAMETER;\r
-                  break;\r
-                default:\r
-                  ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_UK), gShellLevel2HiiHandle, Status);\r
-                  ShellStatus = SHELL_UNSUPPORTED;\r
+              if (Status == EFI_ACCESS_DENIED) {\r
+                ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_AD), gShellLevel2HiiHandle, L"map");  \r
+                ShellStatus = SHELL_ACCESS_DENIED;\r
+              } else if (Status == EFI_NOT_FOUND) {\r
+                ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_MAP_NF), gShellLevel2HiiHandle, L"map", SName);  \r
+                ShellStatus = SHELL_INVALID_PARAMETER;\r
+              } else {\r
+                ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_UK), gShellLevel2HiiHandle, L"map", Status);  \r
+                ShellStatus = SHELL_UNSUPPORTED;\r
               }\r
             }\r
           } else {\r
-            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel2HiiHandle);\r
+            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel2HiiHandle, L"map");  \r
             ShellStatus = SHELL_INVALID_PARAMETER;\r
           }\r
         }\r
@@ -991,13 +1144,14 @@ ShellCommandRunMap (
                || ShellCommandLineGetFlag(Package, L"-u")\r
                || ShellCommandLineGetFlag(Package, L"-t")\r
               ){\r
+        ProbeForMediaChange ();\r
         if ( ShellCommandLineGetFlag(Package, L"-r")) {\r
           //\r
           // Do the reset\r
           //\r
           Status = ShellCommandCreateInitialMappingsAndPaths();\r
           if (EFI_ERROR(Status)) {\r
-            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_UK), gShellLevel2HiiHandle, Status);\r
+            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_UK), gShellLevel2HiiHandle, L"map", Status);  \r
             ShellStatus = SHELL_UNSUPPORTED;\r
           }\r
         }\r
@@ -1005,9 +1159,9 @@ ShellCommandRunMap (
           //\r
           // Do the Update\r
           //\r
-          Status = UpdateMapping();\r
+          Status = ShellCommandUpdateMapping ();\r
           if (EFI_ERROR(Status)) {\r
-            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_UK), gShellLevel2HiiHandle, Status);\r
+            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_UK), gShellLevel2HiiHandle, L"map", Status);  \r
             ShellStatus = SHELL_UNSUPPORTED;\r
           }\r
         }\r
@@ -1082,44 +1236,56 @@ ShellCommandRunMap (
            );\r
         } else {\r
           if (ShellIsHexOrDecimalNumber(Mapping, TRUE, FALSE)) {\r
-            MappingAsHandle = ConvertHandleIndexToHandle(StrHexToUintn(Mapping));\r
+            MapAsHandle = ConvertHandleIndexToHandle(ShellStrToUintn(Mapping));\r
           } else {\r
-            MappingAsHandle = NULL;\r
+            MapAsHandle = NULL;\r
           }\r
-          if (MappingAsHandle == NULL && Mapping[StrLen(Mapping)-1] != L':') {\r
-            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, Mapping);\r
+          if (MapAsHandle == NULL && Mapping[StrLen(Mapping)-1] != L':') {\r
+            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellLevel2HiiHandle, L"map", Mapping);  \r
             ShellStatus = SHELL_INVALID_PARAMETER;\r
           } else {\r
-            if (MappingAsHandle != NULL) {\r
-              ShellStatus = AddMappingFromHandle(MappingAsHandle, SName);\r
-            } else {\r
-              ShellStatus = AddMappingFromMapping(Mapping, SName);\r
+            TempStringLength = StrLen(SName);\r
+            if (!IsNumberLetterOnly(SName, TempStringLength-(SName[TempStringLength-1]==L':'?1:0))) {\r
+              ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellLevel2HiiHandle, L"map", SName);\r
+              ShellStatus = SHELL_INVALID_PARAMETER;\r
             }\r
-            if (ShellStatus != SHELL_SUCCESS) {\r
-              switch (ShellStatus) {\r
-                case SHELL_ACCESS_DENIED:\r
-                  ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_AD), gShellLevel2HiiHandle);\r
-                  break;\r
-                case SHELL_INVALID_PARAMETER:\r
-                  ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellLevel2HiiHandle);\r
-                  break;\r
-                default:\r
-                  ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_UK), gShellLevel2HiiHandle, ShellStatus|MAX_BIT);\r
+\r
+            if (ShellStatus == SHELL_SUCCESS) {\r
+              if (MapAsHandle != NULL) {\r
+                ShellStatus = AddMappingFromHandle(MapAsHandle, SName);\r
+              } else {\r
+                ShellStatus = AddMappingFromMapping(Mapping, SName);\r
               }\r
-            } else {\r
-              //\r
-              // now do the display...\r
-              //\r
-              ShellStatus = PerformMappingDisplay(\r
-                FALSE,\r
-                FALSE,\r
-                FALSE,\r
-                NULL,\r
-                SfoMode,\r
-                SName,\r
-                TRUE\r
-               );\r
-            } // we were sucessful so do an output\r
+\r
+              if (ShellStatus != SHELL_SUCCESS) {\r
+                switch (ShellStatus) {\r
+                  case SHELL_ACCESS_DENIED:\r
+                    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_AD), gShellLevel2HiiHandle, L"map");\r
+                    break;\r
+                  case SHELL_INVALID_PARAMETER:\r
+                    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellLevel2HiiHandle, L"map", Mapping);\r
+                    break;\r
+                  case SHELL_DEVICE_ERROR:\r
+                    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_MAP_NOF), gShellLevel2HiiHandle, L"map", Mapping);\r
+                    break;\r
+                  default:\r
+                    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_UK), gShellLevel2HiiHandle, L"map", ShellStatus|MAX_BIT);\r
+                }\r
+              } else {\r
+                //\r
+                // now do the display...\r
+                //\r
+                ShellStatus = PerformMappingDisplay(\r
+                  FALSE,\r
+                  FALSE,\r
+                  FALSE,\r
+                  NULL,\r
+                  SfoMode,\r
+                  SName,\r
+                  TRUE\r
+                 );\r
+              } // we were sucessful so do an output\r
+            }\r
           } // got a valid map target\r
         } // got 2 variables\r
       } // we are adding a mapping\r