udk2010.up2.shell initial release.
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10874 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
jcarsey
599
ShellPkg/Library/UefiShellLevel2CommandsLib/Cp.c
Normal file
599
ShellPkg/Library/UefiShellLevel2CommandsLib/Cp.c
Normal file
@ -0,0 +1,599 @@
|
||||
/** @file
|
||||
Main file for cp shell level 2 function.
|
||||
|
||||
Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>
|
||||
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"
|
||||
|
||||
// this is later in the file.
|
||||
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) {
|
||||
Status = ShellPromptForResponseHii(ShellPromptResponseTypeYesNoAllCancel, STRING_TOKEN (STR_CP_PROMPT), gShellLevel2HiiHandle, &Response);
|
||||
}
|
||||
//
|
||||
// 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;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
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.
|
||||
|
||||
@param[in] FileList A LIST_ENTRY* based list of files to move
|
||||
@param[in] DestDir the destination location
|
||||
|
||||
@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);
|
||||
NewSize += StrSize(Node->FileName);
|
||||
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);
|
||||
DestPath = AllocatePool(PathLen);
|
||||
|
||||
//
|
||||
// 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
|
||||
) {
|
||||
ASSERT(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 {
|
||||
//
|
||||
// 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);
|
||||
if (Resp != NULL) {
|
||||
SHELL_FREE_NON_NULL(Response);
|
||||
}
|
||||
|
||||
return (ShellStatus);
|
||||
}
|
||||
|
||||
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(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)) {
|
||||
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_NF), gShellLevel2HiiHandle, ShellCommandLineGetRawValue(Package, 1));
|
||||
ShellStatus = SHELL_NOT_FOUND;
|
||||
}
|
||||
}
|
||||
//
|
||||
// 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, 1), ShellStatus|MAX_BIT);
|
||||
ShellStatus = SHELL_ACCESS_DENIED;
|
||||
}
|
||||
}
|
||||
|
||||
break;
|
||||
} // switch on parameter count
|
||||
|
||||
if (FileList != NULL) {
|
||||
ShellCloseFileMetaArg(&FileList);
|
||||
}
|
||||
|
||||
//
|
||||
// free the command line package
|
||||
//
|
||||
ShellCommandLineFreeVarList (Package);
|
||||
}
|
||||
|
||||
if (ShellGetExecutionBreakFlag()) {
|
||||
return (SHELL_ABORTED);
|
||||
}
|
||||
|
||||
return (ShellStatus);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user