[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
  if (HiiOutput == NULL || HiiOutput == NULL || HiiResultOk == NULL) {\r
    SHELL_FREE_NON_NULL(DestPath);\r
    SHELL_FREE_NON_NULL(HiiOutput);\r
    SHELL_FREE_NON_NULL(HiiResultOk);\r
    return (SHELL_OUT_OF_RESOURCES);\r
  }\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
9ea69f8a	290	if (HiiOutput == NULL \|\| HiiOutput == NULL \|\| HiiResultOk == NULL) {\r
	291	SHELL_FREE_NON_NULL(DestPath);\r
	292	SHELL_FREE_NON_NULL(HiiOutput);\r
	293	SHELL_FREE_NON_NULL(HiiResultOk);\r
	294	return (SHELL_OUT_OF_RESOURCES);\r
	295	}\r
	296	\r
a405b86d	297	//\r
	298	// Go through the list of files to copy...\r
	299	//\r
	300	for (Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&FileList->Link)\r
	301	; !IsNull(&FileList->Link, &Node->Link)\r
	302	; Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&FileList->Link, &Node->Link)\r
	303	){\r
	304	if (ShellGetExecutionBreakFlag()) {\r
	305	break;\r
	306	}\r
	307	ASSERT(Node->FileName != NULL);\r
	308	ASSERT(Node->FullName != NULL);\r
	309	\r
	310	//\r
	311	// skip the directory traversing stuff...\r
	312	//\r
	313	if (StrCmp(Node->FileName, L".") == 0 \|\| StrCmp(Node->FileName, L"..") == 0) {\r
	314	continue;\r
	315	}\r
	316	\r
	317	if (FileList->Link.ForwardLink == FileList->Link.BackLink // 1 item\r
	318	&& EFI_ERROR(ShellIsDirectory(DestDir)) // not an existing directory\r
	319	) {\r
	320	ASSERT(StrStr(DestDir, L":") == NULL);\r
	321	//\r
	322	// simple copy of a single file\r
	323	//\r
	324	StrCpy(DestPath, Cwd);\r
	325	if (DestPath[StrLen(DestPath)-1] != L'\\' && DestDir[0] != L'\\') {\r
	326	StrCat(DestPath, L"\\");\r
	327	} else if (DestPath[StrLen(DestPath)-1] == L'\\' && DestDir[0] == L'\\') {\r
	328	((CHAR16*)DestPath)[StrLen(DestPath)-1] = CHAR_NULL;\r
	329	}\r
	330	StrCat(DestPath, DestDir);\r
	331	} else {\r
	332	//\r
	333	// we have multiple files or a directory in the DestDir\r
	334	//\r
	335	if (StrStr(DestDir, L":") == NULL) {\r
	336	StrCpy(DestPath, Cwd);\r
	337	if (DestPath[StrLen(DestPath)-1] != L'\\' && DestDir[0] != L'\\') {\r
	338	StrCat(DestPath, L"\\");\r
	339	} else if (DestPath[StrLen(DestPath)-1] == L'\\' && DestDir[0] == L'\\') {\r
	340	((CHAR16*)DestPath)[StrLen(DestPath)-1] = CHAR_NULL;\r
	341	}\r
	342	StrCat(DestPath, DestDir);\r
	343	if (DestDir[StrLen(DestDir)-1] != L'\\' && Node->FileName[0] != L'\\') {\r
	344	StrCat(DestPath, L"\\");\r
	345	} else if (DestDir[StrLen(DestDir)-1] == L'\\' && Node->FileName[0] == L'\\') {\r
	346	((CHAR16*)DestPath)[StrLen(DestPath)-1] = CHAR_NULL;\r
	347	}\r
	348	StrCat(DestPath, Node->FileName);\r
	349	\r
	350	} else {\r
	351	StrCpy(DestPath, DestDir);\r
	352	if (DestDir[StrLen(DestDir)-1] != L'\\' && Node->FileName[0] != L'\\') {\r
	353	StrCat(DestPath, L"\\");\r
	354	} else if (DestDir[StrLen(DestDir)-1] == L'\\' && Node->FileName[0] == L'\\') {\r
	355	((CHAR16*)DestDir)[StrLen(DestDir)-1] = CHAR_NULL;\r
	356	}\r
	357	StrCat(DestPath, Node->FileName);\r
	358	}\r
	359	}\r
	360	\r
361	//\r
362	// Make sure the path exists\r
363	//\r
364	if (EFI_ERROR(VerifyIntermediateDirectories(DestPath))) {\r
365	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_DIR_WNF), gShellLevel2HiiHandle);\r
366	ShellStatus = SHELL_DEVICE_ERROR;\r
367	break;\r
368	}\r
369	\r
370	if ( !EFI_ERROR(ShellIsDirectory(Node->FullName))\r
371	&& !EFI_ERROR(ShellIsDirectory(DestPath))\r
372	&& StrniCmp(Node->FullName, DestPath, StrLen(DestPath)) == NULL\r
373	){\r
374	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_SD_PARENT), gShellLevel2HiiHandle);\r
375	ShellStatus = SHELL_INVALID_PARAMETER;\r
376	break;\r
377	}\r
378	if (StringNoCaseCompare(&Node->FullName, &DestPath) == 0) {\r
379	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_SD_SAME), gShellLevel2HiiHandle);\r
380	ShellStatus = SHELL_INVALID_PARAMETER;\r
381	break;\r
382	}\r
383	\r
384	if ((TempLocation = StrniCmp(Node->FullName, DestPath, StrLen(Node->FullName))) == 0\r
385	&& (DestPath[StrLen(Node->FullName)] == CHAR_NULL \|\| DestPath[StrLen(Node->FullName)] == L'\\')\r
386	) {\r
387	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_SD_SAME), gShellLevel2HiiHandle);\r
388	ShellStatus = SHELL_INVALID_PARAMETER;\r
389	break;\r
390	}\r
391	\r
392	CleanPath(DestPath);\r
393	\r
394	ShellPrintEx(-1, -1, HiiOutput, Node->FullName, DestPath);\r
395	\r
396	//\r
397	// copy single file...\r
398	//\r
399	ShellStatus = CopySingleFile(Node->FullName, DestPath, &Response, SilentMode);\r
400	if (ShellStatus != SHELL_SUCCESS) {\r
401	break;\r
402	}\r
403	}\r
404	if (ShellStatus == SHELL_SUCCESS && Resp == NULL) {\r
405	ShellPrintEx(-1, -1, L"%s", HiiResultOk);\r
406	}\r
407	\r
408	SHELL_FREE_NON_NULL(DestPath);\r
409	SHELL_FREE_NON_NULL(HiiOutput);\r
410	SHELL_FREE_NON_NULL(HiiResultOk);\r
411	if (Resp != NULL) {\r
412	SHELL_FREE_NON_NULL(Response);\r
413	}\r
414	\r
415	return (ShellStatus);\r
416	}\r
417	\r
418	SHELL_STATUS\r
419	EFIAPI\r
420	ProcessValidateAndCopyFiles(\r
421	IN EFI_SHELL_FILE_INFO *FileList,\r
422	IN CONST CHAR16 *DestDir,\r
423	IN BOOLEAN SilentMode,\r
424	IN BOOLEAN RecursiveMode\r
425	)\r
426	{\r
427	SHELL_STATUS ShellStatus;\r
428	EFI_SHELL_FILE_INFO *List;\r
429	EFI_STATUS Status;\r
430	EFI_FILE_INFO *FileInfo;\r
431	\r
432	List = NULL;\r
433	\r
434	Status = ShellOpenFileMetaArg((CHAR16*)DestDir, EFI_FILE_MODE_READ, &List);\r
435	if (List != NULL && List->Link.ForwardLink != List->Link.BackLink) {\r
436	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_MARG_ERROR), gShellLevel2HiiHandle, DestDir);\r
437	ShellStatus = SHELL_INVALID_PARAMETER;\r
438	ShellCloseFileMetaArg(&List);\r
439	} else if (List != NULL) {\r
440	ASSERT(List != NULL);\r
441	ASSERT(((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink) != NULL);\r
442	ASSERT(((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink)->FullName != NULL);\r
443	FileInfo = NULL;\r
444	FileInfo = gEfiShellProtocol->GetFileInfo(((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink)->Handle);\r
445	ASSERT(FileInfo != NULL);\r
446	if ((FileInfo->Attribute & EFI_FILE_READ_ONLY) == 0) {\r
447	ShellStatus = ValidateAndCopyFiles(FileList, ((EFI_SHELL_FILE_INFO *)List->Link.ForwardLink)->FullName, SilentMode, RecursiveMode, NULL);\r
448	} else {\r
449	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_CP_DEST_ERROR), gShellLevel2HiiHandle);\r
450	ShellStatus = SHELL_ACCESS_DENIED;\r
451	}\r
452	SHELL_FREE_NON_NULL(FileInfo);\r
453	ShellCloseFileMetaArg(&List);\r
454	} else {\r
455	ShellStatus = ValidateAndCopyFiles(FileList, DestDir, SilentMode, RecursiveMode, NULL);\r
456	}\r
457	\r
458	return (ShellStatus);\r
459	}\r
460	\r
461	STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
462	{L"-r", TypeFlag},\r
463	{L"-q", TypeFlag},\r
464	{NULL, TypeMax}\r
465	};\r
466	\r
467	/**\r
468	Function for 'cp' command.\r
469	\r
470	@param[in] ImageHandle Handle to the Image (NULL if Internal).\r
471	@param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
472	**/\r
473	SHELL_STATUS\r
474	EFIAPI\r
475	ShellCommandRunCp (\r
476	IN EFI_HANDLE ImageHandle,\r
477	IN EFI_SYSTEM_TABLE *SystemTable\r
478	)\r
479	{\r
480	EFI_STATUS Status;\r
481	LIST_ENTRY *Package;\r
482	CHAR16 *ProblemParam;\r
483	SHELL_STATUS ShellStatus;\r
484	UINTN ParamCount;\r
485	UINTN LoopCounter;\r
486	EFI_SHELL_FILE_INFO *FileList;\r
487	BOOLEAN SilentMode;\r
488	BOOLEAN RecursiveMode;\r
489	CONST CHAR16 *Cwd;\r
490	\r
491	ProblemParam = NULL;\r
492	ShellStatus = SHELL_SUCCESS;\r
493	ParamCount = 0;\r
494	FileList = NULL;\r
495	\r
496	//\r
497	// initialize the shell lib (we must be in non-auto-init...)\r
498	//\r
499	Status = ShellInitialize();\r
500	ASSERT_EFI_ERROR(Status);\r
501	\r
502	Status = CommandInit();\r
503	ASSERT_EFI_ERROR(Status);\r
504	\r
505	//\r
506	// parse the command line\r
507	//\r
508	Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
509	if (EFI_ERROR(Status)) {\r
510	if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
511	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, ProblemParam);\r
512	FreePool(ProblemParam);\r
513	ShellStatus = SHELL_INVALID_PARAMETER;\r
514	} else {\r
515	ASSERT(FALSE);\r
516	}\r
517	} else {\r
518	//\r
519	// check for "-?"\r
520	//\r
521	if (ShellCommandLineGetFlag(Package, L"-?")) {\r
522	ASSERT(FALSE);\r
523	}\r
524	\r
525	//\r
526	// Initialize SilentMode and RecursiveMode\r
527	//\r
528	if (gEfiShellProtocol->BatchIsActive()) {\r
529	SilentMode = TRUE;\r
530	} else {\r
531	SilentMode = ShellCommandLineGetFlag(Package, L"-q");\r
532	}\r
533	RecursiveMode = ShellCommandLineGetFlag(Package, L"-r");\r
534	\r
535	switch (ParamCount = ShellCommandLineGetCount(Package)) {\r
536	case 0:\r
537	case 1:\r
538	//\r
539	// we have insufficient parameters\r
540	//\r
541	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel2HiiHandle);\r
542	ShellStatus = SHELL_INVALID_PARAMETER;\r
543	break;\r
544	case 2:\r
545	//\r
546	// must have valid CWD for single parameter...\r
547	//\r
548	Cwd = ShellGetCurrentDir(NULL);\r
549	if (Cwd == NULL){\r
550	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellLevel2HiiHandle);\r
551	ShellStatus = SHELL_INVALID_PARAMETER;\r
552	} else {\r
553	Status = ShellOpenFileMetaArg((CHAR16*)ShellCommandLineGetRawValue(Package, 1), EFI_FILE_MODE_WRITE\|EFI_FILE_MODE_READ, &FileList);\r
554	if (FileList == NULL \|\| IsListEmpty(&FileList->Link) \|\| EFI_ERROR(Status)) {\r
555	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_NF), gShellLevel2HiiHandle, ShellCommandLineGetRawValue(Package, 1));\r
556	ShellStatus = SHELL_NOT_FOUND;\r
557	} else {\r
558	ShellStatus = ProcessValidateAndCopyFiles(FileList, Cwd, SilentMode, RecursiveMode);\r
559	}\r
560	}\r
561	\r
562	break;\r
563	default:\r
564	//\r
565	// Make a big list of all the files...\r
566	//\r
567	for (ParamCount--, LoopCounter = 1 ; LoopCounter < ParamCount && ShellStatus == SHELL_SUCCESS ; LoopCounter++) {\r
568	if (ShellGetExecutionBreakFlag()) {\r
569	break;\r
570	}\r
571	Status = ShellOpenFileMetaArg((CHAR16*)ShellCommandLineGetRawValue(Package, LoopCounter), EFI_FILE_MODE_WRITE\|EFI_FILE_MODE_READ, &FileList);\r
572	if (EFI_ERROR(Status) \|\| FileList == NULL \|\| IsListEmpty(&FileList->Link)) {\r
573	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_NF), gShellLevel2HiiHandle, ShellCommandLineGetRawValue(Package, 1));\r
574	ShellStatus = SHELL_NOT_FOUND;\r
575	}\r
576	}\r
577	//\r
578	// now copy them all...\r
579	//\r
580	if (FileList != NULL && !IsListEmpty(&FileList->Link)) {\r
581	ShellStatus = ProcessValidateAndCopyFiles(FileList, ShellCommandCleanPath((CHAR16*)ShellCommandLineGetRawValue(Package, ParamCount)), SilentMode, RecursiveMode);\r
582	Status = ShellCloseFileMetaArg(&FileList);\r
583	if (EFI_ERROR(Status) && ShellStatus == SHELL_SUCCESS) {\r
584	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ERR_FILE), gShellLevel2HiiHandle, ShellCommandLineGetRawValue(Package, 1), ShellStatus\|MAX_BIT);\r
585	ShellStatus = SHELL_ACCESS_DENIED;\r
586	}\r
587	}\r
588	\r
589	break;\r
590	} // switch on parameter count\r
591	\r
592	if (FileList != NULL) {\r
593	ShellCloseFileMetaArg(&FileList);\r
594	}\r
595	\r
596	//\r
597	// free the command line package\r
598	//\r
599	ShellCommandLineFreeVarList (Package);\r
600	}\r
601	\r
602	if (ShellGetExecutionBreakFlag()) {\r
603	return (SHELL_ABORTED);\r
604	}\r
605	\r
606	return (ShellStatus);\r
607	}\r
608	\r