[mirror_edk2.git] / ShellPkg / Library / UefiShellLevel2CommandsLib / Cp.c

/** @file\r
  Main file for cp shell level 2 function.\r
\r
  Copyright (c) 2009 - 2010, 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
  http://opensource.org/licenses/bsd-license.php\r
\r
  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
\r
#include "UefiShellLevel2CommandsLib.h"\r
\r
// this is later in the file.\r
SHELL_STATUS\r
EFIAPI\r
ValidateAndCopyFiles(\r
  IN CONST EFI_SHELL_FILE_INFO  *FileList,\r
  IN CONST CHAR16               *DestDir,\r
  IN BOOLEAN                    SilentMode,\r
  IN BOOLEAN                    RecursiveMode,\r
  IN VOID                       **Resp\r
  );\r
\r
/**\r
  Function to Copy one file to another location\r
\r
  If the destination exists the user will be prompted and the result put into *resp\r
\r
  @param[in] Source     pointer to source file name\r
  @param[in] Dest       pointer to destination file name\r
  @param[out] Resp      pointer to response from question.  Pass back on looped calling\r
  @param[in] SilentMode whether to run in quiet mode or not\r
\r
  @retval SHELL_SUCCESS   The source file was copied to the destination\r
**/\r
SHELL_STATUS\r
EFIAPI\r
CopySingleFile(\r
  IN CONST CHAR16 *Source,\r
  IN CONST CHAR16 *Dest,\r
  OUT VOID        **Resp,\r
  IN BOOLEAN      SilentMode\r
  )\r
{\r
  VOID                *Response;\r
  UINTN               ReadSize;\r
  SHELL_FILE_HANDLE   SourceHandle;\r
  SHELL_FILE_HANDLE   DestHandle;\r
  EFI_STATUS          Status;\r
  VOID                *Buffer;\r
  CHAR16              *TempName;\r
  UINTN               Size;\r
  EFI_SHELL_FILE_INFO *List;\r
  SHELL_STATUS        ShellStatus;\r
\r
\r
  ASSERT(Resp != NULL);\r
\r
  SourceHandle  = NULL;\r
  DestHandle    = NULL;\r
  Response      = *Resp;\r
  List          = NULL;\r
\r
  ReadSize = PcdGet16(PcdShellFileOperationSize);\r
  // Why bother copying a file to itself\r
  if (StrCmp(Source, Dest) == 0) {\r
    return (SHELL_SUCCESS);\r
  }\r
\r
  //\r
  // Open destination file without create\r
  //\r
  Status = ShellOpenFileByName(Dest, &DestHandle, EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE, 0);\r
\r
  //\r
  // close file\r
  //\r
  if (DestHandle != NULL) {\r
    ShellCloseFile(&DestHandle);\r
    DestHandle   = NULL;\r
  }\r
\r
  //\r
  // if the destination file existed check response and possibly prompt user\r
  //\r
  if (!EFI_ERROR(Status)) {\r
    if (Response == NULL && !SilentMode) {\r
      Status = ShellPromptForResponseHii(ShellPromptResponseTypeYesNoAllCancel, STRING_TOKEN (STR_CP_PROMPT), gShellLevel2HiiHandle, &Response);\r
    }\r
    //\r
    // possibly return based on response\r
    //\r
    if (!SilentMode) {\r
      switch (*(SHELL_PROMPT_RESPONSE*)Response) {\r
        case ShellPromptResponseNo:\r
          //\r
          // return success here so we dont stop the process\r
          //\r
          return (SHELL_SUCCESS);\r
        case ShellPromptResponseCancel:\r
          *Resp = Response;\r
          //\r
          // indicate to stop everything\r
          //\r
          return (SHELL_ABORTED);\r
        case ShellPromptResponseAll:\r
          *Resp = Response;\r
        case ShellPromptResponseYes:\r
          break;\r
        default:\r
          return SHELL_ABORTED;\r
      }\r
    }\r
  }\r
\r
  if (ShellIsDirectory(Source) == EFI_SUCCESS) {\r
    Status = ShellCreateDirectory(Dest, &DestHandle);\r
    if (EFI_ERROR(Status)) {\r
      return (SHELL_ACCESS_DENIED);\r
    }\r
\r
    //\r
    // Now copy all the files under the directory...\r
    //\r
    TempName    = NULL;\r
    Size          = 0;\r
    StrnCatGrow(&TempName, &Size, Source, 0);\r
    StrnCatGrow(&TempName, &Size, L"\\*", 0);\r
    ShellOpenFileMetaArg((CHAR16*)TempName, EFI_FILE_MODE_READ, &List);\r
    TempName = NULL;\r
    StrnCatGrow(&TempName, &Size, Dest, 0);\r
    StrnCatGrow(&TempName, &Size, L"\\", 0);\r
    ShellStatus = ValidateAndCopyFiles(List, TempName, SilentMode, TRUE, Resp);\r
    ShellCloseFileMetaArg(&List);\r
    FreePool(TempName);\r
    Size = 0;\r
  } else {\r
    //\r
    // open file with create enabled\r
    //\r
    Status = ShellOpenFileByName(Dest, &DestHandle, EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE|EFI_FILE_MODE_CREATE, 0);\r
    if (EFI_ERROR(Status)) {\r
      return (SHELL_ACCESS_DENIED);\r
    }\r
\r
    //\r
    // open source file\r
    //\r
    Status = ShellOpenFileByName(Source, &SourceHandle, EFI_FILE_MODE_READ, 0);\r
    ASSERT_EFI_ERROR(Status);\r
\r
    //\r
    // copy data between files\r
    //\r
    Buffer = AllocateZeroPool(ReadSize);\r
    ASSERT(Buffer != NULL);\r
    while (ReadSize == PcdGet16(PcdShellFileOperationSize) && !EFI_ERROR(Status)) {\r
      Status = ShellReadFile(SourceHandle, &ReadSize, Buffer);\r
      ASSERT_EFI_ERROR(Status);\r
      Status = ShellWriteFile(DestHandle, &ReadSize, Buffer);\r
    }\r
  }\r
\r
  //\r
  // close files\r
  //\r
  if (DestHandle != NULL) {\r
    ShellCloseFile(&DestHandle);\r
    DestHandle   = NULL;\r
  }\r
  if (SourceHandle != NULL) {\r
    ShellCloseFile(&SourceHandle);\r
    SourceHandle = NULL;\r
  }\r
\r
  //\r
  // return\r
  //\r
  return (SHELL_SUCCESS);\r
}\r
\r
/**\r
  function to take a list of files to copy and a destination location and do\r
  the verification and copying of those files to that location.  This function\r
  will report any errors to the user and halt.\r
\r
  The key is to have this function called ONLY once.  this allows for the parameter\r
  verification to happen correctly.\r
\r
  @param[in] FileList           A LIST_ENTRY* based list of files to move\r
  @param[in] DestDir            the destination location\r
\r
  @retval SHELL_SUCCESS             the files were all moved.\r
  @retval SHELL_INVALID_PARAMETER   a parameter was invalid\r
  @retval SHELL_SECURITY_VIOLATION  a security violation ocurred\r
  @retval SHELL_WRITE_PROTECTED     the destination was write protected\r
  @retval SHELL_OUT_OF_RESOURCES    a memory allocation failed\r
**/\r
SHELL_STATUS\r
EFIAPI\r
ValidateAndCopyFiles(\r
  IN CONST EFI_SHELL_FILE_INFO  *FileList,\r
  IN CONST CHAR16               *DestDir,\r
  IN BOOLEAN                    SilentMode,\r
  IN BOOLEAN                    RecursiveMode,\r
  IN VOID                       **Resp\r
  )\r
{\r
  CHAR16                    *HiiOutput;\r
  CHAR16                    *HiiResultOk;\r
  CONST EFI_SHELL_FILE_INFO *Node;\r
  SHELL_STATUS              ShellStatus;\r
  CHAR16                    *DestPath;\r
  VOID                      *Response;\r
  UINTN                     PathLen;\r
  CONST CHAR16              *Cwd;\r
  CONST CHAR16              *TempLocation;\r
  UINTN                     NewSize;\r
\r
  if (Resp == NULL) {\r
    Response = NULL;\r
  } else {\r
    Response = *Resp;\r
  }\r
\r
  DestPath    = NULL;\r
  ShellStatus = SHELL_SUCCESS;\r
  PathLen     = 0;\r
  Cwd         = ShellGetCurrentDir(NULL);\r
\r
  ASSERT(FileList != NULL);\r
  ASSERT(DestDir  != NULL);\r
\r
  //\r
  // If we are trying to copy multiple files... make sure we got a directory for the target...\r
  //\r
  if (EFI_ERROR(ShellIsDirectory(DestDir)) && FileList->Link.ForwardLink != FileList->Link.BackLink) {\r
    //\r
    // Error for destination not a directory\r
    //\r
    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NOT_DIR), gShellLevel2HiiHandle, DestDir);\r
    return (SHELL_INVALID_PARAMETER);\r
  }\r
  for (Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&FileList->Link)\r
    ;  !IsNull(&FileList->Link, &Node->Link)\r
    ;  Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&FileList->Link, &Node->Link)\r
   ){\r
    //\r
    // skip the directory traversing stuff...\r
    //\r
    if (StrCmp(Node->FileName, L".") == 0 || StrCmp(Node->FileName, L"..") == 0) {\r
      continue;\r
    }\r
\r
    NewSize =  StrSize(DestDir);\r
    NewSize += StrSize(Node->FileName);\r
    NewSize += StrSize(Cwd);\r
    if (NewSize > PathLen) {\r
      PathLen = NewSize;\r
    }\r
\r
    //\r
    // Make sure got -r if required\r
    //\r
    if (!RecursiveMode && !EFI_ERROR(ShellIsDirectory(Node->FullName))) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_DIR_REQ), gShellLevel2HiiHandle);\r
      return (SHELL_INVALID_PARAMETER);\r
    }\r
