| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752 |
- /*
- LZ4io.c - LZ4 File/Stream Interface
- Copyright (C) Yann Collet 2011-2020
- 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 code of the LZ4 library.
- - The license of LZ4 library is BSD.
- - The license of xxHash library is BSD.
- - The license of this source file is GPLv2.
- */
- /*-************************************
- * Compiler options
- **************************************/
- #ifdef _MSC_VER /* Visual Studio */
- # pragma warning(disable : 4127) /* disable: C4127: conditional expression is constant */
- #endif
- #if defined(__MINGW32__) && !defined(_POSIX_SOURCE)
- # define _POSIX_SOURCE 1 /* disable %llu warnings with MinGW on Windows */
- #endif
- /*****************************
- * Includes
- *****************************/
- #include "platform.h" /* Large File Support, SET_BINARY_MODE, SET_SPARSE_FILE_MODE, PLATFORM_POSIX_VERSION, __64BIT__ */
- #include "util.h" /* UTIL_getFileStat, UTIL_setFileStat */
- #include <stdio.h> /* fprintf, fopen, fread, stdin, stdout, fflush, getchar */
- #include <stdlib.h> /* malloc, free */
- #include <string.h> /* strerror, strcmp, strlen */
- #include <time.h> /* clock */
- #include <sys/types.h> /* stat64 */
- #include <sys/stat.h> /* stat64 */
- #include "lz4.h" /* still required for legacy format */
- #include "lz4hc.h" /* still required for legacy format */
- #define LZ4F_STATIC_LINKING_ONLY
- #include "lz4frame.h"
- #include "lz4io.h"
- /*****************************
- * Constants
- *****************************/
- #define KB *(1 <<10)
- #define MB *(1 <<20)
- #define GB *(1U<<30)
- #define _1BIT 0x01
- #define _2BITS 0x03
- #define _3BITS 0x07
- #define _4BITS 0x0F
- #define _8BITS 0xFF
- #define MAGICNUMBER_SIZE 4
- #define LZ4IO_MAGICNUMBER 0x184D2204
- #define LZ4IO_SKIPPABLE0 0x184D2A50
- #define LZ4IO_SKIPPABLEMASK 0xFFFFFFF0
- #define LEGACY_MAGICNUMBER 0x184C2102
- #define CACHELINE 64
- #define LEGACY_BLOCKSIZE (8 MB)
- #define MIN_STREAM_BUFSIZE (192 KB)
- #define LZ4IO_BLOCKSIZEID_DEFAULT 7
- #define LZ4_MAX_DICT_SIZE (64 KB)
- /**************************************
- * Macros
- **************************************/
- #define DISPLAY(...) fprintf(stderr, __VA_ARGS__)
- #define DISPLAYOUT(...) fprintf(stdout, __VA_ARGS__)
- #define DISPLAYLEVEL(l, ...) if (g_displayLevel>=l) { DISPLAY(__VA_ARGS__); }
- static int g_displayLevel = 0; /* 0 : no display ; 1: errors ; 2 : + result + interaction + warnings ; 3 : + progression; 4 : + information */
- #define DISPLAYUPDATE(l, ...) if (g_displayLevel>=l) { \
- if ( ((clock() - g_time) > refreshRate) \
- || (g_displayLevel>=4) ) { \
- g_time = clock(); \
- DISPLAY(__VA_ARGS__); \
- if (g_displayLevel>=4) fflush(stderr); \
- } }
- static const clock_t refreshRate = CLOCKS_PER_SEC / 6;
- static clock_t g_time = 0;
- #define LZ4IO_STATIC_ASSERT(c) { enum { LZ4IO_static_assert = 1/(int)(!!(c)) }; } /* use after variable declarations */
- /**************************************
- * Exceptions
- ***************************************/
- #ifndef DEBUG
- # define DEBUG 0
- #endif
- #define DEBUGOUTPUT(...) if (DEBUG) DISPLAY(__VA_ARGS__);
- #define END_PROCESS(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); \
- }
- /* ************************************************** */
- /* ****************** Parameters ******************** */
- /* ************************************************** */
- struct LZ4IO_prefs_s {
- int passThrough;
- int overwrite;
- int testMode;
- int blockSizeId;
- size_t blockSize;
- int blockChecksum;
- int streamChecksum;
- int blockIndependence;
- int sparseFileSupport;
- int contentSizeFlag;
- int useDictionary;
- unsigned favorDecSpeed;
- const char* dictionaryFilename;
- int removeSrcFile;
- };
- LZ4IO_prefs_t* LZ4IO_defaultPreferences(void)
- {
- LZ4IO_prefs_t* const ret = (LZ4IO_prefs_t*)malloc(sizeof(*ret));
- if (!ret) END_PROCESS(21, "Allocation error : not enough memory");
- ret->passThrough = 0;
- ret->overwrite = 1;
- ret->testMode = 0;
- ret->blockSizeId = LZ4IO_BLOCKSIZEID_DEFAULT;
- ret->blockSize = 0;
- ret->blockChecksum = 0;
- ret->streamChecksum = 1;
- ret->blockIndependence = 1;
- ret->sparseFileSupport = 1;
- ret->contentSizeFlag = 0;
- ret->useDictionary = 0;
- ret->favorDecSpeed = 0;
- ret->dictionaryFilename = NULL;
- ret->removeSrcFile = 0;
- return ret;
- }
- void LZ4IO_freePreferences(LZ4IO_prefs_t* prefs)
- {
- free(prefs);
- }
- int LZ4IO_setDictionaryFilename(LZ4IO_prefs_t* const prefs, const char* dictionaryFilename)
- {
- prefs->dictionaryFilename = dictionaryFilename;
- prefs->useDictionary = dictionaryFilename != NULL;
- return prefs->useDictionary;
- }
- /* Default setting : passThrough = 0; return : passThrough mode (0/1) */
- int LZ4IO_setPassThrough(LZ4IO_prefs_t* const prefs, int yes)
- {
- prefs->passThrough = (yes!=0);
- return prefs->passThrough;
- }
- /* Default setting : overwrite = 1; return : overwrite mode (0/1) */
- int LZ4IO_setOverwrite(LZ4IO_prefs_t* const prefs, int yes)
- {
- prefs->overwrite = (yes!=0);
- return prefs->overwrite;
- }
- /* Default setting : testMode = 0; return : testMode (0/1) */
- int LZ4IO_setTestMode(LZ4IO_prefs_t* const prefs, int yes)
- {
- prefs->testMode = (yes!=0);
- return prefs->testMode;
- }
- /* blockSizeID : valid values : 4-5-6-7 */
- size_t LZ4IO_setBlockSizeID(LZ4IO_prefs_t* const prefs, unsigned bsid)
- {
- static const size_t blockSizeTable[] = { 64 KB, 256 KB, 1 MB, 4 MB };
- static const unsigned minBlockSizeID = 4;
- static const unsigned maxBlockSizeID = 7;
- if ((bsid < minBlockSizeID) || (bsid > maxBlockSizeID)) return 0;
- prefs->blockSizeId = (int)bsid;
- prefs->blockSize = blockSizeTable[(unsigned)prefs->blockSizeId-minBlockSizeID];
- return prefs->blockSize;
- }
- size_t LZ4IO_setBlockSize(LZ4IO_prefs_t* const prefs, size_t blockSize)
- {
- static const size_t minBlockSize = 32;
- static const size_t maxBlockSize = 4 MB;
- unsigned bsid = 0;
- if (blockSize < minBlockSize) blockSize = minBlockSize;
- if (blockSize > maxBlockSize) blockSize = maxBlockSize;
- prefs->blockSize = blockSize;
- blockSize--;
- /* find which of { 64k, 256k, 1MB, 4MB } is closest to blockSize */
- while (blockSize >>= 2)
- bsid++;
- if (bsid < 7) bsid = 7;
- prefs->blockSizeId = (int)(bsid-3);
- return prefs->blockSize;
- }
- /* Default setting : 1 == independent blocks */
- int LZ4IO_setBlockMode(LZ4IO_prefs_t* const prefs, LZ4IO_blockMode_t blockMode)
- {
- prefs->blockIndependence = (blockMode == LZ4IO_blockIndependent);
- return prefs->blockIndependence;
- }
- /* Default setting : 0 == no block checksum */
- int LZ4IO_setBlockChecksumMode(LZ4IO_prefs_t* const prefs, int enable)
- {
- prefs->blockChecksum = (enable != 0);
- return prefs->blockChecksum;
- }
- /* Default setting : 1 == checksum enabled */
- int LZ4IO_setStreamChecksumMode(LZ4IO_prefs_t* const prefs, int enable)
- {
- prefs->streamChecksum = (enable != 0);
- return prefs->streamChecksum;
- }
- /* Default setting : 0 (no notification) */
- int LZ4IO_setNotificationLevel(int level)
- {
- g_displayLevel = level;
- return g_displayLevel;
- }
- /* Default setting : 1 (auto: enabled on file, disabled on stdout) */
- int LZ4IO_setSparseFile(LZ4IO_prefs_t* const prefs, int enable)
- {
- prefs->sparseFileSupport = 2*(enable!=0); /* 2==force enable */
- return prefs->sparseFileSupport;
- }
- /* Default setting : 0 (disabled) */
- int LZ4IO_setContentSize(LZ4IO_prefs_t* const prefs, int enable)
- {
- prefs->contentSizeFlag = (enable!=0);
- return prefs->contentSizeFlag;
- }
- /* Default setting : 0 (disabled) */
- void LZ4IO_favorDecSpeed(LZ4IO_prefs_t* const prefs, int favor)
- {
- prefs->favorDecSpeed = (favor!=0);
- }
- void LZ4IO_setRemoveSrcFile(LZ4IO_prefs_t* const prefs, unsigned flag)
- {
- prefs->removeSrcFile = (flag>0);
- }
- /* ************************************************************************ **
- ** ********************** String functions ********************* **
- ** ************************************************************************ */
- static int LZ4IO_isDevNull(const char* s)
- {
- return UTIL_sameString(s, nulmark);
- }
- static int LZ4IO_isStdin(const char* s)
- {
- return UTIL_sameString(s, stdinmark);
- }
- static int LZ4IO_isStdout(const char* s)
- {
- return UTIL_sameString(s, stdoutmark);
- }
- /* ************************************************************************ **
- ** ********************** LZ4 File / Pipe compression ********************* **
- ** ************************************************************************ */
- static int LZ4IO_isSkippableMagicNumber(unsigned int magic) {
- return (magic & LZ4IO_SKIPPABLEMASK) == LZ4IO_SKIPPABLE0;
- }
- /** LZ4IO_openSrcFile() :
- * condition : `srcFileName` must be non-NULL.
- * @result : FILE* to `dstFileName`, or NULL if it fails */
- static FILE* LZ4IO_openSrcFile(const char* srcFileName)
- {
- FILE* f;
- if (LZ4IO_isStdin(srcFileName)) {
- DISPLAYLEVEL(4,"Using stdin for input \n");
- f = stdin;
- SET_BINARY_MODE(stdin);
- } else {
- f = fopen(srcFileName, "rb");
- if (f==NULL) DISPLAYLEVEL(1, "%s: %s \n", srcFileName, strerror(errno));
- }
- return f;
- }
- /** FIO_openDstFile() :
- * prefs is writable, because sparseFileSupport might be updated.
- * condition : `dstFileName` must be non-NULL.
- * @result : FILE* to `dstFileName`, or NULL if it fails */
- static FILE* LZ4IO_openDstFile(const char* dstFileName, const LZ4IO_prefs_t* const prefs)
- {
- FILE* f;
- assert(dstFileName != NULL);
- if (LZ4IO_isStdout(dstFileName)) {
- DISPLAYLEVEL(4, "Using stdout for output \n");
- f = stdout;
- SET_BINARY_MODE(stdout);
- if (prefs->sparseFileSupport==1) {
- DISPLAYLEVEL(4, "Sparse File Support automatically disabled on stdout ;"
- " to force-enable it, add --sparse command \n");
- }
- } else {
- if (!prefs->overwrite && !LZ4IO_isDevNull(dstFileName)) {
- /* Check if destination file already exists */
- FILE* const testf = fopen( dstFileName, "rb" );
- if (testf != NULL) { /* dest exists, prompt for overwrite authorization */
- fclose(testf);
- if (g_displayLevel <= 1) { /* No interaction possible */
- DISPLAY("%s already exists; not overwritten \n", dstFileName);
- return NULL;
- }
- DISPLAY("%s already exists; do you want to overwrite (y/N) ? ", dstFileName);
- { int ch = getchar();
- if ((ch!='Y') && (ch!='y')) {
- DISPLAY(" not overwritten \n");
- return NULL;
- }
- while ((ch!=EOF) && (ch!='\n')) ch = getchar(); /* flush rest of input line */
- } } }
- f = fopen( dstFileName, "wb" );
- if (f==NULL) DISPLAYLEVEL(1, "%s: %s\n", dstFileName, strerror(errno));
- }
- /* sparse file */
- { int const sparseMode = (prefs->sparseFileSupport - (f==stdout)) > 0;
- if (f && sparseMode) { SET_SPARSE_FILE_MODE(f); }
- }
- return f;
- }
- /***************************************
- * Legacy Compression
- ***************************************/
- /* Size in bytes of a legacy block header in little-endian format */
- #define LZ4IO_LEGACY_BLOCK_HEADER_SIZE 4
- #define LZ4IO_LEGACY_BLOCK_SIZE_MAX (8 MB)
- /* unoptimized version; solves endianness & alignment issues */
- static void LZ4IO_writeLE32 (void* p, unsigned value32)
- {
- unsigned char* const dstPtr = (unsigned char*)p;
- dstPtr[0] = (unsigned char)value32;
- dstPtr[1] = (unsigned char)(value32 >> 8);
- dstPtr[2] = (unsigned char)(value32 >> 16);
- dstPtr[3] = (unsigned char)(value32 >> 24);
- }
- static int LZ4IO_LZ4_compress(const char* src, char* dst, int srcSize, int dstSize, int cLevel)
- {
- (void)cLevel;
- return LZ4_compress_fast(src, dst, srcSize, dstSize, 1);
- }
- /* LZ4IO_compressFilename_Legacy :
- * This function is intentionally "hidden" (not published in .h)
- * It generates compressed streams using the old 'legacy' format */
- int LZ4IO_compressFilename_Legacy(const char* input_filename, const char* output_filename,
- int compressionlevel, const LZ4IO_prefs_t* prefs)
- {
- typedef int (*compress_f)(const char* src, char* dst, int srcSize, int dstSize, int cLevel);
- compress_f const compressionFunction = (compressionlevel < 3) ? LZ4IO_LZ4_compress : LZ4_compress_HC;
- unsigned long long filesize = 0;
- unsigned long long compressedfilesize = MAGICNUMBER_SIZE;
- char* in_buff;
- char* out_buff;
- const int outBuffSize = LZ4_compressBound(LEGACY_BLOCKSIZE);
- FILE* const finput = LZ4IO_openSrcFile(input_filename);
- FILE* foutput;
- clock_t clockEnd;
- /* Init */
- clock_t const clockStart = clock();
- if (finput == NULL)
- END_PROCESS(20, "%s : open file error ", input_filename);
- foutput = LZ4IO_openDstFile(output_filename, prefs);
- if (foutput == NULL) {
- fclose(finput);
- END_PROCESS(20, "%s : open file error ", input_filename);
- }
- /* Allocate Memory */
- in_buff = (char*)malloc(LEGACY_BLOCKSIZE);
- out_buff = (char*)malloc((size_t)outBuffSize + 4);
- if (!in_buff || !out_buff)
- END_PROCESS(21, "Allocation error : not enough memory");
- /* Write Archive Header */
- LZ4IO_writeLE32(out_buff, LEGACY_MAGICNUMBER);
- if (fwrite(out_buff, 1, MAGICNUMBER_SIZE, foutput) != MAGICNUMBER_SIZE)
- END_PROCESS(22, "Write error : cannot write header");
- /* Main Loop */
- while (1) {
- int outSize;
- /* Read Block */
- size_t const inSize = fread(in_buff, (size_t)1, (size_t)LEGACY_BLOCKSIZE, finput);
- if (inSize == 0) break;
- assert(inSize <= LEGACY_BLOCKSIZE);
- filesize += inSize;
- /* Compress Block */
- outSize = compressionFunction(in_buff, out_buff+4, (int)inSize, outBuffSize, compressionlevel);
- assert(outSize >= 0);
- compressedfilesize += (unsigned long long)outSize+4;
- DISPLAYUPDATE(2, "\rRead : %i MiB ==> %.2f%% ",
- (int)(filesize>>20), (double)compressedfilesize/filesize*100);
- /* Write Block */
- assert(outSize > 0);
- assert(outSize < outBuffSize);
- LZ4IO_writeLE32(out_buff, (unsigned)outSize);
- if (fwrite(out_buff, 1, (size_t)outSize+4, foutput) != (size_t)(outSize+4)) {
- END_PROCESS(24, "Write error : cannot write compressed block");
- } }
- if (ferror(finput)) END_PROCESS(24, "Error while reading %s ", input_filename);
- /* Status */
- clockEnd = clock();
- clockEnd += (clockEnd==clockStart); /* avoid division by zero (speed) */
- filesize += !filesize; /* avoid division by zero (ratio) */
- DISPLAYLEVEL(2, "\r%79s\r", ""); /* blank line */
- DISPLAYLEVEL(2,"Compressed %llu bytes into %llu bytes ==> %.2f%%\n",
- filesize, compressedfilesize, (double)compressedfilesize / filesize * 100);
- { double const seconds = (double)(clockEnd - clockStart) / CLOCKS_PER_SEC;
- DISPLAYLEVEL(4,"Done in %.2f s ==> %.2f MiB/s\n", seconds,
- (double)filesize / seconds / 1024 / 1024);
- }
- /* Close & Free */
- free(in_buff);
- free(out_buff);
- fclose(finput);
- if (!LZ4IO_isStdout(output_filename)) fclose(foutput); /* do not close stdout */
- return 0;
- }
- #define FNSPACE 30
- /* LZ4IO_compressMultipleFilenames_Legacy :
- * This function is intentionally "hidden" (not published in .h)
- * It generates multiple compressed streams using the old 'legacy' format */
- int LZ4IO_compressMultipleFilenames_Legacy(
- const char** inFileNamesTable, int ifntSize,
- const char* suffix,
- int compressionLevel, const LZ4IO_prefs_t* prefs)
- {
- int i;
- int missed_files = 0;
- char* dstFileName = (char*)malloc(FNSPACE);
- size_t ofnSize = FNSPACE;
- const size_t suffixSize = strlen(suffix);
- if (dstFileName == NULL) return ifntSize; /* not enough memory */
- /* loop on each file */
- for (i=0; i<ifntSize; i++) {
- size_t const ifnSize = strlen(inFileNamesTable[i]);
- if (LZ4IO_isStdout(suffix)) {
- missed_files += LZ4IO_compressFilename_Legacy(
- inFileNamesTable[i], stdoutmark,
- compressionLevel, prefs);
- continue;
- }
- if (ofnSize <= ifnSize+suffixSize+1) {
- free(dstFileName);
- ofnSize = ifnSize + 20;
- dstFileName = (char*)malloc(ofnSize);
- if (dstFileName==NULL) {
- return ifntSize;
- } }
- strcpy(dstFileName, inFileNamesTable[i]);
- strcat(dstFileName, suffix);
- missed_files += LZ4IO_compressFilename_Legacy(
- inFileNamesTable[i], dstFileName,
- compressionLevel, prefs);
- }
- /* Close & Free */
- free(dstFileName);
- return missed_files;
- }
- /*********************************************
- * Compression using Frame format
- *********************************************/
- typedef struct {
- void* srcBuffer;
- size_t srcBufferSize;
- void* dstBuffer;
- size_t dstBufferSize;
- LZ4F_compressionContext_t ctx;
- LZ4F_CDict* cdict;
- } cRess_t;
- static void* LZ4IO_createDict(size_t* dictSize, const char* const dictFilename)
- {
- size_t readSize;
- size_t dictEnd = 0;
- size_t dictLen = 0;
- size_t dictStart;
- size_t circularBufSize = LZ4_MAX_DICT_SIZE;
- char* circularBuf = (char*)malloc(circularBufSize);
- char* dictBuf;
- FILE* dictFile;
- if (!circularBuf) END_PROCESS(25, "Allocation error : not enough memory for circular buffer");
- if (!dictFilename) END_PROCESS(26, "Dictionary error : no filename provided");
- dictFile = LZ4IO_openSrcFile(dictFilename);
- if (!dictFile) END_PROCESS(27, "Dictionary error : could not open dictionary file");
- /* opportunistically seek to the part of the file we care about.
- * If this fails it's not a problem since we'll just read everything anyways. */
- if (!LZ4IO_isStdin(dictFilename)) {
- (void)UTIL_fseek(dictFile, -LZ4_MAX_DICT_SIZE, SEEK_END);
- }
- do {
- readSize = fread(circularBuf + dictEnd, 1, circularBufSize - dictEnd, dictFile);
- dictEnd = (dictEnd + readSize) % circularBufSize;
- dictLen += readSize;
- } while (readSize>0);
- if (dictLen > LZ4_MAX_DICT_SIZE) {
- dictLen = LZ4_MAX_DICT_SIZE;
- }
- *dictSize = dictLen;
- dictStart = (circularBufSize + dictEnd - dictLen) % circularBufSize;
- if (dictStart == 0) {
- /* We're in the simple case where the dict starts at the beginning of our circular buffer. */
- dictBuf = circularBuf;
- circularBuf = NULL;
- } else {
- /* Otherwise, we will alloc a new buffer and copy our dict into that. */
- dictBuf = (char *)malloc(dictLen ? dictLen : 1);
- if (!dictBuf) END_PROCESS(28, "Allocation error : not enough memory");
- memcpy(dictBuf, circularBuf + dictStart, circularBufSize - dictStart);
- memcpy(dictBuf + circularBufSize - dictStart, circularBuf, dictLen - (circularBufSize - dictStart));
- }
- fclose(dictFile);
- free(circularBuf);
- return dictBuf;
- }
- static LZ4F_CDict* LZ4IO_createCDict(const LZ4IO_prefs_t* const prefs)
- {
- size_t dictionarySize;
- void* dictionaryBuffer;
- LZ4F_CDict* cdict;
- if (!prefs->useDictionary) return NULL;
- dictionaryBuffer = LZ4IO_createDict(&dictionarySize, prefs->dictionaryFilename);
- if (!dictionaryBuffer) END_PROCESS(29, "Dictionary error : could not create dictionary");
- cdict = LZ4F_createCDict(dictionaryBuffer, dictionarySize);
- free(dictionaryBuffer);
- return cdict;
- }
- static cRess_t LZ4IO_createCResources(const LZ4IO_prefs_t* const prefs)
- {
- const size_t blockSize = prefs->blockSize;
- cRess_t ress;
- LZ4F_errorCode_t const errorCode = LZ4F_createCompressionContext(&(ress.ctx), LZ4F_VERSION);
- if (LZ4F_isError(errorCode)) END_PROCESS(30, "Allocation error : can't create LZ4F context : %s", LZ4F_getErrorName(errorCode));
- /* Allocate Memory */
- ress.srcBuffer = malloc(blockSize);
- ress.srcBufferSize = blockSize;
- ress.dstBufferSize = LZ4F_compressFrameBound(blockSize, NULL); /* cover worst case */
- ress.dstBuffer = malloc(ress.dstBufferSize);
- if (!ress.srcBuffer || !ress.dstBuffer) END_PROCESS(31, "Allocation error : not enough memory");
- ress.cdict = LZ4IO_createCDict(prefs);
- return ress;
- }
- static void LZ4IO_freeCResources(cRess_t ress)
- {
- free(ress.srcBuffer);
- free(ress.dstBuffer);
- LZ4F_freeCDict(ress.cdict);
- ress.cdict = NULL;
- { LZ4F_errorCode_t const errorCode = LZ4F_freeCompressionContext(ress.ctx);
- if (LZ4F_isError(errorCode)) END_PROCESS(35, "Error : can't free LZ4F context resource : %s", LZ4F_getErrorName(errorCode)); }
- }
- /*
- * LZ4IO_compressFilename_extRess()
- * result : 0 : compression completed correctly
- * 1 : missing or pb opening srcFileName
- */
- static int
- LZ4IO_compressFilename_extRess(cRess_t ress,
- const char* srcFileName, const char* dstFileName,
- int compressionLevel, const LZ4IO_prefs_t* const io_prefs)
- {
- unsigned long long filesize = 0;
- unsigned long long compressedfilesize = 0;
- FILE* dstFile;
- void* const srcBuffer = ress.srcBuffer;
- void* const dstBuffer = ress.dstBuffer;
- const size_t dstBufferSize = ress.dstBufferSize;
- const size_t blockSize = io_prefs->blockSize;
- size_t readSize;
- LZ4F_compressionContext_t ctx = ress.ctx; /* just a pointer */
- LZ4F_preferences_t prefs;
- /* Init */
- FILE* const srcFile = LZ4IO_openSrcFile(srcFileName);
- if (srcFile == NULL) return 1;
- dstFile = LZ4IO_openDstFile(dstFileName, io_prefs);
- if (dstFile == NULL) { fclose(srcFile); return 1; }
- memset(&prefs, 0, sizeof(prefs));
- /* Set compression parameters */
- prefs.autoFlush = 1;
- prefs.compressionLevel = compressionLevel;
- prefs.frameInfo.blockMode = (LZ4F_blockMode_t)io_prefs->blockIndependence;
- prefs.frameInfo.blockSizeID = (LZ4F_blockSizeID_t)io_prefs->blockSizeId;
- prefs.frameInfo.blockChecksumFlag = (LZ4F_blockChecksum_t)io_prefs->blockChecksum;
- prefs.frameInfo.contentChecksumFlag = (LZ4F_contentChecksum_t)io_prefs->streamChecksum;
- prefs.favorDecSpeed = io_prefs->favorDecSpeed;
- if (io_prefs->contentSizeFlag) {
- U64 const fileSize = UTIL_getOpenFileSize(srcFile);
- prefs.frameInfo.contentSize = fileSize; /* == 0 if input == stdin */
- if (fileSize==0)
- DISPLAYLEVEL(3, "Warning : cannot determine input content size \n");
- }
- /* read first block */
- readSize = fread(srcBuffer, (size_t)1, blockSize, srcFile);
- if (ferror(srcFile)) END_PROCESS(40, "Error reading %s ", srcFileName);
- filesize += readSize;
- /* single-block file */
- if (readSize < blockSize) {
- /* Compress in single pass */
- size_t const cSize = LZ4F_compressFrame_usingCDict(ctx, dstBuffer, dstBufferSize, srcBuffer, readSize, ress.cdict, &prefs);
- if (LZ4F_isError(cSize))
- END_PROCESS(41, "Compression failed : %s", LZ4F_getErrorName(cSize));
- compressedfilesize = cSize;
- DISPLAYUPDATE(2, "\rRead : %u MiB ==> %.2f%% ",
- (unsigned)(filesize>>20), (double)compressedfilesize/(filesize+!filesize)*100); /* avoid division by zero */
- /* Write Block */
- if (fwrite(dstBuffer, 1, cSize, dstFile) != cSize) {
- END_PROCESS(42, "Write error : failed writing single-block compressed frame");
- } }
- else
- /* multiple-blocks file */
- {
- /* Write Frame Header */
- size_t const headerSize = LZ4F_compressBegin_usingCDict(ctx, dstBuffer, dstBufferSize, ress.cdict, &prefs);
- if (LZ4F_isError(headerSize)) END_PROCESS(43, "File header generation failed : %s", LZ4F_getErrorName(headerSize));
- if (fwrite(dstBuffer, 1, headerSize, dstFile) != headerSize)
- END_PROCESS(44, "Write error : cannot write header");
- compressedfilesize += headerSize;
- /* Main Loop - one block at a time */
- while (readSize>0) {
- size_t const outSize = LZ4F_compressUpdate(ctx, dstBuffer, dstBufferSize, srcBuffer, readSize, NULL);
- if (LZ4F_isError(outSize))
- END_PROCESS(45, "Compression failed : %s", LZ4F_getErrorName(outSize));
- compressedfilesize += outSize;
- DISPLAYUPDATE(2, "\rRead : %u MiB ==> %.2f%% ",
- (unsigned)(filesize>>20), (double)compressedfilesize/filesize*100);
- /* Write Block */
- if (fwrite(dstBuffer, 1, outSize, dstFile) != outSize)
- END_PROCESS(46, "Write error : cannot write compressed block");
- /* Read next block */
- readSize = fread(srcBuffer, (size_t)1, (size_t)blockSize, srcFile);
- filesize += readSize;
- }
- if (ferror(srcFile)) END_PROCESS(47, "Error reading %s ", srcFileName);
- /* End of Frame mark */
- { size_t const endSize = LZ4F_compressEnd(ctx, dstBuffer, dstBufferSize, NULL);
- if (LZ4F_isError(endSize))
- END_PROCESS(48, "End of frame error : %s", LZ4F_getErrorName(endSize));
- if (fwrite(dstBuffer, 1, endSize, dstFile) != endSize)
- END_PROCESS(49, "Write error : cannot write end of frame");
- compressedfilesize += endSize;
- } }
- /* Release file handlers */
- fclose (srcFile);
- if (!LZ4IO_isStdout(dstFileName)) fclose(dstFile); /* do not close stdout */
- /* Copy owner, file permissions and modification time */
- { stat_t statbuf;
- if (!LZ4IO_isStdin(srcFileName)
- && !LZ4IO_isStdout(dstFileName)
- && !LZ4IO_isDevNull(dstFileName)
- && UTIL_getFileStat(srcFileName, &statbuf)) {
- UTIL_setFileStat(dstFileName, &statbuf);
- } }
- if (io_prefs->removeSrcFile) { /* remove source file : --rm */
- if (remove(srcFileName))
- END_PROCESS(50, "Remove error : %s: %s", srcFileName, strerror(errno));
- }
- /* Final Status */
- DISPLAYLEVEL(2, "\r%79s\r", "");
- DISPLAYLEVEL(2, "Compressed %llu bytes into %llu bytes ==> %.2f%%\n",
- filesize, compressedfilesize,
- (double)compressedfilesize / (filesize + !filesize /* avoid division by zero */ ) * 100);
- return 0;
- }
- int LZ4IO_compressFilename(const char* srcFileName, const char* dstFileName, int compressionLevel, const LZ4IO_prefs_t* prefs)
- {
- UTIL_time_t const timeStart = UTIL_getTime();
- clock_t const cpuStart = clock();
- cRess_t const ress = LZ4IO_createCResources(prefs);
- int const result = LZ4IO_compressFilename_extRess(ress, srcFileName, dstFileName, compressionLevel, prefs);
- /* Free resources */
- LZ4IO_freeCResources(ress);
- /* Final Status */
- { clock_t const cpuEnd = clock();
- double const cpuLoad_s = (double)(cpuEnd - cpuStart) / CLOCKS_PER_SEC;
- U64 const timeLength_ns = UTIL_clockSpanNano(timeStart);
- double const timeLength_s = (double)timeLength_ns / 1000000000;
- DISPLAYLEVEL(4, "Completed in %.2f sec (cpu load : %.0f%%)\n",
- timeLength_s, (cpuLoad_s / timeLength_s) * 100);
- }
- return result;
- }
- int LZ4IO_compressMultipleFilenames(
- const char** inFileNamesTable, int ifntSize,
- const char* suffix,
- int compressionLevel,
- const LZ4IO_prefs_t* prefs)
- {
- int i;
- int missed_files = 0;
- char* dstFileName = (char*)malloc(FNSPACE);
- size_t ofnSize = FNSPACE;
- const size_t suffixSize = strlen(suffix);
- cRess_t ress;
- if (dstFileName == NULL) return ifntSize; /* not enough memory */
- ress = LZ4IO_createCResources(prefs);
- /* loop on each file */
- for (i=0; i<ifntSize; i++) {
- size_t const ifnSize = strlen(inFileNamesTable[i]);
- if (LZ4IO_isStdout(suffix)) {
- missed_files += LZ4IO_compressFilename_extRess(ress,
- inFileNamesTable[i], stdoutmark,
- compressionLevel, prefs);
- continue;
- }
- /* suffix != stdout => compress into a file => generate its name */
- if (ofnSize <= ifnSize+suffixSize+1) {
- free(dstFileName);
- ofnSize = ifnSize + 20;
- dstFileName = (char*)malloc(ofnSize);
- if (dstFileName==NULL) {
- LZ4IO_freeCResources(ress);
- return ifntSize;
- } }
- strcpy(dstFileName, inFileNamesTable[i]);
- strcat(dstFileName, suffix);
- missed_files += LZ4IO_compressFilename_extRess(ress,
- inFileNamesTable[i], dstFileName,
- compressionLevel, prefs);
- }
- /* Close & Free */
- LZ4IO_freeCResources(ress);
- free(dstFileName);
- return missed_files;
- }
- /* ********************************************************************* */
- /* ********************** LZ4 file-stream Decompression **************** */
- /* ********************************************************************* */
- /* It's presumed that s points to a memory space of size >= 4 */
- static unsigned LZ4IO_readLE32 (const void* s)
- {
- const unsigned char* const srcPtr = (const unsigned char*)s;
- unsigned value32 = srcPtr[0];
- value32 += (unsigned)srcPtr[1] << 8;
- value32 += (unsigned)srcPtr[2] << 16;
- value32 += (unsigned)srcPtr[3] << 24;
- return value32;
- }
- static unsigned
- LZ4IO_fwriteSparse(FILE* file,
- const void* buffer, size_t bufferSize,
- int sparseFileSupport,
- unsigned storedSkips)
- {
- const size_t sizeT = sizeof(size_t);
- const size_t maskT = sizeT -1 ;
- const size_t* const bufferT = (const size_t*)buffer; /* Buffer is supposed malloc'ed, hence aligned on size_t */
- const size_t* ptrT = bufferT;
- size_t bufferSizeT = bufferSize / sizeT;
- const size_t* const bufferTEnd = bufferT + bufferSizeT;
- const size_t segmentSizeT = (32 KB) / sizeT;
- int const sparseMode = (sparseFileSupport - (file==stdout)) > 0;
- if (!sparseMode) { /* normal write */
- size_t const sizeCheck = fwrite(buffer, 1, bufferSize, file);
- if (sizeCheck != bufferSize) END_PROCESS(70, "Write error : cannot write decoded block");
- return 0;
- }
- /* avoid int overflow */
- if (storedSkips > 1 GB) {
- int const seekResult = UTIL_fseek(file, 1 GB, SEEK_CUR);
- if (seekResult != 0) END_PROCESS(71, "1 GB skip error (sparse file support)");
- storedSkips -= 1 GB;
- }
- while (ptrT < bufferTEnd) {
- size_t seg0SizeT = segmentSizeT;
- size_t nb0T;
- /* count leading zeros */
- if (seg0SizeT > bufferSizeT) seg0SizeT = bufferSizeT;
- bufferSizeT -= seg0SizeT;
- for (nb0T=0; (nb0T < seg0SizeT) && (ptrT[nb0T] == 0); nb0T++) ;
- storedSkips += (unsigned)(nb0T * sizeT);
- if (nb0T != seg0SizeT) { /* not all 0s */
- errno = 0;
- { int const seekResult = UTIL_fseek(file, storedSkips, SEEK_CUR);
- if (seekResult) END_PROCESS(72, "Sparse skip error(%d): %s ; try --no-sparse", (int)errno, strerror(errno));
- }
- storedSkips = 0;
- seg0SizeT -= nb0T;
- ptrT += nb0T;
- { size_t const sizeCheck = fwrite(ptrT, sizeT, seg0SizeT, file);
- if (sizeCheck != seg0SizeT) END_PROCESS(73, "Write error : cannot write decoded block");
- } }
- ptrT += seg0SizeT;
- }
- if (bufferSize & maskT) { /* size not multiple of sizeT : implies end of block */
- const char* const restStart = (const char*)bufferTEnd;
- const char* restPtr = restStart;
- size_t const restSize = bufferSize & maskT;
- const char* const restEnd = restStart + restSize;
- for (; (restPtr < restEnd) && (*restPtr == 0); restPtr++) ;
- storedSkips += (unsigned) (restPtr - restStart);
- if (restPtr != restEnd) {
- int const seekResult = UTIL_fseek(file, storedSkips, SEEK_CUR);
- if (seekResult) END_PROCESS(74, "Sparse skip error ; try --no-sparse");
- storedSkips = 0;
- { size_t const sizeCheck = fwrite(restPtr, 1, (size_t)(restEnd - restPtr), file);
- if (sizeCheck != (size_t)(restEnd - restPtr)) END_PROCESS(75, "Write error : cannot write decoded end of block");
- } }
- }
- return storedSkips;
- }
- static void LZ4IO_fwriteSparseEnd(FILE* file, unsigned storedSkips)
- {
- if (storedSkips>0) { /* implies sparseFileSupport>0 */
- const char lastZeroByte[1] = { 0 };
- if (UTIL_fseek(file, storedSkips-1, SEEK_CUR) != 0)
- END_PROCESS(68, "Final skip error (sparse file)\n");
- if (fwrite(lastZeroByte, 1, 1, file) != 1)
- END_PROCESS(69, "Write error : cannot write last zero\n");
- }
- }
- static unsigned g_magicRead = 0; /* out-parameter of LZ4IO_decodeLegacyStream() */
- static unsigned long long
- LZ4IO_decodeLegacyStream(FILE* finput, FILE* foutput, const LZ4IO_prefs_t* prefs)
- {
- unsigned long long streamSize = 0;
- unsigned storedSkips = 0;
- /* Allocate Memory */
- char* const in_buff = (char*)malloc((size_t)LZ4_compressBound(LEGACY_BLOCKSIZE));
- char* const out_buff = (char*)malloc(LEGACY_BLOCKSIZE);
- if (!in_buff || !out_buff) END_PROCESS(51, "Allocation error : not enough memory");
- /* Main Loop */
- while (1) {
- unsigned int blockSize;
- /* Block Size */
- { size_t const sizeCheck = fread(in_buff, 1, LZ4IO_LEGACY_BLOCK_HEADER_SIZE, finput);
- if (sizeCheck == 0) break; /* Nothing to read : file read is completed */
- if (sizeCheck != LZ4IO_LEGACY_BLOCK_HEADER_SIZE) END_PROCESS(52, "Read error : cannot access block size ");
- }
- blockSize = LZ4IO_readLE32(in_buff); /* Convert to Little Endian */
- if (blockSize > LZ4_COMPRESSBOUND(LEGACY_BLOCKSIZE)) {
- /* Cannot read next block : maybe new stream ? */
- g_magicRead = blockSize;
- break;
- }
- /* Read Block */
- { size_t const sizeCheck = fread(in_buff, 1, blockSize, finput);
- if (sizeCheck != blockSize) END_PROCESS(53, "Read error : cannot access compressed block !"); }
- /* Decode Block */
- { int const decodeSize = LZ4_decompress_safe(in_buff, out_buff, (int)blockSize, LEGACY_BLOCKSIZE);
- if (decodeSize < 0) END_PROCESS(54, "Decoding Failed ! Corrupted input detected !");
- streamSize += (unsigned long long)decodeSize;
- /* Write Block */
- storedSkips = LZ4IO_fwriteSparse(foutput, out_buff, (size_t)decodeSize, prefs->sparseFileSupport, storedSkips); /* success or die */
- } }
- if (ferror(finput)) END_PROCESS(55, "Read error : ferror");
- LZ4IO_fwriteSparseEnd(foutput, storedSkips);
- /* Free */
- free(in_buff);
- free(out_buff);
- return streamSize;
- }
- typedef struct {
- void* srcBuffer;
- size_t srcBufferSize;
- void* dstBuffer;
- size_t dstBufferSize;
- FILE* dstFile;
- LZ4F_decompressionContext_t dCtx;
- void* dictBuffer;
- size_t dictBufferSize;
- } dRess_t;
- static void LZ4IO_loadDDict(dRess_t* ress, const LZ4IO_prefs_t* const prefs)
- {
- if (!prefs->useDictionary) {
- ress->dictBuffer = NULL;
- ress->dictBufferSize = 0;
- return;
- }
- ress->dictBuffer = LZ4IO_createDict(&ress->dictBufferSize, prefs->dictionaryFilename);
- if (!ress->dictBuffer) END_PROCESS(25, "Dictionary error : could not create dictionary");
- }
- static const size_t LZ4IO_dBufferSize = 64 KB;
- static dRess_t LZ4IO_createDResources(const LZ4IO_prefs_t* const prefs)
- {
- dRess_t ress;
- /* init */
- LZ4F_errorCode_t const errorCode = LZ4F_createDecompressionContext(&ress.dCtx, LZ4F_VERSION);
- if (LZ4F_isError(errorCode)) END_PROCESS(60, "Can't create LZ4F context : %s", LZ4F_getErrorName(errorCode));
- /* Allocate Memory */
- ress.srcBufferSize = LZ4IO_dBufferSize;
- ress.srcBuffer = malloc(ress.srcBufferSize);
- ress.dstBufferSize = LZ4IO_dBufferSize;
- ress.dstBuffer = malloc(ress.dstBufferSize);
- if (!ress.srcBuffer || !ress.dstBuffer) END_PROCESS(61, "Allocation error : not enough memory");
- LZ4IO_loadDDict(&ress, prefs);
- ress.dstFile = NULL;
- return ress;
- }
- static void LZ4IO_freeDResources(dRess_t ress)
- {
- LZ4F_errorCode_t errorCode = LZ4F_freeDecompressionContext(ress.dCtx);
- if (LZ4F_isError(errorCode)) END_PROCESS(69, "Error : can't free LZ4F context resource : %s", LZ4F_getErrorName(errorCode));
- free(ress.srcBuffer);
- free(ress.dstBuffer);
- free(ress.dictBuffer);
- }
- static unsigned long long
- LZ4IO_decompressLZ4F(dRess_t ress,
- FILE* const srcFile, FILE* const dstFile,
- const LZ4IO_prefs_t* const prefs)
- {
- unsigned long long filesize = 0;
- LZ4F_errorCode_t nextToLoad;
- unsigned storedSkips = 0;
- LZ4F_decompressOptions_t const dOpt_skipCrc = { 0, 1, 0, 0 };
- const LZ4F_decompressOptions_t* const dOptPtr =
- ((prefs->blockChecksum==0) && (prefs->streamChecksum==0)) ?
- &dOpt_skipCrc : NULL;
- /* Init feed with magic number (already consumed from FILE* sFile) */
- { size_t inSize = MAGICNUMBER_SIZE;
- size_t outSize= 0;
- LZ4IO_writeLE32(ress.srcBuffer, LZ4IO_MAGICNUMBER);
- nextToLoad = LZ4F_decompress_usingDict(ress.dCtx,
- ress.dstBuffer, &outSize,
- ress.srcBuffer, &inSize,
- ress.dictBuffer, ress.dictBufferSize,
- dOptPtr); /* set it once, it's enough */
- if (LZ4F_isError(nextToLoad))
- END_PROCESS(62, "Header error : %s", LZ4F_getErrorName(nextToLoad));
- }
- /* Main Loop */
- for (;nextToLoad;) {
- size_t readSize;
- size_t pos = 0;
- size_t decodedBytes = ress.dstBufferSize;
- /* Read input */
- if (nextToLoad > ress.srcBufferSize) nextToLoad = ress.srcBufferSize;
- readSize = fread(ress.srcBuffer, 1, nextToLoad, srcFile);
- if (!readSize) break; /* reached end of file or stream */
- while ((pos < readSize) || (decodedBytes == ress.dstBufferSize)) { /* still to read, or still to flush */
- /* Decode Input (at least partially) */
- size_t remaining = readSize - pos;
- decodedBytes = ress.dstBufferSize;
- nextToLoad = LZ4F_decompress_usingDict(ress.dCtx,
- ress.dstBuffer, &decodedBytes,
- (char*)(ress.srcBuffer)+pos, &remaining,
- ress.dictBuffer, ress.dictBufferSize,
- NULL);
- if (LZ4F_isError(nextToLoad))
- END_PROCESS(66, "Decompression error : %s", LZ4F_getErrorName(nextToLoad));
- pos += remaining;
- /* Write Block */
- if (decodedBytes) {
- if (!prefs->testMode)
- storedSkips = LZ4IO_fwriteSparse(dstFile, ress.dstBuffer, decodedBytes, prefs->sparseFileSupport, storedSkips);
- filesize += decodedBytes;
- DISPLAYUPDATE(2, "\rDecompressed : %u MiB ", (unsigned)(filesize>>20));
- }
- if (!nextToLoad) break;
- }
- }
- /* can be out because readSize == 0, which could be an fread() error */
- if (ferror(srcFile)) END_PROCESS(67, "Read error");
- if (!prefs->testMode) LZ4IO_fwriteSparseEnd(dstFile, storedSkips);
- if (nextToLoad!=0) END_PROCESS(68, "Unfinished stream");
- return filesize;
- }
- /* LZ4IO_passThrough:
- * just output the same content as input, no decoding.
- * This is a capability of zcat, and by extension lz4cat
- * MNstore : contain the first MAGICNUMBER_SIZE bytes already read from finput
- */
- #define PTSIZE (64 KB)
- #define PTSIZET (PTSIZE / sizeof(size_t))
- static unsigned long long
- LZ4IO_passThrough(FILE* finput, FILE* foutput,
- unsigned char MNstore[MAGICNUMBER_SIZE],
- int sparseFileSupport)
- {
- size_t buffer[PTSIZET];
- size_t readBytes = 1;
- unsigned long long total = MAGICNUMBER_SIZE;
- unsigned storedSkips = 0;
- if (fwrite(MNstore, 1, MAGICNUMBER_SIZE, foutput) != MAGICNUMBER_SIZE) {
- END_PROCESS(50, "Pass-through write error");
- }
- while (readBytes) {
- readBytes = fread(buffer, 1, sizeof(buffer), finput);
- total += readBytes;
- storedSkips = LZ4IO_fwriteSparse(foutput, buffer, readBytes, sparseFileSupport, storedSkips);
- }
- if (ferror(finput)) END_PROCESS(51, "Read Error");
- LZ4IO_fwriteSparseEnd(foutput, storedSkips);
- return total;
- }
- /* when fseek() doesn't work (pipe scenario),
- * read and forget from input.
- **/
- #define SKIP_BUFF_SIZE (16 KB)
- #define MIN(a,b) ( ((a)<(b)) ? (a) : (b) )
- static int skipStream(FILE* f, unsigned offset)
- {
- char buf[SKIP_BUFF_SIZE];
- while (offset > 0) {
- size_t const tr = MIN(offset, sizeof(buf));
- size_t const r = fread(buf, 1, tr, f);
- if (r != tr) return 1; /* error reading f */
- offset -= (unsigned)tr;
- }
- assert(offset == 0);
- return 0;
- }
- /** Safely handle cases when (unsigned)offset > LONG_MAX */
- static int fseek_u32(FILE *fp, unsigned offset, int where)
- {
- const unsigned stepMax = 1U << 30;
- int errorNb = 0;
- if (where != SEEK_CUR) return -1; /* Only allows SEEK_CUR */
- while (offset > 0) {
- unsigned s = offset;
- if (s > stepMax) s = stepMax;
- errorNb = UTIL_fseek(fp, (long)s, SEEK_CUR);
- if (errorNb==0) { offset -= s; continue; }
- errorNb = skipStream(fp, offset);
- offset = 0;
- }
- return errorNb;
- }
- #define ENDOFSTREAM ((unsigned long long)-1)
- #define DECODING_ERROR ((unsigned long long)-2)
- static unsigned long long
- selectDecoder(dRess_t ress,
- FILE* finput, FILE* foutput,
- const LZ4IO_prefs_t* const prefs)
- {
- unsigned char MNstore[MAGICNUMBER_SIZE];
- unsigned magicNumber;
- static unsigned nbFrames = 0;
- /* init */
- nbFrames++;
- /* Check Archive Header */
- if (g_magicRead) { /* magic number already read from finput (see legacy frame)*/
- magicNumber = g_magicRead;
- g_magicRead = 0;
- } else {
- size_t const nbReadBytes = fread(MNstore, 1, MAGICNUMBER_SIZE, finput);
- if (nbReadBytes==0) { nbFrames = 0; return ENDOFSTREAM; } /* EOF */
- if (nbReadBytes != MAGICNUMBER_SIZE)
- END_PROCESS(40, "Unrecognized header : Magic Number unreadable");
- magicNumber = LZ4IO_readLE32(MNstore); /* Little Endian format */
- }
- if (LZ4IO_isSkippableMagicNumber(magicNumber))
- magicNumber = LZ4IO_SKIPPABLE0; /* fold skippable magic numbers */
- switch(magicNumber)
- {
- case LZ4IO_MAGICNUMBER:
- return LZ4IO_decompressLZ4F(ress, finput, foutput, prefs);
- case LEGACY_MAGICNUMBER:
- DISPLAYLEVEL(4, "Detected : Legacy format \n");
- return LZ4IO_decodeLegacyStream(finput, foutput, prefs);
- case LZ4IO_SKIPPABLE0:
- DISPLAYLEVEL(4, "Skipping detected skippable area \n");
- { size_t const nbReadBytes = fread(MNstore, 1, 4, finput);
- if (nbReadBytes != 4)
- END_PROCESS(42, "Stream error : skippable size unreadable");
- }
- { unsigned const size = LZ4IO_readLE32(MNstore);
- int const errorNb = fseek_u32(finput, size, SEEK_CUR);
- if (errorNb != 0)
- END_PROCESS(43, "Stream error : cannot skip skippable area");
- }
- return 0;
- default:
- if (nbFrames == 1) { /* just started */
- /* Wrong magic number at the beginning of 1st stream */
- if (!prefs->testMode && prefs->overwrite && prefs->passThrough) {
- nbFrames = 0;
- return LZ4IO_passThrough(finput, foutput, MNstore, prefs->sparseFileSupport);
- }
- END_PROCESS(44,"Unrecognized header : file cannot be decoded");
- }
- { long int const position = ftell(finput); /* only works for files < 2 GB */
- DISPLAYLEVEL(2, "Stream followed by undecodable data ");
- if (position != -1L)
- DISPLAYLEVEL(2, "at position %i ", (int)position);
- DISPLAYLEVEL(2, "\n");
- }
- return DECODING_ERROR;
- }
- }
- static int
- LZ4IO_decompressSrcFile(dRess_t ress,
- const char* input_filename, const char* output_filename,
- const LZ4IO_prefs_t* const prefs)
- {
- FILE* const foutput = ress.dstFile;
- unsigned long long filesize = 0;
- int result = 0;
- /* Init */
- FILE* const finput = LZ4IO_openSrcFile(input_filename);
- if (finput==NULL) return 1;
- assert(foutput != NULL);
- /* Loop over multiple streams */
- for ( ; ; ) { /* endless loop, see break condition */
- unsigned long long const decodedSize =
- selectDecoder(ress, finput, foutput, prefs);
- if (decodedSize == ENDOFSTREAM) break;
- if (decodedSize == DECODING_ERROR) { result=1; break; }
- filesize += decodedSize;
- }
- /* Close input */
- fclose(finput);
- if (prefs->removeSrcFile) { /* --rm */
- if (remove(input_filename))
- END_PROCESS(45, "Remove error : %s: %s", input_filename, strerror(errno));
- }
- /* Final Status */
- DISPLAYLEVEL(2, "\r%79s\r", "");
- DISPLAYLEVEL(2, "%-20.20s : decoded %llu bytes \n", input_filename, filesize);
- (void)output_filename;
- return result;
- }
- static int
- LZ4IO_decompressDstFile(dRess_t ress,
- const char* input_filename, const char* output_filename,
- const LZ4IO_prefs_t* const prefs)
- {
- int result;
- stat_t statbuf;
- int stat_result = 0;
- FILE* const foutput = LZ4IO_openDstFile(output_filename, prefs);
- if (foutput==NULL) return 1; /* failure */
- if ( !LZ4IO_isStdin(input_filename)
- && UTIL_getFileStat(input_filename, &statbuf))
- stat_result = 1;
- ress.dstFile = foutput;
- result = LZ4IO_decompressSrcFile(ress, input_filename, output_filename, prefs);
- fclose(foutput);
- /* Copy owner, file permissions and modification time */
- if ( stat_result != 0
- && !LZ4IO_isStdout(output_filename)
- && !LZ4IO_isDevNull(output_filename)) {
- UTIL_setFileStat(output_filename, &statbuf);
- /* should return value be read ? or is silent fail good enough ? */
- }
- return result;
- }
- /* Note : LZ4IO_decompressFilename()
- * can provide total decompression time for the specified fileName.
- * This information is not available with LZ4IO_decompressMultipleFilenames().
- */
- int LZ4IO_decompressFilename(const char* input_filename, const char* output_filename, const LZ4IO_prefs_t* prefs)
- {
- dRess_t const ress = LZ4IO_createDResources(prefs);
- clock_t const start = clock();
- int const status = LZ4IO_decompressDstFile(ress, input_filename, output_filename, prefs);
- clock_t const end = clock();
- double const seconds = (double)(end - start) / CLOCKS_PER_SEC;
- DISPLAYLEVEL(4, "Done in %.2f sec \n", seconds);
- LZ4IO_freeDResources(ress);
- return status;
- }
- int LZ4IO_decompressMultipleFilenames(
- const char** inFileNamesTable, int ifntSize,
- const char* suffix,
- const LZ4IO_prefs_t* prefs)
- {
- int i;
- int skippedFiles = 0;
- int missingFiles = 0;
- char* outFileName = (char*)malloc(FNSPACE);
- size_t ofnSize = FNSPACE;
- size_t const suffixSize = strlen(suffix);
- dRess_t ress = LZ4IO_createDResources(prefs);
- if (outFileName==NULL) END_PROCESS(70, "Memory allocation error");
- if (prefs->blockChecksum==0 && prefs->streamChecksum==0) {
- DISPLAYLEVEL(4, "disabling checksum validation during decoding \n");
- }
- ress.dstFile = LZ4IO_openDstFile(stdoutmark, prefs);
- for (i=0; i<ifntSize; i++) {
- size_t const ifnSize = strlen(inFileNamesTable[i]);
- const char* const suffixPtr = inFileNamesTable[i] + ifnSize - suffixSize;
- if (LZ4IO_isStdout(suffix) || LZ4IO_isDevNull(suffix)) {
- missingFiles += LZ4IO_decompressSrcFile(ress, inFileNamesTable[i], suffix, prefs);
- continue;
- }
- if (ofnSize <= ifnSize-suffixSize+1) {
- free(outFileName);
- ofnSize = ifnSize + 20;
- outFileName = (char*)malloc(ofnSize);
- if (outFileName==NULL) END_PROCESS(71, "Memory allocation error");
- }
- if (ifnSize <= suffixSize || !UTIL_sameString(suffixPtr, suffix) ) {
- DISPLAYLEVEL(1, "File extension doesn't match expected LZ4_EXTENSION (%4s); will not process file: %s\n", suffix, inFileNamesTable[i]);
- skippedFiles++;
- continue;
- }
- memcpy(outFileName, inFileNamesTable[i], ifnSize - suffixSize);
- outFileName[ifnSize-suffixSize] = '\0';
- missingFiles += LZ4IO_decompressDstFile(ress, inFileNamesTable[i], outFileName, prefs);
- }
- LZ4IO_freeDResources(ress);
- free(outFileName);
- return missingFiles + skippedFiles;
- }
- /* ********************************************************************* */
- /* ********************** LZ4 --list command *********************** */
- /* ********************************************************************* */
- typedef enum
- {
- lz4Frame = 0,
- legacyFrame,
- skippableFrame
- } LZ4IO_frameType_t;
- typedef struct {
- LZ4F_frameInfo_t lz4FrameInfo;
- LZ4IO_frameType_t frameType;
- } LZ4IO_frameInfo_t;
- #define LZ4IO_INIT_FRAMEINFO { LZ4F_INIT_FRAMEINFO, lz4Frame }
- typedef struct {
- const char* fileName;
- unsigned long long fileSize;
- unsigned long long frameCount;
- LZ4IO_frameInfo_t frameSummary;
- unsigned short eqFrameTypes;
- unsigned short eqBlockTypes;
- unsigned short allContentSize;
- } LZ4IO_cFileInfo_t;
- #define LZ4IO_INIT_CFILEINFO { NULL, 0ULL, 0, LZ4IO_INIT_FRAMEINFO, 1, 1, 1 }
- typedef enum { LZ4IO_LZ4F_OK, LZ4IO_format_not_known, LZ4IO_not_a_file } LZ4IO_infoResult;
- static const char * LZ4IO_frameTypeNames[] = {"LZ4Frame", "LegacyFrame", "SkippableFrame" };
- /* Read block headers and skip block data
- Return total blocks size for this frame including block headers,
- block checksums and content checksums.
- returns 0 in case it can't successfully skip block data.
- Assumes SEEK_CUR after frame header.
- */
- static unsigned long long
- LZ4IO_skipBlocksData(FILE* finput,
- const LZ4F_blockChecksum_t blockChecksumFlag,
- const LZ4F_contentChecksum_t contentChecksumFlag)
- {
- unsigned char blockInfo[LZ4F_BLOCK_HEADER_SIZE];
- unsigned long long totalBlocksSize = 0;
- for (;;) {
- if (!fread(blockInfo, 1, LZ4F_BLOCK_HEADER_SIZE, finput)) {
- if (feof(finput)) return totalBlocksSize;
- return 0;
- }
- totalBlocksSize += LZ4F_BLOCK_HEADER_SIZE;
- { const unsigned long nextCBlockSize = LZ4IO_readLE32(&blockInfo) & 0x7FFFFFFFU;
- const unsigned long nextBlock = nextCBlockSize + (blockChecksumFlag * LZ4F_BLOCK_CHECKSUM_SIZE);
- if (nextCBlockSize == 0) {
- /* Reached EndMark */
- if (contentChecksumFlag) {
- /* Skip content checksum */
- if (UTIL_fseek(finput, LZ4F_CONTENT_CHECKSUM_SIZE, SEEK_CUR) != 0) {
- return 0;
- }
- totalBlocksSize += LZ4F_CONTENT_CHECKSUM_SIZE;
- }
- break;
- }
- totalBlocksSize += nextBlock;
- /* skip to the next block */
- assert(nextBlock < LONG_MAX);
- if (UTIL_fseek(finput, (long)nextBlock, SEEK_CUR) != 0) return 0;
- } }
- return totalBlocksSize;
- }
- static const unsigned long long legacyFrameUndecodable = (0ULL-1);
- /* For legacy frames only.
- Read block headers and skip block data.
- Return total blocks size for this frame including block headers.
- or legacyFrameUndecodable in case it can't successfully skip block data.
- This works as long as legacy block header size = magic number size.
- Assumes SEEK_CUR after frame header.
- */
- static unsigned long long LZ4IO_skipLegacyBlocksData(FILE* finput)
- {
- unsigned char blockInfo[LZ4IO_LEGACY_BLOCK_HEADER_SIZE];
- unsigned long long totalBlocksSize = 0;
- LZ4IO_STATIC_ASSERT(LZ4IO_LEGACY_BLOCK_HEADER_SIZE == MAGICNUMBER_SIZE);
- for (;;) {
- size_t const bhs = fread(blockInfo, 1, LZ4IO_LEGACY_BLOCK_HEADER_SIZE, finput);
- if (bhs == 0) {
- if (feof(finput)) return totalBlocksSize;
- return legacyFrameUndecodable;
- }
- if (bhs != 4) {
- return legacyFrameUndecodable;
- }
- { const unsigned int nextCBlockSize = LZ4IO_readLE32(&blockInfo);
- if ( nextCBlockSize == LEGACY_MAGICNUMBER
- || nextCBlockSize == LZ4IO_MAGICNUMBER
- || LZ4IO_isSkippableMagicNumber(nextCBlockSize) ) {
- /* Rewind back. we want cursor at the beginning of next frame */
- if (UTIL_fseek(finput, -LZ4IO_LEGACY_BLOCK_HEADER_SIZE, SEEK_CUR) != 0) {
- END_PROCESS(37, "impossible to skip backward");
- }
- break;
- }
- if (nextCBlockSize > LZ4IO_LEGACY_BLOCK_SIZE_MAX) {
- DISPLAYLEVEL(4, "Error : block in legacy frame is too large \n");
- return legacyFrameUndecodable;
- }
- totalBlocksSize += LZ4IO_LEGACY_BLOCK_HEADER_SIZE + nextCBlockSize;
- /* skip to the next block
- * note : this won't fail if nextCBlockSize is too large, skipping past the end of finput */
- if (UTIL_fseek(finput, nextCBlockSize, SEEK_CUR) != 0) {
- return legacyFrameUndecodable;
- } } }
- return totalBlocksSize;
- }
- /* LZ4IO_blockTypeID:
- * return human-readable block type, following command line convention
- * buffer : must be a valid memory area of at least 4 bytes */
- const char* LZ4IO_blockTypeID(LZ4F_blockSizeID_t sizeID, LZ4F_blockMode_t blockMode, char buffer[4])
- {
- buffer[0] = 'B';
- assert(sizeID >= 4); assert(sizeID <= 7);
- buffer[1] = (char)(sizeID + '0');
- buffer[2] = (blockMode == LZ4F_blockIndependent) ? 'I' : 'D';
- buffer[3] = 0;
- return buffer;
- }
- /* buffer : must be valid memory area of at least 10 bytes */
- static const char* LZ4IO_toHuman(long double size, char *buf)
- {
- const char units[] = {"\0KMGTPEZY"};
- size_t i = 0;
- for (; size >= 1024; i++) size /= 1024;
- sprintf(buf, "%.2Lf%c", size, units[i]);
- return buf;
- }
- /* Get filename without path prefix */
- static const char* LZ4IO_baseName(const char* input_filename)
- {
- const char* b = strrchr(input_filename, '/');
- if (!b) b = strrchr(input_filename, '\\');
- if (!b) return input_filename;
- return b + 1;
- }
- /* Report frame/s information (--list) in verbose mode (-v).
- * Will populate file info with fileName and frameSummary where applicable.
- * - TODO :
- * + report nb of blocks, hence max. possible decompressed size (when not reported in header)
- */
- static LZ4IO_infoResult
- LZ4IO_getCompressedFileInfo(LZ4IO_cFileInfo_t* cfinfo, const char* input_filename)
- {
- LZ4IO_infoResult result = LZ4IO_format_not_known; /* default result (error) */
- unsigned char buffer[LZ4F_HEADER_SIZE_MAX];
- FILE* const finput = LZ4IO_openSrcFile(input_filename);
- if (finput == NULL) return LZ4IO_not_a_file;
- cfinfo->fileSize = UTIL_getOpenFileSize(finput);
- while (!feof(finput)) {
- LZ4IO_frameInfo_t frameInfo = LZ4IO_INIT_FRAMEINFO;
- unsigned magicNumber;
- /* Get MagicNumber */
- { size_t const nbReadBytes = fread(buffer, 1, MAGICNUMBER_SIZE, finput);
- if (nbReadBytes == 0) { break; } /* EOF */
- result = LZ4IO_format_not_known; /* default result (error) */
- if (nbReadBytes != MAGICNUMBER_SIZE) {
- END_PROCESS(40, "Unrecognized header : Magic Number unreadable");
- } }
- magicNumber = LZ4IO_readLE32(buffer); /* Little Endian format */
- if (LZ4IO_isSkippableMagicNumber(magicNumber))
- magicNumber = LZ4IO_SKIPPABLE0; /* fold skippable magic numbers */
- switch (magicNumber) {
- case LZ4IO_MAGICNUMBER:
- if (cfinfo->frameSummary.frameType != lz4Frame) cfinfo->eqFrameTypes = 0;
- /* Get frame info */
- { const size_t readBytes = fread(buffer + MAGICNUMBER_SIZE, 1, LZ4F_HEADER_SIZE_MIN - MAGICNUMBER_SIZE, finput);
- if (!readBytes || ferror(finput)) END_PROCESS(71, "Error reading %s", input_filename);
- }
- { size_t hSize = LZ4F_headerSize(&buffer, LZ4F_HEADER_SIZE_MIN);
- if (LZ4F_isError(hSize)) break;
- if (hSize > (LZ4F_HEADER_SIZE_MIN + MAGICNUMBER_SIZE)) {
- /* We've already read LZ4F_HEADER_SIZE_MIN so read any extra until hSize*/
- const size_t readBytes = fread(buffer + LZ4F_HEADER_SIZE_MIN, 1, hSize - LZ4F_HEADER_SIZE_MIN, finput);
- if (!readBytes || ferror(finput)) END_PROCESS(72, "Error reading %s", input_filename);
- }
- /* Create decompression context */
- { LZ4F_dctx* dctx;
- if ( LZ4F_isError(LZ4F_createDecompressionContext(&dctx, LZ4F_VERSION)) ) break;
- { unsigned const frameInfoError = LZ4F_isError(LZ4F_getFrameInfo(dctx, &frameInfo.lz4FrameInfo, buffer, &hSize));
- LZ4F_freeDecompressionContext(dctx);
- if (frameInfoError) break;
- if ((cfinfo->frameSummary.lz4FrameInfo.blockSizeID != frameInfo.lz4FrameInfo.blockSizeID ||
- cfinfo->frameSummary.lz4FrameInfo.blockMode != frameInfo.lz4FrameInfo.blockMode)
- && cfinfo->frameCount != 0)
- cfinfo->eqBlockTypes = 0;
- { const unsigned long long totalBlocksSize = LZ4IO_skipBlocksData(finput,
- frameInfo.lz4FrameInfo.blockChecksumFlag,
- frameInfo.lz4FrameInfo.contentChecksumFlag);
- if (totalBlocksSize) {
- char bTypeBuffer[5];
- LZ4IO_blockTypeID(frameInfo.lz4FrameInfo.blockSizeID, frameInfo.lz4FrameInfo.blockMode, bTypeBuffer);
- DISPLAYLEVEL(3, " %6llu %14s %5s %8s",
- cfinfo->frameCount + 1,
- LZ4IO_frameTypeNames[frameInfo.frameType],
- bTypeBuffer,
- frameInfo.lz4FrameInfo.contentChecksumFlag ? "XXH32" : "-");
- if (frameInfo.lz4FrameInfo.contentSize) {
- { double const ratio = (double)(totalBlocksSize + hSize) / frameInfo.lz4FrameInfo.contentSize * 100;
- DISPLAYLEVEL(3, " %20llu %20llu %9.2f%%\n",
- totalBlocksSize + hSize,
- frameInfo.lz4FrameInfo.contentSize,
- ratio);
- }
- /* Now we've consumed frameInfo we can use it to store the total contentSize */
- frameInfo.lz4FrameInfo.contentSize += cfinfo->frameSummary.lz4FrameInfo.contentSize;
- }
- else {
- DISPLAYLEVEL(3, " %20llu %20s %9s \n", totalBlocksSize + hSize, "-", "-");
- cfinfo->allContentSize = 0;
- }
- result = LZ4IO_LZ4F_OK;
- } } } } }
- break;
- case LEGACY_MAGICNUMBER:
- frameInfo.frameType = legacyFrame;
- if (cfinfo->frameSummary.frameType != legacyFrame && cfinfo->frameCount != 0) cfinfo->eqFrameTypes = 0;
- cfinfo->eqBlockTypes = 0;
- cfinfo->allContentSize = 0;
- { const unsigned long long totalBlocksSize = LZ4IO_skipLegacyBlocksData(finput);
- if (totalBlocksSize == legacyFrameUndecodable) {
- DISPLAYLEVEL(1, "Corrupted legacy frame \n");
- result = LZ4IO_format_not_known;
- break;
- }
- if (totalBlocksSize) {
- DISPLAYLEVEL(3, " %6llu %14s %5s %8s %20llu %20s %9s\n",
- cfinfo->frameCount + 1,
- LZ4IO_frameTypeNames[frameInfo.frameType],
- "-", "-",
- totalBlocksSize + 4,
- "-", "-");
- result = LZ4IO_LZ4F_OK;
- } }
- break;
- case LZ4IO_SKIPPABLE0:
- frameInfo.frameType = skippableFrame;
- if (cfinfo->frameSummary.frameType != skippableFrame && cfinfo->frameCount != 0) cfinfo->eqFrameTypes = 0;
- cfinfo->eqBlockTypes = 0;
- cfinfo->allContentSize = 0;
- { size_t const nbReadBytes = fread(buffer, 1, 4, finput);
- if (nbReadBytes != 4)
- END_PROCESS(42, "Stream error : skippable size unreadable");
- }
- { unsigned const size = LZ4IO_readLE32(buffer);
- int const errorNb = fseek_u32(finput, size, SEEK_CUR);
- if (errorNb != 0)
- END_PROCESS(43, "Stream error : cannot skip skippable area");
- DISPLAYLEVEL(3, " %6llu %14s %5s %8s %20u %20s %9s\n",
- cfinfo->frameCount + 1,
- "SkippableFrame",
- "-", "-", size + 8, "-", "-");
- result = LZ4IO_LZ4F_OK;
- }
- break;
- default:
- { long int const position = ftell(finput); /* only works for files < 2 GB */
- DISPLAYLEVEL(3, "Stream followed by undecodable data ");
- if (position != -1L)
- DISPLAYLEVEL(3, "at position %i ", (int)position);
- result = LZ4IO_format_not_known;
- DISPLAYLEVEL(3, "\n");
- }
- break;
- }
- if (result != LZ4IO_LZ4F_OK) break;
- cfinfo->frameSummary = frameInfo;
- cfinfo->frameCount++;
- } /* while (!feof(finput)) */
- fclose(finput);
- return result;
- }
- int LZ4IO_displayCompressedFilesInfo(const char** inFileNames, size_t ifnIdx)
- {
- int result = 0;
- size_t idx = 0;
- if (g_displayLevel < 3) {
- DISPLAYOUT("%10s %14s %5s %11s %13s %9s %s\n",
- "Frames", "Type", "Block", "Compressed", "Uncompressed", "Ratio", "Filename");
- }
- for (; idx < ifnIdx; idx++) {
- /* Get file info */
- LZ4IO_cFileInfo_t cfinfo = LZ4IO_INIT_CFILEINFO;
- cfinfo.fileName = LZ4IO_baseName(inFileNames[idx]);
- if (LZ4IO_isStdin(inFileNames[idx]) ? !UTIL_isRegFD(0) : !UTIL_isRegFile(inFileNames[idx])) {
- DISPLAYLEVEL(1, "lz4: %s is not a regular file \n", inFileNames[idx]);
- return 1;
- }
- DISPLAYLEVEL(3, "%s(%llu/%llu)\n", cfinfo.fileName, (unsigned long long)idx + 1, (unsigned long long)ifnIdx);
- DISPLAYLEVEL(3, " %6s %14s %5s %8s %20s %20s %9s\n",
- "Frame", "Type", "Block", "Checksum", "Compressed", "Uncompressed", "Ratio")
- { LZ4IO_infoResult const op_result = LZ4IO_getCompressedFileInfo(&cfinfo, inFileNames[idx]);
- if (op_result != LZ4IO_LZ4F_OK) {
- assert(op_result == LZ4IO_format_not_known);
- DISPLAYLEVEL(1, "lz4: %s: File format not recognized \n", inFileNames[idx]);
- return 1;
- } }
- DISPLAYLEVEL(3, "\n");
- if (g_displayLevel < 3) {
- /* Display Summary */
- { char buffers[3][10];
- DISPLAYOUT("%10llu %14s %5s %11s %13s ",
- cfinfo.frameCount,
- cfinfo.eqFrameTypes ? LZ4IO_frameTypeNames[cfinfo.frameSummary.frameType] : "-" ,
- cfinfo.eqBlockTypes ? LZ4IO_blockTypeID(cfinfo.frameSummary.lz4FrameInfo.blockSizeID,
- cfinfo.frameSummary.lz4FrameInfo.blockMode, buffers[0]) : "-",
- LZ4IO_toHuman((long double)cfinfo.fileSize, buffers[1]),
- cfinfo.allContentSize ? LZ4IO_toHuman((long double)cfinfo.frameSummary.lz4FrameInfo.contentSize, buffers[2]) : "-");
- if (cfinfo.allContentSize) {
- double const ratio = (double)cfinfo.fileSize / cfinfo.frameSummary.lz4FrameInfo.contentSize * 100;
- DISPLAYOUT("%9.2f%% %s \n", ratio, cfinfo.fileName);
- } else {
- DISPLAYOUT("%9s %s\n",
- "-",
- cfinfo.fileName);
- } } } /* if (g_displayLevel < 3) */
- } /* for (; idx < ifnIdx; idx++) */
- return result;
- }
|