]> git.proxmox.com Git - mirror_edk2.git/blob - ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c
renaming the BaseFileHandleLib to UefiFileHandleLib.
[mirror_edk2.git] / ShellPkg / Library / UefiFileHandleLib / UefiFileHandleLib.c
1 /** @file
2 Provides interface to EFI_FILE_HANDLE functionality.
3
4 Copyright (c) 2006 - 2011, Intel Corporation. All rights reserved. <BR>
5 This program and the accompanying materials
6 are licensed and made available under the terms and conditions of the BSD License
7 which accompanies this distribution. The full text of the license may be found at
8 http://opensource.org/licenses/bsd-license.php
9
10 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
12
13 **/
14
15 #include <Uefi.h>
16
17 #include <Protocol/SimpleFileSystem.h>
18 #include <Protocol/UnicodeCollation.h>
19
20 #include <Guid/FileInfo.h>
21
22 #include <Library/DebugLib.h>
23 #include <Library/MemoryAllocationLib.h>
24 #include <Library/BaseLib.h>
25 #include <Library/BaseMemoryLib.h>
26 #include <Library/FileHandleLib.h>
27 #include <Library/PcdLib.h>
28 #include <Library/PrintLib.h>
29
30 CONST UINT16 gUnicodeFileTag = EFI_UNICODE_BYTE_ORDER_MARK;
31
32 #define MAX_FILE_NAME_LEN 522 // (20 * (6+5+2))+1) unicode characters from EFI FAT spec (doubled for bytes)
33 #define FIND_XXXXX_FILE_BUFFER_SIZE (SIZE_OF_EFI_FILE_INFO + MAX_FILE_NAME_LEN)
34
35 /**
36 This function will retrieve the information about the file for the handle
37 specified and store it in allocated pool memory.
38
39 This function allocates a buffer to store the file's information. It is the
40 caller's responsibility to free the buffer
41
42 @param FileHandle The file handle of the file for which information is
43 being requested.
44
45 @retval NULL information could not be retrieved.
46
47 @return the information about the file
48 **/
49 EFI_FILE_INFO*
50 EFIAPI
51 FileHandleGetInfo (
52 IN EFI_FILE_HANDLE FileHandle
53 )
54 {
55 EFI_FILE_INFO *FileInfo;
56 UINTN FileInfoSize;
57 EFI_STATUS Status;
58
59 if (FileHandle == NULL) {
60 return (NULL);
61 }
62
63 //
64 // Get the required size to allocate
65 //
66 FileInfoSize = 0;
67 FileInfo = NULL;
68 Status = FileHandle->GetInfo(FileHandle,
69 &gEfiFileInfoGuid,
70 &FileInfoSize,
71 NULL);
72 if (Status == EFI_BUFFER_TOO_SMALL){
73 //
74 // error is expected. getting size to allocate
75 //
76 FileInfo = AllocateZeroPool(FileInfoSize);
77 //
78 // now get the information
79 //
80 Status = FileHandle->GetInfo(FileHandle,
81 &gEfiFileInfoGuid,
82 &FileInfoSize,
83 FileInfo);
84 //
85 // if we got an error free the memory and return NULL
86 //
87 if (EFI_ERROR(Status)) {
88 FreePool(FileInfo);
89 return NULL;
90 }
91 }
92 return (FileInfo);
93 }
94
95 /**
96 This function sets the information about the file for the opened handle
97 specified.
98
99 @param[in] FileHandle The file handle of the file for which information
100 is being set.
101
102 @param[in] FileInfo The information to set.
103
104 @retval EFI_SUCCESS The information was set.
105 @retval EFI_INVALID_PARAMETER A parameter was out of range or invalid.
106 @retval EFI_UNSUPPORTED The FileHandle does not support FileInfo.
107 @retval EFI_NO_MEDIA The device has no medium.
108 @retval EFI_DEVICE_ERROR The device reported an error.
109 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
110 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
111 @retval EFI_ACCESS_DENIED The file was opened read only.
112 @retval EFI_VOLUME_FULL The volume is full.
113 **/
114 EFI_STATUS
115 EFIAPI
116 FileHandleSetInfo (
117 IN EFI_FILE_HANDLE FileHandle,
118 IN CONST EFI_FILE_INFO *FileInfo
119 )
120 {
121
122 //
123 // ASSERT if the FileHandle or FileInfo is NULL
124 //
125 ASSERT (FileHandle != NULL);
126 ASSERT (FileInfo != NULL);
127
128 //
129 // Set the info
130 //
131 return (FileHandle->SetInfo(FileHandle,
132 &gEfiFileInfoGuid,
133 (UINTN)FileInfo->Size,
134 (EFI_FILE_INFO*)FileInfo));
135 }
136
137 /**
138 This function reads information from an opened file.
139
140 If FileHandle is not a directory, the function reads the requested number of
141 bytes from the file at the file's current position and returns them in Buffer.
142 If the read goes beyond the end of the file, the read length is truncated to the
143 end of the file. The file's current position is increased by the number of bytes
144 returned. If FileHandle is a directory, the function reads the directory entry
145 at the file's current position and returns the entry in Buffer. If the Buffer
146 is not large enough to hold the current directory entry, then
147 EFI_BUFFER_TOO_SMALL is returned and the current file position is not updated.
148 BufferSize is set to be the size of the buffer needed to read the entry. On
149 success, the current position is updated to the next directory entry. If there
150 are no more directory entries, the read returns a zero-length buffer.
151 EFI_FILE_INFO is the structure returned as the directory entry.
152
153 @param FileHandle the opened file handle
154 @param BufferSize on input the size of buffer in bytes. on return
155 the number of bytes written.
156 @param Buffer the buffer to put read data into.
157
158 @retval EFI_SUCCESS Data was read.
159 @retval EFI_NO_MEDIA The device has no media.
160 @retval EFI_DEVICE_ERROR The device reported an error.
161 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
162 @retval EFI_BUFFER_TO_SMALL Buffer is too small. ReadSize contains required
163 size.
164
165 **/
166 EFI_STATUS
167 EFIAPI
168 FileHandleRead(
169 IN EFI_FILE_HANDLE FileHandle,
170 IN OUT UINTN *BufferSize,
171 OUT VOID *Buffer
172 )
173 {
174 //
175 // ASSERT if FileHandle is NULL
176 //
177 ASSERT (FileHandle != NULL);
178
179 //
180 // Perform the read based on EFI_FILE_PROTOCOL
181 //
182 return (FileHandle->Read(FileHandle, BufferSize, Buffer));
183 }
184
185
186 /**
187 Write data to a file.
188
189 This function writes the specified number of bytes to the file at the current
190 file position. The current file position is advanced the actual number of bytes
191 written, which is returned in BufferSize. Partial writes only occur when there
192 has been a data error during the write attempt (such as "volume space full").
193 The file is automatically grown to hold the data if required. Direct writes to
194 opened directories are not supported.
195
196 @param FileHandle The opened file for writing
197 @param BufferSize on input the number of bytes in Buffer. On output
198 the number of bytes written.
199 @param Buffer the buffer containing data to write is stored.
200
201 @retval EFI_SUCCESS Data was written.
202 @retval EFI_UNSUPPORTED Writes to an open directory are not supported.
203 @retval EFI_NO_MEDIA The device has no media.
204 @retval EFI_DEVICE_ERROR The device reported an error.
205 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
206 @retval EFI_WRITE_PROTECTED The device is write-protected.
207 @retval EFI_ACCESS_DENIED The file was open for read only.
208 @retval EFI_VOLUME_FULL The volume is full.
209 **/
210 EFI_STATUS
211 EFIAPI
212 FileHandleWrite(
213 IN EFI_FILE_HANDLE FileHandle,
214 IN OUT UINTN *BufferSize,
215 IN VOID *Buffer
216 )
217 {
218 //
219 // ASSERT if FileHandle is NULL
220 //
221 ASSERT (FileHandle != NULL);
222 //
223 // Perform the write based on EFI_FILE_PROTOCOL
224 //
225 return (FileHandle->Write(FileHandle, BufferSize, Buffer));
226 }
227
228 /**
229 Close an open file handle.
230
231 This function closes a specified file handle. All "dirty" cached file data is
232 flushed to the device, and the file is closed. In all cases the handle is
233 closed.
234
235 @param FileHandle the file handle to close.
236
237 @retval EFI_SUCCESS the file handle was closed sucessfully.
238 **/
239 EFI_STATUS
240 EFIAPI
241 FileHandleClose (
242 IN EFI_FILE_HANDLE FileHandle
243 )
244 {
245 EFI_STATUS Status;
246 //
247 // ASSERT if FileHandle is NULL
248 //
249 ASSERT (FileHandle != NULL);
250 //
251 // Perform the Close based on EFI_FILE_PROTOCOL
252 //
253 Status = FileHandle->Close(FileHandle);
254 return Status;
255 }
256
257 /**
258 Delete a file and close the handle
259
260 This function closes and deletes a file. In all cases the file handle is closed.
261 If the file cannot be deleted, the warning code EFI_WARN_DELETE_FAILURE is
262 returned, but the handle is still closed.
263
264 @param FileHandle the file handle to delete
265
266 @retval EFI_SUCCESS the file was closed sucessfully
267 @retval EFI_WARN_DELETE_FAILURE the handle was closed, but the file was not
268 deleted
269 @retval INVALID_PARAMETER One of the parameters has an invalid value.
270 **/
271 EFI_STATUS
272 EFIAPI
273 FileHandleDelete (
274 IN EFI_FILE_HANDLE FileHandle
275 )
276 {
277 EFI_STATUS Status;
278 //
279 // ASSERT if FileHandle is NULL
280 //
281 ASSERT (FileHandle != NULL);
282 //
283 // Perform the Delete based on EFI_FILE_PROTOCOL
284 //
285 Status = FileHandle->Delete(FileHandle);
286 return Status;
287 }
288
289 /**
290 Set the current position in a file.
291
292 This function sets the current file position for the handle to the position
293 supplied. With the exception of seeking to position 0xFFFFFFFFFFFFFFFF, only
294 absolute positioning is supported, and seeking past the end of the file is
295 allowed (a subsequent write would grow the file). Seeking to position
296 0xFFFFFFFFFFFFFFFF causes the current position to be set to the end of the file.
297 If FileHandle is a directory, the only position that may be set is zero. This
298 has the effect of starting the read process of the directory entries over.
299
300 @param FileHandle The file handle on which the position is being set
301 @param Position Byte position from begining of file
302
303 @retval EFI_SUCCESS Operation completed sucessfully.
304 @retval EFI_UNSUPPORTED the seek request for non-zero is not valid on
305 directories.
306 @retval INVALID_PARAMETER One of the parameters has an invalid value.
307 **/
308 EFI_STATUS
309 EFIAPI
310 FileHandleSetPosition (
311 IN EFI_FILE_HANDLE FileHandle,
312 IN UINT64 Position
313 )
314 {
315 //
316 // ASSERT if FileHandle is NULL
317 //
318 ASSERT (FileHandle != NULL);
319 //
320 // Perform the SetPosition based on EFI_FILE_PROTOCOL
321 //
322 return (FileHandle->SetPosition(FileHandle, Position));
323 }
324
325 /**
326 Gets a file's current position
327
328 This function retrieves the current file position for the file handle. For
329 directories, the current file position has no meaning outside of the file
330 system driver and as such the operation is not supported. An error is returned
331 if FileHandle is a directory.
332
333 @param FileHandle The open file handle on which to get the position.
334 @param Position Byte position from begining of file.
335
336 @retval EFI_SUCCESS the operation completed sucessfully.
337 @retval INVALID_PARAMETER One of the parameters has an invalid value.
338 @retval EFI_UNSUPPORTED the request is not valid on directories.
339 **/
340 EFI_STATUS
341 EFIAPI
342 FileHandleGetPosition (
343 IN EFI_FILE_HANDLE FileHandle,
344 OUT UINT64 *Position
345 )
346 {
347 if (Position == NULL) {
348 return (EFI_INVALID_PARAMETER);
349 }
350 //
351 // ASSERT if FileHandle is NULL
352 //
353 ASSERT (FileHandle != NULL);
354 //
355 // Perform the GetPosition based on EFI_FILE_PROTOCOL
356 //
357 return (FileHandle->GetPosition(FileHandle, Position));
358 }
359 /**
360 Flushes data on a file
361
362 This function flushes all modified data associated with a file to a device.
363
364 @param FileHandle The file handle on which to flush data
365
366 @retval EFI_SUCCESS The data was flushed.
367 @retval EFI_NO_MEDIA The device has no media.
368 @retval EFI_DEVICE_ERROR The device reported an error.
369 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
370 @retval EFI_WRITE_PROTECTED The file or medium is write protected.
371 @retval EFI_ACCESS_DENIED The file was opened for read only.
372 **/
373 EFI_STATUS
374 EFIAPI
375 FileHandleFlush (
376 IN EFI_FILE_HANDLE FileHandle
377 )
378 {
379 //
380 // ASSERT if FileHandle is NULL
381 //
382 ASSERT (FileHandle != NULL);
383 //
384 // Perform the Flush based on EFI_FILE_PROTOCOL
385 //
386 return (FileHandle->Flush(FileHandle));
387 }
388
389 /**
390 function to determine if a given handle is a directory handle
391
392 if DirHandle is NULL then ASSERT()
393
394 open the file information on the DirHandle and verify that the Attribute
395 includes EFI_FILE_DIRECTORY bit set.
396
397 @param DirHandle Handle to open file
398
399 @retval EFI_SUCCESS DirHandle is a directory
400 @retval EFI_INVALID_PARAMETER DirHandle did not have EFI_FILE_INFO available
401 @retval EFI_NOT_FOUND DirHandle is not a directory
402 **/
403 EFI_STATUS
404 EFIAPI
405 FileHandleIsDirectory (
406 IN EFI_FILE_HANDLE DirHandle
407 )
408 {
409 EFI_FILE_INFO *DirInfo;
410
411 //
412 // ASSERT if DirHandle is NULL
413 //
414 ASSERT(DirHandle != NULL);
415
416 //
417 // get the file information for DirHandle
418 //
419 DirInfo = FileHandleGetInfo (DirHandle);
420
421 //
422 // Parse DirInfo
423 //
424 if (DirInfo == NULL) {
425 //
426 // We got nothing...
427 //
428 return (EFI_INVALID_PARAMETER);
429 }
430 if ((DirInfo->Attribute & EFI_FILE_DIRECTORY) == 0) {
431 //
432 // Attributes say this is not a directory
433 //
434 FreePool (DirInfo);
435 return (EFI_NOT_FOUND);
436 }
437 //
438 // all good...
439 //
440 FreePool (DirInfo);
441 return (EFI_SUCCESS);
442 }
443
444 /**
445 Retrieves the first file from a directory
446
447 This function opens a directory and gets the first file's info in the
448 directory. Caller can use FileHandleFindNextFile() to get other files. When
449 complete the caller is responsible for calling FreePool() on Buffer.
450
451 @param DirHandle The file handle of the directory to search
452 @param Buffer Pointer to buffer for file's information
453
454 @retval EFI_SUCCESS Found the first file.
455 @retval EFI_NOT_FOUND Cannot find the directory.
456 @retval EFI_NO_MEDIA The device has no media.
457 @retval EFI_DEVICE_ERROR The device reported an error.
458 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
459 @return Others status of FileHandleGetInfo, FileHandleSetPosition,
460 or FileHandleRead
461 **/
462 EFI_STATUS
463 EFIAPI
464 FileHandleFindFirstFile (
465 IN EFI_FILE_HANDLE DirHandle,
466 OUT EFI_FILE_INFO **Buffer
467 )
468 {
469 EFI_STATUS Status;
470 UINTN BufferSize;
471
472 if (Buffer == NULL || DirHandle == NULL) {
473 return (EFI_INVALID_PARAMETER);
474 }
475
476 //
477 // verify that DirHandle is a directory
478 //
479 Status = FileHandleIsDirectory(DirHandle);
480 if (EFI_ERROR(Status)) {
481 return (Status);
482 }
483
484 //
485 // Allocate a buffer sized to struct size + enough for the string at the end
486 //
487 BufferSize = FIND_XXXXX_FILE_BUFFER_SIZE;
488 *Buffer = AllocateZeroPool(BufferSize);
489 if (*Buffer == NULL){
490 return (EFI_OUT_OF_RESOURCES);
491 }
492
493 //
494 // reset to the begining of the directory
495 //
496 Status = FileHandleSetPosition(DirHandle, 0);
497 if (EFI_ERROR(Status)) {
498 FreePool(*Buffer);
499 *Buffer = NULL;
500 return (Status);
501 }
502
503 //
504 // read in the info about the first file
505 //
506 Status = FileHandleRead (DirHandle, &BufferSize, *Buffer);
507 ASSERT(Status != EFI_BUFFER_TOO_SMALL);
508 if (EFI_ERROR(Status)) {
509 FreePool(*Buffer);
510 *Buffer = NULL;
511 return (Status);
512 }
513 return (EFI_SUCCESS);
514 }
515 /**
516 Retrieves the next file in a directory.
517
518 To use this function, caller must call the FileHandleFindFirstFile() to get the
519 first file, and then use this function get other files. This function can be
520 called for several times to get each file's information in the directory. If
521 the call of FileHandleFindNextFile() got the last file in the directory, the next
522 call of this function has no file to get. *NoFile will be set to TRUE and the
523 Buffer memory will be automatically freed.
524
525 @param DirHandle the file handle of the directory
526 @param Buffer pointer to buffer for file's information
527 @param NoFile pointer to boolean when last file is found
528
529 @retval EFI_SUCCESS Found the next file, or reached last file
530 @retval EFI_NO_MEDIA The device has no media.
531 @retval EFI_DEVICE_ERROR The device reported an error.
532 @retval EFI_VOLUME_CORRUPTED The file system structures are corrupted.
533 **/
534 EFI_STATUS
535 EFIAPI
536 FileHandleFindNextFile(
537 IN EFI_FILE_HANDLE DirHandle,
538 OUT EFI_FILE_INFO *Buffer,
539 OUT BOOLEAN *NoFile
540 )
541 {
542 EFI_STATUS Status;
543 UINTN BufferSize;
544
545 //
546 // ASSERTs for DirHandle or Buffer or NoFile poitners being NULL
547 //
548 ASSERT (DirHandle != NULL);
549 ASSERT (Buffer != NULL);
550 ASSERT (NoFile != NULL);
551
552 //
553 // This BufferSize MUST stay equal to the originally allocated one in GetFirstFile
554 //
555 BufferSize = FIND_XXXXX_FILE_BUFFER_SIZE;
556
557 //
558 // read in the info about the next file
559 //
560 Status = FileHandleRead (DirHandle, &BufferSize, Buffer);
561 ASSERT(Status != EFI_BUFFER_TOO_SMALL);
562 if (EFI_ERROR(Status)) {
563 return (Status);
564 }
565
566 //
567 // If we read 0 bytes (but did not have erros) we already read in the last file.
568 //
569 if (BufferSize == 0) {
570 FreePool(Buffer);
571 *NoFile = TRUE;
572 }
573
574 return (EFI_SUCCESS);
575 }
576
577 /**
578 Retrieve the size of a file.
579
580 if FileHandle is NULL then ASSERT()
581 if Size is NULL then ASSERT()
582
583 This function extracts the file size info from the FileHandle's EFI_FILE_INFO
584 data.
585
586 @param FileHandle file handle from which size is retrieved
587 @param Size pointer to size
588
589 @retval EFI_SUCCESS operation was completed sucessfully
590 @retval EFI_DEVICE_ERROR cannot access the file
591 **/
592 EFI_STATUS
593 EFIAPI
594 FileHandleGetSize (
595 IN EFI_FILE_HANDLE FileHandle,
596 OUT UINT64 *Size
597 )
598 {
599 EFI_FILE_INFO *FileInfo;
600
601 //
602 // ASSERT for FileHandle or Size being NULL
603 //
604 ASSERT (FileHandle != NULL);
605 ASSERT (Size != NULL);
606
607 //
608 // get the FileInfo structure
609 //
610 FileInfo = FileHandleGetInfo(FileHandle);
611 if (FileInfo == NULL) {
612 return (EFI_DEVICE_ERROR);
613 }
614
615 //
616 // Assign the Size pointer to the correct value
617 //
618 *Size = FileInfo->FileSize;
619
620 //
621 // free the FileInfo memory
622 //
623 FreePool(FileInfo);
624
625 return (EFI_SUCCESS);
626 }
627
628 /**
629 Set the size of a file.
630
631 If FileHandle is NULL then ASSERT().
632
633 This function changes the file size info from the FileHandle's EFI_FILE_INFO
634 data.
635
636 @param FileHandle File handle whose size is to be changed.
637 @param Size New size.
638
639 @retval EFI_SUCCESS operation was completed sucessfully.
640 @retval EFI_DEVICE_ERROR cannot access the file.
641 **/
642 EFI_STATUS
643 EFIAPI
644 FileHandleSetSize (
645 IN EFI_FILE_HANDLE FileHandle,
646 IN UINT64 Size
647 )
648 {
649 EFI_FILE_INFO *FileInfo;
650 EFI_STATUS Status;
651
652 //
653 // ASSERT for FileHandle or Size being NULL
654 //
655 ASSERT (FileHandle != NULL);
656
657 //
658 // get the FileInfo structure
659 //
660 FileInfo = FileHandleGetInfo(FileHandle);
661 if (FileInfo == NULL) {
662 return (EFI_DEVICE_ERROR);
663 }
664
665 //
666 // Assign the FileSize pointer to the new value
667 //
668 FileInfo->FileSize = Size;
669
670 Status = FileHandleSetInfo(FileHandle, FileInfo);
671 //
672 // free the FileInfo memory
673 //
674 FreePool(FileInfo);
675
676 return (Status);
677 }
678
679 /**
680 Safely append (on the left) with automatic string resizing given length of Destination and
681 desired length of copy from Source.
682
683 append the first D characters of Source to the end of Destination, where D is
684 the lesser of Count and the StrLen() of Source. If appending those D characters
685 will fit within Destination (whose Size is given as CurrentSize) and
686 still leave room for a NULL terminator, then those characters are appended,
687 starting at the original terminating NULL of Destination, and a new terminating
688 NULL is appended.
689
690 If appending D characters onto Destination will result in a overflow of the size
691 given in CurrentSize the string will be grown such that the copy can be performed
692 and CurrentSize will be updated to the new size.
693
694 If Source is NULL, there is nothing to append, just return the current buffer in
695 Destination.
696
697 if Destination is NULL, then ASSERT()
698 if Destination's current length (including NULL terminator) is already more then
699 CurrentSize, then ASSERT()
700
701 @param[in,out] Destination The String to append onto
702 @param[in,out] CurrentSize on call the number of bytes in Destination. On
703 return possibly the new size (still in bytes). if NULL
704 then allocate whatever is needed.
705 @param[in] Source The String to append from
706 @param[in] Count Maximum number of characters to append. if 0 then
707 all are appended.
708
709 @return Destination return the resultant string.
710 **/
711 CHAR16*
712 EFIAPI
713 StrnCatGrowLeft (
714 IN OUT CHAR16 **Destination,
715 IN OUT UINTN *CurrentSize,
716 IN CONST CHAR16 *Source,
717 IN UINTN Count
718 )
719 {
720 UINTN DestinationStartSize;
721 UINTN NewSize;
722 UINTN CopySize;
723
724 //
725 // ASSERTs
726 //
727 ASSERT(Destination != NULL);
728
729 //
730 // If there's nothing to do then just return Destination
731 //
732 if (Source == NULL) {
733 return (*Destination);
734 }
735
736 //
737 // allow for NULL pointers address as Destination
738 //
739 if (*Destination != NULL) {
740 ASSERT(CurrentSize != 0);
741 DestinationStartSize = StrSize(*Destination);
742 ASSERT(DestinationStartSize <= *CurrentSize);
743 } else {
744 DestinationStartSize = 0;
745 // ASSERT(*CurrentSize == 0);
746 }
747
748 //
749 // Append all of Source?
750 //
751 if (Count == 0) {
752 Count = StrSize(Source);
753 }
754
755 //
756 // Test and grow if required
757 //
758 if (CurrentSize != NULL) {
759 NewSize = *CurrentSize;
760 while (NewSize < (DestinationStartSize + Count)) {
761 NewSize += 2 * Count;
762 }
763 *Destination = ReallocatePool(*CurrentSize, NewSize, *Destination);
764 *CurrentSize = NewSize;
765 } else {
766 *Destination = AllocateZeroPool(Count+sizeof(CHAR16));
767 }
768
769 CopySize = StrSize(*Destination);
770 CopyMem((*Destination)+((Count-2)/sizeof(CHAR16)), *Destination, CopySize);
771 CopyMem(*Destination, Source, Count-2);
772 return (*Destination);
773 }
774
775 /**
776 Function to get a full filename given a EFI_FILE_HANDLE somewhere lower on the
777 directory 'stack'.
778
779 if Handle is NULL, return EFI_INVALID_PARAMETER
780
781 @param[in] Handle Handle to the Directory or File to create path to.
782 @param[out] FullFileName pointer to pointer to generated full file name. It
783 is the responsibility of the caller to free this memory
784 with a call to FreePool().
785 @retval EFI_SUCCESS the operation was sucessful and the FullFileName is valid.
786 @retval EFI_INVALID_PARAMETER Handle was NULL.
787 @retval EFI_INVALID_PARAMETER FullFileName was NULL.
788 @retval EFI_OUT_OF_RESOURCES a memory allocation failed.
789 **/
790 EFI_STATUS
791 EFIAPI
792 FileHandleGetFileName (
793 IN CONST EFI_FILE_HANDLE Handle,
794 OUT CHAR16 **FullFileName
795 )
796 {
797 EFI_STATUS Status;
798 UINTN Size;
799 EFI_FILE_HANDLE CurrentHandle;
800 EFI_FILE_HANDLE NextHigherHandle;
801 EFI_FILE_INFO *FileInfo;
802
803 Size = 0;
804
805 //
806 // Check our parameters
807 //
808 if (FullFileName == NULL || Handle == NULL) {
809 return (EFI_INVALID_PARAMETER);
810 }
811
812 *FullFileName = NULL;
813 CurrentHandle = NULL;
814
815 Status = Handle->Open(Handle, &CurrentHandle, L".", EFI_FILE_MODE_READ, 0);
816 if (!EFI_ERROR(Status)) {
817 //
818 // Reverse out the current directory on the device
819 //
820 for (;;) {
821 FileInfo = FileHandleGetInfo(CurrentHandle);
822 if (FileInfo == NULL) {
823 Status = EFI_OUT_OF_RESOURCES;
824 break;
825 } else {
826 //
827 // We got info... do we have a name? if yes preceed the current path with it...
828 //
829 if (StrLen (FileInfo->FileName) == 0) {
830 if (*FullFileName == NULL) {
831 ASSERT((*FullFileName == NULL && Size == 0) || (*FullFileName != NULL));
832 *FullFileName = StrnCatGrowLeft(FullFileName, &Size, L"\\", 0);
833 }
834 FreePool(FileInfo);
835 break;
836 } else {
837 if (*FullFileName == NULL) {
838 ASSERT((*FullFileName == NULL && Size == 0) || (*FullFileName != NULL));
839 *FullFileName = StrnCatGrowLeft(FullFileName, &Size, L"\\", 0);
840 }
841 ASSERT((*FullFileName == NULL && Size == 0) || (*FullFileName != NULL));
842 *FullFileName = StrnCatGrowLeft(FullFileName, &Size, FileInfo->FileName, 0);
843 *FullFileName = StrnCatGrowLeft(FullFileName, &Size, L"\\", 0);
844 FreePool(FileInfo);
845 }
846 }
847 //
848 // Move to the parent directory
849 //
850 Status = CurrentHandle->Open (CurrentHandle, &NextHigherHandle, L"..", EFI_FILE_MODE_READ, 0);
851 if (EFI_ERROR (Status)) {
852 break;
853 }
854
855 FileHandleClose(CurrentHandle);
856 CurrentHandle = NextHigherHandle;
857 }
858 } else if (Status == EFI_NOT_FOUND) {
859 Status = EFI_SUCCESS;
860 ASSERT((*FullFileName == NULL && Size == 0) || (*FullFileName != NULL));
861 *FullFileName = StrnCatGrowLeft(FullFileName, &Size, L"\\", 0);
862 }
863
864 if (CurrentHandle != NULL) {
865 CurrentHandle->Close (CurrentHandle);
866 }
867
868 if (EFI_ERROR(Status) && *FullFileName != NULL) {
869 FreePool(*FullFileName);
870 }
871
872 return (Status);
873 }
874
875 /**
876 Function to read a single line from a file. The \n is not included in the returned
877 buffer. The returned buffer must be callee freed.
878
879 If the position upon start is 0, then the Ascii Boolean will be set. This should be
880 maintained and not changed for all operations with the same file.
881
882 @param[in] Handle FileHandle to read from.
883 @param[in,out] Ascii Boolean value for indicating whether the file is Ascii (TRUE) or UCS2 (FALSE);
884
885 @return The line of text from the file.
886
887 @sa FileHandleReadLine
888 **/
889 CHAR16*
890 EFIAPI
891 FileHandleReturnLine(
892 IN EFI_FILE_HANDLE Handle,
893 IN OUT BOOLEAN *Ascii
894 )
895 {
896 CHAR16 *RetVal;
897 UINTN Size;
898 EFI_STATUS Status;
899
900 Size = 0;
901 RetVal = NULL;
902
903 Status = FileHandleReadLine(Handle, RetVal, &Size, FALSE, Ascii);
904 if (Status == EFI_BUFFER_TOO_SMALL) {
905 RetVal = AllocateZeroPool(Size);
906 Status = FileHandleReadLine(Handle, RetVal, &Size, FALSE, Ascii);
907 }
908 ASSERT_EFI_ERROR(Status);
909 if (EFI_ERROR(Status) && (RetVal != NULL)) {
910 FreePool(RetVal);
911 RetVal = NULL;
912 }
913 return (RetVal);
914 }
915
916 /**
917 Function to read a single line (up to but not including the \n) from a EFI_FILE_HANDLE.
918
919 If the position upon start is 0, then the Ascii Boolean will be set. This should be
920 maintained and not changed for all operations with the same file.
921
922 @param[in] Handle FileHandle to read from
923 @param[in,out] Buffer pointer to buffer to read into
924 @param[in,out] Size pointer to number of bytes in buffer
925 @param[in] Truncate if TRUE then allows for truncation of the line to fit.
926 if FALSE will reset the position to the begining of the
927 line if the buffer is not large enough.
928 @param[in,out] Ascii Boolean value for indicating whether the file is Ascii (TRUE) or UCS2 (FALSE);
929
930 @retval EFI_SUCCESS the operation was sucessful. the line is stored in
931 Buffer.
932 @retval EFI_INVALID_PARAMETER Handle was NULL.
933 @retval EFI_INVALID_PARAMETER Size was NULL.
934 @retval EFI_BUFFER_TOO_SMALL Size was not enough space to store the line.
935 Size was updated to minimum space required.
936 @sa FileHandleRead
937 **/
938 EFI_STATUS
939 EFIAPI
940 FileHandleReadLine(
941 IN EFI_FILE_HANDLE Handle,
942 IN OUT CHAR16 *Buffer,
943 IN OUT UINTN *Size,
944 IN BOOLEAN Truncate,
945 IN OUT BOOLEAN *Ascii
946 )
947 {
948 EFI_STATUS Status;
949 CHAR16 CharBuffer;
950 UINTN CharSize;
951 UINTN CountSoFar;
952 UINT64 OriginalFilePosition;
953
954
955 if (Handle == NULL
956 ||Size == NULL
957 ){
958 return (EFI_INVALID_PARAMETER);
959 }
960 if (Buffer == NULL) {
961 ASSERT(*Size == 0);
962 } else {
963 *Buffer = CHAR_NULL;
964 }
965 FileHandleGetPosition(Handle, &OriginalFilePosition);
966 if (OriginalFilePosition == 0) {
967 CharSize = sizeof(CHAR16);
968 Status = FileHandleRead(Handle, &CharSize, &CharBuffer);
969 ASSERT_EFI_ERROR(Status);
970 if (CharBuffer == gUnicodeFileTag) {
971 *Ascii = FALSE;
972 } else {
973 *Ascii = TRUE;
974 FileHandleSetPosition(Handle, OriginalFilePosition);
975 }
976 }
977
978 for (CountSoFar = 0;;CountSoFar++){
979 CharBuffer = 0;
980 if (*Ascii) {
981 CharSize = sizeof(CHAR8);
982 } else {
983 CharSize = sizeof(CHAR16);
984 }
985 Status = FileHandleRead(Handle, &CharSize, &CharBuffer);
986 if ( EFI_ERROR(Status)
987 || CharSize == 0
988 || (CharBuffer == L'\n' && !(*Ascii))
989 || (CharBuffer == '\n' && *Ascii)
990 ){
991 break;
992 }
993 //
994 // if we have space save it...
995 //
996 if ((CountSoFar+1)*sizeof(CHAR16) < *Size){
997 ASSERT(Buffer != NULL);
998 ((CHAR16*)Buffer)[CountSoFar] = CharBuffer;
999 ((CHAR16*)Buffer)[CountSoFar+1] = CHAR_NULL;
1000 }
1001 }
1002
1003 //
1004 // if we ran out of space tell when...
1005 //
1006 if ((CountSoFar+1)*sizeof(CHAR16) > *Size){
1007 *Size = (CountSoFar+1)*sizeof(CHAR16);
1008 if (!Truncate) {
1009 FileHandleSetPosition(Handle, OriginalFilePosition);
1010 } else {
1011 DEBUG((DEBUG_WARN, "The line was truncated in FileHandleReadLine"));
1012 }
1013 return (EFI_BUFFER_TOO_SMALL);
1014 }
1015 while(Buffer[StrLen(Buffer)-1] == L'\r') {
1016 Buffer[StrLen(Buffer)-1] = CHAR_NULL;
1017 }
1018
1019 return (Status);
1020 }
1021
1022 /**
1023 function to write a line of unicode text to a file.
1024
1025 if Handle is NULL, ASSERT.
1026 if Buffer is NULL, do nothing. (return SUCCESS)
1027
1028 @param[in] Handle FileHandle to write to
1029 @param[in] Buffer Buffer to write
1030
1031 @retval EFI_SUCCESS the data was written.
1032 @retval other failure.
1033
1034 @sa FileHandleWrite
1035 **/
1036 EFI_STATUS
1037 EFIAPI
1038 FileHandleWriteLine(
1039 IN EFI_FILE_HANDLE Handle,
1040 IN CHAR16 *Buffer
1041 )
1042 {
1043 EFI_STATUS Status;
1044 UINTN Size;
1045
1046 ASSERT(Handle != NULL);
1047
1048 if (Buffer == NULL) {
1049 return (EFI_SUCCESS);
1050 }
1051
1052 Size = StrSize(Buffer) - sizeof(Buffer[0]);
1053 Status = FileHandleWrite(Handle, &Size, Buffer);
1054 if (EFI_ERROR(Status)) {
1055 return (Status);
1056 }
1057 Size = StrSize(L"\r\n") - sizeof(CHAR16);
1058 return FileHandleWrite(Handle, &Size, L"\r\n");
1059 }
1060
1061 /**
1062 function to take a formatted argument and print it to a file.
1063
1064 @param[in] Handle the file handle for the file to write to
1065 @param[in] Format the format argument (see printlib for format specifier)
1066 @param[in] ... the variable arguments for the format
1067
1068 @retval EFI_SUCCESS the operation was sucessful
1069 @return other a return value from FileHandleWriteLine
1070
1071 @sa FileHandleWriteLine
1072 **/
1073 EFI_STATUS
1074 EFIAPI
1075 FileHandlePrintLine(
1076 IN EFI_FILE_HANDLE Handle,
1077 IN CONST CHAR16 *Format,
1078 ...
1079 )
1080 {
1081 VA_LIST Marker;
1082 CHAR16 *Buffer;
1083 EFI_STATUS Status;
1084
1085 VA_START (Marker, Format);
1086
1087 //
1088 // Get a buffer to print into
1089 //
1090 Buffer = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
1091 ASSERT (Buffer != NULL);
1092
1093 //
1094 // Print into our buffer
1095 //
1096 UnicodeVSPrint (Buffer, PcdGet16 (PcdShellPrintBufferSize), Format, Marker);
1097
1098 //
1099 // Print buffer into file
1100 //
1101 Status = FileHandleWriteLine(Handle, Buffer);
1102
1103 //
1104 // Cleanup and return
1105 //
1106 FreePool(Buffer);
1107 return (Status);
1108 }
1109
1110 /**
1111 Function to determine if a FILE_HANDLE is at the end of the file.
1112
1113 This will NOT work on directories.
1114
1115 If Handle is NULL, then ASSERT.
1116
1117 @param[in] Handle the file handle
1118
1119 @retval TRUE the position is at the end of the file
1120 @retval FALSE the position is not at the end of the file
1121 **/
1122 BOOLEAN
1123 EFIAPI
1124 FileHandleEof(
1125 IN EFI_FILE_HANDLE Handle
1126 )
1127 {
1128 EFI_FILE_INFO *Info;
1129 UINT64 Pos;
1130 BOOLEAN RetVal;
1131
1132 //
1133 // ASSERT if Handle is NULL
1134 //
1135 ASSERT(Handle != NULL);
1136
1137 FileHandleGetPosition(Handle, &Pos);
1138 Info = FileHandleGetInfo (Handle);
1139 ASSERT(Info != NULL);
1140 FileHandleSetPosition(Handle, Pos);
1141
1142 if (Info == NULL) {
1143 return (FALSE);
1144 }
1145
1146 if (Pos == Info->FileSize) {
1147 RetVal = TRUE;
1148 } else {
1149 RetVal = FALSE;
1150 }
1151
1152 FreePool (Info);
1153
1154 return (RetVal);
1155 }