Adding Driver1 profile commands to the UEFI Shell 2.0.

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

1 files changed, 220 insertions, 0 deletions

diff --git a/ShellPkg/Library/UefiShellDriver1CommandsLib/Unload.c b/ShellPkg/Library/UefiShellDriver1CommandsLib/Unload.c
new file mode 100644
index 0000000000..239ac4aae2
--- /dev/null
+++ b/ShellPkg/Library/UefiShellDriver1CommandsLib/Unload.c
@@ -0,0 +1,220 @@
+/** @file

+  Main file for Unload shell Driver1 function.

+

+  Copyright (c) 2010, 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

+  http://opensource.org/licenses/bsd-license.php

+

+  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,

+  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.

+

+**/

+

+#include "UefiShellDriver1CommandsLib.h"

+

+/**

+  Function to translate the EFI_MEMORY_TYPE into a string.

+

+  @param[in] Memory     The memory type.

+

+  @retval               A string representation of the type allocated from BS Pool.

+**/

+CHAR16*

+EFIAPI

+ConvertMemoryType (

+  IN CONST EFI_MEMORY_TYPE Memory

+  )

+{

+  CHAR16 *RetVal;

+  RetVal = NULL;

+

+  switch (Memory) {

+  case EfiReservedMemoryType:       StrnCatGrow(&RetVal, NULL, L"EfiReservedMemoryType", 0);        break;

+  case EfiLoaderCode:               StrnCatGrow(&RetVal, NULL, L"EfiLoaderCode", 0);                break;

+  case EfiLoaderData:               StrnCatGrow(&RetVal, NULL, L"EfiLoaderData", 0);                break;

+  case EfiBootServicesCode:         StrnCatGrow(&RetVal, NULL, L"EfiBootServicesCode", 0);          break;

+  case EfiBootServicesData:         StrnCatGrow(&RetVal, NULL, L"EfiBootServicesData", 0);          break;

+  case EfiRuntimeServicesCode:      StrnCatGrow(&RetVal, NULL, L"EfiRuntimeServicesCode", 0);       break;

+  case EfiRuntimeServicesData:      StrnCatGrow(&RetVal, NULL, L"EfiRuntimeServicesData", 0);       break;

+  case EfiConventionalMemory:       StrnCatGrow(&RetVal, NULL, L"EfiConventionalMemory", 0);        break;

+  case EfiUnusableMemory:           StrnCatGrow(&RetVal, NULL, L"EfiUnusableMemory", 0);            break;

+  case EfiACPIReclaimMemory:        StrnCatGrow(&RetVal, NULL, L"EfiACPIReclaimMemory", 0);         break;

+  case EfiACPIMemoryNVS:            StrnCatGrow(&RetVal, NULL, L"EfiACPIMemoryNVS", 0);             break;

+  case EfiMemoryMappedIO:           StrnCatGrow(&RetVal, NULL, L"EfiMemoryMappedIO", 0);            break;

+  case EfiMemoryMappedIOPortSpace:  StrnCatGrow(&RetVal, NULL, L"EfiMemoryMappedIOPortSpace", 0);   break;

+  case EfiPalCode:                  StrnCatGrow(&RetVal, NULL, L"EfiPalCode", 0);                   break;

+  case EfiMaxMemoryType:            StrnCatGrow(&RetVal, NULL, L"EfiMaxMemoryType", 0);             break;

+  default: ASSERT(FALSE);

+  }

+  return (RetVal);

+}

+

+/**

+  Function to dump LoadedImage info from TheHandle.

+

+  @param[in] TheHandle              The handle to dump info from.

+

+  @retval EFI_SUCCESS               The info was dumped.

+  @retval EFI_INVALID_PARAMETER     The handle did not have LoadedImage

+**/

+EFI_STATUS

+EFIAPI

+DumpLoadedImageProtocolInfo (

+  IN EFI_HANDLE   TheHandle

+  )

+{

+  EFI_LOADED_IMAGE_PROTOCOL         *Image;

+  EFI_STATUS                        Status;

+  EFI_DEVICE_PATH_TO_TEXT_PROTOCOL  *DevicePathToText;

+  CHAR16                            *DevicePathText;

+  CHAR16                            *CodeTypeText;

+  CHAR16                            *DataTypeText;

+  CHAR8                             *PdbPointer;

+

+  Image = NULL;

+

+  Status = gBS->OpenProtocol(TheHandle, &gEfiLoadedImageProtocolGuid, (VOID**)&Image, gImageHandle, NULL, EFI_OPEN_PROTOCOL_GET_PROTOCOL);

+  if (EFI_ERROR(Status)) {

+    return (EFI_INVALID_PARAMETER);

+  }

+

+  Status = gBS->LocateProtocol(

+    &gEfiDevicePathToTextProtocolGuid,

+    NULL,

+    (VOID**)&DevicePathToText);

+  //

+  // we now have the device path to text protocol

+  //

+  if (!EFI_ERROR(Status)) {

+    DevicePathText = DevicePathToText->ConvertDevicePathToText(Image->FilePath, TRUE, TRUE);

+  } else {

+    DevicePathText = NULL;

+  }

+

+  CodeTypeText = ConvertMemoryType(Image->ImageCodeType);

+  DataTypeText = ConvertMemoryType(Image->ImageDataType);

+  PdbPointer = (CHAR8*)PeCoffLoaderGetPdbPointer(Image->ImageBase);

+  ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_UNLOAD_VERBOSE), gShellDriver1HiiHandle,

