2014-02-04 22:11:10 +08:00
|
|
|
/*
|
|
|
|
LZ4io.c - LZ4 File/Stream Interface
|
2015-03-07 20:23:00 +08:00
|
|
|
Copyright (C) Yann Collet 2011-2015
|
2015-03-13 09:24:08 +08:00
|
|
|
|
2014-02-04 22:11:10 +08:00
|
|
|
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 :
|
2015-03-13 09:24:08 +08:00
|
|
|
- LZ4 source repository : https://github.com/Cyan4973/lz4
|
2014-02-04 22:11:10 +08:00
|
|
|
- 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.
|
|
|
|
*/
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/**************************************
|
|
|
|
* Compiler Options
|
2015-03-14 03:36:59 +08:00
|
|
|
**************************************/
|
2014-02-04 22:11:10 +08:00
|
|
|
#ifdef _MSC_VER /* Visual Studio */
|
|
|
|
# define _CRT_SECURE_NO_WARNINGS
|
2014-12-17 05:03:16 +08:00
|
|
|
# define _CRT_SECURE_NO_DEPRECATE /* VS2005 */
|
|
|
|
# pragma warning(disable : 4127) /* disable: C4127: conditional expression is constant */
|
2014-02-04 22:11:10 +08:00
|
|
|
#endif
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
#define _LARGE_FILES /* Large file support on 32-bits AIX */
|
|
|
|
#define _FILE_OFFSET_BITS 64 /* Large file support on 32-bits unix */
|
2014-02-04 22:11:10 +08:00
|
|
|
|
|
|
|
|
2015-03-14 03:36:59 +08:00
|
|
|
/*****************************
|
2014-12-17 05:03:16 +08:00
|
|
|
* Includes
|
|
|
|
*****************************/
|
2015-03-23 09:20:42 +08:00
|
|
|
#include <stdio.h> /* fprintf, fopen, fread, stdin, stdout */
|
|
|
|
#include <stdlib.h> /* malloc, free */
|
|
|
|
#include <string.h> /* strcmp, strlen */
|
|
|
|
#include <time.h> /* clock */
|
|
|
|
#include <sys/types.h> /* stat64 */
|
|
|
|
#include <sys/stat.h> /* stat64 */
|
2014-02-04 22:11:10 +08:00
|
|
|
#include "lz4io.h"
|
2015-04-11 17:25:08 +08:00
|
|
|
#include "lz4.h" /* still required for legacy format */
|
|
|
|
#include "lz4hc.h" /* still required for legacy format */
|
2014-12-01 00:59:31 +08:00
|
|
|
#include "lz4frame.h"
|
2014-02-04 22:11:10 +08:00
|
|
|
|
|
|
|
|
2015-03-14 05:15:08 +08:00
|
|
|
/******************************
|
2014-12-17 05:03:16 +08:00
|
|
|
* OS-specific Includes
|
2015-03-14 05:15:08 +08:00
|
|
|
******************************/
|
2015-04-02 02:47:24 +08:00
|
|
|
#if defined(MSDOS) || defined(OS2) || defined(WIN32) || defined(_WIN32)
|
2014-12-17 05:03:16 +08:00
|
|
|
# include <fcntl.h> /* _O_BINARY */
|
2015-03-14 05:15:08 +08:00
|
|
|
# include <io.h> /* _setmode, _fileno, _get_osfhandle */
|
2015-03-10 23:57:42 +08:00
|
|
|
# define SET_BINARY_MODE(file) _setmode(_fileno(file), _O_BINARY)
|
2015-03-15 08:42:27 +08:00
|
|
|
# include <Windows.h> /* DeviceIoControl, HANDLE, FSCTL_SET_SPARSE */
|
2015-03-14 05:15:08 +08:00
|
|
|
# define SET_SPARSE_FILE_MODE(file) { DWORD dw; DeviceIoControl((HANDLE) _get_osfhandle(_fileno(file)), FSCTL_SET_SPARSE, 0, 0, 0, 0, &dw, 0); }
|
2015-02-14 11:48:11 +08:00
|
|
|
# if defined(_MSC_VER) && (_MSC_VER >= 1400) /* Avoid MSVC fseek()'s 2GiB barrier */
|
|
|
|
# define fseek _fseeki64
|
|
|
|
# endif
|
2014-02-04 22:11:10 +08:00
|
|
|
#else
|
|
|
|
# define SET_BINARY_MODE(file)
|
2015-03-14 05:15:08 +08:00
|
|
|
# define SET_SPARSE_FILE_MODE(file)
|
2014-02-04 22:11:10 +08:00
|
|
|
#endif
|
|
|
|
|
2015-03-31 21:47:23 +08:00
|
|
|
#if !defined(S_ISREG)
|
|
|
|
# define S_ISREG(x) (((x) & S_IFMT) == S_IFREG)
|
|
|
|
#endif
|
|
|
|
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2015-03-14 05:15:08 +08:00
|
|
|
/*****************************
|
2014-12-17 05:03:16 +08:00
|
|
|
* Constants
|
|
|
|
*****************************/
|
2014-12-01 00:59:31 +08:00
|
|
|
#define KB *(1 <<10)
|
|
|
|
#define MB *(1 <<20)
|
2014-02-04 22:11:10 +08:00
|
|
|
#define GB *(1U<<30)
|
|
|
|
|
|
|
|
#define _1BIT 0x01
|
|
|
|
#define _2BITS 0x03
|
|
|
|
#define _3BITS 0x07
|
|
|
|
#define _4BITS 0x0F
|
|
|
|
#define _8BITS 0xFF
|
|
|
|
|
2015-03-18 01:02:01 +08:00
|
|
|
#define MAGICNUMBER_SIZE 4
|
|
|
|
#define LZ4IO_MAGICNUMBER 0x184D2204
|
|
|
|
#define LZ4IO_SKIPPABLE0 0x184D2A50
|
|
|
|
#define LZ4IO_SKIPPABLEMASK 0xFFFFFFF0
|
|
|
|
#define LEGACY_MAGICNUMBER 0x184C2102
|
2014-02-04 22:11:10 +08:00
|
|
|
|
|
|
|
#define CACHELINE 64
|
|
|
|
#define LEGACY_BLOCKSIZE (8 MB)
|
2014-06-18 04:41:59 +08:00
|
|
|
#define MIN_STREAM_BUFSIZE (192 KB)
|
2015-03-18 01:02:01 +08:00
|
|
|
#define LZ4IO_BLOCKSIZEID_DEFAULT 7
|
|
|
|
|
|
|
|
#define sizeT sizeof(size_t)
|
|
|
|
#define maskT (sizeT - 1)
|
2014-02-04 22:11:10 +08:00
|
|
|
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/**************************************
|
|
|
|
* Macros
|
2015-03-14 03:36:59 +08:00
|
|
|
**************************************/
|
2014-02-04 22:11:10 +08:00
|
|
|
#define DISPLAY(...) fprintf(stderr, __VA_ARGS__)
|
2015-03-12 02:42:37 +08:00
|
|
|
#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 ((LZ4IO_GetMilliSpan(g_time) > refreshRate) || (g_displayLevel>=4)) \
|
2014-12-01 00:59:31 +08:00
|
|
|
{ g_time = clock(); DISPLAY(__VA_ARGS__); \
|
2015-03-12 02:42:37 +08:00
|
|
|
if (g_displayLevel>=4) fflush(stdout); } }
|
2014-12-01 00:59:31 +08:00
|
|
|
static const unsigned refreshRate = 150;
|
|
|
|
static clock_t g_time = 0;
|
2014-02-04 22:11:10 +08:00
|
|
|
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/**************************************
|
|
|
|
* Local Parameters
|
2015-03-14 03:36:59 +08:00
|
|
|
**************************************/
|
2015-03-12 02:42:37 +08:00
|
|
|
static int g_overwrite = 1;
|
2015-03-18 01:02:01 +08:00
|
|
|
static int g_blockSizeId = LZ4IO_BLOCKSIZEID_DEFAULT;
|
2015-03-12 02:42:37 +08:00
|
|
|
static int g_blockChecksum = 0;
|
|
|
|
static int g_streamChecksum = 1;
|
|
|
|
static int g_blockIndependence = 1;
|
|
|
|
static int g_sparseFileSupport = 0;
|
2015-03-23 09:20:42 +08:00
|
|
|
static int g_contentSizeFlag = 0;
|
2014-02-04 22:11:10 +08:00
|
|
|
|
|
|
|
static const int minBlockSizeID = 4;
|
|
|
|
static const int maxBlockSizeID = 7;
|
|
|
|
|
2014-12-07 00:10:54 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/**************************************
|
|
|
|
* Exceptions
|
|
|
|
***************************************/
|
2014-02-04 22:11:10 +08:00
|
|
|
#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); \
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/**************************************
|
|
|
|
* Version modifiers
|
2015-03-18 01:02:01 +08:00
|
|
|
**************************************/
|
2014-02-04 22:11:10 +08:00
|
|
|
#define EXTENDED_ARGUMENTS
|
|
|
|
#define EXTENDED_HELP
|
|
|
|
#define EXTENDED_FORMAT
|
|
|
|
#define DEFAULT_DECOMPRESSOR decodeLZ4S
|
|
|
|
|
|
|
|
|
|
|
|
/* ************************************************** */
|
|
|
|
/* ****************** Parameters ******************** */
|
|
|
|
/* ************************************************** */
|
|
|
|
|
|
|
|
/* Default setting : overwrite = 1; return : overwrite mode (0/1) */
|
|
|
|
int LZ4IO_setOverwrite(int yes)
|
|
|
|
{
|
2015-03-12 02:42:37 +08:00
|
|
|
g_overwrite = (yes!=0);
|
|
|
|
return g_overwrite;
|
2014-02-04 22:11:10 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/* blockSizeID : valid values : 4-5-6-7 */
|
|
|
|
int LZ4IO_setBlockSizeID(int bsid)
|
|
|
|
{
|
|
|
|
static const int blockSizeTable[] = { 64 KB, 256 KB, 1 MB, 4 MB };
|
|
|
|
if ((bsid < minBlockSizeID) || (bsid > maxBlockSizeID)) return -1;
|
2015-03-12 02:42:37 +08:00
|
|
|
g_blockSizeId = bsid;
|
|
|
|
return blockSizeTable[g_blockSizeId-minBlockSizeID];
|
2014-02-04 22:11:10 +08:00
|
|
|
}
|
|
|
|
|
2014-12-01 00:59:31 +08:00
|
|
|
int LZ4IO_setBlockMode(LZ4IO_blockMode_t blockMode)
|
2014-02-04 22:11:10 +08:00
|
|
|
{
|
2015-03-12 02:42:37 +08:00
|
|
|
g_blockIndependence = (blockMode == LZ4IO_blockIndependent);
|
|
|
|
return g_blockIndependence;
|
2014-02-04 22:11:10 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Default setting : no checksum */
|
|
|
|
int LZ4IO_setBlockChecksumMode(int xxhash)
|
|
|
|
{
|
2015-03-12 02:42:37 +08:00
|
|
|
g_blockChecksum = (xxhash != 0);
|
|
|
|
return g_blockChecksum;
|
2014-02-04 22:11:10 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Default setting : checksum enabled */
|
|
|
|
int LZ4IO_setStreamChecksumMode(int xxhash)
|
|
|
|
{
|
2015-03-12 02:42:37 +08:00
|
|
|
g_streamChecksum = (xxhash != 0);
|
|
|
|
return g_streamChecksum;
|
2014-02-04 22:11:10 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Default setting : 0 (no notification) */
|
|
|
|
int LZ4IO_setNotificationLevel(int level)
|
|
|
|
{
|
2015-03-12 02:42:37 +08:00
|
|
|
g_displayLevel = level;
|
|
|
|
return g_displayLevel;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Default setting : 0 (disabled) */
|
|
|
|
int LZ4IO_setSparseFile(int enable)
|
|
|
|
{
|
2015-03-17 03:38:25 +08:00
|
|
|
g_sparseFileSupport = (enable!=0);
|
2015-03-12 02:42:37 +08:00
|
|
|
return g_sparseFileSupport;
|
2014-02-04 22:11:10 +08:00
|
|
|
}
|
|
|
|
|
2015-03-23 09:20:42 +08:00
|
|
|
/* Default setting : 0 (disabled) */
|
|
|
|
int LZ4IO_setContentSize(int enable)
|
|
|
|
{
|
|
|
|
g_contentSizeFlag = (enable!=0);
|
|
|
|
return g_contentSizeFlag;
|
|
|
|
}
|
|
|
|
|
2014-12-01 00:59:31 +08:00
|
|
|
static unsigned LZ4IO_GetMilliSpan(clock_t nPrevious)
|
|
|
|
{
|
|
|
|
clock_t nCurrent = clock();
|
2014-12-07 00:10:54 +08:00
|
|
|
unsigned nSpan = (unsigned)(((nCurrent - nPrevious) * 1000) / CLOCKS_PER_SEC);
|
2014-12-01 00:59:31 +08:00
|
|
|
return nSpan;
|
|
|
|
}
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2015-03-23 09:20:42 +08:00
|
|
|
static unsigned long long LZ4IO_GetFileSize(const char* infilename)
|
|
|
|
{
|
|
|
|
int r;
|
|
|
|
#if defined(_MSC_VER)
|
|
|
|
struct _stat64 statbuf;
|
|
|
|
r = _stat64(infilename, &statbuf);
|
|
|
|
#else
|
|
|
|
struct stat statbuf;
|
|
|
|
r = stat(infilename, &statbuf);
|
|
|
|
#endif
|
|
|
|
if (r || !S_ISREG(statbuf.st_mode)) return 0; /* No good... */
|
|
|
|
return (unsigned long long)statbuf.st_size;
|
|
|
|
}
|
|
|
|
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2015-03-14 03:36:59 +08:00
|
|
|
/* ************************************************************************ **
|
|
|
|
** ********************** LZ4 File / Pipe compression ********************* **
|
|
|
|
** ************************************************************************ */
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2015-03-18 01:02:01 +08:00
|
|
|
static int LZ4IO_GetBlockSize_FromBlockId (int id) { return (1 << (8 + (2 * id))); }
|
|
|
|
static int LZ4IO_isSkippableMagicNumber(unsigned int magic) { return (magic & LZ4IO_SKIPPABLEMASK) == LZ4IO_SKIPPABLE0; }
|
2014-02-04 22:11:10 +08:00
|
|
|
|
|
|
|
|
2015-03-07 20:23:00 +08:00
|
|
|
static int get_fileHandle(const char* input_filename, const char* output_filename, FILE** pfinput, FILE** pfoutput)
|
2014-02-04 22:11:10 +08:00
|
|
|
{
|
|
|
|
|
|
|
|
if (!strcmp (input_filename, stdinmark))
|
|
|
|
{
|
|
|
|
DISPLAYLEVEL(4,"Using stdin for input\n");
|
|
|
|
*pfinput = stdin;
|
|
|
|
SET_BINARY_MODE(stdin);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
*pfinput = fopen(input_filename, "rb");
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!strcmp (output_filename, stdoutmark))
|
|
|
|
{
|
|
|
|
DISPLAYLEVEL(4,"Using stdout for output\n");
|
|
|
|
*pfoutput = stdout;
|
|
|
|
SET_BINARY_MODE(stdout);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Check if destination file already exists */
|
2014-02-04 22:11:10 +08:00
|
|
|
*pfoutput=0;
|
|
|
|
if (output_filename != nulmark) *pfoutput = fopen( output_filename, "rb" );
|
|
|
|
if (*pfoutput!=0)
|
|
|
|
{
|
|
|
|
fclose(*pfoutput);
|
2015-03-12 02:42:37 +08:00
|
|
|
if (!g_overwrite)
|
2014-02-04 22:11:10 +08:00
|
|
|
{
|
|
|
|
char ch;
|
|
|
|
DISPLAYLEVEL(2, "Warning : %s already exists\n", output_filename);
|
|
|
|
DISPLAYLEVEL(2, "Overwrite ? (Y/N) : ");
|
2015-03-12 02:42:37 +08:00
|
|
|
if (g_displayLevel <= 1) EXM_THROW(11, "Operation aborted : %s already exists", output_filename); /* No interaction possible */
|
2014-02-04 22:11:10 +08:00
|
|
|
ch = (char)getchar();
|
|
|
|
if ((ch!='Y') && (ch!='y')) EXM_THROW(11, "Operation aborted : %s already exists", output_filename);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
*pfoutput = fopen( output_filename, "wb" );
|
|
|
|
}
|
|
|
|
|
|
|
|
if ( *pfinput==0 ) EXM_THROW(12, "Pb opening %s", input_filename);
|
|
|
|
if ( *pfoutput==0) EXM_THROW(13, "Pb opening %s", output_filename);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2014-12-01 00:59:31 +08:00
|
|
|
|
|
|
|
/***************************************
|
2015-03-14 03:36:59 +08:00
|
|
|
* Legacy Compression
|
|
|
|
***************************************/
|
2014-12-01 00:59:31 +08:00
|
|
|
|
|
|
|
/* unoptimized version; solves endianess & alignment issues */
|
|
|
|
static void LZ4IO_writeLE32 (void* p, unsigned value32)
|
|
|
|
{
|
2015-03-10 23:57:42 +08:00
|
|
|
unsigned char* dstPtr = (unsigned char*)p;
|
2014-12-01 00:59:31 +08:00
|
|
|
dstPtr[0] = (unsigned char)value32;
|
|
|
|
dstPtr[1] = (unsigned char)(value32 >> 8);
|
|
|
|
dstPtr[2] = (unsigned char)(value32 >> 16);
|
|
|
|
dstPtr[3] = (unsigned char)(value32 >> 24);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* LZ4IO_compressFilename_Legacy :
|
|
|
|
* This function is intentionally "hidden" (not published in .h)
|
|
|
|
* It generates compressed streams using the old 'legacy' format */
|
2015-03-07 20:23:00 +08:00
|
|
|
int LZ4IO_compressFilename_Legacy(const char* input_filename, const char* output_filename, int compressionlevel)
|
2014-02-04 22:11:10 +08:00
|
|
|
{
|
|
|
|
int (*compressionFunction)(const char*, char*, int);
|
|
|
|
unsigned long long filesize = 0;
|
|
|
|
unsigned long long compressedfilesize = MAGICNUMBER_SIZE;
|
|
|
|
char* in_buff;
|
|
|
|
char* out_buff;
|
|
|
|
FILE* finput;
|
|
|
|
FILE* foutput;
|
|
|
|
clock_t start, end;
|
|
|
|
size_t sizeCheck;
|
|
|
|
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Init */
|
2014-02-04 22:11:10 +08:00
|
|
|
start = clock();
|
2014-12-01 00:59:31 +08:00
|
|
|
if (compressionlevel < 3) compressionFunction = LZ4_compress; else compressionFunction = LZ4_compressHC;
|
|
|
|
|
2014-02-04 22:11:10 +08:00
|
|
|
get_fileHandle(input_filename, output_filename, &finput, &foutput);
|
2015-03-12 02:42:37 +08:00
|
|
|
if ((g_displayLevel==2) && (compressionlevel==1)) g_displayLevel=3;
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Allocate Memory */
|
2014-02-04 22:11:10 +08:00
|
|
|
in_buff = (char*)malloc(LEGACY_BLOCKSIZE);
|
|
|
|
out_buff = (char*)malloc(LZ4_compressBound(LEGACY_BLOCKSIZE));
|
|
|
|
if (!in_buff || !out_buff) EXM_THROW(21, "Allocation error : not enough memory");
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Write Archive Header */
|
2014-12-01 00:59:31 +08:00
|
|
|
LZ4IO_writeLE32(out_buff, LEGACY_MAGICNUMBER);
|
2014-02-04 22:11:10 +08:00
|
|
|
sizeCheck = fwrite(out_buff, 1, MAGICNUMBER_SIZE, foutput);
|
|
|
|
if (sizeCheck!=MAGICNUMBER_SIZE) EXM_THROW(22, "Write error : cannot write header");
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Main Loop */
|
2014-02-04 22:11:10 +08:00
|
|
|
while (1)
|
|
|
|
{
|
|
|
|
unsigned int outSize;
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Read Block */
|
2014-02-04 22:11:10 +08:00
|
|
|
int inSize = (int) fread(in_buff, (size_t)1, (size_t)LEGACY_BLOCKSIZE, finput);
|
|
|
|
if( inSize<=0 ) break;
|
|
|
|
filesize += inSize;
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Compress Block */
|
2014-02-04 22:11:10 +08:00
|
|
|
outSize = compressionFunction(in_buff, out_buff+4, inSize);
|
|
|
|
compressedfilesize += outSize+4;
|
2014-12-01 00:59:31 +08:00
|
|
|
DISPLAYUPDATE(3, "\rRead : %i MB ==> %.2f%% ", (int)(filesize>>20), (double)compressedfilesize/filesize*100);
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Write Block */
|
2014-12-01 00:59:31 +08:00
|
|
|
LZ4IO_writeLE32(out_buff, outSize);
|
2014-02-04 22:11:10 +08:00
|
|
|
sizeCheck = fwrite(out_buff, 1, outSize+4, foutput);
|
|
|
|
if (sizeCheck!=(size_t)(outSize+4)) EXM_THROW(23, "Write error : cannot write compressed block");
|
|
|
|
}
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Status */
|
2014-02-04 22:11:10 +08:00
|
|
|
end = clock();
|
|
|
|
DISPLAYLEVEL(2, "\r%79s\r", "");
|
2015-04-12 15:21:35 +08:00
|
|
|
filesize += !filesize; /* avoid divide by zero */
|
2014-02-04 22:11:10 +08:00
|
|
|
DISPLAYLEVEL(2,"Compressed %llu bytes into %llu bytes ==> %.2f%%\n",
|
|
|
|
(unsigned long long) filesize, (unsigned long long) compressedfilesize, (double)compressedfilesize/filesize*100);
|
|
|
|
{
|
|
|
|
double seconds = (double)(end - start)/CLOCKS_PER_SEC;
|
|
|
|
DISPLAYLEVEL(4,"Done in %.2f s ==> %.2f MB/s\n", seconds, (double)filesize / seconds / 1024 / 1024);
|
|
|
|
}
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Close & Free */
|
2014-02-04 22:11:10 +08:00
|
|
|
free(in_buff);
|
|
|
|
free(out_buff);
|
|
|
|
fclose(finput);
|
|
|
|
fclose(foutput);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-03-14 03:36:59 +08:00
|
|
|
/*********************************************
|
|
|
|
* Compression using Frame format
|
|
|
|
*********************************************/
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2015-03-07 20:23:00 +08:00
|
|
|
int LZ4IO_compressFilename(const char* input_filename, const char* output_filename, int compressionLevel)
|
2014-02-04 22:11:10 +08:00
|
|
|
{
|
|
|
|
unsigned long long filesize = 0;
|
|
|
|
unsigned long long compressedfilesize = 0;
|
|
|
|
char* in_buff;
|
|
|
|
char* out_buff;
|
|
|
|
FILE* finput;
|
|
|
|
FILE* foutput;
|
|
|
|
clock_t start, end;
|
|
|
|
int blockSize;
|
2014-12-01 00:59:31 +08:00
|
|
|
size_t sizeCheck, headerSize, readSize, outBuffSize;
|
|
|
|
LZ4F_compressionContext_t ctx;
|
|
|
|
LZ4F_errorCode_t errorCode;
|
2015-03-10 23:57:42 +08:00
|
|
|
LZ4F_preferences_t prefs;
|
2014-02-04 22:11:10 +08:00
|
|
|
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Init */
|
2014-02-04 22:11:10 +08:00
|
|
|
start = clock();
|
2015-03-10 23:57:42 +08:00
|
|
|
memset(&prefs, 0, sizeof(prefs));
|
2014-12-01 00:59:31 +08:00
|
|
|
errorCode = LZ4F_createCompressionContext(&ctx, LZ4F_VERSION);
|
|
|
|
if (LZ4F_isError(errorCode)) EXM_THROW(30, "Allocation error : can't create LZ4F context : %s", LZ4F_getErrorName(errorCode));
|
2014-02-04 22:11:10 +08:00
|
|
|
get_fileHandle(input_filename, output_filename, &finput, &foutput);
|
2015-03-18 01:02:01 +08:00
|
|
|
blockSize = LZ4IO_GetBlockSize_FromBlockId (g_blockSizeId);
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Set compression parameters */
|
2014-12-01 00:59:31 +08:00
|
|
|
prefs.autoFlush = 1;
|
|
|
|
prefs.compressionLevel = compressionLevel;
|
2015-03-12 02:42:37 +08:00
|
|
|
prefs.frameInfo.blockMode = (blockMode_t)g_blockIndependence;
|
|
|
|
prefs.frameInfo.blockSizeID = (blockSizeID_t)g_blockSizeId;
|
|
|
|
prefs.frameInfo.contentChecksumFlag = (contentChecksum_t)g_streamChecksum;
|
2015-03-23 09:20:42 +08:00
|
|
|
if (g_contentSizeFlag)
|
|
|
|
{
|
|
|
|
unsigned long long fileSize = LZ4IO_GetFileSize(input_filename);
|
2015-03-29 18:20:09 +08:00
|
|
|
prefs.frameInfo.contentSize = fileSize; /* == 0 if input == stdin */
|
2015-04-02 20:25:28 +08:00
|
|
|
if (fileSize==0)
|
|
|
|
DISPLAYLEVEL(3, "Warning : cannot determine uncompressed frame content size \n");
|
2015-03-23 09:20:42 +08:00
|
|
|
}
|
2014-12-01 00:59:31 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Allocate Memory */
|
2014-02-04 22:11:10 +08:00
|
|
|
in_buff = (char*)malloc(blockSize);
|
2014-12-01 00:59:31 +08:00
|
|
|
outBuffSize = LZ4F_compressBound(blockSize, &prefs);
|
|
|
|
out_buff = (char*)malloc(outBuffSize);
|
|
|
|
if (!in_buff || !out_buff) EXM_THROW(31, "Allocation error : not enough memory");
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Write Archive Header */
|
2014-12-01 00:59:31 +08:00
|
|
|
headerSize = LZ4F_compressBegin(ctx, out_buff, outBuffSize, &prefs);
|
|
|
|
if (LZ4F_isError(headerSize)) EXM_THROW(32, "File header generation failed : %s", LZ4F_getErrorName(headerSize));
|
|
|
|
sizeCheck = fwrite(out_buff, 1, headerSize, foutput);
|
|
|
|
if (sizeCheck!=headerSize) EXM_THROW(33, "Write error : cannot write header");
|
|
|
|
compressedfilesize += headerSize;
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* read first block */
|
2014-02-04 22:11:10 +08:00
|
|
|
readSize = fread(in_buff, (size_t)1, (size_t)blockSize, finput);
|
2014-12-01 00:59:31 +08:00
|
|
|
filesize += readSize;
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Main Loop */
|
2014-02-04 22:11:10 +08:00
|
|
|
while (readSize>0)
|
|
|
|
{
|
2014-12-01 00:59:31 +08:00
|
|
|
size_t outSize;
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Compress Block */
|
2014-12-01 00:59:31 +08:00
|
|
|
outSize = LZ4F_compressUpdate(ctx, out_buff, outBuffSize, in_buff, readSize, NULL);
|
|
|
|
if (LZ4F_isError(outSize)) EXM_THROW(34, "Compression failed : %s", LZ4F_getErrorName(outSize));
|
|
|
|
compressedfilesize += outSize;
|
2015-04-02 20:25:28 +08:00
|
|
|
DISPLAYUPDATE(2, "\rRead : %i MB ==> %.2f%% ", (int)(filesize>>20), (double)compressedfilesize/filesize*100);
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Write Block */
|
2014-12-01 00:59:31 +08:00
|
|
|
sizeCheck = fwrite(out_buff, 1, outSize, foutput);
|
|
|
|
if (sizeCheck!=outSize) EXM_THROW(35, "Write error : cannot write compressed block");
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Read next block */
|
2014-02-04 22:11:10 +08:00
|
|
|
readSize = fread(in_buff, (size_t)1, (size_t)blockSize, finput);
|
2014-12-01 00:59:31 +08:00
|
|
|
filesize += readSize;
|
2014-02-04 22:11:10 +08:00
|
|
|
}
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* End of Stream mark */
|
2014-12-01 00:59:31 +08:00
|
|
|
headerSize = LZ4F_compressEnd(ctx, out_buff, outBuffSize, NULL);
|
|
|
|
if (LZ4F_isError(headerSize)) EXM_THROW(36, "End of file generation failed : %s", LZ4F_getErrorName(headerSize));
|
|
|
|
|
|
|
|
sizeCheck = fwrite(out_buff, 1, headerSize, foutput);
|
|
|
|
if (sizeCheck!=headerSize) EXM_THROW(37, "Write error : cannot write end of stream");
|
|
|
|
compressedfilesize += headerSize;
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Close & Free */
|
2014-02-04 22:11:10 +08:00
|
|
|
free(in_buff);
|
|
|
|
free(out_buff);
|
|
|
|
fclose(finput);
|
|
|
|
fclose(foutput);
|
2014-12-01 00:59:31 +08:00
|
|
|
errorCode = LZ4F_freeCompressionContext(ctx);
|
|
|
|
if (LZ4F_isError(errorCode)) EXM_THROW(38, "Error : can't free LZ4F context resource : %s", LZ4F_getErrorName(errorCode));
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Final Status */
|
2014-02-04 22:11:10 +08:00
|
|
|
end = clock();
|
|
|
|
DISPLAYLEVEL(2, "\r%79s\r", "");
|
2015-04-10 05:59:07 +08:00
|
|
|
if (filesize == 0)
|
|
|
|
{
|
|
|
|
DISPLAYLEVEL(2, "Null size input; converted into %u lz4 stream\n", (unsigned)compressedfilesize);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2015-04-10 21:24:13 +08:00
|
|
|
double seconds = (double)(end - start)/CLOCKS_PER_SEC;
|
2015-04-10 05:59:07 +08:00
|
|
|
DISPLAYLEVEL(2, "Compressed %llu bytes into %llu bytes ==> %.2f%%\n",
|
2014-02-04 22:11:10 +08:00
|
|
|
(unsigned long long) filesize, (unsigned long long) compressedfilesize, (double)compressedfilesize/filesize*100);
|
|
|
|
DISPLAYLEVEL(4, "Done in %.2f s ==> %.2f MB/s\n", seconds, (double)filesize / seconds / 1024 / 1024);
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-03-15 08:42:27 +08:00
|
|
|
#define FNSPACE 30
|
2015-03-07 20:23:00 +08:00
|
|
|
int LZ4IO_compressMultipleFilenames(const char** inFileNamesTable, int ifntSize, const char* suffix, int compressionlevel)
|
|
|
|
{
|
|
|
|
int i;
|
2015-04-13 07:41:55 +08:00
|
|
|
int missing_files = 0;
|
2015-03-15 08:42:27 +08:00
|
|
|
char* outFileName = (char*)malloc(FNSPACE);
|
|
|
|
size_t ofnSize = FNSPACE;
|
2015-03-07 20:23:00 +08:00
|
|
|
const size_t suffixSize = strlen(suffix);
|
|
|
|
|
|
|
|
for (i=0; i<ifntSize; i++)
|
|
|
|
{
|
2015-04-14 03:43:54 +08:00
|
|
|
size_t ifnSize;
|
|
|
|
FILE* ifp = fopen(inFileNamesTable[i], "r");
|
2015-04-13 07:41:55 +08:00
|
|
|
if (ifp == NULL) {
|
|
|
|
DISPLAYLEVEL(2, "Unable to access file for processing: %s\n", inFileNamesTable[i]);
|
|
|
|
missing_files++;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
fclose(ifp);
|
2015-04-14 03:43:54 +08:00
|
|
|
ifnSize = strlen(inFileNamesTable[i]);
|
2015-03-10 23:57:42 +08:00
|
|
|
if (ofnSize <= ifnSize+suffixSize+1) { free(outFileName); ofnSize = ifnSize + 20; outFileName = (char*)malloc(ofnSize); }
|
2015-03-07 20:23:00 +08:00
|
|
|
strcpy(outFileName, inFileNamesTable[i]);
|
|
|
|
strcat(outFileName, suffix);
|
|
|
|
LZ4IO_compressFilename(inFileNamesTable[i], outFileName, compressionlevel);
|
|
|
|
}
|
|
|
|
free(outFileName);
|
2015-04-13 07:41:55 +08:00
|
|
|
if (missing_files > 0) return 1;
|
2015-03-07 20:23:00 +08:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2015-04-13 06:28:13 +08:00
|
|
|
int LZ4IO_decompressMultipleFilenames(const char** inFileNamesTable, int ifntSize, const char* suffix)
|
|
|
|
{
|
|
|
|
int i;
|
|
|
|
int skipped_files = 0;
|
2015-04-13 07:41:55 +08:00
|
|
|
int missing_files = 0;
|
2015-04-13 06:28:13 +08:00
|
|
|
char* outFileName = (char*)malloc(FNSPACE);
|
|
|
|
size_t ofnSize = FNSPACE;
|
2015-04-14 03:43:54 +08:00
|
|
|
const size_t suffixSize = strlen(suffix);
|
2015-04-13 06:28:13 +08:00
|
|
|
char* ifnSuffix = (char*)malloc(suffixSize + 1);
|
|
|
|
|
|
|
|
for (i=0; i<ifntSize; i++)
|
|
|
|
{
|
2015-04-14 03:43:54 +08:00
|
|
|
size_t ifnSize;
|
|
|
|
FILE* ifp = fopen(inFileNamesTable[i], "r");
|
|
|
|
if (ifp == NULL)
|
|
|
|
{
|
2015-04-13 07:41:55 +08:00
|
|
|
DISPLAYLEVEL(2, "Unable to access file for processing: %s\n", inFileNamesTable[i]);
|
|
|
|
missing_files++;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
fclose(ifp);
|
2015-04-14 03:43:54 +08:00
|
|
|
ifnSize = strlen(inFileNamesTable[i]);
|
2015-04-13 06:28:13 +08:00
|
|
|
strcpy(ifnSuffix, inFileNamesTable[i] + ifnSize - suffixSize);
|
|
|
|
if (ofnSize <= ifnSize-suffixSize+1) { free(outFileName); ofnSize = ifnSize + 20; outFileName = (char*)malloc(ofnSize); }
|
2015-04-14 03:43:54 +08:00
|
|
|
if (ifnSize <= suffixSize || strcmp(ifnSuffix, suffix) != 0)
|
|
|
|
{
|
2015-04-13 06:28:13 +08:00
|
|
|
DISPLAYLEVEL(2, "File extension doesn't match expected LZ4_EXTENSION (%4s); will not process file: %s\n", suffix, inFileNamesTable[i]);
|
2015-04-13 07:41:55 +08:00
|
|
|
skipped_files++;
|
2015-04-13 06:28:13 +08:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
memcpy(outFileName, inFileNamesTable[i], ifnSize - suffixSize);
|
|
|
|
outFileName[ifnSize-suffixSize] = '\0';
|
|
|
|
LZ4IO_decompressFilename(inFileNamesTable[i], outFileName);
|
|
|
|
}
|
|
|
|
free(outFileName);
|
2015-04-14 03:43:54 +08:00
|
|
|
free(ifnSuffix);
|
2015-04-13 07:41:55 +08:00
|
|
|
if (skipped_files > 0) return 1;
|
|
|
|
if (missing_files > 0) return 1;
|
2015-04-13 06:28:13 +08:00
|
|
|
return 0;
|
|
|
|
}
|
2015-03-07 20:23:00 +08:00
|
|
|
|
2014-02-04 22:11:10 +08:00
|
|
|
/* ********************************************************************* */
|
2015-03-07 20:23:00 +08:00
|
|
|
/* ********************** LZ4 file-stream Decompression **************** */
|
2014-02-04 22:11:10 +08:00
|
|
|
/* ********************************************************************* */
|
|
|
|
|
2014-12-01 00:59:31 +08:00
|
|
|
static unsigned LZ4IO_readLE32 (const void* s)
|
|
|
|
{
|
2015-03-10 23:57:42 +08:00
|
|
|
const unsigned char* srcPtr = (const unsigned char*)s;
|
2014-12-01 00:59:31 +08:00
|
|
|
unsigned value32 = srcPtr[0];
|
|
|
|
value32 += (srcPtr[1]<<8);
|
|
|
|
value32 += (srcPtr[2]<<16);
|
|
|
|
value32 += (srcPtr[3]<<24);
|
|
|
|
return value32;
|
|
|
|
}
|
|
|
|
|
2015-04-12 15:21:35 +08:00
|
|
|
static unsigned g_magicRead = 0;
|
2014-02-04 22:11:10 +08:00
|
|
|
static unsigned long long decodeLegacyStream(FILE* finput, FILE* foutput)
|
|
|
|
{
|
|
|
|
unsigned long long filesize = 0;
|
|
|
|
char* in_buff;
|
|
|
|
char* out_buff;
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Allocate Memory */
|
2014-02-04 22:11:10 +08:00
|
|
|
in_buff = (char*)malloc(LZ4_compressBound(LEGACY_BLOCKSIZE));
|
|
|
|
out_buff = (char*)malloc(LEGACY_BLOCKSIZE);
|
|
|
|
if (!in_buff || !out_buff) EXM_THROW(51, "Allocation error : not enough memory");
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Main Loop */
|
2014-02-04 22:11:10 +08:00
|
|
|
while (1)
|
|
|
|
{
|
|
|
|
int decodeSize;
|
|
|
|
size_t sizeCheck;
|
2014-12-07 00:10:54 +08:00
|
|
|
unsigned int blockSize;
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Block Size */
|
2014-12-01 00:59:31 +08:00
|
|
|
sizeCheck = fread(in_buff, 1, 4, finput);
|
2014-12-17 05:03:16 +08:00
|
|
|
if (sizeCheck==0) break; /* Nothing to read : file read is completed */
|
|
|
|
blockSize = LZ4IO_readLE32(in_buff); /* Convert to Little Endian */
|
2014-02-04 22:11:10 +08:00
|
|
|
if (blockSize > LZ4_COMPRESSBOUND(LEGACY_BLOCKSIZE))
|
2014-12-17 05:03:16 +08:00
|
|
|
{ /* Cannot read next block : maybe new stream ? */
|
2015-04-12 15:21:35 +08:00
|
|
|
g_magicRead = blockSize;
|
2014-02-04 22:11:10 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Read Block */
|
2014-02-04 22:11:10 +08:00
|
|
|
sizeCheck = fread(in_buff, 1, blockSize, finput);
|
2015-03-15 08:42:27 +08:00
|
|
|
if (sizeCheck!=blockSize) EXM_THROW(52, "Read error : cannot access compressed block !");
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Decode Block */
|
2014-02-04 22:11:10 +08:00
|
|
|
decodeSize = LZ4_decompress_safe(in_buff, out_buff, blockSize, LEGACY_BLOCKSIZE);
|
2015-03-15 08:42:27 +08:00
|
|
|
if (decodeSize < 0) EXM_THROW(53, "Decoding Failed ! Corrupted input detected !");
|
2014-02-04 22:11:10 +08:00
|
|
|
filesize += decodeSize;
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Write Block */
|
2014-02-04 22:11:10 +08:00
|
|
|
sizeCheck = fwrite(out_buff, 1, decodeSize, foutput);
|
2015-03-15 08:42:27 +08:00
|
|
|
if (sizeCheck != (size_t)decodeSize) EXM_THROW(54, "Write error : cannot write decoded block into output\n");
|
2014-02-04 22:11:10 +08:00
|
|
|
}
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Free */
|
2014-02-04 22:11:10 +08:00
|
|
|
free(in_buff);
|
|
|
|
free(out_buff);
|
|
|
|
|
|
|
|
return filesize;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static unsigned long long decodeLZ4S(FILE* finput, FILE* foutput)
|
|
|
|
{
|
|
|
|
unsigned long long filesize = 0;
|
2015-03-12 02:42:37 +08:00
|
|
|
void* inBuff;
|
|
|
|
void* outBuff;
|
2015-03-23 09:20:42 +08:00
|
|
|
size_t sizeCheck;
|
|
|
|
const size_t inBuffSize = 256 KB;
|
|
|
|
const size_t outBuffSize = 256 KB;
|
2014-12-01 00:59:31 +08:00
|
|
|
LZ4F_decompressionContext_t ctx;
|
|
|
|
LZ4F_errorCode_t errorCode;
|
2015-03-13 09:24:08 +08:00
|
|
|
unsigned storedSkips = 0;
|
2014-06-18 04:41:59 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* init */
|
2014-12-01 00:59:31 +08:00
|
|
|
errorCode = LZ4F_createDecompressionContext(&ctx, LZ4F_VERSION);
|
2015-03-23 09:20:42 +08:00
|
|
|
if (LZ4F_isError(errorCode)) EXM_THROW(60, "Can't create context : %s", LZ4F_getErrorName(errorCode));
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Allocate Memory */
|
2015-03-23 09:20:42 +08:00
|
|
|
inBuff = malloc(256 KB);
|
|
|
|
outBuff = malloc(256 KB);
|
|
|
|
if (!inBuff || !outBuff) EXM_THROW(61, "Allocation error : not enough memory");
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2015-03-23 09:20:42 +08:00
|
|
|
/* Init feed with magic number (already consumed from FILE) */
|
2014-02-04 22:11:10 +08:00
|
|
|
{
|
2015-03-23 09:20:42 +08:00
|
|
|
size_t inSize = 4;
|
|
|
|
size_t outSize=0;
|
|
|
|
LZ4IO_writeLE32(inBuff, LZ4IO_MAGICNUMBER);
|
|
|
|
errorCode = LZ4F_decompress(ctx, outBuff, &outSize, inBuff, &inSize, NULL);
|
|
|
|
if (LZ4F_isError(errorCode)) EXM_THROW(62, "Header error : %s", LZ4F_getErrorName(errorCode));
|
|
|
|
}
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-06-18 04:41:59 +08:00
|
|
|
|
2015-03-23 09:20:42 +08:00
|
|
|
/* Main Loop */
|
|
|
|
for (;;)
|
|
|
|
{
|
|
|
|
size_t readSize;
|
|
|
|
size_t pos = 0;
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2015-03-23 09:20:42 +08:00
|
|
|
/* Read input */
|
|
|
|
readSize = fread(inBuff, 1, inBuffSize, finput);
|
|
|
|
if (!readSize) break; /* empty file or stream */
|
|
|
|
|
|
|
|
while (pos < readSize)
|
2015-03-12 02:42:37 +08:00
|
|
|
{
|
2015-03-23 09:20:42 +08:00
|
|
|
/* Decode Input (at least partially) */
|
|
|
|
size_t remaining = readSize - pos;
|
|
|
|
size_t decodedBytes = outBuffSize;
|
|
|
|
errorCode = LZ4F_decompress(ctx, outBuff, &decodedBytes, (char*)inBuff+pos, &remaining, NULL);
|
|
|
|
if (LZ4F_isError(errorCode)) EXM_THROW(66, "Decompression error : %s", LZ4F_getErrorName(errorCode));
|
|
|
|
pos += remaining;
|
|
|
|
|
|
|
|
if (decodedBytes)
|
2015-03-13 09:24:08 +08:00
|
|
|
{
|
2015-03-23 09:20:42 +08:00
|
|
|
/* Write Block */
|
|
|
|
filesize += decodedBytes;
|
2015-04-02 20:25:28 +08:00
|
|
|
DISPLAYUPDATE(2, "\rDecompressed : %u MB ", (unsigned)(filesize>>20));
|
2015-03-23 09:20:42 +08:00
|
|
|
if (g_sparseFileSupport)
|
2015-03-14 03:36:59 +08:00
|
|
|
{
|
2015-03-23 09:20:42 +08:00
|
|
|
size_t* const oBuffStartT = (size_t*)outBuff; /* since outBuff is malloc'ed, it's aligned on size_t */
|
|
|
|
size_t* oBuffPosT = oBuffStartT;
|
|
|
|
size_t oBuffSizeT = decodedBytes / sizeT;
|
|
|
|
size_t* const oBuffEndT = oBuffStartT + oBuffSizeT;
|
|
|
|
static const size_t bs0T = (32 KB) / sizeT;
|
|
|
|
while (oBuffPosT < oBuffEndT)
|
|
|
|
{
|
|
|
|
size_t seg0SizeT = bs0T;
|
|
|
|
size_t nb0T;
|
|
|
|
int seekResult;
|
|
|
|
if (seg0SizeT > oBuffSizeT) seg0SizeT = oBuffSizeT;
|
|
|
|
oBuffSizeT -= seg0SizeT;
|
|
|
|
for (nb0T=0; (nb0T < seg0SizeT) && (oBuffPosT[nb0T] == 0); nb0T++) ;
|
|
|
|
storedSkips += (unsigned)(nb0T * sizeT);
|
|
|
|
if (storedSkips > 1 GB) /* avoid int overflow */
|
|
|
|
{
|
|
|
|
seekResult = fseek(foutput, 1 GB, SEEK_CUR);
|
|
|
|
if (seekResult != 0) EXM_THROW(68, "1 GB skip error (sparse file)");
|
|
|
|
storedSkips -= 1 GB;
|
|
|
|
}
|
|
|
|
if (nb0T != seg0SizeT) /* not all 0s */
|
|
|
|
{
|
|
|
|
seekResult = fseek(foutput, storedSkips, SEEK_CUR);
|
|
|
|
if (seekResult) EXM_THROW(68, "Skip error (sparse file)");
|
|
|
|
storedSkips = 0;
|
|
|
|
seg0SizeT -= nb0T;
|
|
|
|
oBuffPosT += nb0T;
|
|
|
|
sizeCheck = fwrite(oBuffPosT, sizeT, seg0SizeT, foutput);
|
|
|
|
if (sizeCheck != seg0SizeT) EXM_THROW(68, "Write error : cannot write decoded block");
|
|
|
|
}
|
|
|
|
oBuffPosT += seg0SizeT;
|
|
|
|
}
|
|
|
|
if (decodedBytes & maskT) /* size not multiple of sizeT (necessarily end of block) */
|
|
|
|
{
|
|
|
|
const char* const restStart = (char*)oBuffEndT;
|
|
|
|
const char* restPtr = restStart;
|
|
|
|
size_t restSize = decodedBytes & maskT;
|
|
|
|
const char* const restEnd = restStart + restSize;
|
|
|
|
for (; (restPtr < restEnd) && (*restPtr == 0); restPtr++) ;
|
|
|
|
storedSkips += (unsigned) (restPtr - restStart);
|
|
|
|
if (restPtr != restEnd)
|
|
|
|
{
|
|
|
|
int seekResult = fseek(foutput, storedSkips, SEEK_CUR);
|
|
|
|
if (seekResult) EXM_THROW(68, "Skip error (end of block)");
|
|
|
|
storedSkips = 0;
|
|
|
|
sizeCheck = fwrite(restPtr, 1, restEnd - restPtr, foutput);
|
|
|
|
if (sizeCheck != (size_t)(restEnd - restPtr)) EXM_THROW(68, "Write error : cannot write decoded end of block");
|
|
|
|
}
|
|
|
|
}
|
2015-03-14 03:36:59 +08:00
|
|
|
}
|
2015-03-23 09:20:42 +08:00
|
|
|
else
|
2015-03-14 03:36:59 +08:00
|
|
|
{
|
2015-03-23 09:20:42 +08:00
|
|
|
sizeCheck = fwrite(outBuff, 1, decodedBytes, foutput);
|
|
|
|
if (sizeCheck != decodedBytes) EXM_THROW(68, "Write error : cannot write decoded block");
|
2015-03-14 03:36:59 +08:00
|
|
|
}
|
2015-03-13 09:24:08 +08:00
|
|
|
}
|
2015-03-12 02:42:37 +08:00
|
|
|
}
|
2015-03-23 09:20:42 +08:00
|
|
|
|
2015-03-13 09:24:08 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
if ((g_sparseFileSupport) && (storedSkips>0))
|
|
|
|
{
|
|
|
|
int seekResult;
|
|
|
|
storedSkips --;
|
|
|
|
seekResult = fseek(foutput, storedSkips, SEEK_CUR);
|
2015-03-14 03:36:59 +08:00
|
|
|
if (seekResult != 0) EXM_THROW(69, "Final skip error (sparse file)\n");
|
2015-03-13 09:24:08 +08:00
|
|
|
memset(outBuff, 0, 1);
|
|
|
|
sizeCheck = fwrite(outBuff, 1, 1, foutput);
|
2015-03-14 03:36:59 +08:00
|
|
|
if (sizeCheck != 1) EXM_THROW(69, "Write error : cannot write last zero\n");
|
2014-02-04 22:11:10 +08:00
|
|
|
}
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Free */
|
2014-12-01 00:59:31 +08:00
|
|
|
free(inBuff);
|
|
|
|
free(outBuff);
|
|
|
|
errorCode = LZ4F_freeDecompressionContext(ctx);
|
|
|
|
if (LZ4F_isError(errorCode)) EXM_THROW(69, "Error : can't free LZ4F context resource : %s", LZ4F_getErrorName(errorCode));
|
2014-02-04 22:11:10 +08:00
|
|
|
|
|
|
|
return filesize;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-03-17 03:38:25 +08:00
|
|
|
static unsigned long long LZ4IO_passThrough(FILE* finput, FILE* foutput, unsigned char U32store[MAGICNUMBER_SIZE])
|
|
|
|
{
|
|
|
|
void* buffer = malloc(64 KB);
|
|
|
|
size_t read = 1, sizeCheck;
|
|
|
|
unsigned long long total = MAGICNUMBER_SIZE;
|
|
|
|
|
|
|
|
sizeCheck = fwrite(U32store, 1, MAGICNUMBER_SIZE, foutput);
|
|
|
|
if (sizeCheck != MAGICNUMBER_SIZE) EXM_THROW(50, "Pass-through error at start");
|
|
|
|
|
|
|
|
while (read)
|
|
|
|
{
|
|
|
|
read = fread(buffer, 1, 64 KB, finput);
|
|
|
|
total += read;
|
|
|
|
sizeCheck = fwrite(buffer, 1, read, foutput);
|
|
|
|
if (sizeCheck != read) EXM_THROW(50, "Pass-through error");
|
|
|
|
}
|
|
|
|
|
|
|
|
free(buffer);
|
|
|
|
return total;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2014-11-04 19:11:14 +08:00
|
|
|
#define ENDOFSTREAM ((unsigned long long)-1)
|
2014-02-04 22:11:10 +08:00
|
|
|
static unsigned long long selectDecoder( FILE* finput, FILE* foutput)
|
|
|
|
{
|
2014-12-01 00:59:31 +08:00
|
|
|
unsigned char U32store[MAGICNUMBER_SIZE];
|
|
|
|
unsigned magicNumber, size;
|
2014-02-04 22:11:10 +08:00
|
|
|
int errorNb;
|
|
|
|
size_t nbReadBytes;
|
2015-03-17 03:38:25 +08:00
|
|
|
static unsigned nbCalls = 0;
|
|
|
|
|
|
|
|
/* init */
|
|
|
|
nbCalls++;
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Check Archive Header */
|
2015-04-12 15:21:35 +08:00
|
|
|
if (g_magicRead)
|
|
|
|
{
|
|
|
|
magicNumber = g_magicRead;
|
|
|
|
g_magicRead = 0;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
nbReadBytes = fread(U32store, 1, MAGICNUMBER_SIZE, finput);
|
|
|
|
if (nbReadBytes==0) return ENDOFSTREAM; /* EOF */
|
|
|
|
if (nbReadBytes != MAGICNUMBER_SIZE) EXM_THROW(40, "Unrecognized header : Magic Number unreadable");
|
|
|
|
magicNumber = LZ4IO_readLE32(U32store); /* Little Endian format */
|
|
|
|
}
|
2015-03-18 01:02:01 +08:00
|
|
|
if (LZ4IO_isSkippableMagicNumber(magicNumber)) magicNumber = LZ4IO_SKIPPABLE0; /* fold skippable magic numbers */
|
2014-02-04 22:11:10 +08:00
|
|
|
|
|
|
|
switch(magicNumber)
|
|
|
|
{
|
2015-03-18 01:02:01 +08:00
|
|
|
case LZ4IO_MAGICNUMBER:
|
2014-02-04 22:11:10 +08:00
|
|
|
return DEFAULT_DECOMPRESSOR(finput, foutput);
|
|
|
|
case LEGACY_MAGICNUMBER:
|
|
|
|
DISPLAYLEVEL(4, "Detected : Legacy format \n");
|
|
|
|
return decodeLegacyStream(finput, foutput);
|
2015-03-18 01:02:01 +08:00
|
|
|
case LZ4IO_SKIPPABLE0:
|
2014-02-04 22:11:10 +08:00
|
|
|
DISPLAYLEVEL(4, "Skipping detected skippable area \n");
|
2014-12-01 00:59:31 +08:00
|
|
|
nbReadBytes = fread(U32store, 1, 4, finput);
|
2014-02-04 22:11:10 +08:00
|
|
|
if (nbReadBytes != 4) EXM_THROW(42, "Stream error : skippable size unreadable");
|
2014-12-17 05:03:16 +08:00
|
|
|
size = LZ4IO_readLE32(U32store); /* Little Endian format */
|
2014-02-04 22:11:10 +08:00
|
|
|
errorNb = fseek(finput, size, SEEK_CUR);
|
|
|
|
if (errorNb != 0) EXM_THROW(43, "Stream error : cannot skip skippable area");
|
|
|
|
return selectDecoder(finput, foutput);
|
|
|
|
EXTENDED_FORMAT;
|
|
|
|
default:
|
2015-03-17 03:38:25 +08:00
|
|
|
if (nbCalls == 1) /* just started */
|
|
|
|
{
|
|
|
|
if (g_overwrite)
|
|
|
|
return LZ4IO_passThrough(finput, foutput, U32store);
|
|
|
|
EXM_THROW(44,"Unrecognized header : file cannot be decoded"); /* Wrong magic number at the beginning of 1st stream */
|
|
|
|
}
|
2014-02-04 22:11:10 +08:00
|
|
|
DISPLAYLEVEL(2, "Stream followed by unrecognized data\n");
|
2014-11-04 19:11:14 +08:00
|
|
|
return ENDOFSTREAM;
|
2014-02-04 22:11:10 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-03-07 20:23:00 +08:00
|
|
|
int LZ4IO_decompressFilename(const char* input_filename, const char* output_filename)
|
2014-02-04 22:11:10 +08:00
|
|
|
{
|
|
|
|
unsigned long long filesize = 0, decodedSize=0;
|
|
|
|
FILE* finput;
|
|
|
|
FILE* foutput;
|
|
|
|
clock_t start, end;
|
|
|
|
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Init */
|
2014-02-04 22:11:10 +08:00
|
|
|
start = clock();
|
|
|
|
get_fileHandle(input_filename, output_filename, &finput, &foutput);
|
|
|
|
|
2015-03-14 05:15:08 +08:00
|
|
|
/* sparse file */
|
2015-04-12 15:21:35 +08:00
|
|
|
if (g_sparseFileSupport) { SET_SPARSE_FILE_MODE(foutput); }
|
2015-03-14 05:15:08 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Loop over multiple streams */
|
2014-02-04 22:11:10 +08:00
|
|
|
do
|
|
|
|
{
|
|
|
|
decodedSize = selectDecoder(finput, foutput);
|
2014-11-04 19:11:14 +08:00
|
|
|
if (decodedSize != ENDOFSTREAM)
|
|
|
|
filesize += decodedSize;
|
|
|
|
} while (decodedSize != ENDOFSTREAM);
|
2014-02-04 22:11:10 +08:00
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Final Status */
|
2014-02-04 22:11:10 +08:00
|
|
|
end = clock();
|
|
|
|
DISPLAYLEVEL(2, "\r%79s\r", "");
|
|
|
|
DISPLAYLEVEL(2, "Successfully decoded %llu bytes \n", filesize);
|
2015-04-11 17:25:08 +08:00
|
|
|
if (end==start) end=start+1;
|
2014-02-04 22:11:10 +08:00
|
|
|
{
|
|
|
|
double seconds = (double)(end - start)/CLOCKS_PER_SEC;
|
|
|
|
DISPLAYLEVEL(4, "Done in %.2f s ==> %.2f MB/s\n", seconds, (double)filesize / seconds / 1024 / 1024);
|
|
|
|
}
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Close */
|
2014-02-04 22:11:10 +08:00
|
|
|
fclose(finput);
|
|
|
|
fclose(foutput);
|
|
|
|
|
2014-12-17 05:03:16 +08:00
|
|
|
/* Error status = OK */
|
2014-02-04 22:11:10 +08:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|