2 Main file for cp shell level 2 function.
4 Copyright (c) 2009 - 2014, 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
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.
15 #include "UefiShellLevel2CommandsLib.h"
16 #include <Guid/FileSystemInfo.h>
17 #include <Guid/FileSystemVolumeLabelInfo.h>
20 Function to take a list of files to copy and a destination location and do
21 the verification and copying of those files to that location. This function
22 will report any errors to the user and halt.
24 @param[in] FileList A LIST_ENTRY* based list of files to move.
25 @param[in] DestDir The destination location.
26 @param[in] SilentMode TRUE to eliminate screen output.
27 @param[in] RecursiveMode TRUE to copy directories.
28 @param[in] Resp The response to the overwrite query (if always).
30 @retval SHELL_SUCCESS the files were all moved.
31 @retval SHELL_INVALID_PARAMETER a parameter was invalid
32 @retval SHELL_SECURITY_VIOLATION a security violation ocurred
33 @retval SHELL_WRITE_PROTECTED the destination was write protected
34 @retval SHELL_OUT_OF_RESOURCES a memory allocation failed
39 IN CONST EFI_SHELL_FILE_INFO
*FileList
,
40 IN CONST CHAR16
*DestDir
,
41 IN BOOLEAN SilentMode
,
42 IN BOOLEAN RecursiveMode
,
47 Function to Copy one file to another location
49 If the destination exists the user will be prompted and the result put into *resp
51 @param[in] Source pointer to source file name
52 @param[in] Dest pointer to destination file name
53 @param[out] Resp pointer to response from question. Pass back on looped calling
54 @param[in] SilentMode whether to run in quiet mode or not
56 @retval SHELL_SUCCESS The source file was copied to the destination
61 IN CONST CHAR16
*Source
,
62 IN CONST CHAR16
*Dest
,
69 SHELL_FILE_HANDLE SourceHandle
;
70 SHELL_FILE_HANDLE DestHandle
;
75 EFI_SHELL_FILE_INFO
*List
;
76 SHELL_STATUS ShellStatus
;
77 UINT64 SourceFileSize
;
79 EFI_FILE_PROTOCOL
*DestVolumeFP
;
80 EFI_FILE_SYSTEM_INFO
*DestVolumeInfo
;
81 UINTN DestVolumeInfoSize
;
89 DestVolumeInfo
= NULL
;
90 ShellStatus
= SHELL_SUCCESS
;
92 ReadSize
= PcdGet32(PcdShellFileOperationSize
);
93 // Why bother copying a file to itself
94 if (StrCmp(Source
, Dest
) == 0) {
95 return (SHELL_SUCCESS
);
99 // if the destination file existed check response and possibly prompt user
101 if (ShellFileExists(Dest
) == EFI_SUCCESS
) {
102 if (Response
== NULL
&& !SilentMode
) {
103 Status
= ShellPromptForResponseHii(ShellPromptResponseTypeYesNoAllCancel
, STRING_TOKEN (STR_GEN_DEST_EXIST_OVR
), gShellLevel2HiiHandle
, &Response
);
106 // possibly return based on response
109 switch (*(SHELL_PROMPT_RESPONSE
*)Response
) {
110 case ShellPromptResponseNo
:
112 // return success here so we dont stop the process
114 return (SHELL_SUCCESS
);
115 case ShellPromptResponseCancel
:
118 // indicate to stop everything
120 return (SHELL_ABORTED
);
121 case ShellPromptResponseAll
:
123 case ShellPromptResponseYes
:
126 return SHELL_ABORTED
;
131 if (ShellIsDirectory(Source
) == EFI_SUCCESS
) {
132 Status
= ShellCreateDirectory(Dest
, &DestHandle
);
133 if (EFI_ERROR(Status
)) {
134 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_CP_DEST_DIR_FAIL
), gShellLevel2HiiHandle
, Dest
);
135 return (SHELL_ACCESS_DENIED
);
139 // Now copy all the files under the directory...
143 StrnCatGrow(&TempName
, &Size
, Source
, 0);
144 StrnCatGrow(&TempName
, &Size
, L
"\\*", 0);
145 if (TempName
!= NULL
) {
146 ShellOpenFileMetaArg((CHAR16
*)TempName
, EFI_FILE_MODE_READ
, &List
);
147 *TempName
= CHAR_NULL
;
148 StrnCatGrow(&TempName
, &Size
, Dest
, 0);
149 StrnCatGrow(&TempName
, &Size
, L
"\\", 0);
150 ShellStatus
= ValidateAndCopyFiles(List
, TempName
, SilentMode
, TRUE
, Resp
);
151 ShellCloseFileMetaArg(&List
);
152 SHELL_FREE_NON_NULL(TempName
);
156 Status
= ShellDeleteFileByName(Dest
);
159 // open file with create enabled
161 Status
= ShellOpenFileByName(Dest
, &DestHandle
, EFI_FILE_MODE_READ
|EFI_FILE_MODE_WRITE
|EFI_FILE_MODE_CREATE
, 0);
162 if (EFI_ERROR(Status
)) {
163 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_CP_DEST_OPEN_FAIL
), gShellLevel2HiiHandle
, Dest
);
164 return (SHELL_ACCESS_DENIED
);
170 Status
= ShellOpenFileByName(Source
, &SourceHandle
, EFI_FILE_MODE_READ
, 0);
171 ASSERT_EFI_ERROR(Status
);
174 //get file size of source file and freespace available on destination volume
176 ShellGetFileSize(SourceHandle
, &SourceFileSize
);
177 ShellGetFileSize(DestHandle
, &DestFileSize
);
180 //if the destination file already exists then it will be replaced, meaning the sourcefile effectively needs less storage space
182 if(DestFileSize
< SourceFileSize
){
183 SourceFileSize
-= DestFileSize
;
189 //get the system volume info to check the free space
191 DestVolumeFP
= ConvertShellHandleToEfiFileProtocol(DestHandle
);
192 DestVolumeInfo
= NULL
;
193 DestVolumeInfoSize
= 0;
194 Status
= DestVolumeFP
->GetInfo(
196 &gEfiFileSystemInfoGuid
,
201 if (Status
== EFI_BUFFER_TOO_SMALL
) {
202 DestVolumeInfo
= AllocateZeroPool(DestVolumeInfoSize
);
203 Status
= DestVolumeFP
->GetInfo(
205 &gEfiFileSystemInfoGuid
,
212 //check if enough space available on destination drive to complete copy
214 if (DestVolumeInfo
!= NULL
&& (DestVolumeInfo
->FreeSpace
< SourceFileSize
)) {
216 //not enough space on destination directory to copy file
218 SHELL_FREE_NON_NULL(DestVolumeInfo
);
219 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_CPY_FAIL
), gShellLevel2HiiHandle
);
220 return(SHELL_VOLUME_FULL
);
223 // copy data between files
225 Buffer
= AllocateZeroPool(ReadSize
);
226 ASSERT(Buffer
!= NULL
);
227 while (ReadSize
== PcdGet32(PcdShellFileOperationSize
) && !EFI_ERROR(Status
)) {
228 Status
= ShellReadFile(SourceHandle
, &ReadSize
, Buffer
);
229 if (!EFI_ERROR(Status
)) {
230 Status
= ShellWriteFile(DestHandle
, &ReadSize
, Buffer
);
231 if (EFI_ERROR(Status
)) {
232 ShellStatus
= (SHELL_STATUS
) (Status
& (~MAX_BIT
));
233 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_CPY_WRITE_ERROR
), gShellLevel2HiiHandle
, Dest
);
237 ShellStatus
= (SHELL_STATUS
) (Status
& (~MAX_BIT
));
238 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_CPY_READ_ERROR
), gShellLevel2HiiHandle
, Source
);
243 SHELL_FREE_NON_NULL(DestVolumeInfo
);
249 if (DestHandle
!= NULL
) {
250 ShellCloseFile(&DestHandle
);
253 if (SourceHandle
!= NULL
) {
254 ShellCloseFile(&SourceHandle
);
265 function to take a list of files to copy and a destination location and do
266 the verification and copying of those files to that location. This function
267 will report any errors to the user and halt.
269 The key is to have this function called ONLY once. this allows for the parameter
270 verification to happen correctly.
272 @param[in] FileList A LIST_ENTRY* based list of files to move.
273 @param[in] DestDir The destination location.
274 @param[in] SilentMode TRUE to eliminate screen output.
275 @param[in] RecursiveMode TRUE to copy directories.
276 @param[in] Resp The response to the overwrite query (if always).
278 @retval SHELL_SUCCESS the files were all moved.
279 @retval SHELL_INVALID_PARAMETER a parameter was invalid
280 @retval SHELL_SECURITY_VIOLATION a security violation ocurred
281 @retval SHELL_WRITE_PROTECTED the destination was write protected
282 @retval SHELL_OUT_OF_RESOURCES a memory allocation failed
286 ValidateAndCopyFiles(
287 IN CONST EFI_SHELL_FILE_INFO
*FileList
,
288 IN CONST CHAR16
*DestDir
,
289 IN BOOLEAN SilentMode
,
290 IN BOOLEAN RecursiveMode
,
296 CONST EFI_SHELL_FILE_INFO
*Node
;
297 SHELL_STATUS ShellStatus
;
304 CHAR16
*CleanFilePathStr
;
313 ShellStatus
= SHELL_SUCCESS
;
315 Cwd
= ShellGetCurrentDir(NULL
);
316 CleanFilePathStr
= NULL
;
318 ASSERT(FileList
!= NULL
);
319 ASSERT(DestDir
!= NULL
);
322 Status
= ShellLevel2StripQuotes (DestDir
, &CleanFilePathStr
);
323 if (EFI_ERROR (Status
)) {
324 if (Status
== EFI_OUT_OF_RESOURCES
) {
325 return SHELL_OUT_OF_RESOURCES
;
327 return SHELL_INVALID_PARAMETER
;
331 ASSERT (CleanFilePathStr
!= NULL
);
334 // If we are trying to copy multiple files... make sure we got a directory for the target...
336 if (EFI_ERROR(ShellIsDirectory(CleanFilePathStr
)) && FileList
->Link
.ForwardLink
!= FileList
->Link
.BackLink
) {
338 // Error for destination not a directory
340 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_NOT_DIR
), gShellLevel2HiiHandle
, CleanFilePathStr
);
341 FreePool (CleanFilePathStr
);
342 return (SHELL_INVALID_PARAMETER
);
344 for (Node
= (EFI_SHELL_FILE_INFO
*)GetFirstNode(&FileList
->Link
)
345 ; !IsNull(&FileList
->Link
, &Node
->Link
)
346 ; Node
= (EFI_SHELL_FILE_INFO
*)GetNextNode(&FileList
->Link
, &Node
->Link
)
349 // skip the directory traversing stuff...
351 if (StrCmp(Node
->FileName
, L
".") == 0 || StrCmp(Node
->FileName
, L
"..") == 0) {
355 NewSize
= StrSize(CleanFilePathStr
);
356 NewSize
+= StrSize(Node
->FullName
);
357 NewSize
+= (Cwd
== NULL
)? 0 : StrSize(Cwd
);
358 if (NewSize
> PathSize
) {
363 // Make sure got -r if required
365 if (!RecursiveMode
&& !EFI_ERROR(ShellIsDirectory(Node
->FullName
))) {
366 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_CP_DIR_REQ
), gShellLevel2HiiHandle
);
367 FreePool (CleanFilePathStr
);
368 return (SHELL_INVALID_PARAMETER
);
372 // make sure got dest as dir if needed
374 if (!EFI_ERROR(ShellIsDirectory(Node
->FullName
)) && EFI_ERROR(ShellIsDirectory(CleanFilePathStr
))) {
376 // Error for destination not a directory
378 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_NOT_DIR
), gShellLevel2HiiHandle
, CleanFilePathStr
);
379 FreePool (CleanFilePathStr
);
380 return (SHELL_INVALID_PARAMETER
);
384 HiiOutput
= HiiGetString (gShellLevel2HiiHandle
, STRING_TOKEN (STR_CP_OUTPUT
), NULL
);
385 HiiResultOk
= HiiGetString (gShellLevel2HiiHandle
, STRING_TOKEN (STR_GEN_RES_OK
), NULL
);
386 DestPath
= AllocateZeroPool(PathSize
);
388 if (DestPath
== NULL
|| HiiOutput
== NULL
|| HiiResultOk
== NULL
) {
389 SHELL_FREE_NON_NULL(DestPath
);
390 SHELL_FREE_NON_NULL(HiiOutput
);
391 SHELL_FREE_NON_NULL(HiiResultOk
);
392 FreePool (CleanFilePathStr
);
393 return (SHELL_OUT_OF_RESOURCES
);
397 // Go through the list of files to copy...
399 for (Node
= (EFI_SHELL_FILE_INFO
*)GetFirstNode(&FileList
->Link
)
400 ; !IsNull(&FileList
->Link
, &Node
->Link
)
401 ; Node
= (EFI_SHELL_FILE_INFO
*)GetNextNode(&FileList
->Link
, &Node
->Link
)
403 if (ShellGetExecutionBreakFlag()) {
406 ASSERT(Node
->FileName
!= NULL
);
407 ASSERT(Node
->FullName
!= NULL
);
410 // skip the directory traversing stuff...
412 if (StrCmp(Node
->FileName
, L
".") == 0 || StrCmp(Node
->FileName
, L
"..") == 0) {
416 if (FileList
->Link
.ForwardLink
== FileList
->Link
.BackLink
// 1 item
417 && EFI_ERROR(ShellIsDirectory(CleanFilePathStr
)) // not an existing directory
419 if (StrStr(CleanFilePathStr
, L
":") == NULL
) {
421 // simple copy of a single file
424 StrnCpy(DestPath
, Cwd
, PathSize
/sizeof(CHAR16
)-1);
426 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_DIR_NF
), gShellLevel2HiiHandle
, CleanFilePathStr
);
427 FreePool (CleanFilePathStr
);
428 return (SHELL_INVALID_PARAMETER
);
430 if (DestPath
[StrLen(DestPath
)-1] != L
'\\' && CleanFilePathStr
[0] != L
'\\') {
431 StrnCat(DestPath
, L
"\\", PathSize
/sizeof(CHAR16
) - StrLen(DestPath
) -1);
432 } else if (DestPath
[StrLen(DestPath
)-1] == L
'\\' && CleanFilePathStr
[0] == L
'\\') {
433 ((CHAR16
*)DestPath
)[StrLen(DestPath
)-1] = CHAR_NULL
;
435 StrnCat(DestPath
, CleanFilePathStr
, PathSize
/sizeof(CHAR16
) - StrLen(DestPath
) -1);
437 StrnCpy(DestPath
, CleanFilePathStr
, PathSize
/sizeof(CHAR16
) -1);
441 // we have multiple files or a directory in the DestDir
445 // Check for leading slash
447 if (CleanFilePathStr
[0] == L
'\\') {
449 // Copy to the root of CWD
452 StrnCpy(DestPath
, Cwd
, PathSize
/sizeof(CHAR16
) -1);
454 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_DIR_NF
), gShellLevel2HiiHandle
, CleanFilePathStr
);
455 FreePool(CleanFilePathStr
);
456 return (SHELL_INVALID_PARAMETER
);
458 while (PathRemoveLastItem(DestPath
));
459 StrnCat(DestPath
, CleanFilePathStr
+1, PathSize
/sizeof(CHAR16
) - StrLen(DestPath
) -1);
460 StrnCat(DestPath
, Node
->FileName
, PathSize
/sizeof(CHAR16
) - StrLen(DestPath
) -1);
461 } else if (StrStr(CleanFilePathStr
, L
":") == NULL
) {
463 StrnCpy(DestPath
, Cwd
, PathSize
/sizeof(CHAR16
) -1);
465 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_DIR_NF
), gShellLevel2HiiHandle
, CleanFilePathStr
);
466 FreePool(CleanFilePathStr
);
467 return (SHELL_INVALID_PARAMETER
);
469 if (DestPath
[StrLen(DestPath
)-1] != L
'\\' && CleanFilePathStr
[0] != L
'\\') {
470 StrnCat(DestPath
, L
"\\", PathSize
/sizeof(CHAR16
) - StrLen(DestPath
) -1);
471 } else if (DestPath
[StrLen(DestPath
)-1] == L
'\\' && CleanFilePathStr
[0] == L
'\\') {
472 ((CHAR16
*)DestPath
)[StrLen(DestPath
)-1] = CHAR_NULL
;
474 StrnCat(DestPath
, CleanFilePathStr
, PathSize
/sizeof(CHAR16
) - StrLen(DestPath
) -1);
475 if (CleanFilePathStr
[StrLen(CleanFilePathStr
)-1] != L
'\\' && Node
->FileName
[0] != L
'\\') {
476 StrnCat(DestPath
, L
"\\", PathSize
/sizeof(CHAR16
) - StrLen(DestPath
) -1);
477 } else if (CleanFilePathStr
[StrLen(CleanFilePathStr
)-1] == L
'\\' && Node
->FileName
[0] == L
'\\') {
478 ((CHAR16
*)DestPath
)[StrLen(DestPath
)-1] = CHAR_NULL
;
480 StrnCat(DestPath
, Node
->FileName
, PathSize
/sizeof(CHAR16
) - StrLen(DestPath
) -1);
483 StrnCpy(DestPath
, CleanFilePathStr
, PathSize
/sizeof(CHAR16
) -1);
484 if (CleanFilePathStr
[StrLen(CleanFilePathStr
)-1] != L
'\\' && Node
->FileName
[0] != L
'\\') {
485 StrnCat(DestPath
, L
"\\", PathSize
/sizeof(CHAR16
) - StrLen(DestPath
) -1);
486 } else if (CleanFilePathStr
[StrLen(CleanFilePathStr
)-1] == L
'\\' && Node
->FileName
[0] == L
'\\') {
487 ((CHAR16
*)CleanFilePathStr
)[StrLen(CleanFilePathStr
)-1] = CHAR_NULL
;
489 StrnCat(DestPath
, Node
->FileName
, PathSize
/sizeof(CHAR16
) - StrLen(DestPath
) -1);
494 // Make sure the path exists
496 if (EFI_ERROR(VerifyIntermediateDirectories(DestPath
))) {
497 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_CP_DIR_WNF
), gShellLevel2HiiHandle
);
498 ShellStatus
= SHELL_DEVICE_ERROR
;
502 if ( !EFI_ERROR(ShellIsDirectory(Node
->FullName
))
503 && !EFI_ERROR(ShellIsDirectory(DestPath
))
504 && StrniCmp(Node
->FullName
, DestPath
, StrLen(DestPath
)) == NULL
506 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_CP_SD_PARENT
), gShellLevel2HiiHandle
);
507 ShellStatus
= SHELL_INVALID_PARAMETER
;
510 if (StringNoCaseCompare(&Node
->FullName
, &DestPath
) == 0) {
511 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_CP_SD_SAME
), gShellLevel2HiiHandle
);
512 ShellStatus
= SHELL_INVALID_PARAMETER
;
516 if ((StrniCmp(Node
->FullName
, DestPath
, StrLen(Node
->FullName
)) == 0)
517 && (DestPath
[StrLen(Node
->FullName
)] == CHAR_NULL
|| DestPath
[StrLen(Node
->FullName
)] == L
'\\')
519 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_CP_SD_SAME
), gShellLevel2HiiHandle
);
520 ShellStatus
= SHELL_INVALID_PARAMETER
;
524 PathCleanUpDirectories(DestPath
);
527 ShellPrintEx(-1, -1, HiiOutput
, Node
->FullName
, DestPath
);
531 // copy single file...
533 ShellStatus
= CopySingleFile(Node
->FullName
, DestPath
, &Response
, SilentMode
);
534 if (ShellStatus
!= SHELL_SUCCESS
) {
538 if (ShellStatus
== SHELL_SUCCESS
&& Resp
== NULL
) {
539 ShellPrintEx(-1, -1, L
"%s", HiiResultOk
);
542 SHELL_FREE_NON_NULL(DestPath
);
543 SHELL_FREE_NON_NULL(HiiOutput
);
544 SHELL_FREE_NON_NULL(HiiResultOk
);
545 SHELL_FREE_NON_NULL(CleanFilePathStr
);
547 SHELL_FREE_NON_NULL(Response
);
550 return (ShellStatus
);
555 Validate and if successful copy all the files from the list into
556 destination directory.
558 @param[in] FileList The list of files to copy.
559 @param[in] DestDir The directory to copy files to.
560 @param[in] SilentMode TRUE to eliminate screen output.
561 @param[in] RecursiveMode TRUE to copy directories.
563 @retval SHELL_INVALID_PARAMETER A parameter was invalid.
564 @retval SHELL_SUCCESS The operation was successful.
568 ProcessValidateAndCopyFiles(
569 IN EFI_SHELL_FILE_INFO
*FileList
,
570 IN CONST CHAR16
*DestDir
,
571 IN BOOLEAN SilentMode
,
572 IN BOOLEAN RecursiveMode
575 SHELL_STATUS ShellStatus
;
576 EFI_SHELL_FILE_INFO
*List
;
577 EFI_FILE_INFO
*FileInfo
;
584 ShellOpenFileMetaArg((CHAR16
*)DestDir
, EFI_FILE_MODE_READ
, &List
);
585 if (List
!= NULL
&& List
->Link
.ForwardLink
!= List
->Link
.BackLink
) {
586 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_MARG_ERROR
), gShellLevel2HiiHandle
, DestDir
);
587 ShellStatus
= SHELL_INVALID_PARAMETER
;
588 ShellCloseFileMetaArg(&List
);
589 } else if (List
!= NULL
) {
590 ASSERT(((EFI_SHELL_FILE_INFO
*)List
->Link
.ForwardLink
) != NULL
);
591 ASSERT(((EFI_SHELL_FILE_INFO
*)List
->Link
.ForwardLink
)->FullName
!= NULL
);
592 FileInfo
= gEfiShellProtocol
->GetFileInfo(((EFI_SHELL_FILE_INFO
*)List
->Link
.ForwardLink
)->Handle
);
593 ASSERT(FileInfo
!= NULL
);
594 StrnCatGrow(&FullName
, NULL
, ((EFI_SHELL_FILE_INFO
*)List
->Link
.ForwardLink
)->FullName
, 0);
595 ShellCloseFileMetaArg(&List
);
596 if ((FileInfo
->Attribute
& EFI_FILE_READ_ONLY
) == 0) {
597 ShellStatus
= ValidateAndCopyFiles(FileList
, FullName
, SilentMode
, RecursiveMode
, NULL
);
599 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_CP_DEST_ERROR
), gShellLevel2HiiHandle
);
600 ShellStatus
= SHELL_ACCESS_DENIED
;
603 ShellCloseFileMetaArg(&List
);
604 ShellStatus
= ValidateAndCopyFiles(FileList
, DestDir
, SilentMode
, RecursiveMode
, NULL
);
607 SHELL_FREE_NON_NULL(FileInfo
);
608 SHELL_FREE_NON_NULL(FullName
);
609 return (ShellStatus
);
612 STATIC CONST SHELL_PARAM_ITEM ParamList
[] = {
619 Function for 'cp' command.
621 @param[in] ImageHandle Handle to the Image (NULL if Internal).
622 @param[in] SystemTable Pointer to the System Table (NULL if Internal).
627 IN EFI_HANDLE ImageHandle
,
628 IN EFI_SYSTEM_TABLE
*SystemTable
633 CHAR16
*ProblemParam
;
634 SHELL_STATUS ShellStatus
;
637 EFI_SHELL_FILE_INFO
*FileList
;
639 BOOLEAN RecursiveMode
;
643 ShellStatus
= SHELL_SUCCESS
;
648 // initialize the shell lib (we must be in non-auto-init...)
650 Status
= ShellInitialize();
651 ASSERT_EFI_ERROR(Status
);
653 Status
= CommandInit();
654 ASSERT_EFI_ERROR(Status
);
657 // parse the command line
659 Status
= ShellCommandLineParse (ParamList
, &Package
, &ProblemParam
, TRUE
);
660 if (EFI_ERROR(Status
)) {
661 if (Status
== EFI_VOLUME_CORRUPTED
&& ProblemParam
!= NULL
) {
662 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PROBLEM
), gShellLevel2HiiHandle
, ProblemParam
);
663 FreePool(ProblemParam
);
664 ShellStatus
= SHELL_INVALID_PARAMETER
;
672 if (ShellCommandLineGetFlag(Package
, L
"-?")) {
677 // Initialize SilentMode and RecursiveMode
679 if (gEfiShellProtocol
->BatchIsActive()) {
682 SilentMode
= ShellCommandLineGetFlag(Package
, L
"-q");
684 RecursiveMode
= ShellCommandLineGetFlag(Package
, L
"-r");
686 switch (ParamCount
= ShellCommandLineGetCount(Package
)) {
690 // we have insufficient parameters
692 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_TOO_FEW
), gShellLevel2HiiHandle
);
693 ShellStatus
= SHELL_INVALID_PARAMETER
;
697 // must have valid CWD for single parameter...
699 Cwd
= ShellGetCurrentDir(NULL
);
701 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_NO_CWD
), gShellLevel2HiiHandle
);
702 ShellStatus
= SHELL_INVALID_PARAMETER
;
704 Status
= ShellOpenFileMetaArg((CHAR16
*)ShellCommandLineGetRawValue(Package
, 1), EFI_FILE_MODE_WRITE
|EFI_FILE_MODE_READ
, &FileList
);
705 if (FileList
== NULL
|| IsListEmpty(&FileList
->Link
) || EFI_ERROR(Status
)) {
706 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_FILE_NF
), gShellLevel2HiiHandle
, ShellCommandLineGetRawValue(Package
, 1));
707 ShellStatus
= SHELL_NOT_FOUND
;
709 ShellStatus
= ProcessValidateAndCopyFiles(FileList
, Cwd
, SilentMode
, RecursiveMode
);
716 // Make a big list of all the files...
718 for (ParamCount
--, LoopCounter
= 1 ; LoopCounter
< ParamCount
&& ShellStatus
== SHELL_SUCCESS
; LoopCounter
++) {
719 if (ShellGetExecutionBreakFlag()) {
722 Status
= ShellOpenFileMetaArg((CHAR16
*)ShellCommandLineGetRawValue(Package
, LoopCounter
), EFI_FILE_MODE_WRITE
|EFI_FILE_MODE_READ
, &FileList
);
723 if (EFI_ERROR(Status
) || FileList
== NULL
|| IsListEmpty(&FileList
->Link
)) {
724 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_FILE_NF
), gShellLevel2HiiHandle
, ShellCommandLineGetRawValue(Package
, LoopCounter
));
725 ShellStatus
= SHELL_NOT_FOUND
;
728 if (ShellStatus
!= SHELL_SUCCESS
) {
729 Status
= ShellCloseFileMetaArg(&FileList
);
732 // now copy them all...
734 if (FileList
!= NULL
&& !IsListEmpty(&FileList
->Link
)) {
735 ShellStatus
= ProcessValidateAndCopyFiles(FileList
, PathCleanUpDirectories((CHAR16
*)ShellCommandLineGetRawValue(Package
, ParamCount
)), SilentMode
, RecursiveMode
);
736 Status
= ShellCloseFileMetaArg(&FileList
);
737 if (EFI_ERROR(Status
) && ShellStatus
== SHELL_SUCCESS
) {
738 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_ERR_FILE
), gShellLevel2HiiHandle
, ShellCommandLineGetRawValue(Package
, ParamCount
), ShellStatus
|MAX_BIT
);
739 ShellStatus
= SHELL_ACCESS_DENIED
;
744 } // switch on parameter count
746 if (FileList
!= NULL
) {
747 ShellCloseFileMetaArg(&FileList
);
751 // free the command line package
753 ShellCommandLineFreeVarList (Package
);
756 if (ShellGetExecutionBreakFlag()) {
757 return (SHELL_ABORTED
);
760 return (ShellStatus
);