+    ConvertHandleToHandleIndex(TheHandle),

+    TheHandle,

+    Image,

+    Image->ParentHandle,

+    Image->SystemTable,

+    Image->DeviceHandle,

+    DevicePathText,

+    PdbPointer,

+    Image->ImageBase,

+    Image->ImageSize,

+    CodeTypeText,

+    DataTypeText

+    );

+

+  SHELL_FREE_NON_NULL(DevicePathText);

+  SHELL_FREE_NON_NULL(CodeTypeText);

+  SHELL_FREE_NON_NULL(DataTypeText);

+

+  return (EFI_SUCCESS);

+}

+

+STATIC CONST SHELL_PARAM_ITEM ParamList[] = {

+  {L"-n", TypeFlag},

+  {L"-v", TypeFlag},

+  {L"-verbose", TypeFlag},

+  {NULL, TypeMax}

+  };

+

+/**

+  Function for 'unload' command.

+

+  @param[in] ImageHandle  Handle to the Image (NULL if Internal).

+  @param[in] SystemTable  Pointer to the System Table (NULL if Internal).

+**/

+SHELL_STATUS

+EFIAPI

+ShellCommandRunUnload (

+  IN EFI_HANDLE        ImageHandle,

+  IN EFI_SYSTEM_TABLE  *SystemTable

+  )

+{

+  EFI_STATUS            Status;

+  LIST_ENTRY            *Package;

+  CHAR16                *ProblemParam;

+  SHELL_STATUS          ShellStatus;

+  EFI_HANDLE            TheHandle;

+  CONST CHAR16          *Param1;

+  SHELL_PROMPT_RESPONSE *Resp;

+

+  ShellStatus         = SHELL_SUCCESS;

+  Package             = NULL;

+  Resp                = NULL;

+

+  //

+  // initialize the shell lib (we must be in non-auto-init...)

+  //

+  Status = ShellInitialize();

+  ASSERT_EFI_ERROR(Status);

+

+  //

+  // parse the command line

+  //

+  Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);

+  if (EFI_ERROR(Status)) {

+    if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {

+      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDriver1HiiHandle, ProblemParam);

+      FreePool(ProblemParam);

+      ShellStatus = SHELL_INVALID_PARAMETER;

+    } else {

+      ASSERT(FALSE);

+    }

+  } else {

+    if (ShellCommandLineGetCount(Package) > 2){

+      //

+      // error for too many parameters

+      //

+      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDriver1HiiHandle);

+      ShellStatus = SHELL_INVALID_PARAMETER;

+    } else if (ShellCommandLineGetCount(Package) < 2) {

+      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDriver1HiiHandle);

+      ShellStatus = SHELL_INVALID_PARAMETER;

+    } else {

+      Param1 = ShellCommandLineGetRawValue(Package, 1);

+      if (Param1 == NULL || ConvertHandleIndexToHandle(StrHexToUintn(Param1)) == NULL){

+        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_HANDLE), gShellDriver1HiiHandle, Param1);

+        ShellStatus = SHELL_INVALID_PARAMETER;

+      } else {

+        TheHandle = ConvertHandleIndexToHandle(StrHexToUintn(Param1));

+        ASSERT(TheHandle != NULL);

+        if (ShellCommandLineGetFlag(Package, L"-v") || ShellCommandLineGetFlag(Package, L"-verbose")) {

+          DumpLoadedImageProtocolInfo(TheHandle);

+        }

+        

+        if (!ShellCommandLineGetFlag(Package, L"-n")) {

+          Status = ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN(STR_UNLOAD_CONF), gShellDriver1HiiHandle, StrHexToUintn(Param1));

+          Status = ShellPromptForResponse(ShellPromptResponseTypeYesNo, NULL, (VOID**)&Resp);

+        }

+        if (ShellCommandLineGetFlag(Package, L"-n") || (Resp != NULL && *Resp == ShellPromptResponseYes)) {

+          Status = gBS->UnloadImage(TheHandle);

+          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HANDLE_RESULT), gShellDriver1HiiHandle, L"Unload", StrHexToUintn(Param1), Status);

+        }

+        SHELL_FREE_NON_NULL(Resp);

+      }

+    }

+  }

+  if (ShellStatus == SHELL_SUCCESS) {

+    if (Status == EFI_SECURITY_VIOLATION) {

+      ShellStatus = SHELL_SECURITY_VIOLATION;

+    } else if (Status == EFI_INVALID_PARAMETER) {

+      ShellStatus = SHELL_INVALID_PARAMETER;

+    } else if (EFI_ERROR(Status)) {

+      ShellStatus = SHELL_NOT_FOUND;

+    }

+  }

+

+  if (Package != NULL) {

+    ShellCommandLineFreeVarList(Package);

+  }

+

+  return (ShellStatus);

+}