From 6b80310f34199d1f62e45e40fa902734735091fa Mon Sep 17 00:00:00 2001 From: "Zhang, Shenglei" Date: Tue, 17 Sep 2019 13:59:05 +0800 Subject: [PATCH] BaseTools/LzmaCompress: Add two switches As is requested in the BZ 2077, add two switches to support setting compression mode and dictionary size. (https://bugzilla.tianocore.org/show_bug.cgi?id=2077) Cc: Bob Feng Cc: Liming Gao Signed-off-by: Shenglei Zhang Reviewed-by: Liming Gao --- BaseTools/Source/C/LzmaCompress/GNUmakefile | 4 +- .../Source/C/LzmaCompress/LzmaCompress.c | 41 +++++++++++++++---- BaseTools/Source/C/LzmaCompress/Makefile | 4 +- 3 files changed, 38 insertions(+), 11 deletions(-) diff --git a/BaseTools/Source/C/LzmaCompress/GNUmakefile b/BaseTools/Source/C/LzmaCompress/GNUmakefile index 533f0a59a0..c837e77823 100644 --- a/BaseTools/Source/C/LzmaCompress/GNUmakefile +++ b/BaseTools/Source/C/LzmaCompress/GNUmakefile @@ -1,13 +1,15 @@ ## @file # GNU/Linux makefile for 'LzmaCompress' module build. # -# Copyright (c) 2009 - 2018, Intel Corporation. All rights reserved.
+# Copyright (c) 2009 - 2019, Intel Corporation. All rights reserved.
# SPDX-License-Identifier: BSD-2-Clause-Patent # MAKEROOT ?= .. APPNAME = LzmaCompress +LIBS = -lCommon + SDK_C = Sdk/C OBJECTS = \ diff --git a/BaseTools/Source/C/LzmaCompress/LzmaCompress.c b/BaseTools/Source/C/LzmaCompress/LzmaCompress.c index a3607f9b20..856fcf9ffb 100644 --- a/BaseTools/Source/C/LzmaCompress/LzmaCompress.c +++ b/BaseTools/Source/C/LzmaCompress/LzmaCompress.c @@ -5,7 +5,7 @@ LzmaUtil.c -- Test application for LZMA compression 2018-04-30 : Igor Pavlov : Public domain - Copyright (c) 2006 - 2018, Intel Corporation. All rights reserved.
+ Copyright (c) 2006 - 2019, Intel Corporation. All rights reserved.
SPDX-License-Identifier: BSD-2-Clause-Patent **/ @@ -23,6 +23,7 @@ #include "Sdk/C/LzmaEnc.h" #include "Sdk/C/Bra.h" #include "CommonLib.h" +#include "ParseInf.h" #define LZMA_HEADER_SIZE (LZMA_PROPS_SIZE + 8) @@ -36,10 +37,14 @@ const char *kCantReadMessage = "Can not read input file"; const char *kCantWriteMessage = "Can not write output file"; const char *kCantAllocateMessage = "Can not allocate memory"; const char *kDataErrorMessage = "Data error"; +const char *kInvalidParamValMessage = "Invalid parameter value"; static Bool mQuietMode = False; static CONVERTER_TYPE mConType = NoConverter; +UINT64 mDictionarySize = 31; +UINT64 mCompressionMode = 2; + #define UTILITY_NAME "LzmaCompress" #define UTILITY_MAJOR_VERSION 0 #define UTILITY_MINOR_VERSION 2 @@ -58,6 +63,8 @@ void PrintHelp(char *buffer) " -v, --verbose: increase output messages\n" " -q, --quiet: reduce output messages\n" " --debug [0-9]: set debug level\n" + " -a: set compression mode 0 = fast, 1 = normal, default: 1 (normal)\n" + " d: sets Dictionary size - [0, 30], default: 23 (8MB)\n" " --version: display the program version and exit\n" " -h, --help: display this help text\n" ); @@ -87,7 +94,7 @@ void PrintVersion(char *buffer) sprintf (buffer, "%s Version %d.%d %s ", UTILITY_NAME, UTILITY_MAJOR_VERSION, UTILITY_MINOR_VERSION, __BUILD_VERSION); } -static SRes Encode(ISeqOutStream *outStream, ISeqInStream *inStream, UInt64 fileSize) +static SRes Encode(ISeqOutStream *outStream, ISeqInStream *inStream, UInt64 fileSize, CLzmaEncProps *props) { SRes res; size_t inSize = (size_t)fileSize; @@ -95,10 +102,6 @@ static SRes Encode(ISeqOutStream *outStream, ISeqInStream *inStream, UInt64 file Byte *outBuffer = 0; Byte *filteredStream = 0; size_t outSize; - CLzmaEncProps props; - - LzmaEncProps_Init(&props); - LzmaEncProps_Normalize(&props); if (inSize != 0) { inBuffer = (Byte *)MyAlloc(inSize); @@ -151,7 +154,7 @@ static SRes Encode(ISeqOutStream *outStream, ISeqInStream *inStream, UInt64 file res = LzmaEncode(outBuffer + LZMA_HEADER_SIZE, &outSizeProcessed, mConType != NoConverter ? filteredStream : inBuffer, inSize, - &props, outBuffer, &outPropsSize, 0, + props, outBuffer, &outPropsSize, 0, NULL, &g_Alloc, &g_Alloc); if (res != SZ_OK) @@ -246,6 +249,10 @@ int main2(int numArgs, const char *args[], char *rs) const char *outputFile = "file.tmp"; int param; UInt64 fileSize; + CLzmaEncProps props; + + LzmaEncProps_Init(&props); + LzmaEncProps_Normalize(&props); FileSeqInStream_CreateVTable(&inStream); File_Construct(&inStream.file); @@ -280,6 +287,24 @@ int main2(int numArgs, const char *args[], char *rs) // parameter compatibility with other build tools. // param++; + } else if (strcmp(args[param], "-a") == 0) { + AsciiStringToUint64(args[param + 1],FALSE,&mCompressionMode); + if ((mCompressionMode == 0)||(mCompressionMode == 1)){ + props.algo = (int)mCompressionMode; + param++; + continue; + } else { + return PrintError(rs, kInvalidParamValMessage); + } + } else if (strcmp(args[param], "d") == 0) { + AsciiStringToUint64(args[param + 1],FALSE,&mDictionarySize); + if (mDictionarySize <= 30){ + props.dictSize = (UINT32)mDictionarySize; + param++; + continue; + } else { + return PrintError(rs, kInvalidParamValMessage); + } } else if ( strcmp(args[param], "-h") == 0 || strcmp(args[param], "--help") == 0 @@ -335,7 +360,7 @@ int main2(int numArgs, const char *args[], char *rs) if (!mQuietMode) { printf("Encoding\n"); } - res = Encode(&outStream.vt, &inStream.vt, fileSize); + res = Encode(&outStream.vt, &inStream.vt, fileSize, &props); } else { diff --git a/BaseTools/Source/C/LzmaCompress/Makefile b/BaseTools/Source/C/LzmaCompress/Makefile index 12be48de29..055f5d3ac3 100644 --- a/BaseTools/Source/C/LzmaCompress/Makefile +++ b/BaseTools/Source/C/LzmaCompress/Makefile @@ -1,14 +1,14 @@ ## @file # Windows makefile for 'LzmaCompress' module build. # -# Copyright (c) 2009 - 2016, Intel Corporation. All rights reserved.
+# Copyright (c) 2009 - 2019, Intel Corporation. All rights reserved.
# SPDX-License-Identifier: BSD-2-Clause-Patent # !INCLUDE ..\Makefiles\ms.common APPNAME = LzmaCompress -#LIBS = $(LIB_PATH)\Common.lib +LIBS = $(LIB_PATH)\Common.lib SDK_C = Sdk\C -- 2.39.2