udk2010.up2.shell initial release.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10874 6f19259b-4bc3-4df7-8a09-765794883524

1 files changed, 329 insertions, 0 deletions

diff --git a/ShellPkg/Library/UefiShellLevel2CommandsLib/UefiShellLevel2CommandsLib.c b/ShellPkg/Library/UefiShellLevel2CommandsLib/UefiShellLevel2CommandsLib.c
new file mode 100644
index 0000000000..7fe5a1102c
--- /dev/null
+++ b/ShellPkg/Library/UefiShellLevel2CommandsLib/UefiShellLevel2CommandsLib.c
@@ -0,0 +1,329 @@
+/** @file

+  Main file for NULL named library for level 2 shell command functions.

+

+  these functions are:

+  attrib,

+  cd,

+  cp,

+  date*,

+  time*,

+  load,

+  ls,

+  map,

+  mkdir,

+  mv,

+  parse,

+  rm,

+  reset,

+  set,

+  timezone*

+

+  * functions are non-interactive only

+

+

+  Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>

+  This program and the accompanying materials

+  are licensed and made available under the terms and conditions of the BSD License

+  which accompanies this distribution.  The full text of the license may be found at

+  http://opensource.org/licenses/bsd-license.php

+

+  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,

+  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.

+

+**/

+#include "UefiShellLevel2CommandsLib.h"

+

+CONST CHAR16 mFileName[] = L"ShellCommands";

+EFI_HANDLE gShellLevel2HiiHandle = NULL;

+CONST EFI_GUID gShellLevel2HiiGuid = \

+  { \

+    0xf95a7ccc, 0x4c55, 0x4426, { 0xa7, 0xb4, 0xdc, 0x89, 0x61, 0x95, 0xb, 0xae } \

+  };

+

+CONST CHAR16*

+EFIAPI

+ShellCommandGetManFileNameLevel2 (

+  VOID

+  )

+{

+  return (mFileName);

+}

+

+/**

+  Constructor for the Shell Level 2 Commands library.

+

+  Install the handlers for level 2 UEFI Shell 2.0 commands.

+

+  @param ImageHandle    the image handle of the process

+  @param SystemTable    the EFI System Table pointer

+

+  @retval EFI_SUCCESS        the shell command handlers were installed sucessfully

+  @retval EFI_UNSUPPORTED    the shell level required was not found.

+**/

+EFI_STATUS

+EFIAPI

+ShellLevel2CommandsLibConstructor (

+  IN EFI_HANDLE        ImageHandle,

+  IN EFI_SYSTEM_TABLE  *SystemTable

+  )

+{

+  //

+  // if shell level is less than 2 do nothing

+  //

+  if (PcdGet8(PcdShellSupportLevel) < 2) {

+    return (EFI_UNSUPPORTED);

+  }

+

+  gShellLevel2HiiHandle = HiiAddPackages (&gShellLevel2HiiGuid, gImageHandle, UefiShellLevel2CommandsLibStrings, NULL);

+  if (gShellLevel2HiiHandle == NULL) {

+    return (EFI_DEVICE_ERROR);

+  }

+

+  //

+  // install our shell command handlers that are always installed

+  //

+  ShellCommandRegisterCommandName(L"attrib",   ShellCommandRunAttrib  , ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_ATTRIB) );

+  ShellCommandRegisterCommandName(L"cd",       ShellCommandRunCd      , ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_CD)     );

+  ShellCommandRegisterCommandName(L"cp",       ShellCommandRunCp      , ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_CP)     );

+  ShellCommandRegisterCommandName(L"load",     ShellCommandRunLoad    , ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_LOAD)   );

+  ShellCommandRegisterCommandName(L"map",      ShellCommandRunMap     , ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_MAP)    );

+  ShellCommandRegisterCommandName(L"mkdir",    ShellCommandRunMkDir   , ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_MKDIR)  );

+  ShellCommandRegisterCommandName(L"mv",       ShellCommandRunMv      , ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_MV)     );

+  ShellCommandRegisterCommandName(L"parse",    ShellCommandRunParse   , ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_PARSE)  );

+  ShellCommandRegisterCommandName(L"reset",    ShellCommandRunReset   , ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_RESET)  );

+  ShellCommandRegisterCommandName(L"set",      ShellCommandRunSet     , ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_SET)    );

+  ShellCommandRegisterCommandName(L"ls",       ShellCommandRunLs      , ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_LS)     );

+  ShellCommandRegisterCommandName(L"rm",       ShellCommandRunRm      , ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_RM)     );

