mirror of
https://github.com/lz4/lz4.git
synced 2024-12-11 10:49:23 +08:00
629 lines
27 KiB
C
629 lines
27 KiB
C
/*
|
|
LZ4cli - LZ4 Command Line Interface
|
|
Copyright (C) Yann Collet 2011-2016
|
|
|
|
GPL v2 License
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License along
|
|
with this program; if not, write to the Free Software Foundation, Inc.,
|
|
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
|
|
You can contact the author at :
|
|
- LZ4 source repository : https://github.com/lz4/lz4
|
|
- LZ4 public forum : https://groups.google.com/forum/#!forum/lz4c
|
|
*/
|
|
/*
|
|
Note : this is stand-alone program.
|
|
It is not part of LZ4 compression library, it is a user program of the LZ4 library.
|
|
The license of LZ4 library is BSD.
|
|
The license of xxHash library is BSD.
|
|
The license of this compression CLI program is GPLv2.
|
|
*/
|
|
|
|
/**************************************
|
|
* Tuning parameters
|
|
***************************************/
|
|
/* ENABLE_LZ4C_LEGACY_OPTIONS :
|
|
Control the availability of -c0, -c1 and -hc legacy arguments
|
|
Default : Legacy options are disabled */
|
|
/* #define ENABLE_LZ4C_LEGACY_OPTIONS */
|
|
|
|
|
|
/**************************************
|
|
* Compiler Options
|
|
***************************************/
|
|
/* cf. http://man7.org/linux/man-pages/man7/feature_test_macros.7.html */
|
|
#define _XOPEN_VERSION 600 /* POSIX.2001, for fileno() within <stdio.h> on unix */
|
|
|
|
|
|
/****************************
|
|
* Includes
|
|
*****************************/
|
|
#include "util.h" /* Compiler options, UTIL_HAS_CREATEFILELIST */
|
|
#include <stdio.h> /* fprintf, getchar */
|
|
#include <stdlib.h> /* exit, calloc, free */
|
|
#include <string.h> /* strcmp, strlen */
|
|
#include "bench.h" /* BMK_benchFile, BMK_SetNbIterations, BMK_SetBlocksize, BMK_SetPause */
|
|
#include "lz4io.h" /* LZ4IO_compressFilename, LZ4IO_decompressFilename, LZ4IO_compressMultipleFilenames */
|
|
#include "lz4hc.h" /* LZ4HC_DEFAULT_CLEVEL */
|
|
#include "lz4.h" /* LZ4_VERSION_STRING */
|
|
|
|
|
|
/*-************************************
|
|
* OS-specific Includes
|
|
**************************************/
|
|
#if defined(_POSIX_SOURCE) || defined(_POSIX_C_SOURCE) || defined(_XOPEN_SOURCE) || (defined(__APPLE__) && defined(__MACH__)) || defined(__DJGPP__) /* https://sourceforge.net/p/predef/wiki/OperatingSystems/ */
|
|
# include <unistd.h> /* isatty */
|
|
# define IS_CONSOLE(stdStream) isatty(fileno(stdStream))
|
|
#elif defined(MSDOS) || defined(OS2) || defined(WIN32) || defined(_WIN32) || defined(__CYGWIN__)
|
|
# include <io.h> /* _isatty */
|
|
# define IS_CONSOLE(stdStream) _isatty(_fileno(stdStream))
|
|
#else
|
|
# define IS_CONSOLE(stdStream) 0
|
|
#endif
|
|
|
|
|
|
/*****************************
|
|
* Constants
|
|
******************************/
|
|
#define COMPRESSOR_NAME "LZ4 command line interface"
|
|
#define AUTHOR "Yann Collet"
|
|
#define WELCOME_MESSAGE "*** %s %i-bits v%s, by %s ***\n", COMPRESSOR_NAME, (int)(sizeof(void*)*8), LZ4_versionString(), AUTHOR
|
|
#define LZ4_EXTENSION ".lz4"
|
|
#define LZ4CAT "lz4cat"
|
|
#define UNLZ4 "unlz4"
|
|
|
|
#define KB *(1U<<10)
|
|
#define MB *(1U<<20)
|
|
#define GB *(1U<<30)
|
|
|
|
#define LZ4_BLOCKSIZEID_DEFAULT 7
|
|
|
|
|
|
/*-************************************
|
|
* Macros
|
|
***************************************/
|
|
#define DISPLAY(...) fprintf(stderr, __VA_ARGS__)
|
|
#define DISPLAYLEVEL(l, ...) if (displayLevel>=l) { DISPLAY(__VA_ARGS__); }
|
|
static unsigned displayLevel = 2; /* 0 : no display ; 1: errors only ; 2 : downgradable normal ; 3 : non-downgradable normal; 4 : + information */
|
|
|
|
|
|
/*-************************************
|
|
* Exceptions
|
|
***************************************/
|
|
#define DEBUG 0
|
|
#define DEBUGOUTPUT(...) if (DEBUG) DISPLAY(__VA_ARGS__);
|
|
#define EXM_THROW(error, ...) \
|
|
{ \
|
|
DEBUGOUTPUT("Error defined at %s, line %i : \n", __FILE__, __LINE__); \
|
|
DISPLAYLEVEL(1, "Error %i : ", error); \
|
|
DISPLAYLEVEL(1, __VA_ARGS__); \
|
|
DISPLAYLEVEL(1, "\n"); \
|
|
exit(error); \
|
|
}
|
|
|
|
|
|
/*-************************************
|
|
* Version modifiers
|
|
***************************************/
|
|
#define EXTENDED_ARGUMENTS
|
|
#define EXTENDED_HELP
|
|
#define EXTENDED_FORMAT
|
|
#define DEFAULT_COMPRESSOR LZ4IO_compressFilename
|
|
#define DEFAULT_DECOMPRESSOR LZ4IO_decompressFilename
|
|
int LZ4IO_compressFilename_Legacy(const char* input_filename, const char* output_filename, int compressionlevel); /* hidden function */
|
|
|
|
|
|
/*-***************************
|
|
* Functions
|
|
*****************************/
|
|
static int usage(const char* exeName)
|
|
{
|
|
DISPLAY( "Usage : \n");
|
|
DISPLAY( " %s [arg] [input] [output] \n", exeName);
|
|
DISPLAY( "\n");
|
|
DISPLAY( "input : a filename \n");
|
|
DISPLAY( " with no FILE, or when FILE is - or %s, read standard input\n", stdinmark);
|
|
DISPLAY( "Arguments : \n");
|
|
DISPLAY( " -1 : Fast compression (default) \n");
|
|
DISPLAY( " -9 : High compression \n");
|
|
DISPLAY( " -d : decompression (default for %s extension)\n", LZ4_EXTENSION);
|
|
DISPLAY( " -z : force compression \n");
|
|
DISPLAY( " -f : overwrite output without prompting \n");
|
|
DISPLAY( " -k : preserve source files(s) (default) \n");
|
|
DISPLAY( "--rm : remove source file(s) after successful de/compression \n");
|
|
DISPLAY( " -h/-H : display help/long help and exit \n");
|
|
return 0;
|
|
}
|
|
|
|
static int usage_advanced(const char* exeName)
|
|
{
|
|
DISPLAY(WELCOME_MESSAGE);
|
|
usage(exeName);
|
|
DISPLAY( "\n");
|
|
DISPLAY( "Advanced arguments :\n");
|
|
DISPLAY( " -V : display Version number and exit \n");
|
|
DISPLAY( " -v : verbose mode \n");
|
|
DISPLAY( " -q : suppress warnings; specify twice to suppress errors too\n");
|
|
DISPLAY( " -c : force write to standard output, even if it is the console\n");
|
|
DISPLAY( " -t : test compressed file integrity\n");
|
|
DISPLAY( " -m : multiple input files (implies automatic output filenames)\n");
|
|
#ifdef UTIL_HAS_CREATEFILELIST
|
|
DISPLAY( " -r : operate recursively on directories (sets also -m) \n");
|
|
#endif
|
|
DISPLAY( " -l : compress using Legacy format (Linux kernel compression)\n");
|
|
DISPLAY( " -B# : Block size [4-7] (default : 7) \n");
|
|
DISPLAY( " -BD : Block dependency (improve compression ratio) \n");
|
|
/* DISPLAY( " -BX : enable block checksum (default:disabled)\n"); *//* Option currently inactive */
|
|
DISPLAY( "--no-frame-crc : disable stream checksum (default:enabled) \n");
|
|
DISPLAY( "--content-size : compressed frame includes original size (default:not present)\n");
|
|
DISPLAY( "--[no-]sparse : sparse mode (default:enabled on file, disabled on stdout)\n");
|
|
DISPLAY( "Benchmark arguments : \n");
|
|
DISPLAY( " -b# : benchmark file(s), using # compression level (default : 1) \n");
|
|
DISPLAY( " -e# : test all compression levels from -bX to # (default : 1)\n");
|
|
DISPLAY( " -i# : minimum evaluation time in seconds (default : 3s) \n");
|
|
DISPLAY( " -B# : cut file into independent blocks of size # bytes [32+] \n");
|
|
DISPLAY( " or predefined block size [4-7] (default: 7) \n");
|
|
#if defined(ENABLE_LZ4C_LEGACY_OPTIONS)
|
|
DISPLAY( "Legacy arguments : \n");
|
|
DISPLAY( " -c0 : fast compression \n");
|
|
DISPLAY( " -c1 : high compression \n");
|
|
DISPLAY( " -hc : high compression \n");
|
|
DISPLAY( " -y : overwrite output without prompting \n");
|
|
#endif /* ENABLE_LZ4C_LEGACY_OPTIONS */
|
|
EXTENDED_HELP;
|
|
return 0;
|
|
}
|
|
|
|
static int usage_longhelp(const char* exeName)
|
|
{
|
|
usage_advanced(exeName);
|
|
DISPLAY( "\n");
|
|
DISPLAY( "****************************\n");
|
|
DISPLAY( "***** Advanced comment *****\n");
|
|
DISPLAY( "****************************\n");
|
|
DISPLAY( "\n");
|
|
DISPLAY( "Which values can [output] have ? \n");
|
|
DISPLAY( "---------------------------------\n");
|
|
DISPLAY( "[output] : a filename \n");
|
|
DISPLAY( " '%s', or '-' for standard output (pipe mode)\n", stdoutmark);
|
|
DISPLAY( " '%s' to discard output (test mode) \n", NULL_OUTPUT);
|
|
DISPLAY( "[output] can be left empty. In this case, it receives the following value :\n");
|
|
DISPLAY( " - if stdout is not the console, then [output] = stdout \n");
|
|
DISPLAY( " - if stdout is console : \n");
|
|
DISPLAY( " + for compression, output to filename%s \n", LZ4_EXTENSION);
|
|
DISPLAY( " + for decompression, output to filename without '%s'\n", LZ4_EXTENSION);
|
|
DISPLAY( " > if input filename has no '%s' extension : error \n", LZ4_EXTENSION);
|
|
DISPLAY( "\n");
|
|
DISPLAY( "Compression levels : \n");
|
|
DISPLAY( "---------------------\n");
|
|
DISPLAY( "-0 ... -2 => Fast compression, all identicals\n");
|
|
DISPLAY( "-3 ... -%d => High compression; higher number == more compression but slower\n", LZ4HC_MAX_CLEVEL);
|
|
DISPLAY( "\n");
|
|
DISPLAY( "stdin, stdout and the console : \n");
|
|
DISPLAY( "--------------------------------\n");
|
|
DISPLAY( "To protect the console from binary flooding (bad argument mistake)\n");
|
|
DISPLAY( "%s will refuse to read from console, or write to console \n", exeName);
|
|
DISPLAY( "except if '-c' command is specified, to force output to console \n");
|
|
DISPLAY( "\n");
|
|
DISPLAY( "Simple example :\n");
|
|
DISPLAY( "----------------\n");
|
|
DISPLAY( "1 : compress 'filename' fast, using default output name 'filename.lz4'\n");
|
|
DISPLAY( " %s filename\n", exeName);
|
|
DISPLAY( "\n");
|
|
DISPLAY( "Short arguments can be aggregated. For example :\n");
|
|
DISPLAY( "----------------------------------\n");
|
|
DISPLAY( "2 : compress 'filename' in high compression mode, overwrite output if exists\n");
|
|
DISPLAY( " %s -9 -f filename \n", exeName);
|
|
DISPLAY( " is equivalent to :\n");
|
|
DISPLAY( " %s -9f filename \n", exeName);
|
|
DISPLAY( "\n");
|
|
DISPLAY( "%s can be used in 'pure pipe mode'. For example :\n", exeName);
|
|
DISPLAY( "-------------------------------------\n");
|
|
DISPLAY( "3 : compress data stream from 'generator', send result to 'consumer'\n");
|
|
DISPLAY( " generator | %s | consumer \n", exeName);
|
|
#if defined(ENABLE_LZ4C_LEGACY_OPTIONS)
|
|
DISPLAY( "\n");
|
|
DISPLAY( "***** Warning *****\n");
|
|
DISPLAY( "Legacy arguments take precedence. Therefore : \n");
|
|
DISPLAY( "---------------------------------\n");
|
|
DISPLAY( " %s -hc filename\n", exeName);
|
|
DISPLAY( "means 'compress filename in high compression mode'\n");
|
|
DISPLAY( "It is not equivalent to :\n");
|
|
DISPLAY( " %s -h -c filename\n", exeName);
|
|
DISPLAY( "which would display help text and exit\n");
|
|
#endif /* ENABLE_LZ4C_LEGACY_OPTIONS */
|
|
return 0;
|
|
}
|
|
|
|
static int badusage(const char* exeName)
|
|
{
|
|
DISPLAYLEVEL(1, "Incorrect parameters\n");
|
|
if (displayLevel >= 1) usage(exeName);
|
|
exit(1);
|
|
}
|
|
|
|
|
|
static void waitEnter(void)
|
|
{
|
|
DISPLAY("Press enter to continue...\n");
|
|
(void)getchar();
|
|
}
|
|
|
|
|
|
/*! readU32FromChar() :
|
|
@return : unsigned integer value reach from input in `char` format
|
|
Will also modify `*stringPtr`, advancing it to position where it stopped reading.
|
|
Note : this function can overflow if result > MAX_UINT */
|
|
static unsigned readU32FromChar(const char** stringPtr)
|
|
{
|
|
unsigned result = 0;
|
|
while ((**stringPtr >='0') && (**stringPtr <='9'))
|
|
result *= 10, result += **stringPtr - '0', (*stringPtr)++ ;
|
|
return result;
|
|
}
|
|
|
|
typedef enum { om_auto, om_compress, om_decompress, om_test, om_bench } operationMode_e;
|
|
|
|
int main(int argc, const char** argv)
|
|
{
|
|
int i,
|
|
cLevel=1,
|
|
cLevelLast=1,
|
|
legacy_format=0,
|
|
forceStdout=0,
|
|
main_pause=0,
|
|
multiple_inputs=0,
|
|
operationResult=0;
|
|
operationMode_e mode = om_auto;
|
|
const char* input_filename = NULL;
|
|
const char* output_filename= NULL;
|
|
char* dynNameSpace = NULL;
|
|
const char** inFileNames = (const char**) calloc(argc, sizeof(char*));
|
|
unsigned ifnIdx=0;
|
|
const char nullOutput[] = NULL_OUTPUT;
|
|
const char extension[] = LZ4_EXTENSION;
|
|
size_t blockSize = LZ4IO_setBlockSizeID(LZ4_BLOCKSIZEID_DEFAULT);
|
|
const char* const exeName = argv[0];
|
|
#ifdef UTIL_HAS_CREATEFILELIST
|
|
const char** extendedFileList = NULL;
|
|
char* fileNamesBuf = NULL;
|
|
unsigned fileNamesNb, recursive=0;
|
|
#endif
|
|
|
|
/* Init */
|
|
if (inFileNames==NULL) {
|
|
DISPLAY("Allocation error : not enough memory \n");
|
|
return 1;
|
|
}
|
|
LZ4IO_setOverwrite(0);
|
|
|
|
/* lz4cat predefined behavior */
|
|
if (!strcmp(exeName, LZ4CAT)) {
|
|
mode = om_decompress;
|
|
LZ4IO_setOverwrite(1);
|
|
forceStdout=1;
|
|
output_filename=stdoutmark;
|
|
displayLevel=1;
|
|
multiple_inputs=1;
|
|
}
|
|
if (!strcmp(exeName, UNLZ4)) { mode = om_decompress; }
|
|
|
|
/* command switches */
|
|
for(i=1; i<argc; i++) {
|
|
const char* argument = argv[i];
|
|
|
|
if(!argument) continue; /* Protection if argument empty */
|
|
|
|
/* Short commands (note : aggregated short commands are allowed) */
|
|
if (argument[0]=='-') {
|
|
/* '-' means stdin/stdout */
|
|
if (argument[1]==0) {
|
|
if (!input_filename) input_filename=stdinmark;
|
|
else output_filename=stdoutmark;
|
|
continue;
|
|
}
|
|
|
|
/* long commands (--long-word) */
|
|
if (argument[1]=='-') {
|
|
if (!strcmp(argument, "--compress")) { mode = om_compress; continue; }
|
|
if ((!strcmp(argument, "--decompress"))
|
|
|| (!strcmp(argument, "--uncompress"))) { mode = om_decompress; continue; }
|
|
if (!strcmp(argument, "--multiple")) { multiple_inputs = 1; continue; }
|
|
if (!strcmp(argument, "--test")) { mode = om_test; continue; }
|
|
if (!strcmp(argument, "--force")) { LZ4IO_setOverwrite(1); continue; }
|
|
if (!strcmp(argument, "--no-force")) { LZ4IO_setOverwrite(0); continue; }
|
|
if ((!strcmp(argument, "--stdout"))
|
|
|| (!strcmp(argument, "--to-stdout"))) { forceStdout=1; output_filename=stdoutmark; continue; }
|
|
if (!strcmp(argument, "--frame-crc")) { LZ4IO_setStreamChecksumMode(1); continue; }
|
|
if (!strcmp(argument, "--no-frame-crc")) { LZ4IO_setStreamChecksumMode(0); continue; }
|
|
if (!strcmp(argument, "--content-size")) { LZ4IO_setContentSize(1); continue; }
|
|
if (!strcmp(argument, "--no-content-size")) { LZ4IO_setContentSize(0); continue; }
|
|
if (!strcmp(argument, "--sparse")) { LZ4IO_setSparseFile(2); continue; }
|
|
if (!strcmp(argument, "--no-sparse")) { LZ4IO_setSparseFile(0); continue; }
|
|
if (!strcmp(argument, "--verbose")) { displayLevel++; continue; }
|
|
if (!strcmp(argument, "--quiet")) { if (displayLevel) displayLevel--; continue; }
|
|
if (!strcmp(argument, "--version")) { DISPLAY(WELCOME_MESSAGE); return 0; }
|
|
if (!strcmp(argument, "--help")) { usage_advanced(exeName); goto _cleanup; }
|
|
if (!strcmp(argument, "--keep")) { LZ4IO_setRemoveSrcFile(0); continue; } /* keep source file (default) */
|
|
if (!strcmp(argument, "--rm")) { LZ4IO_setRemoveSrcFile(1); continue; }
|
|
}
|
|
|
|
while (argument[1]!=0) {
|
|
argument ++;
|
|
|
|
#if defined(ENABLE_LZ4C_LEGACY_OPTIONS)
|
|
/* Legacy arguments (-c0, -c1, -hc, -y, -s) */
|
|
if ((argument[0]=='c') && (argument[1]=='0')) { cLevel=0; argument++; continue; } /* -c0 (fast compression) */
|
|
if ((argument[0]=='c') && (argument[1]=='1')) { cLevel=9; argument++; continue; } /* -c1 (high compression) */
|
|
if ((argument[0]=='h') && (argument[1]=='c')) { cLevel=9; argument++; continue; } /* -hc (high compression) */
|
|
if (*argument=='y') { LZ4IO_setOverwrite(1); continue; } /* -y (answer 'yes' to overwrite permission) */
|
|
#endif /* ENABLE_LZ4C_LEGACY_OPTIONS */
|
|
|
|
if ((*argument>='0') && (*argument<='9')) {
|
|
cLevel = readU32FromChar(&argument);
|
|
argument--;
|
|
continue;
|
|
}
|
|
|
|
|
|
switch(argument[0])
|
|
{
|
|
/* Display help */
|
|
case 'V': DISPLAY(WELCOME_MESSAGE); goto _cleanup; /* Version */
|
|
case 'h': usage_advanced(exeName); goto _cleanup;
|
|
case 'H': usage_longhelp(exeName); goto _cleanup;
|
|
|
|
case 'e':
|
|
argument++;
|
|
cLevelLast = readU32FromChar(&argument);
|
|
argument--;
|
|
break;
|
|
|
|
/* Compression (default) */
|
|
case 'z': mode = om_compress; break;
|
|
|
|
/* Use Legacy format (ex : Linux kernel compression) */
|
|
case 'l': legacy_format = 1; blockSize = 8 MB; break;
|
|
|
|
/* Decoding */
|
|
case 'd': mode = om_decompress; break;
|
|
|
|
/* Force stdout, even if stdout==console */
|
|
case 'c': forceStdout=1; output_filename=stdoutmark; break;
|
|
|
|
/* Test integrity */
|
|
case 't': mode = om_test; break;
|
|
|
|
/* Overwrite */
|
|
case 'f': LZ4IO_setOverwrite(1); break;
|
|
|
|
/* Verbose mode */
|
|
case 'v': displayLevel++; break;
|
|
|
|
/* Quiet mode */
|
|
case 'q': if (displayLevel) displayLevel--; break;
|
|
|
|
/* keep source file (default anyway, so useless) (for xz/lzma compatibility) */
|
|
case 'k': LZ4IO_setRemoveSrcFile(0); break;
|
|
|
|
/* Modify Block Properties */
|
|
case 'B':
|
|
while (argument[1]!=0) {
|
|
int exitBlockProperties=0;
|
|
switch(argument[1])
|
|
{
|
|
case 'D': LZ4IO_setBlockMode(LZ4IO_blockLinked); argument++; break;
|
|
case 'X': LZ4IO_setBlockChecksumMode(1); argument ++; break; /* disabled by default */
|
|
default :
|
|
if (argument[1] < '0' || argument[1] > '9') {
|
|
exitBlockProperties=1;
|
|
break;
|
|
} else {
|
|
unsigned B;
|
|
argument++;
|
|
B = readU32FromChar(&argument);
|
|
argument--;
|
|
if (B < 4) badusage(exeName);
|
|
if (B <= 7) {
|
|
blockSize = LZ4IO_setBlockSizeID(B);
|
|
BMK_SetBlockSize(blockSize);
|
|
DISPLAYLEVEL(2, "using blocks of size %u KB \n", (U32)(blockSize>>10));
|
|
} else {
|
|
if (B < 32) badusage(exeName);
|
|
BMK_SetBlockSize(B);
|
|
if (B >= 1024) {
|
|
DISPLAYLEVEL(2, "bench: using blocks of size %u KB \n", (U32)(B>>10));
|
|
} else {
|
|
DISPLAYLEVEL(2, "bench: using blocks of size %u bytes \n", (U32)(B));
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
if (exitBlockProperties) break;
|
|
}
|
|
break;
|
|
|
|
/* Benchmark */
|
|
case 'b': mode = om_bench; multiple_inputs=1;
|
|
break;
|
|
|
|
#ifdef UTIL_HAS_CREATEFILELIST
|
|
/* recursive */
|
|
case 'r': recursive=1; /* without break */
|
|
#endif
|
|
/* Treat non-option args as input files. See https://code.google.com/p/lz4/issues/detail?id=151 */
|
|
case 'm': multiple_inputs=1;
|
|
break;
|
|
|
|
/* Modify Nb Seconds (benchmark only) */
|
|
case 'i':
|
|
{ unsigned iters;
|
|
argument++;
|
|
iters = readU32FromChar(&argument);
|
|
argument--;
|
|
BMK_setNotificationLevel(displayLevel);
|
|
BMK_SetNbSeconds(iters);
|
|
}
|
|
break;
|
|
|
|
/* Pause at the end (hidden option) */
|
|
case 'p': main_pause=1; break;
|
|
|
|
/* Specific commands for customized versions */
|
|
EXTENDED_ARGUMENTS;
|
|
|
|
/* Unrecognised command */
|
|
default : badusage(exeName);
|
|
}
|
|
}
|
|
continue;
|
|
}
|
|
|
|
/* Store in *inFileNames[] if -m is used. */
|
|
if (multiple_inputs) { inFileNames[ifnIdx++]=argument; continue; }
|
|
|
|
/* Store first non-option arg in input_filename to preserve original cli logic. */
|
|
if (!input_filename) { input_filename=argument; continue; }
|
|
|
|
/* Second non-option arg in output_filename to preserve original cli logic. */
|
|
if (!output_filename) {
|
|
output_filename=argument;
|
|
if (!strcmp (output_filename, nullOutput)) output_filename = nulmark;
|
|
continue;
|
|
}
|
|
|
|
/* 3rd non-option arg should not exist */
|
|
DISPLAYLEVEL(1, "Warning : %s won't be used ! Do you want multiple input files (-m) ? \n", argument);
|
|
}
|
|
|
|
DISPLAYLEVEL(3, WELCOME_MESSAGE);
|
|
if ((mode == om_compress) || (mode == om_bench)) DISPLAYLEVEL(4, "Blocks size : %i KB\n", (U32)(blockSize>>10));
|
|
|
|
if (multiple_inputs) {
|
|
input_filename = inFileNames[0];
|
|
#ifdef UTIL_HAS_CREATEFILELIST
|
|
if (recursive) { /* at this stage, filenameTable is a list of paths, which can contain both files and directories */
|
|
extendedFileList = UTIL_createFileList(inFileNames, ifnIdx, &fileNamesBuf, &fileNamesNb);
|
|
if (extendedFileList) {
|
|
unsigned u;
|
|
for (u=0; u<fileNamesNb; u++) DISPLAYLEVEL(4, "%u %s\n", u, extendedFileList[u]);
|
|
free((void*)inFileNames);
|
|
inFileNames = extendedFileList;
|
|
ifnIdx = fileNamesNb;
|
|
}
|
|
}
|
|
#endif
|
|
}
|
|
|
|
/* benchmark and test modes */
|
|
if (mode == om_bench) {
|
|
BMK_setNotificationLevel(displayLevel);
|
|
operationResult = BMK_benchFiles(inFileNames, ifnIdx, cLevel, cLevelLast);
|
|
goto _cleanup;
|
|
}
|
|
|
|
if (mode == om_test) {
|
|
LZ4IO_setTestMode(1);
|
|
output_filename = nulmark;
|
|
mode = om_decompress; /* defer to decompress */
|
|
}
|
|
|
|
/* compress or decompress */
|
|
if (!input_filename) input_filename = stdinmark;
|
|
/* Check if input is defined as console; trigger an error in this case */
|
|
if (!strcmp(input_filename, stdinmark) && IS_CONSOLE(stdin) ) {
|
|
DISPLAYLEVEL(1, "refusing to read from a console\n");
|
|
exit(1);
|
|
}
|
|
|
|
/* No output filename ==> try to select one automatically (when possible) */
|
|
while ((!output_filename) && (multiple_inputs==0)) {
|
|
if (!IS_CONSOLE(stdout)) { output_filename=stdoutmark; break; } /* Default to stdout whenever possible (i.e. not a console) */
|
|
if (mode == om_auto) { /* auto-determine compression or decompression, based on file extension */
|
|
size_t const inSize = strlen(input_filename);
|
|
size_t const extSize = strlen(LZ4_EXTENSION);
|
|
size_t const extStart= (inSize > extSize) ? inSize-extSize : 0;
|
|
if (!strcmp(input_filename+extStart, LZ4_EXTENSION)) mode = om_decompress;
|
|
else mode = om_compress;
|
|
}
|
|
if (mode == om_compress) { /* compression to file */
|
|
size_t const l = strlen(input_filename);
|
|
dynNameSpace = (char*)calloc(1,l+5);
|
|
if (dynNameSpace==NULL) { perror(exeName); exit(1); }
|
|
strcpy(dynNameSpace, input_filename);
|
|
strcat(dynNameSpace, LZ4_EXTENSION);
|
|
output_filename = dynNameSpace;
|
|
DISPLAYLEVEL(2, "Compressed filename will be : %s \n", output_filename);
|
|
break;
|
|
}
|
|
if (mode == om_decompress) {/* decompression to file (automatic name will work only if input filename has correct format extension) */
|
|
size_t outl;
|
|
size_t const inl = strlen(input_filename);
|
|
dynNameSpace = (char*)calloc(1,inl+1);
|
|
if (dynNameSpace==NULL) { perror(exeName); exit(1); }
|
|
strcpy(dynNameSpace, input_filename);
|
|
outl = inl;
|
|
if (inl>4)
|
|
while ((outl >= inl-4) && (input_filename[outl] == extension[outl-inl+4])) dynNameSpace[outl--]=0;
|
|
if (outl != inl-5) { DISPLAYLEVEL(1, "Cannot determine an output filename\n"); badusage(exeName); }
|
|
output_filename = dynNameSpace;
|
|
DISPLAYLEVEL(2, "Decoding file %s \n", output_filename);
|
|
}
|
|
break;
|
|
}
|
|
|
|
if (!output_filename) output_filename = "*\\dummy^!//";
|
|
|
|
/* Check if output is defined as console; trigger an error in this case */
|
|
if (!strcmp(output_filename,stdoutmark) && IS_CONSOLE(stdout) && !forceStdout) {
|
|
DISPLAYLEVEL(1, "refusing to write to console without -c\n");
|
|
exit(1);
|
|
}
|
|
|
|
/* Downgrade notification level in stdout and multiple file mode */
|
|
if (!strcmp(output_filename,stdoutmark) && (displayLevel==2)) displayLevel=1;
|
|
if ((multiple_inputs) && (displayLevel==2)) displayLevel=1;
|
|
|
|
/* IO Stream/File */
|
|
LZ4IO_setNotificationLevel(displayLevel);
|
|
if (mode == om_decompress) {
|
|
if (multiple_inputs)
|
|
operationResult = LZ4IO_decompressMultipleFilenames(inFileNames, ifnIdx, !strcmp(output_filename,stdoutmark) ? stdoutmark : LZ4_EXTENSION);
|
|
else
|
|
operationResult = DEFAULT_DECOMPRESSOR(input_filename, output_filename);
|
|
} else { /* compression is default action */
|
|
if (legacy_format) {
|
|
DISPLAYLEVEL(3, "! Generating compressed LZ4 using Legacy format (deprecated) ! \n");
|
|
LZ4IO_compressFilename_Legacy(input_filename, output_filename, cLevel);
|
|
} else {
|
|
if (multiple_inputs)
|
|
operationResult = LZ4IO_compressMultipleFilenames(inFileNames, ifnIdx, LZ4_EXTENSION, cLevel);
|
|
else
|
|
operationResult = DEFAULT_COMPRESSOR(input_filename, output_filename, cLevel);
|
|
}
|
|
}
|
|
|
|
_cleanup:
|
|
if (main_pause) waitEnter();
|
|
if (dynNameSpace) free(dynNameSpace);
|
|
#ifdef UTIL_HAS_CREATEFILELIST
|
|
if (extendedFileList)
|
|
UTIL_freeFileList(extendedFileList, fileNamesBuf);
|
|
else
|
|
#endif
|
|
free((void*)inFileNames);
|
|
return operationResult;
|
|
}
|