diff options
| -rw-r--r-- | Library/UefiShellUfmCommandLib/UefiShellUfmCommandLib.inf | 2 | ||||
| -rw-r--r-- | Library/UefiShellUfmCommandLib/cmds.c | 42 |
2 files changed, 43 insertions, 1 deletions
diff --git a/Library/UefiShellUfmCommandLib/UefiShellUfmCommandLib.inf b/Library/UefiShellUfmCommandLib/UefiShellUfmCommandLib.inf index e53e3b3..d37516e 100644 --- a/Library/UefiShellUfmCommandLib/UefiShellUfmCommandLib.inf +++ b/Library/UefiShellUfmCommandLib/UefiShellUfmCommandLib.inf @@ -34,6 +34,7 @@ DevicePathLib FileHandleLib MemoryAllocationLib + PcdLib PrintLib ShellCommandLib ShellLib @@ -52,3 +53,4 @@ [FeaturePcd] [Pcd] + gEfiShellPkgTokenSpaceGuid.PcdShellFileOperationSize diff --git a/Library/UefiShellUfmCommandLib/cmds.c b/Library/UefiShellUfmCommandLib/cmds.c index 8a9c425..ed1168c 100644 --- a/Library/UefiShellUfmCommandLib/cmds.c +++ b/Library/UefiShellUfmCommandLib/cmds.c @@ -2,6 +2,7 @@ #include <Library/DebugLib.h> #include <Library/FileHandleLib.h> #include <Library/MemoryAllocationLib.h> +#include <Library/PcdLib.h> #include "cmds.h" @@ -76,6 +77,45 @@ EFI_STATUS delete_file(EFI_SHELL_FILE_INFO *node) return status; } +STATIC EFI_STATUS copy_single_file(CONST CHAR16 *src, CONST CHAR16 *dest) +{ + SHELL_FILE_HANDLE src_handle = NULL, dest_handle = NULL; + UINTN read_size = PcdGet32(PcdShellFileOperationSize); + VOID *buffer; + EFI_STATUS status; + + ShellDeleteFileByName(dest); + + status = ShellOpenFileByName(dest, &dest_handle, EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE|EFI_FILE_MODE_CREATE, 0); + if(EFI_ERROR(status)) + return EFI_ACCESS_DENIED; + + status = ShellOpenFileByName(src, &src_handle, EFI_FILE_MODE_READ, 0); + if(EFI_ERROR(status)) + return EFI_ACCESS_DENIED; + + buffer = AllocateZeroPool(read_size); + if(buffer == NULL) + return EFI_OUT_OF_RESOURCES; + + while(read_size == PcdGet32(PcdShellFileOperationSize) && !EFI_ERROR(status)) + { + status = ShellReadFile(src_handle, &read_size, buffer); + if(EFI_ERROR(status)) + break; + + status = ShellWriteFile(dest_handle, &read_size, buffer); + if(EFI_ERROR(status)) + break; + } + + if(dest_handle != NULL) + ShellCloseFile(&dest_handle); + if(src_handle != NULL) + ShellCloseFile(&src_handle); + return status; +} + EFI_STATUS copy_file(CONST CHAR16 *src, CONST CHAR16 *dest) { SHELL_FILE_HANDLE dest_handle = NULL; @@ -124,7 +164,7 @@ EFI_STATUS copy_file(CONST CHAR16 *src, CONST CHAR16 *dest) StrnCatGrow(&correct_dest, &size, path_last_item, 0); } - // TODO: copy single file + status = copy_single_file(src, correct_dest); SHELL_FREE_NON_NULL(correct_dest); return status; } |