]> git.proxmox.com Git - mirror_edk2.git/blob - ShellPkg/Library/UefiShellLib/UefiShellLib.c
remove unnecessary PCD usage
[mirror_edk2.git] / ShellPkg / Library / UefiShellLib / UefiShellLib.c
1 /** @file
2 Provides interface to shell functionality for shell commands and applications.
3
4 Copyright (c) 2006 - 2010, Intel Corporation. All rights reserved.<BR>
5 This program and the accompanying materials
6 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
9
10 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.
12
13 **/
14
15 #include "UefiShellLib.h"
16
17 #define MAX_FILE_NAME_LEN 522 // (20 * (6+5+2))+1) unicode characters from EFI FAT spec (doubled for bytes)
18 #define FIND_XXXXX_FILE_BUFFER_SIZE (SIZE_OF_EFI_FILE_INFO + MAX_FILE_NAME_LEN)
19
20 //
21 // This is not static since it's extern in the .h file
22 //
23 SHELL_PARAM_ITEM EmptyParamList[] = {
24 {NULL, TypeMax}
25 };
26
27 //
28 // Static file globals for the shell library
29 //
30 STATIC EFI_SHELL_ENVIRONMENT2 *mEfiShellEnvironment2;
31 STATIC EFI_SHELL_INTERFACE *mEfiShellInterface;
32 STATIC EFI_SHELL_PROTOCOL *mEfiShellProtocol;
33 STATIC EFI_SHELL_PARAMETERS_PROTOCOL *mEfiShellParametersProtocol;
34 STATIC EFI_HANDLE mEfiShellEnvironment2Handle;
35 STATIC FILE_HANDLE_FUNCTION_MAP FileFunctionMap;
36 STATIC UINTN mTotalParameterCount;
37 STATIC CHAR16 *mPostReplaceFormat;
38 STATIC CHAR16 *mPostReplaceFormat2;
39
40 /**
41 Check if a Unicode character is a hexadecimal character.
42
43 This internal function checks if a Unicode character is a
44 decimal character. The valid hexadecimal character is
45 L'0' to L'9', L'a' to L'f', or L'A' to L'F'.
46
47
48 @param Char The character to check against.
49
50 @retval TRUE If the Char is a hexadecmial character.
51 @retval FALSE If the Char is not a hexadecmial character.
52
53 **/
54 BOOLEAN
55 EFIAPI
56 ShellIsHexaDecimalDigitCharacter (
57 IN CHAR16 Char
58 ) {
59 return (BOOLEAN) ((Char >= L'0' && Char <= L'9') || (Char >= L'A' && Char <= L'F') || (Char >= L'a' && Char <= L'f'));
60 }
61
62 /**
63 helper function to find ShellEnvironment2 for constructor
64 **/
65 EFI_STATUS
66 EFIAPI
67 ShellFindSE2 (
68 IN EFI_HANDLE ImageHandle
69 ) {
70 EFI_STATUS Status;
71 EFI_HANDLE *Buffer;
72 UINTN BufferSize;
73 UINTN HandleIndex;
74
75 BufferSize = 0;
76 Buffer = NULL;
77 Status = gBS->OpenProtocol(ImageHandle,
78 &gEfiShellEnvironment2Guid,
79 (VOID **)&mEfiShellEnvironment2,
80 ImageHandle,
81 NULL,
82 EFI_OPEN_PROTOCOL_GET_PROTOCOL
83 );
84 //
85 // look for the mEfiShellEnvironment2 protocol at a higher level
86 //
87 if (EFI_ERROR (Status) || !(CompareGuid (&mEfiShellEnvironment2->SESGuid, &gEfiShellEnvironment2ExtGuid) != FALSE)){
88 //
89 // figure out how big of a buffer we need.
90 //
91 Status = gBS->LocateHandle (ByProtocol,
92 &gEfiShellEnvironment2Guid,
93 NULL, // ignored for ByProtocol
94 &BufferSize,
95 Buffer
96 );
97 //
98 // maybe it's not there???
99 //
100 if (Status == EFI_BUFFER_TOO_SMALL) {
101 Buffer = (EFI_HANDLE*)AllocatePool(BufferSize);
102 ASSERT(Buffer != NULL);
103 Status = gBS->LocateHandle (ByProtocol,
104 &gEfiShellEnvironment2Guid,
105 NULL, // ignored for ByProtocol
106 &BufferSize,
107 Buffer
108 );
109 }
110 if (!EFI_ERROR (Status) && Buffer != NULL) {
111 //
112 // now parse the list of returned handles
113 //
114 Status = EFI_NOT_FOUND;
115 for (HandleIndex = 0; HandleIndex < (BufferSize/sizeof(Buffer[0])); HandleIndex++) {
116 Status = gBS->OpenProtocol(Buffer[HandleIndex],
117 &gEfiShellEnvironment2Guid,
118 (VOID **)&mEfiShellEnvironment2,
119 ImageHandle,
120 NULL,
121 EFI_OPEN_PROTOCOL_GET_PROTOCOL
122 );
123 if (CompareGuid (&mEfiShellEnvironment2->SESGuid, &gEfiShellEnvironment2ExtGuid) != FALSE) {
124 mEfiShellEnvironment2Handle = Buffer[HandleIndex];
125 Status = EFI_SUCCESS;
126 break;
127 }
128 }
129 }
130 }
131 if (Buffer != NULL) {
132 FreePool (Buffer);
133 }
134 return (Status);
135 }
136
137 EFI_STATUS
138 EFIAPI
139 ShellLibConstructorWorker (
140 IN EFI_HANDLE ImageHandle,
141 IN EFI_SYSTEM_TABLE *SystemTable
142 ) {
143 EFI_STATUS Status;
144
145 mPostReplaceFormat = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
146 ASSERT (mPostReplaceFormat != NULL);
147 mPostReplaceFormat2 = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
148 ASSERT (mPostReplaceFormat2 != NULL);
149
150 //
151 // Set the parameter count to an invalid number
152 //
153 mTotalParameterCount = (UINTN)(-1);
154
155 //
156 // UEFI 2.0 shell interfaces (used preferentially)
157 //
158 Status = gBS->OpenProtocol(ImageHandle,
159 &gEfiShellProtocolGuid,
160 (VOID **)&mEfiShellProtocol,
161 ImageHandle,
162 NULL,
163 EFI_OPEN_PROTOCOL_GET_PROTOCOL
164 );
165 if (EFI_ERROR(Status)) {
166 mEfiShellProtocol = NULL;
167 }
168 Status = gBS->OpenProtocol(ImageHandle,
169 &gEfiShellParametersProtocolGuid,
170 (VOID **)&mEfiShellParametersProtocol,
171 ImageHandle,
172 NULL,
173 EFI_OPEN_PROTOCOL_GET_PROTOCOL
174 );
175 if (EFI_ERROR(Status)) {
176 mEfiShellParametersProtocol = NULL;
177 }
178
179 if (mEfiShellParametersProtocol == NULL || mEfiShellProtocol == NULL) {
180 //
181 // Moved to seperate function due to complexity
182 //
183 Status = ShellFindSE2(ImageHandle);
184
185 if (EFI_ERROR(Status)) {
186 DEBUG((DEBUG_ERROR, "Status: 0x%08x\r\n", Status));
187 mEfiShellEnvironment2 = NULL;
188 }
189 Status = gBS->OpenProtocol(ImageHandle,
190 &gEfiShellInterfaceGuid,
191 (VOID **)&mEfiShellInterface,
192 ImageHandle,
193 NULL,
194 EFI_OPEN_PROTOCOL_GET_PROTOCOL
195 );
196 if (EFI_ERROR(Status)) {
197 mEfiShellInterface = NULL;
198 }
199 }
200 //
201 // only success getting 2 of either the old or new, but no 1/2 and 1/2
202 //
203 if ((mEfiShellEnvironment2 != NULL && mEfiShellInterface != NULL) ||
204 (mEfiShellProtocol != NULL && mEfiShellParametersProtocol != NULL) ) {
205 if (mEfiShellProtocol != NULL) {
206 FileFunctionMap.GetFileInfo = mEfiShellProtocol->GetFileInfo;
207 FileFunctionMap.SetFileInfo = mEfiShellProtocol->SetFileInfo;
208 FileFunctionMap.ReadFile = mEfiShellProtocol->ReadFile;
209 FileFunctionMap.WriteFile = mEfiShellProtocol->WriteFile;
210 FileFunctionMap.CloseFile = mEfiShellProtocol->CloseFile;
211 FileFunctionMap.DeleteFile = mEfiShellProtocol->DeleteFile;
212 FileFunctionMap.GetFilePosition = mEfiShellProtocol->GetFilePosition;
213 FileFunctionMap.SetFilePosition = mEfiShellProtocol->SetFilePosition;
214 FileFunctionMap.FlushFile = mEfiShellProtocol->FlushFile;
215 FileFunctionMap.GetFileSize = mEfiShellProtocol->GetFileSize;
216 } else {
217 FileFunctionMap.GetFileInfo = FileHandleGetInfo;
218 FileFunctionMap.SetFileInfo = FileHandleSetInfo;
219 FileFunctionMap.ReadFile = FileHandleRead;
220 FileFunctionMap.WriteFile = FileHandleWrite;
221 FileFunctionMap.CloseFile = FileHandleClose;
222 FileFunctionMap.DeleteFile = FileHandleDelete;
223 FileFunctionMap.GetFilePosition = FileHandleGetPosition;
224 FileFunctionMap.SetFilePosition = FileHandleSetPosition;
225 FileFunctionMap.FlushFile = FileHandleFlush;
226 FileFunctionMap.GetFileSize = FileHandleGetSize;
227 }
228 return (EFI_SUCCESS);
229 }
230 return (EFI_NOT_FOUND);
231 }
232 /**
233 Constructor for the Shell library.
234
235 Initialize the library and determine if the underlying is a UEFI Shell 2.0 or an EFI shell.
236
237 @param ImageHandle the image handle of the process
238 @param SystemTable the EFI System Table pointer
239
240 @retval EFI_SUCCESS the initialization was complete sucessfully
241 @return others an error ocurred during initialization
242 **/
243 EFI_STATUS
244 EFIAPI
245 ShellLibConstructor (
246 IN EFI_HANDLE ImageHandle,
247 IN EFI_SYSTEM_TABLE *SystemTable
248 ) {
249
250
251 mEfiShellEnvironment2 = NULL;
252 mEfiShellProtocol = NULL;
253 mEfiShellParametersProtocol = NULL;
254 mEfiShellInterface = NULL;
255 mEfiShellEnvironment2Handle = NULL;
256 mPostReplaceFormat = NULL;
257 mPostReplaceFormat2 = NULL;
258
259 //
260 // verify that auto initialize is not set false
261 //
262 if (PcdGetBool(PcdShellLibAutoInitialize) == 0) {
263 return (EFI_SUCCESS);
264 }
265
266 return (ShellLibConstructorWorker(ImageHandle, SystemTable));
267 }
268
269 /**
270 Destructory for the library. free any resources.
271 **/
272 EFI_STATUS
273 EFIAPI
274 ShellLibDestructor (
275 IN EFI_HANDLE ImageHandle,
276 IN EFI_SYSTEM_TABLE *SystemTable
277 ) {
278 if (mEfiShellEnvironment2 != NULL) {
279 gBS->CloseProtocol(mEfiShellEnvironment2Handle==NULL?ImageHandle:mEfiShellEnvironment2Handle,
280 &gEfiShellEnvironment2Guid,
281 ImageHandle,
282 NULL);
283 mEfiShellEnvironment2 = NULL;
284 }
285 if (mEfiShellInterface != NULL) {
286 gBS->CloseProtocol(ImageHandle,
287 &gEfiShellInterfaceGuid,
288 ImageHandle,
289 NULL);
290 mEfiShellInterface = NULL;
291 }
292 if (mEfiShellProtocol != NULL) {
293 gBS->CloseProtocol(ImageHandle,
294 &gEfiShellProtocolGuid,
295 ImageHandle,
296 NULL);
297 mEfiShellProtocol = NULL;
298 }
299 if (mEfiShellParametersProtocol != NULL) {
300 gBS->CloseProtocol(ImageHandle,
301 &gEfiShellParametersProtocolGuid,
302 ImageHandle,
303 NULL);
304 mEfiShellParametersProtocol = NULL;
305 }
306 mEfiShellEnvironment2Handle = NULL;
307
308 if (mPostReplaceFormat != NULL) {
309 FreePool(mPostReplaceFormat);
310 }
311 if (mPostReplaceFormat2 != NULL) {
312 FreePool(mPostReplaceFormat2);
313 }
314 mPostReplaceFormat = NULL;
315 mPostReplaceFormat2 = NULL;
316
317 return (EFI_SUCCESS);
318 }
319
320 /**
321 This function causes the shell library to initialize itself. If the shell library
322 is already initialized it will de-initialize all the current protocol poitners and
323 re-populate them again.
324
325 When the library is used with PcdShellLibAutoInitialize set to true this function
326 will return EFI_SUCCESS and perform no actions.
327
328 This function is intended for internal access for shell commands only.
329
330 @retval EFI_SUCCESS the initialization was complete sucessfully
331
332 **/
333 EFI_STATUS
334 EFIAPI
335 ShellInitialize (
336 ) {
337 //
338 // if auto initialize is not false then skip
339 //
340 if (PcdGetBool(PcdShellLibAutoInitialize) != 0) {
341 return (EFI_SUCCESS);
342 }
343
344 //
345 // deinit the current stuff
346 //
347 ASSERT_EFI_ERROR(ShellLibDestructor(gImageHandle, gST));
348
349 //
350 // init the new stuff
351 //
352 return (ShellLibConstructorWorker(gImageHandle, gST));
353 }
354
355 /**
356 This function will retrieve the information about the file for the handle
357 specified and store it in allocated pool memory.
358
359 This function allocates a buffer to store the file's information. It is the
360 caller's responsibility to free the buffer
361
362 @param FileHandle The file handle of the file for which information is
363 being requested.
364
365 @retval NULL information could not be retrieved.
366
367 @return the information about the file
368 **/
369 EFI_FILE_INFO*
370 EFIAPI
371 ShellGetFileInfo (
372 IN EFI_FILE_HANDLE FileHandle
373 ) {
374 return (FileFunctionMap.GetFileInfo(FileHandle));
375 }
376
377 /**
378 This function will set the information about the file for the opened handle
379 specified.
380
381 @param FileHandle The file handle of the file for which information
382 is being set
383
384 @param FileInfo The infotmation to set.
385
386 @retval EFI_SUCCESS The information was set.
387 @retval EFI_UNSUPPORTED The InformationType is not known.
388 @retval EFI_NO_MEDIA The device has no medium.
389 @retval EFI_DEVICE_ERROR The device reported an error.
390 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
391 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
392 @retval EFI_ACCESS_DENIED The file was opened read only.
393 @retval EFI_VOLUME_FULL The volume is full.
394 **/
395 EFI_STATUS
396 EFIAPI
397 ShellSetFileInfo (
398 IN EFI_FILE_HANDLE FileHandle,
399 IN EFI_FILE_INFO *FileInfo
400 ) {
401 return (FileFunctionMap.SetFileInfo(FileHandle, FileInfo));
402 }
403
404 /**
405 This function will open a file or directory referenced by DevicePath.
406
407 This function opens a file with the open mode according to the file path. The
408 Attributes is valid only for EFI_FILE_MODE_CREATE.
409
410 @param FilePath on input the device path to the file. On output
411 the remaining device path.
412 @param DeviceHandle pointer to the system device handle.
413 @param FileHandle pointer to the file handle.
414 @param OpenMode the mode to open the file with.
415 @param Attributes the file's file attributes.
416
417 @retval EFI_SUCCESS The information was set.
418 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
419 @retval EFI_UNSUPPORTED Could not open the file path.
420 @retval EFI_NOT_FOUND The specified file could not be found on the
421 device or the file system could not be found on
422 the device.
423 @retval EFI_NO_MEDIA The device has no medium.
424 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
425 medium is no longer supported.
426 @retval EFI_DEVICE_ERROR The device reported an error.
427 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
428 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
429 @retval EFI_ACCESS_DENIED The file was opened read only.
430 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
431 file.
432 @retval EFI_VOLUME_FULL The volume is full.
433 **/
434 EFI_STATUS
435 EFIAPI
436 ShellOpenFileByDevicePath(
437 IN OUT EFI_DEVICE_PATH_PROTOCOL **FilePath,
438 OUT EFI_HANDLE *DeviceHandle,
439 OUT EFI_FILE_HANDLE *FileHandle,
440 IN UINT64 OpenMode,
441 IN UINT64 Attributes
442 ) {
443 CHAR16 *FileName;
444 EFI_STATUS Status;
445 EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *EfiSimpleFileSystemProtocol;
446 EFI_FILE_HANDLE LastHandle;
447
448 //
449 // ASERT for FileHandle, FilePath, and DeviceHandle being NULL
450 //
451 ASSERT(FilePath != NULL);
452 ASSERT(FileHandle != NULL);
453 ASSERT(DeviceHandle != NULL);
454 //
455 // which shell interface should we use
456 //
457 if (mEfiShellProtocol != NULL) {
458 //
459 // use UEFI Shell 2.0 method.
460 //
461 FileName = mEfiShellProtocol->GetFilePathFromDevicePath(*FilePath);
462 if (FileName == NULL) {
463 return (EFI_INVALID_PARAMETER);
464 }
465 Status = ShellOpenFileByName(FileName, FileHandle, OpenMode, Attributes);
466 FreePool(FileName);
467 return (Status);
468 }
469
470
471 //
472 // use old shell method.
473 //
474 Status = gBS->LocateDevicePath (&gEfiSimpleFileSystemProtocolGuid,
475 FilePath,
476 DeviceHandle);
477 if (EFI_ERROR (Status)) {
478 return Status;
479 }
480 Status = gBS->OpenProtocol(*DeviceHandle,
481 &gEfiSimpleFileSystemProtocolGuid,
482 (VOID**)&EfiSimpleFileSystemProtocol,
483 gImageHandle,
484 NULL,
485 EFI_OPEN_PROTOCOL_GET_PROTOCOL);
486 if (EFI_ERROR (Status)) {
487 return Status;
488 }
489 Status = EfiSimpleFileSystemProtocol->OpenVolume(EfiSimpleFileSystemProtocol, FileHandle);
490 if (EFI_ERROR (Status)) {
491 FileHandle = NULL;
492 return Status;
493 }
494
495 //
496 // go down directories one node at a time.
497 //
498 while (!IsDevicePathEnd (*FilePath)) {
499 //
500 // For file system access each node should be a file path component
501 //
502 if (DevicePathType (*FilePath) != MEDIA_DEVICE_PATH ||
503 DevicePathSubType (*FilePath) != MEDIA_FILEPATH_DP
504 ) {
505 FileHandle = NULL;
506 return (EFI_INVALID_PARAMETER);
507 }
508 //
509 // Open this file path node
510 //
511 LastHandle = *FileHandle;
512 *FileHandle = NULL;
513
514 //
515 // Try to test opening an existing file
516 //
517 Status = LastHandle->Open (
518 LastHandle,
519 FileHandle,
520 ((FILEPATH_DEVICE_PATH*)*FilePath)->PathName,
521 OpenMode &~EFI_FILE_MODE_CREATE,
522 0
523 );
524
525 //
526 // see if the error was that it needs to be created
527 //
528 if ((EFI_ERROR (Status)) && (OpenMode != (OpenMode &~EFI_FILE_MODE_CREATE))) {
529 Status = LastHandle->Open (
530 LastHandle,
531 FileHandle,
532 ((FILEPATH_DEVICE_PATH*)*FilePath)->PathName,
533 OpenMode,
534 Attributes
535 );
536 }
537 //
538 // Close the last node
539 //
540 LastHandle->Close (LastHandle);
541
542 if (EFI_ERROR(Status)) {
543 return (Status);
544 }
545
546 //
547 // Get the next node
548 //
549 *FilePath = NextDevicePathNode (*FilePath);
550 }
551 return (EFI_SUCCESS);
552 }
553
554 /**
555 This function will open a file or directory referenced by filename.
556
557 If return is EFI_SUCCESS, the Filehandle is the opened file's handle;
558 otherwise, the Filehandle is NULL. The Attributes is valid only for
559 EFI_FILE_MODE_CREATE.
560
561 if FileNAme is NULL then ASSERT()
562
563 @param FileName pointer to file name
564 @param FileHandle pointer to the file handle.
565 @param OpenMode the mode to open the file with.
566 @param Attributes the file's file attributes.
567
568 @retval EFI_SUCCESS The information was set.
569 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
570 @retval EFI_UNSUPPORTED Could not open the file path.
571 @retval EFI_NOT_FOUND The specified file could not be found on the
572 device or the file system could not be found
573 on the device.
574 @retval EFI_NO_MEDIA The device has no medium.
575 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
576 medium is no longer supported.
577 @retval EFI_DEVICE_ERROR The device reported an error.
578 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
579 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
580 @retval EFI_ACCESS_DENIED The file was opened read only.
581 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
582 file.
583 @retval EFI_VOLUME_FULL The volume is full.
584 **/
585 EFI_STATUS
586 EFIAPI
587 ShellOpenFileByName(
588 IN CONST CHAR16 *FileName,
589 OUT EFI_FILE_HANDLE *FileHandle,
590 IN UINT64 OpenMode,
591 IN UINT64 Attributes
592 ) {
593 EFI_HANDLE DeviceHandle;
594 EFI_DEVICE_PATH_PROTOCOL *FilePath;
595 EFI_STATUS Status;
596 EFI_FILE_INFO *FileInfo;
597
598 //
599 // ASSERT if FileName is NULL
600 //
601 ASSERT(FileName != NULL);
602
603 if (mEfiShellProtocol != NULL) {
604 //
605 // Use UEFI Shell 2.0 method
606 //
607 Status = mEfiShellProtocol->OpenFileByName(FileName,
608 FileHandle,
609 OpenMode);
610 if (!EFI_ERROR(Status) && ((OpenMode & EFI_FILE_MODE_CREATE) != 0)){
611 FileInfo = FileFunctionMap.GetFileInfo(*FileHandle);
612 ASSERT(FileInfo != NULL);
613 FileInfo->Attribute = Attributes;
614 Status = FileFunctionMap.SetFileInfo(*FileHandle, FileInfo);
615 FreePool(FileInfo);
616 }
617 return (Status);
618 }
619 //
620 // Using EFI Shell version
621 // this means convert name to path and call that function
622 // since this will use EFI method again that will open it.
623 //
624 ASSERT(mEfiShellEnvironment2 != NULL);
625 FilePath = mEfiShellEnvironment2->NameToPath ((CHAR16*)FileName);
626 if (FileDevicePath != NULL) {
627 return (ShellOpenFileByDevicePath(&FilePath,
628 &DeviceHandle,
629 FileHandle,
630 OpenMode,
631 Attributes ));
632 }
633 return (EFI_DEVICE_ERROR);
634 }
635 /**
636 This function create a directory
637
638 If return is EFI_SUCCESS, the Filehandle is the opened directory's handle;
639 otherwise, the Filehandle is NULL. If the directory already existed, this
640 function opens the existing directory.
641
642 @param DirectoryName pointer to directory name
643 @param FileHandle pointer to the file handle.
644
645 @retval EFI_SUCCESS The information was set.
646 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
647 @retval EFI_UNSUPPORTED Could not open the file path.
648 @retval EFI_NOT_FOUND The specified file could not be found on the
649 device or the file system could not be found
650 on the device.
651 @retval EFI_NO_MEDIA The device has no medium.
652 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
653 medium is no longer supported.
654 @retval EFI_DEVICE_ERROR The device reported an error.
655 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
656 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
657 @retval EFI_ACCESS_DENIED The file was opened read only.
658 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
659 file.
660 @retval EFI_VOLUME_FULL The volume is full.
661 @sa ShellOpenFileByName
662 **/
663 EFI_STATUS
664 EFIAPI
665 ShellCreateDirectory(
666 IN CONST CHAR16 *DirectoryName,
667 OUT EFI_FILE_HANDLE *FileHandle
668 ) {
669 if (mEfiShellProtocol != NULL) {
670 //
671 // Use UEFI Shell 2.0 method
672 //
673 return (mEfiShellProtocol->CreateFile(DirectoryName,
674 EFI_FILE_DIRECTORY,
675 FileHandle
676 ));
677 } else {
678 return (ShellOpenFileByName(DirectoryName,
679 FileHandle,
680 EFI_FILE_MODE_READ | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE,
681 EFI_FILE_DIRECTORY
682 ));
683 }
684 }
685
686 /**
687 This function reads information from an opened file.
688
689 If FileHandle is not a directory, the function reads the requested number of
690 bytes from the file at the file's current position and returns them in Buffer.
691 If the read goes beyond the end of the file, the read length is truncated to the
692 end of the file. The file's current position is increased by the number of bytes
693 returned. If FileHandle is a directory, the function reads the directory entry
694 at the file's current position and returns the entry in Buffer. If the Buffer
695 is not large enough to hold the current directory entry, then
696 EFI_BUFFER_TOO_SMALL is returned and the current file position is not updated.
697 BufferSize is set to be the size of the buffer needed to read the entry. On
698 success, the current position is updated to the next directory entry. If there
699 are no more directory entries, the read returns a zero-length buffer.
700 EFI_FILE_INFO is the structure returned as the directory entry.
701
702 @param FileHandle the opened file handle
703 @param BufferSize on input the size of buffer in bytes. on return
704 the number of bytes written.
705 @param Buffer the buffer to put read data into.
706
707 @retval EFI_SUCCESS Data was read.
708 @retval EFI_NO_MEDIA The device has no media.
709 @retval EFI_DEVICE_ERROR The device reported an error.
710 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
711 @retval EFI_BUFFER_TO_SMALL Buffer is too small. ReadSize contains required
712 size.
713
714 **/
715 EFI_STATUS
716 EFIAPI
717 ShellReadFile(
718 IN EFI_FILE_HANDLE FileHandle,
719 IN OUT UINTN *BufferSize,
720 OUT VOID *Buffer
721 ) {
722 return (FileFunctionMap.ReadFile(FileHandle, BufferSize, Buffer));
723 }
724
725
726 /**
727 Write data to a file.
728
729 This function writes the specified number of bytes to the file at the current
730 file position. The current file position is advanced the actual number of bytes
731 written, which is returned in BufferSize. Partial writes only occur when there
732 has been a data error during the write attempt (such as "volume space full").
733 The file is automatically grown to hold the data if required. Direct writes to
734 opened directories are not supported.
735
736 @param FileHandle The opened file for writing
737 @param BufferSize on input the number of bytes in Buffer. On output
738 the number of bytes written.
739 @param Buffer the buffer containing data to write is stored.
740
741 @retval EFI_SUCCESS Data was written.
742 @retval EFI_UNSUPPORTED Writes to an open directory are not supported.
743 @retval EFI_NO_MEDIA The device has no media.
744 @retval EFI_DEVICE_ERROR The device reported an error.
745 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
746 @retval EFI_WRITE_PROTECTED The device is write-protected.
747 @retval EFI_ACCESS_DENIED The file was open for read only.
748 @retval EFI_VOLUME_FULL The volume is full.
749 **/
750 EFI_STATUS
751 EFIAPI
752 ShellWriteFile(
753 IN EFI_FILE_HANDLE FileHandle,
754 IN OUT UINTN *BufferSize,
755 IN VOID *Buffer
756 ) {
757 return (FileFunctionMap.WriteFile(FileHandle, BufferSize, Buffer));
758 }
759
760 /**
761 Close an open file handle.
762
763 This function closes a specified file handle. All "dirty" cached file data is
764 flushed to the device, and the file is closed. In all cases the handle is
765 closed.
766
767 @param FileHandle the file handle to close.
768
769 @retval EFI_SUCCESS the file handle was closed sucessfully.
770 **/
771 EFI_STATUS
772 EFIAPI
773 ShellCloseFile (
774 IN EFI_FILE_HANDLE *FileHandle
775 ) {
776 return (FileFunctionMap.CloseFile(*FileHandle));
777 }
778
779 /**
780 Delete a file and close the handle
781
782 This function closes and deletes a file. In all cases the file handle is closed.
783 If the file cannot be deleted, the warning code EFI_WARN_DELETE_FAILURE is
784 returned, but the handle is still closed.
785
786 @param FileHandle the file handle to delete
787
788 @retval EFI_SUCCESS the file was closed sucessfully
789 @retval EFI_WARN_DELETE_FAILURE the handle was closed, but the file was not
790 deleted
791 @retval INVALID_PARAMETER One of the parameters has an invalid value.
792 **/
793 EFI_STATUS
794 EFIAPI
795 ShellDeleteFile (
796 IN EFI_FILE_HANDLE *FileHandle
797 ) {
798 return (FileFunctionMap.DeleteFile(*FileHandle));
799 }
800
801 /**
802 Set the current position in a file.
803
804 This function sets the current file position for the handle to the position
805 supplied. With the exception of seeking to position 0xFFFFFFFFFFFFFFFF, only
806 absolute positioning is supported, and seeking past the end of the file is
807 allowed (a subsequent write would grow the file). Seeking to position
808 0xFFFFFFFFFFFFFFFF causes the current position to be set to the end of the file.
809 If FileHandle is a directory, the only position that may be set is zero. This
810 has the effect of starting the read process of the directory entries over.
811
812 @param FileHandle The file handle on which the position is being set
813 @param Position Byte position from begining of file
814
815 @retval EFI_SUCCESS Operation completed sucessfully.
816 @retval EFI_UNSUPPORTED the seek request for non-zero is not valid on
817 directories.
818 @retval INVALID_PARAMETER One of the parameters has an invalid value.
819 **/
820 EFI_STATUS
821 EFIAPI
822 ShellSetFilePosition (
823 IN EFI_FILE_HANDLE FileHandle,
824 IN UINT64 Position
825 ) {
826 return (FileFunctionMap.SetFilePosition(FileHandle, Position));
827 }
828
829 /**
830 Gets a file's current position
831
832 This function retrieves the current file position for the file handle. For
833 directories, the current file position has no meaning outside of the file
834 system driver and as such the operation is not supported. An error is returned
835 if FileHandle is a directory.
836
837 @param FileHandle The open file handle on which to get the position.
838 @param Position Byte position from begining of file.
839
840 @retval EFI_SUCCESS the operation completed sucessfully.
841 @retval INVALID_PARAMETER One of the parameters has an invalid value.
842 @retval EFI_UNSUPPORTED the request is not valid on directories.
843 **/
844 EFI_STATUS
845 EFIAPI
846 ShellGetFilePosition (
847 IN EFI_FILE_HANDLE FileHandle,
848 OUT UINT64 *Position
849 ) {
850 return (FileFunctionMap.GetFilePosition(FileHandle, Position));
851 }
852 /**
853 Flushes data on a file
854
855 This function flushes all modified data associated with a file to a device.
856
857 @param FileHandle The file handle on which to flush data
858
859 @retval EFI_SUCCESS The data was flushed.
860 @retval EFI_NO_MEDIA The device has no media.
861 @retval EFI_DEVICE_ERROR The device reported an error.
862 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
863 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
864 @retval EFI_ACCESS_DENIED The file was opened for read only.
865 **/
866 EFI_STATUS
867 EFIAPI
868 ShellFlushFile (
869 IN EFI_FILE_HANDLE FileHandle
870 ) {
871 return (FileFunctionMap.FlushFile(FileHandle));
872 }
873
874 /**
875 Retrieves the first file from a directory
876
877 This function opens a directory and gets the first file's info in the
878 directory. Caller can use ShellFindNextFile() to get other files. When
879 complete the caller is responsible for calling FreePool() on Buffer.
880
881 @param DirHandle The file handle of the directory to search
882 @param Buffer Pointer to buffer for file's information
883
884 @retval EFI_SUCCESS Found the first file.
885 @retval EFI_NOT_FOUND Cannot find the directory.
886 @retval EFI_NO_MEDIA The device has no media.
887 @retval EFI_DEVICE_ERROR The device reported an error.
888 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
889 @return Others status of ShellGetFileInfo, ShellSetFilePosition,
890 or ShellReadFile
891 **/
892 EFI_STATUS
893 EFIAPI
894 ShellFindFirstFile (
895 IN EFI_FILE_HANDLE DirHandle,
896 OUT EFI_FILE_INFO **Buffer
897 ) {
898 //
899 // pass to file handle lib
900 //
901 return (FileHandleFindFirstFile(DirHandle, Buffer));
902 }
903 /**
904 Retrieves the next file in a directory.
905
906 To use this function, caller must call the LibFindFirstFile() to get the
907 first file, and then use this function get other files. This function can be
908 called for several times to get each file's information in the directory. If
909 the call of ShellFindNextFile() got the last file in the directory, the next
910 call of this function has no file to get. *NoFile will be set to TRUE and the
911 Buffer memory will be automatically freed.
912
913 @param DirHandle the file handle of the directory
914 @param Buffer pointer to buffer for file's information
915 @param NoFile pointer to boolean when last file is found
916
917 @retval EFI_SUCCESS Found the next file, or reached last file
918 @retval EFI_NO_MEDIA The device has no media.
919 @retval EFI_DEVICE_ERROR The device reported an error.
920 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
921 **/
922 EFI_STATUS
923 EFIAPI
924 ShellFindNextFile(
925 IN EFI_FILE_HANDLE DirHandle,
926 OUT EFI_FILE_INFO *Buffer,
927 OUT BOOLEAN *NoFile
928 ) {
929 //
930 // pass to file handle lib
931 //
932 return (FileHandleFindNextFile(DirHandle, Buffer, NoFile));
933 }
934 /**
935 Retrieve the size of a file.
936
937 if FileHandle is NULL then ASSERT()
938 if Size is NULL then ASSERT()
939
940 This function extracts the file size info from the FileHandle's EFI_FILE_INFO
941 data.
942
943 @param FileHandle file handle from which size is retrieved
944 @param Size pointer to size
945
946 @retval EFI_SUCCESS operation was completed sucessfully
947 @retval EFI_DEVICE_ERROR cannot access the file
948 **/
949 EFI_STATUS
950 EFIAPI
951 ShellGetFileSize (
952 IN EFI_FILE_HANDLE FileHandle,
953 OUT UINT64 *Size
954 ) {
955 return (FileFunctionMap.GetFileSize(FileHandle, Size));
956 }
957 /**
958 Retrieves the status of the break execution flag
959
960 this function is useful to check whether the application is being asked to halt by the shell.
961
962 @retval TRUE the execution break is enabled
963 @retval FALSE the execution break is not enabled
964 **/
965 BOOLEAN
966 EFIAPI
967 ShellGetExecutionBreakFlag(
968 VOID
969 )
970 {
971 //
972 // Check for UEFI Shell 2.0 protocols
973 //
974 if (mEfiShellProtocol != NULL) {
975
976 //
977 // We are using UEFI Shell 2.0; see if the event has been triggered
978 //
979 if (gBS->CheckEvent(mEfiShellProtocol->ExecutionBreak) != EFI_SUCCESS) {
980 return (FALSE);
981 }
982 return (TRUE);
983 }
984
985 //
986 // using EFI Shell; call the function to check
987 //
988 ASSERT(mEfiShellEnvironment2 != NULL);
989 return (mEfiShellEnvironment2->GetExecutionBreak());
990 }
991 /**
992 return the value of an environment variable
993
994 this function gets the value of the environment variable set by the
995 ShellSetEnvironmentVariable function
996
997 @param EnvKey The key name of the environment variable.
998
999 @retval NULL the named environment variable does not exist.
1000 @return != NULL pointer to the value of the environment variable
1001 **/
1002 CONST CHAR16*
1003 EFIAPI
1004 ShellGetEnvironmentVariable (
1005 IN CONST CHAR16 *EnvKey
1006 )
1007 {
1008 //
1009 // Check for UEFI Shell 2.0 protocols
1010 //
1011 if (mEfiShellProtocol != NULL) {
1012 return (mEfiShellProtocol->GetEnv(EnvKey));
1013 }
1014
1015 //
1016 // ASSERT that we must have EFI shell
1017 //
1018 ASSERT(mEfiShellEnvironment2 != NULL);
1019
1020 //
1021 // using EFI Shell
1022 //
1023 return (mEfiShellEnvironment2->GetEnv((CHAR16*)EnvKey));
1024 }
1025 /**
1026 set the value of an environment variable
1027
1028 This function changes the current value of the specified environment variable. If the
1029 environment variable exists and the Value is an empty string, then the environment
1030 variable is deleted. If the environment variable exists and the Value is not an empty
1031 string, then the value of the environment variable is changed. If the environment
1032 variable does not exist and the Value is an empty string, there is no action. If the
1033 environment variable does not exist and the Value is a non-empty string, then the
1034 environment variable is created and assigned the specified value.
1035
1036 This is not supported pre-UEFI Shell 2.0.
1037
1038 @param EnvKey The key name of the environment variable.
1039 @param EnvVal The Value of the environment variable
1040 @param Volatile Indicates whether the variable is non-volatile (FALSE) or volatile (TRUE).
1041
1042 @retval EFI_SUCCESS the operation was completed sucessfully
1043 @retval EFI_UNSUPPORTED This operation is not allowed in pre UEFI 2.0 Shell environments
1044 **/
1045 EFI_STATUS
1046 EFIAPI
1047 ShellSetEnvironmentVariable (
1048 IN CONST CHAR16 *EnvKey,
1049 IN CONST CHAR16 *EnvVal,
1050 IN BOOLEAN Volatile
1051 )
1052 {
1053 //
1054 // Check for UEFI Shell 2.0 protocols
1055 //
1056 if (mEfiShellProtocol != NULL) {
1057 return (mEfiShellProtocol->SetEnv(EnvKey, EnvVal, Volatile));
1058 }
1059
1060 //
1061 // This feature does not exist under EFI shell
1062 //
1063 return (EFI_UNSUPPORTED);
1064 }
1065 /**
1066 cause the shell to parse and execute a command line.
1067
1068 This function creates a nested instance of the shell and executes the specified
1069 command (CommandLine) with the specified environment (Environment). Upon return,
1070 the status code returned by the specified command is placed in StatusCode.
1071 If Environment is NULL, then the current environment is used and all changes made
1072 by the commands executed will be reflected in the current environment. If the
1073 Environment is non-NULL, then the changes made will be discarded.
1074 The CommandLine is executed from the current working directory on the current
1075 device.
1076
1077 EnvironmentVariables and Status are only supported for UEFI Shell 2.0.
1078 Output is only supported for pre-UEFI Shell 2.0
1079
1080 @param ImageHandle Parent image that is starting the operation
1081 @param CommandLine pointer to NULL terminated command line.
1082 @param Output true to display debug output. false to hide it.
1083 @param EnvironmentVariables optional pointer to array of environment variables
1084 in the form "x=y". if NULL current set is used.
1085 @param Status the status of the run command line.
1086
1087 @retval EFI_SUCCESS the operation completed sucessfully. Status
1088 contains the status code returned.
1089 @retval EFI_INVALID_PARAMETER a parameter contains an invalid value
1090 @retval EFI_OUT_OF_RESOURCES out of resources
1091 @retval EFI_UNSUPPORTED the operation is not allowed.
1092 **/
1093 EFI_STATUS
1094 EFIAPI
1095 ShellExecute (
1096 IN EFI_HANDLE *ParentHandle,
1097 IN CHAR16 *CommandLine OPTIONAL,
1098 IN BOOLEAN Output OPTIONAL,
1099 IN CHAR16 **EnvironmentVariables OPTIONAL,
1100 OUT EFI_STATUS *Status OPTIONAL
1101 )
1102 {
1103 //
1104 // Check for UEFI Shell 2.0 protocols
1105 //
1106 if (mEfiShellProtocol != NULL) {
1107 //
1108 // Call UEFI Shell 2.0 version (not using Output parameter)
1109 //
1110 return (mEfiShellProtocol->Execute(ParentHandle,
1111 CommandLine,
1112 EnvironmentVariables,
1113 Status));
1114 }
1115 //
1116 // ASSERT that we must have EFI shell
1117 //
1118 ASSERT(mEfiShellEnvironment2 != NULL);
1119 //
1120 // Call EFI Shell version (not using EnvironmentVariables or Status parameters)
1121 // Due to oddity in the EFI shell we want to dereference the ParentHandle here
1122 //
1123 return (mEfiShellEnvironment2->Execute(*ParentHandle,
1124 CommandLine,
1125 Output));
1126 }
1127 /**
1128 Retreives the current directory path
1129
1130 If the DeviceName is NULL, it returns the current device's current directory
1131 name. If the DeviceName is not NULL, it returns the current directory name
1132 on specified drive.
1133
1134 @param DeviceName the name of the drive to get directory on
1135
1136 @retval NULL the directory does not exist
1137 @return != NULL the directory
1138 **/
1139 CONST CHAR16*
1140 EFIAPI
1141 ShellGetCurrentDir (
1142 IN CHAR16 *DeviceName OPTIONAL
1143 )
1144 {
1145 //
1146 // Check for UEFI Shell 2.0 protocols
1147 //
1148 if (mEfiShellProtocol != NULL) {
1149 return (mEfiShellProtocol->GetCurDir(DeviceName));
1150 }
1151 //
1152 // ASSERT that we must have EFI shell
1153 //
1154 ASSERT(mEfiShellEnvironment2 != NULL);
1155 return (mEfiShellEnvironment2->CurDir(DeviceName));
1156 }
1157 /**
1158 sets (enabled or disabled) the page break mode
1159
1160 when page break mode is enabled the screen will stop scrolling
1161 and wait for operator input before scrolling a subsequent screen.
1162
1163 @param CurrentState TRUE to enable and FALSE to disable
1164 **/
1165 VOID
1166 EFIAPI
1167 ShellSetPageBreakMode (
1168 IN BOOLEAN CurrentState
1169 )
1170 {
1171 //
1172 // check for enabling
1173 //
1174 if (CurrentState != 0x00) {
1175 //
1176 // check for UEFI Shell 2.0
1177 //
1178 if (mEfiShellProtocol != NULL) {
1179 //
1180 // Enable with UEFI 2.0 Shell
1181 //
1182 mEfiShellProtocol->EnablePageBreak();
1183 return;
1184 } else {
1185 //
1186 // ASSERT that must have EFI Shell
1187 //
1188 ASSERT(mEfiShellEnvironment2 != NULL);
1189 //
1190 // Enable with EFI Shell
1191 //
1192 mEfiShellEnvironment2->EnablePageBreak (DEFAULT_INIT_ROW, DEFAULT_AUTO_LF);
1193 return;
1194 }
1195 } else {
1196 //
1197 // check for UEFI Shell 2.0
1198 //
1199 if (mEfiShellProtocol != NULL) {
1200 //
1201 // Disable with UEFI 2.0 Shell
1202 //
1203 mEfiShellProtocol->DisablePageBreak();
1204 return;
1205 } else {
1206 //
1207 // ASSERT that must have EFI Shell
1208 //
1209 ASSERT(mEfiShellEnvironment2 != NULL);
1210 //
1211 // Disable with EFI Shell
1212 //
1213 mEfiShellEnvironment2->DisablePageBreak ();
1214 return;
1215 }
1216 }
1217 }
1218
1219 ///
1220 /// version of EFI_SHELL_FILE_INFO struct, except has no CONST pointers.
1221 /// This allows for the struct to be populated.
1222 ///
1223 typedef struct {
1224 LIST_ENTRY Link;
1225 EFI_STATUS Status;
1226 CHAR16 *FullName;
1227 CHAR16 *FileName;
1228 EFI_FILE_HANDLE Handle;
1229 EFI_FILE_INFO *Info;
1230 } EFI_SHELL_FILE_INFO_NO_CONST;
1231
1232 /**
1233 Converts a EFI shell list of structures to the coresponding UEFI Shell 2.0 type of list.
1234
1235 if OldStyleFileList is NULL then ASSERT()
1236
1237 this function will convert a SHELL_FILE_ARG based list into a callee allocated
1238 EFI_SHELL_FILE_INFO based list. it is up to the caller to free the memory via
1239 the ShellCloseFileMetaArg function.
1240
1241 @param[in] FileList the EFI shell list type
1242 @param[in,out] ListHead the list to add to
1243
1244 @retval the resultant head of the double linked new format list;
1245 **/
1246 LIST_ENTRY*
1247 EFIAPI
1248 InternalShellConvertFileListType (
1249 IN LIST_ENTRY *FileList,
1250 IN OUT LIST_ENTRY *ListHead
1251 )
1252 {
1253 SHELL_FILE_ARG *OldInfo;
1254 LIST_ENTRY *Link;
1255 EFI_SHELL_FILE_INFO_NO_CONST *NewInfo;
1256
1257 //
1258 // ASSERTs
1259 //
1260 ASSERT(FileList != NULL);
1261 ASSERT(ListHead != NULL);
1262
1263 //
1264 // enumerate through each member of the old list and copy
1265 //
1266 for (Link = FileList->ForwardLink; Link != FileList; Link = Link->ForwardLink) {
1267 OldInfo = CR (Link, SHELL_FILE_ARG, Link, SHELL_FILE_ARG_SIGNATURE);
1268
1269 //
1270 // make sure the old list was valid
1271 //
1272 ASSERT(OldInfo != NULL);
1273 ASSERT(OldInfo->Info != NULL);
1274 ASSERT(OldInfo->FullName != NULL);
1275 ASSERT(OldInfo->FileName != NULL);
1276
1277 //
1278 // allocate a new EFI_SHELL_FILE_INFO object
1279 //
1280 NewInfo = AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1281
1282 //
1283 // copy the simple items
1284 //
1285 NewInfo->Handle = OldInfo->Handle;
1286 NewInfo->Status = OldInfo->Status;
1287
1288 // old shell checks for 0 not NULL
1289 OldInfo->Handle = 0;
1290
1291 //
1292 // allocate new space to copy strings and structure
1293 //
1294 NewInfo->FullName = AllocateZeroPool(StrSize(OldInfo->FullName));
1295 NewInfo->FileName = AllocateZeroPool(StrSize(OldInfo->FileName));
1296 NewInfo->Info = AllocateZeroPool((UINTN)OldInfo->Info->Size);
1297
1298 //
1299 // make sure all the memory allocations were sucessful
1300 //
1301 ASSERT(NewInfo->FullName != NULL);
1302 ASSERT(NewInfo->FileName != NULL);
1303 ASSERT(NewInfo->Info != NULL);
1304
1305 //
1306 // Copt the strings and structure
1307 //
1308 StrCpy(NewInfo->FullName, OldInfo->FullName);
1309 StrCpy(NewInfo->FileName, OldInfo->FileName);
1310 gBS->CopyMem (NewInfo->Info, OldInfo->Info, (UINTN)OldInfo->Info->Size);
1311
1312 //
1313 // add that to the list
1314 //
1315 InsertTailList(ListHead, &NewInfo->Link);
1316 }
1317 return (ListHead);
1318 }
1319 /**
1320 Opens a group of files based on a path.
1321
1322 This function uses the Arg to open all the matching files. Each matched
1323 file has a SHELL_FILE_ARG structure to record the file information. These
1324 structures are placed on the list ListHead. Users can get the SHELL_FILE_ARG
1325 structures from ListHead to access each file. This function supports wildcards
1326 and will process '?' and '*' as such. the list must be freed with a call to
1327 ShellCloseFileMetaArg().
1328
1329 If you are NOT appending to an existing list *ListHead must be NULL. If
1330 *ListHead is NULL then it must be callee freed.
1331
1332 @param Arg pointer to path string
1333 @param OpenMode mode to open files with
1334 @param ListHead head of linked list of results
1335
1336 @retval EFI_SUCCESS the operation was sucessful and the list head
1337 contains the list of opened files
1338 #retval EFI_UNSUPPORTED a previous ShellOpenFileMetaArg must be closed first.
1339 *ListHead is set to NULL.
1340 @return != EFI_SUCCESS the operation failed
1341
1342 @sa InternalShellConvertFileListType
1343 **/
1344 EFI_STATUS
1345 EFIAPI
1346 ShellOpenFileMetaArg (
1347 IN CHAR16 *Arg,
1348 IN UINT64 OpenMode,
1349 IN OUT EFI_SHELL_FILE_INFO **ListHead
1350 )
1351 {
1352 EFI_STATUS Status;
1353 LIST_ENTRY mOldStyleFileList;
1354
1355 //
1356 // ASSERT that Arg and ListHead are not NULL
1357 //
1358 ASSERT(Arg != NULL);
1359 ASSERT(ListHead != NULL);
1360
1361 //
1362 // Check for UEFI Shell 2.0 protocols
1363 //
1364 if (mEfiShellProtocol != NULL) {
1365 if (*ListHead == NULL) {
1366 *ListHead = (EFI_SHELL_FILE_INFO*)AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1367 if (*ListHead == NULL) {
1368 return (EFI_OUT_OF_RESOURCES);
1369 }
1370 InitializeListHead(&((*ListHead)->Link));
1371 }
1372 Status = mEfiShellProtocol->OpenFileList(Arg,
1373 OpenMode,
1374 ListHead);
1375 if (EFI_ERROR(Status)) {
1376 mEfiShellProtocol->RemoveDupInFileList(ListHead);
1377 } else {
1378 Status = mEfiShellProtocol->RemoveDupInFileList(ListHead);
1379 }
1380 return (Status);
1381 }
1382
1383 //
1384 // ASSERT that we must have EFI shell
1385 //
1386 ASSERT(mEfiShellEnvironment2 != NULL);
1387
1388 //
1389 // make sure the list head is initialized
1390 //
1391 InitializeListHead(&mOldStyleFileList);
1392
1393 //
1394 // Get the EFI Shell list of files
1395 //
1396 Status = mEfiShellEnvironment2->FileMetaArg(Arg, &mOldStyleFileList);
1397 if (EFI_ERROR(Status)) {
1398 *ListHead = NULL;
1399 return (Status);
1400 }
1401
1402 if (*ListHead == NULL) {
1403 *ListHead = (EFI_SHELL_FILE_INFO *)AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1404 if (*ListHead == NULL) {
1405 return (EFI_OUT_OF_RESOURCES);
1406 }
1407 }
1408
1409 //
1410 // Convert that to equivalent of UEFI Shell 2.0 structure
1411 //
1412 InternalShellConvertFileListType(&mOldStyleFileList, &(*ListHead)->Link);
1413
1414 //
1415 // Free the EFI Shell version that was converted.
1416 //
1417 mEfiShellEnvironment2->FreeFileList(&mOldStyleFileList);
1418
1419 return (Status);
1420 }
1421 /**
1422 Free the linked list returned from ShellOpenFileMetaArg
1423
1424 if ListHead is NULL then ASSERT()
1425
1426 @param ListHead the pointer to free
1427
1428 @retval EFI_SUCCESS the operation was sucessful
1429 **/
1430 EFI_STATUS
1431 EFIAPI
1432 ShellCloseFileMetaArg (
1433 IN OUT EFI_SHELL_FILE_INFO **ListHead
1434 )
1435 {
1436 LIST_ENTRY *Node;
1437
1438 //
1439 // ASSERT that ListHead is not NULL
1440 //
1441 ASSERT(ListHead != NULL);
1442
1443 //
1444 // Check for UEFI Shell 2.0 protocols
1445 //
1446 if (mEfiShellProtocol != NULL) {
1447 return (mEfiShellProtocol->FreeFileList(ListHead));
1448 } else {
1449 //
1450 // Since this is EFI Shell version we need to free our internally made copy
1451 // of the list
1452 //
1453 for ( Node = GetFirstNode(&(*ListHead)->Link)
1454 ; IsListEmpty(&(*ListHead)->Link) == FALSE
1455 ; Node = GetFirstNode(&(*ListHead)->Link)) {
1456 RemoveEntryList(Node);
1457 ((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->Handle->Close(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->Handle);
1458 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->FullName);
1459 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->FileName);
1460 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->Info);
1461 FreePool((EFI_SHELL_FILE_INFO_NO_CONST*)Node);
1462 }
1463 return EFI_SUCCESS;
1464 }
1465 }
1466
1467 /**
1468 Find a file by searching the CWD and then the path.
1469
1470 If FileName is NULL then ASSERT.
1471
1472 If the return value is not NULL then the memory must be caller freed.
1473
1474 @param FileName Filename string.
1475
1476 @retval NULL the file was not found
1477 @return !NULL the full path to the file.
1478 **/
1479 CHAR16 *
1480 EFIAPI
1481 ShellFindFilePath (
1482 IN CONST CHAR16 *FileName
1483 )
1484 {
1485 CONST CHAR16 *Path;
1486 EFI_FILE_HANDLE Handle;
1487 EFI_STATUS Status;
1488 CHAR16 *RetVal;
1489 CHAR16 *TestPath;
1490 CONST CHAR16 *Walker;
1491 UINTN Size;
1492 CHAR16 *TempChar;
1493
1494 RetVal = NULL;
1495
1496 Path = ShellGetEnvironmentVariable(L"cwd");
1497 if (Path != NULL) {
1498 Size = StrSize(Path);
1499 Size += StrSize(FileName);
1500 TestPath = AllocateZeroPool(Size);
1501 StrCpy(TestPath, Path);
1502 StrCat(TestPath, FileName);
1503 Status = ShellOpenFileByName(TestPath, &Handle, EFI_FILE_MODE_READ, 0);
1504 if (!EFI_ERROR(Status)){
1505 RetVal = StrnCatGrow(&RetVal, NULL, TestPath, 0);
1506 ShellCloseFile(&Handle);
1507 FreePool(TestPath);
1508 return (RetVal);
1509 }
1510 FreePool(TestPath);
1511 }
1512 Path = ShellGetEnvironmentVariable(L"path");
1513 if (Path != NULL) {
1514 Size = StrSize(Path);
1515 Size += StrSize(FileName);
1516 TestPath = AllocateZeroPool(Size);
1517 Walker = (CHAR16*)Path;
1518 do {
1519 CopyMem(TestPath, Walker, StrSize(Walker));
1520 TempChar = StrStr(TestPath, L";");
1521 if (TempChar != NULL) {
1522 *TempChar = CHAR_NULL;
1523 }
1524 StrCat(TestPath, FileName);
1525 if (StrStr(Walker, L";") != NULL) {
1526 Walker = StrStr(Walker, L";") + 1;
1527 } else {
1528 Walker = NULL;
1529 }
1530 Status = ShellOpenFileByName(TestPath, &Handle, EFI_FILE_MODE_READ, 0);
1531 if (!EFI_ERROR(Status)){
1532 RetVal = StrnCatGrow(&RetVal, NULL, TestPath, 0);
1533 ShellCloseFile(&Handle);
1534 break;
1535 }
1536 } while (Walker != NULL && Walker[0] != CHAR_NULL);
1537 FreePool(TestPath);
1538 }
1539 return (RetVal);
1540 }
1541
1542 /**
1543 Find a file by searching the CWD and then the path with a variable set of file
1544 extensions. If the file is not found it will append each extension in the list
1545 in the order provided and return the first one that is successful.
1546
1547 If FileName is NULL, then ASSERT.
1548 If FileExtension is NULL, then behavior is identical to ShellFindFilePath.
1549
1550 If the return value is not NULL then the memory must be caller freed.
1551
1552 @param[in] FileName Filename string.
1553 @param[in] FileExtension Semi-colon delimeted list of possible extensions.
1554
1555 @retval NULL The file was not found.
1556 @retval !NULL The path to the file.
1557 **/
1558 CHAR16 *
1559 EFIAPI
1560 ShellFindFilePathEx (
1561 IN CONST CHAR16 *FileName,
1562 IN CONST CHAR16 *FileExtension
1563 )
1564 {
1565 CHAR16 *TestPath;
1566 CHAR16 *RetVal;
1567 CONST CHAR16 *ExtensionWalker;
1568 UINTN Size;
1569 CHAR16 *TempChar;
1570
1571 ASSERT(FileName != NULL);
1572 if (FileExtension == NULL) {
1573 return (ShellFindFilePath(FileName));
1574 }
1575 RetVal = ShellFindFilePath(FileName);
1576 if (RetVal != NULL) {
1577 return (RetVal);
1578 }
1579 Size = StrSize(FileName);
1580 Size += StrSize(FileExtension);
1581 TestPath = AllocateZeroPool(Size);
1582 for (ExtensionWalker = FileExtension ; ; ExtensionWalker = StrStr(ExtensionWalker, L";") + 1 ){
1583 StrCpy(TestPath, FileName);
1584 if (ExtensionWalker != NULL) {
1585 StrCat(TestPath, ExtensionWalker);
1586 }
1587 TempChar = StrStr(TestPath, L";");
1588 if (TempChar != NULL) {
1589 *TempChar = CHAR_NULL;
1590 }
1591 RetVal = ShellFindFilePath(TestPath);
1592 if (RetVal != NULL) {
1593 break;
1594 }
1595 //
1596 // Must be after first loop...
1597 //
1598 if (StrStr(ExtensionWalker, L";") == NULL) {
1599 break;
1600 }
1601 }
1602 FreePool(TestPath);
1603 return (RetVal);
1604 }
1605
1606 typedef struct {
1607 LIST_ENTRY Link;
1608 CHAR16 *Name;
1609 ParamType Type;
1610 CHAR16 *Value;
1611 UINTN OriginalPosition;
1612 } SHELL_PARAM_PACKAGE;
1613
1614 /**
1615 Checks the list of valid arguments and returns TRUE if the item was found. If the
1616 return value is TRUE then the type parameter is set also.
1617
1618 if CheckList is NULL then ASSERT();
1619 if Name is NULL then ASSERT();
1620 if Type is NULL then ASSERT();
1621
1622 @param Type pointer to type of parameter if it was found
1623 @param Name pointer to Name of parameter found
1624 @param CheckList List to check against
1625
1626 @retval TRUE the Parameter was found. Type is valid.
1627 @retval FALSE the Parameter was not found. Type is not valid.
1628 **/
1629 BOOLEAN
1630 EFIAPI
1631 InternalIsOnCheckList (
1632 IN CONST CHAR16 *Name,
1633 IN CONST SHELL_PARAM_ITEM *CheckList,
1634 OUT ParamType *Type
1635 ) {
1636 SHELL_PARAM_ITEM *TempListItem;
1637
1638 //
1639 // ASSERT that all 3 pointer parameters aren't NULL
1640 //
1641 ASSERT(CheckList != NULL);
1642 ASSERT(Type != NULL);
1643 ASSERT(Name != NULL);
1644
1645 //
1646 // question mark and page break mode are always supported
1647 //
1648 if ((StrCmp(Name, L"-?") == 0) ||
1649 (StrCmp(Name, L"-b") == 0)
1650 ) {
1651 return (TRUE);
1652 }
1653
1654 //
1655 // Enumerate through the list
1656 //
1657 for (TempListItem = (SHELL_PARAM_ITEM*)CheckList ; TempListItem->Name != NULL ; TempListItem++) {
1658 //
1659 // If the Type is TypeStart only check the first characters of the passed in param
1660 // If it matches set the type and return TRUE
1661 //
1662 if (TempListItem->Type == TypeStart && StrnCmp(Name, TempListItem->Name, StrLen(TempListItem->Name)) == 0) {
1663 *Type = TempListItem->Type;
1664 return (TRUE);
1665 } else if (StrCmp(Name, TempListItem->Name) == 0) {
1666 *Type = TempListItem->Type;
1667 return (TRUE);
1668 }
1669 }
1670
1671 return (FALSE);
1672 }
1673 /**
1674 Checks the string for indicators of "flag" status. this is a leading '/', '-', or '+'
1675
1676 @param Name pointer to Name of parameter found
1677
1678 @retval TRUE the Parameter is a flag.
1679 @retval FALSE the Parameter not a flag
1680 **/
1681 BOOLEAN
1682 EFIAPI
1683 InternalIsFlag (
1684 IN CONST CHAR16 *Name,
1685 IN BOOLEAN AlwaysAllowNumbers
1686 )
1687 {
1688 //
1689 // ASSERT that Name isn't NULL
1690 //
1691 ASSERT(Name != NULL);
1692
1693 //
1694 // If we accept numbers then dont return TRUE. (they will be values)
1695 //
1696 if (((Name[0] == L'-' || Name[0] == L'+') && ShellIsHexaDecimalDigitCharacter(Name[1])) && AlwaysAllowNumbers != FALSE) {
1697 return (FALSE);
1698 }
1699
1700 //
1701 // If the Name has a / or - as the first character return TRUE
1702 //
1703 if ((Name[0] == L'/') ||
1704 (Name[0] == L'-') ||
1705 (Name[0] == L'+')
1706 ) {
1707 return (TRUE);
1708 }
1709 return (FALSE);
1710 }
1711
1712 /**
1713 Checks the command line arguments passed against the list of valid ones.
1714
1715 If no initialization is required, then return RETURN_SUCCESS.
1716
1717 @param CheckList pointer to list of parameters to check
1718 @param CheckPackage pointer to pointer to list checked values
1719 @param ProblemParam optional pointer to pointer to unicode string for
1720 the paramater that caused failure. If used then the
1721 caller is responsible for freeing the memory.
1722 @param AutoPageBreak will automatically set PageBreakEnabled for "b" parameter
1723 @param Argc Count of parameters in Argv
1724 @param Argv pointer to array of parameters
1725
1726 @retval EFI_SUCCESS The operation completed sucessfully.
1727 @retval EFI_OUT_OF_RESOURCES A memory allocation failed
1728 @retval EFI_INVALID_PARAMETER A parameter was invalid
1729 @retval EFI_VOLUME_CORRUPTED the command line was corrupt. an argument was
1730 duplicated. the duplicated command line argument
1731 was returned in ProblemParam if provided.
1732 @retval EFI_NOT_FOUND a argument required a value that was missing.
1733 the invalid command line argument was returned in
1734 ProblemParam if provided.
1735 **/
1736 STATIC
1737 EFI_STATUS
1738 EFIAPI
1739 InternalCommandLineParse (
1740 IN CONST SHELL_PARAM_ITEM *CheckList,
1741 OUT LIST_ENTRY **CheckPackage,
1742 OUT CHAR16 **ProblemParam OPTIONAL,
1743 IN BOOLEAN AutoPageBreak,
1744 IN CONST CHAR16 **Argv,
1745 IN UINTN Argc,
1746 IN BOOLEAN AlwaysAllowNumbers
1747 ) {
1748 UINTN LoopCounter;
1749 ParamType CurrentItemType;
1750 SHELL_PARAM_PACKAGE *CurrentItemPackage;
1751 UINTN GetItemValue;
1752 UINTN ValueSize;
1753
1754 CurrentItemPackage = NULL;
1755 mTotalParameterCount = 0;
1756 GetItemValue = 0;
1757 ValueSize = 0;
1758
1759 //
1760 // If there is only 1 item we dont need to do anything
1761 //
1762 if (Argc <= 1) {
1763 *CheckPackage = NULL;
1764 return (EFI_SUCCESS);
1765 }
1766
1767 //
1768 // ASSERTs
1769 //
1770 ASSERT(CheckList != NULL);
1771 ASSERT(Argv != NULL);
1772
1773 //
1774 // initialize the linked list
1775 //
1776 *CheckPackage = (LIST_ENTRY*)AllocateZeroPool(sizeof(LIST_ENTRY));
1777 InitializeListHead(*CheckPackage);
1778
1779 //
1780 // loop through each of the arguments
1781 //
1782 for (LoopCounter = 0 ; LoopCounter < Argc ; ++LoopCounter) {
1783 if (Argv[LoopCounter] == NULL) {
1784 //
1785 // do nothing for NULL argv
1786 //
1787 } else if (InternalIsOnCheckList(Argv[LoopCounter], CheckList, &CurrentItemType) != FALSE) {
1788 //
1789 // We might have leftover if last parameter didnt have optional value
1790 //
1791 if (GetItemValue != 0) {
1792 GetItemValue = 0;
1793 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
1794 }
1795 //
1796 // this is a flag
1797 //
1798 CurrentItemPackage = AllocatePool(sizeof(SHELL_PARAM_PACKAGE));
1799 ASSERT(CurrentItemPackage != NULL);
1800 CurrentItemPackage->Name = AllocatePool(StrSize(Argv[LoopCounter]));
1801 ASSERT(CurrentItemPackage->Name != NULL);
1802 StrCpy(CurrentItemPackage->Name, Argv[LoopCounter]);
1803 CurrentItemPackage->Type = CurrentItemType;
1804 CurrentItemPackage->OriginalPosition = (UINTN)(-1);
1805 CurrentItemPackage->Value = NULL;
1806
1807 //
1808 // Does this flag require a value
1809 //
1810 switch (CurrentItemPackage->Type) {
1811 //
1812 // possibly trigger the next loop(s) to populate the value of this item
1813 //
1814 case TypeValue:
1815 GetItemValue = 1;
1816 ValueSize = 0;
1817 break;
1818 case TypeDoubleValue:
1819 GetItemValue = 2;
1820 ValueSize = 0;
1821 break;
1822 case TypeMaxValue:
1823 GetItemValue = (UINTN)(-1);
1824 ValueSize = 0;
1825 break;
1826 default:
1827 //
1828 // this item has no value expected; we are done
1829 //
1830 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
1831 ASSERT(GetItemValue == 0);
1832 break;
1833 }
1834 } else if (GetItemValue != 0 && InternalIsFlag(Argv[LoopCounter], AlwaysAllowNumbers) == FALSE) {
1835 ASSERT(CurrentItemPackage != NULL);
1836 //
1837 // get the item VALUE for a previous flag
1838 //
1839 CurrentItemPackage->Value = ReallocatePool(ValueSize, ValueSize + StrSize(Argv[LoopCounter]) + sizeof(CHAR16), CurrentItemPackage->Value);
1840 ASSERT(CurrentItemPackage->Value != NULL);
1841 if (ValueSize == 0) {
1842 StrCpy(CurrentItemPackage->Value, Argv[LoopCounter]);
1843 } else {
1844 StrCat(CurrentItemPackage->Value, L" ");
1845 StrCat(CurrentItemPackage->Value, Argv[LoopCounter]);
1846 }
1847 ValueSize += StrSize(Argv[LoopCounter]) + sizeof(CHAR16);
1848 GetItemValue--;
1849 if (GetItemValue == 0) {
1850 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
1851 }
1852 } else if (InternalIsFlag(Argv[LoopCounter], AlwaysAllowNumbers) == FALSE) {
1853 //
1854 // add this one as a non-flag
1855 //
1856 CurrentItemPackage = AllocatePool(sizeof(SHELL_PARAM_PACKAGE));
1857 ASSERT(CurrentItemPackage != NULL);
1858 CurrentItemPackage->Name = NULL;
1859 CurrentItemPackage->Type = TypePosition;
1860 CurrentItemPackage->Value = AllocatePool(StrSize(Argv[LoopCounter]));
1861 ASSERT(CurrentItemPackage->Value != NULL);
1862 StrCpy(CurrentItemPackage->Value, Argv[LoopCounter]);
1863 CurrentItemPackage->OriginalPosition = mTotalParameterCount++;
1864 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
1865 } else if (ProblemParam) {
1866 //
1867 // this was a non-recognised flag... error!
1868 //
1869 *ProblemParam = AllocatePool(StrSize(Argv[LoopCounter]));
1870 ASSERT(*ProblemParam != NULL);
1871 StrCpy(*ProblemParam, Argv[LoopCounter]);
1872 ShellCommandLineFreeVarList(*CheckPackage);
1873 *CheckPackage = NULL;
1874 return (EFI_VOLUME_CORRUPTED);
1875 } else {
1876 ShellCommandLineFreeVarList(*CheckPackage);
1877 *CheckPackage = NULL;
1878 return (EFI_VOLUME_CORRUPTED);
1879 }
1880 }
1881 if (GetItemValue != 0) {
1882 GetItemValue = 0;
1883 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
1884 }
1885 //
1886 // support for AutoPageBreak
1887 //
1888 if (AutoPageBreak && ShellCommandLineGetFlag(*CheckPackage, L"-b")) {
1889 ShellSetPageBreakMode(TRUE);
1890 }
1891 return (EFI_SUCCESS);
1892 }
1893
1894 /**
1895 Checks the command line arguments passed against the list of valid ones.
1896 Optionally removes NULL values first.
1897
1898 If no initialization is required, then return RETURN_SUCCESS.
1899
1900 @param CheckList pointer to list of parameters to check
1901 @param CheckPackage pointer to pointer to list checked values
1902 @param ProblemParam optional pointer to pointer to unicode string for
1903 the paramater that caused failure.
1904 @param AutoPageBreak will automatically set PageBreakEnabled for "b" parameter
1905
1906 @retval EFI_SUCCESS The operation completed sucessfully.
1907 @retval EFI_OUT_OF_RESOURCES A memory allocation failed
1908 @retval EFI_INVALID_PARAMETER A parameter was invalid
1909 @retval EFI_VOLUME_CORRUPTED the command line was corrupt. an argument was
1910 duplicated. the duplicated command line argument
1911 was returned in ProblemParam if provided.
1912 @retval EFI_DEVICE_ERROR the commands contained 2 opposing arguments. one
1913 of the command line arguments was returned in
1914 ProblemParam if provided.
1915 @retval EFI_NOT_FOUND a argument required a value that was missing.
1916 the invalid command line argument was returned in
1917 ProblemParam if provided.
1918 **/
1919 EFI_STATUS
1920 EFIAPI
1921 ShellCommandLineParseEx (
1922 IN CONST SHELL_PARAM_ITEM *CheckList,
1923 OUT LIST_ENTRY **CheckPackage,
1924 OUT CHAR16 **ProblemParam OPTIONAL,
1925 IN BOOLEAN AutoPageBreak,
1926 IN BOOLEAN AlwaysAllowNumbers
1927 ) {
1928 //
1929 // ASSERT that CheckList and CheckPackage aren't NULL
1930 //
1931 ASSERT(CheckList != NULL);
1932 ASSERT(CheckPackage != NULL);
1933
1934 //
1935 // Check for UEFI Shell 2.0 protocols
1936 //
1937 if (mEfiShellParametersProtocol != NULL) {
1938 return (InternalCommandLineParse(CheckList,
1939 CheckPackage,
1940 ProblemParam,
1941 AutoPageBreak,
1942 (CONST CHAR16**) mEfiShellParametersProtocol->Argv,
1943 mEfiShellParametersProtocol->Argc,
1944 AlwaysAllowNumbers));
1945 }
1946
1947 //
1948 // ASSERT That EFI Shell is not required
1949 //
1950 ASSERT (mEfiShellInterface != NULL);
1951 return (InternalCommandLineParse(CheckList,
1952 CheckPackage,
1953 ProblemParam,
1954 AutoPageBreak,
1955 (CONST CHAR16**) mEfiShellInterface->Argv,
1956 mEfiShellInterface->Argc,
1957 AlwaysAllowNumbers));
1958 }
1959
1960 /**
1961 Frees shell variable list that was returned from ShellCommandLineParse.
1962
1963 This function will free all the memory that was used for the CheckPackage
1964 list of postprocessed shell arguments.
1965
1966 this function has no return value.
1967
1968 if CheckPackage is NULL, then return
1969
1970 @param CheckPackage the list to de-allocate
1971 **/
1972 VOID
1973 EFIAPI
1974 ShellCommandLineFreeVarList (
1975 IN LIST_ENTRY *CheckPackage
1976 ) {
1977 LIST_ENTRY *Node;
1978
1979 //
1980 // check for CheckPackage == NULL
1981 //
1982 if (CheckPackage == NULL) {
1983 return;
1984 }
1985
1986 //
1987 // for each node in the list
1988 //
1989 for ( Node = GetFirstNode(CheckPackage)
1990 ; IsListEmpty(CheckPackage) == FALSE
1991 ; Node = GetFirstNode(CheckPackage)
1992 ){
1993 //
1994 // Remove it from the list
1995 //
1996 RemoveEntryList(Node);
1997
1998 //
1999 // if it has a name free the name
2000 //
2001 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
2002 FreePool(((SHELL_PARAM_PACKAGE*)Node)->Name);
2003 }
2004
2005 //
2006 // if it has a value free the value
2007 //
2008 if (((SHELL_PARAM_PACKAGE*)Node)->Value != NULL) {
2009 FreePool(((SHELL_PARAM_PACKAGE*)Node)->Value);
2010 }
2011
2012 //
2013 // free the node structure
2014 //
2015 FreePool((SHELL_PARAM_PACKAGE*)Node);
2016 }
2017 //
2018 // free the list head node
2019 //
2020 FreePool(CheckPackage);
2021 }
2022 /**
2023 Checks for presence of a flag parameter
2024
2025 flag arguments are in the form of "-<Key>" or "/<Key>", but do not have a value following the key
2026
2027 if CheckPackage is NULL then return FALSE.
2028 if KeyString is NULL then ASSERT()
2029
2030 @param CheckPackage The package of parsed command line arguments
2031 @param KeyString the Key of the command line argument to check for
2032
2033 @retval TRUE the flag is on the command line
2034 @retval FALSE the flag is not on the command line
2035 **/
2036 BOOLEAN
2037 EFIAPI
2038 ShellCommandLineGetFlag (
2039 IN CONST LIST_ENTRY *CheckPackage,
2040 IN CHAR16 *KeyString
2041 ) {
2042 LIST_ENTRY *Node;
2043
2044 //
2045 // ASSERT that both CheckPackage and KeyString aren't NULL
2046 //
2047 ASSERT(KeyString != NULL);
2048
2049 //
2050 // return FALSE for no package
2051 //
2052 if (CheckPackage == NULL) {
2053 return (FALSE);
2054 }
2055
2056 //
2057 // enumerate through the list of parametrs
2058 //
2059 for ( Node = GetFirstNode(CheckPackage)
2060 ; !IsNull (CheckPackage, Node)
2061 ; Node = GetNextNode(CheckPackage, Node)
2062 ){
2063 //
2064 // If the Name matches, return TRUE (and there may be NULL name)
2065 //
2066 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
2067 //
2068 // If Type is TypeStart then only compare the begining of the strings
2069 //
2070 if ( ((SHELL_PARAM_PACKAGE*)Node)->Type == TypeStart
2071 && StrnCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name, StrLen(KeyString)) == 0
2072 ){
2073 return (TRUE);
2074 } else if (StrCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2075 return (TRUE);
2076 }
2077 }
2078 }
2079 return (FALSE);
2080 }
2081 /**
2082 returns value from command line argument
2083
2084 value parameters are in the form of "-<Key> value" or "/<Key> value"
2085
2086 if CheckPackage is NULL, then return NULL;
2087
2088 @param CheckPackage The package of parsed command line arguments
2089 @param KeyString the Key of the command line argument to check for
2090
2091 @retval NULL the flag is not on the command line
2092 @return !=NULL pointer to unicode string of the value
2093 **/
2094 CONST CHAR16*
2095 EFIAPI
2096 ShellCommandLineGetValue (
2097 IN CONST LIST_ENTRY *CheckPackage,
2098 IN CHAR16 *KeyString
2099 ) {
2100 LIST_ENTRY *Node;
2101
2102 //
2103 // check for CheckPackage == NULL
2104 //
2105 if (CheckPackage == NULL) {
2106 return (NULL);
2107 }
2108
2109 //
2110 // enumerate through the list of parametrs
2111 //
2112 for ( Node = GetFirstNode(CheckPackage)
2113 ; !IsNull (CheckPackage, Node)
2114 ; Node = GetNextNode(CheckPackage, Node)
2115 ){
2116 //
2117 // If the Name matches, return the value (name can be NULL)
2118 //
2119 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
2120 //
2121 // If Type is TypeStart then only compare the begining of the strings
2122 //
2123 if ( ((SHELL_PARAM_PACKAGE*)Node)->Type == TypeStart
2124 && StrnCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name, StrLen(KeyString)) == 0
2125 ){
2126 //
2127 // return the string part after the flag
2128 //
2129 return (((SHELL_PARAM_PACKAGE*)Node)->Name + StrLen(KeyString));
2130 } else if (StrCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2131 //
2132 // return the value
2133 //
2134 return (((SHELL_PARAM_PACKAGE*)Node)->Value);
2135 }
2136 }
2137 }
2138 return (NULL);
2139 }
2140 /**
2141 returns raw value from command line argument
2142
2143 raw value parameters are in the form of "value" in a specific position in the list
2144
2145 if CheckPackage is NULL, then return NULL;
2146
2147 @param CheckPackage The package of parsed command line arguments
2148 @param Position the position of the value
2149
2150 @retval NULL the flag is not on the command line
2151 @return !=NULL pointer to unicode string of the value
2152 **/
2153 CONST CHAR16*
2154 EFIAPI
2155 ShellCommandLineGetRawValue (
2156 IN CONST LIST_ENTRY *CheckPackage,
2157 IN UINT32 Position
2158 ) {
2159 LIST_ENTRY *Node;
2160
2161 //
2162 // check for CheckPackage == NULL
2163 //
2164 if (CheckPackage == NULL) {
2165 return (NULL);
2166 }
2167
2168 //
2169 // enumerate through the list of parametrs
2170 //
2171 for ( Node = GetFirstNode(CheckPackage)
2172 ; !IsNull (CheckPackage, Node)
2173 ; Node = GetNextNode(CheckPackage, Node)
2174 ){
2175 //
2176 // If the position matches, return the value
2177 //
2178 if (((SHELL_PARAM_PACKAGE*)Node)->OriginalPosition == Position) {
2179 return (((SHELL_PARAM_PACKAGE*)Node)->Value);
2180 }
2181 }
2182 return (NULL);
2183 }
2184
2185 /**
2186 returns the number of command line value parameters that were parsed.
2187
2188 this will not include flags.
2189
2190 @retval (UINTN)-1 No parsing has ocurred
2191 @return other The number of value parameters found
2192 **/
2193 UINTN
2194 EFIAPI
2195 ShellCommandLineGetCount(
2196 VOID
2197 )
2198 {
2199 return (mTotalParameterCount);
2200 }
2201
2202 /**
2203 Determins if a parameter is duplicated.
2204
2205 If Param is not NULL then it will point to a callee allocated string buffer
2206 with the parameter value if a duplicate is found.
2207
2208 If CheckPackage is NULL, then ASSERT.
2209
2210 @param[in] CheckPackage The package of parsed command line arguments.
2211 @param[out] Param Upon finding one, a pointer to the duplicated parameter.
2212
2213 @retval EFI_SUCCESS No parameters were duplicated.
2214 @retval EFI_DEVICE_ERROR A duplicate was found.
2215 **/
2216 EFI_STATUS
2217 EFIAPI
2218 ShellCommandLineCheckDuplicate (
2219 IN CONST LIST_ENTRY *CheckPackage,
2220 OUT CHAR16 **Param
2221 )
2222 {
2223 LIST_ENTRY *Node1;
2224 LIST_ENTRY *Node2;
2225
2226 ASSERT(CheckPackage != NULL);
2227
2228 for ( Node1 = GetFirstNode(CheckPackage)
2229 ; !IsNull (CheckPackage, Node1)
2230 ; Node1 = GetNextNode(CheckPackage, Node1)
2231 ){
2232 for ( Node2 = GetNextNode(CheckPackage, Node1)
2233 ; !IsNull (CheckPackage, Node2)
2234 ; Node2 = GetNextNode(CheckPackage, Node2)
2235 ){
2236 if (StrCmp(((SHELL_PARAM_PACKAGE*)Node1)->Name, ((SHELL_PARAM_PACKAGE*)Node2)->Name) == 0) {
2237 if (Param != NULL) {
2238 *Param = NULL;
2239 *Param = StrnCatGrow(Param, NULL, ((SHELL_PARAM_PACKAGE*)Node1)->Name, 0);
2240 }
2241 return (EFI_DEVICE_ERROR);
2242 }
2243 }
2244 }
2245 return (EFI_SUCCESS);
2246 }
2247
2248 /**
2249 This is a find and replace function. Upon successful return the NewString is a copy of
2250 SourceString with each instance of FindTarget replaced with ReplaceWith.
2251
2252 If SourceString and NewString overlap the behavior is undefined.
2253
2254 If the string would grow bigger than NewSize it will halt and return error.
2255
2256 @param[in] SourceString String with source buffer
2257 @param[in,out] NewString String with resultant buffer
2258 @param[in] NewSize Size in bytes of NewString
2259 @param[in] FindTarget String to look for
2260 @param[in[ ReplaceWith String to replace FindTarget with
2261 @param[in] SkipPreCarrot If TRUE will skip a FindTarget that has a '^'
2262 immediately before it.
2263
2264 @retval EFI_INVALID_PARAMETER SourceString was NULL.
2265 @retval EFI_INVALID_PARAMETER NewString was NULL.
2266 @retval EFI_INVALID_PARAMETER FindTarget was NULL.
2267 @retval EFI_INVALID_PARAMETER ReplaceWith was NULL.
2268 @retval EFI_INVALID_PARAMETER FindTarget had length < 1.
2269 @retval EFI_INVALID_PARAMETER SourceString had length < 1.
2270 @retval EFI_BUFFER_TOO_SMALL NewSize was less than the minimum size to hold
2271 the new string (truncation occurred).
2272 @retval EFI_SUCCESS the string was sucessfully copied with replacement.
2273 **/
2274
2275 EFI_STATUS
2276 EFIAPI
2277 ShellCopySearchAndReplace2(
2278 IN CHAR16 CONST *SourceString,
2279 IN CHAR16 *NewString,
2280 IN UINTN NewSize,
2281 IN CONST CHAR16 *FindTarget,
2282 IN CONST CHAR16 *ReplaceWith,
2283 IN CONST BOOLEAN SkipPreCarrot
2284 )
2285 {
2286 UINTN Size;
2287 if ( (SourceString == NULL)
2288 || (NewString == NULL)
2289 || (FindTarget == NULL)
2290 || (ReplaceWith == NULL)
2291 || (StrLen(FindTarget) < 1)
2292 || (StrLen(SourceString) < 1)
2293 ){
2294 return (EFI_INVALID_PARAMETER);
2295 }
2296 NewString = SetMem16(NewString, NewSize, CHAR_NULL);
2297 while (*SourceString != CHAR_NULL) {
2298 //
2299 // if we find the FindTarget and either Skip == FALSE or Skip == TRUE and we
2300 // dont have a carrot do a replace...
2301 //
2302 if (StrnCmp(SourceString, FindTarget, StrLen(FindTarget)) == 0
2303 && ((SkipPreCarrot && *(SourceString-1) != L'^') || SkipPreCarrot == FALSE)
2304 ){
2305 SourceString += StrLen(FindTarget);
2306 Size = StrSize(NewString);
2307 if ((Size + (StrLen(ReplaceWith)*sizeof(CHAR16))) > NewSize) {
2308 return (EFI_BUFFER_TOO_SMALL);
2309 }
2310 StrCat(NewString, ReplaceWith);
2311 } else {
2312 Size = StrSize(NewString);
2313 if (Size + sizeof(CHAR16) > NewSize) {
2314 return (EFI_BUFFER_TOO_SMALL);
2315 }
2316 StrnCat(NewString, SourceString, 1);
2317 SourceString++;
2318 }
2319 }
2320 return (EFI_SUCCESS);
2321 }
2322
2323 /**
2324 Internal worker function to output a string.
2325
2326 This function will output a string to the correct StdOut.
2327
2328 @param[in] String The string to print out.
2329
2330 @retval EFI_SUCCESS The operation was sucessful.
2331 @retval !EFI_SUCCESS The operation failed.
2332 **/
2333 EFI_STATUS
2334 EFIAPI
2335 InternalPrintTo (
2336 IN CONST CHAR16 *String
2337 )
2338 {
2339 UINTN Size;
2340 Size = StrSize(String) - sizeof(CHAR16);
2341 if (mEfiShellParametersProtocol != NULL) {
2342 return (mEfiShellParametersProtocol->StdOut->Write(mEfiShellParametersProtocol->StdOut, &Size, (VOID*)String));
2343 }
2344 if (mEfiShellInterface != NULL) {
2345 //
2346 // Divide in half for old shell. Must be string length not size.
2347 //
2348 Size /= 2;
2349 return ( mEfiShellInterface->StdOut->Write(mEfiShellInterface->StdOut, &Size, (VOID*)String));
2350 }
2351 ASSERT(FALSE);
2352 return (EFI_UNSUPPORTED);
2353 }
2354
2355 /**
2356 Print at a specific location on the screen.
2357
2358 This function will move the cursor to a given screen location and print the specified string
2359
2360 If -1 is specified for either the Row or Col the current screen location for BOTH
2361 will be used.
2362
2363 if either Row or Col is out of range for the current console, then ASSERT
2364 if Format is NULL, then ASSERT
2365
2366 In addition to the standard %-based flags as supported by UefiLib Print() this supports
2367 the following additional flags:
2368 %N - Set output attribute to normal
2369 %H - Set output attribute to highlight
2370 %E - Set output attribute to error
2371 %B - Set output attribute to blue color
2372 %V - Set output attribute to green color
2373
2374 Note: The background color is controlled by the shell command cls.
2375
2376 @param[in] Row the row to print at
2377 @param[in] Col the column to print at
2378 @param[in] Format the format string
2379 @param[in] Marker the marker for the variable argument list
2380
2381 @return the number of characters printed to the screen
2382 **/
2383
2384 UINTN
2385 EFIAPI
2386 InternalShellPrintWorker(
2387 IN INT32 Col OPTIONAL,
2388 IN INT32 Row OPTIONAL,
2389 IN CONST CHAR16 *Format,
2390 VA_LIST Marker
2391 )
2392 {
2393 UINTN Return;
2394 EFI_STATUS Status;
2395 UINTN NormalAttribute;
2396 CHAR16 *ResumeLocation;
2397 CHAR16 *FormatWalker;
2398
2399 //
2400 // Back and forth each time fixing up 1 of our flags...
2401 //
2402 Status = ShellLibCopySearchAndReplace(Format, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%N", L"%%N");
2403 ASSERT_EFI_ERROR(Status);
2404 Status = ShellLibCopySearchAndReplace(mPostReplaceFormat, mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), L"%E", L"%%E");
2405 ASSERT_EFI_ERROR(Status);
2406 Status = ShellLibCopySearchAndReplace(mPostReplaceFormat2, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%H", L"%%H");
2407 ASSERT_EFI_ERROR(Status);
2408 Status = ShellLibCopySearchAndReplace(mPostReplaceFormat, mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), L"%B", L"%%B");
2409 ASSERT_EFI_ERROR(Status);
2410 Status = ShellLibCopySearchAndReplace(mPostReplaceFormat2, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%V", L"%%V");
2411 ASSERT_EFI_ERROR(Status);
2412
2413 //
2414 // Use the last buffer from replacing to print from...
2415 //
2416 Return = UnicodeVSPrint (mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), mPostReplaceFormat, Marker);
2417
2418 if (Col != -1 && Row != -1) {
2419 Status = gST->ConOut->SetCursorPosition(gST->ConOut, Col, Row);
2420 ASSERT_EFI_ERROR(Status);
2421 }
2422
2423 NormalAttribute = gST->ConOut->Mode->Attribute;
2424 FormatWalker = mPostReplaceFormat2;
2425 while (*FormatWalker != CHAR_NULL) {
2426 //
2427 // Find the next attribute change request
2428 //
2429 ResumeLocation = StrStr(FormatWalker, L"%");
2430 if (ResumeLocation != NULL) {
2431 *ResumeLocation = CHAR_NULL;
2432 }
2433 //
2434 // print the current FormatWalker string
2435 //
2436 Status = InternalPrintTo(FormatWalker);
2437 ASSERT_EFI_ERROR(Status);
2438 //
2439 // update the attribute
2440 //
2441 if (ResumeLocation != NULL) {
2442 switch (*(ResumeLocation+1)) {
2443 case (L'N'):
2444 gST->ConOut->SetAttribute(gST->ConOut, NormalAttribute);
2445 break;
2446 case (L'E'):
2447 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_YELLOW, ((NormalAttribute&(BIT4|BIT5|BIT6))>>4)));
2448 break;
2449 case (L'H'):
2450 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_WHITE, ((NormalAttribute&(BIT4|BIT5|BIT6))>>4)));
2451 break;
2452 case (L'B'):
2453 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_BLUE, ((NormalAttribute&(BIT4|BIT5|BIT6))>>4)));
2454 break;
2455 case (L'V'):
2456 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_GREEN, ((NormalAttribute&(BIT4|BIT5|BIT6))>>4)));
2457 break;
2458 default:
2459 //
2460 // Print a simple '%' symbol
2461 //
2462 Status = InternalPrintTo(L"%");
2463 ASSERT_EFI_ERROR(Status);
2464 ResumeLocation = ResumeLocation - 1;
2465 break;
2466 }
2467 } else {
2468 //
2469 // reset to normal now...
2470 //
2471 gST->ConOut->SetAttribute(gST->ConOut, NormalAttribute);
2472 break;
2473 }
2474
2475 //
2476 // update FormatWalker to Resume + 2 (skip the % and the indicator)
2477 //
2478 FormatWalker = ResumeLocation + 2;
2479 }
2480
2481 return (Return);
2482 }
2483
2484 /**
2485 Print at a specific location on the screen.
2486
2487 This function will move the cursor to a given screen location and print the specified string.
2488
2489 If -1 is specified for either the Row or Col the current screen location for BOTH
2490 will be used.
2491
2492 If either Row or Col is out of range for the current console, then ASSERT.
2493 If Format is NULL, then ASSERT.
2494
2495 In addition to the standard %-based flags as supported by UefiLib Print() this supports
2496 the following additional flags:
2497 %N - Set output attribute to normal
2498 %H - Set output attribute to highlight
2499 %E - Set output attribute to error
2500 %B - Set output attribute to blue color
2501 %V - Set output attribute to green color
2502
2503 Note: The background color is controlled by the shell command cls.
2504
2505 @param[in] Row the row to print at
2506 @param[in] Col the column to print at
2507 @param[in] Format the format string
2508
2509 @return the number of characters printed to the screen
2510 **/
2511
2512 UINTN
2513 EFIAPI
2514 ShellPrintEx(
2515 IN INT32 Col OPTIONAL,
2516 IN INT32 Row OPTIONAL,
2517 IN CONST CHAR16 *Format,
2518 ...
2519 )
2520 {
2521 VA_LIST Marker;
2522 EFI_STATUS Status;
2523 VA_START (Marker, Format);
2524 Status = InternalShellPrintWorker(Col, Row, Format, Marker);
2525 VA_END(Marker);
2526 return(Status);
2527 }
2528
2529 /**
2530 Print at a specific location on the screen.
2531
2532 This function will move the cursor to a given screen location and print the specified string.
2533
2534 If -1 is specified for either the Row or Col the current screen location for BOTH
2535 will be used.
2536
2537 If either Row or Col is out of range for the current console, then ASSERT.
2538 If Format is NULL, then ASSERT.
2539
2540 In addition to the standard %-based flags as supported by UefiLib Print() this supports
2541 the following additional flags:
2542 %N - Set output attribute to normal.
2543 %H - Set output attribute to highlight.
2544 %E - Set output attribute to error.
2545 %B - Set output attribute to blue color.
2546 %V - Set output attribute to green color.
2547
2548 Note: The background color is controlled by the shell command cls.
2549
2550 @param[in] Row The row to print at.
2551 @param[in] Col The column to print at.
2552 @param[in] Language The language of the string to retrieve. If this parameter
2553 is NULL, then the current platform language is used.
2554 @param[in] HiiFormatStringId The format string Id for getting from Hii.
2555 @param[in] HiiFormatHandle The format string Handle for getting from Hii.
2556
2557 @return the number of characters printed to the screen.
2558 **/
2559 UINTN
2560 EFIAPI
2561 ShellPrintHiiEx(
2562 IN INT32 Col OPTIONAL,
2563 IN INT32 Row OPTIONAL,
2564 IN CONST CHAR8 *Language OPTIONAL,
2565 IN CONST EFI_STRING_ID HiiFormatStringId,
2566 IN CONST EFI_HANDLE HiiFormatHandle,
2567 ...
2568 )
2569 {
2570 VA_LIST Marker;
2571 CHAR16 *HiiFormatString;
2572 UINTN RetVal;
2573
2574 VA_START (Marker, HiiFormatHandle);
2575 HiiFormatString = HiiGetString(HiiFormatHandle, HiiFormatStringId, Language);
2576 ASSERT(HiiFormatString != NULL);
2577
2578 RetVal = InternalShellPrintWorker(Col, Row, HiiFormatString, Marker);
2579
2580 FreePool(HiiFormatString);
2581 VA_END(Marker);
2582
2583 return (RetVal);
2584 }
2585
2586 /**
2587 Function to determine if a given filename represents a file or a directory.
2588
2589 @param[in] DirName Path to directory to test.
2590
2591 @retval EFI_SUCCESS The Path represents a directory
2592 @retval EFI_NOT_FOUND The Path does not represent a directory
2593 @return other The path failed to open
2594 **/
2595 EFI_STATUS
2596 EFIAPI
2597 ShellIsDirectory(
2598 IN CONST CHAR16 *DirName
2599 )
2600 {
2601 EFI_STATUS Status;
2602 EFI_FILE_HANDLE Handle;
2603
2604 ASSERT(DirName != NULL);
2605
2606 Handle = NULL;
2607
2608 Status = ShellOpenFileByName(DirName, &Handle, EFI_FILE_MODE_READ, 0);
2609 if (EFI_ERROR(Status)) {
2610 return (Status);
2611 }
2612
2613 if (FileHandleIsDirectory(Handle) == EFI_SUCCESS) {
2614 ShellCloseFile(&Handle);
2615 return (EFI_SUCCESS);
2616 }
2617 ShellCloseFile(&Handle);
2618 return (EFI_NOT_FOUND);
2619 }
2620
2621 /**
2622 Function to determine if a given filename represents a file.
2623
2624 @param[in] Name Path to file to test.
2625
2626 @retval EFI_SUCCESS The Path represents a file.
2627 @retval EFI_NOT_FOUND The Path does not represent a file.
2628 @retval other The path failed to open.
2629 **/
2630 EFI_STATUS
2631 EFIAPI
2632 ShellIsFile(
2633 IN CONST CHAR16 *Name
2634 )
2635 {
2636 EFI_STATUS Status;
2637 EFI_FILE_HANDLE Handle;
2638
2639 ASSERT(Name != NULL);
2640
2641 Handle = NULL;
2642
2643 Status = ShellOpenFileByName(Name, &Handle, EFI_FILE_MODE_READ, 0);
2644 if (EFI_ERROR(Status)) {
2645 return (Status);
2646 }
2647
2648 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
2649 ShellCloseFile(&Handle);
2650 return (EFI_SUCCESS);
2651 }
2652 ShellCloseFile(&Handle);
2653 return (EFI_NOT_FOUND);
2654 }
2655
2656 /**
2657 Function to determine if a given filename represents a file.
2658
2659 This will search the CWD and then the Path.
2660
2661 If Name is NULL, then ASSERT.
2662
2663 @param[in] Name Path to file to test.
2664
2665 @retval EFI_SUCCESS The Path represents a file.
2666 @retval EFI_NOT_FOUND The Path does not represent a file.
2667 @retval other The path failed to open.
2668 **/
2669 EFI_STATUS
2670 EFIAPI
2671 ShellIsFileInPath(
2672 IN CONST CHAR16 *Name
2673 ) {
2674 CHAR16 *NewName;
2675 EFI_STATUS Status;
2676
2677 if (!EFI_ERROR(ShellIsFile(Name))) {
2678 return (TRUE);
2679 }
2680
2681 NewName = ShellFindFilePath(Name);
2682 if (NewName == NULL) {
2683 return (EFI_NOT_FOUND);
2684 }
2685 Status = ShellIsFile(NewName);
2686 FreePool(NewName);
2687 return (Status);
2688 }
2689 /**
2690 Function to determine whether a string is decimal or hex representation of a number
2691 and return the number converted from the string.
2692
2693 @param[in] String String representation of a number
2694
2695 @retval all the number
2696 **/
2697 UINTN
2698 EFIAPI
2699 ShellStrToUintn(
2700 IN CONST CHAR16 *String
2701 )
2702 {
2703 CONST CHAR16 *Walker;
2704 for (Walker = String; Walker != NULL && *Walker != CHAR_NULL && *Walker == L' '; Walker++);
2705 if (Walker == NULL || *Walker == CHAR_NULL) {
2706 ASSERT(FALSE);
2707 return ((UINTN)(-1));
2708 } else {
2709 if (StrnCmp(Walker, L"0x", 2) == 0 || StrnCmp(Walker, L"0X", 2) == 0){
2710 return (StrHexToUintn(Walker));
2711 }
2712 return (StrDecimalToUintn(Walker));
2713 }
2714 }
2715
2716 /**
2717 Safely append with automatic string resizing given length of Destination and
2718 desired length of copy from Source.
2719
2720 append the first D characters of Source to the end of Destination, where D is
2721 the lesser of Count and the StrLen() of Source. If appending those D characters
2722 will fit within Destination (whose Size is given as CurrentSize) and
2723 still leave room for a NULL terminator, then those characters are appended,
2724 starting at the original terminating NULL of Destination, and a new terminating
2725 NULL is appended.
2726
2727 If appending D characters onto Destination will result in a overflow of the size
2728 given in CurrentSize the string will be grown such that the copy can be performed
2729 and CurrentSize will be updated to the new size.
2730
2731 If Source is NULL, there is nothing to append, just return the current buffer in
2732 Destination.
2733
2734 if Destination is NULL, then ASSERT()
2735 if Destination's current length (including NULL terminator) is already more then
2736 CurrentSize, then ASSERT()
2737
2738 @param[in,out] Destination The String to append onto
2739 @param[in,out] CurrentSize on call the number of bytes in Destination. On
2740 return possibly the new size (still in bytes). if NULL
2741 then allocate whatever is needed.
2742 @param[in] Source The String to append from
2743 @param[in] Count Maximum number of characters to append. if 0 then
2744 all are appended.
2745
2746 @return Destination return the resultant string.
2747 **/
2748 CHAR16*
2749 EFIAPI
2750 StrnCatGrow (
2751 IN OUT CHAR16 **Destination,
2752 IN OUT UINTN *CurrentSize,
2753 IN CONST CHAR16 *Source,
2754 IN UINTN Count
2755 )
2756 {
2757 UINTN DestinationStartSize;
2758 UINTN NewSize;
2759
2760 //
2761 // ASSERTs
2762 //
2763 ASSERT(Destination != NULL);
2764
2765 //
2766 // If there's nothing to do then just return Destination
2767 //
2768 if (Source == NULL) {
2769 return (*Destination);
2770 }
2771
2772 //
2773 // allow for un-initialized pointers, based on size being 0
2774 //
2775 if (CurrentSize != NULL && *CurrentSize == 0) {
2776 *Destination = NULL;
2777 }
2778
2779 //
2780 // allow for NULL pointers address as Destination
2781 //
2782 if (*Destination != NULL) {
2783 ASSERT(CurrentSize != 0);
2784 DestinationStartSize = StrSize(*Destination);
2785 ASSERT(DestinationStartSize <= *CurrentSize);
2786 } else {
2787 DestinationStartSize = 0;
2788 // ASSERT(*CurrentSize == 0);
2789 }
2790
2791 //
2792 // Append all of Source?
2793 //
2794 if (Count == 0) {
2795 Count = StrLen(Source);
2796 }
2797
2798 //
2799 // Test and grow if required
2800 //
2801 if (CurrentSize != NULL) {
2802 NewSize = *CurrentSize;
2803 while (NewSize < (DestinationStartSize + (Count*sizeof(CHAR16)))) {
2804 NewSize += 2 * Count * sizeof(CHAR16);
2805 }
2806 *Destination = ReallocatePool(*CurrentSize, NewSize, *Destination);
2807 *CurrentSize = NewSize;
2808 } else {
2809 *Destination = AllocateZeroPool((Count+1)*sizeof(CHAR16));
2810 }
2811
2812 //
2813 // Now use standard StrnCat on a big enough buffer
2814 //
2815 return StrnCat(*Destination, Source, Count);
2816 }