+

+  //

+  // support for permenant (built in) aliases

+  //

+  ShellCommandRegisterAlias(L"rm", L"del");

+  ShellCommandRegisterAlias(L"ls", L"dir");

+  ShellCommandRegisterAlias(L"cp", L"copy");

+  ShellCommandRegisterAlias(L"mkdir", L"md");

+  ShellCommandRegisterAlias(L"cd ..", L"cd..");

+  ShellCommandRegisterAlias(L"cd \\", L"cd\\");

+  //

+  // These are installed in level 2 or 3...

+  //

+  if (PcdGet8(PcdShellSupportLevel) == 2 || PcdGet8(PcdShellSupportLevel) == 3) {

+    ShellCommandRegisterCommandName(L"date",     ShellCommandRunDate    , ShellCommandGetManFileNameLevel2, PcdGet8(PcdShellSupportLevel), L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_DATE)   );

+    ShellCommandRegisterCommandName(L"time",     ShellCommandRunTime    , ShellCommandGetManFileNameLevel2, PcdGet8(PcdShellSupportLevel), L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_TIME)   );

+    ShellCommandRegisterCommandName(L"timezone", ShellCommandRunTimeZone, ShellCommandGetManFileNameLevel2, PcdGet8(PcdShellSupportLevel), L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_TIMEZONE));

+  } else {

+    DEBUG_CODE_BEGIN();

+    //

+    // we want to be able to test these so install them under a different name in debug mode...

+    //

+    ShellCommandRegisterCommandName(L"l2date",     ShellCommandRunDate    , ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_DATE)   );

+    ShellCommandRegisterCommandName(L"l2time",     ShellCommandRunTime    , ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_TIME)   );

+    ShellCommandRegisterCommandName(L"l2timezone", ShellCommandRunTimeZone, ShellCommandGetManFileNameLevel2, 2, L"", TRUE, gShellLevel2HiiHandle, STRING_TOKEN(STR_GET_HELP_TIMEZONE));

+    DEBUG_CODE_END();

+  }

+

+  return (EFI_SUCCESS);

+}

+

+/**

+  Destructor for the library.  free any resources.

+

+  @param ImageHandle    The image handle of the process.

+  @param SystemTable    The EFI System Table pointer.

+

+  @retval EFI_SUCCESS   Always returned.

+**/

+EFI_STATUS

+EFIAPI

+ShellLevel2CommandsLibDestructor (

+  IN EFI_HANDLE        ImageHandle,

+  IN EFI_SYSTEM_TABLE  *SystemTable

+  )

+{

+  if (gShellLevel2HiiHandle != NULL) {

+    HiiRemovePackages(gShellLevel2HiiHandle);

+  }

+  return (EFI_SUCCESS);

+}

+

+/**

+  Function to clean up paths.  Removes the following items:

+    single periods in the path (no need for the current directory tag)

+    double periods in the path and removes a single parent directory.

+

+  This will be done inline and the resultant string may be be 'too big'.

+

+  @param[in] PathToReturn  The pointer to the string containing the path.

+

+  @return PathToReturn is always returned.

+**/

+CHAR16*

+EFIAPI

+CleanPath(

+  IN CHAR16 *PathToReturn

+  )

+{

+  CHAR16  *TempString;

+  UINTN   TempSize;

+  if (PathToReturn==NULL) {

+    return(NULL);

+  }

+  //

+  // Fix up the directory name

+  //

+  while ((TempString = StrStr(PathToReturn, L"\\..\\")) != NULL) {

+    *TempString = CHAR_NULL;

+    TempString  += 4;

+    ChopLastSlash(PathToReturn);

+    TempSize = StrSize(TempString);

+    CopyMem(PathToReturn+StrLen(PathToReturn), TempString, TempSize);

+  }

+  if ((TempString = StrStr(PathToReturn, L"\\..")) != NULL && *(TempString + 3) == CHAR_NULL) {

+    *TempString = CHAR_NULL;

+    ChopLastSlash(PathToReturn);

+  }

+  while ((TempString = StrStr(PathToReturn, L"\\.\\")) != NULL) {

+    *TempString = CHAR_NULL;

+    TempString  += 2;

+    TempSize = StrSize(TempString);

+    CopyMem(PathToReturn+StrLen(PathToReturn), TempString, TempSize);

+  }

+  if ((TempString = StrStr(PathToReturn, L"\\.")) != NULL && *(TempString + 2) == CHAR_NULL) {

+    *TempString = CHAR_NULL;

+  }

+  return (PathToReturn);

+}

