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