/** @file | |
Main file for mv shell level 2 function. | |
(C) Copyright 2013-2015 Hewlett-Packard Development Company, L.P.<BR> | |
Copyright (c) 2009 - 2014, 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" | |
/** | |
function to determine if a move is between file systems. | |
@param FullName [in] The name of the file to move. | |
@param Cwd [in] The current working directory | |
@param DestPath [in] The target location to move to | |
@retval TRUE The move is across file system. | |
@retval FALSE The move is within a file system. | |
**/ | |
BOOLEAN | |
EFIAPI | |
IsBetweenFileSystem( | |
IN CONST CHAR16 *FullName, | |
IN CONST CHAR16 *Cwd, | |
IN CONST CHAR16 *DestPath | |
) | |
{ | |
CHAR16 *Test; | |
CHAR16 *Test1; | |
UINTN Result; | |
Test = StrStr(FullName, L":"); | |
if (Test == NULL && Cwd != NULL) { | |
Test = StrStr(Cwd, L":"); | |
} | |
Test1 = StrStr(DestPath, L":"); | |
if (Test1 == NULL && Cwd != NULL) { | |
Test1 = StrStr(Cwd, L":"); | |
} | |
if (Test1 != NULL && Test != NULL) { | |
*Test = CHAR_NULL; | |
*Test1 = CHAR_NULL; | |
Result = StringNoCaseCompare(&FullName, &DestPath); | |
*Test = L':'; | |
*Test1 = L':'; | |
if (Result != 0) { | |
return (TRUE); | |
} | |
} | |
return (FALSE); | |
} | |
/** | |
Function to validate that moving a specific file (FileName) to a specific | |
location (DestPath) is valid. | |
This function will verify that the destination is not a subdirectory of | |
FullName, that the Current working Directory is not being moved, and that | |
the directory is not read only. | |
if the move is invalid this function will report the error to StdOut. | |
@param SourcePath [in] The name of the file to move. | |
@param Cwd [in] The current working directory | |
@param DestPath [in] The target location to move to | |
@param Attribute [in] The Attribute of the file | |
@param DestAttr [in] The Attribute of the destination | |
@param FileStatus [in] The Status of the file when opened | |
@retval TRUE The move is valid | |
@retval FALSE The move is not | |
**/ | |
BOOLEAN | |
EFIAPI | |
IsValidMove( | |
IN CONST CHAR16 *SourcePath, | |
IN CONST CHAR16 *Cwd, | |
IN CONST CHAR16 *DestPath, | |
IN CONST UINT64 Attribute, | |
IN CONST UINT64 DestAttr, | |
IN CONST EFI_STATUS FileStatus | |
) | |
{ | |
CHAR16 *DestPathCopy; | |
CHAR16 *DestPathWalker; | |
if (Cwd != NULL && StrCmp(SourcePath, Cwd) == 0) { | |
// | |
// Invalid move | |
// | |
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_MV_INV_CWD), gShellLevel2HiiHandle); | |
return (FALSE); | |
} | |
// | |
// invalid to move read only or move to a read only destination | |
// | |
if (((Attribute & EFI_FILE_READ_ONLY) != 0) | |
|| (FileStatus == EFI_WRITE_PROTECTED) | |
|| ((DestAttr & EFI_FILE_READ_ONLY) != 0) | |
) { | |
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_MV_INV_RO), gShellLevel2HiiHandle, SourcePath); | |
return (FALSE); | |
} | |
DestPathCopy = AllocateCopyPool(StrSize(DestPath), DestPath); | |
if (DestPathCopy == NULL) { | |
return (FALSE); | |
} | |
for (DestPathWalker = DestPathCopy; *DestPathWalker == L'\\'; DestPathWalker++) ; | |
while(DestPathWalker != NULL && DestPathWalker[StrLen(DestPathWalker)-1] == L'\\') { | |
DestPathWalker[StrLen(DestPathWalker)-1] = CHAR_NULL; | |
} | |
ASSERT(DestPathWalker != NULL); | |
ASSERT(SourcePath != NULL); | |
// | |
// If they're the same, or if source is "above" dest on file path tree | |
// | |
if ( StrCmp(DestPathWalker, SourcePath) == 0 | |
|| StrStr(DestPathWalker, SourcePath) == DestPathWalker | |
) { | |
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_MV_INV_SUB), gShellLevel2HiiHandle); | |
FreePool(DestPathCopy); | |
return (FALSE); | |
} | |
FreePool(DestPathCopy); | |
return (TRUE); | |
} | |
/** | |
Function to take a destination path that might contain wildcards and verify | |
that there is only a single possible target (IE we cant have wildcards that | |
have 2 possible destination). | |
if the result is sucessful the caller must free *DestPathPointer. | |
@param[in] DestParameter The original path to the destination. | |
@param[in, out] DestPathPointer A pointer to the callee allocated final path. | |
@param[in] Cwd A pointer to the current working directory. | |
@param[in] SingleSource TRUE to have only one source file. | |
@param[in, out] DestAttr A pointer to the destination information attribute. | |
@retval SHELL_INVALID_PARAMETER The DestParameter could not be resolved to a location. | |
@retval SHELL_INVALID_PARAMETER The DestParameter could be resolved to more than 1 location. | |
@retval SHELL_INVALID_PARAMETER Cwd is required and is NULL. | |
@retval SHELL_SUCCESS The operation was sucessful. | |
**/ | |
SHELL_STATUS | |
EFIAPI | |
GetDestinationLocation( | |
IN CONST CHAR16 *DestParameter, | |
IN OUT CHAR16 **DestPathPointer, | |
IN CONST CHAR16 *Cwd, | |
IN CONST BOOLEAN SingleSource, | |
IN OUT UINT64 *DestAttr | |
) | |
{ | |
EFI_SHELL_FILE_INFO *DestList; | |
EFI_SHELL_FILE_INFO *Node; | |
CHAR16 *DestPath; | |
UINTN NewSize; | |
UINTN CurrentSize; | |
DestList = NULL; | |
DestPath = NULL; | |
ASSERT(DestAttr != NULL); | |
if (StrStr(DestParameter, L"\\") == DestParameter) { | |
if (Cwd == NULL) { | |
return SHELL_INVALID_PARAMETER; | |
} | |
DestPath = AllocateZeroPool(StrSize(Cwd)); | |
if (DestPath == NULL) { | |
return (SHELL_OUT_OF_RESOURCES); | |
} | |
StrCpy(DestPath, Cwd); | |
while (PathRemoveLastItem(DestPath)) ; | |
// | |
// Append DestParameter beyond '\' which may be present | |
// | |
CurrentSize = StrSize(DestPath); | |
StrnCatGrow(&DestPath, &CurrentSize, &DestParameter[1], 0); | |
*DestPathPointer = DestPath; | |
return (SHELL_SUCCESS); | |
} | |
// | |
// get the destination path | |
// | |
ShellOpenFileMetaArg((CHAR16*)DestParameter, EFI_FILE_MODE_WRITE|EFI_FILE_MODE_READ|EFI_FILE_MODE_CREATE, &DestList); | |
if (DestList == NULL || IsListEmpty(&DestList->Link)) { | |
// | |
// Not existing... must be renaming | |
// | |
if (StrStr(DestParameter, L":") == NULL) { | |
if (Cwd == NULL) { | |
ShellCloseFileMetaArg(&DestList); | |
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellLevel2HiiHandle); | |
return (SHELL_INVALID_PARAMETER); | |
} | |
NewSize = StrSize(Cwd); | |
NewSize += StrSize(DestParameter); | |
DestPath = AllocateZeroPool(NewSize); | |
if (DestPath == NULL) { | |
ShellCloseFileMetaArg(&DestList); | |
return (SHELL_OUT_OF_RESOURCES); | |
} | |
StrCpy(DestPath, Cwd); | |
if (DestPath[StrLen(DestPath)-1] != L'\\' && DestParameter[0] != L'\\') { | |
StrCat(DestPath, L"\\"); | |
} else if (DestPath[StrLen(DestPath)-1] == L'\\' && DestParameter[0] == L'\\') { | |
((CHAR16*)DestPath)[StrLen(DestPath)-1] = CHAR_NULL; | |
} | |
StrCat(DestPath, DestParameter); | |
} else { | |
ASSERT(DestPath == NULL); | |
DestPath = StrnCatGrow(&DestPath, NULL, DestParameter, 0); | |
if (DestPath == NULL) { | |
ShellCloseFileMetaArg(&DestList); | |
return (SHELL_OUT_OF_RESOURCES); | |
} | |
} | |
} else { | |
Node = (EFI_SHELL_FILE_INFO*)GetFirstNode(&DestList->Link); | |
*DestAttr = Node->Info->Attribute; | |
// | |
// Make sure there is only 1 node in the list. | |
// | |
if (!IsNodeAtEnd(&DestList->Link, &Node->Link)) { | |
ShellCloseFileMetaArg(&DestList); | |
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_MARG_ERROR), gShellLevel2HiiHandle, L"mv", DestParameter); | |
return (SHELL_INVALID_PARAMETER); | |
} | |
// | |
// If we are a directory or a single file, then one node is fine. | |
// | |
if (ShellIsDirectory(Node->FullName)==EFI_SUCCESS || SingleSource) { | |
DestPath = AllocateZeroPool(StrSize(Node->FullName)+sizeof(CHAR16)); | |
if (DestPath == NULL) { | |
ShellCloseFileMetaArg(&DestList); | |
return (SHELL_OUT_OF_RESOURCES); | |
} | |
StrCpy(DestPath, Node->FullName); | |
StrCat(DestPath, L"\\"); | |
} else { | |
// | |
// cant move multiple files onto a single file. | |
// | |
ShellCloseFileMetaArg(&DestList); | |
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_ERROR), gShellLevel2HiiHandle, L"mv", DestParameter); | |
return (SHELL_INVALID_PARAMETER); | |
} | |
} | |
*DestPathPointer = DestPath; | |
ShellCloseFileMetaArg(&DestList); | |
return (SHELL_SUCCESS); | |
} | |
/** | |
Function to do a move across file systems. | |
@param[in] Node A pointer to the file to be removed. | |
@param[in] DestPath A pointer to the destination file path. | |
@param[out] Resp A pointer to response from question. Pass back on looped calling | |
@retval SHELL_SUCCESS The source file was moved to the destination. | |
**/ | |
EFI_STATUS | |
EFIAPI | |
MoveBetweenFileSystems( | |
IN EFI_SHELL_FILE_INFO *Node, | |
IN CONST CHAR16 *DestPath, | |
OUT VOID **Resp | |
) | |
{ | |
EFI_STATUS Status; | |
// | |
// First we copy the file | |
// | |
Status = CopySingleFile(Node->FullName, DestPath, Resp, TRUE); | |
// | |
// Check our result | |
// | |
if (!EFI_ERROR(Status)) { | |
// | |
// The copy was successful. delete the source file. | |
// | |
CascadeDelete(Node, TRUE); | |
Node->Handle = NULL; | |
} | |
return (Status); | |
} | |
/** | |
Function to take the destination path and target file name to generate the full destination path. | |
@param[in] DestPath A pointer to the destination file path string. | |
@param[out] FullDestPath A pointer to the full destination path string. | |
@param[in] FileName Name string of the targe file. | |
@retval SHELL_SUCCESS the files were all moved. | |
@retval SHELL_INVALID_PARAMETER a parameter was invalid | |
@retval SHELL_OUT_OF_RESOURCES a memory allocation failed | |
**/ | |
EFI_STATUS | |
EFIAPI | |
CreateFullDestPath( | |
IN CONST CHAR16 **DestPath, | |
OUT CHAR16 **FullDestPath, | |
IN CONST CHAR16 *FileName | |
) | |
{ | |
UINTN Size; | |
if (FullDestPath == NULL || FileName == NULL || DestPath == NULL || *DestPath == NULL){ | |
return (EFI_INVALID_PARAMETER); | |
} | |
Size = StrSize(*DestPath) + StrSize(FileName); | |
*FullDestPath = AllocateZeroPool(Size); | |
if (*FullDestPath == NULL){ | |
return (EFI_OUT_OF_RESOURCES); | |
} | |
StrnCpy(*FullDestPath, *DestPath, Size / sizeof(CHAR16) - 1); | |
if ((*FullDestPath)[StrLen(*FullDestPath)-1] != L'\\' && FileName[0] != L'\\') { | |
StrnCat(*FullDestPath, L"\\",Size / sizeof(CHAR16) - 1 - StrLen(*FullDestPath)); | |
} | |
StrnCat(*FullDestPath, FileName, Size / sizeof(CHAR16) - 1 - StrLen(*FullDestPath)); | |
return (EFI_SUCCESS); | |
} | |
/** | |
Function to do a move within a file system. | |
@param[in] Node A pointer to the file to be removed. | |
@param[in] DestPath A pointer to the destination file path. | |
@param[out] Resp A pointer to response from question. Pass back on looped calling. | |
@retval SHELL_SUCCESS The source file was moved to the destination. | |
@retval SHELL_OUT_OF_RESOURCES A memory allocation failed. | |
**/ | |
EFI_STATUS | |
EFIAPI | |
MoveWithinFileSystems( | |
IN EFI_SHELL_FILE_INFO *Node, | |
IN CHAR16 *DestPath, | |
OUT VOID **Resp | |
) | |
{ | |
EFI_FILE_INFO *NewFileInfo; | |
CHAR16 *TempLocation; | |
UINTN NewSize; | |
UINTN Length; | |
EFI_STATUS Status; | |
// | |
// Chop off map info from DestPath | |
// | |
if ((TempLocation = StrStr(DestPath, L":")) != NULL) { | |
CopyMem(DestPath, TempLocation+1, StrSize(TempLocation+1)); | |
} | |
// | |
// construct the new file info block | |
// | |
NewSize = StrSize(DestPath); | |
NewSize += StrSize(Node->FileName) + SIZE_OF_EFI_FILE_INFO + sizeof(CHAR16); | |
NewFileInfo = AllocateZeroPool(NewSize); | |
if (NewFileInfo == NULL) { | |
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_MEM), gShellLevel2HiiHandle); | |
Status = EFI_OUT_OF_RESOURCES; | |
} else { | |
CopyMem(NewFileInfo, Node->Info, SIZE_OF_EFI_FILE_INFO); | |
if (DestPath[0] != L'\\') { | |
StrCpy(NewFileInfo->FileName, L"\\"); | |
StrCat(NewFileInfo->FileName, DestPath); | |
} else { | |
StrCpy(NewFileInfo->FileName, DestPath); | |
} | |
Length = StrLen(NewFileInfo->FileName); | |
if (Length > 0) { | |
Length--; | |
} | |
if (NewFileInfo->FileName[Length] == L'\\') { | |
if (Node->FileName[0] == L'\\') { | |
// | |
// Don't allow for double slashes. Eliminate one of them. | |
// | |
NewFileInfo->FileName[Length] = CHAR_NULL; | |
} | |
StrCat(NewFileInfo->FileName, Node->FileName); | |
} | |
NewFileInfo->Size = SIZE_OF_EFI_FILE_INFO + StrSize(NewFileInfo->FileName); | |
// | |
// Perform the move operation | |
// | |
Status = ShellSetFileInfo(Node->Handle, NewFileInfo); | |
// | |
// Free the info object we used... | |
// | |
FreePool(NewFileInfo); | |
} | |
return (Status); | |
} | |
/** | |
function to take a list of files to move and a destination location and do | |
the verification and moving of those files to that location. This function | |
will report any errors to the user and continue to move the rest of the files. | |
@param[in] FileList A LIST_ENTRY* based list of files to move | |
@param[out] Resp pointer to response from question. Pass back on looped calling | |
@param[in] DestParameter the originally specified 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 | |
ValidateAndMoveFiles( | |
IN EFI_SHELL_FILE_INFO *FileList, | |
OUT VOID **Resp, | |
IN CONST CHAR16 *DestParameter | |
) | |
{ | |
EFI_STATUS Status; | |
CHAR16 *HiiOutput; | |
CHAR16 *HiiResultOk; | |
CHAR16 *DestPath; | |
CHAR16 *FullDestPath; | |
CONST CHAR16 *Cwd; | |
SHELL_STATUS ShellStatus; | |
EFI_SHELL_FILE_INFO *Node; | |
VOID *Response; | |
UINT64 Attr; | |
CHAR16 *CleanFilePathStr; | |
ASSERT(FileList != NULL); | |
ASSERT(DestParameter != NULL); | |
DestPath = NULL; | |
FullDestPath = NULL; | |
Cwd = ShellGetCurrentDir(NULL); | |
Response = *Resp; | |
Attr = 0; | |
CleanFilePathStr = NULL; | |
Status = ShellLevel2StripQuotes (DestParameter, &CleanFilePathStr); | |
if (EFI_ERROR (Status)) { | |
if (Status == EFI_OUT_OF_RESOURCES) { | |
return SHELL_OUT_OF_RESOURCES; | |
} else { | |
return SHELL_INVALID_PARAMETER; | |
} | |
} | |
ASSERT (CleanFilePathStr != NULL); | |
// | |
// Get and validate the destination location | |
// | |
ShellStatus = GetDestinationLocation(CleanFilePathStr, &DestPath, Cwd, (BOOLEAN)(FileList->Link.ForwardLink == FileList->Link.BackLink), &Attr); | |
FreePool (CleanFilePathStr); | |
if (ShellStatus != SHELL_SUCCESS) { | |
return (ShellStatus); | |
} | |
DestPath = PathCleanUpDirectories(DestPath); | |
if (DestPath == NULL) { | |
return (SHELL_OUT_OF_RESOURCES); | |
} | |
HiiOutput = HiiGetString (gShellLevel2HiiHandle, STRING_TOKEN (STR_MV_OUTPUT), NULL); | |
HiiResultOk = HiiGetString (gShellLevel2HiiHandle, STRING_TOKEN (STR_GEN_RES_OK), NULL); | |
if (HiiOutput == NULL || HiiResultOk == NULL) { | |
SHELL_FREE_NON_NULL(DestPath); | |
SHELL_FREE_NON_NULL(HiiOutput); | |
SHELL_FREE_NON_NULL(HiiResultOk); | |
return (SHELL_OUT_OF_RESOURCES); | |
} | |
// | |
// Go through the list of files and directories to move... | |
// | |
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; | |
} | |
// | |
// These should never be NULL | |
// | |
ASSERT(Node->FileName != NULL); | |
ASSERT(Node->FullName != NULL); | |
ASSERT(Node->Info != NULL); | |
// | |
// skip the directory traversing stuff... | |
// | |
if (StrCmp(Node->FileName, L".") == 0 || StrCmp(Node->FileName, L"..") == 0) { | |
continue; | |
} | |
SHELL_FREE_NON_NULL(FullDestPath); | |
FullDestPath = NULL; | |
if (ShellIsDirectory(DestPath)==EFI_SUCCESS) { | |
CreateFullDestPath((CONST CHAR16 **)&DestPath, &FullDestPath, Node->FileName); | |
} | |
// | |
// Validate that the move is valid | |
// | |
if (!IsValidMove(Node->FullName, Cwd, FullDestPath!=NULL? FullDestPath:DestPath, Node->Info->Attribute, Attr, Node->Status)) { | |
ShellStatus = SHELL_INVALID_PARAMETER; | |
continue; | |
} | |
ShellPrintEx(-1, -1, HiiOutput, Node->FullName, FullDestPath!=NULL? FullDestPath:DestPath); | |
// | |
// See if destination exists | |
// | |
if (!EFI_ERROR(ShellFileExists(FullDestPath!=NULL? FullDestPath:DestPath))) { | |
if (Response == NULL) { | |
ShellPromptForResponseHii(ShellPromptResponseTypeYesNoAllCancel, STRING_TOKEN (STR_GEN_DEST_EXIST_OVR), gShellLevel2HiiHandle, &Response); | |
} | |
switch (*(SHELL_PROMPT_RESPONSE*)Response) { | |
case ShellPromptResponseNo: | |
FreePool(Response); | |
Response = NULL; | |
continue; | |
case ShellPromptResponseCancel: | |
*Resp = Response; | |
// | |
// indicate to stop everything | |
// | |
return (SHELL_ABORTED); | |
case ShellPromptResponseAll: | |
*Resp = Response; | |
break; | |
case ShellPromptResponseYes: | |
FreePool(Response); | |
Response = NULL; | |
break; | |
default: | |
FreePool(Response); | |
return SHELL_ABORTED; | |
} | |
Status = ShellDeleteFileByName(FullDestPath!=NULL? FullDestPath:DestPath); | |
} | |
if (IsBetweenFileSystem(Node->FullName, Cwd, DestPath)) { | |
while (FullDestPath == NULL && DestPath != NULL && DestPath[0] != CHAR_NULL && DestPath[StrLen(DestPath) - 1] == L'\\') { | |
DestPath[StrLen(DestPath) - 1] = CHAR_NULL; | |
} | |
Status = MoveBetweenFileSystems(Node, FullDestPath!=NULL? FullDestPath:DestPath, &Response); | |
} else { | |
Status = MoveWithinFileSystems(Node, DestPath, &Response); | |
} | |
// | |
// Check our result | |
// | |
if (EFI_ERROR(Status)) { | |
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_UK), gShellLevel2HiiHandle, Status); | |
ShellStatus = SHELL_INVALID_PARAMETER; | |
if (Status == EFI_SECURITY_VIOLATION) { | |
ShellStatus = SHELL_SECURITY_VIOLATION; | |
} else if (Status == EFI_WRITE_PROTECTED) { | |
ShellStatus = SHELL_WRITE_PROTECTED; | |
} else if (Status == EFI_OUT_OF_RESOURCES) { | |
ShellStatus = SHELL_OUT_OF_RESOURCES; | |
} else if (Status == EFI_DEVICE_ERROR) { | |
ShellStatus = SHELL_DEVICE_ERROR; | |
} else if (Status == EFI_ACCESS_DENIED) { | |
ShellStatus = SHELL_ACCESS_DENIED; | |
} | |
} else { | |
ShellPrintEx(-1, -1, L"%s", HiiResultOk); | |
} | |
} // main for loop | |
SHELL_FREE_NON_NULL(FullDestPath); | |
SHELL_FREE_NON_NULL(DestPath); | |
SHELL_FREE_NON_NULL(HiiOutput); | |
SHELL_FREE_NON_NULL(HiiResultOk); | |
return (ShellStatus); | |
} | |
/** | |
Function for 'mv' 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 | |
ShellCommandRunMv ( | |
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; | |
VOID *Response; | |
ProblemParam = NULL; | |
ShellStatus = SHELL_SUCCESS; | |
ParamCount = 0; | |
FileList = NULL; | |
Response = NULL; | |
// | |
// initialize the shell lib (we must be in non-auto-init...) | |
// | |
Status = ShellInitialize(); | |
ASSERT_EFI_ERROR(Status); | |
// | |
// parse the command line | |
// | |
Status = ShellCommandLineParse (EmptyParamList, &Package, &ProblemParam, TRUE); | |
if (EFI_ERROR(Status)) { | |
if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) { | |
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, L"mv", ProblemParam); | |
FreePool(ProblemParam); | |
ShellStatus = SHELL_INVALID_PARAMETER; | |
} else { | |
ASSERT(FALSE); | |
} | |
} else { | |
// | |
// check for "-?" | |
// | |
if (ShellCommandLineGetFlag(Package, L"-?")) { | |
ASSERT(FALSE); | |
} | |
switch (ParamCount = ShellCommandLineGetCount(Package)) { | |
case 0: | |
case 1: | |
// | |
// we have insufficient parameters | |
// | |
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel2HiiHandle, L"mv"); | |
ShellStatus = SHELL_INVALID_PARAMETER; | |
break; | |
case 2: | |
// | |
// must have valid CWD for single parameter... | |
// | |
if (ShellGetCurrentDir(NULL) == NULL){ | |
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellLevel2HiiHandle, L"mv"); | |
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, L"mv", ShellCommandLineGetRawValue(Package, 1)); | |
ShellStatus = SHELL_NOT_FOUND; | |
} else { | |
// | |
// ValidateAndMoveFiles will report errors to the screen itself | |
// | |
ShellStatus = ValidateAndMoveFiles(FileList, &Response, ShellGetCurrentDir(NULL)); | |
} | |
} | |
break; | |
default: | |
///@todo make sure this works with error half way through and continues... | |
for (ParamCount--, LoopCounter = 1 ; LoopCounter < ParamCount ; LoopCounter++) { | |
if (ShellGetExecutionBreakFlag()) { | |
break; | |
} | |
Status = ShellOpenFileMetaArg((CHAR16*)ShellCommandLineGetRawValue(Package, LoopCounter), 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, L"mv", ShellCommandLineGetRawValue(Package, LoopCounter)); | |
ShellStatus = SHELL_NOT_FOUND; | |
} else { | |
// | |
// ValidateAndMoveFiles will report errors to the screen itself | |
// Only change ShellStatus if it's sucessful | |
// | |
if (ShellStatus == SHELL_SUCCESS) { | |
ShellStatus = ValidateAndMoveFiles(FileList, &Response, ShellCommandLineGetRawValue(Package, ParamCount)); | |
} else { | |
ValidateAndMoveFiles(FileList, &Response, ShellCommandLineGetRawValue(Package, ParamCount)); | |
} | |
} | |
if (FileList != NULL && !IsListEmpty(&FileList->Link)) { | |
Status = ShellCloseFileMetaArg(&FileList); | |
if (EFI_ERROR(Status) && ShellStatus == SHELL_SUCCESS) { | |
ShellStatus = SHELL_ACCESS_DENIED; | |
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_FILE), gShellLevel2HiiHandle, L"mv", ShellCommandLineGetRawValue(Package, 1), ShellStatus|MAX_BIT); | |
} | |
} | |
} | |
break; | |
} // switch on parameter count | |
if (FileList != NULL) { | |
ShellCloseFileMetaArg(&FileList); | |
} | |
// | |
// free the command line package | |
// | |
ShellCommandLineFreeVarList (Package); | |
} | |
SHELL_FREE_NON_NULL(Response); | |
if (ShellGetExecutionBreakFlag()) { | |
return (SHELL_ABORTED); | |
} | |
return (ShellStatus); | |
} |