1 /** @file
2   Main file for Disconnect shell Driver1 function.
3 
4   (C) Copyright 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
DisconnectAll(VOID)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
ShellCommandRunDisconnect(IN EFI_HANDLE ImageHandle,IN EFI_SYSTEM_TABLE * SystemTable)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 }
190