ShellPkg/Library/UefiShellDriver1CommandsLib/Disconnect.c

   1 /** @file
   2   Main file for Disconnect shell Driver1 function.
   3
   4   (C) Copyright 2016 Hewlett Packard Enterprise Development LP<BR>
   5   (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>
   6   Copyright (c) 2010 - 2012, Intel Corporation. All rights reserved.<BR>
   7   This program and the accompanying materials
   8   are licensed and made available under the terms and conditions of the BSD License
   9   which accompanies this distribution.  The full text of the license may be found at
  10   http://opensource.org/licenses/bsd-license.php
  11
  12   THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
  13   WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
  14
  15 **/
  16
  17 #include "UefiShellDriver1CommandsLib.h"
  18
  19 STATIC CONST SHELL_PARAM_ITEM ParamList[] = {
  20   {L"-r", TypeFlag},
  21   {L"-nc", TypeFlag},
  22   {NULL, TypeMax}
  23   };
  24
  25 /**
  26   Disconnect everything.
  27
  28   @retval EFI_SUCCESS     The operation was successful.
  29 **/
  30 EFI_STATUS
  31 EFIAPI
  32 DisconnectAll(
  33   VOID
  34   )
  35 {
  36   //
  37   // Stolen from UEFI 2.3 spec (May 2009 version)
  38   // Pages 171/172
  39   // Removed gBS local definition
  40   //
  41
  42   //
  43   // Disconnect All Handles Example
  44   // The following example recusively disconnects all drivers from all
  45   // controllers in a platform.
  46   //
  47   EFI_STATUS Status;
  48 //  EFI_BOOT_SERVICES *gBS;
  49   UINTN HandleCount;
  50   EFI_HANDLE *HandleBuffer;
  51   UINTN HandleIndex;
  52   //
  53   // Retrieve the list of all handles from the handle database
  54   //
  55   Status = gBS->LocateHandleBuffer (
  56     AllHandles,
  57     NULL,
  58     NULL,
  59     &HandleCount,
  60     &HandleBuffer
  61    );
  62   if (!EFI_ERROR (Status)) {
  63     for (HandleIndex = 0; HandleIndex < HandleCount; HandleIndex++) {
  64       Status = gBS->DisconnectController (
  65         HandleBuffer[HandleIndex],
  66         NULL,
  67         NULL
  68        );
  69     }
  70     gBS->FreePool(HandleBuffer);
  71     //
  72     // end of stealing
  73     //
  74   }
  75   return (EFI_SUCCESS);
  76 }
  77
  78 /**
  79   Function for 'disconnect' command.
  80
  81   @param[in] ImageHandle  Handle to the Image (NULL if Internal).
  82   @param[in] SystemTable  Pointer to the System Table (NULL if Internal).
  83 **/
  84 SHELL_STATUS
  85 EFIAPI
  86 ShellCommandRunDisconnect (
  87   IN EFI_HANDLE        ImageHandle,
  88   IN EFI_SYSTEM_TABLE  *SystemTable
  89   )
  90 {
  91   EFI_STATUS          Status;
  92   LIST_ENTRY          *Package;
  93   CHAR16              *ProblemParam;
  94   SHELL_STATUS        ShellStatus;
  95   CONST CHAR16        *Param1;
  96   CONST CHAR16        *Param2;
  97   CONST CHAR16        *Param3;
  98   EFI_HANDLE          Handle1;
  99   EFI_HANDLE          Handle2;
 100   EFI_HANDLE          Handle3;
 101   UINT64              Intermediate1;
 102   UINT64              Intermediate2;
 103   UINT64              Intermediate3;
 104
 105   ShellStatus         = SHELL_SUCCESS;
 106
 107   //
 108   // initialize the shell lib (we must be in non-auto-init...)
 109   //
 110   Status = ShellInitialize();
 111   ASSERT_EFI_ERROR(Status);
 112
 113   Status = CommandInit();
 114   ASSERT_EFI_ERROR(Status);
 115
 116   //
 117   // parse the command line
 118   //
 119   Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);
 120   if (EFI_ERROR(Status)) {
 121     if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {
 122       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDriver1HiiHandle, L"disconnect", ProblemParam);
 123       FreePool(ProblemParam);
 124       ShellStatus = SHELL_INVALID_PARAMETER;
 125     } else {
 126       ASSERT(FALSE);
 127     }
 128   } else {
 129     if (ShellCommandLineGetFlag(Package, L"-r")){
 130       if (ShellCommandLineGetCount(Package) > 1){
 131         ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDriver1HiiHandle, L"disconnect");
 132         ShellStatus = SHELL_INVALID_PARAMETER;
 133       } else if (ShellCommandLineGetCount(Package) < 1) {
 134         ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDriver1HiiHandle, L"disconnect");
 135         ShellStatus = SHELL_INVALID_PARAMETER;
 136       } else {
 137          Status = DisconnectAll ();
 138          //
 139          // Reconnect all consoles if -nc is not provided
 140          //
 141          if (!ShellCommandLineGetFlag (Package, L"-nc")){
 142            ShellConnectFromDevPaths (L"ConInDev");
 143            ShellConnectFromDevPaths (L"ConOutDev");
 144            ShellConnectFromDevPaths (L"ErrOutDev");
 145            ShellConnectFromDevPaths (L"ErrOut");
 146            ShellConnectFromDevPaths (L"ConIn");
 147            ShellConnectFromDevPaths (L"ConOut");
 148          }
 149       }
 150     } else if (ShellCommandLineGetFlag (Package, L"-nc")) {
 151       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDriver1HiiHandle, L"disconnect");
 152       ShellStatus = SHELL_INVALID_PARAMETER;
 153     } else {
 154       if (ShellCommandLineGetCount(Package) > 4){
 155         ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDriver1HiiHandle, L"disconnect");
 156         ShellStatus = SHELL_INVALID_PARAMETER;
 157       } else if (ShellCommandLineGetCount(Package) < 2) {
 158         ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDriver1HiiHandle, L"disconnect");
 159         ShellStatus = SHELL_INVALID_PARAMETER;
 160       } else {
 161         //
 162         // must have between 1 and 3 handles passed in ...
 163         //
 164         Param1  = ShellCommandLineGetRawValue(Package, 1);
 165         Param2  = ShellCommandLineGetRawValue(Package, 2);
 166         Param3  = ShellCommandLineGetRawValue(Package, 3);
 167         ShellConvertStringToUint64(Param1, &Intermediate1, TRUE, FALSE);
 168         Handle1 = Param1!=NULL?ConvertHandleIndexToHandle((UINTN)Intermediate1):NULL;
 169         ShellConvertStringToUint64(Param2, &Intermediate2, TRUE, FALSE);
 170         Handle2 = Param2!=NULL?ConvertHandleIndexToHandle((UINTN)Intermediate2):NULL;
 171         ShellConvertStringToUint64(Param3, &Intermediate3, TRUE, FALSE);
 172         Handle3 = Param3!=NULL?ConvertHandleIndexToHandle((UINTN)Intermediate3):NULL;
 173
 174         if (Param1 != NULL && Handle1 == NULL) {
 175           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_HANDLE), gShellDriver1HiiHandle, L"disconnect", Param1);
 176           ShellStatus = SHELL_INVALID_PARAMETER;
 177         } else if (Param2 != NULL && Handle2 == NULL) {
 178           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_HANDLE), gShellDriver1HiiHandle, L"disconnect", Param2);
 179           ShellStatus = SHELL_INVALID_PARAMETER;
 180         } else if (Param3 != NULL && Handle3 == NULL) {
 181           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_HANDLE), gShellDriver1HiiHandle, L"disconnect", Param3);
 182           ShellStatus = SHELL_INVALID_PARAMETER;
 183         } else if (Handle2 != NULL && EFI_ERROR(gBS->OpenProtocol(Handle2, &gEfiDriverBindingProtocolGuid, NULL, gImageHandle, NULL, EFI_OPEN_PROTOCOL_TEST_PROTOCOL))) {
 184           ASSERT(Param2 != NULL);
 185           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_HANDLE_NOT), gShellDriver1HiiHandle, L"disconnect", ShellStrToUintn(Param2), L"driver handle");
 186           ShellStatus = SHELL_INVALID_PARAMETER;
 187         } else {
 188           ASSERT(Param1 != NULL);
 189           Status = gBS->DisconnectController(Handle1, Handle2, Handle3);
 190           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_3P_RESULT), gShellDriver1HiiHandle, L"Disconnect", (UINTN)Intermediate1, (UINTN)Intermediate2, (UINTN)Intermediate3, Status);
 191         }
 192       }
 193     }
 194   }
 195   if (ShellStatus == SHELL_SUCCESS) {
 196     if (Status == EFI_SECURITY_VIOLATION) {
 197       ShellStatus = SHELL_SECURITY_VIOLATION;
 198     } else if (Status == EFI_INVALID_PARAMETER) {
 199       ShellStatus = SHELL_INVALID_PARAMETER;
 200     } else if (EFI_ERROR(Status)) {
 201       ShellStatus = SHELL_NOT_FOUND;
 202     }
 203   }
 204   return (ShellStatus);
 205 }