ShellPkg/Library/UefiShellDriver1CommandsLib/Disconnect.c

   1 /** @file
   2   Main file for Disconnect shell Driver1 function.
   3
   4   Copyright (c) 2015, Hewlett-Packard Development Company, L.P.<BR>
   5   Copyright (c) 2010 - 2012, Intel Corporation. All rights reserved.<BR>
   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 "UefiShellDriver1CommandsLib.h"
  17
  18 STATIC CONST SHELL_PARAM_ITEM ParamList[] = {
  19   {L"-r", TypeFlag},
  20   {NULL, TypeMax}
  21   };
  22
  23 /**
  24   Disconnect everything.
  25
  26   @retval EFI_SUCCESS     The operation was successful.
  27 **/
  28 EFI_STATUS
  29 EFIAPI
  30 DisconnectAll(
  31   VOID
  32   )
  33 {
  34   //
  35   // Stolen from UEFI 2.3 spec (May 2009 version)
  36   // Pages 171/172
  37   // Removed gBS local definition
  38   //
  39
  40   //
  41   // Disconnect All Handles Example
  42   // The following example recusively disconnects all drivers from all
  43   // controllers in a platform.
  44   //
  45   EFI_STATUS Status;
  46 //  EFI_BOOT_SERVICES *gBS;
  47   UINTN HandleCount;
  48   EFI_HANDLE *HandleBuffer;
  49   UINTN HandleIndex;
  50   //
  51   // Retrieve the list of all handles from the handle database
  52   //
  53   Status = gBS->LocateHandleBuffer (
  54     AllHandles,
  55     NULL,
  56     NULL,
  57     &HandleCount,
  58     &HandleBuffer
  59    );
  60   if (!EFI_ERROR (Status)) {
  61     for (HandleIndex = 0; HandleIndex < HandleCount; HandleIndex++) {
  62       Status = gBS->DisconnectController (
  63         HandleBuffer[HandleIndex],
  64         NULL,
  65         NULL
  66        );
  67     }
  68     gBS->FreePool(HandleBuffer);
  69     //
  70     // end of stealing
  71     //
  72   }
  73   return (EFI_SUCCESS);
  74 }
  75
  76 /**
  77   Function for 'disconnect' command.
  78
  79   @param[in] ImageHandle  Handle to the Image (NULL if Internal).
  80   @param[in] SystemTable  Pointer to the System Table (NULL if Internal).
  81 **/
  82 SHELL_STATUS
  83 EFIAPI
  84 ShellCommandRunDisconnect (
  85   IN EFI_HANDLE        ImageHandle,
  86   IN EFI_SYSTEM_TABLE  *SystemTable
  87   )
  88 {
  89   EFI_STATUS          Status;
  90   LIST_ENTRY          *Package;
  91   CHAR16              *ProblemParam;
  92   SHELL_STATUS        ShellStatus;
  93   CONST CHAR16        *Param1;
  94   CONST CHAR16        *Param2;
  95   CONST CHAR16        *Param3;
  96   EFI_HANDLE          Handle1;
  97   EFI_HANDLE          Handle2;
  98   EFI_HANDLE          Handle3;
  99   UINT64              Intermediate1;
 100   UINT64              Intermediate2;
 101   UINT64              Intermediate3;
 102
 103   ShellStatus         = SHELL_SUCCESS;
 104
 105   //
 106   // initialize the shell lib (we must be in non-auto-init...)
 107   //
 108   Status = ShellInitialize();
 109   ASSERT_EFI_ERROR(Status);
 110
 111   Status = CommandInit();
 112   ASSERT_EFI_ERROR(Status);
 113
 114   //
 115   // parse the command line
 116   //
 117   Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);
 118   if (EFI_ERROR(Status)) {
 119     if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {
 120       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDriver1HiiHandle, L"disconnect", ProblemParam);
 121       FreePool(ProblemParam);
 122       ShellStatus = SHELL_INVALID_PARAMETER;
 123     } else {
 124       ASSERT(FALSE);
 125     }
 126   } else {
 127     if (ShellCommandLineGetFlag(Package, L"-r")){
 128       if (ShellCommandLineGetCount(Package) > 1){
 129         ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDriver1HiiHandle, L"disconnect");
 130         ShellStatus = SHELL_INVALID_PARAMETER;
 131       } else if (ShellCommandLineGetCount(Package) < 1) {
 132         ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDriver1HiiHandle, L"disconnect");
 133         ShellStatus = SHELL_INVALID_PARAMETER;
 134       } else {
 135         Status = DisconnectAll();
 136       }
 137     } else {
 138       if (ShellCommandLineGetCount(Package) > 4){
 139         ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDriver1HiiHandle, L"disconnect");
 140         ShellStatus = SHELL_INVALID_PARAMETER;
 141       } else if (ShellCommandLineGetCount(Package) < 2) {
 142         ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDriver1HiiHandle, L"disconnect");
 143         ShellStatus = SHELL_INVALID_PARAMETER;
 144       } else {
 145         //
 146         // must have between 1 and 3 handles passed in ...
 147         //
 148         Param1  = ShellCommandLineGetRawValue(Package, 1);
 149         Param2  = ShellCommandLineGetRawValue(Package, 2);
 150         Param3  = ShellCommandLineGetRawValue(Package, 3);
 151         ShellConvertStringToUint64(Param1, &Intermediate1, TRUE, FALSE);
 152         Handle1 = Param1!=NULL?ConvertHandleIndexToHandle((UINTN)Intermediate1):NULL;
 153         ShellConvertStringToUint64(Param2, &Intermediate2, TRUE, FALSE);
 154         Handle2 = Param2!=NULL?ConvertHandleIndexToHandle((UINTN)Intermediate2):NULL;
 155         ShellConvertStringToUint64(Param3, &Intermediate3, TRUE, FALSE);
 156         Handle3 = Param3!=NULL?ConvertHandleIndexToHandle((UINTN)Intermediate3):NULL;
 157
 158         if (Param1 != NULL && Handle1 == NULL) {
 159           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_HANDLE), gShellDriver1HiiHandle, L"disconnect", Param1);
 160           ShellStatus = SHELL_INVALID_PARAMETER;
 161         } else if (Param2 != NULL && Handle2 == NULL) {
 162           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_HANDLE), gShellDriver1HiiHandle, L"disconnect", Param2);
 163           ShellStatus = SHELL_INVALID_PARAMETER;
 164         } else if (Param3 != NULL && Handle3 == NULL) {
 165           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_HANDLE), gShellDriver1HiiHandle, L"disconnect", Param3);
 166           ShellStatus = SHELL_INVALID_PARAMETER;
 167         } else if (Handle2 != NULL && EFI_ERROR(gBS->OpenProtocol(Handle2, &gEfiDriverBindingProtocolGuid, NULL, gImageHandle, NULL, EFI_OPEN_PROTOCOL_TEST_PROTOCOL))) {
 168           ASSERT(Param2 != NULL);
 169           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_HANDLE_NOT), gShellDriver1HiiHandle, L"disconnect", ShellStrToUintn(Param2), L"driver handle");
 170           ShellStatus = SHELL_INVALID_PARAMETER;
 171         } else {
 172           ASSERT(Param1 != NULL);
 173           Status = gBS->DisconnectController(Handle1, Handle2, Handle3);
 174           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_3P_RESULT), gShellDriver1HiiHandle, L"Disconnect", (UINTN)Intermediate1, (UINTN)Intermediate2, (UINTN)Intermediate3, Status);
 175         }
 176       }
 177     }
 178   }
 179   if (ShellStatus == SHELL_SUCCESS) {
 180     if (Status == EFI_SECURITY_VIOLATION) {
 181       ShellStatus = SHELL_SECURITY_VIOLATION;
 182     } else if (Status == EFI_INVALID_PARAMETER) {
 183       ShellStatus = SHELL_INVALID_PARAMETER;
 184     } else if (EFI_ERROR(Status)) {
 185       ShellStatus = SHELL_NOT_FOUND;
 186     }
 187   }
 188   return (ShellStatus);
 189 }