\r
    //\r
    // make sure got dest as dir if needed\r
    //\r
    if (!EFI_ERROR(ShellIsDirectory(Node->FullName)) && EFI_ERROR(ShellIsDirectory(DestDir))) {\r
      //\r
      // Error for destination not a directory\r
      //\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NOT_DIR), gShellLevel2HiiHandle, DestDir);\r
      return (SHELL_INVALID_PARAMETER);\r
    }\r
  }\r
\r
  HiiOutput   = HiiGetString (gShellLevel2HiiHandle, STRING_TOKEN (STR_CP_OUTPUT), NULL);\r
  HiiResultOk = HiiGetString (gShellLevel2HiiHandle, STRING_TOKEN (STR_GEN_RES_OK), NULL);\r
  DestPath    = AllocatePool(PathLen);\r
\r
  //\r
  // Go through the list of files to copy...\r
  //\r
  for (Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&FileList->Link)\r
    ;  !IsNull(&FileList->Link, &Node->Link)\r
    ;  Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&FileList->Link, &Node->Link)\r
   ){\r
    if (ShellGetExecutionBreakFlag()) {\r
      break;\r
    }\r
    ASSERT(Node->FileName != NULL);\r
    ASSERT(Node->FullName != NULL);\r
\r
    //\r
    // skip the directory traversing stuff...\r
    //\r
    if (StrCmp(Node->FileName, L".") == 0 || StrCmp(Node->FileName, L"..") == 0) {\r
      continue;\r
    }\r
\r
    if (FileList->Link.ForwardLink == FileList->Link.BackLink // 1 item\r
      && EFI_ERROR(ShellIsDirectory(DestDir))                 // not an existing directory\r
     ) {\r
      ASSERT(StrStr(DestDir, L":") == NULL);\r
      //\r
      // simple copy of a single file\r
      //\r
      StrCpy(DestPath, Cwd);\r
      if (DestPath[StrLen(DestPath)-1] != L'\\' && DestDir[0] != L'\\') {\r
        StrCat(DestPath, L"\\");\r
      } else if (DestPath[StrLen(DestPath)-1] == L'\\' && DestDir[0] == L'\\') {\r
        ((CHAR16*)DestPath)[StrLen(DestPath)-1] = CHAR_NULL;\r
      }\r
      StrCat(DestPath, DestDir);\r
    } else {\r
      //\r
      // we have multiple files or a directory in the DestDir\r
      //\r
      if (StrStr(DestDir, L":") == NULL) {\r
        StrCpy(DestPath, Cwd);\r
        if (DestPath[StrLen(DestPath)-1] != L'\\' && DestDir[0] != L'\\') {\r
          StrCat(DestPath, L"\\");\r
        } else if (DestPath[StrLen(DestPath)-1] == L'\\' && DestDir[0] == L'\\') {\r
          ((CHAR16*)DestPath)[StrLen(DestPath)-1] = CHAR_NULL;\r
        }\r
        StrCat(DestPath, DestDir);\r
        if (DestDir[StrLen(DestDir)-1] != L'\\' && Node->FileName[0] != L'\\') {\r
          StrCat(DestPath, L"\\");\r
        } else if (DestDir[StrLen(DestDir)-1] == L'\\' && Node->FileName[0] == L'\\') {\r
          ((CHAR16*)DestPath)[StrLen(DestPath)-1] = CHAR_NULL;\r
        }\r
        StrCat(DestPath, Node->FileName);\r
\r
      } else {\r
        StrCpy(DestPath, DestDir);\r
        if (DestDir[StrLen(DestDir)-1] != L'\\' && Node->FileName[0] != L'\\') {\r
          StrCat(DestPath, L"\\");\r
        } else if (DestDir[StrLen(DestDir)-1] == L'\\' && Node->FileName[0] == L'\\') {\r
          ((CHAR16*)DestDir)[StrLen(DestDir)-1] = CHAR_NULL;\r
        }\r
        StrCat(DestPath, Node->FileName);\r
      }\r
    }\r
\r
    //\r
    // Make sure the path exists\r
    //\r
    if (EFI_ERROR(VerifyIntermediateDirectories(DestPath))) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_DIR_WNF), gShellLevel2HiiHandle);\r
      ShellStatus = SHELL_DEVICE_ERROR;\r
      break;\r
    }\r
