ShellPkg/Library/UefiShellDebug1CommandsLib/EfiCompress.c

   1 /** @file
   2   Main file for EfiCompress shell Debug1 function.
   3
   4   Copyright (c) 2005 - 2010, Intel Corporation. All rights reserved.<BR>
   5   This program and the accompanying materials
   6   are licensed and made available under the terms and conditions of the BSD License
   7   which accompanies this distribution.  The full text of the license may be found at
   8   http://opensource.org/licenses/bsd-license.php
   9
  10   THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
  11   WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
  12
  13 **/
  14
  15 #include "UefiShellDebug1CommandsLib.h"
  16 #include "Compress.h"
  17
  18 SHELL_STATUS
  19 EFIAPI
  20 ShellCommandRunEfiCompress (
  21   IN EFI_HANDLE        ImageHandle,
  22   IN EFI_SYSTEM_TABLE  *SystemTable
  23   )
  24 {
  25   EFI_STATUS          Status;
  26   LIST_ENTRY          *Package;
  27   CHAR16              *ProblemParam;
  28   SHELL_STATUS        ShellStatus;
  29   SHELL_FILE_HANDLE   InShellFileHandle;
  30   SHELL_FILE_HANDLE   OutShellFileHandle;
  31   UINT64              OutSize;
  32   VOID                *OutBuffer;
  33   UINT64              InSize;
  34   VOID                *InBuffer;
  35   CHAR16              *InFileName;
  36   CONST CHAR16        *OutFileName;
  37
  38   InFileName          = NULL;
  39   OutFileName         = NULL;
  40   OutSize             = 0;
  41   ShellStatus         = SHELL_SUCCESS;
  42   Status              = EFI_SUCCESS;
  43   OutBuffer           = NULL;
  44   InShellFileHandle   = NULL;
  45   OutShellFileHandle  = NULL;
  46   InBuffer            = NULL;
  47
  48   //
  49   // initialize the shell lib (we must be in non-auto-init...)
  50   //
  51   Status = ShellInitialize();
  52   ASSERT_EFI_ERROR(Status);
  53
  54   Status = CommandInit();
  55   ASSERT_EFI_ERROR(Status);
  56
  57   //
  58   // parse the command line
  59   //
  60   Status = ShellCommandLineParse (EmptyParamList, &Package, &ProblemParam, TRUE);
  61   if (EFI_ERROR(Status)) {
  62     if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {
  63       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, ProblemParam);
  64       FreePool(ProblemParam);
  65       ShellStatus = SHELL_INVALID_PARAMETER;
  66     } else {
  67       ASSERT(FALSE);
  68     }
  69   } else {
  70     if (ShellCommandLineGetCount(Package) > 3) {
  71       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);
  72       ShellStatus = SHELL_INVALID_PARAMETER;
  73     } else if (ShellCommandLineGetCount(Package) < 3) {
  74       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);
  75       ShellStatus = SHELL_INVALID_PARAMETER;
  76     } else {
  77       InFileName = ShellFindFilePath(ShellCommandLineGetRawValue(Package, 1));
  78       OutFileName = ShellCommandLineGetRawValue(Package, 2);
  79       Status = ShellOpenFileByName(InFileName, &InShellFileHandle, EFI_FILE_MODE_READ, 0);
  80       if (EFI_ERROR(Status)) {
  81         ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, ShellCommandLineGetRawValue(Package, 1), Status);
  82         ShellStatus = SHELL_NOT_FOUND;
  83       }
  84       Status = ShellOpenFileByName(OutFileName, &OutShellFileHandle, EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE|EFI_FILE_MODE_CREATE, 0);
  85       if (EFI_ERROR(Status)) {
  86         ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, ShellCommandLineGetRawValue(Package, 2), Status);
  87         ShellStatus = SHELL_NOT_FOUND;
  88       }
  89       if (FileHandleIsDirectory(InShellFileHandle) == EFI_SUCCESS){
  90         ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_NOT_DIR), gShellDebug1HiiHandle, InFileName);
  91         ShellStatus = SHELL_INVALID_PARAMETER;
  92       }
  93       if (FileHandleIsDirectory(OutShellFileHandle) == EFI_SUCCESS){
  94         ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_NOT_DIR), gShellDebug1HiiHandle, OutFileName);
  95         ShellStatus = SHELL_INVALID_PARAMETER;
  96       }
  97       Status = gEfiShellProtocol->GetFileSize(InShellFileHandle, &InSize);
  98       ASSERT_EFI_ERROR(Status);
  99       InBuffer = AllocateZeroPool((UINTN)InSize);
 100       ASSERT(InBuffer != NULL);
 101       Status = gEfiShellProtocol->ReadFile(InShellFileHandle, &((UINTN)InSize), InBuffer);
 102       ASSERT_EFI_ERROR(Status);
 103       Status = Compress(InBuffer, InSize, OutBuffer, &OutSize);
 104       if (Status == EFI_BUFFER_TOO_SMALL) {
 105         OutBuffer = AllocateZeroPool((UINTN)OutSize);
 106         ASSERT(OutBuffer != NULL);
 107         Status = Compress(InBuffer, InSize, OutBuffer, &OutSize);
 108       }
 109       if (EFI_ERROR(Status)) {
 110         ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_EFI_COMPRESS_FAIL), gShellDebug1HiiHandle, Status);
 111         ShellStatus = SHELL_DEVICE_ERROR;
 112       } else {
 113         Status = gEfiShellProtocol->WriteFile(OutShellFileHandle, &((UINTN)OutSize), OutBuffer);
 114         if (EFI_ERROR(Status)) {
 115           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_WRITE_FAIL), gShellDebug1HiiHandle, OutFileName, Status);
 116           ShellStatus = SHELL_DEVICE_ERROR;
 117         }
 118       }
 119     }
 120
 121     ShellCommandLineFreeVarList (Package);
 122   }
 123   if (InFileName != NULL) {
 124     FreePool(InFileName);
 125   }
 126   if (InShellFileHandle != NULL) {
 127     gEfiShellProtocol->CloseFile(InShellFileHandle);
 128   }
 129   if (OutShellFileHandle != NULL) {
 130     gEfiShellProtocol->CloseFile(OutShellFileHandle);
 131   }
 132   if (InBuffer != NULL) {
 133     FreePool(InBuffer);
 134   }
 135   if (OutBuffer != NULL) {
 136     FreePool(OutBuffer);
 137   }
 138
 139   return (ShellStatus);
 140 }