blob: 819c9f03ac16f4ae779d6acaa343a3416b25a046 [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;
jcarsey1e6e84c2010-01-25 20:05:08 +00001497
jcarsey94b17fa2009-05-07 18:46:18 +00001498 //
1499 // ASSERT that Arg and ListHead are not NULL
1500 //
1501 ASSERT(Arg != NULL);
1502 ASSERT(ListHead != NULL);
1503
jcarsey1e6e84c2010-01-25 20:05:08 +00001504 //
jcarsey94b17fa2009-05-07 18:46:18 +00001505 // Check for UEFI Shell 2.0 protocols
1506 //
jcarsey366f81a2011-06-27 21:04:22 +00001507 if (gEfiShellProtocol != NULL) {
jcarsey5f7431d2009-07-10 18:06:01 +00001508 if (*ListHead == NULL) {
1509 *ListHead = (EFI_SHELL_FILE_INFO*)AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1510 if (*ListHead == NULL) {
1511 return (EFI_OUT_OF_RESOURCES);
1512 }
1513 InitializeListHead(&((*ListHead)->Link));
jcarsey1e6e84c2010-01-25 20:05:08 +00001514 }
jcarsey366f81a2011-06-27 21:04:22 +00001515 Status = gEfiShellProtocol->OpenFileList(Arg,
jcarsey1e6e84c2010-01-25 20:05:08 +00001516 OpenMode,
jcarsey2247dde2009-11-09 18:08:58 +00001517 ListHead);
1518 if (EFI_ERROR(Status)) {
jcarsey366f81a2011-06-27 21:04:22 +00001519 gEfiShellProtocol->RemoveDupInFileList(ListHead);
jcarsey2247dde2009-11-09 18:08:58 +00001520 } else {
jcarsey366f81a2011-06-27 21:04:22 +00001521 Status = gEfiShellProtocol->RemoveDupInFileList(ListHead);
jcarsey2247dde2009-11-09 18:08:58 +00001522 }
jcarseya405b862010-09-14 05:18:09 +00001523 if (*ListHead != NULL && IsListEmpty(&(*ListHead)->Link)) {
1524 FreePool(*ListHead);
1525 *ListHead = NULL;
1526 return (EFI_NOT_FOUND);
1527 }
jcarsey2247dde2009-11-09 18:08:58 +00001528 return (Status);
jcarsey1e6e84c2010-01-25 20:05:08 +00001529 }
jcarsey94b17fa2009-05-07 18:46:18 +00001530
1531 //
jcarsey92a54472011-06-27 20:33:13 +00001532 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001533 //
jcarsey92a54472011-06-27 20:33:13 +00001534 if (mEfiShellEnvironment2 != NULL) {
1535 //
1536 // make sure the list head is initialized
1537 //
1538 InitializeListHead(&mOldStyleFileList);
jcarsey94b17fa2009-05-07 18:46:18 +00001539
jcarsey92a54472011-06-27 20:33:13 +00001540 //
1541 // Get the EFI Shell list of files
1542 //
1543 Status = mEfiShellEnvironment2->FileMetaArg(Arg, &mOldStyleFileList);
1544 if (EFI_ERROR(Status)) {
1545 *ListHead = NULL;
1546 return (Status);
1547 }
jcarsey94b17fa2009-05-07 18:46:18 +00001548
jcarsey92a54472011-06-27 20:33:13 +00001549 if (*ListHead == NULL) {
1550 *ListHead = (EFI_SHELL_FILE_INFO *)AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1551 if (*ListHead == NULL) {
1552 return (EFI_OUT_OF_RESOURCES);
1553 }
1554 InitializeListHead(&((*ListHead)->Link));
1555 }
1556
1557 //
1558 // Convert that to equivalent of UEFI Shell 2.0 structure
1559 //
1560 InternalShellConvertFileListType(&mOldStyleFileList, &(*ListHead)->Link);
1561
1562 //
1563 // Free the EFI Shell version that was converted.
1564 //
1565 mEfiShellEnvironment2->FreeFileList(&mOldStyleFileList);
1566
1567 if ((*ListHead)->Link.ForwardLink == (*ListHead)->Link.BackLink && (*ListHead)->Link.BackLink == &((*ListHead)->Link)) {
1568 FreePool(*ListHead);
1569 *ListHead = NULL;
1570 Status = EFI_NOT_FOUND;
1571 }
jcarsey94b17fa2009-05-07 18:46:18 +00001572 return (Status);
1573 }
1574
jcarsey92a54472011-06-27 20:33:13 +00001575 return (EFI_UNSUPPORTED);
jcarsey94b17fa2009-05-07 18:46:18 +00001576}
1577/**
jcarseya405b862010-09-14 05:18:09 +00001578 Free the linked list returned from ShellOpenFileMetaArg.
jcarsey94b17fa2009-05-07 18:46:18 +00001579
jcarseya405b862010-09-14 05:18:09 +00001580 if ListHead is NULL then ASSERT().
jcarsey94b17fa2009-05-07 18:46:18 +00001581
jcarseya405b862010-09-14 05:18:09 +00001582 @param ListHead the pointer to free.
jcarsey94b17fa2009-05-07 18:46:18 +00001583
jcarseya405b862010-09-14 05:18:09 +00001584 @retval EFI_SUCCESS the operation was sucessful.
jcarsey94b17fa2009-05-07 18:46:18 +00001585**/
1586EFI_STATUS
1587EFIAPI
1588ShellCloseFileMetaArg (
1589 IN OUT EFI_SHELL_FILE_INFO **ListHead
1590 )
1591{
1592 LIST_ENTRY *Node;
1593
1594 //
1595 // ASSERT that ListHead is not NULL
1596 //
1597 ASSERT(ListHead != NULL);
1598
jcarsey1e6e84c2010-01-25 20:05:08 +00001599 //
jcarsey94b17fa2009-05-07 18:46:18 +00001600 // Check for UEFI Shell 2.0 protocols
1601 //
jcarsey366f81a2011-06-27 21:04:22 +00001602 if (gEfiShellProtocol != NULL) {
1603 return (gEfiShellProtocol->FreeFileList(ListHead));
jcarsey92a54472011-06-27 20:33:13 +00001604 } else if (mEfiShellEnvironment2 != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001605 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001606 // Since this is EFI Shell version we need to free our internally made copy
jcarsey94b17fa2009-05-07 18:46:18 +00001607 // of the list
1608 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001609 for ( Node = GetFirstNode(&(*ListHead)->Link)
jcarseya405b862010-09-14 05:18:09 +00001610 ; *ListHead != NULL && !IsListEmpty(&(*ListHead)->Link)
jcarsey9b3bf082009-06-23 21:15:07 +00001611 ; Node = GetFirstNode(&(*ListHead)->Link)) {
jcarsey94b17fa2009-05-07 18:46:18 +00001612 RemoveEntryList(Node);
jcarseya405b862010-09-14 05:18:09 +00001613 ((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 +00001614 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->FullName);
1615 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->FileName);
1616 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->Info);
1617 FreePool((EFI_SHELL_FILE_INFO_NO_CONST*)Node);
1618 }
Jaben Carsey75a5e2e2014-01-10 16:42:45 +00001619 SHELL_FREE_NON_NULL(*ListHead);
jcarsey94b17fa2009-05-07 18:46:18 +00001620 return EFI_SUCCESS;
1621 }
jcarsey92a54472011-06-27 20:33:13 +00001622
1623 return (EFI_UNSUPPORTED);
jcarsey94b17fa2009-05-07 18:46:18 +00001624}
1625
jcarsey125c2cf2009-11-18 21:36:50 +00001626/**
1627 Find a file by searching the CWD and then the path.
1628
jcarseyb3011f42010-01-11 21:49:04 +00001629 If FileName is NULL then ASSERT.
jcarsey125c2cf2009-11-18 21:36:50 +00001630
jcarseyb3011f42010-01-11 21:49:04 +00001631 If the return value is not NULL then the memory must be caller freed.
jcarsey125c2cf2009-11-18 21:36:50 +00001632
1633 @param FileName Filename string.
1634
1635 @retval NULL the file was not found
1636 @return !NULL the full path to the file.
1637**/
1638CHAR16 *
1639EFIAPI
1640ShellFindFilePath (
1641 IN CONST CHAR16 *FileName
1642 )
1643{
1644 CONST CHAR16 *Path;
jcarseya405b862010-09-14 05:18:09 +00001645 SHELL_FILE_HANDLE Handle;
jcarsey125c2cf2009-11-18 21:36:50 +00001646 EFI_STATUS Status;
1647 CHAR16 *RetVal;
1648 CHAR16 *TestPath;
1649 CONST CHAR16 *Walker;
jcarsey36a9d672009-11-20 21:13:41 +00001650 UINTN Size;
jcarsey1cd45e72010-01-29 15:07:44 +00001651 CHAR16 *TempChar;
jcarsey125c2cf2009-11-18 21:36:50 +00001652
1653 RetVal = NULL;
1654
jcarseya405b862010-09-14 05:18:09 +00001655 //
1656 // First make sure its not an absolute path.
1657 //
1658 Status = ShellOpenFileByName(FileName, &Handle, EFI_FILE_MODE_READ, 0);
1659 if (!EFI_ERROR(Status)){
1660 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1661 ASSERT(RetVal == NULL);
1662 RetVal = StrnCatGrow(&RetVal, NULL, FileName, 0);
1663 ShellCloseFile(&Handle);
1664 return (RetVal);
1665 } else {
1666 ShellCloseFile(&Handle);
1667 }
1668 }
1669
jcarsey125c2cf2009-11-18 21:36:50 +00001670 Path = ShellGetEnvironmentVariable(L"cwd");
1671 if (Path != NULL) {
jcarsey36a9d672009-11-20 21:13:41 +00001672 Size = StrSize(Path);
1673 Size += StrSize(FileName);
1674 TestPath = AllocateZeroPool(Size);
jcarseyc9d92df2010-02-03 15:37:54 +00001675 if (TestPath == NULL) {
1676 return (NULL);
1677 }
Jaben Carsey98c16be2014-08-19 21:00:34 +00001678 StrnCpy(TestPath, Path, Size/sizeof(CHAR16) - 1);
1679 StrnCat(TestPath, FileName, Size/sizeof(CHAR16) - 1 - StrLen(TestPath));
jcarsey125c2cf2009-11-18 21:36:50 +00001680 Status = ShellOpenFileByName(TestPath, &Handle, EFI_FILE_MODE_READ, 0);
1681 if (!EFI_ERROR(Status)){
jcarseya405b862010-09-14 05:18:09 +00001682 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1683 ASSERT(RetVal == NULL);
1684 RetVal = StrnCatGrow(&RetVal, NULL, TestPath, 0);
1685 ShellCloseFile(&Handle);
1686 FreePool(TestPath);
1687 return (RetVal);
1688 } else {
1689 ShellCloseFile(&Handle);
1690 }
jcarsey125c2cf2009-11-18 21:36:50 +00001691 }
1692 FreePool(TestPath);
1693 }
1694 Path = ShellGetEnvironmentVariable(L"path");
1695 if (Path != NULL) {
jcarseya405b862010-09-14 05:18:09 +00001696 Size = StrSize(Path)+sizeof(CHAR16);
jcarsey36a9d672009-11-20 21:13:41 +00001697 Size += StrSize(FileName);
1698 TestPath = AllocateZeroPool(Size);
jcarsey3e082d52010-10-04 16:44:57 +00001699 if (TestPath == NULL) {
1700 return (NULL);
1701 }
jcarsey1e6e84c2010-01-25 20:05:08 +00001702 Walker = (CHAR16*)Path;
jcarsey125c2cf2009-11-18 21:36:50 +00001703 do {
1704 CopyMem(TestPath, Walker, StrSize(Walker));
jcarsey3e082d52010-10-04 16:44:57 +00001705 if (TestPath != NULL) {
1706 TempChar = StrStr(TestPath, L";");
1707 if (TempChar != NULL) {
1708 *TempChar = CHAR_NULL;
1709 }
1710 if (TestPath[StrLen(TestPath)-1] != L'\\') {
Jaben Carsey98c16be2014-08-19 21:00:34 +00001711 StrnCat(TestPath, L"\\", Size/sizeof(CHAR16) - 1 - StrLen(TestPath));
jcarsey3e082d52010-10-04 16:44:57 +00001712 }
jcarsey89e85372011-04-13 23:37:50 +00001713 if (FileName[0] == L'\\') {
1714 FileName++;
1715 }
Jaben Carsey98c16be2014-08-19 21:00:34 +00001716 StrnCat(TestPath, FileName, Size/sizeof(CHAR16) - 1 - StrLen(TestPath));
jcarsey3e082d52010-10-04 16:44:57 +00001717 if (StrStr(Walker, L";") != NULL) {
1718 Walker = StrStr(Walker, L";") + 1;
jcarseya405b862010-09-14 05:18:09 +00001719 } else {
jcarsey3e082d52010-10-04 16:44:57 +00001720 Walker = NULL;
1721 }
1722 Status = ShellOpenFileByName(TestPath, &Handle, EFI_FILE_MODE_READ, 0);
1723 if (!EFI_ERROR(Status)){
1724 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1725 ASSERT(RetVal == NULL);
1726 RetVal = StrnCatGrow(&RetVal, NULL, TestPath, 0);
1727 ShellCloseFile(&Handle);
1728 break;
1729 } else {
1730 ShellCloseFile(&Handle);
1731 }
jcarseya405b862010-09-14 05:18:09 +00001732 }
jcarsey125c2cf2009-11-18 21:36:50 +00001733 }
1734 } while (Walker != NULL && Walker[0] != CHAR_NULL);
1735 FreePool(TestPath);
1736 }
1737 return (RetVal);
1738}
1739
jcarseyb3011f42010-01-11 21:49:04 +00001740/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001741 Find a file by searching the CWD and then the path with a variable set of file
1742 extensions. If the file is not found it will append each extension in the list
jcarseyb3011f42010-01-11 21:49:04 +00001743 in the order provided and return the first one that is successful.
1744
1745 If FileName is NULL, then ASSERT.
1746 If FileExtension is NULL, then behavior is identical to ShellFindFilePath.
1747
1748 If the return value is not NULL then the memory must be caller freed.
1749
1750 @param[in] FileName Filename string.
1751 @param[in] FileExtension Semi-colon delimeted list of possible extensions.
1752
1753 @retval NULL The file was not found.
1754 @retval !NULL The path to the file.
1755**/
1756CHAR16 *
1757EFIAPI
1758ShellFindFilePathEx (
1759 IN CONST CHAR16 *FileName,
1760 IN CONST CHAR16 *FileExtension
1761 )
1762{
1763 CHAR16 *TestPath;
1764 CHAR16 *RetVal;
1765 CONST CHAR16 *ExtensionWalker;
jcarsey9e926b62010-01-14 20:26:39 +00001766 UINTN Size;
jcarsey1cd45e72010-01-29 15:07:44 +00001767 CHAR16 *TempChar;
jcarseyc9d92df2010-02-03 15:37:54 +00001768 CHAR16 *TempChar2;
jcarsey1cd45e72010-01-29 15:07:44 +00001769
jcarseyb3011f42010-01-11 21:49:04 +00001770 ASSERT(FileName != NULL);
1771 if (FileExtension == NULL) {
1772 return (ShellFindFilePath(FileName));
1773 }
1774 RetVal = ShellFindFilePath(FileName);
1775 if (RetVal != NULL) {
1776 return (RetVal);
1777 }
jcarsey9e926b62010-01-14 20:26:39 +00001778 Size = StrSize(FileName);
1779 Size += StrSize(FileExtension);
1780 TestPath = AllocateZeroPool(Size);
jcarseyc9d92df2010-02-03 15:37:54 +00001781 if (TestPath == NULL) {
1782 return (NULL);
1783 }
jcarseya405b862010-09-14 05:18:09 +00001784 for (ExtensionWalker = FileExtension, TempChar2 = (CHAR16*)FileExtension; TempChar2 != NULL ; ExtensionWalker = TempChar2 + 1){
Jaben Carsey98c16be2014-08-19 21:00:34 +00001785 StrnCpy(TestPath, FileName, Size/sizeof(CHAR16) - 1);
jcarseya405b862010-09-14 05:18:09 +00001786 if (ExtensionWalker != NULL) {
Jaben Carsey98c16be2014-08-19 21:00:34 +00001787 StrnCat(TestPath, ExtensionWalker, Size/sizeof(CHAR16) - 1 - StrLen(TestPath));
jcarseya405b862010-09-14 05:18:09 +00001788 }
jcarsey1cd45e72010-01-29 15:07:44 +00001789 TempChar = StrStr(TestPath, L";");
1790 if (TempChar != NULL) {
1791 *TempChar = CHAR_NULL;
jcarseyb3011f42010-01-11 21:49:04 +00001792 }
1793 RetVal = ShellFindFilePath(TestPath);
1794 if (RetVal != NULL) {
1795 break;
1796 }
jcarseya405b862010-09-14 05:18:09 +00001797 ASSERT(ExtensionWalker != NULL);
jcarseyc9d92df2010-02-03 15:37:54 +00001798 TempChar2 = StrStr(ExtensionWalker, L";");
jcarseyb3011f42010-01-11 21:49:04 +00001799 }
1800 FreePool(TestPath);
1801 return (RetVal);
1802}
1803
jcarsey94b17fa2009-05-07 18:46:18 +00001804typedef struct {
jcarsey9b3bf082009-06-23 21:15:07 +00001805 LIST_ENTRY Link;
jcarsey94b17fa2009-05-07 18:46:18 +00001806 CHAR16 *Name;
jcarseya405b862010-09-14 05:18:09 +00001807 SHELL_PARAM_TYPE Type;
jcarsey94b17fa2009-05-07 18:46:18 +00001808 CHAR16 *Value;
1809 UINTN OriginalPosition;
1810} SHELL_PARAM_PACKAGE;
1811
1812/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001813 Checks the list of valid arguments and returns TRUE if the item was found. If the
jcarsey94b17fa2009-05-07 18:46:18 +00001814 return value is TRUE then the type parameter is set also.
jcarsey1e6e84c2010-01-25 20:05:08 +00001815
jcarsey94b17fa2009-05-07 18:46:18 +00001816 if CheckList is NULL then ASSERT();
1817 if Name is NULL then ASSERT();
1818 if Type is NULL then ASSERT();
1819
jcarsey94b17fa2009-05-07 18:46:18 +00001820 @param Name pointer to Name of parameter found
1821 @param CheckList List to check against
jcarseya405b862010-09-14 05:18:09 +00001822 @param Type pointer to type of parameter if it was found
jcarsey94b17fa2009-05-07 18:46:18 +00001823
1824 @retval TRUE the Parameter was found. Type is valid.
1825 @retval FALSE the Parameter was not found. Type is not valid.
1826**/
1827BOOLEAN
1828EFIAPI
jcarseyd2b45642009-05-11 18:02:16 +00001829InternalIsOnCheckList (
jcarsey94b17fa2009-05-07 18:46:18 +00001830 IN CONST CHAR16 *Name,
1831 IN CONST SHELL_PARAM_ITEM *CheckList,
jcarsey252d9452011-03-25 20:49:53 +00001832 OUT SHELL_PARAM_TYPE *Type
jcarseya405b862010-09-14 05:18:09 +00001833 )
1834{
jcarsey94b17fa2009-05-07 18:46:18 +00001835 SHELL_PARAM_ITEM *TempListItem;
jcarsey252d9452011-03-25 20:49:53 +00001836 CHAR16 *TempString;
jcarsey94b17fa2009-05-07 18:46:18 +00001837
1838 //
1839 // ASSERT that all 3 pointer parameters aren't NULL
1840 //
1841 ASSERT(CheckList != NULL);
1842 ASSERT(Type != NULL);
1843 ASSERT(Name != NULL);
1844
1845 //
jcarseyd2b45642009-05-11 18:02:16 +00001846 // question mark and page break mode are always supported
1847 //
1848 if ((StrCmp(Name, L"-?") == 0) ||
1849 (StrCmp(Name, L"-b") == 0)
jcarseya405b862010-09-14 05:18:09 +00001850 ) {
jcarsey252d9452011-03-25 20:49:53 +00001851 *Type = TypeFlag;
jcarseyd2b45642009-05-11 18:02:16 +00001852 return (TRUE);
1853 }
1854
1855 //
jcarsey94b17fa2009-05-07 18:46:18 +00001856 // Enumerate through the list
1857 //
1858 for (TempListItem = (SHELL_PARAM_ITEM*)CheckList ; TempListItem->Name != NULL ; TempListItem++) {
1859 //
jcarsey9eb53ac2009-07-08 17:26:58 +00001860 // If the Type is TypeStart only check the first characters of the passed in param
1861 // If it matches set the type and return TRUE
jcarsey94b17fa2009-05-07 18:46:18 +00001862 //
darylm503b0934ac2011-11-12 00:35:11 +00001863 if (TempListItem->Type == TypeStart) {
jcarsey252d9452011-03-25 20:49:53 +00001864 if (StrnCmp(Name, TempListItem->Name, StrLen(TempListItem->Name)) == 0) {
1865 *Type = TempListItem->Type;
1866 return (TRUE);
1867 }
1868 TempString = NULL;
1869 TempString = StrnCatGrow(&TempString, NULL, Name, StrLen(TempListItem->Name));
1870 if (TempString != NULL) {
1871 if (StringNoCaseCompare(&TempString, &TempListItem->Name) == 0) {
1872 *Type = TempListItem->Type;
1873 FreePool(TempString);
1874 return (TRUE);
1875 }
1876 FreePool(TempString);
1877 }
1878 } else if (StringNoCaseCompare(&Name, &TempListItem->Name) == 0) {
jcarsey94b17fa2009-05-07 18:46:18 +00001879 *Type = TempListItem->Type;
1880 return (TRUE);
1881 }
1882 }
jcarsey2247dde2009-11-09 18:08:58 +00001883
jcarsey94b17fa2009-05-07 18:46:18 +00001884 return (FALSE);
1885}
1886/**
jcarseyd2b45642009-05-11 18:02:16 +00001887 Checks the string for indicators of "flag" status. this is a leading '/', '-', or '+'
jcarsey94b17fa2009-05-07 18:46:18 +00001888
jcarseya405b862010-09-14 05:18:09 +00001889 @param[in] Name pointer to Name of parameter found
1890 @param[in] AlwaysAllowNumbers TRUE to allow numbers, FALSE to not.
jcarsey94b17fa2009-05-07 18:46:18 +00001891
1892 @retval TRUE the Parameter is a flag.
jcarseya405b862010-09-14 05:18:09 +00001893 @retval FALSE the Parameter not a flag.
jcarsey94b17fa2009-05-07 18:46:18 +00001894**/
1895BOOLEAN
1896EFIAPI
jcarseyd2b45642009-05-11 18:02:16 +00001897InternalIsFlag (
jcarsey2247dde2009-11-09 18:08:58 +00001898 IN CONST CHAR16 *Name,
1899 IN BOOLEAN AlwaysAllowNumbers
jcarsey94b17fa2009-05-07 18:46:18 +00001900 )
1901{
1902 //
1903 // ASSERT that Name isn't NULL
1904 //
1905 ASSERT(Name != NULL);
1906
1907 //
jcarsey2247dde2009-11-09 18:08:58 +00001908 // If we accept numbers then dont return TRUE. (they will be values)
1909 //
Jaben Carsey8af89da2014-08-19 20:58:03 +00001910 if (((Name[0] == L'-' || Name[0] == L'+') && InternalShellIsHexOrDecimalNumber(Name+1, FALSE, FALSE)) && AlwaysAllowNumbers) {
jcarsey2247dde2009-11-09 18:08:58 +00001911 return (FALSE);
1912 }
1913
1914 //
jcarseya405b862010-09-14 05:18:09 +00001915 // If the Name has a /, +, or - as the first character return TRUE
jcarsey94b17fa2009-05-07 18:46:18 +00001916 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001917 if ((Name[0] == L'/') ||
jcarseyd2b45642009-05-11 18:02:16 +00001918 (Name[0] == L'-') ||
1919 (Name[0] == L'+')
jcarseya405b862010-09-14 05:18:09 +00001920 ) {
jcarsey94b17fa2009-05-07 18:46:18 +00001921 return (TRUE);
1922 }
1923 return (FALSE);
1924}
1925
1926/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001927 Checks the command line arguments passed against the list of valid ones.
jcarsey94b17fa2009-05-07 18:46:18 +00001928
1929 If no initialization is required, then return RETURN_SUCCESS.
jcarsey1e6e84c2010-01-25 20:05:08 +00001930
jcarseya405b862010-09-14 05:18:09 +00001931 @param[in] CheckList pointer to list of parameters to check
1932 @param[out] CheckPackage pointer to pointer to list checked values
1933 @param[out] ProblemParam optional pointer to pointer to unicode string for
jcarseyd2b45642009-05-11 18:02:16 +00001934 the paramater that caused failure. If used then the
1935 caller is responsible for freeing the memory.
jcarseya405b862010-09-14 05:18:09 +00001936 @param[in] AutoPageBreak will automatically set PageBreakEnabled for "b" parameter
1937 @param[in] Argv pointer to array of parameters
1938 @param[in] Argc Count of parameters in Argv
1939 @param[in] AlwaysAllowNumbers TRUE to allow numbers always, FALSE otherwise.
jcarsey94b17fa2009-05-07 18:46:18 +00001940
1941 @retval EFI_SUCCESS The operation completed sucessfully.
1942 @retval EFI_OUT_OF_RESOURCES A memory allocation failed
1943 @retval EFI_INVALID_PARAMETER A parameter was invalid
jcarsey1e6e84c2010-01-25 20:05:08 +00001944 @retval EFI_VOLUME_CORRUPTED the command line was corrupt. an argument was
1945 duplicated. the duplicated command line argument
jcarsey94b17fa2009-05-07 18:46:18 +00001946 was returned in ProblemParam if provided.
jcarsey1e6e84c2010-01-25 20:05:08 +00001947 @retval EFI_NOT_FOUND a argument required a value that was missing.
jcarsey94b17fa2009-05-07 18:46:18 +00001948 the invalid command line argument was returned in
1949 ProblemParam if provided.
1950**/
1951EFI_STATUS
1952EFIAPI
1953InternalCommandLineParse (
1954 IN CONST SHELL_PARAM_ITEM *CheckList,
1955 OUT LIST_ENTRY **CheckPackage,
1956 OUT CHAR16 **ProblemParam OPTIONAL,
1957 IN BOOLEAN AutoPageBreak,
1958 IN CONST CHAR16 **Argv,
jcarsey2247dde2009-11-09 18:08:58 +00001959 IN UINTN Argc,
1960 IN BOOLEAN AlwaysAllowNumbers
jcarseya405b862010-09-14 05:18:09 +00001961 )
1962{
jcarsey94b17fa2009-05-07 18:46:18 +00001963 UINTN LoopCounter;
jcarsey252d9452011-03-25 20:49:53 +00001964 SHELL_PARAM_TYPE CurrentItemType;
jcarsey94b17fa2009-05-07 18:46:18 +00001965 SHELL_PARAM_PACKAGE *CurrentItemPackage;
jcarsey125c2cf2009-11-18 21:36:50 +00001966 UINTN GetItemValue;
1967 UINTN ValueSize;
jcarseya405b862010-09-14 05:18:09 +00001968 UINTN Count;
jcarsey252d9452011-03-25 20:49:53 +00001969 CONST CHAR16 *TempPointer;
Jaben Carsey98c16be2014-08-19 21:00:34 +00001970 UINTN CurrentValueSize;
jcarsey94b17fa2009-05-07 18:46:18 +00001971
1972 CurrentItemPackage = NULL;
jcarsey125c2cf2009-11-18 21:36:50 +00001973 GetItemValue = 0;
1974 ValueSize = 0;
jcarseya405b862010-09-14 05:18:09 +00001975 Count = 0;
jcarsey94b17fa2009-05-07 18:46:18 +00001976
1977 //
1978 // If there is only 1 item we dont need to do anything
1979 //
jcarseya405b862010-09-14 05:18:09 +00001980 if (Argc < 1) {
jcarsey94b17fa2009-05-07 18:46:18 +00001981 *CheckPackage = NULL;
1982 return (EFI_SUCCESS);
1983 }
1984
1985 //
jcarsey2247dde2009-11-09 18:08:58 +00001986 // ASSERTs
1987 //
1988 ASSERT(CheckList != NULL);
1989 ASSERT(Argv != NULL);
1990
1991 //
jcarsey94b17fa2009-05-07 18:46:18 +00001992 // initialize the linked list
1993 //
1994 *CheckPackage = (LIST_ENTRY*)AllocateZeroPool(sizeof(LIST_ENTRY));
sfu502a758c2011-10-08 02:55:30 +00001995 if (*CheckPackage == NULL) {
jcarseybeab0fc2011-10-10 17:26:25 +00001996 return (EFI_OUT_OF_RESOURCES);
sfu502a758c2011-10-08 02:55:30 +00001997 }
jcarseybeab0fc2011-10-10 17:26:25 +00001998
jcarsey94b17fa2009-05-07 18:46:18 +00001999 InitializeListHead(*CheckPackage);
2000
2001 //
2002 // loop through each of the arguments
2003 //
2004 for (LoopCounter = 0 ; LoopCounter < Argc ; ++LoopCounter) {
2005 if (Argv[LoopCounter] == NULL) {
2006 //
2007 // do nothing for NULL argv
2008 //
jcarseya405b862010-09-14 05:18:09 +00002009 } else if (InternalIsOnCheckList(Argv[LoopCounter], CheckList, &CurrentItemType)) {
jcarsey94b17fa2009-05-07 18:46:18 +00002010 //
jcarsey2247dde2009-11-09 18:08:58 +00002011 // We might have leftover if last parameter didnt have optional value
2012 //
jcarsey125c2cf2009-11-18 21:36:50 +00002013 if (GetItemValue != 0) {
2014 GetItemValue = 0;
jcarsey2247dde2009-11-09 18:08:58 +00002015 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2016 }
2017 //
jcarsey94b17fa2009-05-07 18:46:18 +00002018 // this is a flag
2019 //
jcarsey252d9452011-03-25 20:49:53 +00002020 CurrentItemPackage = AllocateZeroPool(sizeof(SHELL_PARAM_PACKAGE));
jcarseybeab0fc2011-10-10 17:26:25 +00002021 if (CurrentItemPackage == NULL) {
2022 ShellCommandLineFreeVarList(*CheckPackage);
2023 *CheckPackage = NULL;
2024 return (EFI_OUT_OF_RESOURCES);
2025 }
Jaben Carsey98c16be2014-08-19 21:00:34 +00002026 CurrentItemPackage->Name = AllocateCopyPool(StrSize(Argv[LoopCounter]), Argv[LoopCounter]);
jcarseybeab0fc2011-10-10 17:26:25 +00002027 if (CurrentItemPackage->Name == NULL) {
2028 ShellCommandLineFreeVarList(*CheckPackage);
2029 *CheckPackage = NULL;
2030 return (EFI_OUT_OF_RESOURCES);
2031 }
jcarsey94b17fa2009-05-07 18:46:18 +00002032 CurrentItemPackage->Type = CurrentItemType;
2033 CurrentItemPackage->OriginalPosition = (UINTN)(-1);
jcarseyb1f95a02009-06-16 00:23:19 +00002034 CurrentItemPackage->Value = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +00002035
2036 //
2037 // Does this flag require a value
2038 //
jcarsey125c2cf2009-11-18 21:36:50 +00002039 switch (CurrentItemPackage->Type) {
jcarsey94b17fa2009-05-07 18:46:18 +00002040 //
jcarsey125c2cf2009-11-18 21:36:50 +00002041 // possibly trigger the next loop(s) to populate the value of this item
jcarsey1e6e84c2010-01-25 20:05:08 +00002042 //
jcarsey125c2cf2009-11-18 21:36:50 +00002043 case TypeValue:
jcarsey1e6e84c2010-01-25 20:05:08 +00002044 GetItemValue = 1;
jcarsey125c2cf2009-11-18 21:36:50 +00002045 ValueSize = 0;
2046 break;
2047 case TypeDoubleValue:
2048 GetItemValue = 2;
2049 ValueSize = 0;
2050 break;
2051 case TypeMaxValue:
2052 GetItemValue = (UINTN)(-1);
2053 ValueSize = 0;
2054 break;
2055 default:
2056 //
2057 // this item has no value expected; we are done
2058 //
2059 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2060 ASSERT(GetItemValue == 0);
2061 break;
jcarsey94b17fa2009-05-07 18:46:18 +00002062 }
jcarseya405b862010-09-14 05:18:09 +00002063 } else if (GetItemValue != 0 && !InternalIsFlag(Argv[LoopCounter], AlwaysAllowNumbers)) {
jcarseyb1f95a02009-06-16 00:23:19 +00002064 ASSERT(CurrentItemPackage != NULL);
2065 //
jcarsey125c2cf2009-11-18 21:36:50 +00002066 // get the item VALUE for a previous flag
jcarseyb1f95a02009-06-16 00:23:19 +00002067 //
jcarsey49bd4982012-01-27 18:42:43 +00002068 if (StrStr(Argv[LoopCounter], L" ") == NULL) {
Jaben Carsey98c16be2014-08-19 21:00:34 +00002069 CurrentValueSize = ValueSize + StrSize(Argv[LoopCounter]) + sizeof(CHAR16);
2070 CurrentItemPackage->Value = ReallocatePool(ValueSize, CurrentValueSize, CurrentItemPackage->Value);
jcarsey49bd4982012-01-27 18:42:43 +00002071 ASSERT(CurrentItemPackage->Value != NULL);
2072 if (ValueSize == 0) {
Jaben Carsey98c16be2014-08-19 21:00:34 +00002073 StrnCpy(CurrentItemPackage->Value, Argv[LoopCounter], CurrentValueSize/sizeof(CHAR16) - 1);
jcarsey49bd4982012-01-27 18:42:43 +00002074 } else {
Jaben Carsey98c16be2014-08-19 21:00:34 +00002075 StrnCat(CurrentItemPackage->Value, L" ", CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
2076 StrnCat(CurrentItemPackage->Value, Argv[LoopCounter], CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
jcarsey49bd4982012-01-27 18:42:43 +00002077 }
2078 ValueSize += StrSize(Argv[LoopCounter]) + sizeof(CHAR16);
jcarsey125c2cf2009-11-18 21:36:50 +00002079 } else {
jcarsey49bd4982012-01-27 18:42:43 +00002080 //
2081 // the parameter has spaces. must be quoted.
2082 //
Jaben Carsey98c16be2014-08-19 21:00:34 +00002083 CurrentValueSize = ValueSize + StrSize(Argv[LoopCounter]) + sizeof(CHAR16) + sizeof(CHAR16) + sizeof(CHAR16);
2084 CurrentItemPackage->Value = ReallocatePool(ValueSize, CurrentValueSize, CurrentItemPackage->Value);
jcarsey49bd4982012-01-27 18:42:43 +00002085 ASSERT(CurrentItemPackage->Value != NULL);
2086 if (ValueSize == 0) {
Jaben Carsey98c16be2014-08-19 21:00:34 +00002087 StrnCpy(CurrentItemPackage->Value, L"\"", CurrentValueSize/sizeof(CHAR16) - 1);
2088 StrnCat(CurrentItemPackage->Value, Argv[LoopCounter], CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
2089 StrnCat(CurrentItemPackage->Value, L"\"", CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
jcarsey49bd4982012-01-27 18:42:43 +00002090 } else {
Jaben Carsey98c16be2014-08-19 21:00:34 +00002091 StrnCat(CurrentItemPackage->Value, L" ", CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
2092 StrnCat(CurrentItemPackage->Value, L"\"", CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
2093 StrnCat(CurrentItemPackage->Value, Argv[LoopCounter], CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
2094 StrnCat(CurrentItemPackage->Value, L"\"", CurrentValueSize/sizeof(CHAR16) - 1 - StrLen(CurrentItemPackage->Value));
jcarsey49bd4982012-01-27 18:42:43 +00002095 }
2096 ValueSize += StrSize(Argv[LoopCounter]) + sizeof(CHAR16);
jcarsey125c2cf2009-11-18 21:36:50 +00002097 }
jcarsey125c2cf2009-11-18 21:36:50 +00002098 GetItemValue--;
2099 if (GetItemValue == 0) {
2100 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2101 }
jcarseya405b862010-09-14 05:18:09 +00002102 } else if (!InternalIsFlag(Argv[LoopCounter], AlwaysAllowNumbers) ){ //|| ProblemParam == NULL) {
jcarseyb1f95a02009-06-16 00:23:19 +00002103 //
2104 // add this one as a non-flag
2105 //
jcarsey252d9452011-03-25 20:49:53 +00002106
2107 TempPointer = Argv[LoopCounter];
darylm503b0934ac2011-11-12 00:35:11 +00002108 if ((*TempPointer == L'^' && *(TempPointer+1) == L'-')
jcarsey252d9452011-03-25 20:49:53 +00002109 || (*TempPointer == L'^' && *(TempPointer+1) == L'/')
2110 || (*TempPointer == L'^' && *(TempPointer+1) == L'+')
2111 ){
2112 TempPointer++;
2113 }
2114 CurrentItemPackage = AllocateZeroPool(sizeof(SHELL_PARAM_PACKAGE));
jcarseybeab0fc2011-10-10 17:26:25 +00002115 if (CurrentItemPackage == NULL) {
2116 ShellCommandLineFreeVarList(*CheckPackage);
2117 *CheckPackage = NULL;
2118 return (EFI_OUT_OF_RESOURCES);
2119 }
jcarseyb1f95a02009-06-16 00:23:19 +00002120 CurrentItemPackage->Name = NULL;
2121 CurrentItemPackage->Type = TypePosition;
Jaben Carsey98c16be2014-08-19 21:00:34 +00002122 CurrentItemPackage->Value = AllocateCopyPool(StrSize(TempPointer), TempPointer);
jcarseybeab0fc2011-10-10 17:26:25 +00002123 if (CurrentItemPackage->Value == NULL) {
2124 ShellCommandLineFreeVarList(*CheckPackage);
2125 *CheckPackage = NULL;
2126 return (EFI_OUT_OF_RESOURCES);
2127 }
jcarseya405b862010-09-14 05:18:09 +00002128 CurrentItemPackage->OriginalPosition = Count++;
jcarsey9b3bf082009-06-23 21:15:07 +00002129 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
jcarsey252d9452011-03-25 20:49:53 +00002130 } else {
jcarsey94b17fa2009-05-07 18:46:18 +00002131 //
2132 // this was a non-recognised flag... error!
2133 //
jcarsey252d9452011-03-25 20:49:53 +00002134 if (ProblemParam != NULL) {
Jaben Carsey98c16be2014-08-19 21:00:34 +00002135 *ProblemParam = AllocateCopyPool(StrSize(Argv[LoopCounter]), Argv[LoopCounter]);
jcarsey252d9452011-03-25 20:49:53 +00002136 }
jcarsey94b17fa2009-05-07 18:46:18 +00002137 ShellCommandLineFreeVarList(*CheckPackage);
2138 *CheckPackage = NULL;
2139 return (EFI_VOLUME_CORRUPTED);
jcarsey94b17fa2009-05-07 18:46:18 +00002140 }
2141 }
jcarsey125c2cf2009-11-18 21:36:50 +00002142 if (GetItemValue != 0) {
2143 GetItemValue = 0;
2144 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2145 }
jcarsey94b17fa2009-05-07 18:46:18 +00002146 //
2147 // support for AutoPageBreak
2148 //
2149 if (AutoPageBreak && ShellCommandLineGetFlag(*CheckPackage, L"-b")) {
2150 ShellSetPageBreakMode(TRUE);
2151 }
2152 return (EFI_SUCCESS);
2153}
2154
2155/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002156 Checks the command line arguments passed against the list of valid ones.
jcarsey94b17fa2009-05-07 18:46:18 +00002157 Optionally removes NULL values first.
jcarsey1e6e84c2010-01-25 20:05:08 +00002158
jcarsey94b17fa2009-05-07 18:46:18 +00002159 If no initialization is required, then return RETURN_SUCCESS.
jcarsey1e6e84c2010-01-25 20:05:08 +00002160
jcarseya405b862010-09-14 05:18:09 +00002161 @param[in] CheckList The pointer to list of parameters to check.
2162 @param[out] CheckPackage The package of checked values.
2163 @param[out] ProblemParam Optional pointer to pointer to unicode string for
jcarsey94b17fa2009-05-07 18:46:18 +00002164 the paramater that caused failure.
jcarseya405b862010-09-14 05:18:09 +00002165 @param[in] AutoPageBreak Will automatically set PageBreakEnabled.
2166 @param[in] AlwaysAllowNumbers Will never fail for number based flags.
jcarsey94b17fa2009-05-07 18:46:18 +00002167
2168 @retval EFI_SUCCESS The operation completed sucessfully.
jcarseya405b862010-09-14 05:18:09 +00002169 @retval EFI_OUT_OF_RESOURCES A memory allocation failed.
2170 @retval EFI_INVALID_PARAMETER A parameter was invalid.
2171 @retval EFI_VOLUME_CORRUPTED The command line was corrupt.
2172 @retval EFI_DEVICE_ERROR The commands contained 2 opposing arguments. One
jcarsey1e6e84c2010-01-25 20:05:08 +00002173 of the command line arguments was returned in
jcarsey94b17fa2009-05-07 18:46:18 +00002174 ProblemParam if provided.
jcarseya405b862010-09-14 05:18:09 +00002175 @retval EFI_NOT_FOUND A argument required a value that was missing.
2176 The invalid command line argument was returned in
jcarsey94b17fa2009-05-07 18:46:18 +00002177 ProblemParam if provided.
2178**/
2179EFI_STATUS
2180EFIAPI
jcarsey2247dde2009-11-09 18:08:58 +00002181ShellCommandLineParseEx (
jcarsey94b17fa2009-05-07 18:46:18 +00002182 IN CONST SHELL_PARAM_ITEM *CheckList,
2183 OUT LIST_ENTRY **CheckPackage,
2184 OUT CHAR16 **ProblemParam OPTIONAL,
jcarsey2247dde2009-11-09 18:08:58 +00002185 IN BOOLEAN AutoPageBreak,
2186 IN BOOLEAN AlwaysAllowNumbers
jcarseya405b862010-09-14 05:18:09 +00002187 )
2188{
jcarsey1e6e84c2010-01-25 20:05:08 +00002189 //
jcarsey94b17fa2009-05-07 18:46:18 +00002190 // ASSERT that CheckList and CheckPackage aren't NULL
2191 //
2192 ASSERT(CheckList != NULL);
2193 ASSERT(CheckPackage != NULL);
2194
jcarsey1e6e84c2010-01-25 20:05:08 +00002195 //
jcarsey94b17fa2009-05-07 18:46:18 +00002196 // Check for UEFI Shell 2.0 protocols
2197 //
jcarsey366f81a2011-06-27 21:04:22 +00002198 if (gEfiShellParametersProtocol != NULL) {
jcarsey1e6e84c2010-01-25 20:05:08 +00002199 return (InternalCommandLineParse(CheckList,
2200 CheckPackage,
2201 ProblemParam,
2202 AutoPageBreak,
jcarsey366f81a2011-06-27 21:04:22 +00002203 (CONST CHAR16**) gEfiShellParametersProtocol->Argv,
2204 gEfiShellParametersProtocol->Argc,
jcarsey2247dde2009-11-09 18:08:58 +00002205 AlwaysAllowNumbers));
jcarsey94b17fa2009-05-07 18:46:18 +00002206 }
2207
jcarsey1e6e84c2010-01-25 20:05:08 +00002208 //
jcarsey94b17fa2009-05-07 18:46:18 +00002209 // ASSERT That EFI Shell is not required
2210 //
2211 ASSERT (mEfiShellInterface != NULL);
jcarsey1e6e84c2010-01-25 20:05:08 +00002212 return (InternalCommandLineParse(CheckList,
2213 CheckPackage,
2214 ProblemParam,
2215 AutoPageBreak,
jljusten08d7f8e2009-06-15 18:42:13 +00002216 (CONST CHAR16**) mEfiShellInterface->Argv,
jcarsey2247dde2009-11-09 18:08:58 +00002217 mEfiShellInterface->Argc,
2218 AlwaysAllowNumbers));
jcarsey94b17fa2009-05-07 18:46:18 +00002219}
2220
2221/**
2222 Frees shell variable list that was returned from ShellCommandLineParse.
2223
2224 This function will free all the memory that was used for the CheckPackage
2225 list of postprocessed shell arguments.
2226
2227 this function has no return value.
2228
2229 if CheckPackage is NULL, then return
2230
2231 @param CheckPackage the list to de-allocate
2232 **/
2233VOID
2234EFIAPI
2235ShellCommandLineFreeVarList (
2236 IN LIST_ENTRY *CheckPackage
jcarseya405b862010-09-14 05:18:09 +00002237 )
2238{
jcarsey94b17fa2009-05-07 18:46:18 +00002239 LIST_ENTRY *Node;
2240
2241 //
2242 // check for CheckPackage == NULL
2243 //
2244 if (CheckPackage == NULL) {
2245 return;
2246 }
2247
2248 //
2249 // for each node in the list
2250 //
jcarsey9eb53ac2009-07-08 17:26:58 +00002251 for ( Node = GetFirstNode(CheckPackage)
jcarseya405b862010-09-14 05:18:09 +00002252 ; !IsListEmpty(CheckPackage)
jcarsey9eb53ac2009-07-08 17:26:58 +00002253 ; Node = GetFirstNode(CheckPackage)
jcarseya405b862010-09-14 05:18:09 +00002254 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002255 //
2256 // Remove it from the list
2257 //
2258 RemoveEntryList(Node);
2259
2260 //
2261 // if it has a name free the name
2262 //
2263 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
2264 FreePool(((SHELL_PARAM_PACKAGE*)Node)->Name);
2265 }
2266
2267 //
2268 // if it has a value free the value
2269 //
2270 if (((SHELL_PARAM_PACKAGE*)Node)->Value != NULL) {
2271 FreePool(((SHELL_PARAM_PACKAGE*)Node)->Value);
2272 }
jcarsey1e6e84c2010-01-25 20:05:08 +00002273
jcarsey94b17fa2009-05-07 18:46:18 +00002274 //
2275 // free the node structure
2276 //
2277 FreePool((SHELL_PARAM_PACKAGE*)Node);
2278 }
2279 //
2280 // free the list head node
2281 //
2282 FreePool(CheckPackage);
2283}
2284/**
2285 Checks for presence of a flag parameter
2286
2287 flag arguments are in the form of "-<Key>" or "/<Key>", but do not have a value following the key
2288
2289 if CheckPackage is NULL then return FALSE.
2290 if KeyString is NULL then ASSERT()
jcarsey1e6e84c2010-01-25 20:05:08 +00002291
jcarsey94b17fa2009-05-07 18:46:18 +00002292 @param CheckPackage The package of parsed command line arguments
2293 @param KeyString the Key of the command line argument to check for
2294
2295 @retval TRUE the flag is on the command line
2296 @retval FALSE the flag is not on the command line
2297 **/
2298BOOLEAN
2299EFIAPI
2300ShellCommandLineGetFlag (
jcarseya405b862010-09-14 05:18:09 +00002301 IN CONST LIST_ENTRY * CONST CheckPackage,
2302 IN CONST CHAR16 * CONST KeyString
2303 )
2304{
jcarsey94b17fa2009-05-07 18:46:18 +00002305 LIST_ENTRY *Node;
jcarsey252d9452011-03-25 20:49:53 +00002306 CHAR16 *TempString;
jcarsey94b17fa2009-05-07 18:46:18 +00002307
2308 //
ydong100c1950b2011-10-13 02:37:35 +00002309 // return FALSE for no package or KeyString is NULL
jcarsey94b17fa2009-05-07 18:46:18 +00002310 //
ydong100c1950b2011-10-13 02:37:35 +00002311 if (CheckPackage == NULL || KeyString == NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00002312 return (FALSE);
2313 }
2314
2315 //
2316 // enumerate through the list of parametrs
2317 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002318 for ( Node = GetFirstNode(CheckPackage)
2319 ; !IsNull (CheckPackage, Node)
2320 ; Node = GetNextNode(CheckPackage, Node)
jcarsey252d9452011-03-25 20:49:53 +00002321 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002322 //
2323 // If the Name matches, return TRUE (and there may be NULL name)
2324 //
2325 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
jcarsey9eb53ac2009-07-08 17:26:58 +00002326 //
2327 // If Type is TypeStart then only compare the begining of the strings
2328 //
jcarsey252d9452011-03-25 20:49:53 +00002329 if (((SHELL_PARAM_PACKAGE*)Node)->Type == TypeStart) {
2330 if (StrnCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name, StrLen(KeyString)) == 0) {
2331 return (TRUE);
2332 }
2333 TempString = NULL;
2334 TempString = StrnCatGrow(&TempString, NULL, KeyString, StrLen(((SHELL_PARAM_PACKAGE*)Node)->Name));
2335 if (TempString != NULL) {
2336 if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2337 FreePool(TempString);
2338 return (TRUE);
2339 }
2340 FreePool(TempString);
2341 }
2342 } else if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
jcarsey94b17fa2009-05-07 18:46:18 +00002343 return (TRUE);
2344 }
2345 }
2346 }
2347 return (FALSE);
2348}
2349/**
jcarseya405b862010-09-14 05:18:09 +00002350 Returns value from command line argument.
jcarsey94b17fa2009-05-07 18:46:18 +00002351
jcarseya405b862010-09-14 05:18:09 +00002352 Value parameters are in the form of "-<Key> value" or "/<Key> value".
jcarsey1e6e84c2010-01-25 20:05:08 +00002353
jcarseya405b862010-09-14 05:18:09 +00002354 If CheckPackage is NULL, then return NULL.
jcarsey94b17fa2009-05-07 18:46:18 +00002355
jcarseya405b862010-09-14 05:18:09 +00002356 @param[in] CheckPackage The package of parsed command line arguments.
2357 @param[in] KeyString The Key of the command line argument to check for.
jcarsey94b17fa2009-05-07 18:46:18 +00002358
jcarseya405b862010-09-14 05:18:09 +00002359 @retval NULL The flag is not on the command line.
2360 @retval !=NULL The pointer to unicode string of the value.
2361**/
jcarsey94b17fa2009-05-07 18:46:18 +00002362CONST CHAR16*
2363EFIAPI
2364ShellCommandLineGetValue (
2365 IN CONST LIST_ENTRY *CheckPackage,
2366 IN CHAR16 *KeyString
jcarseya405b862010-09-14 05:18:09 +00002367 )
2368{
jcarsey94b17fa2009-05-07 18:46:18 +00002369 LIST_ENTRY *Node;
jcarsey252d9452011-03-25 20:49:53 +00002370 CHAR16 *TempString;
jcarsey94b17fa2009-05-07 18:46:18 +00002371
2372 //
ydong100c1950b2011-10-13 02:37:35 +00002373 // return NULL for no package or KeyString is NULL
jcarsey94b17fa2009-05-07 18:46:18 +00002374 //
ydong100c1950b2011-10-13 02:37:35 +00002375 if (CheckPackage == NULL || KeyString == NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00002376 return (NULL);
2377 }
2378
2379 //
2380 // enumerate through the list of parametrs
2381 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002382 for ( Node = GetFirstNode(CheckPackage)
2383 ; !IsNull (CheckPackage, Node)
2384 ; Node = GetNextNode(CheckPackage, Node)
jcarsey252d9452011-03-25 20:49:53 +00002385 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002386 //
jcarsey252d9452011-03-25 20:49:53 +00002387 // If the Name matches, return TRUE (and there may be NULL name)
jcarsey94b17fa2009-05-07 18:46:18 +00002388 //
2389 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
jcarsey9eb53ac2009-07-08 17:26:58 +00002390 //
2391 // If Type is TypeStart then only compare the begining of the strings
2392 //
jcarsey252d9452011-03-25 20:49:53 +00002393 if (((SHELL_PARAM_PACKAGE*)Node)->Type == TypeStart) {
2394 if (StrnCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name, StrLen(KeyString)) == 0) {
2395 return (((SHELL_PARAM_PACKAGE*)Node)->Name + StrLen(KeyString));
2396 }
2397 TempString = NULL;
2398 TempString = StrnCatGrow(&TempString, NULL, KeyString, StrLen(((SHELL_PARAM_PACKAGE*)Node)->Name));
2399 if (TempString != NULL) {
2400 if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2401 FreePool(TempString);
2402 return (((SHELL_PARAM_PACKAGE*)Node)->Name + StrLen(KeyString));
2403 }
2404 FreePool(TempString);
2405 }
2406 } else if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
jcarsey94b17fa2009-05-07 18:46:18 +00002407 return (((SHELL_PARAM_PACKAGE*)Node)->Value);
2408 }
2409 }
2410 }
2411 return (NULL);
2412}
jcarseya405b862010-09-14 05:18:09 +00002413
jcarsey94b17fa2009-05-07 18:46:18 +00002414/**
jcarseya405b862010-09-14 05:18:09 +00002415 Returns raw value from command line argument.
jcarsey94b17fa2009-05-07 18:46:18 +00002416
jcarseya405b862010-09-14 05:18:09 +00002417 Raw value parameters are in the form of "value" in a specific position in the list.
jcarsey1e6e84c2010-01-25 20:05:08 +00002418
jcarseya405b862010-09-14 05:18:09 +00002419 If CheckPackage is NULL, then return NULL.
jcarsey94b17fa2009-05-07 18:46:18 +00002420
jcarseya405b862010-09-14 05:18:09 +00002421 @param[in] CheckPackage The package of parsed command line arguments.
2422 @param[in] Position The position of the value.
jcarsey94b17fa2009-05-07 18:46:18 +00002423
jcarseya405b862010-09-14 05:18:09 +00002424 @retval NULL The flag is not on the command line.
2425 @retval !=NULL The pointer to unicode string of the value.
jcarsey94b17fa2009-05-07 18:46:18 +00002426 **/
2427CONST CHAR16*
2428EFIAPI
2429ShellCommandLineGetRawValue (
jcarseya405b862010-09-14 05:18:09 +00002430 IN CONST LIST_ENTRY * CONST CheckPackage,
2431 IN UINTN Position
2432 )
2433{
jcarsey94b17fa2009-05-07 18:46:18 +00002434 LIST_ENTRY *Node;
2435
2436 //
2437 // check for CheckPackage == NULL
2438 //
2439 if (CheckPackage == NULL) {
2440 return (NULL);
2441 }
2442
2443 //
2444 // enumerate through the list of parametrs
2445 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002446 for ( Node = GetFirstNode(CheckPackage)
2447 ; !IsNull (CheckPackage, Node)
2448 ; Node = GetNextNode(CheckPackage, Node)
jcarseya405b862010-09-14 05:18:09 +00002449 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002450 //
2451 // If the position matches, return the value
2452 //
2453 if (((SHELL_PARAM_PACKAGE*)Node)->OriginalPosition == Position) {
2454 return (((SHELL_PARAM_PACKAGE*)Node)->Value);
2455 }
2456 }
2457 return (NULL);
jcarseyb1f95a02009-06-16 00:23:19 +00002458}
jcarsey2247dde2009-11-09 18:08:58 +00002459
2460/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002461 returns the number of command line value parameters that were parsed.
2462
jcarsey2247dde2009-11-09 18:08:58 +00002463 this will not include flags.
2464
jcarseya405b862010-09-14 05:18:09 +00002465 @param[in] CheckPackage The package of parsed command line arguments.
2466
jcarsey2247dde2009-11-09 18:08:58 +00002467 @retval (UINTN)-1 No parsing has ocurred
2468 @return other The number of value parameters found
2469**/
2470UINTN
2471EFIAPI
2472ShellCommandLineGetCount(
jcarseya405b862010-09-14 05:18:09 +00002473 IN CONST LIST_ENTRY *CheckPackage
jcarsey125c2cf2009-11-18 21:36:50 +00002474 )
2475{
jcarseya405b862010-09-14 05:18:09 +00002476 LIST_ENTRY *Node1;
2477 UINTN Count;
2478
2479 if (CheckPackage == NULL) {
2480 return (0);
2481 }
2482 for ( Node1 = GetFirstNode(CheckPackage), Count = 0
2483 ; !IsNull (CheckPackage, Node1)
2484 ; Node1 = GetNextNode(CheckPackage, Node1)
2485 ){
2486 if (((SHELL_PARAM_PACKAGE*)Node1)->Name == NULL) {
2487 Count++;
2488 }
2489 }
2490 return (Count);
jcarsey2247dde2009-11-09 18:08:58 +00002491}
2492
jcarsey975136a2009-06-16 19:03:54 +00002493/**
jcarsey36a9d672009-11-20 21:13:41 +00002494 Determins if a parameter is duplicated.
2495
jcarsey1e6e84c2010-01-25 20:05:08 +00002496 If Param is not NULL then it will point to a callee allocated string buffer
jcarsey36a9d672009-11-20 21:13:41 +00002497 with the parameter value if a duplicate is found.
2498
2499 If CheckPackage is NULL, then ASSERT.
2500
2501 @param[in] CheckPackage The package of parsed command line arguments.
2502 @param[out] Param Upon finding one, a pointer to the duplicated parameter.
2503
2504 @retval EFI_SUCCESS No parameters were duplicated.
2505 @retval EFI_DEVICE_ERROR A duplicate was found.
2506 **/
2507EFI_STATUS
2508EFIAPI
2509ShellCommandLineCheckDuplicate (
2510 IN CONST LIST_ENTRY *CheckPackage,
2511 OUT CHAR16 **Param
2512 )
2513{
2514 LIST_ENTRY *Node1;
2515 LIST_ENTRY *Node2;
jcarsey1e6e84c2010-01-25 20:05:08 +00002516
jcarsey36a9d672009-11-20 21:13:41 +00002517 ASSERT(CheckPackage != NULL);
2518
jcarsey1e6e84c2010-01-25 20:05:08 +00002519 for ( Node1 = GetFirstNode(CheckPackage)
2520 ; !IsNull (CheckPackage, Node1)
2521 ; Node1 = GetNextNode(CheckPackage, Node1)
jcarseya405b862010-09-14 05:18:09 +00002522 ){
jcarsey1e6e84c2010-01-25 20:05:08 +00002523 for ( Node2 = GetNextNode(CheckPackage, Node1)
2524 ; !IsNull (CheckPackage, Node2)
2525 ; Node2 = GetNextNode(CheckPackage, Node2)
jcarseya405b862010-09-14 05:18:09 +00002526 ){
2527 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 +00002528 if (Param != NULL) {
2529 *Param = NULL;
2530 *Param = StrnCatGrow(Param, NULL, ((SHELL_PARAM_PACKAGE*)Node1)->Name, 0);
2531 }
2532 return (EFI_DEVICE_ERROR);
2533 }
2534 }
2535 }
2536 return (EFI_SUCCESS);
2537}
2538
2539/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002540 This is a find and replace function. Upon successful return the NewString is a copy of
jcarsey975136a2009-06-16 19:03:54 +00002541 SourceString with each instance of FindTarget replaced with ReplaceWith.
2542
jcarseyb3011f42010-01-11 21:49:04 +00002543 If SourceString and NewString overlap the behavior is undefined.
2544
jcarsey975136a2009-06-16 19:03:54 +00002545 If the string would grow bigger than NewSize it will halt and return error.
2546
ydong104ff7e372011-09-02 08:05:34 +00002547 @param[in] SourceString The string with source buffer.
2548 @param[in, out] NewString The string with resultant buffer.
2549 @param[in] NewSize The size in bytes of NewString.
2550 @param[in] FindTarget The string to look for.
2551 @param[in] ReplaceWith The string to replace FindTarget with.
2552 @param[in] SkipPreCarrot If TRUE will skip a FindTarget that has a '^'
2553 immediately before it.
2554 @param[in] ParameterReplacing If TRUE will add "" around items with spaces.
jcarsey975136a2009-06-16 19:03:54 +00002555
jcarsey969c7832010-01-13 16:46:33 +00002556 @retval EFI_INVALID_PARAMETER SourceString was NULL.
2557 @retval EFI_INVALID_PARAMETER NewString was NULL.
2558 @retval EFI_INVALID_PARAMETER FindTarget was NULL.
2559 @retval EFI_INVALID_PARAMETER ReplaceWith was NULL.
2560 @retval EFI_INVALID_PARAMETER FindTarget had length < 1.
2561 @retval EFI_INVALID_PARAMETER SourceString had length < 1.
jcarsey1e6e84c2010-01-25 20:05:08 +00002562 @retval EFI_BUFFER_TOO_SMALL NewSize was less than the minimum size to hold
jcarsey969c7832010-01-13 16:46:33 +00002563 the new string (truncation occurred).
jcarseya405b862010-09-14 05:18:09 +00002564 @retval EFI_SUCCESS The string was successfully copied with replacement.
jcarsey975136a2009-06-16 19:03:54 +00002565**/
jcarsey975136a2009-06-16 19:03:54 +00002566EFI_STATUS
2567EFIAPI
jcarseya405b862010-09-14 05:18:09 +00002568ShellCopySearchAndReplace(
jcarsey975136a2009-06-16 19:03:54 +00002569 IN CHAR16 CONST *SourceString,
jcarseya405b862010-09-14 05:18:09 +00002570 IN OUT CHAR16 *NewString,
jcarsey975136a2009-06-16 19:03:54 +00002571 IN UINTN NewSize,
2572 IN CONST CHAR16 *FindTarget,
jcarsey969c7832010-01-13 16:46:33 +00002573 IN CONST CHAR16 *ReplaceWith,
jcarseya405b862010-09-14 05:18:09 +00002574 IN CONST BOOLEAN SkipPreCarrot,
2575 IN CONST BOOLEAN ParameterReplacing
jcarsey1e6e84c2010-01-25 20:05:08 +00002576 )
jcarsey2247dde2009-11-09 18:08:58 +00002577{
jcarsey01582942009-07-10 19:46:17 +00002578 UINTN Size;
jcarseya405b862010-09-14 05:18:09 +00002579 CHAR16 *Replace;
2580
jcarsey975136a2009-06-16 19:03:54 +00002581 if ( (SourceString == NULL)
2582 || (NewString == NULL)
2583 || (FindTarget == NULL)
2584 || (ReplaceWith == NULL)
2585 || (StrLen(FindTarget) < 1)
2586 || (StrLen(SourceString) < 1)
jcarseya405b862010-09-14 05:18:09 +00002587 ){
jcarsey975136a2009-06-16 19:03:54 +00002588 return (EFI_INVALID_PARAMETER);
2589 }
jcarseya405b862010-09-14 05:18:09 +00002590 Replace = NULL;
2591 if (StrStr(ReplaceWith, L" ") == NULL || !ParameterReplacing) {
2592 Replace = StrnCatGrow(&Replace, NULL, ReplaceWith, 0);
2593 } else {
2594 Replace = AllocateZeroPool(StrSize(ReplaceWith) + 2*sizeof(CHAR16));
jcarseybeab0fc2011-10-10 17:26:25 +00002595 if (Replace != NULL) {
2596 UnicodeSPrint(Replace, StrSize(ReplaceWith) + 2*sizeof(CHAR16), L"\"%s\"", ReplaceWith);
2597 }
jcarseya405b862010-09-14 05:18:09 +00002598 }
jcarsey3e082d52010-10-04 16:44:57 +00002599 if (Replace == NULL) {
2600 return (EFI_OUT_OF_RESOURCES);
2601 }
Jaben Carsey98c16be2014-08-19 21:00:34 +00002602 NewString = ZeroMem(NewString, NewSize);
jcarsey2247dde2009-11-09 18:08:58 +00002603 while (*SourceString != CHAR_NULL) {
jcarsey969c7832010-01-13 16:46:33 +00002604 //
jcarseya405b862010-09-14 05:18:09 +00002605 // if we find the FindTarget and either Skip == FALSE or Skip and we
jcarsey969c7832010-01-13 16:46:33 +00002606 // dont have a carrot do a replace...
2607 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002608 if (StrnCmp(SourceString, FindTarget, StrLen(FindTarget)) == 0
jcarseya405b862010-09-14 05:18:09 +00002609 && ((SkipPreCarrot && *(SourceString-1) != L'^') || !SkipPreCarrot)
2610 ){
jcarsey975136a2009-06-16 19:03:54 +00002611 SourceString += StrLen(FindTarget);
jcarsey01582942009-07-10 19:46:17 +00002612 Size = StrSize(NewString);
jcarseya405b862010-09-14 05:18:09 +00002613 if ((Size + (StrLen(Replace)*sizeof(CHAR16))) > NewSize) {
2614 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002615 return (EFI_BUFFER_TOO_SMALL);
2616 }
Jaben Carsey98c16be2014-08-19 21:00:34 +00002617 StrnCat(NewString, Replace, NewSize/sizeof(CHAR16) - 1 - StrLen(NewString));
jcarsey975136a2009-06-16 19:03:54 +00002618 } else {
jcarsey01582942009-07-10 19:46:17 +00002619 Size = StrSize(NewString);
2620 if (Size + sizeof(CHAR16) > NewSize) {
jcarseya405b862010-09-14 05:18:09 +00002621 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002622 return (EFI_BUFFER_TOO_SMALL);
2623 }
2624 StrnCat(NewString, SourceString, 1);
2625 SourceString++;
2626 }
2627 }
jcarseya405b862010-09-14 05:18:09 +00002628 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002629 return (EFI_SUCCESS);
2630}
jcarseyb1f95a02009-06-16 00:23:19 +00002631
2632/**
jcarseye2f82972009-12-01 05:40:24 +00002633 Internal worker function to output a string.
2634
2635 This function will output a string to the correct StdOut.
2636
2637 @param[in] String The string to print out.
2638
2639 @retval EFI_SUCCESS The operation was sucessful.
2640 @retval !EFI_SUCCESS The operation failed.
2641**/
2642EFI_STATUS
2643EFIAPI
2644InternalPrintTo (
2645 IN CONST CHAR16 *String
2646 )
2647{
2648 UINTN Size;
2649 Size = StrSize(String) - sizeof(CHAR16);
jcarseya405b862010-09-14 05:18:09 +00002650 if (Size == 0) {
2651 return (EFI_SUCCESS);
2652 }
jcarsey366f81a2011-06-27 21:04:22 +00002653 if (gEfiShellParametersProtocol != NULL) {
2654 return (gEfiShellProtocol->WriteFile(gEfiShellParametersProtocol->StdOut, &Size, (VOID*)String));
jcarseye2f82972009-12-01 05:40:24 +00002655 }
2656 if (mEfiShellInterface != NULL) {
jcarsey06c355b2012-03-26 21:00:39 +00002657 if (mEfiShellInterface->RedirArgc == 0) {
jcarsey49bd4982012-01-27 18:42:43 +00002658 //
2659 // Divide in half for old shell. Must be string length not size.
jcarsey06c355b2012-03-26 21:00:39 +00002660 //
2661 Size /=2; // Divide in half only when no redirection.
2662 }
jcarseya405b862010-09-14 05:18:09 +00002663 return (mEfiShellInterface->StdOut->Write(mEfiShellInterface->StdOut, &Size, (VOID*)String));
jcarseye2f82972009-12-01 05:40:24 +00002664 }
2665 ASSERT(FALSE);
2666 return (EFI_UNSUPPORTED);
2667}
2668
2669/**
jcarseyb1f95a02009-06-16 00:23:19 +00002670 Print at a specific location on the screen.
2671
jcarseyf1b87e72009-06-17 00:52:11 +00002672 This function will move the cursor to a given screen location and print the specified string
jcarsey1e6e84c2010-01-25 20:05:08 +00002673
2674 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarseyf1b87e72009-06-17 00:52:11 +00002675 will be used.
jcarseyb1f95a02009-06-16 00:23:19 +00002676
2677 if either Row or Col is out of range for the current console, then ASSERT
2678 if Format is NULL, then ASSERT
2679
jcarsey1e6e84c2010-01-25 20:05:08 +00002680 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarseyb1f95a02009-06-16 00:23:19 +00002681 the following additional flags:
2682 %N - Set output attribute to normal
2683 %H - Set output attribute to highlight
2684 %E - Set output attribute to error
2685 %B - Set output attribute to blue color
2686 %V - Set output attribute to green color
2687
2688 Note: The background color is controlled by the shell command cls.
2689
jcarseyb1f95a02009-06-16 00:23:19 +00002690 @param[in] Col the column to print at
jcarsey252d9452011-03-25 20:49:53 +00002691 @param[in] Row the row to print at
jcarseyb1f95a02009-06-16 00:23:19 +00002692 @param[in] Format the format string
jcarsey2247dde2009-11-09 18:08:58 +00002693 @param[in] Marker the marker for the variable argument list
jcarseyb1f95a02009-06-16 00:23:19 +00002694
jcarseya405b862010-09-14 05:18:09 +00002695 @return EFI_SUCCESS The operation was successful.
2696 @return EFI_DEVICE_ERROR The console device reported an error.
jcarseyb1f95a02009-06-16 00:23:19 +00002697**/
jcarseya405b862010-09-14 05:18:09 +00002698EFI_STATUS
jcarseyb1f95a02009-06-16 00:23:19 +00002699EFIAPI
jcarsey2247dde2009-11-09 18:08:58 +00002700InternalShellPrintWorker(
jcarseyb1f95a02009-06-16 00:23:19 +00002701 IN INT32 Col OPTIONAL,
2702 IN INT32 Row OPTIONAL,
2703 IN CONST CHAR16 *Format,
jcarsey252d9452011-03-25 20:49:53 +00002704 IN VA_LIST Marker
jcarsey1e6e84c2010-01-25 20:05:08 +00002705 )
jcarsey2247dde2009-11-09 18:08:58 +00002706{
jcarseyb1f95a02009-06-16 00:23:19 +00002707 EFI_STATUS Status;
jcarsey975136a2009-06-16 19:03:54 +00002708 CHAR16 *ResumeLocation;
2709 CHAR16 *FormatWalker;
jcarseya405b862010-09-14 05:18:09 +00002710 UINTN OriginalAttribute;
jcarsey89e85372011-04-13 23:37:50 +00002711 CHAR16 *mPostReplaceFormat;
2712 CHAR16 *mPostReplaceFormat2;
2713
2714 mPostReplaceFormat = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
2715 mPostReplaceFormat2 = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
jcarseya405b862010-09-14 05:18:09 +00002716
jcarseyf8d3e682011-04-19 17:54:42 +00002717 if (mPostReplaceFormat == NULL || mPostReplaceFormat2 == NULL) {
2718 SHELL_FREE_NON_NULL(mPostReplaceFormat);
2719 SHELL_FREE_NON_NULL(mPostReplaceFormat2);
2720 return (EFI_OUT_OF_RESOURCES);
2721 }
2722
jcarseya405b862010-09-14 05:18:09 +00002723 Status = EFI_SUCCESS;
2724 OriginalAttribute = gST->ConOut->Mode->Attribute;
jcarsey1e6e84c2010-01-25 20:05:08 +00002725
jcarsey975136a2009-06-16 19:03:54 +00002726 //
2727 // Back and forth each time fixing up 1 of our flags...
2728 //
jcarseya405b862010-09-14 05:18:09 +00002729 Status = ShellCopySearchAndReplace(Format, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%N", L"%%N", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002730 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002731 Status = ShellCopySearchAndReplace(mPostReplaceFormat, mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), L"%E", L"%%E", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002732 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002733 Status = ShellCopySearchAndReplace(mPostReplaceFormat2, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%H", L"%%H", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002734 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002735 Status = ShellCopySearchAndReplace(mPostReplaceFormat, mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), L"%B", L"%%B", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002736 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002737 Status = ShellCopySearchAndReplace(mPostReplaceFormat2, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%V", L"%%V", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002738 ASSERT_EFI_ERROR(Status);
2739
2740 //
2741 // Use the last buffer from replacing to print from...
2742 //
jcarseya405b862010-09-14 05:18:09 +00002743 UnicodeVSPrint (mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), mPostReplaceFormat, Marker);
jcarseyb1f95a02009-06-16 00:23:19 +00002744
2745 if (Col != -1 && Row != -1) {
jcarseyb1f95a02009-06-16 00:23:19 +00002746 Status = gST->ConOut->SetCursorPosition(gST->ConOut, Col, Row);
jcarsey975136a2009-06-16 19:03:54 +00002747 }
2748
jcarseyecd3d592009-12-07 18:05:00 +00002749 FormatWalker = mPostReplaceFormat2;
jcarsey2247dde2009-11-09 18:08:58 +00002750 while (*FormatWalker != CHAR_NULL) {
jcarsey975136a2009-06-16 19:03:54 +00002751 //
2752 // Find the next attribute change request
2753 //
2754 ResumeLocation = StrStr(FormatWalker, L"%");
2755 if (ResumeLocation != NULL) {
jcarsey2247dde2009-11-09 18:08:58 +00002756 *ResumeLocation = CHAR_NULL;
jcarsey975136a2009-06-16 19:03:54 +00002757 }
2758 //
2759 // print the current FormatWalker string
2760 //
jcarseya405b862010-09-14 05:18:09 +00002761 if (StrLen(FormatWalker)>0) {
2762 Status = InternalPrintTo(FormatWalker);
2763 if (EFI_ERROR(Status)) {
2764 break;
2765 }
2766 }
2767
jcarsey975136a2009-06-16 19:03:54 +00002768 //
2769 // update the attribute
2770 //
2771 if (ResumeLocation != NULL) {
jcarsey5d46f172011-08-23 15:34:23 +00002772 if (*(ResumeLocation-1) == L'^') {
2773 //
jcarsey8bb74412012-01-30 18:44:41 +00002774 // Move cursor back 1 position to overwrite the ^
2775 //
2776 gST->ConOut->SetCursorPosition(gST->ConOut, gST->ConOut->Mode->CursorColumn - 1, gST->ConOut->Mode->CursorRow);
2777
2778 //
jcarsey5d46f172011-08-23 15:34:23 +00002779 // Print a simple '%' symbol
2780 //
2781 Status = InternalPrintTo(L"%");
2782 ResumeLocation = ResumeLocation - 1;
2783 } else {
2784 switch (*(ResumeLocation+1)) {
2785 case (L'N'):
2786 gST->ConOut->SetAttribute(gST->ConOut, OriginalAttribute);
jcarseya405b862010-09-14 05:18:09 +00002787 break;
jcarsey5d46f172011-08-23 15:34:23 +00002788 case (L'E'):
2789 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_YELLOW, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2790 break;
2791 case (L'H'):
2792 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_WHITE, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2793 break;
2794 case (L'B'):
2795 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_BLUE, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2796 break;
2797 case (L'V'):
2798 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_GREEN, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2799 break;
2800 default:
2801 //
2802 // Print a simple '%' symbol
2803 //
2804 Status = InternalPrintTo(L"%");
2805 if (EFI_ERROR(Status)) {
2806 break;
2807 }
2808 ResumeLocation = ResumeLocation - 1;
2809 break;
2810 }
jcarsey975136a2009-06-16 19:03:54 +00002811 }
2812 } else {
2813 //
2814 // reset to normal now...
2815 //
jcarsey975136a2009-06-16 19:03:54 +00002816 break;
2817 }
2818
2819 //
2820 // update FormatWalker to Resume + 2 (skip the % and the indicator)
2821 //
2822 FormatWalker = ResumeLocation + 2;
2823 }
jcarseyb1f95a02009-06-16 00:23:19 +00002824
jcarseya405b862010-09-14 05:18:09 +00002825 gST->ConOut->SetAttribute(gST->ConOut, OriginalAttribute);
jcarsey89e85372011-04-13 23:37:50 +00002826
2827 SHELL_FREE_NON_NULL(mPostReplaceFormat);
2828 SHELL_FREE_NON_NULL(mPostReplaceFormat2);
jcarseya405b862010-09-14 05:18:09 +00002829 return (Status);
jcarsey5f7431d2009-07-10 18:06:01 +00002830}
jcarsey2247dde2009-11-09 18:08:58 +00002831
2832/**
2833 Print at a specific location on the screen.
2834
jcarseye2f82972009-12-01 05:40:24 +00002835 This function will move the cursor to a given screen location and print the specified string.
jcarsey1e6e84c2010-01-25 20:05:08 +00002836
2837 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarsey2247dde2009-11-09 18:08:58 +00002838 will be used.
2839
jcarseye2f82972009-12-01 05:40:24 +00002840 If either Row or Col is out of range for the current console, then ASSERT.
2841 If Format is NULL, then ASSERT.
jcarsey2247dde2009-11-09 18:08:58 +00002842
jcarsey1e6e84c2010-01-25 20:05:08 +00002843 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarsey2247dde2009-11-09 18:08:58 +00002844 the following additional flags:
2845 %N - Set output attribute to normal
2846 %H - Set output attribute to highlight
2847 %E - Set output attribute to error
2848 %B - Set output attribute to blue color
2849 %V - Set output attribute to green color
2850
2851 Note: The background color is controlled by the shell command cls.
2852
jcarsey2247dde2009-11-09 18:08:58 +00002853 @param[in] Col the column to print at
jcarseya405b862010-09-14 05:18:09 +00002854 @param[in] Row the row to print at
jcarsey2247dde2009-11-09 18:08:58 +00002855 @param[in] Format the format string
jcarseya405b862010-09-14 05:18:09 +00002856 @param[in] ... The variable argument list.
jcarsey2247dde2009-11-09 18:08:58 +00002857
jcarseya405b862010-09-14 05:18:09 +00002858 @return EFI_SUCCESS The printing was successful.
2859 @return EFI_DEVICE_ERROR The console device reported an error.
jcarsey2247dde2009-11-09 18:08:58 +00002860**/
jcarseya405b862010-09-14 05:18:09 +00002861EFI_STATUS
jcarsey2247dde2009-11-09 18:08:58 +00002862EFIAPI
2863ShellPrintEx(
2864 IN INT32 Col OPTIONAL,
2865 IN INT32 Row OPTIONAL,
2866 IN CONST CHAR16 *Format,
2867 ...
jcarsey1e6e84c2010-01-25 20:05:08 +00002868 )
jcarsey2247dde2009-11-09 18:08:58 +00002869{
2870 VA_LIST Marker;
jcarseya405b862010-09-14 05:18:09 +00002871 EFI_STATUS RetVal;
jcarsey3e082d52010-10-04 16:44:57 +00002872 if (Format == NULL) {
2873 return (EFI_INVALID_PARAMETER);
2874 }
jcarsey2247dde2009-11-09 18:08:58 +00002875 VA_START (Marker, Format);
jcarseya405b862010-09-14 05:18:09 +00002876 RetVal = InternalShellPrintWorker(Col, Row, Format, Marker);
jcarseye2f82972009-12-01 05:40:24 +00002877 VA_END(Marker);
jcarseya405b862010-09-14 05:18:09 +00002878 return(RetVal);
jcarsey2247dde2009-11-09 18:08:58 +00002879}
2880
2881/**
2882 Print at a specific location on the screen.
2883
jcarseye2f82972009-12-01 05:40:24 +00002884 This function will move the cursor to a given screen location and print the specified string.
jcarsey1e6e84c2010-01-25 20:05:08 +00002885
2886 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarseye2f82972009-12-01 05:40:24 +00002887 will be used.
jcarsey2247dde2009-11-09 18:08:58 +00002888
jcarseye2f82972009-12-01 05:40:24 +00002889 If either Row or Col is out of range for the current console, then ASSERT.
2890 If Format is NULL, then ASSERT.
jcarsey2247dde2009-11-09 18:08:58 +00002891
jcarsey1e6e84c2010-01-25 20:05:08 +00002892 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarsey2247dde2009-11-09 18:08:58 +00002893 the following additional flags:
jcarsey1e6e84c2010-01-25 20:05:08 +00002894 %N - Set output attribute to normal.
2895 %H - Set output attribute to highlight.
2896 %E - Set output attribute to error.
2897 %B - Set output attribute to blue color.
2898 %V - Set output attribute to green color.
jcarsey2247dde2009-11-09 18:08:58 +00002899
2900 Note: The background color is controlled by the shell command cls.
2901
jcarsey1e6e84c2010-01-25 20:05:08 +00002902 @param[in] Col The column to print at.
jcarseya405b862010-09-14 05:18:09 +00002903 @param[in] Row The row to print at.
jcarsey1e6e84c2010-01-25 20:05:08 +00002904 @param[in] Language The language of the string to retrieve. If this parameter
2905 is NULL, then the current platform language is used.
2906 @param[in] HiiFormatStringId The format string Id for getting from Hii.
2907 @param[in] HiiFormatHandle The format string Handle for getting from Hii.
jcarseya405b862010-09-14 05:18:09 +00002908 @param[in] ... The variable argument list.
jcarsey2247dde2009-11-09 18:08:58 +00002909
jcarseya405b862010-09-14 05:18:09 +00002910 @return EFI_SUCCESS The printing was successful.
2911 @return EFI_DEVICE_ERROR The console device reported an error.
jcarsey2247dde2009-11-09 18:08:58 +00002912**/
jcarseya405b862010-09-14 05:18:09 +00002913EFI_STATUS
jcarsey2247dde2009-11-09 18:08:58 +00002914EFIAPI
2915ShellPrintHiiEx(
2916 IN INT32 Col OPTIONAL,
2917 IN INT32 Row OPTIONAL,
jcarsey1e6e84c2010-01-25 20:05:08 +00002918 IN CONST CHAR8 *Language OPTIONAL,
jcarsey2247dde2009-11-09 18:08:58 +00002919 IN CONST EFI_STRING_ID HiiFormatStringId,
2920 IN CONST EFI_HANDLE HiiFormatHandle,
2921 ...
2922 )
2923{
2924 VA_LIST Marker;
2925 CHAR16 *HiiFormatString;
jcarseya405b862010-09-14 05:18:09 +00002926 EFI_STATUS RetVal;
jcarsey2247dde2009-11-09 18:08:58 +00002927
2928 VA_START (Marker, HiiFormatHandle);
jcarsey1e6e84c2010-01-25 20:05:08 +00002929 HiiFormatString = HiiGetString(HiiFormatHandle, HiiFormatStringId, Language);
jcarsey2247dde2009-11-09 18:08:58 +00002930 ASSERT(HiiFormatString != NULL);
2931
2932 RetVal = InternalShellPrintWorker(Col, Row, HiiFormatString, Marker);
2933
jcarseya405b862010-09-14 05:18:09 +00002934 SHELL_FREE_NON_NULL(HiiFormatString);
jcarseye2f82972009-12-01 05:40:24 +00002935 VA_END(Marker);
jcarsey2247dde2009-11-09 18:08:58 +00002936
2937 return (RetVal);
2938}
2939
2940/**
2941 Function to determine if a given filename represents a file or a directory.
2942
2943 @param[in] DirName Path to directory to test.
2944
jcarseyc8c22592011-10-17 17:49:21 +00002945 @retval EFI_SUCCESS The Path represents a directory
2946 @retval EFI_NOT_FOUND The Path does not represent a directory
2947 @retval EFI_OUT_OF_RESOURCES A memory allocation failed.
2948 @return The path failed to open
jcarsey2247dde2009-11-09 18:08:58 +00002949**/
2950EFI_STATUS
2951EFIAPI
2952ShellIsDirectory(
2953 IN CONST CHAR16 *DirName
2954 )
2955{
2956 EFI_STATUS Status;
jcarseya405b862010-09-14 05:18:09 +00002957 SHELL_FILE_HANDLE Handle;
jcarsey3e082d52010-10-04 16:44:57 +00002958 CHAR16 *TempLocation;
2959 CHAR16 *TempLocation2;
jcarsey2247dde2009-11-09 18:08:58 +00002960
jcarseyecd3d592009-12-07 18:05:00 +00002961 ASSERT(DirName != NULL);
2962
jcarseya405b862010-09-14 05:18:09 +00002963 Handle = NULL;
2964 TempLocation = NULL;
jcarsey2247dde2009-11-09 18:08:58 +00002965
2966 Status = ShellOpenFileByName(DirName, &Handle, EFI_FILE_MODE_READ, 0);
2967 if (EFI_ERROR(Status)) {
jcarseya405b862010-09-14 05:18:09 +00002968 //
2969 // try good logic first.
2970 //
jcarsey366f81a2011-06-27 21:04:22 +00002971 if (gEfiShellProtocol != NULL) {
jcarsey3e082d52010-10-04 16:44:57 +00002972 TempLocation = StrnCatGrow(&TempLocation, NULL, DirName, 0);
jcarseyc8c22592011-10-17 17:49:21 +00002973 if (TempLocation == NULL) {
2974 ShellCloseFile(&Handle);
2975 return (EFI_OUT_OF_RESOURCES);
2976 }
jcarsey3e082d52010-10-04 16:44:57 +00002977 TempLocation2 = StrStr(TempLocation, L":");
2978 if (TempLocation2 != NULL && StrLen(StrStr(TempLocation, L":")) == 2) {
2979 *(TempLocation2+1) = CHAR_NULL;
jcarseya405b862010-09-14 05:18:09 +00002980 }
jcarsey366f81a2011-06-27 21:04:22 +00002981 if (gEfiShellProtocol->GetDevicePathFromMap(TempLocation) != NULL) {
jcarseya405b862010-09-14 05:18:09 +00002982 FreePool(TempLocation);
2983 return (EFI_SUCCESS);
2984 }
2985 FreePool(TempLocation);
2986 } else {
2987 //
2988 // probably a map name?!?!!?
2989 //
2990 TempLocation = StrStr(DirName, L"\\");
2991 if (TempLocation != NULL && *(TempLocation+1) == CHAR_NULL) {
2992 return (EFI_SUCCESS);
2993 }
2994 }
jcarsey2247dde2009-11-09 18:08:58 +00002995 return (Status);
2996 }
2997
2998 if (FileHandleIsDirectory(Handle) == EFI_SUCCESS) {
2999 ShellCloseFile(&Handle);
3000 return (EFI_SUCCESS);
3001 }
3002 ShellCloseFile(&Handle);
3003 return (EFI_NOT_FOUND);
3004}
3005
jcarsey125c2cf2009-11-18 21:36:50 +00003006/**
jcarsey36a9d672009-11-20 21:13:41 +00003007 Function to determine if a given filename represents a file.
3008
3009 @param[in] Name Path to file to test.
3010
3011 @retval EFI_SUCCESS The Path represents a file.
3012 @retval EFI_NOT_FOUND The Path does not represent a file.
3013 @retval other The path failed to open.
3014**/
3015EFI_STATUS
3016EFIAPI
3017ShellIsFile(
3018 IN CONST CHAR16 *Name
3019 )
3020{
3021 EFI_STATUS Status;
jcarseya405b862010-09-14 05:18:09 +00003022 SHELL_FILE_HANDLE Handle;
jcarsey36a9d672009-11-20 21:13:41 +00003023
jcarseyecd3d592009-12-07 18:05:00 +00003024 ASSERT(Name != NULL);
3025
jcarsey36a9d672009-11-20 21:13:41 +00003026 Handle = NULL;
3027
3028 Status = ShellOpenFileByName(Name, &Handle, EFI_FILE_MODE_READ, 0);
3029 if (EFI_ERROR(Status)) {
3030 return (Status);
3031 }
3032
3033 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
3034 ShellCloseFile(&Handle);
3035 return (EFI_SUCCESS);
3036 }
3037 ShellCloseFile(&Handle);
3038 return (EFI_NOT_FOUND);
3039}
3040
3041/**
jcarseyb3011f42010-01-11 21:49:04 +00003042 Function to determine if a given filename represents a file.
3043
3044 This will search the CWD and then the Path.
3045
3046 If Name is NULL, then ASSERT.
3047
3048 @param[in] Name Path to file to test.
3049
3050 @retval EFI_SUCCESS The Path represents a file.
3051 @retval EFI_NOT_FOUND The Path does not represent a file.
3052 @retval other The path failed to open.
3053**/
3054EFI_STATUS
3055EFIAPI
3056ShellIsFileInPath(
3057 IN CONST CHAR16 *Name
jcarseya405b862010-09-14 05:18:09 +00003058 )
3059{
jcarseyb3011f42010-01-11 21:49:04 +00003060 CHAR16 *NewName;
3061 EFI_STATUS Status;
3062
3063 if (!EFI_ERROR(ShellIsFile(Name))) {
jcarseya405b862010-09-14 05:18:09 +00003064 return (EFI_SUCCESS);
jcarseyb3011f42010-01-11 21:49:04 +00003065 }
3066
3067 NewName = ShellFindFilePath(Name);
3068 if (NewName == NULL) {
3069 return (EFI_NOT_FOUND);
3070 }
3071 Status = ShellIsFile(NewName);
3072 FreePool(NewName);
3073 return (Status);
3074}
jcarsey252d9452011-03-25 20:49:53 +00003075
jcarseyb3011f42010-01-11 21:49:04 +00003076/**
Jaben Carsey74b0fb82013-11-22 21:37:34 +00003077 Function return the number converted from a hex representation of a number.
3078
3079 Note: this function cannot be used when (UINTN)(-1), (0xFFFFFFFF) may be a valid
3080 result. Use ShellConvertStringToUint64 instead.
3081
3082 @param[in] String String representation of a number.
3083
3084 @return The unsigned integer result of the conversion.
3085 @retval (UINTN)(-1) An error occured.
3086**/
3087UINTN
3088EFIAPI
3089ShellHexStrToUintn(
3090 IN CONST CHAR16 *String
3091 )
3092{
3093 UINT64 RetVal;
3094
3095 if (!EFI_ERROR(ShellConvertStringToUint64(String, &RetVal, TRUE, TRUE))) {
3096 return ((UINTN)RetVal);
3097 }
3098
3099 return ((UINTN)(-1));
3100}
3101
3102/**
jcarsey1e6e84c2010-01-25 20:05:08 +00003103 Function to determine whether a string is decimal or hex representation of a number
Jaben Carseyd59fc242013-11-14 23:52:43 +00003104 and return the number converted from the string. Spaces are always skipped.
jcarsey125c2cf2009-11-18 21:36:50 +00003105
3106 @param[in] String String representation of a number
3107
jcarsey252d9452011-03-25 20:49:53 +00003108 @return the number
3109 @retval (UINTN)(-1) An error ocurred.
jcarsey125c2cf2009-11-18 21:36:50 +00003110**/
3111UINTN
3112EFIAPI
3113ShellStrToUintn(
3114 IN CONST CHAR16 *String
3115 )
3116{
jcarsey252d9452011-03-25 20:49:53 +00003117 UINT64 RetVal;
3118 BOOLEAN Hex;
3119
3120 Hex = FALSE;
3121
3122 if (!InternalShellIsHexOrDecimalNumber(String, Hex, TRUE)) {
3123 Hex = TRUE;
jcarsey125c2cf2009-11-18 21:36:50 +00003124 }
jcarsey252d9452011-03-25 20:49:53 +00003125
3126 if (!EFI_ERROR(ShellConvertStringToUint64(String, &RetVal, Hex, TRUE))) {
3127 return ((UINTN)RetVal);
3128 }
3129 return ((UINTN)(-1));
jcarsey125c2cf2009-11-18 21:36:50 +00003130}
3131
3132/**
3133 Safely append with automatic string resizing given length of Destination and
3134 desired length of copy from Source.
3135
3136 append the first D characters of Source to the end of Destination, where D is
3137 the lesser of Count and the StrLen() of Source. If appending those D characters
3138 will fit within Destination (whose Size is given as CurrentSize) and
jcarsey1e6e84c2010-01-25 20:05:08 +00003139 still leave room for a NULL terminator, then those characters are appended,
3140 starting at the original terminating NULL of Destination, and a new terminating
3141 NULL is appended.
jcarsey125c2cf2009-11-18 21:36:50 +00003142
3143 If appending D characters onto Destination will result in a overflow of the size
3144 given in CurrentSize the string will be grown such that the copy can be performed
3145 and CurrentSize will be updated to the new size.
3146
3147 If Source is NULL, there is nothing to append, just return the current buffer in
3148 Destination.
3149
3150 if Destination is NULL, then ASSERT()
3151 if Destination's current length (including NULL terminator) is already more then
3152 CurrentSize, then ASSERT()
3153
ydong104ff7e372011-09-02 08:05:34 +00003154 @param[in, out] Destination The String to append onto
3155 @param[in, out] CurrentSize on call the number of bytes in Destination. On
jcarsey125c2cf2009-11-18 21:36:50 +00003156 return possibly the new size (still in bytes). if NULL
3157 then allocate whatever is needed.
3158 @param[in] Source The String to append from
3159 @param[in] Count Maximum number of characters to append. if 0 then
3160 all are appended.
3161
3162 @return Destination return the resultant string.
3163**/
3164CHAR16*
3165EFIAPI
3166StrnCatGrow (
3167 IN OUT CHAR16 **Destination,
3168 IN OUT UINTN *CurrentSize,
3169 IN CONST CHAR16 *Source,
3170 IN UINTN Count
3171 )
3172{
3173 UINTN DestinationStartSize;
3174 UINTN NewSize;
3175
3176 //
3177 // ASSERTs
3178 //
3179 ASSERT(Destination != NULL);
3180
3181 //
3182 // If there's nothing to do then just return Destination
3183 //
3184 if (Source == NULL) {
3185 return (*Destination);
3186 }
3187
3188 //
3189 // allow for un-initialized pointers, based on size being 0
3190 //
3191 if (CurrentSize != NULL && *CurrentSize == 0) {
3192 *Destination = NULL;
3193 }
3194
3195 //
3196 // allow for NULL pointers address as Destination
3197 //
3198 if (*Destination != NULL) {
3199 ASSERT(CurrentSize != 0);
3200 DestinationStartSize = StrSize(*Destination);
3201 ASSERT(DestinationStartSize <= *CurrentSize);
3202 } else {
3203 DestinationStartSize = 0;
3204// ASSERT(*CurrentSize == 0);
3205 }
3206
3207 //
3208 // Append all of Source?
3209 //
3210 if (Count == 0) {
3211 Count = StrLen(Source);
3212 }
3213
3214 //
3215 // Test and grow if required
3216 //
3217 if (CurrentSize != NULL) {
3218 NewSize = *CurrentSize;
ydong10f480fdc2012-09-07 01:55:33 +00003219 if (NewSize < DestinationStartSize + (Count * sizeof(CHAR16))) {
3220 while (NewSize < (DestinationStartSize + (Count*sizeof(CHAR16)))) {
3221 NewSize += 2 * Count * sizeof(CHAR16);
3222 }
3223 *Destination = ReallocatePool(*CurrentSize, NewSize, *Destination);
3224 *CurrentSize = NewSize;
jcarsey125c2cf2009-11-18 21:36:50 +00003225 }
jcarsey125c2cf2009-11-18 21:36:50 +00003226 } else {
3227 *Destination = AllocateZeroPool((Count+1)*sizeof(CHAR16));
3228 }
3229
3230 //
3231 // Now use standard StrnCat on a big enough buffer
3232 //
jcarseyc9d92df2010-02-03 15:37:54 +00003233 if (*Destination == NULL) {
3234 return (NULL);
3235 }
jcarsey125c2cf2009-11-18 21:36:50 +00003236 return StrnCat(*Destination, Source, Count);
3237}
jcarseyc9d92df2010-02-03 15:37:54 +00003238
3239/**
3240 Prompt the user and return the resultant answer to the requestor.
3241
3242 This function will display the requested question on the shell prompt and then
3243 wait for an apropriate answer to be input from the console.
3244
jcarseya405b862010-09-14 05:18:09 +00003245 if the SHELL_PROMPT_REQUEST_TYPE is SHELL_PROMPT_REQUEST_TYPE_YESNO, ShellPromptResponseTypeQuitContinue
jcarseyc9d92df2010-02-03 15:37:54 +00003246 or SHELL_PROMPT_REQUEST_TYPE_YESNOCANCEL then *Response is of type SHELL_PROMPT_RESPONSE.
3247
jcarseya405b862010-09-14 05:18:09 +00003248 if the SHELL_PROMPT_REQUEST_TYPE is ShellPromptResponseTypeFreeform then *Response is of type
jcarseyc9d92df2010-02-03 15:37:54 +00003249 CHAR16*.
3250
3251 In either case *Response must be callee freed if Response was not NULL;
3252
3253 @param Type What type of question is asked. This is used to filter the input
3254 to prevent invalid answers to question.
3255 @param Prompt Pointer to string prompt to use to request input.
3256 @param Response Pointer to Response which will be populated upon return.
3257
3258 @retval EFI_SUCCESS The operation was sucessful.
3259 @retval EFI_UNSUPPORTED The operation is not supported as requested.
3260 @retval EFI_INVALID_PARAMETER A parameter was invalid.
3261 @return other The operation failed.
3262**/
3263EFI_STATUS
3264EFIAPI
3265ShellPromptForResponse (
3266 IN SHELL_PROMPT_REQUEST_TYPE Type,
3267 IN CHAR16 *Prompt OPTIONAL,
3268 IN OUT VOID **Response OPTIONAL
3269 )
3270{
3271 EFI_STATUS Status;
3272 EFI_INPUT_KEY Key;
3273 UINTN EventIndex;
3274 SHELL_PROMPT_RESPONSE *Resp;
jcarseya405b862010-09-14 05:18:09 +00003275 UINTN Size;
3276 CHAR16 *Buffer;
jcarseyc9d92df2010-02-03 15:37:54 +00003277
jcarseya405b862010-09-14 05:18:09 +00003278 Status = EFI_UNSUPPORTED;
3279 Resp = NULL;
3280 Buffer = NULL;
3281 Size = 0;
3282 if (Type != ShellPromptResponseTypeFreeform) {
jcarsey252d9452011-03-25 20:49:53 +00003283 Resp = (SHELL_PROMPT_RESPONSE*)AllocateZeroPool(sizeof(SHELL_PROMPT_RESPONSE));
jcarsey3e082d52010-10-04 16:44:57 +00003284 if (Resp == NULL) {
3285 return (EFI_OUT_OF_RESOURCES);
3286 }
xdu290bfa222010-07-19 05:21:27 +00003287 }
jcarseyc9d92df2010-02-03 15:37:54 +00003288
3289 switch(Type) {
jcarseya405b862010-09-14 05:18:09 +00003290 case ShellPromptResponseTypeQuitContinue:
jcarseyc9d92df2010-02-03 15:37:54 +00003291 if (Prompt != NULL) {
3292 ShellPrintEx(-1, -1, L"%s", Prompt);
3293 }
3294 //
3295 // wait for valid response
3296 //
3297 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3298 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
Jaben Carsey31b018a2014-01-16 16:52:39 +00003299 if (EFI_ERROR(Status)) {
3300 break;
3301 }
jcarseyc9d92df2010-02-03 15:37:54 +00003302 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3303 if (Key.UnicodeChar == L'Q' || Key.UnicodeChar ==L'q') {
jcarseya405b862010-09-14 05:18:09 +00003304 *Resp = ShellPromptResponseQuit;
jcarseyc9d92df2010-02-03 15:37:54 +00003305 } else {
jcarseya405b862010-09-14 05:18:09 +00003306 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003307 }
3308 break;
jcarseya405b862010-09-14 05:18:09 +00003309 case ShellPromptResponseTypeYesNoCancel:
jcarseyc9d92df2010-02-03 15:37:54 +00003310 if (Prompt != NULL) {
3311 ShellPrintEx(-1, -1, L"%s", Prompt);
3312 }
3313 //
3314 // wait for valid response
3315 //
jcarseya405b862010-09-14 05:18:09 +00003316 *Resp = ShellPromptResponseMax;
3317 while (*Resp == ShellPromptResponseMax) {
Jaben Carsey194ae482013-12-09 22:55:13 +00003318 if (ShellGetExecutionBreakFlag()) {
3319 Status = EFI_ABORTED;
3320 break;
3321 }
jcarseyc9d92df2010-02-03 15:37:54 +00003322 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3323 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
Jaben Carsey31b018a2014-01-16 16:52:39 +00003324 if (EFI_ERROR(Status)) {
3325 break;
3326 }
jcarseyc9d92df2010-02-03 15:37:54 +00003327 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3328 switch (Key.UnicodeChar) {
3329 case L'Y':
3330 case L'y':
jcarseya405b862010-09-14 05:18:09 +00003331 *Resp = ShellPromptResponseYes;
jcarseyc9d92df2010-02-03 15:37:54 +00003332 break;
3333 case L'N':
3334 case L'n':
jcarseya405b862010-09-14 05:18:09 +00003335 *Resp = ShellPromptResponseNo;
jcarseyc9d92df2010-02-03 15:37:54 +00003336 break;
3337 case L'C':
3338 case L'c':
jcarseya405b862010-09-14 05:18:09 +00003339 *Resp = ShellPromptResponseCancel;
3340 break;
3341 }
3342 }
3343 break; case ShellPromptResponseTypeYesNoAllCancel:
3344 if (Prompt != NULL) {
3345 ShellPrintEx(-1, -1, L"%s", Prompt);
3346 }
3347 //
3348 // wait for valid response
3349 //
3350 *Resp = ShellPromptResponseMax;
3351 while (*Resp == ShellPromptResponseMax) {
Jaben Carsey194ae482013-12-09 22:55:13 +00003352 if (ShellGetExecutionBreakFlag()) {
3353 Status = EFI_ABORTED;
3354 break;
3355 }
jcarseya405b862010-09-14 05:18:09 +00003356 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3357 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
Jaben Carsey31b018a2014-01-16 16:52:39 +00003358 if (EFI_ERROR(Status)) {
3359 break;
3360 }
jcarseya405b862010-09-14 05:18:09 +00003361 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3362 switch (Key.UnicodeChar) {
3363 case L'Y':
3364 case L'y':
3365 *Resp = ShellPromptResponseYes;
3366 break;
3367 case L'N':
3368 case L'n':
3369 *Resp = ShellPromptResponseNo;
3370 break;
3371 case L'A':
3372 case L'a':
3373 *Resp = ShellPromptResponseAll;
3374 break;
3375 case L'C':
3376 case L'c':
3377 *Resp = ShellPromptResponseCancel;
jcarseyc9d92df2010-02-03 15:37:54 +00003378 break;
3379 }
3380 }
3381 break;
jcarseya405b862010-09-14 05:18:09 +00003382 case ShellPromptResponseTypeEnterContinue:
3383 case ShellPromptResponseTypeAnyKeyContinue:
jcarseyc9d92df2010-02-03 15:37:54 +00003384 if (Prompt != NULL) {
3385 ShellPrintEx(-1, -1, L"%s", Prompt);
3386 }
3387 //
3388 // wait for valid response
3389 //
jcarseya405b862010-09-14 05:18:09 +00003390 *Resp = ShellPromptResponseMax;
3391 while (*Resp == ShellPromptResponseMax) {
Jaben Carsey194ae482013-12-09 22:55:13 +00003392 if (ShellGetExecutionBreakFlag()) {
3393 Status = EFI_ABORTED;
3394 break;
3395 }
jcarseyc9d92df2010-02-03 15:37:54 +00003396 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
jcarseya405b862010-09-14 05:18:09 +00003397 if (Type == ShellPromptResponseTypeEnterContinue) {
jcarseyc9d92df2010-02-03 15:37:54 +00003398 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
Jaben Carsey31b018a2014-01-16 16:52:39 +00003399 if (EFI_ERROR(Status)) {
3400 break;
3401 }
jcarseyc9d92df2010-02-03 15:37:54 +00003402 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3403 if (Key.UnicodeChar == CHAR_CARRIAGE_RETURN) {
jcarseya405b862010-09-14 05:18:09 +00003404 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003405 break;
3406 }
3407 }
jcarseya405b862010-09-14 05:18:09 +00003408 if (Type == ShellPromptResponseTypeAnyKeyContinue) {
3409 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3410 ASSERT_EFI_ERROR(Status);
3411 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003412 break;
3413 }
3414 }
3415 break;
jcarseya405b862010-09-14 05:18:09 +00003416 case ShellPromptResponseTypeYesNo:
3417 if (Prompt != NULL) {
3418 ShellPrintEx(-1, -1, L"%s", Prompt);
3419 }
3420 //
3421 // wait for valid response
3422 //
3423 *Resp = ShellPromptResponseMax;
3424 while (*Resp == ShellPromptResponseMax) {
Jaben Carsey194ae482013-12-09 22:55:13 +00003425 if (ShellGetExecutionBreakFlag()) {
3426 Status = EFI_ABORTED;
3427 break;
3428 }
jcarseya405b862010-09-14 05:18:09 +00003429 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3430 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
Jaben Carsey31b018a2014-01-16 16:52:39 +00003431 if (EFI_ERROR(Status)) {
3432 break;
3433 }
jcarseya405b862010-09-14 05:18:09 +00003434 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3435 switch (Key.UnicodeChar) {
3436 case L'Y':
3437 case L'y':
3438 *Resp = ShellPromptResponseYes;
3439 break;
3440 case L'N':
3441 case L'n':
3442 *Resp = ShellPromptResponseNo;
3443 break;
3444 }
3445 }
3446 break;
3447 case ShellPromptResponseTypeFreeform:
3448 if (Prompt != NULL) {
3449 ShellPrintEx(-1, -1, L"%s", Prompt);
3450 }
3451 while(1) {
Jaben Carsey194ae482013-12-09 22:55:13 +00003452 if (ShellGetExecutionBreakFlag()) {
3453 Status = EFI_ABORTED;
3454 break;
3455 }
jcarseya405b862010-09-14 05:18:09 +00003456 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3457 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
Jaben Carsey31b018a2014-01-16 16:52:39 +00003458 if (EFI_ERROR(Status)) {
3459 break;
3460 }
jcarseya405b862010-09-14 05:18:09 +00003461 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3462 if (Key.UnicodeChar == CHAR_CARRIAGE_RETURN) {
3463 break;
3464 }
3465 ASSERT((Buffer == NULL && Size == 0) || (Buffer != NULL));
3466 StrnCatGrow(&Buffer, &Size, &Key.UnicodeChar, 1);
3467 }
3468 break;
3469 //
3470 // This is the location to add new prompt types.
Jaben Carsey194ae482013-12-09 22:55:13 +00003471 // If your new type loops remember to add ExecutionBreak support.
jcarseya405b862010-09-14 05:18:09 +00003472 //
jcarseyc9d92df2010-02-03 15:37:54 +00003473 default:
jcarseya405b862010-09-14 05:18:09 +00003474 ASSERT(FALSE);
jcarseyc9d92df2010-02-03 15:37:54 +00003475 }
3476
3477 if (Response != NULL) {
jcarseya405b862010-09-14 05:18:09 +00003478 if (Resp != NULL) {
3479 *Response = Resp;
3480 } else if (Buffer != NULL) {
3481 *Response = Buffer;
3482 }
jcarseyc9d92df2010-02-03 15:37:54 +00003483 } else {
jcarseya405b862010-09-14 05:18:09 +00003484 if (Resp != NULL) {
3485 FreePool(Resp);
3486 }
3487 if (Buffer != NULL) {
3488 FreePool(Buffer);
3489 }
jcarseyc9d92df2010-02-03 15:37:54 +00003490 }
3491
jcarseya405b862010-09-14 05:18:09 +00003492 ShellPrintEx(-1, -1, L"\r\n");
jcarseyc9d92df2010-02-03 15:37:54 +00003493 return (Status);
3494}
3495
3496/**
3497 Prompt the user and return the resultant answer to the requestor.
3498
3499 This function is the same as ShellPromptForResponse, except that the prompt is
3500 automatically pulled from HII.
3501
3502 @param Type What type of question is asked. This is used to filter the input
3503 to prevent invalid answers to question.
jcarseya405b862010-09-14 05:18:09 +00003504 @param[in] HiiFormatStringId The format string Id for getting from Hii.
3505 @param[in] HiiFormatHandle The format string Handle for getting from Hii.
3506 @param Response Pointer to Response which will be populated upon return.
jcarseyc9d92df2010-02-03 15:37:54 +00003507
3508 @retval EFI_SUCCESS the operation was sucessful.
3509 @return other the operation failed.
3510
3511 @sa ShellPromptForResponse
3512**/
3513EFI_STATUS
3514EFIAPI
3515ShellPromptForResponseHii (
3516 IN SHELL_PROMPT_REQUEST_TYPE Type,
3517 IN CONST EFI_STRING_ID HiiFormatStringId,
3518 IN CONST EFI_HANDLE HiiFormatHandle,
3519 IN OUT VOID **Response
3520 )
3521{
3522 CHAR16 *Prompt;
3523 EFI_STATUS Status;
3524
3525 Prompt = HiiGetString(HiiFormatHandle, HiiFormatStringId, NULL);
3526 Status = ShellPromptForResponse(Type, Prompt, Response);
3527 FreePool(Prompt);
3528 return (Status);
3529}
3530
jcarseya405b862010-09-14 05:18:09 +00003531/**
3532 Function to determin if an entire string is a valid number.
jcarseyc9d92df2010-02-03 15:37:54 +00003533
jcarseya405b862010-09-14 05:18:09 +00003534 If Hex it must be preceeded with a 0x or has ForceHex, set TRUE.
3535
3536 @param[in] String The string to evaluate.
3537 @param[in] ForceHex TRUE - always assume hex.
3538 @param[in] StopAtSpace TRUE to halt upon finding a space, FALSE to keep going.
3539
3540 @retval TRUE It is all numeric (dec/hex) characters.
3541 @retval FALSE There is a non-numeric character.
3542**/
3543BOOLEAN
3544EFIAPI
jcarsey252d9452011-03-25 20:49:53 +00003545InternalShellIsHexOrDecimalNumber (
jcarseya405b862010-09-14 05:18:09 +00003546 IN CONST CHAR16 *String,
3547 IN CONST BOOLEAN ForceHex,
3548 IN CONST BOOLEAN StopAtSpace
3549 )
3550{
3551 BOOLEAN Hex;
3552
3553 //
3554 // chop off a single negative sign
3555 //
3556 if (String != NULL && *String == L'-') {
3557 String++;
3558 }
darylm503b0934ac2011-11-12 00:35:11 +00003559
jcarseya405b862010-09-14 05:18:09 +00003560 if (String == NULL) {
3561 return (FALSE);
3562 }
3563
3564 //
3565 // chop leading zeroes
3566 //
3567 while(String != NULL && *String == L'0'){
3568 String++;
3569 }
3570 //
3571 // allow '0x' or '0X', but not 'x' or 'X'
3572 //
3573 if (String != NULL && (*String == L'x' || *String == L'X')) {
3574 if (*(String-1) != L'0') {
3575 //
3576 // we got an x without a preceeding 0
3577 //
3578 return (FALSE);
3579 }
3580 String++;
3581 Hex = TRUE;
3582 } else if (ForceHex) {
3583 Hex = TRUE;
3584 } else {
3585 Hex = FALSE;
3586 }
3587
3588 //
3589 // loop through the remaining characters and use the lib function
3590 //
3591 for ( ; String != NULL && *String != CHAR_NULL && !(StopAtSpace && *String == L' ') ; String++){
3592 if (Hex) {
3593 if (!ShellIsHexaDecimalDigitCharacter(*String)) {
3594 return (FALSE);
3595 }
3596 } else {
3597 if (!ShellIsDecimalDigitCharacter(*String)) {
3598 return (FALSE);
3599 }
3600 }
3601 }
jcarsey252d9452011-03-25 20:49:53 +00003602
jcarseya405b862010-09-14 05:18:09 +00003603 return (TRUE);
3604}
3605
3606/**
3607 Function to determine if a given filename exists.
3608
3609 @param[in] Name Path to test.
3610
3611 @retval EFI_SUCCESS The Path represents a file.
3612 @retval EFI_NOT_FOUND The Path does not represent a file.
3613 @retval other The path failed to open.
3614**/
3615EFI_STATUS
3616EFIAPI
3617ShellFileExists(
3618 IN CONST CHAR16 *Name
3619 )
3620{
3621 EFI_STATUS Status;
3622 EFI_SHELL_FILE_INFO *List;
3623
3624 ASSERT(Name != NULL);
3625
3626 List = NULL;
3627 Status = ShellOpenFileMetaArg((CHAR16*)Name, EFI_FILE_MODE_READ, &List);
3628 if (EFI_ERROR(Status)) {
3629 return (Status);
3630 }
3631
3632 ShellCloseFileMetaArg(&List);
3633
3634 return (EFI_SUCCESS);
3635}
jcarsey252d9452011-03-25 20:49:53 +00003636
3637/**
darylm503b0934ac2011-11-12 00:35:11 +00003638 Convert a Unicode character to upper case only if
jcarsey252d9452011-03-25 20:49:53 +00003639 it maps to a valid small-case ASCII character.
3640
3641 This internal function only deal with Unicode character
3642 which maps to a valid small-case ASCII character, i.e.
3643 L'a' to L'z'. For other Unicode character, the input character
3644 is returned directly.
3645
3646 @param Char The character to convert.
3647
3648 @retval LowerCharacter If the Char is with range L'a' to L'z'.
3649 @retval Unchanged Otherwise.
3650
3651**/
3652CHAR16
3653EFIAPI
3654InternalShellCharToUpper (
3655 IN CHAR16 Char
3656 )
3657{
3658 if (Char >= L'a' && Char <= L'z') {
3659 return (CHAR16) (Char - (L'a' - L'A'));
3660 }
3661
3662 return Char;
3663}
3664
3665/**
3666 Convert a Unicode character to numerical value.
3667
3668 This internal function only deal with Unicode character
3669 which maps to a valid hexadecimal ASII character, i.e.
darylm503b0934ac2011-11-12 00:35:11 +00003670 L'0' to L'9', L'a' to L'f' or L'A' to L'F'. For other
jcarsey252d9452011-03-25 20:49:53 +00003671 Unicode character, the value returned does not make sense.
3672
3673 @param Char The character to convert.
3674
3675 @return The numerical value converted.
3676
3677**/
3678UINTN
3679EFIAPI
3680InternalShellHexCharToUintn (
3681 IN CHAR16 Char
3682 )
3683{
3684 if (ShellIsDecimalDigitCharacter (Char)) {
3685 return Char - L'0';
3686 }
3687
3688 return (UINTN) (10 + InternalShellCharToUpper (Char) - L'A');
3689}
3690
3691/**
3692 Convert a Null-terminated Unicode hexadecimal string to a value of type UINT64.
3693
3694 This function returns a value of type UINTN by interpreting the contents
3695 of the Unicode string specified by String as a hexadecimal number.
3696 The format of the input Unicode string String is:
3697
3698 [spaces][zeros][x][hexadecimal digits].
3699
3700 The valid hexadecimal digit character is in the range [0-9], [a-f] and [A-F].
3701 The prefix "0x" is optional. Both "x" and "X" is allowed in "0x" prefix.
3702 If "x" appears in the input string, it must be prefixed with at least one 0.
3703 The function will ignore the pad space, which includes spaces or tab characters,
3704 before [zeros], [x] or [hexadecimal digit]. The running zero before [x] or
3705 [hexadecimal digit] will be ignored. Then, the decoding starts after [x] or the
3706 first valid hexadecimal digit. Then, the function stops at the first character that is
3707 a not a valid hexadecimal character or NULL, whichever one comes first.
3708
3709 If String has only pad spaces, then zero is returned.
3710 If String has no leading pad spaces, leading zeros or valid hexadecimal digits,
3711 then zero is returned.
3712
3713 @param[in] String A pointer to a Null-terminated Unicode string.
3714 @param[out] Value Upon a successful return the value of the conversion.
3715 @param[in] StopAtSpace FALSE to skip spaces.
3716
3717 @retval EFI_SUCCESS The conversion was successful.
3718 @retval EFI_INVALID_PARAMETER A parameter was NULL or invalid.
3719 @retval EFI_DEVICE_ERROR An overflow occured.
3720**/
3721EFI_STATUS
3722EFIAPI
3723InternalShellStrHexToUint64 (
3724 IN CONST CHAR16 *String,
3725 OUT UINT64 *Value,
3726 IN CONST BOOLEAN StopAtSpace
3727 )
3728{
3729 UINT64 Result;
3730
3731 if (String == NULL || StrSize(String) == 0 || Value == NULL) {
3732 return (EFI_INVALID_PARAMETER);
3733 }
darylm503b0934ac2011-11-12 00:35:11 +00003734
jcarsey252d9452011-03-25 20:49:53 +00003735 //
darylm503b0934ac2011-11-12 00:35:11 +00003736 // Ignore the pad spaces (space or tab)
jcarsey252d9452011-03-25 20:49:53 +00003737 //
3738 while ((*String == L' ') || (*String == L'\t')) {
3739 String++;
3740 }
3741
3742 //
3743 // Ignore leading Zeros after the spaces
3744 //
3745 while (*String == L'0') {
3746 String++;
3747 }
3748
3749 if (InternalShellCharToUpper (*String) == L'X') {
3750 if (*(String - 1) != L'0') {
3751 return 0;
3752 }
3753 //
3754 // Skip the 'X'
3755 //
3756 String++;
3757 }
3758
3759 Result = 0;
darylm503b0934ac2011-11-12 00:35:11 +00003760
jcarsey252d9452011-03-25 20:49:53 +00003761 //
ydong1012ea4692012-07-26 05:42:43 +00003762 // Skip spaces if requested
jcarsey252d9452011-03-25 20:49:53 +00003763 //
ydong1012ea4692012-07-26 05:42:43 +00003764 while (StopAtSpace && *String == L' ') {
3765 String++;
jcarsey252d9452011-03-25 20:49:53 +00003766 }
darylm503b0934ac2011-11-12 00:35:11 +00003767
jcarsey252d9452011-03-25 20:49:53 +00003768 while (ShellIsHexaDecimalDigitCharacter (*String)) {
3769 //
darylm503b0934ac2011-11-12 00:35:11 +00003770 // If the Hex Number represented by String overflows according
jcarsey252d9452011-03-25 20:49:53 +00003771 // to the range defined by UINTN, then ASSERT().
3772 //
3773 if (!(Result <= (RShiftU64((((UINT64) ~0) - InternalShellHexCharToUintn (*String)), 4)))) {
3774// if (!(Result <= ((((UINT64) ~0) - InternalShellHexCharToUintn (*String)) >> 4))) {
3775 return (EFI_DEVICE_ERROR);
3776 }
3777
3778 Result = (LShiftU64(Result, 4));
3779 Result += InternalShellHexCharToUintn (*String);
3780 String++;
3781
3782 //
jcarsey49bd4982012-01-27 18:42:43 +00003783 // stop at spaces if requested
jcarsey252d9452011-03-25 20:49:53 +00003784 //
jcarsey49bd4982012-01-27 18:42:43 +00003785 if (StopAtSpace && *String == L' ') {
3786 break;
jcarsey252d9452011-03-25 20:49:53 +00003787 }
3788 }
3789
3790 *Value = Result;
3791 return (EFI_SUCCESS);
3792}
3793
3794/**
3795 Convert a Null-terminated Unicode decimal string to a value of
3796 type UINT64.
3797
3798 This function returns a value of type UINT64 by interpreting the contents
3799 of the Unicode string specified by String as a decimal number. The format
3800 of the input Unicode string String is:
3801
3802 [spaces] [decimal digits].
3803
3804 The valid decimal digit character is in the range [0-9]. The
3805 function will ignore the pad space, which includes spaces or
3806 tab characters, before [decimal digits]. The running zero in the
3807 beginning of [decimal digits] will be ignored. Then, the function
3808 stops at the first character that is a not a valid decimal character
3809 or a Null-terminator, whichever one comes first.
3810
3811 If String has only pad spaces, then 0 is returned.
3812 If String has no pad spaces or valid decimal digits,
3813 then 0 is returned.
3814
3815 @param[in] String A pointer to a Null-terminated Unicode string.
3816 @param[out] Value Upon a successful return the value of the conversion.
3817 @param[in] StopAtSpace FALSE to skip spaces.
3818
3819 @retval EFI_SUCCESS The conversion was successful.
3820 @retval EFI_INVALID_PARAMETER A parameter was NULL or invalid.
3821 @retval EFI_DEVICE_ERROR An overflow occured.
3822**/
3823EFI_STATUS
3824EFIAPI
3825InternalShellStrDecimalToUint64 (
3826 IN CONST CHAR16 *String,
3827 OUT UINT64 *Value,
3828 IN CONST BOOLEAN StopAtSpace
3829 )
3830{
3831 UINT64 Result;
3832
3833 if (String == NULL || StrSize (String) == 0 || Value == NULL) {
3834 return (EFI_INVALID_PARAMETER);
3835 }
3836
3837 //
3838 // Ignore the pad spaces (space or tab)
3839 //
3840 while ((*String == L' ') || (*String == L'\t')) {
3841 String++;
3842 }
3843
3844 //
3845 // Ignore leading Zeros after the spaces
3846 //
3847 while (*String == L'0') {
3848 String++;
3849 }
3850
3851 Result = 0;
3852
3853 //
3854 // Skip spaces if requested
3855 //
3856 while (StopAtSpace && *String == L' ') {
3857 String++;
3858 }
3859 while (ShellIsDecimalDigitCharacter (*String)) {
3860 //
darylm503b0934ac2011-11-12 00:35:11 +00003861 // If the number represented by String overflows according
jcarsey252d9452011-03-25 20:49:53 +00003862 // to the range defined by UINT64, then ASSERT().
3863 //
darylm503b0934ac2011-11-12 00:35:11 +00003864
jcarsey252d9452011-03-25 20:49:53 +00003865 if (!(Result <= (DivU64x32((((UINT64) ~0) - (*String - L'0')),10)))) {
3866 return (EFI_DEVICE_ERROR);
3867 }
3868
3869 Result = MultU64x32(Result, 10) + (*String - L'0');
3870 String++;
3871
3872 //
3873 // Stop at spaces if requested
3874 //
3875 if (StopAtSpace && *String == L' ') {
3876 break;
3877 }
3878 }
3879
3880 *Value = Result;
darylm503b0934ac2011-11-12 00:35:11 +00003881
jcarsey252d9452011-03-25 20:49:53 +00003882 return (EFI_SUCCESS);
3883}
3884
3885/**
3886 Function to verify and convert a string to its numerical value.
3887
3888 If Hex it must be preceeded with a 0x, 0X, or has ForceHex set TRUE.
3889
3890 @param[in] String The string to evaluate.
3891 @param[out] Value Upon a successful return the value of the conversion.
3892 @param[in] ForceHex TRUE - always assume hex.
3893 @param[in] StopAtSpace FALSE to skip spaces.
darylm503b0934ac2011-11-12 00:35:11 +00003894
jcarsey252d9452011-03-25 20:49:53 +00003895 @retval EFI_SUCCESS The conversion was successful.
3896 @retval EFI_INVALID_PARAMETER String contained an invalid character.
3897 @retval EFI_NOT_FOUND String was a number, but Value was NULL.
3898**/
3899EFI_STATUS
3900EFIAPI
3901ShellConvertStringToUint64(
3902 IN CONST CHAR16 *String,
3903 OUT UINT64 *Value,
3904 IN CONST BOOLEAN ForceHex,
3905 IN CONST BOOLEAN StopAtSpace
3906 )
3907{
3908 UINT64 RetVal;
3909 CONST CHAR16 *Walker;
3910 EFI_STATUS Status;
3911 BOOLEAN Hex;
3912
3913 Hex = ForceHex;
3914
3915 if (!InternalShellIsHexOrDecimalNumber(String, Hex, StopAtSpace)) {
3916 if (!Hex) {
3917 Hex = TRUE;
3918 if (!InternalShellIsHexOrDecimalNumber(String, Hex, StopAtSpace)) {
3919 return (EFI_INVALID_PARAMETER);
3920 }
3921 } else {
3922 return (EFI_INVALID_PARAMETER);
3923 }
3924 }
3925
3926 //
3927 // Chop off leading spaces
3928 //
3929 for (Walker = String; Walker != NULL && *Walker != CHAR_NULL && *Walker == L' '; Walker++);
3930
3931 //
3932 // make sure we have something left that is numeric.
3933 //
3934 if (Walker == NULL || *Walker == CHAR_NULL || !InternalShellIsHexOrDecimalNumber(Walker, Hex, StopAtSpace)) {
3935 return (EFI_INVALID_PARAMETER);
darylm503b0934ac2011-11-12 00:35:11 +00003936 }
jcarsey252d9452011-03-25 20:49:53 +00003937
3938 //
3939 // do the conversion.
3940 //
3941 if (Hex || StrnCmp(Walker, L"0x", 2) == 0 || StrnCmp(Walker, L"0X", 2) == 0){
3942 Status = InternalShellStrHexToUint64(Walker, &RetVal, StopAtSpace);
3943 } else {
3944 Status = InternalShellStrDecimalToUint64(Walker, &RetVal, StopAtSpace);
3945 }
3946
3947 if (Value == NULL && !EFI_ERROR(Status)) {
3948 return (EFI_NOT_FOUND);
3949 }
3950
3951 if (Value != NULL) {
3952 *Value = RetVal;
3953 }
3954
3955 return (Status);
3956}
3957
3958/**
3959 Function to determin if an entire string is a valid number.
3960
3961 If Hex it must be preceeded with a 0x or has ForceHex, set TRUE.
3962
3963 @param[in] String The string to evaluate.
3964 @param[in] ForceHex TRUE - always assume hex.
3965 @param[in] StopAtSpace TRUE to halt upon finding a space, FALSE to keep going.
3966
3967 @retval TRUE It is all numeric (dec/hex) characters.
3968 @retval FALSE There is a non-numeric character.
3969**/
3970BOOLEAN
3971EFIAPI
3972ShellIsHexOrDecimalNumber (
3973 IN CONST CHAR16 *String,
3974 IN CONST BOOLEAN ForceHex,
3975 IN CONST BOOLEAN StopAtSpace
3976 )
3977{
3978 if (ShellConvertStringToUint64(String, NULL, ForceHex, StopAtSpace) == EFI_NOT_FOUND) {
3979 return (TRUE);
3980 }
3981 return (FALSE);
3982}
jcarsey4d0a4fc2011-07-06 22:28:36 +00003983
3984/**
3985 Function to read a single line from a SHELL_FILE_HANDLE. The \n is not included in the returned
3986 buffer. The returned buffer must be callee freed.
3987
3988 If the position upon start is 0, then the Ascii Boolean will be set. This should be
3989 maintained and not changed for all operations with the same file.
3990
ydong104ff7e372011-09-02 08:05:34 +00003991 @param[in] Handle SHELL_FILE_HANDLE to read from.
3992 @param[in, out] Ascii Boolean value for indicating whether the file is
3993 Ascii (TRUE) or UCS2 (FALSE).
jcarsey4d0a4fc2011-07-06 22:28:36 +00003994
jcarseybeab0fc2011-10-10 17:26:25 +00003995 @return The line of text from the file.
3996 @retval NULL There was not enough memory available.
jcarsey4d0a4fc2011-07-06 22:28:36 +00003997
3998 @sa ShellFileHandleReadLine
3999**/
4000CHAR16*
4001EFIAPI
4002ShellFileHandleReturnLine(
4003 IN SHELL_FILE_HANDLE Handle,
4004 IN OUT BOOLEAN *Ascii
4005 )
4006{
4007 CHAR16 *RetVal;
4008 UINTN Size;
4009 EFI_STATUS Status;
4010
4011 Size = 0;
4012 RetVal = NULL;
4013
4014 Status = ShellFileHandleReadLine(Handle, RetVal, &Size, FALSE, Ascii);
4015 if (Status == EFI_BUFFER_TOO_SMALL) {
4016 RetVal = AllocateZeroPool(Size);
jcarseybeab0fc2011-10-10 17:26:25 +00004017 if (RetVal == NULL) {
4018 return (NULL);
4019 }
jcarsey4d0a4fc2011-07-06 22:28:36 +00004020 Status = ShellFileHandleReadLine(Handle, RetVal, &Size, FALSE, Ascii);
darylm503b0934ac2011-11-12 00:35:11 +00004021
jcarsey4d0a4fc2011-07-06 22:28:36 +00004022 }
jcarsey4d0a4fc2011-07-06 22:28:36 +00004023 if (EFI_ERROR(Status) && (RetVal != NULL)) {
4024 FreePool(RetVal);
4025 RetVal = NULL;
4026 }
4027 return (RetVal);
4028}
4029
4030/**
4031 Function to read a single line (up to but not including the \n) from a SHELL_FILE_HANDLE.
4032
4033 If the position upon start is 0, then the Ascii Boolean will be set. This should be
4034 maintained and not changed for all operations with the same file.
4035
ydong104ff7e372011-09-02 08:05:34 +00004036 @param[in] Handle SHELL_FILE_HANDLE to read from.
4037 @param[in, out] Buffer The pointer to buffer to read into.
4038 @param[in, out] Size The pointer to number of bytes in Buffer.
4039 @param[in] Truncate If the buffer is large enough, this has no effect.
4040 If the buffer is is too small and Truncate is TRUE,
4041 the line will be truncated.
4042 If the buffer is is too small and Truncate is FALSE,
4043 then no read will occur.
jcarsey4d0a4fc2011-07-06 22:28:36 +00004044
ydong104ff7e372011-09-02 08:05:34 +00004045 @param[in, out] Ascii Boolean value for indicating whether the file is
4046 Ascii (TRUE) or UCS2 (FALSE).
jcarsey4d0a4fc2011-07-06 22:28:36 +00004047
4048 @retval EFI_SUCCESS The operation was successful. The line is stored in
4049 Buffer.
4050 @retval EFI_INVALID_PARAMETER Handle was NULL.
4051 @retval EFI_INVALID_PARAMETER Size was NULL.
4052 @retval EFI_BUFFER_TOO_SMALL Size was not large enough to store the line.
4053 Size was updated to the minimum space required.
4054**/
4055EFI_STATUS
4056EFIAPI
4057ShellFileHandleReadLine(
4058 IN SHELL_FILE_HANDLE Handle,
4059 IN OUT CHAR16 *Buffer,
4060 IN OUT UINTN *Size,
4061 IN BOOLEAN Truncate,
4062 IN OUT BOOLEAN *Ascii
4063 )
4064{
4065 EFI_STATUS Status;
4066 CHAR16 CharBuffer;
4067 UINTN CharSize;
4068 UINTN CountSoFar;
4069 UINT64 OriginalFilePosition;
4070
4071
4072 if (Handle == NULL
4073 ||Size == NULL
4074 ){
4075 return (EFI_INVALID_PARAMETER);
4076 }
4077 if (Buffer == NULL) {
4078 ASSERT(*Size == 0);
4079 } else {
4080 *Buffer = CHAR_NULL;
4081 }
4082 gEfiShellProtocol->GetFilePosition(Handle, &OriginalFilePosition);
4083 if (OriginalFilePosition == 0) {
4084 CharSize = sizeof(CHAR16);
4085 Status = gEfiShellProtocol->ReadFile(Handle, &CharSize, &CharBuffer);
4086 ASSERT_EFI_ERROR(Status);
4087 if (CharBuffer == gUnicodeFileTag) {
4088 *Ascii = FALSE;
4089 } else {
4090 *Ascii = TRUE;
4091 gEfiShellProtocol->SetFilePosition(Handle, OriginalFilePosition);
4092 }
4093 }
4094
4095 for (CountSoFar = 0;;CountSoFar++){
4096 CharBuffer = 0;
4097 if (*Ascii) {
4098 CharSize = sizeof(CHAR8);
4099 } else {
4100 CharSize = sizeof(CHAR16);
4101 }
4102 Status = gEfiShellProtocol->ReadFile(Handle, &CharSize, &CharBuffer);
4103 if ( EFI_ERROR(Status)
4104 || CharSize == 0
4105 || (CharBuffer == L'\n' && !(*Ascii))
4106 || (CharBuffer == '\n' && *Ascii)
4107 ){
4108 break;
4109 }
4110 //
4111 // if we have space save it...
4112 //
4113 if ((CountSoFar+1)*sizeof(CHAR16) < *Size){
4114 ASSERT(Buffer != NULL);
4115 ((CHAR16*)Buffer)[CountSoFar] = CharBuffer;
4116 ((CHAR16*)Buffer)[CountSoFar+1] = CHAR_NULL;
4117 }
4118 }
4119
4120 //
4121 // if we ran out of space tell when...
4122 //
4123 if ((CountSoFar+1)*sizeof(CHAR16) > *Size){
4124 *Size = (CountSoFar+1)*sizeof(CHAR16);
4125 if (!Truncate) {
4126 gEfiShellProtocol->SetFilePosition(Handle, OriginalFilePosition);
4127 } else {
4128 DEBUG((DEBUG_WARN, "The line was truncated in ShellFileHandleReadLine"));
4129 }
4130 return (EFI_BUFFER_TOO_SMALL);
4131 }
4132 while(Buffer[StrLen(Buffer)-1] == L'\r') {
4133 Buffer[StrLen(Buffer)-1] = CHAR_NULL;
4134 }
4135
4136 return (Status);
4137}
jcarseyfb5278e2013-02-20 18:21:14 +00004138
4139/**
jcarsey365aa982013-03-04 21:54:02 +00004140 Function to print help file / man page content in the spec from the UEFI Shell protocol GetHelpText function.
4141
4142 @param[in] CommandToGetHelpOn Pointer to a string containing the command name of help file to be printed.
4143 @param[in] SectionToGetHelpOn Pointer to the section specifier(s).
4144 @param[in] PrintCommandText If TRUE, prints the command followed by the help content, otherwise prints
4145 the help content only.
4146 @retval EFI_DEVICE_ERROR The help data format was incorrect.
4147 @retval EFI_NOT_FOUND The help data could not be found.
4148 @retval EFI_SUCCESS The operation was successful.
4149**/
4150EFI_STATUS
4151EFIAPI
4152ShellPrintHelp (
4153 IN CONST CHAR16 *CommandToGetHelpOn,
4154 IN CONST CHAR16 *SectionToGetHelpOn,
4155 IN BOOLEAN PrintCommandText
4156 )
4157{
4158 EFI_STATUS Status;
4159 CHAR16 *OutText;
4160
4161 OutText = NULL;
4162
4163 //
4164 // Get the string to print based
4165 //
4166 Status = gEfiShellProtocol->GetHelpText (CommandToGetHelpOn, SectionToGetHelpOn, &OutText);
4167
4168 //
4169 // make sure we got a valid string
4170 //
4171 if (EFI_ERROR(Status)){
4172 return Status;
4173 }
4174 if (OutText == NULL || StrLen(OutText) == 0) {
4175 return EFI_NOT_FOUND;
4176 }
4177
4178 //
4179 // Chop off trailing stuff we dont need
4180 //
4181 while (OutText[StrLen(OutText)-1] == L'\r' || OutText[StrLen(OutText)-1] == L'\n' || OutText[StrLen(OutText)-1] == L' ') {
4182 OutText[StrLen(OutText)-1] = CHAR_NULL;
4183 }
4184
4185 //
4186 // Print this out to the console
4187 //
4188 if (PrintCommandText) {
4189 ShellPrintEx(-1, -1, L"%H%-14s%N- %s\r\n", CommandToGetHelpOn, OutText);
4190 } else {
4191 ShellPrintEx(-1, -1, L"%N%s\r\n", OutText);
4192 }
4193
4194 SHELL_FREE_NON_NULL(OutText);
4195
4196 return EFI_SUCCESS;
4197}
4198
4199/**
jcarseyfb5278e2013-02-20 18:21:14 +00004200 Function to delete a file by name
4201
4202 @param[in] FileName Pointer to file name to delete.
4203
4204 @retval EFI_SUCCESS the file was deleted sucessfully
4205 @retval EFI_WARN_DELETE_FAILURE the handle was closed, but the file was not
4206 deleted
4207 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
4208 @retval EFI_NOT_FOUND The specified file could not be found on the
4209 device or the file system could not be found
4210 on the device.
4211 @retval EFI_NO_MEDIA The device has no medium.
4212 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
4213 medium is no longer supported.
4214 @retval EFI_DEVICE_ERROR The device reported an error.
4215 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
4216 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
4217 @retval EFI_ACCESS_DENIED The file was opened read only.
4218 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
4219 file.
4220 @retval other The file failed to open
4221**/
4222EFI_STATUS
4223EFIAPI
4224ShellDeleteFileByName(
4225 IN CONST CHAR16 *FileName
4226 )
4227{
4228 EFI_STATUS Status;
4229 SHELL_FILE_HANDLE FileHandle;
4230
4231 Status = ShellFileExists(FileName);
4232
4233 if (Status == EFI_SUCCESS){
4234 Status = ShellOpenFileByName(FileName, &FileHandle, EFI_FILE_MODE_READ | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE, 0x0);
4235 if (Status == EFI_SUCCESS){
4236 Status = ShellDeleteFile(&FileHandle);
4237 }
4238 }
4239
4240 return(Status);
4241
4242}