]> git.proxmox.com Git - mirror_edk2.git/blob - ShellPkg/Library/UefiShellLib/UefiShellLib.c
Refine comments and two code style.
[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 - 2011, 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 #include <ShellBase.h>
17 #include <Library/SortLib.h>
18
19 #define FIND_XXXXX_FILE_BUFFER_SIZE (SIZE_OF_EFI_FILE_INFO + MAX_FILE_NAME_LEN)
20
21 //
22 // globals...
23 //
24 SHELL_PARAM_ITEM EmptyParamList[] = {
25 {NULL, TypeMax}
26 };
27 SHELL_PARAM_ITEM SfoParamList[] = {
28 {L"-sfo", TypeFlag},
29 {NULL, TypeMax}
30 };
31 EFI_SHELL_ENVIRONMENT2 *mEfiShellEnvironment2;
32 EFI_SHELL_INTERFACE *mEfiShellInterface;
33 EFI_SHELL_PROTOCOL *gEfiShellProtocol;
34 EFI_SHELL_PARAMETERS_PROTOCOL *gEfiShellParametersProtocol;
35 EFI_HANDLE mEfiShellEnvironment2Handle;
36 FILE_HANDLE_FUNCTION_MAP FileFunctionMap;
37
38 /**
39 Check if a Unicode character is a hexadecimal character.
40
41 This internal function checks if a Unicode character is a
42 numeric character. The valid hexadecimal characters are
43 L'0' to L'9', L'a' to L'f', or L'A' to L'F'.
44
45 @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 **/
51 BOOLEAN
52 EFIAPI
53 ShellIsHexaDecimalDigitCharacter (
54 IN CHAR16 Char
55 )
56 {
57 return (BOOLEAN) ((Char >= L'0' && Char <= L'9') || (Char >= L'A' && Char <= L'F') || (Char >= L'a' && Char <= L'f'));
58 }
59
60 /**
61 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 **/
74 BOOLEAN
75 EFIAPI
76 ShellIsDecimalDigitCharacter (
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.
87 **/
88 EFI_STATUS
89 EFIAPI
90 ShellFindSE2 (
91 IN EFI_HANDLE ImageHandle
92 )
93 {
94 EFI_STATUS Status;
95 EFI_HANDLE *Buffer;
96 UINTN BufferSize;
97 UINTN HandleIndex;
98
99 BufferSize = 0;
100 Buffer = NULL;
101 Status = gBS->OpenProtocol(ImageHandle,
102 &gEfiShellEnvironment2Guid,
103 (VOID **)&mEfiShellEnvironment2,
104 ImageHandle,
105 NULL,
106 EFI_OPEN_PROTOCOL_GET_PROTOCOL
107 );
108 //
109 // look for the mEfiShellEnvironment2 protocol at a higher level
110 //
111 if (EFI_ERROR (Status) || !(CompareGuid (&mEfiShellEnvironment2->SESGuid, &gEfiShellEnvironment2ExtGuid))){
112 //
113 // figure out how big of a buffer we need.
114 //
115 Status = gBS->LocateHandle (ByProtocol,
116 &gEfiShellEnvironment2Guid,
117 NULL, // ignored for ByProtocol
118 &BufferSize,
119 Buffer
120 );
121 //
122 // maybe it's not there???
123 //
124 if (Status == EFI_BUFFER_TOO_SMALL) {
125 Buffer = (EFI_HANDLE*)AllocateZeroPool(BufferSize);
126 ASSERT(Buffer != NULL);
127 Status = gBS->LocateHandle (ByProtocol,
128 &gEfiShellEnvironment2Guid,
129 NULL, // ignored for ByProtocol
130 &BufferSize,
131 Buffer
132 );
133 }
134 if (!EFI_ERROR (Status) && Buffer != NULL) {
135 //
136 // now parse the list of returned handles
137 //
138 Status = EFI_NOT_FOUND;
139 for (HandleIndex = 0; HandleIndex < (BufferSize/sizeof(Buffer[0])); HandleIndex++) {
140 Status = gBS->OpenProtocol(Buffer[HandleIndex],
141 &gEfiShellEnvironment2Guid,
142 (VOID **)&mEfiShellEnvironment2,
143 ImageHandle,
144 NULL,
145 EFI_OPEN_PROTOCOL_GET_PROTOCOL
146 );
147 if (CompareGuid (&mEfiShellEnvironment2->SESGuid, &gEfiShellEnvironment2ExtGuid)) {
148 mEfiShellEnvironment2Handle = Buffer[HandleIndex];
149 Status = EFI_SUCCESS;
150 break;
151 }
152 }
153 }
154 }
155 if (Buffer != NULL) {
156 FreePool (Buffer);
157 }
158 return (Status);
159 }
160
161 /**
162 Function to do most of the work of the constructor. Allows for calling
163 multiple times without complete re-initialization.
164
165 @param[in] ImageHandle A copy of the ImageHandle.
166 @param[in] SystemTable A pointer to the SystemTable for the application.
167
168 @retval EFI_SUCCESS The operationw as successful.
169 **/
170 EFI_STATUS
171 EFIAPI
172 ShellLibConstructorWorker (
173 IN EFI_HANDLE ImageHandle,
174 IN EFI_SYSTEM_TABLE *SystemTable
175 )
176 {
177 EFI_STATUS Status;
178
179 //
180 // UEFI 2.0 shell interfaces (used preferentially)
181 //
182 Status = gBS->OpenProtocol(
183 ImageHandle,
184 &gEfiShellProtocolGuid,
185 (VOID **)&gEfiShellProtocol,
186 ImageHandle,
187 NULL,
188 EFI_OPEN_PROTOCOL_GET_PROTOCOL
189 );
190 if (EFI_ERROR(Status)) {
191 //
192 // Search for the shell protocol
193 //
194 Status = gBS->LocateProtocol(
195 &gEfiShellProtocolGuid,
196 NULL,
197 (VOID **)&gEfiShellProtocol
198 );
199 if (EFI_ERROR(Status)) {
200 gEfiShellProtocol = NULL;
201 }
202 }
203 Status = gBS->OpenProtocol(
204 ImageHandle,
205 &gEfiShellParametersProtocolGuid,
206 (VOID **)&gEfiShellParametersProtocol,
207 ImageHandle,
208 NULL,
209 EFI_OPEN_PROTOCOL_GET_PROTOCOL
210 );
211 if (EFI_ERROR(Status)) {
212 gEfiShellParametersProtocol = NULL;
213 }
214
215 if (gEfiShellParametersProtocol == NULL || gEfiShellProtocol == NULL) {
216 //
217 // Moved to seperate function due to complexity
218 //
219 Status = ShellFindSE2(ImageHandle);
220
221 if (EFI_ERROR(Status)) {
222 DEBUG((DEBUG_ERROR, "Status: 0x%08x\r\n", Status));
223 mEfiShellEnvironment2 = NULL;
224 }
225 Status = gBS->OpenProtocol(ImageHandle,
226 &gEfiShellInterfaceGuid,
227 (VOID **)&mEfiShellInterface,
228 ImageHandle,
229 NULL,
230 EFI_OPEN_PROTOCOL_GET_PROTOCOL
231 );
232 if (EFI_ERROR(Status)) {
233 mEfiShellInterface = NULL;
234 }
235 }
236
237 //
238 // only success getting 2 of either the old or new, but no 1/2 and 1/2
239 //
240 if ((mEfiShellEnvironment2 != NULL && mEfiShellInterface != NULL) ||
241 (gEfiShellProtocol != NULL && gEfiShellParametersProtocol != NULL) ) {
242 if (gEfiShellProtocol != NULL) {
243 FileFunctionMap.GetFileInfo = gEfiShellProtocol->GetFileInfo;
244 FileFunctionMap.SetFileInfo = gEfiShellProtocol->SetFileInfo;
245 FileFunctionMap.ReadFile = gEfiShellProtocol->ReadFile;
246 FileFunctionMap.WriteFile = gEfiShellProtocol->WriteFile;
247 FileFunctionMap.CloseFile = gEfiShellProtocol->CloseFile;
248 FileFunctionMap.DeleteFile = gEfiShellProtocol->DeleteFile;
249 FileFunctionMap.GetFilePosition = gEfiShellProtocol->GetFilePosition;
250 FileFunctionMap.SetFilePosition = gEfiShellProtocol->SetFilePosition;
251 FileFunctionMap.FlushFile = gEfiShellProtocol->FlushFile;
252 FileFunctionMap.GetFileSize = gEfiShellProtocol->GetFileSize;
253 } else {
254 FileFunctionMap.GetFileInfo = (EFI_SHELL_GET_FILE_INFO)FileHandleGetInfo;
255 FileFunctionMap.SetFileInfo = (EFI_SHELL_SET_FILE_INFO)FileHandleSetInfo;
256 FileFunctionMap.ReadFile = (EFI_SHELL_READ_FILE)FileHandleRead;
257 FileFunctionMap.WriteFile = (EFI_SHELL_WRITE_FILE)FileHandleWrite;
258 FileFunctionMap.CloseFile = (EFI_SHELL_CLOSE_FILE)FileHandleClose;
259 FileFunctionMap.DeleteFile = (EFI_SHELL_DELETE_FILE)FileHandleDelete;
260 FileFunctionMap.GetFilePosition = (EFI_SHELL_GET_FILE_POSITION)FileHandleGetPosition;
261 FileFunctionMap.SetFilePosition = (EFI_SHELL_SET_FILE_POSITION)FileHandleSetPosition;
262 FileFunctionMap.FlushFile = (EFI_SHELL_FLUSH_FILE)FileHandleFlush;
263 FileFunctionMap.GetFileSize = (EFI_SHELL_GET_FILE_SIZE)FileHandleGetSize;
264 }
265 return (EFI_SUCCESS);
266 }
267 return (EFI_NOT_FOUND);
268 }
269 /**
270 Constructor for the Shell library.
271
272 Initialize the library and determine if the underlying is a UEFI Shell 2.0 or an EFI shell.
273
274 @param ImageHandle the image handle of the process
275 @param SystemTable the EFI System Table pointer
276
277 @retval EFI_SUCCESS the initialization was complete sucessfully
278 @return others an error ocurred during initialization
279 **/
280 EFI_STATUS
281 EFIAPI
282 ShellLibConstructor (
283 IN EFI_HANDLE ImageHandle,
284 IN EFI_SYSTEM_TABLE *SystemTable
285 )
286 {
287 mEfiShellEnvironment2 = NULL;
288 gEfiShellProtocol = NULL;
289 gEfiShellParametersProtocol = NULL;
290 mEfiShellInterface = NULL;
291 mEfiShellEnvironment2Handle = NULL;
292
293 //
294 // verify that auto initialize is not set false
295 //
296 if (PcdGetBool(PcdShellLibAutoInitialize) == 0) {
297 return (EFI_SUCCESS);
298 }
299
300 return (ShellLibConstructorWorker(ImageHandle, SystemTable));
301 }
302
303 /**
304 Destructor for the library. free any resources.
305
306 @param[in] ImageHandle A copy of the ImageHandle.
307 @param[in] SystemTable A pointer to the SystemTable for the application.
308
309 @retval EFI_SUCCESS The operation was successful.
310 @return An error from the CloseProtocol function.
311 **/
312 EFI_STATUS
313 EFIAPI
314 ShellLibDestructor (
315 IN EFI_HANDLE ImageHandle,
316 IN EFI_SYSTEM_TABLE *SystemTable
317 )
318 {
319 if (mEfiShellEnvironment2 != NULL) {
320 gBS->CloseProtocol(mEfiShellEnvironment2Handle==NULL?ImageHandle:mEfiShellEnvironment2Handle,
321 &gEfiShellEnvironment2Guid,
322 ImageHandle,
323 NULL);
324 mEfiShellEnvironment2 = NULL;
325 }
326 if (mEfiShellInterface != NULL) {
327 gBS->CloseProtocol(ImageHandle,
328 &gEfiShellInterfaceGuid,
329 ImageHandle,
330 NULL);
331 mEfiShellInterface = NULL;
332 }
333 if (gEfiShellProtocol != NULL) {
334 gBS->CloseProtocol(ImageHandle,
335 &gEfiShellProtocolGuid,
336 ImageHandle,
337 NULL);
338 gEfiShellProtocol = NULL;
339 }
340 if (gEfiShellParametersProtocol != NULL) {
341 gBS->CloseProtocol(ImageHandle,
342 &gEfiShellParametersProtocolGuid,
343 ImageHandle,
344 NULL);
345 gEfiShellParametersProtocol = NULL;
346 }
347 mEfiShellEnvironment2Handle = NULL;
348
349 return (EFI_SUCCESS);
350 }
351
352 /**
353 This function causes the shell library to initialize itself. If the shell library
354 is already initialized it will de-initialize all the current protocol poitners and
355 re-populate them again.
356
357 When the library is used with PcdShellLibAutoInitialize set to true this function
358 will return EFI_SUCCESS and perform no actions.
359
360 This function is intended for internal access for shell commands only.
361
362 @retval EFI_SUCCESS the initialization was complete sucessfully
363
364 **/
365 EFI_STATUS
366 EFIAPI
367 ShellInitialize (
368 )
369 {
370 //
371 // if auto initialize is not false then skip
372 //
373 if (PcdGetBool(PcdShellLibAutoInitialize) != 0) {
374 return (EFI_SUCCESS);
375 }
376
377 //
378 // deinit the current stuff
379 //
380 ASSERT_EFI_ERROR(ShellLibDestructor(gImageHandle, gST));
381
382 //
383 // init the new stuff
384 //
385 return (ShellLibConstructorWorker(gImageHandle, gST));
386 }
387
388 /**
389 This function will retrieve the information about the file for the handle
390 specified and store it in allocated pool memory.
391
392 This function allocates a buffer to store the file's information. It is the
393 caller's responsibility to free the buffer
394
395 @param FileHandle The file handle of the file for which information is
396 being requested.
397
398 @retval NULL information could not be retrieved.
399
400 @return the information about the file
401 **/
402 EFI_FILE_INFO*
403 EFIAPI
404 ShellGetFileInfo (
405 IN SHELL_FILE_HANDLE FileHandle
406 )
407 {
408 return (FileFunctionMap.GetFileInfo(FileHandle));
409 }
410
411 /**
412 This function sets the information about the file for the opened handle
413 specified.
414
415 @param[in] FileHandle The file handle of the file for which information
416 is being set.
417
418 @param[in] FileInfo The information to set.
419
420 @retval EFI_SUCCESS The information was set.
421 @retval EFI_INVALID_PARAMETER A parameter was out of range or invalid.
422 @retval EFI_UNSUPPORTED The FileHandle does not support FileInfo.
423 @retval EFI_NO_MEDIA The device has no medium.
424 @retval EFI_DEVICE_ERROR The device reported an error.
425 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
426 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
427 @retval EFI_ACCESS_DENIED The file was opened read only.
428 @retval EFI_VOLUME_FULL The volume is full.
429 **/
430 EFI_STATUS
431 EFIAPI
432 ShellSetFileInfo (
433 IN SHELL_FILE_HANDLE FileHandle,
434 IN EFI_FILE_INFO *FileInfo
435 )
436 {
437 return (FileFunctionMap.SetFileInfo(FileHandle, FileInfo));
438 }
439
440 /**
441 This function will open a file or directory referenced by DevicePath.
442
443 This function opens a file with the open mode according to the file path. The
444 Attributes is valid only for EFI_FILE_MODE_CREATE.
445
446 @param FilePath on input the device path to the file. On output
447 the remaining device path.
448 @param DeviceHandle pointer to the system device handle.
449 @param FileHandle pointer to the file handle.
450 @param OpenMode the mode to open the file with.
451 @param Attributes the file's file attributes.
452
453 @retval EFI_SUCCESS The information was set.
454 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
455 @retval EFI_UNSUPPORTED Could not open the file path.
456 @retval EFI_NOT_FOUND The specified file could not be found on the
457 device or the file system could not be found on
458 the device.
459 @retval EFI_NO_MEDIA The device has no medium.
460 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
461 medium is no longer supported.
462 @retval EFI_DEVICE_ERROR The device reported an error.
463 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
464 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
465 @retval EFI_ACCESS_DENIED The file was opened read only.
466 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
467 file.
468 @retval EFI_VOLUME_FULL The volume is full.
469 **/
470 EFI_STATUS
471 EFIAPI
472 ShellOpenFileByDevicePath(
473 IN OUT EFI_DEVICE_PATH_PROTOCOL **FilePath,
474 OUT EFI_HANDLE *DeviceHandle,
475 OUT SHELL_FILE_HANDLE *FileHandle,
476 IN UINT64 OpenMode,
477 IN UINT64 Attributes
478 )
479 {
480 CHAR16 *FileName;
481 EFI_STATUS Status;
482 EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *EfiSimpleFileSystemProtocol;
483 EFI_FILE_PROTOCOL *Handle1;
484 EFI_FILE_PROTOCOL *Handle2;
485
486 if (FilePath == NULL || FileHandle == NULL || DeviceHandle == NULL) {
487 return (EFI_INVALID_PARAMETER);
488 }
489
490 //
491 // which shell interface should we use
492 //
493 if (gEfiShellProtocol != NULL) {
494 //
495 // use UEFI Shell 2.0 method.
496 //
497 FileName = gEfiShellProtocol->GetFilePathFromDevicePath(*FilePath);
498 if (FileName == NULL) {
499 return (EFI_INVALID_PARAMETER);
500 }
501 Status = ShellOpenFileByName(FileName, FileHandle, OpenMode, Attributes);
502 FreePool(FileName);
503 return (Status);
504 }
505
506
507 //
508 // use old shell method.
509 //
510 Status = gBS->LocateDevicePath (&gEfiSimpleFileSystemProtocolGuid,
511 FilePath,
512 DeviceHandle);
513 if (EFI_ERROR (Status)) {
514 return Status;
515 }
516 Status = gBS->OpenProtocol(*DeviceHandle,
517 &gEfiSimpleFileSystemProtocolGuid,
518 (VOID**)&EfiSimpleFileSystemProtocol,
519 gImageHandle,
520 NULL,
521 EFI_OPEN_PROTOCOL_GET_PROTOCOL);
522 if (EFI_ERROR (Status)) {
523 return Status;
524 }
525 Status = EfiSimpleFileSystemProtocol->OpenVolume(EfiSimpleFileSystemProtocol, &Handle1);
526 if (EFI_ERROR (Status)) {
527 FileHandle = NULL;
528 return Status;
529 }
530
531 //
532 // go down directories one node at a time.
533 //
534 while (!IsDevicePathEnd (*FilePath)) {
535 //
536 // For file system access each node should be a file path component
537 //
538 if (DevicePathType (*FilePath) != MEDIA_DEVICE_PATH ||
539 DevicePathSubType (*FilePath) != MEDIA_FILEPATH_DP
540 ) {
541 FileHandle = NULL;
542 return (EFI_INVALID_PARAMETER);
543 }
544 //
545 // Open this file path node
546 //
547 Handle2 = Handle1;
548 Handle1 = NULL;
549
550 //
551 // Try to test opening an existing file
552 //
553 Status = Handle2->Open (
554 Handle2,
555 &Handle1,
556 ((FILEPATH_DEVICE_PATH*)*FilePath)->PathName,
557 OpenMode &~EFI_FILE_MODE_CREATE,
558 0
559 );
560
561 //
562 // see if the error was that it needs to be created
563 //
564 if ((EFI_ERROR (Status)) && (OpenMode != (OpenMode &~EFI_FILE_MODE_CREATE))) {
565 Status = Handle2->Open (
566 Handle2,
567 &Handle1,
568 ((FILEPATH_DEVICE_PATH*)*FilePath)->PathName,
569 OpenMode,
570 Attributes
571 );
572 }
573 //
574 // Close the last node
575 //
576 Handle2->Close (Handle2);
577
578 if (EFI_ERROR(Status)) {
579 return (Status);
580 }
581
582 //
583 // Get the next node
584 //
585 *FilePath = NextDevicePathNode (*FilePath);
586 }
587
588 //
589 // This is a weak spot since if the undefined SHELL_FILE_HANDLE format changes this must change also!
590 //
591 *FileHandle = (VOID*)Handle1;
592 return (EFI_SUCCESS);
593 }
594
595 /**
596 This function will open a file or directory referenced by filename.
597
598 If return is EFI_SUCCESS, the Filehandle is the opened file's handle;
599 otherwise, the Filehandle is NULL. The Attributes is valid only for
600 EFI_FILE_MODE_CREATE.
601
602 if FileName is NULL then ASSERT()
603
604 @param FileName pointer to file name
605 @param FileHandle pointer to the file handle.
606 @param OpenMode the mode to open the file with.
607 @param Attributes the file's file attributes.
608
609 @retval EFI_SUCCESS The information was set.
610 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
611 @retval EFI_UNSUPPORTED Could not open the file path.
612 @retval EFI_NOT_FOUND The specified file could not be found on the
613 device or the file system could not be found
614 on the device.
615 @retval EFI_NO_MEDIA The device has no medium.
616 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
617 medium is no longer supported.
618 @retval EFI_DEVICE_ERROR The device reported an error.
619 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
620 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
621 @retval EFI_ACCESS_DENIED The file was opened read only.
622 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
623 file.
624 @retval EFI_VOLUME_FULL The volume is full.
625 **/
626 EFI_STATUS
627 EFIAPI
628 ShellOpenFileByName(
629 IN CONST CHAR16 *FileName,
630 OUT SHELL_FILE_HANDLE *FileHandle,
631 IN UINT64 OpenMode,
632 IN UINT64 Attributes
633 )
634 {
635 EFI_HANDLE DeviceHandle;
636 EFI_DEVICE_PATH_PROTOCOL *FilePath;
637 EFI_STATUS Status;
638 EFI_FILE_INFO *FileInfo;
639
640 //
641 // ASSERT if FileName is NULL
642 //
643 ASSERT(FileName != NULL);
644
645 if (FileName == NULL) {
646 return (EFI_INVALID_PARAMETER);
647 }
648
649 if (gEfiShellProtocol != NULL) {
650 if ((OpenMode & EFI_FILE_MODE_CREATE) == EFI_FILE_MODE_CREATE && (Attributes & EFI_FILE_DIRECTORY) == EFI_FILE_DIRECTORY) {
651 return ShellCreateDirectory(FileName, FileHandle);
652 }
653 //
654 // Use UEFI Shell 2.0 method
655 //
656 Status = gEfiShellProtocol->OpenFileByName(FileName,
657 FileHandle,
658 OpenMode);
659 if (StrCmp(FileName, L"NUL") != 0 && !EFI_ERROR(Status) && ((OpenMode & EFI_FILE_MODE_CREATE) != 0)){
660 FileInfo = FileFunctionMap.GetFileInfo(*FileHandle);
661 ASSERT(FileInfo != NULL);
662 FileInfo->Attribute = Attributes;
663 Status = FileFunctionMap.SetFileInfo(*FileHandle, FileInfo);
664 FreePool(FileInfo);
665 }
666 return (Status);
667 }
668 //
669 // Using EFI Shell version
670 // this means convert name to path and call that function
671 // since this will use EFI method again that will open it.
672 //
673 ASSERT(mEfiShellEnvironment2 != NULL);
674 FilePath = mEfiShellEnvironment2->NameToPath ((CHAR16*)FileName);
675 if (FilePath != NULL) {
676 return (ShellOpenFileByDevicePath(&FilePath,
677 &DeviceHandle,
678 FileHandle,
679 OpenMode,
680 Attributes));
681 }
682 return (EFI_DEVICE_ERROR);
683 }
684 /**
685 This function create a directory
686
687 If return is EFI_SUCCESS, the Filehandle is the opened directory's handle;
688 otherwise, the Filehandle is NULL. If the directory already existed, this
689 function opens the existing directory.
690
691 @param DirectoryName pointer to directory name
692 @param FileHandle pointer to the file handle.
693
694 @retval EFI_SUCCESS The information was set.
695 @retval EFI_INVALID_PARAMETER One of the parameters has an invalid value.
696 @retval EFI_UNSUPPORTED Could not open the file path.
697 @retval EFI_NOT_FOUND The specified file could not be found on the
698 device or the file system could not be found
699 on the device.
700 @retval EFI_NO_MEDIA The device has no medium.
701 @retval EFI_MEDIA_CHANGED The device has a different medium in it or the
702 medium is no longer supported.
703 @retval EFI_DEVICE_ERROR The device reported an error.
704 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
705 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
706 @retval EFI_ACCESS_DENIED The file was opened read only.
707 @retval EFI_OUT_OF_RESOURCES Not enough resources were available to open the
708 file.
709 @retval EFI_VOLUME_FULL The volume is full.
710 @sa ShellOpenFileByName
711 **/
712 EFI_STATUS
713 EFIAPI
714 ShellCreateDirectory(
715 IN CONST CHAR16 *DirectoryName,
716 OUT SHELL_FILE_HANDLE *FileHandle
717 )
718 {
719 if (gEfiShellProtocol != NULL) {
720 //
721 // Use UEFI Shell 2.0 method
722 //
723 return (gEfiShellProtocol->CreateFile(DirectoryName,
724 EFI_FILE_DIRECTORY,
725 FileHandle
726 ));
727 } else {
728 return (ShellOpenFileByName(DirectoryName,
729 FileHandle,
730 EFI_FILE_MODE_READ | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE,
731 EFI_FILE_DIRECTORY
732 ));
733 }
734 }
735
736 /**
737 This function reads information from an opened file.
738
739 If FileHandle is not a directory, the function reads the requested number of
740 bytes from the file at the file's current position and returns them in Buffer.
741 If the read goes beyond the end of the file, the read length is truncated to the
742 end of the file. The file's current position is increased by the number of bytes
743 returned. If FileHandle is a directory, the function reads the directory entry
744 at the file's current position and returns the entry in Buffer. If the Buffer
745 is not large enough to hold the current directory entry, then
746 EFI_BUFFER_TOO_SMALL is returned and the current file position is not updated.
747 BufferSize is set to be the size of the buffer needed to read the entry. On
748 success, the current position is updated to the next directory entry. If there
749 are no more directory entries, the read returns a zero-length buffer.
750 EFI_FILE_INFO is the structure returned as the directory entry.
751
752 @param FileHandle the opened file handle
753 @param BufferSize on input the size of buffer in bytes. on return
754 the number of bytes written.
755 @param Buffer the buffer to put read data into.
756
757 @retval EFI_SUCCESS Data was read.
758 @retval EFI_NO_MEDIA The device has no media.
759 @retval EFI_DEVICE_ERROR The device reported an error.
760 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
761 @retval EFI_BUFFER_TO_SMALL Buffer is too small. ReadSize contains required
762 size.
763
764 **/
765 EFI_STATUS
766 EFIAPI
767 ShellReadFile(
768 IN SHELL_FILE_HANDLE FileHandle,
769 IN OUT UINTN *BufferSize,
770 OUT VOID *Buffer
771 )
772 {
773 return (FileFunctionMap.ReadFile(FileHandle, BufferSize, Buffer));
774 }
775
776
777 /**
778 Write data to a file.
779
780 This function writes the specified number of bytes to the file at the current
781 file position. The current file position is advanced the actual number of bytes
782 written, which is returned in BufferSize. Partial writes only occur when there
783 has been a data error during the write attempt (such as "volume space full").
784 The file is automatically grown to hold the data if required. Direct writes to
785 opened directories are not supported.
786
787 @param FileHandle The opened file for writing
788 @param BufferSize on input the number of bytes in Buffer. On output
789 the number of bytes written.
790 @param Buffer the buffer containing data to write is stored.
791
792 @retval EFI_SUCCESS Data was written.
793 @retval EFI_UNSUPPORTED Writes to an open directory are not supported.
794 @retval EFI_NO_MEDIA The device has no media.
795 @retval EFI_DEVICE_ERROR The device reported an error.
796 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
797 @retval EFI_WRITE_PROTECTED The device is write-protected.
798 @retval EFI_ACCESS_DENIED The file was open for read only.
799 @retval EFI_VOLUME_FULL The volume is full.
800 **/
801 EFI_STATUS
802 EFIAPI
803 ShellWriteFile(
804 IN SHELL_FILE_HANDLE FileHandle,
805 IN OUT UINTN *BufferSize,
806 IN VOID *Buffer
807 )
808 {
809 return (FileFunctionMap.WriteFile(FileHandle, BufferSize, Buffer));
810 }
811
812 /**
813 Close an open file handle.
814
815 This function closes a specified file handle. All "dirty" cached file data is
816 flushed to the device, and the file is closed. In all cases the handle is
817 closed.
818
819 @param FileHandle the file handle to close.
820
821 @retval EFI_SUCCESS the file handle was closed sucessfully.
822 **/
823 EFI_STATUS
824 EFIAPI
825 ShellCloseFile (
826 IN SHELL_FILE_HANDLE *FileHandle
827 )
828 {
829 return (FileFunctionMap.CloseFile(*FileHandle));
830 }
831
832 /**
833 Delete a file and close the handle
834
835 This function closes and deletes a file. In all cases the file handle is closed.
836 If the file cannot be deleted, the warning code EFI_WARN_DELETE_FAILURE is
837 returned, but the handle is still closed.
838
839 @param FileHandle the file handle to delete
840
841 @retval EFI_SUCCESS the file was closed sucessfully
842 @retval EFI_WARN_DELETE_FAILURE the handle was closed, but the file was not
843 deleted
844 @retval INVALID_PARAMETER One of the parameters has an invalid value.
845 **/
846 EFI_STATUS
847 EFIAPI
848 ShellDeleteFile (
849 IN SHELL_FILE_HANDLE *FileHandle
850 )
851 {
852 return (FileFunctionMap.DeleteFile(*FileHandle));
853 }
854
855 /**
856 Set the current position in a file.
857
858 This function sets the current file position for the handle to the position
859 supplied. With the exception of seeking to position 0xFFFFFFFFFFFFFFFF, only
860 absolute positioning is supported, and seeking past the end of the file is
861 allowed (a subsequent write would grow the file). Seeking to position
862 0xFFFFFFFFFFFFFFFF causes the current position to be set to the end of the file.
863 If FileHandle is a directory, the only position that may be set is zero. This
864 has the effect of starting the read process of the directory entries over.
865
866 @param FileHandle The file handle on which the position is being set
867 @param Position Byte position from begining of file
868
869 @retval EFI_SUCCESS Operation completed sucessfully.
870 @retval EFI_UNSUPPORTED the seek request for non-zero is not valid on
871 directories.
872 @retval INVALID_PARAMETER One of the parameters has an invalid value.
873 **/
874 EFI_STATUS
875 EFIAPI
876 ShellSetFilePosition (
877 IN SHELL_FILE_HANDLE FileHandle,
878 IN UINT64 Position
879 )
880 {
881 return (FileFunctionMap.SetFilePosition(FileHandle, Position));
882 }
883
884 /**
885 Gets a file's current position
886
887 This function retrieves the current file position for the file handle. For
888 directories, the current file position has no meaning outside of the file
889 system driver and as such the operation is not supported. An error is returned
890 if FileHandle is a directory.
891
892 @param FileHandle The open file handle on which to get the position.
893 @param Position Byte position from begining of file.
894
895 @retval EFI_SUCCESS the operation completed sucessfully.
896 @retval INVALID_PARAMETER One of the parameters has an invalid value.
897 @retval EFI_UNSUPPORTED the request is not valid on directories.
898 **/
899 EFI_STATUS
900 EFIAPI
901 ShellGetFilePosition (
902 IN SHELL_FILE_HANDLE FileHandle,
903 OUT UINT64 *Position
904 )
905 {
906 return (FileFunctionMap.GetFilePosition(FileHandle, Position));
907 }
908 /**
909 Flushes data on a file
910
911 This function flushes all modified data associated with a file to a device.
912
913 @param FileHandle The file handle on which to flush data
914
915 @retval EFI_SUCCESS The data was flushed.
916 @retval EFI_NO_MEDIA The device has no media.
917 @retval EFI_DEVICE_ERROR The device reported an error.
918 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
919 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
920 @retval EFI_ACCESS_DENIED The file was opened for read only.
921 **/
922 EFI_STATUS
923 EFIAPI
924 ShellFlushFile (
925 IN SHELL_FILE_HANDLE FileHandle
926 )
927 {
928 return (FileFunctionMap.FlushFile(FileHandle));
929 }
930
931 /**
932 Retrieves the first file from a directory
933
934 This function opens a directory and gets the first file's info in the
935 directory. Caller can use ShellFindNextFile() to get other files. When
936 complete the caller is responsible for calling FreePool() on Buffer.
937
938 @param DirHandle The file handle of the directory to search
939 @param Buffer Pointer to buffer for file's information
940
941 @retval EFI_SUCCESS Found the first file.
942 @retval EFI_NOT_FOUND Cannot find the directory.
943 @retval EFI_NO_MEDIA The device has no media.
944 @retval EFI_DEVICE_ERROR The device reported an error.
945 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
946 @return Others status of ShellGetFileInfo, ShellSetFilePosition,
947 or ShellReadFile
948 **/
949 EFI_STATUS
950 EFIAPI
951 ShellFindFirstFile (
952 IN SHELL_FILE_HANDLE DirHandle,
953 OUT EFI_FILE_INFO **Buffer
954 )
955 {
956 //
957 // pass to file handle lib
958 //
959 return (FileHandleFindFirstFile(DirHandle, Buffer));
960 }
961 /**
962 Retrieves the next file in a directory.
963
964 To use this function, caller must call the ShellFindFirstFile() to get the
965 first file, and then use this function get other files. This function can be
966 called for several times to get each file's information in the directory. If
967 the call of ShellFindNextFile() got the last file in the directory, the next
968 call of this function has no file to get. *NoFile will be set to TRUE and the
969 Buffer memory will be automatically freed.
970
971 @param DirHandle the file handle of the directory
972 @param Buffer pointer to buffer for file's information
973 @param NoFile pointer to boolean when last file is found
974
975 @retval EFI_SUCCESS Found the next file, or reached last file
976 @retval EFI_NO_MEDIA The device has no media.
977 @retval EFI_DEVICE_ERROR The device reported an error.
978 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
979 **/
980 EFI_STATUS
981 EFIAPI
982 ShellFindNextFile(
983 IN SHELL_FILE_HANDLE DirHandle,
984 OUT EFI_FILE_INFO *Buffer,
985 OUT BOOLEAN *NoFile
986 )
987 {
988 //
989 // pass to file handle lib
990 //
991 return (FileHandleFindNextFile(DirHandle, Buffer, NoFile));
992 }
993 /**
994 Retrieve the size of a file.
995
996 if FileHandle is NULL then ASSERT()
997 if Size is NULL then ASSERT()
998
999 This function extracts the file size info from the FileHandle's EFI_FILE_INFO
1000 data.
1001
1002 @param FileHandle file handle from which size is retrieved
1003 @param Size pointer to size
1004
1005 @retval EFI_SUCCESS operation was completed sucessfully
1006 @retval EFI_DEVICE_ERROR cannot access the file
1007 **/
1008 EFI_STATUS
1009 EFIAPI
1010 ShellGetFileSize (
1011 IN SHELL_FILE_HANDLE FileHandle,
1012 OUT UINT64 *Size
1013 )
1014 {
1015 return (FileFunctionMap.GetFileSize(FileHandle, Size));
1016 }
1017 /**
1018 Retrieves the status of the break execution flag
1019
1020 this function is useful to check whether the application is being asked to halt by the shell.
1021
1022 @retval TRUE the execution break is enabled
1023 @retval FALSE the execution break is not enabled
1024 **/
1025 BOOLEAN
1026 EFIAPI
1027 ShellGetExecutionBreakFlag(
1028 VOID
1029 )
1030 {
1031 //
1032 // Check for UEFI Shell 2.0 protocols
1033 //
1034 if (gEfiShellProtocol != NULL) {
1035
1036 //
1037 // We are using UEFI Shell 2.0; see if the event has been triggered
1038 //
1039 if (gBS->CheckEvent(gEfiShellProtocol->ExecutionBreak) != EFI_SUCCESS) {
1040 return (FALSE);
1041 }
1042 return (TRUE);
1043 }
1044
1045 //
1046 // using EFI Shell; call the function to check
1047 //
1048 if (mEfiShellEnvironment2 != NULL) {
1049 return (mEfiShellEnvironment2->GetExecutionBreak());
1050 }
1051
1052 return (FALSE);
1053 }
1054 /**
1055 return the value of an environment variable
1056
1057 this function gets the value of the environment variable set by the
1058 ShellSetEnvironmentVariable function
1059
1060 @param EnvKey The key name of the environment variable.
1061
1062 @retval NULL the named environment variable does not exist.
1063 @return != NULL pointer to the value of the environment variable
1064 **/
1065 CONST CHAR16*
1066 EFIAPI
1067 ShellGetEnvironmentVariable (
1068 IN CONST CHAR16 *EnvKey
1069 )
1070 {
1071 //
1072 // Check for UEFI Shell 2.0 protocols
1073 //
1074 if (gEfiShellProtocol != NULL) {
1075 return (gEfiShellProtocol->GetEnv(EnvKey));
1076 }
1077
1078 //
1079 // Check for EFI shell
1080 //
1081 if (mEfiShellEnvironment2 != NULL) {
1082 return (mEfiShellEnvironment2->GetEnv((CHAR16*)EnvKey));
1083 }
1084
1085 return NULL;
1086 }
1087 /**
1088 set the value of an environment variable
1089
1090 This function changes the current value of the specified environment variable. If the
1091 environment variable exists and the Value is an empty string, then the environment
1092 variable is deleted. If the environment variable exists and the Value is not an empty
1093 string, then the value of the environment variable is changed. If the environment
1094 variable does not exist and the Value is an empty string, there is no action. If the
1095 environment variable does not exist and the Value is a non-empty string, then the
1096 environment variable is created and assigned the specified value.
1097
1098 This is not supported pre-UEFI Shell 2.0.
1099
1100 @param EnvKey The key name of the environment variable.
1101 @param EnvVal The Value of the environment variable
1102 @param Volatile Indicates whether the variable is non-volatile (FALSE) or volatile (TRUE).
1103
1104 @retval EFI_SUCCESS the operation was completed sucessfully
1105 @retval EFI_UNSUPPORTED This operation is not allowed in pre UEFI 2.0 Shell environments
1106 **/
1107 EFI_STATUS
1108 EFIAPI
1109 ShellSetEnvironmentVariable (
1110 IN CONST CHAR16 *EnvKey,
1111 IN CONST CHAR16 *EnvVal,
1112 IN BOOLEAN Volatile
1113 )
1114 {
1115 //
1116 // Check for UEFI Shell 2.0 protocols
1117 //
1118 if (gEfiShellProtocol != NULL) {
1119 return (gEfiShellProtocol->SetEnv(EnvKey, EnvVal, Volatile));
1120 }
1121
1122 //
1123 // This feature does not exist under EFI shell
1124 //
1125 return (EFI_UNSUPPORTED);
1126 }
1127
1128 /**
1129 Cause the shell to parse and execute a command line.
1130
1131 This function creates a nested instance of the shell and executes the specified
1132 command (CommandLine) with the specified environment (Environment). Upon return,
1133 the status code returned by the specified command is placed in StatusCode.
1134 If Environment is NULL, then the current environment is used and all changes made
1135 by the commands executed will be reflected in the current environment. If the
1136 Environment is non-NULL, then the changes made will be discarded.
1137 The CommandLine is executed from the current working directory on the current
1138 device.
1139
1140 The EnvironmentVariables and Status parameters are ignored in a pre-UEFI Shell 2.0
1141 environment. The values pointed to by the parameters will be unchanged by the
1142 ShellExecute() function. The Output parameter has no effect in a
1143 UEFI Shell 2.0 environment.
1144
1145 @param[in] ParentHandle The parent image starting the operation.
1146 @param[in] CommandLine The pointer to a NULL terminated command line.
1147 @param[in] Output True to display debug output. False to hide it.
1148 @param[in] EnvironmentVariables Optional pointer to array of environment variables
1149 in the form "x=y". If NULL, the current set is used.
1150 @param[out] Status The status of the run command line.
1151
1152 @retval EFI_SUCCESS The operation completed sucessfully. Status
1153 contains the status code returned.
1154 @retval EFI_INVALID_PARAMETER A parameter contains an invalid value.
1155 @retval EFI_OUT_OF_RESOURCES Out of resources.
1156 @retval EFI_UNSUPPORTED The operation is not allowed.
1157 **/
1158 EFI_STATUS
1159 EFIAPI
1160 ShellExecute (
1161 IN EFI_HANDLE *ParentHandle,
1162 IN CHAR16 *CommandLine OPTIONAL,
1163 IN BOOLEAN Output OPTIONAL,
1164 IN CHAR16 **EnvironmentVariables OPTIONAL,
1165 OUT EFI_STATUS *Status OPTIONAL
1166 )
1167 {
1168 //
1169 // Check for UEFI Shell 2.0 protocols
1170 //
1171 if (gEfiShellProtocol != NULL) {
1172 //
1173 // Call UEFI Shell 2.0 version (not using Output parameter)
1174 //
1175 return (gEfiShellProtocol->Execute(ParentHandle,
1176 CommandLine,
1177 EnvironmentVariables,
1178 Status));
1179 }
1180
1181 //
1182 // Check for EFI shell
1183 //
1184 if (mEfiShellEnvironment2 != NULL) {
1185 //
1186 // Call EFI Shell version (not using EnvironmentVariables or Status parameters)
1187 // Due to oddity in the EFI shell we want to dereference the ParentHandle here
1188 //
1189 return (mEfiShellEnvironment2->Execute(*ParentHandle,
1190 CommandLine,
1191 Output));
1192 }
1193
1194 return (EFI_UNSUPPORTED);
1195 }
1196 /**
1197 Retreives the current directory path
1198
1199 If the DeviceName is NULL, it returns the current device's current directory
1200 name. If the DeviceName is not NULL, it returns the current directory name
1201 on specified drive.
1202
1203 @param DeviceName the name of the drive to get directory on
1204
1205 @retval NULL the directory does not exist
1206 @return != NULL the directory
1207 **/
1208 CONST CHAR16*
1209 EFIAPI
1210 ShellGetCurrentDir (
1211 IN CHAR16 * CONST DeviceName OPTIONAL
1212 )
1213 {
1214 //
1215 // Check for UEFI Shell 2.0 protocols
1216 //
1217 if (gEfiShellProtocol != NULL) {
1218 return (gEfiShellProtocol->GetCurDir(DeviceName));
1219 }
1220
1221 //
1222 // Check for EFI shell
1223 //
1224 if (mEfiShellEnvironment2 != NULL) {
1225 return (mEfiShellEnvironment2->CurDir(DeviceName));
1226 }
1227
1228 return (NULL);
1229 }
1230 /**
1231 sets (enabled or disabled) the page break mode
1232
1233 when page break mode is enabled the screen will stop scrolling
1234 and wait for operator input before scrolling a subsequent screen.
1235
1236 @param CurrentState TRUE to enable and FALSE to disable
1237 **/
1238 VOID
1239 EFIAPI
1240 ShellSetPageBreakMode (
1241 IN BOOLEAN CurrentState
1242 )
1243 {
1244 //
1245 // check for enabling
1246 //
1247 if (CurrentState != 0x00) {
1248 //
1249 // check for UEFI Shell 2.0
1250 //
1251 if (gEfiShellProtocol != NULL) {
1252 //
1253 // Enable with UEFI 2.0 Shell
1254 //
1255 gEfiShellProtocol->EnablePageBreak();
1256 return;
1257 } else {
1258 //
1259 // Check for EFI shell
1260 //
1261 if (mEfiShellEnvironment2 != NULL) {
1262 //
1263 // Enable with EFI Shell
1264 //
1265 mEfiShellEnvironment2->EnablePageBreak (DEFAULT_INIT_ROW, DEFAULT_AUTO_LF);
1266 return;
1267 }
1268 }
1269 } else {
1270 //
1271 // check for UEFI Shell 2.0
1272 //
1273 if (gEfiShellProtocol != NULL) {
1274 //
1275 // Disable with UEFI 2.0 Shell
1276 //
1277 gEfiShellProtocol->DisablePageBreak();
1278 return;
1279 } else {
1280 //
1281 // Check for EFI shell
1282 //
1283 if (mEfiShellEnvironment2 != NULL) {
1284 //
1285 // Disable with EFI Shell
1286 //
1287 mEfiShellEnvironment2->DisablePageBreak ();
1288 return;
1289 }
1290 }
1291 }
1292 }
1293
1294 ///
1295 /// version of EFI_SHELL_FILE_INFO struct, except has no CONST pointers.
1296 /// This allows for the struct to be populated.
1297 ///
1298 typedef struct {
1299 LIST_ENTRY Link;
1300 EFI_STATUS Status;
1301 CHAR16 *FullName;
1302 CHAR16 *FileName;
1303 SHELL_FILE_HANDLE Handle;
1304 EFI_FILE_INFO *Info;
1305 } EFI_SHELL_FILE_INFO_NO_CONST;
1306
1307 /**
1308 Converts a EFI shell list of structures to the coresponding UEFI Shell 2.0 type of list.
1309
1310 if OldStyleFileList is NULL then ASSERT()
1311
1312 this function will convert a SHELL_FILE_ARG based list into a callee allocated
1313 EFI_SHELL_FILE_INFO based list. it is up to the caller to free the memory via
1314 the ShellCloseFileMetaArg function.
1315
1316 @param[in] FileList the EFI shell list type
1317 @param[in, out] ListHead the list to add to
1318
1319 @retval the resultant head of the double linked new format list;
1320 **/
1321 LIST_ENTRY*
1322 EFIAPI
1323 InternalShellConvertFileListType (
1324 IN LIST_ENTRY *FileList,
1325 IN OUT LIST_ENTRY *ListHead
1326 )
1327 {
1328 SHELL_FILE_ARG *OldInfo;
1329 LIST_ENTRY *Link;
1330 EFI_SHELL_FILE_INFO_NO_CONST *NewInfo;
1331
1332 //
1333 // ASSERTs
1334 //
1335 ASSERT(FileList != NULL);
1336 ASSERT(ListHead != NULL);
1337
1338 //
1339 // enumerate through each member of the old list and copy
1340 //
1341 for (Link = FileList->ForwardLink; Link != FileList; Link = Link->ForwardLink) {
1342 OldInfo = CR (Link, SHELL_FILE_ARG, Link, SHELL_FILE_ARG_SIGNATURE);
1343 ASSERT(OldInfo != NULL);
1344
1345 //
1346 // Skip ones that failed to open...
1347 //
1348 if (OldInfo->Status != EFI_SUCCESS) {
1349 continue;
1350 }
1351
1352 //
1353 // make sure the old list was valid
1354 //
1355 ASSERT(OldInfo->Info != NULL);
1356 ASSERT(OldInfo->FullName != NULL);
1357 ASSERT(OldInfo->FileName != NULL);
1358
1359 //
1360 // allocate a new EFI_SHELL_FILE_INFO object
1361 //
1362 NewInfo = AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1363 ASSERT(NewInfo != NULL);
1364 if (NewInfo == NULL) {
1365 break;
1366 }
1367
1368 //
1369 // copy the simple items
1370 //
1371 NewInfo->Handle = OldInfo->Handle;
1372 NewInfo->Status = OldInfo->Status;
1373
1374 // old shell checks for 0 not NULL
1375 OldInfo->Handle = 0;
1376
1377 //
1378 // allocate new space to copy strings and structure
1379 //
1380 NewInfo->FullName = AllocateZeroPool(StrSize(OldInfo->FullName));
1381 NewInfo->FileName = AllocateZeroPool(StrSize(OldInfo->FileName));
1382 NewInfo->Info = AllocateZeroPool((UINTN)OldInfo->Info->Size);
1383
1384 //
1385 // make sure all the memory allocations were sucessful
1386 //
1387 ASSERT(NewInfo->FullName != NULL);
1388 ASSERT(NewInfo->FileName != NULL);
1389 ASSERT(NewInfo->Info != NULL);
1390
1391 //
1392 // Copt the strings and structure
1393 //
1394 StrCpy(NewInfo->FullName, OldInfo->FullName);
1395 StrCpy(NewInfo->FileName, OldInfo->FileName);
1396 gBS->CopyMem (NewInfo->Info, OldInfo->Info, (UINTN)OldInfo->Info->Size);
1397
1398 //
1399 // add that to the list
1400 //
1401 InsertTailList(ListHead, &NewInfo->Link);
1402 }
1403 return (ListHead);
1404 }
1405 /**
1406 Opens a group of files based on a path.
1407
1408 This function uses the Arg to open all the matching files. Each matched
1409 file has a SHELL_FILE_ARG structure to record the file information. These
1410 structures are placed on the list ListHead. Users can get the SHELL_FILE_ARG
1411 structures from ListHead to access each file. This function supports wildcards
1412 and will process '?' and '*' as such. the list must be freed with a call to
1413 ShellCloseFileMetaArg().
1414
1415 If you are NOT appending to an existing list *ListHead must be NULL. If
1416 *ListHead is NULL then it must be callee freed.
1417
1418 @param Arg pointer to path string
1419 @param OpenMode mode to open files with
1420 @param ListHead head of linked list of results
1421
1422 @retval EFI_SUCCESS the operation was sucessful and the list head
1423 contains the list of opened files
1424 @return != EFI_SUCCESS the operation failed
1425
1426 @sa InternalShellConvertFileListType
1427 **/
1428 EFI_STATUS
1429 EFIAPI
1430 ShellOpenFileMetaArg (
1431 IN CHAR16 *Arg,
1432 IN UINT64 OpenMode,
1433 IN OUT EFI_SHELL_FILE_INFO **ListHead
1434 )
1435 {
1436 EFI_STATUS Status;
1437 LIST_ENTRY mOldStyleFileList;
1438
1439 //
1440 // ASSERT that Arg and ListHead are not NULL
1441 //
1442 ASSERT(Arg != NULL);
1443 ASSERT(ListHead != NULL);
1444
1445 //
1446 // Check for UEFI Shell 2.0 protocols
1447 //
1448 if (gEfiShellProtocol != NULL) {
1449 if (*ListHead == NULL) {
1450 *ListHead = (EFI_SHELL_FILE_INFO*)AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1451 if (*ListHead == NULL) {
1452 return (EFI_OUT_OF_RESOURCES);
1453 }
1454 InitializeListHead(&((*ListHead)->Link));
1455 }
1456 Status = gEfiShellProtocol->OpenFileList(Arg,
1457 OpenMode,
1458 ListHead);
1459 if (EFI_ERROR(Status)) {
1460 gEfiShellProtocol->RemoveDupInFileList(ListHead);
1461 } else {
1462 Status = gEfiShellProtocol->RemoveDupInFileList(ListHead);
1463 }
1464 if (*ListHead != NULL && IsListEmpty(&(*ListHead)->Link)) {
1465 FreePool(*ListHead);
1466 *ListHead = NULL;
1467 return (EFI_NOT_FOUND);
1468 }
1469 return (Status);
1470 }
1471
1472 //
1473 // Check for EFI shell
1474 //
1475 if (mEfiShellEnvironment2 != NULL) {
1476 //
1477 // make sure the list head is initialized
1478 //
1479 InitializeListHead(&mOldStyleFileList);
1480
1481 //
1482 // Get the EFI Shell list of files
1483 //
1484 Status = mEfiShellEnvironment2->FileMetaArg(Arg, &mOldStyleFileList);
1485 if (EFI_ERROR(Status)) {
1486 *ListHead = NULL;
1487 return (Status);
1488 }
1489
1490 if (*ListHead == NULL) {
1491 *ListHead = (EFI_SHELL_FILE_INFO *)AllocateZeroPool(sizeof(EFI_SHELL_FILE_INFO));
1492 if (*ListHead == NULL) {
1493 return (EFI_OUT_OF_RESOURCES);
1494 }
1495 InitializeListHead(&((*ListHead)->Link));
1496 }
1497
1498 //
1499 // Convert that to equivalent of UEFI Shell 2.0 structure
1500 //
1501 InternalShellConvertFileListType(&mOldStyleFileList, &(*ListHead)->Link);
1502
1503 //
1504 // Free the EFI Shell version that was converted.
1505 //
1506 mEfiShellEnvironment2->FreeFileList(&mOldStyleFileList);
1507
1508 if ((*ListHead)->Link.ForwardLink == (*ListHead)->Link.BackLink && (*ListHead)->Link.BackLink == &((*ListHead)->Link)) {
1509 FreePool(*ListHead);
1510 *ListHead = NULL;
1511 Status = EFI_NOT_FOUND;
1512 }
1513 return (Status);
1514 }
1515
1516 return (EFI_UNSUPPORTED);
1517 }
1518 /**
1519 Free the linked list returned from ShellOpenFileMetaArg.
1520
1521 if ListHead is NULL then ASSERT().
1522
1523 @param ListHead the pointer to free.
1524
1525 @retval EFI_SUCCESS the operation was sucessful.
1526 **/
1527 EFI_STATUS
1528 EFIAPI
1529 ShellCloseFileMetaArg (
1530 IN OUT EFI_SHELL_FILE_INFO **ListHead
1531 )
1532 {
1533 LIST_ENTRY *Node;
1534
1535 //
1536 // ASSERT that ListHead is not NULL
1537 //
1538 ASSERT(ListHead != NULL);
1539
1540 //
1541 // Check for UEFI Shell 2.0 protocols
1542 //
1543 if (gEfiShellProtocol != NULL) {
1544 return (gEfiShellProtocol->FreeFileList(ListHead));
1545 } else if (mEfiShellEnvironment2 != NULL) {
1546 //
1547 // Since this is EFI Shell version we need to free our internally made copy
1548 // of the list
1549 //
1550 for ( Node = GetFirstNode(&(*ListHead)->Link)
1551 ; *ListHead != NULL && !IsListEmpty(&(*ListHead)->Link)
1552 ; Node = GetFirstNode(&(*ListHead)->Link)) {
1553 RemoveEntryList(Node);
1554 ((EFI_FILE_PROTOCOL*)((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->Handle)->Close(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->Handle);
1555 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->FullName);
1556 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->FileName);
1557 FreePool(((EFI_SHELL_FILE_INFO_NO_CONST*)Node)->Info);
1558 FreePool((EFI_SHELL_FILE_INFO_NO_CONST*)Node);
1559 }
1560 return EFI_SUCCESS;
1561 }
1562
1563 return (EFI_UNSUPPORTED);
1564 }
1565
1566 /**
1567 Find a file by searching the CWD and then the path.
1568
1569 If FileName is NULL then ASSERT.
1570
1571 If the return value is not NULL then the memory must be caller freed.
1572
1573 @param FileName Filename string.
1574
1575 @retval NULL the file was not found
1576 @return !NULL the full path to the file.
1577 **/
1578 CHAR16 *
1579 EFIAPI
1580 ShellFindFilePath (
1581 IN CONST CHAR16 *FileName
1582 )
1583 {
1584 CONST CHAR16 *Path;
1585 SHELL_FILE_HANDLE Handle;
1586 EFI_STATUS Status;
1587 CHAR16 *RetVal;
1588 CHAR16 *TestPath;
1589 CONST CHAR16 *Walker;
1590 UINTN Size;
1591 CHAR16 *TempChar;
1592
1593 RetVal = NULL;
1594
1595 //
1596 // First make sure its not an absolute path.
1597 //
1598 Status = ShellOpenFileByName(FileName, &Handle, EFI_FILE_MODE_READ, 0);
1599 if (!EFI_ERROR(Status)){
1600 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1601 ASSERT(RetVal == NULL);
1602 RetVal = StrnCatGrow(&RetVal, NULL, FileName, 0);
1603 ShellCloseFile(&Handle);
1604 return (RetVal);
1605 } else {
1606 ShellCloseFile(&Handle);
1607 }
1608 }
1609
1610 Path = ShellGetEnvironmentVariable(L"cwd");
1611 if (Path != NULL) {
1612 Size = StrSize(Path);
1613 Size += StrSize(FileName);
1614 TestPath = AllocateZeroPool(Size);
1615 ASSERT(TestPath != NULL);
1616 if (TestPath == NULL) {
1617 return (NULL);
1618 }
1619 StrCpy(TestPath, Path);
1620 StrCat(TestPath, FileName);
1621 Status = ShellOpenFileByName(TestPath, &Handle, EFI_FILE_MODE_READ, 0);
1622 if (!EFI_ERROR(Status)){
1623 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1624 ASSERT(RetVal == NULL);
1625 RetVal = StrnCatGrow(&RetVal, NULL, TestPath, 0);
1626 ShellCloseFile(&Handle);
1627 FreePool(TestPath);
1628 return (RetVal);
1629 } else {
1630 ShellCloseFile(&Handle);
1631 }
1632 }
1633 FreePool(TestPath);
1634 }
1635 Path = ShellGetEnvironmentVariable(L"path");
1636 if (Path != NULL) {
1637 Size = StrSize(Path)+sizeof(CHAR16);
1638 Size += StrSize(FileName);
1639 TestPath = AllocateZeroPool(Size);
1640 if (TestPath == NULL) {
1641 return (NULL);
1642 }
1643 Walker = (CHAR16*)Path;
1644 do {
1645 CopyMem(TestPath, Walker, StrSize(Walker));
1646 if (TestPath != NULL) {
1647 TempChar = StrStr(TestPath, L";");
1648 if (TempChar != NULL) {
1649 *TempChar = CHAR_NULL;
1650 }
1651 if (TestPath[StrLen(TestPath)-1] != L'\\') {
1652 StrCat(TestPath, L"\\");
1653 }
1654 if (FileName[0] == L'\\') {
1655 FileName++;
1656 }
1657 StrCat(TestPath, FileName);
1658 if (StrStr(Walker, L";") != NULL) {
1659 Walker = StrStr(Walker, L";") + 1;
1660 } else {
1661 Walker = NULL;
1662 }
1663 Status = ShellOpenFileByName(TestPath, &Handle, EFI_FILE_MODE_READ, 0);
1664 if (!EFI_ERROR(Status)){
1665 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
1666 ASSERT(RetVal == NULL);
1667 RetVal = StrnCatGrow(&RetVal, NULL, TestPath, 0);
1668 ShellCloseFile(&Handle);
1669 break;
1670 } else {
1671 ShellCloseFile(&Handle);
1672 }
1673 }
1674 }
1675 } while (Walker != NULL && Walker[0] != CHAR_NULL);
1676 FreePool(TestPath);
1677 }
1678 return (RetVal);
1679 }
1680
1681 /**
1682 Find a file by searching the CWD and then the path with a variable set of file
1683 extensions. If the file is not found it will append each extension in the list
1684 in the order provided and return the first one that is successful.
1685
1686 If FileName is NULL, then ASSERT.
1687 If FileExtension is NULL, then behavior is identical to ShellFindFilePath.
1688
1689 If the return value is not NULL then the memory must be caller freed.
1690
1691 @param[in] FileName Filename string.
1692 @param[in] FileExtension Semi-colon delimeted list of possible extensions.
1693
1694 @retval NULL The file was not found.
1695 @retval !NULL The path to the file.
1696 **/
1697 CHAR16 *
1698 EFIAPI
1699 ShellFindFilePathEx (
1700 IN CONST CHAR16 *FileName,
1701 IN CONST CHAR16 *FileExtension
1702 )
1703 {
1704 CHAR16 *TestPath;
1705 CHAR16 *RetVal;
1706 CONST CHAR16 *ExtensionWalker;
1707 UINTN Size;
1708 CHAR16 *TempChar;
1709 CHAR16 *TempChar2;
1710
1711 ASSERT(FileName != NULL);
1712 if (FileExtension == NULL) {
1713 return (ShellFindFilePath(FileName));
1714 }
1715 RetVal = ShellFindFilePath(FileName);
1716 if (RetVal != NULL) {
1717 return (RetVal);
1718 }
1719 Size = StrSize(FileName);
1720 Size += StrSize(FileExtension);
1721 TestPath = AllocateZeroPool(Size);
1722 ASSERT(TestPath != NULL);
1723 if (TestPath == NULL) {
1724 return (NULL);
1725 }
1726 for (ExtensionWalker = FileExtension, TempChar2 = (CHAR16*)FileExtension; TempChar2 != NULL ; ExtensionWalker = TempChar2 + 1){
1727 StrCpy(TestPath, FileName);
1728 if (ExtensionWalker != NULL) {
1729 StrCat(TestPath, ExtensionWalker);
1730 }
1731 TempChar = StrStr(TestPath, L";");
1732 if (TempChar != NULL) {
1733 *TempChar = CHAR_NULL;
1734 }
1735 RetVal = ShellFindFilePath(TestPath);
1736 if (RetVal != NULL) {
1737 break;
1738 }
1739 ASSERT(ExtensionWalker != NULL);
1740 TempChar2 = StrStr(ExtensionWalker, L";");
1741 }
1742 FreePool(TestPath);
1743 return (RetVal);
1744 }
1745
1746 typedef struct {
1747 LIST_ENTRY Link;
1748 CHAR16 *Name;
1749 SHELL_PARAM_TYPE Type;
1750 CHAR16 *Value;
1751 UINTN OriginalPosition;
1752 } SHELL_PARAM_PACKAGE;
1753
1754 /**
1755 Checks the list of valid arguments and returns TRUE if the item was found. If the
1756 return value is TRUE then the type parameter is set also.
1757
1758 if CheckList is NULL then ASSERT();
1759 if Name is NULL then ASSERT();
1760 if Type is NULL then ASSERT();
1761
1762 @param Name pointer to Name of parameter found
1763 @param CheckList List to check against
1764 @param Type pointer to type of parameter if it was found
1765
1766 @retval TRUE the Parameter was found. Type is valid.
1767 @retval FALSE the Parameter was not found. Type is not valid.
1768 **/
1769 BOOLEAN
1770 EFIAPI
1771 InternalIsOnCheckList (
1772 IN CONST CHAR16 *Name,
1773 IN CONST SHELL_PARAM_ITEM *CheckList,
1774 OUT SHELL_PARAM_TYPE *Type
1775 )
1776 {
1777 SHELL_PARAM_ITEM *TempListItem;
1778 CHAR16 *TempString;
1779
1780 //
1781 // ASSERT that all 3 pointer parameters aren't NULL
1782 //
1783 ASSERT(CheckList != NULL);
1784 ASSERT(Type != NULL);
1785 ASSERT(Name != NULL);
1786
1787 //
1788 // question mark and page break mode are always supported
1789 //
1790 if ((StrCmp(Name, L"-?") == 0) ||
1791 (StrCmp(Name, L"-b") == 0)
1792 ) {
1793 *Type = TypeFlag;
1794 return (TRUE);
1795 }
1796
1797 //
1798 // Enumerate through the list
1799 //
1800 for (TempListItem = (SHELL_PARAM_ITEM*)CheckList ; TempListItem->Name != NULL ; TempListItem++) {
1801 //
1802 // If the Type is TypeStart only check the first characters of the passed in param
1803 // If it matches set the type and return TRUE
1804 //
1805 if (TempListItem->Type == TypeStart) {
1806 if (StrnCmp(Name, TempListItem->Name, StrLen(TempListItem->Name)) == 0) {
1807 *Type = TempListItem->Type;
1808 return (TRUE);
1809 }
1810 TempString = NULL;
1811 TempString = StrnCatGrow(&TempString, NULL, Name, StrLen(TempListItem->Name));
1812 if (TempString != NULL) {
1813 if (StringNoCaseCompare(&TempString, &TempListItem->Name) == 0) {
1814 *Type = TempListItem->Type;
1815 FreePool(TempString);
1816 return (TRUE);
1817 }
1818 FreePool(TempString);
1819 }
1820 } else if (StringNoCaseCompare(&Name, &TempListItem->Name) == 0) {
1821 *Type = TempListItem->Type;
1822 return (TRUE);
1823 }
1824 }
1825
1826 return (FALSE);
1827 }
1828 /**
1829 Checks the string for indicators of "flag" status. this is a leading '/', '-', or '+'
1830
1831 @param[in] Name pointer to Name of parameter found
1832 @param[in] AlwaysAllowNumbers TRUE to allow numbers, FALSE to not.
1833
1834 @retval TRUE the Parameter is a flag.
1835 @retval FALSE the Parameter not a flag.
1836 **/
1837 BOOLEAN
1838 EFIAPI
1839 InternalIsFlag (
1840 IN CONST CHAR16 *Name,
1841 IN BOOLEAN AlwaysAllowNumbers
1842 )
1843 {
1844 //
1845 // ASSERT that Name isn't NULL
1846 //
1847 ASSERT(Name != NULL);
1848
1849 //
1850 // If we accept numbers then dont return TRUE. (they will be values)
1851 //
1852 if (((Name[0] == L'-' || Name[0] == L'+') && ShellIsHexaDecimalDigitCharacter(Name[1])) && AlwaysAllowNumbers) {
1853 return (FALSE);
1854 }
1855
1856 //
1857 // If the Name has a /, +, or - as the first character return TRUE
1858 //
1859 if ((Name[0] == L'/') ||
1860 (Name[0] == L'-') ||
1861 (Name[0] == L'+')
1862 ) {
1863 return (TRUE);
1864 }
1865 return (FALSE);
1866 }
1867
1868 /**
1869 Checks the command line arguments passed against the list of valid ones.
1870
1871 If no initialization is required, then return RETURN_SUCCESS.
1872
1873 @param[in] CheckList pointer to list of parameters to check
1874 @param[out] CheckPackage pointer to pointer to list checked values
1875 @param[out] ProblemParam optional pointer to pointer to unicode string for
1876 the paramater that caused failure. If used then the
1877 caller is responsible for freeing the memory.
1878 @param[in] AutoPageBreak will automatically set PageBreakEnabled for "b" parameter
1879 @param[in] Argv pointer to array of parameters
1880 @param[in] Argc Count of parameters in Argv
1881 @param[in] AlwaysAllowNumbers TRUE to allow numbers always, FALSE otherwise.
1882
1883 @retval EFI_SUCCESS The operation completed sucessfully.
1884 @retval EFI_OUT_OF_RESOURCES A memory allocation failed
1885 @retval EFI_INVALID_PARAMETER A parameter was invalid
1886 @retval EFI_VOLUME_CORRUPTED the command line was corrupt. an argument was
1887 duplicated. the duplicated command line argument
1888 was returned in ProblemParam if provided.
1889 @retval EFI_NOT_FOUND a argument required a value that was missing.
1890 the invalid command line argument was returned in
1891 ProblemParam if provided.
1892 **/
1893 EFI_STATUS
1894 EFIAPI
1895 InternalCommandLineParse (
1896 IN CONST SHELL_PARAM_ITEM *CheckList,
1897 OUT LIST_ENTRY **CheckPackage,
1898 OUT CHAR16 **ProblemParam OPTIONAL,
1899 IN BOOLEAN AutoPageBreak,
1900 IN CONST CHAR16 **Argv,
1901 IN UINTN Argc,
1902 IN BOOLEAN AlwaysAllowNumbers
1903 )
1904 {
1905 UINTN LoopCounter;
1906 SHELL_PARAM_TYPE CurrentItemType;
1907 SHELL_PARAM_PACKAGE *CurrentItemPackage;
1908 UINTN GetItemValue;
1909 UINTN ValueSize;
1910 UINTN Count;
1911 CONST CHAR16 *TempPointer;
1912
1913 CurrentItemPackage = NULL;
1914 GetItemValue = 0;
1915 ValueSize = 0;
1916 Count = 0;
1917
1918 //
1919 // If there is only 1 item we dont need to do anything
1920 //
1921 if (Argc < 1) {
1922 *CheckPackage = NULL;
1923 return (EFI_SUCCESS);
1924 }
1925
1926 //
1927 // ASSERTs
1928 //
1929 ASSERT(CheckList != NULL);
1930 ASSERT(Argv != NULL);
1931
1932 //
1933 // initialize the linked list
1934 //
1935 *CheckPackage = (LIST_ENTRY*)AllocateZeroPool(sizeof(LIST_ENTRY));
1936 InitializeListHead(*CheckPackage);
1937
1938 //
1939 // loop through each of the arguments
1940 //
1941 for (LoopCounter = 0 ; LoopCounter < Argc ; ++LoopCounter) {
1942 if (Argv[LoopCounter] == NULL) {
1943 //
1944 // do nothing for NULL argv
1945 //
1946 } else if (InternalIsOnCheckList(Argv[LoopCounter], CheckList, &CurrentItemType)) {
1947 //
1948 // We might have leftover if last parameter didnt have optional value
1949 //
1950 if (GetItemValue != 0) {
1951 GetItemValue = 0;
1952 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
1953 }
1954 //
1955 // this is a flag
1956 //
1957 CurrentItemPackage = AllocateZeroPool(sizeof(SHELL_PARAM_PACKAGE));
1958 ASSERT(CurrentItemPackage != NULL);
1959 CurrentItemPackage->Name = AllocateZeroPool(StrSize(Argv[LoopCounter]));
1960 ASSERT(CurrentItemPackage->Name != NULL);
1961 StrCpy(CurrentItemPackage->Name, Argv[LoopCounter]);
1962 CurrentItemPackage->Type = CurrentItemType;
1963 CurrentItemPackage->OriginalPosition = (UINTN)(-1);
1964 CurrentItemPackage->Value = NULL;
1965
1966 //
1967 // Does this flag require a value
1968 //
1969 switch (CurrentItemPackage->Type) {
1970 //
1971 // possibly trigger the next loop(s) to populate the value of this item
1972 //
1973 case TypeValue:
1974 GetItemValue = 1;
1975 ValueSize = 0;
1976 break;
1977 case TypeDoubleValue:
1978 GetItemValue = 2;
1979 ValueSize = 0;
1980 break;
1981 case TypeMaxValue:
1982 GetItemValue = (UINTN)(-1);
1983 ValueSize = 0;
1984 break;
1985 default:
1986 //
1987 // this item has no value expected; we are done
1988 //
1989 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
1990 ASSERT(GetItemValue == 0);
1991 break;
1992 }
1993 } else if (GetItemValue != 0 && !InternalIsFlag(Argv[LoopCounter], AlwaysAllowNumbers)) {
1994 ASSERT(CurrentItemPackage != NULL);
1995 //
1996 // get the item VALUE for a previous flag
1997 //
1998 CurrentItemPackage->Value = ReallocatePool(ValueSize, ValueSize + StrSize(Argv[LoopCounter]) + sizeof(CHAR16), CurrentItemPackage->Value);
1999 ASSERT(CurrentItemPackage->Value != NULL);
2000 if (ValueSize == 0) {
2001 StrCpy(CurrentItemPackage->Value, Argv[LoopCounter]);
2002 } else {
2003 StrCat(CurrentItemPackage->Value, L" ");
2004 StrCat(CurrentItemPackage->Value, Argv[LoopCounter]);
2005 }
2006 ValueSize += StrSize(Argv[LoopCounter]) + sizeof(CHAR16);
2007 GetItemValue--;
2008 if (GetItemValue == 0) {
2009 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2010 }
2011 } else if (!InternalIsFlag(Argv[LoopCounter], AlwaysAllowNumbers) ){ //|| ProblemParam == NULL) {
2012 //
2013 // add this one as a non-flag
2014 //
2015
2016 TempPointer = Argv[LoopCounter];
2017 if ((*TempPointer == L'^' && *(TempPointer+1) == L'-')
2018 || (*TempPointer == L'^' && *(TempPointer+1) == L'/')
2019 || (*TempPointer == L'^' && *(TempPointer+1) == L'+')
2020 ){
2021 TempPointer++;
2022 }
2023 CurrentItemPackage = AllocateZeroPool(sizeof(SHELL_PARAM_PACKAGE));
2024 ASSERT(CurrentItemPackage != NULL);
2025 CurrentItemPackage->Name = NULL;
2026 CurrentItemPackage->Type = TypePosition;
2027 CurrentItemPackage->Value = AllocateZeroPool(StrSize(TempPointer));
2028 ASSERT(CurrentItemPackage->Value != NULL);
2029 StrCpy(CurrentItemPackage->Value, TempPointer);
2030 CurrentItemPackage->OriginalPosition = Count++;
2031 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2032 } else {
2033 //
2034 // this was a non-recognised flag... error!
2035 //
2036 if (ProblemParam != NULL) {
2037 *ProblemParam = AllocateZeroPool(StrSize(Argv[LoopCounter]));
2038 ASSERT(*ProblemParam != NULL);
2039 StrCpy(*ProblemParam, Argv[LoopCounter]);
2040 }
2041 ShellCommandLineFreeVarList(*CheckPackage);
2042 *CheckPackage = NULL;
2043 return (EFI_VOLUME_CORRUPTED);
2044 }
2045 }
2046 if (GetItemValue != 0) {
2047 GetItemValue = 0;
2048 InsertHeadList(*CheckPackage, &CurrentItemPackage->Link);
2049 }
2050 //
2051 // support for AutoPageBreak
2052 //
2053 if (AutoPageBreak && ShellCommandLineGetFlag(*CheckPackage, L"-b")) {
2054 ShellSetPageBreakMode(TRUE);
2055 }
2056 return (EFI_SUCCESS);
2057 }
2058
2059 /**
2060 Checks the command line arguments passed against the list of valid ones.
2061 Optionally removes NULL values first.
2062
2063 If no initialization is required, then return RETURN_SUCCESS.
2064
2065 @param[in] CheckList The pointer to list of parameters to check.
2066 @param[out] CheckPackage The package of checked values.
2067 @param[out] ProblemParam Optional pointer to pointer to unicode string for
2068 the paramater that caused failure.
2069 @param[in] AutoPageBreak Will automatically set PageBreakEnabled.
2070 @param[in] AlwaysAllowNumbers Will never fail for number based flags.
2071
2072 @retval EFI_SUCCESS The operation completed sucessfully.
2073 @retval EFI_OUT_OF_RESOURCES A memory allocation failed.
2074 @retval EFI_INVALID_PARAMETER A parameter was invalid.
2075 @retval EFI_VOLUME_CORRUPTED The command line was corrupt.
2076 @retval EFI_DEVICE_ERROR The commands contained 2 opposing arguments. One
2077 of the command line arguments was returned in
2078 ProblemParam if provided.
2079 @retval EFI_NOT_FOUND A argument required a value that was missing.
2080 The invalid command line argument was returned in
2081 ProblemParam if provided.
2082 **/
2083 EFI_STATUS
2084 EFIAPI
2085 ShellCommandLineParseEx (
2086 IN CONST SHELL_PARAM_ITEM *CheckList,
2087 OUT LIST_ENTRY **CheckPackage,
2088 OUT CHAR16 **ProblemParam OPTIONAL,
2089 IN BOOLEAN AutoPageBreak,
2090 IN BOOLEAN AlwaysAllowNumbers
2091 )
2092 {
2093 //
2094 // ASSERT that CheckList and CheckPackage aren't NULL
2095 //
2096 ASSERT(CheckList != NULL);
2097 ASSERT(CheckPackage != NULL);
2098
2099 //
2100 // Check for UEFI Shell 2.0 protocols
2101 //
2102 if (gEfiShellParametersProtocol != NULL) {
2103 return (InternalCommandLineParse(CheckList,
2104 CheckPackage,
2105 ProblemParam,
2106 AutoPageBreak,
2107 (CONST CHAR16**) gEfiShellParametersProtocol->Argv,
2108 gEfiShellParametersProtocol->Argc,
2109 AlwaysAllowNumbers));
2110 }
2111
2112 //
2113 // ASSERT That EFI Shell is not required
2114 //
2115 ASSERT (mEfiShellInterface != NULL);
2116 return (InternalCommandLineParse(CheckList,
2117 CheckPackage,
2118 ProblemParam,
2119 AutoPageBreak,
2120 (CONST CHAR16**) mEfiShellInterface->Argv,
2121 mEfiShellInterface->Argc,
2122 AlwaysAllowNumbers));
2123 }
2124
2125 /**
2126 Frees shell variable list that was returned from ShellCommandLineParse.
2127
2128 This function will free all the memory that was used for the CheckPackage
2129 list of postprocessed shell arguments.
2130
2131 this function has no return value.
2132
2133 if CheckPackage is NULL, then return
2134
2135 @param CheckPackage the list to de-allocate
2136 **/
2137 VOID
2138 EFIAPI
2139 ShellCommandLineFreeVarList (
2140 IN LIST_ENTRY *CheckPackage
2141 )
2142 {
2143 LIST_ENTRY *Node;
2144
2145 //
2146 // check for CheckPackage == NULL
2147 //
2148 if (CheckPackage == NULL) {
2149 return;
2150 }
2151
2152 //
2153 // for each node in the list
2154 //
2155 for ( Node = GetFirstNode(CheckPackage)
2156 ; !IsListEmpty(CheckPackage)
2157 ; Node = GetFirstNode(CheckPackage)
2158 ){
2159 //
2160 // Remove it from the list
2161 //
2162 RemoveEntryList(Node);
2163
2164 //
2165 // if it has a name free the name
2166 //
2167 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
2168 FreePool(((SHELL_PARAM_PACKAGE*)Node)->Name);
2169 }
2170
2171 //
2172 // if it has a value free the value
2173 //
2174 if (((SHELL_PARAM_PACKAGE*)Node)->Value != NULL) {
2175 FreePool(((SHELL_PARAM_PACKAGE*)Node)->Value);
2176 }
2177
2178 //
2179 // free the node structure
2180 //
2181 FreePool((SHELL_PARAM_PACKAGE*)Node);
2182 }
2183 //
2184 // free the list head node
2185 //
2186 FreePool(CheckPackage);
2187 }
2188 /**
2189 Checks for presence of a flag parameter
2190
2191 flag arguments are in the form of "-<Key>" or "/<Key>", but do not have a value following the key
2192
2193 if CheckPackage is NULL then return FALSE.
2194 if KeyString is NULL then ASSERT()
2195
2196 @param CheckPackage The package of parsed command line arguments
2197 @param KeyString the Key of the command line argument to check for
2198
2199 @retval TRUE the flag is on the command line
2200 @retval FALSE the flag is not on the command line
2201 **/
2202 BOOLEAN
2203 EFIAPI
2204 ShellCommandLineGetFlag (
2205 IN CONST LIST_ENTRY * CONST CheckPackage,
2206 IN CONST CHAR16 * CONST KeyString
2207 )
2208 {
2209 LIST_ENTRY *Node;
2210 CHAR16 *TempString;
2211
2212 //
2213 // ASSERT that both CheckPackage and KeyString aren't NULL
2214 //
2215 ASSERT(KeyString != NULL);
2216
2217 //
2218 // return FALSE for no package
2219 //
2220 if (CheckPackage == NULL) {
2221 return (FALSE);
2222 }
2223
2224 //
2225 // enumerate through the list of parametrs
2226 //
2227 for ( Node = GetFirstNode(CheckPackage)
2228 ; !IsNull (CheckPackage, Node)
2229 ; Node = GetNextNode(CheckPackage, Node)
2230 ){
2231 //
2232 // If the Name matches, return TRUE (and there may be NULL name)
2233 //
2234 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
2235 //
2236 // If Type is TypeStart then only compare the begining of the strings
2237 //
2238 if (((SHELL_PARAM_PACKAGE*)Node)->Type == TypeStart) {
2239 if (StrnCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name, StrLen(KeyString)) == 0) {
2240 return (TRUE);
2241 }
2242 TempString = NULL;
2243 TempString = StrnCatGrow(&TempString, NULL, KeyString, StrLen(((SHELL_PARAM_PACKAGE*)Node)->Name));
2244 if (TempString != NULL) {
2245 if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2246 FreePool(TempString);
2247 return (TRUE);
2248 }
2249 FreePool(TempString);
2250 }
2251 } else if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2252 return (TRUE);
2253 }
2254 }
2255 }
2256 return (FALSE);
2257 }
2258 /**
2259 Returns value from command line argument.
2260
2261 Value parameters are in the form of "-<Key> value" or "/<Key> value".
2262
2263 If CheckPackage is NULL, then return NULL.
2264
2265 @param[in] CheckPackage The package of parsed command line arguments.
2266 @param[in] KeyString The Key of the command line argument to check for.
2267
2268 @retval NULL The flag is not on the command line.
2269 @retval !=NULL The pointer to unicode string of the value.
2270 **/
2271 CONST CHAR16*
2272 EFIAPI
2273 ShellCommandLineGetValue (
2274 IN CONST LIST_ENTRY *CheckPackage,
2275 IN CHAR16 *KeyString
2276 )
2277 {
2278 LIST_ENTRY *Node;
2279 CHAR16 *TempString;
2280
2281 //
2282 // check for CheckPackage == NULL
2283 //
2284 if (CheckPackage == NULL) {
2285 return (NULL);
2286 }
2287
2288 //
2289 // enumerate through the list of parametrs
2290 //
2291 for ( Node = GetFirstNode(CheckPackage)
2292 ; !IsNull (CheckPackage, Node)
2293 ; Node = GetNextNode(CheckPackage, Node)
2294 ){
2295 //
2296 // If the Name matches, return TRUE (and there may be NULL name)
2297 //
2298 if (((SHELL_PARAM_PACKAGE*)Node)->Name != NULL) {
2299 //
2300 // If Type is TypeStart then only compare the begining of the strings
2301 //
2302 if (((SHELL_PARAM_PACKAGE*)Node)->Type == TypeStart) {
2303 if (StrnCmp(KeyString, ((SHELL_PARAM_PACKAGE*)Node)->Name, StrLen(KeyString)) == 0) {
2304 return (((SHELL_PARAM_PACKAGE*)Node)->Name + StrLen(KeyString));
2305 }
2306 TempString = NULL;
2307 TempString = StrnCatGrow(&TempString, NULL, KeyString, StrLen(((SHELL_PARAM_PACKAGE*)Node)->Name));
2308 if (TempString != NULL) {
2309 if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2310 FreePool(TempString);
2311 return (((SHELL_PARAM_PACKAGE*)Node)->Name + StrLen(KeyString));
2312 }
2313 FreePool(TempString);
2314 }
2315 } else if (StringNoCaseCompare(&KeyString, &((SHELL_PARAM_PACKAGE*)Node)->Name) == 0) {
2316 return (((SHELL_PARAM_PACKAGE*)Node)->Value);
2317 }
2318 }
2319 }
2320 return (NULL);
2321 }
2322
2323 /**
2324 Returns raw value from command line argument.
2325
2326 Raw value parameters are in the form of "value" in a specific position in the list.
2327
2328 If CheckPackage is NULL, then return NULL.
2329
2330 @param[in] CheckPackage The package of parsed command line arguments.
2331 @param[in] Position The position of the value.
2332
2333 @retval NULL The flag is not on the command line.
2334 @retval !=NULL The pointer to unicode string of the value.
2335 **/
2336 CONST CHAR16*
2337 EFIAPI
2338 ShellCommandLineGetRawValue (
2339 IN CONST LIST_ENTRY * CONST CheckPackage,
2340 IN UINTN Position
2341 )
2342 {
2343 LIST_ENTRY *Node;
2344
2345 //
2346 // check for CheckPackage == NULL
2347 //
2348 if (CheckPackage == NULL) {
2349 return (NULL);
2350 }
2351
2352 //
2353 // enumerate through the list of parametrs
2354 //
2355 for ( Node = GetFirstNode(CheckPackage)
2356 ; !IsNull (CheckPackage, Node)
2357 ; Node = GetNextNode(CheckPackage, Node)
2358 ){
2359 //
2360 // If the position matches, return the value
2361 //
2362 if (((SHELL_PARAM_PACKAGE*)Node)->OriginalPosition == Position) {
2363 return (((SHELL_PARAM_PACKAGE*)Node)->Value);
2364 }
2365 }
2366 return (NULL);
2367 }
2368
2369 /**
2370 returns the number of command line value parameters that were parsed.
2371
2372 this will not include flags.
2373
2374 @param[in] CheckPackage The package of parsed command line arguments.
2375
2376 @retval (UINTN)-1 No parsing has ocurred
2377 @return other The number of value parameters found
2378 **/
2379 UINTN
2380 EFIAPI
2381 ShellCommandLineGetCount(
2382 IN CONST LIST_ENTRY *CheckPackage
2383 )
2384 {
2385 LIST_ENTRY *Node1;
2386 UINTN Count;
2387
2388 if (CheckPackage == NULL) {
2389 return (0);
2390 }
2391 for ( Node1 = GetFirstNode(CheckPackage), Count = 0
2392 ; !IsNull (CheckPackage, Node1)
2393 ; Node1 = GetNextNode(CheckPackage, Node1)
2394 ){
2395 if (((SHELL_PARAM_PACKAGE*)Node1)->Name == NULL) {
2396 Count++;
2397 }
2398 }
2399 return (Count);
2400 }
2401
2402 /**
2403 Determins if a parameter is duplicated.
2404
2405 If Param is not NULL then it will point to a callee allocated string buffer
2406 with the parameter value if a duplicate is found.
2407
2408 If CheckPackage is NULL, then ASSERT.
2409
2410 @param[in] CheckPackage The package of parsed command line arguments.
2411 @param[out] Param Upon finding one, a pointer to the duplicated parameter.
2412
2413 @retval EFI_SUCCESS No parameters were duplicated.
2414 @retval EFI_DEVICE_ERROR A duplicate was found.
2415 **/
2416 EFI_STATUS
2417 EFIAPI
2418 ShellCommandLineCheckDuplicate (
2419 IN CONST LIST_ENTRY *CheckPackage,
2420 OUT CHAR16 **Param
2421 )
2422 {
2423 LIST_ENTRY *Node1;
2424 LIST_ENTRY *Node2;
2425
2426 ASSERT(CheckPackage != NULL);
2427
2428 for ( Node1 = GetFirstNode(CheckPackage)
2429 ; !IsNull (CheckPackage, Node1)
2430 ; Node1 = GetNextNode(CheckPackage, Node1)
2431 ){
2432 for ( Node2 = GetNextNode(CheckPackage, Node1)
2433 ; !IsNull (CheckPackage, Node2)
2434 ; Node2 = GetNextNode(CheckPackage, Node2)
2435 ){
2436 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) {
2437 if (Param != NULL) {
2438 *Param = NULL;
2439 *Param = StrnCatGrow(Param, NULL, ((SHELL_PARAM_PACKAGE*)Node1)->Name, 0);
2440 }
2441 return (EFI_DEVICE_ERROR);
2442 }
2443 }
2444 }
2445 return (EFI_SUCCESS);
2446 }
2447
2448 /**
2449 This is a find and replace function. Upon successful return the NewString is a copy of
2450 SourceString with each instance of FindTarget replaced with ReplaceWith.
2451
2452 If SourceString and NewString overlap the behavior is undefined.
2453
2454 If the string would grow bigger than NewSize it will halt and return error.
2455
2456 @param[in] SourceString The string with source buffer.
2457 @param[in, out] NewString The string with resultant buffer.
2458 @param[in] NewSize The size in bytes of NewString.
2459 @param[in] FindTarget The string to look for.
2460 @param[in] ReplaceWith The string to replace FindTarget with.
2461 @param[in] SkipPreCarrot If TRUE will skip a FindTarget that has a '^'
2462 immediately before it.
2463 @param[in] ParameterReplacing If TRUE will add "" around items with spaces.
2464
2465 @retval EFI_INVALID_PARAMETER SourceString was NULL.
2466 @retval EFI_INVALID_PARAMETER NewString was NULL.
2467 @retval EFI_INVALID_PARAMETER FindTarget was NULL.
2468 @retval EFI_INVALID_PARAMETER ReplaceWith was NULL.
2469 @retval EFI_INVALID_PARAMETER FindTarget had length < 1.
2470 @retval EFI_INVALID_PARAMETER SourceString had length < 1.
2471 @retval EFI_BUFFER_TOO_SMALL NewSize was less than the minimum size to hold
2472 the new string (truncation occurred).
2473 @retval EFI_SUCCESS The string was successfully copied with replacement.
2474 **/
2475 EFI_STATUS
2476 EFIAPI
2477 ShellCopySearchAndReplace(
2478 IN CHAR16 CONST *SourceString,
2479 IN OUT CHAR16 *NewString,
2480 IN UINTN NewSize,
2481 IN CONST CHAR16 *FindTarget,
2482 IN CONST CHAR16 *ReplaceWith,
2483 IN CONST BOOLEAN SkipPreCarrot,
2484 IN CONST BOOLEAN ParameterReplacing
2485 )
2486 {
2487 UINTN Size;
2488 CHAR16 *Replace;
2489
2490 if ( (SourceString == NULL)
2491 || (NewString == NULL)
2492 || (FindTarget == NULL)
2493 || (ReplaceWith == NULL)
2494 || (StrLen(FindTarget) < 1)
2495 || (StrLen(SourceString) < 1)
2496 ){
2497 return (EFI_INVALID_PARAMETER);
2498 }
2499 Replace = NULL;
2500 if (StrStr(ReplaceWith, L" ") == NULL || !ParameterReplacing) {
2501 Replace = StrnCatGrow(&Replace, NULL, ReplaceWith, 0);
2502 } else {
2503 Replace = AllocateZeroPool(StrSize(ReplaceWith) + 2*sizeof(CHAR16));
2504 UnicodeSPrint(Replace, StrSize(ReplaceWith) + 2*sizeof(CHAR16), L"\"%s\"", ReplaceWith);
2505 }
2506 if (Replace == NULL) {
2507 return (EFI_OUT_OF_RESOURCES);
2508 }
2509 NewString = SetMem16(NewString, NewSize, CHAR_NULL);
2510 while (*SourceString != CHAR_NULL) {
2511 //
2512 // if we find the FindTarget and either Skip == FALSE or Skip and we
2513 // dont have a carrot do a replace...
2514 //
2515 if (StrnCmp(SourceString, FindTarget, StrLen(FindTarget)) == 0
2516 && ((SkipPreCarrot && *(SourceString-1) != L'^') || !SkipPreCarrot)
2517 ){
2518 SourceString += StrLen(FindTarget);
2519 Size = StrSize(NewString);
2520 if ((Size + (StrLen(Replace)*sizeof(CHAR16))) > NewSize) {
2521 FreePool(Replace);
2522 return (EFI_BUFFER_TOO_SMALL);
2523 }
2524 StrCat(NewString, Replace);
2525 } else {
2526 Size = StrSize(NewString);
2527 if (Size + sizeof(CHAR16) > NewSize) {
2528 FreePool(Replace);
2529 return (EFI_BUFFER_TOO_SMALL);
2530 }
2531 StrnCat(NewString, SourceString, 1);
2532 SourceString++;
2533 }
2534 }
2535 FreePool(Replace);
2536 return (EFI_SUCCESS);
2537 }
2538
2539 /**
2540 Internal worker function to output a string.
2541
2542 This function will output a string to the correct StdOut.
2543
2544 @param[in] String The string to print out.
2545
2546 @retval EFI_SUCCESS The operation was sucessful.
2547 @retval !EFI_SUCCESS The operation failed.
2548 **/
2549 EFI_STATUS
2550 EFIAPI
2551 InternalPrintTo (
2552 IN CONST CHAR16 *String
2553 )
2554 {
2555 UINTN Size;
2556 Size = StrSize(String) - sizeof(CHAR16);
2557 if (Size == 0) {
2558 return (EFI_SUCCESS);
2559 }
2560 if (gEfiShellParametersProtocol != NULL) {
2561 return (gEfiShellProtocol->WriteFile(gEfiShellParametersProtocol->StdOut, &Size, (VOID*)String));
2562 }
2563 if (mEfiShellInterface != NULL) {
2564 //
2565 // Divide in half for old shell. Must be string length not size.
2566 //
2567 Size /= 2;
2568 return (mEfiShellInterface->StdOut->Write(mEfiShellInterface->StdOut, &Size, (VOID*)String));
2569 }
2570 ASSERT(FALSE);
2571 return (EFI_UNSUPPORTED);
2572 }
2573
2574 /**
2575 Print at a specific location on the screen.
2576
2577 This function will move the cursor to a given screen location and print the specified string
2578
2579 If -1 is specified for either the Row or Col the current screen location for BOTH
2580 will be used.
2581
2582 if either Row or Col is out of range for the current console, then ASSERT
2583 if Format is NULL, then ASSERT
2584
2585 In addition to the standard %-based flags as supported by UefiLib Print() this supports
2586 the following additional flags:
2587 %N - Set output attribute to normal
2588 %H - Set output attribute to highlight
2589 %E - Set output attribute to error
2590 %B - Set output attribute to blue color
2591 %V - Set output attribute to green color
2592
2593 Note: The background color is controlled by the shell command cls.
2594
2595 @param[in] Col the column to print at
2596 @param[in] Row the row to print at
2597 @param[in] Format the format string
2598 @param[in] Marker the marker for the variable argument list
2599
2600 @return EFI_SUCCESS The operation was successful.
2601 @return EFI_DEVICE_ERROR The console device reported an error.
2602 **/
2603 EFI_STATUS
2604 EFIAPI
2605 InternalShellPrintWorker(
2606 IN INT32 Col OPTIONAL,
2607 IN INT32 Row OPTIONAL,
2608 IN CONST CHAR16 *Format,
2609 IN VA_LIST Marker
2610 )
2611 {
2612 EFI_STATUS Status;
2613 CHAR16 *ResumeLocation;
2614 CHAR16 *FormatWalker;
2615 UINTN OriginalAttribute;
2616 CHAR16 *mPostReplaceFormat;
2617 CHAR16 *mPostReplaceFormat2;
2618
2619 mPostReplaceFormat = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
2620 mPostReplaceFormat2 = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
2621
2622 if (mPostReplaceFormat == NULL || mPostReplaceFormat2 == NULL) {
2623 SHELL_FREE_NON_NULL(mPostReplaceFormat);
2624 SHELL_FREE_NON_NULL(mPostReplaceFormat2);
2625 return (EFI_OUT_OF_RESOURCES);
2626 }
2627
2628 Status = EFI_SUCCESS;
2629 OriginalAttribute = gST->ConOut->Mode->Attribute;
2630
2631 //
2632 // Back and forth each time fixing up 1 of our flags...
2633 //
2634 Status = ShellCopySearchAndReplace(Format, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%N", L"%%N", FALSE, FALSE);
2635 ASSERT_EFI_ERROR(Status);
2636 Status = ShellCopySearchAndReplace(mPostReplaceFormat, mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), L"%E", L"%%E", FALSE, FALSE);
2637 ASSERT_EFI_ERROR(Status);
2638 Status = ShellCopySearchAndReplace(mPostReplaceFormat2, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%H", L"%%H", FALSE, FALSE);
2639 ASSERT_EFI_ERROR(Status);
2640 Status = ShellCopySearchAndReplace(mPostReplaceFormat, mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), L"%B", L"%%B", FALSE, FALSE);
2641 ASSERT_EFI_ERROR(Status);
2642 Status = ShellCopySearchAndReplace(mPostReplaceFormat2, mPostReplaceFormat, PcdGet16 (PcdShellPrintBufferSize), L"%V", L"%%V", FALSE, FALSE);
2643 ASSERT_EFI_ERROR(Status);
2644
2645 //
2646 // Use the last buffer from replacing to print from...
2647 //
2648 UnicodeVSPrint (mPostReplaceFormat2, PcdGet16 (PcdShellPrintBufferSize), mPostReplaceFormat, Marker);
2649
2650 if (Col != -1 && Row != -1) {
2651 Status = gST->ConOut->SetCursorPosition(gST->ConOut, Col, Row);
2652 }
2653
2654 FormatWalker = mPostReplaceFormat2;
2655 while (*FormatWalker != CHAR_NULL) {
2656 //
2657 // Find the next attribute change request
2658 //
2659 ResumeLocation = StrStr(FormatWalker, L"%");
2660 if (ResumeLocation != NULL) {
2661 *ResumeLocation = CHAR_NULL;
2662 }
2663 //
2664 // print the current FormatWalker string
2665 //
2666 if (StrLen(FormatWalker)>0) {
2667 Status = InternalPrintTo(FormatWalker);
2668 if (EFI_ERROR(Status)) {
2669 break;
2670 }
2671 }
2672
2673 //
2674 // update the attribute
2675 //
2676 if (ResumeLocation != NULL) {
2677 if (*(ResumeLocation-1) == L'^') {
2678 //
2679 // Print a simple '%' symbol
2680 //
2681 Status = InternalPrintTo(L"%");
2682 ResumeLocation = ResumeLocation - 1;
2683 } else {
2684 switch (*(ResumeLocation+1)) {
2685 case (L'N'):
2686 gST->ConOut->SetAttribute(gST->ConOut, OriginalAttribute);
2687 break;
2688 case (L'E'):
2689 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_YELLOW, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2690 break;
2691 case (L'H'):
2692 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_WHITE, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2693 break;
2694 case (L'B'):
2695 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_BLUE, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2696 break;
2697 case (L'V'):
2698 gST->ConOut->SetAttribute(gST->ConOut, EFI_TEXT_ATTR(EFI_GREEN, ((OriginalAttribute&(BIT4|BIT5|BIT6))>>4)));
2699 break;
2700 default:
2701 //
2702 // Print a simple '%' symbol
2703 //
2704 Status = InternalPrintTo(L"%");
2705 if (EFI_ERROR(Status)) {
2706 break;
2707 }
2708 ResumeLocation = ResumeLocation - 1;
2709 break;
2710 }
2711 }
2712 } else {
2713 //
2714 // reset to normal now...
2715 //
2716 break;
2717 }
2718
2719 //
2720 // update FormatWalker to Resume + 2 (skip the % and the indicator)
2721 //
2722 FormatWalker = ResumeLocation + 2;
2723 }
2724
2725 gST->ConOut->SetAttribute(gST->ConOut, OriginalAttribute);
2726
2727 SHELL_FREE_NON_NULL(mPostReplaceFormat);
2728 SHELL_FREE_NON_NULL(mPostReplaceFormat2);
2729 return (Status);
2730 }
2731
2732 /**
2733 Print at a specific location on the screen.
2734
2735 This function will move the cursor to a given screen location and print the specified string.
2736
2737 If -1 is specified for either the Row or Col the current screen location for BOTH
2738 will be used.
2739
2740 If either Row or Col is out of range for the current console, then ASSERT.
2741 If Format is NULL, then ASSERT.
2742
2743 In addition to the standard %-based flags as supported by UefiLib Print() this supports
2744 the following additional flags:
2745 %N - Set output attribute to normal
2746 %H - Set output attribute to highlight
2747 %E - Set output attribute to error
2748 %B - Set output attribute to blue color
2749 %V - Set output attribute to green color
2750
2751 Note: The background color is controlled by the shell command cls.
2752
2753 @param[in] Col the column to print at
2754 @param[in] Row the row to print at
2755 @param[in] Format the format string
2756 @param[in] ... The variable argument list.
2757
2758 @return EFI_SUCCESS The printing was successful.
2759 @return EFI_DEVICE_ERROR The console device reported an error.
2760 **/
2761 EFI_STATUS
2762 EFIAPI
2763 ShellPrintEx(
2764 IN INT32 Col OPTIONAL,
2765 IN INT32 Row OPTIONAL,
2766 IN CONST CHAR16 *Format,
2767 ...
2768 )
2769 {
2770 VA_LIST Marker;
2771 EFI_STATUS RetVal;
2772 if (Format == NULL) {
2773 return (EFI_INVALID_PARAMETER);
2774 }
2775 VA_START (Marker, Format);
2776 RetVal = InternalShellPrintWorker(Col, Row, Format, Marker);
2777 VA_END(Marker);
2778 return(RetVal);
2779 }
2780
2781 /**
2782 Print at a specific location on the screen.
2783
2784 This function will move the cursor to a given screen location and print the specified string.
2785
2786 If -1 is specified for either the Row or Col the current screen location for BOTH
2787 will be used.
2788
2789 If either Row or Col is out of range for the current console, then ASSERT.
2790 If Format is NULL, then ASSERT.
2791
2792 In addition to the standard %-based flags as supported by UefiLib Print() this supports
2793 the following additional flags:
2794 %N - Set output attribute to normal.
2795 %H - Set output attribute to highlight.
2796 %E - Set output attribute to error.
2797 %B - Set output attribute to blue color.
2798 %V - Set output attribute to green color.
2799
2800 Note: The background color is controlled by the shell command cls.
2801
2802 @param[in] Col The column to print at.
2803 @param[in] Row The row to print at.
2804 @param[in] Language The language of the string to retrieve. If this parameter
2805 is NULL, then the current platform language is used.
2806 @param[in] HiiFormatStringId The format string Id for getting from Hii.
2807 @param[in] HiiFormatHandle The format string Handle for getting from Hii.
2808 @param[in] ... The variable argument list.
2809
2810 @return EFI_SUCCESS The printing was successful.
2811 @return EFI_DEVICE_ERROR The console device reported an error.
2812 **/
2813 EFI_STATUS
2814 EFIAPI
2815 ShellPrintHiiEx(
2816 IN INT32 Col OPTIONAL,
2817 IN INT32 Row OPTIONAL,
2818 IN CONST CHAR8 *Language OPTIONAL,
2819 IN CONST EFI_STRING_ID HiiFormatStringId,
2820 IN CONST EFI_HANDLE HiiFormatHandle,
2821 ...
2822 )
2823 {
2824 VA_LIST Marker;
2825 CHAR16 *HiiFormatString;
2826 EFI_STATUS RetVal;
2827
2828 VA_START (Marker, HiiFormatHandle);
2829 HiiFormatString = HiiGetString(HiiFormatHandle, HiiFormatStringId, Language);
2830 ASSERT(HiiFormatString != NULL);
2831
2832 RetVal = InternalShellPrintWorker(Col, Row, HiiFormatString, Marker);
2833
2834 SHELL_FREE_NON_NULL(HiiFormatString);
2835 VA_END(Marker);
2836
2837 return (RetVal);
2838 }
2839
2840 /**
2841 Function to determine if a given filename represents a file or a directory.
2842
2843 @param[in] DirName Path to directory to test.
2844
2845 @retval EFI_SUCCESS The Path represents a directory
2846 @retval EFI_NOT_FOUND The Path does not represent a directory
2847 @return other The path failed to open
2848 **/
2849 EFI_STATUS
2850 EFIAPI
2851 ShellIsDirectory(
2852 IN CONST CHAR16 *DirName
2853 )
2854 {
2855 EFI_STATUS Status;
2856 SHELL_FILE_HANDLE Handle;
2857 CHAR16 *TempLocation;
2858 CHAR16 *TempLocation2;
2859
2860 ASSERT(DirName != NULL);
2861
2862 Handle = NULL;
2863 TempLocation = NULL;
2864
2865 Status = ShellOpenFileByName(DirName, &Handle, EFI_FILE_MODE_READ, 0);
2866 if (EFI_ERROR(Status)) {
2867 //
2868 // try good logic first.
2869 //
2870 if (gEfiShellProtocol != NULL) {
2871 TempLocation = StrnCatGrow(&TempLocation, NULL, DirName, 0);
2872 TempLocation2 = StrStr(TempLocation, L":");
2873 if (TempLocation2 != NULL && StrLen(StrStr(TempLocation, L":")) == 2) {
2874 *(TempLocation2+1) = CHAR_NULL;
2875 }
2876 if (gEfiShellProtocol->GetDevicePathFromMap(TempLocation) != NULL) {
2877 FreePool(TempLocation);
2878 return (EFI_SUCCESS);
2879 }
2880 FreePool(TempLocation);
2881 } else {
2882 //
2883 // probably a map name?!?!!?
2884 //
2885 TempLocation = StrStr(DirName, L"\\");
2886 if (TempLocation != NULL && *(TempLocation+1) == CHAR_NULL) {
2887 return (EFI_SUCCESS);
2888 }
2889 }
2890 return (Status);
2891 }
2892
2893 if (FileHandleIsDirectory(Handle) == EFI_SUCCESS) {
2894 ShellCloseFile(&Handle);
2895 return (EFI_SUCCESS);
2896 }
2897 ShellCloseFile(&Handle);
2898 return (EFI_NOT_FOUND);
2899 }
2900
2901 /**
2902 Function to determine if a given filename represents a file.
2903
2904 @param[in] Name Path to file to test.
2905
2906 @retval EFI_SUCCESS The Path represents a file.
2907 @retval EFI_NOT_FOUND The Path does not represent a file.
2908 @retval other The path failed to open.
2909 **/
2910 EFI_STATUS
2911 EFIAPI
2912 ShellIsFile(
2913 IN CONST CHAR16 *Name
2914 )
2915 {
2916 EFI_STATUS Status;
2917 SHELL_FILE_HANDLE Handle;
2918
2919 ASSERT(Name != NULL);
2920
2921 Handle = NULL;
2922
2923 Status = ShellOpenFileByName(Name, &Handle, EFI_FILE_MODE_READ, 0);
2924 if (EFI_ERROR(Status)) {
2925 return (Status);
2926 }
2927
2928 if (FileHandleIsDirectory(Handle) != EFI_SUCCESS) {
2929 ShellCloseFile(&Handle);
2930 return (EFI_SUCCESS);
2931 }
2932 ShellCloseFile(&Handle);
2933 return (EFI_NOT_FOUND);
2934 }
2935
2936 /**
2937 Function to determine if a given filename represents a file.
2938
2939 This will search the CWD and then the Path.
2940
2941 If Name is NULL, then ASSERT.
2942
2943 @param[in] Name Path to file to test.
2944
2945 @retval EFI_SUCCESS The Path represents a file.
2946 @retval EFI_NOT_FOUND The Path does not represent a file.
2947 @retval other The path failed to open.
2948 **/
2949 EFI_STATUS
2950 EFIAPI
2951 ShellIsFileInPath(
2952 IN CONST CHAR16 *Name
2953 )
2954 {
2955 CHAR16 *NewName;
2956 EFI_STATUS Status;
2957
2958 if (!EFI_ERROR(ShellIsFile(Name))) {
2959 return (EFI_SUCCESS);
2960 }
2961
2962 NewName = ShellFindFilePath(Name);
2963 if (NewName == NULL) {
2964 return (EFI_NOT_FOUND);
2965 }
2966 Status = ShellIsFile(NewName);
2967 FreePool(NewName);
2968 return (Status);
2969 }
2970
2971 /**
2972 Function to determine whether a string is decimal or hex representation of a number
2973 and return the number converted from the string.
2974
2975 @param[in] String String representation of a number
2976
2977 @return the number
2978 @retval (UINTN)(-1) An error ocurred.
2979 **/
2980 UINTN
2981 EFIAPI
2982 ShellStrToUintn(
2983 IN CONST CHAR16 *String
2984 )
2985 {
2986 UINT64 RetVal;
2987 BOOLEAN Hex;
2988
2989 Hex = FALSE;
2990
2991 if (!InternalShellIsHexOrDecimalNumber(String, Hex, TRUE)) {
2992 Hex = TRUE;
2993 }
2994
2995 if (!EFI_ERROR(ShellConvertStringToUint64(String, &RetVal, Hex, TRUE))) {
2996 return ((UINTN)RetVal);
2997 }
2998 return ((UINTN)(-1));
2999 }
3000
3001 /**
3002 Safely append with automatic string resizing given length of Destination and
3003 desired length of copy from Source.
3004
3005 append the first D characters of Source to the end of Destination, where D is
3006 the lesser of Count and the StrLen() of Source. If appending those D characters
3007 will fit within Destination (whose Size is given as CurrentSize) and
3008 still leave room for a NULL terminator, then those characters are appended,
3009 starting at the original terminating NULL of Destination, and a new terminating
3010 NULL is appended.
3011
3012 If appending D characters onto Destination will result in a overflow of the size
3013 given in CurrentSize the string will be grown such that the copy can be performed
3014 and CurrentSize will be updated to the new size.
3015
3016 If Source is NULL, there is nothing to append, just return the current buffer in
3017 Destination.
3018
3019 if Destination is NULL, then ASSERT()
3020 if Destination's current length (including NULL terminator) is already more then
3021 CurrentSize, then ASSERT()
3022
3023 @param[in, out] Destination The String to append onto
3024 @param[in, out] CurrentSize on call the number of bytes in Destination. On
3025 return possibly the new size (still in bytes). if NULL
3026 then allocate whatever is needed.
3027 @param[in] Source The String to append from
3028 @param[in] Count Maximum number of characters to append. if 0 then
3029 all are appended.
3030
3031 @return Destination return the resultant string.
3032 **/
3033 CHAR16*
3034 EFIAPI
3035 StrnCatGrow (
3036 IN OUT CHAR16 **Destination,
3037 IN OUT UINTN *CurrentSize,
3038 IN CONST CHAR16 *Source,
3039 IN UINTN Count
3040 )
3041 {
3042 UINTN DestinationStartSize;
3043 UINTN NewSize;
3044
3045 //
3046 // ASSERTs
3047 //
3048 ASSERT(Destination != NULL);
3049
3050 //
3051 // If there's nothing to do then just return Destination
3052 //
3053 if (Source == NULL) {
3054 return (*Destination);
3055 }
3056
3057 //
3058 // allow for un-initialized pointers, based on size being 0
3059 //
3060 if (CurrentSize != NULL && *CurrentSize == 0) {
3061 *Destination = NULL;
3062 }
3063
3064 //
3065 // allow for NULL pointers address as Destination
3066 //
3067 if (*Destination != NULL) {
3068 ASSERT(CurrentSize != 0);
3069 DestinationStartSize = StrSize(*Destination);
3070 ASSERT(DestinationStartSize <= *CurrentSize);
3071 } else {
3072 DestinationStartSize = 0;
3073 // ASSERT(*CurrentSize == 0);
3074 }
3075
3076 //
3077 // Append all of Source?
3078 //
3079 if (Count == 0) {
3080 Count = StrLen(Source);
3081 }
3082
3083 //
3084 // Test and grow if required
3085 //
3086 if (CurrentSize != NULL) {
3087 NewSize = *CurrentSize;
3088 while (NewSize < (DestinationStartSize + (Count*sizeof(CHAR16)))) {
3089 NewSize += 2 * Count * sizeof(CHAR16);
3090 }
3091 *Destination = ReallocatePool(*CurrentSize, NewSize, *Destination);
3092 ASSERT(*Destination != NULL);
3093 *CurrentSize = NewSize;
3094 } else {
3095 *Destination = AllocateZeroPool((Count+1)*sizeof(CHAR16));
3096 ASSERT(*Destination != NULL);
3097 }
3098
3099 //
3100 // Now use standard StrnCat on a big enough buffer
3101 //
3102 if (*Destination == NULL) {
3103 return (NULL);
3104 }
3105 return StrnCat(*Destination, Source, Count);
3106 }
3107
3108 /**
3109 Prompt the user and return the resultant answer to the requestor.
3110
3111 This function will display the requested question on the shell prompt and then
3112 wait for an apropriate answer to be input from the console.
3113
3114 if the SHELL_PROMPT_REQUEST_TYPE is SHELL_PROMPT_REQUEST_TYPE_YESNO, ShellPromptResponseTypeQuitContinue
3115 or SHELL_PROMPT_REQUEST_TYPE_YESNOCANCEL then *Response is of type SHELL_PROMPT_RESPONSE.
3116
3117 if the SHELL_PROMPT_REQUEST_TYPE is ShellPromptResponseTypeFreeform then *Response is of type
3118 CHAR16*.
3119
3120 In either case *Response must be callee freed if Response was not NULL;
3121
3122 @param Type What type of question is asked. This is used to filter the input
3123 to prevent invalid answers to question.
3124 @param Prompt Pointer to string prompt to use to request input.
3125 @param Response Pointer to Response which will be populated upon return.
3126
3127 @retval EFI_SUCCESS The operation was sucessful.
3128 @retval EFI_UNSUPPORTED The operation is not supported as requested.
3129 @retval EFI_INVALID_PARAMETER A parameter was invalid.
3130 @return other The operation failed.
3131 **/
3132 EFI_STATUS
3133 EFIAPI
3134 ShellPromptForResponse (
3135 IN SHELL_PROMPT_REQUEST_TYPE Type,
3136 IN CHAR16 *Prompt OPTIONAL,
3137 IN OUT VOID **Response OPTIONAL
3138 )
3139 {
3140 EFI_STATUS Status;
3141 EFI_INPUT_KEY Key;
3142 UINTN EventIndex;
3143 SHELL_PROMPT_RESPONSE *Resp;
3144 UINTN Size;
3145 CHAR16 *Buffer;
3146
3147 Status = EFI_UNSUPPORTED;
3148 Resp = NULL;
3149 Buffer = NULL;
3150 Size = 0;
3151 if (Type != ShellPromptResponseTypeFreeform) {
3152 Resp = (SHELL_PROMPT_RESPONSE*)AllocateZeroPool(sizeof(SHELL_PROMPT_RESPONSE));
3153 if (Resp == NULL) {
3154 return (EFI_OUT_OF_RESOURCES);
3155 }
3156 }
3157
3158 switch(Type) {
3159 case ShellPromptResponseTypeQuitContinue:
3160 if (Prompt != NULL) {
3161 ShellPrintEx(-1, -1, L"%s", Prompt);
3162 }
3163 //
3164 // wait for valid response
3165 //
3166 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3167 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3168 ASSERT_EFI_ERROR(Status);
3169 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3170 if (Key.UnicodeChar == L'Q' || Key.UnicodeChar ==L'q') {
3171 *Resp = ShellPromptResponseQuit;
3172 } else {
3173 *Resp = ShellPromptResponseContinue;
3174 }
3175 break;
3176 case ShellPromptResponseTypeYesNoCancel:
3177 if (Prompt != NULL) {
3178 ShellPrintEx(-1, -1, L"%s", Prompt);
3179 }
3180 //
3181 // wait for valid response
3182 //
3183 *Resp = ShellPromptResponseMax;
3184 while (*Resp == ShellPromptResponseMax) {
3185 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3186 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3187 ASSERT_EFI_ERROR(Status);
3188 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3189 switch (Key.UnicodeChar) {
3190 case L'Y':
3191 case L'y':
3192 *Resp = ShellPromptResponseYes;
3193 break;
3194 case L'N':
3195 case L'n':
3196 *Resp = ShellPromptResponseNo;
3197 break;
3198 case L'C':
3199 case L'c':
3200 *Resp = ShellPromptResponseCancel;
3201 break;
3202 }
3203 }
3204 break; case ShellPromptResponseTypeYesNoAllCancel:
3205 if (Prompt != NULL) {
3206 ShellPrintEx(-1, -1, L"%s", Prompt);
3207 }
3208 //
3209 // wait for valid response
3210 //
3211 *Resp = ShellPromptResponseMax;
3212 while (*Resp == ShellPromptResponseMax) {
3213 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3214 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3215 ASSERT_EFI_ERROR(Status);
3216 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3217 switch (Key.UnicodeChar) {
3218 case L'Y':
3219 case L'y':
3220 *Resp = ShellPromptResponseYes;
3221 break;
3222 case L'N':
3223 case L'n':
3224 *Resp = ShellPromptResponseNo;
3225 break;
3226 case L'A':
3227 case L'a':
3228 *Resp = ShellPromptResponseAll;
3229 break;
3230 case L'C':
3231 case L'c':
3232 *Resp = ShellPromptResponseCancel;
3233 break;
3234 }
3235 }
3236 break;
3237 case ShellPromptResponseTypeEnterContinue:
3238 case ShellPromptResponseTypeAnyKeyContinue:
3239 if (Prompt != NULL) {
3240 ShellPrintEx(-1, -1, L"%s", Prompt);
3241 }
3242 //
3243 // wait for valid response
3244 //
3245 *Resp = ShellPromptResponseMax;
3246 while (*Resp == ShellPromptResponseMax) {
3247 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3248 if (Type == ShellPromptResponseTypeEnterContinue) {
3249 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3250 ASSERT_EFI_ERROR(Status);
3251 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3252 if (Key.UnicodeChar == CHAR_CARRIAGE_RETURN) {
3253 *Resp = ShellPromptResponseContinue;
3254 break;
3255 }
3256 }
3257 if (Type == ShellPromptResponseTypeAnyKeyContinue) {
3258 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3259 ASSERT_EFI_ERROR(Status);
3260 *Resp = ShellPromptResponseContinue;
3261 break;
3262 }
3263 }
3264 break;
3265 case ShellPromptResponseTypeYesNo:
3266 if (Prompt != NULL) {
3267 ShellPrintEx(-1, -1, L"%s", Prompt);
3268 }
3269 //
3270 // wait for valid response
3271 //
3272 *Resp = ShellPromptResponseMax;
3273 while (*Resp == ShellPromptResponseMax) {
3274 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3275 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3276 ASSERT_EFI_ERROR(Status);
3277 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3278 switch (Key.UnicodeChar) {
3279 case L'Y':
3280 case L'y':
3281 *Resp = ShellPromptResponseYes;
3282 break;
3283 case L'N':
3284 case L'n':
3285 *Resp = ShellPromptResponseNo;
3286 break;
3287 }
3288 }
3289 break;
3290 case ShellPromptResponseTypeFreeform:
3291 if (Prompt != NULL) {
3292 ShellPrintEx(-1, -1, L"%s", Prompt);
3293 }
3294 while(1) {
3295 gBS->WaitForEvent (1, &gST->ConIn->WaitForKey, &EventIndex);
3296 Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);
3297 ASSERT_EFI_ERROR(Status);
3298 ShellPrintEx(-1, -1, L"%c", Key.UnicodeChar);
3299 if (Key.UnicodeChar == CHAR_CARRIAGE_RETURN) {
3300 break;
3301 }
3302 ASSERT((Buffer == NULL && Size == 0) || (Buffer != NULL));
3303 StrnCatGrow(&Buffer, &Size, &Key.UnicodeChar, 1);
3304 }
3305 break;
3306 //
3307 // This is the location to add new prompt types.
3308 //
3309 default:
3310 ASSERT(FALSE);
3311 }
3312
3313 if (Response != NULL) {
3314 if (Resp != NULL) {
3315 *Response = Resp;
3316 } else if (Buffer != NULL) {
3317 *Response = Buffer;
3318 }
3319 } else {
3320 if (Resp != NULL) {
3321 FreePool(Resp);
3322 }
3323 if (Buffer != NULL) {
3324 FreePool(Buffer);
3325 }
3326 }
3327
3328 ShellPrintEx(-1, -1, L"\r\n");
3329 return (Status);
3330 }
3331
3332 /**
3333 Prompt the user and return the resultant answer to the requestor.
3334
3335 This function is the same as ShellPromptForResponse, except that the prompt is
3336 automatically pulled from HII.
3337
3338 @param Type What type of question is asked. This is used to filter the input
3339 to prevent invalid answers to question.
3340 @param[in] HiiFormatStringId The format string Id for getting from Hii.
3341 @param[in] HiiFormatHandle The format string Handle for getting from Hii.
3342 @param Response Pointer to Response which will be populated upon return.
3343
3344 @retval EFI_SUCCESS the operation was sucessful.
3345 @return other the operation failed.
3346
3347 @sa ShellPromptForResponse
3348 **/
3349 EFI_STATUS
3350 EFIAPI
3351 ShellPromptForResponseHii (
3352 IN SHELL_PROMPT_REQUEST_TYPE Type,
3353 IN CONST EFI_STRING_ID HiiFormatStringId,
3354 IN CONST EFI_HANDLE HiiFormatHandle,
3355 IN OUT VOID **Response
3356 )
3357 {
3358 CHAR16 *Prompt;
3359 EFI_STATUS Status;
3360
3361 Prompt = HiiGetString(HiiFormatHandle, HiiFormatStringId, NULL);
3362 Status = ShellPromptForResponse(Type, Prompt, Response);
3363 FreePool(Prompt);
3364 return (Status);
3365 }
3366
3367 /**
3368 Function to determin if an entire string is a valid number.
3369
3370 If Hex it must be preceeded with a 0x or has ForceHex, set TRUE.
3371
3372 @param[in] String The string to evaluate.
3373 @param[in] ForceHex TRUE - always assume hex.
3374 @param[in] StopAtSpace TRUE to halt upon finding a space, FALSE to keep going.
3375
3376 @retval TRUE It is all numeric (dec/hex) characters.
3377 @retval FALSE There is a non-numeric character.
3378 **/
3379 BOOLEAN
3380 EFIAPI
3381 InternalShellIsHexOrDecimalNumber (
3382 IN CONST CHAR16 *String,
3383 IN CONST BOOLEAN ForceHex,
3384 IN CONST BOOLEAN StopAtSpace
3385 )
3386 {
3387 BOOLEAN Hex;
3388
3389 //
3390 // chop off a single negative sign
3391 //
3392 if (String != NULL && *String == L'-') {
3393 String++;
3394 }
3395
3396 if (String == NULL) {
3397 return (FALSE);
3398 }
3399
3400 //
3401 // chop leading zeroes
3402 //
3403 while(String != NULL && *String == L'0'){
3404 String++;
3405 }
3406 //
3407 // allow '0x' or '0X', but not 'x' or 'X'
3408 //
3409 if (String != NULL && (*String == L'x' || *String == L'X')) {
3410 if (*(String-1) != L'0') {
3411 //
3412 // we got an x without a preceeding 0
3413 //
3414 return (FALSE);
3415 }
3416 String++;
3417 Hex = TRUE;
3418 } else if (ForceHex) {
3419 Hex = TRUE;
3420 } else {
3421 Hex = FALSE;
3422 }
3423
3424 //
3425 // loop through the remaining characters and use the lib function
3426 //
3427 for ( ; String != NULL && *String != CHAR_NULL && !(StopAtSpace && *String == L' ') ; String++){
3428 if (Hex) {
3429 if (!ShellIsHexaDecimalDigitCharacter(*String)) {
3430 return (FALSE);
3431 }
3432 } else {
3433 if (!ShellIsDecimalDigitCharacter(*String)) {
3434 return (FALSE);
3435 }
3436 }
3437 }
3438
3439 return (TRUE);
3440 }
3441
3442 /**
3443 Function to determine if a given filename exists.
3444
3445 @param[in] Name Path to test.
3446
3447 @retval EFI_SUCCESS The Path represents a file.
3448 @retval EFI_NOT_FOUND The Path does not represent a file.
3449 @retval other The path failed to open.
3450 **/
3451 EFI_STATUS
3452 EFIAPI
3453 ShellFileExists(
3454 IN CONST CHAR16 *Name
3455 )
3456 {
3457 EFI_STATUS Status;
3458 EFI_SHELL_FILE_INFO *List;
3459
3460 ASSERT(Name != NULL);
3461
3462 List = NULL;
3463 Status = ShellOpenFileMetaArg((CHAR16*)Name, EFI_FILE_MODE_READ, &List);
3464 if (EFI_ERROR(Status)) {
3465 return (Status);
3466 }
3467
3468 ShellCloseFileMetaArg(&List);
3469
3470 return (EFI_SUCCESS);
3471 }
3472
3473 /**
3474 Convert a Unicode character to upper case only if
3475 it maps to a valid small-case ASCII character.
3476
3477 This internal function only deal with Unicode character
3478 which maps to a valid small-case ASCII character, i.e.
3479 L'a' to L'z'. For other Unicode character, the input character
3480 is returned directly.
3481
3482 @param Char The character to convert.
3483
3484 @retval LowerCharacter If the Char is with range L'a' to L'z'.
3485 @retval Unchanged Otherwise.
3486
3487 **/
3488 CHAR16
3489 EFIAPI
3490 InternalShellCharToUpper (
3491 IN CHAR16 Char
3492 )
3493 {
3494 if (Char >= L'a' && Char <= L'z') {
3495 return (CHAR16) (Char - (L'a' - L'A'));
3496 }
3497
3498 return Char;
3499 }
3500
3501 /**
3502 Convert a Unicode character to numerical value.
3503
3504 This internal function only deal with Unicode character
3505 which maps to a valid hexadecimal ASII character, i.e.
3506 L'0' to L'9', L'a' to L'f' or L'A' to L'F'. For other
3507 Unicode character, the value returned does not make sense.
3508
3509 @param Char The character to convert.
3510
3511 @return The numerical value converted.
3512
3513 **/
3514 UINTN
3515 EFIAPI
3516 InternalShellHexCharToUintn (
3517 IN CHAR16 Char
3518 )
3519 {
3520 if (ShellIsDecimalDigitCharacter (Char)) {
3521 return Char - L'0';
3522 }
3523
3524 return (UINTN) (10 + InternalShellCharToUpper (Char) - L'A');
3525 }
3526
3527 /**
3528 Convert a Null-terminated Unicode hexadecimal string to a value of type UINT64.
3529
3530 This function returns a value of type UINTN by interpreting the contents
3531 of the Unicode string specified by String as a hexadecimal number.
3532 The format of the input Unicode string String is:
3533
3534 [spaces][zeros][x][hexadecimal digits].
3535
3536 The valid hexadecimal digit character is in the range [0-9], [a-f] and [A-F].
3537 The prefix "0x" is optional. Both "x" and "X" is allowed in "0x" prefix.
3538 If "x" appears in the input string, it must be prefixed with at least one 0.
3539 The function will ignore the pad space, which includes spaces or tab characters,
3540 before [zeros], [x] or [hexadecimal digit]. The running zero before [x] or
3541 [hexadecimal digit] will be ignored. Then, the decoding starts after [x] or the
3542 first valid hexadecimal digit. Then, the function stops at the first character that is
3543 a not a valid hexadecimal character or NULL, whichever one comes first.
3544
3545 If String has only pad spaces, then zero is returned.
3546 If String has no leading pad spaces, leading zeros or valid hexadecimal digits,
3547 then zero is returned.
3548
3549 @param[in] String A pointer to a Null-terminated Unicode string.
3550 @param[out] Value Upon a successful return the value of the conversion.
3551 @param[in] StopAtSpace FALSE to skip spaces.
3552
3553 @retval EFI_SUCCESS The conversion was successful.
3554 @retval EFI_INVALID_PARAMETER A parameter was NULL or invalid.
3555 @retval EFI_DEVICE_ERROR An overflow occured.
3556 **/
3557 EFI_STATUS
3558 EFIAPI
3559 InternalShellStrHexToUint64 (
3560 IN CONST CHAR16 *String,
3561 OUT UINT64 *Value,
3562 IN CONST BOOLEAN StopAtSpace
3563 )
3564 {
3565 UINT64 Result;
3566
3567 if (String == NULL || StrSize(String) == 0 || Value == NULL) {
3568 return (EFI_INVALID_PARAMETER);
3569 }
3570
3571 //
3572 // Ignore the pad spaces (space or tab)
3573 //
3574 while ((*String == L' ') || (*String == L'\t')) {
3575 String++;
3576 }
3577
3578 //
3579 // Ignore leading Zeros after the spaces
3580 //
3581 while (*String == L'0') {
3582 String++;
3583 }
3584
3585 if (InternalShellCharToUpper (*String) == L'X') {
3586 if (*(String - 1) != L'0') {
3587 return 0;
3588 }
3589 //
3590 // Skip the 'X'
3591 //
3592 String++;
3593 }
3594
3595 Result = 0;
3596
3597 //
3598 // Skip spaces if requested
3599 //
3600 while (StopAtSpace && *String == L' ') {
3601 String++;
3602 }
3603
3604 while (ShellIsHexaDecimalDigitCharacter (*String)) {
3605 //
3606 // If the Hex Number represented by String overflows according
3607 // to the range defined by UINTN, then ASSERT().
3608 //
3609 if (!(Result <= (RShiftU64((((UINT64) ~0) - InternalShellHexCharToUintn (*String)), 4)))) {
3610 // if (!(Result <= ((((UINT64) ~0) - InternalShellHexCharToUintn (*String)) >> 4))) {
3611 return (EFI_DEVICE_ERROR);
3612 }
3613
3614 Result = (LShiftU64(Result, 4));
3615 Result += InternalShellHexCharToUintn (*String);
3616 String++;
3617
3618 //
3619 // Skip spaces if requested
3620 //
3621 while (StopAtSpace && *String == L' ') {
3622 String++;
3623 }
3624 }
3625
3626 *Value = Result;
3627 return (EFI_SUCCESS);
3628 }
3629
3630 /**
3631 Convert a Null-terminated Unicode decimal string to a value of
3632 type UINT64.
3633
3634 This function returns a value of type UINT64 by interpreting the contents
3635 of the Unicode string specified by String as a decimal number. The format
3636 of the input Unicode string String is:
3637
3638 [spaces] [decimal digits].
3639
3640 The valid decimal digit character is in the range [0-9]. The
3641 function will ignore the pad space, which includes spaces or
3642 tab characters, before [decimal digits]. The running zero in the
3643 beginning of [decimal digits] will be ignored. Then, the function
3644 stops at the first character that is a not a valid decimal character
3645 or a Null-terminator, whichever one comes first.
3646
3647 If String has only pad spaces, then 0 is returned.
3648 If String has no pad spaces or valid decimal digits,
3649 then 0 is returned.
3650
3651 @param[in] String A pointer to a Null-terminated Unicode string.
3652 @param[out] Value Upon a successful return the value of the conversion.
3653 @param[in] StopAtSpace FALSE to skip spaces.
3654
3655 @retval EFI_SUCCESS The conversion was successful.
3656 @retval EFI_INVALID_PARAMETER A parameter was NULL or invalid.
3657 @retval EFI_DEVICE_ERROR An overflow occured.
3658 **/
3659 EFI_STATUS
3660 EFIAPI
3661 InternalShellStrDecimalToUint64 (
3662 IN CONST CHAR16 *String,
3663 OUT UINT64 *Value,
3664 IN CONST BOOLEAN StopAtSpace
3665 )
3666 {
3667 UINT64 Result;
3668
3669 if (String == NULL || StrSize (String) == 0 || Value == NULL) {
3670 return (EFI_INVALID_PARAMETER);
3671 }
3672
3673 //
3674 // Ignore the pad spaces (space or tab)
3675 //
3676 while ((*String == L' ') || (*String == L'\t')) {
3677 String++;
3678 }
3679
3680 //
3681 // Ignore leading Zeros after the spaces
3682 //
3683 while (*String == L'0') {
3684 String++;
3685 }
3686
3687 Result = 0;
3688
3689 //
3690 // Skip spaces if requested
3691 //
3692 while (StopAtSpace && *String == L' ') {
3693 String++;
3694 }
3695 while (ShellIsDecimalDigitCharacter (*String)) {
3696 //
3697 // If the number represented by String overflows according
3698 // to the range defined by UINT64, then ASSERT().
3699 //
3700
3701 if (!(Result <= (DivU64x32((((UINT64) ~0) - (*String - L'0')),10)))) {
3702 return (EFI_DEVICE_ERROR);
3703 }
3704
3705 Result = MultU64x32(Result, 10) + (*String - L'0');
3706 String++;
3707
3708 //
3709 // Stop at spaces if requested
3710 //
3711 if (StopAtSpace && *String == L' ') {
3712 break;
3713 }
3714 }
3715
3716 *Value = Result;
3717
3718 return (EFI_SUCCESS);
3719 }
3720
3721 /**
3722 Function to verify and convert a string to its numerical value.
3723
3724 If Hex it must be preceeded with a 0x, 0X, or has ForceHex set TRUE.
3725
3726 @param[in] String The string to evaluate.
3727 @param[out] Value Upon a successful return the value of the conversion.
3728 @param[in] ForceHex TRUE - always assume hex.
3729 @param[in] StopAtSpace FALSE to skip spaces.
3730
3731 @retval EFI_SUCCESS The conversion was successful.
3732 @retval EFI_INVALID_PARAMETER String contained an invalid character.
3733 @retval EFI_NOT_FOUND String was a number, but Value was NULL.
3734 **/
3735 EFI_STATUS
3736 EFIAPI
3737 ShellConvertStringToUint64(
3738 IN CONST CHAR16 *String,
3739 OUT UINT64 *Value,
3740 IN CONST BOOLEAN ForceHex,
3741 IN CONST BOOLEAN StopAtSpace
3742 )
3743 {
3744 UINT64 RetVal;
3745 CONST CHAR16 *Walker;
3746 EFI_STATUS Status;
3747 BOOLEAN Hex;
3748
3749 Hex = ForceHex;
3750
3751 if (!InternalShellIsHexOrDecimalNumber(String, Hex, StopAtSpace)) {
3752 if (!Hex) {
3753 Hex = TRUE;
3754 if (!InternalShellIsHexOrDecimalNumber(String, Hex, StopAtSpace)) {
3755 return (EFI_INVALID_PARAMETER);
3756 }
3757 } else {
3758 return (EFI_INVALID_PARAMETER);
3759 }
3760 }
3761
3762 //
3763 // Chop off leading spaces
3764 //
3765 for (Walker = String; Walker != NULL && *Walker != CHAR_NULL && *Walker == L' '; Walker++);
3766
3767 //
3768 // make sure we have something left that is numeric.
3769 //
3770 if (Walker == NULL || *Walker == CHAR_NULL || !InternalShellIsHexOrDecimalNumber(Walker, Hex, StopAtSpace)) {
3771 return (EFI_INVALID_PARAMETER);
3772 }
3773
3774 //
3775 // do the conversion.
3776 //
3777 if (Hex || StrnCmp(Walker, L"0x", 2) == 0 || StrnCmp(Walker, L"0X", 2) == 0){
3778 Status = InternalShellStrHexToUint64(Walker, &RetVal, StopAtSpace);
3779 } else {
3780 Status = InternalShellStrDecimalToUint64(Walker, &RetVal, StopAtSpace);
3781 }
3782
3783 if (Value == NULL && !EFI_ERROR(Status)) {
3784 return (EFI_NOT_FOUND);
3785 }
3786
3787 if (Value != NULL) {
3788 *Value = RetVal;
3789 }
3790
3791 return (Status);
3792 }
3793
3794 /**
3795 Function to determin if an entire string is a valid number.
3796
3797 If Hex it must be preceeded with a 0x or has ForceHex, set TRUE.
3798
3799 @param[in] String The string to evaluate.
3800 @param[in] ForceHex TRUE - always assume hex.
3801 @param[in] StopAtSpace TRUE to halt upon finding a space, FALSE to keep going.
3802
3803 @retval TRUE It is all numeric (dec/hex) characters.
3804 @retval FALSE There is a non-numeric character.
3805 **/
3806 BOOLEAN
3807 EFIAPI
3808 ShellIsHexOrDecimalNumber (
3809 IN CONST CHAR16 *String,
3810 IN CONST BOOLEAN ForceHex,
3811 IN CONST BOOLEAN StopAtSpace
3812 )
3813 {
3814 if (ShellConvertStringToUint64(String, NULL, ForceHex, StopAtSpace) == EFI_NOT_FOUND) {
3815 return (TRUE);
3816 }
3817 return (FALSE);
3818 }
3819
3820 /**
3821 Function to read a single line from a SHELL_FILE_HANDLE. The \n is not included in the returned
3822 buffer. The returned buffer must be callee freed.
3823
3824 If the position upon start is 0, then the Ascii Boolean will be set. This should be
3825 maintained and not changed for all operations with the same file.
3826
3827 @param[in] Handle SHELL_FILE_HANDLE to read from.
3828 @param[in, out] Ascii Boolean value for indicating whether the file is
3829 Ascii (TRUE) or UCS2 (FALSE).
3830
3831 @return The line of text from the file.
3832
3833 @sa ShellFileHandleReadLine
3834 **/
3835 CHAR16*
3836 EFIAPI
3837 ShellFileHandleReturnLine(
3838 IN SHELL_FILE_HANDLE Handle,
3839 IN OUT BOOLEAN *Ascii
3840 )
3841 {
3842 CHAR16 *RetVal;
3843 UINTN Size;
3844 EFI_STATUS Status;
3845
3846 Size = 0;
3847 RetVal = NULL;
3848
3849 Status = ShellFileHandleReadLine(Handle, RetVal, &Size, FALSE, Ascii);
3850 if (Status == EFI_BUFFER_TOO_SMALL) {
3851 RetVal = AllocateZeroPool(Size);
3852 Status = ShellFileHandleReadLine(Handle, RetVal, &Size, FALSE, Ascii);
3853 }
3854 ASSERT_EFI_ERROR(Status);
3855 if (EFI_ERROR(Status) && (RetVal != NULL)) {
3856 FreePool(RetVal);
3857 RetVal = NULL;
3858 }
3859 return (RetVal);
3860 }
3861
3862 /**
3863 Function to read a single line (up to but not including the \n) from a SHELL_FILE_HANDLE.
3864
3865 If the position upon start is 0, then the Ascii Boolean will be set. This should be
3866 maintained and not changed for all operations with the same file.
3867
3868 @param[in] Handle SHELL_FILE_HANDLE to read from.
3869 @param[in, out] Buffer The pointer to buffer to read into.
3870 @param[in, out] Size The pointer to number of bytes in Buffer.
3871 @param[in] Truncate If the buffer is large enough, this has no effect.
3872 If the buffer is is too small and Truncate is TRUE,
3873 the line will be truncated.
3874 If the buffer is is too small and Truncate is FALSE,
3875 then no read will occur.
3876
3877 @param[in, out] Ascii Boolean value for indicating whether the file is
3878 Ascii (TRUE) or UCS2 (FALSE).
3879
3880 @retval EFI_SUCCESS The operation was successful. The line is stored in
3881 Buffer.
3882 @retval EFI_INVALID_PARAMETER Handle was NULL.
3883 @retval EFI_INVALID_PARAMETER Size was NULL.
3884 @retval EFI_BUFFER_TOO_SMALL Size was not large enough to store the line.
3885 Size was updated to the minimum space required.
3886 **/
3887 EFI_STATUS
3888 EFIAPI
3889 ShellFileHandleReadLine(
3890 IN SHELL_FILE_HANDLE Handle,
3891 IN OUT CHAR16 *Buffer,
3892 IN OUT UINTN *Size,
3893 IN BOOLEAN Truncate,
3894 IN OUT BOOLEAN *Ascii
3895 )
3896 {
3897 EFI_STATUS Status;
3898 CHAR16 CharBuffer;
3899 UINTN CharSize;
3900 UINTN CountSoFar;
3901 UINT64 OriginalFilePosition;
3902
3903
3904 if (Handle == NULL
3905 ||Size == NULL
3906 ){
3907 return (EFI_INVALID_PARAMETER);
3908 }
3909 if (Buffer == NULL) {
3910 ASSERT(*Size == 0);
3911 } else {
3912 *Buffer = CHAR_NULL;
3913 }
3914 gEfiShellProtocol->GetFilePosition(Handle, &OriginalFilePosition);
3915 if (OriginalFilePosition == 0) {
3916 CharSize = sizeof(CHAR16);
3917 Status = gEfiShellProtocol->ReadFile(Handle, &CharSize, &CharBuffer);
3918 ASSERT_EFI_ERROR(Status);
3919 if (CharBuffer == gUnicodeFileTag) {
3920 *Ascii = FALSE;
3921 } else {
3922 *Ascii = TRUE;
3923 gEfiShellProtocol->SetFilePosition(Handle, OriginalFilePosition);
3924 }
3925 }
3926
3927 for (CountSoFar = 0;;CountSoFar++){
3928 CharBuffer = 0;
3929 if (*Ascii) {
3930 CharSize = sizeof(CHAR8);
3931 } else {
3932 CharSize = sizeof(CHAR16);
3933 }
3934 Status = gEfiShellProtocol->ReadFile(Handle, &CharSize, &CharBuffer);
3935 if ( EFI_ERROR(Status)
3936 || CharSize == 0
3937 || (CharBuffer == L'\n' && !(*Ascii))
3938 || (CharBuffer == '\n' && *Ascii)
3939 ){
3940 break;
3941 }
3942 //
3943 // if we have space save it...
3944 //
3945 if ((CountSoFar+1)*sizeof(CHAR16) < *Size){
3946 ASSERT(Buffer != NULL);
3947 ((CHAR16*)Buffer)[CountSoFar] = CharBuffer;
3948 ((CHAR16*)Buffer)[CountSoFar+1] = CHAR_NULL;
3949 }
3950 }
3951
3952 //
3953 // if we ran out of space tell when...
3954 //
3955 if ((CountSoFar+1)*sizeof(CHAR16) > *Size){
3956 *Size = (CountSoFar+1)*sizeof(CHAR16);
3957 if (!Truncate) {
3958 gEfiShellProtocol->SetFilePosition(Handle, OriginalFilePosition);
3959 } else {
3960 DEBUG((DEBUG_WARN, "The line was truncated in ShellFileHandleReadLine"));
3961 }
3962 return (EFI_BUFFER_TOO_SMALL);
3963 }
3964 while(Buffer[StrLen(Buffer)-1] == L'\r') {
3965 Buffer[StrLen(Buffer)-1] = CHAR_NULL;
3966 }
3967
3968 return (Status);
3969 }