cmds: make copying FILE to another FILE

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;
 }