REF: https://bugzilla.tianocore.org/show_bug.cgi?id=3737 Apply uncrustify changes to .c/.h files in the ShellPkg package Cc: Andrew Fish <afish@apple.com> Cc: Leif Lindholm <leif@nuviainc.com> Cc: Michael D Kinney <michael.d.kinney@intel.com> Signed-off-by: Michael Kubacki <michael.kubacki@microsoft.com> Reviewed-by: Ray Ni <ray.ni@intel.com>
		
			
				
	
	
		
			892 lines
		
	
	
		
			28 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			892 lines
		
	
	
		
			28 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /** @file
 | |
|   Main file for mv shell level 2 function.
 | |
| 
 | |
|   (C) Copyright 2013-2015 Hewlett-Packard Development Company, L.P.<BR>
 | |
|   Copyright (c) 2009 - 2019, Intel Corporation. All rights reserved.<BR>
 | |
|   SPDX-License-Identifier: BSD-2-Clause-Patent
 | |
| 
 | |
| **/
 | |
| 
 | |
| #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
 | |
| 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 determine if SrcPath is valid to mv.
 | |
| 
 | |
|   if SrcPath equal CWD then it's invalid.
 | |
|   if SrcPath is the parent path of CWD then it's invalid.
 | |
|   is SrcPath is NULL return FALSE.
 | |
| 
 | |
|   if CwdPath is NULL then ASSERT()
 | |
| 
 | |
|   @param SrcPath  [in]    The source path.
 | |
|   @param CwdPath  [in]    The current working directory.
 | |
| 
 | |
|   @retval TRUE            The source path is valid.
 | |
|   @retval FALSE           The source path is invalid.
 | |
| **/
 | |
| BOOLEAN
 | |
| IsSoucePathValid (
 | |
|   IN CONST CHAR16  *SrcPath,
 | |
|   IN CONST CHAR16  *CwdPath
 | |
|   )
 | |
| {
 | |
|   CHAR16   *SrcPathBuffer;
 | |
|   CHAR16   *CwdPathBuffer;
 | |
|   BOOLEAN  Ret;
 | |
| 
 | |
|   ASSERT (CwdPath != NULL);
 | |
|   if (SrcPath == NULL) {
 | |
|     return FALSE;
 | |
|   }
 | |
| 
 | |
|   Ret = TRUE;
 | |
| 
 | |
|   SrcPathBuffer = AllocateCopyPool (StrSize (SrcPath), SrcPath);
 | |
|   if (SrcPathBuffer == NULL) {
 | |
|     return FALSE;
 | |
|   }
 | |
| 
 | |
|   CwdPathBuffer = AllocateCopyPool (StrSize (CwdPath), CwdPath);
 | |
|   if (CwdPathBuffer == NULL) {
 | |
|     FreePool (SrcPathBuffer);
 | |
|     return FALSE;
 | |
|   }
 | |
| 
 | |
|   gUnicodeCollation->StrUpr (gUnicodeCollation, SrcPathBuffer);
 | |
|   gUnicodeCollation->StrUpr (gUnicodeCollation, CwdPathBuffer);
 | |
| 
 | |
|   if (SrcPathBuffer[StrLen (SrcPathBuffer) -1] == L'\\') {
 | |
|     SrcPathBuffer[StrLen (SrcPathBuffer) - 1] = CHAR_NULL;
 | |
|   }
 | |
| 
 | |
|   if (CwdPathBuffer[StrLen (CwdPathBuffer) - 1] == L'\\') {
 | |
|     CwdPathBuffer[StrLen (CwdPathBuffer) - 1] = CHAR_NULL;
 | |
|   }
 | |
| 
 | |
|   if ((StrCmp (CwdPathBuffer, SrcPathBuffer) == 0) ||
 | |
|       ((StrStr (CwdPathBuffer, SrcPathBuffer) == CwdPathBuffer) &&
 | |
|        (CwdPathBuffer[StrLen (SrcPathBuffer)] == L'\\'))
 | |
|       )
 | |
|   {
 | |
|     Ret = FALSE;
 | |
|   }
 | |
| 
 | |
|   FreePool (SrcPathBuffer);
 | |
|   FreePool (CwdPathBuffer);
 | |
| 
 | |
|   return Ret;
 | |
| }
 | |
| 
 | |