\r
    if ( !EFI_ERROR(ShellIsDirectory(Node->FullName))\r
      && !EFI_ERROR(ShellIsDirectory(DestPath))\r
      && StrniCmp(Node->FullName, DestPath, StrLen(DestPath)) == NULL\r
     ){\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_SD_PARENT), gShellLevel2HiiHandle);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
      break;\r
    }\r
    if (StringNoCaseCompare(&Node->FullName, &DestPath) == 0) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_SD_SAME), gShellLevel2HiiHandle);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
      break;\r
    }\r
\r
    if ((TempLocation = StrniCmp(Node->FullName, DestPath, StrLen(Node->FullName))) == 0\r
      && (DestPath[StrLen(Node->FullName)] == CHAR_NULL || DestPath[StrLen(Node->FullName)] == L'\\')\r
     ) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_SD_SAME), gShellLevel2HiiHandle);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
      break;\r
    }\r
\r
    CleanPath(DestPath);\r
\r
    ShellPrintEx(-1, -1, HiiOutput, Node->FullName, DestPath);\r
\r
    //\r
    // copy single file...\r
    //\r
    ShellStatus = CopySingleFile(Node->FullName, DestPath, &Response, SilentMode);\r
    if (ShellStatus != SHELL_SUCCESS) {\r
      break;\r
    }\r
  }\r
  if (ShellStatus == SHELL_SUCCESS && Resp == NULL) {\r
    ShellPrintEx(-1, -1, L"%s", HiiResultOk);\r
  }\r
\r
  SHELL_FREE_NON_NULL(DestPath);\r
  SHELL_FREE_NON_NULL(HiiOutput);\r
  SHELL_FREE_NON_NULL(HiiResultOk);\r
  if (Resp != NULL) {\r
    SHELL_FREE_NON_NULL(Response);\r
  }\r
\r
  return (ShellStatus);\r
}\r
\r
SHELL_STATUS\r
EFIAPI\r
ProcessValidateAndCopyFiles(\r
  IN       EFI_SHELL_FILE_INFO  *FileList,\r
  IN CONST CHAR16               *DestDir,\r
  IN BOOLEAN                    SilentMode,\r
  IN BOOLEAN                    RecursiveMode\r
  )\r
{\r
  SHELL_STATUS        ShellStatus;\r
  EFI_SHELL_FILE_INFO *List;\r
  EFI_STATUS          Status;\r
  EFI_FILE_INFO       *FileInfo;\r
\r
  List = NULL;\r
\r
  Status = ShellOpenFileMetaArg((CHAR16*)DestDir, EFI_FILE_MODE_READ, &List);\r
  if (List != NULL && List->Link.ForwardLink != List->Link.BackLink) {\r
    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_MARG_ERROR), gShellLevel2HiiHandle, DestDir);\r
    ShellStatus = SHELL_INVALID_PARAMETER;\r
    ShellCloseFileMetaArg(&List);\r
  } else if (List != NULL) {\r
    ASSERT(List != NULL);\r
    ASSERT(((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink) != NULL);\r
    ASSERT(((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink)->FullName != NULL);\r
    FileInfo    = NULL;\r
    FileInfo = gEfiShellProtocol->GetFileInfo(((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink)->Handle);\r
    ASSERT(FileInfo != NULL);\r
    if ((FileInfo->Attribute & EFI_FILE_READ_ONLY) == 0) {\r
      ShellStatus = ValidateAndCopyFiles(FileList, ((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink)->FullName, SilentMode, RecursiveMode, NULL);\r
    } else {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_DEST_ERROR), gShellLevel2HiiHandle);\r
      ShellStatus = SHELL_ACCESS_DENIED;\r
    }\r
    SHELL_FREE_NON_NULL(FileInfo);\r
    ShellCloseFileMetaArg(&List);\r
  } else {\r
      ShellStatus = ValidateAndCopyFiles(FileList, DestDir, SilentMode, RecursiveMode, NULL);\r
  }\r
\r
  return (ShellStatus);\r
}\r
\r
STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
  {L"-r", TypeFlag},\r
  {L"-q", TypeFlag},\r
  {NULL, TypeMax}\r
  };\r
