Contributed-under: TianoCore Contribution Agreement 1.0

Signed-off-by: Michael Kinney <michael.d.kinney@intel.com>
Reviewed-by: Jaben Carsey <Jaben.carsey@intel.com>

1) ShellPkg/Library/UefiHandleParsingLib
    a. UefiHandleParsingLib.c – ConvertHandleToHandleIndex() and ConvertHandleIndexToHandle()
        i. Update to work correctly when handles are destroyed due to driver disconnect operations.  Same handle index is never reused.
    b. UefiHandleParsingLib.c – ParseHandleDatabaseByRelationshipWithType() and ParseHandleDatabaseForChildControllers()
        i. Expand to handle Service Binding Protocol usage such as Network stack.
2) ShellPkg/Library/UefiShellDriver1CommandsLib
    a. DevTree.c – Fix bug where the same handle is shown more than once.
    b. Dh.c – Fix use of GetDriverName() and GetDriverImageName().  The status returned must always be evaluated. 
    c. Disconnect.c – Remove requirement that a handle being disconnected must support Device Path Protocol.  This prevents driver model handles without device paths from being disconnected (i.e. Network Stack)
    d. OpenInfo.c – Fix bug when showing open info about a ControllerHandle that has been destroyed due to a disconnect or an unload operation.
    e. UefiShellDriver1CommandsLib.uni – Fix “drivers” command formatting when handle indexes have more than 2 characters.


git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@13786 6f19259b-4bc3-4df7-8a09-765794883524

5 files changed, 54 insertions, 30 deletions

