Main file for mv shell level 2 function.\r
\r
(C) Copyright 2013-2015 Hewlett-Packard Development Company, L.P.<BR>\r
- Copyright (c) 2009 - 2014, Intel Corporation. All rights reserved.<BR>\r
+ Copyright (c) 2009 - 2016, Intel Corporation. All rights reserved.<BR>\r
This program and the accompanying materials\r
are licensed and made available under the terms and conditions of the BSD License\r
which accompanies this distribution. The full text of the license may be found at\r
@retval FALSE The move is within a file system.\r
**/\r
BOOLEAN\r
-EFIAPI\r
IsBetweenFileSystem(\r
IN CONST CHAR16 *FullName,\r
IN CONST CHAR16 *Cwd,\r
return (FALSE);\r
}\r
\r
+/**\r
+ function to determine if SrcPath is valid to mv.\r
+\r
+ if SrcPath equal CWD then it's invalid.\r
+ if SrcPath is the parent path of CWD then it's invalid.\r
+ is SrcPath is NULL return FALSE.\r
+\r
+ if CwdPath is NULL then ASSERT()\r
+\r
+ @param SrcPath [in] The source path.\r
+ @param CwdPath [in] The current working directory.\r
+\r
+ @retval TRUE The source path is valid.\r
+ @retval FALSE The source path is invalid.\r
+**/\r
+BOOLEAN\r
+IsSoucePathValid(\r
+ IN CONST CHAR16* SrcPath,\r
+ IN CONST CHAR16* CwdPath\r
+ )\r
+{\r
+ CHAR16* SrcPathBuffer;\r
+ CHAR16* CwdPathBuffer;\r
+ BOOLEAN Ret;\r
+\r
+ ASSERT (CwdPath != NULL);\r
+ if (SrcPath == NULL) {\r
+ return FALSE;\r
+ }\r
+\r
+ Ret = TRUE;\r
+\r
+ SrcPathBuffer = AllocateCopyPool (StrSize (SrcPath), SrcPath);\r
+ if (SrcPathBuffer == NULL) {\r
+ return FALSE;\r
+ }\r
+\r
+ CwdPathBuffer = AllocateCopyPool (StrSize (CwdPath), CwdPath);\r
+ if (CwdPathBuffer == NULL) {\r
+ FreePool(SrcPathBuffer);\r
+ return FALSE;\r
+ }\r
+\r
+ gUnicodeCollation->StrUpr (gUnicodeCollation, SrcPathBuffer);\r
+ gUnicodeCollation->StrUpr (gUnicodeCollation, CwdPathBuffer);\r
+\r
+ if (SrcPathBuffer[StrLen (SrcPathBuffer) -1 ] == L'\\') {\r
+ SrcPathBuffer[StrLen (SrcPathBuffer) - 1] = CHAR_NULL;\r
+ }\r
+\r
+ if (CwdPathBuffer[StrLen (CwdPathBuffer) - 1] == L'\\') {\r
+ CwdPathBuffer[StrLen (CwdPathBuffer) - 1] = CHAR_NULL;\r
+ }\r
+\r
+ if (StrCmp (CwdPathBuffer, SrcPathBuffer) == 0 ||\r
+ ((StrStr (CwdPathBuffer, SrcPathBuffer) == CwdPathBuffer) &&\r
+ (CwdPathBuffer[StrLen (SrcPathBuffer)] == L'\\'))\r
+ ) {\r
+ Ret = FALSE;\r
+ }\r
+\r
+ FreePool (SrcPathBuffer);\r
+ FreePool (CwdPathBuffer);\r
+\r
+ return Ret;\r
+}\r
+\r
/**\r
Function to validate that moving a specific file (FileName) to a specific\r
location (DestPath) is valid.\r
@retval FALSE The move is not\r
**/\r
BOOLEAN\r
-EFIAPI\r
IsValidMove(\r
IN CONST CHAR16 *SourcePath,\r
IN CONST CHAR16 *Cwd,\r
CHAR16 *DestPathCopy;\r
CHAR16 *DestPathWalker;\r
\r
- if (Cwd != NULL && StrCmp(SourcePath, Cwd) == 0) {\r
- //\r
- // Invalid move\r
- //\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_MV_INV_CWD), gShellLevel2HiiHandle);\r
- return (FALSE);\r
+ if ((Cwd != NULL) && ((Attribute & EFI_FILE_DIRECTORY) == EFI_FILE_DIRECTORY)) {\r
+ if (!IsSoucePathValid (SourcePath, Cwd)) {\r
+ //\r
+ // Invalid move\r
+ //\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN(STR_MV_INV_CWD), gShellLevel2HiiHandle);\r
+ return FALSE;\r
+ }\r
}\r
\r
//\r
//\r
// If they're the same, or if source is "above" dest on file path tree\r
//\r
- if ( StringNoCaseCompare (&DestPathWalker, &SourcePath) == 0 \r
- || StrStr(DestPathWalker, SourcePath) == DestPathWalker \r
- ) {\r
+ if ( StringNoCaseCompare (&DestPathWalker, &SourcePath) == 0 ||\r
+ ((StrStr(DestPathWalker, SourcePath) == DestPathWalker) && \r
+ (DestPathWalker[StrLen(SourcePath)] == '\\')\r
+ )\r
+ ) {\r
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_MV_INV_SUB), gShellLevel2HiiHandle);\r
FreePool(DestPathCopy);\r
return (FALSE);\r
@retval SHELL_SUCCESS The operation was sucessful.\r
**/\r
SHELL_STATUS\r
-EFIAPI\r
GetDestinationLocation(\r
IN CONST CHAR16 *DestParameter,\r
IN OUT CHAR16 **DestPathPointer,\r
if (DestPath == NULL) {\r
return (SHELL_OUT_OF_RESOURCES);\r
}\r
- StrCpy(DestPath, Cwd);\r
+ StrCpyS(DestPath, StrSize(Cwd) / sizeof(CHAR16), Cwd);\r
while (PathRemoveLastItem(DestPath)) ;\r
\r
//\r
ShellCloseFileMetaArg(&DestList);\r
return (SHELL_OUT_OF_RESOURCES);\r
}\r
- StrCpy(DestPath, Cwd);\r
+ StrCpyS(DestPath, NewSize / sizeof(CHAR16), Cwd);\r
if (DestPath[StrLen(DestPath)-1] != L'\\' && DestParameter[0] != L'\\') {\r
- StrCat(DestPath, L"\\");\r
+ StrCatS(DestPath, NewSize / sizeof(CHAR16), L"\\");\r
} else if (DestPath[StrLen(DestPath)-1] == L'\\' && DestParameter[0] == L'\\') {\r
((CHAR16*)DestPath)[StrLen(DestPath)-1] = CHAR_NULL;\r
}\r
- StrCat(DestPath, DestParameter);\r
+ StrCatS(DestPath, NewSize / sizeof(CHAR16), DestParameter);\r
} else {\r
ASSERT(DestPath == NULL);\r
DestPath = StrnCatGrow(&DestPath, NULL, DestParameter, 0);\r
ShellCloseFileMetaArg(&DestList);\r
return (SHELL_OUT_OF_RESOURCES);\r
}\r
- StrCpy(DestPath, Node->FullName);\r
- StrCat(DestPath, L"\\");\r
+ StrCpyS(DestPath, (StrSize(Node->FullName)+sizeof(CHAR16)) / sizeof(CHAR16), Node->FullName);\r
+ StrCatS(DestPath, (StrSize(Node->FullName)+sizeof(CHAR16)) / sizeof(CHAR16), L"\\");\r
} else {\r
//\r
// cant move multiple files onto a single file.\r
@retval SHELL_SUCCESS The source file was moved to the destination.\r
**/\r
EFI_STATUS\r
-EFIAPI\r
MoveBetweenFileSystems(\r
IN EFI_SHELL_FILE_INFO *Node,\r
IN CONST CHAR16 *DestPath,\r
@retval SHELL_OUT_OF_RESOURCES a memory allocation failed\r
**/\r
EFI_STATUS\r
-EFIAPI\r
CreateFullDestPath(\r
IN CONST CHAR16 **DestPath,\r
OUT CHAR16 **FullDestPath, \r
return (EFI_OUT_OF_RESOURCES);\r
}\r
\r
- StrnCpy(*FullDestPath, *DestPath, Size / sizeof(CHAR16) - 1);\r
+ StrCpyS(*FullDestPath, Size / sizeof(CHAR16), *DestPath);\r
if ((*FullDestPath)[StrLen(*FullDestPath)-1] != L'\\' && FileName[0] != L'\\') {\r
- StrnCat(*FullDestPath, L"\\",Size / sizeof(CHAR16) - 1 - StrLen(*FullDestPath));\r
+ StrCatS(*FullDestPath, Size / sizeof(CHAR16), L"\\");\r
}\r
- StrnCat(*FullDestPath, FileName, Size / sizeof(CHAR16) - 1 - StrLen(*FullDestPath));\r
+ StrCatS(*FullDestPath, Size / sizeof(CHAR16), FileName);\r
\r
return (EFI_SUCCESS);\r
}\r
@retval SHELL_OUT_OF_RESOURCES A memory allocation failed.\r
**/\r
EFI_STATUS\r
-EFIAPI\r
MoveWithinFileSystems(\r
IN EFI_SHELL_FILE_INFO *Node,\r
IN CHAR16 *DestPath,\r
} else {\r
CopyMem(NewFileInfo, Node->Info, SIZE_OF_EFI_FILE_INFO);\r
if (DestPath[0] != L'\\') {\r
- StrCpy(NewFileInfo->FileName, L"\\");\r
- StrCat(NewFileInfo->FileName, DestPath);\r
+ StrCpyS(NewFileInfo->FileName, (NewSize - SIZE_OF_EFI_FILE_INFO) / sizeof(CHAR16), L"\\");\r
+ StrCatS(NewFileInfo->FileName, (NewSize - SIZE_OF_EFI_FILE_INFO) / sizeof(CHAR16), DestPath);\r
} else {\r
- StrCpy(NewFileInfo->FileName, DestPath);\r
+ StrCpyS(NewFileInfo->FileName, (NewSize - SIZE_OF_EFI_FILE_INFO) / sizeof(CHAR16), DestPath);\r
}\r
Length = StrLen(NewFileInfo->FileName);\r
if (Length > 0) {\r
//\r
NewFileInfo->FileName[Length] = CHAR_NULL;\r
}\r
- StrCat(NewFileInfo->FileName, Node->FileName);\r
+ StrCatS(NewFileInfo->FileName, (NewSize - SIZE_OF_EFI_FILE_INFO) / sizeof(CHAR16), Node->FileName);\r
}\r
NewFileInfo->Size = SIZE_OF_EFI_FILE_INFO + StrSize(NewFileInfo->FileName);\r
\r
@retval SHELL_OUT_OF_RESOURCES a memory allocation failed\r
**/\r
SHELL_STATUS\r
-EFIAPI\r
ValidateAndMoveFiles(\r
IN EFI_SHELL_FILE_INFO *FileList,\r
OUT VOID **Resp,\r
CHAR16 *DestPath;\r
CHAR16 *FullDestPath;\r
CONST CHAR16 *Cwd;\r
+ CHAR16 *FullCwd;\r
SHELL_STATUS ShellStatus;\r
EFI_SHELL_FILE_INFO *Node;\r
VOID *Response;\r
Response = *Resp;\r
Attr = 0;\r
CleanFilePathStr = NULL;\r
+ FullCwd = NULL;\r
+\r
+ if (Cwd != NULL) {\r
+ FullCwd = AllocateZeroPool(StrSize(Cwd) + sizeof(CHAR16));\r
+ if (FullCwd == NULL) {\r
+ return SHELL_OUT_OF_RESOURCES;\r
+ } else {\r
+ StrCpyS(FullCwd, StrSize(Cwd)/sizeof(CHAR16)+1, Cwd);\r
+ StrCatS(FullCwd, StrSize(Cwd)/sizeof(CHAR16)+1, L"\\");\r
+ }\r
+ } \r
\r
Status = ShellLevel2StripQuotes (DestParameter, &CleanFilePathStr);\r
if (EFI_ERROR (Status)) {\r
+ SHELL_FREE_NON_NULL(FullCwd);\r
if (Status == EFI_OUT_OF_RESOURCES) {\r
return SHELL_OUT_OF_RESOURCES;\r
} else {\r
//\r
// Get and validate the destination location\r
//\r
- ShellStatus = GetDestinationLocation(CleanFilePathStr, &DestPath, Cwd, (BOOLEAN)(FileList->Link.ForwardLink == FileList->Link.BackLink), &Attr);\r
+ ShellStatus = GetDestinationLocation(CleanFilePathStr, &DestPath, FullCwd, (BOOLEAN)(FileList->Link.ForwardLink == FileList->Link.BackLink), &Attr);\r
FreePool (CleanFilePathStr);\r
\r
if (ShellStatus != SHELL_SUCCESS) {\r
+ SHELL_FREE_NON_NULL (FullCwd);\r
return (ShellStatus);\r
}\r
DestPath = PathCleanUpDirectories(DestPath);\r
if (DestPath == NULL) {\r
+ FreePool (FullCwd);\r
return (SHELL_OUT_OF_RESOURCES);\r
}\r
\r
SHELL_FREE_NON_NULL(DestPath);\r
SHELL_FREE_NON_NULL(HiiOutput);\r
SHELL_FREE_NON_NULL(HiiResultOk);\r
+ SHELL_FREE_NON_NULL(FullCwd);\r
return (SHELL_OUT_OF_RESOURCES);\r
}\r
\r
//\r
// Validate that the move is valid\r
//\r
- if (!IsValidMove(Node->FullName, Cwd, FullDestPath!=NULL? FullDestPath:DestPath, Node->Info->Attribute, Attr, Node->Status)) {\r
+ if (!IsValidMove(Node->FullName, FullCwd, FullDestPath!=NULL? FullDestPath:DestPath, Node->Info->Attribute, Attr, Node->Status)) {\r
ShellStatus = SHELL_INVALID_PARAMETER;\r
continue;\r
}\r
//\r
// indicate to stop everything\r
//\r
+ SHELL_FREE_NON_NULL(FullCwd);\r
return (SHELL_ABORTED);\r
case ShellPromptResponseAll:\r
*Resp = Response;\r
break;\r
default:\r
FreePool(Response);\r
+ SHELL_FREE_NON_NULL(FullCwd);\r
return SHELL_ABORTED;\r
}\r
Status = ShellDeleteFileByName(FullDestPath!=NULL? FullDestPath:DestPath);\r
}\r
\r
- if (IsBetweenFileSystem(Node->FullName, Cwd, DestPath)) {\r
+ if (IsBetweenFileSystem(Node->FullName, FullCwd, DestPath)) {\r
while (FullDestPath == NULL && DestPath != NULL && DestPath[0] != CHAR_NULL && DestPath[StrLen(DestPath) - 1] == L'\\') {\r
DestPath[StrLen(DestPath) - 1] = CHAR_NULL;\r
}\r
SHELL_FREE_NON_NULL(DestPath);\r
SHELL_FREE_NON_NULL(HiiOutput);\r
SHELL_FREE_NON_NULL(HiiResultOk);\r
+ SHELL_FREE_NON_NULL(FullCwd);\r
return (ShellStatus);\r
}\r
\r
EFI_STATUS Status;\r
LIST_ENTRY *Package;\r
CHAR16 *ProblemParam;\r
+ CHAR16 *Cwd;\r
+ UINTN CwdSize;\r
SHELL_STATUS ShellStatus;\r
UINTN ParamCount;\r
UINTN LoopCounter;\r
//\r
// ValidateAndMoveFiles will report errors to the screen itself\r
//\r
- ShellStatus = ValidateAndMoveFiles(FileList, &Response, ShellGetCurrentDir(NULL));\r
+ CwdSize = StrSize(ShellGetCurrentDir(NULL)) + sizeof(CHAR16);\r
+ Cwd = AllocateZeroPool(CwdSize);\r
+ if (Cwd == NULL) {\r
+ ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellLevel2HiiHandle, L"mv");\r
+ ShellStatus = SHELL_OUT_OF_RESOURCES;\r
+ } else {\r
+ StrCpyS (Cwd, CwdSize / sizeof (CHAR16), ShellGetCurrentDir (NULL));\r
+ StrCatS (Cwd, CwdSize / sizeof (CHAR16), L"\\");\r
+ ShellStatus = ValidateAndMoveFiles (FileList, &Response, Cwd);\r
+ FreePool (Cwd);\r
+ }\r
}\r
}\r
\r
projects / mirror_edk2.git / blobdiff