blob: 85a5d44f081495b553c64206f7fd31ab65d9948b [file] [log] [blame]
jcarsey94b17fa2009-05-07 18:46:18 +00001/** @file
2 Provides interface to shell functionality for shell commands and applications.
3
jcarsey49bd4982012-01-27 18:42:43 +00004 Copyright (c) 2006 - 2012, Intel Corporation. All rights reserved.<BR>
jcarsey1e6e84c2010-01-25 20:05:08 +00005 This program and the accompanying materials
jcarseyb3011f42010-01-11 21:49:04 +00006 are licensed and made available under the terms and conditions of the BSD License
7 which accompanies this distribution. The full text of the license may be found at
8 http://opensource.org/licenses/bsd-license.php
jcarsey94b17fa2009-05-07 18:46:18 +00009
jcarseyb3011f42010-01-11 21:49:04 +000010 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
jcarsey94b17fa2009-05-07 18:46:18 +000012
13**/
14
jcarseyb1f95a02009-06-16 00:23:19 +000015#include "UefiShellLib.h"
jcarseya405b862010-09-14 05:18:09 +000016#include <ShellBase.h>
jcarsey252d9452011-03-25 20:49:53 +000017#include <Library/SortLib.h>
jcarseyd2b45642009-05-11 18:02:16 +000018
jcarsey94b17fa2009-05-07 18:46:18 +000019#define FIND_XXXXX_FILE_BUFFER_SIZE (SIZE_OF_EFI_FILE_INFO + MAX_FILE_NAME_LEN)
20
jcarseyd2b45642009-05-11 18:02:16 +000021//
jcarseya405b862010-09-14 05:18:09 +000022// globals...
jcarseyd2b45642009-05-11 18:02:16 +000023//
24SHELL_PARAM_ITEM EmptyParamList[] = {
25 {NULL, TypeMax}
26 };
jcarseya405b862010-09-14 05:18:09 +000027SHELL_PARAM_ITEM SfoParamList[] = {
28 {L"-sfo", TypeFlag},
29 {NULL, TypeMax}
30 };
31EFI_SHELL_ENVIRONMENT2 *mEfiShellEnvironment2;
32EFI_SHELL_INTERFACE *mEfiShellInterface;
jcarsey366f81a2011-06-27 21:04:22 +000033EFI_SHELL_PROTOCOL *gEfiShellProtocol;
34EFI_SHELL_PARAMETERS_PROTOCOL *gEfiShellParametersProtocol;
jcarseya405b862010-09-14 05:18:09 +000035EFI_HANDLE mEfiShellEnvironment2Handle;
36FILE_HANDLE_FUNCTION_MAP FileFunctionMap;
jcarseyb3011f42010-01-11 21:49:04 +000037
jcarsey2247dde2009-11-09 18:08:58 +000038/**
39 Check if a Unicode character is a hexadecimal character.
40
jcarsey1e6e84c2010-01-25 20:05:08 +000041 This internal function checks if a Unicode character is a
jcarseya405b862010-09-14 05:18:09 +000042 numeric character. The valid hexadecimal characters are
jcarsey2247dde2009-11-09 18:08:58 +000043 L'0' to L'9', L'a' to L'f', or L'A' to L'F'.
44
jcarsey2247dde2009-11-09 18:08:58 +000045 @param Char The character to check against.
46
47 @retval TRUE If the Char is a hexadecmial character.
48 @retval FALSE If the Char is not a hexadecmial character.
49
50**/
51BOOLEAN
52EFIAPI
jcarsey969c7832010-01-13 16:46:33 +000053ShellIsHexaDecimalDigitCharacter (
jcarsey2247dde2009-11-09 18:08:58 +000054 IN CHAR16 Char
jcarseya405b862010-09-14 05:18:09 +000055 )
56{
jcarsey2247dde2009-11-09 18:08:58 +000057 return (BOOLEAN) ((Char >= L'0' && Char <= L'9') || (Char >= L'A' && Char <= L'F') || (Char >= L'a' && Char <= L'f'));
58}
jcarsey94b17fa2009-05-07 18:46:18 +000059
60/**
jcarseya405b862010-09-14 05:18:09 +000061 Check if a Unicode character is a decimal character.
62
63 This internal function checks if a Unicode character is a
64 decimal character. The valid characters are
65 L'0' to L'9'.
66
67
68 @param Char The character to check against.
69
70 @retval TRUE If the Char is a hexadecmial character.
71 @retval FALSE If the Char is not a hexadecmial character.
72
73**/
74BOOLEAN
75EFIAPI
76ShellIsDecimalDigitCharacter (
77 IN CHAR16 Char
78 )
79{
80 return (BOOLEAN) (Char >= L'0' && Char <= L'9');
81}
82
83/**
84 Helper function to find ShellEnvironment2 for constructor.
85
86 @param[in] ImageHandle A copy of the calling image's handle.
jcarseybeab0fc2011-10-10 17:26:25 +000087
88 @retval EFI_OUT_OF_RESOURCES Memory allocation failed.
jcarsey94b17fa2009-05-07 18:46:18 +000089**/
90EFI_STATUS
91EFIAPI
92ShellFindSE2 (
93 IN EFI_HANDLE ImageHandle
jcarseya405b862010-09-14 05:18:09 +000094 )
95{
jcarsey94b17fa2009-05-07 18:46:18 +000096 EFI_STATUS Status;
97 EFI_HANDLE *Buffer;
98 UINTN BufferSize;
99 UINTN HandleIndex;
100
101 BufferSize = 0;
102 Buffer = NULL;
jcarsey1e6e84c2010-01-25 20:05:08 +0000103 Status = gBS->OpenProtocol(ImageHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000104 &gEfiShellEnvironment2Guid,
105 (VOID **)&mEfiShellEnvironment2,
106 ImageHandle,
107 NULL,
108 EFI_OPEN_PROTOCOL_GET_PROTOCOL
jcarseya405b862010-09-14 05:18:09 +0000109 );
jcarsey94b17fa2009-05-07 18:46:18 +0000110 //
111 // look for the mEfiShellEnvironment2 protocol at a higher level
112 //
jcarseya405b862010-09-14 05:18:09 +0000113 if (EFI_ERROR (Status) || !(CompareGuid (&mEfiShellEnvironment2->SESGuid, &gEfiShellEnvironment2ExtGuid))){
jcarsey94b17fa2009-05-07 18:46:18 +0000114 //
115 // figure out how big of a buffer we need.
116 //
117 Status = gBS->LocateHandle (ByProtocol,
118 &gEfiShellEnvironment2Guid,
119 NULL, // ignored for ByProtocol
120 &BufferSize,
121 Buffer
jcarseya405b862010-09-14 05:18:09 +0000122 );
jcarsey2247dde2009-11-09 18:08:58 +0000123 //
124 // maybe it's not there???
125 //
126 if (Status == EFI_BUFFER_TOO_SMALL) {
jcarsey252d9452011-03-25 20:49:53 +0000127 Buffer = (EFI_HANDLE*)AllocateZeroPool(BufferSize);
jcarseybeab0fc2011-10-10 17:26:25 +0000128 if (Buffer == NULL) {
129 return (EFI_OUT_OF_RESOURCES);
130 }
jcarsey2247dde2009-11-09 18:08:58 +0000131 Status = gBS->LocateHandle (ByProtocol,
132 &gEfiShellEnvironment2Guid,
133 NULL, // ignored for ByProtocol
134 &BufferSize,
135 Buffer
jcarseya405b862010-09-14 05:18:09 +0000136 );
jcarsey2247dde2009-11-09 18:08:58 +0000137 }
jcarsey1cd45e72010-01-29 15:07:44 +0000138 if (!EFI_ERROR (Status) && Buffer != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000139 //
140 // now parse the list of returned handles
141 //
142 Status = EFI_NOT_FOUND;
143 for (HandleIndex = 0; HandleIndex < (BufferSize/sizeof(Buffer[0])); HandleIndex++) {
jcarsey1e6e84c2010-01-25 20:05:08 +0000144 Status = gBS->OpenProtocol(Buffer[HandleIndex],
jcarsey94b17fa2009-05-07 18:46:18 +0000145 &gEfiShellEnvironment2Guid,
146 (VOID **)&mEfiShellEnvironment2,
147 ImageHandle,
148 NULL,
149 EFI_OPEN_PROTOCOL_GET_PROTOCOL
jcarseya405b862010-09-14 05:18:09 +0000150 );
151 if (CompareGuid (&mEfiShellEnvironment2->SESGuid, &gEfiShellEnvironment2ExtGuid)) {
jcarsey94b17fa2009-05-07 18:46:18 +0000152 mEfiShellEnvironment2Handle = Buffer[HandleIndex];
153 Status = EFI_SUCCESS;
154 break;
155 }
156 }
157 }
158 }
159 if (Buffer != NULL) {
160 FreePool (Buffer);
161 }
162 return (Status);
163}
164
jcarsey252d9452011-03-25 20:49:53 +0000165/**
darylm503b0934ac2011-11-12 00:35:11 +0000166 Function to do most of the work of the constructor. Allows for calling
jcarseya405b862010-09-14 05:18:09 +0000167 multiple times without complete re-initialization.
168
169 @param[in] ImageHandle A copy of the ImageHandle.
170 @param[in] SystemTable A pointer to the SystemTable for the application.
jcarsey252d9452011-03-25 20:49:53 +0000171
172 @retval EFI_SUCCESS The operationw as successful.
jcarseya405b862010-09-14 05:18:09 +0000173**/
jcarsey94b17fa2009-05-07 18:46:18 +0000174EFI_STATUS
175EFIAPI
jcarseyd2b45642009-05-11 18:02:16 +0000176ShellLibConstructorWorker (
jcarsey94b17fa2009-05-07 18:46:18 +0000177 IN EFI_HANDLE ImageHandle,
178 IN EFI_SYSTEM_TABLE *SystemTable
jcarseya405b862010-09-14 05:18:09 +0000179 )
180{
181 EFI_STATUS Status;
jcarseyecd3d592009-12-07 18:05:00 +0000182
jcarsey94b17fa2009-05-07 18:46:18 +0000183 //
184 // UEFI 2.0 shell interfaces (used preferentially)
185 //
jcarseya405b862010-09-14 05:18:09 +0000186 Status = gBS->OpenProtocol(
187 ImageHandle,
188 &gEfiShellProtocolGuid,
jcarsey366f81a2011-06-27 21:04:22 +0000189 (VOID **)&gEfiShellProtocol,
jcarseya405b862010-09-14 05:18:09 +0000190 ImageHandle,
191 NULL,
192 EFI_OPEN_PROTOCOL_GET_PROTOCOL
193 );
jcarsey94b17fa2009-05-07 18:46:18 +0000194 if (EFI_ERROR(Status)) {
jcarseya405b862010-09-14 05:18:09 +0000195 //
196 // Search for the shell protocol
197 //
198 Status = gBS->LocateProtocol(
199 &gEfiShellProtocolGuid,
200 NULL,
jcarsey366f81a2011-06-27 21:04:22 +0000201 (VOID **)&gEfiShellProtocol
jcarseya405b862010-09-14 05:18:09 +0000202 );
203 if (EFI_ERROR(Status)) {
jcarsey366f81a2011-06-27 21:04:22 +0000204 gEfiShellProtocol = NULL;
jcarseya405b862010-09-14 05:18:09 +0000205 }
jcarsey94b17fa2009-05-07 18:46:18 +0000206 }
jcarseya405b862010-09-14 05:18:09 +0000207 Status = gBS->OpenProtocol(
208 ImageHandle,
209 &gEfiShellParametersProtocolGuid,
jcarsey366f81a2011-06-27 21:04:22 +0000210 (VOID **)&gEfiShellParametersProtocol,
jcarseya405b862010-09-14 05:18:09 +0000211 ImageHandle,
212 NULL,
213 EFI_OPEN_PROTOCOL_GET_PROTOCOL
214 );
jcarsey94b17fa2009-05-07 18:46:18 +0000215 if (EFI_ERROR(Status)) {
jcarsey366f81a2011-06-27 21:04:22 +0000216 gEfiShellParametersProtocol = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000217 }
218
jcarsey366f81a2011-06-27 21:04:22 +0000219 if (gEfiShellParametersProtocol == NULL || gEfiShellProtocol == NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000220 //
221 // Moved to seperate function due to complexity
222 //
223 Status = ShellFindSE2(ImageHandle);
224
225 if (EFI_ERROR(Status)) {
226 DEBUG((DEBUG_ERROR, "Status: 0x%08x\r\n", Status));
227 mEfiShellEnvironment2 = NULL;
228 }
jcarsey1e6e84c2010-01-25 20:05:08 +0000229 Status = gBS->OpenProtocol(ImageHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000230 &gEfiShellInterfaceGuid,
231 (VOID **)&mEfiShellInterface,
232 ImageHandle,
233 NULL,
234 EFI_OPEN_PROTOCOL_GET_PROTOCOL
jcarseya405b862010-09-14 05:18:09 +0000235 );
jcarsey94b17fa2009-05-07 18:46:18 +0000236 if (EFI_ERROR(Status)) {
237 mEfiShellInterface = NULL;
238 }
239 }
jcarseyc9d92df2010-02-03 15:37:54 +0000240
jcarsey94b17fa2009-05-07 18:46:18 +0000241 //
242 // only success getting 2 of either the old or new, but no 1/2 and 1/2
243 //
jcarsey1e6e84c2010-01-25 20:05:08 +0000244 if ((mEfiShellEnvironment2 != NULL && mEfiShellInterface != NULL) ||
jcarsey366f81a2011-06-27 21:04:22 +0000245 (gEfiShellProtocol != NULL && gEfiShellParametersProtocol != NULL) ) {
246 if (gEfiShellProtocol != NULL) {
247 FileFunctionMap.GetFileInfo = gEfiShellProtocol->GetFileInfo;
248 FileFunctionMap.SetFileInfo = gEfiShellProtocol->SetFileInfo;
249 FileFunctionMap.ReadFile = gEfiShellProtocol->ReadFile;
250 FileFunctionMap.WriteFile = gEfiShellProtocol->WriteFile;
251 FileFunctionMap.CloseFile = gEfiShellProtocol->CloseFile;
252 FileFunctionMap.DeleteFile = gEfiShellProtocol->DeleteFile;
253 FileFunctionMap.GetFilePosition = gEfiShellProtocol->GetFilePosition;
254 FileFunctionMap.SetFilePosition = gEfiShellProtocol->SetFilePosition;
255 FileFunctionMap.FlushFile = gEfiShellProtocol->FlushFile;
256 FileFunctionMap.GetFileSize = gEfiShellProtocol->GetFileSize;
jcarseyd2b45642009-05-11 18:02:16 +0000257 } else {
jcarseya405b862010-09-14 05:18:09 +0000258 FileFunctionMap.GetFileInfo = (EFI_SHELL_GET_FILE_INFO)FileHandleGetInfo;
259 FileFunctionMap.SetFileInfo = (EFI_SHELL_SET_FILE_INFO)FileHandleSetInfo;
260 FileFunctionMap.ReadFile = (EFI_SHELL_READ_FILE)FileHandleRead;
261 FileFunctionMap.WriteFile = (EFI_SHELL_WRITE_FILE)FileHandleWrite;
262 FileFunctionMap.CloseFile = (EFI_SHELL_CLOSE_FILE)FileHandleClose;
263 FileFunctionMap.DeleteFile = (EFI_SHELL_DELETE_FILE)FileHandleDelete;
264 FileFunctionMap.GetFilePosition = (EFI_SHELL_GET_FILE_POSITION)FileHandleGetPosition;
265 FileFunctionMap.SetFilePosition = (EFI_SHELL_SET_FILE_POSITION)FileHandleSetPosition;
266 FileFunctionMap.FlushFile = (EFI_SHELL_FLUSH_FILE)FileHandleFlush;
267 FileFunctionMap.GetFileSize = (EFI_SHELL_GET_FILE_SIZE)FileHandleGetSize;
jcarseyd2b45642009-05-11 18:02:16 +0000268 }
jcarsey94b17fa2009-05-07 18:46:18 +0000269 return (EFI_SUCCESS);
270 }
271 return (EFI_NOT_FOUND);
272}
jcarseyd2b45642009-05-11 18:02:16 +0000273/**
274 Constructor for the Shell library.
275
276 Initialize the library and determine if the underlying is a UEFI Shell 2.0 or an EFI shell.
277
278 @param ImageHandle the image handle of the process
279 @param SystemTable the EFI System Table pointer
280
281 @retval EFI_SUCCESS the initialization was complete sucessfully
282 @return others an error ocurred during initialization
283**/
284EFI_STATUS
285EFIAPI
286ShellLibConstructor (
287 IN EFI_HANDLE ImageHandle,
288 IN EFI_SYSTEM_TABLE *SystemTable
jcarseya405b862010-09-14 05:18:09 +0000289 )
290{
jcarseyd2b45642009-05-11 18:02:16 +0000291 mEfiShellEnvironment2 = NULL;
jcarsey366f81a2011-06-27 21:04:22 +0000292 gEfiShellProtocol = NULL;
293 gEfiShellParametersProtocol = NULL;
jcarseyd2b45642009-05-11 18:02:16 +0000294 mEfiShellInterface = NULL;
295 mEfiShellEnvironment2Handle = NULL;
296
jcarseyd2b45642009-05-11 18:02:16 +0000297 //
298 // verify that auto initialize is not set false
jcarsey1e6e84c2010-01-25 20:05:08 +0000299 //
jcarseyd2b45642009-05-11 18:02:16 +0000300 if (PcdGetBool(PcdShellLibAutoInitialize) == 0) {
301 return (EFI_SUCCESS);
302 }
jcarsey1e6e84c2010-01-25 20:05:08 +0000303
jcarseyd2b45642009-05-11 18:02:16 +0000304 return (ShellLibConstructorWorker(ImageHandle, SystemTable));
305}
jcarsey94b17fa2009-05-07 18:46:18 +0000306
307/**
jcarseya405b862010-09-14 05:18:09 +0000308 Destructor for the library. free any resources.
309
310 @param[in] ImageHandle A copy of the ImageHandle.
311 @param[in] SystemTable A pointer to the SystemTable for the application.
312
313 @retval EFI_SUCCESS The operation was successful.
314 @return An error from the CloseProtocol function.
jcarsey94b17fa2009-05-07 18:46:18 +0000315**/
316EFI_STATUS
317EFIAPI
318ShellLibDestructor (
319 IN EFI_HANDLE ImageHandle,
320 IN EFI_SYSTEM_TABLE *SystemTable
jcarseya405b862010-09-14 05:18:09 +0000321 )
322{
jcarsey94b17fa2009-05-07 18:46:18 +0000323 if (mEfiShellEnvironment2 != NULL) {
324 gBS->CloseProtocol(mEfiShellEnvironment2Handle==NULL?ImageHandle:mEfiShellEnvironment2Handle,
325 &gEfiShellEnvironment2Guid,
326 ImageHandle,
327 NULL);
jcarseyd2b45642009-05-11 18:02:16 +0000328 mEfiShellEnvironment2 = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000329 }
330 if (mEfiShellInterface != NULL) {
331 gBS->CloseProtocol(ImageHandle,
332 &gEfiShellInterfaceGuid,
333 ImageHandle,
jcarsey1e6e84c2010-01-25 20:05:08 +0000334 NULL);
jcarseyd2b45642009-05-11 18:02:16 +0000335 mEfiShellInterface = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000336 }
jcarsey366f81a2011-06-27 21:04:22 +0000337 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000338 gBS->CloseProtocol(ImageHandle,
339 &gEfiShellProtocolGuid,
340 ImageHandle,
jcarsey1e6e84c2010-01-25 20:05:08 +0000341 NULL);
jcarsey366f81a2011-06-27 21:04:22 +0000342 gEfiShellProtocol = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000343 }
jcarsey366f81a2011-06-27 21:04:22 +0000344 if (gEfiShellParametersProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000345 gBS->CloseProtocol(ImageHandle,
346 &gEfiShellParametersProtocolGuid,
347 ImageHandle,
jcarsey1e6e84c2010-01-25 20:05:08 +0000348 NULL);
jcarsey366f81a2011-06-27 21:04:22 +0000349 gEfiShellParametersProtocol = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000350 }
jcarseyd2b45642009-05-11 18:02:16 +0000351 mEfiShellEnvironment2Handle = NULL;
jcarseyecd3d592009-12-07 18:05:00 +0000352
jcarsey94b17fa2009-05-07 18:46:18 +0000353 return (EFI_SUCCESS);
354}
jcarseyd2b45642009-05-11 18:02:16 +0000355
356/**
357 This function causes the shell library to initialize itself. If the shell library
358 is already initialized it will de-initialize all the current protocol poitners and
359 re-populate them again.
360
361 When the library is used with PcdShellLibAutoInitialize set to true this function
362 will return EFI_SUCCESS and perform no actions.
363
364 This function is intended for internal access for shell commands only.
365
366 @retval EFI_SUCCESS the initialization was complete sucessfully
367
368**/
369EFI_STATUS
370EFIAPI
371ShellInitialize (
jcarseya405b862010-09-14 05:18:09 +0000372 )
373{
jcarseyd2b45642009-05-11 18:02:16 +0000374 //
375 // if auto initialize is not false then skip
376 //
377 if (PcdGetBool(PcdShellLibAutoInitialize) != 0) {
378 return (EFI_SUCCESS);
379 }
380
381 //
382 // deinit the current stuff
383 //
384 ASSERT_EFI_ERROR(ShellLibDestructor(gImageHandle, gST));
385
386 //
387 // init the new stuff
388 //
389 return (ShellLibConstructorWorker(gImageHandle, gST));
390}
391
jcarsey94b17fa2009-05-07 18:46:18 +0000392/**
jcarsey1e6e84c2010-01-25 20:05:08 +0000393 This function will retrieve the information about the file for the handle
jcarsey94b17fa2009-05-07 18:46:18 +0000394 specified and store it in allocated pool memory.
395
jcarsey1e6e84c2010-01-25 20:05:08 +0000396 This function allocates a buffer to store the file's information. It is the
qhuang869817bf2009-05-20 14:42:48 +0000397 caller's responsibility to free the buffer
jcarsey94b17fa2009-05-07 18:46:18 +0000398
jcarsey1e6e84c2010-01-25 20:05:08 +0000399 @param FileHandle The file handle of the file for which information is
jcarsey94b17fa2009-05-07 18:46:18 +0000400 being requested.
401
402 @retval NULL information could not be retrieved.
403
404 @return the information about the file
405**/
406EFI_FILE_INFO*
407EFIAPI
408ShellGetFileInfo (
jcarseya405b862010-09-14 05:18:09 +0000409 IN SHELL_FILE_HANDLE FileHandle
410 )
411{
jcarseyd2b45642009-05-11 18:02:16 +0000412 return (FileFunctionMap.GetFileInfo(FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000413}
414
415/**
jcarseya405b862010-09-14 05:18:09 +0000416 This function sets the information about the file for the opened handle
jcarsey94b17fa2009-05-07 18:46:18 +0000417 specified.
418
jcarseya405b862010-09-14 05:18:09 +0000419 @param[in] FileHandle The file handle of the file for which information
420 is being set.
jcarsey94b17fa2009-05-07 18:46:18 +0000421
jcarseya405b862010-09-14 05:18:09 +0000422 @param[in] FileInfo The information to set.
jcarsey94b17fa2009-05-07 18:46:18 +0000423
darylm503b0934ac2011-11-12 00:35:11 +0000424 @retval EFI_SUCCESS The information was set.
jcarseya405b862010-09-14 05:18:09 +0000425 @retval EFI_INVALID_PARAMETER A parameter was out of range or invalid.
426 @retval EFI_UNSUPPORTED The FileHandle does not support FileInfo.
darylm503b0934ac2011-11-12 00:35:11 +0000427 @retval EFI_NO_MEDIA The device has no medium.
428 @retval EFI_DEVICE_ERROR The device reported an error.
429 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
430 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
jcarseya405b862010-09-14 05:18:09 +0000431 @retval EFI_ACCESS_DENIED The file was opened read only.
432 @retval EFI_VOLUME_FULL The volume is full.
jcarsey94b17fa2009-05-07 18:46:18 +0000433**/
434EFI_STATUS
435EFIAPI
436ShellSetFileInfo (
darylm503b0934ac2011-11-12 00:35:11 +0000437 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000438 IN EFI_FILE_INFO *FileInfo
jcarseya405b862010-09-14 05:18:09 +0000439 )
440{
jcarseyd2b45642009-05-11 18:02:16 +0000441 return (FileFunctionMap.SetFileInfo(FileHandle, FileInfo));
jcarsey1e6e84c2010-01-25 20:05:08 +0000442}
443
jcarsey94b17fa2009-05-07 18:46:18 +0000444 /**
445 This function will open a file or directory referenced by DevicePath.
446
jcarsey1e6e84c2010-01-25 20:05:08 +0000447 This function opens a file with the open mode according to the file path. The
jcarsey94b17fa2009-05-07 18:46:18 +0000448 Attributes is valid only for EFI_FILE_MODE_CREATE.
449
darylm503b0934ac2011-11-12 00:35:11 +0000450 @param FilePath on input the device path to the file. On output
jcarsey94b17fa2009-05-07 18:46:18 +0000451 the remaining device path.
darylm503b0934ac2011-11-12 00:35:11 +0000452 @param DeviceHandle pointer to the system device handle.
453 @param FileHandle pointer to the file handle.
454 @param OpenMode the mode to open the file with.
455 @param Attributes the file's file attributes.
jcarsey94b17fa2009-05-07 18:46:18 +0000456
darylm503b0934ac2011-11-12 00:35:11 +0000457 @retval EFI_SUCCESS The information was set.
458 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
459 @retval EFI_UNSUPPORTED Could not open the file path.
460 @retval EFI_NOT_FOUND The specified file could not be found on the
jcarsey1e6e84c2010-01-25 20:05:08 +0000461 device or the file system could not be found on
jcarsey94b17fa2009-05-07 18:46:18 +0000462 the device.
darylm503b0934ac2011-11-12 00:35:11 +0000463 @retval EFI_NO_MEDIA The device has no medium.
464 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
jcarsey94b17fa2009-05-07 18:46:18 +0000465 medium is no longer supported.
darylm503b0934ac2011-11-12 00:35:11 +0000466 @retval EFI_DEVICE_ERROR The device reported an error.
467 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
468 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
469 @retval EFI_ACCESS_DENIED The file was opened read only.
470 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
jcarsey94b17fa2009-05-07 18:46:18 +0000471 file.
darylm503b0934ac2011-11-12 00:35:11 +0000472 @retval EFI_VOLUME_FULL The volume is full.
jcarsey94b17fa2009-05-07 18:46:18 +0000473**/
474EFI_STATUS
475EFIAPI
476ShellOpenFileByDevicePath(
darylm503b0934ac2011-11-12 00:35:11 +0000477 IN OUT EFI_DEVICE_PATH_PROTOCOL **FilePath,
478 OUT EFI_HANDLE *DeviceHandle,
jcarseya405b862010-09-14 05:18:09 +0000479 OUT SHELL_FILE_HANDLE *FileHandle,
darylm503b0934ac2011-11-12 00:35:11 +0000480 IN UINT64 OpenMode,
481 IN UINT64 Attributes
jcarseya405b862010-09-14 05:18:09 +0000482 )
483{
484 CHAR16 *FileName;
485 EFI_STATUS Status;
jcarsey94b17fa2009-05-07 18:46:18 +0000486 EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *EfiSimpleFileSystemProtocol;
jcarseya405b862010-09-14 05:18:09 +0000487 EFI_FILE_PROTOCOL *Handle1;
488 EFI_FILE_PROTOCOL *Handle2;
jcarsey94b17fa2009-05-07 18:46:18 +0000489
jcarsey92a54472011-06-27 20:33:13 +0000490 if (FilePath == NULL || FileHandle == NULL || DeviceHandle == NULL) {
491 return (EFI_INVALID_PARAMETER);
492 }
493
jcarsey1e6e84c2010-01-25 20:05:08 +0000494 //
jcarsey94b17fa2009-05-07 18:46:18 +0000495 // which shell interface should we use
496 //
jcarsey366f81a2011-06-27 21:04:22 +0000497 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000498 //
499 // use UEFI Shell 2.0 method.
500 //
jcarsey366f81a2011-06-27 21:04:22 +0000501 FileName = gEfiShellProtocol->GetFilePathFromDevicePath(*FilePath);
jcarsey94b17fa2009-05-07 18:46:18 +0000502 if (FileName == NULL) {
503 return (EFI_INVALID_PARAMETER);
504 }
505 Status = ShellOpenFileByName(FileName, FileHandle, OpenMode, Attributes);
506 FreePool(FileName);
507 return (Status);
jcarsey1e6e84c2010-01-25 20:05:08 +0000508 }
jcarseyd2b45642009-05-11 18:02:16 +0000509
510
511 //
512 // use old shell method.
513 //
jcarsey1e6e84c2010-01-25 20:05:08 +0000514 Status = gBS->LocateDevicePath (&gEfiSimpleFileSystemProtocolGuid,
515 FilePath,
jcarseyd2b45642009-05-11 18:02:16 +0000516 DeviceHandle);
517 if (EFI_ERROR (Status)) {
518 return Status;
519 }
520 Status = gBS->OpenProtocol(*DeviceHandle,
521 &gEfiSimpleFileSystemProtocolGuid,
jcarseyb1f95a02009-06-16 00:23:19 +0000522 (VOID**)&EfiSimpleFileSystemProtocol,
jcarseyd2b45642009-05-11 18:02:16 +0000523 gImageHandle,
524 NULL,
525 EFI_OPEN_PROTOCOL_GET_PROTOCOL);
526 if (EFI_ERROR (Status)) {
527 return Status;
528 }
jcarseya405b862010-09-14 05:18:09 +0000529 Status = EfiSimpleFileSystemProtocol->OpenVolume(EfiSimpleFileSystemProtocol, &Handle1);
jcarseyd2b45642009-05-11 18:02:16 +0000530 if (EFI_ERROR (Status)) {
531 FileHandle = NULL;
532 return Status;
533 }
534
535 //
536 // go down directories one node at a time.
537 //
538 while (!IsDevicePathEnd (*FilePath)) {
jcarsey94b17fa2009-05-07 18:46:18 +0000539 //
jcarseyd2b45642009-05-11 18:02:16 +0000540 // For file system access each node should be a file path component
jcarsey94b17fa2009-05-07 18:46:18 +0000541 //
jcarseyd2b45642009-05-11 18:02:16 +0000542 if (DevicePathType (*FilePath) != MEDIA_DEVICE_PATH ||
543 DevicePathSubType (*FilePath) != MEDIA_FILEPATH_DP
jcarseya405b862010-09-14 05:18:09 +0000544 ) {
jcarsey94b17fa2009-05-07 18:46:18 +0000545 FileHandle = NULL;
jcarseyd2b45642009-05-11 18:02:16 +0000546 return (EFI_INVALID_PARAMETER);
jcarsey94b17fa2009-05-07 18:46:18 +0000547 }
jcarseyd2b45642009-05-11 18:02:16 +0000548 //
549 // Open this file path node
550 //
jcarseya405b862010-09-14 05:18:09 +0000551 Handle2 = Handle1;
552 Handle1 = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +0000553
554 //
jcarseyd2b45642009-05-11 18:02:16 +0000555 // Try to test opening an existing file
jcarsey94b17fa2009-05-07 18:46:18 +0000556 //
jcarseya405b862010-09-14 05:18:09 +0000557 Status = Handle2->Open (
558 Handle2,
559 &Handle1,
jcarseyd2b45642009-05-11 18:02:16 +0000560 ((FILEPATH_DEVICE_PATH*)*FilePath)->PathName,
561 OpenMode &~EFI_FILE_MODE_CREATE,
562 0
jcarseya405b862010-09-14 05:18:09 +0000563 );
jcarsey94b17fa2009-05-07 18:46:18 +0000564
jcarseyd2b45642009-05-11 18:02:16 +0000565 //
566 // see if the error was that it needs to be created
567 //
568 if ((EFI_ERROR (Status)) && (OpenMode != (OpenMode &~EFI_FILE_MODE_CREATE))) {
jcarseya405b862010-09-14 05:18:09 +0000569 Status = Handle2->Open (
570 Handle2,
571 &Handle1,
jcarsey94b17fa2009-05-07 18:46:18 +0000572 ((FILEPATH_DEVICE_PATH*)*FilePath)->PathName,
jcarseyd2b45642009-05-11 18:02:16 +0000573 OpenMode,
574 Attributes
jcarseya405b862010-09-14 05:18:09 +0000575 );
jcarsey94b17fa2009-05-07 18:46:18 +0000576 }
jcarseyd2b45642009-05-11 18:02:16 +0000577 //
578 // Close the last node
579 //
jcarseya405b862010-09-14 05:18:09 +0000580 Handle2->Close (Handle2);
jcarseyd2b45642009-05-11 18:02:16 +0000581
582 if (EFI_ERROR(Status)) {
583 return (Status);
584 }
585
586 //
587 // Get the next node
588 //
589 *FilePath = NextDevicePathNode (*FilePath);
jcarsey94b17fa2009-05-07 18:46:18 +0000590 }
jcarseya405b862010-09-14 05:18:09 +0000591
592 //
593 // This is a weak spot since if the undefined SHELL_FILE_HANDLE format changes this must change also!
594 //
595 *FileHandle = (VOID*)Handle1;
jcarseyd2b45642009-05-11 18:02:16 +0000596 return (EFI_SUCCESS);
jcarsey94b17fa2009-05-07 18:46:18 +0000597}
598
599/**
600 This function will open a file or directory referenced by filename.
601
jcarsey1e6e84c2010-01-25 20:05:08 +0000602 If return is EFI_SUCCESS, the Filehandle is the opened file's handle;
603 otherwise, the Filehandle is NULL. The Attributes is valid only for
jcarsey94b17fa2009-05-07 18:46:18 +0000604 EFI_FILE_MODE_CREATE.
605
jcarsey92a54472011-06-27 20:33:13 +0000606 if FileName is NULL then ASSERT()
jcarsey94b17fa2009-05-07 18:46:18 +0000607
darylm503b0934ac2011-11-12 00:35:11 +0000608 @param FileName pointer to file name
609 @param FileHandle pointer to the file handle.
610 @param OpenMode the mode to open the file with.
611 @param Attributes the file's file attributes.
jcarsey94b17fa2009-05-07 18:46:18 +0000612
darylm503b0934ac2011-11-12 00:35:11 +0000613 @retval EFI_SUCCESS The information was set.
614 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
615 @retval EFI_UNSUPPORTED Could not open the file path.
616 @retval EFI_NOT_FOUND The specified file could not be found on the
jcarsey1e6e84c2010-01-25 20:05:08 +0000617 device or the file system could not be found
jcarsey94b17fa2009-05-07 18:46:18 +0000618 on the device.
darylm503b0934ac2011-11-12 00:35:11 +0000619 @retval EFI_NO_MEDIA The device has no medium.
620 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
jcarsey94b17fa2009-05-07 18:46:18 +0000621 medium is no longer supported.
darylm503b0934ac2011-11-12 00:35:11 +0000622 @retval EFI_DEVICE_ERROR The device reported an error.
623 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
624 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
625 @retval EFI_ACCESS_DENIED The file was opened read only.
626 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
jcarsey94b17fa2009-05-07 18:46:18 +0000627 file.
darylm503b0934ac2011-11-12 00:35:11 +0000628 @retval EFI_VOLUME_FULL The volume is full.
jcarsey94b17fa2009-05-07 18:46:18 +0000629**/
630EFI_STATUS
631EFIAPI
632ShellOpenFileByName(
darylm503b0934ac2011-11-12 00:35:11 +0000633 IN CONST CHAR16 *FileName,
jcarseya405b862010-09-14 05:18:09 +0000634 OUT SHELL_FILE_HANDLE *FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000635 IN UINT64 OpenMode,
darylm503b0934ac2011-11-12 00:35:11 +0000636 IN UINT64 Attributes
jcarseya405b862010-09-14 05:18:09 +0000637 )
638{
jcarsey94b17fa2009-05-07 18:46:18 +0000639 EFI_HANDLE DeviceHandle;
640 EFI_DEVICE_PATH_PROTOCOL *FilePath;
jcarseyb1f95a02009-06-16 00:23:19 +0000641 EFI_STATUS Status;
642 EFI_FILE_INFO *FileInfo;
jcarsey94b17fa2009-05-07 18:46:18 +0000643
644 //
645 // ASSERT if FileName is NULL
646 //
647 ASSERT(FileName != NULL);
648
jcarseya405b862010-09-14 05:18:09 +0000649 if (FileName == NULL) {
650 return (EFI_INVALID_PARAMETER);
651 }
652
jcarsey366f81a2011-06-27 21:04:22 +0000653 if (gEfiShellProtocol != NULL) {
jcarseya405b862010-09-14 05:18:09 +0000654 if ((OpenMode & EFI_FILE_MODE_CREATE) == EFI_FILE_MODE_CREATE && (Attributes & EFI_FILE_DIRECTORY) == EFI_FILE_DIRECTORY) {
655 return ShellCreateDirectory(FileName, FileHandle);
656 }
jcarsey94b17fa2009-05-07 18:46:18 +0000657 //
658 // Use UEFI Shell 2.0 method
659 //
jcarsey366f81a2011-06-27 21:04:22 +0000660 Status = gEfiShellProtocol->OpenFileByName(FileName,
jcarseyb1f95a02009-06-16 00:23:19 +0000661 FileHandle,
662 OpenMode);
jcarseya405b862010-09-14 05:18:09 +0000663 if (StrCmp(FileName, L"NUL") != 0 && !EFI_ERROR(Status) && ((OpenMode & EFI_FILE_MODE_CREATE) != 0)){
jcarsey2247dde2009-11-09 18:08:58 +0000664 FileInfo = FileFunctionMap.GetFileInfo(*FileHandle);
jcarseyb1f95a02009-06-16 00:23:19 +0000665 ASSERT(FileInfo != NULL);
666 FileInfo->Attribute = Attributes;
jcarsey2247dde2009-11-09 18:08:58 +0000667 Status = FileFunctionMap.SetFileInfo(*FileHandle, FileInfo);
668 FreePool(FileInfo);
jcarseyb1f95a02009-06-16 00:23:19 +0000669 }
670 return (Status);
jcarsey1e6e84c2010-01-25 20:05:08 +0000671 }
jcarsey94b17fa2009-05-07 18:46:18 +0000672 //
673 // Using EFI Shell version
674 // this means convert name to path and call that function
675 // since this will use EFI method again that will open it.
676 //
677 ASSERT(mEfiShellEnvironment2 != NULL);
jcarseyb82bfcc2009-06-29 16:28:23 +0000678 FilePath = mEfiShellEnvironment2->NameToPath ((CHAR16*)FileName);
xdu290bfa222010-07-19 05:21:27 +0000679 if (FilePath != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +0000680 return (ShellOpenFileByDevicePath(&FilePath,
681 &DeviceHandle,
682 FileHandle,
683 OpenMode,
jcarseya405b862010-09-14 05:18:09 +0000684 Attributes));
jcarsey94b17fa2009-05-07 18:46:18 +0000685 }
686 return (EFI_DEVICE_ERROR);
687}
688/**
689 This function create a directory
690
jcarsey1e6e84c2010-01-25 20:05:08 +0000691 If return is EFI_SUCCESS, the Filehandle is the opened directory's handle;
692 otherwise, the Filehandle is NULL. If the directory already existed, this
jcarsey94b17fa2009-05-07 18:46:18 +0000693 function opens the existing directory.
694
darylm503b0934ac2011-11-12 00:35:11 +0000695 @param DirectoryName pointer to directory name
696 @param FileHandle pointer to the file handle.
jcarsey94b17fa2009-05-07 18:46:18 +0000697
darylm503b0934ac2011-11-12 00:35:11 +0000698 @retval EFI_SUCCESS The information was set.
699 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
700 @retval EFI_UNSUPPORTED Could not open the file path.
701 @retval EFI_NOT_FOUND The specified file could not be found on the
jcarsey1e6e84c2010-01-25 20:05:08 +0000702 device or the file system could not be found
jcarsey94b17fa2009-05-07 18:46:18 +0000703 on the device.
darylm503b0934ac2011-11-12 00:35:11 +0000704 @retval EFI_NO_MEDIA The device has no medium.
705 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
jcarsey94b17fa2009-05-07 18:46:18 +0000706 medium is no longer supported.
darylm503b0934ac2011-11-12 00:35:11 +0000707 @retval EFI_DEVICE_ERROR The device reported an error.
708 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
709 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
710 @retval EFI_ACCESS_DENIED The file was opened read only.
711 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
jcarsey94b17fa2009-05-07 18:46:18 +0000712 file.
darylm503b0934ac2011-11-12 00:35:11 +0000713 @retval EFI_VOLUME_FULL The volume is full.
jcarsey94b17fa2009-05-07 18:46:18 +0000714 @sa ShellOpenFileByName
715**/
716EFI_STATUS
717EFIAPI
718ShellCreateDirectory(
jcarseyb82bfcc2009-06-29 16:28:23 +0000719 IN CONST CHAR16 *DirectoryName,
jcarseya405b862010-09-14 05:18:09 +0000720 OUT SHELL_FILE_HANDLE *FileHandle
721 )
722{
jcarsey366f81a2011-06-27 21:04:22 +0000723 if (gEfiShellProtocol != NULL) {
jcarsey2247dde2009-11-09 18:08:58 +0000724 //
725 // Use UEFI Shell 2.0 method
726 //
jcarsey366f81a2011-06-27 21:04:22 +0000727 return (gEfiShellProtocol->CreateFile(DirectoryName,
jcarsey2247dde2009-11-09 18:08:58 +0000728 EFI_FILE_DIRECTORY,
729 FileHandle
jcarseya405b862010-09-14 05:18:09 +0000730 ));
jcarsey2247dde2009-11-09 18:08:58 +0000731 } else {
732 return (ShellOpenFileByName(DirectoryName,
733 FileHandle,
734 EFI_FILE_MODE_READ | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE,
735 EFI_FILE_DIRECTORY
jcarseya405b862010-09-14 05:18:09 +0000736 ));
jcarsey2247dde2009-11-09 18:08:58 +0000737 }
jcarsey94b17fa2009-05-07 18:46:18 +0000738}
739
740/**
741 This function reads information from an opened file.
742
jcarsey1e6e84c2010-01-25 20:05:08 +0000743 If FileHandle is not a directory, the function reads the requested number of
744 bytes from the file at the file's current position and returns them in Buffer.
jcarsey94b17fa2009-05-07 18:46:18 +0000745 If the read goes beyond the end of the file, the read length is truncated to the
jcarsey1e6e84c2010-01-25 20:05:08 +0000746 end of the file. The file's current position is increased by the number of bytes
747 returned. If FileHandle is a directory, the function reads the directory entry
748 at the file's current position and returns the entry in Buffer. If the Buffer
749 is not large enough to hold the current directory entry, then
750 EFI_BUFFER_TOO_SMALL is returned and the current file position is not updated.
751 BufferSize is set to be the size of the buffer needed to read the entry. On
752 success, the current position is updated to the next directory entry. If there
753 are no more directory entries, the read returns a zero-length buffer.
jcarsey94b17fa2009-05-07 18:46:18 +0000754 EFI_FILE_INFO is the structure returned as the directory entry.
755
756 @param FileHandle the opened file handle
jcarsey1e6e84c2010-01-25 20:05:08 +0000757 @param BufferSize on input the size of buffer in bytes. on return
jcarsey94b17fa2009-05-07 18:46:18 +0000758 the number of bytes written.
759 @param Buffer the buffer to put read data into.
760
darylm503b0934ac2011-11-12 00:35:11 +0000761 @retval EFI_SUCCESS Data was read.
762 @retval EFI_NO_MEDIA The device has no media.
763 @retval EFI_DEVICE_ERROR The device reported an error.
764 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
765 @retval EFI_BUFFER_TO_SMALL Buffer is too small. ReadSize contains required
jcarsey94b17fa2009-05-07 18:46:18 +0000766 size.
767
768**/
769EFI_STATUS
770EFIAPI
771ShellReadFile(
jcarseya405b862010-09-14 05:18:09 +0000772 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000773 IN OUT UINTN *BufferSize,
774 OUT VOID *Buffer
jcarseya405b862010-09-14 05:18:09 +0000775 )
776{
jcarseyd2b45642009-05-11 18:02:16 +0000777 return (FileFunctionMap.ReadFile(FileHandle, BufferSize, Buffer));
jcarsey94b17fa2009-05-07 18:46:18 +0000778}
779
780
781/**
782 Write data to a file.
783
jcarsey1e6e84c2010-01-25 20:05:08 +0000784 This function writes the specified number of bytes to the file at the current
785 file position. The current file position is advanced the actual number of bytes
786 written, which is returned in BufferSize. Partial writes only occur when there
787 has been a data error during the write attempt (such as "volume space full").
788 The file is automatically grown to hold the data if required. Direct writes to
jcarsey94b17fa2009-05-07 18:46:18 +0000789 opened directories are not supported.
790
791 @param FileHandle The opened file for writing
792 @param BufferSize on input the number of bytes in Buffer. On output
793 the number of bytes written.
794 @param Buffer the buffer containing data to write is stored.
795
darylm503b0934ac2011-11-12 00:35:11 +0000796 @retval EFI_SUCCESS Data was written.
797 @retval EFI_UNSUPPORTED Writes to an open directory are not supported.
798 @retval EFI_NO_MEDIA The device has no media.
799 @retval EFI_DEVICE_ERROR The device reported an error.
800 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
801 @retval EFI_WRITE_PROTECTED The device is write-protected.
802 @retval EFI_ACCESS_DENIED The file was open for read only.
803 @retval EFI_VOLUME_FULL The volume is full.
jcarsey94b17fa2009-05-07 18:46:18 +0000804**/
805EFI_STATUS
806EFIAPI
807ShellWriteFile(
jcarsey252d9452011-03-25 20:49:53 +0000808 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000809 IN OUT UINTN *BufferSize,
810 IN VOID *Buffer
jcarseya405b862010-09-14 05:18:09 +0000811 )
812{
jcarseyd2b45642009-05-11 18:02:16 +0000813 return (FileFunctionMap.WriteFile(FileHandle, BufferSize, Buffer));
jcarsey94b17fa2009-05-07 18:46:18 +0000814}
815
jcarsey1e6e84c2010-01-25 20:05:08 +0000816/**
jcarsey94b17fa2009-05-07 18:46:18 +0000817 Close an open file handle.
818
jcarsey1e6e84c2010-01-25 20:05:08 +0000819 This function closes a specified file handle. All "dirty" cached file data is
820 flushed to the device, and the file is closed. In all cases the handle is
jcarsey94b17fa2009-05-07 18:46:18 +0000821 closed.
822
823@param FileHandle the file handle to close.
824
825@retval EFI_SUCCESS the file handle was closed sucessfully.
826**/
827EFI_STATUS
828EFIAPI
829ShellCloseFile (
jcarseya405b862010-09-14 05:18:09 +0000830 IN SHELL_FILE_HANDLE *FileHandle
831 )
832{
jcarseyd2b45642009-05-11 18:02:16 +0000833 return (FileFunctionMap.CloseFile(*FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000834}
835
836/**
837 Delete a file and close the handle
838
839 This function closes and deletes a file. In all cases the file handle is closed.
jcarsey1e6e84c2010-01-25 20:05:08 +0000840 If the file cannot be deleted, the warning code EFI_WARN_DELETE_FAILURE is
jcarsey94b17fa2009-05-07 18:46:18 +0000841 returned, but the handle is still closed.
842
843 @param FileHandle the file handle to delete
844
845 @retval EFI_SUCCESS the file was closed sucessfully
jcarsey1e6e84c2010-01-25 20:05:08 +0000846 @retval EFI_WARN_DELETE_FAILURE the handle was closed, but the file was not
jcarsey94b17fa2009-05-07 18:46:18 +0000847 deleted
darylm503b0934ac2011-11-12 00:35:11 +0000848 @retval INVALID_PARAMETER One of the parameters has an invalid value.
jcarsey94b17fa2009-05-07 18:46:18 +0000849**/
850EFI_STATUS
851EFIAPI
852ShellDeleteFile (
darylm503b0934ac2011-11-12 00:35:11 +0000853 IN SHELL_FILE_HANDLE *FileHandle
jcarseya405b862010-09-14 05:18:09 +0000854 )
855{
jcarseyd2b45642009-05-11 18:02:16 +0000856 return (FileFunctionMap.DeleteFile(*FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000857}
858
859/**
860 Set the current position in a file.
861
jcarsey1e6e84c2010-01-25 20:05:08 +0000862 This function sets the current file position for the handle to the position
jcarsey94b17fa2009-05-07 18:46:18 +0000863 supplied. With the exception of seeking to position 0xFFFFFFFFFFFFFFFF, only
jcarsey1e6e84c2010-01-25 20:05:08 +0000864 absolute positioning is supported, and seeking past the end of the file is
865 allowed (a subsequent write would grow the file). Seeking to position
jcarsey94b17fa2009-05-07 18:46:18 +0000866 0xFFFFFFFFFFFFFFFF causes the current position to be set to the end of the file.
jcarsey1e6e84c2010-01-25 20:05:08 +0000867 If FileHandle is a directory, the only position that may be set is zero. This
jcarsey94b17fa2009-05-07 18:46:18 +0000868 has the effect of starting the read process of the directory entries over.
869
870 @param FileHandle The file handle on which the position is being set
871 @param Position Byte position from begining of file
872
873 @retval EFI_SUCCESS Operation completed sucessfully.
jcarsey1e6e84c2010-01-25 20:05:08 +0000874 @retval EFI_UNSUPPORTED the seek request for non-zero is not valid on
jcarsey94b17fa2009-05-07 18:46:18 +0000875 directories.
876 @retval INVALID_PARAMETER One of the parameters has an invalid value.
877**/
878EFI_STATUS
879EFIAPI
880ShellSetFilePosition (
darylm503b0934ac2011-11-12 00:35:11 +0000881 IN SHELL_FILE_HANDLE FileHandle,
882 IN UINT64 Position
jcarseya405b862010-09-14 05:18:09 +0000883 )
884{
jcarseyd2b45642009-05-11 18:02:16 +0000885 return (FileFunctionMap.SetFilePosition(FileHandle, Position));
jcarsey94b17fa2009-05-07 18:46:18 +0000886}
887
jcarsey1e6e84c2010-01-25 20:05:08 +0000888/**
jcarsey94b17fa2009-05-07 18:46:18 +0000889 Gets a file's current position
890
jcarsey1e6e84c2010-01-25 20:05:08 +0000891 This function retrieves the current file position for the file handle. For
892 directories, the current file position has no meaning outside of the file
jcarsey94b17fa2009-05-07 18:46:18 +0000893 system driver and as such the operation is not supported. An error is returned
894 if FileHandle is a directory.
895
896 @param FileHandle The open file handle on which to get the position.
897 @param Position Byte position from begining of file.
898
899 @retval EFI_SUCCESS the operation completed sucessfully.
900 @retval INVALID_PARAMETER One of the parameters has an invalid value.
901 @retval EFI_UNSUPPORTED the request is not valid on directories.
902**/
903EFI_STATUS
904EFIAPI
905ShellGetFilePosition (
jcarseya405b862010-09-14 05:18:09 +0000906 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000907 OUT UINT64 *Position
jcarseya405b862010-09-14 05:18:09 +0000908 )
909{
jcarseyd2b45642009-05-11 18:02:16 +0000910 return (FileFunctionMap.GetFilePosition(FileHandle, Position));
jcarsey94b17fa2009-05-07 18:46:18 +0000911}
912/**
913 Flushes data on a file
jcarsey1e6e84c2010-01-25 20:05:08 +0000914
jcarsey94b17fa2009-05-07 18:46:18 +0000915 This function flushes all modified data associated with a file to a device.
916
917 @param FileHandle The file handle on which to flush data
918
919 @retval EFI_SUCCESS The data was flushed.
920 @retval EFI_NO_MEDIA The device has no media.
921 @retval EFI_DEVICE_ERROR The device reported an error.
922 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
923 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
924 @retval EFI_ACCESS_DENIED The file was opened for read only.
925**/
926EFI_STATUS
927EFIAPI
928ShellFlushFile (
jcarseya405b862010-09-14 05:18:09 +0000929 IN SHELL_FILE_HANDLE FileHandle
930 )
931{
jcarseyd2b45642009-05-11 18:02:16 +0000932 return (FileFunctionMap.FlushFile(FileHandle));
jcarsey94b17fa2009-05-07 18:46:18 +0000933}
934
darylm503b0934ac2011-11-12 00:35:11 +0000935/** Retrieve first entry from a directory.
jcarsey94b17fa2009-05-07 18:46:18 +0000936
darylm503b0934ac2011-11-12 00:35:11 +0000937 This function takes an open directory handle and gets information from the
938 first entry in the directory. A buffer is allocated to contain
939 the information and a pointer to the buffer is returned in *Buffer. The
940 caller can use ShellFindNextFile() to get subsequent directory entries.
jcarsey94b17fa2009-05-07 18:46:18 +0000941
darylm503b0934ac2011-11-12 00:35:11 +0000942 The buffer will be freed by ShellFindNextFile() when the last directory
943 entry is read. Otherwise, the caller must free the buffer, using FreePool,
944 when finished with it.
945
946 @param[in] DirHandle The file handle of the directory to search.
947 @param[out] Buffer The pointer to the buffer for the file's information.
jcarsey94b17fa2009-05-07 18:46:18 +0000948
949 @retval EFI_SUCCESS Found the first file.
950 @retval EFI_NOT_FOUND Cannot find the directory.
951 @retval EFI_NO_MEDIA The device has no media.
952 @retval EFI_DEVICE_ERROR The device reported an error.
953 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
954 @return Others status of ShellGetFileInfo, ShellSetFilePosition,
955 or ShellReadFile
956**/
957EFI_STATUS
958EFIAPI
959ShellFindFirstFile (
jcarseya405b862010-09-14 05:18:09 +0000960 IN SHELL_FILE_HANDLE DirHandle,
jcarseyd2b45642009-05-11 18:02:16 +0000961 OUT EFI_FILE_INFO **Buffer
jcarseya405b862010-09-14 05:18:09 +0000962 )
963{
jcarsey94b17fa2009-05-07 18:46:18 +0000964 //
jcarseyd2b45642009-05-11 18:02:16 +0000965 // pass to file handle lib
jcarsey94b17fa2009-05-07 18:46:18 +0000966 //
jcarseyd2b45642009-05-11 18:02:16 +0000967 return (FileHandleFindFirstFile(DirHandle, Buffer));
jcarsey94b17fa2009-05-07 18:46:18 +0000968}
darylm503b0934ac2011-11-12 00:35:11 +0000969/** Retrieve next entries from a directory.
jcarsey94b17fa2009-05-07 18:46:18 +0000970
darylm503b0934ac2011-11-12 00:35:11 +0000971 To use this function, the caller must first call the ShellFindFirstFile()
972 function to get the first directory entry. Subsequent directory entries are
973 retrieved by using the ShellFindNextFile() function. This function can
974 be called several times to get each entry from the directory. If the call of
975 ShellFindNextFile() retrieved the last directory entry, the next call of
976 this function will set *NoFile to TRUE and free the buffer.
jcarsey94b17fa2009-05-07 18:46:18 +0000977
darylm503b0934ac2011-11-12 00:35:11 +0000978 @param[in] DirHandle The file handle of the directory.
979 @param[out] Buffer The pointer to buffer for file's information.
980 @param[out] NoFile The pointer to boolean when last file is found.
jcarsey94b17fa2009-05-07 18:46:18 +0000981
982 @retval EFI_SUCCESS Found the next file, or reached last file
983 @retval EFI_NO_MEDIA The device has no media.
984 @retval EFI_DEVICE_ERROR The device reported an error.
985 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
986**/
987EFI_STATUS
988EFIAPI
989ShellFindNextFile(
jcarseya405b862010-09-14 05:18:09 +0000990 IN SHELL_FILE_HANDLE DirHandle,
jcarsey94b17fa2009-05-07 18:46:18 +0000991 OUT EFI_FILE_INFO *Buffer,
992 OUT BOOLEAN *NoFile
jcarseya405b862010-09-14 05:18:09 +0000993 )
994{
jcarsey94b17fa2009-05-07 18:46:18 +0000995 //
jcarseyd2b45642009-05-11 18:02:16 +0000996 // pass to file handle lib
jcarsey94b17fa2009-05-07 18:46:18 +0000997 //
jcarseyd2b45642009-05-11 18:02:16 +0000998 return (FileHandleFindNextFile(DirHandle, Buffer, NoFile));
jcarsey94b17fa2009-05-07 18:46:18 +0000999}
1000/**
1001 Retrieve the size of a file.
1002
1003 if FileHandle is NULL then ASSERT()
1004 if Size is NULL then ASSERT()
1005
jcarsey1e6e84c2010-01-25 20:05:08 +00001006 This function extracts the file size info from the FileHandle's EFI_FILE_INFO
jcarsey94b17fa2009-05-07 18:46:18 +00001007 data.
1008
1009 @param FileHandle file handle from which size is retrieved
1010 @param Size pointer to size
1011
1012 @retval EFI_SUCCESS operation was completed sucessfully
1013 @retval EFI_DEVICE_ERROR cannot access the file
1014**/
1015EFI_STATUS
1016EFIAPI
1017ShellGetFileSize (
jcarseya405b862010-09-14 05:18:09 +00001018 IN SHELL_FILE_HANDLE FileHandle,
jcarsey94b17fa2009-05-07 18:46:18 +00001019 OUT UINT64 *Size
jcarseya405b862010-09-14 05:18:09 +00001020 )
1021{
jcarseyd2b45642009-05-11 18:02:16 +00001022 return (FileFunctionMap.GetFileSize(FileHandle, Size));
jcarsey94b17fa2009-05-07 18:46:18 +00001023}
1024/**
1025 Retrieves the status of the break execution flag
1026
1027 this function is useful to check whether the application is being asked to halt by the shell.
1028
1029 @retval TRUE the execution break is enabled
1030 @retval FALSE the execution break is not enabled
1031**/
1032BOOLEAN
1033EFIAPI
1034ShellGetExecutionBreakFlag(
1035 VOID
1036 )
1037{
jcarsey1e6e84c2010-01-25 20:05:08 +00001038 //
jcarsey94b17fa2009-05-07 18:46:18 +00001039 // Check for UEFI Shell 2.0 protocols
1040 //
jcarsey366f81a2011-06-27 21:04:22 +00001041 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001042
1043 //
1044 // We are using UEFI Shell 2.0; see if the event has been triggered
1045 //
jcarsey366f81a2011-06-27 21:04:22 +00001046 if (gBS->CheckEvent(gEfiShellProtocol->ExecutionBreak) != EFI_SUCCESS) {
jcarsey94b17fa2009-05-07 18:46:18 +00001047 return (FALSE);
1048 }
1049 return (TRUE);
jcarsey1e6e84c2010-01-25 20:05:08 +00001050 }
jcarsey94b17fa2009-05-07 18:46:18 +00001051
1052 //
1053 // using EFI Shell; call the function to check
1054 //
jcarsey92a54472011-06-27 20:33:13 +00001055 if (mEfiShellEnvironment2 != NULL) {
1056 return (mEfiShellEnvironment2->GetExecutionBreak());
1057 }
1058
1059 return (FALSE);
jcarsey94b17fa2009-05-07 18:46:18 +00001060}
1061/**
1062 return the value of an environment variable
1063
jcarsey1e6e84c2010-01-25 20:05:08 +00001064 this function gets the value of the environment variable set by the
jcarsey94b17fa2009-05-07 18:46:18 +00001065 ShellSetEnvironmentVariable function
1066
1067 @param EnvKey The key name of the environment variable.
1068
1069 @retval NULL the named environment variable does not exist.
1070 @return != NULL pointer to the value of the environment variable
1071**/
1072CONST CHAR16*
1073EFIAPI
1074ShellGetEnvironmentVariable (
jcarsey9b3bf082009-06-23 21:15:07 +00001075 IN CONST CHAR16 *EnvKey
jcarsey94b17fa2009-05-07 18:46:18 +00001076 )
1077{
jcarsey1e6e84c2010-01-25 20:05:08 +00001078 //
jcarsey94b17fa2009-05-07 18:46:18 +00001079 // Check for UEFI Shell 2.0 protocols
1080 //
jcarsey366f81a2011-06-27 21:04:22 +00001081 if (gEfiShellProtocol != NULL) {
1082 return (gEfiShellProtocol->GetEnv(EnvKey));
jcarsey94b17fa2009-05-07 18:46:18 +00001083 }
1084
1085 //
jcarsey92a54472011-06-27 20:33:13 +00001086 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001087 //
jcarsey92a54472011-06-27 20:33:13 +00001088 if (mEfiShellEnvironment2 != NULL) {
1089 return (mEfiShellEnvironment2->GetEnv((CHAR16*)EnvKey));
1090 }
jcarsey94b17fa2009-05-07 18:46:18 +00001091
jcarsey92a54472011-06-27 20:33:13 +00001092 return NULL;
jcarsey94b17fa2009-05-07 18:46:18 +00001093}
1094/**
1095 set the value of an environment variable
1096
1097This function changes the current value of the specified environment variable. If the
1098environment variable exists and the Value is an empty string, then the environment
1099variable is deleted. If the environment variable exists and the Value is not an empty
1100string, then the value of the environment variable is changed. If the environment
1101variable does not exist and the Value is an empty string, there is no action. If the
1102environment variable does not exist and the Value is a non-empty string, then the
1103environment variable is created and assigned the specified value.
1104
1105 This is not supported pre-UEFI Shell 2.0.
1106
1107 @param EnvKey The key name of the environment variable.
1108 @param EnvVal The Value of the environment variable
1109 @param Volatile Indicates whether the variable is non-volatile (FALSE) or volatile (TRUE).
1110
1111 @retval EFI_SUCCESS the operation was completed sucessfully
1112 @retval EFI_UNSUPPORTED This operation is not allowed in pre UEFI 2.0 Shell environments
1113**/
1114EFI_STATUS
1115EFIAPI
1116ShellSetEnvironmentVariable (
1117 IN CONST CHAR16 *EnvKey,
1118 IN CONST CHAR16 *EnvVal,
1119 IN BOOLEAN Volatile
1120 )
1121{
jcarsey1e6e84c2010-01-25 20:05:08 +00001122 //
jcarsey94b17fa2009-05-07 18:46:18 +00001123 // Check for UEFI Shell 2.0 protocols
1124 //
jcarsey366f81a2011-06-27 21:04:22 +00001125 if (gEfiShellProtocol != NULL) {
1126 return (gEfiShellProtocol->SetEnv(EnvKey, EnvVal, Volatile));
jcarsey1e6e84c2010-01-25 20:05:08 +00001127 }
jcarsey94b17fa2009-05-07 18:46:18 +00001128
1129 //
1130 // This feature does not exist under EFI shell
1131 //
1132 return (EFI_UNSUPPORTED);
1133}
jcarseya405b862010-09-14 05:18:09 +00001134
jcarsey94b17fa2009-05-07 18:46:18 +00001135/**
jcarseya405b862010-09-14 05:18:09 +00001136 Cause the shell to parse and execute a command line.
jcarsey94b17fa2009-05-07 18:46:18 +00001137
1138 This function creates a nested instance of the shell and executes the specified
jcarseya405b862010-09-14 05:18:09 +00001139 command (CommandLine) with the specified environment (Environment). Upon return,
1140 the status code returned by the specified command is placed in StatusCode.
1141 If Environment is NULL, then the current environment is used and all changes made
1142 by the commands executed will be reflected in the current environment. If the
1143 Environment is non-NULL, then the changes made will be discarded.
1144 The CommandLine is executed from the current working directory on the current
1145 device.
jcarsey94b17fa2009-05-07 18:46:18 +00001146
jcarseya405b862010-09-14 05:18:09 +00001147 The EnvironmentVariables and Status parameters are ignored in a pre-UEFI Shell 2.0
1148 environment. The values pointed to by the parameters will be unchanged by the
1149 ShellExecute() function. The Output parameter has no effect in a
1150 UEFI Shell 2.0 environment.
jcarsey94b17fa2009-05-07 18:46:18 +00001151
jcarseya405b862010-09-14 05:18:09 +00001152 @param[in] ParentHandle The parent image starting the operation.
1153 @param[in] CommandLine The pointer to a NULL terminated command line.
1154 @param[in] Output True to display debug output. False to hide it.
1155 @param[in] EnvironmentVariables Optional pointer to array of environment variables
1156 in the form "x=y". If NULL, the current set is used.
1157 @param[out] Status The status of the run command line.
jcarsey94b17fa2009-05-07 18:46:18 +00001158
jcarseya405b862010-09-14 05:18:09 +00001159 @retval EFI_SUCCESS The operation completed sucessfully. Status
1160 contains the status code returned.
1161 @retval EFI_INVALID_PARAMETER A parameter contains an invalid value.
1162 @retval EFI_OUT_OF_RESOURCES Out of resources.
1163 @retval EFI_UNSUPPORTED The operation is not allowed.
jcarsey94b17fa2009-05-07 18:46:18 +00001164**/
1165EFI_STATUS
1166EFIAPI
1167ShellExecute (
1168 IN EFI_HANDLE *ParentHandle,
1169 IN CHAR16 *CommandLine OPTIONAL,
1170 IN BOOLEAN Output OPTIONAL,
1171 IN CHAR16 **EnvironmentVariables OPTIONAL,
1172 OUT EFI_STATUS *Status OPTIONAL
1173 )
1174{
jcarsey1e6e84c2010-01-25 20:05:08 +00001175 //
jcarsey94b17fa2009-05-07 18:46:18 +00001176 // Check for UEFI Shell 2.0 protocols
1177 //
jcarsey366f81a2011-06-27 21:04:22 +00001178 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001179 //
1180 // Call UEFI Shell 2.0 version (not using Output parameter)
1181 //
jcarsey366f81a2011-06-27 21:04:22 +00001182 return (gEfiShellProtocol->Execute(ParentHandle,
jcarsey94b17fa2009-05-07 18:46:18 +00001183 CommandLine,
1184 EnvironmentVariables,
1185 Status));
jcarsey1e6e84c2010-01-25 20:05:08 +00001186 }
jcarsey92a54472011-06-27 20:33:13 +00001187
jcarsey94b17fa2009-05-07 18:46:18 +00001188 //
jcarsey92a54472011-06-27 20:33:13 +00001189 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001190 //
jcarsey92a54472011-06-27 20:33:13 +00001191 if (mEfiShellEnvironment2 != NULL) {
1192 //
1193 // Call EFI Shell version (not using EnvironmentVariables or Status parameters)
1194 // Due to oddity in the EFI shell we want to dereference the ParentHandle here
1195 //
1196 return (mEfiShellEnvironment2->Execute(*ParentHandle,
1197 CommandLine,
1198 Output));
1199 }
1200
1201 return (EFI_UNSUPPORTED);
jcarsey94b17fa2009-05-07 18:46:18 +00001202}
1203/**
1204 Retreives the current directory path
1205
jcarsey1e6e84c2010-01-25 20:05:08 +00001206 If the DeviceName is NULL, it returns the current device's current directory
1207 name. If the DeviceName is not NULL, it returns the current directory name
jcarsey94b17fa2009-05-07 18:46:18 +00001208 on specified drive.
1209
1210 @param DeviceName the name of the drive to get directory on
1211
1212 @retval NULL the directory does not exist
1213 @return != NULL the directory
1214**/
1215CONST CHAR16*
1216EFIAPI
1217ShellGetCurrentDir (
jcarseya405b862010-09-14 05:18:09 +00001218 IN CHAR16 * CONST DeviceName OPTIONAL
jcarsey94b17fa2009-05-07 18:46:18 +00001219 )
1220{
jcarsey1e6e84c2010-01-25 20:05:08 +00001221 //
jcarsey94b17fa2009-05-07 18:46:18 +00001222 // Check for UEFI Shell 2.0 protocols
1223 //
jcarsey366f81a2011-06-27 21:04:22 +00001224 if (gEfiShellProtocol != NULL) {
1225 return (gEfiShellProtocol->GetCurDir(DeviceName));
jcarsey1e6e84c2010-01-25 20:05:08 +00001226 }
jcarsey92a54472011-06-27 20:33:13 +00001227
jcarsey94b17fa2009-05-07 18:46:18 +00001228 //
jcarsey8bd282b2011-06-27 16:45:41 +00001229 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001230 //
jcarsey8bd282b2011-06-27 16:45:41 +00001231 if (mEfiShellEnvironment2 != NULL) {
1232 return (mEfiShellEnvironment2->CurDir(DeviceName));
1233 }
1234
1235 return (NULL);
jcarsey94b17fa2009-05-07 18:46:18 +00001236}
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 //
jcarsey366f81a2011-06-27 21:04:22 +00001258 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001259 //
1260 // Enable with UEFI 2.0 Shell
1261 //
jcarsey366f81a2011-06-27 21:04:22 +00001262 gEfiShellProtocol->EnablePageBreak();
jcarsey94b17fa2009-05-07 18:46:18 +00001263 return;
1264 } else {
jcarsey1e6e84c2010-01-25 20:05:08 +00001265 //
jcarsey92a54472011-06-27 20:33:13 +00001266 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001267 //
jcarsey92a54472011-06-27 20:33:13 +00001268 if (mEfiShellEnvironment2 != NULL) {
1269 //
1270 // Enable with EFI Shell
1271 //
1272 mEfiShellEnvironment2->EnablePageBreak (DEFAULT_INIT_ROW, DEFAULT_AUTO_LF);
1273 return;
1274 }
jcarsey94b17fa2009-05-07 18:46:18 +00001275 }
1276 } else {
jcarsey1e6e84c2010-01-25 20:05:08 +00001277 //
jcarsey94b17fa2009-05-07 18:46:18 +00001278 // check for UEFI Shell 2.0
1279 //
jcarsey366f81a2011-06-27 21:04:22 +00001280 if (gEfiShellProtocol != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001281 //
1282 // Disable with UEFI 2.0 Shell
1283 //
jcarsey366f81a2011-06-27 21:04:22 +00001284 gEfiShellProtocol->DisablePageBreak();
jcarsey94b17fa2009-05-07 18:46:18 +00001285 return;
1286 } else {
jcarsey1e6e84c2010-01-25 20:05:08 +00001287 //
jcarsey92a54472011-06-27 20:33:13 +00001288 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001289 //
jcarsey92a54472011-06-27 20:33:13 +00001290 if (mEfiShellEnvironment2 != NULL) {
1291 //
1292 // Disable with EFI Shell
1293 //
1294 mEfiShellEnvironment2->DisablePageBreak ();
1295 return;
1296 }
jcarsey94b17fa2009-05-07 18:46:18 +00001297 }
1298 }
1299}
1300
1301///
1302/// version of EFI_SHELL_FILE_INFO struct, except has no CONST pointers.
1303/// This allows for the struct to be populated.
1304///
1305typedef struct {
jcarseyd2b45642009-05-11 18:02:16 +00001306 LIST_ENTRY Link;
jcarsey94b17fa2009-05-07 18:46:18 +00001307 EFI_STATUS Status;
1308 CHAR16 *FullName;
1309 CHAR16 *FileName;
jcarseya405b862010-09-14 05:18:09 +00001310 SHELL_FILE_HANDLE Handle;
jcarsey94b17fa2009-05-07 18:46:18 +00001311 EFI_FILE_INFO *Info;
1312} EFI_SHELL_FILE_INFO_NO_CONST;
1313
1314/**
1315 Converts a EFI shell list of structures to the coresponding UEFI Shell 2.0 type of list.
1316
1317 if OldStyleFileList is NULL then ASSERT()
1318
jcarsey1e6e84c2010-01-25 20:05:08 +00001319 this function will convert a SHELL_FILE_ARG based list into a callee allocated
jcarsey94b17fa2009-05-07 18:46:18 +00001320 EFI_SHELL_FILE_INFO based list. it is up to the caller to free the memory via
1321 the ShellCloseFileMetaArg function.
1322
ydong104ff7e372011-09-02 08:05:34 +00001323 @param[in] FileList the EFI shell list type
1324 @param[in, out] ListHead the list to add to
jcarsey94b17fa2009-05-07 18:46:18 +00001325
1326 @retval the resultant head of the double linked new format list;
1327**/
1328LIST_ENTRY*
1329EFIAPI
1330InternalShellConvertFileListType (
jcarsey9b3bf082009-06-23 21:15:07 +00001331 IN LIST_ENTRY *FileList,
1332 IN OUT LIST_ENTRY *ListHead
jcarsey125c2cf2009-11-18 21:36:50 +00001333 )
1334{
jcarsey94b17fa2009-05-07 18:46:18 +00001335 SHELL_FILE_ARG *OldInfo;
jcarsey9b3bf082009-06-23 21:15:07 +00001336 LIST_ENTRY *Link;
jcarsey94b17fa2009-05-07 18:46:18 +00001337 EFI_SHELL_FILE_INFO_NO_CONST *NewInfo;
1338
1339 //
jcarsey9b3bf082009-06-23 21:15:07 +00001340 // ASSERTs
jcarsey94b17fa2009-05-07 18:46:18 +00001341 //
jcarsey9b3bf082009-06-23 21:15:07 +00001342 ASSERT(FileList != NULL);
1343 ASSERT(ListHead != NULL);
jcarsey94b17fa2009-05-07 18:46:18 +00001344
1345 //
1346 // enumerate through each member of the old list and copy
1347 //
jcarseyd2b45642009-05-11 18:02:16 +00001348 for (Link = FileList->ForwardLink; Link != FileList; Link = Link->ForwardLink) {
jcarsey94b17fa2009-05-07 18:46:18 +00001349 OldInfo = CR (Link, SHELL_FILE_ARG, Link, SHELL_FILE_ARG_SIGNATURE);
jcarseya405b862010-09-14 05:18:09 +00001350 ASSERT(OldInfo != NULL);
1351
1352 //
1353 // Skip ones that failed to open...
1354 //
1355 if (OldInfo->Status != EFI_SUCCESS) {
1356 continue;
1357 }
jcarsey94b17fa2009-05-07 18:46:18 +00001358
1359 //
1360 // make sure the old list was valid
1361 //
jcarsey94b17fa2009-05-07 18:46:18 +00001362 ASSERT(OldInfo->Info != NULL);
1363 ASSERT(OldInfo->FullName != NULL);
1364 ASSERT(OldInfo->FileName != NULL);
1365
1366 //
1367 // allocate a new EFI_SHELL_FILE_INFO object
1368 //
1369 NewInfo = AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
jcarseyc9d92df2010-02-03 15:37:54 +00001370 if (NewInfo == NULL) {
jcarsey7a95efd2011-10-13 16:08:18 +00001371 ShellCloseFileMetaArg((EFI_SHELL_FILE_INFO**)(&ListHead));
jcarseybeab0fc2011-10-10 17:26:25 +00001372 ListHead = NULL;
jcarseyc9d92df2010-02-03 15:37:54 +00001373 break;
1374 }
jcarsey1e6e84c2010-01-25 20:05:08 +00001375
1376 //
jcarsey94b17fa2009-05-07 18:46:18 +00001377 // copy the simple items
1378 //
1379 NewInfo->Handle = OldInfo->Handle;
1380 NewInfo->Status = OldInfo->Status;
1381
jcarseyd2b45642009-05-11 18:02:16 +00001382 // old shell checks for 0 not NULL
1383 OldInfo->Handle = 0;
1384
jcarsey94b17fa2009-05-07 18:46:18 +00001385 //
1386 // allocate new space to copy strings and structure
1387 //
1388 NewInfo->FullName = AllocateZeroPool(StrSize(OldInfo->FullName));
1389 NewInfo->FileName = AllocateZeroPool(StrSize(OldInfo->FileName));
1390 NewInfo->Info = AllocateZeroPool((UINTN)OldInfo->Info->Size);
jcarsey1e6e84c2010-01-25 20:05:08 +00001391
jcarsey94b17fa2009-05-07 18:46:18 +00001392 //
1393 // make sure all the memory allocations were sucessful
1394 //
jcarseybeab0fc2011-10-10 17:26:25 +00001395 if (NULL == NewInfo->FullName || NewInfo->FileName == NULL || NewInfo->Info == NULL) {
jcarsey7a95efd2011-10-13 16:08:18 +00001396 ShellCloseFileMetaArg((EFI_SHELL_FILE_INFO**)(&ListHead));
jcarseybeab0fc2011-10-10 17:26:25 +00001397 ListHead = NULL;
1398 break;
1399 }
jcarsey94b17fa2009-05-07 18:46:18 +00001400
1401 //
1402 // Copt the strings and structure
1403 //
1404 StrCpy(NewInfo->FullName, OldInfo->FullName);
1405 StrCpy(NewInfo->FileName, OldInfo->FileName);
1406 gBS->CopyMem (NewInfo->Info, OldInfo->Info, (UINTN)OldInfo->Info->Size);
1407
1408 //
1409 // add that to the list
1410 //
jcarsey9b3bf082009-06-23 21:15:07 +00001411 InsertTailList(ListHead, &NewInfo->Link);
jcarsey94b17fa2009-05-07 18:46:18 +00001412 }
1413 return (ListHead);
1414}
1415/**
1416 Opens a group of files based on a path.
1417
jcarsey1e6e84c2010-01-25 20:05:08 +00001418 This function uses the Arg to open all the matching files. Each matched
1419 file has a SHELL_FILE_ARG structure to record the file information. These
1420 structures are placed on the list ListHead. Users can get the SHELL_FILE_ARG
jcarsey94b17fa2009-05-07 18:46:18 +00001421 structures from ListHead to access each file. This function supports wildcards
jcarsey1e6e84c2010-01-25 20:05:08 +00001422 and will process '?' and '*' as such. the list must be freed with a call to
jcarsey94b17fa2009-05-07 18:46:18 +00001423 ShellCloseFileMetaArg().
1424
jcarsey1e6e84c2010-01-25 20:05:08 +00001425 If you are NOT appending to an existing list *ListHead must be NULL. If
jcarsey5f7431d2009-07-10 18:06:01 +00001426 *ListHead is NULL then it must be callee freed.
jcarsey94b17fa2009-05-07 18:46:18 +00001427
1428 @param Arg pointer to path string
1429 @param OpenMode mode to open files with
1430 @param ListHead head of linked list of results
1431
jcarsey1e6e84c2010-01-25 20:05:08 +00001432 @retval EFI_SUCCESS the operation was sucessful and the list head
jcarsey94b17fa2009-05-07 18:46:18 +00001433 contains the list of opened files
jcarsey94b17fa2009-05-07 18:46:18 +00001434 @return != EFI_SUCCESS the operation failed
1435
1436 @sa InternalShellConvertFileListType
1437**/
1438EFI_STATUS
1439EFIAPI
1440ShellOpenFileMetaArg (
1441 IN CHAR16 *Arg,
1442 IN UINT64 OpenMode,
1443 IN OUT EFI_SHELL_FILE_INFO **ListHead
1444 )
1445{
1446 EFI_STATUS Status;
jcarsey9b3bf082009-06-23 21:15:07 +00001447 LIST_ENTRY mOldStyleFileList;
jcarsey1e6e84c2010-01-25 20:05:08 +00001448
jcarsey94b17fa2009-05-07 18:46:18 +00001449 //
1450 // ASSERT that Arg and ListHead are not NULL
1451 //
1452 ASSERT(Arg != NULL);
1453 ASSERT(ListHead != NULL);
1454
jcarsey1e6e84c2010-01-25 20:05:08 +00001455 //
jcarsey94b17fa2009-05-07 18:46:18 +00001456 // Check for UEFI Shell 2.0 protocols
1457 //
jcarsey366f81a2011-06-27 21:04:22 +00001458 if (gEfiShellProtocol != NULL) {
jcarsey5f7431d2009-07-10 18:06:01 +00001459 if (*ListHead == NULL) {
1460 *ListHead = (EFI_SHELL_FILE_INFO*)AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1461 if (*ListHead == NULL) {
1462 return (EFI_OUT_OF_RESOURCES);
1463 }
1464 InitializeListHead(&((*ListHead)->Link));
jcarsey1e6e84c2010-01-25 20:05:08 +00001465 }
jcarsey366f81a2011-06-27 21:04:22 +00001466 Status = gEfiShellProtocol->OpenFileList(Arg,
jcarsey1e6e84c2010-01-25 20:05:08 +00001467 OpenMode,
jcarsey2247dde2009-11-09 18:08:58 +00001468 ListHead);
1469 if (EFI_ERROR(Status)) {
jcarsey366f81a2011-06-27 21:04:22 +00001470 gEfiShellProtocol->RemoveDupInFileList(ListHead);
jcarsey2247dde2009-11-09 18:08:58 +00001471 } else {
jcarsey366f81a2011-06-27 21:04:22 +00001472 Status = gEfiShellProtocol->RemoveDupInFileList(ListHead);
jcarsey2247dde2009-11-09 18:08:58 +00001473 }
jcarseya405b862010-09-14 05:18:09 +00001474 if (*ListHead != NULL && IsListEmpty(&(*ListHead)->Link)) {
1475 FreePool(*ListHead);
1476 *ListHead = NULL;
1477 return (EFI_NOT_FOUND);
1478 }
jcarsey2247dde2009-11-09 18:08:58 +00001479 return (Status);
jcarsey1e6e84c2010-01-25 20:05:08 +00001480 }
jcarsey94b17fa2009-05-07 18:46:18 +00001481
1482 //
jcarsey92a54472011-06-27 20:33:13 +00001483 // Check for EFI shell
jcarsey94b17fa2009-05-07 18:46:18 +00001484 //
jcarsey92a54472011-06-27 20:33:13 +00001485 if (mEfiShellEnvironment2 != NULL) {
1486 //
1487 // make sure the list head is initialized
1488 //
1489 InitializeListHead(&mOldStyleFileList);
jcarsey94b17fa2009-05-07 18:46:18 +00001490
jcarsey92a54472011-06-27 20:33:13 +00001491 //
1492 // Get the EFI Shell list of files
1493 //
1494 Status = mEfiShellEnvironment2->FileMetaArg(Arg, &mOldStyleFileList);
1495 if (EFI_ERROR(Status)) {
1496 *ListHead = NULL;
1497 return (Status);
1498 }
jcarsey94b17fa2009-05-07 18:46:18 +00001499
jcarsey92a54472011-06-27 20:33:13 +00001500 if (*ListHead == NULL) {
1501 *ListHead = (EFI_SHELL_FILE_INFO *)AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1502 if (*ListHead == NULL) {
1503 return (EFI_OUT_OF_RESOURCES);
1504 }
1505 InitializeListHead(&((*ListHead)->Link));
1506 }
1507
1508 //
1509 // Convert that to equivalent of UEFI Shell 2.0 structure
1510 //
1511 InternalShellConvertFileListType(&mOldStyleFileList, &(*ListHead)->Link);
1512
1513 //
1514 // Free the EFI Shell version that was converted.
1515 //
1516 mEfiShellEnvironment2->FreeFileList(&mOldStyleFileList);
1517
1518 if ((*ListHead)->Link.ForwardLink == (*ListHead)->Link.BackLink && (*ListHead)->Link.BackLink == &((*ListHead)->Link)) {
1519 FreePool(*ListHead);
1520 *ListHead = NULL;
1521 Status = EFI_NOT_FOUND;
1522 }
jcarsey94b17fa2009-05-07 18:46:18 +00001523 return (Status);
1524 }
1525
jcarsey92a54472011-06-27 20:33:13 +00001526 return (EFI_UNSUPPORTED);
jcarsey94b17fa2009-05-07 18:46:18 +00001527}
1528/**
jcarseya405b862010-09-14 05:18:09 +00001529 Free the linked list returned from ShellOpenFileMetaArg.
jcarsey94b17fa2009-05-07 18:46:18 +00001530
jcarseya405b862010-09-14 05:18:09 +00001531 if ListHead is NULL then ASSERT().
jcarsey94b17fa2009-05-07 18:46:18 +00001532
jcarseya405b862010-09-14 05:18:09 +00001533 @param ListHead the pointer to free.
jcarsey94b17fa2009-05-07 18:46:18 +00001534
jcarseya405b862010-09-14 05:18:09 +00001535 @retval EFI_SUCCESS the operation was sucessful.
jcarsey94b17fa2009-05-07 18:46:18 +00001536**/
1537EFI_STATUS
1538EFIAPI
1539ShellCloseFileMetaArg (
1540 IN OUT EFI_SHELL_FILE_INFO **ListHead
1541 )
1542{
1543 LIST_ENTRY *Node;
1544
1545 //
1546 // ASSERT that ListHead is not NULL
1547 //
1548 ASSERT(ListHead != NULL);
1549
jcarsey1e6e84c2010-01-25 20:05:08 +00001550 //
jcarsey94b17fa2009-05-07 18:46:18 +00001551 // Check for UEFI Shell 2.0 protocols
1552 //
jcarsey366f81a2011-06-27 21:04:22 +00001553 if (gEfiShellProtocol != NULL) {
1554 return (gEfiShellProtocol->FreeFileList(ListHead));
jcarsey92a54472011-06-27 20:33:13 +00001555 } else if (mEfiShellEnvironment2 != NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00001556 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001557 // Since this is EFI Shell version we need to free our internally made copy
jcarsey94b17fa2009-05-07 18:46:18 +00001558 // of the list
1559 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001560 for ( Node = GetFirstNode(&(*ListHead)->Link)
jcarseya405b862010-09-14 05:18:09 +00001561 ; *ListHead != NULL && !IsListEmpty(&(*ListHead)->Link)
jcarsey9b3bf082009-06-23 21:15:07 +00001562 ; Node = GetFirstNode(&(*ListHead)->Link)) {
jcarsey94b17fa2009-05-07 18:46:18 +00001563 RemoveEntryList(Node);
jcarseya405b862010-09-14 05:18:09 +00001564 ((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 +00001565 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->FullName);
1566 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->FileName);
1567 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->Info);
1568 FreePool((EFI_SHELL_FILE_INFO_NO_CONST*)Node);
1569 }
1570 return EFI_SUCCESS;
1571 }
jcarsey92a54472011-06-27 20:33:13 +00001572
1573 return (EFI_UNSUPPORTED);
jcarsey94b17fa2009-05-07 18:46:18 +00001574}
1575
jcarsey125c2cf2009-11-18 21:36:50 +00001576/**
1577 Find a file by searching the CWD and then the path.
1578
jcarseyb3011f42010-01-11 21:49:04 +00001579 If FileName is NULL then ASSERT.
jcarsey125c2cf2009-11-18 21:36:50 +00001580
jcarseyb3011f42010-01-11 21:49:04 +00001581 If the return value is not NULL then the memory must be caller freed.
jcarsey125c2cf2009-11-18 21:36:50 +00001582
1583 @param FileName Filename string.
1584
1585 @retval NULL the file was not found
1586 @return !NULL the full path to the file.
1587**/
1588CHAR16 *
1589EFIAPI
1590ShellFindFilePath (
1591 IN CONST CHAR16 *FileName
1592 )
1593{
1594 CONST CHAR16 *Path;
jcarseya405b862010-09-14 05:18:09 +00001595 SHELL_FILE_HANDLE Handle;
jcarsey125c2cf2009-11-18 21:36:50 +00001596 EFI_STATUS Status;
1597 CHAR16 *RetVal;
1598 CHAR16 *TestPath;
1599 CONST CHAR16 *Walker;
jcarsey36a9d672009-11-20 21:13:41 +00001600 UINTN Size;
jcarsey1cd45e72010-01-29 15:07:44 +00001601 CHAR16 *TempChar;
jcarsey125c2cf2009-11-18 21:36:50 +00001602
1603 RetVal = NULL;
1604
jcarseya405b862010-09-14 05:18:09 +00001605 //
1606 // First make sure its not an absolute path.
1607 //
1608 Status = ShellOpenFileByName(FileName, &Handle, EFI_FILE_MODE_READ, 0);
1609 if (!EFI_ERROR(Status)){
1610 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1611 ASSERT(RetVal == NULL);
1612 RetVal = StrnCatGrow(&RetVal, NULL, FileName, 0);
1613 ShellCloseFile(&Handle);
1614 return (RetVal);
1615 } else {
1616 ShellCloseFile(&Handle);
1617 }
1618 }
1619
jcarsey125c2cf2009-11-18 21:36:50 +00001620 Path = ShellGetEnvironmentVariable(L"cwd");
1621 if (Path != NULL) {
jcarsey36a9d672009-11-20 21:13:41 +00001622 Size = StrSize(Path);
1623 Size += StrSize(FileName);
1624 TestPath = AllocateZeroPool(Size);
jcarseyc9d92df2010-02-03 15:37:54 +00001625 if (TestPath == NULL) {
1626 return (NULL);
1627 }
jcarsey125c2cf2009-11-18 21:36:50 +00001628 StrCpy(TestPath, Path);
1629 StrCat(TestPath, FileName);
1630 Status = ShellOpenFileByName(TestPath, &Handle, EFI_FILE_MODE_READ, 0);
1631 if (!EFI_ERROR(Status)){
jcarseya405b862010-09-14 05:18:09 +00001632 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1633 ASSERT(RetVal == NULL);
1634 RetVal = StrnCatGrow(&RetVal, NULL, TestPath, 0);
1635 ShellCloseFile(&Handle);
1636 FreePool(TestPath);
1637 return (RetVal);
1638 } else {
1639 ShellCloseFile(&Handle);
1640 }
jcarsey125c2cf2009-11-18 21:36:50 +00001641 }
1642 FreePool(TestPath);
1643 }
1644 Path = ShellGetEnvironmentVariable(L"path");
1645 if (Path != NULL) {
jcarseya405b862010-09-14 05:18:09 +00001646 Size = StrSize(Path)+sizeof(CHAR16);
jcarsey36a9d672009-11-20 21:13:41 +00001647 Size += StrSize(FileName);
1648 TestPath = AllocateZeroPool(Size);
jcarsey3e082d52010-10-04 16:44:57 +00001649 if (TestPath == NULL) {
1650 return (NULL);
1651 }
jcarsey1e6e84c2010-01-25 20:05:08 +00001652 Walker = (CHAR16*)Path;
jcarsey125c2cf2009-11-18 21:36:50 +00001653 do {
1654 CopyMem(TestPath, Walker, StrSize(Walker));
jcarsey3e082d52010-10-04 16:44:57 +00001655 if (TestPath != NULL) {
1656 TempChar = StrStr(TestPath, L";");
1657 if (TempChar != NULL) {
1658 *TempChar = CHAR_NULL;
1659 }
1660 if (TestPath[StrLen(TestPath)-1] != L'\\') {
1661 StrCat(TestPath, L"\\");
1662 }
jcarsey89e85372011-04-13 23:37:50 +00001663 if (FileName[0] == L'\\') {
1664 FileName++;
1665 }
jcarsey3e082d52010-10-04 16:44:57 +00001666 StrCat(TestPath, FileName);
1667 if (StrStr(Walker, L";") != NULL) {
1668 Walker = StrStr(Walker, L";") + 1;
jcarseya405b862010-09-14 05:18:09 +00001669 } else {
jcarsey3e082d52010-10-04 16:44:57 +00001670 Walker = NULL;
1671 }
1672 Status = ShellOpenFileByName(TestPath, &Handle, EFI_FILE_MODE_READ, 0);
1673 if (!EFI_ERROR(Status)){
1674 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1675 ASSERT(RetVal == NULL);
1676 RetVal = StrnCatGrow(&RetVal, NULL, TestPath, 0);
1677 ShellCloseFile(&Handle);
1678 break;
1679 } else {
1680 ShellCloseFile(&Handle);
1681 }
jcarseya405b862010-09-14 05:18:09 +00001682 }
jcarsey125c2cf2009-11-18 21:36:50 +00001683 }
1684 } while (Walker != NULL && Walker[0] != CHAR_NULL);
1685 FreePool(TestPath);
1686 }
1687 return (RetVal);
1688}
1689
jcarseyb3011f42010-01-11 21:49:04 +00001690/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001691 Find a file by searching the CWD and then the path with a variable set of file
1692 extensions. If the file is not found it will append each extension in the list
jcarseyb3011f42010-01-11 21:49:04 +00001693 in the order provided and return the first one that is successful.
1694
1695 If FileName is NULL, then ASSERT.
1696 If FileExtension is NULL, then behavior is identical to ShellFindFilePath.
1697
1698 If the return value is not NULL then the memory must be caller freed.
1699
1700 @param[in] FileName Filename string.
1701 @param[in] FileExtension Semi-colon delimeted list of possible extensions.
1702
1703 @retval NULL The file was not found.
1704 @retval !NULL The path to the file.
1705**/
1706CHAR16 *
1707EFIAPI
1708ShellFindFilePathEx (
1709 IN CONST CHAR16 *FileName,
1710 IN CONST CHAR16 *FileExtension
1711 )
1712{
1713 CHAR16 *TestPath;
1714 CHAR16 *RetVal;
1715 CONST CHAR16 *ExtensionWalker;
jcarsey9e926b62010-01-14 20:26:39 +00001716 UINTN Size;
jcarsey1cd45e72010-01-29 15:07:44 +00001717 CHAR16 *TempChar;
jcarseyc9d92df2010-02-03 15:37:54 +00001718 CHAR16 *TempChar2;
jcarsey1cd45e72010-01-29 15:07:44 +00001719
jcarseyb3011f42010-01-11 21:49:04 +00001720 ASSERT(FileName != NULL);
1721 if (FileExtension == NULL) {
1722 return (ShellFindFilePath(FileName));
1723 }
1724 RetVal = ShellFindFilePath(FileName);
1725 if (RetVal != NULL) {
1726 return (RetVal);
1727 }
jcarsey9e926b62010-01-14 20:26:39 +00001728 Size = StrSize(FileName);
1729 Size += StrSize(FileExtension);
1730 TestPath = AllocateZeroPool(Size);
jcarseyc9d92df2010-02-03 15:37:54 +00001731 if (TestPath == NULL) {
1732 return (NULL);
1733 }
jcarseya405b862010-09-14 05:18:09 +00001734 for (ExtensionWalker = FileExtension, TempChar2 = (CHAR16*)FileExtension; TempChar2 != NULL ; ExtensionWalker = TempChar2 + 1){
jcarseyb3011f42010-01-11 21:49:04 +00001735 StrCpy(TestPath, FileName);
jcarseya405b862010-09-14 05:18:09 +00001736 if (ExtensionWalker != NULL) {
1737 StrCat(TestPath, ExtensionWalker);
1738 }
jcarsey1cd45e72010-01-29 15:07:44 +00001739 TempChar = StrStr(TestPath, L";");
1740 if (TempChar != NULL) {
1741 *TempChar = CHAR_NULL;
jcarseyb3011f42010-01-11 21:49:04 +00001742 }
1743 RetVal = ShellFindFilePath(TestPath);
1744 if (RetVal != NULL) {
1745 break;
1746 }
jcarseya405b862010-09-14 05:18:09 +00001747 ASSERT(ExtensionWalker != NULL);
jcarseyc9d92df2010-02-03 15:37:54 +00001748 TempChar2 = StrStr(ExtensionWalker, L";");
jcarseyb3011f42010-01-11 21:49:04 +00001749 }
1750 FreePool(TestPath);
1751 return (RetVal);
1752}
1753
jcarsey94b17fa2009-05-07 18:46:18 +00001754typedef struct {
jcarsey9b3bf082009-06-23 21:15:07 +00001755 LIST_ENTRY Link;
jcarsey94b17fa2009-05-07 18:46:18 +00001756 CHAR16 *Name;
jcarseya405b862010-09-14 05:18:09 +00001757 SHELL_PARAM_TYPE Type;
jcarsey94b17fa2009-05-07 18:46:18 +00001758 CHAR16 *Value;
1759 UINTN OriginalPosition;
1760} SHELL_PARAM_PACKAGE;
1761
1762/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001763 Checks the list of valid arguments and returns TRUE if the item was found. If the
jcarsey94b17fa2009-05-07 18:46:18 +00001764 return value is TRUE then the type parameter is set also.
jcarsey1e6e84c2010-01-25 20:05:08 +00001765
jcarsey94b17fa2009-05-07 18:46:18 +00001766 if CheckList is NULL then ASSERT();
1767 if Name is NULL then ASSERT();
1768 if Type is NULL then ASSERT();
1769
jcarsey94b17fa2009-05-07 18:46:18 +00001770 @param Name pointer to Name of parameter found
1771 @param CheckList List to check against
jcarseya405b862010-09-14 05:18:09 +00001772 @param Type pointer to type of parameter if it was found
jcarsey94b17fa2009-05-07 18:46:18 +00001773
1774 @retval TRUE the Parameter was found. Type is valid.
1775 @retval FALSE the Parameter was not found. Type is not valid.
1776**/
1777BOOLEAN
1778EFIAPI
jcarseyd2b45642009-05-11 18:02:16 +00001779InternalIsOnCheckList (
jcarsey94b17fa2009-05-07 18:46:18 +00001780 IN CONST CHAR16 *Name,
1781 IN CONST SHELL_PARAM_ITEM *CheckList,
jcarsey252d9452011-03-25 20:49:53 +00001782 OUT SHELL_PARAM_TYPE *Type
jcarseya405b862010-09-14 05:18:09 +00001783 )
1784{
jcarsey94b17fa2009-05-07 18:46:18 +00001785 SHELL_PARAM_ITEM *TempListItem;
jcarsey252d9452011-03-25 20:49:53 +00001786 CHAR16 *TempString;
jcarsey94b17fa2009-05-07 18:46:18 +00001787
1788 //
1789 // ASSERT that all 3 pointer parameters aren't NULL
1790 //
1791 ASSERT(CheckList != NULL);
1792 ASSERT(Type != NULL);
1793 ASSERT(Name != NULL);
1794
1795 //
jcarseyd2b45642009-05-11 18:02:16 +00001796 // question mark and page break mode are always supported
1797 //
1798 if ((StrCmp(Name, L"-?") == 0) ||
1799 (StrCmp(Name, L"-b") == 0)
jcarseya405b862010-09-14 05:18:09 +00001800 ) {
jcarsey252d9452011-03-25 20:49:53 +00001801 *Type = TypeFlag;
jcarseyd2b45642009-05-11 18:02:16 +00001802 return (TRUE);
1803 }
1804
1805 //
jcarsey94b17fa2009-05-07 18:46:18 +00001806 // Enumerate through the list
1807 //
1808 for (TempListItem = (SHELL_PARAM_ITEM*)CheckList ; TempListItem->Name != NULL ; TempListItem++) {
1809 //
jcarsey9eb53ac2009-07-08 17:26:58 +00001810 // If the Type is TypeStart only check the first characters of the passed in param
1811 // If it matches set the type and return TRUE
jcarsey94b17fa2009-05-07 18:46:18 +00001812 //
darylm503b0934ac2011-11-12 00:35:11 +00001813 if (TempListItem->Type == TypeStart) {
jcarsey252d9452011-03-25 20:49:53 +00001814 if (StrnCmp(Name, TempListItem->Name, StrLen(TempListItem->Name)) == 0) {
1815 *Type = TempListItem->Type;
1816 return (TRUE);
1817 }
1818 TempString = NULL;
1819 TempString = StrnCatGrow(&TempString, NULL, Name, StrLen(TempListItem->Name));
1820 if (TempString != NULL) {
1821 if (StringNoCaseCompare(&TempString, &TempListItem->Name) == 0) {
1822 *Type = TempListItem->Type;
1823 FreePool(TempString);
1824 return (TRUE);
1825 }
1826 FreePool(TempString);
1827 }
1828 } else if (StringNoCaseCompare(&Name, &TempListItem->Name) == 0) {
jcarsey94b17fa2009-05-07 18:46:18 +00001829 *Type = TempListItem->Type;
1830 return (TRUE);
1831 }
1832 }
jcarsey2247dde2009-11-09 18:08:58 +00001833
jcarsey94b17fa2009-05-07 18:46:18 +00001834 return (FALSE);
1835}
1836/**
jcarseyd2b45642009-05-11 18:02:16 +00001837 Checks the string for indicators of "flag" status. this is a leading '/', '-', or '+'
jcarsey94b17fa2009-05-07 18:46:18 +00001838
jcarseya405b862010-09-14 05:18:09 +00001839 @param[in] Name pointer to Name of parameter found
1840 @param[in] AlwaysAllowNumbers TRUE to allow numbers, FALSE to not.
jcarsey94b17fa2009-05-07 18:46:18 +00001841
1842 @retval TRUE the Parameter is a flag.
jcarseya405b862010-09-14 05:18:09 +00001843 @retval FALSE the Parameter not a flag.
jcarsey94b17fa2009-05-07 18:46:18 +00001844**/
1845BOOLEAN
1846EFIAPI
jcarseyd2b45642009-05-11 18:02:16 +00001847InternalIsFlag (
jcarsey2247dde2009-11-09 18:08:58 +00001848 IN CONST CHAR16 *Name,
1849 IN BOOLEAN AlwaysAllowNumbers
jcarsey94b17fa2009-05-07 18:46:18 +00001850 )
1851{
1852 //
1853 // ASSERT that Name isn't NULL
1854 //
1855 ASSERT(Name != NULL);
1856
1857 //
jcarsey2247dde2009-11-09 18:08:58 +00001858 // If we accept numbers then dont return TRUE. (they will be values)
1859 //
jcarseya405b862010-09-14 05:18:09 +00001860 if (((Name[0] == L'-' || Name[0] == L'+') && ShellIsHexaDecimalDigitCharacter(Name[1])) && AlwaysAllowNumbers) {
jcarsey2247dde2009-11-09 18:08:58 +00001861 return (FALSE);
1862 }
1863
1864 //
jcarseya405b862010-09-14 05:18:09 +00001865 // If the Name has a /, +, or - as the first character return TRUE
jcarsey94b17fa2009-05-07 18:46:18 +00001866 //
jcarsey1e6e84c2010-01-25 20:05:08 +00001867 if ((Name[0] == L'/') ||
jcarseyd2b45642009-05-11 18:02:16 +00001868 (Name[0] == L'-') ||
1869 (Name[0] == L'+')
jcarseya405b862010-09-14 05:18:09 +00001870 ) {
jcarsey94b17fa2009-05-07 18:46:18 +00001871 return (TRUE);
1872 }
1873 return (FALSE);
1874}
1875
1876/**
jcarsey1e6e84c2010-01-25 20:05:08 +00001877 Checks the command line arguments passed against the list of valid ones.
jcarsey94b17fa2009-05-07 18:46:18 +00001878
1879 If no initialization is required, then return RETURN_SUCCESS.
jcarsey1e6e84c2010-01-25 20:05:08 +00001880
jcarseya405b862010-09-14 05:18:09 +00001881 @param[in] CheckList pointer to list of parameters to check
1882 @param[out] CheckPackage pointer to pointer to list checked values
1883 @param[out] ProblemParam optional pointer to pointer to unicode string for
jcarseyd2b45642009-05-11 18:02:16 +00001884 the paramater that caused failure. If used then the
1885 caller is responsible for freeing the memory.
jcarseya405b862010-09-14 05:18:09 +00001886 @param[in] AutoPageBreak will automatically set PageBreakEnabled for "b" parameter
1887 @param[in] Argv pointer to array of parameters
1888 @param[in] Argc Count of parameters in Argv
1889 @param[in] AlwaysAllowNumbers TRUE to allow numbers always, FALSE otherwise.
jcarsey94b17fa2009-05-07 18:46:18 +00001890
1891 @retval EFI_SUCCESS The operation completed sucessfully.
1892 @retval EFI_OUT_OF_RESOURCES A memory allocation failed
1893 @retval EFI_INVALID_PARAMETER A parameter was invalid
jcarsey1e6e84c2010-01-25 20:05:08 +00001894 @retval EFI_VOLUME_CORRUPTED the command line was corrupt. an argument was
1895 duplicated. the duplicated command line argument
jcarsey94b17fa2009-05-07 18:46:18 +00001896 was returned in ProblemParam if provided.
jcarsey1e6e84c2010-01-25 20:05:08 +00001897 @retval EFI_NOT_FOUND a argument required a value that was missing.
jcarsey94b17fa2009-05-07 18:46:18 +00001898 the invalid command line argument was returned in
1899 ProblemParam if provided.
1900**/
1901EFI_STATUS
1902EFIAPI
1903InternalCommandLineParse (
1904 IN CONST SHELL_PARAM_ITEM *CheckList,
1905 OUT LIST_ENTRY **CheckPackage,
1906 OUT CHAR16 **ProblemParam OPTIONAL,
1907 IN BOOLEAN AutoPageBreak,
1908 IN CONST CHAR16 **Argv,
jcarsey2247dde2009-11-09 18:08:58 +00001909 IN UINTN Argc,
1910 IN BOOLEAN AlwaysAllowNumbers
jcarseya405b862010-09-14 05:18:09 +00001911 )
1912{
jcarsey94b17fa2009-05-07 18:46:18 +00001913 UINTN LoopCounter;
jcarsey252d9452011-03-25 20:49:53 +00001914 SHELL_PARAM_TYPE CurrentItemType;
jcarsey94b17fa2009-05-07 18:46:18 +00001915 SHELL_PARAM_PACKAGE *CurrentItemPackage;
jcarsey125c2cf2009-11-18 21:36:50 +00001916 UINTN GetItemValue;
1917 UINTN ValueSize;
jcarseya405b862010-09-14 05:18:09 +00001918 UINTN Count;
jcarsey252d9452011-03-25 20:49:53 +00001919 CONST CHAR16 *TempPointer;
jcarsey94b17fa2009-05-07 18:46:18 +00001920
1921 CurrentItemPackage = NULL;
jcarsey125c2cf2009-11-18 21:36:50 +00001922 GetItemValue = 0;
1923 ValueSize = 0;
jcarseya405b862010-09-14 05:18:09 +00001924 Count = 0;
jcarsey94b17fa2009-05-07 18:46:18 +00001925
1926 //
1927 // If there is only 1 item we dont need to do anything
1928 //
jcarseya405b862010-09-14 05:18:09 +00001929 if (Argc < 1) {
jcarsey94b17fa2009-05-07 18:46:18 +00001930 *CheckPackage = NULL;
1931 return (EFI_SUCCESS);
1932 }
1933
1934 //
jcarsey2247dde2009-11-09 18:08:58 +00001935 // ASSERTs
1936 //
1937 ASSERT(CheckList != NULL);
1938 ASSERT(Argv != NULL);
1939
1940 //
jcarsey94b17fa2009-05-07 18:46:18 +00001941 // initialize the linked list
1942 //
1943 *CheckPackage = (LIST_ENTRY*)AllocateZeroPool(sizeof(LIST_ENTRY));
sfu502a758c2011-10-08 02:55:30 +00001944 if (*CheckPackage == NULL) {
jcarseybeab0fc2011-10-10 17:26:25 +00001945 return (EFI_OUT_OF_RESOURCES);
sfu502a758c2011-10-08 02:55:30 +00001946 }
jcarseybeab0fc2011-10-10 17:26:25 +00001947
jcarsey94b17fa2009-05-07 18:46:18 +00001948 InitializeListHead(*CheckPackage);
1949
1950 //
1951 // loop through each of the arguments
1952 //
1953 for (LoopCounter = 0 ; LoopCounter < Argc ; ++LoopCounter) {
1954 if (Argv[LoopCounter] == NULL) {
1955 //
1956 // do nothing for NULL argv
1957 //
jcarseya405b862010-09-14 05:18:09 +00001958 } else if (InternalIsOnCheckList(Argv[LoopCounter], CheckList, &CurrentItemType)) {
jcarsey94b17fa2009-05-07 18:46:18 +00001959 //
jcarsey2247dde2009-11-09 18:08:58 +00001960 // We might have leftover if last parameter didnt have optional value
1961 //
jcarsey125c2cf2009-11-18 21:36:50 +00001962 if (GetItemValue != 0) {
1963 GetItemValue = 0;
jcarsey2247dde2009-11-09 18:08:58 +00001964 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
1965 }
1966 //
jcarsey94b17fa2009-05-07 18:46:18 +00001967 // this is a flag
1968 //
jcarsey252d9452011-03-25 20:49:53 +00001969 CurrentItemPackage = AllocateZeroPool(sizeof(SHELL_PARAM_PACKAGE));
jcarseybeab0fc2011-10-10 17:26:25 +00001970 if (CurrentItemPackage == NULL) {
1971 ShellCommandLineFreeVarList(*CheckPackage);
1972 *CheckPackage = NULL;
1973 return (EFI_OUT_OF_RESOURCES);
1974 }
jcarsey252d9452011-03-25 20:49:53 +00001975 CurrentItemPackage->Name = AllocateZeroPool(StrSize(Argv[LoopCounter]));
jcarseybeab0fc2011-10-10 17:26:25 +00001976 if (CurrentItemPackage->Name == NULL) {
1977 ShellCommandLineFreeVarList(*CheckPackage);
1978 *CheckPackage = NULL;
1979 return (EFI_OUT_OF_RESOURCES);
1980 }
jcarsey94b17fa2009-05-07 18:46:18 +00001981 StrCpy(CurrentItemPackage->Name, Argv[LoopCounter]);
1982 CurrentItemPackage->Type = CurrentItemType;
1983 CurrentItemPackage->OriginalPosition = (UINTN)(-1);
jcarseyb1f95a02009-06-16 00:23:19 +00001984 CurrentItemPackage->Value = NULL;
jcarsey94b17fa2009-05-07 18:46:18 +00001985
1986 //
1987 // Does this flag require a value
1988 //
jcarsey125c2cf2009-11-18 21:36:50 +00001989 switch (CurrentItemPackage->Type) {
jcarsey94b17fa2009-05-07 18:46:18 +00001990 //
jcarsey125c2cf2009-11-18 21:36:50 +00001991 // possibly trigger the next loop(s) to populate the value of this item
jcarsey1e6e84c2010-01-25 20:05:08 +00001992 //
jcarsey125c2cf2009-11-18 21:36:50 +00001993 case TypeValue:
jcarsey1e6e84c2010-01-25 20:05:08 +00001994 GetItemValue = 1;
jcarsey125c2cf2009-11-18 21:36:50 +00001995 ValueSize = 0;
1996 break;
1997 case TypeDoubleValue:
1998 GetItemValue = 2;
1999 ValueSize = 0;
2000 break;
2001 case TypeMaxValue:
2002 GetItemValue = (UINTN)(-1);
2003 ValueSize = 0;
2004 break;
2005 default:
2006 //
2007 // this item has no value expected; we are done
2008 //
2009 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2010 ASSERT(GetItemValue == 0);
2011 break;
jcarsey94b17fa2009-05-07 18:46:18 +00002012 }
jcarseya405b862010-09-14 05:18:09 +00002013 } else if (GetItemValue != 0 && !InternalIsFlag(Argv[LoopCounter], AlwaysAllowNumbers)) {
jcarseyb1f95a02009-06-16 00:23:19 +00002014 ASSERT(CurrentItemPackage != NULL);
2015 //
jcarsey125c2cf2009-11-18 21:36:50 +00002016 // get the item VALUE for a previous flag
jcarseyb1f95a02009-06-16 00:23:19 +00002017 //
jcarsey49bd4982012-01-27 18:42:43 +00002018 if (StrStr(Argv[LoopCounter], L" ") == NULL) {
2019 CurrentItemPackage->Value = ReallocatePool(ValueSize, ValueSize + StrSize(Argv[LoopCounter]) + sizeof(CHAR16), CurrentItemPackage->Value);
2020 ASSERT(CurrentItemPackage->Value != NULL);
2021 if (ValueSize == 0) {
2022 StrCpy(CurrentItemPackage->Value, Argv[LoopCounter]);
2023 } else {
2024 StrCat(CurrentItemPackage->Value, L" ");
2025 StrCat(CurrentItemPackage->Value, Argv[LoopCounter]);
2026 }
2027 ValueSize += StrSize(Argv[LoopCounter]) + sizeof(CHAR16);
jcarsey125c2cf2009-11-18 21:36:50 +00002028 } else {
jcarsey49bd4982012-01-27 18:42:43 +00002029 //
2030 // the parameter has spaces. must be quoted.
2031 //
2032 CurrentItemPackage->Value = ReallocatePool(ValueSize, ValueSize + StrSize(Argv[LoopCounter]) + sizeof(CHAR16) + sizeof(CHAR16) + sizeof(CHAR16), CurrentItemPackage->Value);
2033 ASSERT(CurrentItemPackage->Value != NULL);
2034 if (ValueSize == 0) {
2035 StrCpy(CurrentItemPackage->Value, L"\"");
2036 StrCat(CurrentItemPackage->Value, Argv[LoopCounter]);
2037 StrCat(CurrentItemPackage->Value, L"\"");
2038 } else {
2039 StrCat(CurrentItemPackage->Value, L" ");
2040 StrCat(CurrentItemPackage->Value, L"\"");
2041 StrCat(CurrentItemPackage->Value, Argv[LoopCounter]);
2042 StrCat(CurrentItemPackage->Value, L"\"");
2043 }
2044 ValueSize += StrSize(Argv[LoopCounter]) + sizeof(CHAR16);
jcarsey125c2cf2009-11-18 21:36:50 +00002045 }
jcarsey125c2cf2009-11-18 21:36:50 +00002046 GetItemValue--;
2047 if (GetItemValue == 0) {
2048 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2049 }
jcarseya405b862010-09-14 05:18:09 +00002050 } else if (!InternalIsFlag(Argv[LoopCounter], AlwaysAllowNumbers) ){ //|| ProblemParam == NULL) {
jcarseyb1f95a02009-06-16 00:23:19 +00002051 //
2052 // add this one as a non-flag
2053 //
jcarsey252d9452011-03-25 20:49:53 +00002054
2055 TempPointer = Argv[LoopCounter];
darylm503b0934ac2011-11-12 00:35:11 +00002056 if ((*TempPointer == L'^' && *(TempPointer+1) == L'-')
jcarsey252d9452011-03-25 20:49:53 +00002057 || (*TempPointer == L'^' && *(TempPointer+1) == L'/')
2058 || (*TempPointer == L'^' && *(TempPointer+1) == L'+')
2059 ){
2060 TempPointer++;
2061 }
2062 CurrentItemPackage = AllocateZeroPool(sizeof(SHELL_PARAM_PACKAGE));
jcarseybeab0fc2011-10-10 17:26:25 +00002063 if (CurrentItemPackage == NULL) {
2064 ShellCommandLineFreeVarList(*CheckPackage);
2065 *CheckPackage = NULL;
2066 return (EFI_OUT_OF_RESOURCES);
2067 }
jcarseyb1f95a02009-06-16 00:23:19 +00002068 CurrentItemPackage->Name = NULL;
2069 CurrentItemPackage->Type = TypePosition;
jcarsey252d9452011-03-25 20:49:53 +00002070 CurrentItemPackage->Value = AllocateZeroPool(StrSize(TempPointer));
jcarseybeab0fc2011-10-10 17:26:25 +00002071 if (CurrentItemPackage->Value == NULL) {
2072 ShellCommandLineFreeVarList(*CheckPackage);
2073 *CheckPackage = NULL;
2074 return (EFI_OUT_OF_RESOURCES);
2075 }
jcarsey252d9452011-03-25 20:49:53 +00002076 StrCpy(CurrentItemPackage->Value, TempPointer);
jcarseya405b862010-09-14 05:18:09 +00002077 CurrentItemPackage->OriginalPosition = Count++;
jcarsey9b3bf082009-06-23 21:15:07 +00002078 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
jcarsey252d9452011-03-25 20:49:53 +00002079 } else {
jcarsey94b17fa2009-05-07 18:46:18 +00002080 //
2081 // this was a non-recognised flag... error!
2082 //
jcarsey252d9452011-03-25 20:49:53 +00002083 if (ProblemParam != NULL) {
2084 *ProblemParam = AllocateZeroPool(StrSize(Argv[LoopCounter]));
jcarseybeab0fc2011-10-10 17:26:25 +00002085 if (*ProblemParam != NULL) {
darylm503b0934ac2011-11-12 00:35:11 +00002086 StrCpy(*ProblemParam, Argv[LoopCounter]);
jcarseybeab0fc2011-10-10 17:26:25 +00002087 }
jcarsey252d9452011-03-25 20:49:53 +00002088 }
jcarsey94b17fa2009-05-07 18:46:18 +00002089 ShellCommandLineFreeVarList(*CheckPackage);
2090 *CheckPackage = NULL;
2091 return (EFI_VOLUME_CORRUPTED);
jcarsey94b17fa2009-05-07 18:46:18 +00002092 }
2093 }
jcarsey125c2cf2009-11-18 21:36:50 +00002094 if (GetItemValue != 0) {
2095 GetItemValue = 0;
2096 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2097 }
jcarsey94b17fa2009-05-07 18:46:18 +00002098 //
2099 // support for AutoPageBreak
2100 //
2101 if (AutoPageBreak && ShellCommandLineGetFlag(*CheckPackage, L"-b")) {
2102 ShellSetPageBreakMode(TRUE);
2103 }
2104 return (EFI_SUCCESS);
2105}
2106
2107/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002108 Checks the command line arguments passed against the list of valid ones.
jcarsey94b17fa2009-05-07 18:46:18 +00002109 Optionally removes NULL values first.
jcarsey1e6e84c2010-01-25 20:05:08 +00002110
jcarsey94b17fa2009-05-07 18:46:18 +00002111 If no initialization is required, then return RETURN_SUCCESS.
jcarsey1e6e84c2010-01-25 20:05:08 +00002112
jcarseya405b862010-09-14 05:18:09 +00002113 @param[in] CheckList The pointer to list of parameters to check.
2114 @param[out] CheckPackage The package of checked values.
2115 @param[out] ProblemParam Optional pointer to pointer to unicode string for
jcarsey94b17fa2009-05-07 18:46:18 +00002116 the paramater that caused failure.
jcarseya405b862010-09-14 05:18:09 +00002117 @param[in] AutoPageBreak Will automatically set PageBreakEnabled.
2118 @param[in] AlwaysAllowNumbers Will never fail for number based flags.
jcarsey94b17fa2009-05-07 18:46:18 +00002119
2120 @retval EFI_SUCCESS The operation completed sucessfully.
jcarseya405b862010-09-14 05:18:09 +00002121 @retval EFI_OUT_OF_RESOURCES A memory allocation failed.
2122 @retval EFI_INVALID_PARAMETER A parameter was invalid.
2123 @retval EFI_VOLUME_CORRUPTED The command line was corrupt.
2124 @retval EFI_DEVICE_ERROR The commands contained 2 opposing arguments. One
jcarsey1e6e84c2010-01-25 20:05:08 +00002125 of the command line arguments was returned in
jcarsey94b17fa2009-05-07 18:46:18 +00002126 ProblemParam if provided.
jcarseya405b862010-09-14 05:18:09 +00002127 @retval EFI_NOT_FOUND A argument required a value that was missing.
2128 The invalid command line argument was returned in
jcarsey94b17fa2009-05-07 18:46:18 +00002129 ProblemParam if provided.
2130**/
2131EFI_STATUS
2132EFIAPI
jcarsey2247dde2009-11-09 18:08:58 +00002133ShellCommandLineParseEx (
jcarsey94b17fa2009-05-07 18:46:18 +00002134 IN CONST SHELL_PARAM_ITEM *CheckList,
2135 OUT LIST_ENTRY **CheckPackage,
2136 OUT CHAR16 **ProblemParam OPTIONAL,
jcarsey2247dde2009-11-09 18:08:58 +00002137 IN BOOLEAN AutoPageBreak,
2138 IN BOOLEAN AlwaysAllowNumbers
jcarseya405b862010-09-14 05:18:09 +00002139 )
2140{
jcarsey1e6e84c2010-01-25 20:05:08 +00002141 //
jcarsey94b17fa2009-05-07 18:46:18 +00002142 // ASSERT that CheckList and CheckPackage aren't NULL
2143 //
2144 ASSERT(CheckList != NULL);
2145 ASSERT(CheckPackage != NULL);
2146
jcarsey1e6e84c2010-01-25 20:05:08 +00002147 //
jcarsey94b17fa2009-05-07 18:46:18 +00002148 // Check for UEFI Shell 2.0 protocols
2149 //
jcarsey366f81a2011-06-27 21:04:22 +00002150 if (gEfiShellParametersProtocol != NULL) {
jcarsey1e6e84c2010-01-25 20:05:08 +00002151 return (InternalCommandLineParse(CheckList,
2152 CheckPackage,
2153 ProblemParam,
2154 AutoPageBreak,
jcarsey366f81a2011-06-27 21:04:22 +00002155 (CONST CHAR16**) gEfiShellParametersProtocol->Argv,
2156 gEfiShellParametersProtocol->Argc,
jcarsey2247dde2009-11-09 18:08:58 +00002157 AlwaysAllowNumbers));
jcarsey94b17fa2009-05-07 18:46:18 +00002158 }
2159
jcarsey1e6e84c2010-01-25 20:05:08 +00002160 //
jcarsey94b17fa2009-05-07 18:46:18 +00002161 // ASSERT That EFI Shell is not required
2162 //
2163 ASSERT (mEfiShellInterface != NULL);
jcarsey1e6e84c2010-01-25 20:05:08 +00002164 return (InternalCommandLineParse(CheckList,
2165 CheckPackage,
2166 ProblemParam,
2167 AutoPageBreak,
jljusten08d7f8e2009-06-15 18:42:13 +00002168 (CONST CHAR16**) mEfiShellInterface->Argv,
jcarsey2247dde2009-11-09 18:08:58 +00002169 mEfiShellInterface->Argc,
2170 AlwaysAllowNumbers));
jcarsey94b17fa2009-05-07 18:46:18 +00002171}
2172
2173/**
2174 Frees shell variable list that was returned from ShellCommandLineParse.
2175
2176 This function will free all the memory that was used for the CheckPackage
2177 list of postprocessed shell arguments.
2178
2179 this function has no return value.
2180
2181 if CheckPackage is NULL, then return
2182
2183 @param CheckPackage the list to de-allocate
2184 **/
2185VOID
2186EFIAPI
2187ShellCommandLineFreeVarList (
2188 IN LIST_ENTRY *CheckPackage
jcarseya405b862010-09-14 05:18:09 +00002189 )
2190{
jcarsey94b17fa2009-05-07 18:46:18 +00002191 LIST_ENTRY *Node;
2192
2193 //
2194 // check for CheckPackage == NULL
2195 //
2196 if (CheckPackage == NULL) {
2197 return;
2198 }
2199
2200 //
2201 // for each node in the list
2202 //
jcarsey9eb53ac2009-07-08 17:26:58 +00002203 for ( Node = GetFirstNode(CheckPackage)
jcarseya405b862010-09-14 05:18:09 +00002204 ; !IsListEmpty(CheckPackage)
jcarsey9eb53ac2009-07-08 17:26:58 +00002205 ; Node = GetFirstNode(CheckPackage)
jcarseya405b862010-09-14 05:18:09 +00002206 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002207 //
2208 // Remove it from the list
2209 //
2210 RemoveEntryList(Node);
2211
2212 //
2213 // if it has a name free the name
2214 //
2215 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
2216 FreePool(((SHELL_PARAM_PACKAGE*)Node)->Name);
2217 }
2218
2219 //
2220 // if it has a value free the value
2221 //
2222 if (((SHELL_PARAM_PACKAGE*)Node)->Value != NULL) {
2223 FreePool(((SHELL_PARAM_PACKAGE*)Node)->Value);
2224 }
jcarsey1e6e84c2010-01-25 20:05:08 +00002225
jcarsey94b17fa2009-05-07 18:46:18 +00002226 //
2227 // free the node structure
2228 //
2229 FreePool((SHELL_PARAM_PACKAGE*)Node);
2230 }
2231 //
2232 // free the list head node
2233 //
2234 FreePool(CheckPackage);
2235}
2236/**
2237 Checks for presence of a flag parameter
2238
2239 flag arguments are in the form of "-<Key>" or "/<Key>", but do not have a value following the key
2240
2241 if CheckPackage is NULL then return FALSE.
2242 if KeyString is NULL then ASSERT()
jcarsey1e6e84c2010-01-25 20:05:08 +00002243
jcarsey94b17fa2009-05-07 18:46:18 +00002244 @param CheckPackage The package of parsed command line arguments
2245 @param KeyString the Key of the command line argument to check for
2246
2247 @retval TRUE the flag is on the command line
2248 @retval FALSE the flag is not on the command line
2249 **/
2250BOOLEAN
2251EFIAPI
2252ShellCommandLineGetFlag (
jcarseya405b862010-09-14 05:18:09 +00002253 IN CONST LIST_ENTRY * CONST CheckPackage,
2254 IN CONST CHAR16 * CONST KeyString
2255 )
2256{
jcarsey94b17fa2009-05-07 18:46:18 +00002257 LIST_ENTRY *Node;
jcarsey252d9452011-03-25 20:49:53 +00002258 CHAR16 *TempString;
jcarsey94b17fa2009-05-07 18:46:18 +00002259
2260 //
ydong100c1950b2011-10-13 02:37:35 +00002261 // return FALSE for no package or KeyString is NULL
jcarsey94b17fa2009-05-07 18:46:18 +00002262 //
ydong100c1950b2011-10-13 02:37:35 +00002263 if (CheckPackage == NULL || KeyString == NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00002264 return (FALSE);
2265 }
2266
2267 //
2268 // enumerate through the list of parametrs
2269 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002270 for ( Node = GetFirstNode(CheckPackage)
2271 ; !IsNull (CheckPackage, Node)
2272 ; Node = GetNextNode(CheckPackage, Node)
jcarsey252d9452011-03-25 20:49:53 +00002273 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002274 //
2275 // If the Name matches, return TRUE (and there may be NULL name)
2276 //
2277 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
jcarsey9eb53ac2009-07-08 17:26:58 +00002278 //
2279 // If Type is TypeStart then only compare the begining of the strings
2280 //
jcarsey252d9452011-03-25 20:49:53 +00002281 if (((SHELL_PARAM_PACKAGE*)Node)->Type == TypeStart) {
2282 if (StrnCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name, StrLen(KeyString)) == 0) {
2283 return (TRUE);
2284 }
2285 TempString = NULL;
2286 TempString = StrnCatGrow(&TempString, NULL, KeyString, StrLen(((SHELL_PARAM_PACKAGE*)Node)->Name));
2287 if (TempString != NULL) {
2288 if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2289 FreePool(TempString);
2290 return (TRUE);
2291 }
2292 FreePool(TempString);
2293 }
2294 } else if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
jcarsey94b17fa2009-05-07 18:46:18 +00002295 return (TRUE);
2296 }
2297 }
2298 }
2299 return (FALSE);
2300}
2301/**
jcarseya405b862010-09-14 05:18:09 +00002302 Returns value from command line argument.
jcarsey94b17fa2009-05-07 18:46:18 +00002303
jcarseya405b862010-09-14 05:18:09 +00002304 Value parameters are in the form of "-<Key> value" or "/<Key> value".
jcarsey1e6e84c2010-01-25 20:05:08 +00002305
jcarseya405b862010-09-14 05:18:09 +00002306 If CheckPackage is NULL, then return NULL.
jcarsey94b17fa2009-05-07 18:46:18 +00002307
jcarseya405b862010-09-14 05:18:09 +00002308 @param[in] CheckPackage The package of parsed command line arguments.
2309 @param[in] KeyString The Key of the command line argument to check for.
jcarsey94b17fa2009-05-07 18:46:18 +00002310
jcarseya405b862010-09-14 05:18:09 +00002311 @retval NULL The flag is not on the command line.
2312 @retval !=NULL The pointer to unicode string of the value.
2313**/
jcarsey94b17fa2009-05-07 18:46:18 +00002314CONST CHAR16*
2315EFIAPI
2316ShellCommandLineGetValue (
2317 IN CONST LIST_ENTRY *CheckPackage,
2318 IN CHAR16 *KeyString
jcarseya405b862010-09-14 05:18:09 +00002319 )
2320{
jcarsey94b17fa2009-05-07 18:46:18 +00002321 LIST_ENTRY *Node;
jcarsey252d9452011-03-25 20:49:53 +00002322 CHAR16 *TempString;
jcarsey94b17fa2009-05-07 18:46:18 +00002323
2324 //
ydong100c1950b2011-10-13 02:37:35 +00002325 // return NULL for no package or KeyString is NULL
jcarsey94b17fa2009-05-07 18:46:18 +00002326 //
ydong100c1950b2011-10-13 02:37:35 +00002327 if (CheckPackage == NULL || KeyString == NULL) {
jcarsey94b17fa2009-05-07 18:46:18 +00002328 return (NULL);
2329 }
2330
2331 //
2332 // enumerate through the list of parametrs
2333 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002334 for ( Node = GetFirstNode(CheckPackage)
2335 ; !IsNull (CheckPackage, Node)
2336 ; Node = GetNextNode(CheckPackage, Node)
jcarsey252d9452011-03-25 20:49:53 +00002337 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002338 //
jcarsey252d9452011-03-25 20:49:53 +00002339 // If the Name matches, return TRUE (and there may be NULL name)
jcarsey94b17fa2009-05-07 18:46:18 +00002340 //
2341 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
jcarsey9eb53ac2009-07-08 17:26:58 +00002342 //
2343 // If Type is TypeStart then only compare the begining of the strings
2344 //
jcarsey252d9452011-03-25 20:49:53 +00002345 if (((SHELL_PARAM_PACKAGE*)Node)->Type == TypeStart) {
2346 if (StrnCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name, StrLen(KeyString)) == 0) {
2347 return (((SHELL_PARAM_PACKAGE*)Node)->Name + StrLen(KeyString));
2348 }
2349 TempString = NULL;
2350 TempString = StrnCatGrow(&TempString, NULL, KeyString, StrLen(((SHELL_PARAM_PACKAGE*)Node)->Name));
2351 if (TempString != NULL) {
2352 if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2353 FreePool(TempString);
2354 return (((SHELL_PARAM_PACKAGE*)Node)->Name + StrLen(KeyString));
2355 }
2356 FreePool(TempString);
2357 }
2358 } else if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
jcarsey94b17fa2009-05-07 18:46:18 +00002359 return (((SHELL_PARAM_PACKAGE*)Node)->Value);
2360 }
2361 }
2362 }
2363 return (NULL);
2364}
jcarseya405b862010-09-14 05:18:09 +00002365
jcarsey94b17fa2009-05-07 18:46:18 +00002366/**
jcarseya405b862010-09-14 05:18:09 +00002367 Returns raw value from command line argument.
jcarsey94b17fa2009-05-07 18:46:18 +00002368
jcarseya405b862010-09-14 05:18:09 +00002369 Raw value parameters are in the form of "value" in a specific position in the list.
jcarsey1e6e84c2010-01-25 20:05:08 +00002370
jcarseya405b862010-09-14 05:18:09 +00002371 If CheckPackage is NULL, then return NULL.
jcarsey94b17fa2009-05-07 18:46:18 +00002372
jcarseya405b862010-09-14 05:18:09 +00002373 @param[in] CheckPackage The package of parsed command line arguments.
2374 @param[in] Position The position of the value.
jcarsey94b17fa2009-05-07 18:46:18 +00002375
jcarseya405b862010-09-14 05:18:09 +00002376 @retval NULL The flag is not on the command line.
2377 @retval !=NULL The pointer to unicode string of the value.
jcarsey94b17fa2009-05-07 18:46:18 +00002378 **/
2379CONST CHAR16*
2380EFIAPI
2381ShellCommandLineGetRawValue (
jcarseya405b862010-09-14 05:18:09 +00002382 IN CONST LIST_ENTRY * CONST CheckPackage,
2383 IN UINTN Position
2384 )
2385{
jcarsey94b17fa2009-05-07 18:46:18 +00002386 LIST_ENTRY *Node;
2387
2388 //
2389 // check for CheckPackage == NULL
2390 //
2391 if (CheckPackage == NULL) {
2392 return (NULL);
2393 }
2394
2395 //
2396 // enumerate through the list of parametrs
2397 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002398 for ( Node = GetFirstNode(CheckPackage)
2399 ; !IsNull (CheckPackage, Node)
2400 ; Node = GetNextNode(CheckPackage, Node)
jcarseya405b862010-09-14 05:18:09 +00002401 ){
jcarsey94b17fa2009-05-07 18:46:18 +00002402 //
2403 // If the position matches, return the value
2404 //
2405 if (((SHELL_PARAM_PACKAGE*)Node)->OriginalPosition == Position) {
2406 return (((SHELL_PARAM_PACKAGE*)Node)->Value);
2407 }
2408 }
2409 return (NULL);
jcarseyb1f95a02009-06-16 00:23:19 +00002410}
jcarsey2247dde2009-11-09 18:08:58 +00002411
2412/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002413 returns the number of command line value parameters that were parsed.
2414
jcarsey2247dde2009-11-09 18:08:58 +00002415 this will not include flags.
2416
jcarseya405b862010-09-14 05:18:09 +00002417 @param[in] CheckPackage The package of parsed command line arguments.
2418
jcarsey2247dde2009-11-09 18:08:58 +00002419 @retval (UINTN)-1 No parsing has ocurred
2420 @return other The number of value parameters found
2421**/
2422UINTN
2423EFIAPI
2424ShellCommandLineGetCount(
jcarseya405b862010-09-14 05:18:09 +00002425 IN CONST LIST_ENTRY *CheckPackage
jcarsey125c2cf2009-11-18 21:36:50 +00002426 )
2427{
jcarseya405b862010-09-14 05:18:09 +00002428 LIST_ENTRY *Node1;
2429 UINTN Count;
2430
2431 if (CheckPackage == NULL) {
2432 return (0);
2433 }
2434 for ( Node1 = GetFirstNode(CheckPackage), Count = 0
2435 ; !IsNull (CheckPackage, Node1)
2436 ; Node1 = GetNextNode(CheckPackage, Node1)
2437 ){
2438 if (((SHELL_PARAM_PACKAGE*)Node1)->Name == NULL) {
2439 Count++;
2440 }
2441 }
2442 return (Count);
jcarsey2247dde2009-11-09 18:08:58 +00002443}
2444
jcarsey975136a2009-06-16 19:03:54 +00002445/**
jcarsey36a9d672009-11-20 21:13:41 +00002446 Determins if a parameter is duplicated.
2447
jcarsey1e6e84c2010-01-25 20:05:08 +00002448 If Param is not NULL then it will point to a callee allocated string buffer
jcarsey36a9d672009-11-20 21:13:41 +00002449 with the parameter value if a duplicate is found.
2450
2451 If CheckPackage is NULL, then ASSERT.
2452
2453 @param[in] CheckPackage The package of parsed command line arguments.
2454 @param[out] Param Upon finding one, a pointer to the duplicated parameter.
2455
2456 @retval EFI_SUCCESS No parameters were duplicated.
2457 @retval EFI_DEVICE_ERROR A duplicate was found.
2458 **/
2459EFI_STATUS
2460EFIAPI
2461ShellCommandLineCheckDuplicate (
2462 IN CONST LIST_ENTRY *CheckPackage,
2463 OUT CHAR16 **Param
2464 )
2465{
2466 LIST_ENTRY *Node1;
2467 LIST_ENTRY *Node2;
jcarsey1e6e84c2010-01-25 20:05:08 +00002468
jcarsey36a9d672009-11-20 21:13:41 +00002469 ASSERT(CheckPackage != NULL);
2470
jcarsey1e6e84c2010-01-25 20:05:08 +00002471 for ( Node1 = GetFirstNode(CheckPackage)
2472 ; !IsNull (CheckPackage, Node1)
2473 ; Node1 = GetNextNode(CheckPackage, Node1)
jcarseya405b862010-09-14 05:18:09 +00002474 ){
jcarsey1e6e84c2010-01-25 20:05:08 +00002475 for ( Node2 = GetNextNode(CheckPackage, Node1)
2476 ; !IsNull (CheckPackage, Node2)
2477 ; Node2 = GetNextNode(CheckPackage, Node2)
jcarseya405b862010-09-14 05:18:09 +00002478 ){
2479 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 +00002480 if (Param != NULL) {
2481 *Param = NULL;
2482 *Param = StrnCatGrow(Param, NULL, ((SHELL_PARAM_PACKAGE*)Node1)->Name, 0);
2483 }
2484 return (EFI_DEVICE_ERROR);
2485 }
2486 }
2487 }
2488 return (EFI_SUCCESS);
2489}
2490
2491/**
jcarsey1e6e84c2010-01-25 20:05:08 +00002492 This is a find and replace function. Upon successful return the NewString is a copy of
jcarsey975136a2009-06-16 19:03:54 +00002493 SourceString with each instance of FindTarget replaced with ReplaceWith.
2494
jcarseyb3011f42010-01-11 21:49:04 +00002495 If SourceString and NewString overlap the behavior is undefined.
2496
jcarsey975136a2009-06-16 19:03:54 +00002497 If the string would grow bigger than NewSize it will halt and return error.
2498
ydong104ff7e372011-09-02 08:05:34 +00002499 @param[in] SourceString The string with source buffer.
2500 @param[in, out] NewString The string with resultant buffer.
2501 @param[in] NewSize The size in bytes of NewString.
2502 @param[in] FindTarget The string to look for.
2503 @param[in] ReplaceWith The string to replace FindTarget with.
2504 @param[in] SkipPreCarrot If TRUE will skip a FindTarget that has a '^'
2505 immediately before it.
2506 @param[in] ParameterReplacing If TRUE will add "" around items with spaces.
jcarsey975136a2009-06-16 19:03:54 +00002507
jcarsey969c7832010-01-13 16:46:33 +00002508 @retval EFI_INVALID_PARAMETER SourceString was NULL.
2509 @retval EFI_INVALID_PARAMETER NewString was NULL.
2510 @retval EFI_INVALID_PARAMETER FindTarget was NULL.
2511 @retval EFI_INVALID_PARAMETER ReplaceWith was NULL.
2512 @retval EFI_INVALID_PARAMETER FindTarget had length < 1.
2513 @retval EFI_INVALID_PARAMETER SourceString had length < 1.
jcarsey1e6e84c2010-01-25 20:05:08 +00002514 @retval EFI_BUFFER_TOO_SMALL NewSize was less than the minimum size to hold
jcarsey969c7832010-01-13 16:46:33 +00002515 the new string (truncation occurred).
jcarseya405b862010-09-14 05:18:09 +00002516 @retval EFI_SUCCESS The string was successfully copied with replacement.
jcarsey975136a2009-06-16 19:03:54 +00002517**/
jcarsey975136a2009-06-16 19:03:54 +00002518EFI_STATUS
2519EFIAPI
jcarseya405b862010-09-14 05:18:09 +00002520ShellCopySearchAndReplace(
jcarsey975136a2009-06-16 19:03:54 +00002521 IN CHAR16 CONST *SourceString,
jcarseya405b862010-09-14 05:18:09 +00002522 IN OUT CHAR16 *NewString,
jcarsey975136a2009-06-16 19:03:54 +00002523 IN UINTN NewSize,
2524 IN CONST CHAR16 *FindTarget,
jcarsey969c7832010-01-13 16:46:33 +00002525 IN CONST CHAR16 *ReplaceWith,
jcarseya405b862010-09-14 05:18:09 +00002526 IN CONST BOOLEAN SkipPreCarrot,
2527 IN CONST BOOLEAN ParameterReplacing
jcarsey1e6e84c2010-01-25 20:05:08 +00002528 )
jcarsey2247dde2009-11-09 18:08:58 +00002529{
jcarsey01582942009-07-10 19:46:17 +00002530 UINTN Size;
jcarseya405b862010-09-14 05:18:09 +00002531 CHAR16 *Replace;
2532
jcarsey975136a2009-06-16 19:03:54 +00002533 if ( (SourceString == NULL)
2534 || (NewString == NULL)
2535 || (FindTarget == NULL)
2536 || (ReplaceWith == NULL)
2537 || (StrLen(FindTarget) < 1)
2538 || (StrLen(SourceString) < 1)
jcarseya405b862010-09-14 05:18:09 +00002539 ){
jcarsey975136a2009-06-16 19:03:54 +00002540 return (EFI_INVALID_PARAMETER);
2541 }
jcarseya405b862010-09-14 05:18:09 +00002542 Replace = NULL;
2543 if (StrStr(ReplaceWith, L" ") == NULL || !ParameterReplacing) {
2544 Replace = StrnCatGrow(&Replace, NULL, ReplaceWith, 0);
2545 } else {
2546 Replace = AllocateZeroPool(StrSize(ReplaceWith) + 2*sizeof(CHAR16));
jcarseybeab0fc2011-10-10 17:26:25 +00002547 if (Replace != NULL) {
2548 UnicodeSPrint(Replace, StrSize(ReplaceWith) + 2*sizeof(CHAR16), L"\"%s\"", ReplaceWith);
2549 }
jcarseya405b862010-09-14 05:18:09 +00002550 }
jcarsey3e082d52010-10-04 16:44:57 +00002551 if (Replace == NULL) {
2552 return (EFI_OUT_OF_RESOURCES);
2553 }
jcarsey2247dde2009-11-09 18:08:58 +00002554 NewString = SetMem16(NewString, NewSize, CHAR_NULL);
2555 while (*SourceString != CHAR_NULL) {
jcarsey969c7832010-01-13 16:46:33 +00002556 //
jcarseya405b862010-09-14 05:18:09 +00002557 // if we find the FindTarget and either Skip == FALSE or Skip and we
jcarsey969c7832010-01-13 16:46:33 +00002558 // dont have a carrot do a replace...
2559 //
jcarsey1e6e84c2010-01-25 20:05:08 +00002560 if (StrnCmp(SourceString, FindTarget, StrLen(FindTarget)) == 0
jcarseya405b862010-09-14 05:18:09 +00002561 && ((SkipPreCarrot && *(SourceString-1) != L'^') || !SkipPreCarrot)
2562 ){
jcarsey975136a2009-06-16 19:03:54 +00002563 SourceString += StrLen(FindTarget);
jcarsey01582942009-07-10 19:46:17 +00002564 Size = StrSize(NewString);
jcarseya405b862010-09-14 05:18:09 +00002565 if ((Size + (StrLen(Replace)*sizeof(CHAR16))) > NewSize) {
2566 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002567 return (EFI_BUFFER_TOO_SMALL);
2568 }
jcarseya405b862010-09-14 05:18:09 +00002569 StrCat(NewString, Replace);
jcarsey975136a2009-06-16 19:03:54 +00002570 } else {
jcarsey01582942009-07-10 19:46:17 +00002571 Size = StrSize(NewString);
2572 if (Size + sizeof(CHAR16) > NewSize) {
jcarseya405b862010-09-14 05:18:09 +00002573 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002574 return (EFI_BUFFER_TOO_SMALL);
2575 }
2576 StrnCat(NewString, SourceString, 1);
2577 SourceString++;
2578 }
2579 }
jcarseya405b862010-09-14 05:18:09 +00002580 FreePool(Replace);
jcarsey975136a2009-06-16 19:03:54 +00002581 return (EFI_SUCCESS);
2582}
jcarseyb1f95a02009-06-16 00:23:19 +00002583
2584/**
jcarseye2f82972009-12-01 05:40:24 +00002585 Internal worker function to output a string.
2586
2587 This function will output a string to the correct StdOut.
2588
2589 @param[in] String The string to print out.
2590
2591 @retval EFI_SUCCESS The operation was sucessful.
2592 @retval !EFI_SUCCESS The operation failed.
2593**/
2594EFI_STATUS
2595EFIAPI
2596InternalPrintTo (
2597 IN CONST CHAR16 *String
2598 )
2599{
2600 UINTN Size;
2601 Size = StrSize(String) - sizeof(CHAR16);
jcarseya405b862010-09-14 05:18:09 +00002602 if (Size == 0) {
2603 return (EFI_SUCCESS);
2604 }
jcarsey366f81a2011-06-27 21:04:22 +00002605 if (gEfiShellParametersProtocol != NULL) {
2606 return (gEfiShellProtocol->WriteFile(gEfiShellParametersProtocol->StdOut, &Size, (VOID*)String));
jcarseye2f82972009-12-01 05:40:24 +00002607 }
2608 if (mEfiShellInterface != NULL) {
jcarsey49bd4982012-01-27 18:42:43 +00002609 //
2610 // Divide in half for old shell. Must be string length not size.
2611 //
2612 Size /= 2;
jcarseya405b862010-09-14 05:18:09 +00002613 return (mEfiShellInterface->StdOut->Write(mEfiShellInterface->StdOut, &Size, (VOID*)String));
jcarseye2f82972009-12-01 05:40:24 +00002614 }
2615 ASSERT(FALSE);
2616 return (EFI_UNSUPPORTED);
2617}
2618
2619/**
jcarseyb1f95a02009-06-16 00:23:19 +00002620 Print at a specific location on the screen.
2621
jcarseyf1b87e72009-06-17 00:52:11 +00002622 This function will move the cursor to a given screen location and print the specified string
jcarsey1e6e84c2010-01-25 20:05:08 +00002623
2624 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarseyf1b87e72009-06-17 00:52:11 +00002625 will be used.
jcarseyb1f95a02009-06-16 00:23:19 +00002626
2627 if either Row or Col is out of range for the current console, then ASSERT
2628 if Format is NULL, then ASSERT
2629
jcarsey1e6e84c2010-01-25 20:05:08 +00002630 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarseyb1f95a02009-06-16 00:23:19 +00002631 the following additional flags:
2632 %N - Set output attribute to normal
2633 %H - Set output attribute to highlight
2634 %E - Set output attribute to error
2635 %B - Set output attribute to blue color
2636 %V - Set output attribute to green color
2637
2638 Note: The background color is controlled by the shell command cls.
2639
jcarseyb1f95a02009-06-16 00:23:19 +00002640 @param[in] Col the column to print at
jcarsey252d9452011-03-25 20:49:53 +00002641 @param[in] Row the row to print at
jcarseyb1f95a02009-06-16 00:23:19 +00002642 @param[in] Format the format string
jcarsey2247dde2009-11-09 18:08:58 +00002643 @param[in] Marker the marker for the variable argument list
jcarseyb1f95a02009-06-16 00:23:19 +00002644
jcarseya405b862010-09-14 05:18:09 +00002645 @return EFI_SUCCESS The operation was successful.
2646 @return EFI_DEVICE_ERROR The console device reported an error.
jcarseyb1f95a02009-06-16 00:23:19 +00002647**/
jcarseya405b862010-09-14 05:18:09 +00002648EFI_STATUS
jcarseyb1f95a02009-06-16 00:23:19 +00002649EFIAPI
jcarsey2247dde2009-11-09 18:08:58 +00002650InternalShellPrintWorker(
jcarseyb1f95a02009-06-16 00:23:19 +00002651 IN INT32 Col OPTIONAL,
2652 IN INT32 Row OPTIONAL,
2653 IN CONST CHAR16 *Format,
jcarsey252d9452011-03-25 20:49:53 +00002654 IN VA_LIST Marker
jcarsey1e6e84c2010-01-25 20:05:08 +00002655 )
jcarsey2247dde2009-11-09 18:08:58 +00002656{
jcarseyb1f95a02009-06-16 00:23:19 +00002657 EFI_STATUS Status;
jcarsey975136a2009-06-16 19:03:54 +00002658 CHAR16 *ResumeLocation;
2659 CHAR16 *FormatWalker;
jcarseya405b862010-09-14 05:18:09 +00002660 UINTN OriginalAttribute;
jcarsey89e85372011-04-13 23:37:50 +00002661 CHAR16 *mPostReplaceFormat;
2662 CHAR16 *mPostReplaceFormat2;
2663
2664 mPostReplaceFormat = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
2665 mPostReplaceFormat2 = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
jcarseya405b862010-09-14 05:18:09 +00002666
jcarseyf8d3e682011-04-19 17:54:42 +00002667 if (mPostReplaceFormat == NULL || mPostReplaceFormat2 == NULL) {
2668 SHELL_FREE_NON_NULL(mPostReplaceFormat);
2669 SHELL_FREE_NON_NULL(mPostReplaceFormat2);
2670 return (EFI_OUT_OF_RESOURCES);
2671 }
2672
jcarseya405b862010-09-14 05:18:09 +00002673 Status = EFI_SUCCESS;
2674 OriginalAttribute = gST->ConOut->Mode->Attribute;
jcarsey1e6e84c2010-01-25 20:05:08 +00002675
jcarsey975136a2009-06-16 19:03:54 +00002676 //
2677 // Back and forth each time fixing up 1 of our flags...
2678 //
jcarseya405b862010-09-14 05:18:09 +00002679 Status = ShellCopySearchAndReplace(Format, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%N", L"%%N", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002680 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002681 Status = ShellCopySearchAndReplace(mPostReplaceFormat, mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), L"%E", L"%%E", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002682 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002683 Status = ShellCopySearchAndReplace(mPostReplaceFormat2, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%H", L"%%H", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002684 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002685 Status = ShellCopySearchAndReplace(mPostReplaceFormat, mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), L"%B", L"%%B", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002686 ASSERT_EFI_ERROR(Status);
jcarseya405b862010-09-14 05:18:09 +00002687 Status = ShellCopySearchAndReplace(mPostReplaceFormat2, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%V", L"%%V", FALSE, FALSE);
jcarsey975136a2009-06-16 19:03:54 +00002688 ASSERT_EFI_ERROR(Status);
2689
2690 //
2691 // Use the last buffer from replacing to print from...
2692 //
jcarseya405b862010-09-14 05:18:09 +00002693 UnicodeVSPrint (mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), mPostReplaceFormat, Marker);
jcarseyb1f95a02009-06-16 00:23:19 +00002694
2695 if (Col != -1 && Row != -1) {
jcarseyb1f95a02009-06-16 00:23:19 +00002696 Status = gST->ConOut->SetCursorPosition(gST->ConOut, Col, Row);
jcarsey975136a2009-06-16 19:03:54 +00002697 }
2698
jcarseyecd3d592009-12-07 18:05:00 +00002699 FormatWalker = mPostReplaceFormat2;
jcarsey2247dde2009-11-09 18:08:58 +00002700 while (*FormatWalker != CHAR_NULL) {
jcarsey975136a2009-06-16 19:03:54 +00002701 //
2702 // Find the next attribute change request
2703 //
2704 ResumeLocation = StrStr(FormatWalker, L"%");
2705 if (ResumeLocation != NULL) {
jcarsey2247dde2009-11-09 18:08:58 +00002706 *ResumeLocation = CHAR_NULL;
jcarsey975136a2009-06-16 19:03:54 +00002707 }
2708 //
2709 // print the current FormatWalker string
2710 //
jcarseya405b862010-09-14 05:18:09 +00002711 if (StrLen(FormatWalker)>0) {
2712 Status = InternalPrintTo(FormatWalker);
2713 if (EFI_ERROR(Status)) {
2714 break;
2715 }
2716 }
2717
jcarsey975136a2009-06-16 19:03:54 +00002718 //
2719 // update the attribute
2720 //
2721 if (ResumeLocation != NULL) {
jcarsey5d46f172011-08-23 15:34:23 +00002722 if (*(ResumeLocation-1) == L'^') {
2723 //
jcarsey8bb74412012-01-30 18:44:41 +00002724 // Move cursor back 1 position to overwrite the ^
2725 //
2726 gST->ConOut->SetCursorPosition(gST->ConOut, gST->ConOut->Mode->CursorColumn - 1, gST->ConOut->Mode->CursorRow);
2727
2728 //
jcarsey5d46f172011-08-23 15:34:23 +00002729 // Print a simple '%' symbol
2730 //
2731 Status = InternalPrintTo(L"%");
2732 ResumeLocation = ResumeLocation - 1;
2733 } else {
2734 switch (*(ResumeLocation+1)) {
2735 case (L'N'):
2736 gST->ConOut->SetAttribute(gST->ConOut, OriginalAttribute);
jcarseya405b862010-09-14 05:18:09 +00002737 break;
jcarsey5d46f172011-08-23 15:34:23 +00002738 case (L'E'):
2739 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_YELLOW, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2740 break;
2741 case (L'H'):
2742 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_WHITE, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2743 break;
2744 case (L'B'):
2745 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_BLUE, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2746 break;
2747 case (L'V'):
2748 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_GREEN, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2749 break;
2750 default:
2751 //
2752 // Print a simple '%' symbol
2753 //
2754 Status = InternalPrintTo(L"%");
2755 if (EFI_ERROR(Status)) {
2756 break;
2757 }
2758 ResumeLocation = ResumeLocation - 1;
2759 break;
2760 }
jcarsey975136a2009-06-16 19:03:54 +00002761 }
2762 } else {
2763 //
2764 // reset to normal now...
2765 //
jcarsey975136a2009-06-16 19:03:54 +00002766 break;
2767 }
2768
2769 //
2770 // update FormatWalker to Resume + 2 (skip the % and the indicator)
2771 //
2772 FormatWalker = ResumeLocation + 2;
2773 }
jcarseyb1f95a02009-06-16 00:23:19 +00002774
jcarseya405b862010-09-14 05:18:09 +00002775 gST->ConOut->SetAttribute(gST->ConOut, OriginalAttribute);
jcarsey89e85372011-04-13 23:37:50 +00002776
2777 SHELL_FREE_NON_NULL(mPostReplaceFormat);
2778 SHELL_FREE_NON_NULL(mPostReplaceFormat2);
jcarseya405b862010-09-14 05:18:09 +00002779 return (Status);
jcarsey5f7431d2009-07-10 18:06:01 +00002780}
jcarsey2247dde2009-11-09 18:08:58 +00002781
2782/**
2783 Print at a specific location on the screen.
2784
jcarseye2f82972009-12-01 05:40:24 +00002785 This function will move the cursor to a given screen location and print the specified string.
jcarsey1e6e84c2010-01-25 20:05:08 +00002786
2787 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarsey2247dde2009-11-09 18:08:58 +00002788 will be used.
2789
jcarseye2f82972009-12-01 05:40:24 +00002790 If either Row or Col is out of range for the current console, then ASSERT.
2791 If Format is NULL, then ASSERT.
jcarsey2247dde2009-11-09 18:08:58 +00002792
jcarsey1e6e84c2010-01-25 20:05:08 +00002793 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarsey2247dde2009-11-09 18:08:58 +00002794 the following additional flags:
2795 %N - Set output attribute to normal
2796 %H - Set output attribute to highlight
2797 %E - Set output attribute to error
2798 %B - Set output attribute to blue color
2799 %V - Set output attribute to green color
2800
2801 Note: The background color is controlled by the shell command cls.
2802
jcarsey2247dde2009-11-09 18:08:58 +00002803 @param[in] Col the column to print at
jcarseya405b862010-09-14 05:18:09 +00002804 @param[in] Row the row to print at
jcarsey2247dde2009-11-09 18:08:58 +00002805 @param[in] Format the format string
jcarseya405b862010-09-14 05:18:09 +00002806 @param[in] ... The variable argument list.
jcarsey2247dde2009-11-09 18:08:58 +00002807
jcarseya405b862010-09-14 05:18:09 +00002808 @return EFI_SUCCESS The printing was successful.
2809 @return EFI_DEVICE_ERROR The console device reported an error.
jcarsey2247dde2009-11-09 18:08:58 +00002810**/
jcarseya405b862010-09-14 05:18:09 +00002811EFI_STATUS
jcarsey2247dde2009-11-09 18:08:58 +00002812EFIAPI
2813ShellPrintEx(
2814 IN INT32 Col OPTIONAL,
2815 IN INT32 Row OPTIONAL,
2816 IN CONST CHAR16 *Format,
2817 ...
jcarsey1e6e84c2010-01-25 20:05:08 +00002818 )
jcarsey2247dde2009-11-09 18:08:58 +00002819{
2820 VA_LIST Marker;
jcarseya405b862010-09-14 05:18:09 +00002821 EFI_STATUS RetVal;
jcarsey3e082d52010-10-04 16:44:57 +00002822 if (Format == NULL) {
2823 return (EFI_INVALID_PARAMETER);
2824 }
jcarsey2247dde2009-11-09 18:08:58 +00002825 VA_START (Marker, Format);
jcarseya405b862010-09-14 05:18:09 +00002826 RetVal = InternalShellPrintWorker(Col, Row, Format, Marker);
jcarseye2f82972009-12-01 05:40:24 +00002827 VA_END(Marker);
jcarseya405b862010-09-14 05:18:09 +00002828 return(RetVal);
jcarsey2247dde2009-11-09 18:08:58 +00002829}
2830
2831/**
2832 Print at a specific location on the screen.
2833
jcarseye2f82972009-12-01 05:40:24 +00002834 This function will move the cursor to a given screen location and print the specified string.
jcarsey1e6e84c2010-01-25 20:05:08 +00002835
2836 If -1 is specified for either the Row or Col the current screen location for BOTH
jcarseye2f82972009-12-01 05:40:24 +00002837 will be used.
jcarsey2247dde2009-11-09 18:08:58 +00002838
jcarseye2f82972009-12-01 05:40:24 +00002839 If either Row or Col is out of range for the current console, then ASSERT.
2840 If Format is NULL, then ASSERT.
jcarsey2247dde2009-11-09 18:08:58 +00002841
jcarsey1e6e84c2010-01-25 20:05:08 +00002842 In addition to the standard %-based flags as supported by UefiLib Print() this supports
jcarsey2247dde2009-11-09 18:08:58 +00002843 the following additional flags:
jcarsey1e6e84c2010-01-25 20:05:08 +00002844 %N - Set output attribute to normal.
2845 %H - Set output attribute to highlight.
2846 %E - Set output attribute to error.
2847 %B - Set output attribute to blue color.
2848 %V - Set output attribute to green color.
jcarsey2247dde2009-11-09 18:08:58 +00002849
2850 Note: The background color is controlled by the shell command cls.
2851
jcarsey1e6e84c2010-01-25 20:05:08 +00002852 @param[in] Col The column to print at.
jcarseya405b862010-09-14 05:18:09 +00002853 @param[in] Row The row to print at.
jcarsey1e6e84c2010-01-25 20:05:08 +00002854 @param[in] Language The language of the string to retrieve. If this parameter
2855 is NULL, then the current platform language is used.
2856 @param[in] HiiFormatStringId The format string Id for getting from Hii.
2857 @param[in] HiiFormatHandle The format string Handle for getting from Hii.
jcarseya405b862010-09-14 05:18:09 +00002858 @param[in] ... The variable argument list.
jcarsey2247dde2009-11-09 18:08:58 +00002859
jcarseya405b862010-09-14 05:18:09 +00002860 @return EFI_SUCCESS The printing was successful.
2861 @return EFI_DEVICE_ERROR The console device reported an error.
jcarsey2247dde2009-11-09 18:08:58 +00002862**/
jcarseya405b862010-09-14 05:18:09 +00002863EFI_STATUS
jcarsey2247dde2009-11-09 18:08:58 +00002864EFIAPI
2865ShellPrintHiiEx(
2866 IN INT32 Col OPTIONAL,
2867 IN INT32 Row OPTIONAL,
jcarsey1e6e84c2010-01-25 20:05:08 +00002868 IN CONST CHAR8 *Language OPTIONAL,
jcarsey2247dde2009-11-09 18:08:58 +00002869 IN CONST EFI_STRING_ID HiiFormatStringId,
2870 IN CONST EFI_HANDLE HiiFormatHandle,
2871 ...
2872 )
2873{
2874 VA_LIST Marker;
2875 CHAR16 *HiiFormatString;
jcarseya405b862010-09-14 05:18:09 +00002876 EFI_STATUS RetVal;
jcarsey2247dde2009-11-09 18:08:58 +00002877
2878 VA_START (Marker, HiiFormatHandle);
jcarsey1e6e84c2010-01-25 20:05:08 +00002879 HiiFormatString = HiiGetString(HiiFormatHandle, HiiFormatStringId, Language);
jcarsey2247dde2009-11-09 18:08:58 +00002880 ASSERT(HiiFormatString != NULL);
2881
2882 RetVal = InternalShellPrintWorker(Col, Row, HiiFormatString, Marker);
2883
jcarseya405b862010-09-14 05:18:09 +00002884 SHELL_FREE_NON_NULL(HiiFormatString);
jcarseye2f82972009-12-01 05:40:24 +00002885 VA_END(Marker);
jcarsey2247dde2009-11-09 18:08:58 +00002886
2887 return (RetVal);
2888}
2889
2890/**
2891 Function to determine if a given filename represents a file or a directory.
2892
2893 @param[in] DirName Path to directory to test.
2894
jcarseyc8c22592011-10-17 17:49:21 +00002895 @retval EFI_SUCCESS The Path represents a directory
2896 @retval EFI_NOT_FOUND The Path does not represent a directory
2897 @retval EFI_OUT_OF_RESOURCES A memory allocation failed.
2898 @return The path failed to open
jcarsey2247dde2009-11-09 18:08:58 +00002899**/
2900EFI_STATUS
2901EFIAPI
2902ShellIsDirectory(
2903 IN CONST CHAR16 *DirName
2904 )
2905{
2906 EFI_STATUS Status;
jcarseya405b862010-09-14 05:18:09 +00002907 SHELL_FILE_HANDLE Handle;
jcarsey3e082d52010-10-04 16:44:57 +00002908 CHAR16 *TempLocation;
2909 CHAR16 *TempLocation2;
jcarsey2247dde2009-11-09 18:08:58 +00002910
jcarseyecd3d592009-12-07 18:05:00 +00002911 ASSERT(DirName != NULL);
2912
jcarseya405b862010-09-14 05:18:09 +00002913 Handle = NULL;
2914 TempLocation = NULL;
jcarsey2247dde2009-11-09 18:08:58 +00002915
2916 Status = ShellOpenFileByName(DirName, &Handle, EFI_FILE_MODE_READ, 0);
2917 if (EFI_ERROR(Status)) {
jcarseya405b862010-09-14 05:18:09 +00002918 //
2919 // try good logic first.
2920 //
jcarsey366f81a2011-06-27 21:04:22 +00002921 if (gEfiShellProtocol != NULL) {
jcarsey3e082d52010-10-04 16:44:57 +00002922 TempLocation = StrnCatGrow(&TempLocation, NULL, DirName, 0);
jcarseyc8c22592011-10-17 17:49:21 +00002923 if (TempLocation == NULL) {
2924 ShellCloseFile(&Handle);
2925 return (EFI_OUT_OF_RESOURCES);
2926 }
jcarsey3e082d52010-10-04 16:44:57 +00002927 TempLocation2 = StrStr(TempLocation, L":");
2928 if (TempLocation2 != NULL && StrLen(StrStr(TempLocation, L":")) == 2) {
2929 *(TempLocation2+1) = CHAR_NULL;
jcarseya405b862010-09-14 05:18:09 +00002930 }
jcarsey366f81a2011-06-27 21:04:22 +00002931 if (gEfiShellProtocol->GetDevicePathFromMap(TempLocation) != NULL) {
jcarseya405b862010-09-14 05:18:09 +00002932 FreePool(TempLocation);
2933 return (EFI_SUCCESS);
2934 }
2935 FreePool(TempLocation);
2936 } else {
2937 //
2938 // probably a map name?!?!!?
2939 //
2940 TempLocation = StrStr(DirName, L"\\");
2941 if (TempLocation != NULL && *(TempLocation+1) == CHAR_NULL) {
2942 return (EFI_SUCCESS);
2943 }
2944 }
jcarsey2247dde2009-11-09 18:08:58 +00002945 return (Status);
2946 }
2947
2948 if (FileHandleIsDirectory(Handle) == EFI_SUCCESS) {
2949 ShellCloseFile(&Handle);
2950 return (EFI_SUCCESS);
2951 }
2952 ShellCloseFile(&Handle);
2953 return (EFI_NOT_FOUND);
2954}
2955
jcarsey125c2cf2009-11-18 21:36:50 +00002956/**
jcarsey36a9d672009-11-20 21:13:41 +00002957 Function to determine if a given filename represents a file.
2958
2959 @param[in] Name Path to file to test.
2960
2961 @retval EFI_SUCCESS The Path represents a file.
2962 @retval EFI_NOT_FOUND The Path does not represent a file.
2963 @retval other The path failed to open.
2964**/
2965EFI_STATUS
2966EFIAPI
2967ShellIsFile(
2968 IN CONST CHAR16 *Name
2969 )
2970{
2971 EFI_STATUS Status;
jcarseya405b862010-09-14 05:18:09 +00002972 SHELL_FILE_HANDLE Handle;
jcarsey36a9d672009-11-20 21:13:41 +00002973
jcarseyecd3d592009-12-07 18:05:00 +00002974 ASSERT(Name != NULL);
2975
jcarsey36a9d672009-11-20 21:13:41 +00002976 Handle = NULL;
2977
2978 Status = ShellOpenFileByName(Name, &Handle, EFI_FILE_MODE_READ, 0);
2979 if (EFI_ERROR(Status)) {
2980 return (Status);
2981 }
2982
2983 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
2984 ShellCloseFile(&Handle);
2985 return (EFI_SUCCESS);
2986 }
2987 ShellCloseFile(&Handle);
2988 return (EFI_NOT_FOUND);
2989}
2990
2991/**
jcarseyb3011f42010-01-11 21:49:04 +00002992 Function to determine if a given filename represents a file.
2993
2994 This will search the CWD and then the Path.
2995
2996 If Name is NULL, then ASSERT.
2997
2998 @param[in] Name Path to file to test.
2999
3000 @retval EFI_SUCCESS The Path represents a file.
3001 @retval EFI_NOT_FOUND The Path does not represent a file.
3002 @retval other The path failed to open.
3003**/
3004EFI_STATUS
3005EFIAPI
3006ShellIsFileInPath(
3007 IN CONST CHAR16 *Name
jcarseya405b862010-09-14 05:18:09 +00003008 )
3009{
jcarseyb3011f42010-01-11 21:49:04 +00003010 CHAR16 *NewName;
3011 EFI_STATUS Status;
3012
3013 if (!EFI_ERROR(ShellIsFile(Name))) {
jcarseya405b862010-09-14 05:18:09 +00003014 return (EFI_SUCCESS);
jcarseyb3011f42010-01-11 21:49:04 +00003015 }
3016
3017 NewName = ShellFindFilePath(Name);
3018 if (NewName == NULL) {
3019 return (EFI_NOT_FOUND);
3020 }
3021 Status = ShellIsFile(NewName);
3022 FreePool(NewName);
3023 return (Status);
3024}
jcarsey252d9452011-03-25 20:49:53 +00003025
jcarseyb3011f42010-01-11 21:49:04 +00003026/**
jcarsey1e6e84c2010-01-25 20:05:08 +00003027 Function to determine whether a string is decimal or hex representation of a number
jcarsey125c2cf2009-11-18 21:36:50 +00003028 and return the number converted from the string.
3029
3030 @param[in] String String representation of a number
3031
jcarsey252d9452011-03-25 20:49:53 +00003032 @return the number
3033 @retval (UINTN)(-1) An error ocurred.
jcarsey125c2cf2009-11-18 21:36:50 +00003034**/
3035UINTN
3036EFIAPI
3037ShellStrToUintn(
3038 IN CONST CHAR16 *String
3039 )
3040{
jcarsey252d9452011-03-25 20:49:53 +00003041 UINT64 RetVal;
3042 BOOLEAN Hex;
3043
3044 Hex = FALSE;
3045
3046 if (!InternalShellIsHexOrDecimalNumber(String, Hex, TRUE)) {
3047 Hex = TRUE;
jcarsey125c2cf2009-11-18 21:36:50 +00003048 }
jcarsey252d9452011-03-25 20:49:53 +00003049
3050 if (!EFI_ERROR(ShellConvertStringToUint64(String, &RetVal, Hex, TRUE))) {
3051 return ((UINTN)RetVal);
3052 }
3053 return ((UINTN)(-1));
jcarsey125c2cf2009-11-18 21:36:50 +00003054}
3055
3056/**
3057 Safely append with automatic string resizing given length of Destination and
3058 desired length of copy from Source.
3059
3060 append the first D characters of Source to the end of Destination, where D is
3061 the lesser of Count and the StrLen() of Source. If appending those D characters
3062 will fit within Destination (whose Size is given as CurrentSize) and
jcarsey1e6e84c2010-01-25 20:05:08 +00003063 still leave room for a NULL terminator, then those characters are appended,
3064 starting at the original terminating NULL of Destination, and a new terminating
3065 NULL is appended.
jcarsey125c2cf2009-11-18 21:36:50 +00003066
3067 If appending D characters onto Destination will result in a overflow of the size
3068 given in CurrentSize the string will be grown such that the copy can be performed
3069 and CurrentSize will be updated to the new size.
3070
3071 If Source is NULL, there is nothing to append, just return the current buffer in
3072 Destination.
3073
3074 if Destination is NULL, then ASSERT()
3075 if Destination's current length (including NULL terminator) is already more then
3076 CurrentSize, then ASSERT()
3077
ydong104ff7e372011-09-02 08:05:34 +00003078 @param[in, out] Destination The String to append onto
3079 @param[in, out] CurrentSize on call the number of bytes in Destination. On
jcarsey125c2cf2009-11-18 21:36:50 +00003080 return possibly the new size (still in bytes). if NULL
3081 then allocate whatever is needed.
3082 @param[in] Source The String to append from
3083 @param[in] Count Maximum number of characters to append. if 0 then
3084 all are appended.
3085
3086 @return Destination return the resultant string.
3087**/
3088CHAR16*
3089EFIAPI
3090StrnCatGrow (
3091 IN OUT CHAR16 **Destination,
3092 IN OUT UINTN *CurrentSize,
3093 IN CONST CHAR16 *Source,
3094 IN UINTN Count
3095 )
3096{
3097 UINTN DestinationStartSize;
3098 UINTN NewSize;
3099
3100 //
3101 // ASSERTs
3102 //
3103 ASSERT(Destination != NULL);
3104
3105 //
3106 // If there's nothing to do then just return Destination
3107 //
3108 if (Source == NULL) {
3109 return (*Destination);
3110 }
3111
3112 //
3113 // allow for un-initialized pointers, based on size being 0
3114 //
3115 if (CurrentSize != NULL && *CurrentSize == 0) {
3116 *Destination = NULL;
3117 }
3118
3119 //
3120 // allow for NULL pointers address as Destination
3121 //
3122 if (*Destination != NULL) {
3123 ASSERT(CurrentSize != 0);
3124 DestinationStartSize = StrSize(*Destination);
3125 ASSERT(DestinationStartSize <= *CurrentSize);
3126 } else {
3127 DestinationStartSize = 0;
3128// ASSERT(*CurrentSize == 0);
3129 }
3130
3131 //
3132 // Append all of Source?
3133 //
3134 if (Count == 0) {
3135 Count = StrLen(Source);
3136 }
3137
3138 //
3139 // Test and grow if required
3140 //
3141 if (CurrentSize != NULL) {
3142 NewSize = *CurrentSize;
3143 while (NewSize < (DestinationStartSize + (Count*sizeof(CHAR16)))) {
3144 NewSize += 2 * Count * sizeof(CHAR16);
3145 }
3146 *Destination = ReallocatePool(*CurrentSize, NewSize, *Destination);
3147 *CurrentSize = NewSize;
3148 } else {
3149 *Destination = AllocateZeroPool((Count+1)*sizeof(CHAR16));
3150 }
3151
3152 //
3153 // Now use standard StrnCat on a big enough buffer
3154 //
jcarseyc9d92df2010-02-03 15:37:54 +00003155 if (*Destination == NULL) {
3156 return (NULL);
3157 }
jcarsey125c2cf2009-11-18 21:36:50 +00003158 return StrnCat(*Destination, Source, Count);
3159}
jcarseyc9d92df2010-02-03 15:37:54 +00003160
3161/**
3162 Prompt the user and return the resultant answer to the requestor.
3163
3164 This function will display the requested question on the shell prompt and then
3165 wait for an apropriate answer to be input from the console.
3166
jcarseya405b862010-09-14 05:18:09 +00003167 if the SHELL_PROMPT_REQUEST_TYPE is SHELL_PROMPT_REQUEST_TYPE_YESNO, ShellPromptResponseTypeQuitContinue
jcarseyc9d92df2010-02-03 15:37:54 +00003168 or SHELL_PROMPT_REQUEST_TYPE_YESNOCANCEL then *Response is of type SHELL_PROMPT_RESPONSE.
3169
jcarseya405b862010-09-14 05:18:09 +00003170 if the SHELL_PROMPT_REQUEST_TYPE is ShellPromptResponseTypeFreeform then *Response is of type
jcarseyc9d92df2010-02-03 15:37:54 +00003171 CHAR16*.
3172
3173 In either case *Response must be callee freed if Response was not NULL;
3174
3175 @param Type What type of question is asked. This is used to filter the input
3176 to prevent invalid answers to question.
3177 @param Prompt Pointer to string prompt to use to request input.
3178 @param Response Pointer to Response which will be populated upon return.
3179
3180 @retval EFI_SUCCESS The operation was sucessful.
3181 @retval EFI_UNSUPPORTED The operation is not supported as requested.
3182 @retval EFI_INVALID_PARAMETER A parameter was invalid.
3183 @return other The operation failed.
3184**/
3185EFI_STATUS
3186EFIAPI
3187ShellPromptForResponse (
3188 IN SHELL_PROMPT_REQUEST_TYPE Type,
3189 IN CHAR16 *Prompt OPTIONAL,
3190 IN OUT VOID **Response OPTIONAL
3191 )
3192{
3193 EFI_STATUS Status;
3194 EFI_INPUT_KEY Key;
3195 UINTN EventIndex;
3196 SHELL_PROMPT_RESPONSE *Resp;
jcarseya405b862010-09-14 05:18:09 +00003197 UINTN Size;
3198 CHAR16 *Buffer;
jcarseyc9d92df2010-02-03 15:37:54 +00003199
jcarseya405b862010-09-14 05:18:09 +00003200 Status = EFI_UNSUPPORTED;
3201 Resp = NULL;
3202 Buffer = NULL;
3203 Size = 0;
3204 if (Type != ShellPromptResponseTypeFreeform) {
jcarsey252d9452011-03-25 20:49:53 +00003205 Resp = (SHELL_PROMPT_RESPONSE*)AllocateZeroPool(sizeof(SHELL_PROMPT_RESPONSE));
jcarsey3e082d52010-10-04 16:44:57 +00003206 if (Resp == NULL) {
3207 return (EFI_OUT_OF_RESOURCES);
3208 }
xdu290bfa222010-07-19 05:21:27 +00003209 }
jcarseyc9d92df2010-02-03 15:37:54 +00003210
3211 switch(Type) {
jcarseya405b862010-09-14 05:18:09 +00003212 case ShellPromptResponseTypeQuitContinue:
jcarseyc9d92df2010-02-03 15:37:54 +00003213 if (Prompt != NULL) {
3214 ShellPrintEx(-1, -1, L"%s", Prompt);
3215 }
3216 //
3217 // wait for valid response
3218 //
3219 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3220 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3221 ASSERT_EFI_ERROR(Status);
3222 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3223 if (Key.UnicodeChar == L'Q' || Key.UnicodeChar ==L'q') {
jcarseya405b862010-09-14 05:18:09 +00003224 *Resp = ShellPromptResponseQuit;
jcarseyc9d92df2010-02-03 15:37:54 +00003225 } else {
jcarseya405b862010-09-14 05:18:09 +00003226 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003227 }
3228 break;
jcarseya405b862010-09-14 05:18:09 +00003229 case ShellPromptResponseTypeYesNoCancel:
jcarseyc9d92df2010-02-03 15:37:54 +00003230 if (Prompt != NULL) {
3231 ShellPrintEx(-1, -1, L"%s", Prompt);
3232 }
3233 //
3234 // wait for valid response
3235 //
jcarseya405b862010-09-14 05:18:09 +00003236 *Resp = ShellPromptResponseMax;
3237 while (*Resp == ShellPromptResponseMax) {
jcarseyc9d92df2010-02-03 15:37:54 +00003238 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3239 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3240 ASSERT_EFI_ERROR(Status);
3241 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3242 switch (Key.UnicodeChar) {
3243 case L'Y':
3244 case L'y':
jcarseya405b862010-09-14 05:18:09 +00003245 *Resp = ShellPromptResponseYes;
jcarseyc9d92df2010-02-03 15:37:54 +00003246 break;
3247 case L'N':
3248 case L'n':
jcarseya405b862010-09-14 05:18:09 +00003249 *Resp = ShellPromptResponseNo;
jcarseyc9d92df2010-02-03 15:37:54 +00003250 break;
3251 case L'C':
3252 case L'c':
jcarseya405b862010-09-14 05:18:09 +00003253 *Resp = ShellPromptResponseCancel;
3254 break;
3255 }
3256 }
3257 break; case ShellPromptResponseTypeYesNoAllCancel:
3258 if (Prompt != NULL) {
3259 ShellPrintEx(-1, -1, L"%s", Prompt);
3260 }
3261 //
3262 // wait for valid response
3263 //
3264 *Resp = ShellPromptResponseMax;
3265 while (*Resp == ShellPromptResponseMax) {
3266 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3267 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3268 ASSERT_EFI_ERROR(Status);
3269 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3270 switch (Key.UnicodeChar) {
3271 case L'Y':
3272 case L'y':
3273 *Resp = ShellPromptResponseYes;
3274 break;
3275 case L'N':
3276 case L'n':
3277 *Resp = ShellPromptResponseNo;
3278 break;
3279 case L'A':
3280 case L'a':
3281 *Resp = ShellPromptResponseAll;
3282 break;
3283 case L'C':
3284 case L'c':
3285 *Resp = ShellPromptResponseCancel;
jcarseyc9d92df2010-02-03 15:37:54 +00003286 break;
3287 }
3288 }
3289 break;
jcarseya405b862010-09-14 05:18:09 +00003290 case ShellPromptResponseTypeEnterContinue:
3291 case ShellPromptResponseTypeAnyKeyContinue:
jcarseyc9d92df2010-02-03 15:37:54 +00003292 if (Prompt != NULL) {
3293 ShellPrintEx(-1, -1, L"%s", Prompt);
3294 }
3295 //
3296 // wait for valid response
3297 //
jcarseya405b862010-09-14 05:18:09 +00003298 *Resp = ShellPromptResponseMax;
3299 while (*Resp == ShellPromptResponseMax) {
jcarseyc9d92df2010-02-03 15:37:54 +00003300 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
jcarseya405b862010-09-14 05:18:09 +00003301 if (Type == ShellPromptResponseTypeEnterContinue) {
jcarseyc9d92df2010-02-03 15:37:54 +00003302 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3303 ASSERT_EFI_ERROR(Status);
3304 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3305 if (Key.UnicodeChar == CHAR_CARRIAGE_RETURN) {
jcarseya405b862010-09-14 05:18:09 +00003306 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003307 break;
3308 }
3309 }
jcarseya405b862010-09-14 05:18:09 +00003310 if (Type == ShellPromptResponseTypeAnyKeyContinue) {
3311 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3312 ASSERT_EFI_ERROR(Status);
3313 *Resp = ShellPromptResponseContinue;
jcarseyc9d92df2010-02-03 15:37:54 +00003314 break;
3315 }
3316 }
3317 break;
jcarseya405b862010-09-14 05:18:09 +00003318 case ShellPromptResponseTypeYesNo:
3319 if (Prompt != NULL) {
3320 ShellPrintEx(-1, -1, L"%s", Prompt);
3321 }
3322 //
3323 // wait for valid response
3324 //
3325 *Resp = ShellPromptResponseMax;
3326 while (*Resp == ShellPromptResponseMax) {
3327 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3328 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3329 ASSERT_EFI_ERROR(Status);
3330 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3331 switch (Key.UnicodeChar) {
3332 case L'Y':
3333 case L'y':
3334 *Resp = ShellPromptResponseYes;
3335 break;
3336 case L'N':
3337 case L'n':
3338 *Resp = ShellPromptResponseNo;
3339 break;
3340 }
3341 }
3342 break;
3343 case ShellPromptResponseTypeFreeform:
3344 if (Prompt != NULL) {
3345 ShellPrintEx(-1, -1, L"%s", Prompt);
3346 }
3347 while(1) {
3348 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3349 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3350 ASSERT_EFI_ERROR(Status);
3351 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3352 if (Key.UnicodeChar == CHAR_CARRIAGE_RETURN) {
3353 break;
3354 }
3355 ASSERT((Buffer == NULL && Size == 0) || (Buffer != NULL));
3356 StrnCatGrow(&Buffer, &Size, &Key.UnicodeChar, 1);
3357 }
3358 break;
3359 //
3360 // This is the location to add new prompt types.
3361 //
jcarseyc9d92df2010-02-03 15:37:54 +00003362 default:
jcarseya405b862010-09-14 05:18:09 +00003363 ASSERT(FALSE);
jcarseyc9d92df2010-02-03 15:37:54 +00003364 }
3365
3366 if (Response != NULL) {
jcarseya405b862010-09-14 05:18:09 +00003367 if (Resp != NULL) {
3368 *Response = Resp;
3369 } else if (Buffer != NULL) {
3370 *Response = Buffer;
3371 }
jcarseyc9d92df2010-02-03 15:37:54 +00003372 } else {
jcarseya405b862010-09-14 05:18:09 +00003373 if (Resp != NULL) {
3374 FreePool(Resp);
3375 }
3376 if (Buffer != NULL) {
3377 FreePool(Buffer);
3378 }
jcarseyc9d92df2010-02-03 15:37:54 +00003379 }
3380
jcarseya405b862010-09-14 05:18:09 +00003381 ShellPrintEx(-1, -1, L"\r\n");
jcarseyc9d92df2010-02-03 15:37:54 +00003382 return (Status);
3383}
3384
3385/**
3386 Prompt the user and return the resultant answer to the requestor.
3387
3388 This function is the same as ShellPromptForResponse, except that the prompt is
3389 automatically pulled from HII.
3390
3391 @param Type What type of question is asked. This is used to filter the input
3392 to prevent invalid answers to question.
jcarseya405b862010-09-14 05:18:09 +00003393 @param[in] HiiFormatStringId The format string Id for getting from Hii.
3394 @param[in] HiiFormatHandle The format string Handle for getting from Hii.
3395 @param Response Pointer to Response which will be populated upon return.
jcarseyc9d92df2010-02-03 15:37:54 +00003396
3397 @retval EFI_SUCCESS the operation was sucessful.
3398 @return other the operation failed.
3399
3400 @sa ShellPromptForResponse
3401**/
3402EFI_STATUS
3403EFIAPI
3404ShellPromptForResponseHii (
3405 IN SHELL_PROMPT_REQUEST_TYPE Type,
3406 IN CONST EFI_STRING_ID HiiFormatStringId,
3407 IN CONST EFI_HANDLE HiiFormatHandle,
3408 IN OUT VOID **Response
3409 )
3410{
3411 CHAR16 *Prompt;
3412 EFI_STATUS Status;
3413
3414 Prompt = HiiGetString(HiiFormatHandle, HiiFormatStringId, NULL);
3415 Status = ShellPromptForResponse(Type, Prompt, Response);
3416 FreePool(Prompt);
3417 return (Status);
3418}
3419
jcarseya405b862010-09-14 05:18:09 +00003420/**
3421 Function to determin if an entire string is a valid number.
jcarseyc9d92df2010-02-03 15:37:54 +00003422
jcarseya405b862010-09-14 05:18:09 +00003423 If Hex it must be preceeded with a 0x or has ForceHex, set TRUE.
3424
3425 @param[in] String The string to evaluate.
3426 @param[in] ForceHex TRUE - always assume hex.
3427 @param[in] StopAtSpace TRUE to halt upon finding a space, FALSE to keep going.
3428
3429 @retval TRUE It is all numeric (dec/hex) characters.
3430 @retval FALSE There is a non-numeric character.
3431**/
3432BOOLEAN
3433EFIAPI
jcarsey252d9452011-03-25 20:49:53 +00003434InternalShellIsHexOrDecimalNumber (
jcarseya405b862010-09-14 05:18:09 +00003435 IN CONST CHAR16 *String,
3436 IN CONST BOOLEAN ForceHex,
3437 IN CONST BOOLEAN StopAtSpace
3438 )
3439{
3440 BOOLEAN Hex;
3441
3442 //
3443 // chop off a single negative sign
3444 //
3445 if (String != NULL && *String == L'-') {
3446 String++;
3447 }
darylm503b0934ac2011-11-12 00:35:11 +00003448
jcarseya405b862010-09-14 05:18:09 +00003449 if (String == NULL) {
3450 return (FALSE);
3451 }
3452
3453 //
3454 // chop leading zeroes
3455 //
3456 while(String != NULL && *String == L'0'){
3457 String++;
3458 }
3459 //
3460 // allow '0x' or '0X', but not 'x' or 'X'
3461 //
3462 if (String != NULL && (*String == L'x' || *String == L'X')) {
3463 if (*(String-1) != L'0') {
3464 //
3465 // we got an x without a preceeding 0
3466 //
3467 return (FALSE);
3468 }
3469 String++;
3470 Hex = TRUE;
3471 } else if (ForceHex) {
3472 Hex = TRUE;
3473 } else {
3474 Hex = FALSE;
3475 }
3476
3477 //
3478 // loop through the remaining characters and use the lib function
3479 //
3480 for ( ; String != NULL && *String != CHAR_NULL && !(StopAtSpace && *String == L' ') ; String++){
3481 if (Hex) {
3482 if (!ShellIsHexaDecimalDigitCharacter(*String)) {
3483 return (FALSE);
3484 }
3485 } else {
3486 if (!ShellIsDecimalDigitCharacter(*String)) {
3487 return (FALSE);
3488 }
3489 }
3490 }
jcarsey252d9452011-03-25 20:49:53 +00003491
jcarseya405b862010-09-14 05:18:09 +00003492 return (TRUE);
3493}
3494
3495/**
3496 Function to determine if a given filename exists.
3497
3498 @param[in] Name Path to test.
3499
3500 @retval EFI_SUCCESS The Path represents a file.
3501 @retval EFI_NOT_FOUND The Path does not represent a file.
3502 @retval other The path failed to open.
3503**/
3504EFI_STATUS
3505EFIAPI
3506ShellFileExists(
3507 IN CONST CHAR16 *Name
3508 )
3509{
3510 EFI_STATUS Status;
3511 EFI_SHELL_FILE_INFO *List;
3512
3513 ASSERT(Name != NULL);
3514
3515 List = NULL;
3516 Status = ShellOpenFileMetaArg((CHAR16*)Name, EFI_FILE_MODE_READ, &List);
3517 if (EFI_ERROR(Status)) {
3518 return (Status);
3519 }
3520
3521 ShellCloseFileMetaArg(&List);
3522
3523 return (EFI_SUCCESS);
3524}
jcarsey252d9452011-03-25 20:49:53 +00003525
3526/**
darylm503b0934ac2011-11-12 00:35:11 +00003527 Convert a Unicode character to upper case only if
jcarsey252d9452011-03-25 20:49:53 +00003528 it maps to a valid small-case ASCII character.
3529
3530 This internal function only deal with Unicode character
3531 which maps to a valid small-case ASCII character, i.e.
3532 L'a' to L'z'. For other Unicode character, the input character
3533 is returned directly.
3534
3535 @param Char The character to convert.
3536
3537 @retval LowerCharacter If the Char is with range L'a' to L'z'.
3538 @retval Unchanged Otherwise.
3539
3540**/
3541CHAR16
3542EFIAPI
3543InternalShellCharToUpper (
3544 IN CHAR16 Char
3545 )
3546{
3547 if (Char >= L'a' && Char <= L'z') {
3548 return (CHAR16) (Char - (L'a' - L'A'));
3549 }
3550
3551 return Char;
3552}
3553
3554/**
3555 Convert a Unicode character to numerical value.
3556
3557 This internal function only deal with Unicode character
3558 which maps to a valid hexadecimal ASII character, i.e.
darylm503b0934ac2011-11-12 00:35:11 +00003559 L'0' to L'9', L'a' to L'f' or L'A' to L'F'. For other
jcarsey252d9452011-03-25 20:49:53 +00003560 Unicode character, the value returned does not make sense.
3561
3562 @param Char The character to convert.
3563
3564 @return The numerical value converted.
3565
3566**/
3567UINTN
3568EFIAPI
3569InternalShellHexCharToUintn (
3570 IN CHAR16 Char
3571 )
3572{
3573 if (ShellIsDecimalDigitCharacter (Char)) {
3574 return Char - L'0';
3575 }
3576
3577 return (UINTN) (10 + InternalShellCharToUpper (Char) - L'A');
3578}
3579
3580/**
3581 Convert a Null-terminated Unicode hexadecimal string to a value of type UINT64.
3582
3583 This function returns a value of type UINTN by interpreting the contents
3584 of the Unicode string specified by String as a hexadecimal number.
3585 The format of the input Unicode string String is:
3586
3587 [spaces][zeros][x][hexadecimal digits].
3588
3589 The valid hexadecimal digit character is in the range [0-9], [a-f] and [A-F].
3590 The prefix "0x" is optional. Both "x" and "X" is allowed in "0x" prefix.
3591 If "x" appears in the input string, it must be prefixed with at least one 0.
3592 The function will ignore the pad space, which includes spaces or tab characters,
3593 before [zeros], [x] or [hexadecimal digit]. The running zero before [x] or
3594 [hexadecimal digit] will be ignored. Then, the decoding starts after [x] or the
3595 first valid hexadecimal digit. Then, the function stops at the first character that is
3596 a not a valid hexadecimal character or NULL, whichever one comes first.
3597
3598 If String has only pad spaces, then zero is returned.
3599 If String has no leading pad spaces, leading zeros or valid hexadecimal digits,
3600 then zero is returned.
3601
3602 @param[in] String A pointer to a Null-terminated Unicode string.
3603 @param[out] Value Upon a successful return the value of the conversion.
3604 @param[in] StopAtSpace FALSE to skip spaces.
3605
3606 @retval EFI_SUCCESS The conversion was successful.
3607 @retval EFI_INVALID_PARAMETER A parameter was NULL or invalid.
3608 @retval EFI_DEVICE_ERROR An overflow occured.
3609**/
3610EFI_STATUS
3611EFIAPI
3612InternalShellStrHexToUint64 (
3613 IN CONST CHAR16 *String,
3614 OUT UINT64 *Value,
3615 IN CONST BOOLEAN StopAtSpace
3616 )
3617{
3618 UINT64 Result;
3619
3620 if (String == NULL || StrSize(String) == 0 || Value == NULL) {
3621 return (EFI_INVALID_PARAMETER);
3622 }
darylm503b0934ac2011-11-12 00:35:11 +00003623
jcarsey252d9452011-03-25 20:49:53 +00003624 //
darylm503b0934ac2011-11-12 00:35:11 +00003625 // Ignore the pad spaces (space or tab)
jcarsey252d9452011-03-25 20:49:53 +00003626 //
3627 while ((*String == L' ') || (*String == L'\t')) {
3628 String++;
3629 }
3630
3631 //
3632 // Ignore leading Zeros after the spaces
3633 //
3634 while (*String == L'0') {
3635 String++;
3636 }
3637
3638 if (InternalShellCharToUpper (*String) == L'X') {
3639 if (*(String - 1) != L'0') {
3640 return 0;
3641 }
3642 //
3643 // Skip the 'X'
3644 //
3645 String++;
3646 }
3647
3648 Result = 0;
darylm503b0934ac2011-11-12 00:35:11 +00003649
jcarsey252d9452011-03-25 20:49:53 +00003650 //
3651 // Skip spaces if requested
3652 //
3653 while (StopAtSpace && *String == L' ') {
3654 String++;
3655 }
darylm503b0934ac2011-11-12 00:35:11 +00003656
jcarsey252d9452011-03-25 20:49:53 +00003657 while (ShellIsHexaDecimalDigitCharacter (*String)) {
3658 //
darylm503b0934ac2011-11-12 00:35:11 +00003659 // If the Hex Number represented by String overflows according
jcarsey252d9452011-03-25 20:49:53 +00003660 // to the range defined by UINTN, then ASSERT().
3661 //
3662 if (!(Result <= (RShiftU64((((UINT64) ~0) - InternalShellHexCharToUintn (*String)), 4)))) {
3663// if (!(Result <= ((((UINT64) ~0) - InternalShellHexCharToUintn (*String)) >> 4))) {
3664 return (EFI_DEVICE_ERROR);
3665 }
3666
3667 Result = (LShiftU64(Result, 4));
3668 Result += InternalShellHexCharToUintn (*String);
3669 String++;
3670
3671 //
jcarsey49bd4982012-01-27 18:42:43 +00003672 // stop at spaces if requested
jcarsey252d9452011-03-25 20:49:53 +00003673 //
jcarsey49bd4982012-01-27 18:42:43 +00003674 if (StopAtSpace && *String == L' ') {
3675 break;
jcarsey252d9452011-03-25 20:49:53 +00003676 }
3677 }
3678
3679 *Value = Result;
3680 return (EFI_SUCCESS);
3681}
3682
3683/**
3684 Convert a Null-terminated Unicode decimal string to a value of
3685 type UINT64.
3686
3687 This function returns a value of type UINT64 by interpreting the contents
3688 of the Unicode string specified by String as a decimal number. The format
3689 of the input Unicode string String is:
3690
3691 [spaces] [decimal digits].
3692
3693 The valid decimal digit character is in the range [0-9]. The
3694 function will ignore the pad space, which includes spaces or
3695 tab characters, before [decimal digits]. The running zero in the
3696 beginning of [decimal digits] will be ignored. Then, the function
3697 stops at the first character that is a not a valid decimal character
3698 or a Null-terminator, whichever one comes first.
3699
3700 If String has only pad spaces, then 0 is returned.
3701 If String has no pad spaces or valid decimal digits,
3702 then 0 is returned.
3703
3704 @param[in] String A pointer to a Null-terminated Unicode string.
3705 @param[out] Value Upon a successful return the value of the conversion.
3706 @param[in] StopAtSpace FALSE to skip spaces.
3707
3708 @retval EFI_SUCCESS The conversion was successful.
3709 @retval EFI_INVALID_PARAMETER A parameter was NULL or invalid.
3710 @retval EFI_DEVICE_ERROR An overflow occured.
3711**/
3712EFI_STATUS
3713EFIAPI
3714InternalShellStrDecimalToUint64 (
3715 IN CONST CHAR16 *String,
3716 OUT UINT64 *Value,
3717 IN CONST BOOLEAN StopAtSpace
3718 )
3719{
3720 UINT64 Result;
3721
3722 if (String == NULL || StrSize (String) == 0 || Value == NULL) {
3723 return (EFI_INVALID_PARAMETER);
3724 }
3725
3726 //
3727 // Ignore the pad spaces (space or tab)
3728 //
3729 while ((*String == L' ') || (*String == L'\t')) {
3730 String++;
3731 }
3732
3733 //
3734 // Ignore leading Zeros after the spaces
3735 //
3736 while (*String == L'0') {
3737 String++;
3738 }
3739
3740 Result = 0;
3741
3742 //
3743 // Skip spaces if requested
3744 //
3745 while (StopAtSpace && *String == L' ') {
3746 String++;
3747 }
3748 while (ShellIsDecimalDigitCharacter (*String)) {
3749 //
darylm503b0934ac2011-11-12 00:35:11 +00003750 // If the number represented by String overflows according
jcarsey252d9452011-03-25 20:49:53 +00003751 // to the range defined by UINT64, then ASSERT().
3752 //
darylm503b0934ac2011-11-12 00:35:11 +00003753
jcarsey252d9452011-03-25 20:49:53 +00003754 if (!(Result <= (DivU64x32((((UINT64) ~0) - (*String - L'0')),10)))) {
3755 return (EFI_DEVICE_ERROR);
3756 }
3757
3758 Result = MultU64x32(Result, 10) + (*String - L'0');
3759 String++;
3760
3761 //
3762 // Stop at spaces if requested
3763 //
3764 if (StopAtSpace && *String == L' ') {
3765 break;
3766 }
3767 }
3768
3769 *Value = Result;
darylm503b0934ac2011-11-12 00:35:11 +00003770
jcarsey252d9452011-03-25 20:49:53 +00003771 return (EFI_SUCCESS);
3772}
3773
3774/**
3775 Function to verify and convert a string to its numerical value.
3776
3777 If Hex it must be preceeded with a 0x, 0X, or has ForceHex set TRUE.
3778
3779 @param[in] String The string to evaluate.
3780 @param[out] Value Upon a successful return the value of the conversion.
3781 @param[in] ForceHex TRUE - always assume hex.
3782 @param[in] StopAtSpace FALSE to skip spaces.
darylm503b0934ac2011-11-12 00:35:11 +00003783
jcarsey252d9452011-03-25 20:49:53 +00003784 @retval EFI_SUCCESS The conversion was successful.
3785 @retval EFI_INVALID_PARAMETER String contained an invalid character.
3786 @retval EFI_NOT_FOUND String was a number, but Value was NULL.
3787**/
3788EFI_STATUS
3789EFIAPI
3790ShellConvertStringToUint64(
3791 IN CONST CHAR16 *String,
3792 OUT UINT64 *Value,
3793 IN CONST BOOLEAN ForceHex,
3794 IN CONST BOOLEAN StopAtSpace
3795 )
3796{
3797 UINT64 RetVal;
3798 CONST CHAR16 *Walker;
3799 EFI_STATUS Status;
3800 BOOLEAN Hex;
3801
3802 Hex = ForceHex;
3803
3804 if (!InternalShellIsHexOrDecimalNumber(String, Hex, StopAtSpace)) {
3805 if (!Hex) {
3806 Hex = TRUE;
3807 if (!InternalShellIsHexOrDecimalNumber(String, Hex, StopAtSpace)) {
3808 return (EFI_INVALID_PARAMETER);
3809 }
3810 } else {
3811 return (EFI_INVALID_PARAMETER);
3812 }
3813 }
3814
3815 //
3816 // Chop off leading spaces
3817 //
3818 for (Walker = String; Walker != NULL && *Walker != CHAR_NULL && *Walker == L' '; Walker++);
3819
3820 //
3821 // make sure we have something left that is numeric.
3822 //
3823 if (Walker == NULL || *Walker == CHAR_NULL || !InternalShellIsHexOrDecimalNumber(Walker, Hex, StopAtSpace)) {
3824 return (EFI_INVALID_PARAMETER);
darylm503b0934ac2011-11-12 00:35:11 +00003825 }
jcarsey252d9452011-03-25 20:49:53 +00003826
3827 //
3828 // do the conversion.
3829 //
3830 if (Hex || StrnCmp(Walker, L"0x", 2) == 0 || StrnCmp(Walker, L"0X", 2) == 0){
3831 Status = InternalShellStrHexToUint64(Walker, &RetVal, StopAtSpace);
3832 } else {
3833 Status = InternalShellStrDecimalToUint64(Walker, &RetVal, StopAtSpace);
3834 }
3835
3836 if (Value == NULL && !EFI_ERROR(Status)) {
3837 return (EFI_NOT_FOUND);
3838 }
3839
3840 if (Value != NULL) {
3841 *Value = RetVal;
3842 }
3843
3844 return (Status);
3845}
3846
3847/**
3848 Function to determin if an entire string is a valid number.
3849
3850 If Hex it must be preceeded with a 0x or has ForceHex, set TRUE.
3851
3852 @param[in] String The string to evaluate.
3853 @param[in] ForceHex TRUE - always assume hex.
3854 @param[in] StopAtSpace TRUE to halt upon finding a space, FALSE to keep going.
3855
3856 @retval TRUE It is all numeric (dec/hex) characters.
3857 @retval FALSE There is a non-numeric character.
3858**/
3859BOOLEAN
3860EFIAPI
3861ShellIsHexOrDecimalNumber (
3862 IN CONST CHAR16 *String,
3863 IN CONST BOOLEAN ForceHex,
3864 IN CONST BOOLEAN StopAtSpace
3865 )
3866{
3867 if (ShellConvertStringToUint64(String, NULL, ForceHex, StopAtSpace) == EFI_NOT_FOUND) {
3868 return (TRUE);
3869 }
3870 return (FALSE);
3871}
jcarsey4d0a4fc2011-07-06 22:28:36 +00003872
3873/**
3874 Function to read a single line from a SHELL_FILE_HANDLE. The \n is not included in the returned
3875 buffer. The returned buffer must be callee freed.
3876
3877 If the position upon start is 0, then the Ascii Boolean will be set. This should be
3878 maintained and not changed for all operations with the same file.
3879
ydong104ff7e372011-09-02 08:05:34 +00003880 @param[in] Handle SHELL_FILE_HANDLE to read from.
3881 @param[in, out] Ascii Boolean value for indicating whether the file is
3882 Ascii (TRUE) or UCS2 (FALSE).
jcarsey4d0a4fc2011-07-06 22:28:36 +00003883
jcarseybeab0fc2011-10-10 17:26:25 +00003884 @return The line of text from the file.
3885 @retval NULL There was not enough memory available.
jcarsey4d0a4fc2011-07-06 22:28:36 +00003886
3887 @sa ShellFileHandleReadLine
3888**/
3889CHAR16*
3890EFIAPI
3891ShellFileHandleReturnLine(
3892 IN SHELL_FILE_HANDLE Handle,
3893 IN OUT BOOLEAN *Ascii
3894 )
3895{
3896 CHAR16 *RetVal;
3897 UINTN Size;
3898 EFI_STATUS Status;
3899
3900 Size = 0;
3901 RetVal = NULL;
3902
3903 Status = ShellFileHandleReadLine(Handle, RetVal, &Size, FALSE, Ascii);
3904 if (Status == EFI_BUFFER_TOO_SMALL) {
3905 RetVal = AllocateZeroPool(Size);
jcarseybeab0fc2011-10-10 17:26:25 +00003906 if (RetVal == NULL) {
3907 return (NULL);
3908 }
jcarsey4d0a4fc2011-07-06 22:28:36 +00003909 Status = ShellFileHandleReadLine(Handle, RetVal, &Size, FALSE, Ascii);
darylm503b0934ac2011-11-12 00:35:11 +00003910
jcarsey4d0a4fc2011-07-06 22:28:36 +00003911 }
jcarsey4d0a4fc2011-07-06 22:28:36 +00003912 if (EFI_ERROR(Status) && (RetVal != NULL)) {
3913 FreePool(RetVal);
3914 RetVal = NULL;
3915 }
3916 return (RetVal);
3917}
3918
3919/**
3920 Function to read a single line (up to but not including the \n) from a SHELL_FILE_HANDLE.
3921
3922 If the position upon start is 0, then the Ascii Boolean will be set. This should be
3923 maintained and not changed for all operations with the same file.
3924
ydong104ff7e372011-09-02 08:05:34 +00003925 @param[in] Handle SHELL_FILE_HANDLE to read from.
3926 @param[in, out] Buffer The pointer to buffer to read into.
3927 @param[in, out] Size The pointer to number of bytes in Buffer.
3928 @param[in] Truncate If the buffer is large enough, this has no effect.
3929 If the buffer is is too small and Truncate is TRUE,
3930 the line will be truncated.
3931 If the buffer is is too small and Truncate is FALSE,
3932 then no read will occur.
jcarsey4d0a4fc2011-07-06 22:28:36 +00003933
ydong104ff7e372011-09-02 08:05:34 +00003934 @param[in, out] Ascii Boolean value for indicating whether the file is
3935 Ascii (TRUE) or UCS2 (FALSE).
jcarsey4d0a4fc2011-07-06 22:28:36 +00003936
3937 @retval EFI_SUCCESS The operation was successful. The line is stored in
3938 Buffer.
3939 @retval EFI_INVALID_PARAMETER Handle was NULL.
3940 @retval EFI_INVALID_PARAMETER Size was NULL.
3941 @retval EFI_BUFFER_TOO_SMALL Size was not large enough to store the line.
3942 Size was updated to the minimum space required.
3943**/
3944EFI_STATUS
3945EFIAPI
3946ShellFileHandleReadLine(
3947 IN SHELL_FILE_HANDLE Handle,
3948 IN OUT CHAR16 *Buffer,
3949 IN OUT UINTN *Size,
3950 IN BOOLEAN Truncate,
3951 IN OUT BOOLEAN *Ascii
3952 )
3953{
3954 EFI_STATUS Status;
3955 CHAR16 CharBuffer;
3956 UINTN CharSize;
3957 UINTN CountSoFar;
3958 UINT64 OriginalFilePosition;
3959
3960
3961 if (Handle == NULL
3962 ||Size == NULL
3963 ){
3964 return (EFI_INVALID_PARAMETER);
3965 }
3966 if (Buffer == NULL) {
3967 ASSERT(*Size == 0);
3968 } else {
3969 *Buffer = CHAR_NULL;
3970 }
3971 gEfiShellProtocol->GetFilePosition(Handle, &OriginalFilePosition);
3972 if (OriginalFilePosition == 0) {
3973 CharSize = sizeof(CHAR16);
3974 Status = gEfiShellProtocol->ReadFile(Handle, &CharSize, &CharBuffer);
3975 ASSERT_EFI_ERROR(Status);
3976 if (CharBuffer == gUnicodeFileTag) {
3977 *Ascii = FALSE;
3978 } else {
3979 *Ascii = TRUE;
3980 gEfiShellProtocol->SetFilePosition(Handle, OriginalFilePosition);
3981 }
3982 }
3983
3984 for (CountSoFar = 0;;CountSoFar++){
3985 CharBuffer = 0;
3986 if (*Ascii) {
3987 CharSize = sizeof(CHAR8);
3988 } else {
3989 CharSize = sizeof(CHAR16);
3990 }
3991 Status = gEfiShellProtocol->ReadFile(Handle, &CharSize, &CharBuffer);
3992 if ( EFI_ERROR(Status)
3993 || CharSize == 0
3994 || (CharBuffer == L'\n' && !(*Ascii))
3995 || (CharBuffer == '\n' && *Ascii)
3996 ){
3997 break;
3998 }
3999 //
4000 // if we have space save it...
4001 //
4002 if ((CountSoFar+1)*sizeof(CHAR16) < *Size){
4003 ASSERT(Buffer != NULL);
4004 ((CHAR16*)Buffer)[CountSoFar] = CharBuffer;
4005 ((CHAR16*)Buffer)[CountSoFar+1] = CHAR_NULL;
4006 }
4007 }
4008
4009 //
4010 // if we ran out of space tell when...
4011 //
4012 if ((CountSoFar+1)*sizeof(CHAR16) > *Size){
4013 *Size = (CountSoFar+1)*sizeof(CHAR16);
4014 if (!Truncate) {
4015 gEfiShellProtocol->SetFilePosition(Handle, OriginalFilePosition);
4016 } else {
4017 DEBUG((DEBUG_WARN, "The line was truncated in ShellFileHandleReadLine"));
4018 }
4019 return (EFI_BUFFER_TOO_SMALL);
4020 }
4021 while(Buffer[StrLen(Buffer)-1] == L'\r') {
4022 Buffer[StrLen(Buffer)-1] = CHAR_NULL;
4023 }
4024
4025 return (Status);
4026}