diff --git a/ShellPkg/Library/UefiShellDriver1CommandsLib/DevTree.c b/ShellPkg/Library/UefiShellDriver1CommandsLib/DevTree.c
index a66bdec..1548770 100644
--- a/ShellPkg/Library/UefiShellDriver1CommandsLib/DevTree.c
+++ b/ShellPkg/Library/UefiShellDriver1CommandsLib/DevTree.c
@@ -1,7 +1,7 @@
 /** @file

   Main file for DevTree shell Driver1 function.

 

-  Copyright (c) 2010 - 2011, Intel Corporation. All rights reserved.<BR>

+  Copyright (c) 2010 - 2012, Intel Corporation. All rights reserved.<BR>

   This program and the accompanying materials

   are licensed and made available under the terms and conditions of the BSD License

   which accompanies this distribution.  The full text of the license may be found at

@@ -84,23 +84,6 @@ DoDevTreeForHandle(
     return SHELL_SUCCESS;

   }

 

-  //

-  // If we are at the begining then we want root handles they have no parents and do have device path.

-  //

-  if (IndentCharCount == 0) {

-    Status = gBS->OpenProtocol (

-                  TheHandle,

-                  &gEfiDevicePathProtocolGuid,

-                  NULL,

-                  NULL,

-                  NULL,

-                  EFI_OPEN_PROTOCOL_TEST_PROTOCOL

-                 );

-    if (EFI_ERROR (Status)) {

-      return SHELL_SUCCESS;

-    }

-  }

-

   FormatString        = AllocateZeroPool(StrSize(HiiString) + (10)*sizeof(FormatString[0]));

 

   ASSERT(HiiString    != NULL);

@@ -169,6 +152,8 @@ ShellCommandRunDevTree (
   EFI_HANDLE          TheHandle;

   BOOLEAN             FlagD;

   UINT64              Intermediate;

+  UINTN               ParentControllerHandleCount;

+  EFI_HANDLE          *ParentControllerHandleBuffer;

 

   ShellStatus         = SHELL_SUCCESS;

   Status              = EFI_SUCCESS;

@@ -226,6 +211,39 @@ ShellCommandRunDevTree (
         if (TheHandle == NULL){

           break;

         }

+

+        //

+        // Skip handles that do not have device path protocol

+        //

+        Status = gBS->OpenProtocol (

+                      TheHandle,

+                      &gEfiDevicePathProtocolGuid,

+                      NULL,

+                      NULL,

+                      NULL,

+                      EFI_OPEN_PROTOCOL_TEST_PROTOCOL

+                     );

+        if (EFI_ERROR (Status)) {

+          continue;

+        }

+

+        //

+        // Skip handles that do have parents

+        //

+        ParentControllerHandleBuffer = NULL;

+        Status = PARSE_HANDLE_DATABASE_PARENTS (

+                   TheHandle,

+                   &ParentControllerHandleCount,

+                   &ParentControllerHandleBuffer

+                   );

+        SHELL_FREE_NON_NULL (ParentControllerHandleBuffer);

+        if (ParentControllerHandleCount > 0) {

+          continue;

+        }

+

+        //

+        // Start a devtree from TheHandle that has a device path and no parents

+        //

         ShellStatus = DoDevTreeForHandle(TheHandle, Language, FlagD, 0, HiiString);

       }

     } else {

diff --git a/ShellPkg/Library/UefiShellDriver1CommandsLib/Dh.c b/ShellPkg/Library/UefiShellDriver1CommandsLib/Dh.c
index 1b42b23..a0dc139 100644
--- a/ShellPkg/Library/UefiShellDriver1CommandsLib/Dh.c
+++ b/ShellPkg/Library/UefiShellDriver1CommandsLib/Dh.c
@@ -1,7 +1,7 @@
 /** @file

   Main file for Dh shell Driver1 function.

 

-  Copyright (c) 2010 - 2011, Intel Corporation. All rights reserved.<BR>

+  Copyright (c) 2010 - 2012, Intel Corporation. All rights reserved.<BR>

   This program and the accompanying materials

   are licensed and made available under the terms and conditions of the BSD License

   which accompanies this distribution.  The full text of the license may be found at

@@ -394,11 +394,14 @@ DisplayDriverModelHandle (
                   Language,

                   &DriverName

                   );

-        if (DriverName == NULL) {

+        if (EFI_ERROR (Status)) {

           Status = GetDriverImageName (

                     DriverBindingHandleBuffer[Index],

                     &DriverName

                     );

+          if (EFI_ERROR (Status)) {

+            DriverName = NULL;

+          }

         }

 

         if (Image) {

@@ -537,6 +540,9 @@ DisplayDriverModelHandle (
   }

 

   Status = GetDriverName (Handle, Language, &DriverName);

+  if (EFI_ERROR (Status)) {

+    DriverName = NULL;

+  }

 

   ShellPrintHiiEx(

     -1, 

@@ -548,11 +554,13 @@ DisplayDriverModelHandle (
     DriverName!=NULL?DriverName:L"<Unknown>"

     );

   SHELL_FREE_NON_NULL(DriverName);

-  DriverName = NULL;

   Status = GetDriverImageName (

             Handle,

             &DriverName

             );

+  if (EFI_ERROR (Status)) {

+    DriverName = NULL;

+  }

   ShellPrintHiiEx(

     -1, 

     -1, 

diff --git a/ShellPkg/Library/UefiShellDriver1CommandsLib/Disconnect.c b/ShellPkg/Library/UefiShellDriver1CommandsLib/Disconnect.c
index 96233f9..a33bed9 100644
--- a/ShellPkg/Library/UefiShellDriver1CommandsLib/Disconnect.c
+++ b/ShellPkg/Library/UefiShellDriver1CommandsLib/Disconnect.c
@@ -1,7 +1,7 @@
 /** @file

   Main file for Disconnect shell Driver1 function.

 

-  Copyright (c) 2010 - 2011, Intel Corporation. All rights reserved.<BR>

+  Copyright (c) 2010 - 2012, Intel Corporation. All rights reserved.<BR>

   This program and the accompanying materials

   are licensed and made available under the terms and conditions of the BSD License

   which accompanies this distribution.  The full text of the license may be found at

@@ -163,10 +163,6 @@ ShellCommandRunDisconnect (
         } else if (Param3 != NULL && Handle3 == NULL) {

           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_HANDLE), gShellDriver1HiiHandle, Param3);

           ShellStatus = SHELL_INVALID_PARAMETER;

-        } else if (EFI_ERROR(gBS->OpenProtocol(Handle1, &gEfiDevicePathProtocolGuid, NULL, gImageHandle, NULL, EFI_OPEN_PROTOCOL_TEST_PROTOCOL))) {

-          ASSERT(Param1 != NULL);

-          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_HANDLE_NOT), gShellDriver1HiiHandle, ShellStrToUintn(Param1), L"controller handle");

-          ShellStatus = SHELL_INVALID_PARAMETER;

         } else if (Handle2 != NULL && EFI_ERROR(gBS->OpenProtocol(Handle2, &gEfiDriverBindingProtocolGuid, NULL, gImageHandle, NULL, EFI_OPEN_PROTOCOL_TEST_PROTOCOL))) {

           ASSERT(Param2 != NULL);

           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_HANDLE_NOT), gShellDriver1HiiHandle, ShellStrToUintn(Param2), L"driver handle");

diff --git a/ShellPkg/Library/UefiShellDriver1CommandsLib/OpenInfo.c b/ShellPkg/Library/UefiShellDriver1CommandsLib/OpenInfo.c
index 3689483..2922f42 100644
--- a/ShellPkg/Library/UefiShellDriver1CommandsLib/OpenInfo.c
+++ b/ShellPkg/Library/UefiShellDriver1CommandsLib/OpenInfo.c
@@ -48,6 +48,7 @@ TraverseHandleDatabase (
   CHAR16                              *TempString;

   UINTN                               HandleIndex;

   CONST CHAR16                        *Name;

+  UINTN                               ControllerIndex;

 

   if (TheHandle == NULL) {

     return (EFI_INVALID_PARAMETER);

@@ -98,9 +99,10 @@ TraverseHandleDatabase (
                                                         OpenTypeString = StringDriverEx;  break;

             default:                                    OpenTypeString = StringUnknown;   break;

           }

-          HandleIndex = ConvertHandleToHandleIndex(OpenInfo[OpenInfoIndex].AgentHandle);

-          Name        = GetStringNameFromHandle(OpenInfo[OpenInfoIndex].AgentHandle, NULL);

-          if (OpenInfo[OpenInfoIndex].ControllerHandle!=NULL) {

+          HandleIndex     = ConvertHandleToHandleIndex(OpenInfo[OpenInfoIndex].AgentHandle);

+          Name            = GetStringNameFromHandle(OpenInfo[OpenInfoIndex].AgentHandle, NULL);

+          ControllerIndex = ConvertHandleToHandleIndex(OpenInfo[OpenInfoIndex].ControllerHandle);

+          if (ControllerIndex != 0) {

             ShellPrintHiiEx(

               -1,

               -1,

@@ -108,7 +110,7 @@ TraverseHandleDatabase (
               STRING_TOKEN(STR_OPENINFO_LINE),

               gShellDriver1HiiHandle,

               HandleIndex,

-              ConvertHandleToHandleIndex(OpenInfo[OpenInfoIndex].ControllerHandle),

+              ControllerIndex,

               OpenInfo[OpenInfoIndex].OpenCount,

               OpenTypeString,

               Name

diff --git a/ShellPkg/Library/UefiShellDriver1CommandsLib/UefiShellDriver1CommandsLib.uni b/ShellPkg/Library/UefiShellDriver1CommandsLib/UefiShellDriver1CommandsLib.uni
index 6c66979..9ffad7e 100644
--- a/ShellPkg/Library/UefiShellDriver1CommandsLib/UefiShellDriver1CommandsLib.uni
+++ b/ShellPkg/Library/UefiShellDriver1CommandsLib/UefiShellDriver1CommandsLib.uni