blob: 06e2386378296df769c07a29ea9234ec752e7728 [file] [log] [blame]
jcarsey94b17fa2009-05-07 18:46:18 +00001/** @file
2 Provides interface to shell functionality for shell commands and applications.
3
jcarsey252d9452011-03-25 20:49:53 +00004 Copyright (c) 2006 - 2011, 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.
jcarsey94b17fa2009-05-07 18:46:18 +000087**/
88EFI_STATUS
89EFIAPI
90ShellFindSE2 (
91 IN EFI_HANDLE ImageHandle
jcarseya405b862010-09-14 05:18:09 +000092 )
93{
jcarsey94b17fa2009-05-07 18:46:18 +000094 EFI_STATUS Status;
95 EFI_HANDLE *Buffer;
96 UINTN BufferSize;
97 UINTN HandleIndex;
98
99 BufferSize = 0;
100 Buffer = NULL;
jcarsey1e6e84c2010-01-25 20:05:08 +0000101 Status = gBS->OpenProtocol(ImageHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000102 &gEfiShellEnvironment2Guid,
103 (VOID **)&mEfiShellEnvironment2,
104 ImageHandle,
105 NULL,
106 EFI_OPEN_PROTOCOL_GET_PROTOCOL
jcarseya405b862010-09-14 05:18:09 +0000107 );
jcarsey94b17fa2009-05-07 18:46:18 +0000108 //
109 // look for the mEfiShellEnvironment2 protocol at a higher level
110 //
jcarseya405b862010-09-14 05:18:09 +0000111 if (EFI_ERROR (Status) || !(CompareGuid (&mEfiShellEnvironment2->SESGuid, &gEfiShellEnvironment2ExtGuid))){
jcarsey94b17fa2009-05-07 18:46:18 +0000112 //
113 // figure out how big of a buffer we need.
114 //
115 Status = gBS->LocateHandle (ByProtocol,
116 &gEfiShellEnvironment2Guid,
117 NULL, // ignored for ByProtocol
118 &BufferSize,
119 Buffer
jcarseya405b862010-09-14 05:18:09 +0000120 );
jcarsey2247dde2009-11-09 18:08:58 +0000121 //
122 // maybe it's not there???
123 //
124 if (Status == EFI_BUFFER_TOO_SMALL) {
jcarsey252d9452011-03-25 20:49:53 +0000125 Buffer = (EFI_HANDLE*)AllocateZeroPool(BufferSize);
jcarsey2247dde2009-11-09 18:08:58 +0000126 ASSERT(Buffer != NULL);
127 Status = gBS->LocateHandle (ByProtocol,
128 &gEfiShellEnvironment2Guid,
129 NULL, // ignored for ByProtocol
130 &BufferSize,
131 Buffer
jcarseya405b862010-09-14 05:18:09 +0000132 );
jcarsey2247dde2009-11-09 18:08:58 +0000133 }
jcarsey1cd45e72010-01-29 15:07:44 +0000134 if (!EFI_ERROR (Status) && Buffer != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000135 //
136 // now parse the list of returned handles
137 //
138 Status = EFI_NOT_FOUND;
139 for (HandleIndex = 0; HandleIndex < (BufferSize/sizeof(Buffer[0])); HandleIndex++) {
jcarsey1e6e84c2010-01-25 20:05:08 +0000140 Status = gBS->OpenProtocol(Buffer[HandleIndex],
jcarsey94b17fa2009-05-07 18:46:18 +0000141 &gEfiShellEnvironment2Guid,
142 (VOID **)&mEfiShellEnvironment2,
143 ImageHandle,
144 NULL,
145 EFI_OPEN_PROTOCOL_GET_PROTOCOL
jcarseya405b862010-09-14 05:18:09 +0000146 );
147 if (CompareGuid (&mEfiShellEnvironment2->SESGuid, &gEfiShellEnvironment2ExtGuid)) {
jcarsey94b17fa2009-05-07 18:46:18 +0000148 mEfiShellEnvironment2Handle = Buffer[HandleIndex];
149 Status = EFI_SUCCESS;
150 break;
151 }
152 }
153 }
154 }
155 if (Buffer != NULL) {
156 FreePool (Buffer);
157 }
158 return (Status);
159}
160
jcarsey252d9452011-03-25 20:49:53 +0000161/**
jcarseya405b862010-09-14 05:18:09 +0000162 Function to do most of the work of the constructor. Allows for calling
163 multiple times without complete re-initialization.
164
165 @param[in] ImageHandle A copy of the ImageHandle.
166 @param[in] SystemTable A pointer to the SystemTable for the application.
jcarsey252d9452011-03-25 20:49:53 +0000167
168 @retval EFI_SUCCESS The operationw as successful.
jcarseya405b862010-09-14 05:18:09 +0000169**/
jcarsey94b17fa2009-05-07 18:46:18 +0000170EFI_STATUS
171EFIAPI
jcarseyd2b45642009-05-11 18:02:16 +0000172ShellLibConstructorWorker (
jcarsey94b17fa2009-05-07 18:46:18 +0000173 IN EFI_HANDLE ImageHandle,
174 IN EFI_SYSTEM_TABLE *SystemTable
jcarseya405b862010-09-14 05:18:09 +0000175 )
176{
177 EFI_STATUS Status;
jcarseyecd3d592009-12-07 18:05:00 +0000178
jcarsey94b17fa2009-05-07 18:46:18 +0000179 //
180 // UEFI 2.0 shell interfaces (used preferentially)
181 //
jcarseya405b862010-09-14 05:18:09 +0000182 Status = gBS->OpenProtocol(
183 ImageHandle,
184 &gEfiShellProtocolGuid,
jcarsey366f81a2011-06-27 21:04:22 +0000185 (VOID **)&gEfiShellProtocol,
jcarseya405b862010-09-14 05:18:09 +0000186 ImageHandle,
187 NULL,
188 EFI_OPEN_PROTOCOL_GET_PROTOCOL
189 );
jcarsey94b17fa2009-05-07 18:46:18 +0000190 if (EFI_ERROR(Status)) {
jcarseya405b862010-09-14 05:18:09 +0000191 //
192 // Search for the shell protocol
193 //
194 Status = gBS->LocateProtocol(
195 &gEfiShellProtocolGuid,
196 NULL,
jcarsey366f81a2011-06-27 21:04:22 +0000197 (VOID **)&gEfiShellProtocol
jcarseya405b862010-09-14 05:18:09 +0000198 );
199 if (EFI_ERROR(Status)) {
jcarsey366f81a2011-06-27 21:04:22 +0000200 gEfiShellProtocol = NULL;
jcarseya405b862010-09-14 05:18:09 +0000201 }
jcarsey94b17fa2009-05-07 18:46:18 +0000202 }
jcarseya405b862010-09-14 05:18:09 +0000203 Status = gBS->OpenProtocol(
204 ImageHandle,
205 &gEfiShellParametersProtocolGuid,
jcarsey366f81a2011-06-27 21:04:22 +0000206 (VOID **)&gEfiShellParametersProtocol,
jcarseya405b862010-09-14 05:18:09 +0000207 ImageHandle,
208 NULL,
209 EFI_OPEN_PROTOCOL_GET_PROTOCOL
210 );
jcarsey94b17fa2009-05-07 18:46:18 +0000211 if (EFI_ERROR(Status)) {
jcarsey366f81a2011-06-27 21:04:22 +0000212 gEfiShellParametersProtocol = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000213 }
214
jcarsey366f81a2011-06-27 21:04:22 +0000215 if (gEfiShellParametersProtocol == NULL || gEfiShellProtocol == NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000216 //
217 // Moved to seperate function due to complexity
218 //
219 Status = ShellFindSE2(ImageHandle);
220
221 if (EFI_ERROR(Status)) {
222 DEBUG((DEBUG_ERROR, "Status: 0x%08x\r\n", Status));
223 mEfiShellEnvironment2 = NULL;
224 }
jcarsey1e6e84c2010-01-25 20:05:08 +0000225 Status = gBS->OpenProtocol(ImageHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000226 &gEfiShellInterfaceGuid,
227 (VOID **)&mEfiShellInterface,
228 ImageHandle,
229 NULL,
230 EFI_OPEN_PROTOCOL_GET_PROTOCOL
jcarseya405b862010-09-14 05:18:09 +0000231 );
jcarsey94b17fa2009-05-07 18:46:18 +0000232 if (EFI_ERROR(Status)) {
233 mEfiShellInterface = NULL;
234 }
235 }
jcarseyc9d92df2010-02-03 15:37:54 +0000236
jcarsey94b17fa2009-05-07 18:46:18 +0000237 //
238 // only success getting 2 of either the old or new, but no 1/2 and 1/2
239 //
jcarsey1e6e84c2010-01-25 20:05:08 +0000240 if ((mEfiShellEnvironment2 != NULL && mEfiShellInterface != NULL) ||
jcarsey366f81a2011-06-27 21:04:22 +0000241 (gEfiShellProtocol != NULL && gEfiShellParametersProtocol != NULL) ) {
242 if (gEfiShellProtocol != NULL) {
243 FileFunctionMap.GetFileInfo = gEfiShellProtocol->GetFileInfo;
244 FileFunctionMap.SetFileInfo = gEfiShellProtocol->SetFileInfo;
245 FileFunctionMap.ReadFile = gEfiShellProtocol->ReadFile;
246 FileFunctionMap.WriteFile = gEfiShellProtocol->WriteFile;
247 FileFunctionMap.CloseFile = gEfiShellProtocol->CloseFile;
248 FileFunctionMap.DeleteFile = gEfiShellProtocol->DeleteFile;
249 FileFunctionMap.GetFilePosition = gEfiShellProtocol->GetFilePosition;
250 FileFunctionMap.SetFilePosition = gEfiShellProtocol->SetFilePosition;
251 FileFunctionMap.FlushFile = gEfiShellProtocol->FlushFile;
252 FileFunctionMap.GetFileSize = gEfiShellProtocol->GetFileSize;
jcarseyd2b45642009-05-11 18:02:16 +0000253 } else {
jcarseya405b862010-09-14 05:18:09 +0000254 FileFunctionMap.GetFileInfo = (EFI_SHELL_GET_FILE_INFO)FileHandleGetInfo;
255 FileFunctionMap.SetFileInfo = (EFI_SHELL_SET_FILE_INFO)FileHandleSetInfo;
256 FileFunctionMap.ReadFile = (EFI_SHELL_READ_FILE)FileHandleRead;
257 FileFunctionMap.WriteFile = (EFI_SHELL_WRITE_FILE)FileHandleWrite;
258 FileFunctionMap.CloseFile = (EFI_SHELL_CLOSE_FILE)FileHandleClose;
259 FileFunctionMap.DeleteFile = (EFI_SHELL_DELETE_FILE)FileHandleDelete;
260 FileFunctionMap.GetFilePosition = (EFI_SHELL_GET_FILE_POSITION)FileHandleGetPosition;
261 FileFunctionMap.SetFilePosition = (EFI_SHELL_SET_FILE_POSITION)FileHandleSetPosition;
262 FileFunctionMap.FlushFile = (EFI_SHELL_FLUSH_FILE)FileHandleFlush;
263 FileFunctionMap.GetFileSize = (EFI_SHELL_GET_FILE_SIZE)FileHandleGetSize;
jcarseyd2b45642009-05-11 18:02:16 +0000264 }
jcarsey94b17fa2009-05-07 18:46:18 +0000265 return (EFI_SUCCESS);
266 }
267 return (EFI_NOT_FOUND);
268}
jcarseyd2b45642009-05-11 18:02:16 +0000269/**
270 Constructor for the Shell library.
271
272 Initialize the library and determine if the underlying is a UEFI Shell 2.0 or an EFI shell.
273
274 @param ImageHandle the image handle of the process
275 @param SystemTable the EFI System Table pointer
276
277 @retval EFI_SUCCESS the initialization was complete sucessfully
278 @return others an error ocurred during initialization
279**/
280EFI_STATUS
281EFIAPI
282ShellLibConstructor (
283 IN EFI_HANDLE ImageHandle,
284 IN EFI_SYSTEM_TABLE *SystemTable
jcarseya405b862010-09-14 05:18:09 +0000285 )
286{
jcarseyd2b45642009-05-11 18:02:16 +0000287 mEfiShellEnvironment2 = NULL;
jcarsey366f81a2011-06-27 21:04:22 +0000288 gEfiShellProtocol = NULL;
289 gEfiShellParametersProtocol = NULL;
jcarseyd2b45642009-05-11 18:02:16 +0000290 mEfiShellInterface = NULL;
291 mEfiShellEnvironment2Handle = NULL;
292
jcarseyd2b45642009-05-11 18:02:16 +0000293 //
294 // verify that auto initialize is not set false
jcarsey1e6e84c2010-01-25 20:05:08 +0000295 //
jcarseyd2b45642009-05-11 18:02:16 +0000296 if (PcdGetBool(PcdShellLibAutoInitialize) == 0) {
297 return (EFI_SUCCESS);
298 }
jcarsey1e6e84c2010-01-25 20:05:08 +0000299
jcarseyd2b45642009-05-11 18:02:16 +0000300 return (ShellLibConstructorWorker(ImageHandle, SystemTable));
301}
jcarsey94b17fa2009-05-07 18:46:18 +0000302
303/**
jcarseya405b862010-09-14 05:18:09 +0000304 Destructor for the library. free any resources.
305
306 @param[in] ImageHandle A copy of the ImageHandle.
307 @param[in] SystemTable A pointer to the SystemTable for the application.
308
309 @retval EFI_SUCCESS The operation was successful.
310 @return An error from the CloseProtocol function.
jcarsey94b17fa2009-05-07 18:46:18 +0000311**/
312EFI_STATUS
313EFIAPI
314ShellLibDestructor (
315 IN EFI_HANDLE ImageHandle,
316 IN EFI_SYSTEM_TABLE *SystemTable
jcarseya405b862010-09-14 05:18:09 +0000317 )
318{
jcarsey94b17fa2009-05-07 18:46:18 +0000319 if (mEfiShellEnvironment2 != NULL) {
320 gBS->CloseProtocol(mEfiShellEnvironment2Handle==NULL?ImageHandle:mEfiShellEnvironment2Handle,
321 &gEfiShellEnvironment2Guid,
322 ImageHandle,
323 NULL);
jcarseyd2b45642009-05-11 18:02:16 +0000324 mEfiShellEnvironment2 = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000325 }
326 if (mEfiShellInterface != NULL) {
327 gBS->CloseProtocol(ImageHandle,
328 &gEfiShellInterfaceGuid,
329 ImageHandle,
jcarsey1e6e84c2010-01-25 20:05:08 +0000330 NULL);
jcarseyd2b45642009-05-11 18:02:16 +0000331 mEfiShellInterface = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000332 }
jcarsey366f81a2011-06-27 21:04:22 +0000333 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000334 gBS->CloseProtocol(ImageHandle,
335 &gEfiShellProtocolGuid,
336 ImageHandle,
jcarsey1e6e84c2010-01-25 20:05:08 +0000337 NULL);
jcarsey366f81a2011-06-27 21:04:22 +0000338 gEfiShellProtocol = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000339 }
jcarsey366f81a2011-06-27 21:04:22 +0000340 if (gEfiShellParametersProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000341 gBS->CloseProtocol(ImageHandle,
342 &gEfiShellParametersProtocolGuid,
343 ImageHandle,
jcarsey1e6e84c2010-01-25 20:05:08 +0000344 NULL);
jcarsey366f81a2011-06-27 21:04:22 +0000345 gEfiShellParametersProtocol = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000346 }
jcarseyd2b45642009-05-11 18:02:16 +0000347 mEfiShellEnvironment2Handle = NULL;
jcarseyecd3d592009-12-07 18:05:00 +0000348
jcarsey94b17fa2009-05-07 18:46:18 +0000349 return (EFI_SUCCESS);
350}
jcarseyd2b45642009-05-11 18:02:16 +0000351
352/**
353 This function causes the shell library to initialize itself. If the shell library
354 is already initialized it will de-initialize all the current protocol poitners and
355 re-populate them again.
356
357 When the library is used with PcdShellLibAutoInitialize set to true this function
358 will return EFI_SUCCESS and perform no actions.
359
360 This function is intended for internal access for shell commands only.
361
362 @retval EFI_SUCCESS the initialization was complete sucessfully
363
364**/
365EFI_STATUS
366EFIAPI
367ShellInitialize (
jcarseya405b862010-09-14 05:18:09 +0000368 )
369{
jcarseyd2b45642009-05-11 18:02:16 +0000370 //
371 // if auto initialize is not false then skip
372 //
373 if (PcdGetBool(PcdShellLibAutoInitialize) != 0) {
374 return (EFI_SUCCESS);
375 }
376
377 //
378 // deinit the current stuff
379 //
380 ASSERT_EFI_ERROR(ShellLibDestructor(gImageHandle, gST));
381
382 //
383 // init the new stuff
384 //
385 return (ShellLibConstructorWorker(gImageHandle, gST));
386}
387
jcarsey94b17fa2009-05-07 18:46:18 +0000388/**
jcarsey1e6e84c2010-01-25 20:05:08 +0000389 This function will retrieve the information about the file for the handle
jcarsey94b17fa2009-05-07 18:46:18 +0000390 specified and store it in allocated pool memory.
391
jcarsey1e6e84c2010-01-25 20:05:08 +0000392 This function allocates a buffer to store the file's information. It is the
qhuang869817bf2009-05-20 14:42:48 +0000393 caller's responsibility to free the buffer
jcarsey94b17fa2009-05-07 18:46:18 +0000394
jcarsey1e6e84c2010-01-25 20:05:08 +0000395 @param FileHandle The file handle of the file for which information is
jcarsey94b17fa2009-05-07 18:46:18 +0000396 being requested.
397
398 @retval NULL information could not be retrieved.
399
400 @return the information about the file
401**/
402EFI_FILE_INFO*
403EFIAPI
404ShellGetFileInfo (
jcarseya405b862010-09-14 05:18:09 +0000405 IN SHELL_FILE_HANDLE FileHandle
406 )
407{
jcarseyd2b45642009-05-11 18:02:16 +0000408 return (FileFunctionMap.GetFileInfo(FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000409}
410
411/**
jcarseya405b862010-09-14 05:18:09 +0000412 This function sets the information about the file for the opened handle
jcarsey94b17fa2009-05-07 18:46:18 +0000413 specified.
414
jcarseya405b862010-09-14 05:18:09 +0000415 @param[in] FileHandle The file handle of the file for which information
416 is being set.
jcarsey94b17fa2009-05-07 18:46:18 +0000417
jcarseya405b862010-09-14 05:18:09 +0000418 @param[in] FileInfo The information to set.
jcarsey94b17fa2009-05-07 18:46:18 +0000419
jcarseya405b862010-09-14 05:18:09 +0000420 @retval EFI_SUCCESS The information was set.
421 @retval EFI_INVALID_PARAMETER A parameter was out of range or invalid.
422 @retval EFI_UNSUPPORTED The FileHandle does not support FileInfo.
423 @retval EFI_NO_MEDIA The device has no medium.
424 @retval EFI_DEVICE_ERROR The device reported an error.
jcarsey94b17fa2009-05-07 18:46:18 +0000425 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
jcarseya405b862010-09-14 05:18:09 +0000426 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
427 @retval EFI_ACCESS_DENIED The file was opened read only.
428 @retval EFI_VOLUME_FULL The volume is full.
jcarsey94b17fa2009-05-07 18:46:18 +0000429**/
430EFI_STATUS
431EFIAPI
432ShellSetFileInfo (
jcarseya405b862010-09-14 05:18:09 +0000433 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000434 IN EFI_FILE_INFO *FileInfo
jcarseya405b862010-09-14 05:18:09 +0000435 )
436{
jcarseyd2b45642009-05-11 18:02:16 +0000437 return (FileFunctionMap.SetFileInfo(FileHandle, FileInfo));
jcarsey1e6e84c2010-01-25 20:05:08 +0000438}
439
jcarsey94b17fa2009-05-07 18:46:18 +0000440 /**
441 This function will open a file or directory referenced by DevicePath.
442
jcarsey1e6e84c2010-01-25 20:05:08 +0000443 This function opens a file with the open mode according to the file path. The
jcarsey94b17fa2009-05-07 18:46:18 +0000444 Attributes is valid only for EFI_FILE_MODE_CREATE.
445
jcarsey1e6e84c2010-01-25 20:05:08 +0000446 @param FilePath on input the device path to the file. On output
jcarsey94b17fa2009-05-07 18:46:18 +0000447 the remaining device path.
448 @param DeviceHandle pointer to the system device handle.
449 @param FileHandle pointer to the file handle.
450 @param OpenMode the mode to open the file with.
451 @param Attributes the file's file attributes.
452
453 @retval EFI_SUCCESS The information was set.
454 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
jcarsey1e6e84c2010-01-25 20:05:08 +0000455 @retval EFI_UNSUPPORTED Could not open the file path.
456 @retval EFI_NOT_FOUND The specified file could not be found on the
457 device or the file system could not be found on
jcarsey94b17fa2009-05-07 18:46:18 +0000458 the device.
459 @retval EFI_NO_MEDIA The device has no medium.
jcarsey1e6e84c2010-01-25 20:05:08 +0000460 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
jcarsey94b17fa2009-05-07 18:46:18 +0000461 medium is no longer supported.
462 @retval EFI_DEVICE_ERROR The device reported an error.
463 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
464 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
jcarseya405b862010-09-14 05:18:09 +0000465 @retval EFI_ACCESS_DENIED The file was opened read only.
jcarsey1e6e84c2010-01-25 20:05:08 +0000466 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
jcarsey94b17fa2009-05-07 18:46:18 +0000467 file.
468 @retval EFI_VOLUME_FULL The volume is full.
469**/
470EFI_STATUS
471EFIAPI
472ShellOpenFileByDevicePath(
473 IN OUT EFI_DEVICE_PATH_PROTOCOL **FilePath,
474 OUT EFI_HANDLE *DeviceHandle,
jcarseya405b862010-09-14 05:18:09 +0000475 OUT SHELL_FILE_HANDLE *FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000476 IN UINT64 OpenMode,
477 IN UINT64 Attributes
jcarseya405b862010-09-14 05:18:09 +0000478 )
479{
480 CHAR16 *FileName;
481 EFI_STATUS Status;
jcarsey94b17fa2009-05-07 18:46:18 +0000482 EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *EfiSimpleFileSystemProtocol;
jcarseya405b862010-09-14 05:18:09 +0000483 EFI_FILE_PROTOCOL *Handle1;
484 EFI_FILE_PROTOCOL *Handle2;
jcarsey94b17fa2009-05-07 18:46:18 +0000485
jcarsey92a54472011-06-27 20:33:13 +0000486 if (FilePath == NULL || FileHandle == NULL || DeviceHandle == NULL) {
487 return (EFI_INVALID_PARAMETER);
488 }
489
jcarsey1e6e84c2010-01-25 20:05:08 +0000490 //
jcarsey94b17fa2009-05-07 18:46:18 +0000491 // which shell interface should we use
492 //
jcarsey366f81a2011-06-27 21:04:22 +0000493 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000494 //
495 // use UEFI Shell 2.0 method.
496 //
jcarsey366f81a2011-06-27 21:04:22 +0000497 FileName = gEfiShellProtocol->GetFilePathFromDevicePath(*FilePath);
jcarsey94b17fa2009-05-07 18:46:18 +0000498 if (FileName == NULL) {
499 return (EFI_INVALID_PARAMETER);
500 }
501 Status = ShellOpenFileByName(FileName, FileHandle, OpenMode, Attributes);
502 FreePool(FileName);
503 return (Status);
jcarsey1e6e84c2010-01-25 20:05:08 +0000504 }
jcarseyd2b45642009-05-11 18:02:16 +0000505
506
507 //
508 // use old shell method.
509 //
jcarsey1e6e84c2010-01-25 20:05:08 +0000510 Status = gBS->LocateDevicePath (&gEfiSimpleFileSystemProtocolGuid,
511 FilePath,
jcarseyd2b45642009-05-11 18:02:16 +0000512 DeviceHandle);
513 if (EFI_ERROR (Status)) {
514 return Status;
515 }
516 Status = gBS->OpenProtocol(*DeviceHandle,
517 &gEfiSimpleFileSystemProtocolGuid,
jcarseyb1f95a02009-06-16 00:23:19 +0000518 (VOID**)&EfiSimpleFileSystemProtocol,
jcarseyd2b45642009-05-11 18:02:16 +0000519 gImageHandle,
520 NULL,
521 EFI_OPEN_PROTOCOL_GET_PROTOCOL);
522 if (EFI_ERROR (Status)) {
523 return Status;
524 }
jcarseya405b862010-09-14 05:18:09 +0000525 Status = EfiSimpleFileSystemProtocol->OpenVolume(EfiSimpleFileSystemProtocol, &Handle1);
jcarseyd2b45642009-05-11 18:02:16 +0000526 if (EFI_ERROR (Status)) {
527 FileHandle = NULL;
528 return Status;
529 }
530
531 //
532 // go down directories one node at a time.
533 //
534 while (!IsDevicePathEnd (*FilePath)) {
jcarsey94b17fa2009-05-07 18:46:18 +0000535 //
jcarseyd2b45642009-05-11 18:02:16 +0000536 // For file system access each node should be a file path component
jcarsey94b17fa2009-05-07 18:46:18 +0000537 //
jcarseyd2b45642009-05-11 18:02:16 +0000538 if (DevicePathType (*FilePath) != MEDIA_DEVICE_PATH ||
539 DevicePathSubType (*FilePath) != MEDIA_FILEPATH_DP
jcarseya405b862010-09-14 05:18:09 +0000540 ) {
jcarsey94b17fa2009-05-07 18:46:18 +0000541 FileHandle = NULL;
jcarseyd2b45642009-05-11 18:02:16 +0000542 return (EFI_INVALID_PARAMETER);
jcarsey94b17fa2009-05-07 18:46:18 +0000543 }
jcarseyd2b45642009-05-11 18:02:16 +0000544 //
545 // Open this file path node
546 //
jcarseya405b862010-09-14 05:18:09 +0000547 Handle2 = Handle1;
548 Handle1 = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000549
550 //
jcarseyd2b45642009-05-11 18:02:16 +0000551 // Try to test opening an existing file
jcarsey94b17fa2009-05-07 18:46:18 +0000552 //
jcarseya405b862010-09-14 05:18:09 +0000553 Status = Handle2->Open (
554 Handle2,
555 &Handle1,
jcarseyd2b45642009-05-11 18:02:16 +0000556 ((FILEPATH_DEVICE_PATH*)*FilePath)->PathName,
557 OpenMode &~EFI_FILE_MODE_CREATE,
558 0
jcarseya405b862010-09-14 05:18:09 +0000559 );
jcarsey94b17fa2009-05-07 18:46:18 +0000560
jcarseyd2b45642009-05-11 18:02:16 +0000561 //
562 // see if the error was that it needs to be created
563 //
564 if ((EFI_ERROR (Status)) && (OpenMode != (OpenMode &~EFI_FILE_MODE_CREATE))) {
jcarseya405b862010-09-14 05:18:09 +0000565 Status = Handle2->Open (
566 Handle2,
567 &Handle1,
jcarsey94b17fa2009-05-07 18:46:18 +0000568 ((FILEPATH_DEVICE_PATH*)*FilePath)->PathName,
jcarseyd2b45642009-05-11 18:02:16 +0000569 OpenMode,
570 Attributes
jcarseya405b862010-09-14 05:18:09 +0000571 );
jcarsey94b17fa2009-05-07 18:46:18 +0000572 }
jcarseyd2b45642009-05-11 18:02:16 +0000573 //
574 // Close the last node
575 //
jcarseya405b862010-09-14 05:18:09 +0000576 Handle2->Close (Handle2);
jcarseyd2b45642009-05-11 18:02:16 +0000577
578 if (EFI_ERROR(Status)) {
579 return (Status);
580 }
581
582 //
583 // Get the next node
584 //
585 *FilePath = NextDevicePathNode (*FilePath);
jcarsey94b17fa2009-05-07 18:46:18 +0000586 }
jcarseya405b862010-09-14 05:18:09 +0000587
588 //
589 // This is a weak spot since if the undefined SHELL_FILE_HANDLE format changes this must change also!
590 //
591 *FileHandle = (VOID*)Handle1;
jcarseyd2b45642009-05-11 18:02:16 +0000592 return (EFI_SUCCESS);
jcarsey94b17fa2009-05-07 18:46:18 +0000593}
594
595/**
596 This function will open a file or directory referenced by filename.
597
jcarsey1e6e84c2010-01-25 20:05:08 +0000598 If return is EFI_SUCCESS, the Filehandle is the opened file's handle;
599 otherwise, the Filehandle is NULL. The Attributes is valid only for
jcarsey94b17fa2009-05-07 18:46:18 +0000600 EFI_FILE_MODE_CREATE.
601
jcarsey92a54472011-06-27 20:33:13 +0000602 if FileName is NULL then ASSERT()
jcarsey94b17fa2009-05-07 18:46:18 +0000603
604 @param FileName pointer to file name
605 @param FileHandle pointer to the file handle.
606 @param OpenMode the mode to open the file with.
607 @param Attributes the file's file attributes.
608
609 @retval EFI_SUCCESS The information was set.
610 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
jcarsey1e6e84c2010-01-25 20:05:08 +0000611 @retval EFI_UNSUPPORTED Could not open the file path.
612 @retval EFI_NOT_FOUND The specified file could not be found on the
613 device or the file system could not be found
jcarsey94b17fa2009-05-07 18:46:18 +0000614 on the device.
615 @retval EFI_NO_MEDIA The device has no medium.
jcarsey1e6e84c2010-01-25 20:05:08 +0000616 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
jcarsey94b17fa2009-05-07 18:46:18 +0000617 medium is no longer supported.
618 @retval EFI_DEVICE_ERROR The device reported an error.
619 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
620 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
621 @retval EFI_ACCESS_DENIED The file was opened read only.
jcarsey1e6e84c2010-01-25 20:05:08 +0000622 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
jcarsey94b17fa2009-05-07 18:46:18 +0000623 file.
624 @retval EFI_VOLUME_FULL The volume is full.
625**/
626EFI_STATUS
627EFIAPI
628ShellOpenFileByName(
jcarseyb82bfcc2009-06-29 16:28:23 +0000629 IN CONST CHAR16 *FileName,
jcarseya405b862010-09-14 05:18:09 +0000630 OUT SHELL_FILE_HANDLE *FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000631 IN UINT64 OpenMode,
632 IN UINT64 Attributes
jcarseya405b862010-09-14 05:18:09 +0000633 )
634{
jcarsey94b17fa2009-05-07 18:46:18 +0000635 EFI_HANDLE DeviceHandle;
636 EFI_DEVICE_PATH_PROTOCOL *FilePath;
jcarseyb1f95a02009-06-16 00:23:19 +0000637 EFI_STATUS Status;
638 EFI_FILE_INFO *FileInfo;
jcarsey94b17fa2009-05-07 18:46:18 +0000639
640 //
641 // ASSERT if FileName is NULL
642 //
643 ASSERT(FileName != NULL);
644
jcarseya405b862010-09-14 05:18:09 +0000645 if (FileName == NULL) {
646 return (EFI_INVALID_PARAMETER);
647 }
648
jcarsey366f81a2011-06-27 21:04:22 +0000649 if (gEfiShellProtocol != NULL) {
jcarseya405b862010-09-14 05:18:09 +0000650 if ((OpenMode & EFI_FILE_MODE_CREATE) == EFI_FILE_MODE_CREATE && (Attributes & EFI_FILE_DIRECTORY) == EFI_FILE_DIRECTORY) {
651 return ShellCreateDirectory(FileName, FileHandle);
652 }
jcarsey94b17fa2009-05-07 18:46:18 +0000653 //
654 // Use UEFI Shell 2.0 method
655 //
jcarsey366f81a2011-06-27 21:04:22 +0000656 Status = gEfiShellProtocol->OpenFileByName(FileName,
jcarseyb1f95a02009-06-16 00:23:19 +0000657 FileHandle,
658 OpenMode);
jcarseya405b862010-09-14 05:18:09 +0000659 if (StrCmp(FileName, L"NUL") != 0 && !EFI_ERROR(Status) && ((OpenMode & EFI_FILE_MODE_CREATE) != 0)){
jcarsey2247dde2009-11-09 18:08:58 +0000660 FileInfo = FileFunctionMap.GetFileInfo(*FileHandle);
jcarseyb1f95a02009-06-16 00:23:19 +0000661 ASSERT(FileInfo != NULL);
662 FileInfo->Attribute = Attributes;
jcarsey2247dde2009-11-09 18:08:58 +0000663 Status = FileFunctionMap.SetFileInfo(*FileHandle, FileInfo);
664 FreePool(FileInfo);
jcarseyb1f95a02009-06-16 00:23:19 +0000665 }
666 return (Status);
jcarsey1e6e84c2010-01-25 20:05:08 +0000667 }
jcarsey94b17fa2009-05-07 18:46:18 +0000668 //
669 // Using EFI Shell version
670 // this means convert name to path and call that function
671 // since this will use EFI method again that will open it.
672 //
673 ASSERT(mEfiShellEnvironment2 != NULL);
jcarseyb82bfcc2009-06-29 16:28:23 +0000674 FilePath = mEfiShellEnvironment2->NameToPath ((CHAR16*)FileName);
xdu290bfa222010-07-19 05:21:27 +0000675 if (FilePath != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000676 return (ShellOpenFileByDevicePath(&FilePath,
677 &DeviceHandle,
678 FileHandle,
679 OpenMode,
jcarseya405b862010-09-14 05:18:09 +0000680 Attributes));
jcarsey94b17fa2009-05-07 18:46:18 +0000681 }
682 return (EFI_DEVICE_ERROR);
683}
684/**
685 This function create a directory
686
jcarsey1e6e84c2010-01-25 20:05:08 +0000687 If return is EFI_SUCCESS, the Filehandle is the opened directory's handle;
688 otherwise, the Filehandle is NULL. If the directory already existed, this
jcarsey94b17fa2009-05-07 18:46:18 +0000689 function opens the existing directory.
690
691 @param DirectoryName pointer to directory name
692 @param FileHandle pointer to the file handle.
693
694 @retval EFI_SUCCESS The information was set.
695 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
jcarsey1e6e84c2010-01-25 20:05:08 +0000696 @retval EFI_UNSUPPORTED Could not open the file path.
697 @retval EFI_NOT_FOUND The specified file could not be found on the
698 device or the file system could not be found
jcarsey94b17fa2009-05-07 18:46:18 +0000699 on the device.
700 @retval EFI_NO_MEDIA The device has no medium.
jcarsey1e6e84c2010-01-25 20:05:08 +0000701 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
jcarsey94b17fa2009-05-07 18:46:18 +0000702 medium is no longer supported.
703 @retval EFI_DEVICE_ERROR The device reported an error.
704 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
705 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
706 @retval EFI_ACCESS_DENIED The file was opened read only.
jcarsey1e6e84c2010-01-25 20:05:08 +0000707 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
jcarsey94b17fa2009-05-07 18:46:18 +0000708 file.
709 @retval EFI_VOLUME_FULL The volume is full.
710 @sa ShellOpenFileByName
711**/
712EFI_STATUS
713EFIAPI
714ShellCreateDirectory(
jcarseyb82bfcc2009-06-29 16:28:23 +0000715 IN CONST CHAR16 *DirectoryName,
jcarseya405b862010-09-14 05:18:09 +0000716 OUT SHELL_FILE_HANDLE *FileHandle
717 )
718{
jcarsey366f81a2011-06-27 21:04:22 +0000719 if (gEfiShellProtocol != NULL) {
jcarsey2247dde2009-11-09 18:08:58 +0000720 //
721 // Use UEFI Shell 2.0 method
722 //
jcarsey366f81a2011-06-27 21:04:22 +0000723 return (gEfiShellProtocol->CreateFile(DirectoryName,
jcarsey2247dde2009-11-09 18:08:58 +0000724 EFI_FILE_DIRECTORY,
725 FileHandle
jcarseya405b862010-09-14 05:18:09 +0000726 ));
jcarsey2247dde2009-11-09 18:08:58 +0000727 } else {
728 return (ShellOpenFileByName(DirectoryName,
729 FileHandle,
730 EFI_FILE_MODE_READ | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE,
731 EFI_FILE_DIRECTORY
jcarseya405b862010-09-14 05:18:09 +0000732 ));
jcarsey2247dde2009-11-09 18:08:58 +0000733 }
jcarsey94b17fa2009-05-07 18:46:18 +0000734}
735
736/**
737 This function reads information from an opened file.
738
jcarsey1e6e84c2010-01-25 20:05:08 +0000739 If FileHandle is not a directory, the function reads the requested number of
740 bytes from the file at the file's current position and returns them in Buffer.
jcarsey94b17fa2009-05-07 18:46:18 +0000741 If the read goes beyond the end of the file, the read length is truncated to the
jcarsey1e6e84c2010-01-25 20:05:08 +0000742 end of the file. The file's current position is increased by the number of bytes
743 returned. If FileHandle is a directory, the function reads the directory entry
744 at the file's current position and returns the entry in Buffer. If the Buffer
745 is not large enough to hold the current directory entry, then
746 EFI_BUFFER_TOO_SMALL is returned and the current file position is not updated.
747 BufferSize is set to be the size of the buffer needed to read the entry. On
748 success, the current position is updated to the next directory entry. If there
749 are no more directory entries, the read returns a zero-length buffer.
jcarsey94b17fa2009-05-07 18:46:18 +0000750 EFI_FILE_INFO is the structure returned as the directory entry.
751
752 @param FileHandle the opened file handle
jcarsey1e6e84c2010-01-25 20:05:08 +0000753 @param BufferSize on input the size of buffer in bytes. on return
jcarsey94b17fa2009-05-07 18:46:18 +0000754 the number of bytes written.
755 @param Buffer the buffer to put read data into.
756
757 @retval EFI_SUCCESS Data was read.
758 @retval EFI_NO_MEDIA The device has no media.
759 @retval EFI_DEVICE_ERROR The device reported an error.
760 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
jcarsey1e6e84c2010-01-25 20:05:08 +0000761 @retval EFI_BUFFER_TO_SMALL Buffer is too small. ReadSize contains required
jcarsey94b17fa2009-05-07 18:46:18 +0000762 size.
763
764**/
765EFI_STATUS
766EFIAPI
767ShellReadFile(
jcarseya405b862010-09-14 05:18:09 +0000768 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000769 IN OUT UINTN *BufferSize,
770 OUT VOID *Buffer
jcarseya405b862010-09-14 05:18:09 +0000771 )
772{
jcarseyd2b45642009-05-11 18:02:16 +0000773 return (FileFunctionMap.ReadFile(FileHandle, BufferSize, Buffer));
jcarsey94b17fa2009-05-07 18:46:18 +0000774}
775
776
777/**
778 Write data to a file.
779
jcarsey1e6e84c2010-01-25 20:05:08 +0000780 This function writes the specified number of bytes to the file at the current
781 file position. The current file position is advanced the actual number of bytes
782 written, which is returned in BufferSize. Partial writes only occur when there
783 has been a data error during the write attempt (such as "volume space full").
784 The file is automatically grown to hold the data if required. Direct writes to
jcarsey94b17fa2009-05-07 18:46:18 +0000785 opened directories are not supported.
786
787 @param FileHandle The opened file for writing
788 @param BufferSize on input the number of bytes in Buffer. On output
789 the number of bytes written.
790 @param Buffer the buffer containing data to write is stored.
791
792 @retval EFI_SUCCESS Data was written.
793 @retval EFI_UNSUPPORTED Writes to an open directory are not supported.
794 @retval EFI_NO_MEDIA The device has no media.
795 @retval EFI_DEVICE_ERROR The device reported an error.
796 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
797 @retval EFI_WRITE_PROTECTED The device is write-protected.
798 @retval EFI_ACCESS_DENIED The file was open for read only.
799 @retval EFI_VOLUME_FULL The volume is full.
800**/
801EFI_STATUS
802EFIAPI
803ShellWriteFile(
jcarsey252d9452011-03-25 20:49:53 +0000804 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000805 IN OUT UINTN *BufferSize,
806 IN VOID *Buffer
jcarseya405b862010-09-14 05:18:09 +0000807 )
808{
jcarseyd2b45642009-05-11 18:02:16 +0000809 return (FileFunctionMap.WriteFile(FileHandle, BufferSize, Buffer));
jcarsey94b17fa2009-05-07 18:46:18 +0000810}
811
jcarsey1e6e84c2010-01-25 20:05:08 +0000812/**
jcarsey94b17fa2009-05-07 18:46:18 +0000813 Close an open file handle.
814
jcarsey1e6e84c2010-01-25 20:05:08 +0000815 This function closes a specified file handle. All "dirty" cached file data is
816 flushed to the device, and the file is closed. In all cases the handle is
jcarsey94b17fa2009-05-07 18:46:18 +0000817 closed.
818
819@param FileHandle the file handle to close.
820
821@retval EFI_SUCCESS the file handle was closed sucessfully.
822**/
823EFI_STATUS
824EFIAPI
825ShellCloseFile (
jcarseya405b862010-09-14 05:18:09 +0000826 IN SHELL_FILE_HANDLE *FileHandle
827 )
828{
jcarseyd2b45642009-05-11 18:02:16 +0000829 return (FileFunctionMap.CloseFile(*FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000830}
831
832/**
833 Delete a file and close the handle
834
835 This function closes and deletes a file. In all cases the file handle is closed.
jcarsey1e6e84c2010-01-25 20:05:08 +0000836 If the file cannot be deleted, the warning code EFI_WARN_DELETE_FAILURE is
jcarsey94b17fa2009-05-07 18:46:18 +0000837 returned, but the handle is still closed.
838
839 @param FileHandle the file handle to delete
840
841 @retval EFI_SUCCESS the file was closed sucessfully
jcarsey1e6e84c2010-01-25 20:05:08 +0000842 @retval EFI_WARN_DELETE_FAILURE the handle was closed, but the file was not
jcarsey94b17fa2009-05-07 18:46:18 +0000843 deleted
844 @retval INVALID_PARAMETER One of the parameters has an invalid value.
845**/
846EFI_STATUS
847EFIAPI
848ShellDeleteFile (
jcarseya405b862010-09-14 05:18:09 +0000849 IN SHELL_FILE_HANDLE *FileHandle
850 )
851{
jcarseyd2b45642009-05-11 18:02:16 +0000852 return (FileFunctionMap.DeleteFile(*FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000853}
854
855/**
856 Set the current position in a file.
857
jcarsey1e6e84c2010-01-25 20:05:08 +0000858 This function sets the current file position for the handle to the position
jcarsey94b17fa2009-05-07 18:46:18 +0000859 supplied. With the exception of seeking to position 0xFFFFFFFFFFFFFFFF, only
jcarsey1e6e84c2010-01-25 20:05:08 +0000860 absolute positioning is supported, and seeking past the end of the file is
861 allowed (a subsequent write would grow the file). Seeking to position
jcarsey94b17fa2009-05-07 18:46:18 +0000862 0xFFFFFFFFFFFFFFFF causes the current position to be set to the end of the file.
jcarsey1e6e84c2010-01-25 20:05:08 +0000863 If FileHandle is a directory, the only position that may be set is zero. This
jcarsey94b17fa2009-05-07 18:46:18 +0000864 has the effect of starting the read process of the directory entries over.
865
866 @param FileHandle The file handle on which the position is being set
867 @param Position Byte position from begining of file
868
869 @retval EFI_SUCCESS Operation completed sucessfully.
jcarsey1e6e84c2010-01-25 20:05:08 +0000870 @retval EFI_UNSUPPORTED the seek request for non-zero is not valid on
jcarsey94b17fa2009-05-07 18:46:18 +0000871 directories.
872 @retval INVALID_PARAMETER One of the parameters has an invalid value.
873**/
874EFI_STATUS
875EFIAPI
876ShellSetFilePosition (
jcarseya405b862010-09-14 05:18:09 +0000877 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000878 IN UINT64 Position
jcarseya405b862010-09-14 05:18:09 +0000879 )
880{
jcarseyd2b45642009-05-11 18:02:16 +0000881 return (FileFunctionMap.SetFilePosition(FileHandle, Position));
jcarsey94b17fa2009-05-07 18:46:18 +0000882}
883
jcarsey1e6e84c2010-01-25 20:05:08 +0000884/**
jcarsey94b17fa2009-05-07 18:46:18 +0000885 Gets a file's current position
886
jcarsey1e6e84c2010-01-25 20:05:08 +0000887 This function retrieves the current file position for the file handle. For
888 directories, the current file position has no meaning outside of the file
jcarsey94b17fa2009-05-07 18:46:18 +0000889 system driver and as such the operation is not supported. An error is returned
890 if FileHandle is a directory.
891
892 @param FileHandle The open file handle on which to get the position.
893 @param Position Byte position from begining of file.
894
895 @retval EFI_SUCCESS the operation completed sucessfully.
896 @retval INVALID_PARAMETER One of the parameters has an invalid value.
897 @retval EFI_UNSUPPORTED the request is not valid on directories.
898**/
899EFI_STATUS
900EFIAPI
901ShellGetFilePosition (
jcarseya405b862010-09-14 05:18:09 +0000902 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000903 OUT UINT64 *Position
jcarseya405b862010-09-14 05:18:09 +0000904 )
905{
jcarseyd2b45642009-05-11 18:02:16 +0000906 return (FileFunctionMap.GetFilePosition(FileHandle, Position));
jcarsey94b17fa2009-05-07 18:46:18 +0000907}
908/**
909 Flushes data on a file
jcarsey1e6e84c2010-01-25 20:05:08 +0000910
jcarsey94b17fa2009-05-07 18:46:18 +0000911 This function flushes all modified data associated with a file to a device.
912
913 @param FileHandle The file handle on which to flush data
914
915 @retval EFI_SUCCESS The data was flushed.
916 @retval EFI_NO_MEDIA The device has no media.
917 @retval EFI_DEVICE_ERROR The device reported an error.
918 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
919 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
920 @retval EFI_ACCESS_DENIED The file was opened for read only.
921**/
922EFI_STATUS
923EFIAPI
924ShellFlushFile (
jcarseya405b862010-09-14 05:18:09 +0000925 IN SHELL_FILE_HANDLE FileHandle
926 )
927{
jcarseyd2b45642009-05-11 18:02:16 +0000928 return (FileFunctionMap.FlushFile(FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000929}
930
931/**
932 Retrieves the first file from a directory
933
jcarsey1e6e84c2010-01-25 20:05:08 +0000934 This function opens a directory and gets the first file's info in the
935 directory. Caller can use ShellFindNextFile() to get other files. When
jcarsey94b17fa2009-05-07 18:46:18 +0000936 complete the caller is responsible for calling FreePool() on Buffer.
937
938 @param DirHandle The file handle of the directory to search
939 @param Buffer Pointer to buffer for file's information
940
941 @retval EFI_SUCCESS Found the first file.
942 @retval EFI_NOT_FOUND Cannot find the directory.
943 @retval EFI_NO_MEDIA The device has no media.
944 @retval EFI_DEVICE_ERROR The device reported an error.
945 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
946 @return Others status of ShellGetFileInfo, ShellSetFilePosition,
947 or ShellReadFile
948**/
949EFI_STATUS
950EFIAPI
951ShellFindFirstFile (
jcarseya405b862010-09-14 05:18:09 +0000952 IN SHELL_FILE_HANDLE DirHandle,
jcarseyd2b45642009-05-11 18:02:16 +0000953 OUT EFI_FILE_INFO **Buffer
jcarseya405b862010-09-14 05:18:09 +0000954 )
955{
jcarsey94b17fa2009-05-07 18:46:18 +0000956 //
jcarseyd2b45642009-05-11 18:02:16 +0000957 // pass to file handle lib
jcarsey94b17fa2009-05-07 18:46:18 +0000958 //
jcarseyd2b45642009-05-11 18:02:16 +0000959 return (FileHandleFindFirstFile(DirHandle, Buffer));
jcarsey94b17fa2009-05-07 18:46:18 +0000960}
961/**
962 Retrieves the next file in a directory.
963
jcarseya405b862010-09-14 05:18:09 +0000964 To use this function, caller must call the ShellFindFirstFile() to get the
jcarsey1e6e84c2010-01-25 20:05:08 +0000965 first file, and then use this function get other files. This function can be
966 called for several times to get each file's information in the directory. If
967 the call of ShellFindNextFile() got the last file in the directory, the next
968 call of this function has no file to get. *NoFile will be set to TRUE and the
969 Buffer memory will be automatically freed.
jcarsey94b17fa2009-05-07 18:46:18 +0000970
971 @param DirHandle the file handle of the directory
972 @param Buffer pointer to buffer for file's information
973 @param NoFile pointer to boolean when last file is found
974
975 @retval EFI_SUCCESS Found the next file, or reached last file
976 @retval EFI_NO_MEDIA The device has no media.
977 @retval EFI_DEVICE_ERROR The device reported an error.
978 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
979**/
980EFI_STATUS
981EFIAPI
982ShellFindNextFile(
jcarseya405b862010-09-14 05:18:09 +0000983 IN SHELL_FILE_HANDLE DirHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000984 OUT EFI_FILE_INFO *Buffer,
985 OUT BOOLEAN *NoFile
jcarseya405b862010-09-14 05:18:09 +0000986 )
987{
jcarsey94b17fa2009-05-07 18:46:18 +0000988 //
jcarseyd2b45642009-05-11 18:02:16 +0000989 // pass to file handle lib
jcarsey94b17fa2009-05-07 18:46:18 +0000990 //
jcarseyd2b45642009-05-11 18:02:16 +0000991 return (FileHandleFindNextFile(DirHandle, Buffer, NoFile));
jcarsey94b17fa2009-05-07 18:46:18 +0000992}
993/**
994 Retrieve the size of a file.
995
996 if FileHandle is NULL then ASSERT()
997 if Size is NULL then ASSERT()
998
jcarsey1e6e84c2010-01-25 20:05:08 +0000999 This function extracts the file size info from the FileHandle's EFI_FILE_INFO
jcarsey94b17fa2009-05-07 18:46:18 +00001000 data.
1001
1002 @param FileHandle file handle from which size is retrieved
1003 @param Size pointer to size
1004
1005 @retval EFI_SUCCESS operation was completed sucessfully
1006 @retval EFI_DEVICE_ERROR cannot access the file
1007**/
1008EFI_STATUS
1009EFIAPI
1010ShellGetFileSize (
jcarseya405b862010-09-14 05:18:09 +00001011 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +00001012 OUT UINT64 *Size
jcarseya405b862010-09-14 05:18:09 +00001013 )
1014{
jcarseyd2b45642009-05-11 18:02:16 +00001015 return (FileFunctionMap.GetFileSize(FileHandle, Size));
jcarsey94b17fa2009-05-07 18:46:18 +00001016}
1017/**
1018 Retrieves the status of the break execution flag
1019
1020 this function is useful to check whether the application is being asked to halt by the shell.
1021
1022 @retval TRUE the execution break is enabled
1023 @retval FALSE the execution break is not enabled
1024**/
1025BOOLEAN
1026EFIAPI
1027ShellGetExecutionBreakFlag(
1028 VOID
1029 )
1030{
jcarsey1e6e84c2010-01-25 20:05:08 +00001031 //
jcarsey94b17fa2009-05-07 18:46:18 +00001032 // Check for UEFI Shell 2.0 protocols
1033 //
jcarsey366f81a2011-06-27 21:04:22 +00001034 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001035
1036 //
1037 // We are using UEFI Shell 2.0; see if the event has been triggered
1038 //
jcarsey366f81a2011-06-27 21:04:22 +00001039 if (gBS->CheckEvent(gEfiShellProtocol->ExecutionBreak) != EFI_SUCCESS) {
jcarsey94b17fa2009-05-07 18:46:18 +00001040 return (FALSE);
1041 }
1042 return (TRUE);
jcarsey1e6e84c2010-01-25 20:05:08 +00001043 }
jcarsey94b17fa2009-05-07 18:46:18 +00001044
1045 //
1046 // using EFI Shell; call the function to check
1047 //
jcarsey92a54472011-06-27 20:33:13 +00001048 if (mEfiShellEnvironment2 != NULL) {
1049 return (mEfiShellEnvironment2->GetExecutionBreak());
1050 }
1051
1052 return (FALSE);
jcarsey94b17fa2009-05-07 18:46:18 +00001053}
1054/**
1055 return the value of an environment variable
1056
jcarsey1e6e84c2010-01-25 20:05:08 +00001057 this function gets the value of the environment variable set by the
jcarsey94b17fa2009-05-07 18:46:18 +00001058 ShellSetEnvironmentVariable function
1059
1060 @param EnvKey The key name of the environment variable.
1061
1062 @retval NULL the named environment variable does not exist.
1063 @return != NULL pointer to the value of the environment variable
1064**/
1065CONST CHAR16*
1066EFIAPI
1067ShellGetEnvironmentVariable (
jcarsey9b3bf082009-06-23 21:15:07 +00001068 IN CONST CHAR16 *EnvKey
jcarsey94b17fa2009-05-07 18:46:18 +00001069 )
1070{
jcarsey1e6e84c2010-01-25 20:05:08 +00001071 //
jcarsey94b17fa2009-05-07 18:46:18 +00001072 // Check for UEFI Shell 2.0 protocols
1073 //
jcarsey366f81a2011-06-27 21:04:22 +00001074 if (gEfiShellProtocol != NULL) {
1075 return (gEfiShellProtocol->GetEnv(EnvKey));
jcarsey94b17fa2009-05-07 18:46:18 +00001076 }
1077
1078 //
jcarsey92a54472011-06-27 20:33:13 +00001079 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001080 //
jcarsey92a54472011-06-27 20:33:13 +00001081 if (mEfiShellEnvironment2 != NULL) {
1082 return (mEfiShellEnvironment2->GetEnv((CHAR16*)EnvKey));
1083 }
jcarsey94b17fa2009-05-07 18:46:18 +00001084
jcarsey92a54472011-06-27 20:33:13 +00001085 return NULL;
jcarsey94b17fa2009-05-07 18:46:18 +00001086}
1087/**
1088 set the value of an environment variable
1089
1090This function changes the current value of the specified environment variable. If the
1091environment variable exists and the Value is an empty string, then the environment
1092variable is deleted. If the environment variable exists and the Value is not an empty
1093string, then the value of the environment variable is changed. If the environment
1094variable does not exist and the Value is an empty string, there is no action. If the
1095environment variable does not exist and the Value is a non-empty string, then the
1096environment variable is created and assigned the specified value.
1097
1098 This is not supported pre-UEFI Shell 2.0.
1099
1100 @param EnvKey The key name of the environment variable.
1101 @param EnvVal The Value of the environment variable
1102 @param Volatile Indicates whether the variable is non-volatile (FALSE) or volatile (TRUE).
1103
1104 @retval EFI_SUCCESS the operation was completed sucessfully
1105 @retval EFI_UNSUPPORTED This operation is not allowed in pre UEFI 2.0 Shell environments
1106**/
1107EFI_STATUS
1108EFIAPI
1109ShellSetEnvironmentVariable (
1110 IN CONST CHAR16 *EnvKey,
1111 IN CONST CHAR16 *EnvVal,
1112 IN BOOLEAN Volatile
1113 )
1114{
jcarsey1e6e84c2010-01-25 20:05:08 +00001115 //
jcarsey94b17fa2009-05-07 18:46:18 +00001116 // Check for UEFI Shell 2.0 protocols
1117 //
jcarsey366f81a2011-06-27 21:04:22 +00001118 if (gEfiShellProtocol != NULL) {
1119 return (gEfiShellProtocol->SetEnv(EnvKey, EnvVal, Volatile));
jcarsey1e6e84c2010-01-25 20:05:08 +00001120 }
jcarsey94b17fa2009-05-07 18:46:18 +00001121
1122 //
1123 // This feature does not exist under EFI shell
1124 //
1125 return (EFI_UNSUPPORTED);
1126}
jcarseya405b862010-09-14 05:18:09 +00001127
jcarsey94b17fa2009-05-07 18:46:18 +00001128/**
jcarseya405b862010-09-14 05:18:09 +00001129 Cause the shell to parse and execute a command line.
jcarsey94b17fa2009-05-07 18:46:18 +00001130
1131 This function creates a nested instance of the shell and executes the specified
jcarseya405b862010-09-14 05:18:09 +00001132 command (CommandLine) with the specified environment (Environment). Upon return,
1133 the status code returned by the specified command is placed in StatusCode.
1134 If Environment is NULL, then the current environment is used and all changes made
1135 by the commands executed will be reflected in the current environment. If the
1136 Environment is non-NULL, then the changes made will be discarded.
1137 The CommandLine is executed from the current working directory on the current
1138 device.
jcarsey94b17fa2009-05-07 18:46:18 +00001139
jcarseya405b862010-09-14 05:18:09 +00001140 The EnvironmentVariables and Status parameters are ignored in a pre-UEFI Shell 2.0
1141 environment. The values pointed to by the parameters will be unchanged by the
1142 ShellExecute() function. The Output parameter has no effect in a
1143 UEFI Shell 2.0 environment.
jcarsey94b17fa2009-05-07 18:46:18 +00001144
jcarseya405b862010-09-14 05:18:09 +00001145 @param[in] ParentHandle The parent image starting the operation.
1146 @param[in] CommandLine The pointer to a NULL terminated command line.
1147 @param[in] Output True to display debug output. False to hide it.
1148 @param[in] EnvironmentVariables Optional pointer to array of environment variables
1149 in the form "x=y". If NULL, the current set is used.
1150 @param[out] Status The status of the run command line.
jcarsey94b17fa2009-05-07 18:46:18 +00001151
jcarseya405b862010-09-14 05:18:09 +00001152 @retval EFI_SUCCESS The operation completed sucessfully. Status
1153 contains the status code returned.
1154 @retval EFI_INVALID_PARAMETER A parameter contains an invalid value.
1155 @retval EFI_OUT_OF_RESOURCES Out of resources.
1156 @retval EFI_UNSUPPORTED The operation is not allowed.
jcarsey94b17fa2009-05-07 18:46:18 +00001157**/
1158EFI_STATUS
1159EFIAPI
1160ShellExecute (
1161 IN EFI_HANDLE *ParentHandle,
1162 IN CHAR16 *CommandLine OPTIONAL,
1163 IN BOOLEAN Output OPTIONAL,
1164 IN CHAR16 **EnvironmentVariables OPTIONAL,
1165 OUT EFI_STATUS *Status OPTIONAL
1166 )
1167{
jcarsey1e6e84c2010-01-25 20:05:08 +00001168 //
jcarsey94b17fa2009-05-07 18:46:18 +00001169 // Check for UEFI Shell 2.0 protocols
1170 //
jcarsey366f81a2011-06-27 21:04:22 +00001171 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001172 //
1173 // Call UEFI Shell 2.0 version (not using Output parameter)
1174 //
jcarsey366f81a2011-06-27 21:04:22 +00001175 return (gEfiShellProtocol->Execute(ParentHandle,
jcarsey94b17fa2009-05-07 18:46:18 +00001176 CommandLine,
1177 EnvironmentVariables,
1178 Status));
jcarsey1e6e84c2010-01-25 20:05:08 +00001179 }
jcarsey92a54472011-06-27 20:33:13 +00001180
jcarsey94b17fa2009-05-07 18:46:18 +00001181 //
jcarsey92a54472011-06-27 20:33:13 +00001182 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001183 //
jcarsey92a54472011-06-27 20:33:13 +00001184 if (mEfiShellEnvironment2 != NULL) {
1185 //
1186 // Call EFI Shell version (not using EnvironmentVariables or Status parameters)
1187 // Due to oddity in the EFI shell we want to dereference the ParentHandle here
1188 //
1189 return (mEfiShellEnvironment2->Execute(*ParentHandle,
1190 CommandLine,
1191 Output));
1192 }
1193
1194 return (EFI_UNSUPPORTED);
jcarsey94b17fa2009-05-07 18:46:18 +00001195}
1196/**
1197 Retreives the current directory path
1198
jcarsey1e6e84c2010-01-25 20:05:08 +00001199 If the DeviceName is NULL, it returns the current device's current directory
1200 name. If the DeviceName is not NULL, it returns the current directory name
jcarsey94b17fa2009-05-07 18:46:18 +00001201 on specified drive.
1202
1203 @param DeviceName the name of the drive to get directory on
1204
1205 @retval NULL the directory does not exist
1206 @return != NULL the directory
1207**/
1208CONST CHAR16*
1209EFIAPI
1210ShellGetCurrentDir (
jcarseya405b862010-09-14 05:18:09 +00001211 IN CHAR16 * CONST DeviceName OPTIONAL
jcarsey94b17fa2009-05-07 18:46:18 +00001212 )
1213{
jcarsey1e6e84c2010-01-25 20:05:08 +00001214 //
jcarsey94b17fa2009-05-07 18:46:18 +00001215 // Check for UEFI Shell 2.0 protocols
1216 //
jcarsey366f81a2011-06-27 21:04:22 +00001217 if (gEfiShellProtocol != NULL) {
1218 return (gEfiShellProtocol->GetCurDir(DeviceName));
jcarsey1e6e84c2010-01-25 20:05:08 +00001219 }
jcarsey92a54472011-06-27 20:33:13 +00001220
jcarsey94b17fa2009-05-07 18:46:18 +00001221 //
jcarsey8bd282b2011-06-27 16:45:41 +00001222 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001223 //
jcarsey8bd282b2011-06-27 16:45:41 +00001224 if (mEfiShellEnvironment2 != NULL) {
1225 return (mEfiShellEnvironment2->CurDir(DeviceName));
1226 }
1227
1228 return (NULL);
jcarsey94b17fa2009-05-07 18:46:18 +00001229}
1230/**
1231 sets (enabled or disabled) the page break mode
1232
jcarsey1e6e84c2010-01-25 20:05:08 +00001233 when page break mode is enabled the screen will stop scrolling
jcarsey94b17fa2009-05-07 18:46:18 +00001234 and wait for operator input before scrolling a subsequent screen.
1235
1236 @param CurrentState TRUE to enable and FALSE to disable
1237**/
jcarsey1e6e84c2010-01-25 20:05:08 +00001238VOID
jcarsey94b17fa2009-05-07 18:46:18 +00001239EFIAPI
1240ShellSetPageBreakMode (
1241 IN BOOLEAN CurrentState
1242 )
1243{
1244 //
1245 // check for enabling
1246 //
1247 if (CurrentState != 0x00) {
jcarsey1e6e84c2010-01-25 20:05:08 +00001248 //
jcarsey94b17fa2009-05-07 18:46:18 +00001249 // check for UEFI Shell 2.0
1250 //
jcarsey366f81a2011-06-27 21:04:22 +00001251 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001252 //
1253 // Enable with UEFI 2.0 Shell
1254 //
jcarsey366f81a2011-06-27 21:04:22 +00001255 gEfiShellProtocol->EnablePageBreak();
jcarsey94b17fa2009-05-07 18:46:18 +00001256 return;
1257 } else {
jcarsey1e6e84c2010-01-25 20:05:08 +00001258 //
jcarsey92a54472011-06-27 20:33:13 +00001259 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001260 //
jcarsey92a54472011-06-27 20:33:13 +00001261 if (mEfiShellEnvironment2 != NULL) {
1262 //
1263 // Enable with EFI Shell
1264 //
1265 mEfiShellEnvironment2->EnablePageBreak (DEFAULT_INIT_ROW, DEFAULT_AUTO_LF);
1266 return;
1267 }
jcarsey94b17fa2009-05-07 18:46:18 +00001268 }
1269 } else {
jcarsey1e6e84c2010-01-25 20:05:08 +00001270 //
jcarsey94b17fa2009-05-07 18:46:18 +00001271 // check for UEFI Shell 2.0
1272 //
jcarsey366f81a2011-06-27 21:04:22 +00001273 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001274 //
1275 // Disable with UEFI 2.0 Shell
1276 //
jcarsey366f81a2011-06-27 21:04:22 +00001277 gEfiShellProtocol->DisablePageBreak();
jcarsey94b17fa2009-05-07 18:46:18 +00001278 return;
1279 } else {
jcarsey1e6e84c2010-01-25 20:05:08 +00001280 //
jcarsey92a54472011-06-27 20:33:13 +00001281 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001282 //
jcarsey92a54472011-06-27 20:33:13 +00001283 if (mEfiShellEnvironment2 != NULL) {
1284 //
1285 // Disable with EFI Shell
1286 //
1287 mEfiShellEnvironment2->DisablePageBreak ();
1288 return;
1289 }
jcarsey94b17fa2009-05-07 18:46:18 +00001290 }
1291 }
1292}
1293
1294///
1295/// version of EFI_SHELL_FILE_INFO struct, except has no CONST pointers.
1296/// This allows for the struct to be populated.
1297///
1298typedef struct {
jcarseyd2b45642009-05-11 18:02:16 +00001299 LIST_ENTRY Link;
jcarsey94b17fa2009-05-07 18:46:18 +00001300 EFI_STATUS Status;
1301 CHAR16 *FullName;
1302 CHAR16 *FileName;
jcarseya405b862010-09-14 05:18:09 +00001303 SHELL_FILE_HANDLE Handle;
jcarsey94b17fa2009-05-07 18:46:18 +00001304 EFI_FILE_INFO *Info;
1305} EFI_SHELL_FILE_INFO_NO_CONST;
1306
1307/**
1308 Converts a EFI shell list of structures to the coresponding UEFI Shell 2.0 type of list.
1309
1310 if OldStyleFileList is NULL then ASSERT()
1311
jcarsey1e6e84c2010-01-25 20:05:08 +00001312 this function will convert a SHELL_FILE_ARG based list into a callee allocated
jcarsey94b17fa2009-05-07 18:46:18 +00001313 EFI_SHELL_FILE_INFO based list. it is up to the caller to free the memory via
1314 the ShellCloseFileMetaArg function.
1315
ydong104ff7e372011-09-02 08:05:34 +00001316 @param[in] FileList the EFI shell list type
1317 @param[in, out] ListHead the list to add to
jcarsey94b17fa2009-05-07 18:46:18 +00001318
1319 @retval the resultant head of the double linked new format list;
1320**/
1321LIST_ENTRY*
1322EFIAPI
1323InternalShellConvertFileListType (
jcarsey9b3bf082009-06-23 21:15:07 +00001324 IN LIST_ENTRY *FileList,
1325 IN OUT LIST_ENTRY *ListHead
jcarsey125c2cf2009-11-18 21:36:50 +00001326 )
1327{
jcarsey94b17fa2009-05-07 18:46:18 +00001328 SHELL_FILE_ARG *OldInfo;
jcarsey9b3bf082009-06-23 21:15:07 +00001329 LIST_ENTRY *Link;
jcarsey94b17fa2009-05-07 18:46:18 +00001330 EFI_SHELL_FILE_INFO_NO_CONST *NewInfo;
1331
1332 //
jcarsey9b3bf082009-06-23 21:15:07 +00001333 // ASSERTs
jcarsey94b17fa2009-05-07 18:46:18 +00001334 //
jcarsey9b3bf082009-06-23 21:15:07 +00001335 ASSERT(FileList != NULL);
1336 ASSERT(ListHead != NULL);
jcarsey94b17fa2009-05-07 18:46:18 +00001337
1338 //
1339 // enumerate through each member of the old list and copy
1340 //
jcarseyd2b45642009-05-11 18:02:16 +00001341 for (Link = FileList->ForwardLink; Link != FileList; Link = Link->ForwardLink) {
jcarsey94b17fa2009-05-07 18:46:18 +00001342 OldInfo = CR (Link, SHELL_FILE_ARG, Link, SHELL_FILE_ARG_SIGNATURE);
jcarseya405b862010-09-14 05:18:09 +00001343 ASSERT(OldInfo != NULL);
1344
1345 //
1346 // Skip ones that failed to open...
1347 //
1348 if (OldInfo->Status != EFI_SUCCESS) {
1349 continue;
1350 }
jcarsey94b17fa2009-05-07 18:46:18 +00001351
1352 //
1353 // make sure the old list was valid
1354 //
jcarsey94b17fa2009-05-07 18:46:18 +00001355 ASSERT(OldInfo->Info != NULL);
1356 ASSERT(OldInfo->FullName != NULL);
1357 ASSERT(OldInfo->FileName != NULL);
1358
1359 //
1360 // allocate a new EFI_SHELL_FILE_INFO object
1361 //
1362 NewInfo = AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
jcarseyc9d92df2010-02-03 15:37:54 +00001363 ASSERT(NewInfo != NULL);
1364 if (NewInfo == NULL) {
1365 break;
1366 }
jcarsey1e6e84c2010-01-25 20:05:08 +00001367
1368 //
jcarsey94b17fa2009-05-07 18:46:18 +00001369 // copy the simple items
1370 //
1371 NewInfo->Handle = OldInfo->Handle;
1372 NewInfo->Status = OldInfo->Status;
1373
jcarseyd2b45642009-05-11 18:02:16 +00001374 // old shell checks for 0 not NULL
1375 OldInfo->Handle = 0;
1376
jcarsey94b17fa2009-05-07 18:46:18 +00001377 //
1378 // allocate new space to copy strings and structure
1379 //
1380 NewInfo->FullName = AllocateZeroPool(StrSize(OldInfo->FullName));
1381 NewInfo->FileName = AllocateZeroPool(StrSize(OldInfo->FileName));
1382 NewInfo->Info = AllocateZeroPool((UINTN)OldInfo->Info->Size);
jcarsey1e6e84c2010-01-25 20:05:08 +00001383
jcarsey94b17fa2009-05-07 18:46:18 +00001384 //
1385 // make sure all the memory allocations were sucessful
1386 //
1387 ASSERT(NewInfo->FullName != NULL);
1388 ASSERT(NewInfo->FileName != NULL);
1389 ASSERT(NewInfo->Info != NULL);
1390
1391 //
1392 // Copt the strings and structure
1393 //
1394 StrCpy(NewInfo->FullName, OldInfo->FullName);
1395 StrCpy(NewInfo->FileName, OldInfo->FileName);
1396 gBS->CopyMem (NewInfo->Info, OldInfo->Info, (UINTN)OldInfo->Info->Size);
1397
1398 //
1399 // add that to the list
1400 //
jcarsey9b3bf082009-06-23 21:15:07 +00001401 InsertTailList(ListHead, &NewInfo->Link);
jcarsey94b17fa2009-05-07 18:46:18 +00001402 }
1403 return (ListHead);
1404}
1405/**
1406 Opens a group of files based on a path.
1407
jcarsey1e6e84c2010-01-25 20:05:08 +00001408 This function uses the Arg to open all the matching files. Each matched
1409 file has a SHELL_FILE_ARG structure to record the file information. These
1410 structures are placed on the list ListHead. Users can get the SHELL_FILE_ARG
jcarsey94b17fa2009-05-07 18:46:18 +00001411 structures from ListHead to access each file. This function supports wildcards
jcarsey1e6e84c2010-01-25 20:05:08 +00001412 and will process '?' and '*' as such. the list must be freed with a call to
jcarsey94b17fa2009-05-07 18:46:18 +00001413 ShellCloseFileMetaArg().
1414
jcarsey1e6e84c2010-01-25 20:05:08 +00001415 If you are NOT appending to an existing list *ListHead must be NULL. If
jcarsey5f7431d2009-07-10 18:06:01 +00001416 *ListHead is NULL then it must be callee freed.
jcarsey94b17fa2009-05-07 18:46:18 +00001417
1418 @param Arg pointer to path string
1419 @param OpenMode mode to open files with
1420 @param ListHead head of linked list of results
1421
jcarsey1e6e84c2010-01-25 20:05:08 +00001422 @retval EFI_SUCCESS the operation was sucessful and the list head
jcarsey94b17fa2009-05-07 18:46:18 +00001423 contains the list of opened files
jcarsey94b17fa2009-05-07 18:46:18 +00001424 @return != EFI_SUCCESS the operation failed
1425
1426 @sa InternalShellConvertFileListType
1427**/
1428EFI_STATUS
1429EFIAPI
1430ShellOpenFileMetaArg (
1431 IN CHAR16 *Arg,
1432 IN UINT64 OpenMode,
1433 IN OUT EFI_SHELL_FILE_INFO **ListHead
1434 )
1435{
1436 EFI_STATUS Status;
jcarsey9b3bf082009-06-23 21:15:07 +00001437 LIST_ENTRY mOldStyleFileList;
jcarsey1e6e84c2010-01-25 20:05:08 +00001438
jcarsey94b17fa2009-05-07 18:46:18 +00001439 //
1440 // ASSERT that Arg and ListHead are not NULL
1441 //
1442 ASSERT(Arg != NULL);
1443 ASSERT(ListHead != NULL);
1444
jcarsey1e6e84c2010-01-25 20:05:08 +00001445 //
jcarsey94b17fa2009-05-07 18:46:18 +00001446 // Check for UEFI Shell 2.0 protocols
1447 //
jcarsey366f81a2011-06-27 21:04:22 +00001448 if (gEfiShellProtocol != NULL) {
jcarsey5f7431d2009-07-10 18:06:01 +00001449 if (*ListHead == NULL) {
1450 *ListHead = (EFI_SHELL_FILE_INFO*)AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1451 if (*ListHead == NULL) {
1452 return (EFI_OUT_OF_RESOURCES);
1453 }
1454 InitializeListHead(&((*ListHead)->Link));
jcarsey1e6e84c2010-01-25 20:05:08 +00001455 }
jcarsey366f81a2011-06-27 21:04:22 +00001456 Status = gEfiShellProtocol->OpenFileList(Arg,
jcarsey1e6e84c2010-01-25 20:05:08 +00001457 OpenMode,
jcarsey2247dde2009-11-09 18:08:58 +00001458 ListHead);
1459 if (EFI_ERROR(Status)) {
jcarsey366f81a2011-06-27 21:04:22 +00001460 gEfiShellProtocol->RemoveDupInFileList(ListHead);
jcarsey2247dde2009-11-09 18:08:58 +00001461 } else {
jcarsey366f81a2011-06-27 21:04:22 +00001462 Status = gEfiShellProtocol->RemoveDupInFileList(ListHead);
jcarsey2247dde2009-11-09 18:08:58 +00001463 }
jcarseya405b862010-09-14 05:18:09 +00001464 if (*ListHead != NULL && IsListEmpty(&(*ListHead)->Link)) {
1465 FreePool(*ListHead);
1466 *ListHead = NULL;
1467 return (EFI_NOT_FOUND);
1468 }
jcarsey2247dde2009-11-09 18:08:58 +00001469 return (Status);
jcarsey1e6e84c2010-01-25 20:05:08 +00001470 }
jcarsey94b17fa2009-05-07 18:46:18 +00001471
1472 //
jcarsey92a54472011-06-27 20:33:13 +00001473 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001474 //
jcarsey92a54472011-06-27 20:33:13 +00001475 if (mEfiShellEnvironment2 != NULL) {
1476 //
1477 // make sure the list head is initialized
1478 //
1479 InitializeListHead(&mOldStyleFileList);
jcarsey94b17fa2009-05-07 18:46:18 +00001480
jcarsey92a54472011-06-27 20:33:13 +00001481 //
1482 // Get the EFI Shell list of files
1483 //
1484 Status = mEfiShellEnvironment2->FileMetaArg(Arg, &mOldStyleFileList);
1485 if (EFI_ERROR(Status)) {
1486 *ListHead = NULL;
1487 return (Status);
1488 }
jcarsey94b17fa2009-05-07 18:46:18 +00001489
jcarsey92a54472011-06-27 20:33:13 +00001490 if (*ListHead == NULL) {
1491 *ListHead = (EFI_SHELL_FILE_INFO *)AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1492 if (*ListHead == NULL) {
1493 return (EFI_OUT_OF_RESOURCES);
1494 }
1495 InitializeListHead(&((*ListHead)->Link));
1496 }
1497
1498 //
1499 // Convert that to equivalent of UEFI Shell 2.0 structure
1500 //
1501 InternalShellConvertFileListType(&mOldStyleFileList, &(*ListHead)->Link);
1502
1503 //
1504 // Free the EFI Shell version that was converted.
1505 //
1506 mEfiShellEnvironment2->FreeFileList(&mOldStyleFileList);
1507
1508 if ((*ListHead)->Link.ForwardLink == (*ListHead)->Link.BackLink && (*ListHead)->Link.BackLink == &((*ListHead)->Link)) {
1509 FreePool(*ListHead);
1510 *ListHead = NULL;
1511 Status = EFI_NOT_FOUND;
1512 }
jcarsey94b17fa2009-05-07 18:46:18 +00001513 return (Status);
1514 }
1515
jcarsey92a54472011-06-27 20:33:13 +00001516 return (EFI_UNSUPPORTED);
jcarsey94b17fa2009-05-07 18:46:18 +00001517}
1518/**
jcarseya405b862010-09-14 05:18:09 +00001519 Free the linked list returned from ShellOpenFileMetaArg.
jcarsey94b17fa2009-05-07 18:46:18 +00001520
jcarseya405b862010-09-14 05:18:09 +00001521 if ListHead is NULL then ASSERT().
jcarsey94b17fa2009-05-07 18:46:18 +00001522
jcarseya405b862010-09-14 05:18:09 +00001523 @param ListHead the pointer to free.
jcarsey94b17fa2009-05-07 18:46:18 +00001524
jcarseya405b862010-09-14 05:18:09 +00001525 @retval EFI_SUCCESS the operation was sucessful.
jcarsey94b17fa2009-05-07 18:46:18 +00001526**/
1527EFI_STATUS
1528EFIAPI
1529ShellCloseFileMetaArg (
1530 IN OUT EFI_SHELL_FILE_INFO **ListHead
1531 )
1532{
1533 LIST_ENTRY *Node;
1534
1535 //
1536 // ASSERT that ListHead is not NULL
1537 //
1538 ASSERT(ListHead != NULL);
1539
jcarsey1e6e84c2010-01-25 20:05:08 +00001540 //
jcarsey94b17fa2009-05-07 18:46:18 +00001541 // Check for UEFI Shell 2.0 protocols
1542 //
jcarsey366f81a2011-06-27 21:04:22 +00001543 if (gEfiShellProtocol != NULL) {
1544 return (gEfiShellProtocol->FreeFileList(ListHead));
jcarsey92a54472011-06-27 20:33:13 +00001545 } else if (mEfiShellEnvironment2 != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001546 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001547 // Since this is EFI Shell version we need to free our internally made copy
jcarsey94b17fa2009-05-07 18:46:18 +00001548 // of the list
1549 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001550 for ( Node = GetFirstNode(&(*ListHead)->Link)
jcarseya405b862010-09-14 05:18:09 +00001551 ; *ListHead != NULL && !IsListEmpty(&(*ListHead)->Link)
jcarsey9b3bf082009-06-23 21:15:07 +00001552 ; Node = GetFirstNode(&(*ListHead)->Link)) {
jcarsey94b17fa2009-05-07 18:46:18 +00001553 RemoveEntryList(Node);
jcarseya405b862010-09-14 05:18:09 +00001554 ((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 +00001555 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->FullName);
1556 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->FileName);
1557 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->Info);
1558 FreePool((EFI_SHELL_FILE_INFO_NO_CONST*)Node);
1559 }
1560 return EFI_SUCCESS;
1561 }
jcarsey92a54472011-06-27 20:33:13 +00001562
1563 return (EFI_UNSUPPORTED);
jcarsey94b17fa2009-05-07 18:46:18 +00001564}
1565
jcarsey125c2cf2009-11-18 21:36:50 +00001566/**
1567 Find a file by searching the CWD and then the path.
1568
jcarseyb3011f42010-01-11 21:49:04 +00001569 If FileName is NULL then ASSERT.
jcarsey125c2cf2009-11-18 21:36:50 +00001570
jcarseyb3011f42010-01-11 21:49:04 +00001571 If the return value is not NULL then the memory must be caller freed.
jcarsey125c2cf2009-11-18 21:36:50 +00001572
1573 @param FileName Filename string.
1574
1575 @retval NULL the file was not found
1576 @return !NULL the full path to the file.
1577**/
1578CHAR16 *
1579EFIAPI
1580ShellFindFilePath (
1581 IN CONST CHAR16 *FileName
1582 )
1583{
1584 CONST CHAR16 *Path;
jcarseya405b862010-09-14 05:18:09 +00001585 SHELL_FILE_HANDLE Handle;
jcarsey125c2cf2009-11-18 21:36:50 +00001586 EFI_STATUS Status;
1587 CHAR16 *RetVal;
1588 CHAR16 *TestPath;
1589 CONST CHAR16 *Walker;
jcarsey36a9d672009-11-20 21:13:41 +00001590 UINTN Size;
jcarsey1cd45e72010-01-29 15:07:44 +00001591 CHAR16 *TempChar;
jcarsey125c2cf2009-11-18 21:36:50 +00001592
1593 RetVal = NULL;
1594
jcarseya405b862010-09-14 05:18:09 +00001595 //
1596 // First make sure its not an absolute path.
1597 //
1598 Status = ShellOpenFileByName(FileName, &Handle, EFI_FILE_MODE_READ, 0);
1599 if (!EFI_ERROR(Status)){
1600 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1601 ASSERT(RetVal == NULL);
1602 RetVal = StrnCatGrow(&RetVal, NULL, FileName, 0);
1603 ShellCloseFile(&Handle);
1604 return (RetVal);
1605 } else {
1606 ShellCloseFile(&Handle);
1607 }
1608 }
1609
jcarsey125c2cf2009-11-18 21:36:50 +00001610 Path = ShellGetEnvironmentVariable(L"cwd");
1611 if (Path != NULL) {
jcarsey36a9d672009-11-20 21:13:41 +00001612 Size = StrSize(Path);
1613 Size += StrSize(FileName);
1614 TestPath = AllocateZeroPool(Size);
jcarseyc9d92df2010-02-03 15:37:54 +00001615 ASSERT(TestPath != NULL);
1616 if (TestPath == NULL) {
1617 return (NULL);
1618 }
jcarsey125c2cf2009-11-18 21:36:50 +00001619 StrCpy(TestPath, Path);
1620 StrCat(TestPath, FileName);
1621 Status = ShellOpenFileByName(TestPath, &Handle, EFI_FILE_MODE_READ, 0);
1622 if (!EFI_ERROR(Status)){
jcarseya405b862010-09-14 05:18:09 +00001623 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1624 ASSERT(RetVal == NULL);
1625 RetVal = StrnCatGrow(&RetVal, NULL, TestPath, 0);
1626 ShellCloseFile(&Handle);
1627 FreePool(TestPath);
1628 return (RetVal);
1629 } else {
1630 ShellCloseFile(&Handle);
1631 }
jcarsey125c2cf2009-11-18 21:36:50 +00001632 }
1633 FreePool(TestPath);
1634 }
1635 Path = ShellGetEnvironmentVariable(L"path");
1636 if (Path != NULL) {
jcarseya405b862010-09-14 05:18:09 +00001637 Size = StrSize(Path)+sizeof(CHAR16);
jcarsey36a9d672009-11-20 21:13:41 +00001638 Size += StrSize(FileName);
1639 TestPath = AllocateZeroPool(Size);
jcarsey3e082d52010-10-04 16:44:57 +00001640 if (TestPath == NULL) {
1641 return (NULL);
1642 }
jcarsey1e6e84c2010-01-25 20:05:08 +00001643 Walker = (CHAR16*)Path;
jcarsey125c2cf2009-11-18 21:36:50 +00001644 do {
1645 CopyMem(TestPath, Walker, StrSize(Walker));
jcarsey3e082d52010-10-04 16:44:57 +00001646 if (TestPath != NULL) {
1647 TempChar = StrStr(TestPath, L";");
1648 if (TempChar != NULL) {
1649 *TempChar = CHAR_NULL;
1650 }
1651 if (TestPath[StrLen(TestPath)-1] != L'\\') {
1652 StrCat(TestPath, L"\\");
1653 }
jcarsey89e85372011-04-13 23:37:50 +00001654 if (FileName[0] == L'\\') {
1655 FileName++;
1656 }
jcarsey3e082d52010-10-04 16:44:57 +00001657 StrCat(TestPath, FileName);
1658 if (StrStr(Walker, L";") != NULL) {
1659 Walker = StrStr(Walker, L";") + 1;
jcarseya405b862010-09-14 05:18:09 +00001660 } else {
jcarsey3e082d52010-10-04 16:44:57 +00001661 Walker = NULL;
1662 }
1663 Status = ShellOpenFileByName(TestPath, &Handle, EFI_FILE_MODE_READ, 0);
1664 if (!EFI_ERROR(Status)){
1665 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1666 ASSERT(RetVal == NULL);
1667 RetVal = StrnCatGrow(&RetVal, NULL, TestPath, 0);
1668 ShellCloseFile(&Handle);
1669 break;
1670 } else {
1671 ShellCloseFile(&Handle);
1672 }
jcarseya405b862010-09-14 05:18:09 +00001673 }
jcarsey125c2cf2009-11-18 21:36:50 +00001674 }
1675 } while (Walker != NULL && Walker[0] != CHAR_NULL);
1676 FreePool(TestPath);
1677 }
1678 return (RetVal);
1679}
1680
jcarseyb3011f42010-01-11 21:49:04 +00001681/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001682 Find a file by searching the CWD and then the path with a variable set of file
1683 extensions. If the file is not found it will append each extension in the list
jcarseyb3011f42010-01-11 21:49:04 +00001684 in the order provided and return the first one that is successful.
1685
1686 If FileName is NULL, then ASSERT.
1687 If FileExtension is NULL, then behavior is identical to ShellFindFilePath.
1688
1689 If the return value is not NULL then the memory must be caller freed.
1690
1691 @param[in] FileName Filename string.
1692 @param[in] FileExtension Semi-colon delimeted list of possible extensions.
1693
1694 @retval NULL The file was not found.
1695 @retval !NULL The path to the file.
1696**/
1697CHAR16 *
1698EFIAPI
1699ShellFindFilePathEx (
1700 IN CONST CHAR16 *FileName,
1701 IN CONST CHAR16 *FileExtension
1702 )
1703{
1704 CHAR16 *TestPath;
1705 CHAR16 *RetVal;
1706 CONST CHAR16 *ExtensionWalker;
jcarsey9e926b62010-01-14 20:26:39 +00001707 UINTN Size;
jcarsey1cd45e72010-01-29 15:07:44 +00001708 CHAR16 *TempChar;
jcarseyc9d92df2010-02-03 15:37:54 +00001709 CHAR16 *TempChar2;
jcarsey1cd45e72010-01-29 15:07:44 +00001710
jcarseyb3011f42010-01-11 21:49:04 +00001711 ASSERT(FileName != NULL);
1712 if (FileExtension == NULL) {
1713 return (ShellFindFilePath(FileName));
1714 }
1715 RetVal = ShellFindFilePath(FileName);
1716 if (RetVal != NULL) {
1717 return (RetVal);
1718 }
jcarsey9e926b62010-01-14 20:26:39 +00001719 Size = StrSize(FileName);
1720 Size += StrSize(FileExtension);
1721 TestPath = AllocateZeroPool(Size);
jcarseyc9d92df2010-02-03 15:37:54 +00001722 ASSERT(TestPath != NULL);
1723 if (TestPath == NULL) {
1724 return (NULL);
1725 }
jcarseya405b862010-09-14 05:18:09 +00001726 for (ExtensionWalker = FileExtension, TempChar2 = (CHAR16*)FileExtension; TempChar2 != NULL ; ExtensionWalker = TempChar2 + 1){
jcarseyb3011f42010-01-11 21:49:04 +00001727 StrCpy(TestPath, FileName);
jcarseya405b862010-09-14 05:18:09 +00001728 if (ExtensionWalker != NULL) {
1729 StrCat(TestPath, ExtensionWalker);
1730 }
jcarsey1cd45e72010-01-29 15:07:44 +00001731 TempChar = StrStr(TestPath, L";");
1732 if (TempChar != NULL) {
1733 *TempChar = CHAR_NULL;
jcarseyb3011f42010-01-11 21:49:04 +00001734 }
1735 RetVal = ShellFindFilePath(TestPath);
1736 if (RetVal != NULL) {
1737 break;
1738 }
jcarseya405b862010-09-14 05:18:09 +00001739 ASSERT(ExtensionWalker != NULL);
jcarseyc9d92df2010-02-03 15:37:54 +00001740 TempChar2 = StrStr(ExtensionWalker, L";");
jcarseyb3011f42010-01-11 21:49:04 +00001741 }
1742 FreePool(TestPath);
1743 return (RetVal);
1744}
1745
jcarsey94b17fa2009-05-07 18:46:18 +00001746typedef struct {
jcarsey9b3bf082009-06-23 21:15:07 +00001747 LIST_ENTRY Link;
jcarsey94b17fa2009-05-07 18:46:18 +00001748 CHAR16 *Name;
jcarseya405b862010-09-14 05:18:09 +00001749 SHELL_PARAM_TYPE Type;
jcarsey94b17fa2009-05-07 18:46:18 +00001750 CHAR16 *Value;
1751 UINTN OriginalPosition;
1752} SHELL_PARAM_PACKAGE;
1753
1754/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001755 Checks the list of valid arguments and returns TRUE if the item was found. If the
jcarsey94b17fa2009-05-07 18:46:18 +00001756 return value is TRUE then the type parameter is set also.
jcarsey1e6e84c2010-01-25 20:05:08 +00001757
jcarsey94b17fa2009-05-07 18:46:18 +00001758 if CheckList is NULL then ASSERT();
1759 if Name is NULL then ASSERT();
1760 if Type is NULL then ASSERT();
1761
jcarsey94b17fa2009-05-07 18:46:18 +00001762 @param Name pointer to Name of parameter found
1763 @param CheckList List to check against
jcarseya405b862010-09-14 05:18:09 +00001764 @param Type pointer to type of parameter if it was found
jcarsey94b17fa2009-05-07 18:46:18 +00001765
1766 @retval TRUE the Parameter was found. Type is valid.
1767 @retval FALSE the Parameter was not found. Type is not valid.
1768**/
1769BOOLEAN
1770EFIAPI
jcarseyd2b45642009-05-11 18:02:16 +00001771InternalIsOnCheckList (
jcarsey94b17fa2009-05-07 18:46:18 +00001772 IN CONST CHAR16 *Name,
1773 IN CONST SHELL_PARAM_ITEM *CheckList,
jcarsey252d9452011-03-25 20:49:53 +00001774 OUT SHELL_PARAM_TYPE *Type
jcarseya405b862010-09-14 05:18:09 +00001775 )
1776{
jcarsey94b17fa2009-05-07 18:46:18 +00001777 SHELL_PARAM_ITEM *TempListItem;
jcarsey252d9452011-03-25 20:49:53 +00001778 CHAR16 *TempString;
jcarsey94b17fa2009-05-07 18:46:18 +00001779
1780 //
1781 // ASSERT that all 3 pointer parameters aren't NULL
1782 //
1783 ASSERT(CheckList != NULL);
1784 ASSERT(Type != NULL);
1785 ASSERT(Name != NULL);
1786
1787 //
jcarseyd2b45642009-05-11 18:02:16 +00001788 // question mark and page break mode are always supported
1789 //
1790 if ((StrCmp(Name, L"-?") == 0) ||
1791 (StrCmp(Name, L"-b") == 0)
jcarseya405b862010-09-14 05:18:09 +00001792 ) {
jcarsey252d9452011-03-25 20:49:53 +00001793 *Type = TypeFlag;
jcarseyd2b45642009-05-11 18:02:16 +00001794 return (TRUE);
1795 }
1796
1797 //
jcarsey94b17fa2009-05-07 18:46:18 +00001798 // Enumerate through the list
1799 //
1800 for (TempListItem = (SHELL_PARAM_ITEM*)CheckList ; TempListItem->Name != NULL ; TempListItem++) {
1801 //
jcarsey9eb53ac2009-07-08 17:26:58 +00001802 // If the Type is TypeStart only check the first characters of the passed in param
1803 // If it matches set the type and return TRUE
jcarsey94b17fa2009-05-07 18:46:18 +00001804 //
jcarsey252d9452011-03-25 20:49:53 +00001805 if (TempListItem->Type == TypeStart) {
1806 if (StrnCmp(Name, TempListItem->Name, StrLen(TempListItem->Name)) == 0) {
1807 *Type = TempListItem->Type;
1808 return (TRUE);
1809 }
1810 TempString = NULL;
1811 TempString = StrnCatGrow(&TempString, NULL, Name, StrLen(TempListItem->Name));
1812 if (TempString != NULL) {
1813 if (StringNoCaseCompare(&TempString, &TempListItem->Name) == 0) {
1814 *Type = TempListItem->Type;
1815 FreePool(TempString);
1816 return (TRUE);
1817 }
1818 FreePool(TempString);
1819 }
1820 } else if (StringNoCaseCompare(&Name, &TempListItem->Name) == 0) {
jcarsey94b17fa2009-05-07 18:46:18 +00001821 *Type = TempListItem->Type;
1822 return (TRUE);
1823 }
1824 }
jcarsey2247dde2009-11-09 18:08:58 +00001825
jcarsey94b17fa2009-05-07 18:46:18 +00001826 return (FALSE);
1827}
1828/**
jcarseyd2b45642009-05-11 18:02:16 +00001829 Checks the string for indicators of "flag" status. this is a leading '/', '-', or '+'
jcarsey94b17fa2009-05-07 18:46:18 +00001830
jcarseya405b862010-09-14 05:18:09 +00001831 @param[in] Name pointer to Name of parameter found
1832 @param[in] AlwaysAllowNumbers TRUE to allow numbers, FALSE to not.
jcarsey94b17fa2009-05-07 18:46:18 +00001833
1834 @retval TRUE the Parameter is a flag.
jcarseya405b862010-09-14 05:18:09 +00001835 @retval FALSE the Parameter not a flag.
jcarsey94b17fa2009-05-07 18:46:18 +00001836**/
1837BOOLEAN
1838EFIAPI
jcarseyd2b45642009-05-11 18:02:16 +00001839InternalIsFlag (
jcarsey2247dde2009-11-09 18:08:58 +00001840 IN CONST CHAR16 *Name,
1841 IN BOOLEAN AlwaysAllowNumbers
jcarsey94b17fa2009-05-07 18:46:18 +00001842 )
1843{
1844 //
1845 // ASSERT that Name isn't NULL
1846 //
1847 ASSERT(Name != NULL);
1848
1849 //
jcarsey2247dde2009-11-09 18:08:58 +00001850 // If we accept numbers then dont return TRUE. (they will be values)
1851 //
jcarseya405b862010-09-14 05:18:09 +00001852 if (((Name[0] == L'-' || Name[0] == L'+') && ShellIsHexaDecimalDigitCharacter(Name[1])) && AlwaysAllowNumbers) {
jcarsey2247dde2009-11-09 18:08:58 +00001853 return (FALSE);
1854 }
1855
1856 //
jcarseya405b862010-09-14 05:18:09 +00001857 // If the Name has a /, +, or - as the first character return TRUE
jcarsey94b17fa2009-05-07 18:46:18 +00001858 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001859 if ((Name[0] == L'/') ||
jcarseyd2b45642009-05-11 18:02:16 +00001860 (Name[0] == L'-') ||
1861 (Name[0] == L'+')
jcarseya405b862010-09-14 05:18:09 +00001862 ) {
jcarsey94b17fa2009-05-07 18:46:18 +00001863 return (TRUE);
1864 }
1865 return (FALSE);
1866}
1867
1868/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001869 Checks the command line arguments passed against the list of valid ones.
jcarsey94b17fa2009-05-07 18:46:18 +00001870
1871 If no initialization is required, then return RETURN_SUCCESS.
jcarsey1e6e84c2010-01-25 20:05:08 +00001872
jcarseya405b862010-09-14 05:18:09 +00001873 @param[in] CheckList pointer to list of parameters to check
1874 @param[out] CheckPackage pointer to pointer to list checked values
1875 @param[out] ProblemParam optional pointer to pointer to unicode string for
jcarseyd2b45642009-05-11 18:02:16 +00001876 the paramater that caused failure. If used then the
1877 caller is responsible for freeing the memory.
jcarseya405b862010-09-14 05:18:09 +00001878 @param[in] AutoPageBreak will automatically set PageBreakEnabled for "b" parameter
1879 @param[in] Argv pointer to array of parameters
1880 @param[in] Argc Count of parameters in Argv
1881 @param[in] AlwaysAllowNumbers TRUE to allow numbers always, FALSE otherwise.
jcarsey94b17fa2009-05-07 18:46:18 +00001882
1883 @retval EFI_SUCCESS The operation completed sucessfully.
1884 @retval EFI_OUT_OF_RESOURCES A memory allocation failed
1885 @retval EFI_INVALID_PARAMETER A parameter was invalid
jcarsey1e6e84c2010-01-25 20:05:08 +00001886 @retval EFI_VOLUME_CORRUPTED the command line was corrupt. an argument was
1887 duplicated. the duplicated command line argument
jcarsey94b17fa2009-05-07 18:46:18 +00001888 was returned in ProblemParam if provided.
jcarsey1e6e84c2010-01-25 20:05:08 +00001889 @retval EFI_NOT_FOUND a argument required a value that was missing.
jcarsey94b17fa2009-05-07 18:46:18 +00001890 the invalid command line argument was returned in
1891 ProblemParam if provided.
1892**/
1893EFI_STATUS
1894EFIAPI
1895InternalCommandLineParse (
1896 IN CONST SHELL_PARAM_ITEM *CheckList,
1897 OUT LIST_ENTRY **CheckPackage,
1898 OUT CHAR16 **ProblemParam OPTIONAL,
1899 IN BOOLEAN AutoPageBreak,
1900 IN CONST CHAR16 **Argv,
jcarsey2247dde2009-11-09 18:08:58 +00001901 IN UINTN Argc,
1902 IN BOOLEAN AlwaysAllowNumbers
jcarseya405b862010-09-14 05:18:09 +00001903 )
1904{
jcarsey94b17fa2009-05-07 18:46:18 +00001905 UINTN LoopCounter;
jcarsey252d9452011-03-25 20:49:53 +00001906 SHELL_PARAM_TYPE CurrentItemType;
jcarsey94b17fa2009-05-07 18:46:18 +00001907 SHELL_PARAM_PACKAGE *CurrentItemPackage;
jcarsey125c2cf2009-11-18 21:36:50 +00001908 UINTN GetItemValue;
1909 UINTN ValueSize;
jcarseya405b862010-09-14 05:18:09 +00001910 UINTN Count;
jcarsey252d9452011-03-25 20:49:53 +00001911 CONST CHAR16 *TempPointer;
jcarsey94b17fa2009-05-07 18:46:18 +00001912
1913 CurrentItemPackage = NULL;
jcarsey125c2cf2009-11-18 21:36:50 +00001914 GetItemValue = 0;
1915 ValueSize = 0;
jcarseya405b862010-09-14 05:18:09 +00001916 Count = 0;
jcarsey94b17fa2009-05-07 18:46:18 +00001917
1918 //
1919 // If there is only 1 item we dont need to do anything
1920 //
jcarseya405b862010-09-14 05:18:09 +00001921 if (Argc < 1) {
jcarsey94b17fa2009-05-07 18:46:18 +00001922 *CheckPackage = NULL;
1923 return (EFI_SUCCESS);
1924 }
1925
1926 //
jcarsey2247dde2009-11-09 18:08:58 +00001927 // ASSERTs
1928 //
1929 ASSERT(CheckList != NULL);
1930 ASSERT(Argv != NULL);
1931
1932 //
jcarsey94b17fa2009-05-07 18:46:18 +00001933 // initialize the linked list
1934 //
1935 *CheckPackage = (LIST_ENTRY*)AllocateZeroPool(sizeof(LIST_ENTRY));
sfu502a758c2011-10-08 02:55:30 +00001936 if (*CheckPackage == NULL) {
1937 return EFI_OUT_OF_RESOURCES;
1938 }
jcarsey94b17fa2009-05-07 18:46:18 +00001939 InitializeListHead(*CheckPackage);
1940
1941 //
1942 // loop through each of the arguments
1943 //
1944 for (LoopCounter = 0 ; LoopCounter < Argc ; ++LoopCounter) {
1945 if (Argv[LoopCounter] == NULL) {
1946 //
1947 // do nothing for NULL argv
1948 //
jcarseya405b862010-09-14 05:18:09 +00001949 } else if (InternalIsOnCheckList(Argv[LoopCounter], CheckList, &CurrentItemType)) {
jcarsey94b17fa2009-05-07 18:46:18 +00001950 //
jcarsey2247dde2009-11-09 18:08:58 +00001951 // We might have leftover if last parameter didnt have optional value
1952 //
jcarsey125c2cf2009-11-18 21:36:50 +00001953 if (GetItemValue != 0) {
1954 GetItemValue = 0;
jcarsey2247dde2009-11-09 18:08:58 +00001955 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
1956 }
1957 //
jcarsey94b17fa2009-05-07 18:46:18 +00001958 // this is a flag
1959 //
jcarsey252d9452011-03-25 20:49:53 +00001960 CurrentItemPackage = AllocateZeroPool(sizeof(SHELL_PARAM_PACKAGE));
jcarsey94b17fa2009-05-07 18:46:18 +00001961 ASSERT(CurrentItemPackage != NULL);
jcarsey252d9452011-03-25 20:49:53 +00001962 CurrentItemPackage->Name = AllocateZeroPool(StrSize(Argv[LoopCounter]));
jcarsey94b17fa2009-05-07 18:46:18 +00001963 ASSERT(CurrentItemPackage->Name != NULL);
1964 StrCpy(CurrentItemPackage->Name, Argv[LoopCounter]);
1965 CurrentItemPackage->Type = CurrentItemType;
1966 CurrentItemPackage->OriginalPosition = (UINTN)(-1);
jcarseyb1f95a02009-06-16 00:23:19 +00001967 CurrentItemPackage->Value = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +00001968
1969 //
1970 // Does this flag require a value
1971 //
jcarsey125c2cf2009-11-18 21:36:50 +00001972 switch (CurrentItemPackage->Type) {
jcarsey94b17fa2009-05-07 18:46:18 +00001973 //
jcarsey125c2cf2009-11-18 21:36:50 +00001974 // possibly trigger the next loop(s) to populate the value of this item
jcarsey1e6e84c2010-01-25 20:05:08 +00001975 //
jcarsey125c2cf2009-11-18 21:36:50 +00001976 case TypeValue:
jcarsey1e6e84c2010-01-25 20:05:08 +00001977 GetItemValue = 1;
jcarsey125c2cf2009-11-18 21:36:50 +00001978 ValueSize = 0;
1979 break;
1980 case TypeDoubleValue:
1981 GetItemValue = 2;
1982 ValueSize = 0;
1983 break;
1984 case TypeMaxValue:
1985 GetItemValue = (UINTN)(-1);
1986 ValueSize = 0;
1987 break;
1988 default:
1989 //
1990 // this item has no value expected; we are done
1991 //
1992 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
1993 ASSERT(GetItemValue == 0);
1994 break;
jcarsey94b17fa2009-05-07 18:46:18 +00001995 }
jcarseya405b862010-09-14 05:18:09 +00001996 } else if (GetItemValue != 0 && !InternalIsFlag(Argv[LoopCounter], AlwaysAllowNumbers)) {
jcarseyb1f95a02009-06-16 00:23:19 +00001997 ASSERT(CurrentItemPackage != NULL);
1998 //
jcarsey125c2cf2009-11-18 21:36:50 +00001999 // get the item VALUE for a previous flag
jcarseyb1f95a02009-06-16 00:23:19 +00002000 //
jcarsey125c2cf2009-11-18 21:36:50 +00002001 CurrentItemPackage->Value = ReallocatePool(ValueSize, ValueSize + StrSize(Argv[LoopCounter]) + sizeof(CHAR16), CurrentItemPackage->Value);
jcarseyb1f95a02009-06-16 00:23:19 +00002002 ASSERT(CurrentItemPackage->Value != NULL);
jcarsey125c2cf2009-11-18 21:36:50 +00002003 if (ValueSize == 0) {
2004 StrCpy(CurrentItemPackage->Value, Argv[LoopCounter]);
2005 } else {
2006 StrCat(CurrentItemPackage->Value, L" ");
2007 StrCat(CurrentItemPackage->Value, Argv[LoopCounter]);
2008 }
2009 ValueSize += StrSize(Argv[LoopCounter]) + sizeof(CHAR16);
2010 GetItemValue--;
2011 if (GetItemValue == 0) {
2012 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2013 }
jcarseya405b862010-09-14 05:18:09 +00002014 } else if (!InternalIsFlag(Argv[LoopCounter], AlwaysAllowNumbers) ){ //|| ProblemParam == NULL) {
jcarseyb1f95a02009-06-16 00:23:19 +00002015 //
2016 // add this one as a non-flag
2017 //
jcarsey252d9452011-03-25 20:49:53 +00002018
2019 TempPointer = Argv[LoopCounter];
2020 if ((*TempPointer == L'^' && *(TempPointer+1) == L'-')
2021 || (*TempPointer == L'^' && *(TempPointer+1) == L'/')
2022 || (*TempPointer == L'^' && *(TempPointer+1) == L'+')
2023 ){
2024 TempPointer++;
2025 }
2026 CurrentItemPackage = AllocateZeroPool(sizeof(SHELL_PARAM_PACKAGE));
jcarseyb1f95a02009-06-16 00:23:19 +00002027 ASSERT(CurrentItemPackage != NULL);
2028 CurrentItemPackage->Name = NULL;
2029 CurrentItemPackage->Type = TypePosition;
jcarsey252d9452011-03-25 20:49:53 +00002030 CurrentItemPackage->Value = AllocateZeroPool(StrSize(TempPointer));
jcarseyb1f95a02009-06-16 00:23:19 +00002031 ASSERT(CurrentItemPackage->Value != NULL);
jcarsey252d9452011-03-25 20:49:53 +00002032 StrCpy(CurrentItemPackage->Value, TempPointer);
jcarseya405b862010-09-14 05:18:09 +00002033 CurrentItemPackage->OriginalPosition = Count++;
jcarsey9b3bf082009-06-23 21:15:07 +00002034 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
jcarsey252d9452011-03-25 20:49:53 +00002035 } else {
jcarsey94b17fa2009-05-07 18:46:18 +00002036 //
2037 // this was a non-recognised flag... error!
2038 //
jcarsey252d9452011-03-25 20:49:53 +00002039 if (ProblemParam != NULL) {
2040 *ProblemParam = AllocateZeroPool(StrSize(Argv[LoopCounter]));
2041 ASSERT(*ProblemParam != NULL);
2042 StrCpy(*ProblemParam, Argv[LoopCounter]);
2043 }
jcarsey94b17fa2009-05-07 18:46:18 +00002044 ShellCommandLineFreeVarList(*CheckPackage);
2045 *CheckPackage = NULL;
2046 return (EFI_VOLUME_CORRUPTED);
jcarsey94b17fa2009-05-07 18:46:18 +00002047 }
2048 }
jcarsey125c2cf2009-11-18 21:36:50 +00002049 if (GetItemValue != 0) {
2050 GetItemValue = 0;
2051 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2052 }
jcarsey94b17fa2009-05-07 18:46:18 +00002053 //
2054 // support for AutoPageBreak
2055 //
2056 if (AutoPageBreak && ShellCommandLineGetFlag(*CheckPackage, L"-b")) {
2057 ShellSetPageBreakMode(TRUE);
2058 }
2059 return (EFI_SUCCESS);
2060}
2061
2062/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002063 Checks the command line arguments passed against the list of valid ones.
jcarsey94b17fa2009-05-07 18:46:18 +00002064 Optionally removes NULL values first.
jcarsey1e6e84c2010-01-25 20:05:08 +00002065
jcarsey94b17fa2009-05-07 18:46:18 +00002066 If no initialization is required, then return RETURN_SUCCESS.
jcarsey1e6e84c2010-01-25 20:05:08 +00002067
jcarseya405b862010-09-14 05:18:09 +00002068 @param[in] CheckList The pointer to list of parameters to check.
2069 @param[out] CheckPackage The package of checked values.
2070 @param[out] ProblemParam Optional pointer to pointer to unicode string for
jcarsey94b17fa2009-05-07 18:46:18 +00002071 the paramater that caused failure.
jcarseya405b862010-09-14 05:18:09 +00002072 @param[in] AutoPageBreak Will automatically set PageBreakEnabled.
2073 @param[in] AlwaysAllowNumbers Will never fail for number based flags.
jcarsey94b17fa2009-05-07 18:46:18 +00002074
2075 @retval EFI_SUCCESS The operation completed sucessfully.
jcarseya405b862010-09-14 05:18:09 +00002076 @retval EFI_OUT_OF_RESOURCES A memory allocation failed.
2077 @retval EFI_INVALID_PARAMETER A parameter was invalid.
2078 @retval EFI_VOLUME_CORRUPTED The command line was corrupt.
2079 @retval EFI_DEVICE_ERROR The commands contained 2 opposing arguments. One
jcarsey1e6e84c2010-01-25 20:05:08 +00002080 of the command line arguments was returned in
jcarsey94b17fa2009-05-07 18:46:18 +00002081 ProblemParam if provided.
jcarseya405b862010-09-14 05:18:09 +00002082 @retval EFI_NOT_FOUND A argument required a value that was missing.
2083 The invalid command line argument was returned in
jcarsey94b17fa2009-05-07 18:46:18 +00002084 ProblemParam if provided.
2085**/
2086EFI_STATUS
2087EFIAPI
jcarsey2247dde2009-11-09 18:08:58 +00002088ShellCommandLineParseEx (
jcarsey94b17fa2009-05-07 18:46:18 +00002089 IN CONST SHELL_PARAM_ITEM *CheckList,
2090 OUT LIST_ENTRY **CheckPackage,
2091 OUT CHAR16 **ProblemParam OPTIONAL,
jcarsey2247dde2009-11-09 18:08:58 +00002092 IN BOOLEAN AutoPageBreak,
2093 IN BOOLEAN AlwaysAllowNumbers
jcarseya405b862010-09-14 05:18:09 +00002094 )
2095{
jcarsey1e6e84c2010-01-25 20:05:08 +00002096 //
jcarsey94b17fa2009-05-07 18:46:18 +00002097 // ASSERT that CheckList and CheckPackage aren't NULL
2098 //
2099 ASSERT(CheckList != NULL);
2100 ASSERT(CheckPackage != NULL);
2101
jcarsey1e6e84c2010-01-25 20:05:08 +00002102 //
jcarsey94b17fa2009-05-07 18:46:18 +00002103 // Check for UEFI Shell 2.0 protocols
2104 //
jcarsey366f81a2011-06-27 21:04:22 +00002105 if (gEfiShellParametersProtocol != NULL) {
jcarsey1e6e84c2010-01-25 20:05:08 +00002106 return (InternalCommandLineParse(CheckList,
2107 CheckPackage,
2108 ProblemParam,
2109 AutoPageBreak,
jcarsey366f81a2011-06-27 21:04:22 +00002110 (CONST CHAR16**) gEfiShellParametersProtocol->Argv,
2111 gEfiShellParametersProtocol->Argc,
jcarsey2247dde2009-11-09 18:08:58 +00002112 AlwaysAllowNumbers));
jcarsey94b17fa2009-05-07 18:46:18 +00002113 }
2114
jcarsey1e6e84c2010-01-25 20:05:08 +00002115 //
jcarsey94b17fa2009-05-07 18:46:18 +00002116 // ASSERT That EFI Shell is not required
2117 //
2118 ASSERT (mEfiShellInterface != NULL);
jcarsey1e6e84c2010-01-25 20:05:08 +00002119 return (InternalCommandLineParse(CheckList,
2120 CheckPackage,
2121 ProblemParam,
2122 AutoPageBreak,
jljusten08d7f8e2009-06-15 18:42:13 +00002123 (CONST CHAR16**) mEfiShellInterface->Argv,
jcarsey2247dde2009-11-09 18:08:58 +00002124 mEfiShellInterface->Argc,
2125 AlwaysAllowNumbers));
jcarsey94b17fa2009-05-07 18:46:18 +00002126}
2127
2128/**
2129 Frees shell variable list that was returned from ShellCommandLineParse.
2130
2131 This function will free all the memory that was used for the CheckPackage
2132 list of postprocessed shell arguments.
2133
2134 this function has no return value.
2135
2136 if CheckPackage is NULL, then return
2137
2138 @param CheckPackage the list to de-allocate
2139 **/
2140VOID
2141EFIAPI
2142ShellCommandLineFreeVarList (
2143 IN LIST_ENTRY *CheckPackage
jcarseya405b862010-09-14 05:18:09 +00002144 )
2145{
jcarsey94b17fa2009-05-07 18:46:18 +00002146 LIST_ENTRY *Node;
2147
2148 //
2149 // check for CheckPackage == NULL
2150 //
2151 if (CheckPackage == NULL) {
2152 return;
2153 }
2154
2155 //
2156 // for each node in the list
2157 //
jcarsey9eb53ac2009-07-08 17:26:58 +00002158 for ( Node = GetFirstNode(CheckPackage)
jcarseya405b862010-09-14 05:18:09 +00002159 ; !IsListEmpty(CheckPackage)
jcarsey9eb53ac2009-07-08 17:26:58 +00002160 ; Node = GetFirstNode(CheckPackage)
jcarseya405b862010-09-14 05:18:09 +00002161 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002162 //
2163 // Remove it from the list
2164 //
2165 RemoveEntryList(Node);
2166
2167 //
2168 // if it has a name free the name
2169 //
2170 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
2171 FreePool(((SHELL_PARAM_PACKAGE*)Node)->Name);
2172 }
2173
2174 //
2175 // if it has a value free the value
2176 //
2177 if (((SHELL_PARAM_PACKAGE*)Node)->Value != NULL) {
2178 FreePool(((SHELL_PARAM_PACKAGE*)Node)->Value);
2179 }
jcarsey1e6e84c2010-01-25 20:05:08 +00002180
jcarsey94b17fa2009-05-07 18:46:18 +00002181 //
2182 // free the node structure
2183 //
2184 FreePool((SHELL_PARAM_PACKAGE*)Node);
2185 }
2186 //
2187 // free the list head node
2188 //
2189 FreePool(CheckPackage);
2190}
2191/**
2192 Checks for presence of a flag parameter
2193
2194 flag arguments are in the form of "-<Key>" or "/<Key>", but do not have a value following the key
2195
2196 if CheckPackage is NULL then return FALSE.
2197 if KeyString is NULL then ASSERT()
jcarsey1e6e84c2010-01-25 20:05:08 +00002198
jcarsey94b17fa2009-05-07 18:46:18 +00002199 @param CheckPackage The package of parsed command line arguments
2200 @param KeyString the Key of the command line argument to check for
2201
2202 @retval TRUE the flag is on the command line
2203 @retval FALSE the flag is not on the command line
2204 **/
2205BOOLEAN
2206EFIAPI
2207ShellCommandLineGetFlag (
jcarseya405b862010-09-14 05:18:09 +00002208 IN CONST LIST_ENTRY * CONST CheckPackage,
2209 IN CONST CHAR16 * CONST KeyString
2210 )
2211{
jcarsey94b17fa2009-05-07 18:46:18 +00002212 LIST_ENTRY *Node;
jcarsey252d9452011-03-25 20:49:53 +00002213 CHAR16 *TempString;
jcarsey94b17fa2009-05-07 18:46:18 +00002214
2215 //
2216 // ASSERT that both CheckPackage and KeyString aren't NULL
2217 //
2218 ASSERT(KeyString != NULL);
2219
2220 //
2221 // return FALSE for no package
2222 //
2223 if (CheckPackage == NULL) {
2224 return (FALSE);
2225 }
2226
2227 //
2228 // enumerate through the list of parametrs
2229 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002230 for ( Node = GetFirstNode(CheckPackage)
2231 ; !IsNull (CheckPackage, Node)
2232 ; Node = GetNextNode(CheckPackage, Node)
jcarsey252d9452011-03-25 20:49:53 +00002233 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002234 //
2235 // If the Name matches, return TRUE (and there may be NULL name)
2236 //
2237 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
jcarsey9eb53ac2009-07-08 17:26:58 +00002238 //
2239 // If Type is TypeStart then only compare the begining of the strings
2240 //
jcarsey252d9452011-03-25 20:49:53 +00002241 if (((SHELL_PARAM_PACKAGE*)Node)->Type == TypeStart) {
2242 if (StrnCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name, StrLen(KeyString)) == 0) {
2243 return (TRUE);
2244 }
2245 TempString = NULL;
2246 TempString = StrnCatGrow(&TempString, NULL, KeyString, StrLen(((SHELL_PARAM_PACKAGE*)Node)->Name));
2247 if (TempString != NULL) {
2248 if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2249 FreePool(TempString);
2250 return (TRUE);
2251 }
2252 FreePool(TempString);
2253 }
2254 } else if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
jcarsey94b17fa2009-05-07 18:46:18 +00002255 return (TRUE);
2256 }
2257 }
2258 }
2259 return (FALSE);
2260}
2261/**
jcarseya405b862010-09-14 05:18:09 +00002262 Returns value from command line argument.
jcarsey94b17fa2009-05-07 18:46:18 +00002263
jcarseya405b862010-09-14 05:18:09 +00002264 Value parameters are in the form of "-<Key> value" or "/<Key> value".
jcarsey1e6e84c2010-01-25 20:05:08 +00002265
jcarseya405b862010-09-14 05:18:09 +00002266 If CheckPackage is NULL, then return NULL.
jcarsey94b17fa2009-05-07 18:46:18 +00002267
jcarseya405b862010-09-14 05:18:09 +00002268 @param[in] CheckPackage The package of parsed command line arguments.
2269 @param[in] KeyString The Key of the command line argument to check for.
jcarsey94b17fa2009-05-07 18:46:18 +00002270
jcarseya405b862010-09-14 05:18:09 +00002271 @retval NULL The flag is not on the command line.
2272 @retval !=NULL The pointer to unicode string of the value.
2273**/
jcarsey94b17fa2009-05-07 18:46:18 +00002274CONST CHAR16*
2275EFIAPI
2276ShellCommandLineGetValue (
2277 IN CONST LIST_ENTRY *CheckPackage,
2278 IN CHAR16 *KeyString
jcarseya405b862010-09-14 05:18:09 +00002279 )
2280{
jcarsey94b17fa2009-05-07 18:46:18 +00002281 LIST_ENTRY *Node;
jcarsey252d9452011-03-25 20:49:53 +00002282 CHAR16 *TempString;
jcarsey94b17fa2009-05-07 18:46:18 +00002283
2284 //
2285 // check for CheckPackage == NULL
2286 //
2287 if (CheckPackage == NULL) {
2288 return (NULL);
2289 }
2290
2291 //
2292 // enumerate through the list of parametrs
2293 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002294 for ( Node = GetFirstNode(CheckPackage)
2295 ; !IsNull (CheckPackage, Node)
2296 ; Node = GetNextNode(CheckPackage, Node)
jcarsey252d9452011-03-25 20:49:53 +00002297 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002298 //
jcarsey252d9452011-03-25 20:49:53 +00002299 // If the Name matches, return TRUE (and there may be NULL name)
jcarsey94b17fa2009-05-07 18:46:18 +00002300 //
2301 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
jcarsey9eb53ac2009-07-08 17:26:58 +00002302 //
2303 // If Type is TypeStart then only compare the begining of the strings
2304 //
jcarsey252d9452011-03-25 20:49:53 +00002305 if (((SHELL_PARAM_PACKAGE*)Node)->Type == TypeStart) {
2306 if (StrnCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name, StrLen(KeyString)) == 0) {
2307 return (((SHELL_PARAM_PACKAGE*)Node)->Name + StrLen(KeyString));
2308 }
2309 TempString = NULL;
2310 TempString = StrnCatGrow(&TempString, NULL, KeyString, StrLen(((SHELL_PARAM_PACKAGE*)Node)->Name));
2311 if (TempString != NULL) {
2312 if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2313 FreePool(TempString);
2314 return (((SHELL_PARAM_PACKAGE*)Node)->Name + StrLen(KeyString));
2315 }
2316 FreePool(TempString);
2317 }
2318 } else if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
jcarsey94b17fa2009-05-07 18:46:18 +00002319 return (((SHELL_PARAM_PACKAGE*)Node)->Value);
2320 }
2321 }
2322 }
2323 return (NULL);
2324}
jcarseya405b862010-09-14 05:18:09 +00002325
jcarsey94b17fa2009-05-07 18:46:18 +00002326/**
jcarseya405b862010-09-14 05:18:09 +00002327 Returns raw value from command line argument.
jcarsey94b17fa2009-05-07 18:46:18 +00002328
jcarseya405b862010-09-14 05:18:09 +00002329 Raw value parameters are in the form of "value" in a specific position in the list.
jcarsey1e6e84c2010-01-25 20:05:08 +00002330
jcarseya405b862010-09-14 05:18:09 +00002331 If CheckPackage is NULL, then return NULL.
jcarsey94b17fa2009-05-07 18:46:18 +00002332
jcarseya405b862010-09-14 05:18:09 +00002333 @param[in] CheckPackage The package of parsed command line arguments.
2334 @param[in] Position The position of the value.
jcarsey94b17fa2009-05-07 18:46:18 +00002335
jcarseya405b862010-09-14 05:18:09 +00002336 @retval NULL The flag is not on the command line.
2337 @retval !=NULL The pointer to unicode string of the value.
jcarsey94b17fa2009-05-07 18:46:18 +00002338 **/
2339CONST CHAR16*
2340EFIAPI
2341ShellCommandLineGetRawValue (
jcarseya405b862010-09-14 05:18:09 +00002342 IN CONST LIST_ENTRY * CONST CheckPackage,
2343 IN UINTN Position
2344 )
2345{
jcarsey94b17fa2009-05-07 18:46:18 +00002346 LIST_ENTRY *Node;
2347
2348 //
2349 // check for CheckPackage == NULL
2350 //
2351 if (CheckPackage == NULL) {
2352 return (NULL);
2353 }
2354
2355 //
2356 // enumerate through the list of parametrs
2357 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002358 for ( Node = GetFirstNode(CheckPackage)
2359 ; !IsNull (CheckPackage, Node)
2360 ; Node = GetNextNode(CheckPackage, Node)
jcarseya405b862010-09-14 05:18:09 +00002361 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002362 //
2363 // If the position matches, return the value
2364 //
2365 if (((SHELL_PARAM_PACKAGE*)Node)->OriginalPosition == Position) {
2366 return (((SHELL_PARAM_PACKAGE*)Node)->Value);
2367 }
2368 }
2369 return (NULL);
jcarseyb1f95a02009-06-16 00:23:19 +00002370}
jcarsey2247dde2009-11-09 18:08:58 +00002371
2372/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002373 returns the number of command line value parameters that were parsed.
2374
jcarsey2247dde2009-11-09 18:08:58 +00002375 this will not include flags.
2376
jcarseya405b862010-09-14 05:18:09 +00002377 @param[in] CheckPackage The package of parsed command line arguments.
2378
jcarsey2247dde2009-11-09 18:08:58 +00002379 @retval (UINTN)-1 No parsing has ocurred
2380 @return other The number of value parameters found
2381**/
2382UINTN
2383EFIAPI
2384ShellCommandLineGetCount(
jcarseya405b862010-09-14 05:18:09 +00002385 IN CONST LIST_ENTRY *CheckPackage
jcarsey125c2cf2009-11-18 21:36:50 +00002386 )
2387{
jcarseya405b862010-09-14 05:18:09 +00002388 LIST_ENTRY *Node1;
2389 UINTN Count;
2390
2391 if (CheckPackage == NULL) {
2392 return (0);
2393 }
2394 for ( Node1 = GetFirstNode(CheckPackage), Count = 0
2395 ; !IsNull (CheckPackage, Node1)
2396 ; Node1 = GetNextNode(CheckPackage, Node1)
2397 ){
2398 if (((SHELL_PARAM_PACKAGE*)Node1)->Name == NULL) {
2399 Count++;
2400 }
2401 }
2402 return (Count);
jcarsey2247dde2009-11-09 18:08:58 +00002403}
2404
jcarsey975136a2009-06-16 19:03:54 +00002405/**
jcarsey36a9d672009-11-20 21:13:41 +00002406 Determins if a parameter is duplicated.
2407
jcarsey1e6e84c2010-01-25 20:05:08 +00002408 If Param is not NULL then it will point to a callee allocated string buffer
jcarsey36a9d672009-11-20 21:13:41 +00002409 with the parameter value if a duplicate is found.
2410
2411 If CheckPackage is NULL, then ASSERT.
2412
2413 @param[in] CheckPackage The package of parsed command line arguments.
2414 @param[out] Param Upon finding one, a pointer to the duplicated parameter.
2415
2416 @retval EFI_SUCCESS No parameters were duplicated.
2417 @retval EFI_DEVICE_ERROR A duplicate was found.
2418 **/
2419EFI_STATUS
2420EFIAPI
2421ShellCommandLineCheckDuplicate (
2422 IN CONST LIST_ENTRY *CheckPackage,
2423 OUT CHAR16 **Param
2424 )
2425{
2426 LIST_ENTRY *Node1;
2427 LIST_ENTRY *Node2;
jcarsey1e6e84c2010-01-25 20:05:08 +00002428
jcarsey36a9d672009-11-20 21:13:41 +00002429 ASSERT(CheckPackage != NULL);
2430
jcarsey1e6e84c2010-01-25 20:05:08 +00002431 for ( Node1 = GetFirstNode(CheckPackage)
2432 ; !IsNull (CheckPackage, Node1)
2433 ; Node1 = GetNextNode(CheckPackage, Node1)
jcarseya405b862010-09-14 05:18:09 +00002434 ){
jcarsey1e6e84c2010-01-25 20:05:08 +00002435 for ( Node2 = GetNextNode(CheckPackage, Node1)
2436 ; !IsNull (CheckPackage, Node2)
2437 ; Node2 = GetNextNode(CheckPackage, Node2)
jcarseya405b862010-09-14 05:18:09 +00002438 ){
2439 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 +00002440 if (Param != NULL) {
2441 *Param = NULL;
2442 *Param = StrnCatGrow(Param, NULL, ((SHELL_PARAM_PACKAGE*)Node1)->Name, 0);
2443 }
2444 return (EFI_DEVICE_ERROR);
2445 }
2446 }
2447 }
2448 return (EFI_SUCCESS);
2449}
2450
2451/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002452 This is a find and replace function. Upon successful return the NewString is a copy of
jcarsey975136a2009-06-16 19:03:54 +00002453 SourceString with each instance of FindTarget replaced with ReplaceWith.
2454
jcarseyb3011f42010-01-11 21:49:04 +00002455 If SourceString and NewString overlap the behavior is undefined.
2456
jcarsey975136a2009-06-16 19:03:54 +00002457 If the string would grow bigger than NewSize it will halt and return error.
2458
ydong104ff7e372011-09-02 08:05:34 +00002459 @param[in] SourceString The string with source buffer.
2460 @param[in, out] NewString The string with resultant buffer.
2461 @param[in] NewSize The size in bytes of NewString.
2462 @param[in] FindTarget The string to look for.
2463 @param[in] ReplaceWith The string to replace FindTarget with.
2464 @param[in] SkipPreCarrot If TRUE will skip a FindTarget that has a '^'
2465 immediately before it.
2466 @param[in] ParameterReplacing If TRUE will add "" around items with spaces.
jcarsey975136a2009-06-16 19:03:54 +00002467
jcarsey969c7832010-01-13 16:46:33 +00002468 @retval EFI_INVALID_PARAMETER SourceString was NULL.
2469 @retval EFI_INVALID_PARAMETER NewString was NULL.
2470 @retval EFI_INVALID_PARAMETER FindTarget was NULL.
2471 @retval EFI_INVALID_PARAMETER ReplaceWith was NULL.
2472 @retval EFI_INVALID_PARAMETER FindTarget had length < 1.
2473 @retval EFI_INVALID_PARAMETER SourceString had length < 1.
jcarsey1e6e84c2010-01-25 20:05:08 +00002474 @retval EFI_BUFFER_TOO_SMALL NewSize was less than the minimum size to hold
jcarsey969c7832010-01-13 16:46:33 +00002475 the new string (truncation occurred).
jcarseya405b862010-09-14 05:18:09 +00002476 @retval EFI_SUCCESS The string was successfully copied with replacement.
jcarsey975136a2009-06-16 19:03:54 +00002477**/
jcarsey975136a2009-06-16 19:03:54 +00002478EFI_STATUS
2479EFIAPI
jcarseya405b862010-09-14 05:18:09 +00002480ShellCopySearchAndReplace(
jcarsey975136a2009-06-16 19:03:54 +00002481 IN CHAR16 CONST *SourceString,
jcarseya405b862010-09-14 05:18:09 +00002482 IN OUT CHAR16 *NewString,
jcarsey975136a2009-06-16 19:03:54 +00002483 IN UINTN NewSize,
2484 IN CONST CHAR16 *FindTarget,
jcarsey969c7832010-01-13 16:46:33 +00002485 IN CONST CHAR16 *ReplaceWith,
jcarseya405b862010-09-14 05:18:09 +00002486 IN CONST BOOLEAN SkipPreCarrot,
2487 IN CONST BOOLEAN ParameterReplacing
jcarsey1e6e84c2010-01-25 20:05:08 +00002488 )
jcarsey2247dde2009-11-09 18:08:58 +00002489{
jcarsey01582942009-07-10 19:46:17 +00002490 UINTN Size;
jcarseya405b862010-09-14 05:18:09 +00002491 CHAR16 *Replace;
2492
jcarsey975136a2009-06-16 19:03:54 +00002493 if ( (SourceString == NULL)
2494 || (NewString == NULL)
2495 || (FindTarget == NULL)
2496 || (ReplaceWith == NULL)
2497 || (StrLen(FindTarget) < 1)
2498 || (StrLen(SourceString) < 1)
jcarseya405b862010-09-14 05:18:09 +00002499 ){
jcarsey975136a2009-06-16 19:03:54 +00002500 return (EFI_INVALID_PARAMETER);
2501 }
jcarseya405b862010-09-14 05:18:09 +00002502 Replace = NULL;
2503 if (StrStr(ReplaceWith, L" ") == NULL || !ParameterReplacing) {
2504 Replace = StrnCatGrow(&Replace, NULL, ReplaceWith, 0);
2505 } else {
2506 Replace = AllocateZeroPool(StrSize(ReplaceWith) + 2*sizeof(CHAR16));
2507 UnicodeSPrint(Replace, StrSize(ReplaceWith) + 2*sizeof(CHAR16), L"\"%s\"", ReplaceWith);
2508 }
jcarsey3e082d52010-10-04 16:44:57 +00002509 if (Replace == NULL) {
2510 return (EFI_OUT_OF_RESOURCES);
2511 }
jcarsey2247dde2009-11-09 18:08:58 +00002512 NewString = SetMem16(NewString, NewSize, CHAR_NULL);
2513 while (*SourceString != CHAR_NULL) {
jcarsey969c7832010-01-13 16:46:33 +00002514 //
jcarseya405b862010-09-14 05:18:09 +00002515 // if we find the FindTarget and either Skip == FALSE or Skip and we
jcarsey969c7832010-01-13 16:46:33 +00002516 // dont have a carrot do a replace...
2517 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002518 if (StrnCmp(SourceString, FindTarget, StrLen(FindTarget)) == 0
jcarseya405b862010-09-14 05:18:09 +00002519 && ((SkipPreCarrot && *(SourceString-1) != L'^') || !SkipPreCarrot)
2520 ){
jcarsey975136a2009-06-16 19:03:54 +00002521 SourceString += StrLen(FindTarget);
jcarsey01582942009-07-10 19:46:17 +00002522 Size = StrSize(NewString);
jcarseya405b862010-09-14 05:18:09 +00002523 if ((Size + (StrLen(Replace)*sizeof(CHAR16))) > NewSize) {
2524 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002525 return (EFI_BUFFER_TOO_SMALL);
2526 }
jcarseya405b862010-09-14 05:18:09 +00002527 StrCat(NewString, Replace);
jcarsey975136a2009-06-16 19:03:54 +00002528 } else {
jcarsey01582942009-07-10 19:46:17 +00002529 Size = StrSize(NewString);
2530 if (Size + sizeof(CHAR16) > NewSize) {
jcarseya405b862010-09-14 05:18:09 +00002531 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002532 return (EFI_BUFFER_TOO_SMALL);
2533 }
2534 StrnCat(NewString, SourceString, 1);
2535 SourceString++;
2536 }
2537 }
jcarseya405b862010-09-14 05:18:09 +00002538 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002539 return (EFI_SUCCESS);
2540}
jcarseyb1f95a02009-06-16 00:23:19 +00002541
2542/**
jcarseye2f82972009-12-01 05:40:24 +00002543 Internal worker function to output a string.
2544
2545 This function will output a string to the correct StdOut.
2546
2547 @param[in] String The string to print out.
2548
2549 @retval EFI_SUCCESS The operation was sucessful.
2550 @retval !EFI_SUCCESS The operation failed.
2551**/
2552EFI_STATUS
2553EFIAPI
2554InternalPrintTo (
2555 IN CONST CHAR16 *String
2556 )
2557{
2558 UINTN Size;
2559 Size = StrSize(String) - sizeof(CHAR16);
jcarseya405b862010-09-14 05:18:09 +00002560 if (Size == 0) {
2561 return (EFI_SUCCESS);
2562 }
jcarsey366f81a2011-06-27 21:04:22 +00002563 if (gEfiShellParametersProtocol != NULL) {
2564 return (gEfiShellProtocol->WriteFile(gEfiShellParametersProtocol->StdOut, &Size, (VOID*)String));
jcarseye2f82972009-12-01 05:40:24 +00002565 }
2566 if (mEfiShellInterface != NULL) {
jcarseyecd3d592009-12-07 18:05:00 +00002567 //
2568 // Divide in half for old shell. Must be string length not size.
2569 //
2570 Size /= 2;
jcarseya405b862010-09-14 05:18:09 +00002571 return (mEfiShellInterface->StdOut->Write(mEfiShellInterface->StdOut, &Size, (VOID*)String));
jcarseye2f82972009-12-01 05:40:24 +00002572 }
2573 ASSERT(FALSE);
2574 return (EFI_UNSUPPORTED);
2575}
2576
2577/**
jcarseyb1f95a02009-06-16 00:23:19 +00002578 Print at a specific location on the screen.
2579
jcarseyf1b87e72009-06-17 00:52:11 +00002580 This function will move the cursor to a given screen location and print the specified string
jcarsey1e6e84c2010-01-25 20:05:08 +00002581
2582 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarseyf1b87e72009-06-17 00:52:11 +00002583 will be used.
jcarseyb1f95a02009-06-16 00:23:19 +00002584
2585 if either Row or Col is out of range for the current console, then ASSERT
2586 if Format is NULL, then ASSERT
2587
jcarsey1e6e84c2010-01-25 20:05:08 +00002588 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarseyb1f95a02009-06-16 00:23:19 +00002589 the following additional flags:
2590 %N - Set output attribute to normal
2591 %H - Set output attribute to highlight
2592 %E - Set output attribute to error
2593 %B - Set output attribute to blue color
2594 %V - Set output attribute to green color
2595
2596 Note: The background color is controlled by the shell command cls.
2597
jcarseyb1f95a02009-06-16 00:23:19 +00002598 @param[in] Col the column to print at
jcarsey252d9452011-03-25 20:49:53 +00002599 @param[in] Row the row to print at
jcarseyb1f95a02009-06-16 00:23:19 +00002600 @param[in] Format the format string
jcarsey2247dde2009-11-09 18:08:58 +00002601 @param[in] Marker the marker for the variable argument list
jcarseyb1f95a02009-06-16 00:23:19 +00002602
jcarseya405b862010-09-14 05:18:09 +00002603 @return EFI_SUCCESS The operation was successful.
2604 @return EFI_DEVICE_ERROR The console device reported an error.
jcarseyb1f95a02009-06-16 00:23:19 +00002605**/
jcarseya405b862010-09-14 05:18:09 +00002606EFI_STATUS
jcarseyb1f95a02009-06-16 00:23:19 +00002607EFIAPI
jcarsey2247dde2009-11-09 18:08:58 +00002608InternalShellPrintWorker(
jcarseyb1f95a02009-06-16 00:23:19 +00002609 IN INT32 Col OPTIONAL,
2610 IN INT32 Row OPTIONAL,
2611 IN CONST CHAR16 *Format,
jcarsey252d9452011-03-25 20:49:53 +00002612 IN VA_LIST Marker
jcarsey1e6e84c2010-01-25 20:05:08 +00002613 )
jcarsey2247dde2009-11-09 18:08:58 +00002614{
jcarseyb1f95a02009-06-16 00:23:19 +00002615 EFI_STATUS Status;
jcarsey975136a2009-06-16 19:03:54 +00002616 CHAR16 *ResumeLocation;
2617 CHAR16 *FormatWalker;
jcarseya405b862010-09-14 05:18:09 +00002618 UINTN OriginalAttribute;
jcarsey89e85372011-04-13 23:37:50 +00002619 CHAR16 *mPostReplaceFormat;
2620 CHAR16 *mPostReplaceFormat2;
2621
2622 mPostReplaceFormat = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
2623 mPostReplaceFormat2 = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
jcarseya405b862010-09-14 05:18:09 +00002624
jcarseyf8d3e682011-04-19 17:54:42 +00002625 if (mPostReplaceFormat == NULL || mPostReplaceFormat2 == NULL) {
2626 SHELL_FREE_NON_NULL(mPostReplaceFormat);
2627 SHELL_FREE_NON_NULL(mPostReplaceFormat2);
2628 return (EFI_OUT_OF_RESOURCES);
2629 }
2630
jcarseya405b862010-09-14 05:18:09 +00002631 Status = EFI_SUCCESS;
2632 OriginalAttribute = gST->ConOut->Mode->Attribute;
jcarsey1e6e84c2010-01-25 20:05:08 +00002633
jcarsey975136a2009-06-16 19:03:54 +00002634 //
2635 // Back and forth each time fixing up 1 of our flags...
2636 //
jcarseya405b862010-09-14 05:18:09 +00002637 Status = ShellCopySearchAndReplace(Format, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%N", L"%%N", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002638 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002639 Status = ShellCopySearchAndReplace(mPostReplaceFormat, mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), L"%E", L"%%E", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002640 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002641 Status = ShellCopySearchAndReplace(mPostReplaceFormat2, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%H", L"%%H", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002642 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002643 Status = ShellCopySearchAndReplace(mPostReplaceFormat, mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), L"%B", L"%%B", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002644 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002645 Status = ShellCopySearchAndReplace(mPostReplaceFormat2, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%V", L"%%V", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002646 ASSERT_EFI_ERROR(Status);
2647
2648 //
2649 // Use the last buffer from replacing to print from...
2650 //
jcarseya405b862010-09-14 05:18:09 +00002651 UnicodeVSPrint (mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), mPostReplaceFormat, Marker);
jcarseyb1f95a02009-06-16 00:23:19 +00002652
2653 if (Col != -1 && Row != -1) {
jcarseyb1f95a02009-06-16 00:23:19 +00002654 Status = gST->ConOut->SetCursorPosition(gST->ConOut, Col, Row);
jcarsey975136a2009-06-16 19:03:54 +00002655 }
2656
jcarseyecd3d592009-12-07 18:05:00 +00002657 FormatWalker = mPostReplaceFormat2;
jcarsey2247dde2009-11-09 18:08:58 +00002658 while (*FormatWalker != CHAR_NULL) {
jcarsey975136a2009-06-16 19:03:54 +00002659 //
2660 // Find the next attribute change request
2661 //
2662 ResumeLocation = StrStr(FormatWalker, L"%");
2663 if (ResumeLocation != NULL) {
jcarsey2247dde2009-11-09 18:08:58 +00002664 *ResumeLocation = CHAR_NULL;
jcarsey975136a2009-06-16 19:03:54 +00002665 }
2666 //
2667 // print the current FormatWalker string
2668 //
jcarseya405b862010-09-14 05:18:09 +00002669 if (StrLen(FormatWalker)>0) {
2670 Status = InternalPrintTo(FormatWalker);
2671 if (EFI_ERROR(Status)) {
2672 break;
2673 }
2674 }
2675
jcarsey975136a2009-06-16 19:03:54 +00002676 //
2677 // update the attribute
2678 //
2679 if (ResumeLocation != NULL) {
jcarsey5d46f172011-08-23 15:34:23 +00002680 if (*(ResumeLocation-1) == L'^') {
2681 //
2682 // Print a simple '%' symbol
2683 //
2684 Status = InternalPrintTo(L"%");
2685 ResumeLocation = ResumeLocation - 1;
2686 } else {
2687 switch (*(ResumeLocation+1)) {
2688 case (L'N'):
2689 gST->ConOut->SetAttribute(gST->ConOut, OriginalAttribute);
jcarseya405b862010-09-14 05:18:09 +00002690 break;
jcarsey5d46f172011-08-23 15:34:23 +00002691 case (L'E'):
2692 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_YELLOW, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2693 break;
2694 case (L'H'):
2695 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_WHITE, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2696 break;
2697 case (L'B'):
2698 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_BLUE, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2699 break;
2700 case (L'V'):
2701 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_GREEN, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2702 break;
2703 default:
2704 //
2705 // Print a simple '%' symbol
2706 //
2707 Status = InternalPrintTo(L"%");
2708 if (EFI_ERROR(Status)) {
2709 break;
2710 }
2711 ResumeLocation = ResumeLocation - 1;
2712 break;
2713 }
jcarsey975136a2009-06-16 19:03:54 +00002714 }
2715 } else {
2716 //
2717 // reset to normal now...
2718 //
jcarsey975136a2009-06-16 19:03:54 +00002719 break;
2720 }
2721
2722 //
2723 // update FormatWalker to Resume + 2 (skip the % and the indicator)
2724 //
2725 FormatWalker = ResumeLocation + 2;
2726 }
jcarseyb1f95a02009-06-16 00:23:19 +00002727
jcarseya405b862010-09-14 05:18:09 +00002728 gST->ConOut->SetAttribute(gST->ConOut, OriginalAttribute);
jcarsey89e85372011-04-13 23:37:50 +00002729
2730 SHELL_FREE_NON_NULL(mPostReplaceFormat);
2731 SHELL_FREE_NON_NULL(mPostReplaceFormat2);
jcarseya405b862010-09-14 05:18:09 +00002732 return (Status);
jcarsey5f7431d2009-07-10 18:06:01 +00002733}
jcarsey2247dde2009-11-09 18:08:58 +00002734
2735/**
2736 Print at a specific location on the screen.
2737
jcarseye2f82972009-12-01 05:40:24 +00002738 This function will move the cursor to a given screen location and print the specified string.
jcarsey1e6e84c2010-01-25 20:05:08 +00002739
2740 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarsey2247dde2009-11-09 18:08:58 +00002741 will be used.
2742
jcarseye2f82972009-12-01 05:40:24 +00002743 If either Row or Col is out of range for the current console, then ASSERT.
2744 If Format is NULL, then ASSERT.
jcarsey2247dde2009-11-09 18:08:58 +00002745
jcarsey1e6e84c2010-01-25 20:05:08 +00002746 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarsey2247dde2009-11-09 18:08:58 +00002747 the following additional flags:
2748 %N - Set output attribute to normal
2749 %H - Set output attribute to highlight
2750 %E - Set output attribute to error
2751 %B - Set output attribute to blue color
2752 %V - Set output attribute to green color
2753
2754 Note: The background color is controlled by the shell command cls.
2755
jcarsey2247dde2009-11-09 18:08:58 +00002756 @param[in] Col the column to print at
jcarseya405b862010-09-14 05:18:09 +00002757 @param[in] Row the row to print at
jcarsey2247dde2009-11-09 18:08:58 +00002758 @param[in] Format the format string
jcarseya405b862010-09-14 05:18:09 +00002759 @param[in] ... The variable argument list.
jcarsey2247dde2009-11-09 18:08:58 +00002760
jcarseya405b862010-09-14 05:18:09 +00002761 @return EFI_SUCCESS The printing was successful.
2762 @return EFI_DEVICE_ERROR The console device reported an error.
jcarsey2247dde2009-11-09 18:08:58 +00002763**/
jcarseya405b862010-09-14 05:18:09 +00002764EFI_STATUS
jcarsey2247dde2009-11-09 18:08:58 +00002765EFIAPI
2766ShellPrintEx(
2767 IN INT32 Col OPTIONAL,
2768 IN INT32 Row OPTIONAL,
2769 IN CONST CHAR16 *Format,
2770 ...
jcarsey1e6e84c2010-01-25 20:05:08 +00002771 )
jcarsey2247dde2009-11-09 18:08:58 +00002772{
2773 VA_LIST Marker;
jcarseya405b862010-09-14 05:18:09 +00002774 EFI_STATUS RetVal;
jcarsey3e082d52010-10-04 16:44:57 +00002775 if (Format == NULL) {
2776 return (EFI_INVALID_PARAMETER);
2777 }
jcarsey2247dde2009-11-09 18:08:58 +00002778 VA_START (Marker, Format);
jcarseya405b862010-09-14 05:18:09 +00002779 RetVal = InternalShellPrintWorker(Col, Row, Format, Marker);
jcarseye2f82972009-12-01 05:40:24 +00002780 VA_END(Marker);
jcarseya405b862010-09-14 05:18:09 +00002781 return(RetVal);
jcarsey2247dde2009-11-09 18:08:58 +00002782}
2783
2784/**
2785 Print at a specific location on the screen.
2786
jcarseye2f82972009-12-01 05:40:24 +00002787 This function will move the cursor to a given screen location and print the specified string.
jcarsey1e6e84c2010-01-25 20:05:08 +00002788
2789 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarseye2f82972009-12-01 05:40:24 +00002790 will be used.
jcarsey2247dde2009-11-09 18:08:58 +00002791
jcarseye2f82972009-12-01 05:40:24 +00002792 If either Row or Col is out of range for the current console, then ASSERT.
2793 If Format is NULL, then ASSERT.
jcarsey2247dde2009-11-09 18:08:58 +00002794
jcarsey1e6e84c2010-01-25 20:05:08 +00002795 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarsey2247dde2009-11-09 18:08:58 +00002796 the following additional flags:
jcarsey1e6e84c2010-01-25 20:05:08 +00002797 %N - Set output attribute to normal.
2798 %H - Set output attribute to highlight.
2799 %E - Set output attribute to error.
2800 %B - Set output attribute to blue color.
2801 %V - Set output attribute to green color.
jcarsey2247dde2009-11-09 18:08:58 +00002802
2803 Note: The background color is controlled by the shell command cls.
2804
jcarsey1e6e84c2010-01-25 20:05:08 +00002805 @param[in] Col The column to print at.
jcarseya405b862010-09-14 05:18:09 +00002806 @param[in] Row The row to print at.
jcarsey1e6e84c2010-01-25 20:05:08 +00002807 @param[in] Language The language of the string to retrieve. If this parameter
2808 is NULL, then the current platform language is used.
2809 @param[in] HiiFormatStringId The format string Id for getting from Hii.
2810 @param[in] HiiFormatHandle The format string Handle for getting from Hii.
jcarseya405b862010-09-14 05:18:09 +00002811 @param[in] ... The variable argument list.
jcarsey2247dde2009-11-09 18:08:58 +00002812
jcarseya405b862010-09-14 05:18:09 +00002813 @return EFI_SUCCESS The printing was successful.
2814 @return EFI_DEVICE_ERROR The console device reported an error.
jcarsey2247dde2009-11-09 18:08:58 +00002815**/
jcarseya405b862010-09-14 05:18:09 +00002816EFI_STATUS
jcarsey2247dde2009-11-09 18:08:58 +00002817EFIAPI
2818ShellPrintHiiEx(
2819 IN INT32 Col OPTIONAL,
2820 IN INT32 Row OPTIONAL,
jcarsey1e6e84c2010-01-25 20:05:08 +00002821 IN CONST CHAR8 *Language OPTIONAL,
jcarsey2247dde2009-11-09 18:08:58 +00002822 IN CONST EFI_STRING_ID HiiFormatStringId,
2823 IN CONST EFI_HANDLE HiiFormatHandle,
2824 ...
2825 )
2826{
2827 VA_LIST Marker;
2828 CHAR16 *HiiFormatString;
jcarseya405b862010-09-14 05:18:09 +00002829 EFI_STATUS RetVal;
jcarsey2247dde2009-11-09 18:08:58 +00002830
2831 VA_START (Marker, HiiFormatHandle);
jcarsey1e6e84c2010-01-25 20:05:08 +00002832 HiiFormatString = HiiGetString(HiiFormatHandle, HiiFormatStringId, Language);
jcarsey2247dde2009-11-09 18:08:58 +00002833 ASSERT(HiiFormatString != NULL);
2834
2835 RetVal = InternalShellPrintWorker(Col, Row, HiiFormatString, Marker);
2836
jcarseya405b862010-09-14 05:18:09 +00002837 SHELL_FREE_NON_NULL(HiiFormatString);
jcarseye2f82972009-12-01 05:40:24 +00002838 VA_END(Marker);
jcarsey2247dde2009-11-09 18:08:58 +00002839
2840 return (RetVal);
2841}
2842
2843/**
2844 Function to determine if a given filename represents a file or a directory.
2845
2846 @param[in] DirName Path to directory to test.
2847
2848 @retval EFI_SUCCESS The Path represents a directory
2849 @retval EFI_NOT_FOUND The Path does not represent a directory
2850 @return other The path failed to open
2851**/
2852EFI_STATUS
2853EFIAPI
2854ShellIsDirectory(
2855 IN CONST CHAR16 *DirName
2856 )
2857{
2858 EFI_STATUS Status;
jcarseya405b862010-09-14 05:18:09 +00002859 SHELL_FILE_HANDLE Handle;
jcarsey3e082d52010-10-04 16:44:57 +00002860 CHAR16 *TempLocation;
2861 CHAR16 *TempLocation2;
jcarsey2247dde2009-11-09 18:08:58 +00002862
jcarseyecd3d592009-12-07 18:05:00 +00002863 ASSERT(DirName != NULL);
2864
jcarseya405b862010-09-14 05:18:09 +00002865 Handle = NULL;
2866 TempLocation = NULL;
jcarsey2247dde2009-11-09 18:08:58 +00002867
2868 Status = ShellOpenFileByName(DirName, &Handle, EFI_FILE_MODE_READ, 0);
2869 if (EFI_ERROR(Status)) {
jcarseya405b862010-09-14 05:18:09 +00002870 //
2871 // try good logic first.
2872 //
jcarsey366f81a2011-06-27 21:04:22 +00002873 if (gEfiShellProtocol != NULL) {
jcarsey3e082d52010-10-04 16:44:57 +00002874 TempLocation = StrnCatGrow(&TempLocation, NULL, DirName, 0);
2875 TempLocation2 = StrStr(TempLocation, L":");
2876 if (TempLocation2 != NULL && StrLen(StrStr(TempLocation, L":")) == 2) {
2877 *(TempLocation2+1) = CHAR_NULL;
jcarseya405b862010-09-14 05:18:09 +00002878 }
jcarsey366f81a2011-06-27 21:04:22 +00002879 if (gEfiShellProtocol->GetDevicePathFromMap(TempLocation) != NULL) {
jcarseya405b862010-09-14 05:18:09 +00002880 FreePool(TempLocation);
2881 return (EFI_SUCCESS);
2882 }
2883 FreePool(TempLocation);
2884 } else {
2885 //
2886 // probably a map name?!?!!?
2887 //
2888 TempLocation = StrStr(DirName, L"\\");
2889 if (TempLocation != NULL && *(TempLocation+1) == CHAR_NULL) {
2890 return (EFI_SUCCESS);
2891 }
2892 }
jcarsey2247dde2009-11-09 18:08:58 +00002893 return (Status);
2894 }
2895
2896 if (FileHandleIsDirectory(Handle) == EFI_SUCCESS) {
2897 ShellCloseFile(&Handle);
2898 return (EFI_SUCCESS);
2899 }
2900 ShellCloseFile(&Handle);
2901 return (EFI_NOT_FOUND);
2902}
2903
jcarsey125c2cf2009-11-18 21:36:50 +00002904/**
jcarsey36a9d672009-11-20 21:13:41 +00002905 Function to determine if a given filename represents a file.
2906
2907 @param[in] Name Path to file to test.
2908
2909 @retval EFI_SUCCESS The Path represents a file.
2910 @retval EFI_NOT_FOUND The Path does not represent a file.
2911 @retval other The path failed to open.
2912**/
2913EFI_STATUS
2914EFIAPI
2915ShellIsFile(
2916 IN CONST CHAR16 *Name
2917 )
2918{
2919 EFI_STATUS Status;
jcarseya405b862010-09-14 05:18:09 +00002920 SHELL_FILE_HANDLE Handle;
jcarsey36a9d672009-11-20 21:13:41 +00002921
jcarseyecd3d592009-12-07 18:05:00 +00002922 ASSERT(Name != NULL);
2923
jcarsey36a9d672009-11-20 21:13:41 +00002924 Handle = NULL;
2925
2926 Status = ShellOpenFileByName(Name, &Handle, EFI_FILE_MODE_READ, 0);
2927 if (EFI_ERROR(Status)) {
2928 return (Status);
2929 }
2930
2931 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
2932 ShellCloseFile(&Handle);
2933 return (EFI_SUCCESS);
2934 }
2935 ShellCloseFile(&Handle);
2936 return (EFI_NOT_FOUND);
2937}
2938
2939/**
jcarseyb3011f42010-01-11 21:49:04 +00002940 Function to determine if a given filename represents a file.
2941
2942 This will search the CWD and then the Path.
2943
2944 If Name is NULL, then ASSERT.
2945
2946 @param[in] Name Path to file to test.
2947
2948 @retval EFI_SUCCESS The Path represents a file.
2949 @retval EFI_NOT_FOUND The Path does not represent a file.
2950 @retval other The path failed to open.
2951**/
2952EFI_STATUS
2953EFIAPI
2954ShellIsFileInPath(
2955 IN CONST CHAR16 *Name
jcarseya405b862010-09-14 05:18:09 +00002956 )
2957{
jcarseyb3011f42010-01-11 21:49:04 +00002958 CHAR16 *NewName;
2959 EFI_STATUS Status;
2960
2961 if (!EFI_ERROR(ShellIsFile(Name))) {
jcarseya405b862010-09-14 05:18:09 +00002962 return (EFI_SUCCESS);
jcarseyb3011f42010-01-11 21:49:04 +00002963 }
2964
2965 NewName = ShellFindFilePath(Name);
2966 if (NewName == NULL) {
2967 return (EFI_NOT_FOUND);
2968 }
2969 Status = ShellIsFile(NewName);
2970 FreePool(NewName);
2971 return (Status);
2972}
jcarsey252d9452011-03-25 20:49:53 +00002973
jcarseyb3011f42010-01-11 21:49:04 +00002974/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002975 Function to determine whether a string is decimal or hex representation of a number
jcarsey125c2cf2009-11-18 21:36:50 +00002976 and return the number converted from the string.
2977
2978 @param[in] String String representation of a number
2979
jcarsey252d9452011-03-25 20:49:53 +00002980 @return the number
2981 @retval (UINTN)(-1) An error ocurred.
jcarsey125c2cf2009-11-18 21:36:50 +00002982**/
2983UINTN
2984EFIAPI
2985ShellStrToUintn(
2986 IN CONST CHAR16 *String
2987 )
2988{
jcarsey252d9452011-03-25 20:49:53 +00002989 UINT64 RetVal;
2990 BOOLEAN Hex;
2991
2992 Hex = FALSE;
2993
2994 if (!InternalShellIsHexOrDecimalNumber(String, Hex, TRUE)) {
2995 Hex = TRUE;
jcarsey125c2cf2009-11-18 21:36:50 +00002996 }
jcarsey252d9452011-03-25 20:49:53 +00002997
2998 if (!EFI_ERROR(ShellConvertStringToUint64(String, &RetVal, Hex, TRUE))) {
2999 return ((UINTN)RetVal);
3000 }
3001 return ((UINTN)(-1));
jcarsey125c2cf2009-11-18 21:36:50 +00003002}
3003
3004/**
3005 Safely append with automatic string resizing given length of Destination and
3006 desired length of copy from Source.
3007
3008 append the first D characters of Source to the end of Destination, where D is
3009 the lesser of Count and the StrLen() of Source. If appending those D characters
3010 will fit within Destination (whose Size is given as CurrentSize) and
jcarsey1e6e84c2010-01-25 20:05:08 +00003011 still leave room for a NULL terminator, then those characters are appended,
3012 starting at the original terminating NULL of Destination, and a new terminating
3013 NULL is appended.
jcarsey125c2cf2009-11-18 21:36:50 +00003014
3015 If appending D characters onto Destination will result in a overflow of the size
3016 given in CurrentSize the string will be grown such that the copy can be performed
3017 and CurrentSize will be updated to the new size.
3018
3019 If Source is NULL, there is nothing to append, just return the current buffer in
3020 Destination.
3021
3022 if Destination is NULL, then ASSERT()
3023 if Destination's current length (including NULL terminator) is already more then
3024 CurrentSize, then ASSERT()
3025
ydong104ff7e372011-09-02 08:05:34 +00003026 @param[in, out] Destination The String to append onto
3027 @param[in, out] CurrentSize on call the number of bytes in Destination. On
jcarsey125c2cf2009-11-18 21:36:50 +00003028 return possibly the new size (still in bytes). if NULL
3029 then allocate whatever is needed.
3030 @param[in] Source The String to append from
3031 @param[in] Count Maximum number of characters to append. if 0 then
3032 all are appended.
3033
3034 @return Destination return the resultant string.
3035**/
3036CHAR16*
3037EFIAPI
3038StrnCatGrow (
3039 IN OUT CHAR16 **Destination,
3040 IN OUT UINTN *CurrentSize,
3041 IN CONST CHAR16 *Source,
3042 IN UINTN Count
3043 )
3044{
3045 UINTN DestinationStartSize;
3046 UINTN NewSize;
3047
3048 //
3049 // ASSERTs
3050 //
3051 ASSERT(Destination != NULL);
3052
3053 //
3054 // If there's nothing to do then just return Destination
3055 //
3056 if (Source == NULL) {
3057 return (*Destination);
3058 }
3059
3060 //
3061 // allow for un-initialized pointers, based on size being 0
3062 //
3063 if (CurrentSize != NULL && *CurrentSize == 0) {
3064 *Destination = NULL;
3065 }
3066
3067 //
3068 // allow for NULL pointers address as Destination
3069 //
3070 if (*Destination != NULL) {
3071 ASSERT(CurrentSize != 0);
3072 DestinationStartSize = StrSize(*Destination);
3073 ASSERT(DestinationStartSize <= *CurrentSize);
3074 } else {
3075 DestinationStartSize = 0;
3076// ASSERT(*CurrentSize == 0);
3077 }
3078
3079 //
3080 // Append all of Source?
3081 //
3082 if (Count == 0) {
3083 Count = StrLen(Source);
3084 }
3085
3086 //
3087 // Test and grow if required
3088 //
3089 if (CurrentSize != NULL) {
3090 NewSize = *CurrentSize;
3091 while (NewSize < (DestinationStartSize + (Count*sizeof(CHAR16)))) {
3092 NewSize += 2 * Count * sizeof(CHAR16);
3093 }
3094 *Destination = ReallocatePool(*CurrentSize, NewSize, *Destination);
jcarseyc9d92df2010-02-03 15:37:54 +00003095 ASSERT(*Destination != NULL);
jcarsey125c2cf2009-11-18 21:36:50 +00003096 *CurrentSize = NewSize;
3097 } else {
3098 *Destination = AllocateZeroPool((Count+1)*sizeof(CHAR16));
jcarseyc9d92df2010-02-03 15:37:54 +00003099 ASSERT(*Destination != NULL);
jcarsey125c2cf2009-11-18 21:36:50 +00003100 }
3101
3102 //
3103 // Now use standard StrnCat on a big enough buffer
3104 //
jcarseyc9d92df2010-02-03 15:37:54 +00003105 if (*Destination == NULL) {
3106 return (NULL);
3107 }
jcarsey125c2cf2009-11-18 21:36:50 +00003108 return StrnCat(*Destination, Source, Count);
3109}
jcarseyc9d92df2010-02-03 15:37:54 +00003110
3111/**
3112 Prompt the user and return the resultant answer to the requestor.
3113
3114 This function will display the requested question on the shell prompt and then
3115 wait for an apropriate answer to be input from the console.
3116
jcarseya405b862010-09-14 05:18:09 +00003117 if the SHELL_PROMPT_REQUEST_TYPE is SHELL_PROMPT_REQUEST_TYPE_YESNO, ShellPromptResponseTypeQuitContinue
jcarseyc9d92df2010-02-03 15:37:54 +00003118 or SHELL_PROMPT_REQUEST_TYPE_YESNOCANCEL then *Response is of type SHELL_PROMPT_RESPONSE.
3119
jcarseya405b862010-09-14 05:18:09 +00003120 if the SHELL_PROMPT_REQUEST_TYPE is ShellPromptResponseTypeFreeform then *Response is of type
jcarseyc9d92df2010-02-03 15:37:54 +00003121 CHAR16*.
3122
3123 In either case *Response must be callee freed if Response was not NULL;
3124
3125 @param Type What type of question is asked. This is used to filter the input
3126 to prevent invalid answers to question.
3127 @param Prompt Pointer to string prompt to use to request input.
3128 @param Response Pointer to Response which will be populated upon return.
3129
3130 @retval EFI_SUCCESS The operation was sucessful.
3131 @retval EFI_UNSUPPORTED The operation is not supported as requested.
3132 @retval EFI_INVALID_PARAMETER A parameter was invalid.
3133 @return other The operation failed.
3134**/
3135EFI_STATUS
3136EFIAPI
3137ShellPromptForResponse (
3138 IN SHELL_PROMPT_REQUEST_TYPE Type,
3139 IN CHAR16 *Prompt OPTIONAL,
3140 IN OUT VOID **Response OPTIONAL
3141 )
3142{
3143 EFI_STATUS Status;
3144 EFI_INPUT_KEY Key;
3145 UINTN EventIndex;
3146 SHELL_PROMPT_RESPONSE *Resp;
jcarseya405b862010-09-14 05:18:09 +00003147 UINTN Size;
3148 CHAR16 *Buffer;
jcarseyc9d92df2010-02-03 15:37:54 +00003149
jcarseya405b862010-09-14 05:18:09 +00003150 Status = EFI_UNSUPPORTED;
3151 Resp = NULL;
3152 Buffer = NULL;
3153 Size = 0;
3154 if (Type != ShellPromptResponseTypeFreeform) {
jcarsey252d9452011-03-25 20:49:53 +00003155 Resp = (SHELL_PROMPT_RESPONSE*)AllocateZeroPool(sizeof(SHELL_PROMPT_RESPONSE));
jcarsey3e082d52010-10-04 16:44:57 +00003156 if (Resp == NULL) {
3157 return (EFI_OUT_OF_RESOURCES);
3158 }
xdu290bfa222010-07-19 05:21:27 +00003159 }
jcarseyc9d92df2010-02-03 15:37:54 +00003160
3161 switch(Type) {
jcarseya405b862010-09-14 05:18:09 +00003162 case ShellPromptResponseTypeQuitContinue:
jcarseyc9d92df2010-02-03 15:37:54 +00003163 if (Prompt != NULL) {
3164 ShellPrintEx(-1, -1, L"%s", Prompt);
3165 }
3166 //
3167 // wait for valid response
3168 //
3169 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3170 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3171 ASSERT_EFI_ERROR(Status);
3172 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3173 if (Key.UnicodeChar == L'Q' || Key.UnicodeChar ==L'q') {
jcarseya405b862010-09-14 05:18:09 +00003174 *Resp = ShellPromptResponseQuit;
jcarseyc9d92df2010-02-03 15:37:54 +00003175 } else {
jcarseya405b862010-09-14 05:18:09 +00003176 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003177 }
3178 break;
jcarseya405b862010-09-14 05:18:09 +00003179 case ShellPromptResponseTypeYesNoCancel:
jcarseyc9d92df2010-02-03 15:37:54 +00003180 if (Prompt != NULL) {
3181 ShellPrintEx(-1, -1, L"%s", Prompt);
3182 }
3183 //
3184 // wait for valid response
3185 //
jcarseya405b862010-09-14 05:18:09 +00003186 *Resp = ShellPromptResponseMax;
3187 while (*Resp == ShellPromptResponseMax) {
jcarseyc9d92df2010-02-03 15:37:54 +00003188 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3189 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3190 ASSERT_EFI_ERROR(Status);
3191 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3192 switch (Key.UnicodeChar) {
3193 case L'Y':
3194 case L'y':
jcarseya405b862010-09-14 05:18:09 +00003195 *Resp = ShellPromptResponseYes;
jcarseyc9d92df2010-02-03 15:37:54 +00003196 break;
3197 case L'N':
3198 case L'n':
jcarseya405b862010-09-14 05:18:09 +00003199 *Resp = ShellPromptResponseNo;
jcarseyc9d92df2010-02-03 15:37:54 +00003200 break;
3201 case L'C':
3202 case L'c':
jcarseya405b862010-09-14 05:18:09 +00003203 *Resp = ShellPromptResponseCancel;
3204 break;
3205 }
3206 }
3207 break; case ShellPromptResponseTypeYesNoAllCancel:
3208 if (Prompt != NULL) {
3209 ShellPrintEx(-1, -1, L"%s", Prompt);
3210 }
3211 //
3212 // wait for valid response
3213 //
3214 *Resp = ShellPromptResponseMax;
3215 while (*Resp == ShellPromptResponseMax) {
3216 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3217 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3218 ASSERT_EFI_ERROR(Status);
3219 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3220 switch (Key.UnicodeChar) {
3221 case L'Y':
3222 case L'y':
3223 *Resp = ShellPromptResponseYes;
3224 break;
3225 case L'N':
3226 case L'n':
3227 *Resp = ShellPromptResponseNo;
3228 break;
3229 case L'A':
3230 case L'a':
3231 *Resp = ShellPromptResponseAll;
3232 break;
3233 case L'C':
3234 case L'c':
3235 *Resp = ShellPromptResponseCancel;
jcarseyc9d92df2010-02-03 15:37:54 +00003236 break;
3237 }
3238 }
3239 break;
jcarseya405b862010-09-14 05:18:09 +00003240 case ShellPromptResponseTypeEnterContinue:
3241 case ShellPromptResponseTypeAnyKeyContinue:
jcarseyc9d92df2010-02-03 15:37:54 +00003242 if (Prompt != NULL) {
3243 ShellPrintEx(-1, -1, L"%s", Prompt);
3244 }
3245 //
3246 // wait for valid response
3247 //
jcarseya405b862010-09-14 05:18:09 +00003248 *Resp = ShellPromptResponseMax;
3249 while (*Resp == ShellPromptResponseMax) {
jcarseyc9d92df2010-02-03 15:37:54 +00003250 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
jcarseya405b862010-09-14 05:18:09 +00003251 if (Type == ShellPromptResponseTypeEnterContinue) {
jcarseyc9d92df2010-02-03 15:37:54 +00003252 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3253 ASSERT_EFI_ERROR(Status);
3254 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3255 if (Key.UnicodeChar == CHAR_CARRIAGE_RETURN) {
jcarseya405b862010-09-14 05:18:09 +00003256 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003257 break;
3258 }
3259 }
jcarseya405b862010-09-14 05:18:09 +00003260 if (Type == ShellPromptResponseTypeAnyKeyContinue) {
3261 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3262 ASSERT_EFI_ERROR(Status);
3263 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003264 break;
3265 }
3266 }
3267 break;
jcarseya405b862010-09-14 05:18:09 +00003268 case ShellPromptResponseTypeYesNo:
3269 if (Prompt != NULL) {
3270 ShellPrintEx(-1, -1, L"%s", Prompt);
3271 }
3272 //
3273 // wait for valid response
3274 //
3275 *Resp = ShellPromptResponseMax;
3276 while (*Resp == ShellPromptResponseMax) {
3277 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3278 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3279 ASSERT_EFI_ERROR(Status);
3280 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3281 switch (Key.UnicodeChar) {
3282 case L'Y':
3283 case L'y':
3284 *Resp = ShellPromptResponseYes;
3285 break;
3286 case L'N':
3287 case L'n':
3288 *Resp = ShellPromptResponseNo;
3289 break;
3290 }
3291 }
3292 break;
3293 case ShellPromptResponseTypeFreeform:
3294 if (Prompt != NULL) {
3295 ShellPrintEx(-1, -1, L"%s", Prompt);
3296 }
3297 while(1) {
3298 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3299 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3300 ASSERT_EFI_ERROR(Status);
3301 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3302 if (Key.UnicodeChar == CHAR_CARRIAGE_RETURN) {
3303 break;
3304 }
3305 ASSERT((Buffer == NULL && Size == 0) || (Buffer != NULL));
3306 StrnCatGrow(&Buffer, &Size, &Key.UnicodeChar, 1);
3307 }
3308 break;
3309 //
3310 // This is the location to add new prompt types.
3311 //
jcarseyc9d92df2010-02-03 15:37:54 +00003312 default:
jcarseya405b862010-09-14 05:18:09 +00003313 ASSERT(FALSE);
jcarseyc9d92df2010-02-03 15:37:54 +00003314 }
3315
3316 if (Response != NULL) {
jcarseya405b862010-09-14 05:18:09 +00003317 if (Resp != NULL) {
3318 *Response = Resp;
3319 } else if (Buffer != NULL) {
3320 *Response = Buffer;
3321 }
jcarseyc9d92df2010-02-03 15:37:54 +00003322 } else {
jcarseya405b862010-09-14 05:18:09 +00003323 if (Resp != NULL) {
3324 FreePool(Resp);
3325 }
3326 if (Buffer != NULL) {
3327 FreePool(Buffer);
3328 }
jcarseyc9d92df2010-02-03 15:37:54 +00003329 }
3330
jcarseya405b862010-09-14 05:18:09 +00003331 ShellPrintEx(-1, -1, L"\r\n");
jcarseyc9d92df2010-02-03 15:37:54 +00003332 return (Status);
3333}
3334
3335/**
3336 Prompt the user and return the resultant answer to the requestor.
3337
3338 This function is the same as ShellPromptForResponse, except that the prompt is
3339 automatically pulled from HII.
3340
3341 @param Type What type of question is asked. This is used to filter the input
3342 to prevent invalid answers to question.
jcarseya405b862010-09-14 05:18:09 +00003343 @param[in] HiiFormatStringId The format string Id for getting from Hii.
3344 @param[in] HiiFormatHandle The format string Handle for getting from Hii.
3345 @param Response Pointer to Response which will be populated upon return.
jcarseyc9d92df2010-02-03 15:37:54 +00003346
3347 @retval EFI_SUCCESS the operation was sucessful.
3348 @return other the operation failed.
3349
3350 @sa ShellPromptForResponse
3351**/
3352EFI_STATUS
3353EFIAPI
3354ShellPromptForResponseHii (
3355 IN SHELL_PROMPT_REQUEST_TYPE Type,
3356 IN CONST EFI_STRING_ID HiiFormatStringId,
3357 IN CONST EFI_HANDLE HiiFormatHandle,
3358 IN OUT VOID **Response
3359 )
3360{
3361 CHAR16 *Prompt;
3362 EFI_STATUS Status;
3363
3364 Prompt = HiiGetString(HiiFormatHandle, HiiFormatStringId, NULL);
3365 Status = ShellPromptForResponse(Type, Prompt, Response);
3366 FreePool(Prompt);
3367 return (Status);
3368}
3369
jcarseya405b862010-09-14 05:18:09 +00003370/**
3371 Function to determin if an entire string is a valid number.
jcarseyc9d92df2010-02-03 15:37:54 +00003372
jcarseya405b862010-09-14 05:18:09 +00003373 If Hex it must be preceeded with a 0x or has ForceHex, set TRUE.
3374
3375 @param[in] String The string to evaluate.
3376 @param[in] ForceHex TRUE - always assume hex.
3377 @param[in] StopAtSpace TRUE to halt upon finding a space, FALSE to keep going.
3378
3379 @retval TRUE It is all numeric (dec/hex) characters.
3380 @retval FALSE There is a non-numeric character.
3381**/
3382BOOLEAN
3383EFIAPI
jcarsey252d9452011-03-25 20:49:53 +00003384InternalShellIsHexOrDecimalNumber (
jcarseya405b862010-09-14 05:18:09 +00003385 IN CONST CHAR16 *String,
3386 IN CONST BOOLEAN ForceHex,
3387 IN CONST BOOLEAN StopAtSpace
3388 )
3389{
3390 BOOLEAN Hex;
3391
3392 //
3393 // chop off a single negative sign
3394 //
3395 if (String != NULL && *String == L'-') {
3396 String++;
3397 }
3398
3399 if (String == NULL) {
3400 return (FALSE);
3401 }
3402
3403 //
3404 // chop leading zeroes
3405 //
3406 while(String != NULL && *String == L'0'){
3407 String++;
3408 }
3409 //
3410 // allow '0x' or '0X', but not 'x' or 'X'
3411 //
3412 if (String != NULL && (*String == L'x' || *String == L'X')) {
3413 if (*(String-1) != L'0') {
3414 //
3415 // we got an x without a preceeding 0
3416 //
3417 return (FALSE);
3418 }
3419 String++;
3420 Hex = TRUE;
3421 } else if (ForceHex) {
3422 Hex = TRUE;
3423 } else {
3424 Hex = FALSE;
3425 }
3426
3427 //
3428 // loop through the remaining characters and use the lib function
3429 //
3430 for ( ; String != NULL && *String != CHAR_NULL && !(StopAtSpace && *String == L' ') ; String++){
3431 if (Hex) {
3432 if (!ShellIsHexaDecimalDigitCharacter(*String)) {
3433 return (FALSE);
3434 }
3435 } else {
3436 if (!ShellIsDecimalDigitCharacter(*String)) {
3437 return (FALSE);
3438 }
3439 }
3440 }
jcarsey252d9452011-03-25 20:49:53 +00003441
jcarseya405b862010-09-14 05:18:09 +00003442 return (TRUE);
3443}
3444
3445/**
3446 Function to determine if a given filename exists.
3447
3448 @param[in] Name Path to test.
3449
3450 @retval EFI_SUCCESS The Path represents a file.
3451 @retval EFI_NOT_FOUND The Path does not represent a file.
3452 @retval other The path failed to open.
3453**/
3454EFI_STATUS
3455EFIAPI
3456ShellFileExists(
3457 IN CONST CHAR16 *Name
3458 )
3459{
3460 EFI_STATUS Status;
3461 EFI_SHELL_FILE_INFO *List;
3462
3463 ASSERT(Name != NULL);
3464
3465 List = NULL;
3466 Status = ShellOpenFileMetaArg((CHAR16*)Name, EFI_FILE_MODE_READ, &List);
3467 if (EFI_ERROR(Status)) {
3468 return (Status);
3469 }
3470
3471 ShellCloseFileMetaArg(&List);
3472
3473 return (EFI_SUCCESS);
3474}
jcarsey252d9452011-03-25 20:49:53 +00003475
3476/**
3477 Convert a Unicode character to upper case only if
3478 it maps to a valid small-case ASCII character.
3479
3480 This internal function only deal with Unicode character
3481 which maps to a valid small-case ASCII character, i.e.
3482 L'a' to L'z'. For other Unicode character, the input character
3483 is returned directly.
3484
3485 @param Char The character to convert.
3486
3487 @retval LowerCharacter If the Char is with range L'a' to L'z'.
3488 @retval Unchanged Otherwise.
3489
3490**/
3491CHAR16
3492EFIAPI
3493InternalShellCharToUpper (
3494 IN CHAR16 Char
3495 )
3496{
3497 if (Char >= L'a' && Char <= L'z') {
3498 return (CHAR16) (Char - (L'a' - L'A'));
3499 }
3500
3501 return Char;
3502}
3503
3504/**
3505 Convert a Unicode character to numerical value.
3506
3507 This internal function only deal with Unicode character
3508 which maps to a valid hexadecimal ASII character, i.e.
3509 L'0' to L'9', L'a' to L'f' or L'A' to L'F'. For other
3510 Unicode character, the value returned does not make sense.
3511
3512 @param Char The character to convert.
3513
3514 @return The numerical value converted.
3515
3516**/
3517UINTN
3518EFIAPI
3519InternalShellHexCharToUintn (
3520 IN CHAR16 Char
3521 )
3522{
3523 if (ShellIsDecimalDigitCharacter (Char)) {
3524 return Char - L'0';
3525 }
3526
3527 return (UINTN) (10 + InternalShellCharToUpper (Char) - L'A');
3528}
3529
3530/**
3531 Convert a Null-terminated Unicode hexadecimal string to a value of type UINT64.
3532
3533 This function returns a value of type UINTN by interpreting the contents
3534 of the Unicode string specified by String as a hexadecimal number.
3535 The format of the input Unicode string String is:
3536
3537 [spaces][zeros][x][hexadecimal digits].
3538
3539 The valid hexadecimal digit character is in the range [0-9], [a-f] and [A-F].
3540 The prefix "0x" is optional. Both "x" and "X" is allowed in "0x" prefix.
3541 If "x" appears in the input string, it must be prefixed with at least one 0.
3542 The function will ignore the pad space, which includes spaces or tab characters,
3543 before [zeros], [x] or [hexadecimal digit]. The running zero before [x] or
3544 [hexadecimal digit] will be ignored. Then, the decoding starts after [x] or the
3545 first valid hexadecimal digit. Then, the function stops at the first character that is
3546 a not a valid hexadecimal character or NULL, whichever one comes first.
3547
3548 If String has only pad spaces, then zero is returned.
3549 If String has no leading pad spaces, leading zeros or valid hexadecimal digits,
3550 then zero is returned.
3551
3552 @param[in] String A pointer to a Null-terminated Unicode string.
3553 @param[out] Value Upon a successful return the value of the conversion.
3554 @param[in] StopAtSpace FALSE to skip spaces.
3555
3556 @retval EFI_SUCCESS The conversion was successful.
3557 @retval EFI_INVALID_PARAMETER A parameter was NULL or invalid.
3558 @retval EFI_DEVICE_ERROR An overflow occured.
3559**/
3560EFI_STATUS
3561EFIAPI
3562InternalShellStrHexToUint64 (
3563 IN CONST CHAR16 *String,
3564 OUT UINT64 *Value,
3565 IN CONST BOOLEAN StopAtSpace
3566 )
3567{
3568 UINT64 Result;
3569
3570 if (String == NULL || StrSize(String) == 0 || Value == NULL) {
3571 return (EFI_INVALID_PARAMETER);
3572 }
3573
3574 //
3575 // Ignore the pad spaces (space or tab)
3576 //
3577 while ((*String == L' ') || (*String == L'\t')) {
3578 String++;
3579 }
3580
3581 //
3582 // Ignore leading Zeros after the spaces
3583 //
3584 while (*String == L'0') {
3585 String++;
3586 }
3587
3588 if (InternalShellCharToUpper (*String) == L'X') {
3589 if (*(String - 1) != L'0') {
3590 return 0;
3591 }
3592 //
3593 // Skip the 'X'
3594 //
3595 String++;
3596 }
3597
3598 Result = 0;
3599
3600 //
3601 // Skip spaces if requested
3602 //
3603 while (StopAtSpace && *String == L' ') {
3604 String++;
3605 }
3606
3607 while (ShellIsHexaDecimalDigitCharacter (*String)) {
3608 //
3609 // If the Hex Number represented by String overflows according
3610 // to the range defined by UINTN, then ASSERT().
3611 //
3612 if (!(Result <= (RShiftU64((((UINT64) ~0) - InternalShellHexCharToUintn (*String)), 4)))) {
3613// if (!(Result <= ((((UINT64) ~0) - InternalShellHexCharToUintn (*String)) >> 4))) {
3614 return (EFI_DEVICE_ERROR);
3615 }
3616
3617 Result = (LShiftU64(Result, 4));
3618 Result += InternalShellHexCharToUintn (*String);
3619 String++;
3620
3621 //
3622 // Skip spaces if requested
3623 //
3624 while (StopAtSpace && *String == L' ') {
3625 String++;
3626 }
3627 }
3628
3629 *Value = Result;
3630 return (EFI_SUCCESS);
3631}
3632
3633/**
3634 Convert a Null-terminated Unicode decimal string to a value of
3635 type UINT64.
3636
3637 This function returns a value of type UINT64 by interpreting the contents
3638 of the Unicode string specified by String as a decimal number. The format
3639 of the input Unicode string String is:
3640
3641 [spaces] [decimal digits].
3642
3643 The valid decimal digit character is in the range [0-9]. The
3644 function will ignore the pad space, which includes spaces or
3645 tab characters, before [decimal digits]. The running zero in the
3646 beginning of [decimal digits] will be ignored. Then, the function
3647 stops at the first character that is a not a valid decimal character
3648 or a Null-terminator, whichever one comes first.
3649
3650 If String has only pad spaces, then 0 is returned.
3651 If String has no pad spaces or valid decimal digits,
3652 then 0 is returned.
3653
3654 @param[in] String A pointer to a Null-terminated Unicode string.
3655 @param[out] Value Upon a successful return the value of the conversion.
3656 @param[in] StopAtSpace FALSE to skip spaces.
3657
3658 @retval EFI_SUCCESS The conversion was successful.
3659 @retval EFI_INVALID_PARAMETER A parameter was NULL or invalid.
3660 @retval EFI_DEVICE_ERROR An overflow occured.
3661**/
3662EFI_STATUS
3663EFIAPI
3664InternalShellStrDecimalToUint64 (
3665 IN CONST CHAR16 *String,
3666 OUT UINT64 *Value,
3667 IN CONST BOOLEAN StopAtSpace
3668 )
3669{
3670 UINT64 Result;
3671
3672 if (String == NULL || StrSize (String) == 0 || Value == NULL) {
3673 return (EFI_INVALID_PARAMETER);
3674 }
3675
3676 //
3677 // Ignore the pad spaces (space or tab)
3678 //
3679 while ((*String == L' ') || (*String == L'\t')) {
3680 String++;
3681 }
3682
3683 //
3684 // Ignore leading Zeros after the spaces
3685 //
3686 while (*String == L'0') {
3687 String++;
3688 }
3689
3690 Result = 0;
3691
3692 //
3693 // Skip spaces if requested
3694 //
3695 while (StopAtSpace && *String == L' ') {
3696 String++;
3697 }
3698 while (ShellIsDecimalDigitCharacter (*String)) {
3699 //
3700 // If the number represented by String overflows according
3701 // to the range defined by UINT64, then ASSERT().
3702 //
3703
3704 if (!(Result <= (DivU64x32((((UINT64) ~0) - (*String - L'0')),10)))) {
3705 return (EFI_DEVICE_ERROR);
3706 }
3707
3708 Result = MultU64x32(Result, 10) + (*String - L'0');
3709 String++;
3710
3711 //
3712 // Stop at spaces if requested
3713 //
3714 if (StopAtSpace && *String == L' ') {
3715 break;
3716 }
3717 }
3718
3719 *Value = Result;
3720
3721 return (EFI_SUCCESS);
3722}
3723
3724/**
3725 Function to verify and convert a string to its numerical value.
3726
3727 If Hex it must be preceeded with a 0x, 0X, or has ForceHex set TRUE.
3728
3729 @param[in] String The string to evaluate.
3730 @param[out] Value Upon a successful return the value of the conversion.
3731 @param[in] ForceHex TRUE - always assume hex.
3732 @param[in] StopAtSpace FALSE to skip spaces.
3733
3734 @retval EFI_SUCCESS The conversion was successful.
3735 @retval EFI_INVALID_PARAMETER String contained an invalid character.
3736 @retval EFI_NOT_FOUND String was a number, but Value was NULL.
3737**/
3738EFI_STATUS
3739EFIAPI
3740ShellConvertStringToUint64(
3741 IN CONST CHAR16 *String,
3742 OUT UINT64 *Value,
3743 IN CONST BOOLEAN ForceHex,
3744 IN CONST BOOLEAN StopAtSpace
3745 )
3746{
3747 UINT64 RetVal;
3748 CONST CHAR16 *Walker;
3749 EFI_STATUS Status;
3750 BOOLEAN Hex;
3751
3752 Hex = ForceHex;
3753
3754 if (!InternalShellIsHexOrDecimalNumber(String, Hex, StopAtSpace)) {
3755 if (!Hex) {
3756 Hex = TRUE;
3757 if (!InternalShellIsHexOrDecimalNumber(String, Hex, StopAtSpace)) {
3758 return (EFI_INVALID_PARAMETER);
3759 }
3760 } else {
3761 return (EFI_INVALID_PARAMETER);
3762 }
3763 }
3764
3765 //
3766 // Chop off leading spaces
3767 //
3768 for (Walker = String; Walker != NULL && *Walker != CHAR_NULL && *Walker == L' '; Walker++);
3769
3770 //
3771 // make sure we have something left that is numeric.
3772 //
3773 if (Walker == NULL || *Walker == CHAR_NULL || !InternalShellIsHexOrDecimalNumber(Walker, Hex, StopAtSpace)) {
3774 return (EFI_INVALID_PARAMETER);
3775 }
3776
3777 //
3778 // do the conversion.
3779 //
3780 if (Hex || StrnCmp(Walker, L"0x", 2) == 0 || StrnCmp(Walker, L"0X", 2) == 0){
3781 Status = InternalShellStrHexToUint64(Walker, &RetVal, StopAtSpace);
3782 } else {
3783 Status = InternalShellStrDecimalToUint64(Walker, &RetVal, StopAtSpace);
3784 }
3785
3786 if (Value == NULL && !EFI_ERROR(Status)) {
3787 return (EFI_NOT_FOUND);
3788 }
3789
3790 if (Value != NULL) {
3791 *Value = RetVal;
3792 }
3793
3794 return (Status);
3795}
3796
3797/**
3798 Function to determin if an entire string is a valid number.
3799
3800 If Hex it must be preceeded with a 0x or has ForceHex, set TRUE.
3801
3802 @param[in] String The string to evaluate.
3803 @param[in] ForceHex TRUE - always assume hex.
3804 @param[in] StopAtSpace TRUE to halt upon finding a space, FALSE to keep going.
3805
3806 @retval TRUE It is all numeric (dec/hex) characters.
3807 @retval FALSE There is a non-numeric character.
3808**/
3809BOOLEAN
3810EFIAPI
3811ShellIsHexOrDecimalNumber (
3812 IN CONST CHAR16 *String,
3813 IN CONST BOOLEAN ForceHex,
3814 IN CONST BOOLEAN StopAtSpace
3815 )
3816{
3817 if (ShellConvertStringToUint64(String, NULL, ForceHex, StopAtSpace) == EFI_NOT_FOUND) {
3818 return (TRUE);
3819 }
3820 return (FALSE);
3821}
jcarsey4d0a4fc2011-07-06 22:28:36 +00003822
3823/**
3824 Function to read a single line from a SHELL_FILE_HANDLE. The \n is not included in the returned
3825 buffer. The returned buffer must be callee freed.
3826
3827 If the position upon start is 0, then the Ascii Boolean will be set. This should be
3828 maintained and not changed for all operations with the same file.
3829
ydong104ff7e372011-09-02 08:05:34 +00003830 @param[in] Handle SHELL_FILE_HANDLE to read from.
3831 @param[in, out] Ascii Boolean value for indicating whether the file is
3832 Ascii (TRUE) or UCS2 (FALSE).
jcarsey4d0a4fc2011-07-06 22:28:36 +00003833
3834 @return The line of text from the file.
3835
3836 @sa ShellFileHandleReadLine
3837**/
3838CHAR16*
3839EFIAPI
3840ShellFileHandleReturnLine(
3841 IN SHELL_FILE_HANDLE Handle,
3842 IN OUT BOOLEAN *Ascii
3843 )
3844{
3845 CHAR16 *RetVal;
3846 UINTN Size;
3847 EFI_STATUS Status;
3848
3849 Size = 0;
3850 RetVal = NULL;
3851
3852 Status = ShellFileHandleReadLine(Handle, RetVal, &Size, FALSE, Ascii);
3853 if (Status == EFI_BUFFER_TOO_SMALL) {
3854 RetVal = AllocateZeroPool(Size);
3855 Status = ShellFileHandleReadLine(Handle, RetVal, &Size, FALSE, Ascii);
3856 }
3857 ASSERT_EFI_ERROR(Status);
3858 if (EFI_ERROR(Status) && (RetVal != NULL)) {
3859 FreePool(RetVal);
3860 RetVal = NULL;
3861 }
3862 return (RetVal);
3863}
3864
3865/**
3866 Function to read a single line (up to but not including the \n) from a SHELL_FILE_HANDLE.
3867
3868 If the position upon start is 0, then the Ascii Boolean will be set. This should be
3869 maintained and not changed for all operations with the same file.
3870
ydong104ff7e372011-09-02 08:05:34 +00003871 @param[in] Handle SHELL_FILE_HANDLE to read from.
3872 @param[in, out] Buffer The pointer to buffer to read into.
3873 @param[in, out] Size The pointer to number of bytes in Buffer.
3874 @param[in] Truncate If the buffer is large enough, this has no effect.
3875 If the buffer is is too small and Truncate is TRUE,
3876 the line will be truncated.
3877 If the buffer is is too small and Truncate is FALSE,
3878 then no read will occur.
jcarsey4d0a4fc2011-07-06 22:28:36 +00003879
ydong104ff7e372011-09-02 08:05:34 +00003880 @param[in, out] Ascii Boolean value for indicating whether the file is
3881 Ascii (TRUE) or UCS2 (FALSE).
jcarsey4d0a4fc2011-07-06 22:28:36 +00003882
3883 @retval EFI_SUCCESS The operation was successful. The line is stored in
3884 Buffer.
3885 @retval EFI_INVALID_PARAMETER Handle was NULL.
3886 @retval EFI_INVALID_PARAMETER Size was NULL.
3887 @retval EFI_BUFFER_TOO_SMALL Size was not large enough to store the line.
3888 Size was updated to the minimum space required.
3889**/
3890EFI_STATUS
3891EFIAPI
3892ShellFileHandleReadLine(
3893 IN SHELL_FILE_HANDLE Handle,
3894 IN OUT CHAR16 *Buffer,
3895 IN OUT UINTN *Size,
3896 IN BOOLEAN Truncate,
3897 IN OUT BOOLEAN *Ascii
3898 )
3899{
3900 EFI_STATUS Status;
3901 CHAR16 CharBuffer;
3902 UINTN CharSize;
3903 UINTN CountSoFar;
3904 UINT64 OriginalFilePosition;
3905
3906
3907 if (Handle == NULL
3908 ||Size == NULL
3909 ){
3910 return (EFI_INVALID_PARAMETER);
3911 }
3912 if (Buffer == NULL) {
3913 ASSERT(*Size == 0);
3914 } else {
3915 *Buffer = CHAR_NULL;
3916 }
3917 gEfiShellProtocol->GetFilePosition(Handle, &OriginalFilePosition);
3918 if (OriginalFilePosition == 0) {
3919 CharSize = sizeof(CHAR16);
3920 Status = gEfiShellProtocol->ReadFile(Handle, &CharSize, &CharBuffer);
3921 ASSERT_EFI_ERROR(Status);
3922 if (CharBuffer == gUnicodeFileTag) {
3923 *Ascii = FALSE;
3924 } else {
3925 *Ascii = TRUE;
3926 gEfiShellProtocol->SetFilePosition(Handle, OriginalFilePosition);
3927 }
3928 }
3929
3930 for (CountSoFar = 0;;CountSoFar++){
3931 CharBuffer = 0;
3932 if (*Ascii) {
3933 CharSize = sizeof(CHAR8);
3934 } else {
3935 CharSize = sizeof(CHAR16);
3936 }
3937 Status = gEfiShellProtocol->ReadFile(Handle, &CharSize, &CharBuffer);
3938 if ( EFI_ERROR(Status)
3939 || CharSize == 0
3940 || (CharBuffer == L'\n' && !(*Ascii))
3941 || (CharBuffer == '\n' && *Ascii)
3942 ){
3943 break;
3944 }
3945 //
3946 // if we have space save it...
3947 //
3948 if ((CountSoFar+1)*sizeof(CHAR16) < *Size){
3949 ASSERT(Buffer != NULL);
3950 ((CHAR16*)Buffer)[CountSoFar] = CharBuffer;
3951 ((CHAR16*)Buffer)[CountSoFar+1] = CHAR_NULL;
3952 }
3953 }
3954
3955 //
3956 // if we ran out of space tell when...
3957 //
3958 if ((CountSoFar+1)*sizeof(CHAR16) > *Size){
3959 *Size = (CountSoFar+1)*sizeof(CHAR16);
3960 if (!Truncate) {
3961 gEfiShellProtocol->SetFilePosition(Handle, OriginalFilePosition);
3962 } else {
3963 DEBUG((DEBUG_WARN, "The line was truncated in ShellFileHandleReadLine"));
3964 }
3965 return (EFI_BUFFER_TOO_SMALL);
3966 }
3967 while(Buffer[StrLen(Buffer)-1] == L'\r') {
3968 Buffer[StrLen(Buffer)-1] = CHAR_NULL;
3969 }
3970
3971 return (Status);
3972}