EmulatorPkg: Remove framework pkgs dependency from EmulatorPkg

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

/** @file\r
  Main file for attrib shell level 2 function.\r
\r
  (C) Copyright 2016 Hewlett Packard Enterprise Development LP<BR>\r
  (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>\r
  Copyright (c) 2009 - 2018, Intel Corporation. All rights reserved.<BR>\r
  Copyright (c) 2018, Dell Technologies. All rights reserved.<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "UefiShellLevel2CommandsLib.h"\r
\r
/**\r
  Function will replace drive identifier with CWD.\r
\r
  If FullPath begining with ':' is invalid path, then ASSERT.\r
  If FullPath not include dirve identifier , then do nothing.\r
  If FullPath likes "fs0:\xx" or "fs0:/xx" , then do nothing.\r
  If FullPath likes "fs0:xxx" or "fs0:", the drive replaced by CWD.\r
\r
  @param[in, out]   FullPath    The pointer to the string containing the path.\r
  @param[in]        Cwd         Current directory.\r
\r
  @retval   EFI_SUCCESS         Success.\r
  @retval   EFI_OUT_OF_SOURCES  A memory allocation failed.\r
**/\r
EFI_STATUS\r
ReplaceDriveWithCwd (\r
  IN OUT    CHAR16  **FullPath,\r
  IN CONST  CHAR16  *Cwd\r
  )\r
{\r
  CHAR16        *Splitter;\r
  CHAR16        *TempBuffer;\r
  UINTN         TotalSize;\r
\r
  Splitter   = NULL;\r
  TempBuffer = NULL;\r
  TotalSize  = 0;\r
\r
  if (FullPath == NULL || *FullPath == NULL) {\r
    return EFI_SUCCESS;\r
  }\r
\r
  Splitter = StrStr (*FullPath, L":");\r
  ASSERT(Splitter != *FullPath);\r
\r
  if (Splitter != NULL && *(Splitter + 1) != L'\\' && *(Splitter + 1) != L'/') {\r
    TotalSize = StrSize (Cwd) + StrSize (Splitter + 1);\r
    TempBuffer = AllocateZeroPool (TotalSize);\r
    if (TempBuffer == NULL) {\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
\r
    StrCpyS (TempBuffer, TotalSize / sizeof(CHAR16), Cwd);\r
    StrCatS (TempBuffer, TotalSize / sizeof(CHAR16), L"\\");\r
    StrCatS (TempBuffer, TotalSize / sizeof(CHAR16), Splitter + 1);\r
\r
    FreePool(*FullPath);\r
    *FullPath = TempBuffer;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  function to determine if FullPath is under current filesystem.\r
\r
  @param[in]    FullPath    The target location to determine.\r
  @param[in]    Cwd         Current directory.\r
\r
  @retval       TRUE        The FullPath is in the current filesystem.\r
  @retval       FALSE       The FullPaht isn't in the current filesystem.\r
**/\r
BOOLEAN\r
IsCurrentFileSystem (\r
  IN CONST CHAR16   *FullPath,\r
  IN CONST CHAR16   *Cwd\r
  )\r
{\r
  CHAR16 *Splitter1;\r
  CHAR16 *Splitter2;\r
\r
  Splitter1 = NULL;\r
  Splitter2 = NULL;\r
\r
  ASSERT(FullPath != NULL);\r
\r
  Splitter1 = StrStr (FullPath, L":");\r
  if (Splitter1 == NULL) {\r
    return TRUE;\r
  }\r
\r
  Splitter2 = StrStr (Cwd, L":");\r
\r
  if (((UINTN) Splitter1 - (UINTN) FullPath) != ((UINTN) Splitter2 - (UINTN) Cwd)) {\r
    return FALSE;\r
  } else {\r
    if (StrniCmp (FullPath, Cwd, ((UINTN) Splitter1 - (UINTN) FullPath) / sizeof (CHAR16)) == 0) {\r
      return TRUE;\r
    } else {\r
      return FALSE;\r
    }\r
  }\r
}\r
\r
/**\r
  Extract drive string and path string from FullPath.\r
\r
  The caller must be free Drive and Path.\r
\r
  @param[in]    FullPath    A path to be extracted.\r
  @param[out]   Drive       Buffer to save drive identifier.\r
  @param[out]   Path        Buffer to save path.\r
\r
  @retval       EFI_SUCCESS           Success.\r
  @retval       EFI_OUT_OF_RESOUCES   A memory allocation failed.\r
**/\r
EFI_STATUS\r
ExtractDriveAndPath (\r
  IN CONST CHAR16   *FullPath,\r
  OUT CHAR16        **Drive,\r
  OUT CHAR16        **Path\r
  )\r
{\r
  CHAR16 *Splitter;\r
\r
  ASSERT (FullPath != NULL);\r
\r
  Splitter = StrStr (FullPath, L":");\r
\r
  if (Splitter == NULL) {\r
    *Drive = NULL;\r
    *Path = AllocateCopyPool (StrSize (FullPath), FullPath);\r
    if (*Path == NULL) {\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
  } else {\r
    if (*(Splitter + 1) == CHAR_NULL) {\r
      *Drive = AllocateCopyPool (StrSize (FullPath), FullPath);\r
      *Path = NULL;\r
      if (*Drive == NULL) {\r
        return EFI_OUT_OF_RESOURCES;\r
      }\r
    } else {\r
      *Drive = AllocateCopyPool ((Splitter - FullPath + 2) * sizeof(CHAR16), FullPath);\r
      if (*Drive == NULL) {\r
        return EFI_OUT_OF_RESOURCES;\r
      }\r
      (*Drive)[Splitter - FullPath + 1] = CHAR_NULL;\r
\r
      *Path = AllocateCopyPool (StrSize (Splitter + 1), Splitter + 1);\r
      if (*Path == NULL) {\r
        FreePool (*Drive);\r
        return EFI_OUT_OF_RESOURCES;\r
      }\r
    }\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Function for 'cd' 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
ShellCommandRunCd (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  EFI_STATUS        Status;\r
  LIST_ENTRY        *Package;\r
  CONST CHAR16      *Cwd;\r
  CHAR16            *Path;\r
  CHAR16            *Drive;\r
  CHAR16            *ProblemParam;\r
  SHELL_STATUS      ShellStatus;\r
  CONST CHAR16      *Param1;\r
  CHAR16            *Param1Copy;\r
  CHAR16            *Walker;\r
  CHAR16            *Splitter;\r
  CHAR16            *TempBuffer;\r
  UINTN             TotalSize;\r
\r
  ProblemParam  = NULL;\r
  ShellStatus   = SHELL_SUCCESS;\r
  Cwd           = NULL;\r
  Path          = NULL;\r
  Drive         = NULL;\r
  Splitter      = NULL;\r
  TempBuffer    = NULL;\r
  TotalSize     = 0;\r
\r
  Status = CommandInit();\r
  ASSERT_EFI_ERROR(Status);\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
  //\r
  // parse the command line\r
  //\r
  Status = ShellCommandLineParse (EmptyParamList, &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, L"cd", ProblemParam);\r
      FreePool(ProblemParam);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } else {\r
      ASSERT(FALSE);\r
    }\r
  }\r
\r
  //\r
  // check for "-?"\r
  //\r
  if (ShellCommandLineGetFlag(Package, L"-?")) {\r
    ASSERT(FALSE);\r
  } else if (ShellCommandLineGetRawValue(Package, 2) != NULL) {\r
    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel2HiiHandle, L"cd");\r
    ShellStatus = SHELL_INVALID_PARAMETER;\r
  } else {\r
    //\r
    // remember that param 0 is the command name\r
    // If there are 0 value parameters, then print the current directory\r
    // else If there are 2 value parameters, then print the error message\r
    // else If there is  1 value paramerer , then change the directory\r
    //\r
    Cwd = ShellGetCurrentDir (NULL);\r
    if (Cwd == NULL) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN(STR_GEN_NO_CWD), gShellLevel2HiiHandle, L"cd");\r
      ShellStatus = SHELL_NOT_FOUND;\r
    } else {\r
      Param1 = ShellCommandLineGetRawValue (Package, 1);\r
      if (Param1 == NULL) {\r
        //\r
        // display the current directory\r
        //\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN(STR_CD_PRINT), gShellLevel2HiiHandle, Cwd);\r
      } else {\r
        Param1Copy = CatSPrint (NULL, L"%s", Param1, NULL);\r
        for (Walker = Param1Copy; Walker != NULL && *Walker != CHAR_NULL; Walker++) {\r
          if (*Walker == L'\"') {\r
            CopyMem (Walker, Walker + 1, StrSize(Walker) - sizeof(Walker[0]));\r
          }\r
        }\r
\r
        if (Param1Copy != NULL && IsCurrentFileSystem (Param1Copy, Cwd)) {\r
          Status = ReplaceDriveWithCwd (&Param1Copy,Cwd);\r
        } else {\r
          //\r
          // Can't use cd command to change filesystem.\r
          //\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_CD_NF), gShellLevel2HiiHandle, L"cd");\r
          Status = EFI_NOT_FOUND;\r
        }\r
\r
        if (!EFI_ERROR(Status) && Param1Copy != NULL) {\r
          Splitter = StrStr (Cwd, L":");\r
          if (Param1Copy[0] == L'\\') {\r
            //\r
            // Absolute Path on current drive letter.\r
            //\r
            TotalSize = ((Splitter - Cwd + 1) * sizeof(CHAR16)) + StrSize(Param1Copy);\r
            TempBuffer = AllocateZeroPool (TotalSize);\r
            if (TempBuffer == NULL) {\r
              Status = EFI_OUT_OF_RESOURCES;\r
            } else {\r
              StrnCpyS (TempBuffer, TotalSize / sizeof(CHAR16), Cwd, (Splitter - Cwd + 1));\r
              StrCatS (TempBuffer, TotalSize / sizeof(CHAR16), Param1Copy);\r
\r
              FreePool (Param1Copy);\r
              Param1Copy = TempBuffer;\r
              TempBuffer = NULL;\r
            }\r
          } else {\r
            if (StrStr (Param1Copy,L":") == NULL) {\r
              TotalSize = StrSize (Cwd) + StrSize (Param1Copy);\r
              TempBuffer = AllocateZeroPool (TotalSize);\r
              if (TempBuffer == NULL) {\r
                Status = EFI_OUT_OF_RESOURCES;\r
              } else {\r
                StrCpyS (TempBuffer, TotalSize / sizeof (CHAR16), Cwd);\r
                StrCatS (TempBuffer, TotalSize / sizeof (CHAR16), L"\\");\r
                StrCatS (TempBuffer, TotalSize / sizeof (CHAR16), Param1Copy);\r
\r
                FreePool (Param1Copy);\r
                Param1Copy = TempBuffer;\r
                TempBuffer = NULL;\r
              }\r
            }\r
          }\r
        }\r
\r
        if (!EFI_ERROR(Status)) {\r
          Param1Copy = PathCleanUpDirectories (Param1Copy);\r
          Status = ExtractDriveAndPath (Param1Copy, &Drive, &Path);\r
        }\r
\r
        if (!EFI_ERROR (Status) && Drive != NULL && Path != NULL) {\r
          if (EFI_ERROR(ShellIsDirectory (Param1Copy))) {\r
            ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN(STR_GEN_NOT_DIR), gShellLevel2HiiHandle, L"cd", Param1Copy);\r
            ShellStatus = SHELL_NOT_FOUND;\r
          } else {\r
            Status = gEfiShellProtocol->SetCurDir (Drive, Path + 1);\r
            if (EFI_ERROR (Status)) {\r
              ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN(STR_GEN_DIR_NF), gShellLevel2HiiHandle, L"cd", Param1Copy);\r
              ShellStatus = SHELL_NOT_FOUND;\r
            }\r
          }\r
        }\r
\r
        if (Drive != NULL) {\r
          FreePool (Drive);\r
        }\r
\r
        if (Path != NULL) {\r
          FreePool (Path);\r
        }\r
\r
        FreePool (Param1Copy);\r
      }\r
    }\r
  }\r
\r
  //\r
  // free the command line package\r
  //\r
  ShellCommandLineFreeVarList (Package);\r
\r
  //\r
  // return the status\r
  //\r
  return (ShellStatus);\r
}\r
\r