2010-09-14 07:18:09 +02:00
|
|
|
/** @file
|
|
|
|
Main file for cp shell level 2 function.
|
|
|
|
|
2011-03-25 21:58:08 +01:00
|
|
|
Copyright (c) 2009 - 2011, Intel Corporation. All rights reserved.<BR>
|
2010-09-14 07:18:09 +02:00
|
|
|
This program and the accompanying materials
|
|
|
|
are licensed and made available under the terms and conditions of the BSD License
|
|
|
|
which accompanies this distribution. The full text of the license may be found at
|
|
|
|
http://opensource.org/licenses/bsd-license.php
|
|
|
|
|
|
|
|
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
|
|
|
|
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
|
|
|
|
|
|
|
|
**/
|
|
|
|
|
|
|
|
#include "UefiShellLevel2CommandsLib.h"
|
|
|
|
|
2011-03-25 21:58:08 +01:00
|
|
|
/**
|
|
|
|
Function to take a list of files to copy and a destination location and do
|
|
|
|
the verification and copying of those files to that location. This function
|
|
|
|
will report any errors to the user and halt.
|
|
|
|
|
|
|
|
@param[in] FileList A LIST_ENTRY* based list of files to move.
|
|
|
|
@param[in] DestDir The destination location.
|
|
|
|
@param[in] SilentMode TRUE to eliminate screen output.
|
|
|
|
@param[in] RecursiveMode TRUE to copy directories.
|
|
|
|
@param[in] Resp The response to the overwrite query (if always).
|
|
|
|
|
|
|
|
@retval SHELL_SUCCESS the files were all moved.
|
|
|
|
@retval SHELL_INVALID_PARAMETER a parameter was invalid
|
|
|
|
@retval SHELL_SECURITY_VIOLATION a security violation ocurred
|
|
|
|
@retval SHELL_WRITE_PROTECTED the destination was write protected
|
|
|
|
@retval SHELL_OUT_OF_RESOURCES a memory allocation failed
|
|
|
|
**/
|
2010-09-14 07:18:09 +02:00
|
|
|
SHELL_STATUS
|
|
|
|
EFIAPI
|
|
|
|
ValidateAndCopyFiles(
|
|
|
|
IN CONST EFI_SHELL_FILE_INFO *FileList,
|
|
|
|
IN CONST CHAR16 *DestDir,
|
|
|
|
IN BOOLEAN SilentMode,
|
|
|
|
IN BOOLEAN RecursiveMode,
|
|
|
|
IN VOID **Resp
|
|
|
|
);
|
|
|
|
|
|
|
|
/**
|
|
|
|
Function to Copy one file to another location
|
|
|
|
|
|
|
|
If the destination exists the user will be prompted and the result put into *resp
|
|
|
|
|
|
|
|
@param[in] Source pointer to source file name
|
|
|
|
@param[in] Dest pointer to destination file name
|
|
|
|
@param[out] Resp pointer to response from question. Pass back on looped calling
|
|
|
|
@param[in] SilentMode whether to run in quiet mode or not
|
|
|
|
|
|
|
|
@retval SHELL_SUCCESS The source file was copied to the destination
|
|
|
|
**/
|
|
|
|
SHELL_STATUS
|
|
|
|
EFIAPI
|
|
|
|
CopySingleFile(
|
|
|
|
IN CONST CHAR16 *Source,
|
|
|
|
IN CONST CHAR16 *Dest,
|
|
|
|
OUT VOID **Resp,
|
|
|
|
IN BOOLEAN SilentMode
|
|
|
|
)
|
|
|
|
{
|
|
|
|
VOID *Response;
|
|
|
|
UINTN ReadSize;
|
|
|
|
SHELL_FILE_HANDLE SourceHandle;
|
|
|
|
SHELL_FILE_HANDLE DestHandle;
|
|
|
|
EFI_STATUS Status;
|
|
|
|
VOID *Buffer;
|
|
|
|
CHAR16 *TempName;
|
|
|
|
UINTN Size;
|
|
|
|
EFI_SHELL_FILE_INFO *List;
|
|
|
|
SHELL_STATUS ShellStatus;
|
|
|
|
|
|
|
|
|
|
|
|
ASSERT(Resp != NULL);
|
|
|
|
|
|
|
|
SourceHandle = NULL;
|
|
|
|
DestHandle = NULL;
|
|
|
|
Response = *Resp;
|
|
|
|
List = NULL;
|
|
|
|
|
|
|
|
ReadSize = PcdGet16(PcdShellFileOperationSize);
|
|
|
|
// Why bother copying a file to itself
|
|
|
|
if (StrCmp(Source, Dest) == 0) {
|
|
|
|
return (SHELL_SUCCESS);
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// Open destination file without create
|
|
|
|
//
|
|
|
|
Status = ShellOpenFileByName(Dest, &DestHandle, EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE, 0);
|
|
|
|
|
|
|
|
//
|
|
|
|
// close file
|
|
|
|
//
|
|
|
|
if (DestHandle != NULL) {
|
|
|
|
ShellCloseFile(&DestHandle);
|
|
|
|
DestHandle = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// if the destination file existed check response and possibly prompt user
|
|
|
|
//
|
|
|
|
if (!EFI_ERROR(Status)) {
|
|
|
|
if (Response == NULL && !SilentMode) {
|
2011-03-25 21:58:08 +01:00
|
|
|
Status = ShellPromptForResponseHii(ShellPromptResponseTypeYesNoAllCancel, STRING_TOKEN (STR_GEN_DEST_EXIST_OVR), gShellLevel2HiiHandle, &Response);
|
2010-09-14 07:18:09 +02:00
|
|
|
}
|
|
|
|
//
|
|
|
|
// possibly return based on response
|
|
|
|
//
|
|
|
|
if (!SilentMode) {
|
|
|
|
switch (*(SHELL_PROMPT_RESPONSE*)Response) {
|
|
|
|
case ShellPromptResponseNo:
|
|
|
|
//
|
|
|
|
// return success here so we dont stop the process
|
|
|
|
//
|
|
|
|
return (SHELL_SUCCESS);
|
|
|
|
case ShellPromptResponseCancel:
|
|
|
|
*Resp = Response;
|
|
|
|
//
|
|
|
|
// indicate to stop everything
|
|
|
|
//
|
|
|
|
return (SHELL_ABORTED);
|
|
|
|
case ShellPromptResponseAll:
|
|
|
|
*Resp = Response;
|
|
|
|
case ShellPromptResponseYes:
|
|
|
|
break;
|
2010-09-17 22:09:48 +02:00
|
|
|
default:
|
|
|
|
return SHELL_ABORTED;
|
2010-09-14 07:18:09 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (ShellIsDirectory(Source) == EFI_SUCCESS) {
|
|
|
|
Status = ShellCreateDirectory(Dest, &DestHandle);
|
|
|
|
if (EFI_ERROR(Status)) {
|
|
|
|
return (SHELL_ACCESS_DENIED);
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// Now copy all the files under the directory...
|
|
|
|
//
|
|
|
|
TempName = NULL;
|
|
|
|
Size = 0;
|
|
|
|
StrnCatGrow(&TempName, &Size, Source, 0);
|
|
|
|
StrnCatGrow(&TempName, &Size, L"\\*", 0);
|
|
|
|
ShellOpenFileMetaArg((CHAR16*)TempName, EFI_FILE_MODE_READ, &List);
|
|
|
|
TempName = NULL;
|
|
|
|
StrnCatGrow(&TempName, &Size, Dest, 0);
|
|
|
|
StrnCatGrow(&TempName, &Size, L"\\", 0);
|
|
|
|
ShellStatus = ValidateAndCopyFiles(List, TempName, SilentMode, TRUE, Resp);
|
|
|
|
ShellCloseFileMetaArg(&List);
|
|
|
|
FreePool(TempName);
|
|
|
|
Size = 0;
|
|
|
|
} else {
|
|
|
|
//
|
|
|
|
// open file with create enabled
|
|
|
|
//
|
|
|
|
Status = ShellOpenFileByName(Dest, &DestHandle, EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE|EFI_FILE_MODE_CREATE, 0);
|
|
|
|
if (EFI_ERROR(Status)) {
|
|
|
|
return (SHELL_ACCESS_DENIED);
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// open source file
|
|
|
|
//
|
|
|
|
Status = ShellOpenFileByName(Source, &SourceHandle, EFI_FILE_MODE_READ, 0);
|
|
|
|
ASSERT_EFI_ERROR(Status);
|
|
|
|
|
|
|
|
//
|
|
|
|
// copy data between files
|
|
|
|
//
|
|
|
|
Buffer = AllocateZeroPool(ReadSize);
|
|
|
|
ASSERT(Buffer != NULL);
|
|
|
|
while (ReadSize == PcdGet16(PcdShellFileOperationSize) && !EFI_ERROR(Status)) {
|
|
|
|
Status = ShellReadFile(SourceHandle, &ReadSize, Buffer);
|
|
|
|
ASSERT_EFI_ERROR(Status);
|
|
|
|
Status = ShellWriteFile(DestHandle, &ReadSize, Buffer);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// close files
|
|
|
|
//
|
|
|
|
if (DestHandle != NULL) {
|
|
|
|
ShellCloseFile(&DestHandle);
|
|
|
|
DestHandle = NULL;
|
|
|
|
}
|
|
|
|
if (SourceHandle != NULL) {
|
|
|
|
ShellCloseFile(&SourceHandle);
|
|
|
|
SourceHandle = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// return
|
|
|
|
//
|
|
|
|
return (SHELL_SUCCESS);
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
function to take a list of files to copy and a destination location and do
|
|
|
|
the verification and copying of those files to that location. This function
|
|
|
|
will report any errors to the user and halt.
|
|
|
|
|
|
|
|
The key is to have this function called ONLY once. this allows for the parameter
|
|
|
|
verification to happen correctly.
|
|
|
|
|
2011-03-25 21:58:08 +01:00
|
|
|
@param[in] FileList A LIST_ENTRY* based list of files to move.
|
|
|
|
@param[in] DestDir The destination location.
|
|
|
|
@param[in] SilentMode TRUE to eliminate screen output.
|
|
|
|
@param[in] RecursiveMode TRUE to copy directories.
|
|
|
|
@param[in] Resp The response to the overwrite query (if always).
|
2010-09-14 07:18:09 +02:00
|
|
|
|
|
|
|
@retval SHELL_SUCCESS the files were all moved.
|
|
|
|
@retval SHELL_INVALID_PARAMETER a parameter was invalid
|
|
|
|
@retval SHELL_SECURITY_VIOLATION a security violation ocurred
|
|
|
|
@retval SHELL_WRITE_PROTECTED the destination was write protected
|
|
|
|
@retval SHELL_OUT_OF_RESOURCES a memory allocation failed
|
|
|
|
**/
|
|
|
|
SHELL_STATUS
|
|
|
|
EFIAPI
|
|
|
|
ValidateAndCopyFiles(
|
|
|
|
IN CONST EFI_SHELL_FILE_INFO *FileList,
|
|
|
|
IN CONST CHAR16 *DestDir,
|
|
|
|
IN BOOLEAN SilentMode,
|
|
|
|
IN BOOLEAN RecursiveMode,
|
|
|
|
IN VOID **Resp
|
|
|
|
)
|
|
|
|
{
|
|
|
|
CHAR16 *HiiOutput;
|
|
|
|
CHAR16 *HiiResultOk;
|
|
|
|
CONST EFI_SHELL_FILE_INFO *Node;
|
|
|
|
SHELL_STATUS ShellStatus;
|
|
|
|
CHAR16 *DestPath;
|
|
|
|
VOID *Response;
|
|
|
|
UINTN PathLen;
|
|
|
|
CONST CHAR16 *Cwd;
|
|
|
|
CONST CHAR16 *TempLocation;
|
|
|
|
UINTN NewSize;
|
|
|
|
|
|
|
|
if (Resp == NULL) {
|
|
|
|
Response = NULL;
|
|
|
|
} else {
|
|
|
|
Response = *Resp;
|
|
|
|
}
|
|
|
|
|
|
|
|
DestPath = NULL;
|
|
|
|
ShellStatus = SHELL_SUCCESS;
|
|
|
|
PathLen = 0;
|
|
|
|
Cwd = ShellGetCurrentDir(NULL);
|
|
|
|
|
|
|
|
ASSERT(FileList != NULL);
|
|
|
|
ASSERT(DestDir != NULL);
|
|
|
|
|
|
|
|
//
|
|
|
|
// If we are trying to copy multiple files... make sure we got a directory for the target...
|
|
|
|
//
|
|
|
|
if (EFI_ERROR(ShellIsDirectory(DestDir)) && FileList->Link.ForwardLink != FileList->Link.BackLink) {
|
|
|
|
//
|
|
|
|
// Error for destination not a directory
|
|
|
|
//
|
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NOT_DIR), gShellLevel2HiiHandle, DestDir);
|
|
|
|
return (SHELL_INVALID_PARAMETER);
|
|
|
|
}
|
|
|
|
for (Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&FileList->Link)
|
|
|
|
; !IsNull(&FileList->Link, &Node->Link)
|
|
|
|
; Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&FileList->Link, &Node->Link)
|
|
|
|
){
|
|
|
|
//
|
|
|
|
// skip the directory traversing stuff...
|
|
|
|
//
|
|
|
|
if (StrCmp(Node->FileName, L".") == 0 || StrCmp(Node->FileName, L"..") == 0) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
NewSize = StrSize(DestDir);
|
2011-03-25 21:58:08 +01:00
|
|
|
NewSize += StrSize(Node->FullName);
|
2010-09-14 07:18:09 +02:00
|
|
|
NewSize += StrSize(Cwd);
|
|
|
|
if (NewSize > PathLen) {
|
|
|
|
PathLen = NewSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// Make sure got -r if required
|
|
|
|
//
|
|
|
|
if (!RecursiveMode && !EFI_ERROR(ShellIsDirectory(Node->FullName))) {
|
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_DIR_REQ), gShellLevel2HiiHandle);
|
|
|
|
return (SHELL_INVALID_PARAMETER);
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// make sure got dest as dir if needed
|
|
|
|
//
|
|
|
|
if (!EFI_ERROR(ShellIsDirectory(Node->FullName)) && EFI_ERROR(ShellIsDirectory(DestDir))) {
|
|
|
|
//
|
|
|
|
// Error for destination not a directory
|
|
|
|
//
|
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NOT_DIR), gShellLevel2HiiHandle, DestDir);
|
|
|
|
return (SHELL_INVALID_PARAMETER);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
HiiOutput = HiiGetString (gShellLevel2HiiHandle, STRING_TOKEN (STR_CP_OUTPUT), NULL);
|
|
|
|
HiiResultOk = HiiGetString (gShellLevel2HiiHandle, STRING_TOKEN (STR_GEN_RES_OK), NULL);
|
2011-03-25 21:58:08 +01:00
|
|
|
DestPath = AllocateZeroPool(PathLen);
|
2010-09-14 07:18:09 +02:00
|
|
|
|
2010-10-04 18:44:57 +02:00
|
|
|
if (DestPath == NULL || HiiOutput == NULL || HiiResultOk == NULL) {
|
2010-10-04 18:30:40 +02:00
|
|
|
SHELL_FREE_NON_NULL(DestPath);
|
|
|
|
SHELL_FREE_NON_NULL(HiiOutput);
|
|
|
|
SHELL_FREE_NON_NULL(HiiResultOk);
|
|
|
|
return (SHELL_OUT_OF_RESOURCES);
|
|
|
|
}
|
|
|
|
|
2010-09-14 07:18:09 +02:00
|
|
|
//
|
|
|
|
// Go through the list of files to copy...
|
|
|
|
//
|
|
|
|
for (Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&FileList->Link)
|
|
|
|
; !IsNull(&FileList->Link, &Node->Link)
|
|
|
|
; Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&FileList->Link, &Node->Link)
|
|
|
|
){
|
|
|
|
if (ShellGetExecutionBreakFlag()) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
ASSERT(Node->FileName != NULL);
|
|
|
|
ASSERT(Node->FullName != NULL);
|
|
|
|
|
|
|
|
//
|
|
|
|
// skip the directory traversing stuff...
|
|
|
|
//
|
|
|
|
if (StrCmp(Node->FileName, L".") == 0 || StrCmp(Node->FileName, L"..") == 0) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (FileList->Link.ForwardLink == FileList->Link.BackLink // 1 item
|
|
|
|
&& EFI_ERROR(ShellIsDirectory(DestDir)) // not an existing directory
|
|
|
|
) {
|
2011-03-25 21:58:08 +01:00
|
|
|
if (StrStr(DestDir, L":") == NULL) {
|
|
|
|
//
|
|
|
|
// simple copy of a single file
|
|
|
|
//
|
|
|
|
StrCpy(DestPath, Cwd);
|
|
|
|
if (DestPath[StrLen(DestPath)-1] != L'\\' && DestDir[0] != L'\\') {
|
|
|
|
StrCat(DestPath, L"\\");
|
|
|
|
} else if (DestPath[StrLen(DestPath)-1] == L'\\' && DestDir[0] == L'\\') {
|
|
|
|
((CHAR16*)DestPath)[StrLen(DestPath)-1] = CHAR_NULL;
|
|
|
|
}
|
|
|
|
StrCat(DestPath, DestDir);
|
|
|
|
} else {
|
|
|
|
StrCpy(DestPath, DestDir);
|
2010-09-14 07:18:09 +02:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
//
|
|
|
|
// we have multiple files or a directory in the DestDir
|
|
|
|
//
|
|
|
|
if (StrStr(DestDir, L":") == NULL) {
|
|
|
|
StrCpy(DestPath, Cwd);
|
|
|
|
if (DestPath[StrLen(DestPath)-1] != L'\\' && DestDir[0] != L'\\') {
|
|
|
|
StrCat(DestPath, L"\\");
|
|
|
|
} else if (DestPath[StrLen(DestPath)-1] == L'\\' && DestDir[0] == L'\\') {
|
|
|
|
((CHAR16*)DestPath)[StrLen(DestPath)-1] = CHAR_NULL;
|
|
|
|
}
|
|
|
|
StrCat(DestPath, DestDir);
|
|
|
|
if (DestDir[StrLen(DestDir)-1] != L'\\' && Node->FileName[0] != L'\\') {
|
|
|
|
StrCat(DestPath, L"\\");
|
|
|
|
} else if (DestDir[StrLen(DestDir)-1] == L'\\' && Node->FileName[0] == L'\\') {
|
|
|
|
((CHAR16*)DestPath)[StrLen(DestPath)-1] = CHAR_NULL;
|
|
|
|
}
|
|
|
|
StrCat(DestPath, Node->FileName);
|
|
|
|
|
|
|
|
} else {
|
|
|
|
StrCpy(DestPath, DestDir);
|
|
|
|
if (DestDir[StrLen(DestDir)-1] != L'\\' && Node->FileName[0] != L'\\') {
|
|
|
|
StrCat(DestPath, L"\\");
|
|
|
|
} else if (DestDir[StrLen(DestDir)-1] == L'\\' && Node->FileName[0] == L'\\') {
|
|
|
|
((CHAR16*)DestDir)[StrLen(DestDir)-1] = CHAR_NULL;
|
|
|
|
}
|
|
|
|
StrCat(DestPath, Node->FileName);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// Make sure the path exists
|
|
|
|
//
|
|
|
|
if (EFI_ERROR(VerifyIntermediateDirectories(DestPath))) {
|
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_DIR_WNF), gShellLevel2HiiHandle);
|
|
|
|
ShellStatus = SHELL_DEVICE_ERROR;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
if ( !EFI_ERROR(ShellIsDirectory(Node->FullName))
|
|
|
|
&& !EFI_ERROR(ShellIsDirectory(DestPath))
|
|
|
|
&& StrniCmp(Node->FullName, DestPath, StrLen(DestPath)) == NULL
|
|
|
|
){
|
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_SD_PARENT), gShellLevel2HiiHandle);
|
|
|
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (StringNoCaseCompare(&Node->FullName, &DestPath) == 0) {
|
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_SD_SAME), gShellLevel2HiiHandle);
|
|
|
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
if ((TempLocation = StrniCmp(Node->FullName, DestPath, StrLen(Node->FullName))) == 0
|
|
|
|
&& (DestPath[StrLen(Node->FullName)] == CHAR_NULL || DestPath[StrLen(Node->FullName)] == L'\\')
|
|
|
|
) {
|
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_SD_SAME), gShellLevel2HiiHandle);
|
|
|
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
CleanPath(DestPath);
|
|
|
|
|
|
|
|
ShellPrintEx(-1, -1, HiiOutput, Node->FullName, DestPath);
|
|
|
|
|
|
|
|
//
|
|
|
|
// copy single file...
|
|
|
|
//
|
|
|
|
ShellStatus = CopySingleFile(Node->FullName, DestPath, &Response, SilentMode);
|
|
|
|
if (ShellStatus != SHELL_SUCCESS) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (ShellStatus == SHELL_SUCCESS && Resp == NULL) {
|
|
|
|
ShellPrintEx(-1, -1, L"%s", HiiResultOk);
|
|
|
|
}
|
|
|
|
|
|
|
|
SHELL_FREE_NON_NULL(DestPath);
|
|
|
|
SHELL_FREE_NON_NULL(HiiOutput);
|
|
|
|
SHELL_FREE_NON_NULL(HiiResultOk);
|
2011-03-28 20:42:51 +02:00
|
|
|
if (Resp == NULL) {
|
2010-09-14 07:18:09 +02:00
|
|
|
SHELL_FREE_NON_NULL(Response);
|
|
|
|
}
|
|
|
|
|
|
|
|
return (ShellStatus);
|
|
|
|
}
|
|
|
|
|
2011-03-25 21:58:08 +01:00
|
|
|
/**
|
|
|
|
Validate and if successful copy all the files from the list into
|
|
|
|
destination directory.
|
|
|
|
|
|
|
|
@param[in] FileList The list of files to copy.
|
|
|
|
@param[in] DestDir The directory to copy files to.
|
|
|
|
@param[in] SilentMode TRUE to eliminate screen output.
|
|
|
|
@param[in] RecursiveMode TRUE to copy directories.
|
|
|
|
|
|
|
|
@retval SHELL_INVALID_PARAMETER A parameter was invalid.
|
|
|
|
@retval SHELL_SUCCESS The operation was successful.
|
|
|
|
**/
|
2010-09-14 07:18:09 +02:00
|
|
|
SHELL_STATUS
|
|
|
|
EFIAPI
|
|
|
|
ProcessValidateAndCopyFiles(
|
|
|
|
IN EFI_SHELL_FILE_INFO *FileList,
|
|
|
|
IN CONST CHAR16 *DestDir,
|
|
|
|
IN BOOLEAN SilentMode,
|
|
|
|
IN BOOLEAN RecursiveMode
|
|
|
|
)
|
|
|
|
{
|
|
|
|
SHELL_STATUS ShellStatus;
|
|
|
|
EFI_SHELL_FILE_INFO *List;
|
|
|
|
EFI_STATUS Status;
|
|
|
|
EFI_FILE_INFO *FileInfo;
|
|
|
|
|
|
|
|
List = NULL;
|
|
|
|
|
|
|
|
Status = ShellOpenFileMetaArg((CHAR16*)DestDir, EFI_FILE_MODE_READ, &List);
|
|
|
|
if (List != NULL && List->Link.ForwardLink != List->Link.BackLink) {
|
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_MARG_ERROR), gShellLevel2HiiHandle, DestDir);
|
|
|
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
|
|
|
ShellCloseFileMetaArg(&List);
|
|
|
|
} else if (List != NULL) {
|
|
|
|
ASSERT(((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink) != NULL);
|
|
|
|
ASSERT(((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink)->FullName != NULL);
|
|
|
|
FileInfo = NULL;
|
|
|
|
FileInfo = gEfiShellProtocol->GetFileInfo(((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink)->Handle);
|
|
|
|
ASSERT(FileInfo != NULL);
|
|
|
|
if ((FileInfo->Attribute & EFI_FILE_READ_ONLY) == 0) {
|
|
|
|
ShellStatus = ValidateAndCopyFiles(FileList, ((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink)->FullName, SilentMode, RecursiveMode, NULL);
|
|
|
|
} else {
|
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_DEST_ERROR), gShellLevel2HiiHandle);
|
|
|
|
ShellStatus = SHELL_ACCESS_DENIED;
|
|
|
|
}
|
|
|
|
SHELL_FREE_NON_NULL(FileInfo);
|
|
|
|
ShellCloseFileMetaArg(&List);
|
|
|
|
} else {
|
|
|
|
ShellStatus = ValidateAndCopyFiles(FileList, DestDir, SilentMode, RecursiveMode, NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
return (ShellStatus);
|
|
|
|
}
|
|
|
|
|
|
|
|
STATIC CONST SHELL_PARAM_ITEM ParamList[] = {
|
|
|
|
{L"-r", TypeFlag},
|
|
|
|
{L"-q", TypeFlag},
|
|
|
|
{NULL, TypeMax}
|
|
|
|
};
|
|
|
|
|
|
|
|
/**
|
|
|
|
Function for 'cp' command.
|
|
|
|
|
|
|
|
@param[in] ImageHandle Handle to the Image (NULL if Internal).
|
|
|
|
@param[in] SystemTable Pointer to the System Table (NULL if Internal).
|
|
|
|
**/
|
|
|
|
SHELL_STATUS
|
|
|
|
EFIAPI
|
|
|
|
ShellCommandRunCp (
|
|
|
|
IN EFI_HANDLE ImageHandle,
|
|
|
|
IN EFI_SYSTEM_TABLE *SystemTable
|
|
|
|
)
|
|
|
|
{
|
|
|
|
EFI_STATUS Status;
|
|
|
|
LIST_ENTRY *Package;
|
|
|
|
CHAR16 *ProblemParam;
|
|
|
|
SHELL_STATUS ShellStatus;
|
|
|
|
UINTN ParamCount;
|
|
|
|
UINTN LoopCounter;
|
|
|
|
EFI_SHELL_FILE_INFO *FileList;
|
|
|
|
BOOLEAN SilentMode;
|
|
|
|
BOOLEAN RecursiveMode;
|
|
|
|
CONST CHAR16 *Cwd;
|
|
|
|
|
|
|
|
ProblemParam = NULL;
|
|
|
|
ShellStatus = SHELL_SUCCESS;
|
|
|
|
ParamCount = 0;
|
|
|
|
FileList = NULL;
|
|
|
|
|
|
|
|
//
|
|
|
|
// initialize the shell lib (we must be in non-auto-init...)
|
|
|
|
//
|
|
|
|
Status = ShellInitialize();
|
|
|
|
ASSERT_EFI_ERROR(Status);
|
|
|
|
|
|
|
|
Status = CommandInit();
|
|
|
|
ASSERT_EFI_ERROR(Status);
|
|
|
|
|
|
|
|
//
|
|
|
|
// parse the command line
|
|
|
|
//
|
|
|
|
Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);
|
|
|
|
if (EFI_ERROR(Status)) {
|
|
|
|
if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {
|
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, ProblemParam);
|
|
|
|
FreePool(ProblemParam);
|
|
|
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
|
|
|
} else {
|
|
|
|
ASSERT(FALSE);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
//
|
|
|
|
// check for "-?"
|
|
|
|
//
|
|
|
|
if (ShellCommandLineGetFlag(Package, L"-?")) {
|
|
|
|
ASSERT(FALSE);
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// Initialize SilentMode and RecursiveMode
|
|
|
|
//
|
|
|
|
if (gEfiShellProtocol->BatchIsActive()) {
|
|
|
|
SilentMode = TRUE;
|
|
|
|
} else {
|
|
|
|
SilentMode = ShellCommandLineGetFlag(Package, L"-q");
|
|
|
|
}
|
|
|
|
RecursiveMode = ShellCommandLineGetFlag(Package, L"-r");
|
|
|
|
|
|
|
|
switch (ParamCount = ShellCommandLineGetCount(Package)) {
|
|
|
|
case 0:
|
|
|
|
case 1:
|
|
|
|
//
|
|
|
|
// we have insufficient parameters
|
|
|
|
//
|
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel2HiiHandle);
|
|
|
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
|
|
|
break;
|
|
|
|
case 2:
|
|
|
|
//
|
|
|
|
// must have valid CWD for single parameter...
|
|
|
|
//
|
|
|
|
Cwd = ShellGetCurrentDir(NULL);
|
|
|
|
if (Cwd == NULL){
|
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellLevel2HiiHandle);
|
|
|
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
|
|
|
} else {
|
|
|
|
Status = ShellOpenFileMetaArg((CHAR16*)ShellCommandLineGetRawValue(Package, 1), EFI_FILE_MODE_WRITE|EFI_FILE_MODE_READ, &FileList);
|
|
|
|
if (FileList == NULL || IsListEmpty(&FileList->Link) || EFI_ERROR(Status)) {
|
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_NF), gShellLevel2HiiHandle, ShellCommandLineGetRawValue(Package, 1));
|
|
|
|
ShellStatus = SHELL_NOT_FOUND;
|
|
|
|
} else {
|
|
|
|
ShellStatus = ProcessValidateAndCopyFiles(FileList, Cwd, SilentMode, RecursiveMode);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
//
|
|
|
|
// Make a big list of all the files...
|
|
|
|
//
|
|
|
|
for (ParamCount--, LoopCounter = 1 ; LoopCounter < ParamCount && ShellStatus == SHELL_SUCCESS ; LoopCounter++) {
|
|
|
|
if (ShellGetExecutionBreakFlag()) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
Status = ShellOpenFileMetaArg((CHAR16*)ShellCommandLineGetRawValue(Package, LoopCounter), EFI_FILE_MODE_WRITE|EFI_FILE_MODE_READ, &FileList);
|
|
|
|
if (EFI_ERROR(Status) || FileList == NULL || IsListEmpty(&FileList->Link)) {
|
2011-03-30 23:37:01 +02:00
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_NF), gShellLevel2HiiHandle, ShellCommandLineGetRawValue(Package, LoopCounter));
|
2010-09-14 07:18:09 +02:00
|
|
|
ShellStatus = SHELL_NOT_FOUND;
|
|
|
|
}
|
|
|
|
}
|
2011-03-30 23:37:01 +02:00
|
|
|
if (ShellStatus != SHELL_SUCCESS) {
|
2010-09-14 07:18:09 +02:00
|
|
|
Status = ShellCloseFileMetaArg(&FileList);
|
2011-03-30 23:37:01 +02:00
|
|
|
} else {
|
|
|
|
//
|
|
|
|
// now copy them all...
|
|
|
|
//
|
|
|
|
if (FileList != NULL && !IsListEmpty(&FileList->Link)) {
|
|
|
|
ShellStatus = ProcessValidateAndCopyFiles(FileList, ShellCommandCleanPath((CHAR16*)ShellCommandLineGetRawValue(Package, ParamCount)), SilentMode, RecursiveMode);
|
|
|
|
Status = ShellCloseFileMetaArg(&FileList);
|
|
|
|
if (EFI_ERROR(Status) && ShellStatus == SHELL_SUCCESS) {
|
|
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_FILE), gShellLevel2HiiHandle, ShellCommandLineGetRawValue(Package, ParamCount), ShellStatus|MAX_BIT);
|
|
|
|
ShellStatus = SHELL_ACCESS_DENIED;
|
|
|
|
}
|
2010-09-14 07:18:09 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
} // switch on parameter count
|
|
|
|
|
|
|
|
if (FileList != NULL) {
|
|
|
|
ShellCloseFileMetaArg(&FileList);
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// free the command line package
|
|
|
|
//
|
|
|
|
ShellCommandLineFreeVarList (Package);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (ShellGetExecutionBreakFlag()) {
|
|
|
|
return (SHELL_ABORTED);
|
|
|
|
}
|
|
|
|
|
|
|
|
return (ShellStatus);
|
|
|
|
}
|
|
|
|
|