2015-11-11 04:43:58 -08:00
|
|
|
/*
|
|
|
|
zstd - standard compression library
|
2016-02-02 17:46:46 -08:00
|
|
|
Copyright (C) 2014-2016, Yann Collet.
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
BSD 2-Clause License (http://www.opensource.org/licenses/bsd-license.php)
|
|
|
|
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
|
|
modification, are permitted provided that the following conditions are
|
|
|
|
met:
|
|
|
|
* Redistributions of source code must retain the above copyright
|
|
|
|
notice, this list of conditions and the following disclaimer.
|
|
|
|
* Redistributions in binary form must reproduce the above
|
|
|
|
copyright notice, this list of conditions and the following disclaimer
|
|
|
|
in the documentation and/or other materials provided with the
|
|
|
|
distribution.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
|
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
|
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
|
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
|
|
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
|
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
|
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
|
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
|
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
|
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
|
|
|
|
You can contact the author at :
|
2016-03-19 06:14:31 -07:00
|
|
|
- zstd homepage : http://www.zstd.net
|
2015-11-11 04:43:58 -08:00
|
|
|
*/
|
|
|
|
|
|
|
|
/* ***************************************************************
|
|
|
|
* Tuning parameters
|
|
|
|
*****************************************************************/
|
|
|
|
/*!
|
|
|
|
* HEAPMODE :
|
2016-01-11 03:56:11 -08:00
|
|
|
* Select how default decompression function ZSTD_decompress() will allocate memory,
|
|
|
|
* in memory stack (0), or in memory heap (1, requires malloc())
|
2015-11-11 04:43:58 -08:00
|
|
|
*/
|
|
|
|
#ifndef ZSTD_HEAPMODE
|
|
|
|
# define ZSTD_HEAPMODE 1
|
2016-01-11 03:56:11 -08:00
|
|
|
#endif
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
/*!
|
|
|
|
* LEGACY_SUPPORT :
|
2016-02-02 17:46:46 -08:00
|
|
|
* if set to 1, ZSTD_decompress() can decode older formats (v0.1+)
|
2015-11-11 04:43:58 -08:00
|
|
|
*/
|
|
|
|
#ifndef ZSTD_LEGACY_SUPPORT
|
2016-01-18 03:03:27 -08:00
|
|
|
# define ZSTD_LEGACY_SUPPORT 0
|
2015-11-11 04:43:58 -08:00
|
|
|
#endif
|
|
|
|
|
|
|
|
|
2016-01-27 15:18:06 -08:00
|
|
|
/*-*******************************************************
|
2016-02-04 06:28:14 -08:00
|
|
|
* Dependencies
|
2015-11-11 04:43:58 -08:00
|
|
|
*********************************************************/
|
|
|
|
#include <string.h> /* memcpy, memmove */
|
2016-02-04 06:28:14 -08:00
|
|
|
#include <stdio.h> /* debug only : printf */
|
2015-11-11 04:43:58 -08:00
|
|
|
#include "mem.h" /* low level memory routines */
|
2016-05-31 09:13:56 -07:00
|
|
|
#define XXH_STATIC_LINKING_ONLY /* XXH64_state_t */
|
|
|
|
#include "xxhash.h" /* XXH64_* */
|
2015-11-11 04:43:58 -08:00
|
|
|
#include "zstd_internal.h"
|
|
|
|
#include "fse_static.h"
|
2016-04-22 03:43:18 -07:00
|
|
|
#include "huf_static.h"
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
#if defined(ZSTD_LEGACY_SUPPORT) && (ZSTD_LEGACY_SUPPORT==1)
|
|
|
|
# include "zstd_legacy.h"
|
|
|
|
#endif
|
|
|
|
|
2016-01-27 15:18:06 -08:00
|
|
|
|
|
|
|
/*-*******************************************************
|
2015-11-11 04:43:58 -08:00
|
|
|
* Compiler specifics
|
|
|
|
*********************************************************/
|
|
|
|
#ifdef _MSC_VER /* Visual Studio */
|
|
|
|
# define FORCE_INLINE static __forceinline
|
|
|
|
# include <intrin.h> /* For Visual 2005 */
|
|
|
|
# pragma warning(disable : 4127) /* disable: C4127: conditional expression is constant */
|
|
|
|
# pragma warning(disable : 4324) /* disable: C4324: padded structure */
|
|
|
|
#else
|
|
|
|
# ifdef __GNUC__
|
|
|
|
# define FORCE_INLINE static inline __attribute__((always_inline))
|
|
|
|
# else
|
|
|
|
# define FORCE_INLINE static inline
|
|
|
|
# endif
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
2016-05-12 06:55:26 -07:00
|
|
|
/*-*************************************
|
|
|
|
* Macros
|
|
|
|
***************************************/
|
|
|
|
#define ZSTD_isError ERR_isError /* for inlining */
|
|
|
|
#define FSE_isError ERR_isError
|
|
|
|
#define HUF_isError ERR_isError
|
|
|
|
|
|
|
|
|
2016-03-11 16:25:40 -08:00
|
|
|
/*_*******************************************************
|
2015-11-11 04:43:58 -08:00
|
|
|
* Memory operations
|
|
|
|
**********************************************************/
|
|
|
|
static void ZSTD_copy4(void* dst, const void* src) { memcpy(dst, src, 4); }
|
|
|
|
|
|
|
|
|
2016-03-11 16:25:40 -08:00
|
|
|
/*-*************************************************************
|
2015-11-12 06:36:05 -08:00
|
|
|
* Context management
|
2015-11-11 04:43:58 -08:00
|
|
|
***************************************************************/
|
2015-11-25 12:09:17 -08:00
|
|
|
typedef enum { ZSTDds_getFrameHeaderSize, ZSTDds_decodeFrameHeader,
|
2016-05-31 03:43:46 -07:00
|
|
|
ZSTDds_decodeBlockHeader, ZSTDds_decompressBlock,
|
|
|
|
ZSTDds_decodeSkippableHeader, ZSTDds_skipFrame } ZSTD_dStage;
|
2015-11-25 05:42:45 -08:00
|
|
|
|
2015-11-11 04:43:58 -08:00
|
|
|
struct ZSTD_DCtx_s
|
|
|
|
{
|
2016-01-27 15:18:06 -08:00
|
|
|
FSE_DTable LLTable[FSE_DTABLE_SIZE_U32(LLFSELog)];
|
|
|
|
FSE_DTable OffTable[FSE_DTABLE_SIZE_U32(OffFSELog)];
|
|
|
|
FSE_DTable MLTable[FSE_DTABLE_SIZE_U32(MLFSELog)];
|
|
|
|
unsigned hufTableX4[HUF_DTABLE_SIZE(HufLog)];
|
2015-12-04 08:16:37 -08:00
|
|
|
const void* previousDstEnd;
|
|
|
|
const void* base;
|
|
|
|
const void* vBase;
|
|
|
|
const void* dictEnd;
|
2015-11-11 04:43:58 -08:00
|
|
|
size_t expected;
|
2015-11-25 05:42:45 -08:00
|
|
|
size_t headerSize;
|
2016-03-11 12:58:04 -08:00
|
|
|
ZSTD_frameParams fParams;
|
2016-05-31 09:13:56 -07:00
|
|
|
XXH64_state_t xxhState;
|
2016-06-02 04:04:18 -07:00
|
|
|
ZSTD_customMem customMem;
|
2016-01-27 15:18:06 -08:00
|
|
|
blockType_t bType; /* used in ZSTD_decompressContinue(), to transfer blockType between header decoding and block decoding stages */
|
2015-11-25 05:42:45 -08:00
|
|
|
ZSTD_dStage stage;
|
2016-05-28 20:01:04 -07:00
|
|
|
U32 dictID;
|
2016-04-09 08:26:22 -07:00
|
|
|
U32 flagRepeatTable;
|
2015-11-11 04:43:58 -08:00
|
|
|
const BYTE* litPtr;
|
|
|
|
size_t litBufSize;
|
|
|
|
size_t litSize;
|
2016-03-16 07:26:51 -07:00
|
|
|
BYTE litBuffer[ZSTD_BLOCKSIZE_MAX + WILDCOPY_OVERLENGTH];
|
2016-03-11 12:58:04 -08:00
|
|
|
BYTE headerBuffer[ZSTD_FRAMEHEADERSIZE_MAX];
|
2015-12-04 08:16:37 -08:00
|
|
|
}; /* typedef'd to ZSTD_DCtx within "zstd_static.h" */
|
2015-11-11 04:43:58 -08:00
|
|
|
|
2016-04-09 00:42:27 -07:00
|
|
|
size_t ZSTD_sizeofDCtx (void) { return sizeof(ZSTD_DCtx); } /* non published interface */
|
2016-01-27 15:18:06 -08:00
|
|
|
|
2016-01-26 06:58:49 -08:00
|
|
|
size_t ZSTD_decompressBegin(ZSTD_DCtx* dctx)
|
2015-11-12 06:36:05 -08:00
|
|
|
{
|
2015-11-25 05:42:45 -08:00
|
|
|
dctx->expected = ZSTD_frameHeaderSize_min;
|
|
|
|
dctx->stage = ZSTDds_getFrameHeaderSize;
|
2015-11-12 06:36:05 -08:00
|
|
|
dctx->previousDstEnd = NULL;
|
|
|
|
dctx->base = NULL;
|
|
|
|
dctx->vBase = NULL;
|
|
|
|
dctx->dictEnd = NULL;
|
2016-01-25 18:14:20 -08:00
|
|
|
dctx->hufTableX4[0] = HufLog;
|
2016-04-09 08:26:22 -07:00
|
|
|
dctx->flagRepeatTable = 0;
|
2016-05-28 20:01:04 -07:00
|
|
|
dctx->dictID = 0;
|
2015-11-12 06:36:05 -08:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2016-05-23 07:24:52 -07:00
|
|
|
ZSTD_DCtx* ZSTD_createDCtx_advanced(ZSTD_customMem customMem)
|
|
|
|
{
|
|
|
|
ZSTD_DCtx* dctx;
|
|
|
|
|
2016-06-03 04:23:04 -07:00
|
|
|
if (!customMem.customAlloc && !customMem.customFree)
|
|
|
|
customMem = defaultCustomMem;
|
2016-05-23 07:24:52 -07:00
|
|
|
|
2016-05-23 08:04:23 -07:00
|
|
|
if (!customMem.customAlloc || !customMem.customFree)
|
|
|
|
return NULL;
|
|
|
|
|
2016-06-02 04:04:18 -07:00
|
|
|
dctx = (ZSTD_DCtx*) customMem.customAlloc(customMem.opaque, sizeof(ZSTD_DCtx));
|
2016-05-23 07:24:52 -07:00
|
|
|
if (!dctx) return NULL;
|
2016-06-02 04:04:18 -07:00
|
|
|
memcpy(&dctx->customMem, &customMem, sizeof(ZSTD_customMem));
|
2016-01-26 06:58:49 -08:00
|
|
|
ZSTD_decompressBegin(dctx);
|
2015-11-12 06:36:05 -08:00
|
|
|
return dctx;
|
|
|
|
}
|
|
|
|
|
2016-05-31 15:18:28 -07:00
|
|
|
ZSTD_DCtx* ZSTD_createDCtx(void)
|
|
|
|
{
|
2016-06-03 04:28:20 -07:00
|
|
|
return ZSTD_createDCtx_advanced(defaultCustomMem);
|
2016-05-31 15:18:28 -07:00
|
|
|
}
|
|
|
|
|
2016-05-23 07:24:52 -07:00
|
|
|
|
2015-11-12 06:36:05 -08:00
|
|
|
size_t ZSTD_freeDCtx(ZSTD_DCtx* dctx)
|
|
|
|
{
|
2016-06-02 04:04:18 -07:00
|
|
|
dctx->customMem.customFree(dctx->customMem.opaque, dctx);
|
2016-02-04 17:33:10 -08:00
|
|
|
return 0; /* reserved as a potential error code in the future */
|
2015-11-12 06:36:05 -08:00
|
|
|
}
|
|
|
|
|
2016-01-26 06:58:49 -08:00
|
|
|
void ZSTD_copyDCtx(ZSTD_DCtx* dstDCtx, const ZSTD_DCtx* srcDCtx)
|
|
|
|
{
|
|
|
|
memcpy(dstDCtx, srcDCtx,
|
2016-03-16 07:26:51 -07:00
|
|
|
sizeof(ZSTD_DCtx) - (ZSTD_BLOCKSIZE_MAX+WILDCOPY_OVERLENGTH + ZSTD_frameHeaderSize_max)); /* no need to copy workspace */
|
2016-01-26 06:58:49 -08:00
|
|
|
}
|
|
|
|
|
2015-11-12 06:36:05 -08:00
|
|
|
|
2016-03-11 16:25:40 -08:00
|
|
|
/*-*************************************************************
|
2015-11-12 06:36:05 -08:00
|
|
|
* Decompression section
|
|
|
|
***************************************************************/
|
2016-01-23 10:28:41 -08:00
|
|
|
|
|
|
|
/* Frame format description
|
|
|
|
Frame Header - [ Block Header - Block ] - Frame End
|
|
|
|
1) Frame Header
|
2016-03-11 16:25:40 -08:00
|
|
|
- 4 bytes - Magic Number : ZSTD_MAGICNUMBER (defined within zstd_static.h)
|
|
|
|
- 1 byte - Frame Descriptor
|
2016-01-23 10:28:41 -08:00
|
|
|
2) Block Header
|
|
|
|
- 3 bytes, starting with a 2-bits descriptor
|
|
|
|
Uncompressed, Compressed, Frame End, unused
|
|
|
|
3) Block
|
|
|
|
See Block Format Description
|
|
|
|
4) Frame End
|
|
|
|
- 3 bytes, compatible with Block Header
|
|
|
|
*/
|
|
|
|
|
2016-03-11 16:25:40 -08:00
|
|
|
|
|
|
|
/* Frame descriptor
|
|
|
|
|
2016-05-28 20:01:04 -07:00
|
|
|
1 byte - Alloc :
|
2016-03-11 16:25:40 -08:00
|
|
|
bit 0-3 : windowLog - ZSTD_WINDOWLOG_ABSOLUTEMIN (see zstd_internal.h)
|
2016-05-28 20:01:04 -07:00
|
|
|
bit 4 : reserved for windowLog (must be zero)
|
2016-03-11 16:25:40 -08:00
|
|
|
bit 5 : reserved (must be zero)
|
|
|
|
bit 6-7 : Frame content size : unknown, 1 byte, 2 bytes, 8 bytes
|
2016-03-13 03:08:40 -07:00
|
|
|
|
2016-05-28 20:01:04 -07:00
|
|
|
1 byte - checker :
|
|
|
|
bit 0-1 : dictID (0, 1, 2 or 4 bytes)
|
|
|
|
bit 2-7 : reserved (must be zero)
|
|
|
|
|
|
|
|
Optional : dictID (0, 1, 2 or 4 bytes)
|
|
|
|
Automatic adaptation
|
|
|
|
0 : no dictID
|
|
|
|
1 : 1 - 255
|
|
|
|
2 : 256 - 65535
|
|
|
|
4 : all other values
|
|
|
|
|
2016-03-13 03:08:40 -07:00
|
|
|
Optional : content size (0, 1, 2 or 8 bytes)
|
|
|
|
0 : unknown
|
|
|
|
1 : 0-255 bytes
|
|
|
|
2 : 256 - 65535+256
|
|
|
|
8 : up to 16 exa
|
2016-03-11 16:25:40 -08:00
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
|
|
/* Compressed Block, format description
|
2016-01-27 15:18:06 -08:00
|
|
|
|
|
|
|
Block = Literal Section - Sequences Section
|
|
|
|
Prerequisite : size of (compressed) block, maximum size of regenerated data
|
|
|
|
|
2016-01-23 10:28:41 -08:00
|
|
|
1) Literal Section
|
2016-01-27 15:18:06 -08:00
|
|
|
|
|
|
|
1.1) Header : 1-5 bytes
|
|
|
|
flags: 2 bits
|
2016-01-23 10:28:41 -08:00
|
|
|
00 compressed by Huff0
|
2016-01-27 15:18:06 -08:00
|
|
|
01 unused
|
|
|
|
10 is Raw (uncompressed)
|
|
|
|
11 is Rle
|
|
|
|
Note : using 01 => Huff0 with precomputed table ?
|
2016-01-23 10:28:41 -08:00
|
|
|
Note : delta map ? => compressed ?
|
2016-01-27 15:18:06 -08:00
|
|
|
|
|
|
|
1.1.1) Huff0-compressed literal block : 3-5 bytes
|
|
|
|
srcSize < 1 KB => 3 bytes (2-2-10-10) => single stream
|
|
|
|
srcSize < 1 KB => 3 bytes (2-2-10-10)
|
|
|
|
srcSize < 16KB => 4 bytes (2-2-14-14)
|
|
|
|
else => 5 bytes (2-2-18-18)
|
|
|
|
big endian convention
|
|
|
|
|
|
|
|
1.1.2) Raw (uncompressed) literal block header : 1-3 bytes
|
|
|
|
size : 5 bits: (IS_RAW<<6) + (0<<4) + size
|
|
|
|
12 bits: (IS_RAW<<6) + (2<<4) + (size>>8)
|
|
|
|
size&255
|
|
|
|
20 bits: (IS_RAW<<6) + (3<<4) + (size>>16)
|
|
|
|
size>>8&255
|
|
|
|
size&255
|
|
|
|
|
|
|
|
1.1.3) Rle (repeated single byte) literal block header : 1-3 bytes
|
|
|
|
size : 5 bits: (IS_RLE<<6) + (0<<4) + size
|
|
|
|
12 bits: (IS_RLE<<6) + (2<<4) + (size>>8)
|
|
|
|
size&255
|
|
|
|
20 bits: (IS_RLE<<6) + (3<<4) + (size>>16)
|
|
|
|
size>>8&255
|
|
|
|
size&255
|
|
|
|
|
|
|
|
1.1.4) Huff0-compressed literal block, using precomputed CTables : 3-5 bytes
|
|
|
|
srcSize < 1 KB => 3 bytes (2-2-10-10) => single stream
|
|
|
|
srcSize < 1 KB => 3 bytes (2-2-10-10)
|
|
|
|
srcSize < 16KB => 4 bytes (2-2-14-14)
|
|
|
|
else => 5 bytes (2-2-18-18)
|
|
|
|
big endian convention
|
|
|
|
|
|
|
|
1- CTable available (stored into workspace ?)
|
|
|
|
2- Small input (fast heuristic ? Full comparison ? depend on clevel ?)
|
|
|
|
|
|
|
|
|
|
|
|
1.2) Literal block content
|
2016-01-23 10:28:41 -08:00
|
|
|
|
|
|
|
1.2.1) Huff0 block, using sizes from header
|
|
|
|
See Huff0 format
|
|
|
|
|
2016-01-27 15:18:06 -08:00
|
|
|
1.2.2) Huff0 block, using prepared table
|
|
|
|
|
|
|
|
1.2.3) Raw content
|
|
|
|
|
|
|
|
1.2.4) single byte
|
2016-01-23 10:28:41 -08:00
|
|
|
|
|
|
|
|
|
|
|
2) Sequences section
|
|
|
|
TO DO
|
|
|
|
*/
|
|
|
|
|
2016-03-11 16:25:40 -08:00
|
|
|
/** ZSTD_frameHeaderSize() :
|
|
|
|
* srcSize must be >= ZSTD_frameHeaderSize_min.
|
|
|
|
* @return : size of the Frame Header */
|
|
|
|
static size_t ZSTD_frameHeaderSize(const void* src, size_t srcSize)
|
2015-11-25 05:42:45 -08:00
|
|
|
{
|
2016-03-11 16:25:40 -08:00
|
|
|
if (srcSize < ZSTD_frameHeaderSize_min) return ERROR(srcSize_wrong);
|
2016-05-28 20:01:04 -07:00
|
|
|
{ U32 const fcsId = (((const BYTE*)src)[4]) >> 6;
|
|
|
|
U32 const dictID =(((const BYTE*)src)[5]) & 3;
|
|
|
|
return ZSTD_frameHeaderSize_min + ZSTD_fcs_fieldSize[fcsId] + ZSTD_did_fieldSize[dictID];
|
|
|
|
}
|
2015-11-25 05:42:45 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-03-11 16:25:40 -08:00
|
|
|
/** ZSTD_getFrameParams() :
|
|
|
|
* decode Frame Header, or provide expected `srcSize`.
|
|
|
|
* @return : 0, `fparamsPtr` is correctly filled,
|
2016-03-14 17:33:36 -07:00
|
|
|
* >0, `srcSize` is too small, result is expected `srcSize`,
|
2016-03-11 16:25:40 -08:00
|
|
|
* or an error code, which can be tested using ZSTD_isError() */
|
2016-03-15 04:56:03 -07:00
|
|
|
size_t ZSTD_getFrameParams(ZSTD_frameParams* fparamsPtr, const void* src, size_t srcSize)
|
2015-11-25 05:42:45 -08:00
|
|
|
{
|
2016-03-11 16:25:40 -08:00
|
|
|
const BYTE* ip = (const BYTE*)src;
|
|
|
|
|
|
|
|
if (srcSize < ZSTD_frameHeaderSize_min) return ZSTD_frameHeaderSize_min;
|
2016-05-31 03:43:46 -07:00
|
|
|
if (MEM_readLE32(src) != ZSTD_MAGICNUMBER) {
|
|
|
|
if ((MEM_readLE32(src) & 0xFFFFFFF0U) == ZSTD_MAGIC_SKIPPABLE_START) {
|
|
|
|
if (srcSize < ZSTD_skippableHeaderSize) return ZSTD_skippableHeaderSize; /* magic number + skippable frame length */
|
2016-06-01 05:44:31 -07:00
|
|
|
memset(fparamsPtr, 0, sizeof(*fparamsPtr));
|
|
|
|
fparamsPtr->frameContentSize = MEM_readLE32((const char *)src + 4);
|
2016-06-01 09:47:04 -07:00
|
|
|
fparamsPtr->windowLog = 0; /* windowLog==0 means a frame is skippable */
|
2016-05-31 03:43:46 -07:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return ERROR(prefix_unknown);
|
|
|
|
}
|
2016-03-11 16:25:40 -08:00
|
|
|
|
2016-03-14 17:33:36 -07:00
|
|
|
/* ensure there is enough `srcSize` to fully read/decode frame header */
|
2016-03-14 17:24:33 -07:00
|
|
|
{ size_t const fhsize = ZSTD_frameHeaderSize(src, srcSize);
|
2016-03-11 16:25:40 -08:00
|
|
|
if (srcSize < fhsize) return fhsize; }
|
|
|
|
|
2016-03-11 12:58:04 -08:00
|
|
|
memset(fparamsPtr, 0, sizeof(*fparamsPtr));
|
2016-05-28 20:01:04 -07:00
|
|
|
{ BYTE const allocByte = ip[4];
|
|
|
|
BYTE const checkByte = ip[5];
|
|
|
|
size_t pos = ZSTD_frameHeaderSize_min;
|
|
|
|
U32 const dictIDSizeCode = checkByte&3;
|
|
|
|
if ((allocByte & 0x30) != 0) return ERROR(frameParameter_unsupported); /* reserved bits */
|
2016-05-31 09:13:56 -07:00
|
|
|
if ((checkByte & 0xEC) != 0) return ERROR(frameParameter_unsupported); /* reserved bits */
|
2016-05-31 15:18:28 -07:00
|
|
|
fparamsPtr->windowLog = (allocByte & 0xF) + ZSTD_WINDOWLOG_ABSOLUTEMIN;
|
2016-05-31 09:13:56 -07:00
|
|
|
fparamsPtr->checksumFlag = checkByte & 0x10;
|
2016-05-28 20:01:04 -07:00
|
|
|
switch(dictIDSizeCode) /* fcsId */
|
|
|
|
{
|
|
|
|
default: /* impossible */
|
|
|
|
case 0 : fparamsPtr->dictID = 0; break;
|
|
|
|
case 1 : fparamsPtr->dictID = ip[pos]; pos++; break;
|
|
|
|
case 2 : fparamsPtr->dictID = MEM_readLE16(ip+pos); pos+=2; break;
|
|
|
|
case 3 : fparamsPtr->dictID = MEM_readLE32(ip+pos); pos+=4; break;
|
|
|
|
}
|
|
|
|
switch(allocByte >> 6) /* fcsId */
|
2016-03-14 17:33:36 -07:00
|
|
|
{
|
|
|
|
default: /* impossible */
|
|
|
|
case 0 : fparamsPtr->frameContentSize = 0; break;
|
2016-05-28 20:01:04 -07:00
|
|
|
case 1 : fparamsPtr->frameContentSize = ip[pos]; break;
|
|
|
|
case 2 : fparamsPtr->frameContentSize = MEM_readLE16(ip+pos)+256; break;
|
|
|
|
case 3 : fparamsPtr->frameContentSize = MEM_readLE64(ip+pos); break;
|
2016-03-14 17:33:36 -07:00
|
|
|
} }
|
2015-11-25 05:42:45 -08:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2016-03-11 16:25:40 -08:00
|
|
|
|
|
|
|
/** ZSTD_decodeFrameHeader() :
|
2016-03-14 17:33:36 -07:00
|
|
|
* `srcSize` must be the size provided by ZSTD_frameHeaderSize().
|
2016-03-19 06:14:31 -07:00
|
|
|
* @return : 0 if success, or an error code, which can be tested using ZSTD_isError() */
|
2016-05-30 07:17:33 -07:00
|
|
|
static size_t ZSTD_decodeFrameHeader(ZSTD_DCtx* dctx, const void* src, size_t srcSize)
|
2015-11-26 03:43:28 -08:00
|
|
|
{
|
2016-05-30 07:17:33 -07:00
|
|
|
size_t const result = ZSTD_getFrameParams(&(dctx->fParams), src, srcSize);
|
|
|
|
if ((MEM_32bits()) && (dctx->fParams.windowLog > 25)) return ERROR(frameParameter_unsupportedBy32bits);
|
|
|
|
if (dctx->fParams.dictID && (dctx->dictID != dctx->fParams.dictID)) return ERROR(dictionary_wrong);
|
2016-05-31 09:13:56 -07:00
|
|
|
if (dctx->fParams.checksumFlag) XXH64_reset(&dctx->xxhState, 0);
|
2015-11-28 07:03:22 -08:00
|
|
|
return result;
|
2015-11-26 03:43:28 -08:00
|
|
|
}
|
|
|
|
|
2015-11-11 04:43:58 -08:00
|
|
|
|
2016-03-19 06:14:31 -07:00
|
|
|
typedef struct
|
|
|
|
{
|
|
|
|
blockType_t blockType;
|
|
|
|
U32 origSize;
|
|
|
|
} blockProperties_t;
|
|
|
|
|
|
|
|
/*! ZSTD_getcBlockSize() :
|
|
|
|
* Provides the size of compressed block from block header `src` */
|
2015-11-11 04:43:58 -08:00
|
|
|
size_t ZSTD_getcBlockSize(const void* src, size_t srcSize, blockProperties_t* bpPtr)
|
|
|
|
{
|
|
|
|
const BYTE* const in = (const BYTE* const)src;
|
|
|
|
U32 cSize;
|
|
|
|
|
2016-03-19 06:14:31 -07:00
|
|
|
if (srcSize < ZSTD_blockHeaderSize) return ERROR(srcSize_wrong);
|
2015-11-11 04:43:58 -08:00
|
|
|
|
2016-03-19 06:14:31 -07:00
|
|
|
bpPtr->blockType = (blockType_t)((*in) >> 6);
|
2015-11-11 04:43:58 -08:00
|
|
|
cSize = in[2] + (in[1]<<8) + ((in[0] & 7)<<16);
|
|
|
|
bpPtr->origSize = (bpPtr->blockType == bt_rle) ? cSize : 0;
|
|
|
|
|
|
|
|
if (bpPtr->blockType == bt_end) return 0;
|
|
|
|
if (bpPtr->blockType == bt_rle) return 1;
|
|
|
|
return cSize;
|
|
|
|
}
|
|
|
|
|
2016-01-23 10:28:41 -08:00
|
|
|
|
2016-03-19 06:14:31 -07:00
|
|
|
static size_t ZSTD_copyRawBlock(void* dst, size_t dstCapacity, const void* src, size_t srcSize)
|
2015-11-11 04:43:58 -08:00
|
|
|
{
|
2016-03-19 06:14:31 -07:00
|
|
|
if (srcSize > dstCapacity) return ERROR(dstSize_tooSmall);
|
2015-11-11 04:43:58 -08:00
|
|
|
memcpy(dst, src, srcSize);
|
|
|
|
return srcSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-02-04 06:28:14 -08:00
|
|
|
/*! ZSTD_decodeLiteralsBlock() :
|
2015-11-11 12:38:21 -08:00
|
|
|
@return : nb of bytes read from src (< srcSize ) */
|
2015-11-12 06:36:05 -08:00
|
|
|
size_t ZSTD_decodeLiteralsBlock(ZSTD_DCtx* dctx,
|
2015-11-11 04:43:58 -08:00
|
|
|
const void* src, size_t srcSize) /* note : srcSize < BLOCKSIZE */
|
|
|
|
{
|
|
|
|
const BYTE* const istart = (const BYTE*) src;
|
|
|
|
|
|
|
|
/* any compressed block with literals segment must be at least this size */
|
|
|
|
if (srcSize < MIN_CBLOCK_SIZE) return ERROR(corruption_detected);
|
|
|
|
|
2016-01-23 10:28:41 -08:00
|
|
|
switch(istart[0]>> 6)
|
2015-11-11 04:43:58 -08:00
|
|
|
{
|
2016-01-23 10:28:41 -08:00
|
|
|
case IS_HUF:
|
2016-04-11 05:22:56 -07:00
|
|
|
{ size_t litSize, litCSize, singleStream=0;
|
2016-01-23 10:28:41 -08:00
|
|
|
U32 lhSize = ((istart[0]) >> 4) & 3;
|
2016-04-11 05:22:56 -07:00
|
|
|
if (srcSize < 5) return ERROR(corruption_detected); /* srcSize >= MIN_CBLOCK_SIZE == 3; here we need up to 5 for lhSize, + cSize (+nbSeq) */
|
2016-01-23 10:28:41 -08:00
|
|
|
switch(lhSize)
|
|
|
|
{
|
|
|
|
case 0: case 1: default: /* note : default is impossible, since lhSize into [0..3] */
|
|
|
|
/* 2 - 2 - 10 - 10 */
|
|
|
|
lhSize=3;
|
2016-01-24 19:10:46 -08:00
|
|
|
singleStream = istart[0] & 16;
|
2016-01-23 10:28:41 -08:00
|
|
|
litSize = ((istart[0] & 15) << 6) + (istart[1] >> 2);
|
|
|
|
litCSize = ((istart[1] & 3) << 8) + istart[2];
|
|
|
|
break;
|
|
|
|
case 2:
|
|
|
|
/* 2 - 2 - 14 - 14 */
|
|
|
|
lhSize=4;
|
|
|
|
litSize = ((istart[0] & 15) << 10) + (istart[1] << 2) + (istart[2] >> 6);
|
|
|
|
litCSize = ((istart[2] & 63) << 8) + istart[3];
|
|
|
|
break;
|
|
|
|
case 3:
|
|
|
|
/* 2 - 2 - 18 - 18 */
|
|
|
|
lhSize=5;
|
|
|
|
litSize = ((istart[0] & 15) << 14) + (istart[1] << 6) + (istart[2] >> 2);
|
|
|
|
litCSize = ((istart[2] & 3) << 16) + (istart[3] << 8) + istart[4];
|
|
|
|
break;
|
|
|
|
}
|
2016-03-16 07:26:51 -07:00
|
|
|
if (litSize > ZSTD_BLOCKSIZE_MAX) return ERROR(corruption_detected);
|
2016-04-07 10:35:23 -07:00
|
|
|
if (litCSize + lhSize > srcSize) return ERROR(corruption_detected);
|
2016-01-23 10:28:41 -08:00
|
|
|
|
2016-01-24 19:10:46 -08:00
|
|
|
if (HUF_isError(singleStream ?
|
|
|
|
HUF_decompress1X2(dctx->litBuffer, litSize, istart+lhSize, litCSize) :
|
|
|
|
HUF_decompress (dctx->litBuffer, litSize, istart+lhSize, litCSize) ))
|
2016-01-23 10:28:41 -08:00
|
|
|
return ERROR(corruption_detected);
|
|
|
|
|
2015-11-11 04:43:58 -08:00
|
|
|
dctx->litPtr = dctx->litBuffer;
|
2016-03-16 07:26:51 -07:00
|
|
|
dctx->litBufSize = ZSTD_BLOCKSIZE_MAX+8;
|
2015-11-11 04:43:58 -08:00
|
|
|
dctx->litSize = litSize;
|
2016-01-23 10:28:41 -08:00
|
|
|
return litCSize + lhSize;
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
2016-01-25 18:14:20 -08:00
|
|
|
case IS_PCH:
|
2016-04-11 05:22:56 -07:00
|
|
|
{ size_t litSize, litCSize;
|
2016-01-25 18:14:20 -08:00
|
|
|
U32 lhSize = ((istart[0]) >> 4) & 3;
|
|
|
|
if (lhSize != 1) /* only case supported for now : small litSize, single stream */
|
|
|
|
return ERROR(corruption_detected);
|
2016-04-09 08:26:22 -07:00
|
|
|
if (!dctx->flagRepeatTable)
|
2016-01-26 06:58:49 -08:00
|
|
|
return ERROR(dictionary_corrupted);
|
2016-01-25 18:14:20 -08:00
|
|
|
|
|
|
|
/* 2 - 2 - 10 - 10 */
|
|
|
|
lhSize=3;
|
|
|
|
litSize = ((istart[0] & 15) << 6) + (istart[1] >> 2);
|
|
|
|
litCSize = ((istart[1] & 3) << 8) + istart[2];
|
|
|
|
|
2016-04-11 05:22:56 -07:00
|
|
|
{ size_t const errorCode = HUF_decompress1X4_usingDTable(dctx->litBuffer, litSize, istart+lhSize, litCSize, dctx->hufTableX4);
|
|
|
|
if (HUF_isError(errorCode)) return ERROR(corruption_detected);
|
|
|
|
}
|
2016-01-25 18:14:20 -08:00
|
|
|
dctx->litPtr = dctx->litBuffer;
|
2016-03-16 07:26:51 -07:00
|
|
|
dctx->litBufSize = ZSTD_BLOCKSIZE_MAX+WILDCOPY_OVERLENGTH;
|
2016-01-25 18:14:20 -08:00
|
|
|
dctx->litSize = litSize;
|
|
|
|
return litCSize + lhSize;
|
|
|
|
}
|
2015-11-11 04:43:58 -08:00
|
|
|
case IS_RAW:
|
2016-04-11 05:22:56 -07:00
|
|
|
{ size_t litSize;
|
2016-01-23 10:28:41 -08:00
|
|
|
U32 lhSize = ((istart[0]) >> 4) & 3;
|
|
|
|
switch(lhSize)
|
2015-11-11 04:43:58 -08:00
|
|
|
{
|
2016-01-23 10:28:41 -08:00
|
|
|
case 0: case 1: default: /* note : default is impossible, since lhSize into [0..3] */
|
|
|
|
lhSize=1;
|
|
|
|
litSize = istart[0] & 31;
|
|
|
|
break;
|
|
|
|
case 2:
|
|
|
|
litSize = ((istart[0] & 15) << 8) + istart[1];
|
|
|
|
break;
|
|
|
|
case 3:
|
|
|
|
litSize = ((istart[0] & 15) << 16) + (istart[1] << 8) + istart[2];
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2016-01-30 17:04:15 -08:00
|
|
|
if (lhSize+litSize+WILDCOPY_OVERLENGTH > srcSize) { /* risk reading beyond src buffer with wildcopy */
|
2016-02-03 03:39:34 -08:00
|
|
|
if (litSize+lhSize > srcSize) return ERROR(corruption_detected);
|
2016-01-23 10:28:41 -08:00
|
|
|
memcpy(dctx->litBuffer, istart+lhSize, litSize);
|
2015-11-11 04:43:58 -08:00
|
|
|
dctx->litPtr = dctx->litBuffer;
|
2016-03-16 07:26:51 -07:00
|
|
|
dctx->litBufSize = ZSTD_BLOCKSIZE_MAX+8;
|
2015-11-11 04:43:58 -08:00
|
|
|
dctx->litSize = litSize;
|
2016-01-25 08:26:01 -08:00
|
|
|
return lhSize+litSize;
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
|
|
|
/* direct reference into compressed stream */
|
2016-01-23 10:28:41 -08:00
|
|
|
dctx->litPtr = istart+lhSize;
|
|
|
|
dctx->litBufSize = srcSize-lhSize;
|
2015-11-11 04:43:58 -08:00
|
|
|
dctx->litSize = litSize;
|
2016-01-23 10:28:41 -08:00
|
|
|
return lhSize+litSize;
|
|
|
|
}
|
2015-11-11 04:43:58 -08:00
|
|
|
case IS_RLE:
|
2016-04-11 05:22:56 -07:00
|
|
|
{ size_t litSize;
|
2016-01-23 10:28:41 -08:00
|
|
|
U32 lhSize = ((istart[0]) >> 4) & 3;
|
|
|
|
switch(lhSize)
|
|
|
|
{
|
|
|
|
case 0: case 1: default: /* note : default is impossible, since lhSize into [0..3] */
|
|
|
|
lhSize = 1;
|
|
|
|
litSize = istart[0] & 31;
|
|
|
|
break;
|
|
|
|
case 2:
|
|
|
|
litSize = ((istart[0] & 15) << 8) + istart[1];
|
|
|
|
break;
|
|
|
|
case 3:
|
|
|
|
litSize = ((istart[0] & 15) << 16) + (istart[1] << 8) + istart[2];
|
2016-04-11 05:22:56 -07:00
|
|
|
if (srcSize<4) return ERROR(corruption_detected); /* srcSize >= MIN_CBLOCK_SIZE == 3; here we need lhSize+1 = 4 */
|
2016-01-23 10:28:41 -08:00
|
|
|
break;
|
|
|
|
}
|
2016-03-16 07:26:51 -07:00
|
|
|
if (litSize > ZSTD_BLOCKSIZE_MAX) return ERROR(corruption_detected);
|
2016-01-23 10:28:41 -08:00
|
|
|
memset(dctx->litBuffer, istart[lhSize], litSize);
|
2015-11-11 04:43:58 -08:00
|
|
|
dctx->litPtr = dctx->litBuffer;
|
2016-03-16 07:26:51 -07:00
|
|
|
dctx->litBufSize = ZSTD_BLOCKSIZE_MAX+WILDCOPY_OVERLENGTH;
|
2015-11-11 04:43:58 -08:00
|
|
|
dctx->litSize = litSize;
|
2016-01-23 10:28:41 -08:00
|
|
|
return lhSize+1;
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
2016-01-25 18:14:20 -08:00
|
|
|
default:
|
|
|
|
return ERROR(corruption_detected); /* impossible */
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-03-18 14:23:49 -07:00
|
|
|
/*! ZSTD_buildSeqTable() :
|
|
|
|
@return : nb bytes read from src,
|
|
|
|
or an error code if it fails, testable with ZSTD_isError()
|
|
|
|
*/
|
2016-03-22 17:32:41 -07:00
|
|
|
FORCE_INLINE size_t ZSTD_buildSeqTable(FSE_DTable* DTable, U32 type, U32 max, U32 maxLog,
|
|
|
|
const void* src, size_t srcSize,
|
2016-04-09 08:26:22 -07:00
|
|
|
const S16* defaultNorm, U32 defaultLog, U32 flagRepeatTable)
|
2016-03-21 05:24:16 -07:00
|
|
|
{
|
|
|
|
switch(type)
|
|
|
|
{
|
|
|
|
case FSE_ENCODING_RLE :
|
|
|
|
if (!srcSize) return ERROR(srcSize_wrong);
|
|
|
|
if ( (*(const BYTE*)src) > max) return ERROR(corruption_detected);
|
|
|
|
FSE_buildDTable_rle(DTable, *(const BYTE*)src); /* if *src > max, data is corrupted */
|
|
|
|
return 1;
|
|
|
|
case FSE_ENCODING_RAW :
|
2016-03-22 17:32:41 -07:00
|
|
|
FSE_buildDTable(DTable, defaultNorm, max, defaultLog);
|
2016-03-22 15:19:28 -07:00
|
|
|
return 0;
|
|
|
|
case FSE_ENCODING_STATIC:
|
2016-04-09 08:26:22 -07:00
|
|
|
if (!flagRepeatTable) return ERROR(corruption_detected);
|
2016-03-22 15:19:28 -07:00
|
|
|
return 0;
|
|
|
|
default : /* impossible */
|
|
|
|
case FSE_ENCODING_DYNAMIC :
|
|
|
|
{ U32 tableLog;
|
|
|
|
S16 norm[MaxSeq+1];
|
|
|
|
size_t const headerSize = FSE_readNCount(norm, &max, &tableLog, src, srcSize);
|
|
|
|
if (FSE_isError(headerSize)) return ERROR(corruption_detected);
|
|
|
|
if (tableLog > maxLog) return ERROR(corruption_detected);
|
|
|
|
FSE_buildDTable(DTable, norm, max, tableLog);
|
|
|
|
return headerSize;
|
|
|
|
} }
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-03-23 17:27:55 -07:00
|
|
|
size_t ZSTD_decodeSeqHeaders(int* nbSeqPtr,
|
2016-04-09 08:26:22 -07:00
|
|
|
FSE_DTable* DTableLL, FSE_DTable* DTableML, FSE_DTable* DTableOffb, U32 flagRepeatTable,
|
2016-03-22 15:19:28 -07:00
|
|
|
const void* src, size_t srcSize)
|
2015-11-11 04:43:58 -08:00
|
|
|
{
|
|
|
|
const BYTE* const istart = (const BYTE* const)src;
|
|
|
|
const BYTE* const iend = istart + srcSize;
|
2016-03-23 17:27:55 -07:00
|
|
|
const BYTE* ip = istart;
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
/* check */
|
2016-03-18 14:23:49 -07:00
|
|
|
if (srcSize < MIN_SEQUENCES_SIZE) return ERROR(srcSize_wrong);
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
/* SeqHead */
|
2016-03-23 17:27:55 -07:00
|
|
|
{ int nbSeq = *ip++;
|
|
|
|
if (!nbSeq) { *nbSeqPtr=0; return 1; }
|
2016-04-01 06:49:41 -07:00
|
|
|
if (nbSeq > 0x7F) {
|
2016-03-23 17:27:55 -07:00
|
|
|
if (nbSeq == 0xFF)
|
|
|
|
nbSeq = MEM_readLE16(ip) + LONGNBSEQ, ip+=2;
|
|
|
|
else
|
|
|
|
nbSeq = ((nbSeq-0x80)<<8) + *ip++;
|
|
|
|
}
|
|
|
|
*nbSeqPtr = nbSeq;
|
2016-03-04 05:45:31 -08:00
|
|
|
}
|
2016-01-30 15:58:06 -08:00
|
|
|
|
2016-03-04 05:45:31 -08:00
|
|
|
/* FSE table descriptors */
|
2016-03-23 17:27:55 -07:00
|
|
|
{ U32 const LLtype = *ip >> 6;
|
|
|
|
U32 const Offtype = (*ip >> 4) & 3;
|
|
|
|
U32 const MLtype = (*ip >> 2) & 3;
|
|
|
|
ip++;
|
2015-11-11 04:43:58 -08:00
|
|
|
|
2016-03-23 17:27:55 -07:00
|
|
|
/* check */
|
|
|
|
if (ip > iend-3) return ERROR(srcSize_wrong); /* min : all 3 are "raw", hence no header, but at least xxLog bits per type */
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
/* Build DTables */
|
2016-04-09 08:26:22 -07:00
|
|
|
{ size_t const bhSize = ZSTD_buildSeqTable(DTableLL, LLtype, MaxLL, LLFSELog, ip, iend-ip, LL_defaultNorm, LL_defaultNormLog, flagRepeatTable);
|
2016-03-23 17:27:55 -07:00
|
|
|
if (ZSTD_isError(bhSize)) return ERROR(corruption_detected);
|
|
|
|
ip += bhSize;
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
2016-04-09 08:26:22 -07:00
|
|
|
{ size_t const bhSize = ZSTD_buildSeqTable(DTableOffb, Offtype, MaxOff, OffFSELog, ip, iend-ip, OF_defaultNorm, OF_defaultNormLog, flagRepeatTable);
|
2016-03-23 17:27:55 -07:00
|
|
|
if (ZSTD_isError(bhSize)) return ERROR(corruption_detected);
|
|
|
|
ip += bhSize;
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
2016-04-09 08:26:22 -07:00
|
|
|
{ size_t const bhSize = ZSTD_buildSeqTable(DTableML, MLtype, MaxML, MLFSELog, ip, iend-ip, ML_defaultNorm, ML_defaultNormLog, flagRepeatTable);
|
2016-03-23 17:27:55 -07:00
|
|
|
if (ZSTD_isError(bhSize)) return ERROR(corruption_detected);
|
|
|
|
ip += bhSize;
|
2016-01-27 15:18:06 -08:00
|
|
|
} }
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
return ip-istart;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
typedef struct {
|
|
|
|
size_t litLength;
|
|
|
|
size_t matchLength;
|
2016-01-30 15:58:06 -08:00
|
|
|
size_t offset;
|
2015-11-11 04:43:58 -08:00
|
|
|
} seq_t;
|
|
|
|
|
|
|
|
typedef struct {
|
|
|
|
BIT_DStream_t DStream;
|
|
|
|
FSE_DState_t stateLL;
|
|
|
|
FSE_DState_t stateOffb;
|
|
|
|
FSE_DState_t stateML;
|
2016-03-18 03:03:43 -07:00
|
|
|
size_t prevOffset[ZSTD_REP_INIT];
|
2015-11-11 04:43:58 -08:00
|
|
|
} seqState_t;
|
|
|
|
|
2016-02-03 03:53:07 -08:00
|
|
|
|
2016-04-04 07:28:40 -07:00
|
|
|
|
2016-04-06 00:46:01 -07:00
|
|
|
static void ZSTD_decodeSequence(seq_t* seq, seqState_t* seqState)
|
2015-11-11 04:43:58 -08:00
|
|
|
{
|
|
|
|
/* Literal length */
|
2016-03-22 17:54:25 -07:00
|
|
|
U32 const llCode = FSE_peekSymbol(&(seqState->stateLL));
|
|
|
|
U32 const mlCode = FSE_peekSymbol(&(seqState->stateML));
|
|
|
|
U32 const ofCode = FSE_peekSymbol(&(seqState->stateOffb)); /* <= maxOff, by table construction */
|
|
|
|
|
|
|
|
U32 const llBits = LL_bits[llCode];
|
|
|
|
U32 const mlBits = ML_bits[mlCode];
|
2016-03-23 18:31:27 -07:00
|
|
|
U32 const ofBits = ofCode;
|
|
|
|
U32 const totalBits = llBits+mlBits+ofBits;
|
2016-03-22 17:54:25 -07:00
|
|
|
|
|
|
|
static const U32 LL_base[MaxLL+1] = {
|
2016-03-21 05:24:16 -07:00
|
|
|
0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
|
|
|
|
16, 18, 20, 22, 24, 28, 32, 40, 48, 64, 0x80, 0x100, 0x200, 0x400, 0x800, 0x1000,
|
|
|
|
0x2000, 0x4000, 0x8000, 0x10000 };
|
2015-11-11 04:43:58 -08:00
|
|
|
|
2016-03-22 17:54:25 -07:00
|
|
|
static const U32 ML_base[MaxML+1] = {
|
2016-03-22 17:32:41 -07:00
|
|
|
0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
|
|
|
|
16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31,
|
|
|
|
32, 34, 36, 38, 40, 44, 48, 56, 64, 80, 96, 0x80, 0x100, 0x200, 0x400, 0x800,
|
|
|
|
0x1000, 0x2000, 0x4000, 0x8000, 0x10000 };
|
|
|
|
|
2016-03-22 17:54:25 -07:00
|
|
|
static const U32 OF_base[MaxOff+1] = {
|
2016-03-23 18:31:27 -07:00
|
|
|
0, 1, 3, 7, 0xF, 0x1F, 0x3F, 0x7F,
|
|
|
|
0xFF, 0x1FF, 0x3FF, 0x7FF, 0xFFF, 0x1FFF, 0x3FFF, 0x7FFF,
|
|
|
|
0xFFFF, 0x1FFFF, 0x3FFFF, 0x7FFFF, 0xFFFFF, 0x1FFFFF, 0x3FFFFF, 0x7FFFFF,
|
2016-04-07 06:24:29 -07:00
|
|
|
0xFFFFFF, 0x1FFFFFF, 0x3FFFFFF, /*fake*/ 1, 1 };
|
2016-03-22 17:54:25 -07:00
|
|
|
|
2016-03-23 06:00:09 -07:00
|
|
|
/* sequence */
|
2016-04-05 00:34:13 -07:00
|
|
|
{ size_t offset;
|
2016-04-05 01:08:25 -07:00
|
|
|
if (!ofCode)
|
|
|
|
offset = 0;
|
|
|
|
else {
|
|
|
|
offset = OF_base[ofCode] + BIT_readBits(&(seqState->DStream), ofBits); /* <= 26 bits */
|
|
|
|
if (MEM_32bits()) BIT_reloadDStream(&(seqState->DStream));
|
2016-03-17 03:21:00 -07:00
|
|
|
}
|
2016-04-07 06:24:29 -07:00
|
|
|
|
2016-04-05 01:30:05 -07:00
|
|
|
if (offset < ZSTD_REP_NUM) {
|
|
|
|
if (llCode == 0 && offset <= 1) offset = 1-offset;
|
|
|
|
|
|
|
|
if (offset != 0) {
|
|
|
|
size_t temp = seqState->prevOffset[offset];
|
|
|
|
if (offset != 1) {
|
|
|
|
seqState->prevOffset[2] = seqState->prevOffset[1];
|
2016-04-05 01:08:25 -07:00
|
|
|
}
|
2016-04-05 01:30:05 -07:00
|
|
|
seqState->prevOffset[1] = seqState->prevOffset[0];
|
|
|
|
seqState->prevOffset[0] = offset = temp;
|
|
|
|
|
2016-04-05 01:08:25 -07:00
|
|
|
} else {
|
2016-04-05 01:30:05 -07:00
|
|
|
offset = seqState->prevOffset[0];
|
2016-04-05 01:08:25 -07:00
|
|
|
}
|
2016-04-05 01:30:05 -07:00
|
|
|
} else {
|
|
|
|
offset -= ZSTD_REP_MOVE;
|
|
|
|
seqState->prevOffset[2] = seqState->prevOffset[1];
|
2016-04-07 06:24:29 -07:00
|
|
|
seqState->prevOffset[1] = seqState->prevOffset[0];
|
2016-04-05 01:30:05 -07:00
|
|
|
seqState->prevOffset[0] = offset;
|
2016-04-05 01:08:25 -07:00
|
|
|
}
|
2016-04-05 00:34:13 -07:00
|
|
|
seq->offset = offset;
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
|
|
|
|
2016-04-06 00:46:01 -07:00
|
|
|
seq->matchLength = ML_base[mlCode] + MINMATCH + ((mlCode>31) ? BIT_readBits(&(seqState->DStream), mlBits) : 0); /* <= 16 bits */
|
2016-03-26 09:18:11 -07:00
|
|
|
if (MEM_32bits() && (mlBits+llBits>24)) BIT_reloadDStream(&(seqState->DStream));
|
2015-11-11 04:43:58 -08:00
|
|
|
|
2016-03-26 09:18:11 -07:00
|
|
|
seq->litLength = LL_base[llCode] + ((llCode>15) ? BIT_readBits(&(seqState->DStream), llBits) : 0); /* <= 16 bits */
|
|
|
|
if (MEM_32bits() ||
|
|
|
|
(totalBits > 64 - 7 - (LLFSELog+MLFSELog+OffFSELog)) ) BIT_reloadDStream(&(seqState->DStream));
|
2016-01-27 15:18:06 -08:00
|
|
|
|
2016-03-22 17:32:41 -07:00
|
|
|
/* ANS state update */
|
2016-03-26 09:18:11 -07:00
|
|
|
FSE_updateState(&(seqState->stateLL), &(seqState->DStream)); /* <= 9 bits */
|
|
|
|
FSE_updateState(&(seqState->stateML), &(seqState->DStream)); /* <= 9 bits */
|
|
|
|
if (MEM_32bits()) BIT_reloadDStream(&(seqState->DStream)); /* <= 18 bits */
|
|
|
|
FSE_updateState(&(seqState->stateOffb), &(seqState->DStream)); /* <= 8 bits */
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-05-12 05:06:31 -07:00
|
|
|
FORCE_INLINE
|
2016-05-10 05:14:19 -07:00
|
|
|
size_t ZSTD_execSequence(BYTE* op,
|
2015-11-19 08:13:19 -08:00
|
|
|
BYTE* const oend, seq_t sequence,
|
2015-11-11 04:43:58 -08:00
|
|
|
const BYTE** litPtr, const BYTE* const litLimit_8,
|
2015-12-04 08:16:37 -08:00
|
|
|
const BYTE* const base, const BYTE* const vBase, const BYTE* const dictEnd)
|
2015-11-11 04:43:58 -08:00
|
|
|
{
|
|
|
|
BYTE* const oLitEnd = op + sequence.litLength;
|
2016-03-23 17:27:55 -07:00
|
|
|
size_t const sequenceLength = sequence.litLength + sequence.matchLength;
|
2015-11-19 08:13:19 -08:00
|
|
|
BYTE* const oMatchEnd = op + sequenceLength; /* risk : address space overflow (32-bits) */
|
2015-11-11 04:43:58 -08:00
|
|
|
BYTE* const oend_8 = oend-8;
|
2016-04-03 11:24:25 -07:00
|
|
|
const BYTE* const iLitEnd = *litPtr + sequence.litLength;
|
2015-11-19 08:13:19 -08:00
|
|
|
const BYTE* match = oLitEnd - sequence.offset;
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
/* check */
|
|
|
|
if (oLitEnd > oend_8) return ERROR(dstSize_tooSmall); /* last match must start at a minimum distance of 8 from oend */
|
|
|
|
if (oMatchEnd > oend) return ERROR(dstSize_tooSmall); /* overwrite beyond dst buffer */
|
2016-04-03 11:24:25 -07:00
|
|
|
if (iLitEnd > litLimit_8) return ERROR(corruption_detected); /* over-read beyond lit buffer */
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
/* copy Literals */
|
|
|
|
ZSTD_wildcopy(op, *litPtr, sequence.litLength); /* note : oLitEnd <= oend-8 : no risk of overwrite beyond oend */
|
|
|
|
op = oLitEnd;
|
2016-04-03 11:24:25 -07:00
|
|
|
*litPtr = iLitEnd; /* update for next sequence */
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
/* copy Match */
|
2016-01-26 06:58:49 -08:00
|
|
|
if (sequence.offset > (size_t)(oLitEnd - base)) {
|
2015-11-30 14:13:56 -08:00
|
|
|
/* offset beyond prefix */
|
2016-03-23 17:27:55 -07:00
|
|
|
if (sequence.offset > (size_t)(oLitEnd - vBase)) return ERROR(corruption_detected);
|
2015-11-30 14:13:56 -08:00
|
|
|
match = dictEnd - (base-match);
|
2016-01-26 06:58:49 -08:00
|
|
|
if (match + sequence.matchLength <= dictEnd) {
|
2015-12-06 04:18:37 -08:00
|
|
|
memmove(oLitEnd, match, sequence.matchLength);
|
2015-11-30 14:13:56 -08:00
|
|
|
return sequenceLength;
|
|
|
|
}
|
|
|
|
/* span extDict & currentPrefixSegment */
|
2016-03-19 04:12:07 -07:00
|
|
|
{ size_t const length1 = dictEnd - match;
|
2015-12-06 04:18:37 -08:00
|
|
|
memmove(oLitEnd, match, length1);
|
2015-11-30 14:13:56 -08:00
|
|
|
op = oLitEnd + length1;
|
|
|
|
sequence.matchLength -= length1;
|
|
|
|
match = base;
|
2016-01-27 15:18:06 -08:00
|
|
|
} }
|
2015-11-30 14:13:56 -08:00
|
|
|
|
|
|
|
/* match within prefix */
|
2016-01-26 06:58:49 -08:00
|
|
|
if (sequence.offset < 8) {
|
2015-11-30 14:13:56 -08:00
|
|
|
/* close range match, overlap */
|
2016-03-23 17:27:55 -07:00
|
|
|
static const U32 dec32table[] = { 0, 1, 2, 1, 4, 4, 4, 4 }; /* added */
|
|
|
|
static const int dec64table[] = { 8, 8, 8, 7, 8, 9,10,11 }; /* substracted */
|
|
|
|
int const sub2 = dec64table[sequence.offset];
|
2015-11-30 14:13:56 -08:00
|
|
|
op[0] = match[0];
|
|
|
|
op[1] = match[1];
|
|
|
|
op[2] = match[2];
|
|
|
|
op[3] = match[3];
|
|
|
|
match += dec32table[sequence.offset];
|
|
|
|
ZSTD_copy4(op+4, match);
|
|
|
|
match -= sub2;
|
2016-01-27 15:18:06 -08:00
|
|
|
} else {
|
2015-11-30 14:13:56 -08:00
|
|
|
ZSTD_copy8(op, match);
|
|
|
|
}
|
|
|
|
op += 8; match += 8;
|
|
|
|
|
2016-04-06 00:46:01 -07:00
|
|
|
if (oMatchEnd > oend-(16-MINMATCH)) {
|
2016-01-26 06:58:49 -08:00
|
|
|
if (op < oend_8) {
|
2015-11-30 14:13:56 -08:00
|
|
|
ZSTD_wildcopy(op, match, oend_8 - op);
|
|
|
|
match += oend_8 - op;
|
|
|
|
op = oend_8;
|
|
|
|
}
|
2016-03-19 06:14:31 -07:00
|
|
|
while (op < oMatchEnd) *op++ = *match++;
|
2016-01-27 15:18:06 -08:00
|
|
|
} else {
|
2015-11-30 14:13:56 -08:00
|
|
|
ZSTD_wildcopy(op, match, sequence.matchLength-8); /* works even if matchLength < 8 */
|
|
|
|
}
|
|
|
|
return sequenceLength;
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
|
|
|
|
2015-11-19 08:13:19 -08:00
|
|
|
|
2015-11-11 04:43:58 -08:00
|
|
|
static size_t ZSTD_decompressSequences(
|
2015-11-12 06:36:05 -08:00
|
|
|
ZSTD_DCtx* dctx,
|
2015-11-11 04:43:58 -08:00
|
|
|
void* dst, size_t maxDstSize,
|
|
|
|
const void* seqStart, size_t seqSize)
|
|
|
|
{
|
|
|
|
const BYTE* ip = (const BYTE*)seqStart;
|
|
|
|
const BYTE* const iend = ip + seqSize;
|
|
|
|
BYTE* const ostart = (BYTE* const)dst;
|
|
|
|
BYTE* const oend = ostart + maxDstSize;
|
2016-04-07 17:02:12 -07:00
|
|
|
BYTE* op = ostart;
|
2015-11-11 04:43:58 -08:00
|
|
|
const BYTE* litPtr = dctx->litPtr;
|
|
|
|
const BYTE* const litLimit_8 = litPtr + dctx->litBufSize - 8;
|
|
|
|
const BYTE* const litEnd = litPtr + dctx->litSize;
|
2016-04-07 17:02:12 -07:00
|
|
|
FSE_DTable* DTableLL = dctx->LLTable;
|
|
|
|
FSE_DTable* DTableML = dctx->MLTable;
|
|
|
|
FSE_DTable* DTableOffb = dctx->OffTable;
|
2015-12-04 08:16:37 -08:00
|
|
|
const BYTE* const base = (const BYTE*) (dctx->base);
|
|
|
|
const BYTE* const vBase = (const BYTE*) (dctx->vBase);
|
|
|
|
const BYTE* const dictEnd = (const BYTE*) (dctx->dictEnd);
|
2016-04-03 11:24:25 -07:00
|
|
|
int nbSeq;
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
/* Build Decoding Tables */
|
2016-04-09 08:26:22 -07:00
|
|
|
{ size_t const seqHSize = ZSTD_decodeSeqHeaders(&nbSeq, DTableLL, DTableML, DTableOffb, dctx->flagRepeatTable, ip, seqSize);
|
2016-04-03 11:24:25 -07:00
|
|
|
if (ZSTD_isError(seqHSize)) return seqHSize;
|
|
|
|
ip += seqHSize;
|
2016-04-10 05:42:26 -07:00
|
|
|
dctx->flagRepeatTable = 0;
|
2016-03-21 06:22:38 -07:00
|
|
|
}
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
/* Regen sequences */
|
2016-01-30 15:58:06 -08:00
|
|
|
if (nbSeq) {
|
2015-11-11 04:43:58 -08:00
|
|
|
seq_t sequence;
|
|
|
|
seqState_t seqState;
|
|
|
|
|
|
|
|
memset(&sequence, 0, sizeof(sequence));
|
2016-01-30 17:04:15 -08:00
|
|
|
sequence.offset = REPCODE_STARTVALUE;
|
2016-04-07 17:02:12 -07:00
|
|
|
{ U32 i; for (i=0; i<ZSTD_REP_INIT; i++) seqState.prevOffset[i] = REPCODE_STARTVALUE; }
|
2016-03-21 06:22:38 -07:00
|
|
|
{ size_t const errorCode = BIT_initDStream(&(seqState.DStream), ip, iend-ip);
|
|
|
|
if (ERR_isError(errorCode)) return ERROR(corruption_detected); }
|
2015-11-11 04:43:58 -08:00
|
|
|
FSE_initDState(&(seqState.stateLL), &(seqState.DStream), DTableLL);
|
|
|
|
FSE_initDState(&(seqState.stateOffb), &(seqState.DStream), DTableOffb);
|
|
|
|
FSE_initDState(&(seqState.stateML), &(seqState.DStream), DTableML);
|
|
|
|
|
2016-04-08 16:51:36 -07:00
|
|
|
for ( ; (BIT_reloadDStream(&(seqState.DStream)) <= BIT_DStream_completed) && nbSeq ; ) {
|
|
|
|
nbSeq--;
|
2016-04-06 00:46:01 -07:00
|
|
|
ZSTD_decodeSequence(&sequence, &seqState);
|
2016-04-11 07:25:56 -07:00
|
|
|
|
|
|
|
#if 0 /* debug */
|
|
|
|
static BYTE* start = NULL;
|
|
|
|
if (start==NULL) start = op;
|
|
|
|
size_t pos = (size_t)(op-start);
|
|
|
|
if ((pos >= 5810037) && (pos < 5810400))
|
|
|
|
printf("Dpos %6u :%5u literals & match %3u bytes at distance %6u \n",
|
|
|
|
pos, (U32)sequence.litLength, (U32)sequence.matchLength, (U32)sequence.offset);
|
|
|
|
#endif
|
|
|
|
|
2016-04-07 17:02:12 -07:00
|
|
|
{ size_t const oneSeqSize = ZSTD_execSequence(op, oend, sequence, &litPtr, litLimit_8, base, vBase, dictEnd);
|
|
|
|
if (ZSTD_isError(oneSeqSize)) return oneSeqSize;
|
|
|
|
op += oneSeqSize;
|
|
|
|
} }
|
2015-11-19 08:13:19 -08:00
|
|
|
|
|
|
|
/* check if reached exact end */
|
2016-01-30 17:51:03 -08:00
|
|
|
if (nbSeq) return ERROR(corruption_detected);
|
2016-01-30 15:58:06 -08:00
|
|
|
}
|
2015-11-19 08:13:19 -08:00
|
|
|
|
2016-01-30 15:58:06 -08:00
|
|
|
/* last literal segment */
|
2016-03-19 06:14:31 -07:00
|
|
|
{ size_t const lastLLSize = litEnd - litPtr;
|
2016-01-30 15:58:06 -08:00
|
|
|
if (litPtr > litEnd) return ERROR(corruption_detected); /* too many literals already used */
|
|
|
|
if (op+lastLLSize > oend) return ERROR(dstSize_tooSmall);
|
|
|
|
memcpy(op, litPtr, lastLLSize);
|
|
|
|
op += lastLLSize;
|
|
|
|
}
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
return op-ostart;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-01-08 17:00:10 -08:00
|
|
|
static void ZSTD_checkContinuity(ZSTD_DCtx* dctx, const void* dst)
|
|
|
|
{
|
2016-01-26 06:58:49 -08:00
|
|
|
if (dst != dctx->previousDstEnd) { /* not contiguous */
|
2016-01-08 17:00:10 -08:00
|
|
|
dctx->dictEnd = dctx->previousDstEnd;
|
|
|
|
dctx->vBase = (const char*)dst - ((const char*)(dctx->previousDstEnd) - (const char*)(dctx->base));
|
|
|
|
dctx->base = dst;
|
|
|
|
dctx->previousDstEnd = dst;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static size_t ZSTD_decompressBlock_internal(ZSTD_DCtx* dctx,
|
2016-02-03 03:39:34 -08:00
|
|
|
void* dst, size_t dstCapacity,
|
2015-11-11 04:43:58 -08:00
|
|
|
const void* src, size_t srcSize)
|
2016-02-03 03:39:34 -08:00
|
|
|
{ /* blockType == blockCompressed */
|
2015-11-11 04:43:58 -08:00
|
|
|
const BYTE* ip = (const BYTE*)src;
|
2016-02-03 03:39:34 -08:00
|
|
|
|
2016-03-16 07:26:51 -07:00
|
|
|
if (srcSize >= ZSTD_BLOCKSIZE_MAX) return ERROR(srcSize_wrong);
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
/* Decode literals sub-block */
|
2016-04-07 17:02:12 -07:00
|
|
|
{ size_t const litCSize = ZSTD_decodeLiteralsBlock(dctx, src, srcSize);
|
|
|
|
if (ZSTD_isError(litCSize)) return litCSize;
|
|
|
|
ip += litCSize;
|
|
|
|
srcSize -= litCSize;
|
|
|
|
}
|
2016-02-03 03:39:34 -08:00
|
|
|
return ZSTD_decompressSequences(dctx, dst, dstCapacity, ip, srcSize);
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-01-08 17:00:10 -08:00
|
|
|
size_t ZSTD_decompressBlock(ZSTD_DCtx* dctx,
|
2016-02-03 03:39:34 -08:00
|
|
|
void* dst, size_t dstCapacity,
|
2016-01-08 17:00:10 -08:00
|
|
|
const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
ZSTD_checkContinuity(dctx, dst);
|
2016-02-03 03:39:34 -08:00
|
|
|
return ZSTD_decompressBlock_internal(dctx, dst, dstCapacity, src, srcSize);
|
2016-01-08 17:00:10 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-05-06 07:43:23 -07:00
|
|
|
/*! ZSTD_decompressFrame() :
|
|
|
|
* `dctx` must be properly initialized */
|
2016-03-11 12:58:04 -08:00
|
|
|
static size_t ZSTD_decompressFrame(ZSTD_DCtx* dctx,
|
2016-03-19 06:14:31 -07:00
|
|
|
void* dst, size_t dstCapacity,
|
2016-01-26 06:58:49 -08:00
|
|
|
const void* src, size_t srcSize)
|
2015-11-11 04:43:58 -08:00
|
|
|
{
|
|
|
|
const BYTE* ip = (const BYTE*)src;
|
2016-05-06 07:43:23 -07:00
|
|
|
const BYTE* const iend = ip + srcSize;
|
2015-11-11 04:43:58 -08:00
|
|
|
BYTE* const ostart = (BYTE* const)dst;
|
|
|
|
BYTE* op = ostart;
|
2016-03-19 06:14:31 -07:00
|
|
|
BYTE* const oend = ostart + dstCapacity;
|
2015-11-11 04:43:58 -08:00
|
|
|
size_t remainingSize = srcSize;
|
2016-04-22 09:22:30 -07:00
|
|
|
blockProperties_t blockProperties = { bt_compressed, 0 };
|
2015-11-11 04:43:58 -08:00
|
|
|
|
2016-03-19 06:14:31 -07:00
|
|
|
/* check */
|
|
|
|
if (srcSize < ZSTD_frameHeaderSize_min+ZSTD_blockHeaderSize) return ERROR(srcSize_wrong);
|
|
|
|
|
|
|
|
/* Frame Header */
|
|
|
|
{ size_t const frameHeaderSize = ZSTD_frameHeaderSize(src, ZSTD_frameHeaderSize_min);
|
2015-11-25 05:42:45 -08:00
|
|
|
if (ZSTD_isError(frameHeaderSize)) return frameHeaderSize;
|
|
|
|
if (srcSize < frameHeaderSize+ZSTD_blockHeaderSize) return ERROR(srcSize_wrong);
|
2016-03-19 06:14:31 -07:00
|
|
|
if (ZSTD_decodeFrameHeader(dctx, src, frameHeaderSize)) return ERROR(corruption_detected);
|
2015-11-25 05:42:45 -08:00
|
|
|
ip += frameHeaderSize; remainingSize -= frameHeaderSize;
|
|
|
|
}
|
2015-11-11 04:43:58 -08:00
|
|
|
|
|
|
|
/* Loop on each block */
|
2016-03-19 06:14:31 -07:00
|
|
|
while (1) {
|
2015-11-11 04:43:58 -08:00
|
|
|
size_t decodedSize=0;
|
2016-03-23 17:27:55 -07:00
|
|
|
size_t const cBlockSize = ZSTD_getcBlockSize(ip, iend-ip, &blockProperties);
|
2015-11-11 04:43:58 -08:00
|
|
|
if (ZSTD_isError(cBlockSize)) return cBlockSize;
|
|
|
|
|
|
|
|
ip += ZSTD_blockHeaderSize;
|
|
|
|
remainingSize -= ZSTD_blockHeaderSize;
|
|
|
|
if (cBlockSize > remainingSize) return ERROR(srcSize_wrong);
|
|
|
|
|
|
|
|
switch(blockProperties.blockType)
|
|
|
|
{
|
|
|
|
case bt_compressed:
|
2016-01-25 07:54:05 -08:00
|
|
|
decodedSize = ZSTD_decompressBlock_internal(dctx, op, oend-op, ip, cBlockSize);
|
2015-11-11 04:43:58 -08:00
|
|
|
break;
|
|
|
|
case bt_raw :
|
2015-11-12 07:19:30 -08:00
|
|
|
decodedSize = ZSTD_copyRawBlock(op, oend-op, ip, cBlockSize);
|
2015-11-11 04:43:58 -08:00
|
|
|
break;
|
|
|
|
case bt_rle :
|
|
|
|
return ERROR(GENERIC); /* not yet supported */
|
|
|
|
break;
|
|
|
|
case bt_end :
|
|
|
|
/* end of frame */
|
|
|
|
if (remainingSize) return ERROR(srcSize_wrong);
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
return ERROR(GENERIC); /* impossible */
|
|
|
|
}
|
|
|
|
if (cBlockSize == 0) break; /* bt_end */
|
|
|
|
|
|
|
|
if (ZSTD_isError(decodedSize)) return decodedSize;
|
2016-05-31 15:18:28 -07:00
|
|
|
if (dctx->fParams.checksumFlag) XXH64_update(&dctx->xxhState, op, decodedSize);
|
2015-11-11 04:43:58 -08:00
|
|
|
op += decodedSize;
|
|
|
|
ip += cBlockSize;
|
|
|
|
remainingSize -= cBlockSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
return op-ostart;
|
|
|
|
}
|
|
|
|
|
2015-12-17 16:26:48 -08:00
|
|
|
|
2016-01-26 06:58:49 -08:00
|
|
|
size_t ZSTD_decompress_usingPreparedDCtx(ZSTD_DCtx* dctx, const ZSTD_DCtx* refDCtx,
|
2016-03-19 06:14:31 -07:00
|
|
|
void* dst, size_t dstCapacity,
|
2016-01-26 06:58:49 -08:00
|
|
|
const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
ZSTD_copyDCtx(dctx, refDCtx);
|
|
|
|
ZSTD_checkContinuity(dctx, dst);
|
2016-03-19 06:14:31 -07:00
|
|
|
return ZSTD_decompressFrame(dctx, dst, dstCapacity, src, srcSize);
|
2016-01-26 06:58:49 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
size_t ZSTD_decompress_usingDict(ZSTD_DCtx* dctx,
|
2016-03-19 06:14:31 -07:00
|
|
|
void* dst, size_t dstCapacity,
|
2016-01-26 06:58:49 -08:00
|
|
|
const void* src, size_t srcSize,
|
|
|
|
const void* dict, size_t dictSize)
|
|
|
|
{
|
2016-05-06 07:43:23 -07:00
|
|
|
#if defined(ZSTD_LEGACY_SUPPORT) && (ZSTD_LEGACY_SUPPORT==1)
|
|
|
|
{ const U32 magicNumber = MEM_readLE32(src);
|
|
|
|
if (ZSTD_isLegacy(magicNumber))
|
|
|
|
return ZSTD_decompressLegacy(dst, dstCapacity, src, srcSize, dict, dictSize, magicNumber);
|
|
|
|
}
|
|
|
|
#endif
|
2016-01-26 06:58:49 -08:00
|
|
|
ZSTD_decompressBegin_usingDict(dctx, dict, dictSize);
|
|
|
|
ZSTD_checkContinuity(dctx, dst);
|
2016-03-19 06:14:31 -07:00
|
|
|
return ZSTD_decompressFrame(dctx, dst, dstCapacity, src, srcSize);
|
2016-01-26 06:58:49 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-03-19 06:14:31 -07:00
|
|
|
size_t ZSTD_decompressDCtx(ZSTD_DCtx* dctx, void* dst, size_t dstCapacity, const void* src, size_t srcSize)
|
2015-12-17 16:26:48 -08:00
|
|
|
{
|
2016-03-19 06:14:31 -07:00
|
|
|
return ZSTD_decompress_usingDict(dctx, dst, dstCapacity, src, srcSize, NULL, 0);
|
2015-12-17 16:26:48 -08:00
|
|
|
}
|
|
|
|
|
2016-03-11 12:58:04 -08:00
|
|
|
|
2016-03-19 06:14:31 -07:00
|
|
|
size_t ZSTD_decompress(void* dst, size_t dstCapacity, const void* src, size_t srcSize)
|
2015-11-11 04:43:58 -08:00
|
|
|
{
|
2016-01-11 03:56:11 -08:00
|
|
|
#if defined(ZSTD_HEAPMODE) && (ZSTD_HEAPMODE==1)
|
|
|
|
size_t regenSize;
|
|
|
|
ZSTD_DCtx* dctx = ZSTD_createDCtx();
|
|
|
|
if (dctx==NULL) return ERROR(memory_allocation);
|
2016-03-19 06:14:31 -07:00
|
|
|
regenSize = ZSTD_decompressDCtx(dctx, dst, dstCapacity, src, srcSize);
|
2016-01-11 03:56:11 -08:00
|
|
|
ZSTD_freeDCtx(dctx);
|
|
|
|
return regenSize;
|
2016-05-06 07:43:23 -07:00
|
|
|
#else /* stack mode */
|
2015-12-17 16:26:48 -08:00
|
|
|
ZSTD_DCtx dctx;
|
2016-03-19 06:14:31 -07:00
|
|
|
return ZSTD_decompressDCtx(&dctx, dst, dstCapacity, src, srcSize);
|
2016-01-21 07:04:35 -08:00
|
|
|
#endif
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-03-15 07:24:52 -07:00
|
|
|
/*_******************************
|
2015-11-11 04:43:58 -08:00
|
|
|
* Streaming Decompression API
|
|
|
|
********************************/
|
|
|
|
size_t ZSTD_nextSrcSizeToDecompress(ZSTD_DCtx* dctx)
|
|
|
|
{
|
|
|
|
return dctx->expected;
|
|
|
|
}
|
|
|
|
|
2016-05-31 03:43:46 -07:00
|
|
|
int ZSTD_isSkipFrame(ZSTD_DCtx* dctx)
|
|
|
|
{
|
|
|
|
return dctx->stage == ZSTDds_skipFrame;
|
|
|
|
}
|
|
|
|
|
2016-05-31 09:13:56 -07:00
|
|
|
/** ZSTD_decompressContinue() :
|
|
|
|
* @return : nb of bytes generated into `dst` (necessarily <= `dstCapacity)
|
|
|
|
* or an error code, which can be tested using ZSTD_isError() */
|
2016-05-10 05:14:19 -07:00
|
|
|
size_t ZSTD_decompressContinue(ZSTD_DCtx* dctx, void* dst, size_t dstCapacity, const void* src, size_t srcSize)
|
2015-11-11 04:43:58 -08:00
|
|
|
{
|
|
|
|
/* Sanity check */
|
2016-01-25 07:54:05 -08:00
|
|
|
if (srcSize != dctx->expected) return ERROR(srcSize_wrong);
|
2016-05-10 05:14:19 -07:00
|
|
|
if (dstCapacity) ZSTD_checkContinuity(dctx, dst);
|
2015-11-11 04:43:58 -08:00
|
|
|
|
2015-11-25 05:42:45 -08:00
|
|
|
/* Decompress : frame header; part 1 */
|
2016-01-25 07:54:05 -08:00
|
|
|
switch (dctx->stage)
|
2015-11-11 04:43:58 -08:00
|
|
|
{
|
2015-11-25 05:42:45 -08:00
|
|
|
case ZSTDds_getFrameHeaderSize :
|
2016-05-10 05:14:19 -07:00
|
|
|
if (srcSize != ZSTD_frameHeaderSize_min) return ERROR(srcSize_wrong); /* impossible */
|
2016-05-31 03:43:46 -07:00
|
|
|
if ((MEM_readLE32(src) & 0xFFFFFFF0U) == ZSTD_MAGIC_SKIPPABLE_START) {
|
|
|
|
memcpy(dctx->headerBuffer, src, ZSTD_frameHeaderSize_min);
|
|
|
|
dctx->expected = ZSTD_skippableHeaderSize - ZSTD_frameHeaderSize_min; /* magic number + skippable frame length */
|
|
|
|
dctx->stage = ZSTDds_decodeSkippableHeader;
|
|
|
|
return 0;
|
|
|
|
}
|
2016-05-10 05:14:19 -07:00
|
|
|
dctx->headerSize = ZSTD_frameHeaderSize(src, ZSTD_frameHeaderSize_min);
|
|
|
|
if (ZSTD_isError(dctx->headerSize)) return dctx->headerSize;
|
|
|
|
memcpy(dctx->headerBuffer, src, ZSTD_frameHeaderSize_min);
|
|
|
|
if (dctx->headerSize > ZSTD_frameHeaderSize_min) {
|
|
|
|
dctx->expected = dctx->headerSize - ZSTD_frameHeaderSize_min;
|
|
|
|
dctx->stage = ZSTDds_decodeFrameHeader;
|
|
|
|
return 0;
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
2016-05-10 05:14:19 -07:00
|
|
|
dctx->expected = 0; /* not necessary to copy more */
|
|
|
|
|
2015-11-25 05:42:45 -08:00
|
|
|
case ZSTDds_decodeFrameHeader:
|
2016-05-10 05:14:19 -07:00
|
|
|
{ size_t result;
|
2016-01-25 07:54:05 -08:00
|
|
|
memcpy(dctx->headerBuffer + ZSTD_frameHeaderSize_min, src, dctx->expected);
|
2016-03-11 16:25:40 -08:00
|
|
|
result = ZSTD_decodeFrameHeader(dctx, dctx->headerBuffer, dctx->headerSize);
|
2015-11-25 05:42:45 -08:00
|
|
|
if (ZSTD_isError(result)) return result;
|
2016-01-25 07:54:05 -08:00
|
|
|
dctx->expected = ZSTD_blockHeaderSize;
|
|
|
|
dctx->stage = ZSTDds_decodeBlockHeader;
|
2015-11-25 05:42:45 -08:00
|
|
|
return 0;
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
2015-11-25 05:42:45 -08:00
|
|
|
case ZSTDds_decodeBlockHeader:
|
2016-05-10 05:14:19 -07:00
|
|
|
{ blockProperties_t bp;
|
2016-03-23 17:27:55 -07:00
|
|
|
size_t const cBlockSize = ZSTD_getcBlockSize(src, ZSTD_blockHeaderSize, &bp);
|
|
|
|
if (ZSTD_isError(cBlockSize)) return cBlockSize;
|
2016-01-26 06:58:49 -08:00
|
|
|
if (bp.blockType == bt_end) {
|
2016-05-31 15:18:28 -07:00
|
|
|
if (dctx->fParams.checksumFlag) {
|
|
|
|
U64 const h64 = XXH64_digest(&dctx->xxhState);
|
|
|
|
U32 const h32 = (U32)(h64>>11) & ((1<<22)-1);
|
|
|
|
const BYTE* const ip = (const BYTE*)src;
|
|
|
|
U32 const check32 = ip[2] + (ip[1] << 8) + ((ip[0] & 0x3F) << 16);
|
|
|
|
if (check32 != h32) return ERROR(checksum_wrong);
|
|
|
|
}
|
2016-01-25 07:54:05 -08:00
|
|
|
dctx->expected = 0;
|
|
|
|
dctx->stage = ZSTDds_getFrameHeaderSize;
|
2016-03-23 17:27:55 -07:00
|
|
|
} else {
|
|
|
|
dctx->expected = cBlockSize;
|
2016-01-25 07:54:05 -08:00
|
|
|
dctx->bType = bp.blockType;
|
|
|
|
dctx->stage = ZSTDds_decompressBlock;
|
2015-11-25 05:42:45 -08:00
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
2015-12-04 08:16:37 -08:00
|
|
|
case ZSTDds_decompressBlock:
|
2016-05-10 05:14:19 -07:00
|
|
|
{ size_t rSize;
|
2016-01-25 07:54:05 -08:00
|
|
|
switch(dctx->bType)
|
2015-11-25 05:42:45 -08:00
|
|
|
{
|
|
|
|
case bt_compressed:
|
2016-05-10 05:14:19 -07:00
|
|
|
rSize = ZSTD_decompressBlock_internal(dctx, dst, dstCapacity, src, srcSize);
|
2015-11-25 05:42:45 -08:00
|
|
|
break;
|
|
|
|
case bt_raw :
|
2016-05-10 05:14:19 -07:00
|
|
|
rSize = ZSTD_copyRawBlock(dst, dstCapacity, src, srcSize);
|
2015-11-25 05:42:45 -08:00
|
|
|
break;
|
|
|
|
case bt_rle :
|
|
|
|
return ERROR(GENERIC); /* not yet handled */
|
|
|
|
break;
|
|
|
|
case bt_end : /* should never happen (filtered at phase 1) */
|
|
|
|
rSize = 0;
|
|
|
|
break;
|
|
|
|
default:
|
2016-01-26 06:58:49 -08:00
|
|
|
return ERROR(GENERIC); /* impossible */
|
2015-11-25 05:42:45 -08:00
|
|
|
}
|
2016-01-25 07:54:05 -08:00
|
|
|
dctx->stage = ZSTDds_decodeBlockHeader;
|
|
|
|
dctx->expected = ZSTD_blockHeaderSize;
|
|
|
|
dctx->previousDstEnd = (char*)dst + rSize;
|
2016-05-31 15:44:36 -07:00
|
|
|
if (ZSTD_isError(rSize)) return rSize;
|
2016-05-31 15:18:28 -07:00
|
|
|
if (dctx->fParams.checksumFlag) XXH64_update(&dctx->xxhState, dst, rSize);
|
2015-11-25 05:42:45 -08:00
|
|
|
return rSize;
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
2016-05-31 03:43:46 -07:00
|
|
|
case ZSTDds_decodeSkippableHeader:
|
|
|
|
{ memcpy(dctx->headerBuffer + ZSTD_frameHeaderSize_min, src, dctx->expected);
|
|
|
|
dctx->expected = MEM_readLE32(dctx->headerBuffer + 4);
|
|
|
|
dctx->stage = ZSTDds_skipFrame;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case ZSTDds_skipFrame:
|
2016-05-31 10:36:51 -07:00
|
|
|
{ dctx->expected = 0;
|
2016-05-31 03:43:46 -07:00
|
|
|
dctx->stage = ZSTDds_getFrameHeaderSize;
|
|
|
|
return 0;
|
|
|
|
}
|
2015-11-25 05:42:45 -08:00
|
|
|
default:
|
|
|
|
return ERROR(GENERIC); /* impossible */
|
2015-11-11 04:43:58 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-01-25 18:14:20 -08:00
|
|
|
static void ZSTD_refDictContent(ZSTD_DCtx* dctx, const void* dict, size_t dictSize)
|
2015-12-04 08:16:37 -08:00
|
|
|
{
|
2016-01-25 07:54:05 -08:00
|
|
|
dctx->dictEnd = dctx->previousDstEnd;
|
|
|
|
dctx->vBase = (const char*)dict - ((const char*)(dctx->previousDstEnd) - (const char*)(dctx->base));
|
|
|
|
dctx->base = dict;
|
|
|
|
dctx->previousDstEnd = (const char*)dict + dictSize;
|
2015-12-04 08:16:37 -08:00
|
|
|
}
|
2016-01-25 18:14:20 -08:00
|
|
|
|
2016-05-31 15:18:28 -07:00
|
|
|
static size_t ZSTD_loadEntropy(ZSTD_DCtx* dctx, const void* dict, size_t const dictSizeStart)
|
2016-01-25 18:14:20 -08:00
|
|
|
{
|
2016-05-31 15:18:28 -07:00
|
|
|
size_t dictSize = dictSizeStart;
|
2016-01-27 15:18:06 -08:00
|
|
|
|
2016-05-31 15:18:28 -07:00
|
|
|
{ size_t const hSize = HUF_readDTableX4(dctx->hufTableX4, dict, dictSize);
|
|
|
|
if (HUF_isError(hSize)) return ERROR(dictionary_corrupted);
|
|
|
|
dict = (const char*)dict + hSize;
|
|
|
|
dictSize -= hSize;
|
|
|
|
}
|
2016-01-27 15:18:06 -08:00
|
|
|
|
2016-05-06 07:43:23 -07:00
|
|
|
{ short offcodeNCount[MaxOff+1];
|
|
|
|
U32 offcodeMaxValue=MaxOff, offcodeLog=OffFSELog;
|
2016-05-31 15:18:28 -07:00
|
|
|
size_t const offcodeHeaderSize = FSE_readNCount(offcodeNCount, &offcodeMaxValue, &offcodeLog, dict, dictSize);
|
2016-05-06 07:43:23 -07:00
|
|
|
if (FSE_isError(offcodeHeaderSize)) return ERROR(dictionary_corrupted);
|
|
|
|
{ size_t const errorCode = FSE_buildDTable(dctx->OffTable, offcodeNCount, offcodeMaxValue, offcodeLog);
|
|
|
|
if (FSE_isError(errorCode)) return ERROR(dictionary_corrupted); }
|
|
|
|
dict = (const char*)dict + offcodeHeaderSize;
|
|
|
|
dictSize -= offcodeHeaderSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
{ short matchlengthNCount[MaxML+1];
|
|
|
|
unsigned matchlengthMaxValue = MaxML, matchlengthLog = MLFSELog;
|
2016-05-31 15:18:28 -07:00
|
|
|
size_t const matchlengthHeaderSize = FSE_readNCount(matchlengthNCount, &matchlengthMaxValue, &matchlengthLog, dict, dictSize);
|
2016-05-06 07:43:23 -07:00
|
|
|
if (FSE_isError(matchlengthHeaderSize)) return ERROR(dictionary_corrupted);
|
|
|
|
{ size_t const errorCode = FSE_buildDTable(dctx->MLTable, matchlengthNCount, matchlengthMaxValue, matchlengthLog);
|
|
|
|
if (FSE_isError(errorCode)) return ERROR(dictionary_corrupted); }
|
|
|
|
dict = (const char*)dict + matchlengthHeaderSize;
|
|
|
|
dictSize -= matchlengthHeaderSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
{ short litlengthNCount[MaxLL+1];
|
|
|
|
unsigned litlengthMaxValue = MaxLL, litlengthLog = LLFSELog;
|
2016-05-31 15:18:28 -07:00
|
|
|
size_t const litlengthHeaderSize = FSE_readNCount(litlengthNCount, &litlengthMaxValue, &litlengthLog, dict, dictSize);
|
2016-05-06 07:43:23 -07:00
|
|
|
if (FSE_isError(litlengthHeaderSize)) return ERROR(dictionary_corrupted);
|
|
|
|
{ size_t const errorCode = FSE_buildDTable(dctx->LLTable, litlengthNCount, litlengthMaxValue, litlengthLog);
|
|
|
|
if (FSE_isError(errorCode)) return ERROR(dictionary_corrupted); }
|
2016-05-31 15:18:28 -07:00
|
|
|
dictSize -= litlengthHeaderSize;
|
2016-05-06 07:43:23 -07:00
|
|
|
}
|
2016-01-27 15:18:06 -08:00
|
|
|
|
2016-04-09 08:26:22 -07:00
|
|
|
dctx->flagRepeatTable = 1;
|
2016-05-31 15:18:28 -07:00
|
|
|
return dictSizeStart - dictSize;
|
2016-01-25 18:14:20 -08:00
|
|
|
}
|
|
|
|
|
2016-01-26 06:58:49 -08:00
|
|
|
static size_t ZSTD_decompress_insertDictionary(ZSTD_DCtx* dctx, const void* dict, size_t dictSize)
|
2016-01-25 18:14:20 -08:00
|
|
|
{
|
2016-05-28 20:01:04 -07:00
|
|
|
if (dictSize < 8) return ERROR(dictionary_corrupted);
|
|
|
|
{ U32 const magic = MEM_readLE32(dict);
|
|
|
|
if (magic != ZSTD_DICT_MAGIC) {
|
|
|
|
/* pure content mode */
|
|
|
|
ZSTD_refDictContent(dctx, dict, dictSize);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
dctx->dictID = MEM_readLE32((const char*)dict + 4);
|
|
|
|
|
|
|
|
/* load entropy tables */
|
|
|
|
dict = (const char*)dict + 8;
|
|
|
|
dictSize -= 8;
|
|
|
|
{ size_t const eSize = ZSTD_loadEntropy(dctx, dict, dictSize);
|
|
|
|
if (ZSTD_isError(eSize)) return ERROR(dictionary_corrupted);
|
|
|
|
dict = (const char*)dict + eSize;
|
|
|
|
dictSize -= eSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* reference dictionary content */
|
2016-01-25 18:14:20 -08:00
|
|
|
ZSTD_refDictContent(dctx, dict, dictSize);
|
2016-05-28 20:01:04 -07:00
|
|
|
|
2016-01-25 18:14:20 -08:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-01-26 06:58:49 -08:00
|
|
|
|
|
|
|
size_t ZSTD_decompressBegin_usingDict(ZSTD_DCtx* dctx, const void* dict, size_t dictSize)
|
|
|
|
{
|
2016-03-23 17:27:55 -07:00
|
|
|
{ size_t const errorCode = ZSTD_decompressBegin(dctx);
|
|
|
|
if (ZSTD_isError(errorCode)) return errorCode; }
|
2016-01-26 06:58:49 -08:00
|
|
|
|
|
|
|
if (dict && dictSize) {
|
2016-03-23 17:27:55 -07:00
|
|
|
size_t const errorCode = ZSTD_decompress_insertDictionary(dctx, dict, dictSize);
|
2016-01-26 06:58:49 -08:00
|
|
|
if (ZSTD_isError(errorCode)) return ERROR(dictionary_corrupted);
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|