blob: bcb6f2a7e4ca143e3f8db1c8121add3cd11cb018 [file] [log] [blame]
jcarsey94b17fa2009-05-07 18:46:18 +00001/** @file
2 Provides interface to shell functionality for shell commands and applications.
3
jcarsey1e6e84c2010-01-25 20:05:08 +00004 Copyright (c) 2006 - 2010, Intel Corporation. All rights reserved.<BR>
5 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>
jcarseyd2b45642009-05-11 18:02:16 +000017
jcarsey94b17fa2009-05-07 18:46:18 +000018#define FIND_XXXXX_FILE_BUFFER_SIZE (SIZE_OF_EFI_FILE_INFO + MAX_FILE_NAME_LEN)
19
jcarseyd2b45642009-05-11 18:02:16 +000020//
jcarseya405b862010-09-14 05:18:09 +000021// globals...
jcarseyd2b45642009-05-11 18:02:16 +000022//
23SHELL_PARAM_ITEM EmptyParamList[] = {
24 {NULL, TypeMax}
25 };
jcarseya405b862010-09-14 05:18:09 +000026SHELL_PARAM_ITEM SfoParamList[] = {
27 {L"-sfo", TypeFlag},
28 {NULL, TypeMax}
29 };
30EFI_SHELL_ENVIRONMENT2 *mEfiShellEnvironment2;
31EFI_SHELL_INTERFACE *mEfiShellInterface;
32EFI_SHELL_PROTOCOL *mEfiShellProtocol;
33EFI_SHELL_PARAMETERS_PROTOCOL *mEfiShellParametersProtocol;
34EFI_HANDLE mEfiShellEnvironment2Handle;
35FILE_HANDLE_FUNCTION_MAP FileFunctionMap;
36CHAR16 *mPostReplaceFormat;
37CHAR16 *mPostReplaceFormat2;
jcarseyb3011f42010-01-11 21:49:04 +000038
jcarsey2247dde2009-11-09 18:08:58 +000039/**
40 Check if a Unicode character is a hexadecimal character.
41
jcarsey1e6e84c2010-01-25 20:05:08 +000042 This internal function checks if a Unicode character is a
jcarseya405b862010-09-14 05:18:09 +000043 numeric character. The valid hexadecimal characters are
jcarsey2247dde2009-11-09 18:08:58 +000044 L'0' to L'9', L'a' to L'f', or L'A' to L'F'.
45
jcarsey2247dde2009-11-09 18:08:58 +000046 @param Char The character to check against.
47
48 @retval TRUE If the Char is a hexadecmial character.
49 @retval FALSE If the Char is not a hexadecmial character.
50
51**/
52BOOLEAN
53EFIAPI
jcarsey969c7832010-01-13 16:46:33 +000054ShellIsHexaDecimalDigitCharacter (
jcarsey2247dde2009-11-09 18:08:58 +000055 IN CHAR16 Char
jcarseya405b862010-09-14 05:18:09 +000056 )
57{
jcarsey2247dde2009-11-09 18:08:58 +000058 return (BOOLEAN) ((Char >= L'0' && Char <= L'9') || (Char >= L'A' && Char <= L'F') || (Char >= L'a' && Char <= L'f'));
59}
jcarsey94b17fa2009-05-07 18:46:18 +000060
61/**
jcarseya405b862010-09-14 05:18:09 +000062 Check if a Unicode character is a decimal character.
63
64 This internal function checks if a Unicode character is a
65 decimal character. The valid characters are
66 L'0' to L'9'.
67
68
69 @param Char The character to check against.
70
71 @retval TRUE If the Char is a hexadecmial character.
72 @retval FALSE If the Char is not a hexadecmial character.
73
74**/
75BOOLEAN
76EFIAPI
77ShellIsDecimalDigitCharacter (
78 IN CHAR16 Char
79 )
80{
81 return (BOOLEAN) (Char >= L'0' && Char <= L'9');
82}
83
84/**
85 Helper function to find ShellEnvironment2 for constructor.
86
87 @param[in] ImageHandle A copy of the calling image's handle.
jcarsey94b17fa2009-05-07 18:46:18 +000088**/
89EFI_STATUS
90EFIAPI
91ShellFindSE2 (
92 IN EFI_HANDLE ImageHandle
jcarseya405b862010-09-14 05:18:09 +000093 )
94{
jcarsey94b17fa2009-05-07 18:46:18 +000095 EFI_STATUS Status;
96 EFI_HANDLE *Buffer;
97 UINTN BufferSize;
98 UINTN HandleIndex;
99
100 BufferSize = 0;
101 Buffer = NULL;
jcarsey1e6e84c2010-01-25 20:05:08 +0000102 Status = gBS->OpenProtocol(ImageHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000103 &gEfiShellEnvironment2Guid,
104 (VOID **)&mEfiShellEnvironment2,
105 ImageHandle,
106 NULL,
107 EFI_OPEN_PROTOCOL_GET_PROTOCOL
jcarseya405b862010-09-14 05:18:09 +0000108 );
jcarsey94b17fa2009-05-07 18:46:18 +0000109 //
110 // look for the mEfiShellEnvironment2 protocol at a higher level
111 //
jcarseya405b862010-09-14 05:18:09 +0000112 if (EFI_ERROR (Status) || !(CompareGuid (&mEfiShellEnvironment2->SESGuid, &gEfiShellEnvironment2ExtGuid))){
jcarsey94b17fa2009-05-07 18:46:18 +0000113 //
114 // figure out how big of a buffer we need.
115 //
116 Status = gBS->LocateHandle (ByProtocol,
117 &gEfiShellEnvironment2Guid,
118 NULL, // ignored for ByProtocol
119 &BufferSize,
120 Buffer
jcarseya405b862010-09-14 05:18:09 +0000121 );
jcarsey2247dde2009-11-09 18:08:58 +0000122 //
123 // maybe it's not there???
124 //
125 if (Status == EFI_BUFFER_TOO_SMALL) {
126 Buffer = (EFI_HANDLE*)AllocatePool(BufferSize);
127 ASSERT(Buffer != NULL);
128 Status = gBS->LocateHandle (ByProtocol,
129 &gEfiShellEnvironment2Guid,
130 NULL, // ignored for ByProtocol
131 &BufferSize,
132 Buffer
jcarseya405b862010-09-14 05:18:09 +0000133 );
jcarsey2247dde2009-11-09 18:08:58 +0000134 }
jcarsey1cd45e72010-01-29 15:07:44 +0000135 if (!EFI_ERROR (Status) && Buffer != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000136 //
137 // now parse the list of returned handles
138 //
139 Status = EFI_NOT_FOUND;
140 for (HandleIndex = 0; HandleIndex < (BufferSize/sizeof(Buffer[0])); HandleIndex++) {
jcarsey1e6e84c2010-01-25 20:05:08 +0000141 Status = gBS->OpenProtocol(Buffer[HandleIndex],
jcarsey94b17fa2009-05-07 18:46:18 +0000142 &gEfiShellEnvironment2Guid,
143 (VOID **)&mEfiShellEnvironment2,
144 ImageHandle,
145 NULL,
146 EFI_OPEN_PROTOCOL_GET_PROTOCOL
jcarseya405b862010-09-14 05:18:09 +0000147 );
148 if (CompareGuid (&mEfiShellEnvironment2->SESGuid, &gEfiShellEnvironment2ExtGuid)) {
jcarsey94b17fa2009-05-07 18:46:18 +0000149 mEfiShellEnvironment2Handle = Buffer[HandleIndex];
150 Status = EFI_SUCCESS;
151 break;
152 }
153 }
154 }
155 }
156 if (Buffer != NULL) {
157 FreePool (Buffer);
158 }
159 return (Status);
160}
161
jcarseya405b862010-09-14 05:18:09 +0000162/*/
163 Function to do most of the work of the constructor. Allows for calling
164 multiple times without complete re-initialization.
165
166 @param[in] ImageHandle A copy of the ImageHandle.
167 @param[in] SystemTable A pointer to the SystemTable for the application.
168**/
jcarsey94b17fa2009-05-07 18:46:18 +0000169EFI_STATUS
170EFIAPI
jcarseyd2b45642009-05-11 18:02:16 +0000171ShellLibConstructorWorker (
jcarsey94b17fa2009-05-07 18:46:18 +0000172 IN EFI_HANDLE ImageHandle,
173 IN EFI_SYSTEM_TABLE *SystemTable
jcarseya405b862010-09-14 05:18:09 +0000174 )
175{
176 EFI_STATUS Status;
jcarseyb3011f42010-01-11 21:49:04 +0000177 mPostReplaceFormat = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
jcarseyecd3d592009-12-07 18:05:00 +0000178 ASSERT (mPostReplaceFormat != NULL);
jcarseyb3011f42010-01-11 21:49:04 +0000179 mPostReplaceFormat2 = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
jcarseyecd3d592009-12-07 18:05:00 +0000180 ASSERT (mPostReplaceFormat2 != NULL);
181
jcarsey94b17fa2009-05-07 18:46:18 +0000182 //
183 // UEFI 2.0 shell interfaces (used preferentially)
184 //
jcarseya405b862010-09-14 05:18:09 +0000185 Status = gBS->OpenProtocol(
186 ImageHandle,
187 &gEfiShellProtocolGuid,
188 (VOID **)&mEfiShellProtocol,
189 ImageHandle,
190 NULL,
191 EFI_OPEN_PROTOCOL_GET_PROTOCOL
192 );
jcarsey94b17fa2009-05-07 18:46:18 +0000193 if (EFI_ERROR(Status)) {
jcarseya405b862010-09-14 05:18:09 +0000194 //
195 // Search for the shell protocol
196 //
197 Status = gBS->LocateProtocol(
198 &gEfiShellProtocolGuid,
199 NULL,
200 (VOID **)&mEfiShellProtocol
201 );
202 if (EFI_ERROR(Status)) {
203 mEfiShellProtocol = NULL;
204 }
jcarsey94b17fa2009-05-07 18:46:18 +0000205 }
jcarseya405b862010-09-14 05:18:09 +0000206 Status = gBS->OpenProtocol(
207 ImageHandle,
208 &gEfiShellParametersProtocolGuid,
209 (VOID **)&mEfiShellParametersProtocol,
210 ImageHandle,
211 NULL,
212 EFI_OPEN_PROTOCOL_GET_PROTOCOL
213 );
jcarsey94b17fa2009-05-07 18:46:18 +0000214 if (EFI_ERROR(Status)) {
215 mEfiShellParametersProtocol = NULL;
216 }
217
218 if (mEfiShellParametersProtocol == NULL || mEfiShellProtocol == NULL) {
219 //
220 // Moved to seperate function due to complexity
221 //
222 Status = ShellFindSE2(ImageHandle);
223
224 if (EFI_ERROR(Status)) {
225 DEBUG((DEBUG_ERROR, "Status: 0x%08x\r\n", Status));
226 mEfiShellEnvironment2 = NULL;
227 }
jcarsey1e6e84c2010-01-25 20:05:08 +0000228 Status = gBS->OpenProtocol(ImageHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000229 &gEfiShellInterfaceGuid,
230 (VOID **)&mEfiShellInterface,
231 ImageHandle,
232 NULL,
233 EFI_OPEN_PROTOCOL_GET_PROTOCOL
jcarseya405b862010-09-14 05:18:09 +0000234 );
jcarsey94b17fa2009-05-07 18:46:18 +0000235 if (EFI_ERROR(Status)) {
236 mEfiShellInterface = NULL;
237 }
238 }
jcarseyc9d92df2010-02-03 15:37:54 +0000239
jcarsey94b17fa2009-05-07 18:46:18 +0000240 //
241 // only success getting 2 of either the old or new, but no 1/2 and 1/2
242 //
jcarsey1e6e84c2010-01-25 20:05:08 +0000243 if ((mEfiShellEnvironment2 != NULL && mEfiShellInterface != NULL) ||
jcarseya405b862010-09-14 05:18:09 +0000244 (mEfiShellProtocol != NULL && mEfiShellParametersProtocol != NULL) ) {
jcarseyd2b45642009-05-11 18:02:16 +0000245 if (mEfiShellProtocol != NULL) {
246 FileFunctionMap.GetFileInfo = mEfiShellProtocol->GetFileInfo;
247 FileFunctionMap.SetFileInfo = mEfiShellProtocol->SetFileInfo;
248 FileFunctionMap.ReadFile = mEfiShellProtocol->ReadFile;
249 FileFunctionMap.WriteFile = mEfiShellProtocol->WriteFile;
250 FileFunctionMap.CloseFile = mEfiShellProtocol->CloseFile;
251 FileFunctionMap.DeleteFile = mEfiShellProtocol->DeleteFile;
252 FileFunctionMap.GetFilePosition = mEfiShellProtocol->GetFilePosition;
253 FileFunctionMap.SetFilePosition = mEfiShellProtocol->SetFilePosition;
254 FileFunctionMap.FlushFile = mEfiShellProtocol->FlushFile;
255 FileFunctionMap.GetFileSize = mEfiShellProtocol->GetFileSize;
256 } else {
jcarseya405b862010-09-14 05:18:09 +0000257 FileFunctionMap.GetFileInfo = (EFI_SHELL_GET_FILE_INFO)FileHandleGetInfo;
258 FileFunctionMap.SetFileInfo = (EFI_SHELL_SET_FILE_INFO)FileHandleSetInfo;
259 FileFunctionMap.ReadFile = (EFI_SHELL_READ_FILE)FileHandleRead;
260 FileFunctionMap.WriteFile = (EFI_SHELL_WRITE_FILE)FileHandleWrite;
261 FileFunctionMap.CloseFile = (EFI_SHELL_CLOSE_FILE)FileHandleClose;
262 FileFunctionMap.DeleteFile = (EFI_SHELL_DELETE_FILE)FileHandleDelete;
263 FileFunctionMap.GetFilePosition = (EFI_SHELL_GET_FILE_POSITION)FileHandleGetPosition;
264 FileFunctionMap.SetFilePosition = (EFI_SHELL_SET_FILE_POSITION)FileHandleSetPosition;
265 FileFunctionMap.FlushFile = (EFI_SHELL_FLUSH_FILE)FileHandleFlush;
266 FileFunctionMap.GetFileSize = (EFI_SHELL_GET_FILE_SIZE)FileHandleGetSize;
jcarseyd2b45642009-05-11 18:02:16 +0000267 }
jcarsey94b17fa2009-05-07 18:46:18 +0000268 return (EFI_SUCCESS);
269 }
270 return (EFI_NOT_FOUND);
271}
jcarseyd2b45642009-05-11 18:02:16 +0000272/**
273 Constructor for the Shell library.
274
275 Initialize the library and determine if the underlying is a UEFI Shell 2.0 or an EFI shell.
276
277 @param ImageHandle the image handle of the process
278 @param SystemTable the EFI System Table pointer
279
280 @retval EFI_SUCCESS the initialization was complete sucessfully
281 @return others an error ocurred during initialization
282**/
283EFI_STATUS
284EFIAPI
285ShellLibConstructor (
286 IN EFI_HANDLE ImageHandle,
287 IN EFI_SYSTEM_TABLE *SystemTable
jcarseya405b862010-09-14 05:18:09 +0000288 )
289{
jcarseyd2b45642009-05-11 18:02:16 +0000290 mEfiShellEnvironment2 = NULL;
291 mEfiShellProtocol = NULL;
292 mEfiShellParametersProtocol = NULL;
293 mEfiShellInterface = NULL;
294 mEfiShellEnvironment2Handle = NULL;
jcarseyecd3d592009-12-07 18:05:00 +0000295 mPostReplaceFormat = NULL;
296 mPostReplaceFormat2 = NULL;
jcarseyd2b45642009-05-11 18:02:16 +0000297
jcarseyd2b45642009-05-11 18:02:16 +0000298 //
299 // verify that auto initialize is not set false
jcarsey1e6e84c2010-01-25 20:05:08 +0000300 //
jcarseyd2b45642009-05-11 18:02:16 +0000301 if (PcdGetBool(PcdShellLibAutoInitialize) == 0) {
302 return (EFI_SUCCESS);
303 }
jcarsey1e6e84c2010-01-25 20:05:08 +0000304
jcarseyd2b45642009-05-11 18:02:16 +0000305 return (ShellLibConstructorWorker(ImageHandle, SystemTable));
306}
jcarsey94b17fa2009-05-07 18:46:18 +0000307
308/**
jcarseya405b862010-09-14 05:18:09 +0000309 Destructor for the library. free any resources.
310
311 @param[in] ImageHandle A copy of the ImageHandle.
312 @param[in] SystemTable A pointer to the SystemTable for the application.
313
314 @retval EFI_SUCCESS The operation was successful.
315 @return An error from the CloseProtocol function.
jcarsey94b17fa2009-05-07 18:46:18 +0000316**/
317EFI_STATUS
318EFIAPI
319ShellLibDestructor (
320 IN EFI_HANDLE ImageHandle,
321 IN EFI_SYSTEM_TABLE *SystemTable
jcarseya405b862010-09-14 05:18:09 +0000322 )
323{
jcarsey94b17fa2009-05-07 18:46:18 +0000324 if (mEfiShellEnvironment2 != NULL) {
325 gBS->CloseProtocol(mEfiShellEnvironment2Handle==NULL?ImageHandle:mEfiShellEnvironment2Handle,
326 &gEfiShellEnvironment2Guid,
327 ImageHandle,
328 NULL);
jcarseyd2b45642009-05-11 18:02:16 +0000329 mEfiShellEnvironment2 = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000330 }
331 if (mEfiShellInterface != NULL) {
332 gBS->CloseProtocol(ImageHandle,
333 &gEfiShellInterfaceGuid,
334 ImageHandle,
jcarsey1e6e84c2010-01-25 20:05:08 +0000335 NULL);
jcarseyd2b45642009-05-11 18:02:16 +0000336 mEfiShellInterface = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000337 }
338 if (mEfiShellProtocol != NULL) {
339 gBS->CloseProtocol(ImageHandle,
340 &gEfiShellProtocolGuid,
341 ImageHandle,
jcarsey1e6e84c2010-01-25 20:05:08 +0000342 NULL);
jcarseyd2b45642009-05-11 18:02:16 +0000343 mEfiShellProtocol = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000344 }
345 if (mEfiShellParametersProtocol != NULL) {
346 gBS->CloseProtocol(ImageHandle,
347 &gEfiShellParametersProtocolGuid,
348 ImageHandle,
jcarsey1e6e84c2010-01-25 20:05:08 +0000349 NULL);
jcarseyd2b45642009-05-11 18:02:16 +0000350 mEfiShellParametersProtocol = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000351 }
jcarseyd2b45642009-05-11 18:02:16 +0000352 mEfiShellEnvironment2Handle = NULL;
jcarseyecd3d592009-12-07 18:05:00 +0000353
354 if (mPostReplaceFormat != NULL) {
355 FreePool(mPostReplaceFormat);
356 }
357 if (mPostReplaceFormat2 != NULL) {
358 FreePool(mPostReplaceFormat2);
359 }
360 mPostReplaceFormat = NULL;
361 mPostReplaceFormat2 = NULL;
362
jcarsey94b17fa2009-05-07 18:46:18 +0000363 return (EFI_SUCCESS);
364}
jcarseyd2b45642009-05-11 18:02:16 +0000365
366/**
367 This function causes the shell library to initialize itself. If the shell library
368 is already initialized it will de-initialize all the current protocol poitners and
369 re-populate them again.
370
371 When the library is used with PcdShellLibAutoInitialize set to true this function
372 will return EFI_SUCCESS and perform no actions.
373
374 This function is intended for internal access for shell commands only.
375
376 @retval EFI_SUCCESS the initialization was complete sucessfully
377
378**/
379EFI_STATUS
380EFIAPI
381ShellInitialize (
jcarseya405b862010-09-14 05:18:09 +0000382 )
383{
jcarseyd2b45642009-05-11 18:02:16 +0000384 //
385 // if auto initialize is not false then skip
386 //
387 if (PcdGetBool(PcdShellLibAutoInitialize) != 0) {
388 return (EFI_SUCCESS);
389 }
390
391 //
392 // deinit the current stuff
393 //
394 ASSERT_EFI_ERROR(ShellLibDestructor(gImageHandle, gST));
395
396 //
397 // init the new stuff
398 //
399 return (ShellLibConstructorWorker(gImageHandle, gST));
400}
401
jcarsey94b17fa2009-05-07 18:46:18 +0000402/**
jcarsey1e6e84c2010-01-25 20:05:08 +0000403 This function will retrieve the information about the file for the handle
jcarsey94b17fa2009-05-07 18:46:18 +0000404 specified and store it in allocated pool memory.
405
jcarsey1e6e84c2010-01-25 20:05:08 +0000406 This function allocates a buffer to store the file's information. It is the
qhuang869817bf2009-05-20 14:42:48 +0000407 caller's responsibility to free the buffer
jcarsey94b17fa2009-05-07 18:46:18 +0000408
jcarsey1e6e84c2010-01-25 20:05:08 +0000409 @param FileHandle The file handle of the file for which information is
jcarsey94b17fa2009-05-07 18:46:18 +0000410 being requested.
411
412 @retval NULL information could not be retrieved.
413
414 @return the information about the file
415**/
416EFI_FILE_INFO*
417EFIAPI
418ShellGetFileInfo (
jcarseya405b862010-09-14 05:18:09 +0000419 IN SHELL_FILE_HANDLE FileHandle
420 )
421{
jcarseyd2b45642009-05-11 18:02:16 +0000422 return (FileFunctionMap.GetFileInfo(FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000423}
424
425/**
jcarseya405b862010-09-14 05:18:09 +0000426 This function sets the information about the file for the opened handle
jcarsey94b17fa2009-05-07 18:46:18 +0000427 specified.
428
jcarseya405b862010-09-14 05:18:09 +0000429 @param[in] FileHandle The file handle of the file for which information
430 is being set.
jcarsey94b17fa2009-05-07 18:46:18 +0000431
jcarseya405b862010-09-14 05:18:09 +0000432 @param[in] FileInfo The information to set.
jcarsey94b17fa2009-05-07 18:46:18 +0000433
jcarseya405b862010-09-14 05:18:09 +0000434 @retval EFI_SUCCESS The information was set.
435 @retval EFI_INVALID_PARAMETER A parameter was out of range or invalid.
436 @retval EFI_UNSUPPORTED The FileHandle does not support FileInfo.
437 @retval EFI_NO_MEDIA The device has no medium.
438 @retval EFI_DEVICE_ERROR The device reported an error.
jcarsey94b17fa2009-05-07 18:46:18 +0000439 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
jcarseya405b862010-09-14 05:18:09 +0000440 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
441 @retval EFI_ACCESS_DENIED The file was opened read only.
442 @retval EFI_VOLUME_FULL The volume is full.
jcarsey94b17fa2009-05-07 18:46:18 +0000443**/
444EFI_STATUS
445EFIAPI
446ShellSetFileInfo (
jcarseya405b862010-09-14 05:18:09 +0000447 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000448 IN EFI_FILE_INFO *FileInfo
jcarseya405b862010-09-14 05:18:09 +0000449 )
450{
jcarseyd2b45642009-05-11 18:02:16 +0000451 return (FileFunctionMap.SetFileInfo(FileHandle, FileInfo));
jcarsey1e6e84c2010-01-25 20:05:08 +0000452}
453
jcarsey94b17fa2009-05-07 18:46:18 +0000454 /**
455 This function will open a file or directory referenced by DevicePath.
456
jcarsey1e6e84c2010-01-25 20:05:08 +0000457 This function opens a file with the open mode according to the file path. The
jcarsey94b17fa2009-05-07 18:46:18 +0000458 Attributes is valid only for EFI_FILE_MODE_CREATE.
459
jcarsey1e6e84c2010-01-25 20:05:08 +0000460 @param FilePath on input the device path to the file. On output
jcarsey94b17fa2009-05-07 18:46:18 +0000461 the remaining device path.
462 @param DeviceHandle pointer to the system device handle.
463 @param FileHandle pointer to the file handle.
464 @param OpenMode the mode to open the file with.
465 @param Attributes the file's file attributes.
466
467 @retval EFI_SUCCESS The information was set.
468 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
jcarsey1e6e84c2010-01-25 20:05:08 +0000469 @retval EFI_UNSUPPORTED Could not open the file path.
470 @retval EFI_NOT_FOUND The specified file could not be found on the
471 device or the file system could not be found on
jcarsey94b17fa2009-05-07 18:46:18 +0000472 the device.
473 @retval EFI_NO_MEDIA The device has no medium.
jcarsey1e6e84c2010-01-25 20:05:08 +0000474 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
jcarsey94b17fa2009-05-07 18:46:18 +0000475 medium is no longer supported.
476 @retval EFI_DEVICE_ERROR The device reported an error.
477 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
478 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
jcarseya405b862010-09-14 05:18:09 +0000479 @retval EFI_ACCESS_DENIED The file was opened read only.
jcarsey1e6e84c2010-01-25 20:05:08 +0000480 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
jcarsey94b17fa2009-05-07 18:46:18 +0000481 file.
482 @retval EFI_VOLUME_FULL The volume is full.
483**/
484EFI_STATUS
485EFIAPI
486ShellOpenFileByDevicePath(
487 IN OUT EFI_DEVICE_PATH_PROTOCOL **FilePath,
488 OUT EFI_HANDLE *DeviceHandle,
jcarseya405b862010-09-14 05:18:09 +0000489 OUT SHELL_FILE_HANDLE *FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000490 IN UINT64 OpenMode,
491 IN UINT64 Attributes
jcarseya405b862010-09-14 05:18:09 +0000492 )
493{
494 CHAR16 *FileName;
495 EFI_STATUS Status;
jcarsey94b17fa2009-05-07 18:46:18 +0000496 EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *EfiSimpleFileSystemProtocol;
jcarseya405b862010-09-14 05:18:09 +0000497 EFI_FILE_PROTOCOL *Handle1;
498 EFI_FILE_PROTOCOL *Handle2;
jcarsey94b17fa2009-05-07 18:46:18 +0000499
500 //
501 // ASERT for FileHandle, FilePath, and DeviceHandle being NULL
502 //
503 ASSERT(FilePath != NULL);
504 ASSERT(FileHandle != NULL);
505 ASSERT(DeviceHandle != NULL);
jcarsey1e6e84c2010-01-25 20:05:08 +0000506 //
jcarsey94b17fa2009-05-07 18:46:18 +0000507 // which shell interface should we use
508 //
509 if (mEfiShellProtocol != NULL) {
510 //
511 // use UEFI Shell 2.0 method.
512 //
513 FileName = mEfiShellProtocol->GetFilePathFromDevicePath(*FilePath);
514 if (FileName == NULL) {
515 return (EFI_INVALID_PARAMETER);
516 }
517 Status = ShellOpenFileByName(FileName, FileHandle, OpenMode, Attributes);
518 FreePool(FileName);
519 return (Status);
jcarsey1e6e84c2010-01-25 20:05:08 +0000520 }
jcarseyd2b45642009-05-11 18:02:16 +0000521
522
523 //
524 // use old shell method.
525 //
jcarsey1e6e84c2010-01-25 20:05:08 +0000526 Status = gBS->LocateDevicePath (&gEfiSimpleFileSystemProtocolGuid,
527 FilePath,
jcarseyd2b45642009-05-11 18:02:16 +0000528 DeviceHandle);
529 if (EFI_ERROR (Status)) {
530 return Status;
531 }
532 Status = gBS->OpenProtocol(*DeviceHandle,
533 &gEfiSimpleFileSystemProtocolGuid,
jcarseyb1f95a02009-06-16 00:23:19 +0000534 (VOID**)&EfiSimpleFileSystemProtocol,
jcarseyd2b45642009-05-11 18:02:16 +0000535 gImageHandle,
536 NULL,
537 EFI_OPEN_PROTOCOL_GET_PROTOCOL);
538 if (EFI_ERROR (Status)) {
539 return Status;
540 }
jcarseya405b862010-09-14 05:18:09 +0000541 Status = EfiSimpleFileSystemProtocol->OpenVolume(EfiSimpleFileSystemProtocol, &Handle1);
jcarseyd2b45642009-05-11 18:02:16 +0000542 if (EFI_ERROR (Status)) {
543 FileHandle = NULL;
544 return Status;
545 }
546
547 //
548 // go down directories one node at a time.
549 //
550 while (!IsDevicePathEnd (*FilePath)) {
jcarsey94b17fa2009-05-07 18:46:18 +0000551 //
jcarseyd2b45642009-05-11 18:02:16 +0000552 // For file system access each node should be a file path component
jcarsey94b17fa2009-05-07 18:46:18 +0000553 //
jcarseyd2b45642009-05-11 18:02:16 +0000554 if (DevicePathType (*FilePath) != MEDIA_DEVICE_PATH ||
555 DevicePathSubType (*FilePath) != MEDIA_FILEPATH_DP
jcarseya405b862010-09-14 05:18:09 +0000556 ) {
jcarsey94b17fa2009-05-07 18:46:18 +0000557 FileHandle = NULL;
jcarseyd2b45642009-05-11 18:02:16 +0000558 return (EFI_INVALID_PARAMETER);
jcarsey94b17fa2009-05-07 18:46:18 +0000559 }
jcarseyd2b45642009-05-11 18:02:16 +0000560 //
561 // Open this file path node
562 //
jcarseya405b862010-09-14 05:18:09 +0000563 Handle2 = Handle1;
564 Handle1 = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000565
566 //
jcarseyd2b45642009-05-11 18:02:16 +0000567 // Try to test opening an existing file
jcarsey94b17fa2009-05-07 18:46:18 +0000568 //
jcarseya405b862010-09-14 05:18:09 +0000569 Status = Handle2->Open (
570 Handle2,
571 &Handle1,
jcarseyd2b45642009-05-11 18:02:16 +0000572 ((FILEPATH_DEVICE_PATH*)*FilePath)->PathName,
573 OpenMode &~EFI_FILE_MODE_CREATE,
574 0
jcarseya405b862010-09-14 05:18:09 +0000575 );
jcarsey94b17fa2009-05-07 18:46:18 +0000576
jcarseyd2b45642009-05-11 18:02:16 +0000577 //
578 // see if the error was that it needs to be created
579 //
580 if ((EFI_ERROR (Status)) && (OpenMode != (OpenMode &~EFI_FILE_MODE_CREATE))) {
jcarseya405b862010-09-14 05:18:09 +0000581 Status = Handle2->Open (
582 Handle2,
583 &Handle1,
jcarsey94b17fa2009-05-07 18:46:18 +0000584 ((FILEPATH_DEVICE_PATH*)*FilePath)->PathName,
jcarseyd2b45642009-05-11 18:02:16 +0000585 OpenMode,
586 Attributes
jcarseya405b862010-09-14 05:18:09 +0000587 );
jcarsey94b17fa2009-05-07 18:46:18 +0000588 }
jcarseyd2b45642009-05-11 18:02:16 +0000589 //
590 // Close the last node
591 //
jcarseya405b862010-09-14 05:18:09 +0000592 Handle2->Close (Handle2);
jcarseyd2b45642009-05-11 18:02:16 +0000593
594 if (EFI_ERROR(Status)) {
595 return (Status);
596 }
597
598 //
599 // Get the next node
600 //
601 *FilePath = NextDevicePathNode (*FilePath);
jcarsey94b17fa2009-05-07 18:46:18 +0000602 }
jcarseya405b862010-09-14 05:18:09 +0000603
604 //
605 // This is a weak spot since if the undefined SHELL_FILE_HANDLE format changes this must change also!
606 //
607 *FileHandle = (VOID*)Handle1;
jcarseyd2b45642009-05-11 18:02:16 +0000608 return (EFI_SUCCESS);
jcarsey94b17fa2009-05-07 18:46:18 +0000609}
610
611/**
612 This function will open a file or directory referenced by filename.
613
jcarsey1e6e84c2010-01-25 20:05:08 +0000614 If return is EFI_SUCCESS, the Filehandle is the opened file's handle;
615 otherwise, the Filehandle is NULL. The Attributes is valid only for
jcarsey94b17fa2009-05-07 18:46:18 +0000616 EFI_FILE_MODE_CREATE.
617
618 if FileNAme is NULL then ASSERT()
619
620 @param FileName pointer to file name
621 @param FileHandle pointer to the file handle.
622 @param OpenMode the mode to open the file with.
623 @param Attributes the file's file attributes.
624
625 @retval EFI_SUCCESS The information was set.
626 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
jcarsey1e6e84c2010-01-25 20:05:08 +0000627 @retval EFI_UNSUPPORTED Could not open the file path.
628 @retval EFI_NOT_FOUND The specified file could not be found on the
629 device or the file system could not be found
jcarsey94b17fa2009-05-07 18:46:18 +0000630 on the device.
631 @retval EFI_NO_MEDIA The device has no medium.
jcarsey1e6e84c2010-01-25 20:05:08 +0000632 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
jcarsey94b17fa2009-05-07 18:46:18 +0000633 medium is no longer supported.
634 @retval EFI_DEVICE_ERROR The device reported an error.
635 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
636 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
637 @retval EFI_ACCESS_DENIED The file was opened read only.
jcarsey1e6e84c2010-01-25 20:05:08 +0000638 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
jcarsey94b17fa2009-05-07 18:46:18 +0000639 file.
640 @retval EFI_VOLUME_FULL The volume is full.
641**/
642EFI_STATUS
643EFIAPI
644ShellOpenFileByName(
jcarseyb82bfcc2009-06-29 16:28:23 +0000645 IN CONST CHAR16 *FileName,
jcarseya405b862010-09-14 05:18:09 +0000646 OUT SHELL_FILE_HANDLE *FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000647 IN UINT64 OpenMode,
648 IN UINT64 Attributes
jcarseya405b862010-09-14 05:18:09 +0000649 )
650{
jcarsey94b17fa2009-05-07 18:46:18 +0000651 EFI_HANDLE DeviceHandle;
652 EFI_DEVICE_PATH_PROTOCOL *FilePath;
jcarseyb1f95a02009-06-16 00:23:19 +0000653 EFI_STATUS Status;
654 EFI_FILE_INFO *FileInfo;
jcarsey94b17fa2009-05-07 18:46:18 +0000655
656 //
657 // ASSERT if FileName is NULL
658 //
659 ASSERT(FileName != NULL);
660
jcarseya405b862010-09-14 05:18:09 +0000661 if (FileName == NULL) {
662 return (EFI_INVALID_PARAMETER);
663 }
664
jcarsey94b17fa2009-05-07 18:46:18 +0000665 if (mEfiShellProtocol != NULL) {
jcarseya405b862010-09-14 05:18:09 +0000666 if ((OpenMode & EFI_FILE_MODE_CREATE) == EFI_FILE_MODE_CREATE && (Attributes & EFI_FILE_DIRECTORY) == EFI_FILE_DIRECTORY) {
667 return ShellCreateDirectory(FileName, FileHandle);
668 }
jcarsey94b17fa2009-05-07 18:46:18 +0000669 //
670 // Use UEFI Shell 2.0 method
671 //
jcarseyb1f95a02009-06-16 00:23:19 +0000672 Status = mEfiShellProtocol->OpenFileByName(FileName,
673 FileHandle,
674 OpenMode);
jcarseya405b862010-09-14 05:18:09 +0000675 if (StrCmp(FileName, L"NUL") != 0 && !EFI_ERROR(Status) && ((OpenMode & EFI_FILE_MODE_CREATE) != 0)){
jcarsey2247dde2009-11-09 18:08:58 +0000676 FileInfo = FileFunctionMap.GetFileInfo(*FileHandle);
jcarseyb1f95a02009-06-16 00:23:19 +0000677 ASSERT(FileInfo != NULL);
678 FileInfo->Attribute = Attributes;
jcarsey2247dde2009-11-09 18:08:58 +0000679 Status = FileFunctionMap.SetFileInfo(*FileHandle, FileInfo);
680 FreePool(FileInfo);
jcarseyb1f95a02009-06-16 00:23:19 +0000681 }
682 return (Status);
jcarsey1e6e84c2010-01-25 20:05:08 +0000683 }
jcarsey94b17fa2009-05-07 18:46:18 +0000684 //
685 // Using EFI Shell version
686 // this means convert name to path and call that function
687 // since this will use EFI method again that will open it.
688 //
689 ASSERT(mEfiShellEnvironment2 != NULL);
jcarseyb82bfcc2009-06-29 16:28:23 +0000690 FilePath = mEfiShellEnvironment2->NameToPath ((CHAR16*)FileName);
xdu290bfa222010-07-19 05:21:27 +0000691 if (FilePath != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000692 return (ShellOpenFileByDevicePath(&FilePath,
693 &DeviceHandle,
694 FileHandle,
695 OpenMode,
jcarseya405b862010-09-14 05:18:09 +0000696 Attributes));
jcarsey94b17fa2009-05-07 18:46:18 +0000697 }
698 return (EFI_DEVICE_ERROR);
699}
700/**
701 This function create a directory
702
jcarsey1e6e84c2010-01-25 20:05:08 +0000703 If return is EFI_SUCCESS, the Filehandle is the opened directory's handle;
704 otherwise, the Filehandle is NULL. If the directory already existed, this
jcarsey94b17fa2009-05-07 18:46:18 +0000705 function opens the existing directory.
706
707 @param DirectoryName pointer to directory name
708 @param FileHandle pointer to the file handle.
709
710 @retval EFI_SUCCESS The information was set.
711 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
jcarsey1e6e84c2010-01-25 20:05:08 +0000712 @retval EFI_UNSUPPORTED Could not open the file path.
713 @retval EFI_NOT_FOUND The specified file could not be found on the
714 device or the file system could not be found
jcarsey94b17fa2009-05-07 18:46:18 +0000715 on the device.
716 @retval EFI_NO_MEDIA The device has no medium.
jcarsey1e6e84c2010-01-25 20:05:08 +0000717 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
jcarsey94b17fa2009-05-07 18:46:18 +0000718 medium is no longer supported.
719 @retval EFI_DEVICE_ERROR The device reported an error.
720 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
721 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
722 @retval EFI_ACCESS_DENIED The file was opened read only.
jcarsey1e6e84c2010-01-25 20:05:08 +0000723 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
jcarsey94b17fa2009-05-07 18:46:18 +0000724 file.
725 @retval EFI_VOLUME_FULL The volume is full.
726 @sa ShellOpenFileByName
727**/
728EFI_STATUS
729EFIAPI
730ShellCreateDirectory(
jcarseyb82bfcc2009-06-29 16:28:23 +0000731 IN CONST CHAR16 *DirectoryName,
jcarseya405b862010-09-14 05:18:09 +0000732 OUT SHELL_FILE_HANDLE *FileHandle
733 )
734{
jcarsey2247dde2009-11-09 18:08:58 +0000735 if (mEfiShellProtocol != NULL) {
736 //
737 // Use UEFI Shell 2.0 method
738 //
739 return (mEfiShellProtocol->CreateFile(DirectoryName,
740 EFI_FILE_DIRECTORY,
741 FileHandle
jcarseya405b862010-09-14 05:18:09 +0000742 ));
jcarsey2247dde2009-11-09 18:08:58 +0000743 } else {
744 return (ShellOpenFileByName(DirectoryName,
745 FileHandle,
746 EFI_FILE_MODE_READ | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE,
747 EFI_FILE_DIRECTORY
jcarseya405b862010-09-14 05:18:09 +0000748 ));
jcarsey2247dde2009-11-09 18:08:58 +0000749 }
jcarsey94b17fa2009-05-07 18:46:18 +0000750}
751
752/**
753 This function reads information from an opened file.
754
jcarsey1e6e84c2010-01-25 20:05:08 +0000755 If FileHandle is not a directory, the function reads the requested number of
756 bytes from the file at the file's current position and returns them in Buffer.
jcarsey94b17fa2009-05-07 18:46:18 +0000757 If the read goes beyond the end of the file, the read length is truncated to the
jcarsey1e6e84c2010-01-25 20:05:08 +0000758 end of the file. The file's current position is increased by the number of bytes
759 returned. If FileHandle is a directory, the function reads the directory entry
760 at the file's current position and returns the entry in Buffer. If the Buffer
761 is not large enough to hold the current directory entry, then
762 EFI_BUFFER_TOO_SMALL is returned and the current file position is not updated.
763 BufferSize is set to be the size of the buffer needed to read the entry. On
764 success, the current position is updated to the next directory entry. If there
765 are no more directory entries, the read returns a zero-length buffer.
jcarsey94b17fa2009-05-07 18:46:18 +0000766 EFI_FILE_INFO is the structure returned as the directory entry.
767
768 @param FileHandle the opened file handle
jcarsey1e6e84c2010-01-25 20:05:08 +0000769 @param BufferSize on input the size of buffer in bytes. on return
jcarsey94b17fa2009-05-07 18:46:18 +0000770 the number of bytes written.
771 @param Buffer the buffer to put read data into.
772
773 @retval EFI_SUCCESS Data was read.
774 @retval EFI_NO_MEDIA The device has no media.
775 @retval EFI_DEVICE_ERROR The device reported an error.
776 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
jcarsey1e6e84c2010-01-25 20:05:08 +0000777 @retval EFI_BUFFER_TO_SMALL Buffer is too small. ReadSize contains required
jcarsey94b17fa2009-05-07 18:46:18 +0000778 size.
779
780**/
781EFI_STATUS
782EFIAPI
783ShellReadFile(
jcarseya405b862010-09-14 05:18:09 +0000784 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000785 IN OUT UINTN *BufferSize,
786 OUT VOID *Buffer
jcarseya405b862010-09-14 05:18:09 +0000787 )
788{
jcarseyd2b45642009-05-11 18:02:16 +0000789 return (FileFunctionMap.ReadFile(FileHandle, BufferSize, Buffer));
jcarsey94b17fa2009-05-07 18:46:18 +0000790}
791
792
793/**
794 Write data to a file.
795
jcarsey1e6e84c2010-01-25 20:05:08 +0000796 This function writes the specified number of bytes to the file at the current
797 file position. The current file position is advanced the actual number of bytes
798 written, which is returned in BufferSize. Partial writes only occur when there
799 has been a data error during the write attempt (such as "volume space full").
800 The file is automatically grown to hold the data if required. Direct writes to
jcarsey94b17fa2009-05-07 18:46:18 +0000801 opened directories are not supported.
802
803 @param FileHandle The opened file for writing
804 @param BufferSize on input the number of bytes in Buffer. On output
805 the number of bytes written.
806 @param Buffer the buffer containing data to write is stored.
807
808 @retval EFI_SUCCESS Data was written.
809 @retval EFI_UNSUPPORTED Writes to an open directory are not supported.
810 @retval EFI_NO_MEDIA The device has no media.
811 @retval EFI_DEVICE_ERROR The device reported an error.
812 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
813 @retval EFI_WRITE_PROTECTED The device is write-protected.
814 @retval EFI_ACCESS_DENIED The file was open for read only.
815 @retval EFI_VOLUME_FULL The volume is full.
816**/
817EFI_STATUS
818EFIAPI
819ShellWriteFile(
jcarseya405b862010-09-14 05:18:09 +0000820 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000821 IN OUT UINTN *BufferSize,
822 IN VOID *Buffer
jcarseya405b862010-09-14 05:18:09 +0000823 )
824{
jcarseyd2b45642009-05-11 18:02:16 +0000825 return (FileFunctionMap.WriteFile(FileHandle, BufferSize, Buffer));
jcarsey94b17fa2009-05-07 18:46:18 +0000826}
827
jcarsey1e6e84c2010-01-25 20:05:08 +0000828/**
jcarsey94b17fa2009-05-07 18:46:18 +0000829 Close an open file handle.
830
jcarsey1e6e84c2010-01-25 20:05:08 +0000831 This function closes a specified file handle. All "dirty" cached file data is
832 flushed to the device, and the file is closed. In all cases the handle is
jcarsey94b17fa2009-05-07 18:46:18 +0000833 closed.
834
835@param FileHandle the file handle to close.
836
837@retval EFI_SUCCESS the file handle was closed sucessfully.
838**/
839EFI_STATUS
840EFIAPI
841ShellCloseFile (
jcarseya405b862010-09-14 05:18:09 +0000842 IN SHELL_FILE_HANDLE *FileHandle
843 )
844{
jcarseyd2b45642009-05-11 18:02:16 +0000845 return (FileFunctionMap.CloseFile(*FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000846}
847
848/**
849 Delete a file and close the handle
850
851 This function closes and deletes a file. In all cases the file handle is closed.
jcarsey1e6e84c2010-01-25 20:05:08 +0000852 If the file cannot be deleted, the warning code EFI_WARN_DELETE_FAILURE is
jcarsey94b17fa2009-05-07 18:46:18 +0000853 returned, but the handle is still closed.
854
855 @param FileHandle the file handle to delete
856
857 @retval EFI_SUCCESS the file was closed sucessfully
jcarsey1e6e84c2010-01-25 20:05:08 +0000858 @retval EFI_WARN_DELETE_FAILURE the handle was closed, but the file was not
jcarsey94b17fa2009-05-07 18:46:18 +0000859 deleted
860 @retval INVALID_PARAMETER One of the parameters has an invalid value.
861**/
862EFI_STATUS
863EFIAPI
864ShellDeleteFile (
jcarseya405b862010-09-14 05:18:09 +0000865 IN SHELL_FILE_HANDLE *FileHandle
866 )
867{
jcarseyd2b45642009-05-11 18:02:16 +0000868 return (FileFunctionMap.DeleteFile(*FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000869}
870
871/**
872 Set the current position in a file.
873
jcarsey1e6e84c2010-01-25 20:05:08 +0000874 This function sets the current file position for the handle to the position
jcarsey94b17fa2009-05-07 18:46:18 +0000875 supplied. With the exception of seeking to position 0xFFFFFFFFFFFFFFFF, only
jcarsey1e6e84c2010-01-25 20:05:08 +0000876 absolute positioning is supported, and seeking past the end of the file is
877 allowed (a subsequent write would grow the file). Seeking to position
jcarsey94b17fa2009-05-07 18:46:18 +0000878 0xFFFFFFFFFFFFFFFF causes the current position to be set to the end of the file.
jcarsey1e6e84c2010-01-25 20:05:08 +0000879 If FileHandle is a directory, the only position that may be set is zero. This
jcarsey94b17fa2009-05-07 18:46:18 +0000880 has the effect of starting the read process of the directory entries over.
881
882 @param FileHandle The file handle on which the position is being set
883 @param Position Byte position from begining of file
884
885 @retval EFI_SUCCESS Operation completed sucessfully.
jcarsey1e6e84c2010-01-25 20:05:08 +0000886 @retval EFI_UNSUPPORTED the seek request for non-zero is not valid on
jcarsey94b17fa2009-05-07 18:46:18 +0000887 directories.
888 @retval INVALID_PARAMETER One of the parameters has an invalid value.
889**/
890EFI_STATUS
891EFIAPI
892ShellSetFilePosition (
jcarseya405b862010-09-14 05:18:09 +0000893 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000894 IN UINT64 Position
jcarseya405b862010-09-14 05:18:09 +0000895 )
896{
jcarseyd2b45642009-05-11 18:02:16 +0000897 return (FileFunctionMap.SetFilePosition(FileHandle, Position));
jcarsey94b17fa2009-05-07 18:46:18 +0000898}
899
jcarsey1e6e84c2010-01-25 20:05:08 +0000900/**
jcarsey94b17fa2009-05-07 18:46:18 +0000901 Gets a file's current position
902
jcarsey1e6e84c2010-01-25 20:05:08 +0000903 This function retrieves the current file position for the file handle. For
904 directories, the current file position has no meaning outside of the file
jcarsey94b17fa2009-05-07 18:46:18 +0000905 system driver and as such the operation is not supported. An error is returned
906 if FileHandle is a directory.
907
908 @param FileHandle The open file handle on which to get the position.
909 @param Position Byte position from begining of file.
910
911 @retval EFI_SUCCESS the operation completed sucessfully.
912 @retval INVALID_PARAMETER One of the parameters has an invalid value.
913 @retval EFI_UNSUPPORTED the request is not valid on directories.
914**/
915EFI_STATUS
916EFIAPI
917ShellGetFilePosition (
jcarseya405b862010-09-14 05:18:09 +0000918 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000919 OUT UINT64 *Position
jcarseya405b862010-09-14 05:18:09 +0000920 )
921{
jcarseyd2b45642009-05-11 18:02:16 +0000922 return (FileFunctionMap.GetFilePosition(FileHandle, Position));
jcarsey94b17fa2009-05-07 18:46:18 +0000923}
924/**
925 Flushes data on a file
jcarsey1e6e84c2010-01-25 20:05:08 +0000926
jcarsey94b17fa2009-05-07 18:46:18 +0000927 This function flushes all modified data associated with a file to a device.
928
929 @param FileHandle The file handle on which to flush data
930
931 @retval EFI_SUCCESS The data was flushed.
932 @retval EFI_NO_MEDIA The device has no media.
933 @retval EFI_DEVICE_ERROR The device reported an error.
934 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
935 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
936 @retval EFI_ACCESS_DENIED The file was opened for read only.
937**/
938EFI_STATUS
939EFIAPI
940ShellFlushFile (
jcarseya405b862010-09-14 05:18:09 +0000941 IN SHELL_FILE_HANDLE FileHandle
942 )
943{
jcarseyd2b45642009-05-11 18:02:16 +0000944 return (FileFunctionMap.FlushFile(FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000945}
946
947/**
948 Retrieves the first file from a directory
949
jcarsey1e6e84c2010-01-25 20:05:08 +0000950 This function opens a directory and gets the first file's info in the
951 directory. Caller can use ShellFindNextFile() to get other files. When
jcarsey94b17fa2009-05-07 18:46:18 +0000952 complete the caller is responsible for calling FreePool() on Buffer.
953
954 @param DirHandle The file handle of the directory to search
955 @param Buffer Pointer to buffer for file's information
956
957 @retval EFI_SUCCESS Found the first file.
958 @retval EFI_NOT_FOUND Cannot find the directory.
959 @retval EFI_NO_MEDIA The device has no media.
960 @retval EFI_DEVICE_ERROR The device reported an error.
961 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
962 @return Others status of ShellGetFileInfo, ShellSetFilePosition,
963 or ShellReadFile
964**/
965EFI_STATUS
966EFIAPI
967ShellFindFirstFile (
jcarseya405b862010-09-14 05:18:09 +0000968 IN SHELL_FILE_HANDLE DirHandle,
jcarseyd2b45642009-05-11 18:02:16 +0000969 OUT EFI_FILE_INFO **Buffer
jcarseya405b862010-09-14 05:18:09 +0000970 )
971{
jcarsey94b17fa2009-05-07 18:46:18 +0000972 //
jcarseyd2b45642009-05-11 18:02:16 +0000973 // pass to file handle lib
jcarsey94b17fa2009-05-07 18:46:18 +0000974 //
jcarseyd2b45642009-05-11 18:02:16 +0000975 return (FileHandleFindFirstFile(DirHandle, Buffer));
jcarsey94b17fa2009-05-07 18:46:18 +0000976}
977/**
978 Retrieves the next file in a directory.
979
jcarseya405b862010-09-14 05:18:09 +0000980 To use this function, caller must call the ShellFindFirstFile() to get the
jcarsey1e6e84c2010-01-25 20:05:08 +0000981 first file, and then use this function get other files. This function can be
982 called for several times to get each file's information in the directory. If
983 the call of ShellFindNextFile() got the last file in the directory, the next
984 call of this function has no file to get. *NoFile will be set to TRUE and the
985 Buffer memory will be automatically freed.
jcarsey94b17fa2009-05-07 18:46:18 +0000986
987 @param DirHandle the file handle of the directory
988 @param Buffer pointer to buffer for file's information
989 @param NoFile pointer to boolean when last file is found
990
991 @retval EFI_SUCCESS Found the next file, or reached last file
992 @retval EFI_NO_MEDIA The device has no media.
993 @retval EFI_DEVICE_ERROR The device reported an error.
994 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
995**/
996EFI_STATUS
997EFIAPI
998ShellFindNextFile(
jcarseya405b862010-09-14 05:18:09 +0000999 IN SHELL_FILE_HANDLE DirHandle,
jcarsey94b17fa2009-05-07 18:46:18 +00001000 OUT EFI_FILE_INFO *Buffer,
1001 OUT BOOLEAN *NoFile
jcarseya405b862010-09-14 05:18:09 +00001002 )
1003{
jcarsey94b17fa2009-05-07 18:46:18 +00001004 //
jcarseyd2b45642009-05-11 18:02:16 +00001005 // pass to file handle lib
jcarsey94b17fa2009-05-07 18:46:18 +00001006 //
jcarseyd2b45642009-05-11 18:02:16 +00001007 return (FileHandleFindNextFile(DirHandle, Buffer, NoFile));
jcarsey94b17fa2009-05-07 18:46:18 +00001008}
1009/**
1010 Retrieve the size of a file.
1011
1012 if FileHandle is NULL then ASSERT()
1013 if Size is NULL then ASSERT()
1014
jcarsey1e6e84c2010-01-25 20:05:08 +00001015 This function extracts the file size info from the FileHandle's EFI_FILE_INFO
jcarsey94b17fa2009-05-07 18:46:18 +00001016 data.
1017
1018 @param FileHandle file handle from which size is retrieved
1019 @param Size pointer to size
1020
1021 @retval EFI_SUCCESS operation was completed sucessfully
1022 @retval EFI_DEVICE_ERROR cannot access the file
1023**/
1024EFI_STATUS
1025EFIAPI
1026ShellGetFileSize (
jcarseya405b862010-09-14 05:18:09 +00001027 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +00001028 OUT UINT64 *Size
jcarseya405b862010-09-14 05:18:09 +00001029 )
1030{
jcarseyd2b45642009-05-11 18:02:16 +00001031 return (FileFunctionMap.GetFileSize(FileHandle, Size));
jcarsey94b17fa2009-05-07 18:46:18 +00001032}
1033/**
1034 Retrieves the status of the break execution flag
1035
1036 this function is useful to check whether the application is being asked to halt by the shell.
1037
1038 @retval TRUE the execution break is enabled
1039 @retval FALSE the execution break is not enabled
1040**/
1041BOOLEAN
1042EFIAPI
1043ShellGetExecutionBreakFlag(
1044 VOID
1045 )
1046{
jcarsey1e6e84c2010-01-25 20:05:08 +00001047 //
jcarsey94b17fa2009-05-07 18:46:18 +00001048 // Check for UEFI Shell 2.0 protocols
1049 //
1050 if (mEfiShellProtocol != NULL) {
1051
1052 //
1053 // We are using UEFI Shell 2.0; see if the event has been triggered
1054 //
1055 if (gBS->CheckEvent(mEfiShellProtocol->ExecutionBreak) != EFI_SUCCESS) {
1056 return (FALSE);
1057 }
1058 return (TRUE);
jcarsey1e6e84c2010-01-25 20:05:08 +00001059 }
jcarsey94b17fa2009-05-07 18:46:18 +00001060
1061 //
1062 // using EFI Shell; call the function to check
1063 //
1064 ASSERT(mEfiShellEnvironment2 != NULL);
1065 return (mEfiShellEnvironment2->GetExecutionBreak());
1066}
1067/**
1068 return the value of an environment variable
1069
jcarsey1e6e84c2010-01-25 20:05:08 +00001070 this function gets the value of the environment variable set by the
jcarsey94b17fa2009-05-07 18:46:18 +00001071 ShellSetEnvironmentVariable function
1072
1073 @param EnvKey The key name of the environment variable.
1074
1075 @retval NULL the named environment variable does not exist.
1076 @return != NULL pointer to the value of the environment variable
1077**/
1078CONST CHAR16*
1079EFIAPI
1080ShellGetEnvironmentVariable (
jcarsey9b3bf082009-06-23 21:15:07 +00001081 IN CONST CHAR16 *EnvKey
jcarsey94b17fa2009-05-07 18:46:18 +00001082 )
1083{
jcarsey1e6e84c2010-01-25 20:05:08 +00001084 //
jcarsey94b17fa2009-05-07 18:46:18 +00001085 // Check for UEFI Shell 2.0 protocols
1086 //
1087 if (mEfiShellProtocol != NULL) {
1088 return (mEfiShellProtocol->GetEnv(EnvKey));
1089 }
1090
1091 //
1092 // ASSERT that we must have EFI shell
1093 //
1094 ASSERT(mEfiShellEnvironment2 != NULL);
1095
1096 //
1097 // using EFI Shell
1098 //
jcarsey9b3bf082009-06-23 21:15:07 +00001099 return (mEfiShellEnvironment2->GetEnv((CHAR16*)EnvKey));
jcarsey94b17fa2009-05-07 18:46:18 +00001100}
1101/**
1102 set the value of an environment variable
1103
1104This function changes the current value of the specified environment variable. If the
1105environment variable exists and the Value is an empty string, then the environment
1106variable is deleted. If the environment variable exists and the Value is not an empty
1107string, then the value of the environment variable is changed. If the environment
1108variable does not exist and the Value is an empty string, there is no action. If the
1109environment variable does not exist and the Value is a non-empty string, then the
1110environment variable is created and assigned the specified value.
1111
1112 This is not supported pre-UEFI Shell 2.0.
1113
1114 @param EnvKey The key name of the environment variable.
1115 @param EnvVal The Value of the environment variable
1116 @param Volatile Indicates whether the variable is non-volatile (FALSE) or volatile (TRUE).
1117
1118 @retval EFI_SUCCESS the operation was completed sucessfully
1119 @retval EFI_UNSUPPORTED This operation is not allowed in pre UEFI 2.0 Shell environments
1120**/
1121EFI_STATUS
1122EFIAPI
1123ShellSetEnvironmentVariable (
1124 IN CONST CHAR16 *EnvKey,
1125 IN CONST CHAR16 *EnvVal,
1126 IN BOOLEAN Volatile
1127 )
1128{
jcarsey1e6e84c2010-01-25 20:05:08 +00001129 //
jcarsey94b17fa2009-05-07 18:46:18 +00001130 // Check for UEFI Shell 2.0 protocols
1131 //
1132 if (mEfiShellProtocol != NULL) {
1133 return (mEfiShellProtocol->SetEnv(EnvKey, EnvVal, Volatile));
jcarsey1e6e84c2010-01-25 20:05:08 +00001134 }
jcarsey94b17fa2009-05-07 18:46:18 +00001135
1136 //
1137 // This feature does not exist under EFI shell
1138 //
1139 return (EFI_UNSUPPORTED);
1140}
jcarseya405b862010-09-14 05:18:09 +00001141
jcarsey94b17fa2009-05-07 18:46:18 +00001142/**
jcarseya405b862010-09-14 05:18:09 +00001143 Cause the shell to parse and execute a command line.
jcarsey94b17fa2009-05-07 18:46:18 +00001144
1145 This function creates a nested instance of the shell and executes the specified
jcarseya405b862010-09-14 05:18:09 +00001146 command (CommandLine) with the specified environment (Environment). Upon return,
1147 the status code returned by the specified command is placed in StatusCode.
1148 If Environment is NULL, then the current environment is used and all changes made
1149 by the commands executed will be reflected in the current environment. If the
1150 Environment is non-NULL, then the changes made will be discarded.
1151 The CommandLine is executed from the current working directory on the current
1152 device.
jcarsey94b17fa2009-05-07 18:46:18 +00001153
jcarseya405b862010-09-14 05:18:09 +00001154 The EnvironmentVariables and Status parameters are ignored in a pre-UEFI Shell 2.0
1155 environment. The values pointed to by the parameters will be unchanged by the
1156 ShellExecute() function. The Output parameter has no effect in a
1157 UEFI Shell 2.0 environment.
jcarsey94b17fa2009-05-07 18:46:18 +00001158
jcarseya405b862010-09-14 05:18:09 +00001159 @param[in] ParentHandle The parent image starting the operation.
1160 @param[in] CommandLine The pointer to a NULL terminated command line.
1161 @param[in] Output True to display debug output. False to hide it.
1162 @param[in] EnvironmentVariables Optional pointer to array of environment variables
1163 in the form "x=y". If NULL, the current set is used.
1164 @param[out] Status The status of the run command line.
jcarsey94b17fa2009-05-07 18:46:18 +00001165
jcarseya405b862010-09-14 05:18:09 +00001166 @retval EFI_SUCCESS The operation completed sucessfully. Status
1167 contains the status code returned.
1168 @retval EFI_INVALID_PARAMETER A parameter contains an invalid value.
1169 @retval EFI_OUT_OF_RESOURCES Out of resources.
1170 @retval EFI_UNSUPPORTED The operation is not allowed.
jcarsey94b17fa2009-05-07 18:46:18 +00001171**/
jcarseya405b862010-09-14 05:18:09 +00001172
jcarsey94b17fa2009-05-07 18:46:18 +00001173EFI_STATUS
1174EFIAPI
1175ShellExecute (
1176 IN EFI_HANDLE *ParentHandle,
1177 IN CHAR16 *CommandLine OPTIONAL,
1178 IN BOOLEAN Output OPTIONAL,
1179 IN CHAR16 **EnvironmentVariables OPTIONAL,
1180 OUT EFI_STATUS *Status OPTIONAL
1181 )
1182{
jcarsey1e6e84c2010-01-25 20:05:08 +00001183 //
jcarsey94b17fa2009-05-07 18:46:18 +00001184 // Check for UEFI Shell 2.0 protocols
1185 //
1186 if (mEfiShellProtocol != NULL) {
1187 //
1188 // Call UEFI Shell 2.0 version (not using Output parameter)
1189 //
1190 return (mEfiShellProtocol->Execute(ParentHandle,
1191 CommandLine,
1192 EnvironmentVariables,
1193 Status));
jcarsey1e6e84c2010-01-25 20:05:08 +00001194 }
jcarsey94b17fa2009-05-07 18:46:18 +00001195 //
1196 // ASSERT that we must have EFI shell
1197 //
1198 ASSERT(mEfiShellEnvironment2 != NULL);
1199 //
1200 // Call EFI Shell version (not using EnvironmentVariables or Status parameters)
1201 // Due to oddity in the EFI shell we want to dereference the ParentHandle here
1202 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001203 return (mEfiShellEnvironment2->Execute(*ParentHandle,
1204 CommandLine,
jcarsey94b17fa2009-05-07 18:46:18 +00001205 Output));
1206}
1207/**
1208 Retreives the current directory path
1209
jcarsey1e6e84c2010-01-25 20:05:08 +00001210 If the DeviceName is NULL, it returns the current device's current directory
1211 name. If the DeviceName is not NULL, it returns the current directory name
jcarsey94b17fa2009-05-07 18:46:18 +00001212 on specified drive.
1213
1214 @param DeviceName the name of the drive to get directory on
1215
1216 @retval NULL the directory does not exist
1217 @return != NULL the directory
1218**/
1219CONST CHAR16*
1220EFIAPI
1221ShellGetCurrentDir (
jcarseya405b862010-09-14 05:18:09 +00001222 IN CHAR16 * CONST DeviceName OPTIONAL
jcarsey94b17fa2009-05-07 18:46:18 +00001223 )
1224{
jcarsey1e6e84c2010-01-25 20:05:08 +00001225 //
jcarsey94b17fa2009-05-07 18:46:18 +00001226 // Check for UEFI Shell 2.0 protocols
1227 //
1228 if (mEfiShellProtocol != NULL) {
1229 return (mEfiShellProtocol->GetCurDir(DeviceName));
jcarsey1e6e84c2010-01-25 20:05:08 +00001230 }
jcarsey94b17fa2009-05-07 18:46:18 +00001231 //
1232 // ASSERT that we must have EFI shell
1233 //
1234 ASSERT(mEfiShellEnvironment2 != NULL);
1235 return (mEfiShellEnvironment2->CurDir(DeviceName));
1236}
1237/**
1238 sets (enabled or disabled) the page break mode
1239
jcarsey1e6e84c2010-01-25 20:05:08 +00001240 when page break mode is enabled the screen will stop scrolling
jcarsey94b17fa2009-05-07 18:46:18 +00001241 and wait for operator input before scrolling a subsequent screen.
1242
1243 @param CurrentState TRUE to enable and FALSE to disable
1244**/
jcarsey1e6e84c2010-01-25 20:05:08 +00001245VOID
jcarsey94b17fa2009-05-07 18:46:18 +00001246EFIAPI
1247ShellSetPageBreakMode (
1248 IN BOOLEAN CurrentState
1249 )
1250{
1251 //
1252 // check for enabling
1253 //
1254 if (CurrentState != 0x00) {
jcarsey1e6e84c2010-01-25 20:05:08 +00001255 //
jcarsey94b17fa2009-05-07 18:46:18 +00001256 // check for UEFI Shell 2.0
1257 //
1258 if (mEfiShellProtocol != NULL) {
1259 //
1260 // Enable with UEFI 2.0 Shell
1261 //
1262 mEfiShellProtocol->EnablePageBreak();
1263 return;
1264 } else {
jcarsey1e6e84c2010-01-25 20:05:08 +00001265 //
jcarsey94b17fa2009-05-07 18:46:18 +00001266 // ASSERT that must have EFI Shell
1267 //
1268 ASSERT(mEfiShellEnvironment2 != NULL);
1269 //
1270 // Enable with EFI Shell
1271 //
1272 mEfiShellEnvironment2->EnablePageBreak (DEFAULT_INIT_ROW, DEFAULT_AUTO_LF);
1273 return;
1274 }
1275 } else {
jcarsey1e6e84c2010-01-25 20:05:08 +00001276 //
jcarsey94b17fa2009-05-07 18:46:18 +00001277 // check for UEFI Shell 2.0
1278 //
1279 if (mEfiShellProtocol != NULL) {
1280 //
1281 // Disable with UEFI 2.0 Shell
1282 //
1283 mEfiShellProtocol->DisablePageBreak();
1284 return;
1285 } else {
jcarsey1e6e84c2010-01-25 20:05:08 +00001286 //
jcarsey94b17fa2009-05-07 18:46:18 +00001287 // ASSERT that must have EFI Shell
1288 //
1289 ASSERT(mEfiShellEnvironment2 != NULL);
1290 //
1291 // Disable with EFI Shell
1292 //
1293 mEfiShellEnvironment2->DisablePageBreak ();
1294 return;
1295 }
1296 }
1297}
1298
1299///
1300/// version of EFI_SHELL_FILE_INFO struct, except has no CONST pointers.
1301/// This allows for the struct to be populated.
1302///
1303typedef struct {
jcarseyd2b45642009-05-11 18:02:16 +00001304 LIST_ENTRY Link;
jcarsey94b17fa2009-05-07 18:46:18 +00001305 EFI_STATUS Status;
1306 CHAR16 *FullName;
1307 CHAR16 *FileName;
jcarseya405b862010-09-14 05:18:09 +00001308 SHELL_FILE_HANDLE Handle;
jcarsey94b17fa2009-05-07 18:46:18 +00001309 EFI_FILE_INFO *Info;
1310} EFI_SHELL_FILE_INFO_NO_CONST;
1311
1312/**
1313 Converts a EFI shell list of structures to the coresponding UEFI Shell 2.0 type of list.
1314
1315 if OldStyleFileList is NULL then ASSERT()
1316
jcarsey1e6e84c2010-01-25 20:05:08 +00001317 this function will convert a SHELL_FILE_ARG based list into a callee allocated
jcarsey94b17fa2009-05-07 18:46:18 +00001318 EFI_SHELL_FILE_INFO based list. it is up to the caller to free the memory via
1319 the ShellCloseFileMetaArg function.
1320
jcarseya405b862010-09-14 05:18:09 +00001321 @param[in] FileList the EFI shell list type
jcarseyb82bfcc2009-06-29 16:28:23 +00001322 @param[in,out] ListHead the list to add to
jcarsey94b17fa2009-05-07 18:46:18 +00001323
1324 @retval the resultant head of the double linked new format list;
1325**/
1326LIST_ENTRY*
1327EFIAPI
1328InternalShellConvertFileListType (
jcarsey9b3bf082009-06-23 21:15:07 +00001329 IN LIST_ENTRY *FileList,
1330 IN OUT LIST_ENTRY *ListHead
jcarsey125c2cf2009-11-18 21:36:50 +00001331 )
1332{
jcarsey94b17fa2009-05-07 18:46:18 +00001333 SHELL_FILE_ARG *OldInfo;
jcarsey9b3bf082009-06-23 21:15:07 +00001334 LIST_ENTRY *Link;
jcarsey94b17fa2009-05-07 18:46:18 +00001335 EFI_SHELL_FILE_INFO_NO_CONST *NewInfo;
1336
1337 //
jcarsey9b3bf082009-06-23 21:15:07 +00001338 // ASSERTs
jcarsey94b17fa2009-05-07 18:46:18 +00001339 //
jcarsey9b3bf082009-06-23 21:15:07 +00001340 ASSERT(FileList != NULL);
1341 ASSERT(ListHead != NULL);
jcarsey94b17fa2009-05-07 18:46:18 +00001342
1343 //
1344 // enumerate through each member of the old list and copy
1345 //
jcarseyd2b45642009-05-11 18:02:16 +00001346 for (Link = FileList->ForwardLink; Link != FileList; Link = Link->ForwardLink) {
jcarsey94b17fa2009-05-07 18:46:18 +00001347 OldInfo = CR (Link, SHELL_FILE_ARG, Link, SHELL_FILE_ARG_SIGNATURE);
jcarseya405b862010-09-14 05:18:09 +00001348 ASSERT(OldInfo != NULL);
1349
1350 //
1351 // Skip ones that failed to open...
1352 //
1353 if (OldInfo->Status != EFI_SUCCESS) {
1354 continue;
1355 }
jcarsey94b17fa2009-05-07 18:46:18 +00001356
1357 //
1358 // make sure the old list was valid
1359 //
jcarsey94b17fa2009-05-07 18:46:18 +00001360 ASSERT(OldInfo->Info != NULL);
1361 ASSERT(OldInfo->FullName != NULL);
1362 ASSERT(OldInfo->FileName != NULL);
1363
1364 //
1365 // allocate a new EFI_SHELL_FILE_INFO object
1366 //
1367 NewInfo = AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
jcarseyc9d92df2010-02-03 15:37:54 +00001368 ASSERT(NewInfo != NULL);
1369 if (NewInfo == NULL) {
1370 break;
1371 }
jcarsey1e6e84c2010-01-25 20:05:08 +00001372
1373 //
jcarsey94b17fa2009-05-07 18:46:18 +00001374 // copy the simple items
1375 //
1376 NewInfo->Handle = OldInfo->Handle;
1377 NewInfo->Status = OldInfo->Status;
1378
jcarseyd2b45642009-05-11 18:02:16 +00001379 // old shell checks for 0 not NULL
1380 OldInfo->Handle = 0;
1381
jcarsey94b17fa2009-05-07 18:46:18 +00001382 //
1383 // allocate new space to copy strings and structure
1384 //
1385 NewInfo->FullName = AllocateZeroPool(StrSize(OldInfo->FullName));
1386 NewInfo->FileName = AllocateZeroPool(StrSize(OldInfo->FileName));
1387 NewInfo->Info = AllocateZeroPool((UINTN)OldInfo->Info->Size);
jcarsey1e6e84c2010-01-25 20:05:08 +00001388
jcarsey94b17fa2009-05-07 18:46:18 +00001389 //
1390 // make sure all the memory allocations were sucessful
1391 //
1392 ASSERT(NewInfo->FullName != NULL);
1393 ASSERT(NewInfo->FileName != NULL);
1394 ASSERT(NewInfo->Info != NULL);
1395
1396 //
1397 // Copt the strings and structure
1398 //
1399 StrCpy(NewInfo->FullName, OldInfo->FullName);
1400 StrCpy(NewInfo->FileName, OldInfo->FileName);
1401 gBS->CopyMem (NewInfo->Info, OldInfo->Info, (UINTN)OldInfo->Info->Size);
1402
1403 //
1404 // add that to the list
1405 //
jcarsey9b3bf082009-06-23 21:15:07 +00001406 InsertTailList(ListHead, &NewInfo->Link);
jcarsey94b17fa2009-05-07 18:46:18 +00001407 }
1408 return (ListHead);
1409}
1410/**
1411 Opens a group of files based on a path.
1412
jcarsey1e6e84c2010-01-25 20:05:08 +00001413 This function uses the Arg to open all the matching files. Each matched
1414 file has a SHELL_FILE_ARG structure to record the file information. These
1415 structures are placed on the list ListHead. Users can get the SHELL_FILE_ARG
jcarsey94b17fa2009-05-07 18:46:18 +00001416 structures from ListHead to access each file. This function supports wildcards
jcarsey1e6e84c2010-01-25 20:05:08 +00001417 and will process '?' and '*' as such. the list must be freed with a call to
jcarsey94b17fa2009-05-07 18:46:18 +00001418 ShellCloseFileMetaArg().
1419
jcarsey1e6e84c2010-01-25 20:05:08 +00001420 If you are NOT appending to an existing list *ListHead must be NULL. If
jcarsey5f7431d2009-07-10 18:06:01 +00001421 *ListHead is NULL then it must be callee freed.
jcarsey94b17fa2009-05-07 18:46:18 +00001422
1423 @param Arg pointer to path string
1424 @param OpenMode mode to open files with
1425 @param ListHead head of linked list of results
1426
jcarsey1e6e84c2010-01-25 20:05:08 +00001427 @retval EFI_SUCCESS the operation was sucessful and the list head
jcarsey94b17fa2009-05-07 18:46:18 +00001428 contains the list of opened files
jcarsey94b17fa2009-05-07 18:46:18 +00001429 @return != EFI_SUCCESS the operation failed
1430
1431 @sa InternalShellConvertFileListType
1432**/
1433EFI_STATUS
1434EFIAPI
1435ShellOpenFileMetaArg (
1436 IN CHAR16 *Arg,
1437 IN UINT64 OpenMode,
1438 IN OUT EFI_SHELL_FILE_INFO **ListHead
1439 )
1440{
1441 EFI_STATUS Status;
jcarsey9b3bf082009-06-23 21:15:07 +00001442 LIST_ENTRY mOldStyleFileList;
jcarsey1e6e84c2010-01-25 20:05:08 +00001443
jcarsey94b17fa2009-05-07 18:46:18 +00001444 //
1445 // ASSERT that Arg and ListHead are not NULL
1446 //
1447 ASSERT(Arg != NULL);
1448 ASSERT(ListHead != NULL);
1449
jcarsey1e6e84c2010-01-25 20:05:08 +00001450 //
jcarsey94b17fa2009-05-07 18:46:18 +00001451 // Check for UEFI Shell 2.0 protocols
1452 //
1453 if (mEfiShellProtocol != NULL) {
jcarsey5f7431d2009-07-10 18:06:01 +00001454 if (*ListHead == NULL) {
1455 *ListHead = (EFI_SHELL_FILE_INFO*)AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1456 if (*ListHead == NULL) {
1457 return (EFI_OUT_OF_RESOURCES);
1458 }
1459 InitializeListHead(&((*ListHead)->Link));
jcarsey1e6e84c2010-01-25 20:05:08 +00001460 }
1461 Status = mEfiShellProtocol->OpenFileList(Arg,
1462 OpenMode,
jcarsey2247dde2009-11-09 18:08:58 +00001463 ListHead);
1464 if (EFI_ERROR(Status)) {
1465 mEfiShellProtocol->RemoveDupInFileList(ListHead);
1466 } else {
1467 Status = mEfiShellProtocol->RemoveDupInFileList(ListHead);
1468 }
jcarseya405b862010-09-14 05:18:09 +00001469 if (*ListHead != NULL && IsListEmpty(&(*ListHead)->Link)) {
1470 FreePool(*ListHead);
1471 *ListHead = NULL;
1472 return (EFI_NOT_FOUND);
1473 }
jcarsey2247dde2009-11-09 18:08:58 +00001474 return (Status);
jcarsey1e6e84c2010-01-25 20:05:08 +00001475 }
jcarsey94b17fa2009-05-07 18:46:18 +00001476
1477 //
1478 // ASSERT that we must have EFI shell
1479 //
1480 ASSERT(mEfiShellEnvironment2 != NULL);
1481
1482 //
jcarsey94b17fa2009-05-07 18:46:18 +00001483 // make sure the list head is initialized
1484 //
jcarsey9b3bf082009-06-23 21:15:07 +00001485 InitializeListHead(&mOldStyleFileList);
jcarsey94b17fa2009-05-07 18:46:18 +00001486
1487 //
1488 // Get the EFI Shell list of files
1489 //
jcarsey9b3bf082009-06-23 21:15:07 +00001490 Status = mEfiShellEnvironment2->FileMetaArg(Arg, &mOldStyleFileList);
jcarsey94b17fa2009-05-07 18:46:18 +00001491 if (EFI_ERROR(Status)) {
1492 *ListHead = NULL;
1493 return (Status);
1494 }
1495
jcarsey9b3bf082009-06-23 21:15:07 +00001496 if (*ListHead == NULL) {
1497 *ListHead = (EFI_SHELL_FILE_INFO *)AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1498 if (*ListHead == NULL) {
1499 return (EFI_OUT_OF_RESOURCES);
1500 }
jcarseya405b862010-09-14 05:18:09 +00001501 InitializeListHead(&((*ListHead)->Link));
jcarsey9b3bf082009-06-23 21:15:07 +00001502 }
1503
jcarsey94b17fa2009-05-07 18:46:18 +00001504 //
1505 // Convert that to equivalent of UEFI Shell 2.0 structure
1506 //
jcarsey9b3bf082009-06-23 21:15:07 +00001507 InternalShellConvertFileListType(&mOldStyleFileList, &(*ListHead)->Link);
jcarsey94b17fa2009-05-07 18:46:18 +00001508
1509 //
jcarseyd2b45642009-05-11 18:02:16 +00001510 // Free the EFI Shell version that was converted.
1511 //
jcarsey9b3bf082009-06-23 21:15:07 +00001512 mEfiShellEnvironment2->FreeFileList(&mOldStyleFileList);
jcarsey94b17fa2009-05-07 18:46:18 +00001513
jcarseya405b862010-09-14 05:18:09 +00001514 if ((*ListHead)->Link.ForwardLink == (*ListHead)->Link.BackLink && (*ListHead)->Link.BackLink == &((*ListHead)->Link)) {
1515 FreePool(*ListHead);
1516 *ListHead = NULL;
1517 Status = EFI_NOT_FOUND;
1518 }
1519
jcarsey94b17fa2009-05-07 18:46:18 +00001520 return (Status);
1521}
1522/**
jcarseya405b862010-09-14 05:18:09 +00001523 Free the linked list returned from ShellOpenFileMetaArg.
jcarsey94b17fa2009-05-07 18:46:18 +00001524
jcarseya405b862010-09-14 05:18:09 +00001525 if ListHead is NULL then ASSERT().
jcarsey94b17fa2009-05-07 18:46:18 +00001526
jcarseya405b862010-09-14 05:18:09 +00001527 @param ListHead the pointer to free.
jcarsey94b17fa2009-05-07 18:46:18 +00001528
jcarseya405b862010-09-14 05:18:09 +00001529 @retval EFI_SUCCESS the operation was sucessful.
jcarsey94b17fa2009-05-07 18:46:18 +00001530**/
1531EFI_STATUS
1532EFIAPI
1533ShellCloseFileMetaArg (
1534 IN OUT EFI_SHELL_FILE_INFO **ListHead
1535 )
1536{
1537 LIST_ENTRY *Node;
1538
1539 //
1540 // ASSERT that ListHead is not NULL
1541 //
1542 ASSERT(ListHead != NULL);
1543
jcarsey1e6e84c2010-01-25 20:05:08 +00001544 //
jcarsey94b17fa2009-05-07 18:46:18 +00001545 // Check for UEFI Shell 2.0 protocols
1546 //
1547 if (mEfiShellProtocol != NULL) {
1548 return (mEfiShellProtocol->FreeFileList(ListHead));
1549 } else {
1550 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001551 // Since this is EFI Shell version we need to free our internally made copy
jcarsey94b17fa2009-05-07 18:46:18 +00001552 // of the list
1553 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001554 for ( Node = GetFirstNode(&(*ListHead)->Link)
jcarseya405b862010-09-14 05:18:09 +00001555 ; *ListHead != NULL && !IsListEmpty(&(*ListHead)->Link)
jcarsey9b3bf082009-06-23 21:15:07 +00001556 ; Node = GetFirstNode(&(*ListHead)->Link)) {
jcarsey94b17fa2009-05-07 18:46:18 +00001557 RemoveEntryList(Node);
jcarseya405b862010-09-14 05:18:09 +00001558 ((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 +00001559 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->FullName);
1560 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->FileName);
1561 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->Info);
1562 FreePool((EFI_SHELL_FILE_INFO_NO_CONST*)Node);
1563 }
1564 return EFI_SUCCESS;
1565 }
1566}
1567
jcarsey125c2cf2009-11-18 21:36:50 +00001568/**
1569 Find a file by searching the CWD and then the path.
1570
jcarseyb3011f42010-01-11 21:49:04 +00001571 If FileName is NULL then ASSERT.
jcarsey125c2cf2009-11-18 21:36:50 +00001572
jcarseyb3011f42010-01-11 21:49:04 +00001573 If the return value is not NULL then the memory must be caller freed.
jcarsey125c2cf2009-11-18 21:36:50 +00001574
1575 @param FileName Filename string.
1576
1577 @retval NULL the file was not found
1578 @return !NULL the full path to the file.
1579**/
1580CHAR16 *
1581EFIAPI
1582ShellFindFilePath (
1583 IN CONST CHAR16 *FileName
1584 )
1585{
1586 CONST CHAR16 *Path;
jcarseya405b862010-09-14 05:18:09 +00001587 SHELL_FILE_HANDLE Handle;
jcarsey125c2cf2009-11-18 21:36:50 +00001588 EFI_STATUS Status;
1589 CHAR16 *RetVal;
1590 CHAR16 *TestPath;
1591 CONST CHAR16 *Walker;
jcarsey36a9d672009-11-20 21:13:41 +00001592 UINTN Size;
jcarsey1cd45e72010-01-29 15:07:44 +00001593 CHAR16 *TempChar;
jcarsey125c2cf2009-11-18 21:36:50 +00001594
1595 RetVal = NULL;
1596
jcarseya405b862010-09-14 05:18:09 +00001597 //
1598 // First make sure its not an absolute path.
1599 //
1600 Status = ShellOpenFileByName(FileName, &Handle, EFI_FILE_MODE_READ, 0);
1601 if (!EFI_ERROR(Status)){
1602 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1603 ASSERT(RetVal == NULL);
1604 RetVal = StrnCatGrow(&RetVal, NULL, FileName, 0);
1605 ShellCloseFile(&Handle);
1606 return (RetVal);
1607 } else {
1608 ShellCloseFile(&Handle);
1609 }
1610 }
1611
jcarsey125c2cf2009-11-18 21:36:50 +00001612 Path = ShellGetEnvironmentVariable(L"cwd");
1613 if (Path != NULL) {
jcarsey36a9d672009-11-20 21:13:41 +00001614 Size = StrSize(Path);
1615 Size += StrSize(FileName);
1616 TestPath = AllocateZeroPool(Size);
jcarseyc9d92df2010-02-03 15:37:54 +00001617 ASSERT(TestPath != NULL);
1618 if (TestPath == NULL) {
1619 return (NULL);
1620 }
jcarsey125c2cf2009-11-18 21:36:50 +00001621 StrCpy(TestPath, Path);
1622 StrCat(TestPath, FileName);
1623 Status = ShellOpenFileByName(TestPath, &Handle, EFI_FILE_MODE_READ, 0);
1624 if (!EFI_ERROR(Status)){
jcarseya405b862010-09-14 05:18:09 +00001625 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1626 ASSERT(RetVal == NULL);
1627 RetVal = StrnCatGrow(&RetVal, NULL, TestPath, 0);
1628 ShellCloseFile(&Handle);
1629 FreePool(TestPath);
1630 return (RetVal);
1631 } else {
1632 ShellCloseFile(&Handle);
1633 }
jcarsey125c2cf2009-11-18 21:36:50 +00001634 }
1635 FreePool(TestPath);
1636 }
1637 Path = ShellGetEnvironmentVariable(L"path");
1638 if (Path != NULL) {
jcarseya405b862010-09-14 05:18:09 +00001639 Size = StrSize(Path)+sizeof(CHAR16);
jcarsey36a9d672009-11-20 21:13:41 +00001640 Size += StrSize(FileName);
1641 TestPath = AllocateZeroPool(Size);
jcarsey3e082d52010-10-04 16:44:57 +00001642 if (TestPath == NULL) {
1643 return (NULL);
1644 }
jcarsey1e6e84c2010-01-25 20:05:08 +00001645 Walker = (CHAR16*)Path;
jcarsey125c2cf2009-11-18 21:36:50 +00001646 do {
1647 CopyMem(TestPath, Walker, StrSize(Walker));
jcarsey3e082d52010-10-04 16:44:57 +00001648 if (TestPath != NULL) {
1649 TempChar = StrStr(TestPath, L";");
1650 if (TempChar != NULL) {
1651 *TempChar = CHAR_NULL;
1652 }
1653 if (TestPath[StrLen(TestPath)-1] != L'\\') {
1654 StrCat(TestPath, L"\\");
1655 }
1656 StrCat(TestPath, FileName);
1657 if (StrStr(Walker, L";") != NULL) {
1658 Walker = StrStr(Walker, L";") + 1;
jcarseya405b862010-09-14 05:18:09 +00001659 } else {
jcarsey3e082d52010-10-04 16:44:57 +00001660 Walker = NULL;
1661 }
1662 Status = ShellOpenFileByName(TestPath, &Handle, EFI_FILE_MODE_READ, 0);
1663 if (!EFI_ERROR(Status)){
1664 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1665 ASSERT(RetVal == NULL);
1666 RetVal = StrnCatGrow(&RetVal, NULL, TestPath, 0);
1667 ShellCloseFile(&Handle);
1668 break;
1669 } else {
1670 ShellCloseFile(&Handle);
1671 }
jcarseya405b862010-09-14 05:18:09 +00001672 }
jcarsey125c2cf2009-11-18 21:36:50 +00001673 }
1674 } while (Walker != NULL && Walker[0] != CHAR_NULL);
1675 FreePool(TestPath);
1676 }
1677 return (RetVal);
1678}
1679
jcarseyb3011f42010-01-11 21:49:04 +00001680/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001681 Find a file by searching the CWD and then the path with a variable set of file
1682 extensions. If the file is not found it will append each extension in the list
jcarseyb3011f42010-01-11 21:49:04 +00001683 in the order provided and return the first one that is successful.
1684
1685 If FileName is NULL, then ASSERT.
1686 If FileExtension is NULL, then behavior is identical to ShellFindFilePath.
1687
1688 If the return value is not NULL then the memory must be caller freed.
1689
1690 @param[in] FileName Filename string.
1691 @param[in] FileExtension Semi-colon delimeted list of possible extensions.
1692
1693 @retval NULL The file was not found.
1694 @retval !NULL The path to the file.
1695**/
1696CHAR16 *
1697EFIAPI
1698ShellFindFilePathEx (
1699 IN CONST CHAR16 *FileName,
1700 IN CONST CHAR16 *FileExtension
1701 )
1702{
1703 CHAR16 *TestPath;
1704 CHAR16 *RetVal;
1705 CONST CHAR16 *ExtensionWalker;
jcarsey9e926b62010-01-14 20:26:39 +00001706 UINTN Size;
jcarsey1cd45e72010-01-29 15:07:44 +00001707 CHAR16 *TempChar;
jcarseyc9d92df2010-02-03 15:37:54 +00001708 CHAR16 *TempChar2;
jcarsey1cd45e72010-01-29 15:07:44 +00001709
jcarseyb3011f42010-01-11 21:49:04 +00001710 ASSERT(FileName != NULL);
1711 if (FileExtension == NULL) {
1712 return (ShellFindFilePath(FileName));
1713 }
1714 RetVal = ShellFindFilePath(FileName);
1715 if (RetVal != NULL) {
1716 return (RetVal);
1717 }
jcarsey9e926b62010-01-14 20:26:39 +00001718 Size = StrSize(FileName);
1719 Size += StrSize(FileExtension);
1720 TestPath = AllocateZeroPool(Size);
jcarseyc9d92df2010-02-03 15:37:54 +00001721 ASSERT(TestPath != NULL);
1722 if (TestPath == NULL) {
1723 return (NULL);
1724 }
jcarseya405b862010-09-14 05:18:09 +00001725 for (ExtensionWalker = FileExtension, TempChar2 = (CHAR16*)FileExtension; TempChar2 != NULL ; ExtensionWalker = TempChar2 + 1){
jcarseyb3011f42010-01-11 21:49:04 +00001726 StrCpy(TestPath, FileName);
jcarseya405b862010-09-14 05:18:09 +00001727 if (ExtensionWalker != NULL) {
1728 StrCat(TestPath, ExtensionWalker);
1729 }
jcarsey1cd45e72010-01-29 15:07:44 +00001730 TempChar = StrStr(TestPath, L";");
1731 if (TempChar != NULL) {
1732 *TempChar = CHAR_NULL;
jcarseyb3011f42010-01-11 21:49:04 +00001733 }
1734 RetVal = ShellFindFilePath(TestPath);
1735 if (RetVal != NULL) {
1736 break;
1737 }
jcarseya405b862010-09-14 05:18:09 +00001738 ASSERT(ExtensionWalker != NULL);
jcarseyc9d92df2010-02-03 15:37:54 +00001739 TempChar2 = StrStr(ExtensionWalker, L";");
jcarseyb3011f42010-01-11 21:49:04 +00001740 }
1741 FreePool(TestPath);
1742 return (RetVal);
1743}
1744
jcarsey94b17fa2009-05-07 18:46:18 +00001745typedef struct {
jcarsey9b3bf082009-06-23 21:15:07 +00001746 LIST_ENTRY Link;
jcarsey94b17fa2009-05-07 18:46:18 +00001747 CHAR16 *Name;
jcarseya405b862010-09-14 05:18:09 +00001748 SHELL_PARAM_TYPE Type;
jcarsey94b17fa2009-05-07 18:46:18 +00001749 CHAR16 *Value;
1750 UINTN OriginalPosition;
1751} SHELL_PARAM_PACKAGE;
1752
1753/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001754 Checks the list of valid arguments and returns TRUE if the item was found. If the
jcarsey94b17fa2009-05-07 18:46:18 +00001755 return value is TRUE then the type parameter is set also.
jcarsey1e6e84c2010-01-25 20:05:08 +00001756
jcarsey94b17fa2009-05-07 18:46:18 +00001757 if CheckList is NULL then ASSERT();
1758 if Name is NULL then ASSERT();
1759 if Type is NULL then ASSERT();
1760
jcarsey94b17fa2009-05-07 18:46:18 +00001761 @param Name pointer to Name of parameter found
1762 @param CheckList List to check against
jcarseya405b862010-09-14 05:18:09 +00001763 @param Type pointer to type of parameter if it was found
jcarsey94b17fa2009-05-07 18:46:18 +00001764
1765 @retval TRUE the Parameter was found. Type is valid.
1766 @retval FALSE the Parameter was not found. Type is not valid.
1767**/
1768BOOLEAN
1769EFIAPI
jcarseyd2b45642009-05-11 18:02:16 +00001770InternalIsOnCheckList (
jcarsey94b17fa2009-05-07 18:46:18 +00001771 IN CONST CHAR16 *Name,
1772 IN CONST SHELL_PARAM_ITEM *CheckList,
jcarseya405b862010-09-14 05:18:09 +00001773 OUT SHELL_PARAM_TYPE *Type
1774 )
1775{
jcarsey94b17fa2009-05-07 18:46:18 +00001776 SHELL_PARAM_ITEM *TempListItem;
1777
1778 //
1779 // ASSERT that all 3 pointer parameters aren't NULL
1780 //
1781 ASSERT(CheckList != NULL);
1782 ASSERT(Type != NULL);
1783 ASSERT(Name != NULL);
1784
1785 //
jcarseyd2b45642009-05-11 18:02:16 +00001786 // question mark and page break mode are always supported
1787 //
1788 if ((StrCmp(Name, L"-?") == 0) ||
1789 (StrCmp(Name, L"-b") == 0)
jcarseya405b862010-09-14 05:18:09 +00001790 ) {
jcarseyd2b45642009-05-11 18:02:16 +00001791 return (TRUE);
1792 }
1793
1794 //
jcarsey94b17fa2009-05-07 18:46:18 +00001795 // Enumerate through the list
1796 //
1797 for (TempListItem = (SHELL_PARAM_ITEM*)CheckList ; TempListItem->Name != NULL ; TempListItem++) {
1798 //
jcarsey9eb53ac2009-07-08 17:26:58 +00001799 // If the Type is TypeStart only check the first characters of the passed in param
1800 // If it matches set the type and return TRUE
jcarsey94b17fa2009-05-07 18:46:18 +00001801 //
jcarsey9eb53ac2009-07-08 17:26:58 +00001802 if (TempListItem->Type == TypeStart && StrnCmp(Name, TempListItem->Name, StrLen(TempListItem->Name)) == 0) {
1803 *Type = TempListItem->Type;
1804 return (TRUE);
1805 } else if (StrCmp(Name, TempListItem->Name) == 0) {
jcarsey94b17fa2009-05-07 18:46:18 +00001806 *Type = TempListItem->Type;
1807 return (TRUE);
1808 }
1809 }
jcarsey2247dde2009-11-09 18:08:58 +00001810
jcarsey94b17fa2009-05-07 18:46:18 +00001811 return (FALSE);
1812}
1813/**
jcarseyd2b45642009-05-11 18:02:16 +00001814 Checks the string for indicators of "flag" status. this is a leading '/', '-', or '+'
jcarsey94b17fa2009-05-07 18:46:18 +00001815
jcarseya405b862010-09-14 05:18:09 +00001816 @param[in] Name pointer to Name of parameter found
1817 @param[in] AlwaysAllowNumbers TRUE to allow numbers, FALSE to not.
jcarsey94b17fa2009-05-07 18:46:18 +00001818
1819 @retval TRUE the Parameter is a flag.
jcarseya405b862010-09-14 05:18:09 +00001820 @retval FALSE the Parameter not a flag.
jcarsey94b17fa2009-05-07 18:46:18 +00001821**/
1822BOOLEAN
1823EFIAPI
jcarseyd2b45642009-05-11 18:02:16 +00001824InternalIsFlag (
jcarsey2247dde2009-11-09 18:08:58 +00001825 IN CONST CHAR16 *Name,
1826 IN BOOLEAN AlwaysAllowNumbers
jcarsey94b17fa2009-05-07 18:46:18 +00001827 )
1828{
1829 //
1830 // ASSERT that Name isn't NULL
1831 //
1832 ASSERT(Name != NULL);
1833
1834 //
jcarsey2247dde2009-11-09 18:08:58 +00001835 // If we accept numbers then dont return TRUE. (they will be values)
1836 //
jcarseya405b862010-09-14 05:18:09 +00001837 if (((Name[0] == L'-' || Name[0] == L'+') && ShellIsHexaDecimalDigitCharacter(Name[1])) && AlwaysAllowNumbers) {
jcarsey2247dde2009-11-09 18:08:58 +00001838 return (FALSE);
1839 }
1840
1841 //
jcarseya405b862010-09-14 05:18:09 +00001842 // If the Name has a /, +, or - as the first character return TRUE
jcarsey94b17fa2009-05-07 18:46:18 +00001843 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001844 if ((Name[0] == L'/') ||
jcarseyd2b45642009-05-11 18:02:16 +00001845 (Name[0] == L'-') ||
1846 (Name[0] == L'+')
jcarseya405b862010-09-14 05:18:09 +00001847 ) {
jcarsey94b17fa2009-05-07 18:46:18 +00001848 return (TRUE);
1849 }
1850 return (FALSE);
1851}
1852
1853/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001854 Checks the command line arguments passed against the list of valid ones.
jcarsey94b17fa2009-05-07 18:46:18 +00001855
1856 If no initialization is required, then return RETURN_SUCCESS.
jcarsey1e6e84c2010-01-25 20:05:08 +00001857
jcarseya405b862010-09-14 05:18:09 +00001858 @param[in] CheckList pointer to list of parameters to check
1859 @param[out] CheckPackage pointer to pointer to list checked values
1860 @param[out] ProblemParam optional pointer to pointer to unicode string for
jcarseyd2b45642009-05-11 18:02:16 +00001861 the paramater that caused failure. If used then the
1862 caller is responsible for freeing the memory.
jcarseya405b862010-09-14 05:18:09 +00001863 @param[in] AutoPageBreak will automatically set PageBreakEnabled for "b" parameter
1864 @param[in] Argv pointer to array of parameters
1865 @param[in] Argc Count of parameters in Argv
1866 @param[in] AlwaysAllowNumbers TRUE to allow numbers always, FALSE otherwise.
jcarsey94b17fa2009-05-07 18:46:18 +00001867
1868 @retval EFI_SUCCESS The operation completed sucessfully.
1869 @retval EFI_OUT_OF_RESOURCES A memory allocation failed
1870 @retval EFI_INVALID_PARAMETER A parameter was invalid
jcarsey1e6e84c2010-01-25 20:05:08 +00001871 @retval EFI_VOLUME_CORRUPTED the command line was corrupt. an argument was
1872 duplicated. the duplicated command line argument
jcarsey94b17fa2009-05-07 18:46:18 +00001873 was returned in ProblemParam if provided.
jcarsey1e6e84c2010-01-25 20:05:08 +00001874 @retval EFI_NOT_FOUND a argument required a value that was missing.
jcarsey94b17fa2009-05-07 18:46:18 +00001875 the invalid command line argument was returned in
1876 ProblemParam if provided.
1877**/
1878EFI_STATUS
1879EFIAPI
1880InternalCommandLineParse (
1881 IN CONST SHELL_PARAM_ITEM *CheckList,
1882 OUT LIST_ENTRY **CheckPackage,
1883 OUT CHAR16 **ProblemParam OPTIONAL,
1884 IN BOOLEAN AutoPageBreak,
1885 IN CONST CHAR16 **Argv,
jcarsey2247dde2009-11-09 18:08:58 +00001886 IN UINTN Argc,
1887 IN BOOLEAN AlwaysAllowNumbers
jcarseya405b862010-09-14 05:18:09 +00001888 )
1889{
jcarsey94b17fa2009-05-07 18:46:18 +00001890 UINTN LoopCounter;
jcarseya405b862010-09-14 05:18:09 +00001891 SHELL_PARAM_TYPE CurrentItemType;
jcarsey94b17fa2009-05-07 18:46:18 +00001892 SHELL_PARAM_PACKAGE *CurrentItemPackage;
jcarsey125c2cf2009-11-18 21:36:50 +00001893 UINTN GetItemValue;
1894 UINTN ValueSize;
jcarseya405b862010-09-14 05:18:09 +00001895 UINTN Count;
jcarsey94b17fa2009-05-07 18:46:18 +00001896
1897 CurrentItemPackage = NULL;
jcarsey125c2cf2009-11-18 21:36:50 +00001898 GetItemValue = 0;
1899 ValueSize = 0;
jcarseya405b862010-09-14 05:18:09 +00001900 Count = 0;
jcarsey94b17fa2009-05-07 18:46:18 +00001901
1902 //
1903 // If there is only 1 item we dont need to do anything
1904 //
jcarseya405b862010-09-14 05:18:09 +00001905 if (Argc < 1) {
jcarsey94b17fa2009-05-07 18:46:18 +00001906 *CheckPackage = NULL;
1907 return (EFI_SUCCESS);
1908 }
1909
1910 //
jcarsey2247dde2009-11-09 18:08:58 +00001911 // ASSERTs
1912 //
1913 ASSERT(CheckList != NULL);
1914 ASSERT(Argv != NULL);
1915
1916 //
jcarsey94b17fa2009-05-07 18:46:18 +00001917 // initialize the linked list
1918 //
1919 *CheckPackage = (LIST_ENTRY*)AllocateZeroPool(sizeof(LIST_ENTRY));
1920 InitializeListHead(*CheckPackage);
1921
1922 //
1923 // loop through each of the arguments
1924 //
1925 for (LoopCounter = 0 ; LoopCounter < Argc ; ++LoopCounter) {
1926 if (Argv[LoopCounter] == NULL) {
1927 //
1928 // do nothing for NULL argv
1929 //
jcarseya405b862010-09-14 05:18:09 +00001930 } else if (InternalIsOnCheckList(Argv[LoopCounter], CheckList, &CurrentItemType)) {
jcarsey94b17fa2009-05-07 18:46:18 +00001931 //
jcarsey2247dde2009-11-09 18:08:58 +00001932 // We might have leftover if last parameter didnt have optional value
1933 //
jcarsey125c2cf2009-11-18 21:36:50 +00001934 if (GetItemValue != 0) {
1935 GetItemValue = 0;
jcarsey2247dde2009-11-09 18:08:58 +00001936 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
1937 }
1938 //
jcarsey94b17fa2009-05-07 18:46:18 +00001939 // this is a flag
1940 //
1941 CurrentItemPackage = AllocatePool(sizeof(SHELL_PARAM_PACKAGE));
1942 ASSERT(CurrentItemPackage != NULL);
1943 CurrentItemPackage->Name = AllocatePool(StrSize(Argv[LoopCounter]));
1944 ASSERT(CurrentItemPackage->Name != NULL);
1945 StrCpy(CurrentItemPackage->Name, Argv[LoopCounter]);
1946 CurrentItemPackage->Type = CurrentItemType;
1947 CurrentItemPackage->OriginalPosition = (UINTN)(-1);
jcarseyb1f95a02009-06-16 00:23:19 +00001948 CurrentItemPackage->Value = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +00001949
1950 //
1951 // Does this flag require a value
1952 //
jcarsey125c2cf2009-11-18 21:36:50 +00001953 switch (CurrentItemPackage->Type) {
jcarsey94b17fa2009-05-07 18:46:18 +00001954 //
jcarsey125c2cf2009-11-18 21:36:50 +00001955 // possibly trigger the next loop(s) to populate the value of this item
jcarsey1e6e84c2010-01-25 20:05:08 +00001956 //
jcarsey125c2cf2009-11-18 21:36:50 +00001957 case TypeValue:
jcarsey1e6e84c2010-01-25 20:05:08 +00001958 GetItemValue = 1;
jcarsey125c2cf2009-11-18 21:36:50 +00001959 ValueSize = 0;
1960 break;
1961 case TypeDoubleValue:
1962 GetItemValue = 2;
1963 ValueSize = 0;
1964 break;
1965 case TypeMaxValue:
1966 GetItemValue = (UINTN)(-1);
1967 ValueSize = 0;
1968 break;
1969 default:
1970 //
1971 // this item has no value expected; we are done
1972 //
1973 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
1974 ASSERT(GetItemValue == 0);
1975 break;
jcarsey94b17fa2009-05-07 18:46:18 +00001976 }
jcarseya405b862010-09-14 05:18:09 +00001977 } else if (GetItemValue != 0 && !InternalIsFlag(Argv[LoopCounter], AlwaysAllowNumbers)) {
jcarseyb1f95a02009-06-16 00:23:19 +00001978 ASSERT(CurrentItemPackage != NULL);
1979 //
jcarsey125c2cf2009-11-18 21:36:50 +00001980 // get the item VALUE for a previous flag
jcarseyb1f95a02009-06-16 00:23:19 +00001981 //
jcarsey125c2cf2009-11-18 21:36:50 +00001982 CurrentItemPackage->Value = ReallocatePool(ValueSize, ValueSize + StrSize(Argv[LoopCounter]) + sizeof(CHAR16), CurrentItemPackage->Value);
jcarseyb1f95a02009-06-16 00:23:19 +00001983 ASSERT(CurrentItemPackage->Value != NULL);
jcarsey125c2cf2009-11-18 21:36:50 +00001984 if (ValueSize == 0) {
1985 StrCpy(CurrentItemPackage->Value, Argv[LoopCounter]);
1986 } else {
1987 StrCat(CurrentItemPackage->Value, L" ");
1988 StrCat(CurrentItemPackage->Value, Argv[LoopCounter]);
1989 }
1990 ValueSize += StrSize(Argv[LoopCounter]) + sizeof(CHAR16);
1991 GetItemValue--;
1992 if (GetItemValue == 0) {
1993 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
1994 }
jcarseya405b862010-09-14 05:18:09 +00001995 } else if (!InternalIsFlag(Argv[LoopCounter], AlwaysAllowNumbers) ){ //|| ProblemParam == NULL) {
jcarseyb1f95a02009-06-16 00:23:19 +00001996 //
1997 // add this one as a non-flag
1998 //
1999 CurrentItemPackage = AllocatePool(sizeof(SHELL_PARAM_PACKAGE));
2000 ASSERT(CurrentItemPackage != NULL);
2001 CurrentItemPackage->Name = NULL;
2002 CurrentItemPackage->Type = TypePosition;
2003 CurrentItemPackage->Value = AllocatePool(StrSize(Argv[LoopCounter]));
2004 ASSERT(CurrentItemPackage->Value != NULL);
2005 StrCpy(CurrentItemPackage->Value, Argv[LoopCounter]);
jcarseya405b862010-09-14 05:18:09 +00002006 CurrentItemPackage->OriginalPosition = Count++;
jcarsey9b3bf082009-06-23 21:15:07 +00002007 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
jcarseya405b862010-09-14 05:18:09 +00002008 } else if (ProblemParam != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00002009 //
2010 // this was a non-recognised flag... error!
2011 //
jcarseyd2b45642009-05-11 18:02:16 +00002012 *ProblemParam = AllocatePool(StrSize(Argv[LoopCounter]));
2013 ASSERT(*ProblemParam != NULL);
2014 StrCpy(*ProblemParam, Argv[LoopCounter]);
jcarsey94b17fa2009-05-07 18:46:18 +00002015 ShellCommandLineFreeVarList(*CheckPackage);
2016 *CheckPackage = NULL;
2017 return (EFI_VOLUME_CORRUPTED);
2018 } else {
jcarseya405b862010-09-14 05:18:09 +00002019 //ASSERT(FALSE);
jcarsey94b17fa2009-05-07 18:46:18 +00002020 }
2021 }
jcarsey125c2cf2009-11-18 21:36:50 +00002022 if (GetItemValue != 0) {
2023 GetItemValue = 0;
2024 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2025 }
jcarsey94b17fa2009-05-07 18:46:18 +00002026 //
2027 // support for AutoPageBreak
2028 //
2029 if (AutoPageBreak && ShellCommandLineGetFlag(*CheckPackage, L"-b")) {
2030 ShellSetPageBreakMode(TRUE);
2031 }
2032 return (EFI_SUCCESS);
2033}
2034
2035/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002036 Checks the command line arguments passed against the list of valid ones.
jcarsey94b17fa2009-05-07 18:46:18 +00002037 Optionally removes NULL values first.
jcarsey1e6e84c2010-01-25 20:05:08 +00002038
jcarsey94b17fa2009-05-07 18:46:18 +00002039 If no initialization is required, then return RETURN_SUCCESS.
jcarsey1e6e84c2010-01-25 20:05:08 +00002040
jcarseya405b862010-09-14 05:18:09 +00002041 @param[in] CheckList The pointer to list of parameters to check.
2042 @param[out] CheckPackage The package of checked values.
2043 @param[out] ProblemParam Optional pointer to pointer to unicode string for
jcarsey94b17fa2009-05-07 18:46:18 +00002044 the paramater that caused failure.
jcarseya405b862010-09-14 05:18:09 +00002045 @param[in] AutoPageBreak Will automatically set PageBreakEnabled.
2046 @param[in] AlwaysAllowNumbers Will never fail for number based flags.
jcarsey94b17fa2009-05-07 18:46:18 +00002047
2048 @retval EFI_SUCCESS The operation completed sucessfully.
jcarseya405b862010-09-14 05:18:09 +00002049 @retval EFI_OUT_OF_RESOURCES A memory allocation failed.
2050 @retval EFI_INVALID_PARAMETER A parameter was invalid.
2051 @retval EFI_VOLUME_CORRUPTED The command line was corrupt.
2052 @retval EFI_DEVICE_ERROR The commands contained 2 opposing arguments. One
jcarsey1e6e84c2010-01-25 20:05:08 +00002053 of the command line arguments was returned in
jcarsey94b17fa2009-05-07 18:46:18 +00002054 ProblemParam if provided.
jcarseya405b862010-09-14 05:18:09 +00002055 @retval EFI_NOT_FOUND A argument required a value that was missing.
2056 The invalid command line argument was returned in
jcarsey94b17fa2009-05-07 18:46:18 +00002057 ProblemParam if provided.
2058**/
2059EFI_STATUS
2060EFIAPI
jcarsey2247dde2009-11-09 18:08:58 +00002061ShellCommandLineParseEx (
jcarsey94b17fa2009-05-07 18:46:18 +00002062 IN CONST SHELL_PARAM_ITEM *CheckList,
2063 OUT LIST_ENTRY **CheckPackage,
2064 OUT CHAR16 **ProblemParam OPTIONAL,
jcarsey2247dde2009-11-09 18:08:58 +00002065 IN BOOLEAN AutoPageBreak,
2066 IN BOOLEAN AlwaysAllowNumbers
jcarseya405b862010-09-14 05:18:09 +00002067 )
2068{
jcarsey1e6e84c2010-01-25 20:05:08 +00002069 //
jcarsey94b17fa2009-05-07 18:46:18 +00002070 // ASSERT that CheckList and CheckPackage aren't NULL
2071 //
2072 ASSERT(CheckList != NULL);
2073 ASSERT(CheckPackage != NULL);
2074
jcarsey1e6e84c2010-01-25 20:05:08 +00002075 //
jcarsey94b17fa2009-05-07 18:46:18 +00002076 // Check for UEFI Shell 2.0 protocols
2077 //
2078 if (mEfiShellParametersProtocol != NULL) {
jcarsey1e6e84c2010-01-25 20:05:08 +00002079 return (InternalCommandLineParse(CheckList,
2080 CheckPackage,
2081 ProblemParam,
2082 AutoPageBreak,
jljusten08d7f8e2009-06-15 18:42:13 +00002083 (CONST CHAR16**) mEfiShellParametersProtocol->Argv,
jcarsey2247dde2009-11-09 18:08:58 +00002084 mEfiShellParametersProtocol->Argc,
2085 AlwaysAllowNumbers));
jcarsey94b17fa2009-05-07 18:46:18 +00002086 }
2087
jcarsey1e6e84c2010-01-25 20:05:08 +00002088 //
jcarsey94b17fa2009-05-07 18:46:18 +00002089 // ASSERT That EFI Shell is not required
2090 //
2091 ASSERT (mEfiShellInterface != NULL);
jcarsey1e6e84c2010-01-25 20:05:08 +00002092 return (InternalCommandLineParse(CheckList,
2093 CheckPackage,
2094 ProblemParam,
2095 AutoPageBreak,
jljusten08d7f8e2009-06-15 18:42:13 +00002096 (CONST CHAR16**) mEfiShellInterface->Argv,
jcarsey2247dde2009-11-09 18:08:58 +00002097 mEfiShellInterface->Argc,
2098 AlwaysAllowNumbers));
jcarsey94b17fa2009-05-07 18:46:18 +00002099}
2100
2101/**
2102 Frees shell variable list that was returned from ShellCommandLineParse.
2103
2104 This function will free all the memory that was used for the CheckPackage
2105 list of postprocessed shell arguments.
2106
2107 this function has no return value.
2108
2109 if CheckPackage is NULL, then return
2110
2111 @param CheckPackage the list to de-allocate
2112 **/
2113VOID
2114EFIAPI
2115ShellCommandLineFreeVarList (
2116 IN LIST_ENTRY *CheckPackage
jcarseya405b862010-09-14 05:18:09 +00002117 )
2118{
jcarsey94b17fa2009-05-07 18:46:18 +00002119 LIST_ENTRY *Node;
2120
2121 //
2122 // check for CheckPackage == NULL
2123 //
2124 if (CheckPackage == NULL) {
2125 return;
2126 }
2127
2128 //
2129 // for each node in the list
2130 //
jcarsey9eb53ac2009-07-08 17:26:58 +00002131 for ( Node = GetFirstNode(CheckPackage)
jcarseya405b862010-09-14 05:18:09 +00002132 ; !IsListEmpty(CheckPackage)
jcarsey9eb53ac2009-07-08 17:26:58 +00002133 ; Node = GetFirstNode(CheckPackage)
jcarseya405b862010-09-14 05:18:09 +00002134 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002135 //
2136 // Remove it from the list
2137 //
2138 RemoveEntryList(Node);
2139
2140 //
2141 // if it has a name free the name
2142 //
2143 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
2144 FreePool(((SHELL_PARAM_PACKAGE*)Node)->Name);
2145 }
2146
2147 //
2148 // if it has a value free the value
2149 //
2150 if (((SHELL_PARAM_PACKAGE*)Node)->Value != NULL) {
2151 FreePool(((SHELL_PARAM_PACKAGE*)Node)->Value);
2152 }
jcarsey1e6e84c2010-01-25 20:05:08 +00002153
jcarsey94b17fa2009-05-07 18:46:18 +00002154 //
2155 // free the node structure
2156 //
2157 FreePool((SHELL_PARAM_PACKAGE*)Node);
2158 }
2159 //
2160 // free the list head node
2161 //
2162 FreePool(CheckPackage);
2163}
2164/**
2165 Checks for presence of a flag parameter
2166
2167 flag arguments are in the form of "-<Key>" or "/<Key>", but do not have a value following the key
2168
2169 if CheckPackage is NULL then return FALSE.
2170 if KeyString is NULL then ASSERT()
jcarsey1e6e84c2010-01-25 20:05:08 +00002171
jcarsey94b17fa2009-05-07 18:46:18 +00002172 @param CheckPackage The package of parsed command line arguments
2173 @param KeyString the Key of the command line argument to check for
2174
2175 @retval TRUE the flag is on the command line
2176 @retval FALSE the flag is not on the command line
2177 **/
2178BOOLEAN
2179EFIAPI
2180ShellCommandLineGetFlag (
jcarseya405b862010-09-14 05:18:09 +00002181 IN CONST LIST_ENTRY * CONST CheckPackage,
2182 IN CONST CHAR16 * CONST KeyString
2183 )
2184{
jcarsey94b17fa2009-05-07 18:46:18 +00002185 LIST_ENTRY *Node;
2186
2187 //
2188 // ASSERT that both CheckPackage and KeyString aren't NULL
2189 //
2190 ASSERT(KeyString != NULL);
2191
2192 //
2193 // return FALSE for no package
2194 //
2195 if (CheckPackage == NULL) {
2196 return (FALSE);
2197 }
2198
2199 //
2200 // enumerate through the list of parametrs
2201 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002202 for ( Node = GetFirstNode(CheckPackage)
2203 ; !IsNull (CheckPackage, Node)
2204 ; Node = GetNextNode(CheckPackage, Node)
jcarseya405b862010-09-14 05:18:09 +00002205 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002206 //
2207 // If the Name matches, return TRUE (and there may be NULL name)
2208 //
2209 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
jcarsey9eb53ac2009-07-08 17:26:58 +00002210 //
2211 // If Type is TypeStart then only compare the begining of the strings
2212 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002213 if ( ((SHELL_PARAM_PACKAGE*)Node)->Type == TypeStart
jcarsey9eb53ac2009-07-08 17:26:58 +00002214 && StrnCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name, StrLen(KeyString)) == 0
jcarseya405b862010-09-14 05:18:09 +00002215 ){
jcarsey9eb53ac2009-07-08 17:26:58 +00002216 return (TRUE);
2217 } else if (StrCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
jcarsey94b17fa2009-05-07 18:46:18 +00002218 return (TRUE);
2219 }
2220 }
2221 }
2222 return (FALSE);
2223}
2224/**
jcarseya405b862010-09-14 05:18:09 +00002225 Returns value from command line argument.
jcarsey94b17fa2009-05-07 18:46:18 +00002226
jcarseya405b862010-09-14 05:18:09 +00002227 Value parameters are in the form of "-<Key> value" or "/<Key> value".
jcarsey1e6e84c2010-01-25 20:05:08 +00002228
jcarseya405b862010-09-14 05:18:09 +00002229 If CheckPackage is NULL, then return NULL.
jcarsey94b17fa2009-05-07 18:46:18 +00002230
jcarseya405b862010-09-14 05:18:09 +00002231 @param[in] CheckPackage The package of parsed command line arguments.
2232 @param[in] KeyString The Key of the command line argument to check for.
jcarsey94b17fa2009-05-07 18:46:18 +00002233
jcarseya405b862010-09-14 05:18:09 +00002234 @retval NULL The flag is not on the command line.
2235 @retval !=NULL The pointer to unicode string of the value.
2236**/
jcarsey94b17fa2009-05-07 18:46:18 +00002237CONST CHAR16*
2238EFIAPI
2239ShellCommandLineGetValue (
2240 IN CONST LIST_ENTRY *CheckPackage,
2241 IN CHAR16 *KeyString
jcarseya405b862010-09-14 05:18:09 +00002242 )
2243{
jcarsey94b17fa2009-05-07 18:46:18 +00002244 LIST_ENTRY *Node;
2245
2246 //
2247 // check for CheckPackage == NULL
2248 //
2249 if (CheckPackage == NULL) {
2250 return (NULL);
2251 }
2252
2253 //
2254 // enumerate through the list of parametrs
2255 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002256 for ( Node = GetFirstNode(CheckPackage)
2257 ; !IsNull (CheckPackage, Node)
2258 ; Node = GetNextNode(CheckPackage, Node)
jcarseya405b862010-09-14 05:18:09 +00002259 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002260 //
2261 // If the Name matches, return the value (name can be NULL)
2262 //
2263 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
jcarsey9eb53ac2009-07-08 17:26:58 +00002264 //
2265 // If Type is TypeStart then only compare the begining of the strings
2266 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002267 if ( ((SHELL_PARAM_PACKAGE*)Node)->Type == TypeStart
jcarsey9eb53ac2009-07-08 17:26:58 +00002268 && StrnCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name, StrLen(KeyString)) == 0
jcarseya405b862010-09-14 05:18:09 +00002269 ){
jcarsey9eb53ac2009-07-08 17:26:58 +00002270 //
2271 // return the string part after the flag
2272 //
2273 return (((SHELL_PARAM_PACKAGE*)Node)->Name + StrLen(KeyString));
2274 } else if (StrCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2275 //
2276 // return the value
2277 //
jcarsey94b17fa2009-05-07 18:46:18 +00002278 return (((SHELL_PARAM_PACKAGE*)Node)->Value);
2279 }
2280 }
2281 }
2282 return (NULL);
2283}
jcarseya405b862010-09-14 05:18:09 +00002284
jcarsey94b17fa2009-05-07 18:46:18 +00002285/**
jcarseya405b862010-09-14 05:18:09 +00002286 Returns raw value from command line argument.
jcarsey94b17fa2009-05-07 18:46:18 +00002287
jcarseya405b862010-09-14 05:18:09 +00002288 Raw value parameters are in the form of "value" in a specific position in the list.
jcarsey1e6e84c2010-01-25 20:05:08 +00002289
jcarseya405b862010-09-14 05:18:09 +00002290 If CheckPackage is NULL, then return NULL.
jcarsey94b17fa2009-05-07 18:46:18 +00002291
jcarseya405b862010-09-14 05:18:09 +00002292 @param[in] CheckPackage The package of parsed command line arguments.
2293 @param[in] Position The position of the value.
jcarsey94b17fa2009-05-07 18:46:18 +00002294
jcarseya405b862010-09-14 05:18:09 +00002295 @retval NULL The flag is not on the command line.
2296 @retval !=NULL The pointer to unicode string of the value.
jcarsey94b17fa2009-05-07 18:46:18 +00002297 **/
2298CONST CHAR16*
2299EFIAPI
2300ShellCommandLineGetRawValue (
jcarseya405b862010-09-14 05:18:09 +00002301 IN CONST LIST_ENTRY * CONST CheckPackage,
2302 IN UINTN Position
2303 )
2304{
jcarsey94b17fa2009-05-07 18:46:18 +00002305 LIST_ENTRY *Node;
2306
2307 //
2308 // check for CheckPackage == NULL
2309 //
2310 if (CheckPackage == NULL) {
2311 return (NULL);
2312 }
2313
2314 //
2315 // enumerate through the list of parametrs
2316 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002317 for ( Node = GetFirstNode(CheckPackage)
2318 ; !IsNull (CheckPackage, Node)
2319 ; Node = GetNextNode(CheckPackage, Node)
jcarseya405b862010-09-14 05:18:09 +00002320 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002321 //
2322 // If the position matches, return the value
2323 //
2324 if (((SHELL_PARAM_PACKAGE*)Node)->OriginalPosition == Position) {
2325 return (((SHELL_PARAM_PACKAGE*)Node)->Value);
2326 }
2327 }
2328 return (NULL);
jcarseyb1f95a02009-06-16 00:23:19 +00002329}
jcarsey2247dde2009-11-09 18:08:58 +00002330
2331/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002332 returns the number of command line value parameters that were parsed.
2333
jcarsey2247dde2009-11-09 18:08:58 +00002334 this will not include flags.
2335
jcarseya405b862010-09-14 05:18:09 +00002336 @param[in] CheckPackage The package of parsed command line arguments.
2337
jcarsey2247dde2009-11-09 18:08:58 +00002338 @retval (UINTN)-1 No parsing has ocurred
2339 @return other The number of value parameters found
2340**/
2341UINTN
2342EFIAPI
2343ShellCommandLineGetCount(
jcarseya405b862010-09-14 05:18:09 +00002344 IN CONST LIST_ENTRY *CheckPackage
jcarsey125c2cf2009-11-18 21:36:50 +00002345 )
2346{
jcarseya405b862010-09-14 05:18:09 +00002347 LIST_ENTRY *Node1;
2348 UINTN Count;
2349
2350 if (CheckPackage == NULL) {
2351 return (0);
2352 }
2353 for ( Node1 = GetFirstNode(CheckPackage), Count = 0
2354 ; !IsNull (CheckPackage, Node1)
2355 ; Node1 = GetNextNode(CheckPackage, Node1)
2356 ){
2357 if (((SHELL_PARAM_PACKAGE*)Node1)->Name == NULL) {
2358 Count++;
2359 }
2360 }
2361 return (Count);
jcarsey2247dde2009-11-09 18:08:58 +00002362}
2363
jcarsey975136a2009-06-16 19:03:54 +00002364/**
jcarsey36a9d672009-11-20 21:13:41 +00002365 Determins if a parameter is duplicated.
2366
jcarsey1e6e84c2010-01-25 20:05:08 +00002367 If Param is not NULL then it will point to a callee allocated string buffer
jcarsey36a9d672009-11-20 21:13:41 +00002368 with the parameter value if a duplicate is found.
2369
2370 If CheckPackage is NULL, then ASSERT.
2371
2372 @param[in] CheckPackage The package of parsed command line arguments.
2373 @param[out] Param Upon finding one, a pointer to the duplicated parameter.
2374
2375 @retval EFI_SUCCESS No parameters were duplicated.
2376 @retval EFI_DEVICE_ERROR A duplicate was found.
2377 **/
2378EFI_STATUS
2379EFIAPI
2380ShellCommandLineCheckDuplicate (
2381 IN CONST LIST_ENTRY *CheckPackage,
2382 OUT CHAR16 **Param
2383 )
2384{
2385 LIST_ENTRY *Node1;
2386 LIST_ENTRY *Node2;
jcarsey1e6e84c2010-01-25 20:05:08 +00002387
jcarsey36a9d672009-11-20 21:13:41 +00002388 ASSERT(CheckPackage != NULL);
2389
jcarsey1e6e84c2010-01-25 20:05:08 +00002390 for ( Node1 = GetFirstNode(CheckPackage)
2391 ; !IsNull (CheckPackage, Node1)
2392 ; Node1 = GetNextNode(CheckPackage, Node1)
jcarseya405b862010-09-14 05:18:09 +00002393 ){
jcarsey1e6e84c2010-01-25 20:05:08 +00002394 for ( Node2 = GetNextNode(CheckPackage, Node1)
2395 ; !IsNull (CheckPackage, Node2)
2396 ; Node2 = GetNextNode(CheckPackage, Node2)
jcarseya405b862010-09-14 05:18:09 +00002397 ){
2398 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 +00002399 if (Param != NULL) {
2400 *Param = NULL;
2401 *Param = StrnCatGrow(Param, NULL, ((SHELL_PARAM_PACKAGE*)Node1)->Name, 0);
2402 }
2403 return (EFI_DEVICE_ERROR);
2404 }
2405 }
2406 }
2407 return (EFI_SUCCESS);
2408}
2409
2410/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002411 This is a find and replace function. Upon successful return the NewString is a copy of
jcarsey975136a2009-06-16 19:03:54 +00002412 SourceString with each instance of FindTarget replaced with ReplaceWith.
2413
jcarseyb3011f42010-01-11 21:49:04 +00002414 If SourceString and NewString overlap the behavior is undefined.
2415
jcarsey975136a2009-06-16 19:03:54 +00002416 If the string would grow bigger than NewSize it will halt and return error.
2417
jcarseya405b862010-09-14 05:18:09 +00002418 @param[in] SourceString The string with source buffer.
2419 @param[in,out] NewString The string with resultant buffer.
2420 @param[in] NewSize The size in bytes of NewString.
2421 @param[in] FindTarget The string to look for.
2422 @param[in] ReplaceWith The string to replace FindTarget with.
jcarsey969c7832010-01-13 16:46:33 +00002423 @param[in] SkipPreCarrot If TRUE will skip a FindTarget that has a '^'
2424 immediately before it.
jcarseya405b862010-09-14 05:18:09 +00002425 @param[in] ParameterReplacing If TRUE will add "" around items with spaces.
jcarsey975136a2009-06-16 19:03:54 +00002426
jcarsey969c7832010-01-13 16:46:33 +00002427 @retval EFI_INVALID_PARAMETER SourceString was NULL.
2428 @retval EFI_INVALID_PARAMETER NewString was NULL.
2429 @retval EFI_INVALID_PARAMETER FindTarget was NULL.
2430 @retval EFI_INVALID_PARAMETER ReplaceWith was NULL.
2431 @retval EFI_INVALID_PARAMETER FindTarget had length < 1.
2432 @retval EFI_INVALID_PARAMETER SourceString had length < 1.
jcarsey1e6e84c2010-01-25 20:05:08 +00002433 @retval EFI_BUFFER_TOO_SMALL NewSize was less than the minimum size to hold
jcarsey969c7832010-01-13 16:46:33 +00002434 the new string (truncation occurred).
jcarseya405b862010-09-14 05:18:09 +00002435 @retval EFI_SUCCESS The string was successfully copied with replacement.
jcarsey975136a2009-06-16 19:03:54 +00002436**/
jcarsey975136a2009-06-16 19:03:54 +00002437EFI_STATUS
2438EFIAPI
jcarseya405b862010-09-14 05:18:09 +00002439ShellCopySearchAndReplace(
jcarsey975136a2009-06-16 19:03:54 +00002440 IN CHAR16 CONST *SourceString,
jcarseya405b862010-09-14 05:18:09 +00002441 IN OUT CHAR16 *NewString,
jcarsey975136a2009-06-16 19:03:54 +00002442 IN UINTN NewSize,
2443 IN CONST CHAR16 *FindTarget,
jcarsey969c7832010-01-13 16:46:33 +00002444 IN CONST CHAR16 *ReplaceWith,
jcarseya405b862010-09-14 05:18:09 +00002445 IN CONST BOOLEAN SkipPreCarrot,
2446 IN CONST BOOLEAN ParameterReplacing
jcarsey1e6e84c2010-01-25 20:05:08 +00002447 )
jcarsey2247dde2009-11-09 18:08:58 +00002448{
jcarsey01582942009-07-10 19:46:17 +00002449 UINTN Size;
jcarseya405b862010-09-14 05:18:09 +00002450 CHAR16 *Replace;
2451
jcarsey975136a2009-06-16 19:03:54 +00002452 if ( (SourceString == NULL)
2453 || (NewString == NULL)
2454 || (FindTarget == NULL)
2455 || (ReplaceWith == NULL)
2456 || (StrLen(FindTarget) < 1)
2457 || (StrLen(SourceString) < 1)
jcarseya405b862010-09-14 05:18:09 +00002458 ){
jcarsey975136a2009-06-16 19:03:54 +00002459 return (EFI_INVALID_PARAMETER);
2460 }
jcarseya405b862010-09-14 05:18:09 +00002461 Replace = NULL;
2462 if (StrStr(ReplaceWith, L" ") == NULL || !ParameterReplacing) {
2463 Replace = StrnCatGrow(&Replace, NULL, ReplaceWith, 0);
2464 } else {
2465 Replace = AllocateZeroPool(StrSize(ReplaceWith) + 2*sizeof(CHAR16));
2466 UnicodeSPrint(Replace, StrSize(ReplaceWith) + 2*sizeof(CHAR16), L"\"%s\"", ReplaceWith);
2467 }
jcarsey3e082d52010-10-04 16:44:57 +00002468 if (Replace == NULL) {
2469 return (EFI_OUT_OF_RESOURCES);
2470 }
jcarsey2247dde2009-11-09 18:08:58 +00002471 NewString = SetMem16(NewString, NewSize, CHAR_NULL);
2472 while (*SourceString != CHAR_NULL) {
jcarsey969c7832010-01-13 16:46:33 +00002473 //
jcarseya405b862010-09-14 05:18:09 +00002474 // if we find the FindTarget and either Skip == FALSE or Skip and we
jcarsey969c7832010-01-13 16:46:33 +00002475 // dont have a carrot do a replace...
2476 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002477 if (StrnCmp(SourceString, FindTarget, StrLen(FindTarget)) == 0
jcarseya405b862010-09-14 05:18:09 +00002478 && ((SkipPreCarrot && *(SourceString-1) != L'^') || !SkipPreCarrot)
2479 ){
jcarsey975136a2009-06-16 19:03:54 +00002480 SourceString += StrLen(FindTarget);
jcarsey01582942009-07-10 19:46:17 +00002481 Size = StrSize(NewString);
jcarseya405b862010-09-14 05:18:09 +00002482 if ((Size + (StrLen(Replace)*sizeof(CHAR16))) > NewSize) {
2483 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002484 return (EFI_BUFFER_TOO_SMALL);
2485 }
jcarseya405b862010-09-14 05:18:09 +00002486 StrCat(NewString, Replace);
jcarsey975136a2009-06-16 19:03:54 +00002487 } else {
jcarsey01582942009-07-10 19:46:17 +00002488 Size = StrSize(NewString);
2489 if (Size + sizeof(CHAR16) > NewSize) {
jcarseya405b862010-09-14 05:18:09 +00002490 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002491 return (EFI_BUFFER_TOO_SMALL);
2492 }
2493 StrnCat(NewString, SourceString, 1);
2494 SourceString++;
2495 }
2496 }
jcarseya405b862010-09-14 05:18:09 +00002497 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002498 return (EFI_SUCCESS);
2499}
jcarseyb1f95a02009-06-16 00:23:19 +00002500
2501/**
jcarseye2f82972009-12-01 05:40:24 +00002502 Internal worker function to output a string.
2503
2504 This function will output a string to the correct StdOut.
2505
2506 @param[in] String The string to print out.
2507
2508 @retval EFI_SUCCESS The operation was sucessful.
2509 @retval !EFI_SUCCESS The operation failed.
2510**/
2511EFI_STATUS
2512EFIAPI
2513InternalPrintTo (
2514 IN CONST CHAR16 *String
2515 )
2516{
2517 UINTN Size;
2518 Size = StrSize(String) - sizeof(CHAR16);
jcarseya405b862010-09-14 05:18:09 +00002519 if (Size == 0) {
2520 return (EFI_SUCCESS);
2521 }
jcarseye2f82972009-12-01 05:40:24 +00002522 if (mEfiShellParametersProtocol != NULL) {
jcarseya405b862010-09-14 05:18:09 +00002523 return (mEfiShellProtocol->WriteFile(mEfiShellParametersProtocol->StdOut, &Size, (VOID*)String));
jcarseye2f82972009-12-01 05:40:24 +00002524 }
2525 if (mEfiShellInterface != NULL) {
jcarseyecd3d592009-12-07 18:05:00 +00002526 //
2527 // Divide in half for old shell. Must be string length not size.
2528 //
2529 Size /= 2;
jcarseya405b862010-09-14 05:18:09 +00002530 return (mEfiShellInterface->StdOut->Write(mEfiShellInterface->StdOut, &Size, (VOID*)String));
jcarseye2f82972009-12-01 05:40:24 +00002531 }
2532 ASSERT(FALSE);
2533 return (EFI_UNSUPPORTED);
2534}
2535
2536/**
jcarseyb1f95a02009-06-16 00:23:19 +00002537 Print at a specific location on the screen.
2538
jcarseyf1b87e72009-06-17 00:52:11 +00002539 This function will move the cursor to a given screen location and print the specified string
jcarsey1e6e84c2010-01-25 20:05:08 +00002540
2541 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarseyf1b87e72009-06-17 00:52:11 +00002542 will be used.
jcarseyb1f95a02009-06-16 00:23:19 +00002543
2544 if either Row or Col is out of range for the current console, then ASSERT
2545 if Format is NULL, then ASSERT
2546
jcarsey1e6e84c2010-01-25 20:05:08 +00002547 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarseyb1f95a02009-06-16 00:23:19 +00002548 the following additional flags:
2549 %N - Set output attribute to normal
2550 %H - Set output attribute to highlight
2551 %E - Set output attribute to error
2552 %B - Set output attribute to blue color
2553 %V - Set output attribute to green color
2554
2555 Note: The background color is controlled by the shell command cls.
2556
2557 @param[in] Row the row to print at
2558 @param[in] Col the column to print at
2559 @param[in] Format the format string
jcarsey2247dde2009-11-09 18:08:58 +00002560 @param[in] Marker the marker for the variable argument list
jcarseyb1f95a02009-06-16 00:23:19 +00002561
jcarseya405b862010-09-14 05:18:09 +00002562 @return EFI_SUCCESS The operation was successful.
2563 @return EFI_DEVICE_ERROR The console device reported an error.
jcarseyb1f95a02009-06-16 00:23:19 +00002564**/
jcarseya405b862010-09-14 05:18:09 +00002565EFI_STATUS
jcarseyb1f95a02009-06-16 00:23:19 +00002566EFIAPI
jcarsey2247dde2009-11-09 18:08:58 +00002567InternalShellPrintWorker(
jcarseyb1f95a02009-06-16 00:23:19 +00002568 IN INT32 Col OPTIONAL,
2569 IN INT32 Row OPTIONAL,
2570 IN CONST CHAR16 *Format,
jcarsey2247dde2009-11-09 18:08:58 +00002571 VA_LIST Marker
jcarsey1e6e84c2010-01-25 20:05:08 +00002572 )
jcarsey2247dde2009-11-09 18:08:58 +00002573{
jcarseyb1f95a02009-06-16 00:23:19 +00002574 EFI_STATUS Status;
jcarsey975136a2009-06-16 19:03:54 +00002575 CHAR16 *ResumeLocation;
2576 CHAR16 *FormatWalker;
jcarseya405b862010-09-14 05:18:09 +00002577 UINTN OriginalAttribute;
2578
2579 Status = EFI_SUCCESS;
2580 OriginalAttribute = gST->ConOut->Mode->Attribute;
jcarsey1e6e84c2010-01-25 20:05:08 +00002581
jcarsey975136a2009-06-16 19:03:54 +00002582 //
2583 // Back and forth each time fixing up 1 of our flags...
2584 //
jcarseya405b862010-09-14 05:18:09 +00002585 Status = ShellCopySearchAndReplace(Format, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%N", L"%%N", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002586 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002587 Status = ShellCopySearchAndReplace(mPostReplaceFormat, mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), L"%E", L"%%E", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002588 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002589 Status = ShellCopySearchAndReplace(mPostReplaceFormat2, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%H", L"%%H", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002590 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002591 Status = ShellCopySearchAndReplace(mPostReplaceFormat, mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), L"%B", L"%%B", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002592 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002593 Status = ShellCopySearchAndReplace(mPostReplaceFormat2, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%V", L"%%V", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002594 ASSERT_EFI_ERROR(Status);
2595
2596 //
2597 // Use the last buffer from replacing to print from...
2598 //
jcarseya405b862010-09-14 05:18:09 +00002599 UnicodeVSPrint (mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), mPostReplaceFormat, Marker);
jcarseyb1f95a02009-06-16 00:23:19 +00002600
2601 if (Col != -1 && Row != -1) {
jcarseyb1f95a02009-06-16 00:23:19 +00002602 Status = gST->ConOut->SetCursorPosition(gST->ConOut, Col, Row);
2603 ASSERT_EFI_ERROR(Status);
jcarsey975136a2009-06-16 19:03:54 +00002604 }
2605
jcarseyecd3d592009-12-07 18:05:00 +00002606 FormatWalker = mPostReplaceFormat2;
jcarsey2247dde2009-11-09 18:08:58 +00002607 while (*FormatWalker != CHAR_NULL) {
jcarsey975136a2009-06-16 19:03:54 +00002608 //
2609 // Find the next attribute change request
2610 //
2611 ResumeLocation = StrStr(FormatWalker, L"%");
2612 if (ResumeLocation != NULL) {
jcarsey2247dde2009-11-09 18:08:58 +00002613 *ResumeLocation = CHAR_NULL;
jcarsey975136a2009-06-16 19:03:54 +00002614 }
2615 //
2616 // print the current FormatWalker string
2617 //
jcarseya405b862010-09-14 05:18:09 +00002618 if (StrLen(FormatWalker)>0) {
2619 Status = InternalPrintTo(FormatWalker);
2620 if (EFI_ERROR(Status)) {
2621 break;
2622 }
2623 }
2624
jcarsey975136a2009-06-16 19:03:54 +00002625 //
2626 // update the attribute
2627 //
2628 if (ResumeLocation != NULL) {
2629 switch (*(ResumeLocation+1)) {
2630 case (L'N'):
jcarseya405b862010-09-14 05:18:09 +00002631 gST->ConOut->SetAttribute(gST->ConOut, OriginalAttribute);
jcarsey975136a2009-06-16 19:03:54 +00002632 break;
2633 case (L'E'):
jcarseya405b862010-09-14 05:18:09 +00002634 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_YELLOW, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
jcarsey975136a2009-06-16 19:03:54 +00002635 break;
2636 case (L'H'):
jcarseya405b862010-09-14 05:18:09 +00002637 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_WHITE, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
jcarsey975136a2009-06-16 19:03:54 +00002638 break;
2639 case (L'B'):
jcarseya405b862010-09-14 05:18:09 +00002640 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_BLUE, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
jcarsey975136a2009-06-16 19:03:54 +00002641 break;
2642 case (L'V'):
jcarseya405b862010-09-14 05:18:09 +00002643 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_GREEN, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
jcarsey975136a2009-06-16 19:03:54 +00002644 break;
2645 default:
jcarseye2f82972009-12-01 05:40:24 +00002646 //
2647 // Print a simple '%' symbol
2648 //
2649 Status = InternalPrintTo(L"%");
jcarseya405b862010-09-14 05:18:09 +00002650 if (EFI_ERROR(Status)) {
2651 break;
2652 }
jcarseye2f82972009-12-01 05:40:24 +00002653 ResumeLocation = ResumeLocation - 1;
jcarsey975136a2009-06-16 19:03:54 +00002654 break;
2655 }
2656 } else {
2657 //
2658 // reset to normal now...
2659 //
jcarsey975136a2009-06-16 19:03:54 +00002660 break;
2661 }
2662
2663 //
2664 // update FormatWalker to Resume + 2 (skip the % and the indicator)
2665 //
2666 FormatWalker = ResumeLocation + 2;
2667 }
jcarseyb1f95a02009-06-16 00:23:19 +00002668
jcarseya405b862010-09-14 05:18:09 +00002669 gST->ConOut->SetAttribute(gST->ConOut, OriginalAttribute);
2670 return (Status);
jcarsey5f7431d2009-07-10 18:06:01 +00002671}
jcarsey2247dde2009-11-09 18:08:58 +00002672
2673/**
2674 Print at a specific location on the screen.
2675
jcarseye2f82972009-12-01 05:40:24 +00002676 This function will move the cursor to a given screen location and print the specified string.
jcarsey1e6e84c2010-01-25 20:05:08 +00002677
2678 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarsey2247dde2009-11-09 18:08:58 +00002679 will be used.
2680
jcarseye2f82972009-12-01 05:40:24 +00002681 If either Row or Col is out of range for the current console, then ASSERT.
2682 If Format is NULL, then ASSERT.
jcarsey2247dde2009-11-09 18:08:58 +00002683
jcarsey1e6e84c2010-01-25 20:05:08 +00002684 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarsey2247dde2009-11-09 18:08:58 +00002685 the following additional flags:
2686 %N - Set output attribute to normal
2687 %H - Set output attribute to highlight
2688 %E - Set output attribute to error
2689 %B - Set output attribute to blue color
2690 %V - Set output attribute to green color
2691
2692 Note: The background color is controlled by the shell command cls.
2693
jcarsey2247dde2009-11-09 18:08:58 +00002694 @param[in] Col the column to print at
jcarseya405b862010-09-14 05:18:09 +00002695 @param[in] Row the row to print at
jcarsey2247dde2009-11-09 18:08:58 +00002696 @param[in] Format the format string
jcarseya405b862010-09-14 05:18:09 +00002697 @param[in] ... The variable argument list.
jcarsey2247dde2009-11-09 18:08:58 +00002698
jcarseya405b862010-09-14 05:18:09 +00002699 @return EFI_SUCCESS The printing was successful.
2700 @return EFI_DEVICE_ERROR The console device reported an error.
jcarsey2247dde2009-11-09 18:08:58 +00002701**/
jcarseya405b862010-09-14 05:18:09 +00002702EFI_STATUS
jcarsey2247dde2009-11-09 18:08:58 +00002703EFIAPI
2704ShellPrintEx(
2705 IN INT32 Col OPTIONAL,
2706 IN INT32 Row OPTIONAL,
2707 IN CONST CHAR16 *Format,
2708 ...
jcarsey1e6e84c2010-01-25 20:05:08 +00002709 )
jcarsey2247dde2009-11-09 18:08:58 +00002710{
2711 VA_LIST Marker;
jcarseya405b862010-09-14 05:18:09 +00002712 EFI_STATUS RetVal;
jcarsey3e082d52010-10-04 16:44:57 +00002713 if (Format == NULL) {
2714 return (EFI_INVALID_PARAMETER);
2715 }
jcarsey2247dde2009-11-09 18:08:58 +00002716 VA_START (Marker, Format);
jcarseya405b862010-09-14 05:18:09 +00002717 RetVal = InternalShellPrintWorker(Col, Row, Format, Marker);
jcarseye2f82972009-12-01 05:40:24 +00002718 VA_END(Marker);
jcarseya405b862010-09-14 05:18:09 +00002719 return(RetVal);
jcarsey2247dde2009-11-09 18:08:58 +00002720}
2721
2722/**
2723 Print at a specific location on the screen.
2724
jcarseye2f82972009-12-01 05:40:24 +00002725 This function will move the cursor to a given screen location and print the specified string.
jcarsey1e6e84c2010-01-25 20:05:08 +00002726
2727 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarseye2f82972009-12-01 05:40:24 +00002728 will be used.
jcarsey2247dde2009-11-09 18:08:58 +00002729
jcarseye2f82972009-12-01 05:40:24 +00002730 If either Row or Col is out of range for the current console, then ASSERT.
2731 If Format is NULL, then ASSERT.
jcarsey2247dde2009-11-09 18:08:58 +00002732
jcarsey1e6e84c2010-01-25 20:05:08 +00002733 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarsey2247dde2009-11-09 18:08:58 +00002734 the following additional flags:
jcarsey1e6e84c2010-01-25 20:05:08 +00002735 %N - Set output attribute to normal.
2736 %H - Set output attribute to highlight.
2737 %E - Set output attribute to error.
2738 %B - Set output attribute to blue color.
2739 %V - Set output attribute to green color.
jcarsey2247dde2009-11-09 18:08:58 +00002740
2741 Note: The background color is controlled by the shell command cls.
2742
jcarsey1e6e84c2010-01-25 20:05:08 +00002743 @param[in] Col The column to print at.
jcarseya405b862010-09-14 05:18:09 +00002744 @param[in] Row The row to print at.
jcarsey1e6e84c2010-01-25 20:05:08 +00002745 @param[in] Language The language of the string to retrieve. If this parameter
2746 is NULL, then the current platform language is used.
2747 @param[in] HiiFormatStringId The format string Id for getting from Hii.
2748 @param[in] HiiFormatHandle The format string Handle for getting from Hii.
jcarseya405b862010-09-14 05:18:09 +00002749 @param[in] ... The variable argument list.
jcarsey2247dde2009-11-09 18:08:58 +00002750
jcarseya405b862010-09-14 05:18:09 +00002751 @return EFI_SUCCESS The printing was successful.
2752 @return EFI_DEVICE_ERROR The console device reported an error.
jcarsey2247dde2009-11-09 18:08:58 +00002753**/
jcarseya405b862010-09-14 05:18:09 +00002754EFI_STATUS
jcarsey2247dde2009-11-09 18:08:58 +00002755EFIAPI
2756ShellPrintHiiEx(
2757 IN INT32 Col OPTIONAL,
2758 IN INT32 Row OPTIONAL,
jcarsey1e6e84c2010-01-25 20:05:08 +00002759 IN CONST CHAR8 *Language OPTIONAL,
jcarsey2247dde2009-11-09 18:08:58 +00002760 IN CONST EFI_STRING_ID HiiFormatStringId,
2761 IN CONST EFI_HANDLE HiiFormatHandle,
2762 ...
2763 )
2764{
2765 VA_LIST Marker;
2766 CHAR16 *HiiFormatString;
jcarseya405b862010-09-14 05:18:09 +00002767 EFI_STATUS RetVal;
jcarsey2247dde2009-11-09 18:08:58 +00002768
2769 VA_START (Marker, HiiFormatHandle);
jcarsey1e6e84c2010-01-25 20:05:08 +00002770 HiiFormatString = HiiGetString(HiiFormatHandle, HiiFormatStringId, Language);
jcarsey2247dde2009-11-09 18:08:58 +00002771 ASSERT(HiiFormatString != NULL);
2772
2773 RetVal = InternalShellPrintWorker(Col, Row, HiiFormatString, Marker);
2774
jcarseya405b862010-09-14 05:18:09 +00002775 SHELL_FREE_NON_NULL(HiiFormatString);
jcarseye2f82972009-12-01 05:40:24 +00002776 VA_END(Marker);
jcarsey2247dde2009-11-09 18:08:58 +00002777
2778 return (RetVal);
2779}
2780
2781/**
2782 Function to determine if a given filename represents a file or a directory.
2783
2784 @param[in] DirName Path to directory to test.
2785
2786 @retval EFI_SUCCESS The Path represents a directory
2787 @retval EFI_NOT_FOUND The Path does not represent a directory
2788 @return other The path failed to open
2789**/
2790EFI_STATUS
2791EFIAPI
2792ShellIsDirectory(
2793 IN CONST CHAR16 *DirName
2794 )
2795{
2796 EFI_STATUS Status;
jcarseya405b862010-09-14 05:18:09 +00002797 SHELL_FILE_HANDLE Handle;
jcarsey3e082d52010-10-04 16:44:57 +00002798 CHAR16 *TempLocation;
2799 CHAR16 *TempLocation2;
jcarsey2247dde2009-11-09 18:08:58 +00002800
jcarseyecd3d592009-12-07 18:05:00 +00002801 ASSERT(DirName != NULL);
2802
jcarseya405b862010-09-14 05:18:09 +00002803 Handle = NULL;
2804 TempLocation = NULL;
jcarsey2247dde2009-11-09 18:08:58 +00002805
2806 Status = ShellOpenFileByName(DirName, &Handle, EFI_FILE_MODE_READ, 0);
2807 if (EFI_ERROR(Status)) {
jcarseya405b862010-09-14 05:18:09 +00002808 //
2809 // try good logic first.
2810 //
2811 if (mEfiShellProtocol != NULL) {
jcarsey3e082d52010-10-04 16:44:57 +00002812 TempLocation = StrnCatGrow(&TempLocation, NULL, DirName, 0);
2813 TempLocation2 = StrStr(TempLocation, L":");
2814 if (TempLocation2 != NULL && StrLen(StrStr(TempLocation, L":")) == 2) {
2815 *(TempLocation2+1) = CHAR_NULL;
jcarseya405b862010-09-14 05:18:09 +00002816 }
2817 if (mEfiShellProtocol->GetDevicePathFromMap(TempLocation) != NULL) {
2818 FreePool(TempLocation);
2819 return (EFI_SUCCESS);
2820 }
2821 FreePool(TempLocation);
2822 } else {
2823 //
2824 // probably a map name?!?!!?
2825 //
2826 TempLocation = StrStr(DirName, L"\\");
2827 if (TempLocation != NULL && *(TempLocation+1) == CHAR_NULL) {
2828 return (EFI_SUCCESS);
2829 }
2830 }
jcarsey2247dde2009-11-09 18:08:58 +00002831 return (Status);
2832 }
2833
2834 if (FileHandleIsDirectory(Handle) == EFI_SUCCESS) {
2835 ShellCloseFile(&Handle);
2836 return (EFI_SUCCESS);
2837 }
2838 ShellCloseFile(&Handle);
2839 return (EFI_NOT_FOUND);
2840}
2841
jcarsey125c2cf2009-11-18 21:36:50 +00002842/**
jcarsey36a9d672009-11-20 21:13:41 +00002843 Function to determine if a given filename represents a file.
2844
2845 @param[in] Name Path to file to test.
2846
2847 @retval EFI_SUCCESS The Path represents a file.
2848 @retval EFI_NOT_FOUND The Path does not represent a file.
2849 @retval other The path failed to open.
2850**/
2851EFI_STATUS
2852EFIAPI
2853ShellIsFile(
2854 IN CONST CHAR16 *Name
2855 )
2856{
2857 EFI_STATUS Status;
jcarseya405b862010-09-14 05:18:09 +00002858 SHELL_FILE_HANDLE Handle;
jcarsey36a9d672009-11-20 21:13:41 +00002859
jcarseyecd3d592009-12-07 18:05:00 +00002860 ASSERT(Name != NULL);
2861
jcarsey36a9d672009-11-20 21:13:41 +00002862 Handle = NULL;
2863
2864 Status = ShellOpenFileByName(Name, &Handle, EFI_FILE_MODE_READ, 0);
2865 if (EFI_ERROR(Status)) {
2866 return (Status);
2867 }
2868
2869 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
2870 ShellCloseFile(&Handle);
2871 return (EFI_SUCCESS);
2872 }
2873 ShellCloseFile(&Handle);
2874 return (EFI_NOT_FOUND);
2875}
2876
2877/**
jcarseyb3011f42010-01-11 21:49:04 +00002878 Function to determine if a given filename represents a file.
2879
2880 This will search the CWD and then the Path.
2881
2882 If Name is NULL, then ASSERT.
2883
2884 @param[in] Name Path to file to test.
2885
2886 @retval EFI_SUCCESS The Path represents a file.
2887 @retval EFI_NOT_FOUND The Path does not represent a file.
2888 @retval other The path failed to open.
2889**/
2890EFI_STATUS
2891EFIAPI
2892ShellIsFileInPath(
2893 IN CONST CHAR16 *Name
jcarseya405b862010-09-14 05:18:09 +00002894 )
2895{
jcarseyb3011f42010-01-11 21:49:04 +00002896 CHAR16 *NewName;
2897 EFI_STATUS Status;
2898
2899 if (!EFI_ERROR(ShellIsFile(Name))) {
jcarseya405b862010-09-14 05:18:09 +00002900 return (EFI_SUCCESS);
jcarseyb3011f42010-01-11 21:49:04 +00002901 }
2902
2903 NewName = ShellFindFilePath(Name);
2904 if (NewName == NULL) {
2905 return (EFI_NOT_FOUND);
2906 }
2907 Status = ShellIsFile(NewName);
2908 FreePool(NewName);
2909 return (Status);
2910}
2911/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002912 Function to determine whether a string is decimal or hex representation of a number
jcarsey125c2cf2009-11-18 21:36:50 +00002913 and return the number converted from the string.
2914
2915 @param[in] String String representation of a number
2916
2917 @retval all the number
2918**/
2919UINTN
2920EFIAPI
2921ShellStrToUintn(
2922 IN CONST CHAR16 *String
2923 )
2924{
2925 CONST CHAR16 *Walker;
jcarseyb3011f42010-01-11 21:49:04 +00002926 for (Walker = String; Walker != NULL && *Walker != CHAR_NULL && *Walker == L' '; Walker++);
jcarsey1cd45e72010-01-29 15:07:44 +00002927 if (Walker == NULL || *Walker == CHAR_NULL) {
2928 ASSERT(FALSE);
2929 return ((UINTN)(-1));
2930 } else {
2931 if (StrnCmp(Walker, L"0x", 2) == 0 || StrnCmp(Walker, L"0X", 2) == 0){
2932 return (StrHexToUintn(Walker));
2933 }
2934 return (StrDecimalToUintn(Walker));
jcarsey125c2cf2009-11-18 21:36:50 +00002935 }
jcarsey125c2cf2009-11-18 21:36:50 +00002936}
2937
2938/**
2939 Safely append with automatic string resizing given length of Destination and
2940 desired length of copy from Source.
2941
2942 append the first D characters of Source to the end of Destination, where D is
2943 the lesser of Count and the StrLen() of Source. If appending those D characters
2944 will fit within Destination (whose Size is given as CurrentSize) and
jcarsey1e6e84c2010-01-25 20:05:08 +00002945 still leave room for a NULL terminator, then those characters are appended,
2946 starting at the original terminating NULL of Destination, and a new terminating
2947 NULL is appended.
jcarsey125c2cf2009-11-18 21:36:50 +00002948
2949 If appending D characters onto Destination will result in a overflow of the size
2950 given in CurrentSize the string will be grown such that the copy can be performed
2951 and CurrentSize will be updated to the new size.
2952
2953 If Source is NULL, there is nothing to append, just return the current buffer in
2954 Destination.
2955
2956 if Destination is NULL, then ASSERT()
2957 if Destination's current length (including NULL terminator) is already more then
2958 CurrentSize, then ASSERT()
2959
2960 @param[in,out] Destination The String to append onto
2961 @param[in,out] CurrentSize on call the number of bytes in Destination. On
2962 return possibly the new size (still in bytes). if NULL
2963 then allocate whatever is needed.
2964 @param[in] Source The String to append from
2965 @param[in] Count Maximum number of characters to append. if 0 then
2966 all are appended.
2967
2968 @return Destination return the resultant string.
2969**/
2970CHAR16*
2971EFIAPI
2972StrnCatGrow (
2973 IN OUT CHAR16 **Destination,
2974 IN OUT UINTN *CurrentSize,
2975 IN CONST CHAR16 *Source,
2976 IN UINTN Count
2977 )
2978{
2979 UINTN DestinationStartSize;
2980 UINTN NewSize;
2981
2982 //
2983 // ASSERTs
2984 //
2985 ASSERT(Destination != NULL);
2986
2987 //
2988 // If there's nothing to do then just return Destination
2989 //
2990 if (Source == NULL) {
2991 return (*Destination);
2992 }
2993
2994 //
2995 // allow for un-initialized pointers, based on size being 0
2996 //
2997 if (CurrentSize != NULL && *CurrentSize == 0) {
2998 *Destination = NULL;
2999 }
3000
3001 //
3002 // allow for NULL pointers address as Destination
3003 //
3004 if (*Destination != NULL) {
3005 ASSERT(CurrentSize != 0);
3006 DestinationStartSize = StrSize(*Destination);
3007 ASSERT(DestinationStartSize <= *CurrentSize);
3008 } else {
3009 DestinationStartSize = 0;
3010// ASSERT(*CurrentSize == 0);
3011 }
3012
3013 //
3014 // Append all of Source?
3015 //
3016 if (Count == 0) {
3017 Count = StrLen(Source);
3018 }
3019
3020 //
3021 // Test and grow if required
3022 //
3023 if (CurrentSize != NULL) {
3024 NewSize = *CurrentSize;
3025 while (NewSize < (DestinationStartSize + (Count*sizeof(CHAR16)))) {
3026 NewSize += 2 * Count * sizeof(CHAR16);
3027 }
3028 *Destination = ReallocatePool(*CurrentSize, NewSize, *Destination);
jcarseyc9d92df2010-02-03 15:37:54 +00003029 ASSERT(*Destination != NULL);
jcarsey125c2cf2009-11-18 21:36:50 +00003030 *CurrentSize = NewSize;
3031 } else {
3032 *Destination = AllocateZeroPool((Count+1)*sizeof(CHAR16));
jcarseyc9d92df2010-02-03 15:37:54 +00003033 ASSERT(*Destination != NULL);
jcarsey125c2cf2009-11-18 21:36:50 +00003034 }
3035
3036 //
3037 // Now use standard StrnCat on a big enough buffer
3038 //
jcarseyc9d92df2010-02-03 15:37:54 +00003039 if (*Destination == NULL) {
3040 return (NULL);
3041 }
jcarsey125c2cf2009-11-18 21:36:50 +00003042 return StrnCat(*Destination, Source, Count);
3043}
jcarseyc9d92df2010-02-03 15:37:54 +00003044
3045/**
3046 Prompt the user and return the resultant answer to the requestor.
3047
3048 This function will display the requested question on the shell prompt and then
3049 wait for an apropriate answer to be input from the console.
3050
jcarseya405b862010-09-14 05:18:09 +00003051 if the SHELL_PROMPT_REQUEST_TYPE is SHELL_PROMPT_REQUEST_TYPE_YESNO, ShellPromptResponseTypeQuitContinue
jcarseyc9d92df2010-02-03 15:37:54 +00003052 or SHELL_PROMPT_REQUEST_TYPE_YESNOCANCEL then *Response is of type SHELL_PROMPT_RESPONSE.
3053
jcarseya405b862010-09-14 05:18:09 +00003054 if the SHELL_PROMPT_REQUEST_TYPE is ShellPromptResponseTypeFreeform then *Response is of type
jcarseyc9d92df2010-02-03 15:37:54 +00003055 CHAR16*.
3056
3057 In either case *Response must be callee freed if Response was not NULL;
3058
3059 @param Type What type of question is asked. This is used to filter the input
3060 to prevent invalid answers to question.
3061 @param Prompt Pointer to string prompt to use to request input.
3062 @param Response Pointer to Response which will be populated upon return.
3063
3064 @retval EFI_SUCCESS The operation was sucessful.
3065 @retval EFI_UNSUPPORTED The operation is not supported as requested.
3066 @retval EFI_INVALID_PARAMETER A parameter was invalid.
3067 @return other The operation failed.
3068**/
3069EFI_STATUS
3070EFIAPI
3071ShellPromptForResponse (
3072 IN SHELL_PROMPT_REQUEST_TYPE Type,
3073 IN CHAR16 *Prompt OPTIONAL,
3074 IN OUT VOID **Response OPTIONAL
3075 )
3076{
3077 EFI_STATUS Status;
3078 EFI_INPUT_KEY Key;
3079 UINTN EventIndex;
3080 SHELL_PROMPT_RESPONSE *Resp;
jcarseya405b862010-09-14 05:18:09 +00003081 UINTN Size;
3082 CHAR16 *Buffer;
jcarseyc9d92df2010-02-03 15:37:54 +00003083
jcarseya405b862010-09-14 05:18:09 +00003084 Status = EFI_UNSUPPORTED;
3085 Resp = NULL;
3086 Buffer = NULL;
3087 Size = 0;
3088 if (Type != ShellPromptResponseTypeFreeform) {
3089 Resp = (SHELL_PROMPT_RESPONSE*)AllocatePool(sizeof(SHELL_PROMPT_RESPONSE));
jcarsey3e082d52010-10-04 16:44:57 +00003090 if (Resp == NULL) {
3091 return (EFI_OUT_OF_RESOURCES);
3092 }
xdu290bfa222010-07-19 05:21:27 +00003093 }
jcarseyc9d92df2010-02-03 15:37:54 +00003094
3095 switch(Type) {
jcarseya405b862010-09-14 05:18:09 +00003096 case ShellPromptResponseTypeQuitContinue:
jcarseyc9d92df2010-02-03 15:37:54 +00003097 if (Prompt != NULL) {
3098 ShellPrintEx(-1, -1, L"%s", Prompt);
3099 }
3100 //
3101 // wait for valid response
3102 //
3103 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3104 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3105 ASSERT_EFI_ERROR(Status);
3106 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3107 if (Key.UnicodeChar == L'Q' || Key.UnicodeChar ==L'q') {
jcarseya405b862010-09-14 05:18:09 +00003108 *Resp = ShellPromptResponseQuit;
jcarseyc9d92df2010-02-03 15:37:54 +00003109 } else {
jcarseya405b862010-09-14 05:18:09 +00003110 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003111 }
3112 break;
jcarseya405b862010-09-14 05:18:09 +00003113 case ShellPromptResponseTypeYesNoCancel:
jcarseyc9d92df2010-02-03 15:37:54 +00003114 if (Prompt != NULL) {
3115 ShellPrintEx(-1, -1, L"%s", Prompt);
3116 }
3117 //
3118 // wait for valid response
3119 //
jcarseya405b862010-09-14 05:18:09 +00003120 *Resp = ShellPromptResponseMax;
3121 while (*Resp == ShellPromptResponseMax) {
jcarseyc9d92df2010-02-03 15:37:54 +00003122 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3123 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3124 ASSERT_EFI_ERROR(Status);
3125 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3126 switch (Key.UnicodeChar) {
3127 case L'Y':
3128 case L'y':
jcarseya405b862010-09-14 05:18:09 +00003129 *Resp = ShellPromptResponseYes;
jcarseyc9d92df2010-02-03 15:37:54 +00003130 break;
3131 case L'N':
3132 case L'n':
jcarseya405b862010-09-14 05:18:09 +00003133 *Resp = ShellPromptResponseNo;
jcarseyc9d92df2010-02-03 15:37:54 +00003134 break;
3135 case L'C':
3136 case L'c':
jcarseya405b862010-09-14 05:18:09 +00003137 *Resp = ShellPromptResponseCancel;
3138 break;
3139 }
3140 }
3141 break; case ShellPromptResponseTypeYesNoAllCancel:
3142 if (Prompt != NULL) {
3143 ShellPrintEx(-1, -1, L"%s", Prompt);
3144 }
3145 //
3146 // wait for valid response
3147 //
3148 *Resp = ShellPromptResponseMax;
3149 while (*Resp == ShellPromptResponseMax) {
3150 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3151 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3152 ASSERT_EFI_ERROR(Status);
3153 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3154 switch (Key.UnicodeChar) {
3155 case L'Y':
3156 case L'y':
3157 *Resp = ShellPromptResponseYes;
3158 break;
3159 case L'N':
3160 case L'n':
3161 *Resp = ShellPromptResponseNo;
3162 break;
3163 case L'A':
3164 case L'a':
3165 *Resp = ShellPromptResponseAll;
3166 break;
3167 case L'C':
3168 case L'c':
3169 *Resp = ShellPromptResponseCancel;
jcarseyc9d92df2010-02-03 15:37:54 +00003170 break;
3171 }
3172 }
3173 break;
jcarseya405b862010-09-14 05:18:09 +00003174 case ShellPromptResponseTypeEnterContinue:
3175 case ShellPromptResponseTypeAnyKeyContinue:
jcarseyc9d92df2010-02-03 15:37:54 +00003176 if (Prompt != NULL) {
3177 ShellPrintEx(-1, -1, L"%s", Prompt);
3178 }
3179 //
3180 // wait for valid response
3181 //
jcarseya405b862010-09-14 05:18:09 +00003182 *Resp = ShellPromptResponseMax;
3183 while (*Resp == ShellPromptResponseMax) {
jcarseyc9d92df2010-02-03 15:37:54 +00003184 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
jcarseya405b862010-09-14 05:18:09 +00003185 if (Type == ShellPromptResponseTypeEnterContinue) {
jcarseyc9d92df2010-02-03 15:37:54 +00003186 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3187 ASSERT_EFI_ERROR(Status);
3188 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3189 if (Key.UnicodeChar == CHAR_CARRIAGE_RETURN) {
jcarseya405b862010-09-14 05:18:09 +00003190 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003191 break;
3192 }
3193 }
jcarseya405b862010-09-14 05:18:09 +00003194 if (Type == ShellPromptResponseTypeAnyKeyContinue) {
3195 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3196 ASSERT_EFI_ERROR(Status);
3197 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003198 break;
3199 }
3200 }
3201 break;
jcarseya405b862010-09-14 05:18:09 +00003202 case ShellPromptResponseTypeYesNo:
3203 if (Prompt != NULL) {
3204 ShellPrintEx(-1, -1, L"%s", Prompt);
3205 }
3206 //
3207 // wait for valid response
3208 //
3209 *Resp = ShellPromptResponseMax;
3210 while (*Resp == ShellPromptResponseMax) {
3211 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3212 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3213 ASSERT_EFI_ERROR(Status);
3214 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3215 switch (Key.UnicodeChar) {
3216 case L'Y':
3217 case L'y':
3218 *Resp = ShellPromptResponseYes;
3219 break;
3220 case L'N':
3221 case L'n':
3222 *Resp = ShellPromptResponseNo;
3223 break;
3224 }
3225 }
3226 break;
3227 case ShellPromptResponseTypeFreeform:
3228 if (Prompt != NULL) {
3229 ShellPrintEx(-1, -1, L"%s", Prompt);
3230 }
3231 while(1) {
3232 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3233 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3234 ASSERT_EFI_ERROR(Status);
3235 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3236 if (Key.UnicodeChar == CHAR_CARRIAGE_RETURN) {
3237 break;
3238 }
3239 ASSERT((Buffer == NULL && Size == 0) || (Buffer != NULL));
3240 StrnCatGrow(&Buffer, &Size, &Key.UnicodeChar, 1);
3241 }
3242 break;
3243 //
3244 // This is the location to add new prompt types.
3245 //
jcarseyc9d92df2010-02-03 15:37:54 +00003246 default:
jcarseya405b862010-09-14 05:18:09 +00003247 ASSERT(FALSE);
jcarseyc9d92df2010-02-03 15:37:54 +00003248 }
3249
3250 if (Response != NULL) {
jcarseya405b862010-09-14 05:18:09 +00003251 if (Resp != NULL) {
3252 *Response = Resp;
3253 } else if (Buffer != NULL) {
3254 *Response = Buffer;
3255 }
jcarseyc9d92df2010-02-03 15:37:54 +00003256 } else {
jcarseya405b862010-09-14 05:18:09 +00003257 if (Resp != NULL) {
3258 FreePool(Resp);
3259 }
3260 if (Buffer != NULL) {
3261 FreePool(Buffer);
3262 }
jcarseyc9d92df2010-02-03 15:37:54 +00003263 }
3264
jcarseya405b862010-09-14 05:18:09 +00003265 ShellPrintEx(-1, -1, L"\r\n");
jcarseyc9d92df2010-02-03 15:37:54 +00003266 return (Status);
3267}
3268
3269/**
3270 Prompt the user and return the resultant answer to the requestor.
3271
3272 This function is the same as ShellPromptForResponse, except that the prompt is
3273 automatically pulled from HII.
3274
3275 @param Type What type of question is asked. This is used to filter the input
3276 to prevent invalid answers to question.
jcarseya405b862010-09-14 05:18:09 +00003277 @param[in] HiiFormatStringId The format string Id for getting from Hii.
3278 @param[in] HiiFormatHandle The format string Handle for getting from Hii.
3279 @param Response Pointer to Response which will be populated upon return.
jcarseyc9d92df2010-02-03 15:37:54 +00003280
3281 @retval EFI_SUCCESS the operation was sucessful.
3282 @return other the operation failed.
3283
3284 @sa ShellPromptForResponse
3285**/
3286EFI_STATUS
3287EFIAPI
3288ShellPromptForResponseHii (
3289 IN SHELL_PROMPT_REQUEST_TYPE Type,
3290 IN CONST EFI_STRING_ID HiiFormatStringId,
3291 IN CONST EFI_HANDLE HiiFormatHandle,
3292 IN OUT VOID **Response
3293 )
3294{
3295 CHAR16 *Prompt;
3296 EFI_STATUS Status;
3297
3298 Prompt = HiiGetString(HiiFormatHandle, HiiFormatStringId, NULL);
3299 Status = ShellPromptForResponse(Type, Prompt, Response);
3300 FreePool(Prompt);
3301 return (Status);
3302}
3303
jcarseya405b862010-09-14 05:18:09 +00003304/**
3305 Function to determin if an entire string is a valid number.
jcarseyc9d92df2010-02-03 15:37:54 +00003306
jcarseya405b862010-09-14 05:18:09 +00003307 If Hex it must be preceeded with a 0x or has ForceHex, set TRUE.
3308
3309 @param[in] String The string to evaluate.
3310 @param[in] ForceHex TRUE - always assume hex.
3311 @param[in] StopAtSpace TRUE to halt upon finding a space, FALSE to keep going.
3312
3313 @retval TRUE It is all numeric (dec/hex) characters.
3314 @retval FALSE There is a non-numeric character.
3315**/
3316BOOLEAN
3317EFIAPI
3318ShellIsHexOrDecimalNumber (
3319 IN CONST CHAR16 *String,
3320 IN CONST BOOLEAN ForceHex,
3321 IN CONST BOOLEAN StopAtSpace
3322 )
3323{
3324 BOOLEAN Hex;
3325
3326 //
3327 // chop off a single negative sign
3328 //
3329 if (String != NULL && *String == L'-') {
3330 String++;
3331 }
3332
3333 if (String == NULL) {
3334 return (FALSE);
3335 }
3336
3337 //
3338 // chop leading zeroes
3339 //
3340 while(String != NULL && *String == L'0'){
3341 String++;
3342 }
3343 //
3344 // allow '0x' or '0X', but not 'x' or 'X'
3345 //
3346 if (String != NULL && (*String == L'x' || *String == L'X')) {
3347 if (*(String-1) != L'0') {
3348 //
3349 // we got an x without a preceeding 0
3350 //
3351 return (FALSE);
3352 }
3353 String++;
3354 Hex = TRUE;
3355 } else if (ForceHex) {
3356 Hex = TRUE;
3357 } else {
3358 Hex = FALSE;
3359 }
3360
3361 //
3362 // loop through the remaining characters and use the lib function
3363 //
3364 for ( ; String != NULL && *String != CHAR_NULL && !(StopAtSpace && *String == L' ') ; String++){
3365 if (Hex) {
3366 if (!ShellIsHexaDecimalDigitCharacter(*String)) {
3367 return (FALSE);
3368 }
3369 } else {
3370 if (!ShellIsDecimalDigitCharacter(*String)) {
3371 return (FALSE);
3372 }
3373 }
3374 }
3375 return (TRUE);
3376}
3377
3378/**
3379 Function to determine if a given filename exists.
3380
3381 @param[in] Name Path to test.
3382
3383 @retval EFI_SUCCESS The Path represents a file.
3384 @retval EFI_NOT_FOUND The Path does not represent a file.
3385 @retval other The path failed to open.
3386**/
3387EFI_STATUS
3388EFIAPI
3389ShellFileExists(
3390 IN CONST CHAR16 *Name
3391 )
3392{
3393 EFI_STATUS Status;
3394 EFI_SHELL_FILE_INFO *List;
3395
3396 ASSERT(Name != NULL);
3397
3398 List = NULL;
3399 Status = ShellOpenFileMetaArg((CHAR16*)Name, EFI_FILE_MODE_READ, &List);
3400 if (EFI_ERROR(Status)) {
3401 return (Status);
3402 }
3403
3404 ShellCloseFileMetaArg(&List);
3405
3406 return (EFI_SUCCESS);
3407}