blob: dc36db03495ad20de55a9b69bb0da3cdb42e37ca [file] [log] [blame]
jcarsey94b17fa2009-05-07 18:46:18 +00001/** @file
2 Provides interface to shell functionality for shell commands and applications.
3
Jaben Carsey75a5e2e2014-01-10 16:42:45 +00004 Copyright (c) 2006 - 2014, Intel Corporation. All rights reserved.<BR>
jcarsey1e6e84c2010-01-25 20:05:08 +00005 This program and the accompanying materials
jcarseyb3011f42010-01-11 21:49:04 +00006 are licensed and made available under the terms and conditions of the BSD License
7 which accompanies this distribution. The full text of the license may be found at
8 http://opensource.org/licenses/bsd-license.php
jcarsey94b17fa2009-05-07 18:46:18 +00009
jcarseyb3011f42010-01-11 21:49:04 +000010 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
jcarsey94b17fa2009-05-07 18:46:18 +000012
13**/
14
jcarseyb1f95a02009-06-16 00:23:19 +000015#include "UefiShellLib.h"
jcarseya405b862010-09-14 05:18:09 +000016#include <ShellBase.h>
jcarsey252d9452011-03-25 20:49:53 +000017#include <Library/SortLib.h>
jcarseyd2b45642009-05-11 18:02:16 +000018
jcarsey94b17fa2009-05-07 18:46:18 +000019#define FIND_XXXXX_FILE_BUFFER_SIZE (SIZE_OF_EFI_FILE_INFO + MAX_FILE_NAME_LEN)
20
jcarseyd2b45642009-05-11 18:02:16 +000021//
jcarseya405b862010-09-14 05:18:09 +000022// globals...
jcarseyd2b45642009-05-11 18:02:16 +000023//
24SHELL_PARAM_ITEM EmptyParamList[] = {
25 {NULL, TypeMax}
26 };
jcarseya405b862010-09-14 05:18:09 +000027SHELL_PARAM_ITEM SfoParamList[] = {
28 {L"-sfo", TypeFlag},
29 {NULL, TypeMax}
30 };
31EFI_SHELL_ENVIRONMENT2 *mEfiShellEnvironment2;
32EFI_SHELL_INTERFACE *mEfiShellInterface;
jcarsey366f81a2011-06-27 21:04:22 +000033EFI_SHELL_PROTOCOL *gEfiShellProtocol;
34EFI_SHELL_PARAMETERS_PROTOCOL *gEfiShellParametersProtocol;
jcarseya405b862010-09-14 05:18:09 +000035EFI_HANDLE mEfiShellEnvironment2Handle;
36FILE_HANDLE_FUNCTION_MAP FileFunctionMap;
jcarseyb3011f42010-01-11 21:49:04 +000037
jcarsey2247dde2009-11-09 18:08:58 +000038/**
39 Check if a Unicode character is a hexadecimal character.
40
jcarsey1e6e84c2010-01-25 20:05:08 +000041 This internal function checks if a Unicode character is a
jcarseya405b862010-09-14 05:18:09 +000042 numeric character. The valid hexadecimal characters are
jcarsey2247dde2009-11-09 18:08:58 +000043 L'0' to L'9', L'a' to L'f', or L'A' to L'F'.
44
jcarsey2247dde2009-11-09 18:08:58 +000045 @param Char The character to check against.
46
47 @retval TRUE If the Char is a hexadecmial character.
48 @retval FALSE If the Char is not a hexadecmial character.
49
50**/
51BOOLEAN
52EFIAPI
jcarsey969c7832010-01-13 16:46:33 +000053ShellIsHexaDecimalDigitCharacter (
jcarsey2247dde2009-11-09 18:08:58 +000054 IN CHAR16 Char
jcarseya405b862010-09-14 05:18:09 +000055 )
56{
jcarsey2247dde2009-11-09 18:08:58 +000057 return (BOOLEAN) ((Char >= L'0' && Char <= L'9') || (Char >= L'A' && Char <= L'F') || (Char >= L'a' && Char <= L'f'));
58}
jcarsey94b17fa2009-05-07 18:46:18 +000059
60/**
jcarseya405b862010-09-14 05:18:09 +000061 Check if a Unicode character is a decimal character.
62
63 This internal function checks if a Unicode character is a
64 decimal character. The valid characters are
65 L'0' to L'9'.
66
67
68 @param Char The character to check against.
69
70 @retval TRUE If the Char is a hexadecmial character.
71 @retval FALSE If the Char is not a hexadecmial character.
72
73**/
74BOOLEAN
75EFIAPI
76ShellIsDecimalDigitCharacter (
77 IN CHAR16 Char
78 )
79{
80 return (BOOLEAN) (Char >= L'0' && Char <= L'9');
81}
82
83/**
84 Helper function to find ShellEnvironment2 for constructor.
85
86 @param[in] ImageHandle A copy of the calling image's handle.
jcarseybeab0fc2011-10-10 17:26:25 +000087
88 @retval EFI_OUT_OF_RESOURCES Memory allocation failed.
jcarsey94b17fa2009-05-07 18:46:18 +000089**/
90EFI_STATUS
91EFIAPI
92ShellFindSE2 (
93 IN EFI_HANDLE ImageHandle
jcarseya405b862010-09-14 05:18:09 +000094 )
95{
jcarsey94b17fa2009-05-07 18:46:18 +000096 EFI_STATUS Status;
97 EFI_HANDLE *Buffer;
98 UINTN BufferSize;
99 UINTN HandleIndex;
100
101 BufferSize = 0;
102 Buffer = NULL;
jcarsey1e6e84c2010-01-25 20:05:08 +0000103 Status = gBS->OpenProtocol(ImageHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000104 &gEfiShellEnvironment2Guid,
105 (VOID **)&mEfiShellEnvironment2,
106 ImageHandle,
107 NULL,
108 EFI_OPEN_PROTOCOL_GET_PROTOCOL
jcarseya405b862010-09-14 05:18:09 +0000109 );
jcarsey94b17fa2009-05-07 18:46:18 +0000110 //
111 // look for the mEfiShellEnvironment2 protocol at a higher level
112 //
jcarseya405b862010-09-14 05:18:09 +0000113 if (EFI_ERROR (Status) || !(CompareGuid (&mEfiShellEnvironment2->SESGuid, &gEfiShellEnvironment2ExtGuid))){
jcarsey94b17fa2009-05-07 18:46:18 +0000114 //
115 // figure out how big of a buffer we need.
116 //
117 Status = gBS->LocateHandle (ByProtocol,
118 &gEfiShellEnvironment2Guid,
119 NULL, // ignored for ByProtocol
120 &BufferSize,
121 Buffer
jcarseya405b862010-09-14 05:18:09 +0000122 );
jcarsey2247dde2009-11-09 18:08:58 +0000123 //
124 // maybe it's not there???
125 //
126 if (Status == EFI_BUFFER_TOO_SMALL) {
jcarsey252d9452011-03-25 20:49:53 +0000127 Buffer = (EFI_HANDLE*)AllocateZeroPool(BufferSize);
jcarseybeab0fc2011-10-10 17:26:25 +0000128 if (Buffer == NULL) {
129 return (EFI_OUT_OF_RESOURCES);
130 }
jcarsey2247dde2009-11-09 18:08:58 +0000131 Status = gBS->LocateHandle (ByProtocol,
132 &gEfiShellEnvironment2Guid,
133 NULL, // ignored for ByProtocol
134 &BufferSize,
135 Buffer
jcarseya405b862010-09-14 05:18:09 +0000136 );
jcarsey2247dde2009-11-09 18:08:58 +0000137 }
jcarsey1cd45e72010-01-29 15:07:44 +0000138 if (!EFI_ERROR (Status) && Buffer != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000139 //
140 // now parse the list of returned handles
141 //
142 Status = EFI_NOT_FOUND;
143 for (HandleIndex = 0; HandleIndex < (BufferSize/sizeof(Buffer[0])); HandleIndex++) {
jcarsey1e6e84c2010-01-25 20:05:08 +0000144 Status = gBS->OpenProtocol(Buffer[HandleIndex],
jcarsey94b17fa2009-05-07 18:46:18 +0000145 &gEfiShellEnvironment2Guid,
146 (VOID **)&mEfiShellEnvironment2,
147 ImageHandle,
148 NULL,
149 EFI_OPEN_PROTOCOL_GET_PROTOCOL
jcarseya405b862010-09-14 05:18:09 +0000150 );
151 if (CompareGuid (&mEfiShellEnvironment2->SESGuid, &gEfiShellEnvironment2ExtGuid)) {
jcarsey94b17fa2009-05-07 18:46:18 +0000152 mEfiShellEnvironment2Handle = Buffer[HandleIndex];
153 Status = EFI_SUCCESS;
154 break;
155 }
156 }
157 }
158 }
159 if (Buffer != NULL) {
160 FreePool (Buffer);
161 }
162 return (Status);
163}
164
jcarsey252d9452011-03-25 20:49:53 +0000165/**
darylm503b0934ac2011-11-12 00:35:11 +0000166 Function to do most of the work of the constructor. Allows for calling
jcarseya405b862010-09-14 05:18:09 +0000167 multiple times without complete re-initialization.
168
169 @param[in] ImageHandle A copy of the ImageHandle.
170 @param[in] SystemTable A pointer to the SystemTable for the application.
jcarsey252d9452011-03-25 20:49:53 +0000171
172 @retval EFI_SUCCESS The operationw as successful.
jcarseya405b862010-09-14 05:18:09 +0000173**/
jcarsey94b17fa2009-05-07 18:46:18 +0000174EFI_STATUS
175EFIAPI
jcarseyd2b45642009-05-11 18:02:16 +0000176ShellLibConstructorWorker (
jcarsey94b17fa2009-05-07 18:46:18 +0000177 IN EFI_HANDLE ImageHandle,
178 IN EFI_SYSTEM_TABLE *SystemTable
jcarseya405b862010-09-14 05:18:09 +0000179 )
180{
181 EFI_STATUS Status;
jcarseyecd3d592009-12-07 18:05:00 +0000182
jcarsey94b17fa2009-05-07 18:46:18 +0000183 //
184 // UEFI 2.0 shell interfaces (used preferentially)
185 //
jcarseya405b862010-09-14 05:18:09 +0000186 Status = gBS->OpenProtocol(
187 ImageHandle,
188 &gEfiShellProtocolGuid,
jcarsey366f81a2011-06-27 21:04:22 +0000189 (VOID **)&gEfiShellProtocol,
jcarseya405b862010-09-14 05:18:09 +0000190 ImageHandle,
191 NULL,
192 EFI_OPEN_PROTOCOL_GET_PROTOCOL
193 );
jcarsey94b17fa2009-05-07 18:46:18 +0000194 if (EFI_ERROR(Status)) {
jcarseya405b862010-09-14 05:18:09 +0000195 //
196 // Search for the shell protocol
197 //
198 Status = gBS->LocateProtocol(
199 &gEfiShellProtocolGuid,
200 NULL,
jcarsey366f81a2011-06-27 21:04:22 +0000201 (VOID **)&gEfiShellProtocol
jcarseya405b862010-09-14 05:18:09 +0000202 );
203 if (EFI_ERROR(Status)) {
jcarsey366f81a2011-06-27 21:04:22 +0000204 gEfiShellProtocol = NULL;
jcarseya405b862010-09-14 05:18:09 +0000205 }
jcarsey94b17fa2009-05-07 18:46:18 +0000206 }
jcarseya405b862010-09-14 05:18:09 +0000207 Status = gBS->OpenProtocol(
208 ImageHandle,
209 &gEfiShellParametersProtocolGuid,
jcarsey366f81a2011-06-27 21:04:22 +0000210 (VOID **)&gEfiShellParametersProtocol,
jcarseya405b862010-09-14 05:18:09 +0000211 ImageHandle,
212 NULL,
213 EFI_OPEN_PROTOCOL_GET_PROTOCOL
214 );
jcarsey94b17fa2009-05-07 18:46:18 +0000215 if (EFI_ERROR(Status)) {
jcarsey366f81a2011-06-27 21:04:22 +0000216 gEfiShellParametersProtocol = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000217 }
218
jcarsey366f81a2011-06-27 21:04:22 +0000219 if (gEfiShellParametersProtocol == NULL || gEfiShellProtocol == NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000220 //
221 // Moved to seperate function due to complexity
222 //
223 Status = ShellFindSE2(ImageHandle);
224
225 if (EFI_ERROR(Status)) {
226 DEBUG((DEBUG_ERROR, "Status: 0x%08x\r\n", Status));
227 mEfiShellEnvironment2 = NULL;
228 }
jcarsey1e6e84c2010-01-25 20:05:08 +0000229 Status = gBS->OpenProtocol(ImageHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000230 &gEfiShellInterfaceGuid,
231 (VOID **)&mEfiShellInterface,
232 ImageHandle,
233 NULL,
234 EFI_OPEN_PROTOCOL_GET_PROTOCOL
jcarseya405b862010-09-14 05:18:09 +0000235 );
jcarsey94b17fa2009-05-07 18:46:18 +0000236 if (EFI_ERROR(Status)) {
237 mEfiShellInterface = NULL;
238 }
239 }
jcarseyc9d92df2010-02-03 15:37:54 +0000240
jcarsey94b17fa2009-05-07 18:46:18 +0000241 //
242 // only success getting 2 of either the old or new, but no 1/2 and 1/2
243 //
jcarsey1e6e84c2010-01-25 20:05:08 +0000244 if ((mEfiShellEnvironment2 != NULL && mEfiShellInterface != NULL) ||
jcarsey366f81a2011-06-27 21:04:22 +0000245 (gEfiShellProtocol != NULL && gEfiShellParametersProtocol != NULL) ) {
246 if (gEfiShellProtocol != NULL) {
247 FileFunctionMap.GetFileInfo = gEfiShellProtocol->GetFileInfo;
248 FileFunctionMap.SetFileInfo = gEfiShellProtocol->SetFileInfo;
249 FileFunctionMap.ReadFile = gEfiShellProtocol->ReadFile;
250 FileFunctionMap.WriteFile = gEfiShellProtocol->WriteFile;
251 FileFunctionMap.CloseFile = gEfiShellProtocol->CloseFile;
252 FileFunctionMap.DeleteFile = gEfiShellProtocol->DeleteFile;
253 FileFunctionMap.GetFilePosition = gEfiShellProtocol->GetFilePosition;
254 FileFunctionMap.SetFilePosition = gEfiShellProtocol->SetFilePosition;
255 FileFunctionMap.FlushFile = gEfiShellProtocol->FlushFile;
256 FileFunctionMap.GetFileSize = gEfiShellProtocol->GetFileSize;
jcarseyd2b45642009-05-11 18:02:16 +0000257 } else {
jcarseya405b862010-09-14 05:18:09 +0000258 FileFunctionMap.GetFileInfo = (EFI_SHELL_GET_FILE_INFO)FileHandleGetInfo;
259 FileFunctionMap.SetFileInfo = (EFI_SHELL_SET_FILE_INFO)FileHandleSetInfo;
260 FileFunctionMap.ReadFile = (EFI_SHELL_READ_FILE)FileHandleRead;
261 FileFunctionMap.WriteFile = (EFI_SHELL_WRITE_FILE)FileHandleWrite;
262 FileFunctionMap.CloseFile = (EFI_SHELL_CLOSE_FILE)FileHandleClose;
263 FileFunctionMap.DeleteFile = (EFI_SHELL_DELETE_FILE)FileHandleDelete;
264 FileFunctionMap.GetFilePosition = (EFI_SHELL_GET_FILE_POSITION)FileHandleGetPosition;
265 FileFunctionMap.SetFilePosition = (EFI_SHELL_SET_FILE_POSITION)FileHandleSetPosition;
266 FileFunctionMap.FlushFile = (EFI_SHELL_FLUSH_FILE)FileHandleFlush;
267 FileFunctionMap.GetFileSize = (EFI_SHELL_GET_FILE_SIZE)FileHandleGetSize;
jcarseyd2b45642009-05-11 18:02:16 +0000268 }
jcarsey94b17fa2009-05-07 18:46:18 +0000269 return (EFI_SUCCESS);
270 }
271 return (EFI_NOT_FOUND);
272}
jcarseyd2b45642009-05-11 18:02:16 +0000273/**
274 Constructor for the Shell library.
275
276 Initialize the library and determine if the underlying is a UEFI Shell 2.0 or an EFI shell.
277
278 @param ImageHandle the image handle of the process
279 @param SystemTable the EFI System Table pointer
280
281 @retval EFI_SUCCESS the initialization was complete sucessfully
282 @return others an error ocurred during initialization
283**/
284EFI_STATUS
285EFIAPI
286ShellLibConstructor (
287 IN EFI_HANDLE ImageHandle,
288 IN EFI_SYSTEM_TABLE *SystemTable
jcarseya405b862010-09-14 05:18:09 +0000289 )
290{
jcarseyd2b45642009-05-11 18:02:16 +0000291 mEfiShellEnvironment2 = NULL;
jcarsey366f81a2011-06-27 21:04:22 +0000292 gEfiShellProtocol = NULL;
293 gEfiShellParametersProtocol = NULL;
jcarseyd2b45642009-05-11 18:02:16 +0000294 mEfiShellInterface = NULL;
295 mEfiShellEnvironment2Handle = NULL;
296
jcarseyd2b45642009-05-11 18:02:16 +0000297 //
298 // verify that auto initialize is not set false
jcarsey1e6e84c2010-01-25 20:05:08 +0000299 //
jcarseyd2b45642009-05-11 18:02:16 +0000300 if (PcdGetBool(PcdShellLibAutoInitialize) == 0) {
301 return (EFI_SUCCESS);
302 }
jcarsey1e6e84c2010-01-25 20:05:08 +0000303
jcarseyd2b45642009-05-11 18:02:16 +0000304 return (ShellLibConstructorWorker(ImageHandle, SystemTable));
305}
jcarsey94b17fa2009-05-07 18:46:18 +0000306
307/**
jcarseya405b862010-09-14 05:18:09 +0000308 Destructor for the library. free any resources.
309
310 @param[in] ImageHandle A copy of the ImageHandle.
311 @param[in] SystemTable A pointer to the SystemTable for the application.
312
313 @retval EFI_SUCCESS The operation was successful.
314 @return An error from the CloseProtocol function.
jcarsey94b17fa2009-05-07 18:46:18 +0000315**/
316EFI_STATUS
317EFIAPI
318ShellLibDestructor (
319 IN EFI_HANDLE ImageHandle,
320 IN EFI_SYSTEM_TABLE *SystemTable
jcarseya405b862010-09-14 05:18:09 +0000321 )
322{
jcarsey94b17fa2009-05-07 18:46:18 +0000323 if (mEfiShellEnvironment2 != NULL) {
324 gBS->CloseProtocol(mEfiShellEnvironment2Handle==NULL?ImageHandle:mEfiShellEnvironment2Handle,
325 &gEfiShellEnvironment2Guid,
326 ImageHandle,
327 NULL);
jcarseyd2b45642009-05-11 18:02:16 +0000328 mEfiShellEnvironment2 = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000329 }
330 if (mEfiShellInterface != NULL) {
331 gBS->CloseProtocol(ImageHandle,
332 &gEfiShellInterfaceGuid,
333 ImageHandle,
jcarsey1e6e84c2010-01-25 20:05:08 +0000334 NULL);
jcarseyd2b45642009-05-11 18:02:16 +0000335 mEfiShellInterface = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000336 }
jcarsey366f81a2011-06-27 21:04:22 +0000337 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000338 gBS->CloseProtocol(ImageHandle,
339 &gEfiShellProtocolGuid,
340 ImageHandle,
jcarsey1e6e84c2010-01-25 20:05:08 +0000341 NULL);
jcarsey366f81a2011-06-27 21:04:22 +0000342 gEfiShellProtocol = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000343 }
jcarsey366f81a2011-06-27 21:04:22 +0000344 if (gEfiShellParametersProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000345 gBS->CloseProtocol(ImageHandle,
346 &gEfiShellParametersProtocolGuid,
347 ImageHandle,
jcarsey1e6e84c2010-01-25 20:05:08 +0000348 NULL);
jcarsey366f81a2011-06-27 21:04:22 +0000349 gEfiShellParametersProtocol = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000350 }
jcarseyd2b45642009-05-11 18:02:16 +0000351 mEfiShellEnvironment2Handle = NULL;
jcarseyecd3d592009-12-07 18:05:00 +0000352
jcarsey94b17fa2009-05-07 18:46:18 +0000353 return (EFI_SUCCESS);
354}
jcarseyd2b45642009-05-11 18:02:16 +0000355
356/**
357 This function causes the shell library to initialize itself. If the shell library
358 is already initialized it will de-initialize all the current protocol poitners and
359 re-populate them again.
360
361 When the library is used with PcdShellLibAutoInitialize set to true this function
362 will return EFI_SUCCESS and perform no actions.
363
364 This function is intended for internal access for shell commands only.
365
366 @retval EFI_SUCCESS the initialization was complete sucessfully
367
368**/
369EFI_STATUS
370EFIAPI
371ShellInitialize (
jcarseya405b862010-09-14 05:18:09 +0000372 )
373{
jcarseyd2b45642009-05-11 18:02:16 +0000374 //
375 // if auto initialize is not false then skip
376 //
377 if (PcdGetBool(PcdShellLibAutoInitialize) != 0) {
378 return (EFI_SUCCESS);
379 }
380
381 //
382 // deinit the current stuff
383 //
384 ASSERT_EFI_ERROR(ShellLibDestructor(gImageHandle, gST));
385
386 //
387 // init the new stuff
388 //
389 return (ShellLibConstructorWorker(gImageHandle, gST));
390}
391
jcarsey94b17fa2009-05-07 18:46:18 +0000392/**
jcarsey1e6e84c2010-01-25 20:05:08 +0000393 This function will retrieve the information about the file for the handle
jcarsey94b17fa2009-05-07 18:46:18 +0000394 specified and store it in allocated pool memory.
395
jcarsey1e6e84c2010-01-25 20:05:08 +0000396 This function allocates a buffer to store the file's information. It is the
qhuang869817bf2009-05-20 14:42:48 +0000397 caller's responsibility to free the buffer
jcarsey94b17fa2009-05-07 18:46:18 +0000398
jcarsey1e6e84c2010-01-25 20:05:08 +0000399 @param FileHandle The file handle of the file for which information is
jcarsey94b17fa2009-05-07 18:46:18 +0000400 being requested.
401
402 @retval NULL information could not be retrieved.
403
404 @return the information about the file
405**/
406EFI_FILE_INFO*
407EFIAPI
408ShellGetFileInfo (
jcarseya405b862010-09-14 05:18:09 +0000409 IN SHELL_FILE_HANDLE FileHandle
410 )
411{
jcarseyd2b45642009-05-11 18:02:16 +0000412 return (FileFunctionMap.GetFileInfo(FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000413}
414
415/**
jcarseya405b862010-09-14 05:18:09 +0000416 This function sets the information about the file for the opened handle
jcarsey94b17fa2009-05-07 18:46:18 +0000417 specified.
418
jcarseya405b862010-09-14 05:18:09 +0000419 @param[in] FileHandle The file handle of the file for which information
420 is being set.
jcarsey94b17fa2009-05-07 18:46:18 +0000421
jcarseya405b862010-09-14 05:18:09 +0000422 @param[in] FileInfo The information to set.
jcarsey94b17fa2009-05-07 18:46:18 +0000423
darylm503b0934ac2011-11-12 00:35:11 +0000424 @retval EFI_SUCCESS The information was set.
jcarseya405b862010-09-14 05:18:09 +0000425 @retval EFI_INVALID_PARAMETER A parameter was out of range or invalid.
426 @retval EFI_UNSUPPORTED The FileHandle does not support FileInfo.
darylm503b0934ac2011-11-12 00:35:11 +0000427 @retval EFI_NO_MEDIA The device has no medium.
428 @retval EFI_DEVICE_ERROR The device reported an error.
429 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
430 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
jcarseya405b862010-09-14 05:18:09 +0000431 @retval EFI_ACCESS_DENIED The file was opened read only.
432 @retval EFI_VOLUME_FULL The volume is full.
jcarsey94b17fa2009-05-07 18:46:18 +0000433**/
434EFI_STATUS
435EFIAPI
436ShellSetFileInfo (
darylm503b0934ac2011-11-12 00:35:11 +0000437 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000438 IN EFI_FILE_INFO *FileInfo
jcarseya405b862010-09-14 05:18:09 +0000439 )
440{
jcarseyd2b45642009-05-11 18:02:16 +0000441 return (FileFunctionMap.SetFileInfo(FileHandle, FileInfo));
jcarsey1e6e84c2010-01-25 20:05:08 +0000442}
443
jcarsey94b17fa2009-05-07 18:46:18 +0000444 /**
445 This function will open a file or directory referenced by DevicePath.
446
jcarsey1e6e84c2010-01-25 20:05:08 +0000447 This function opens a file with the open mode according to the file path. The
jcarsey94b17fa2009-05-07 18:46:18 +0000448 Attributes is valid only for EFI_FILE_MODE_CREATE.
449
darylm503b0934ac2011-11-12 00:35:11 +0000450 @param FilePath on input the device path to the file. On output
jcarsey94b17fa2009-05-07 18:46:18 +0000451 the remaining device path.
darylm503b0934ac2011-11-12 00:35:11 +0000452 @param DeviceHandle pointer to the system device handle.
453 @param FileHandle pointer to the file handle.
454 @param OpenMode the mode to open the file with.
455 @param Attributes the file's file attributes.
jcarsey94b17fa2009-05-07 18:46:18 +0000456
darylm503b0934ac2011-11-12 00:35:11 +0000457 @retval EFI_SUCCESS The information was set.
458 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
459 @retval EFI_UNSUPPORTED Could not open the file path.
460 @retval EFI_NOT_FOUND The specified file could not be found on the
jcarsey1e6e84c2010-01-25 20:05:08 +0000461 device or the file system could not be found on
jcarsey94b17fa2009-05-07 18:46:18 +0000462 the device.
darylm503b0934ac2011-11-12 00:35:11 +0000463 @retval EFI_NO_MEDIA The device has no medium.
464 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
jcarsey94b17fa2009-05-07 18:46:18 +0000465 medium is no longer supported.
darylm503b0934ac2011-11-12 00:35:11 +0000466 @retval EFI_DEVICE_ERROR The device reported an error.
467 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
468 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
469 @retval EFI_ACCESS_DENIED The file was opened read only.
470 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
jcarsey94b17fa2009-05-07 18:46:18 +0000471 file.
darylm503b0934ac2011-11-12 00:35:11 +0000472 @retval EFI_VOLUME_FULL The volume is full.
jcarsey94b17fa2009-05-07 18:46:18 +0000473**/
474EFI_STATUS
475EFIAPI
476ShellOpenFileByDevicePath(
darylm503b0934ac2011-11-12 00:35:11 +0000477 IN OUT EFI_DEVICE_PATH_PROTOCOL **FilePath,
478 OUT EFI_HANDLE *DeviceHandle,
jcarseya405b862010-09-14 05:18:09 +0000479 OUT SHELL_FILE_HANDLE *FileHandle,
darylm503b0934ac2011-11-12 00:35:11 +0000480 IN UINT64 OpenMode,
481 IN UINT64 Attributes
jcarseya405b862010-09-14 05:18:09 +0000482 )
483{
484 CHAR16 *FileName;
485 EFI_STATUS Status;
jcarsey94b17fa2009-05-07 18:46:18 +0000486 EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *EfiSimpleFileSystemProtocol;
jcarseya405b862010-09-14 05:18:09 +0000487 EFI_FILE_PROTOCOL *Handle1;
488 EFI_FILE_PROTOCOL *Handle2;
ydong100b6cb332013-01-25 02:00:22 +0000489 CHAR16 *FnafPathName;
490 UINTN PathLen;
jcarsey94b17fa2009-05-07 18:46:18 +0000491
jcarsey92a54472011-06-27 20:33:13 +0000492 if (FilePath == NULL || FileHandle == NULL || DeviceHandle == NULL) {
493 return (EFI_INVALID_PARAMETER);
494 }
495
jcarsey1e6e84c2010-01-25 20:05:08 +0000496 //
jcarsey94b17fa2009-05-07 18:46:18 +0000497 // which shell interface should we use
498 //
jcarsey366f81a2011-06-27 21:04:22 +0000499 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000500 //
501 // use UEFI Shell 2.0 method.
502 //
jcarsey366f81a2011-06-27 21:04:22 +0000503 FileName = gEfiShellProtocol->GetFilePathFromDevicePath(*FilePath);
jcarsey94b17fa2009-05-07 18:46:18 +0000504 if (FileName == NULL) {
505 return (EFI_INVALID_PARAMETER);
506 }
507 Status = ShellOpenFileByName(FileName, FileHandle, OpenMode, Attributes);
508 FreePool(FileName);
509 return (Status);
jcarsey1e6e84c2010-01-25 20:05:08 +0000510 }
jcarseyd2b45642009-05-11 18:02:16 +0000511
512
513 //
514 // use old shell method.
515 //
jcarsey1e6e84c2010-01-25 20:05:08 +0000516 Status = gBS->LocateDevicePath (&gEfiSimpleFileSystemProtocolGuid,
517 FilePath,
jcarseyd2b45642009-05-11 18:02:16 +0000518 DeviceHandle);
519 if (EFI_ERROR (Status)) {
520 return Status;
521 }
522 Status = gBS->OpenProtocol(*DeviceHandle,
523 &gEfiSimpleFileSystemProtocolGuid,
jcarseyb1f95a02009-06-16 00:23:19 +0000524 (VOID**)&EfiSimpleFileSystemProtocol,
jcarseyd2b45642009-05-11 18:02:16 +0000525 gImageHandle,
526 NULL,
527 EFI_OPEN_PROTOCOL_GET_PROTOCOL);
528 if (EFI_ERROR (Status)) {
529 return Status;
530 }
jcarseya405b862010-09-14 05:18:09 +0000531 Status = EfiSimpleFileSystemProtocol->OpenVolume(EfiSimpleFileSystemProtocol, &Handle1);
jcarseyd2b45642009-05-11 18:02:16 +0000532 if (EFI_ERROR (Status)) {
533 FileHandle = NULL;
534 return Status;
535 }
536
537 //
538 // go down directories one node at a time.
539 //
540 while (!IsDevicePathEnd (*FilePath)) {
jcarsey94b17fa2009-05-07 18:46:18 +0000541 //
jcarseyd2b45642009-05-11 18:02:16 +0000542 // For file system access each node should be a file path component
jcarsey94b17fa2009-05-07 18:46:18 +0000543 //
jcarseyd2b45642009-05-11 18:02:16 +0000544 if (DevicePathType (*FilePath) != MEDIA_DEVICE_PATH ||
545 DevicePathSubType (*FilePath) != MEDIA_FILEPATH_DP
jcarseya405b862010-09-14 05:18:09 +0000546 ) {
jcarsey94b17fa2009-05-07 18:46:18 +0000547 FileHandle = NULL;
jcarseyd2b45642009-05-11 18:02:16 +0000548 return (EFI_INVALID_PARAMETER);
jcarsey94b17fa2009-05-07 18:46:18 +0000549 }
jcarseyd2b45642009-05-11 18:02:16 +0000550 //
551 // Open this file path node
552 //
jcarseya405b862010-09-14 05:18:09 +0000553 Handle2 = Handle1;
554 Handle1 = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000555
556 //
ydong100b6cb332013-01-25 02:00:22 +0000557 // File Name Alignment Fix (FNAF)
558 // Handle2->Open may be incapable of handling a unaligned CHAR16 data.
559 // The structure pointed to by FilePath may be not CHAR16 aligned.
560 // This code copies the potentially unaligned PathName data from the
561 // FilePath structure to the aligned FnafPathName for use in the
562 // calls to Handl2->Open.
563 //
564
565 //
566 // Determine length of PathName, in bytes.
567 //
568 PathLen = DevicePathNodeLength (*FilePath) - SIZE_OF_FILEPATH_DEVICE_PATH;
569
570 //
571 // Allocate memory for the aligned copy of the string Extra allocation is to allow for forced alignment
572 // Copy bytes from possibly unaligned location to aligned location
573 //
574 FnafPathName = AllocateCopyPool(PathLen, (UINT8 *)((FILEPATH_DEVICE_PATH*)*FilePath)->PathName);
575 if (FnafPathName == NULL) {
576 return EFI_OUT_OF_RESOURCES;
577 }
578
579 //
jcarseyd2b45642009-05-11 18:02:16 +0000580 // Try to test opening an existing file
jcarsey94b17fa2009-05-07 18:46:18 +0000581 //
jcarseya405b862010-09-14 05:18:09 +0000582 Status = Handle2->Open (
583 Handle2,
584 &Handle1,
ydong100b6cb332013-01-25 02:00:22 +0000585 FnafPathName,
jcarseyd2b45642009-05-11 18:02:16 +0000586 OpenMode &~EFI_FILE_MODE_CREATE,
587 0
jcarseya405b862010-09-14 05:18:09 +0000588 );
jcarsey94b17fa2009-05-07 18:46:18 +0000589
jcarseyd2b45642009-05-11 18:02:16 +0000590 //
591 // see if the error was that it needs to be created
592 //
593 if ((EFI_ERROR (Status)) && (OpenMode != (OpenMode &~EFI_FILE_MODE_CREATE))) {
jcarseya405b862010-09-14 05:18:09 +0000594 Status = Handle2->Open (
595 Handle2,
596 &Handle1,
ydong100b6cb332013-01-25 02:00:22 +0000597 FnafPathName,
jcarseyd2b45642009-05-11 18:02:16 +0000598 OpenMode,
599 Attributes
jcarseya405b862010-09-14 05:18:09 +0000600 );
jcarsey94b17fa2009-05-07 18:46:18 +0000601 }
ydong100b6cb332013-01-25 02:00:22 +0000602
603 //
604 // Free the alignment buffer
605 //
606 FreePool(FnafPathName);
607
jcarseyd2b45642009-05-11 18:02:16 +0000608 //
609 // Close the last node
610 //
jcarseya405b862010-09-14 05:18:09 +0000611 Handle2->Close (Handle2);
jcarseyd2b45642009-05-11 18:02:16 +0000612
613 if (EFI_ERROR(Status)) {
614 return (Status);
615 }
616
617 //
618 // Get the next node
619 //
620 *FilePath = NextDevicePathNode (*FilePath);
jcarsey94b17fa2009-05-07 18:46:18 +0000621 }
jcarseya405b862010-09-14 05:18:09 +0000622
623 //
624 // This is a weak spot since if the undefined SHELL_FILE_HANDLE format changes this must change also!
625 //
626 *FileHandle = (VOID*)Handle1;
jcarseyd2b45642009-05-11 18:02:16 +0000627 return (EFI_SUCCESS);
jcarsey94b17fa2009-05-07 18:46:18 +0000628}
629
630/**
631 This function will open a file or directory referenced by filename.
632
jcarsey1e6e84c2010-01-25 20:05:08 +0000633 If return is EFI_SUCCESS, the Filehandle is the opened file's handle;
634 otherwise, the Filehandle is NULL. The Attributes is valid only for
jcarsey94b17fa2009-05-07 18:46:18 +0000635 EFI_FILE_MODE_CREATE.
636
jcarsey92a54472011-06-27 20:33:13 +0000637 if FileName is NULL then ASSERT()
jcarsey94b17fa2009-05-07 18:46:18 +0000638
darylm503b0934ac2011-11-12 00:35:11 +0000639 @param FileName pointer to file name
640 @param FileHandle pointer to the file handle.
641 @param OpenMode the mode to open the file with.
642 @param Attributes the file's file attributes.
jcarsey94b17fa2009-05-07 18:46:18 +0000643
darylm503b0934ac2011-11-12 00:35:11 +0000644 @retval EFI_SUCCESS The information was set.
645 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
646 @retval EFI_UNSUPPORTED Could not open the file path.
647 @retval EFI_NOT_FOUND The specified file could not be found on the
jcarsey1e6e84c2010-01-25 20:05:08 +0000648 device or the file system could not be found
jcarsey94b17fa2009-05-07 18:46:18 +0000649 on the device.
darylm503b0934ac2011-11-12 00:35:11 +0000650 @retval EFI_NO_MEDIA The device has no medium.
651 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
jcarsey94b17fa2009-05-07 18:46:18 +0000652 medium is no longer supported.
darylm503b0934ac2011-11-12 00:35:11 +0000653 @retval EFI_DEVICE_ERROR The device reported an error.
654 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
655 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
656 @retval EFI_ACCESS_DENIED The file was opened read only.
657 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
jcarsey94b17fa2009-05-07 18:46:18 +0000658 file.
darylm503b0934ac2011-11-12 00:35:11 +0000659 @retval EFI_VOLUME_FULL The volume is full.
jcarsey94b17fa2009-05-07 18:46:18 +0000660**/
661EFI_STATUS
662EFIAPI
663ShellOpenFileByName(
darylm503b0934ac2011-11-12 00:35:11 +0000664 IN CONST CHAR16 *FileName,
jcarseya405b862010-09-14 05:18:09 +0000665 OUT SHELL_FILE_HANDLE *FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000666 IN UINT64 OpenMode,
darylm503b0934ac2011-11-12 00:35:11 +0000667 IN UINT64 Attributes
jcarseya405b862010-09-14 05:18:09 +0000668 )
669{
jcarsey94b17fa2009-05-07 18:46:18 +0000670 EFI_HANDLE DeviceHandle;
671 EFI_DEVICE_PATH_PROTOCOL *FilePath;
jcarseyb1f95a02009-06-16 00:23:19 +0000672 EFI_STATUS Status;
673 EFI_FILE_INFO *FileInfo;
jcarsey94b17fa2009-05-07 18:46:18 +0000674
675 //
676 // ASSERT if FileName is NULL
677 //
678 ASSERT(FileName != NULL);
679
jcarseya405b862010-09-14 05:18:09 +0000680 if (FileName == NULL) {
681 return (EFI_INVALID_PARAMETER);
682 }
683
jcarsey366f81a2011-06-27 21:04:22 +0000684 if (gEfiShellProtocol != NULL) {
jcarseya405b862010-09-14 05:18:09 +0000685 if ((OpenMode & EFI_FILE_MODE_CREATE) == EFI_FILE_MODE_CREATE && (Attributes & EFI_FILE_DIRECTORY) == EFI_FILE_DIRECTORY) {
686 return ShellCreateDirectory(FileName, FileHandle);
687 }
jcarsey94b17fa2009-05-07 18:46:18 +0000688 //
689 // Use UEFI Shell 2.0 method
690 //
jcarsey366f81a2011-06-27 21:04:22 +0000691 Status = gEfiShellProtocol->OpenFileByName(FileName,
jcarseyb1f95a02009-06-16 00:23:19 +0000692 FileHandle,
693 OpenMode);
jcarseya405b862010-09-14 05:18:09 +0000694 if (StrCmp(FileName, L"NUL") != 0 && !EFI_ERROR(Status) && ((OpenMode & EFI_FILE_MODE_CREATE) != 0)){
jcarsey2247dde2009-11-09 18:08:58 +0000695 FileInfo = FileFunctionMap.GetFileInfo(*FileHandle);
jcarseyb1f95a02009-06-16 00:23:19 +0000696 ASSERT(FileInfo != NULL);
697 FileInfo->Attribute = Attributes;
jcarsey2247dde2009-11-09 18:08:58 +0000698 Status = FileFunctionMap.SetFileInfo(*FileHandle, FileInfo);
699 FreePool(FileInfo);
jcarseyb1f95a02009-06-16 00:23:19 +0000700 }
701 return (Status);
jcarsey1e6e84c2010-01-25 20:05:08 +0000702 }
jcarsey94b17fa2009-05-07 18:46:18 +0000703 //
704 // Using EFI Shell version
705 // this means convert name to path and call that function
706 // since this will use EFI method again that will open it.
707 //
708 ASSERT(mEfiShellEnvironment2 != NULL);
jcarseyb82bfcc2009-06-29 16:28:23 +0000709 FilePath = mEfiShellEnvironment2->NameToPath ((CHAR16*)FileName);
xdu290bfa222010-07-19 05:21:27 +0000710 if (FilePath != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000711 return (ShellOpenFileByDevicePath(&FilePath,
712 &DeviceHandle,
713 FileHandle,
714 OpenMode,
jcarseya405b862010-09-14 05:18:09 +0000715 Attributes));
jcarsey94b17fa2009-05-07 18:46:18 +0000716 }
717 return (EFI_DEVICE_ERROR);
718}
719/**
720 This function create a directory
721
jcarsey1e6e84c2010-01-25 20:05:08 +0000722 If return is EFI_SUCCESS, the Filehandle is the opened directory's handle;
723 otherwise, the Filehandle is NULL. If the directory already existed, this
jcarsey94b17fa2009-05-07 18:46:18 +0000724 function opens the existing directory.
725
darylm503b0934ac2011-11-12 00:35:11 +0000726 @param DirectoryName pointer to directory name
727 @param FileHandle pointer to the file handle.
jcarsey94b17fa2009-05-07 18:46:18 +0000728
darylm503b0934ac2011-11-12 00:35:11 +0000729 @retval EFI_SUCCESS The information was set.
730 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
731 @retval EFI_UNSUPPORTED Could not open the file path.
732 @retval EFI_NOT_FOUND The specified file could not be found on the
jcarsey1e6e84c2010-01-25 20:05:08 +0000733 device or the file system could not be found
jcarsey94b17fa2009-05-07 18:46:18 +0000734 on the device.
darylm503b0934ac2011-11-12 00:35:11 +0000735 @retval EFI_NO_MEDIA The device has no medium.
736 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
jcarsey94b17fa2009-05-07 18:46:18 +0000737 medium is no longer supported.
darylm503b0934ac2011-11-12 00:35:11 +0000738 @retval EFI_DEVICE_ERROR The device reported an error.
739 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
740 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
741 @retval EFI_ACCESS_DENIED The file was opened read only.
742 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
jcarsey94b17fa2009-05-07 18:46:18 +0000743 file.
darylm503b0934ac2011-11-12 00:35:11 +0000744 @retval EFI_VOLUME_FULL The volume is full.
jcarsey94b17fa2009-05-07 18:46:18 +0000745 @sa ShellOpenFileByName
746**/
747EFI_STATUS
748EFIAPI
749ShellCreateDirectory(
jcarseyb82bfcc2009-06-29 16:28:23 +0000750 IN CONST CHAR16 *DirectoryName,
jcarseya405b862010-09-14 05:18:09 +0000751 OUT SHELL_FILE_HANDLE *FileHandle
752 )
753{
jcarsey366f81a2011-06-27 21:04:22 +0000754 if (gEfiShellProtocol != NULL) {
jcarsey2247dde2009-11-09 18:08:58 +0000755 //
756 // Use UEFI Shell 2.0 method
757 //
jcarsey366f81a2011-06-27 21:04:22 +0000758 return (gEfiShellProtocol->CreateFile(DirectoryName,
jcarsey2247dde2009-11-09 18:08:58 +0000759 EFI_FILE_DIRECTORY,
760 FileHandle
jcarseya405b862010-09-14 05:18:09 +0000761 ));
jcarsey2247dde2009-11-09 18:08:58 +0000762 } else {
763 return (ShellOpenFileByName(DirectoryName,
764 FileHandle,
765 EFI_FILE_MODE_READ | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE,
766 EFI_FILE_DIRECTORY
jcarseya405b862010-09-14 05:18:09 +0000767 ));
jcarsey2247dde2009-11-09 18:08:58 +0000768 }
jcarsey94b17fa2009-05-07 18:46:18 +0000769}
770
771/**
772 This function reads information from an opened file.
773
jcarsey1e6e84c2010-01-25 20:05:08 +0000774 If FileHandle is not a directory, the function reads the requested number of
775 bytes from the file at the file's current position and returns them in Buffer.
jcarsey94b17fa2009-05-07 18:46:18 +0000776 If the read goes beyond the end of the file, the read length is truncated to the
jcarsey1e6e84c2010-01-25 20:05:08 +0000777 end of the file. The file's current position is increased by the number of bytes
778 returned. If FileHandle is a directory, the function reads the directory entry
779 at the file's current position and returns the entry in Buffer. If the Buffer
780 is not large enough to hold the current directory entry, then
781 EFI_BUFFER_TOO_SMALL is returned and the current file position is not updated.
782 BufferSize is set to be the size of the buffer needed to read the entry. On
783 success, the current position is updated to the next directory entry. If there
784 are no more directory entries, the read returns a zero-length buffer.
jcarsey94b17fa2009-05-07 18:46:18 +0000785 EFI_FILE_INFO is the structure returned as the directory entry.
786
787 @param FileHandle the opened file handle
jcarsey1e6e84c2010-01-25 20:05:08 +0000788 @param BufferSize on input the size of buffer in bytes. on return
jcarsey94b17fa2009-05-07 18:46:18 +0000789 the number of bytes written.
790 @param Buffer the buffer to put read data into.
791
darylm503b0934ac2011-11-12 00:35:11 +0000792 @retval EFI_SUCCESS Data was read.
793 @retval EFI_NO_MEDIA The device has no media.
794 @retval EFI_DEVICE_ERROR The device reported an error.
795 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
796 @retval EFI_BUFFER_TO_SMALL Buffer is too small. ReadSize contains required
jcarsey94b17fa2009-05-07 18:46:18 +0000797 size.
798
799**/
800EFI_STATUS
801EFIAPI
802ShellReadFile(
jcarseya405b862010-09-14 05:18:09 +0000803 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000804 IN OUT UINTN *BufferSize,
805 OUT VOID *Buffer
jcarseya405b862010-09-14 05:18:09 +0000806 )
807{
jcarseyd2b45642009-05-11 18:02:16 +0000808 return (FileFunctionMap.ReadFile(FileHandle, BufferSize, Buffer));
jcarsey94b17fa2009-05-07 18:46:18 +0000809}
810
811
812/**
813 Write data to a file.
814
jcarsey1e6e84c2010-01-25 20:05:08 +0000815 This function writes the specified number of bytes to the file at the current
816 file position. The current file position is advanced the actual number of bytes
817 written, which is returned in BufferSize. Partial writes only occur when there
818 has been a data error during the write attempt (such as "volume space full").
819 The file is automatically grown to hold the data if required. Direct writes to
jcarsey94b17fa2009-05-07 18:46:18 +0000820 opened directories are not supported.
821
822 @param FileHandle The opened file for writing
823 @param BufferSize on input the number of bytes in Buffer. On output
824 the number of bytes written.
825 @param Buffer the buffer containing data to write is stored.
826
darylm503b0934ac2011-11-12 00:35:11 +0000827 @retval EFI_SUCCESS Data was written.
828 @retval EFI_UNSUPPORTED Writes to an open directory are not supported.
829 @retval EFI_NO_MEDIA The device has no media.
830 @retval EFI_DEVICE_ERROR The device reported an error.
831 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
832 @retval EFI_WRITE_PROTECTED The device is write-protected.
833 @retval EFI_ACCESS_DENIED The file was open for read only.
834 @retval EFI_VOLUME_FULL The volume is full.
jcarsey94b17fa2009-05-07 18:46:18 +0000835**/
836EFI_STATUS
837EFIAPI
838ShellWriteFile(
jcarsey252d9452011-03-25 20:49:53 +0000839 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000840 IN OUT UINTN *BufferSize,
841 IN VOID *Buffer
jcarseya405b862010-09-14 05:18:09 +0000842 )
843{
jcarseyd2b45642009-05-11 18:02:16 +0000844 return (FileFunctionMap.WriteFile(FileHandle, BufferSize, Buffer));
jcarsey94b17fa2009-05-07 18:46:18 +0000845}
846
jcarsey1e6e84c2010-01-25 20:05:08 +0000847/**
jcarsey94b17fa2009-05-07 18:46:18 +0000848 Close an open file handle.
849
jcarsey1e6e84c2010-01-25 20:05:08 +0000850 This function closes a specified file handle. All "dirty" cached file data is
851 flushed to the device, and the file is closed. In all cases the handle is
jcarsey94b17fa2009-05-07 18:46:18 +0000852 closed.
853
854@param FileHandle the file handle to close.
855
856@retval EFI_SUCCESS the file handle was closed sucessfully.
857**/
858EFI_STATUS
859EFIAPI
860ShellCloseFile (
jcarseya405b862010-09-14 05:18:09 +0000861 IN SHELL_FILE_HANDLE *FileHandle
862 )
863{
jcarseyd2b45642009-05-11 18:02:16 +0000864 return (FileFunctionMap.CloseFile(*FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000865}
866
867/**
868 Delete a file and close the handle
869
870 This function closes and deletes a file. In all cases the file handle is closed.
jcarsey1e6e84c2010-01-25 20:05:08 +0000871 If the file cannot be deleted, the warning code EFI_WARN_DELETE_FAILURE is
jcarsey94b17fa2009-05-07 18:46:18 +0000872 returned, but the handle is still closed.
873
874 @param FileHandle the file handle to delete
875
876 @retval EFI_SUCCESS the file was closed sucessfully
jcarsey1e6e84c2010-01-25 20:05:08 +0000877 @retval EFI_WARN_DELETE_FAILURE the handle was closed, but the file was not
jcarsey94b17fa2009-05-07 18:46:18 +0000878 deleted
darylm503b0934ac2011-11-12 00:35:11 +0000879 @retval INVALID_PARAMETER One of the parameters has an invalid value.
jcarsey94b17fa2009-05-07 18:46:18 +0000880**/
881EFI_STATUS
882EFIAPI
883ShellDeleteFile (
darylm503b0934ac2011-11-12 00:35:11 +0000884 IN SHELL_FILE_HANDLE *FileHandle
jcarseya405b862010-09-14 05:18:09 +0000885 )
886{
jcarseyd2b45642009-05-11 18:02:16 +0000887 return (FileFunctionMap.DeleteFile(*FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000888}
889
890/**
891 Set the current position in a file.
892
jcarsey1e6e84c2010-01-25 20:05:08 +0000893 This function sets the current file position for the handle to the position
jcarsey94b17fa2009-05-07 18:46:18 +0000894 supplied. With the exception of seeking to position 0xFFFFFFFFFFFFFFFF, only
jcarsey1e6e84c2010-01-25 20:05:08 +0000895 absolute positioning is supported, and seeking past the end of the file is
896 allowed (a subsequent write would grow the file). Seeking to position
jcarsey94b17fa2009-05-07 18:46:18 +0000897 0xFFFFFFFFFFFFFFFF causes the current position to be set to the end of the file.
jcarsey1e6e84c2010-01-25 20:05:08 +0000898 If FileHandle is a directory, the only position that may be set is zero. This
jcarsey94b17fa2009-05-07 18:46:18 +0000899 has the effect of starting the read process of the directory entries over.
900
901 @param FileHandle The file handle on which the position is being set
902 @param Position Byte position from begining of file
903
904 @retval EFI_SUCCESS Operation completed sucessfully.
jcarsey1e6e84c2010-01-25 20:05:08 +0000905 @retval EFI_UNSUPPORTED the seek request for non-zero is not valid on
jcarsey94b17fa2009-05-07 18:46:18 +0000906 directories.
907 @retval INVALID_PARAMETER One of the parameters has an invalid value.
908**/
909EFI_STATUS
910EFIAPI
911ShellSetFilePosition (
darylm503b0934ac2011-11-12 00:35:11 +0000912 IN SHELL_FILE_HANDLE FileHandle,
913 IN UINT64 Position
jcarseya405b862010-09-14 05:18:09 +0000914 )
915{
jcarseyd2b45642009-05-11 18:02:16 +0000916 return (FileFunctionMap.SetFilePosition(FileHandle, Position));
jcarsey94b17fa2009-05-07 18:46:18 +0000917}
918
jcarsey1e6e84c2010-01-25 20:05:08 +0000919/**
jcarsey94b17fa2009-05-07 18:46:18 +0000920 Gets a file's current position
921
jcarsey1e6e84c2010-01-25 20:05:08 +0000922 This function retrieves the current file position for the file handle. For
923 directories, the current file position has no meaning outside of the file
jcarsey94b17fa2009-05-07 18:46:18 +0000924 system driver and as such the operation is not supported. An error is returned
925 if FileHandle is a directory.
926
927 @param FileHandle The open file handle on which to get the position.
928 @param Position Byte position from begining of file.
929
930 @retval EFI_SUCCESS the operation completed sucessfully.
931 @retval INVALID_PARAMETER One of the parameters has an invalid value.
932 @retval EFI_UNSUPPORTED the request is not valid on directories.
933**/
934EFI_STATUS
935EFIAPI
936ShellGetFilePosition (
jcarseya405b862010-09-14 05:18:09 +0000937 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000938 OUT UINT64 *Position
jcarseya405b862010-09-14 05:18:09 +0000939 )
940{
jcarseyd2b45642009-05-11 18:02:16 +0000941 return (FileFunctionMap.GetFilePosition(FileHandle, Position));
jcarsey94b17fa2009-05-07 18:46:18 +0000942}
943/**
944 Flushes data on a file
jcarsey1e6e84c2010-01-25 20:05:08 +0000945
jcarsey94b17fa2009-05-07 18:46:18 +0000946 This function flushes all modified data associated with a file to a device.
947
948 @param FileHandle The file handle on which to flush data
949
950 @retval EFI_SUCCESS The data was flushed.
951 @retval EFI_NO_MEDIA The device has no media.
952 @retval EFI_DEVICE_ERROR The device reported an error.
953 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
954 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
955 @retval EFI_ACCESS_DENIED The file was opened for read only.
956**/
957EFI_STATUS
958EFIAPI
959ShellFlushFile (
jcarseya405b862010-09-14 05:18:09 +0000960 IN SHELL_FILE_HANDLE FileHandle
961 )
962{
jcarseyd2b45642009-05-11 18:02:16 +0000963 return (FileFunctionMap.FlushFile(FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000964}
965
darylm503b0934ac2011-11-12 00:35:11 +0000966/** Retrieve first entry from a directory.
jcarsey94b17fa2009-05-07 18:46:18 +0000967
darylm503b0934ac2011-11-12 00:35:11 +0000968 This function takes an open directory handle and gets information from the
969 first entry in the directory. A buffer is allocated to contain
970 the information and a pointer to the buffer is returned in *Buffer. The
971 caller can use ShellFindNextFile() to get subsequent directory entries.
jcarsey94b17fa2009-05-07 18:46:18 +0000972
darylm503b0934ac2011-11-12 00:35:11 +0000973 The buffer will be freed by ShellFindNextFile() when the last directory
974 entry is read. Otherwise, the caller must free the buffer, using FreePool,
975 when finished with it.
976
977 @param[in] DirHandle The file handle of the directory to search.
978 @param[out] Buffer The pointer to the buffer for the file's information.
jcarsey94b17fa2009-05-07 18:46:18 +0000979
980 @retval EFI_SUCCESS Found the first file.
981 @retval EFI_NOT_FOUND Cannot find the directory.
982 @retval EFI_NO_MEDIA The device has no media.
983 @retval EFI_DEVICE_ERROR The device reported an error.
984 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
985 @return Others status of ShellGetFileInfo, ShellSetFilePosition,
986 or ShellReadFile
987**/
988EFI_STATUS
989EFIAPI
990ShellFindFirstFile (
jcarseya405b862010-09-14 05:18:09 +0000991 IN SHELL_FILE_HANDLE DirHandle,
jcarseyd2b45642009-05-11 18:02:16 +0000992 OUT EFI_FILE_INFO **Buffer
jcarseya405b862010-09-14 05:18:09 +0000993 )
994{
jcarsey94b17fa2009-05-07 18:46:18 +0000995 //
jcarseyd2b45642009-05-11 18:02:16 +0000996 // pass to file handle lib
jcarsey94b17fa2009-05-07 18:46:18 +0000997 //
jcarseyd2b45642009-05-11 18:02:16 +0000998 return (FileHandleFindFirstFile(DirHandle, Buffer));
jcarsey94b17fa2009-05-07 18:46:18 +0000999}
darylm503b0934ac2011-11-12 00:35:11 +00001000/** Retrieve next entries from a directory.
jcarsey94b17fa2009-05-07 18:46:18 +00001001
darylm503b0934ac2011-11-12 00:35:11 +00001002 To use this function, the caller must first call the ShellFindFirstFile()
1003 function to get the first directory entry. Subsequent directory entries are
1004 retrieved by using the ShellFindNextFile() function. This function can
1005 be called several times to get each entry from the directory. If the call of
1006 ShellFindNextFile() retrieved the last directory entry, the next call of
1007 this function will set *NoFile to TRUE and free the buffer.
jcarsey94b17fa2009-05-07 18:46:18 +00001008
darylm503b0934ac2011-11-12 00:35:11 +00001009 @param[in] DirHandle The file handle of the directory.
1010 @param[out] Buffer The pointer to buffer for file's information.
1011 @param[out] NoFile The pointer to boolean when last file is found.
jcarsey94b17fa2009-05-07 18:46:18 +00001012
1013 @retval EFI_SUCCESS Found the next file, or reached last file
1014 @retval EFI_NO_MEDIA The device has no media.
1015 @retval EFI_DEVICE_ERROR The device reported an error.
1016 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
1017**/
1018EFI_STATUS
1019EFIAPI
1020ShellFindNextFile(
jcarseya405b862010-09-14 05:18:09 +00001021 IN SHELL_FILE_HANDLE DirHandle,
jcarsey94b17fa2009-05-07 18:46:18 +00001022 OUT EFI_FILE_INFO *Buffer,
1023 OUT BOOLEAN *NoFile
jcarseya405b862010-09-14 05:18:09 +00001024 )
1025{
jcarsey94b17fa2009-05-07 18:46:18 +00001026 //
jcarseyd2b45642009-05-11 18:02:16 +00001027 // pass to file handle lib
jcarsey94b17fa2009-05-07 18:46:18 +00001028 //
jcarseyd2b45642009-05-11 18:02:16 +00001029 return (FileHandleFindNextFile(DirHandle, Buffer, NoFile));
jcarsey94b17fa2009-05-07 18:46:18 +00001030}
1031/**
1032 Retrieve the size of a file.
1033
1034 if FileHandle is NULL then ASSERT()
1035 if Size is NULL then ASSERT()
1036
jcarsey1e6e84c2010-01-25 20:05:08 +00001037 This function extracts the file size info from the FileHandle's EFI_FILE_INFO
jcarsey94b17fa2009-05-07 18:46:18 +00001038 data.
1039
1040 @param FileHandle file handle from which size is retrieved
1041 @param Size pointer to size
1042
1043 @retval EFI_SUCCESS operation was completed sucessfully
1044 @retval EFI_DEVICE_ERROR cannot access the file
1045**/
1046EFI_STATUS
1047EFIAPI
1048ShellGetFileSize (
jcarseya405b862010-09-14 05:18:09 +00001049 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +00001050 OUT UINT64 *Size
jcarseya405b862010-09-14 05:18:09 +00001051 )
1052{
jcarseyd2b45642009-05-11 18:02:16 +00001053 return (FileFunctionMap.GetFileSize(FileHandle, Size));
jcarsey94b17fa2009-05-07 18:46:18 +00001054}
1055/**
1056 Retrieves the status of the break execution flag
1057
1058 this function is useful to check whether the application is being asked to halt by the shell.
1059
1060 @retval TRUE the execution break is enabled
1061 @retval FALSE the execution break is not enabled
1062**/
1063BOOLEAN
1064EFIAPI
1065ShellGetExecutionBreakFlag(
1066 VOID
1067 )
1068{
jcarsey1e6e84c2010-01-25 20:05:08 +00001069 //
jcarsey94b17fa2009-05-07 18:46:18 +00001070 // Check for UEFI Shell 2.0 protocols
1071 //
jcarsey366f81a2011-06-27 21:04:22 +00001072 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001073
1074 //
1075 // We are using UEFI Shell 2.0; see if the event has been triggered
1076 //
jcarsey366f81a2011-06-27 21:04:22 +00001077 if (gBS->CheckEvent(gEfiShellProtocol->ExecutionBreak) != EFI_SUCCESS) {
jcarsey94b17fa2009-05-07 18:46:18 +00001078 return (FALSE);
1079 }
1080 return (TRUE);
jcarsey1e6e84c2010-01-25 20:05:08 +00001081 }
jcarsey94b17fa2009-05-07 18:46:18 +00001082
1083 //
1084 // using EFI Shell; call the function to check
1085 //
jcarsey92a54472011-06-27 20:33:13 +00001086 if (mEfiShellEnvironment2 != NULL) {
1087 return (mEfiShellEnvironment2->GetExecutionBreak());
1088 }
1089
1090 return (FALSE);
jcarsey94b17fa2009-05-07 18:46:18 +00001091}
1092/**
1093 return the value of an environment variable
1094
jcarsey1e6e84c2010-01-25 20:05:08 +00001095 this function gets the value of the environment variable set by the
jcarsey94b17fa2009-05-07 18:46:18 +00001096 ShellSetEnvironmentVariable function
1097
1098 @param EnvKey The key name of the environment variable.
1099
1100 @retval NULL the named environment variable does not exist.
1101 @return != NULL pointer to the value of the environment variable
1102**/
1103CONST CHAR16*
1104EFIAPI
1105ShellGetEnvironmentVariable (
jcarsey9b3bf082009-06-23 21:15:07 +00001106 IN CONST CHAR16 *EnvKey
jcarsey94b17fa2009-05-07 18:46:18 +00001107 )
1108{
jcarsey1e6e84c2010-01-25 20:05:08 +00001109 //
jcarsey94b17fa2009-05-07 18:46:18 +00001110 // Check for UEFI Shell 2.0 protocols
1111 //
jcarsey366f81a2011-06-27 21:04:22 +00001112 if (gEfiShellProtocol != NULL) {
1113 return (gEfiShellProtocol->GetEnv(EnvKey));
jcarsey94b17fa2009-05-07 18:46:18 +00001114 }
1115
1116 //
jcarsey92a54472011-06-27 20:33:13 +00001117 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001118 //
jcarsey92a54472011-06-27 20:33:13 +00001119 if (mEfiShellEnvironment2 != NULL) {
1120 return (mEfiShellEnvironment2->GetEnv((CHAR16*)EnvKey));
1121 }
jcarsey94b17fa2009-05-07 18:46:18 +00001122
jcarsey92a54472011-06-27 20:33:13 +00001123 return NULL;
jcarsey94b17fa2009-05-07 18:46:18 +00001124}
1125/**
1126 set the value of an environment variable
1127
1128This function changes the current value of the specified environment variable. If the
1129environment variable exists and the Value is an empty string, then the environment
1130variable is deleted. If the environment variable exists and the Value is not an empty
1131string, then the value of the environment variable is changed. If the environment
1132variable does not exist and the Value is an empty string, there is no action. If the
1133environment variable does not exist and the Value is a non-empty string, then the
1134environment variable is created and assigned the specified value.
1135
1136 This is not supported pre-UEFI Shell 2.0.
1137
1138 @param EnvKey The key name of the environment variable.
1139 @param EnvVal The Value of the environment variable
1140 @param Volatile Indicates whether the variable is non-volatile (FALSE) or volatile (TRUE).
1141
1142 @retval EFI_SUCCESS the operation was completed sucessfully
1143 @retval EFI_UNSUPPORTED This operation is not allowed in pre UEFI 2.0 Shell environments
1144**/
1145EFI_STATUS
1146EFIAPI
1147ShellSetEnvironmentVariable (
1148 IN CONST CHAR16 *EnvKey,
1149 IN CONST CHAR16 *EnvVal,
1150 IN BOOLEAN Volatile
1151 )
1152{
jcarsey1e6e84c2010-01-25 20:05:08 +00001153 //
jcarsey94b17fa2009-05-07 18:46:18 +00001154 // Check for UEFI Shell 2.0 protocols
1155 //
jcarsey366f81a2011-06-27 21:04:22 +00001156 if (gEfiShellProtocol != NULL) {
1157 return (gEfiShellProtocol->SetEnv(EnvKey, EnvVal, Volatile));
jcarsey1e6e84c2010-01-25 20:05:08 +00001158 }
jcarsey94b17fa2009-05-07 18:46:18 +00001159
1160 //
1161 // This feature does not exist under EFI shell
1162 //
1163 return (EFI_UNSUPPORTED);
1164}
jcarseya405b862010-09-14 05:18:09 +00001165
jcarsey94b17fa2009-05-07 18:46:18 +00001166/**
jcarseya405b862010-09-14 05:18:09 +00001167 Cause the shell to parse and execute a command line.
jcarsey94b17fa2009-05-07 18:46:18 +00001168
1169 This function creates a nested instance of the shell and executes the specified
jcarseya405b862010-09-14 05:18:09 +00001170 command (CommandLine) with the specified environment (Environment). Upon return,
1171 the status code returned by the specified command is placed in StatusCode.
1172 If Environment is NULL, then the current environment is used and all changes made
1173 by the commands executed will be reflected in the current environment. If the
1174 Environment is non-NULL, then the changes made will be discarded.
1175 The CommandLine is executed from the current working directory on the current
1176 device.
jcarsey94b17fa2009-05-07 18:46:18 +00001177
Brendan Jackman3877d0f2014-01-24 22:31:07 +00001178 The EnvironmentVariables pararemeter is ignored in a pre-UEFI Shell 2.0
jcarseya405b862010-09-14 05:18:09 +00001179 environment. The values pointed to by the parameters will be unchanged by the
1180 ShellExecute() function. The Output parameter has no effect in a
1181 UEFI Shell 2.0 environment.
jcarsey94b17fa2009-05-07 18:46:18 +00001182
jcarseya405b862010-09-14 05:18:09 +00001183 @param[in] ParentHandle The parent image starting the operation.
1184 @param[in] CommandLine The pointer to a NULL terminated command line.
1185 @param[in] Output True to display debug output. False to hide it.
1186 @param[in] EnvironmentVariables Optional pointer to array of environment variables
1187 in the form "x=y". If NULL, the current set is used.
1188 @param[out] Status The status of the run command line.
jcarsey94b17fa2009-05-07 18:46:18 +00001189
jcarseya405b862010-09-14 05:18:09 +00001190 @retval EFI_SUCCESS The operation completed sucessfully. Status
1191 contains the status code returned.
1192 @retval EFI_INVALID_PARAMETER A parameter contains an invalid value.
1193 @retval EFI_OUT_OF_RESOURCES Out of resources.
1194 @retval EFI_UNSUPPORTED The operation is not allowed.
jcarsey94b17fa2009-05-07 18:46:18 +00001195**/
1196EFI_STATUS
1197EFIAPI
1198ShellExecute (
1199 IN EFI_HANDLE *ParentHandle,
1200 IN CHAR16 *CommandLine OPTIONAL,
1201 IN BOOLEAN Output OPTIONAL,
1202 IN CHAR16 **EnvironmentVariables OPTIONAL,
1203 OUT EFI_STATUS *Status OPTIONAL
1204 )
1205{
Brendan Jackman3877d0f2014-01-24 22:31:07 +00001206 EFI_STATUS CmdStatus;
jcarsey1e6e84c2010-01-25 20:05:08 +00001207 //
jcarsey94b17fa2009-05-07 18:46:18 +00001208 // Check for UEFI Shell 2.0 protocols
1209 //
jcarsey366f81a2011-06-27 21:04:22 +00001210 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001211 //
1212 // Call UEFI Shell 2.0 version (not using Output parameter)
1213 //
jcarsey366f81a2011-06-27 21:04:22 +00001214 return (gEfiShellProtocol->Execute(ParentHandle,
jcarsey94b17fa2009-05-07 18:46:18 +00001215 CommandLine,
1216 EnvironmentVariables,
1217 Status));
jcarsey1e6e84c2010-01-25 20:05:08 +00001218 }
jcarsey92a54472011-06-27 20:33:13 +00001219
jcarsey94b17fa2009-05-07 18:46:18 +00001220 //
jcarsey92a54472011-06-27 20:33:13 +00001221 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001222 //
jcarsey92a54472011-06-27 20:33:13 +00001223 if (mEfiShellEnvironment2 != NULL) {
1224 //
Brendan Jackman3877d0f2014-01-24 22:31:07 +00001225 // Call EFI Shell version.
jcarsey92a54472011-06-27 20:33:13 +00001226 // Due to oddity in the EFI shell we want to dereference the ParentHandle here
1227 //
Brendan Jackman3877d0f2014-01-24 22:31:07 +00001228 CmdStatus = (mEfiShellEnvironment2->Execute(*ParentHandle,
jcarsey92a54472011-06-27 20:33:13 +00001229 CommandLine,
1230 Output));
Brendan Jackman3877d0f2014-01-24 22:31:07 +00001231 //
1232 // No Status output parameter so just use the returned status
1233 //
1234 if (Status != NULL) {
1235 *Status = CmdStatus;
1236 }
1237 //
1238 // If there was an error, we can't tell if it was from the command or from
1239 // the Execute() function, so we'll just assume the shell ran successfully
1240 // and the error came from the command.
1241 //
1242 return EFI_SUCCESS;
jcarsey92a54472011-06-27 20:33:13 +00001243 }
1244
1245 return (EFI_UNSUPPORTED);
jcarsey94b17fa2009-05-07 18:46:18 +00001246}
Brendan Jackman3877d0f2014-01-24 22:31:07 +00001247
jcarsey94b17fa2009-05-07 18:46:18 +00001248/**
1249 Retreives the current directory path
1250
jcarsey1e6e84c2010-01-25 20:05:08 +00001251 If the DeviceName is NULL, it returns the current device's current directory
1252 name. If the DeviceName is not NULL, it returns the current directory name
jcarsey94b17fa2009-05-07 18:46:18 +00001253 on specified drive.
1254
1255 @param DeviceName the name of the drive to get directory on
1256
1257 @retval NULL the directory does not exist
1258 @return != NULL the directory
1259**/
1260CONST CHAR16*
1261EFIAPI
1262ShellGetCurrentDir (
jcarseya405b862010-09-14 05:18:09 +00001263 IN CHAR16 * CONST DeviceName OPTIONAL
jcarsey94b17fa2009-05-07 18:46:18 +00001264 )
1265{
jcarsey1e6e84c2010-01-25 20:05:08 +00001266 //
jcarsey94b17fa2009-05-07 18:46:18 +00001267 // Check for UEFI Shell 2.0 protocols
1268 //
jcarsey366f81a2011-06-27 21:04:22 +00001269 if (gEfiShellProtocol != NULL) {
1270 return (gEfiShellProtocol->GetCurDir(DeviceName));
jcarsey1e6e84c2010-01-25 20:05:08 +00001271 }
jcarsey92a54472011-06-27 20:33:13 +00001272
jcarsey94b17fa2009-05-07 18:46:18 +00001273 //
jcarsey8bd282b2011-06-27 16:45:41 +00001274 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001275 //
jcarsey8bd282b2011-06-27 16:45:41 +00001276 if (mEfiShellEnvironment2 != NULL) {
1277 return (mEfiShellEnvironment2->CurDir(DeviceName));
1278 }
1279
1280 return (NULL);
jcarsey94b17fa2009-05-07 18:46:18 +00001281}
1282/**
1283 sets (enabled or disabled) the page break mode
1284
jcarsey1e6e84c2010-01-25 20:05:08 +00001285 when page break mode is enabled the screen will stop scrolling
jcarsey94b17fa2009-05-07 18:46:18 +00001286 and wait for operator input before scrolling a subsequent screen.
1287
1288 @param CurrentState TRUE to enable and FALSE to disable
1289**/
jcarsey1e6e84c2010-01-25 20:05:08 +00001290VOID
jcarsey94b17fa2009-05-07 18:46:18 +00001291EFIAPI
1292ShellSetPageBreakMode (
1293 IN BOOLEAN CurrentState
1294 )
1295{
1296 //
1297 // check for enabling
1298 //
1299 if (CurrentState != 0x00) {
jcarsey1e6e84c2010-01-25 20:05:08 +00001300 //
jcarsey94b17fa2009-05-07 18:46:18 +00001301 // check for UEFI Shell 2.0
1302 //
jcarsey366f81a2011-06-27 21:04:22 +00001303 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001304 //
1305 // Enable with UEFI 2.0 Shell
1306 //
jcarsey366f81a2011-06-27 21:04:22 +00001307 gEfiShellProtocol->EnablePageBreak();
jcarsey94b17fa2009-05-07 18:46:18 +00001308 return;
1309 } else {
jcarsey1e6e84c2010-01-25 20:05:08 +00001310 //
jcarsey92a54472011-06-27 20:33:13 +00001311 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001312 //
jcarsey92a54472011-06-27 20:33:13 +00001313 if (mEfiShellEnvironment2 != NULL) {
1314 //
1315 // Enable with EFI Shell
1316 //
1317 mEfiShellEnvironment2->EnablePageBreak (DEFAULT_INIT_ROW, DEFAULT_AUTO_LF);
1318 return;
1319 }
jcarsey94b17fa2009-05-07 18:46:18 +00001320 }
1321 } else {
jcarsey1e6e84c2010-01-25 20:05:08 +00001322 //
jcarsey94b17fa2009-05-07 18:46:18 +00001323 // check for UEFI Shell 2.0
1324 //
jcarsey366f81a2011-06-27 21:04:22 +00001325 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001326 //
1327 // Disable with UEFI 2.0 Shell
1328 //
jcarsey366f81a2011-06-27 21:04:22 +00001329 gEfiShellProtocol->DisablePageBreak();
jcarsey94b17fa2009-05-07 18:46:18 +00001330 return;
1331 } else {
jcarsey1e6e84c2010-01-25 20:05:08 +00001332 //
jcarsey92a54472011-06-27 20:33:13 +00001333 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001334 //
jcarsey92a54472011-06-27 20:33:13 +00001335 if (mEfiShellEnvironment2 != NULL) {
1336 //
1337 // Disable with EFI Shell
1338 //
1339 mEfiShellEnvironment2->DisablePageBreak ();
1340 return;
1341 }
jcarsey94b17fa2009-05-07 18:46:18 +00001342 }
1343 }
1344}
1345
1346///
1347/// version of EFI_SHELL_FILE_INFO struct, except has no CONST pointers.
1348/// This allows for the struct to be populated.
1349///
1350typedef struct {
jcarseyd2b45642009-05-11 18:02:16 +00001351 LIST_ENTRY Link;
jcarsey94b17fa2009-05-07 18:46:18 +00001352 EFI_STATUS Status;
1353 CHAR16 *FullName;
1354 CHAR16 *FileName;
jcarseya405b862010-09-14 05:18:09 +00001355 SHELL_FILE_HANDLE Handle;
jcarsey94b17fa2009-05-07 18:46:18 +00001356 EFI_FILE_INFO *Info;
1357} EFI_SHELL_FILE_INFO_NO_CONST;
1358
1359/**
1360 Converts a EFI shell list of structures to the coresponding UEFI Shell 2.0 type of list.
1361
1362 if OldStyleFileList is NULL then ASSERT()
1363
jcarsey1e6e84c2010-01-25 20:05:08 +00001364 this function will convert a SHELL_FILE_ARG based list into a callee allocated
jcarsey94b17fa2009-05-07 18:46:18 +00001365 EFI_SHELL_FILE_INFO based list. it is up to the caller to free the memory via
1366 the ShellCloseFileMetaArg function.
1367
ydong104ff7e372011-09-02 08:05:34 +00001368 @param[in] FileList the EFI shell list type
1369 @param[in, out] ListHead the list to add to
jcarsey94b17fa2009-05-07 18:46:18 +00001370
1371 @retval the resultant head of the double linked new format list;
1372**/
1373LIST_ENTRY*
1374EFIAPI
1375InternalShellConvertFileListType (
jcarsey9b3bf082009-06-23 21:15:07 +00001376 IN LIST_ENTRY *FileList,
1377 IN OUT LIST_ENTRY *ListHead
jcarsey125c2cf2009-11-18 21:36:50 +00001378 )
1379{
jcarsey94b17fa2009-05-07 18:46:18 +00001380 SHELL_FILE_ARG *OldInfo;
jcarsey9b3bf082009-06-23 21:15:07 +00001381 LIST_ENTRY *Link;
jcarsey94b17fa2009-05-07 18:46:18 +00001382 EFI_SHELL_FILE_INFO_NO_CONST *NewInfo;
1383
1384 //
jcarsey9b3bf082009-06-23 21:15:07 +00001385 // ASSERTs
jcarsey94b17fa2009-05-07 18:46:18 +00001386 //
jcarsey9b3bf082009-06-23 21:15:07 +00001387 ASSERT(FileList != NULL);
1388 ASSERT(ListHead != NULL);
jcarsey94b17fa2009-05-07 18:46:18 +00001389
1390 //
1391 // enumerate through each member of the old list and copy
1392 //
jcarseyd2b45642009-05-11 18:02:16 +00001393 for (Link = FileList->ForwardLink; Link != FileList; Link = Link->ForwardLink) {
jcarsey94b17fa2009-05-07 18:46:18 +00001394 OldInfo = CR (Link, SHELL_FILE_ARG, Link, SHELL_FILE_ARG_SIGNATURE);
jcarseya405b862010-09-14 05:18:09 +00001395 ASSERT(OldInfo != NULL);
1396
1397 //
1398 // Skip ones that failed to open...
1399 //
1400 if (OldInfo->Status != EFI_SUCCESS) {
1401 continue;
1402 }
jcarsey94b17fa2009-05-07 18:46:18 +00001403
1404 //
1405 // make sure the old list was valid
1406 //
jcarsey94b17fa2009-05-07 18:46:18 +00001407 ASSERT(OldInfo->Info != NULL);
1408 ASSERT(OldInfo->FullName != NULL);
1409 ASSERT(OldInfo->FileName != NULL);
1410
1411 //
1412 // allocate a new EFI_SHELL_FILE_INFO object
1413 //
1414 NewInfo = AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
jcarseyc9d92df2010-02-03 15:37:54 +00001415 if (NewInfo == NULL) {
jcarsey7a95efd2011-10-13 16:08:18 +00001416 ShellCloseFileMetaArg((EFI_SHELL_FILE_INFO**)(&ListHead));
jcarseybeab0fc2011-10-10 17:26:25 +00001417 ListHead = NULL;
jcarseyc9d92df2010-02-03 15:37:54 +00001418 break;
1419 }
jcarsey1e6e84c2010-01-25 20:05:08 +00001420
1421 //
jcarsey94b17fa2009-05-07 18:46:18 +00001422 // copy the simple items
1423 //
1424 NewInfo->Handle = OldInfo->Handle;
1425 NewInfo->Status = OldInfo->Status;
1426
jcarseyd2b45642009-05-11 18:02:16 +00001427 // old shell checks for 0 not NULL
1428 OldInfo->Handle = 0;
1429
jcarsey94b17fa2009-05-07 18:46:18 +00001430 //
1431 // allocate new space to copy strings and structure
1432 //
Jaben Carsey98c16be2014-08-19 21:00:34 +00001433 NewInfo->FullName = AllocateCopyPool(StrSize(OldInfo->FullName), OldInfo->FullName);
1434 NewInfo->FileName = AllocateCopyPool(StrSize(OldInfo->FileName), OldInfo->FileName);
1435 NewInfo->Info = AllocateCopyPool((UINTN)OldInfo->Info->Size, OldInfo->Info);
jcarsey1e6e84c2010-01-25 20:05:08 +00001436
jcarsey94b17fa2009-05-07 18:46:18 +00001437 //
1438 // make sure all the memory allocations were sucessful
1439 //
jcarseybeab0fc2011-10-10 17:26:25 +00001440 if (NULL == NewInfo->FullName || NewInfo->FileName == NULL || NewInfo->Info == NULL) {
Jaben Carsey98c16be2014-08-19 21:00:34 +00001441 //
1442 // Free the partially allocated new node
1443 //
1444 SHELL_FREE_NON_NULL(NewInfo->FullName);
1445 SHELL_FREE_NON_NULL(NewInfo->FileName);
1446 SHELL_FREE_NON_NULL(NewInfo->Info);
1447 SHELL_FREE_NON_NULL(NewInfo);
1448
1449 //
1450 // Free the previously converted stuff
1451 //
jcarsey7a95efd2011-10-13 16:08:18 +00001452 ShellCloseFileMetaArg((EFI_SHELL_FILE_INFO**)(&ListHead));
jcarseybeab0fc2011-10-10 17:26:25 +00001453 ListHead = NULL;
1454 break;
1455 }
jcarsey94b17fa2009-05-07 18:46:18 +00001456
1457 //
jcarsey94b17fa2009-05-07 18:46:18 +00001458 // add that to the list
1459 //
jcarsey9b3bf082009-06-23 21:15:07 +00001460 InsertTailList(ListHead, &NewInfo->Link);
jcarsey94b17fa2009-05-07 18:46:18 +00001461 }
1462 return (ListHead);
1463}
1464/**
1465 Opens a group of files based on a path.
1466
jcarsey1e6e84c2010-01-25 20:05:08 +00001467 This function uses the Arg to open all the matching files. Each matched
Brendan Jackman70879312014-01-24 22:29:53 +00001468 file has a SHELL_FILE_INFO structure to record the file information. These
1469 structures are placed on the list ListHead. Users can get the SHELL_FILE_INFO
jcarsey94b17fa2009-05-07 18:46:18 +00001470 structures from ListHead to access each file. This function supports wildcards
jcarsey1e6e84c2010-01-25 20:05:08 +00001471 and will process '?' and '*' as such. the list must be freed with a call to
jcarsey94b17fa2009-05-07 18:46:18 +00001472 ShellCloseFileMetaArg().
1473
jcarsey1e6e84c2010-01-25 20:05:08 +00001474 If you are NOT appending to an existing list *ListHead must be NULL. If
jcarsey5f7431d2009-07-10 18:06:01 +00001475 *ListHead is NULL then it must be callee freed.
jcarsey94b17fa2009-05-07 18:46:18 +00001476
1477 @param Arg pointer to path string
1478 @param OpenMode mode to open files with
1479 @param ListHead head of linked list of results
1480
jcarsey1e6e84c2010-01-25 20:05:08 +00001481 @retval EFI_SUCCESS the operation was sucessful and the list head
jcarsey94b17fa2009-05-07 18:46:18 +00001482 contains the list of opened files
jcarsey94b17fa2009-05-07 18:46:18 +00001483 @return != EFI_SUCCESS the operation failed
1484
1485 @sa InternalShellConvertFileListType
1486**/
1487EFI_STATUS
1488EFIAPI
1489ShellOpenFileMetaArg (
1490 IN CHAR16 *Arg,
1491 IN UINT64 OpenMode,
1492 IN OUT EFI_SHELL_FILE_INFO **ListHead
1493 )
1494{
1495 EFI_STATUS Status;
jcarsey9b3bf082009-06-23 21:15:07 +00001496 LIST_ENTRY mOldStyleFileList;
Qiu Shumin0960ba12014-09-17 07:58:31 +00001497 CHAR16 *CleanFilePathStr;
jcarsey1e6e84c2010-01-25 20:05:08 +00001498
jcarsey94b17fa2009-05-07 18:46:18 +00001499 //
1500 // ASSERT that Arg and ListHead are not NULL
1501 //
1502 ASSERT(Arg != NULL);
1503 ASSERT(ListHead != NULL);
1504
Qiu Shumin0960ba12014-09-17 07:58:31 +00001505 Status = InternalShellStripQuotes (Arg, &CleanFilePathStr);
1506 if (EFI_ERROR (Status)) {
1507 return Status;
1508 }
1509
jcarsey1e6e84c2010-01-25 20:05:08 +00001510 //
jcarsey94b17fa2009-05-07 18:46:18 +00001511 // Check for UEFI Shell 2.0 protocols
1512 //
jcarsey366f81a2011-06-27 21:04:22 +00001513 if (gEfiShellProtocol != NULL) {
jcarsey5f7431d2009-07-10 18:06:01 +00001514 if (*ListHead == NULL) {
1515 *ListHead = (EFI_SHELL_FILE_INFO*)AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1516 if (*ListHead == NULL) {
Qiu Shumin0960ba12014-09-17 07:58:31 +00001517 FreePool(CleanFilePathStr);
jcarsey5f7431d2009-07-10 18:06:01 +00001518 return (EFI_OUT_OF_RESOURCES);
1519 }
1520 InitializeListHead(&((*ListHead)->Link));
jcarsey1e6e84c2010-01-25 20:05:08 +00001521 }
Qiu Shumin0960ba12014-09-17 07:58:31 +00001522 Status = gEfiShellProtocol->OpenFileList(CleanFilePathStr,
jcarsey1e6e84c2010-01-25 20:05:08 +00001523 OpenMode,
jcarsey2247dde2009-11-09 18:08:58 +00001524 ListHead);
1525 if (EFI_ERROR(Status)) {
jcarsey366f81a2011-06-27 21:04:22 +00001526 gEfiShellProtocol->RemoveDupInFileList(ListHead);
jcarsey2247dde2009-11-09 18:08:58 +00001527 } else {
jcarsey366f81a2011-06-27 21:04:22 +00001528 Status = gEfiShellProtocol->RemoveDupInFileList(ListHead);
jcarsey2247dde2009-11-09 18:08:58 +00001529 }
jcarseya405b862010-09-14 05:18:09 +00001530 if (*ListHead != NULL && IsListEmpty(&(*ListHead)->Link)) {
1531 FreePool(*ListHead);
Qiu Shumin0960ba12014-09-17 07:58:31 +00001532 FreePool(CleanFilePathStr);
jcarseya405b862010-09-14 05:18:09 +00001533 *ListHead = NULL;
1534 return (EFI_NOT_FOUND);
1535 }
Qiu Shumin0960ba12014-09-17 07:58:31 +00001536 FreePool(CleanFilePathStr);
jcarsey2247dde2009-11-09 18:08:58 +00001537 return (Status);
jcarsey1e6e84c2010-01-25 20:05:08 +00001538 }
jcarsey94b17fa2009-05-07 18:46:18 +00001539
1540 //
jcarsey92a54472011-06-27 20:33:13 +00001541 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001542 //
jcarsey92a54472011-06-27 20:33:13 +00001543 if (mEfiShellEnvironment2 != NULL) {
1544 //
1545 // make sure the list head is initialized
1546 //
1547 InitializeListHead(&mOldStyleFileList);
jcarsey94b17fa2009-05-07 18:46:18 +00001548
jcarsey92a54472011-06-27 20:33:13 +00001549 //
1550 // Get the EFI Shell list of files
1551 //
Qiu Shumin0960ba12014-09-17 07:58:31 +00001552 Status = mEfiShellEnvironment2->FileMetaArg(CleanFilePathStr, &mOldStyleFileList);
jcarsey92a54472011-06-27 20:33:13 +00001553 if (EFI_ERROR(Status)) {
1554 *ListHead = NULL;
Qiu Shumin0960ba12014-09-17 07:58:31 +00001555 FreePool(CleanFilePathStr);
jcarsey92a54472011-06-27 20:33:13 +00001556 return (Status);
1557 }
jcarsey94b17fa2009-05-07 18:46:18 +00001558
jcarsey92a54472011-06-27 20:33:13 +00001559 if (*ListHead == NULL) {
1560 *ListHead = (EFI_SHELL_FILE_INFO *)AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1561 if (*ListHead == NULL) {
Qiu Shumin0960ba12014-09-17 07:58:31 +00001562 FreePool(CleanFilePathStr);
jcarsey92a54472011-06-27 20:33:13 +00001563 return (EFI_OUT_OF_RESOURCES);
1564 }
1565 InitializeListHead(&((*ListHead)->Link));
1566 }
1567
1568 //
1569 // Convert that to equivalent of UEFI Shell 2.0 structure
1570 //
1571 InternalShellConvertFileListType(&mOldStyleFileList, &(*ListHead)->Link);
1572
1573 //
1574 // Free the EFI Shell version that was converted.
1575 //
1576 mEfiShellEnvironment2->FreeFileList(&mOldStyleFileList);
1577
1578 if ((*ListHead)->Link.ForwardLink == (*ListHead)->Link.BackLink && (*ListHead)->Link.BackLink == &((*ListHead)->Link)) {
1579 FreePool(*ListHead);
1580 *ListHead = NULL;
1581 Status = EFI_NOT_FOUND;
1582 }
Qiu Shumin0960ba12014-09-17 07:58:31 +00001583 FreePool(CleanFilePathStr);
jcarsey94b17fa2009-05-07 18:46:18 +00001584 return (Status);
1585 }
1586
Qiu Shumin0960ba12014-09-17 07:58:31 +00001587 FreePool(CleanFilePathStr);
jcarsey92a54472011-06-27 20:33:13 +00001588 return (EFI_UNSUPPORTED);
jcarsey94b17fa2009-05-07 18:46:18 +00001589}
1590/**
jcarseya405b862010-09-14 05:18:09 +00001591 Free the linked list returned from ShellOpenFileMetaArg.
jcarsey94b17fa2009-05-07 18:46:18 +00001592
jcarseya405b862010-09-14 05:18:09 +00001593 if ListHead is NULL then ASSERT().
jcarsey94b17fa2009-05-07 18:46:18 +00001594
jcarseya405b862010-09-14 05:18:09 +00001595 @param ListHead the pointer to free.
jcarsey94b17fa2009-05-07 18:46:18 +00001596
jcarseya405b862010-09-14 05:18:09 +00001597 @retval EFI_SUCCESS the operation was sucessful.
jcarsey94b17fa2009-05-07 18:46:18 +00001598**/
1599EFI_STATUS
1600EFIAPI
1601ShellCloseFileMetaArg (
1602 IN OUT EFI_SHELL_FILE_INFO **ListHead
1603 )
1604{
1605 LIST_ENTRY *Node;
1606
1607 //
1608 // ASSERT that ListHead is not NULL
1609 //
1610 ASSERT(ListHead != NULL);
1611
jcarsey1e6e84c2010-01-25 20:05:08 +00001612 //
jcarsey94b17fa2009-05-07 18:46:18 +00001613 // Check for UEFI Shell 2.0 protocols
1614 //
jcarsey366f81a2011-06-27 21:04:22 +00001615 if (gEfiShellProtocol != NULL) {
1616 return (gEfiShellProtocol->FreeFileList(ListHead));
jcarsey92a54472011-06-27 20:33:13 +00001617 } else if (mEfiShellEnvironment2 != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001618 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001619 // Since this is EFI Shell version we need to free our internally made copy
jcarsey94b17fa2009-05-07 18:46:18 +00001620 // of the list
1621 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001622 for ( Node = GetFirstNode(&(*ListHead)->Link)
jcarseya405b862010-09-14 05:18:09 +00001623 ; *ListHead != NULL && !IsListEmpty(&(*ListHead)->Link)
jcarsey9b3bf082009-06-23 21:15:07 +00001624 ; Node = GetFirstNode(&(*ListHead)->Link)) {
jcarsey94b17fa2009-05-07 18:46:18 +00001625 RemoveEntryList(Node);
jcarseya405b862010-09-14 05:18:09 +00001626 ((EFI_FILE_PROTOCOL*)((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->Handle)->Close(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->Handle);
jcarsey94b17fa2009-05-07 18:46:18 +00001627 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->FullName);
1628 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->FileName);
1629 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->Info);
1630 FreePool((EFI_SHELL_FILE_INFO_NO_CONST*)Node);
1631 }
Jaben Carsey75a5e2e2014-01-10 16:42:45 +00001632 SHELL_FREE_NON_NULL(*ListHead);
jcarsey94b17fa2009-05-07 18:46:18 +00001633 return EFI_SUCCESS;
1634 }
jcarsey92a54472011-06-27 20:33:13 +00001635
1636 return (EFI_UNSUPPORTED);
jcarsey94b17fa2009-05-07 18:46:18 +00001637}
1638
jcarsey125c2cf2009-11-18 21:36:50 +00001639/**
1640 Find a file by searching the CWD and then the path.
1641
jcarseyb3011f42010-01-11 21:49:04 +00001642 If FileName is NULL then ASSERT.
jcarsey125c2cf2009-11-18 21:36:50 +00001643
jcarseyb3011f42010-01-11 21:49:04 +00001644 If the return value is not NULL then the memory must be caller freed.
jcarsey125c2cf2009-11-18 21:36:50 +00001645
1646 @param FileName Filename string.
1647
1648 @retval NULL the file was not found
1649 @return !NULL the full path to the file.
1650**/
1651CHAR16 *
1652EFIAPI
1653ShellFindFilePath (
1654 IN CONST CHAR16 *FileName
1655 )
1656{
1657 CONST CHAR16 *Path;
jcarseya405b862010-09-14 05:18:09 +00001658 SHELL_FILE_HANDLE Handle;
jcarsey125c2cf2009-11-18 21:36:50 +00001659 EFI_STATUS Status;
1660 CHAR16 *RetVal;
1661 CHAR16 *TestPath;
1662 CONST CHAR16 *Walker;
jcarsey36a9d672009-11-20 21:13:41 +00001663 UINTN Size;
jcarsey1cd45e72010-01-29 15:07:44 +00001664 CHAR16 *TempChar;
jcarsey125c2cf2009-11-18 21:36:50 +00001665
1666 RetVal = NULL;
1667
jcarseya405b862010-09-14 05:18:09 +00001668 //
1669 // First make sure its not an absolute path.
1670 //
1671 Status = ShellOpenFileByName(FileName, &Handle, EFI_FILE_MODE_READ, 0);
1672 if (!EFI_ERROR(Status)){
1673 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1674 ASSERT(RetVal == NULL);
1675 RetVal = StrnCatGrow(&RetVal, NULL, FileName, 0);
1676 ShellCloseFile(&Handle);
1677 return (RetVal);
1678 } else {
1679 ShellCloseFile(&Handle);
1680 }
1681 }
1682
jcarsey125c2cf2009-11-18 21:36:50 +00001683 Path = ShellGetEnvironmentVariable(L"cwd");
1684 if (Path != NULL) {
jcarsey36a9d672009-11-20 21:13:41 +00001685 Size = StrSize(Path);
1686 Size += StrSize(FileName);
1687 TestPath = AllocateZeroPool(Size);
jcarseyc9d92df2010-02-03 15:37:54 +00001688 if (TestPath == NULL) {
1689 return (NULL);
1690 }
Jaben Carsey98c16be2014-08-19 21:00:34 +00001691 StrnCpy(TestPath, Path, Size/sizeof(CHAR16) - 1);
1692 StrnCat(TestPath, FileName, Size/sizeof(CHAR16) - 1 - StrLen(TestPath));
jcarsey125c2cf2009-11-18 21:36:50 +00001693 Status = ShellOpenFileByName(TestPath, &Handle, EFI_FILE_MODE_READ, 0);
1694 if (!EFI_ERROR(Status)){
jcarseya405b862010-09-14 05:18:09 +00001695 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1696 ASSERT(RetVal == NULL);
1697 RetVal = StrnCatGrow(&RetVal, NULL, TestPath, 0);
1698 ShellCloseFile(&Handle);
1699 FreePool(TestPath);
1700 return (RetVal);
1701 } else {
1702 ShellCloseFile(&Handle);
1703 }
jcarsey125c2cf2009-11-18 21:36:50 +00001704 }
1705 FreePool(TestPath);
1706 }
1707 Path = ShellGetEnvironmentVariable(L"path");
1708 if (Path != NULL) {
jcarseya405b862010-09-14 05:18:09 +00001709 Size = StrSize(Path)+sizeof(CHAR16);
jcarsey36a9d672009-11-20 21:13:41 +00001710 Size += StrSize(FileName);
1711 TestPath = AllocateZeroPool(Size);
jcarsey3e082d52010-10-04 16:44:57 +00001712 if (TestPath == NULL) {
1713 return (NULL);
1714 }
jcarsey1e6e84c2010-01-25 20:05:08 +00001715 Walker = (CHAR16*)Path;
jcarsey125c2cf2009-11-18 21:36:50 +00001716 do {
1717 CopyMem(TestPath, Walker, StrSize(Walker));
jcarsey3e082d52010-10-04 16:44:57 +00001718 if (TestPath != NULL) {
1719 TempChar = StrStr(TestPath, L";");
1720 if (TempChar != NULL) {
1721 *TempChar = CHAR_NULL;
1722 }
1723 if (TestPath[StrLen(TestPath)-1] != L'\\') {
Jaben Carsey98c16be2014-08-19 21:00:34 +00001724 StrnCat(TestPath, L"\\", Size/sizeof(CHAR16) - 1 - StrLen(TestPath));
jcarsey3e082d52010-10-04 16:44:57 +00001725 }
jcarsey89e85372011-04-13 23:37:50 +00001726 if (FileName[0] == L'\\') {
1727 FileName++;
1728 }
Jaben Carsey98c16be2014-08-19 21:00:34 +00001729 StrnCat(TestPath, FileName, Size/sizeof(CHAR16) - 1 - StrLen(TestPath));
jcarsey3e082d52010-10-04 16:44:57 +00001730 if (StrStr(Walker, L";") != NULL) {
1731 Walker = StrStr(Walker, L";") + 1;
jcarseya405b862010-09-14 05:18:09 +00001732 } else {
jcarsey3e082d52010-10-04 16:44:57 +00001733 Walker = NULL;
1734 }
1735 Status = ShellOpenFileByName(TestPath, &Handle, EFI_FILE_MODE_READ, 0);
1736 if (!EFI_ERROR(Status)){
1737 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1738 ASSERT(RetVal == NULL);
1739 RetVal = StrnCatGrow(&RetVal, NULL, TestPath, 0);
1740 ShellCloseFile(&Handle);
1741 break;
1742 } else {
1743 ShellCloseFile(&Handle);
1744 }
jcarseya405b862010-09-14 05:18:09 +00001745 }
jcarsey125c2cf2009-11-18 21:36:50 +00001746 }
1747 } while (Walker != NULL && Walker[0] != CHAR_NULL);
1748 FreePool(TestPath);
1749 }
1750 return (RetVal);
1751}
1752
jcarseyb3011f42010-01-11 21:49:04 +00001753/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001754 Find a file by searching the CWD and then the path with a variable set of file
1755 extensions. If the file is not found it will append each extension in the list
jcarseyb3011f42010-01-11 21:49:04 +00001756 in the order provided and return the first one that is successful.
1757
1758 If FileName is NULL, then ASSERT.
1759 If FileExtension is NULL, then behavior is identical to ShellFindFilePath.
1760
1761 If the return value is not NULL then the memory must be caller freed.
1762
1763 @param[in] FileName Filename string.
1764 @param[in] FileExtension Semi-colon delimeted list of possible extensions.
1765
1766 @retval NULL The file was not found.
1767 @retval !NULL The path to the file.
1768**/
1769CHAR16 *
1770EFIAPI
1771ShellFindFilePathEx (
1772 IN CONST CHAR16 *FileName,
1773 IN CONST CHAR16 *FileExtension
1774 )
1775{
1776 CHAR16 *TestPath;
1777 CHAR16 *RetVal;
1778 CONST CHAR16 *ExtensionWalker;
jcarsey9e926b62010-01-14 20:26:39 +00001779 UINTN Size;
jcarsey1cd45e72010-01-29 15:07:44 +00001780 CHAR16 *TempChar;
jcarseyc9d92df2010-02-03 15:37:54 +00001781 CHAR16 *TempChar2;
jcarsey1cd45e72010-01-29 15:07:44 +00001782
jcarseyb3011f42010-01-11 21:49:04 +00001783 ASSERT(FileName != NULL);
1784 if (FileExtension == NULL) {
1785 return (ShellFindFilePath(FileName));
1786 }
1787 RetVal = ShellFindFilePath(FileName);
1788 if (RetVal != NULL) {
1789 return (RetVal);
1790 }
jcarsey9e926b62010-01-14 20:26:39 +00001791 Size = StrSize(FileName);
1792 Size += StrSize(FileExtension);
1793 TestPath = AllocateZeroPool(Size);
jcarseyc9d92df2010-02-03 15:37:54 +00001794 if (TestPath == NULL) {
1795 return (NULL);
1796 }
jcarseya405b862010-09-14 05:18:09 +00001797 for (ExtensionWalker = FileExtension, TempChar2 = (CHAR16*)FileExtension; TempChar2 != NULL ; ExtensionWalker = TempChar2 + 1){
Jaben Carsey98c16be2014-08-19 21:00:34 +00001798 StrnCpy(TestPath, FileName, Size/sizeof(CHAR16) - 1);
jcarseya405b862010-09-14 05:18:09 +00001799 if (ExtensionWalker != NULL) {
Jaben Carsey98c16be2014-08-19 21:00:34 +00001800 StrnCat(TestPath, ExtensionWalker, Size/sizeof(CHAR16) - 1 - StrLen(TestPath));
jcarseya405b862010-09-14 05:18:09 +00001801 }
jcarsey1cd45e72010-01-29 15:07:44 +00001802 TempChar = StrStr(TestPath, L";");
1803 if (TempChar != NULL) {
1804 *TempChar = CHAR_NULL;
jcarseyb3011f42010-01-11 21:49:04 +00001805 }
1806 RetVal = ShellFindFilePath(TestPath);
1807 if (RetVal != NULL) {
1808 break;
1809 }
jcarseya405b862010-09-14 05:18:09 +00001810 ASSERT(ExtensionWalker != NULL);
jcarseyc9d92df2010-02-03 15:37:54 +00001811 TempChar2 = StrStr(ExtensionWalker, L";");
jcarseyb3011f42010-01-11 21:49:04 +00001812 }
1813 FreePool(TestPath);
1814 return (RetVal);
1815}
1816
jcarsey94b17fa2009-05-07 18:46:18 +00001817typedef struct {
jcarsey9b3bf082009-06-23 21:15:07 +00001818 LIST_ENTRY Link;
jcarsey94b17fa2009-05-07 18:46:18 +00001819 CHAR16 *Name;
jcarseya405b862010-09-14 05:18:09 +00001820 SHELL_PARAM_TYPE Type;
jcarsey94b17fa2009-05-07 18:46:18 +00001821 CHAR16 *Value;
1822 UINTN OriginalPosition;
1823} SHELL_PARAM_PACKAGE;
1824
1825/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001826 Checks the list of valid arguments and returns TRUE if the item was found. If the
jcarsey94b17fa2009-05-07 18:46:18 +00001827 return value is TRUE then the type parameter is set also.
jcarsey1e6e84c2010-01-25 20:05:08 +00001828
jcarsey94b17fa2009-05-07 18:46:18 +00001829 if CheckList is NULL then ASSERT();
1830 if Name is NULL then ASSERT();
1831 if Type is NULL then ASSERT();
1832
jcarsey94b17fa2009-05-07 18:46:18 +00001833 @param Name pointer to Name of parameter found
1834 @param CheckList List to check against
jcarseya405b862010-09-14 05:18:09 +00001835 @param Type pointer to type of parameter if it was found
jcarsey94b17fa2009-05-07 18:46:18 +00001836
1837 @retval TRUE the Parameter was found. Type is valid.
1838 @retval FALSE the Parameter was not found. Type is not valid.
1839**/
1840BOOLEAN
1841EFIAPI
jcarseyd2b45642009-05-11 18:02:16 +00001842InternalIsOnCheckList (
jcarsey94b17fa2009-05-07 18:46:18 +00001843 IN CONST CHAR16 *Name,
1844 IN CONST SHELL_PARAM_ITEM *CheckList,
jcarsey252d9452011-03-25 20:49:53 +00001845 OUT SHELL_PARAM_TYPE *Type
jcarseya405b862010-09-14 05:18:09 +00001846 )
1847{
jcarsey94b17fa2009-05-07 18:46:18 +00001848 SHELL_PARAM_ITEM *TempListItem;
jcarsey252d9452011-03-25 20:49:53 +00001849 CHAR16 *TempString;
jcarsey94b17fa2009-05-07 18:46:18 +00001850
1851 //
1852 // ASSERT that all 3 pointer parameters aren't NULL
1853 //
1854 ASSERT(CheckList != NULL);
1855 ASSERT(Type != NULL);
1856 ASSERT(Name != NULL);
1857
1858 //
jcarseyd2b45642009-05-11 18:02:16 +00001859 // question mark and page break mode are always supported
1860 //
1861 if ((StrCmp(Name, L"-?") == 0) ||
1862 (StrCmp(Name, L"-b") == 0)
jcarseya405b862010-09-14 05:18:09 +00001863 ) {
jcarsey252d9452011-03-25 20:49:53 +00001864 *Type = TypeFlag;
jcarseyd2b45642009-05-11 18:02:16 +00001865 return (TRUE);
1866 }
1867
1868 //
jcarsey94b17fa2009-05-07 18:46:18 +00001869 // Enumerate through the list
1870 //
1871 for (TempListItem = (SHELL_PARAM_ITEM*)CheckList ; TempListItem->Name != NULL ; TempListItem++) {
1872 //
jcarsey9eb53ac2009-07-08 17:26:58 +00001873 // If the Type is TypeStart only check the first characters of the passed in param
1874 // If it matches set the type and return TRUE
jcarsey94b17fa2009-05-07 18:46:18 +00001875 //
darylm503b0934ac2011-11-12 00:35:11 +00001876 if (TempListItem->Type == TypeStart) {
jcarsey252d9452011-03-25 20:49:53 +00001877 if (StrnCmp(Name, TempListItem->Name, StrLen(TempListItem->Name)) == 0) {
1878 *Type = TempListItem->Type;
1879 return (TRUE);
1880 }
1881 TempString = NULL;
1882 TempString = StrnCatGrow(&TempString, NULL, Name, StrLen(TempListItem->Name));
1883 if (TempString != NULL) {
1884 if (StringNoCaseCompare(&TempString, &TempListItem->Name) == 0) {
1885 *Type = TempListItem->Type;
1886 FreePool(TempString);
1887 return (TRUE);
1888 }
1889 FreePool(TempString);
1890 }
1891 } else if (StringNoCaseCompare(&Name, &TempListItem->Name) == 0) {
jcarsey94b17fa2009-05-07 18:46:18 +00001892 *Type = TempListItem->Type;
1893 return (TRUE);
1894 }
1895 }
jcarsey2247dde2009-11-09 18:08:58 +00001896
jcarsey94b17fa2009-05-07 18:46:18 +00001897 return (FALSE);
1898}
1899/**
jcarseyd2b45642009-05-11 18:02:16 +00001900 Checks the string for indicators of "flag" status. this is a leading '/', '-', or '+'
jcarsey94b17fa2009-05-07 18:46:18 +00001901
jcarseya405b862010-09-14 05:18:09 +00001902 @param[in] Name pointer to Name of parameter found
1903 @param[in] AlwaysAllowNumbers TRUE to allow numbers, FALSE to not.
jcarsey94b17fa2009-05-07 18:46:18 +00001904
1905 @retval TRUE the Parameter is a flag.
jcarseya405b862010-09-14 05:18:09 +00001906 @retval FALSE the Parameter not a flag.
jcarsey94b17fa2009-05-07 18:46:18 +00001907**/
1908BOOLEAN
1909EFIAPI
jcarseyd2b45642009-05-11 18:02:16 +00001910InternalIsFlag (
jcarsey2247dde2009-11-09 18:08:58 +00001911 IN CONST CHAR16 *Name,
1912 IN BOOLEAN AlwaysAllowNumbers
jcarsey94b17fa2009-05-07 18:46:18 +00001913 )
1914{
1915 //
1916 // ASSERT that Name isn't NULL
1917 //
1918 ASSERT(Name != NULL);
1919
1920 //
jcarsey2247dde2009-11-09 18:08:58 +00001921 // If we accept numbers then dont return TRUE. (they will be values)
1922 //
Jaben Carsey8af89da2014-08-19 20:58:03 +00001923 if (((Name[0] == L'-' || Name[0] == L'+') && InternalShellIsHexOrDecimalNumber(Name+1, FALSE, FALSE)) && AlwaysAllowNumbers) {
jcarsey2247dde2009-11-09 18:08:58 +00001924 return (FALSE);
1925 }
1926
1927 //
jcarseya405b862010-09-14 05:18:09 +00001928 // If the Name has a /, +, or - as the first character return TRUE
jcarsey94b17fa2009-05-07 18:46:18 +00001929 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001930 if ((Name[0] == L'/') ||
jcarseyd2b45642009-05-11 18:02:16 +00001931 (Name[0] == L'-') ||
1932 (Name[0] == L'+')
jcarseya405b862010-09-14 05:18:09 +00001933 ) {
jcarsey94b17fa2009-05-07 18:46:18 +00001934 return (TRUE);
1935 }
1936 return (FALSE);
1937}
1938
1939/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001940 Checks the command line arguments passed against the list of valid ones.
jcarsey94b17fa2009-05-07 18:46:18 +00001941
1942 If no initialization is required, then return RETURN_SUCCESS.
jcarsey1e6e84c2010-01-25 20:05:08 +00001943
jcarseya405b862010-09-14 05:18:09 +00001944 @param[in] CheckList pointer to list of parameters to check
1945 @param[out] CheckPackage pointer to pointer to list checked values
1946 @param[out] ProblemParam optional pointer to pointer to unicode string for
jcarseyd2b45642009-05-11 18:02:16 +00001947 the paramater that caused failure. If used then the
1948 caller is responsible for freeing the memory.
jcarseya405b862010-09-14 05:18:09 +00001949 @param[in] AutoPageBreak will automatically set PageBreakEnabled for "b" parameter
1950 @param[in] Argv pointer to array of parameters
1951 @param[in] Argc Count of parameters in Argv
1952 @param[in] AlwaysAllowNumbers TRUE to allow numbers always, FALSE otherwise.
jcarsey94b17fa2009-05-07 18:46:18 +00001953
1954 @retval EFI_SUCCESS The operation completed sucessfully.
1955 @retval EFI_OUT_OF_RESOURCES A memory allocation failed
1956 @retval EFI_INVALID_PARAMETER A parameter was invalid
jcarsey1e6e84c2010-01-25 20:05:08 +00001957 @retval EFI_VOLUME_CORRUPTED the command line was corrupt. an argument was
1958 duplicated. the duplicated command line argument
jcarsey94b17fa2009-05-07 18:46:18 +00001959 was returned in ProblemParam if provided.
jcarsey1e6e84c2010-01-25 20:05:08 +00001960 @retval EFI_NOT_FOUND a argument required a value that was missing.
jcarsey94b17fa2009-05-07 18:46:18 +00001961 the invalid command line argument was returned in
1962 ProblemParam if provided.
1963**/
1964EFI_STATUS
1965EFIAPI
1966InternalCommandLineParse (
1967 IN CONST SHELL_PARAM_ITEM *CheckList,
1968 OUT LIST_ENTRY **CheckPackage,
1969 OUT CHAR16 **ProblemParam OPTIONAL,
1970 IN BOOLEAN AutoPageBreak,
1971 IN CONST CHAR16 **Argv,
jcarsey2247dde2009-11-09 18:08:58 +00001972 IN UINTN Argc,
1973 IN BOOLEAN AlwaysAllowNumbers
jcarseya405b862010-09-14 05:18:09 +00001974 )
1975{
jcarsey94b17fa2009-05-07 18:46:18 +00001976 UINTN LoopCounter;
jcarsey252d9452011-03-25 20:49:53 +00001977 SHELL_PARAM_TYPE CurrentItemType;
jcarsey94b17fa2009-05-07 18:46:18 +00001978 SHELL_PARAM_PACKAGE *CurrentItemPackage;
jcarsey125c2cf2009-11-18 21:36:50 +00001979 UINTN GetItemValue;
1980 UINTN ValueSize;
jcarseya405b862010-09-14 05:18:09 +00001981 UINTN Count;
jcarsey252d9452011-03-25 20:49:53 +00001982 CONST CHAR16 *TempPointer;
Jaben Carsey98c16be2014-08-19 21:00:34 +00001983 UINTN CurrentValueSize;
jcarsey94b17fa2009-05-07 18:46:18 +00001984
1985 CurrentItemPackage = NULL;
jcarsey125c2cf2009-11-18 21:36:50 +00001986 GetItemValue = 0;
1987 ValueSize = 0;
jcarseya405b862010-09-14 05:18:09 +00001988 Count = 0;
jcarsey94b17fa2009-05-07 18:46:18 +00001989
1990 //
1991 // If there is only 1 item we dont need to do anything
1992 //
jcarseya405b862010-09-14 05:18:09 +00001993 if (Argc < 1) {
jcarsey94b17fa2009-05-07 18:46:18 +00001994 *CheckPackage = NULL;
1995 return (EFI_SUCCESS);
1996 }
1997
1998 //
jcarsey2247dde2009-11-09 18:08:58 +00001999 // ASSERTs
2000 //
2001 ASSERT(CheckList != NULL);
2002 ASSERT(Argv != NULL);
2003
2004 //
jcarsey94b17fa2009-05-07 18:46:18 +00002005 // initialize the linked list
2006 //
2007 *CheckPackage = (LIST_ENTRY*)AllocateZeroPool(sizeof(LIST_ENTRY));
sfu502a758c2011-10-08 02:55:30 +00002008 if (*CheckPackage == NULL) {
jcarseybeab0fc2011-10-10 17:26:25 +00002009 return (EFI_OUT_OF_RESOURCES);
sfu502a758c2011-10-08 02:55:30 +00002010 }
jcarseybeab0fc2011-10-10 17:26:25 +00002011
jcarsey94b17fa2009-05-07 18:46:18 +00002012 InitializeListHead(*CheckPackage);
2013
2014 //
2015 // loop through each of the arguments
2016 //
2017 for (LoopCounter = 0 ; LoopCounter < Argc ; ++LoopCounter) {
2018 if (Argv[LoopCounter] == NULL) {
2019 //
2020 // do nothing for NULL argv
2021 //
jcarseya405b862010-09-14 05:18:09 +00002022 } else if (InternalIsOnCheckList(Argv[LoopCounter], CheckList, &CurrentItemType)) {
jcarsey94b17fa2009-05-07 18:46:18 +00002023 //
jcarsey2247dde2009-11-09 18:08:58 +00002024 // We might have leftover if last parameter didnt have optional value
2025 //
jcarsey125c2cf2009-11-18 21:36:50 +00002026 if (GetItemValue != 0) {
2027 GetItemValue = 0;
jcarsey2247dde2009-11-09 18:08:58 +00002028 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2029 }
2030 //
jcarsey94b17fa2009-05-07 18:46:18 +00002031 // this is a flag
2032 //
jcarsey252d9452011-03-25 20:49:53 +00002033 CurrentItemPackage = AllocateZeroPool(sizeof(SHELL_PARAM_PACKAGE));
jcarseybeab0fc2011-10-10 17:26:25 +00002034 if (CurrentItemPackage == NULL) {
2035 ShellCommandLineFreeVarList(*CheckPackage);
2036 *CheckPackage = NULL;
2037 return (EFI_OUT_OF_RESOURCES);
2038 }
Jaben Carsey98c16be2014-08-19 21:00:34 +00002039 CurrentItemPackage->Name = AllocateCopyPool(StrSize(Argv[LoopCounter]), Argv[LoopCounter]);
jcarseybeab0fc2011-10-10 17:26:25 +00002040 if (CurrentItemPackage->Name == NULL) {
2041 ShellCommandLineFreeVarList(*CheckPackage);
2042 *CheckPackage = NULL;
2043 return (EFI_OUT_OF_RESOURCES);
2044 }
jcarsey94b17fa2009-05-07 18:46:18 +00002045 CurrentItemPackage->Type = CurrentItemType;
2046 CurrentItemPackage->OriginalPosition = (UINTN)(-1);
jcarseyb1f95a02009-06-16 00:23:19 +00002047 CurrentItemPackage->Value = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +00002048
2049 //
2050 // Does this flag require a value
2051 //
jcarsey125c2cf2009-11-18 21:36:50 +00002052 switch (CurrentItemPackage->Type) {
jcarsey94b17fa2009-05-07 18:46:18 +00002053 //
jcarsey125c2cf2009-11-18 21:36:50 +00002054 // possibly trigger the next loop(s) to populate the value of this item
jcarsey1e6e84c2010-01-25 20:05:08 +00002055 //
jcarsey125c2cf2009-11-18 21:36:50 +00002056 case TypeValue:
jcarsey1e6e84c2010-01-25 20:05:08 +00002057 GetItemValue = 1;
jcarsey125c2cf2009-11-18 21:36:50 +00002058 ValueSize = 0;
2059 break;
2060 case TypeDoubleValue:
2061 GetItemValue = 2;
2062 ValueSize = 0;
2063 break;
2064 case TypeMaxValue:
2065 GetItemValue = (UINTN)(-1);
2066 ValueSize = 0;
2067 break;
2068 default:
2069 //
2070 // this item has no value expected; we are done
2071 //
2072 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2073 ASSERT(GetItemValue == 0);
2074 break;
jcarsey94b17fa2009-05-07 18:46:18 +00002075 }
jcarseya405b862010-09-14 05:18:09 +00002076 } else if (GetItemValue != 0 && !InternalIsFlag(Argv[LoopCounter], AlwaysAllowNumbers)) {
jcarseyb1f95a02009-06-16 00:23:19 +00002077 ASSERT(CurrentItemPackage != NULL);
2078 //
jcarsey125c2cf2009-11-18 21:36:50 +00002079 // get the item VALUE for a previous flag
jcarseyb1f95a02009-06-16 00:23:19 +00002080 //
jcarsey49bd4982012-01-27 18:42:43 +00002081 if (StrStr(Argv[LoopCounter], L" ") == NULL) {
Jaben Carsey98c16be2014-08-19 21:00:34 +00002082 CurrentValueSize = ValueSize + StrSize(Argv[LoopCounter]) + sizeof(CHAR16);
2083 CurrentItemPackage->Value = ReallocatePool(ValueSize, CurrentValueSize, CurrentItemPackage->Value);
jcarsey49bd4982012-01-27 18:42:43 +00002084 ASSERT(CurrentItemPackage->Value != NULL);
2085 if (ValueSize == 0) {
Jaben Carsey98c16be2014-08-19 21:00:34 +00002086 StrnCpy(CurrentItemPackage->Value, Argv[LoopCounter], CurrentValueSize/sizeof(CHAR16) - 1);
jcarsey49bd4982012-01-27 18:42:43 +00002087 } else {
Jaben Carsey98c16be2014-08-19 21:00:34 +00002088 StrnCat(CurrentItemPackage->Value, L" ", CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
2089 StrnCat(CurrentItemPackage->Value, Argv[LoopCounter], CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
jcarsey49bd4982012-01-27 18:42:43 +00002090 }
2091 ValueSize += StrSize(Argv[LoopCounter]) + sizeof(CHAR16);
jcarsey125c2cf2009-11-18 21:36:50 +00002092 } else {
jcarsey49bd4982012-01-27 18:42:43 +00002093 //
2094 // the parameter has spaces. must be quoted.
2095 //
Jaben Carsey98c16be2014-08-19 21:00:34 +00002096 CurrentValueSize = ValueSize + StrSize(Argv[LoopCounter]) + sizeof(CHAR16) + sizeof(CHAR16) + sizeof(CHAR16);
2097 CurrentItemPackage->Value = ReallocatePool(ValueSize, CurrentValueSize, CurrentItemPackage->Value);
jcarsey49bd4982012-01-27 18:42:43 +00002098 ASSERT(CurrentItemPackage->Value != NULL);
2099 if (ValueSize == 0) {
Jaben Carsey98c16be2014-08-19 21:00:34 +00002100 StrnCpy(CurrentItemPackage->Value, L"\"", CurrentValueSize/sizeof(CHAR16) - 1);
2101 StrnCat(CurrentItemPackage->Value, Argv[LoopCounter], CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
2102 StrnCat(CurrentItemPackage->Value, L"\"", CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
jcarsey49bd4982012-01-27 18:42:43 +00002103 } else {
Jaben Carsey98c16be2014-08-19 21:00:34 +00002104 StrnCat(CurrentItemPackage->Value, L" ", CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
2105 StrnCat(CurrentItemPackage->Value, L"\"", CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
2106 StrnCat(CurrentItemPackage->Value, Argv[LoopCounter], CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
2107 StrnCat(CurrentItemPackage->Value, L"\"", CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
jcarsey49bd4982012-01-27 18:42:43 +00002108 }
2109 ValueSize += StrSize(Argv[LoopCounter]) + sizeof(CHAR16);
jcarsey125c2cf2009-11-18 21:36:50 +00002110 }
jcarsey125c2cf2009-11-18 21:36:50 +00002111 GetItemValue--;
2112 if (GetItemValue == 0) {
2113 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2114 }
jcarseya405b862010-09-14 05:18:09 +00002115 } else if (!InternalIsFlag(Argv[LoopCounter], AlwaysAllowNumbers) ){ //|| ProblemParam == NULL) {
jcarseyb1f95a02009-06-16 00:23:19 +00002116 //
2117 // add this one as a non-flag
2118 //
jcarsey252d9452011-03-25 20:49:53 +00002119
2120 TempPointer = Argv[LoopCounter];
darylm503b0934ac2011-11-12 00:35:11 +00002121 if ((*TempPointer == L'^' && *(TempPointer+1) == L'-')
jcarsey252d9452011-03-25 20:49:53 +00002122 || (*TempPointer == L'^' && *(TempPointer+1) == L'/')
2123 || (*TempPointer == L'^' && *(TempPointer+1) == L'+')
2124 ){
2125 TempPointer++;
2126 }
2127 CurrentItemPackage = AllocateZeroPool(sizeof(SHELL_PARAM_PACKAGE));
jcarseybeab0fc2011-10-10 17:26:25 +00002128 if (CurrentItemPackage == NULL) {
2129 ShellCommandLineFreeVarList(*CheckPackage);
2130 *CheckPackage = NULL;
2131 return (EFI_OUT_OF_RESOURCES);
2132 }
jcarseyb1f95a02009-06-16 00:23:19 +00002133 CurrentItemPackage->Name = NULL;
2134 CurrentItemPackage->Type = TypePosition;
Jaben Carsey98c16be2014-08-19 21:00:34 +00002135 CurrentItemPackage->Value = AllocateCopyPool(StrSize(TempPointer), TempPointer);
jcarseybeab0fc2011-10-10 17:26:25 +00002136 if (CurrentItemPackage->Value == NULL) {
2137 ShellCommandLineFreeVarList(*CheckPackage);
2138 *CheckPackage = NULL;
2139 return (EFI_OUT_OF_RESOURCES);
2140 }
jcarseya405b862010-09-14 05:18:09 +00002141 CurrentItemPackage->OriginalPosition = Count++;
jcarsey9b3bf082009-06-23 21:15:07 +00002142 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
jcarsey252d9452011-03-25 20:49:53 +00002143 } else {
jcarsey94b17fa2009-05-07 18:46:18 +00002144 //
2145 // this was a non-recognised flag... error!
2146 //
jcarsey252d9452011-03-25 20:49:53 +00002147 if (ProblemParam != NULL) {
Jaben Carsey98c16be2014-08-19 21:00:34 +00002148 *ProblemParam = AllocateCopyPool(StrSize(Argv[LoopCounter]), Argv[LoopCounter]);
jcarsey252d9452011-03-25 20:49:53 +00002149 }
jcarsey94b17fa2009-05-07 18:46:18 +00002150 ShellCommandLineFreeVarList(*CheckPackage);
2151 *CheckPackage = NULL;
2152 return (EFI_VOLUME_CORRUPTED);
jcarsey94b17fa2009-05-07 18:46:18 +00002153 }
2154 }
jcarsey125c2cf2009-11-18 21:36:50 +00002155 if (GetItemValue != 0) {
2156 GetItemValue = 0;
2157 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2158 }
jcarsey94b17fa2009-05-07 18:46:18 +00002159 //
2160 // support for AutoPageBreak
2161 //
2162 if (AutoPageBreak && ShellCommandLineGetFlag(*CheckPackage, L"-b")) {
2163 ShellSetPageBreakMode(TRUE);
2164 }
2165 return (EFI_SUCCESS);
2166}
2167
2168/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002169 Checks the command line arguments passed against the list of valid ones.
jcarsey94b17fa2009-05-07 18:46:18 +00002170 Optionally removes NULL values first.
jcarsey1e6e84c2010-01-25 20:05:08 +00002171
jcarsey94b17fa2009-05-07 18:46:18 +00002172 If no initialization is required, then return RETURN_SUCCESS.
jcarsey1e6e84c2010-01-25 20:05:08 +00002173
jcarseya405b862010-09-14 05:18:09 +00002174 @param[in] CheckList The pointer to list of parameters to check.
2175 @param[out] CheckPackage The package of checked values.
2176 @param[out] ProblemParam Optional pointer to pointer to unicode string for
jcarsey94b17fa2009-05-07 18:46:18 +00002177 the paramater that caused failure.
jcarseya405b862010-09-14 05:18:09 +00002178 @param[in] AutoPageBreak Will automatically set PageBreakEnabled.
2179 @param[in] AlwaysAllowNumbers Will never fail for number based flags.
jcarsey94b17fa2009-05-07 18:46:18 +00002180
2181 @retval EFI_SUCCESS The operation completed sucessfully.
jcarseya405b862010-09-14 05:18:09 +00002182 @retval EFI_OUT_OF_RESOURCES A memory allocation failed.
2183 @retval EFI_INVALID_PARAMETER A parameter was invalid.
2184 @retval EFI_VOLUME_CORRUPTED The command line was corrupt.
2185 @retval EFI_DEVICE_ERROR The commands contained 2 opposing arguments. One
jcarsey1e6e84c2010-01-25 20:05:08 +00002186 of the command line arguments was returned in
jcarsey94b17fa2009-05-07 18:46:18 +00002187 ProblemParam if provided.
jcarseya405b862010-09-14 05:18:09 +00002188 @retval EFI_NOT_FOUND A argument required a value that was missing.
2189 The invalid command line argument was returned in
jcarsey94b17fa2009-05-07 18:46:18 +00002190 ProblemParam if provided.
2191**/
2192EFI_STATUS
2193EFIAPI
jcarsey2247dde2009-11-09 18:08:58 +00002194ShellCommandLineParseEx (
jcarsey94b17fa2009-05-07 18:46:18 +00002195 IN CONST SHELL_PARAM_ITEM *CheckList,
2196 OUT LIST_ENTRY **CheckPackage,
2197 OUT CHAR16 **ProblemParam OPTIONAL,
jcarsey2247dde2009-11-09 18:08:58 +00002198 IN BOOLEAN AutoPageBreak,
2199 IN BOOLEAN AlwaysAllowNumbers
jcarseya405b862010-09-14 05:18:09 +00002200 )
2201{
jcarsey1e6e84c2010-01-25 20:05:08 +00002202 //
jcarsey94b17fa2009-05-07 18:46:18 +00002203 // ASSERT that CheckList and CheckPackage aren't NULL
2204 //
2205 ASSERT(CheckList != NULL);
2206 ASSERT(CheckPackage != NULL);
2207
jcarsey1e6e84c2010-01-25 20:05:08 +00002208 //
jcarsey94b17fa2009-05-07 18:46:18 +00002209 // Check for UEFI Shell 2.0 protocols
2210 //
jcarsey366f81a2011-06-27 21:04:22 +00002211 if (gEfiShellParametersProtocol != NULL) {
jcarsey1e6e84c2010-01-25 20:05:08 +00002212 return (InternalCommandLineParse(CheckList,
2213 CheckPackage,
2214 ProblemParam,
2215 AutoPageBreak,
jcarsey366f81a2011-06-27 21:04:22 +00002216 (CONST CHAR16**) gEfiShellParametersProtocol->Argv,
2217 gEfiShellParametersProtocol->Argc,
jcarsey2247dde2009-11-09 18:08:58 +00002218 AlwaysAllowNumbers));
jcarsey94b17fa2009-05-07 18:46:18 +00002219 }
2220
jcarsey1e6e84c2010-01-25 20:05:08 +00002221 //
jcarsey94b17fa2009-05-07 18:46:18 +00002222 // ASSERT That EFI Shell is not required
2223 //
2224 ASSERT (mEfiShellInterface != NULL);
jcarsey1e6e84c2010-01-25 20:05:08 +00002225 return (InternalCommandLineParse(CheckList,
2226 CheckPackage,
2227 ProblemParam,
2228 AutoPageBreak,
jljusten08d7f8e2009-06-15 18:42:13 +00002229 (CONST CHAR16**) mEfiShellInterface->Argv,
jcarsey2247dde2009-11-09 18:08:58 +00002230 mEfiShellInterface->Argc,
2231 AlwaysAllowNumbers));
jcarsey94b17fa2009-05-07 18:46:18 +00002232}
2233
2234/**
2235 Frees shell variable list that was returned from ShellCommandLineParse.
2236
2237 This function will free all the memory that was used for the CheckPackage
2238 list of postprocessed shell arguments.
2239
2240 this function has no return value.
2241
2242 if CheckPackage is NULL, then return
2243
2244 @param CheckPackage the list to de-allocate
2245 **/
2246VOID
2247EFIAPI
2248ShellCommandLineFreeVarList (
2249 IN LIST_ENTRY *CheckPackage
jcarseya405b862010-09-14 05:18:09 +00002250 )
2251{
jcarsey94b17fa2009-05-07 18:46:18 +00002252 LIST_ENTRY *Node;
2253
2254 //
2255 // check for CheckPackage == NULL
2256 //
2257 if (CheckPackage == NULL) {
2258 return;
2259 }
2260
2261 //
2262 // for each node in the list
2263 //
jcarsey9eb53ac2009-07-08 17:26:58 +00002264 for ( Node = GetFirstNode(CheckPackage)
jcarseya405b862010-09-14 05:18:09 +00002265 ; !IsListEmpty(CheckPackage)
jcarsey9eb53ac2009-07-08 17:26:58 +00002266 ; Node = GetFirstNode(CheckPackage)
jcarseya405b862010-09-14 05:18:09 +00002267 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002268 //
2269 // Remove it from the list
2270 //
2271 RemoveEntryList(Node);
2272
2273 //
2274 // if it has a name free the name
2275 //
2276 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
2277 FreePool(((SHELL_PARAM_PACKAGE*)Node)->Name);
2278 }
2279
2280 //
2281 // if it has a value free the value
2282 //
2283 if (((SHELL_PARAM_PACKAGE*)Node)->Value != NULL) {
2284 FreePool(((SHELL_PARAM_PACKAGE*)Node)->Value);
2285 }
jcarsey1e6e84c2010-01-25 20:05:08 +00002286
jcarsey94b17fa2009-05-07 18:46:18 +00002287 //
2288 // free the node structure
2289 //
2290 FreePool((SHELL_PARAM_PACKAGE*)Node);
2291 }
2292 //
2293 // free the list head node
2294 //
2295 FreePool(CheckPackage);
2296}
2297/**
2298 Checks for presence of a flag parameter
2299
2300 flag arguments are in the form of "-<Key>" or "/<Key>", but do not have a value following the key
2301
2302 if CheckPackage is NULL then return FALSE.
2303 if KeyString is NULL then ASSERT()
jcarsey1e6e84c2010-01-25 20:05:08 +00002304
jcarsey94b17fa2009-05-07 18:46:18 +00002305 @param CheckPackage The package of parsed command line arguments
2306 @param KeyString the Key of the command line argument to check for
2307
2308 @retval TRUE the flag is on the command line
2309 @retval FALSE the flag is not on the command line
2310 **/
2311BOOLEAN
2312EFIAPI
2313ShellCommandLineGetFlag (
jcarseya405b862010-09-14 05:18:09 +00002314 IN CONST LIST_ENTRY * CONST CheckPackage,
2315 IN CONST CHAR16 * CONST KeyString
2316 )
2317{
jcarsey94b17fa2009-05-07 18:46:18 +00002318 LIST_ENTRY *Node;
jcarsey252d9452011-03-25 20:49:53 +00002319 CHAR16 *TempString;
jcarsey94b17fa2009-05-07 18:46:18 +00002320
2321 //
ydong100c1950b2011-10-13 02:37:35 +00002322 // return FALSE for no package or KeyString is NULL
jcarsey94b17fa2009-05-07 18:46:18 +00002323 //
ydong100c1950b2011-10-13 02:37:35 +00002324 if (CheckPackage == NULL || KeyString == NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00002325 return (FALSE);
2326 }
2327
2328 //
2329 // enumerate through the list of parametrs
2330 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002331 for ( Node = GetFirstNode(CheckPackage)
2332 ; !IsNull (CheckPackage, Node)
2333 ; Node = GetNextNode(CheckPackage, Node)
jcarsey252d9452011-03-25 20:49:53 +00002334 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002335 //
2336 // If the Name matches, return TRUE (and there may be NULL name)
2337 //
2338 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
jcarsey9eb53ac2009-07-08 17:26:58 +00002339 //
2340 // If Type is TypeStart then only compare the begining of the strings
2341 //
jcarsey252d9452011-03-25 20:49:53 +00002342 if (((SHELL_PARAM_PACKAGE*)Node)->Type == TypeStart) {
2343 if (StrnCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name, StrLen(KeyString)) == 0) {
2344 return (TRUE);
2345 }
2346 TempString = NULL;
2347 TempString = StrnCatGrow(&TempString, NULL, KeyString, StrLen(((SHELL_PARAM_PACKAGE*)Node)->Name));
2348 if (TempString != NULL) {
2349 if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2350 FreePool(TempString);
2351 return (TRUE);
2352 }
2353 FreePool(TempString);
2354 }
2355 } else if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
jcarsey94b17fa2009-05-07 18:46:18 +00002356 return (TRUE);
2357 }
2358 }
2359 }
2360 return (FALSE);
2361}
2362/**
jcarseya405b862010-09-14 05:18:09 +00002363 Returns value from command line argument.
jcarsey94b17fa2009-05-07 18:46:18 +00002364
jcarseya405b862010-09-14 05:18:09 +00002365 Value parameters are in the form of "-<Key> value" or "/<Key> value".
jcarsey1e6e84c2010-01-25 20:05:08 +00002366
jcarseya405b862010-09-14 05:18:09 +00002367 If CheckPackage is NULL, then return NULL.
jcarsey94b17fa2009-05-07 18:46:18 +00002368
jcarseya405b862010-09-14 05:18:09 +00002369 @param[in] CheckPackage The package of parsed command line arguments.
2370 @param[in] KeyString The Key of the command line argument to check for.
jcarsey94b17fa2009-05-07 18:46:18 +00002371
jcarseya405b862010-09-14 05:18:09 +00002372 @retval NULL The flag is not on the command line.
2373 @retval !=NULL The pointer to unicode string of the value.
2374**/
jcarsey94b17fa2009-05-07 18:46:18 +00002375CONST CHAR16*
2376EFIAPI
2377ShellCommandLineGetValue (
2378 IN CONST LIST_ENTRY *CheckPackage,
2379 IN CHAR16 *KeyString
jcarseya405b862010-09-14 05:18:09 +00002380 )
2381{
jcarsey94b17fa2009-05-07 18:46:18 +00002382 LIST_ENTRY *Node;
jcarsey252d9452011-03-25 20:49:53 +00002383 CHAR16 *TempString;
jcarsey94b17fa2009-05-07 18:46:18 +00002384
2385 //
ydong100c1950b2011-10-13 02:37:35 +00002386 // return NULL for no package or KeyString is NULL
jcarsey94b17fa2009-05-07 18:46:18 +00002387 //
ydong100c1950b2011-10-13 02:37:35 +00002388 if (CheckPackage == NULL || KeyString == NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00002389 return (NULL);
2390 }
2391
2392 //
2393 // enumerate through the list of parametrs
2394 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002395 for ( Node = GetFirstNode(CheckPackage)
2396 ; !IsNull (CheckPackage, Node)
2397 ; Node = GetNextNode(CheckPackage, Node)
jcarsey252d9452011-03-25 20:49:53 +00002398 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002399 //
jcarsey252d9452011-03-25 20:49:53 +00002400 // If the Name matches, return TRUE (and there may be NULL name)
jcarsey94b17fa2009-05-07 18:46:18 +00002401 //
2402 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
jcarsey9eb53ac2009-07-08 17:26:58 +00002403 //
2404 // If Type is TypeStart then only compare the begining of the strings
2405 //
jcarsey252d9452011-03-25 20:49:53 +00002406 if (((SHELL_PARAM_PACKAGE*)Node)->Type == TypeStart) {
2407 if (StrnCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name, StrLen(KeyString)) == 0) {
2408 return (((SHELL_PARAM_PACKAGE*)Node)->Name + StrLen(KeyString));
2409 }
2410 TempString = NULL;
2411 TempString = StrnCatGrow(&TempString, NULL, KeyString, StrLen(((SHELL_PARAM_PACKAGE*)Node)->Name));
2412 if (TempString != NULL) {
2413 if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2414 FreePool(TempString);
2415 return (((SHELL_PARAM_PACKAGE*)Node)->Name + StrLen(KeyString));
2416 }
2417 FreePool(TempString);
2418 }
2419 } else if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
jcarsey94b17fa2009-05-07 18:46:18 +00002420 return (((SHELL_PARAM_PACKAGE*)Node)->Value);
2421 }
2422 }
2423 }
2424 return (NULL);
2425}
jcarseya405b862010-09-14 05:18:09 +00002426
jcarsey94b17fa2009-05-07 18:46:18 +00002427/**
jcarseya405b862010-09-14 05:18:09 +00002428 Returns raw value from command line argument.
jcarsey94b17fa2009-05-07 18:46:18 +00002429
jcarseya405b862010-09-14 05:18:09 +00002430 Raw value parameters are in the form of "value" in a specific position in the list.
jcarsey1e6e84c2010-01-25 20:05:08 +00002431
jcarseya405b862010-09-14 05:18:09 +00002432 If CheckPackage is NULL, then return NULL.
jcarsey94b17fa2009-05-07 18:46:18 +00002433
jcarseya405b862010-09-14 05:18:09 +00002434 @param[in] CheckPackage The package of parsed command line arguments.
2435 @param[in] Position The position of the value.
jcarsey94b17fa2009-05-07 18:46:18 +00002436
jcarseya405b862010-09-14 05:18:09 +00002437 @retval NULL The flag is not on the command line.
2438 @retval !=NULL The pointer to unicode string of the value.
jcarsey94b17fa2009-05-07 18:46:18 +00002439 **/
2440CONST CHAR16*
2441EFIAPI
2442ShellCommandLineGetRawValue (
jcarseya405b862010-09-14 05:18:09 +00002443 IN CONST LIST_ENTRY * CONST CheckPackage,
2444 IN UINTN Position
2445 )
2446{
jcarsey94b17fa2009-05-07 18:46:18 +00002447 LIST_ENTRY *Node;
2448
2449 //
2450 // check for CheckPackage == NULL
2451 //
2452 if (CheckPackage == NULL) {
2453 return (NULL);
2454 }
2455
2456 //
2457 // enumerate through the list of parametrs
2458 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002459 for ( Node = GetFirstNode(CheckPackage)
2460 ; !IsNull (CheckPackage, Node)
2461 ; Node = GetNextNode(CheckPackage, Node)
jcarseya405b862010-09-14 05:18:09 +00002462 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002463 //
2464 // If the position matches, return the value
2465 //
2466 if (((SHELL_PARAM_PACKAGE*)Node)->OriginalPosition == Position) {
2467 return (((SHELL_PARAM_PACKAGE*)Node)->Value);
2468 }
2469 }
2470 return (NULL);
jcarseyb1f95a02009-06-16 00:23:19 +00002471}
jcarsey2247dde2009-11-09 18:08:58 +00002472
2473/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002474 returns the number of command line value parameters that were parsed.
2475
jcarsey2247dde2009-11-09 18:08:58 +00002476 this will not include flags.
2477
jcarseya405b862010-09-14 05:18:09 +00002478 @param[in] CheckPackage The package of parsed command line arguments.
2479
jcarsey2247dde2009-11-09 18:08:58 +00002480 @retval (UINTN)-1 No parsing has ocurred
2481 @return other The number of value parameters found
2482**/
2483UINTN
2484EFIAPI
2485ShellCommandLineGetCount(
jcarseya405b862010-09-14 05:18:09 +00002486 IN CONST LIST_ENTRY *CheckPackage
jcarsey125c2cf2009-11-18 21:36:50 +00002487 )
2488{
jcarseya405b862010-09-14 05:18:09 +00002489 LIST_ENTRY *Node1;
2490 UINTN Count;
2491
2492 if (CheckPackage == NULL) {
2493 return (0);
2494 }
2495 for ( Node1 = GetFirstNode(CheckPackage), Count = 0
2496 ; !IsNull (CheckPackage, Node1)
2497 ; Node1 = GetNextNode(CheckPackage, Node1)
2498 ){
2499 if (((SHELL_PARAM_PACKAGE*)Node1)->Name == NULL) {
2500 Count++;
2501 }
2502 }
2503 return (Count);
jcarsey2247dde2009-11-09 18:08:58 +00002504}
2505
jcarsey975136a2009-06-16 19:03:54 +00002506/**
jcarsey36a9d672009-11-20 21:13:41 +00002507 Determins if a parameter is duplicated.
2508
jcarsey1e6e84c2010-01-25 20:05:08 +00002509 If Param is not NULL then it will point to a callee allocated string buffer
jcarsey36a9d672009-11-20 21:13:41 +00002510 with the parameter value if a duplicate is found.
2511
2512 If CheckPackage is NULL, then ASSERT.
2513
2514 @param[in] CheckPackage The package of parsed command line arguments.
2515 @param[out] Param Upon finding one, a pointer to the duplicated parameter.
2516
2517 @retval EFI_SUCCESS No parameters were duplicated.
2518 @retval EFI_DEVICE_ERROR A duplicate was found.
2519 **/
2520EFI_STATUS
2521EFIAPI
2522ShellCommandLineCheckDuplicate (
2523 IN CONST LIST_ENTRY *CheckPackage,
2524 OUT CHAR16 **Param
2525 )
2526{
2527 LIST_ENTRY *Node1;
2528 LIST_ENTRY *Node2;
jcarsey1e6e84c2010-01-25 20:05:08 +00002529
jcarsey36a9d672009-11-20 21:13:41 +00002530 ASSERT(CheckPackage != NULL);
2531
jcarsey1e6e84c2010-01-25 20:05:08 +00002532 for ( Node1 = GetFirstNode(CheckPackage)
2533 ; !IsNull (CheckPackage, Node1)
2534 ; Node1 = GetNextNode(CheckPackage, Node1)
jcarseya405b862010-09-14 05:18:09 +00002535 ){
jcarsey1e6e84c2010-01-25 20:05:08 +00002536 for ( Node2 = GetNextNode(CheckPackage, Node1)
2537 ; !IsNull (CheckPackage, Node2)
2538 ; Node2 = GetNextNode(CheckPackage, Node2)
jcarseya405b862010-09-14 05:18:09 +00002539 ){
2540 if ((((SHELL_PARAM_PACKAGE*)Node1)->Name != NULL) && (((SHELL_PARAM_PACKAGE*)Node2)->Name != NULL) && StrCmp(((SHELL_PARAM_PACKAGE*)Node1)->Name, ((SHELL_PARAM_PACKAGE*)Node2)->Name) == 0) {
jcarsey36a9d672009-11-20 21:13:41 +00002541 if (Param != NULL) {
2542 *Param = NULL;
2543 *Param = StrnCatGrow(Param, NULL, ((SHELL_PARAM_PACKAGE*)Node1)->Name, 0);
2544 }
2545 return (EFI_DEVICE_ERROR);
2546 }
2547 }
2548 }
2549 return (EFI_SUCCESS);
2550}
2551
2552/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002553 This is a find and replace function. Upon successful return the NewString is a copy of
jcarsey975136a2009-06-16 19:03:54 +00002554 SourceString with each instance of FindTarget replaced with ReplaceWith.
2555
jcarseyb3011f42010-01-11 21:49:04 +00002556 If SourceString and NewString overlap the behavior is undefined.
2557
jcarsey975136a2009-06-16 19:03:54 +00002558 If the string would grow bigger than NewSize it will halt and return error.
2559
ydong104ff7e372011-09-02 08:05:34 +00002560 @param[in] SourceString The string with source buffer.
2561 @param[in, out] NewString The string with resultant buffer.
2562 @param[in] NewSize The size in bytes of NewString.
2563 @param[in] FindTarget The string to look for.
2564 @param[in] ReplaceWith The string to replace FindTarget with.
2565 @param[in] SkipPreCarrot If TRUE will skip a FindTarget that has a '^'
2566 immediately before it.
2567 @param[in] ParameterReplacing If TRUE will add "" around items with spaces.
jcarsey975136a2009-06-16 19:03:54 +00002568
jcarsey969c7832010-01-13 16:46:33 +00002569 @retval EFI_INVALID_PARAMETER SourceString was NULL.
2570 @retval EFI_INVALID_PARAMETER NewString was NULL.
2571 @retval EFI_INVALID_PARAMETER FindTarget was NULL.
2572 @retval EFI_INVALID_PARAMETER ReplaceWith was NULL.
2573 @retval EFI_INVALID_PARAMETER FindTarget had length < 1.
2574 @retval EFI_INVALID_PARAMETER SourceString had length < 1.
jcarsey1e6e84c2010-01-25 20:05:08 +00002575 @retval EFI_BUFFER_TOO_SMALL NewSize was less than the minimum size to hold
jcarsey969c7832010-01-13 16:46:33 +00002576 the new string (truncation occurred).
jcarseya405b862010-09-14 05:18:09 +00002577 @retval EFI_SUCCESS The string was successfully copied with replacement.
jcarsey975136a2009-06-16 19:03:54 +00002578**/
jcarsey975136a2009-06-16 19:03:54 +00002579EFI_STATUS
2580EFIAPI
jcarseya405b862010-09-14 05:18:09 +00002581ShellCopySearchAndReplace(
jcarsey975136a2009-06-16 19:03:54 +00002582 IN CHAR16 CONST *SourceString,
jcarseya405b862010-09-14 05:18:09 +00002583 IN OUT CHAR16 *NewString,
jcarsey975136a2009-06-16 19:03:54 +00002584 IN UINTN NewSize,
2585 IN CONST CHAR16 *FindTarget,
jcarsey969c7832010-01-13 16:46:33 +00002586 IN CONST CHAR16 *ReplaceWith,
jcarseya405b862010-09-14 05:18:09 +00002587 IN CONST BOOLEAN SkipPreCarrot,
2588 IN CONST BOOLEAN ParameterReplacing
jcarsey1e6e84c2010-01-25 20:05:08 +00002589 )
jcarsey2247dde2009-11-09 18:08:58 +00002590{
jcarsey01582942009-07-10 19:46:17 +00002591 UINTN Size;
jcarseya405b862010-09-14 05:18:09 +00002592 CHAR16 *Replace;
2593
jcarsey975136a2009-06-16 19:03:54 +00002594 if ( (SourceString == NULL)
2595 || (NewString == NULL)
2596 || (FindTarget == NULL)
2597 || (ReplaceWith == NULL)
2598 || (StrLen(FindTarget) < 1)
2599 || (StrLen(SourceString) < 1)
jcarseya405b862010-09-14 05:18:09 +00002600 ){
jcarsey975136a2009-06-16 19:03:54 +00002601 return (EFI_INVALID_PARAMETER);
2602 }
jcarseya405b862010-09-14 05:18:09 +00002603 Replace = NULL;
2604 if (StrStr(ReplaceWith, L" ") == NULL || !ParameterReplacing) {
2605 Replace = StrnCatGrow(&Replace, NULL, ReplaceWith, 0);
2606 } else {
2607 Replace = AllocateZeroPool(StrSize(ReplaceWith) + 2*sizeof(CHAR16));
jcarseybeab0fc2011-10-10 17:26:25 +00002608 if (Replace != NULL) {
2609 UnicodeSPrint(Replace, StrSize(ReplaceWith) + 2*sizeof(CHAR16), L"\"%s\"", ReplaceWith);
2610 }
jcarseya405b862010-09-14 05:18:09 +00002611 }
jcarsey3e082d52010-10-04 16:44:57 +00002612 if (Replace == NULL) {
2613 return (EFI_OUT_OF_RESOURCES);
2614 }
Jaben Carsey98c16be2014-08-19 21:00:34 +00002615 NewString = ZeroMem(NewString, NewSize);
jcarsey2247dde2009-11-09 18:08:58 +00002616 while (*SourceString != CHAR_NULL) {
jcarsey969c7832010-01-13 16:46:33 +00002617 //
jcarseya405b862010-09-14 05:18:09 +00002618 // if we find the FindTarget and either Skip == FALSE or Skip and we
jcarsey969c7832010-01-13 16:46:33 +00002619 // dont have a carrot do a replace...
2620 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002621 if (StrnCmp(SourceString, FindTarget, StrLen(FindTarget)) == 0
jcarseya405b862010-09-14 05:18:09 +00002622 && ((SkipPreCarrot && *(SourceString-1) != L'^') || !SkipPreCarrot)
2623 ){
jcarsey975136a2009-06-16 19:03:54 +00002624 SourceString += StrLen(FindTarget);
jcarsey01582942009-07-10 19:46:17 +00002625 Size = StrSize(NewString);
jcarseya405b862010-09-14 05:18:09 +00002626 if ((Size + (StrLen(Replace)*sizeof(CHAR16))) > NewSize) {
2627 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002628 return (EFI_BUFFER_TOO_SMALL);
2629 }
Jaben Carsey98c16be2014-08-19 21:00:34 +00002630 StrnCat(NewString, Replace, NewSize/sizeof(CHAR16) - 1 - StrLen(NewString));
jcarsey975136a2009-06-16 19:03:54 +00002631 } else {
jcarsey01582942009-07-10 19:46:17 +00002632 Size = StrSize(NewString);
2633 if (Size + sizeof(CHAR16) > NewSize) {
jcarseya405b862010-09-14 05:18:09 +00002634 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002635 return (EFI_BUFFER_TOO_SMALL);
2636 }
2637 StrnCat(NewString, SourceString, 1);
2638 SourceString++;
2639 }
2640 }
jcarseya405b862010-09-14 05:18:09 +00002641 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002642 return (EFI_SUCCESS);
2643}
jcarseyb1f95a02009-06-16 00:23:19 +00002644
2645/**
jcarseye2f82972009-12-01 05:40:24 +00002646 Internal worker function to output a string.
2647
2648 This function will output a string to the correct StdOut.
2649
2650 @param[in] String The string to print out.
2651
2652 @retval EFI_SUCCESS The operation was sucessful.
2653 @retval !EFI_SUCCESS The operation failed.
2654**/
2655EFI_STATUS
2656EFIAPI
2657InternalPrintTo (
2658 IN CONST CHAR16 *String
2659 )
2660{
2661 UINTN Size;
2662 Size = StrSize(String) - sizeof(CHAR16);
jcarseya405b862010-09-14 05:18:09 +00002663 if (Size == 0) {
2664 return (EFI_SUCCESS);
2665 }
jcarsey366f81a2011-06-27 21:04:22 +00002666 if (gEfiShellParametersProtocol != NULL) {
2667 return (gEfiShellProtocol->WriteFile(gEfiShellParametersProtocol->StdOut, &Size, (VOID*)String));
jcarseye2f82972009-12-01 05:40:24 +00002668 }
2669 if (mEfiShellInterface != NULL) {
jcarsey06c355b2012-03-26 21:00:39 +00002670 if (mEfiShellInterface->RedirArgc == 0) {
jcarsey49bd4982012-01-27 18:42:43 +00002671 //
2672 // Divide in half for old shell. Must be string length not size.
jcarsey06c355b2012-03-26 21:00:39 +00002673 //
2674 Size /=2; // Divide in half only when no redirection.
2675 }
jcarseya405b862010-09-14 05:18:09 +00002676 return (mEfiShellInterface->StdOut->Write(mEfiShellInterface->StdOut, &Size, (VOID*)String));
jcarseye2f82972009-12-01 05:40:24 +00002677 }
2678 ASSERT(FALSE);
2679 return (EFI_UNSUPPORTED);
2680}
2681
2682/**
jcarseyb1f95a02009-06-16 00:23:19 +00002683 Print at a specific location on the screen.
2684
jcarseyf1b87e72009-06-17 00:52:11 +00002685 This function will move the cursor to a given screen location and print the specified string
jcarsey1e6e84c2010-01-25 20:05:08 +00002686
2687 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarseyf1b87e72009-06-17 00:52:11 +00002688 will be used.
jcarseyb1f95a02009-06-16 00:23:19 +00002689
2690 if either Row or Col is out of range for the current console, then ASSERT
2691 if Format is NULL, then ASSERT
2692
jcarsey1e6e84c2010-01-25 20:05:08 +00002693 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarseyb1f95a02009-06-16 00:23:19 +00002694 the following additional flags:
2695 %N - Set output attribute to normal
2696 %H - Set output attribute to highlight
2697 %E - Set output attribute to error
2698 %B - Set output attribute to blue color
2699 %V - Set output attribute to green color
2700
2701 Note: The background color is controlled by the shell command cls.
2702
jcarseyb1f95a02009-06-16 00:23:19 +00002703 @param[in] Col the column to print at
jcarsey252d9452011-03-25 20:49:53 +00002704 @param[in] Row the row to print at
jcarseyb1f95a02009-06-16 00:23:19 +00002705 @param[in] Format the format string
jcarsey2247dde2009-11-09 18:08:58 +00002706 @param[in] Marker the marker for the variable argument list
jcarseyb1f95a02009-06-16 00:23:19 +00002707
jcarseya405b862010-09-14 05:18:09 +00002708 @return EFI_SUCCESS The operation was successful.
2709 @return EFI_DEVICE_ERROR The console device reported an error.
jcarseyb1f95a02009-06-16 00:23:19 +00002710**/
jcarseya405b862010-09-14 05:18:09 +00002711EFI_STATUS
jcarseyb1f95a02009-06-16 00:23:19 +00002712EFIAPI
jcarsey2247dde2009-11-09 18:08:58 +00002713InternalShellPrintWorker(
jcarseyb1f95a02009-06-16 00:23:19 +00002714 IN INT32 Col OPTIONAL,
2715 IN INT32 Row OPTIONAL,
2716 IN CONST CHAR16 *Format,
jcarsey252d9452011-03-25 20:49:53 +00002717 IN VA_LIST Marker
jcarsey1e6e84c2010-01-25 20:05:08 +00002718 )
jcarsey2247dde2009-11-09 18:08:58 +00002719{
jcarseyb1f95a02009-06-16 00:23:19 +00002720 EFI_STATUS Status;
jcarsey975136a2009-06-16 19:03:54 +00002721 CHAR16 *ResumeLocation;
2722 CHAR16 *FormatWalker;
jcarseya405b862010-09-14 05:18:09 +00002723 UINTN OriginalAttribute;
jcarsey89e85372011-04-13 23:37:50 +00002724 CHAR16 *mPostReplaceFormat;
2725 CHAR16 *mPostReplaceFormat2;
2726
2727 mPostReplaceFormat = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
2728 mPostReplaceFormat2 = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
jcarseya405b862010-09-14 05:18:09 +00002729
jcarseyf8d3e682011-04-19 17:54:42 +00002730 if (mPostReplaceFormat == NULL || mPostReplaceFormat2 == NULL) {
2731 SHELL_FREE_NON_NULL(mPostReplaceFormat);
2732 SHELL_FREE_NON_NULL(mPostReplaceFormat2);
2733 return (EFI_OUT_OF_RESOURCES);
2734 }
2735
jcarseya405b862010-09-14 05:18:09 +00002736 Status = EFI_SUCCESS;
2737 OriginalAttribute = gST->ConOut->Mode->Attribute;
jcarsey1e6e84c2010-01-25 20:05:08 +00002738
jcarsey975136a2009-06-16 19:03:54 +00002739 //
2740 // Back and forth each time fixing up 1 of our flags...
2741 //
jcarseya405b862010-09-14 05:18:09 +00002742 Status = ShellCopySearchAndReplace(Format, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%N", L"%%N", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002743 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002744 Status = ShellCopySearchAndReplace(mPostReplaceFormat, mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), L"%E", L"%%E", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002745 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002746 Status = ShellCopySearchAndReplace(mPostReplaceFormat2, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%H", L"%%H", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002747 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002748 Status = ShellCopySearchAndReplace(mPostReplaceFormat, mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), L"%B", L"%%B", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002749 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002750 Status = ShellCopySearchAndReplace(mPostReplaceFormat2, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%V", L"%%V", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002751 ASSERT_EFI_ERROR(Status);
2752
2753 //
2754 // Use the last buffer from replacing to print from...
2755 //
jcarseya405b862010-09-14 05:18:09 +00002756 UnicodeVSPrint (mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), mPostReplaceFormat, Marker);
jcarseyb1f95a02009-06-16 00:23:19 +00002757
2758 if (Col != -1 && Row != -1) {
jcarseyb1f95a02009-06-16 00:23:19 +00002759 Status = gST->ConOut->SetCursorPosition(gST->ConOut, Col, Row);
jcarsey975136a2009-06-16 19:03:54 +00002760 }
2761
jcarseyecd3d592009-12-07 18:05:00 +00002762 FormatWalker = mPostReplaceFormat2;
jcarsey2247dde2009-11-09 18:08:58 +00002763 while (*FormatWalker != CHAR_NULL) {
jcarsey975136a2009-06-16 19:03:54 +00002764 //
2765 // Find the next attribute change request
2766 //
2767 ResumeLocation = StrStr(FormatWalker, L"%");
2768 if (ResumeLocation != NULL) {
jcarsey2247dde2009-11-09 18:08:58 +00002769 *ResumeLocation = CHAR_NULL;
jcarsey975136a2009-06-16 19:03:54 +00002770 }
2771 //
2772 // print the current FormatWalker string
2773 //
jcarseya405b862010-09-14 05:18:09 +00002774 if (StrLen(FormatWalker)>0) {
2775 Status = InternalPrintTo(FormatWalker);
2776 if (EFI_ERROR(Status)) {
2777 break;
2778 }
2779 }
2780
jcarsey975136a2009-06-16 19:03:54 +00002781 //
2782 // update the attribute
2783 //
2784 if (ResumeLocation != NULL) {
jcarsey5d46f172011-08-23 15:34:23 +00002785 if (*(ResumeLocation-1) == L'^') {
2786 //
jcarsey8bb74412012-01-30 18:44:41 +00002787 // Move cursor back 1 position to overwrite the ^
2788 //
2789 gST->ConOut->SetCursorPosition(gST->ConOut, gST->ConOut->Mode->CursorColumn - 1, gST->ConOut->Mode->CursorRow);
2790
2791 //
jcarsey5d46f172011-08-23 15:34:23 +00002792 // Print a simple '%' symbol
2793 //
2794 Status = InternalPrintTo(L"%");
2795 ResumeLocation = ResumeLocation - 1;
2796 } else {
2797 switch (*(ResumeLocation+1)) {
2798 case (L'N'):
2799 gST->ConOut->SetAttribute(gST->ConOut, OriginalAttribute);
jcarseya405b862010-09-14 05:18:09 +00002800 break;
jcarsey5d46f172011-08-23 15:34:23 +00002801 case (L'E'):
2802 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_YELLOW, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2803 break;
2804 case (L'H'):
2805 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_WHITE, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2806 break;
2807 case (L'B'):
2808 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_BLUE, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2809 break;
2810 case (L'V'):
2811 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_GREEN, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2812 break;
2813 default:
2814 //
2815 // Print a simple '%' symbol
2816 //
2817 Status = InternalPrintTo(L"%");
2818 if (EFI_ERROR(Status)) {
2819 break;
2820 }
2821 ResumeLocation = ResumeLocation - 1;
2822 break;
2823 }
jcarsey975136a2009-06-16 19:03:54 +00002824 }
2825 } else {
2826 //
2827 // reset to normal now...
2828 //
jcarsey975136a2009-06-16 19:03:54 +00002829 break;
2830 }
2831
2832 //
2833 // update FormatWalker to Resume + 2 (skip the % and the indicator)
2834 //
2835 FormatWalker = ResumeLocation + 2;
2836 }
jcarseyb1f95a02009-06-16 00:23:19 +00002837
jcarseya405b862010-09-14 05:18:09 +00002838 gST->ConOut->SetAttribute(gST->ConOut, OriginalAttribute);
jcarsey89e85372011-04-13 23:37:50 +00002839
2840 SHELL_FREE_NON_NULL(mPostReplaceFormat);
2841 SHELL_FREE_NON_NULL(mPostReplaceFormat2);
jcarseya405b862010-09-14 05:18:09 +00002842 return (Status);
jcarsey5f7431d2009-07-10 18:06:01 +00002843}
jcarsey2247dde2009-11-09 18:08:58 +00002844
2845/**
2846 Print at a specific location on the screen.
2847
jcarseye2f82972009-12-01 05:40:24 +00002848 This function will move the cursor to a given screen location and print the specified string.
jcarsey1e6e84c2010-01-25 20:05:08 +00002849
2850 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarsey2247dde2009-11-09 18:08:58 +00002851 will be used.
2852
jcarseye2f82972009-12-01 05:40:24 +00002853 If either Row or Col is out of range for the current console, then ASSERT.
2854 If Format is NULL, then ASSERT.
jcarsey2247dde2009-11-09 18:08:58 +00002855
jcarsey1e6e84c2010-01-25 20:05:08 +00002856 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarsey2247dde2009-11-09 18:08:58 +00002857 the following additional flags:
2858 %N - Set output attribute to normal
2859 %H - Set output attribute to highlight
2860 %E - Set output attribute to error
2861 %B - Set output attribute to blue color
2862 %V - Set output attribute to green color
2863
2864 Note: The background color is controlled by the shell command cls.
2865
jcarsey2247dde2009-11-09 18:08:58 +00002866 @param[in] Col the column to print at
jcarseya405b862010-09-14 05:18:09 +00002867 @param[in] Row the row to print at
jcarsey2247dde2009-11-09 18:08:58 +00002868 @param[in] Format the format string
jcarseya405b862010-09-14 05:18:09 +00002869 @param[in] ... The variable argument list.
jcarsey2247dde2009-11-09 18:08:58 +00002870
jcarseya405b862010-09-14 05:18:09 +00002871 @return EFI_SUCCESS The printing was successful.
2872 @return EFI_DEVICE_ERROR The console device reported an error.
jcarsey2247dde2009-11-09 18:08:58 +00002873**/
jcarseya405b862010-09-14 05:18:09 +00002874EFI_STATUS
jcarsey2247dde2009-11-09 18:08:58 +00002875EFIAPI
2876ShellPrintEx(
2877 IN INT32 Col OPTIONAL,
2878 IN INT32 Row OPTIONAL,
2879 IN CONST CHAR16 *Format,
2880 ...
jcarsey1e6e84c2010-01-25 20:05:08 +00002881 )
jcarsey2247dde2009-11-09 18:08:58 +00002882{
2883 VA_LIST Marker;
jcarseya405b862010-09-14 05:18:09 +00002884 EFI_STATUS RetVal;
jcarsey3e082d52010-10-04 16:44:57 +00002885 if (Format == NULL) {
2886 return (EFI_INVALID_PARAMETER);
2887 }
jcarsey2247dde2009-11-09 18:08:58 +00002888 VA_START (Marker, Format);
jcarseya405b862010-09-14 05:18:09 +00002889 RetVal = InternalShellPrintWorker(Col, Row, Format, Marker);
jcarseye2f82972009-12-01 05:40:24 +00002890 VA_END(Marker);
jcarseya405b862010-09-14 05:18:09 +00002891 return(RetVal);
jcarsey2247dde2009-11-09 18:08:58 +00002892}
2893
2894/**
2895 Print at a specific location on the screen.
2896
jcarseye2f82972009-12-01 05:40:24 +00002897 This function will move the cursor to a given screen location and print the specified string.
jcarsey1e6e84c2010-01-25 20:05:08 +00002898
2899 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarseye2f82972009-12-01 05:40:24 +00002900 will be used.
jcarsey2247dde2009-11-09 18:08:58 +00002901
jcarseye2f82972009-12-01 05:40:24 +00002902 If either Row or Col is out of range for the current console, then ASSERT.
2903 If Format is NULL, then ASSERT.
jcarsey2247dde2009-11-09 18:08:58 +00002904
jcarsey1e6e84c2010-01-25 20:05:08 +00002905 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarsey2247dde2009-11-09 18:08:58 +00002906 the following additional flags:
jcarsey1e6e84c2010-01-25 20:05:08 +00002907 %N - Set output attribute to normal.
2908 %H - Set output attribute to highlight.
2909 %E - Set output attribute to error.
2910 %B - Set output attribute to blue color.
2911 %V - Set output attribute to green color.
jcarsey2247dde2009-11-09 18:08:58 +00002912
2913 Note: The background color is controlled by the shell command cls.
2914
jcarsey1e6e84c2010-01-25 20:05:08 +00002915 @param[in] Col The column to print at.
jcarseya405b862010-09-14 05:18:09 +00002916 @param[in] Row The row to print at.
jcarsey1e6e84c2010-01-25 20:05:08 +00002917 @param[in] Language The language of the string to retrieve. If this parameter
2918 is NULL, then the current platform language is used.
2919 @param[in] HiiFormatStringId The format string Id for getting from Hii.
2920 @param[in] HiiFormatHandle The format string Handle for getting from Hii.
jcarseya405b862010-09-14 05:18:09 +00002921 @param[in] ... The variable argument list.
jcarsey2247dde2009-11-09 18:08:58 +00002922
jcarseya405b862010-09-14 05:18:09 +00002923 @return EFI_SUCCESS The printing was successful.
2924 @return EFI_DEVICE_ERROR The console device reported an error.
jcarsey2247dde2009-11-09 18:08:58 +00002925**/
jcarseya405b862010-09-14 05:18:09 +00002926EFI_STATUS
jcarsey2247dde2009-11-09 18:08:58 +00002927EFIAPI
2928ShellPrintHiiEx(
2929 IN INT32 Col OPTIONAL,
2930 IN INT32 Row OPTIONAL,
jcarsey1e6e84c2010-01-25 20:05:08 +00002931 IN CONST CHAR8 *Language OPTIONAL,
jcarsey2247dde2009-11-09 18:08:58 +00002932 IN CONST EFI_STRING_ID HiiFormatStringId,
2933 IN CONST EFI_HANDLE HiiFormatHandle,
2934 ...
2935 )
2936{
2937 VA_LIST Marker;
2938 CHAR16 *HiiFormatString;
jcarseya405b862010-09-14 05:18:09 +00002939 EFI_STATUS RetVal;
jcarsey2247dde2009-11-09 18:08:58 +00002940
2941 VA_START (Marker, HiiFormatHandle);
jcarsey1e6e84c2010-01-25 20:05:08 +00002942 HiiFormatString = HiiGetString(HiiFormatHandle, HiiFormatStringId, Language);
jcarsey2247dde2009-11-09 18:08:58 +00002943 ASSERT(HiiFormatString != NULL);
2944
2945 RetVal = InternalShellPrintWorker(Col, Row, HiiFormatString, Marker);
2946
jcarseya405b862010-09-14 05:18:09 +00002947 SHELL_FREE_NON_NULL(HiiFormatString);
jcarseye2f82972009-12-01 05:40:24 +00002948 VA_END(Marker);
jcarsey2247dde2009-11-09 18:08:58 +00002949
2950 return (RetVal);
2951}
2952
2953/**
2954 Function to determine if a given filename represents a file or a directory.
2955
2956 @param[in] DirName Path to directory to test.
2957
jcarseyc8c22592011-10-17 17:49:21 +00002958 @retval EFI_SUCCESS The Path represents a directory
2959 @retval EFI_NOT_FOUND The Path does not represent a directory
2960 @retval EFI_OUT_OF_RESOURCES A memory allocation failed.
2961 @return The path failed to open
jcarsey2247dde2009-11-09 18:08:58 +00002962**/
2963EFI_STATUS
2964EFIAPI
2965ShellIsDirectory(
2966 IN CONST CHAR16 *DirName
2967 )
2968{
2969 EFI_STATUS Status;
jcarseya405b862010-09-14 05:18:09 +00002970 SHELL_FILE_HANDLE Handle;
jcarsey3e082d52010-10-04 16:44:57 +00002971 CHAR16 *TempLocation;
2972 CHAR16 *TempLocation2;
jcarsey2247dde2009-11-09 18:08:58 +00002973
jcarseyecd3d592009-12-07 18:05:00 +00002974 ASSERT(DirName != NULL);
2975
jcarseya405b862010-09-14 05:18:09 +00002976 Handle = NULL;
2977 TempLocation = NULL;
jcarsey2247dde2009-11-09 18:08:58 +00002978
2979 Status = ShellOpenFileByName(DirName, &Handle, EFI_FILE_MODE_READ, 0);
2980 if (EFI_ERROR(Status)) {
jcarseya405b862010-09-14 05:18:09 +00002981 //
2982 // try good logic first.
2983 //
jcarsey366f81a2011-06-27 21:04:22 +00002984 if (gEfiShellProtocol != NULL) {
jcarsey3e082d52010-10-04 16:44:57 +00002985 TempLocation = StrnCatGrow(&TempLocation, NULL, DirName, 0);
jcarseyc8c22592011-10-17 17:49:21 +00002986 if (TempLocation == NULL) {
2987 ShellCloseFile(&Handle);
2988 return (EFI_OUT_OF_RESOURCES);
2989 }
jcarsey3e082d52010-10-04 16:44:57 +00002990 TempLocation2 = StrStr(TempLocation, L":");
2991 if (TempLocation2 != NULL && StrLen(StrStr(TempLocation, L":")) == 2) {
2992 *(TempLocation2+1) = CHAR_NULL;
jcarseya405b862010-09-14 05:18:09 +00002993 }
jcarsey366f81a2011-06-27 21:04:22 +00002994 if (gEfiShellProtocol->GetDevicePathFromMap(TempLocation) != NULL) {
jcarseya405b862010-09-14 05:18:09 +00002995 FreePool(TempLocation);
2996 return (EFI_SUCCESS);
2997 }
2998 FreePool(TempLocation);
2999 } else {
3000 //
3001 // probably a map name?!?!!?
3002 //
3003 TempLocation = StrStr(DirName, L"\\");
3004 if (TempLocation != NULL && *(TempLocation+1) == CHAR_NULL) {
3005 return (EFI_SUCCESS);
3006 }
3007 }
jcarsey2247dde2009-11-09 18:08:58 +00003008 return (Status);
3009 }
3010
3011 if (FileHandleIsDirectory(Handle) == EFI_SUCCESS) {
3012 ShellCloseFile(&Handle);
3013 return (EFI_SUCCESS);
3014 }
3015 ShellCloseFile(&Handle);
3016 return (EFI_NOT_FOUND);
3017}
3018
jcarsey125c2cf2009-11-18 21:36:50 +00003019/**
jcarsey36a9d672009-11-20 21:13:41 +00003020 Function to determine if a given filename represents a file.
3021
3022 @param[in] Name Path to file to test.
3023
3024 @retval EFI_SUCCESS The Path represents a file.
3025 @retval EFI_NOT_FOUND The Path does not represent a file.
3026 @retval other The path failed to open.
3027**/
3028EFI_STATUS
3029EFIAPI
3030ShellIsFile(
3031 IN CONST CHAR16 *Name
3032 )
3033{
3034 EFI_STATUS Status;
jcarseya405b862010-09-14 05:18:09 +00003035 SHELL_FILE_HANDLE Handle;
jcarsey36a9d672009-11-20 21:13:41 +00003036
jcarseyecd3d592009-12-07 18:05:00 +00003037 ASSERT(Name != NULL);
3038
jcarsey36a9d672009-11-20 21:13:41 +00003039 Handle = NULL;
3040
3041 Status = ShellOpenFileByName(Name, &Handle, EFI_FILE_MODE_READ, 0);
3042 if (EFI_ERROR(Status)) {
3043 return (Status);
3044 }
3045
3046 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
3047 ShellCloseFile(&Handle);
3048 return (EFI_SUCCESS);
3049 }
3050 ShellCloseFile(&Handle);
3051 return (EFI_NOT_FOUND);
3052}
3053
3054/**
jcarseyb3011f42010-01-11 21:49:04 +00003055 Function to determine if a given filename represents a file.
3056
3057 This will search the CWD and then the Path.
3058
3059 If Name is NULL, then ASSERT.
3060
3061 @param[in] Name Path to file to test.
3062
3063 @retval EFI_SUCCESS The Path represents a file.
3064 @retval EFI_NOT_FOUND The Path does not represent a file.
3065 @retval other The path failed to open.
3066**/
3067EFI_STATUS
3068EFIAPI
3069ShellIsFileInPath(
3070 IN CONST CHAR16 *Name
jcarseya405b862010-09-14 05:18:09 +00003071 )
3072{
jcarseyb3011f42010-01-11 21:49:04 +00003073 CHAR16 *NewName;
3074 EFI_STATUS Status;
3075
3076 if (!EFI_ERROR(ShellIsFile(Name))) {
jcarseya405b862010-09-14 05:18:09 +00003077 return (EFI_SUCCESS);
jcarseyb3011f42010-01-11 21:49:04 +00003078 }
3079
3080 NewName = ShellFindFilePath(Name);
3081 if (NewName == NULL) {
3082 return (EFI_NOT_FOUND);
3083 }
3084 Status = ShellIsFile(NewName);
3085 FreePool(NewName);
3086 return (Status);
3087}
jcarsey252d9452011-03-25 20:49:53 +00003088
jcarseyb3011f42010-01-11 21:49:04 +00003089/**
Jaben Carsey74b0fb82013-11-22 21:37:34 +00003090 Function return the number converted from a hex representation of a number.
3091
3092 Note: this function cannot be used when (UINTN)(-1), (0xFFFFFFFF) may be a valid
3093 result. Use ShellConvertStringToUint64 instead.
3094
3095 @param[in] String String representation of a number.
3096
3097 @return The unsigned integer result of the conversion.
3098 @retval (UINTN)(-1) An error occured.
3099**/
3100UINTN
3101EFIAPI
3102ShellHexStrToUintn(
3103 IN CONST CHAR16 *String
3104 )
3105{
3106 UINT64 RetVal;
3107
3108 if (!EFI_ERROR(ShellConvertStringToUint64(String, &RetVal, TRUE, TRUE))) {
3109 return ((UINTN)RetVal);
3110 }
3111
3112 return ((UINTN)(-1));
3113}
3114
3115/**
jcarsey1e6e84c2010-01-25 20:05:08 +00003116 Function to determine whether a string is decimal or hex representation of a number
Jaben Carseyd59fc242013-11-14 23:52:43 +00003117 and return the number converted from the string. Spaces are always skipped.
jcarsey125c2cf2009-11-18 21:36:50 +00003118
3119 @param[in] String String representation of a number
3120
jcarsey252d9452011-03-25 20:49:53 +00003121 @return the number
3122 @retval (UINTN)(-1) An error ocurred.
jcarsey125c2cf2009-11-18 21:36:50 +00003123**/
3124UINTN
3125EFIAPI
3126ShellStrToUintn(
3127 IN CONST CHAR16 *String
3128 )
3129{
jcarsey252d9452011-03-25 20:49:53 +00003130 UINT64 RetVal;
3131 BOOLEAN Hex;
3132
3133 Hex = FALSE;
3134
3135 if (!InternalShellIsHexOrDecimalNumber(String, Hex, TRUE)) {
3136 Hex = TRUE;
jcarsey125c2cf2009-11-18 21:36:50 +00003137 }
jcarsey252d9452011-03-25 20:49:53 +00003138
3139 if (!EFI_ERROR(ShellConvertStringToUint64(String, &RetVal, Hex, TRUE))) {
3140 return ((UINTN)RetVal);
3141 }
3142 return ((UINTN)(-1));
jcarsey125c2cf2009-11-18 21:36:50 +00003143}
3144
3145/**
3146 Safely append with automatic string resizing given length of Destination and
3147 desired length of copy from Source.
3148
3149 append the first D characters of Source to the end of Destination, where D is
3150 the lesser of Count and the StrLen() of Source. If appending those D characters
3151 will fit within Destination (whose Size is given as CurrentSize) and
jcarsey1e6e84c2010-01-25 20:05:08 +00003152 still leave room for a NULL terminator, then those characters are appended,
3153 starting at the original terminating NULL of Destination, and a new terminating
3154 NULL is appended.
jcarsey125c2cf2009-11-18 21:36:50 +00003155
3156 If appending D characters onto Destination will result in a overflow of the size
3157 given in CurrentSize the string will be grown such that the copy can be performed
3158 and CurrentSize will be updated to the new size.
3159
3160 If Source is NULL, there is nothing to append, just return the current buffer in
3161 Destination.
3162
3163 if Destination is NULL, then ASSERT()
3164 if Destination's current length (including NULL terminator) is already more then
3165 CurrentSize, then ASSERT()
3166
ydong104ff7e372011-09-02 08:05:34 +00003167 @param[in, out] Destination The String to append onto
3168 @param[in, out] CurrentSize on call the number of bytes in Destination. On
jcarsey125c2cf2009-11-18 21:36:50 +00003169 return possibly the new size (still in bytes). if NULL
3170 then allocate whatever is needed.
3171 @param[in] Source The String to append from
3172 @param[in] Count Maximum number of characters to append. if 0 then
3173 all are appended.
3174
3175 @return Destination return the resultant string.
3176**/
3177CHAR16*
3178EFIAPI
3179StrnCatGrow (
3180 IN OUT CHAR16 **Destination,
3181 IN OUT UINTN *CurrentSize,
3182 IN CONST CHAR16 *Source,
3183 IN UINTN Count
3184 )
3185{
3186 UINTN DestinationStartSize;
3187 UINTN NewSize;
3188
3189 //
3190 // ASSERTs
3191 //
3192 ASSERT(Destination != NULL);
3193
3194 //
3195 // If there's nothing to do then just return Destination
3196 //
3197 if (Source == NULL) {
3198 return (*Destination);
3199 }
3200
3201 //
3202 // allow for un-initialized pointers, based on size being 0
3203 //
3204 if (CurrentSize != NULL && *CurrentSize == 0) {
3205 *Destination = NULL;
3206 }
3207
3208 //
3209 // allow for NULL pointers address as Destination
3210 //
3211 if (*Destination != NULL) {
3212 ASSERT(CurrentSize != 0);
3213 DestinationStartSize = StrSize(*Destination);
3214 ASSERT(DestinationStartSize <= *CurrentSize);
3215 } else {
3216 DestinationStartSize = 0;
3217// ASSERT(*CurrentSize == 0);
3218 }
3219
3220 //
3221 // Append all of Source?
3222 //
3223 if (Count == 0) {
3224 Count = StrLen(Source);
3225 }
3226
3227 //
3228 // Test and grow if required
3229 //
3230 if (CurrentSize != NULL) {
3231 NewSize = *CurrentSize;
ydong10f480fdc2012-09-07 01:55:33 +00003232 if (NewSize < DestinationStartSize + (Count * sizeof(CHAR16))) {
3233 while (NewSize < (DestinationStartSize + (Count*sizeof(CHAR16)))) {
3234 NewSize += 2 * Count * sizeof(CHAR16);
3235 }
3236 *Destination = ReallocatePool(*CurrentSize, NewSize, *Destination);
3237 *CurrentSize = NewSize;
jcarsey125c2cf2009-11-18 21:36:50 +00003238 }
jcarsey125c2cf2009-11-18 21:36:50 +00003239 } else {
3240 *Destination = AllocateZeroPool((Count+1)*sizeof(CHAR16));
3241 }
3242
3243 //
3244 // Now use standard StrnCat on a big enough buffer
3245 //
jcarseyc9d92df2010-02-03 15:37:54 +00003246 if (*Destination == NULL) {
3247 return (NULL);
3248 }
jcarsey125c2cf2009-11-18 21:36:50 +00003249 return StrnCat(*Destination, Source, Count);
3250}
jcarseyc9d92df2010-02-03 15:37:54 +00003251
3252/**
3253 Prompt the user and return the resultant answer to the requestor.
3254
3255 This function will display the requested question on the shell prompt and then
3256 wait for an apropriate answer to be input from the console.
3257
jcarseya405b862010-09-14 05:18:09 +00003258 if the SHELL_PROMPT_REQUEST_TYPE is SHELL_PROMPT_REQUEST_TYPE_YESNO, ShellPromptResponseTypeQuitContinue
jcarseyc9d92df2010-02-03 15:37:54 +00003259 or SHELL_PROMPT_REQUEST_TYPE_YESNOCANCEL then *Response is of type SHELL_PROMPT_RESPONSE.
3260
jcarseya405b862010-09-14 05:18:09 +00003261 if the SHELL_PROMPT_REQUEST_TYPE is ShellPromptResponseTypeFreeform then *Response is of type
jcarseyc9d92df2010-02-03 15:37:54 +00003262 CHAR16*.
3263
3264 In either case *Response must be callee freed if Response was not NULL;
3265
3266 @param Type What type of question is asked. This is used to filter the input
3267 to prevent invalid answers to question.
3268 @param Prompt Pointer to string prompt to use to request input.
3269 @param Response Pointer to Response which will be populated upon return.
3270
3271 @retval EFI_SUCCESS The operation was sucessful.
3272 @retval EFI_UNSUPPORTED The operation is not supported as requested.
3273 @retval EFI_INVALID_PARAMETER A parameter was invalid.
3274 @return other The operation failed.
3275**/
3276EFI_STATUS
3277EFIAPI
3278ShellPromptForResponse (
3279 IN SHELL_PROMPT_REQUEST_TYPE Type,
3280 IN CHAR16 *Prompt OPTIONAL,
3281 IN OUT VOID **Response OPTIONAL
3282 )
3283{
3284 EFI_STATUS Status;
3285 EFI_INPUT_KEY Key;
3286 UINTN EventIndex;
3287 SHELL_PROMPT_RESPONSE *Resp;
jcarseya405b862010-09-14 05:18:09 +00003288 UINTN Size;
3289 CHAR16 *Buffer;
jcarseyc9d92df2010-02-03 15:37:54 +00003290
jcarseya405b862010-09-14 05:18:09 +00003291 Status = EFI_UNSUPPORTED;
3292 Resp = NULL;
3293 Buffer = NULL;
3294 Size = 0;
3295 if (Type != ShellPromptResponseTypeFreeform) {
jcarsey252d9452011-03-25 20:49:53 +00003296 Resp = (SHELL_PROMPT_RESPONSE*)AllocateZeroPool(sizeof(SHELL_PROMPT_RESPONSE));
jcarsey3e082d52010-10-04 16:44:57 +00003297 if (Resp == NULL) {
3298 return (EFI_OUT_OF_RESOURCES);
3299 }
xdu290bfa222010-07-19 05:21:27 +00003300 }
jcarseyc9d92df2010-02-03 15:37:54 +00003301
3302 switch(Type) {
jcarseya405b862010-09-14 05:18:09 +00003303 case ShellPromptResponseTypeQuitContinue:
jcarseyc9d92df2010-02-03 15:37:54 +00003304 if (Prompt != NULL) {
3305 ShellPrintEx(-1, -1, L"%s", Prompt);
3306 }
3307 //
3308 // wait for valid response
3309 //
3310 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3311 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
Jaben Carsey31b018a2014-01-16 16:52:39 +00003312 if (EFI_ERROR(Status)) {
3313 break;
3314 }
jcarseyc9d92df2010-02-03 15:37:54 +00003315 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3316 if (Key.UnicodeChar == L'Q' || Key.UnicodeChar ==L'q') {
jcarseya405b862010-09-14 05:18:09 +00003317 *Resp = ShellPromptResponseQuit;
jcarseyc9d92df2010-02-03 15:37:54 +00003318 } else {
jcarseya405b862010-09-14 05:18:09 +00003319 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003320 }
3321 break;
jcarseya405b862010-09-14 05:18:09 +00003322 case ShellPromptResponseTypeYesNoCancel:
jcarseyc9d92df2010-02-03 15:37:54 +00003323 if (Prompt != NULL) {
3324 ShellPrintEx(-1, -1, L"%s", Prompt);
3325 }
3326 //
3327 // wait for valid response
3328 //
jcarseya405b862010-09-14 05:18:09 +00003329 *Resp = ShellPromptResponseMax;
3330 while (*Resp == ShellPromptResponseMax) {
Jaben Carsey194ae482013-12-09 22:55:13 +00003331 if (ShellGetExecutionBreakFlag()) {
3332 Status = EFI_ABORTED;
3333 break;
3334 }
jcarseyc9d92df2010-02-03 15:37:54 +00003335 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3336 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
Jaben Carsey31b018a2014-01-16 16:52:39 +00003337 if (EFI_ERROR(Status)) {
3338 break;
3339 }
jcarseyc9d92df2010-02-03 15:37:54 +00003340 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3341 switch (Key.UnicodeChar) {
3342 case L'Y':
3343 case L'y':
jcarseya405b862010-09-14 05:18:09 +00003344 *Resp = ShellPromptResponseYes;
jcarseyc9d92df2010-02-03 15:37:54 +00003345 break;
3346 case L'N':
3347 case L'n':
jcarseya405b862010-09-14 05:18:09 +00003348 *Resp = ShellPromptResponseNo;
jcarseyc9d92df2010-02-03 15:37:54 +00003349 break;
3350 case L'C':
3351 case L'c':
jcarseya405b862010-09-14 05:18:09 +00003352 *Resp = ShellPromptResponseCancel;
3353 break;
3354 }
3355 }
3356 break; case ShellPromptResponseTypeYesNoAllCancel:
3357 if (Prompt != NULL) {
3358 ShellPrintEx(-1, -1, L"%s", Prompt);
3359 }
3360 //
3361 // wait for valid response
3362 //
3363 *Resp = ShellPromptResponseMax;
3364 while (*Resp == ShellPromptResponseMax) {
Jaben Carsey194ae482013-12-09 22:55:13 +00003365 if (ShellGetExecutionBreakFlag()) {
3366 Status = EFI_ABORTED;
3367 break;
3368 }
jcarseya405b862010-09-14 05:18:09 +00003369 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3370 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
Jaben Carsey31b018a2014-01-16 16:52:39 +00003371 if (EFI_ERROR(Status)) {
3372 break;
3373 }
jcarseya405b862010-09-14 05:18:09 +00003374 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3375 switch (Key.UnicodeChar) {
3376 case L'Y':
3377 case L'y':
3378 *Resp = ShellPromptResponseYes;
3379 break;
3380 case L'N':
3381 case L'n':
3382 *Resp = ShellPromptResponseNo;
3383 break;
3384 case L'A':
3385 case L'a':
3386 *Resp = ShellPromptResponseAll;
3387 break;
3388 case L'C':
3389 case L'c':
3390 *Resp = ShellPromptResponseCancel;
jcarseyc9d92df2010-02-03 15:37:54 +00003391 break;
3392 }
3393 }
3394 break;
jcarseya405b862010-09-14 05:18:09 +00003395 case ShellPromptResponseTypeEnterContinue:
3396 case ShellPromptResponseTypeAnyKeyContinue:
jcarseyc9d92df2010-02-03 15:37:54 +00003397 if (Prompt != NULL) {
3398 ShellPrintEx(-1, -1, L"%s", Prompt);
3399 }
3400 //
3401 // wait for valid response
3402 //
jcarseya405b862010-09-14 05:18:09 +00003403 *Resp = ShellPromptResponseMax;
3404 while (*Resp == ShellPromptResponseMax) {
Jaben Carsey194ae482013-12-09 22:55:13 +00003405 if (ShellGetExecutionBreakFlag()) {
3406 Status = EFI_ABORTED;
3407 break;
3408 }
jcarseyc9d92df2010-02-03 15:37:54 +00003409 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
jcarseya405b862010-09-14 05:18:09 +00003410 if (Type == ShellPromptResponseTypeEnterContinue) {
jcarseyc9d92df2010-02-03 15:37:54 +00003411 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
Jaben Carsey31b018a2014-01-16 16:52:39 +00003412 if (EFI_ERROR(Status)) {
3413 break;
3414 }
jcarseyc9d92df2010-02-03 15:37:54 +00003415 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3416 if (Key.UnicodeChar == CHAR_CARRIAGE_RETURN) {
jcarseya405b862010-09-14 05:18:09 +00003417 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003418 break;
3419 }
3420 }
jcarseya405b862010-09-14 05:18:09 +00003421 if (Type == ShellPromptResponseTypeAnyKeyContinue) {
3422 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3423 ASSERT_EFI_ERROR(Status);
3424 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003425 break;
3426 }
3427 }
3428 break;
jcarseya405b862010-09-14 05:18:09 +00003429 case ShellPromptResponseTypeYesNo:
3430 if (Prompt != NULL) {
3431 ShellPrintEx(-1, -1, L"%s", Prompt);
3432 }
3433 //
3434 // wait for valid response
3435 //
3436 *Resp = ShellPromptResponseMax;
3437 while (*Resp == ShellPromptResponseMax) {
Jaben Carsey194ae482013-12-09 22:55:13 +00003438 if (ShellGetExecutionBreakFlag()) {
3439 Status = EFI_ABORTED;
3440 break;
3441 }
jcarseya405b862010-09-14 05:18:09 +00003442 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3443 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
Jaben Carsey31b018a2014-01-16 16:52:39 +00003444 if (EFI_ERROR(Status)) {
3445 break;
3446 }
jcarseya405b862010-09-14 05:18:09 +00003447 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3448 switch (Key.UnicodeChar) {
3449 case L'Y':
3450 case L'y':
3451 *Resp = ShellPromptResponseYes;
3452 break;
3453 case L'N':
3454 case L'n':
3455 *Resp = ShellPromptResponseNo;
3456 break;
3457 }
3458 }
3459 break;
3460 case ShellPromptResponseTypeFreeform:
3461 if (Prompt != NULL) {
3462 ShellPrintEx(-1, -1, L"%s", Prompt);
3463 }
3464 while(1) {
Jaben Carsey194ae482013-12-09 22:55:13 +00003465 if (ShellGetExecutionBreakFlag()) {
3466 Status = EFI_ABORTED;
3467 break;
3468 }
jcarseya405b862010-09-14 05:18:09 +00003469 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3470 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
Jaben Carsey31b018a2014-01-16 16:52:39 +00003471 if (EFI_ERROR(Status)) {
3472 break;
3473 }
jcarseya405b862010-09-14 05:18:09 +00003474 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3475 if (Key.UnicodeChar == CHAR_CARRIAGE_RETURN) {
3476 break;
3477 }
3478 ASSERT((Buffer == NULL && Size == 0) || (Buffer != NULL));
3479 StrnCatGrow(&Buffer, &Size, &Key.UnicodeChar, 1);
3480 }
3481 break;
3482 //
3483 // This is the location to add new prompt types.
Jaben Carsey194ae482013-12-09 22:55:13 +00003484 // If your new type loops remember to add ExecutionBreak support.
jcarseya405b862010-09-14 05:18:09 +00003485 //
jcarseyc9d92df2010-02-03 15:37:54 +00003486 default:
jcarseya405b862010-09-14 05:18:09 +00003487 ASSERT(FALSE);
jcarseyc9d92df2010-02-03 15:37:54 +00003488 }
3489
3490 if (Response != NULL) {
jcarseya405b862010-09-14 05:18:09 +00003491 if (Resp != NULL) {
3492 *Response = Resp;
3493 } else if (Buffer != NULL) {
3494 *Response = Buffer;
3495 }
jcarseyc9d92df2010-02-03 15:37:54 +00003496 } else {
jcarseya405b862010-09-14 05:18:09 +00003497 if (Resp != NULL) {
3498 FreePool(Resp);
3499 }
3500 if (Buffer != NULL) {
3501 FreePool(Buffer);
3502 }
jcarseyc9d92df2010-02-03 15:37:54 +00003503 }
3504
jcarseya405b862010-09-14 05:18:09 +00003505 ShellPrintEx(-1, -1, L"\r\n");
jcarseyc9d92df2010-02-03 15:37:54 +00003506 return (Status);
3507}
3508
3509/**
3510 Prompt the user and return the resultant answer to the requestor.
3511
3512 This function is the same as ShellPromptForResponse, except that the prompt is
3513 automatically pulled from HII.
3514
3515 @param Type What type of question is asked. This is used to filter the input
3516 to prevent invalid answers to question.
jcarseya405b862010-09-14 05:18:09 +00003517 @param[in] HiiFormatStringId The format string Id for getting from Hii.
3518 @param[in] HiiFormatHandle The format string Handle for getting from Hii.
3519 @param Response Pointer to Response which will be populated upon return.
jcarseyc9d92df2010-02-03 15:37:54 +00003520
3521 @retval EFI_SUCCESS the operation was sucessful.
3522 @return other the operation failed.
3523
3524 @sa ShellPromptForResponse
3525**/
3526EFI_STATUS
3527EFIAPI
3528ShellPromptForResponseHii (
3529 IN SHELL_PROMPT_REQUEST_TYPE Type,
3530 IN CONST EFI_STRING_ID HiiFormatStringId,
3531 IN CONST EFI_HANDLE HiiFormatHandle,
3532 IN OUT VOID **Response
3533 )
3534{
3535 CHAR16 *Prompt;
3536 EFI_STATUS Status;
3537
3538 Prompt = HiiGetString(HiiFormatHandle, HiiFormatStringId, NULL);
3539 Status = ShellPromptForResponse(Type, Prompt, Response);
3540 FreePool(Prompt);
3541 return (Status);
3542}
3543
jcarseya405b862010-09-14 05:18:09 +00003544/**
3545 Function to determin if an entire string is a valid number.
jcarseyc9d92df2010-02-03 15:37:54 +00003546
jcarseya405b862010-09-14 05:18:09 +00003547 If Hex it must be preceeded with a 0x or has ForceHex, set TRUE.
3548
3549 @param[in] String The string to evaluate.
3550 @param[in] ForceHex TRUE - always assume hex.
3551 @param[in] StopAtSpace TRUE to halt upon finding a space, FALSE to keep going.
3552
3553 @retval TRUE It is all numeric (dec/hex) characters.
3554 @retval FALSE There is a non-numeric character.
3555**/
3556BOOLEAN
3557EFIAPI
jcarsey252d9452011-03-25 20:49:53 +00003558InternalShellIsHexOrDecimalNumber (
jcarseya405b862010-09-14 05:18:09 +00003559 IN CONST CHAR16 *String,
3560 IN CONST BOOLEAN ForceHex,
3561 IN CONST BOOLEAN StopAtSpace
3562 )
3563{
3564 BOOLEAN Hex;
3565
3566 //
3567 // chop off a single negative sign
3568 //
3569 if (String != NULL && *String == L'-') {
3570 String++;
3571 }
darylm503b0934ac2011-11-12 00:35:11 +00003572
jcarseya405b862010-09-14 05:18:09 +00003573 if (String == NULL) {
3574 return (FALSE);
3575 }
3576
3577 //
3578 // chop leading zeroes
3579 //
3580 while(String != NULL && *String == L'0'){
3581 String++;
3582 }
3583 //
3584 // allow '0x' or '0X', but not 'x' or 'X'
3585 //
3586 if (String != NULL && (*String == L'x' || *String == L'X')) {
3587 if (*(String-1) != L'0') {
3588 //
3589 // we got an x without a preceeding 0
3590 //
3591 return (FALSE);
3592 }
3593 String++;
3594 Hex = TRUE;
3595 } else if (ForceHex) {
3596 Hex = TRUE;
3597 } else {
3598 Hex = FALSE;
3599 }
3600
3601 //
3602 // loop through the remaining characters and use the lib function
3603 //
3604 for ( ; String != NULL && *String != CHAR_NULL && !(StopAtSpace && *String == L' ') ; String++){
3605 if (Hex) {
3606 if (!ShellIsHexaDecimalDigitCharacter(*String)) {
3607 return (FALSE);
3608 }
3609 } else {
3610 if (!ShellIsDecimalDigitCharacter(*String)) {
3611 return (FALSE);
3612 }
3613 }
3614 }
jcarsey252d9452011-03-25 20:49:53 +00003615
jcarseya405b862010-09-14 05:18:09 +00003616 return (TRUE);
3617}
3618
3619/**
3620 Function to determine if a given filename exists.
3621
3622 @param[in] Name Path to test.
3623
3624 @retval EFI_SUCCESS The Path represents a file.
3625 @retval EFI_NOT_FOUND The Path does not represent a file.
3626 @retval other The path failed to open.
3627**/
3628EFI_STATUS
3629EFIAPI
3630ShellFileExists(
3631 IN CONST CHAR16 *Name
3632 )
3633{
3634 EFI_STATUS Status;
3635 EFI_SHELL_FILE_INFO *List;
3636
3637 ASSERT(Name != NULL);
3638
3639 List = NULL;
3640 Status = ShellOpenFileMetaArg((CHAR16*)Name, EFI_FILE_MODE_READ, &List);
3641 if (EFI_ERROR(Status)) {
3642 return (Status);
3643 }
3644
3645 ShellCloseFileMetaArg(&List);
3646
3647 return (EFI_SUCCESS);
3648}
jcarsey252d9452011-03-25 20:49:53 +00003649
3650/**
darylm503b0934ac2011-11-12 00:35:11 +00003651 Convert a Unicode character to upper case only if
jcarsey252d9452011-03-25 20:49:53 +00003652 it maps to a valid small-case ASCII character.
3653
3654 This internal function only deal with Unicode character
3655 which maps to a valid small-case ASCII character, i.e.
3656 L'a' to L'z'. For other Unicode character, the input character
3657 is returned directly.
3658
3659 @param Char The character to convert.
3660
3661 @retval LowerCharacter If the Char is with range L'a' to L'z'.
3662 @retval Unchanged Otherwise.
3663
3664**/
3665CHAR16
3666EFIAPI
3667InternalShellCharToUpper (
3668 IN CHAR16 Char
3669 )
3670{
3671 if (Char >= L'a' && Char <= L'z') {
3672 return (CHAR16) (Char - (L'a' - L'A'));
3673 }
3674
3675 return Char;
3676}
3677
3678/**
3679 Convert a Unicode character to numerical value.
3680
3681 This internal function only deal with Unicode character
3682 which maps to a valid hexadecimal ASII character, i.e.
darylm503b0934ac2011-11-12 00:35:11 +00003683 L'0' to L'9', L'a' to L'f' or L'A' to L'F'. For other
jcarsey252d9452011-03-25 20:49:53 +00003684 Unicode character, the value returned does not make sense.
3685
3686 @param Char The character to convert.
3687
3688 @return The numerical value converted.
3689
3690**/
3691UINTN
3692EFIAPI
3693InternalShellHexCharToUintn (
3694 IN CHAR16 Char
3695 )
3696{
3697 if (ShellIsDecimalDigitCharacter (Char)) {
3698 return Char - L'0';
3699 }
3700
3701 return (UINTN) (10 + InternalShellCharToUpper (Char) - L'A');
3702}
3703
3704/**
3705 Convert a Null-terminated Unicode hexadecimal string to a value of type UINT64.
3706
3707 This function returns a value of type UINTN by interpreting the contents
3708 of the Unicode string specified by String as a hexadecimal number.
3709 The format of the input Unicode string String is:
3710
3711 [spaces][zeros][x][hexadecimal digits].
3712
3713 The valid hexadecimal digit character is in the range [0-9], [a-f] and [A-F].
3714 The prefix "0x" is optional. Both "x" and "X" is allowed in "0x" prefix.
3715 If "x" appears in the input string, it must be prefixed with at least one 0.
3716 The function will ignore the pad space, which includes spaces or tab characters,
3717 before [zeros], [x] or [hexadecimal digit]. The running zero before [x] or
3718 [hexadecimal digit] will be ignored. Then, the decoding starts after [x] or the
3719 first valid hexadecimal digit. Then, the function stops at the first character that is
3720 a not a valid hexadecimal character or NULL, whichever one comes first.
3721
3722 If String has only pad spaces, then zero is returned.
3723 If String has no leading pad spaces, leading zeros or valid hexadecimal digits,
3724 then zero is returned.
3725
3726 @param[in] String A pointer to a Null-terminated Unicode string.
3727 @param[out] Value Upon a successful return the value of the conversion.
3728 @param[in] StopAtSpace FALSE to skip spaces.
3729
3730 @retval EFI_SUCCESS The conversion was successful.
3731 @retval EFI_INVALID_PARAMETER A parameter was NULL or invalid.
3732 @retval EFI_DEVICE_ERROR An overflow occured.
3733**/
3734EFI_STATUS
3735EFIAPI
3736InternalShellStrHexToUint64 (
3737 IN CONST CHAR16 *String,
3738 OUT UINT64 *Value,
3739 IN CONST BOOLEAN StopAtSpace
3740 )
3741{
3742 UINT64 Result;
3743
3744 if (String == NULL || StrSize(String) == 0 || Value == NULL) {
3745 return (EFI_INVALID_PARAMETER);
3746 }
darylm503b0934ac2011-11-12 00:35:11 +00003747
jcarsey252d9452011-03-25 20:49:53 +00003748 //
darylm503b0934ac2011-11-12 00:35:11 +00003749 // Ignore the pad spaces (space or tab)
jcarsey252d9452011-03-25 20:49:53 +00003750 //
3751 while ((*String == L' ') || (*String == L'\t')) {
3752 String++;
3753 }
3754
3755 //
3756 // Ignore leading Zeros after the spaces
3757 //
3758 while (*String == L'0') {
3759 String++;
3760 }
3761
3762 if (InternalShellCharToUpper (*String) == L'X') {
3763 if (*(String - 1) != L'0') {
3764 return 0;
3765 }
3766 //
3767 // Skip the 'X'
3768 //
3769 String++;
3770 }
3771
3772 Result = 0;
darylm503b0934ac2011-11-12 00:35:11 +00003773
jcarsey252d9452011-03-25 20:49:53 +00003774 //
ydong1012ea4692012-07-26 05:42:43 +00003775 // Skip spaces if requested
jcarsey252d9452011-03-25 20:49:53 +00003776 //
ydong1012ea4692012-07-26 05:42:43 +00003777 while (StopAtSpace && *String == L' ') {
3778 String++;
jcarsey252d9452011-03-25 20:49:53 +00003779 }
darylm503b0934ac2011-11-12 00:35:11 +00003780
jcarsey252d9452011-03-25 20:49:53 +00003781 while (ShellIsHexaDecimalDigitCharacter (*String)) {
3782 //
darylm503b0934ac2011-11-12 00:35:11 +00003783 // If the Hex Number represented by String overflows according
jcarsey252d9452011-03-25 20:49:53 +00003784 // to the range defined by UINTN, then ASSERT().
3785 //
3786 if (!(Result <= (RShiftU64((((UINT64) ~0) - InternalShellHexCharToUintn (*String)), 4)))) {
3787// if (!(Result <= ((((UINT64) ~0) - InternalShellHexCharToUintn (*String)) >> 4))) {
3788 return (EFI_DEVICE_ERROR);
3789 }
3790
3791 Result = (LShiftU64(Result, 4));
3792 Result += InternalShellHexCharToUintn (*String);
3793 String++;
3794
3795 //
jcarsey49bd4982012-01-27 18:42:43 +00003796 // stop at spaces if requested
jcarsey252d9452011-03-25 20:49:53 +00003797 //
jcarsey49bd4982012-01-27 18:42:43 +00003798 if (StopAtSpace && *String == L' ') {
3799 break;
jcarsey252d9452011-03-25 20:49:53 +00003800 }
3801 }
3802
3803 *Value = Result;
3804 return (EFI_SUCCESS);
3805}
3806
3807/**
3808 Convert a Null-terminated Unicode decimal string to a value of
3809 type UINT64.
3810
3811 This function returns a value of type UINT64 by interpreting the contents
3812 of the Unicode string specified by String as a decimal number. The format
3813 of the input Unicode string String is:
3814
3815 [spaces] [decimal digits].
3816
3817 The valid decimal digit character is in the range [0-9]. The
3818 function will ignore the pad space, which includes spaces or
3819 tab characters, before [decimal digits]. The running zero in the
3820 beginning of [decimal digits] will be ignored. Then, the function
3821 stops at the first character that is a not a valid decimal character
3822 or a Null-terminator, whichever one comes first.
3823
3824 If String has only pad spaces, then 0 is returned.
3825 If String has no pad spaces or valid decimal digits,
3826 then 0 is returned.
3827
3828 @param[in] String A pointer to a Null-terminated Unicode string.
3829 @param[out] Value Upon a successful return the value of the conversion.
3830 @param[in] StopAtSpace FALSE to skip spaces.
3831
3832 @retval EFI_SUCCESS The conversion was successful.
3833 @retval EFI_INVALID_PARAMETER A parameter was NULL or invalid.
3834 @retval EFI_DEVICE_ERROR An overflow occured.
3835**/
3836EFI_STATUS
3837EFIAPI
3838InternalShellStrDecimalToUint64 (
3839 IN CONST CHAR16 *String,
3840 OUT UINT64 *Value,
3841 IN CONST BOOLEAN StopAtSpace
3842 )
3843{
3844 UINT64 Result;
3845
3846 if (String == NULL || StrSize (String) == 0 || Value == NULL) {
3847 return (EFI_INVALID_PARAMETER);
3848 }
3849
3850 //
3851 // Ignore the pad spaces (space or tab)
3852 //
3853 while ((*String == L' ') || (*String == L'\t')) {
3854 String++;
3855 }
3856
3857 //
3858 // Ignore leading Zeros after the spaces
3859 //
3860 while (*String == L'0') {
3861 String++;
3862 }
3863
3864 Result = 0;
3865
3866 //
3867 // Skip spaces if requested
3868 //
3869 while (StopAtSpace && *String == L' ') {
3870 String++;
3871 }
3872 while (ShellIsDecimalDigitCharacter (*String)) {
3873 //
darylm503b0934ac2011-11-12 00:35:11 +00003874 // If the number represented by String overflows according
jcarsey252d9452011-03-25 20:49:53 +00003875 // to the range defined by UINT64, then ASSERT().
3876 //
darylm503b0934ac2011-11-12 00:35:11 +00003877
jcarsey252d9452011-03-25 20:49:53 +00003878 if (!(Result <= (DivU64x32((((UINT64) ~0) - (*String - L'0')),10)))) {
3879 return (EFI_DEVICE_ERROR);
3880 }
3881
3882 Result = MultU64x32(Result, 10) + (*String - L'0');
3883 String++;
3884
3885 //
3886 // Stop at spaces if requested
3887 //
3888 if (StopAtSpace && *String == L' ') {
3889 break;
3890 }
3891 }
3892
3893 *Value = Result;
darylm503b0934ac2011-11-12 00:35:11 +00003894
jcarsey252d9452011-03-25 20:49:53 +00003895 return (EFI_SUCCESS);
3896}
3897
3898/**
3899 Function to verify and convert a string to its numerical value.
3900
3901 If Hex it must be preceeded with a 0x, 0X, or has ForceHex set TRUE.
3902
3903 @param[in] String The string to evaluate.
3904 @param[out] Value Upon a successful return the value of the conversion.
3905 @param[in] ForceHex TRUE - always assume hex.
3906 @param[in] StopAtSpace FALSE to skip spaces.
darylm503b0934ac2011-11-12 00:35:11 +00003907
jcarsey252d9452011-03-25 20:49:53 +00003908 @retval EFI_SUCCESS The conversion was successful.
3909 @retval EFI_INVALID_PARAMETER String contained an invalid character.
3910 @retval EFI_NOT_FOUND String was a number, but Value was NULL.
3911**/
3912EFI_STATUS
3913EFIAPI
3914ShellConvertStringToUint64(
3915 IN CONST CHAR16 *String,
3916 OUT UINT64 *Value,
3917 IN CONST BOOLEAN ForceHex,
3918 IN CONST BOOLEAN StopAtSpace
3919 )
3920{
3921 UINT64 RetVal;
3922 CONST CHAR16 *Walker;
3923 EFI_STATUS Status;
3924 BOOLEAN Hex;
3925
3926 Hex = ForceHex;
3927
3928 if (!InternalShellIsHexOrDecimalNumber(String, Hex, StopAtSpace)) {
3929 if (!Hex) {
3930 Hex = TRUE;
3931 if (!InternalShellIsHexOrDecimalNumber(String, Hex, StopAtSpace)) {
3932 return (EFI_INVALID_PARAMETER);
3933 }
3934 } else {
3935 return (EFI_INVALID_PARAMETER);
3936 }
3937 }
3938
3939 //
3940 // Chop off leading spaces
3941 //
3942 for (Walker = String; Walker != NULL && *Walker != CHAR_NULL && *Walker == L' '; Walker++);
3943
3944 //
3945 // make sure we have something left that is numeric.
3946 //
3947 if (Walker == NULL || *Walker == CHAR_NULL || !InternalShellIsHexOrDecimalNumber(Walker, Hex, StopAtSpace)) {
3948 return (EFI_INVALID_PARAMETER);
darylm503b0934ac2011-11-12 00:35:11 +00003949 }
jcarsey252d9452011-03-25 20:49:53 +00003950
3951 //
3952 // do the conversion.
3953 //
3954 if (Hex || StrnCmp(Walker, L"0x", 2) == 0 || StrnCmp(Walker, L"0X", 2) == 0){
3955 Status = InternalShellStrHexToUint64(Walker, &RetVal, StopAtSpace);
3956 } else {
3957 Status = InternalShellStrDecimalToUint64(Walker, &RetVal, StopAtSpace);
3958 }
3959
3960 if (Value == NULL && !EFI_ERROR(Status)) {
3961 return (EFI_NOT_FOUND);
3962 }
3963
3964 if (Value != NULL) {
3965 *Value = RetVal;
3966 }
3967
3968 return (Status);
3969}
3970
3971/**
3972 Function to determin if an entire string is a valid number.
3973
3974 If Hex it must be preceeded with a 0x or has ForceHex, set TRUE.
3975
3976 @param[in] String The string to evaluate.
3977 @param[in] ForceHex TRUE - always assume hex.
3978 @param[in] StopAtSpace TRUE to halt upon finding a space, FALSE to keep going.
3979
3980 @retval TRUE It is all numeric (dec/hex) characters.
3981 @retval FALSE There is a non-numeric character.
3982**/
3983BOOLEAN
3984EFIAPI
3985ShellIsHexOrDecimalNumber (
3986 IN CONST CHAR16 *String,
3987 IN CONST BOOLEAN ForceHex,
3988 IN CONST BOOLEAN StopAtSpace
3989 )
3990{
3991 if (ShellConvertStringToUint64(String, NULL, ForceHex, StopAtSpace) == EFI_NOT_FOUND) {
3992 return (TRUE);
3993 }
3994 return (FALSE);
3995}
jcarsey4d0a4fc2011-07-06 22:28:36 +00003996
3997/**
3998 Function to read a single line from a SHELL_FILE_HANDLE. The \n is not included in the returned
3999 buffer. The returned buffer must be callee freed.
4000
4001 If the position upon start is 0, then the Ascii Boolean will be set. This should be
4002 maintained and not changed for all operations with the same file.
4003
ydong104ff7e372011-09-02 08:05:34 +00004004 @param[in] Handle SHELL_FILE_HANDLE to read from.
4005 @param[in, out] Ascii Boolean value for indicating whether the file is
4006 Ascii (TRUE) or UCS2 (FALSE).
jcarsey4d0a4fc2011-07-06 22:28:36 +00004007
jcarseybeab0fc2011-10-10 17:26:25 +00004008 @return The line of text from the file.
4009 @retval NULL There was not enough memory available.
jcarsey4d0a4fc2011-07-06 22:28:36 +00004010
4011 @sa ShellFileHandleReadLine
4012**/
4013CHAR16*
4014EFIAPI
4015ShellFileHandleReturnLine(
4016 IN SHELL_FILE_HANDLE Handle,
4017 IN OUT BOOLEAN *Ascii
4018 )
4019{
4020 CHAR16 *RetVal;
4021 UINTN Size;
4022 EFI_STATUS Status;
4023
4024 Size = 0;
4025 RetVal = NULL;
4026
4027 Status = ShellFileHandleReadLine(Handle, RetVal, &Size, FALSE, Ascii);
4028 if (Status == EFI_BUFFER_TOO_SMALL) {
4029 RetVal = AllocateZeroPool(Size);
jcarseybeab0fc2011-10-10 17:26:25 +00004030 if (RetVal == NULL) {
4031 return (NULL);
4032 }
jcarsey4d0a4fc2011-07-06 22:28:36 +00004033 Status = ShellFileHandleReadLine(Handle, RetVal, &Size, FALSE, Ascii);
darylm503b0934ac2011-11-12 00:35:11 +00004034
jcarsey4d0a4fc2011-07-06 22:28:36 +00004035 }
jcarsey4d0a4fc2011-07-06 22:28:36 +00004036 if (EFI_ERROR(Status) && (RetVal != NULL)) {
4037 FreePool(RetVal);
4038 RetVal = NULL;
4039 }
4040 return (RetVal);
4041}
4042
4043/**
4044 Function to read a single line (up to but not including the \n) from a SHELL_FILE_HANDLE.
4045
4046 If the position upon start is 0, then the Ascii Boolean will be set. This should be
4047 maintained and not changed for all operations with the same file.
4048
ydong104ff7e372011-09-02 08:05:34 +00004049 @param[in] Handle SHELL_FILE_HANDLE to read from.
4050 @param[in, out] Buffer The pointer to buffer to read into.
4051 @param[in, out] Size The pointer to number of bytes in Buffer.
4052 @param[in] Truncate If the buffer is large enough, this has no effect.
4053 If the buffer is is too small and Truncate is TRUE,
4054 the line will be truncated.
4055 If the buffer is is too small and Truncate is FALSE,
4056 then no read will occur.
jcarsey4d0a4fc2011-07-06 22:28:36 +00004057
ydong104ff7e372011-09-02 08:05:34 +00004058 @param[in, out] Ascii Boolean value for indicating whether the file is
4059 Ascii (TRUE) or UCS2 (FALSE).
jcarsey4d0a4fc2011-07-06 22:28:36 +00004060
4061 @retval EFI_SUCCESS The operation was successful. The line is stored in
4062 Buffer.
4063 @retval EFI_INVALID_PARAMETER Handle was NULL.
4064 @retval EFI_INVALID_PARAMETER Size was NULL.
4065 @retval EFI_BUFFER_TOO_SMALL Size was not large enough to store the line.
4066 Size was updated to the minimum space required.
4067**/
4068EFI_STATUS
4069EFIAPI
4070ShellFileHandleReadLine(
4071 IN SHELL_FILE_HANDLE Handle,
4072 IN OUT CHAR16 *Buffer,
4073 IN OUT UINTN *Size,
4074 IN BOOLEAN Truncate,
4075 IN OUT BOOLEAN *Ascii
4076 )
4077{
4078 EFI_STATUS Status;
4079 CHAR16 CharBuffer;
4080 UINTN CharSize;
4081 UINTN CountSoFar;
4082 UINT64 OriginalFilePosition;
4083
4084
4085 if (Handle == NULL
4086 ||Size == NULL
4087 ){
4088 return (EFI_INVALID_PARAMETER);
4089 }
4090 if (Buffer == NULL) {
4091 ASSERT(*Size == 0);
4092 } else {
4093 *Buffer = CHAR_NULL;
4094 }
4095 gEfiShellProtocol->GetFilePosition(Handle, &OriginalFilePosition);
4096 if (OriginalFilePosition == 0) {
4097 CharSize = sizeof(CHAR16);
4098 Status = gEfiShellProtocol->ReadFile(Handle, &CharSize, &CharBuffer);
4099 ASSERT_EFI_ERROR(Status);
4100 if (CharBuffer == gUnicodeFileTag) {
4101 *Ascii = FALSE;
4102 } else {
4103 *Ascii = TRUE;
4104 gEfiShellProtocol->SetFilePosition(Handle, OriginalFilePosition);
4105 }
4106 }
4107
4108 for (CountSoFar = 0;;CountSoFar++){
4109 CharBuffer = 0;
4110 if (*Ascii) {
4111 CharSize = sizeof(CHAR8);
4112 } else {
4113 CharSize = sizeof(CHAR16);
4114 }
4115 Status = gEfiShellProtocol->ReadFile(Handle, &CharSize, &CharBuffer);
4116 if ( EFI_ERROR(Status)
4117 || CharSize == 0
4118 || (CharBuffer == L'\n' && !(*Ascii))
4119 || (CharBuffer == '\n' && *Ascii)
4120 ){
4121 break;
4122 }
4123 //
4124 // if we have space save it...
4125 //
4126 if ((CountSoFar+1)*sizeof(CHAR16) < *Size){
4127 ASSERT(Buffer != NULL);
4128 ((CHAR16*)Buffer)[CountSoFar] = CharBuffer;
4129 ((CHAR16*)Buffer)[CountSoFar+1] = CHAR_NULL;
4130 }
4131 }
4132
4133 //
4134 // if we ran out of space tell when...
4135 //
4136 if ((CountSoFar+1)*sizeof(CHAR16) > *Size){
4137 *Size = (CountSoFar+1)*sizeof(CHAR16);
4138 if (!Truncate) {
4139 gEfiShellProtocol->SetFilePosition(Handle, OriginalFilePosition);
4140 } else {
4141 DEBUG((DEBUG_WARN, "The line was truncated in ShellFileHandleReadLine"));
4142 }
4143 return (EFI_BUFFER_TOO_SMALL);
4144 }
4145 while(Buffer[StrLen(Buffer)-1] == L'\r') {
4146 Buffer[StrLen(Buffer)-1] = CHAR_NULL;
4147 }
4148
4149 return (Status);
4150}
jcarseyfb5278e2013-02-20 18:21:14 +00004151
4152/**
jcarsey365aa982013-03-04 21:54:02 +00004153 Function to print help file / man page content in the spec from the UEFI Shell protocol GetHelpText function.
4154
4155 @param[in] CommandToGetHelpOn Pointer to a string containing the command name of help file to be printed.
4156 @param[in] SectionToGetHelpOn Pointer to the section specifier(s).
4157 @param[in] PrintCommandText If TRUE, prints the command followed by the help content, otherwise prints
4158 the help content only.
4159 @retval EFI_DEVICE_ERROR The help data format was incorrect.
4160 @retval EFI_NOT_FOUND The help data could not be found.
4161 @retval EFI_SUCCESS The operation was successful.
4162**/
4163EFI_STATUS
4164EFIAPI
4165ShellPrintHelp (
4166 IN CONST CHAR16 *CommandToGetHelpOn,
4167 IN CONST CHAR16 *SectionToGetHelpOn,
4168 IN BOOLEAN PrintCommandText
4169 )
4170{
4171 EFI_STATUS Status;
4172 CHAR16 *OutText;
4173
4174 OutText = NULL;
4175
4176 //
4177 // Get the string to print based
4178 //
4179 Status = gEfiShellProtocol->GetHelpText (CommandToGetHelpOn, SectionToGetHelpOn, &OutText);
4180
4181 //
4182 // make sure we got a valid string
4183 //
4184 if (EFI_ERROR(Status)){
4185 return Status;
4186 }
4187 if (OutText == NULL || StrLen(OutText) == 0) {
4188 return EFI_NOT_FOUND;
4189 }
4190
4191 //
4192 // Chop off trailing stuff we dont need
4193 //
4194 while (OutText[StrLen(OutText)-1] == L'\r' || OutText[StrLen(OutText)-1] == L'\n' || OutText[StrLen(OutText)-1] == L' ') {
4195 OutText[StrLen(OutText)-1] = CHAR_NULL;
4196 }
4197
4198 //
4199 // Print this out to the console
4200 //
4201 if (PrintCommandText) {
4202 ShellPrintEx(-1, -1, L"%H%-14s%N- %s\r\n", CommandToGetHelpOn, OutText);
4203 } else {
4204 ShellPrintEx(-1, -1, L"%N%s\r\n", OutText);
4205 }
4206
4207 SHELL_FREE_NON_NULL(OutText);
4208
4209 return EFI_SUCCESS;
4210}
4211
4212/**
jcarseyfb5278e2013-02-20 18:21:14 +00004213 Function to delete a file by name
4214
4215 @param[in] FileName Pointer to file name to delete.
4216
4217 @retval EFI_SUCCESS the file was deleted sucessfully
4218 @retval EFI_WARN_DELETE_FAILURE the handle was closed, but the file was not
4219 deleted
4220 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
4221 @retval EFI_NOT_FOUND The specified file could not be found on the
4222 device or the file system could not be found
4223 on the device.
4224 @retval EFI_NO_MEDIA The device has no medium.
4225 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
4226 medium is no longer supported.
4227 @retval EFI_DEVICE_ERROR The device reported an error.
4228 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
4229 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
4230 @retval EFI_ACCESS_DENIED The file was opened read only.
4231 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
4232 file.
4233 @retval other The file failed to open
4234**/
4235EFI_STATUS
4236EFIAPI
4237ShellDeleteFileByName(
4238 IN CONST CHAR16 *FileName
4239 )
4240{
4241 EFI_STATUS Status;
4242 SHELL_FILE_HANDLE FileHandle;
4243
4244 Status = ShellFileExists(FileName);
4245
4246 if (Status == EFI_SUCCESS){
4247 Status = ShellOpenFileByName(FileName, &FileHandle, EFI_FILE_MODE_READ | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE, 0x0);
4248 if (Status == EFI_SUCCESS){
4249 Status = ShellDeleteFile(&FileHandle);
4250 }
4251 }
4252
4253 return(Status);
4254
4255}
Qiu Shumin0960ba12014-09-17 07:58:31 +00004256
4257/**
4258 Cleans off all the quotes in the string.
4259
4260 @param[in] OriginalString pointer to the string to be cleaned.
4261 @param[out] CleanString The new string with all quotes removed.
4262 Memory allocated in the function and free
4263 by caller.
4264
4265 @retval EFI_SUCCESS The operation was successful.
4266**/
4267EFI_STATUS
4268EFIAPI
4269InternalShellStripQuotes (
4270 IN CONST CHAR16 *OriginalString,
4271 OUT CHAR16 **CleanString
4272 )
4273{
4274 CHAR16 *Walker;
4275
4276 if (OriginalString == NULL || CleanString == NULL) {
4277 return EFI_INVALID_PARAMETER;
4278 }
4279
4280 *CleanString = AllocateCopyPool (StrSize (OriginalString), OriginalString);
4281 if (*CleanString == NULL) {
4282 return EFI_OUT_OF_RESOURCES;
4283 }
4284
4285 for (Walker = *CleanString; Walker != NULL && *Walker != CHAR_NULL ; Walker++) {
4286 if (*Walker == L'\"') {
4287 CopyMem(Walker, Walker+1, StrSize(Walker) - sizeof(Walker[0]));
4288 }
4289 }
4290
4291 return EFI_SUCCESS;
4292}
4293