| /**
 | |
|   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
 | |
| 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) && ((Attribute & EFI_FILE_DIRECTORY) == EFI_FILE_DIRECTORY)) {
 | |
|     if (!IsSoucePathValid (SourcePath, Cwd)) {
 | |
|       //
 | |
|       // 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 ((StringNoCaseCompare (&DestPathWalker, &SourcePath) == 0) ||
 | |
|       ((StrStr (DestPathWalker, SourcePath) == DestPathWalker) &&
 | |
|        (DestPathWalker[StrLen (SourcePath)] == '\\')
 | |
|       )
 | |
|       )
 | |
|   {
 | |
|     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
 | |
| 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);
 | |
|     }
 | |
| 
 | |
|     StrCpyS (DestPath, StrSize (Cwd) / sizeof (CHAR16), 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);
 | |
|       }
 | |
| 
 | |
|       StrCpyS (DestPath, NewSize / sizeof (CHAR16), Cwd);
 | |
|       if ((DestPath[StrLen (DestPath)-1] != L'\\') && (DestParameter[0] != L'\\')) {
 | |
|         StrCatS (DestPath, NewSize / sizeof (CHAR16), L"\\");
 | |
|       } else if ((DestPath[StrLen (DestPath)-1] == L'\\') && (DestParameter[0] == L'\\')) {
 | |
|         ((CHAR16 *)DestPath)[StrLen (DestPath)-1] = CHAR_NULL;
 | |
|       }
 | |
| 
 | |
|       StrCatS (DestPath, NewSize / sizeof (CHAR16), 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);
 | |
|       }
 | |
| 
 | |
|       StrCpyS (DestPath, (StrSize (Node->FullName)+sizeof (CHAR16)) / sizeof (CHAR16), Node->FullName);
 | |
|       StrCatS (DestPath, (StrSize (Node->FullName)+sizeof (CHAR16)) / sizeof (CHAR16), 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
 | |
| MoveBetweenFileSystems (
 | |
|   IN EFI_SHELL_FILE_INFO  *Node,
 | |
|   IN CONST CHAR16         *DestPath,
 | |
|   OUT VOID                **Resp
 | |
|   )
 | |
| {
 | |
|   SHELL_STATUS  ShellStatus;
 | |
| 
 | |
|   //
 | |
|   // First we copy the file
 | |
|   //
 | |
|   ShellStatus = CopySingleFile (Node->FullName, DestPath, Resp, TRUE, L"mv");
 | |
| 
 | |
|   //
 | |
|   // Check our result
 | |
|   //
 | |
|   if (ShellStatus == SHELL_SUCCESS) {
 | |
|     //
 | |
|     // The copy was successful.  delete the source file.
 | |
|     //
 | |
|     CascadeDelete (Node, TRUE);
 | |
|     Node->Handle = NULL;
 | |
|   } else if (ShellStatus == SHELL_ABORTED) {
 | |
|     return EFI_ABORTED;
 | |
|   } else if (ShellStatus == SHELL_ACCESS_DENIED) {
 | |
|     return EFI_ACCESS_DENIED;
 | |
|   } else if (ShellStatus == SHELL_VOLUME_FULL) {
 | |
|     return EFI_VOLUME_FULL;
 | |
|   } else {
 | |
|     return EFI_UNSUPPORTED;
 | |
|   }
 | |
| 
 | |
|   return (EFI_SUCCESS);
 | |
| }
 | |
| 
 | |
| /**
 | |
|   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
 | |
| 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);
 | |
|   }
 | |
| 
 | |
|   StrCpyS (*FullDestPath, Size / sizeof (CHAR16), *DestPath);
 | |
|   if (((*FullDestPath)[StrLen (*FullDestPath)-1] != L'\\') && (FileName[0] != L'\\')) {
 | |
|     StrCatS (*FullDestPath, Size / sizeof (CHAR16), L"\\");
 | |
|   }
 | |
| 
 | |
|   StrCatS (*FullDestPath, Size / sizeof (CHAR16), FileName);
 | |
| 
 | |
|   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
 | |
| 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'\\') {
 | |
|       StrCpyS (NewFileInfo->FileName, (NewSize - SIZE_OF_EFI_FILE_INFO) / sizeof (CHAR16), L"\\");
 | |
|       StrCatS (NewFileInfo->FileName, (NewSize - SIZE_OF_EFI_FILE_INFO) / sizeof (CHAR16), DestPath);
 | |
|     } else {
 | |
|       StrCpyS (NewFileInfo->FileName, (NewSize - SIZE_OF_EFI_FILE_INFO) / sizeof (CHAR16), 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;
 | |
|       }
 | |
| 
 | |
|       StrCatS (NewFileInfo->FileName, (NewSize - SIZE_OF_EFI_FILE_INFO) / sizeof (CHAR16), 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
 | |
| 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;
 | |
|   CHAR16               *FullCwd;
 | |
|   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;
 | |
|   FullCwd          = NULL;
 | |
| 
 | |
|   if (Cwd != NULL) {
 | |
|     FullCwd = AllocateZeroPool (StrSize (Cwd) + sizeof (CHAR16));
 | |
|     if (FullCwd == NULL) {
 | |
|       return SHELL_OUT_OF_RESOURCES;
 | |
|     } else {
 | |
|       StrCpyS (FullCwd, StrSize (Cwd)/sizeof (CHAR16)+1, Cwd);
 | |
|       StrCatS (FullCwd, StrSize (Cwd)/sizeof (CHAR16)+1, L"\\");
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   Status = ShellLevel2StripQuotes (DestParameter, &CleanFilePathStr);
 | |
|   if (EFI_ERROR (Status)) {
 | |
|     SHELL_FREE_NON_NULL (FullCwd);
 | |
|     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, FullCwd, (BOOLEAN)(FileList->Link.ForwardLink == FileList->Link.BackLink), &Attr);
 | |
|   FreePool (CleanFilePathStr);
 | |
| 
 | |
|   if (ShellStatus != SHELL_SUCCESS) {
 | |
|     SHELL_FREE_NON_NULL (FullCwd);
 | |
|     return (ShellStatus);
 | |
|   }
 | |
| 
 | |
|   DestPath = PathCleanUpDirectories (DestPath);
 | |
|   if (DestPath == NULL) {
 | |
|     FreePool (FullCwd);
 | |
|     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);
 | |
|     SHELL_FREE_NON_NULL (FullCwd);
 | |
|     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, FullCwd, (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);
 | |
|       }
 | |
| 
 | |
|       if (Response == NULL) {
 | |
|         return SHELL_ABORTED;
 | |
|       }
 | |
| 
 | |
|       switch (*(SHELL_PROMPT_RESPONSE *)Response) {
 | |
|         case ShellPromptResponseNo:
 | |
|           FreePool (Response);
 | |
|           Response = NULL;
 | |
|           continue;
 | |
|         case ShellPromptResponseCancel:
 | |
|           *Resp = Response;
 | |
|           //
 | |
|           // indicate to stop everything
 | |
|           //
 | |
|           SHELL_FREE_NON_NULL (FullCwd);
 | |
|           return (SHELL_ABORTED);
 | |
|         case ShellPromptResponseAll:
 | |
|           *Resp = Response;
 | |
|           break;
 | |
|         case ShellPromptResponseYes:
 | |
|           FreePool (Response);
 | |
|           Response = NULL;
 | |
|           break;
 | |
|         default:
 | |
|           FreePool (Response);
 | |
|           SHELL_FREE_NON_NULL (FullCwd);
 | |
|           return SHELL_ABORTED;
 | |
|       }
 | |
| 
 | |
|       Status = ShellDeleteFileByName (FullDestPath != NULL ? FullDestPath : DestPath);
 | |
|     }
 | |
| 
 | |
|     if (IsBetweenFileSystem (Node->FullName, FullCwd, 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);
 | |
|       //
 | |
|       // Display error status
 | |
|       //
 | |
|       if (EFI_ERROR (Status)) {
 | |
|         ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_UK), gShellLevel2HiiHandle, L"mv", Status);
 | |
|       }
 | |
|     }
 | |
| 
 | |
|     //
 | |
|     // Check our result
 | |
|     //
 | |
|     if (EFI_ERROR (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);
 | |
|   SHELL_FREE_NON_NULL (FullCwd);
 | |
|   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;
 | |
|   CHAR16               *Cwd;
 | |
|   UINTN                CwdSize;
 | |
|   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
 | |
|             //
 | |
|             CwdSize = StrSize (ShellGetCurrentDir (NULL)) + sizeof (CHAR16);
 | |
|             Cwd     = AllocateZeroPool (CwdSize);
 | |
|             if (Cwd == NULL) {
 | |
|               ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellLevel2HiiHandle, L"mv");
 | |
|               ShellStatus = SHELL_OUT_OF_RESOURCES;
 | |
|             } else {
 | |
|               StrCpyS (Cwd, CwdSize / sizeof (CHAR16), ShellGetCurrentDir (NULL));
 | |
|               StrCatS (Cwd, CwdSize / sizeof (CHAR16), L"\\");
 | |
|               ShellStatus = ValidateAndMoveFiles (FileList, &Response, Cwd);
 | |
|               FreePool (Cwd);
 | |
|             }
 | |
|           }
 | |
|         }
 | |
| 
 | |
|         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);
 | |
| }
 |