+

+/**

+  returns a fully qualified directory (contains a map drive at the begining)

+  path from a unknown directory path.

+

+  If Path is already fully qualified this will return a duplicat otherwise this

+  will use get the current directory and use that to build the fully qualified

+  version.

+

+  if the return value is not NULL it must be caller freed.

+

+  @param[in] Path         The unknown Path Value

+

+  @retval NULL            A memory allocation failed

+  @retval NULL            a fully qualified path could not be discovered.

+  @retval other           pointer to a fuly qualified path.

+**/

+CHAR16*

+EFIAPI

+GetFullyQualifiedPath(

+  IN CONST CHAR16* Path

+  )

+{

+  CHAR16        *PathToReturn;

+  UINTN         Size;

+  CONST CHAR16  *CurDir;

+

+  PathToReturn  = NULL;

+  Size          = 0;

+

+  ASSERT((PathToReturn == NULL && Size == 0) || (PathToReturn != NULL));

+  //

+  // convert a local path to an absolute path

+  //

+  if (StrStr(Path, L":") == NULL) {

+    CurDir = gEfiShellProtocol->GetCurDir(NULL);

+    StrnCatGrow(&PathToReturn, &Size, CurDir, 0);

+    if (*Path == L'\\') {

+      Path++;

+    }

+  }

+  StrnCatGrow(&PathToReturn, &Size, Path, 0);

+

+  CleanPath(PathToReturn);

+

+  while (PathToReturn[StrLen(PathToReturn)-1] == L'*') {

+    PathToReturn[StrLen(PathToReturn)-1] = CHAR_NULL;

+  }

+

+  return (PathToReturn);

+}

+

+/**

+  Function to verify all intermediate directories in the path.

+

+  @param[in] Path       The pointer to the path to fix.

+

+  @retval EFI_SUCCESS   The operation was successful.

+**/

+EFI_STATUS

+EFIAPI

+VerifyIntermediateDirectories (

+  IN CONST CHAR16 *Path

+  )

+{

+  EFI_STATUS      Status;

+  CHAR16          *PathCopy;

+  CHAR16          *TempSpot;

+  SHELL_FILE_HANDLE          FileHandle;

+

+  ASSERT(Path != NULL);

+

+  Status      = EFI_SUCCESS;

+  PathCopy    = NULL;

+  PathCopy    = StrnCatGrow(&PathCopy, NULL, Path, 0);

+  FileHandle  = NULL;

+

+  for (TempSpot = &PathCopy[StrLen(PathCopy)-1] ; *TempSpot != CHAR_NULL && *TempSpot != L'\\' ; TempSpot = &PathCopy[StrLen(PathCopy)-1]){

+    *TempSpot = CHAR_NULL;

+  }

+  if (*TempSpot == L'\\') {

+    *TempSpot = CHAR_NULL;

+  }

+

+  if (PathCopy != NULL && *PathCopy != CHAR_NULL) {

+    Status = VerifyIntermediateDirectories(PathCopy);

+

+    if (PathCopy[StrLen(PathCopy)-1] != L':') {

+      if (!EFI_ERROR(Status)) {

+        Status = ShellOpenFileByName(PathCopy, &FileHandle, EFI_FILE_MODE_READ, 0);

+        if (FileHandle != NULL) {

+          ShellCloseFile(&FileHandle);

+        }

+      }

+    }

+  }

+

+  SHELL_FREE_NON_NULL(PathCopy);

+

+  return (Status);

+}

+

+// be lazy and borrow from baselib.

+CHAR16

+EFIAPI

+InternalCharToUpper (

+  IN CONST CHAR16                    Char

+  );

+

+CONST CHAR16*

+EFIAPI

+StrniCmp(

+  IN CONST CHAR16 *Source,

+  IN CONST CHAR16 *Target,

+  IN CONST UINTN  Count

+  )

+{

+  UINTN   LoopCount;

+  CHAR16  Char1;

+  CHAR16  Char2;

+

+  ASSERT(Source != NULL);

+  ASSERT(Target != NULL);

+

+  for (LoopCount = 0 ; LoopCount < Count ; LoopCount++) {

+    Char1 = InternalCharToUpper(Source[LoopCount]);

+    Char2 = InternalCharToUpper(Target[LoopCount]);

+    if (Char1 != Char2) {

+      return (&Source[LoopCount]);

+    }

+  }

+  return (NULL);

+}

+