Add "Debug1" profile (all but Edit and HexEdit commands)

[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / EfiCompress.c

diff --git a/ShellPkg/Library/UefiShellDebug1CommandsLib/EfiCompress.c b/ShellPkg/Library/UefiShellDebug1CommandsLib/EfiCompress.c
new file mode 100644 (file)
index 0000000..8d5aabd
--- /dev/null
+++ b/ShellPkg/Library/UefiShellDebug1CommandsLib/EfiCompress.c
@@ -0,0 +1,140 @@
+/** @file\r
+  Main file for EfiCompress shell Debug1 function.\r
+\r
+  Copyright (c) 2005 - 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 "UefiShellDebug1CommandsLib.h"\r
+#include "Compress.h"\r
+\r
+SHELL_STATUS\r
+EFIAPI\r
+ShellCommandRunEfiCompress (\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
+  SHELL_FILE_HANDLE   InShellFileHandle;\r
+  SHELL_FILE_HANDLE   OutShellFileHandle;\r
+  UINT64              OutSize;\r
+  VOID                *OutBuffer;\r
+  UINT64              InSize;\r
+  VOID                *InBuffer;\r
+  CHAR16              *InFileName;\r
+  CONST CHAR16        *OutFileName;\r
+\r
+  InFileName          = NULL;\r
+  OutFileName         = NULL;\r
+  OutSize             = 0;\r
+  ShellStatus         = SHELL_SUCCESS;\r
+  Status              = EFI_SUCCESS;\r
+  OutBuffer           = NULL;\r
+  InShellFileHandle   = NULL;\r
+  OutShellFileHandle  = NULL;\r
+  InBuffer            = 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 (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), gShellDebug1HiiHandle, ProblemParam);\r
+      FreePool(ProblemParam);\r
+      ShellStatus = SHELL_INVALID_PARAMETER;\r
+    } else {\r
+      ASSERT(FALSE);\r
+    }\r
+  } else {\r
+    if (ShellCommandLineGetCount(Package) > 3) {\r
+      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
+      ShellStatus = SHELL_INVALID_PARAMETER;\r
+    } else if (ShellCommandLineGetCount(Package) < 3) {\r
+      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
+      ShellStatus = SHELL_INVALID_PARAMETER;\r
+    } else {\r
+      InFileName = ShellFindFilePath(ShellCommandLineGetRawValue(Package, 1));\r
+      OutFileName = ShellCommandLineGetRawValue(Package, 2);\r
+      Status = ShellOpenFileByName(InFileName, &InShellFileHandle, EFI_FILE_MODE_READ, 0);\r
+      if (EFI_ERROR(Status)) {\r
+        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, ShellCommandLineGetRawValue(Package, 1), Status);\r
+        ShellStatus = SHELL_NOT_FOUND;\r
+      }\r
+      Status = ShellOpenFileByName(OutFileName, &OutShellFileHandle, EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE|EFI_FILE_MODE_CREATE, 0);\r
+      if (EFI_ERROR(Status)) {\r
+        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, ShellCommandLineGetRawValue(Package, 2), Status);\r
+        ShellStatus = SHELL_NOT_FOUND;\r
+      }\r
+      if (FileHandleIsDirectory(InShellFileHandle) == EFI_SUCCESS){\r
+        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_NOT_DIR), gShellDebug1HiiHandle, InFileName);\r
+        ShellStatus = SHELL_INVALID_PARAMETER;\r
+      }\r
+      if (FileHandleIsDirectory(OutShellFileHandle) == EFI_SUCCESS){\r
+        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_NOT_DIR), gShellDebug1HiiHandle, OutFileName);\r
+        ShellStatus = SHELL_INVALID_PARAMETER;\r
+      }\r
+      Status = gEfiShellProtocol->GetFileSize(InShellFileHandle, &InSize);\r
+      ASSERT_EFI_ERROR(Status);\r
+      InBuffer = AllocateZeroPool((UINTN)InSize);\r
+      ASSERT(InBuffer != NULL);\r
+      Status = gEfiShellProtocol->ReadFile(InShellFileHandle, &((UINTN)InSize), InBuffer);\r
+      ASSERT_EFI_ERROR(Status);\r
+      Status = Compress(InBuffer, InSize, OutBuffer, &OutSize);\r
+      if (Status == EFI_BUFFER_TOO_SMALL) {\r
+        OutBuffer = AllocateZeroPool((UINTN)OutSize);\r
+        ASSERT(OutBuffer != NULL);\r
+        Status = Compress(InBuffer, InSize, OutBuffer, &OutSize);\r
+      }\r
+      if (EFI_ERROR(Status)) {\r
+        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_EFI_COMPRESS_FAIL), gShellDebug1HiiHandle, Status);\r
+        ShellStatus = SHELL_DEVICE_ERROR;\r
+      } else {\r
+        Status = gEfiShellProtocol->WriteFile(OutShellFileHandle, &((UINTN)OutSize), OutBuffer);\r
+        if (EFI_ERROR(Status)) {\r
+          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_WRITE_FAIL), gShellDebug1HiiHandle, OutFileName, Status);\r
+          ShellStatus = SHELL_DEVICE_ERROR;\r
+        }\r
+      }\r
+    }\r
+\r
+    ShellCommandLineFreeVarList (Package);\r
+  }\r
+  if (InFileName != NULL) {\r
+    FreePool(InFileName);\r
+  }\r
+  if (InShellFileHandle != NULL) {\r
+    gEfiShellProtocol->CloseFile(InShellFileHandle);\r
+  }\r
+  if (OutShellFileHandle != NULL) {\r
+    gEfiShellProtocol->CloseFile(OutShellFileHandle);\r
+  }\r
+  if (InBuffer != NULL) {\r
+    FreePool(InBuffer);\r
+  }\r
+  if (OutBuffer != NULL) {\r
+    FreePool(OutBuffer);\r
+  }\r
+\r
+  return (ShellStatus);\r
+}\r