\r
/**\r
  Function for 'cp' command.\r
\r
  @param[in] ImageHandle  Handle to the Image (NULL if Internal).\r
  @param[in] SystemTable  Pointer to the System Table (NULL if Internal).\r
**/\r
SHELL_STATUS\r
EFIAPI\r
ShellCommandRunCp (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  EFI_STATUS          Status;\r
  LIST_ENTRY          *Package;\r
  CHAR16              *ProblemParam;\r
  SHELL_STATUS        ShellStatus;\r
  UINTN               ParamCount;\r
  UINTN               LoopCounter;\r
  EFI_SHELL_FILE_INFO *FileList;\r
  BOOLEAN             SilentMode;\r
  BOOLEAN             RecursiveMode;\r
  CONST CHAR16        *Cwd;\r
\r
  ProblemParam        = NULL;\r
  ShellStatus         = SHELL_SUCCESS;\r
  ParamCount          = 0;\r
  FileList            = NULL;\r
\r
  //\r
  // initialize the shell lib (we must be in non-auto-init...)\r
  //\r
  Status = ShellInitialize();\r
  ASSERT_EFI_ERROR(Status);\r
\r
  Status = CommandInit();\r
  ASSERT_EFI_ERROR(Status);\r
\r
  //\r
  // parse the command line\r
  //\r
  Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
  if (EFI_ERROR(Status)) {\r
    if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, ProblemParam);\r
      FreePool(ProblemParam);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } else {\r
      ASSERT(FALSE);\r
    }\r
  } else {\r
    //\r
    // check for "-?"\r
    //\r
    if (ShellCommandLineGetFlag(Package, L"-?")) {\r
      ASSERT(FALSE);\r
    }\r
\r
    //\r
    // Initialize SilentMode and RecursiveMode\r
    //\r
    if (gEfiShellProtocol->BatchIsActive()) {\r
      SilentMode = TRUE;\r
    } else {\r
      SilentMode = ShellCommandLineGetFlag(Package, L"-q");\r
    }\r
    RecursiveMode = ShellCommandLineGetFlag(Package, L"-r");\r
\r
    switch (ParamCount = ShellCommandLineGetCount(Package)) {\r
      case 0:\r
      case 1:\r
        //\r
        // we have insufficient parameters\r
        //\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel2HiiHandle);\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
        break;\r
      case 2:\r
        //\r
        // must have valid CWD for single parameter...\r
        //\r
        Cwd = ShellGetCurrentDir(NULL);\r
        if (Cwd == NULL){\r
          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellLevel2HiiHandle);\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
        } else {\r
          Status = ShellOpenFileMetaArg((CHAR16*)ShellCommandLineGetRawValue(Package, 1), EFI_FILE_MODE_WRITE|EFI_FILE_MODE_READ, &FileList);\r
          if (FileList == NULL || IsListEmpty(&FileList->Link) || EFI_ERROR(Status)) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_NF), gShellLevel2HiiHandle, ShellCommandLineGetRawValue(Package, 1));\r
            ShellStatus = SHELL_NOT_FOUND;\r
          } else  {\r
            ShellStatus = ProcessValidateAndCopyFiles(FileList, Cwd, SilentMode, RecursiveMode);\r
          }\r
        }\r
\r
        break;\r
      default:\r
        //\r
        // Make a big list of all the files...\r
        //\r
        for (ParamCount--, LoopCounter = 1 ; LoopCounter < ParamCount && ShellStatus == SHELL_SUCCESS ; LoopCounter++) {\r
          if (ShellGetExecutionBreakFlag()) {\r
            break;\r
          }\r
          Status = ShellOpenFileMetaArg((CHAR16*)ShellCommandLineGetRawValue(Package, LoopCounter), EFI_FILE_MODE_WRITE|EFI_FILE_MODE_READ, &FileList);\r
          if (EFI_ERROR(Status) || FileList == NULL || IsListEmpty(&FileList->Link)) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_NF), gShellLevel2HiiHandle, ShellCommandLineGetRawValue(Package, 1));\r
            ShellStatus = SHELL_NOT_FOUND;\r
          }\r
        }\r
        //\r
        // now copy them all...\r
        //\r
        if (FileList != NULL && !IsListEmpty(&FileList->Link)) {\r
          ShellStatus = ProcessValidateAndCopyFiles(FileList, ShellCommandCleanPath((CHAR16*)ShellCommandLineGetRawValue(Package, ParamCount)), SilentMode, RecursiveMode);\r
          Status = ShellCloseFileMetaArg(&FileList);\r
          if (EFI_ERROR(Status) && ShellStatus == SHELL_SUCCESS) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_FILE), gShellLevel2HiiHandle, ShellCommandLineGetRawValue(Package, 1), ShellStatus|MAX_BIT);\r
            ShellStatus = SHELL_ACCESS_DENIED;\r
          }\r
        }\r
\r
        break;\r
    } // switch on parameter count\r
\r
    if (FileList != NULL) {\r
      ShellCloseFileMetaArg(&FileList);\r
    }\r
\r
    //\r
    // free the command line package\r
    //\r
    ShellCommandLineFreeVarList (Package);\r
  }\r
\r
  if (ShellGetExecutionBreakFlag()) {\r
    return (SHELL_ABORTED);\r
  }\r
