UefiCpuPkg: Move AsmRelocateApLoopStart from Mpfuncs.nasm to AmdSev.nasm

[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
\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