\r
  return (ShellStatus);\r
}\r
\r
Commit	Line	Data
a405b86d	1	/** @file\r
	2	Main file for cp shell level 2 function.\r
	3	\r
	4	Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
	5	This program and the accompanying materials\r
	6	are licensed and made available under the terms and conditions of the BSD License\r
	7	which accompanies this distribution. The full text of the license may be found at\r
	8	http://opensource.org/licenses/bsd-license.php\r
	9	\r
	10	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	11	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	12	\r
	13	**/\r
	14	\r
	15	#include "UefiShellLevel2CommandsLib.h"\r
	16	\r
	17	// this is later in the file.\r
	18	SHELL_STATUS\r
	19	EFIAPI\r
	20	ValidateAndCopyFiles(\r
	21	IN CONST EFI_SHELL_FILE_INFO *FileList,\r
	22	IN CONST CHAR16 *DestDir,\r
	23	IN BOOLEAN SilentMode,\r
	24	IN BOOLEAN RecursiveMode,\r
	25	IN VOID **Resp\r
	26	);\r
	27	\r
	28	/**\r
	29	Function to Copy one file to another location\r
	30	\r
	31	If the destination exists the user will be prompted and the result put into *resp\r
	32	\r
	33	@param[in] Source pointer to source file name\r
	34	@param[in] Dest pointer to destination file name\r
	35	@param[out] Resp pointer to response from question. Pass back on looped calling\r
	36	@param[in] SilentMode whether to run in quiet mode or not\r
	37	\r
	38	@retval SHELL_SUCCESS The source file was copied to the destination\r
	39	**/\r
	40	SHELL_STATUS\r
	41	EFIAPI\r
	42	CopySingleFile(\r
	43	IN CONST CHAR16 *Source,\r
	44	IN CONST CHAR16 *Dest,\r
	45	OUT VOID **Resp,\r
	46	IN BOOLEAN SilentMode\r
	47	)\r
	48	{\r
	49	VOID *Response;\r
	50	UINTN ReadSize;\r
	51	SHELL_FILE_HANDLE SourceHandle;\r
	52	SHELL_FILE_HANDLE DestHandle;\r
	53	EFI_STATUS Status;\r
	54	VOID *Buffer;\r
	55	CHAR16 *TempName;\r
	56	UINTN Size;\r
	57	EFI_SHELL_FILE_INFO *List;\r
	58	SHELL_STATUS ShellStatus;\r
	59	\r
	60	\r
	61	ASSERT(Resp != NULL);\r
	62	\r
	63	SourceHandle = NULL;\r
	64	DestHandle = NULL;\r
65	Response = *Resp;\r
66	List = NULL;\r
67	\r
68	ReadSize = PcdGet16(PcdShellFileOperationSize);\r
69	// Why bother copying a file to itself\r
70	if (StrCmp(Source, Dest) == 0) {\r
71	return (SHELL_SUCCESS);\r
72	}\r
73	\r
74	//\r
75	// Open destination file without create\r
76	//\r
77	Status = ShellOpenFileByName(Dest, &DestHandle, EFI_FILE_MODE_READ\|EFI_FILE_MODE_WRITE, 0);\r
78	\r
79	//\r
80	// close file\r
81	//\r
82	if (DestHandle != NULL) {\r
83	ShellCloseFile(&DestHandle);\r
84	DestHandle = NULL;\r
85	}\r
86	\r
87	//\r
88	// if the destination file existed check response and possibly prompt user\r
89	//\r
90	if (!EFI_ERROR(Status)) {\r
91	if (Response == NULL && !SilentMode) {\r
92	Status = ShellPromptForResponseHii(ShellPromptResponseTypeYesNoAllCancel, STRING_TOKEN (STR_CP_PROMPT), gShellLevel2HiiHandle, &Response);\r
93	}\r
94	//\r
95	// possibly return based on response\r
96	//\r
97	if (!SilentMode) {\r
98	switch ((SHELL_PROMPT_RESPONSE)Response) {\r
99	case ShellPromptResponseNo:\r
100	//\r
101	// return success here so we dont stop the process\r
102	//\r
103	return (SHELL_SUCCESS);\r
104	case ShellPromptResponseCancel:\r
105	*Resp = Response;\r
106	//\r
107	// indicate to stop everything\r
108	//\r
109	return (SHELL_ABORTED);\r
110	case ShellPromptResponseAll:\r
111	*Resp = Response;\r
112	case ShellPromptResponseYes:\r
113	break;\r
e9723321	114	default:\r
e9723321	115	return SHELL_ABORTED;\r
a405b86d	116	}\r
	117	}\r
	118	}\r
	119	\r
	120	if (ShellIsDirectory(Source) == EFI_SUCCESS) {\r
	121	Status = ShellCreateDirectory(Dest, &DestHandle);\r
	122	if (EFI_ERROR(Status)) {\r
	123	return (SHELL_ACCESS_DENIED);\r
	124	}\r
	125	\r
	126	//\r
	127	// Now copy all the files under the directory...\r
	128	//\r
	129	TempName = NULL;\r
	130	Size = 0;\r
	131	StrnCatGrow(&TempName, &Size, Source, 0);\r
	132	StrnCatGrow(&TempName, &Size, L"\\*", 0);\r
	133	ShellOpenFileMetaArg((CHAR16*)TempName, EFI_FILE_MODE_READ, &List);\r
	134	TempName = NULL;\r
	135	StrnCatGrow(&TempName, &Size, Dest, 0);\r
	136	StrnCatGrow(&TempName, &Size, L"\\", 0);\r
	137	ShellStatus = ValidateAndCopyFiles(List, TempName, SilentMode, TRUE, Resp);\r
	138	ShellCloseFileMetaArg(&List);\r
	139	FreePool(TempName);\r
	140	Size = 0;\r
	141	} else {\r
	142	//\r
	143	// open file with create enabled\r
	144	//\r
	145	Status = ShellOpenFileByName(Dest, &DestHandle, EFI_FILE_MODE_READ\|EFI_FILE_MODE_WRITE\|EFI_FILE_MODE_CREATE, 0);\r
	146	if (EFI_ERROR(Status)) {\r
	147	return (SHELL_ACCESS_DENIED);\r
	148	}\r
	149	\r
	150	//\r
	151	// open source file\r
	152	//\r
	153	Status = ShellOpenFileByName(Source, &SourceHandle, EFI_FILE_MODE_READ, 0);\r
	154	ASSERT_EFI_ERROR(Status);\r
	155	\r
	156	//\r
	157	// copy data between files\r
	158	//\r
	159	Buffer = AllocateZeroPool(ReadSize);\r
	160	ASSERT(Buffer != NULL);\r
	161	while (ReadSize == PcdGet16(PcdShellFileOperationSize) && !EFI_ERROR(Status)) {\r
	162	Status = ShellReadFile(SourceHandle, &ReadSize, Buffer);\r
	163	ASSERT_EFI_ERROR(Status);\r
	164	Status = ShellWriteFile(DestHandle, &ReadSize, Buffer);\r
	165	}\r
	166	}\r
	167	\r
	168	//\r
	169	// close files\r
	170	//\r
	171	if (DestHandle != NULL) {\r
	172	ShellCloseFile(&DestHandle);\r
	173	DestHandle = NULL;\r
	174	}\r
	175	if (SourceHandle != NULL) {\r
	176	ShellCloseFile(&SourceHandle);\r
	177	SourceHandle = NULL;\r
	178	}\r
	179	\r
180	//\r
181	// return\r
182	//\r
183	return (SHELL_SUCCESS);\r
184	}\r
185	\r
186	/**\r
187	function to take a list of files to copy and a destination location and do\r
188	the verification and copying of those files to that location. This function\r
189	will report any errors to the user and halt.\r
190	\r
191	The key is to have this function called ONLY once. this allows for the parameter\r
192	verification to happen correctly.\r
193	\r
194	@param[in] FileList A LIST_ENTRY* based list of files to move\r
195	@param[in] DestDir the destination location\r
196	\r
197	@retval SHELL_SUCCESS the files were all moved.\r
198	@retval SHELL_INVALID_PARAMETER a parameter was invalid\r
199	@retval SHELL_SECURITY_VIOLATION a security violation ocurred\r
200	@retval SHELL_WRITE_PROTECTED the destination was write protected\r
201	@retval SHELL_OUT_OF_RESOURCES a memory allocation failed\r
202	**/\r
203	SHELL_STATUS\r
204	EFIAPI\r
205	ValidateAndCopyFiles(\r
206	IN CONST EFI_SHELL_FILE_INFO *FileList,\r
207	IN CONST CHAR16 *DestDir,\r
208	IN BOOLEAN SilentMode,\r
209	IN BOOLEAN RecursiveMode,\r
210	IN VOID **Resp\r
211	)\r
212	{\r
213	CHAR16 *HiiOutput;\r
214	CHAR16 *HiiResultOk;\r
215	CONST EFI_SHELL_FILE_INFO *Node;\r
216	SHELL_STATUS ShellStatus;\r
217	CHAR16 *DestPath;\r
218	VOID *Response;\r
219	UINTN PathLen;\r
220	CONST CHAR16 *Cwd;\r
221	CONST CHAR16 *TempLocation;\r
222	UINTN NewSize;\r
223	\r
224	if (Resp == NULL) {\r
225	Response = NULL;\r
226	} else {\r
227	Response = *Resp;\r
228	}\r
229	\r
230	DestPath = NULL;\r
231	ShellStatus = SHELL_SUCCESS;\r
232	PathLen = 0;\r
233	Cwd = ShellGetCurrentDir(NULL);\r
234	\r
235	ASSERT(FileList != NULL);\r
236	ASSERT(DestDir != NULL);\r
237	\r
238	//\r
239	// If we are trying to copy multiple files... make sure we got a directory for the target...\r
240	//\r
241	if (EFI_ERROR(ShellIsDirectory(DestDir)) && FileList->Link.ForwardLink != FileList->Link.BackLink) {\r
242	//\r
243	// Error for destination not a directory\r
244	//\r
245	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NOT_DIR), gShellLevel2HiiHandle, DestDir);\r
246	return (SHELL_INVALID_PARAMETER);\r
247	}\r
248	for (Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&FileList->Link)\r
249	; !IsNull(&FileList->Link, &Node->Link)\r
250	; Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&FileList->Link, &Node->Link)\r
251	){\r
252	//\r
253	// skip the directory traversing stuff...\r
254	//\r
255	if (StrCmp(Node->FileName, L".") == 0 \|\| StrCmp(Node->FileName, L"..") == 0) {\r
256	continue;\r
257	}\r
258	\r
259	NewSize = StrSize(DestDir);\r
260	NewSize += StrSize(Node->FileName);\r
261	NewSize += StrSize(Cwd);\r
262	if (NewSize > PathLen) {\r
263	PathLen = NewSize;\r
264	}\r
265	\r
266	//\r
267	// Make sure got -r if required\r
268	//\r
269	if (!RecursiveMode && !EFI_ERROR(ShellIsDirectory(Node->FullName))) {\r
270	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_DIR_REQ), gShellLevel2HiiHandle);\r
271	return (SHELL_INVALID_PARAMETER);\r
272	}\r
273	\r
274	//\r
275	// make sure got dest as dir if needed\r
276	//\r
277	if (!EFI_ERROR(ShellIsDirectory(Node->FullName)) && EFI_ERROR(ShellIsDirectory(DestDir))) {\r
278	//\r
279	// Error for destination not a directory\r
280	//\r
281	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NOT_DIR), gShellLevel2HiiHandle, DestDir);\r
282	return (SHELL_INVALID_PARAMETER);\r
283	}\r
284	}\r
285	\r
286	HiiOutput = HiiGetString (gShellLevel2HiiHandle, STRING_TOKEN (STR_CP_OUTPUT), NULL);\r
287	HiiResultOk = HiiGetString (gShellLevel2HiiHandle, STRING_TOKEN (STR_GEN_RES_OK), NULL);\r
288	DestPath = AllocatePool(PathLen);\r
289	\r
290	//\r
291	// Go through the list of files to copy...\r
292	//\r
293	for (Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&FileList->Link)\r
294	; !IsNull(&FileList->Link, &Node->Link)\r
295	; Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&FileList->Link, &Node->Link)\r
296	){\r
297	if (ShellGetExecutionBreakFlag()) {\r
298	break;\r
299	}\r
300	ASSERT(Node->FileName != NULL);\r
301	ASSERT(Node->FullName != NULL);\r
302	\r
303	//\r
304	// skip the directory traversing stuff...\r
305	//\r
306	if (StrCmp(Node->FileName, L".") == 0 \|\| StrCmp(Node->FileName, L"..") == 0) {\r
307	continue;\r
308	}\r
309	\r
310	if (FileList->Link.ForwardLink == FileList->Link.BackLink // 1 item\r
311	&& EFI_ERROR(ShellIsDirectory(DestDir)) // not an existing directory\r
312	) {\r
313	ASSERT(StrStr(DestDir, L":") == NULL);\r
314	//\r
315	// simple copy of a single file\r
316	//\r
317	StrCpy(DestPath, Cwd);\r
318	if (DestPath[StrLen(DestPath)-1] != L'\\' && DestDir[0] != L'\\') {\r
319	StrCat(DestPath, L"\\");\r
320	} else if (DestPath[StrLen(DestPath)-1] == L'\\' && DestDir[0] == L'\\') {\r
321	((CHAR16*)DestPath)[StrLen(DestPath)-1] = CHAR_NULL;\r
322	}\r
323	StrCat(DestPath, DestDir);\r
324	} else {\r
325	//\r
326	// we have multiple files or a directory in the DestDir\r
327	//\r
328	if (StrStr(DestDir, L":") == NULL) {\r
329	StrCpy(DestPath, Cwd);\r
330	if (DestPath[StrLen(DestPath)-1] != L'\\' && DestDir[0] != L'\\') {\r
331	StrCat(DestPath, L"\\");\r
332	} else if (DestPath[StrLen(DestPath)-1] == L'\\' && DestDir[0] == L'\\') {\r
333	((CHAR16*)DestPath)[StrLen(DestPath)-1] = CHAR_NULL;\r
334	}\r
335	StrCat(DestPath, DestDir);\r
336	if (DestDir[StrLen(DestDir)-1] != L'\\' && Node->FileName[0] != L'\\') {\r
337	StrCat(DestPath, L"\\");\r
338	} else if (DestDir[StrLen(DestDir)-1] == L'\\' && Node->FileName[0] == L'\\') {\r
339	((CHAR16*)DestPath)[StrLen(DestPath)-1] = CHAR_NULL;\r
340	}\r
341	StrCat(DestPath, Node->FileName);\r
342	\r
343	} else {\r
344	StrCpy(DestPath, DestDir);\r
345	if (DestDir[StrLen(DestDir)-1] != L'\\' && Node->FileName[0] != L'\\') {\r
346	StrCat(DestPath, L"\\");\r
347	} else if (DestDir[StrLen(DestDir)-1] == L'\\' && Node->FileName[0] == L'\\') {\r
348	((CHAR16*)DestDir)[StrLen(DestDir)-1] = CHAR_NULL;\r
349	}\r
350	StrCat(DestPath, Node->FileName);\r
351	}\r
352	}\r
353	\r
354	//\r
355	// Make sure the path exists\r
356	//\r
357	if (EFI_ERROR(VerifyIntermediateDirectories(DestPath))) {\r
358	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_DIR_WNF), gShellLevel2HiiHandle);\r
359	ShellStatus = SHELL_DEVICE_ERROR;\r
360	break;\r
361	}\r
362	\r
363	if ( !EFI_ERROR(ShellIsDirectory(Node->FullName))\r
364	&& !EFI_ERROR(ShellIsDirectory(DestPath))\r
365	&& StrniCmp(Node->FullName, DestPath, StrLen(DestPath)) == NULL\r
366	){\r
367	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_SD_PARENT), gShellLevel2HiiHandle);\r
368	ShellStatus = SHELL_INVALID_PARAMETER;\r
369	break;\r
370	}\r
371	if (StringNoCaseCompare(&Node->FullName, &DestPath) == 0) {\r
372	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_SD_SAME), gShellLevel2HiiHandle);\r
373	ShellStatus = SHELL_INVALID_PARAMETER;\r
374	break;\r
375	}\r
376	\r
377	if ((TempLocation = StrniCmp(Node->FullName, DestPath, StrLen(Node->FullName))) == 0\r
378	&& (DestPath[StrLen(Node->FullName)] == CHAR_NULL \|\| DestPath[StrLen(Node->FullName)] == L'\\')\r
379	) {\r
380	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_SD_SAME), gShellLevel2HiiHandle);\r
381	ShellStatus = SHELL_INVALID_PARAMETER;\r
382	break;\r
383	}\r
384	\r
385	CleanPath(DestPath);\r
386	\r
387	ShellPrintEx(-1, -1, HiiOutput, Node->FullName, DestPath);\r
388	\r
389	//\r
390	// copy single file...\r
391	//\r
392	ShellStatus = CopySingleFile(Node->FullName, DestPath, &Response, SilentMode);\r
393	if (ShellStatus != SHELL_SUCCESS) {\r
394	break;\r
395	}\r
396	}\r
397	if (ShellStatus == SHELL_SUCCESS && Resp == NULL) {\r
398	ShellPrintEx(-1, -1, L"%s", HiiResultOk);\r
399	}\r
400	\r
401	SHELL_FREE_NON_NULL(DestPath);\r
402	SHELL_FREE_NON_NULL(HiiOutput);\r
403	SHELL_FREE_NON_NULL(HiiResultOk);\r
404	if (Resp != NULL) {\r
405	SHELL_FREE_NON_NULL(Response);\r
406	}\r
407	\r
408	return (ShellStatus);\r
409	}\r
410	\r
411	SHELL_STATUS\r
412	EFIAPI\r
413	ProcessValidateAndCopyFiles(\r
414	IN EFI_SHELL_FILE_INFO *FileList,\r
415	IN CONST CHAR16 *DestDir,\r
416	IN BOOLEAN SilentMode,\r
417	IN BOOLEAN RecursiveMode\r
418	)\r
419	{\r
420	SHELL_STATUS ShellStatus;\r
421	EFI_SHELL_FILE_INFO *List;\r
422	EFI_STATUS Status;\r
423	EFI_FILE_INFO *FileInfo;\r
424	\r
425	List = NULL;\r
426	\r
427	Status = ShellOpenFileMetaArg((CHAR16*)DestDir, EFI_FILE_MODE_READ, &List);\r
428	if (List != NULL && List->Link.ForwardLink != List->Link.BackLink) {\r
429	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_MARG_ERROR), gShellLevel2HiiHandle, DestDir);\r
430	ShellStatus = SHELL_INVALID_PARAMETER;\r
431	ShellCloseFileMetaArg(&List);\r
432	} else if (List != NULL) {\r
433	ASSERT(List != NULL);\r
434	ASSERT(((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink) != NULL);\r
435	ASSERT(((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink)->FullName != NULL);\r
436	FileInfo = NULL;\r
437	FileInfo = gEfiShellProtocol->GetFileInfo(((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink)->Handle);\r
438	ASSERT(FileInfo != NULL);\r
439	if ((FileInfo->Attribute & EFI_FILE_READ_ONLY) == 0) {\r
440	ShellStatus = ValidateAndCopyFiles(FileList, ((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink)->FullName, SilentMode, RecursiveMode, NULL);\r
441	} else {\r
442	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_DEST_ERROR), gShellLevel2HiiHandle);\r
443	ShellStatus = SHELL_ACCESS_DENIED;\r
444	}\r
445	SHELL_FREE_NON_NULL(FileInfo);\r
446	ShellCloseFileMetaArg(&List);\r
447	} else {\r
448	ShellStatus = ValidateAndCopyFiles(FileList, DestDir, SilentMode, RecursiveMode, NULL);\r
449	}\r
450	\r
451	return (ShellStatus);\r
452	}\r
453	\r
454	STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
455	{L"-r", TypeFlag},\r
456	{L"-q", TypeFlag},\r
457	{NULL, TypeMax}\r
458	};\r
459	\r
460	/**\r
461	Function for 'cp' command.\r
462	\r
463	@param[in] ImageHandle Handle to the Image (NULL if Internal).\r
464	@param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
465	**/\r
466	SHELL_STATUS\r
467	EFIAPI\r
468	ShellCommandRunCp (\r
469	IN EFI_HANDLE ImageHandle,\r
470	IN EFI_SYSTEM_TABLE *SystemTable\r
471	)\r
472	{\r
473	EFI_STATUS Status;\r
474	LIST_ENTRY *Package;\r
475	CHAR16 *ProblemParam;\r
476	SHELL_STATUS ShellStatus;\r
477	UINTN ParamCount;\r
478	UINTN LoopCounter;\r
479	EFI_SHELL_FILE_INFO *FileList;\r
480	BOOLEAN SilentMode;\r
481	BOOLEAN RecursiveMode;\r
482	CONST CHAR16 *Cwd;\r
483	\r
484	ProblemParam = NULL;\r
485	ShellStatus = SHELL_SUCCESS;\r
486	ParamCount = 0;\r
487	FileList = NULL;\r
488	\r
489	//\r
490	// initialize the shell lib (we must be in non-auto-init...)\r
491	//\r
492	Status = ShellInitialize();\r
493	ASSERT_EFI_ERROR(Status);\r
494	\r
495	Status = CommandInit();\r
496	ASSERT_EFI_ERROR(Status);\r
497	\r
498	//\r
499	// parse the command line\r
500	//\r
501	Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
502	if (EFI_ERROR(Status)) {\r
503	if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
504	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, ProblemParam);\r
505	FreePool(ProblemParam);\r
506	ShellStatus = SHELL_INVALID_PARAMETER;\r
507	} else {\r
508	ASSERT(FALSE);\r
509	}\r
510	} else {\r
511	//\r
512	// check for "-?"\r
513	//\r
514	if (ShellCommandLineGetFlag(Package, L"-?")) {\r
515	ASSERT(FALSE);\r
516	}\r
517	\r
518	//\r
519	// Initialize SilentMode and RecursiveMode\r
520	//\r
521	if (gEfiShellProtocol->BatchIsActive()) {\r
522	SilentMode = TRUE;\r
523	} else {\r
524	SilentMode = ShellCommandLineGetFlag(Package, L"-q");\r
525	}\r
526	RecursiveMode = ShellCommandLineGetFlag(Package, L"-r");\r
527	\r
528	switch (ParamCount = ShellCommandLineGetCount(Package)) {\r
529	case 0:\r
530	case 1:\r
531	//\r
532	// we have insufficient parameters\r
533	//\r
534	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel2HiiHandle);\r
535	ShellStatus = SHELL_INVALID_PARAMETER;\r
536	break;\r
537	case 2:\r
538	//\r
539	// must have valid CWD for single parameter...\r
540	//\r
541	Cwd = ShellGetCurrentDir(NULL);\r
542	if (Cwd == NULL){\r
543	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellLevel2HiiHandle);\r
544	ShellStatus = SHELL_INVALID_PARAMETER;\r
545	} else {\r
546	Status = ShellOpenFileMetaArg((CHAR16*)ShellCommandLineGetRawValue(Package, 1), EFI_FILE_MODE_WRITE\|EFI_FILE_MODE_READ, &FileList);\r
547	if (FileList == NULL \|\| IsListEmpty(&FileList->Link) \|\| EFI_ERROR(Status)) {\r
548	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_NF), gShellLevel2HiiHandle, ShellCommandLineGetRawValue(Package, 1));\r
549	ShellStatus = SHELL_NOT_FOUND;\r
550	} else {\r
551	ShellStatus = ProcessValidateAndCopyFiles(FileList, Cwd, SilentMode, RecursiveMode);\r
552	}\r
553	}\r
554	\r
555	break;\r
556	default:\r
557	//\r
558	// Make a big list of all the files...\r
559	//\r
560	for (ParamCount--, LoopCounter = 1 ; LoopCounter < ParamCount && ShellStatus == SHELL_SUCCESS ; LoopCounter++) {\r
561	if (ShellGetExecutionBreakFlag()) {\r
562	break;\r
563	}\r
564	Status = ShellOpenFileMetaArg((CHAR16*)ShellCommandLineGetRawValue(Package, LoopCounter), EFI_FILE_MODE_WRITE\|EFI_FILE_MODE_READ, &FileList);\r
565	if (EFI_ERROR(Status) \|\| FileList == NULL \|\| IsListEmpty(&FileList->Link)) {\r
566	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_NF), gShellLevel2HiiHandle, ShellCommandLineGetRawValue(Package, 1));\r
567	ShellStatus = SHELL_NOT_FOUND;\r
568	}\r
569	}\r
570	//\r
571	// now copy them all...\r
572	//\r
573	if (FileList != NULL && !IsListEmpty(&FileList->Link)) {\r
574	ShellStatus = ProcessValidateAndCopyFiles(FileList, ShellCommandCleanPath((CHAR16*)ShellCommandLineGetRawValue(Package, ParamCount)), SilentMode, RecursiveMode);\r
575	Status = ShellCloseFileMetaArg(&FileList);\r
576	if (EFI_ERROR(Status) && ShellStatus == SHELL_SUCCESS) {\r
577	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_FILE), gShellLevel2HiiHandle, ShellCommandLineGetRawValue(Package, 1), ShellStatus\|MAX_BIT);\r
578	ShellStatus = SHELL_ACCESS_DENIED;\r
579	}\r
580	}\r
581	\r
582	break;\r
583	} // switch on parameter count\r
584	\r
585	if (FileList != NULL) {\r
586	ShellCloseFileMetaArg(&FileList);\r
587	}\r
588	\r
589	//\r
590	// free the command line package\r
591	//\r
592	ShellCommandLineFreeVarList (Package);\r
593	}\r
594	\r
595	if (ShellGetExecutionBreakFlag()) {\r
596	return (SHELL_ABORTED);\r
597	}\r
598	\r
599	return (ShellStatus);\r
600	}\r
601	\r