Path: blob/main/sys/contrib/zstd/lib/legacy/zstd_v03.c
48378 views
/*1* Copyright (c) Yann Collet, Facebook, Inc.2* All rights reserved.3*4* This source code is licensed under both the BSD-style license (found in the5* LICENSE file in the root directory of this source tree) and the GPLv2 (found6* in the COPYING file in the root directory of this source tree).7* You may select, at your option, one of the above-listed licenses.8*/91011#include <stddef.h> /* size_t, ptrdiff_t */12#include "zstd_v03.h"13#include "../common/error_private.h"141516/******************************************17* Compiler-specific18******************************************/19#if defined(_MSC_VER) /* Visual Studio */20# include <stdlib.h> /* _byteswap_ulong */21# include <intrin.h> /* _byteswap_* */22#endif23242526/* ******************************************************************27mem.h28low-level memory access routines29Copyright (C) 2013-2015, Yann Collet.3031BSD 2-Clause License (http://www.opensource.org/licenses/bsd-license.php)3233Redistribution and use in source and binary forms, with or without34modification, are permitted provided that the following conditions are35met:3637* Redistributions of source code must retain the above copyright38notice, this list of conditions and the following disclaimer.39* Redistributions in binary form must reproduce the above40copyright notice, this list of conditions and the following disclaimer41in the documentation and/or other materials provided with the42distribution.4344THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS45"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT46LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR47A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT48OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,49SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT50LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,51DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY52THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT53(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE54OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.5556You can contact the author at :57- FSE source repository : https://github.com/Cyan4973/FiniteStateEntropy58- Public forum : https://groups.google.com/forum/#!forum/lz4c59****************************************************************** */60#ifndef MEM_H_MODULE61#define MEM_H_MODULE6263#if defined (__cplusplus)64extern "C" {65#endif6667/******************************************68* Includes69******************************************/70#include <stddef.h> /* size_t, ptrdiff_t */71#include <string.h> /* memcpy */727374/******************************************75* Compiler-specific76******************************************/77#if defined(__GNUC__)78# define MEM_STATIC static __attribute__((unused))79#elif defined (__cplusplus) || (defined (__STDC_VERSION__) && (__STDC_VERSION__ >= 199901L) /* C99 */)80# define MEM_STATIC static inline81#elif defined(_MSC_VER)82# define MEM_STATIC static __inline83#else84# define MEM_STATIC static /* this version may generate warnings for unused static functions; disable the relevant warning */85#endif868788/****************************************************************89* Basic Types90*****************************************************************/91#if defined (__cplusplus) || (defined (__STDC_VERSION__) && (__STDC_VERSION__ >= 199901L) /* C99 */)92# if defined(_AIX)93# include <inttypes.h>94# else95# include <stdint.h> /* intptr_t */96# endif97typedef uint8_t BYTE;98typedef uint16_t U16;99typedef int16_t S16;100typedef uint32_t U32;101typedef int32_t S32;102typedef uint64_t U64;103typedef int64_t S64;104#else105typedef unsigned char BYTE;106typedef unsigned short U16;107typedef signed short S16;108typedef unsigned int U32;109typedef signed int S32;110typedef unsigned long long U64;111typedef signed long long S64;112#endif113114115/****************************************************************116* Memory I/O117*****************************************************************/118/* MEM_FORCE_MEMORY_ACCESS119* By default, access to unaligned memory is controlled by `memcpy()`, which is safe and portable.120* Unfortunately, on some target/compiler combinations, the generated assembly is sub-optimal.121* The below switch allow to select different access method for improved performance.122* Method 0 (default) : use `memcpy()`. Safe and portable.123* Method 1 : `__packed` statement. It depends on compiler extension (ie, not portable).124* This method is safe if your compiler supports it, and *generally* as fast or faster than `memcpy`.125* Method 2 : direct access. This method is portable but violate C standard.126* It can generate buggy code on targets generating assembly depending on alignment.127* But in some circumstances, it's the only known way to get the most performance (ie GCC + ARMv6)128* See http://fastcompression.blogspot.fr/2015/08/accessing-unaligned-memory.html for details.129* Prefer these methods in priority order (0 > 1 > 2)130*/131#ifndef MEM_FORCE_MEMORY_ACCESS /* can be defined externally, on command line for example */132# if defined(__INTEL_COMPILER) || defined(__GNUC__) || defined(__ICCARM__)133# define MEM_FORCE_MEMORY_ACCESS 1134# endif135#endif136137MEM_STATIC unsigned MEM_32bits(void) { return sizeof(void*)==4; }138MEM_STATIC unsigned MEM_64bits(void) { return sizeof(void*)==8; }139140MEM_STATIC unsigned MEM_isLittleEndian(void)141{142const union { U32 u; BYTE c[4]; } one = { 1 }; /* don't use static : performance detrimental */143return one.c[0];144}145146#if defined(MEM_FORCE_MEMORY_ACCESS) && (MEM_FORCE_MEMORY_ACCESS==2)147148/* violates C standard on structure alignment.149Only use if no other choice to achieve best performance on target platform */150MEM_STATIC U16 MEM_read16(const void* memPtr) { return *(const U16*) memPtr; }151MEM_STATIC U32 MEM_read32(const void* memPtr) { return *(const U32*) memPtr; }152MEM_STATIC U64 MEM_read64(const void* memPtr) { return *(const U64*) memPtr; }153154MEM_STATIC void MEM_write16(void* memPtr, U16 value) { *(U16*)memPtr = value; }155156#elif defined(MEM_FORCE_MEMORY_ACCESS) && (MEM_FORCE_MEMORY_ACCESS==1)157158/* __pack instructions are safer, but compiler specific, hence potentially problematic for some compilers */159/* currently only defined for gcc and icc */160typedef union { U16 u16; U32 u32; U64 u64; } __attribute__((packed)) unalign;161162MEM_STATIC U16 MEM_read16(const void* ptr) { return ((const unalign*)ptr)->u16; }163MEM_STATIC U32 MEM_read32(const void* ptr) { return ((const unalign*)ptr)->u32; }164MEM_STATIC U64 MEM_read64(const void* ptr) { return ((const unalign*)ptr)->u64; }165166MEM_STATIC void MEM_write16(void* memPtr, U16 value) { ((unalign*)memPtr)->u16 = value; }167168#else169170/* default method, safe and standard.171can sometimes prove slower */172173MEM_STATIC U16 MEM_read16(const void* memPtr)174{175U16 val; memcpy(&val, memPtr, sizeof(val)); return val;176}177178MEM_STATIC U32 MEM_read32(const void* memPtr)179{180U32 val; memcpy(&val, memPtr, sizeof(val)); return val;181}182183MEM_STATIC U64 MEM_read64(const void* memPtr)184{185U64 val; memcpy(&val, memPtr, sizeof(val)); return val;186}187188MEM_STATIC void MEM_write16(void* memPtr, U16 value)189{190memcpy(memPtr, &value, sizeof(value));191}192193194#endif /* MEM_FORCE_MEMORY_ACCESS */195196197MEM_STATIC U16 MEM_readLE16(const void* memPtr)198{199if (MEM_isLittleEndian())200return MEM_read16(memPtr);201else202{203const BYTE* p = (const BYTE*)memPtr;204return (U16)(p[0] + (p[1]<<8));205}206}207208MEM_STATIC void MEM_writeLE16(void* memPtr, U16 val)209{210if (MEM_isLittleEndian())211{212MEM_write16(memPtr, val);213}214else215{216BYTE* p = (BYTE*)memPtr;217p[0] = (BYTE)val;218p[1] = (BYTE)(val>>8);219}220}221222MEM_STATIC U32 MEM_readLE24(const void* memPtr)223{224return MEM_readLE16(memPtr) + (((const BYTE*)memPtr)[2] << 16);225}226227MEM_STATIC U32 MEM_readLE32(const void* memPtr)228{229if (MEM_isLittleEndian())230return MEM_read32(memPtr);231else232{233const BYTE* p = (const BYTE*)memPtr;234return (U32)((U32)p[0] + ((U32)p[1]<<8) + ((U32)p[2]<<16) + ((U32)p[3]<<24));235}236}237238MEM_STATIC U64 MEM_readLE64(const void* memPtr)239{240if (MEM_isLittleEndian())241return MEM_read64(memPtr);242else243{244const BYTE* p = (const BYTE*)memPtr;245return (U64)((U64)p[0] + ((U64)p[1]<<8) + ((U64)p[2]<<16) + ((U64)p[3]<<24)246+ ((U64)p[4]<<32) + ((U64)p[5]<<40) + ((U64)p[6]<<48) + ((U64)p[7]<<56));247}248}249250251MEM_STATIC size_t MEM_readLEST(const void* memPtr)252{253if (MEM_32bits())254return (size_t)MEM_readLE32(memPtr);255else256return (size_t)MEM_readLE64(memPtr);257}258259260#if defined (__cplusplus)261}262#endif263264#endif /* MEM_H_MODULE */265266267/* ******************************************************************268bitstream269Part of NewGen Entropy library270header file (to include)271Copyright (C) 2013-2015, Yann Collet.272273BSD 2-Clause License (http://www.opensource.org/licenses/bsd-license.php)274275Redistribution and use in source and binary forms, with or without276modification, are permitted provided that the following conditions are277met:278279* Redistributions of source code must retain the above copyright280notice, this list of conditions and the following disclaimer.281* Redistributions in binary form must reproduce the above282copyright notice, this list of conditions and the following disclaimer283in the documentation and/or other materials provided with the284distribution.285286THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS287"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT288LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR289A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT290OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,291SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT292LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,293DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY294THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT295(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE296OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.297298You can contact the author at :299- Source repository : https://github.com/Cyan4973/FiniteStateEntropy300- Public forum : https://groups.google.com/forum/#!forum/lz4c301****************************************************************** */302#ifndef BITSTREAM_H_MODULE303#define BITSTREAM_H_MODULE304305#if defined (__cplusplus)306extern "C" {307#endif308309310/*311* This API consists of small unitary functions, which highly benefit from being inlined.312* Since link-time-optimization is not available for all compilers,313* these functions are defined into a .h to be included.314*/315316317/**********************************************318* bitStream decompression API (read backward)319**********************************************/320typedef struct321{322size_t bitContainer;323unsigned bitsConsumed;324const char* ptr;325const char* start;326} BIT_DStream_t;327328typedef enum { BIT_DStream_unfinished = 0,329BIT_DStream_endOfBuffer = 1,330BIT_DStream_completed = 2,331BIT_DStream_overflow = 3 } BIT_DStream_status; /* result of BIT_reloadDStream() */332/* 1,2,4,8 would be better for bitmap combinations, but slows down performance a bit ... :( */333334MEM_STATIC size_t BIT_initDStream(BIT_DStream_t* bitD, const void* srcBuffer, size_t srcSize);335MEM_STATIC size_t BIT_readBits(BIT_DStream_t* bitD, unsigned nbBits);336MEM_STATIC BIT_DStream_status BIT_reloadDStream(BIT_DStream_t* bitD);337MEM_STATIC unsigned BIT_endOfDStream(const BIT_DStream_t* bitD);338339340341/******************************************342* unsafe API343******************************************/344MEM_STATIC size_t BIT_readBitsFast(BIT_DStream_t* bitD, unsigned nbBits);345/* faster, but works only if nbBits >= 1 */346347348349/****************************************************************350* Helper functions351****************************************************************/352MEM_STATIC unsigned BIT_highbit32 (U32 val)353{354# if defined(_MSC_VER) /* Visual */355unsigned long r;356return _BitScanReverse(&r, val) ? (unsigned)r : 0;357# elif defined(__GNUC__) && (__GNUC__ >= 3) /* Use GCC Intrinsic */358return __builtin_clz (val) ^ 31;359# else /* Software version */360static const unsigned DeBruijnClz[32] = { 0, 9, 1, 10, 13, 21, 2, 29, 11, 14, 16, 18, 22, 25, 3, 30, 8, 12, 20, 28, 15, 17, 24, 7, 19, 27, 23, 6, 26, 5, 4, 31 };361U32 v = val;362unsigned r;363v |= v >> 1;364v |= v >> 2;365v |= v >> 4;366v |= v >> 8;367v |= v >> 16;368r = DeBruijnClz[ (U32) (v * 0x07C4ACDDU) >> 27];369return r;370# endif371}372373374375/**********************************************************376* bitStream decoding377**********************************************************/378379/*!BIT_initDStream380* Initialize a BIT_DStream_t.381* @bitD : a pointer to an already allocated BIT_DStream_t structure382* @srcBuffer must point at the beginning of a bitStream383* @srcSize must be the exact size of the bitStream384* @result : size of stream (== srcSize) or an errorCode if a problem is detected385*/386MEM_STATIC size_t BIT_initDStream(BIT_DStream_t* bitD, const void* srcBuffer, size_t srcSize)387{388if (srcSize < 1) { memset(bitD, 0, sizeof(*bitD)); return ERROR(srcSize_wrong); }389390if (srcSize >= sizeof(size_t)) /* normal case */391{392U32 contain32;393bitD->start = (const char*)srcBuffer;394bitD->ptr = (const char*)srcBuffer + srcSize - sizeof(size_t);395bitD->bitContainer = MEM_readLEST(bitD->ptr);396contain32 = ((const BYTE*)srcBuffer)[srcSize-1];397if (contain32 == 0) return ERROR(GENERIC); /* endMark not present */398bitD->bitsConsumed = 8 - BIT_highbit32(contain32);399}400else401{402U32 contain32;403bitD->start = (const char*)srcBuffer;404bitD->ptr = bitD->start;405bitD->bitContainer = *(const BYTE*)(bitD->start);406switch(srcSize)407{408case 7: bitD->bitContainer += (size_t)(((const BYTE*)(bitD->start))[6]) << (sizeof(size_t)*8 - 16);409/* fallthrough */410case 6: bitD->bitContainer += (size_t)(((const BYTE*)(bitD->start))[5]) << (sizeof(size_t)*8 - 24);411/* fallthrough */412case 5: bitD->bitContainer += (size_t)(((const BYTE*)(bitD->start))[4]) << (sizeof(size_t)*8 - 32);413/* fallthrough */414case 4: bitD->bitContainer += (size_t)(((const BYTE*)(bitD->start))[3]) << 24;415/* fallthrough */416case 3: bitD->bitContainer += (size_t)(((const BYTE*)(bitD->start))[2]) << 16;417/* fallthrough */418case 2: bitD->bitContainer += (size_t)(((const BYTE*)(bitD->start))[1]) << 8;419/* fallthrough */420default:;421}422contain32 = ((const BYTE*)srcBuffer)[srcSize-1];423if (contain32 == 0) return ERROR(GENERIC); /* endMark not present */424bitD->bitsConsumed = 8 - BIT_highbit32(contain32);425bitD->bitsConsumed += (U32)(sizeof(size_t) - srcSize)*8;426}427428return srcSize;429}430MEM_STATIC size_t BIT_lookBits(BIT_DStream_t* bitD, U32 nbBits)431{432const U32 bitMask = sizeof(bitD->bitContainer)*8 - 1;433return ((bitD->bitContainer << (bitD->bitsConsumed & bitMask)) >> 1) >> ((bitMask-nbBits) & bitMask);434}435436/*! BIT_lookBitsFast :437* unsafe version; only works only if nbBits >= 1 */438MEM_STATIC size_t BIT_lookBitsFast(BIT_DStream_t* bitD, U32 nbBits)439{440const U32 bitMask = sizeof(bitD->bitContainer)*8 - 1;441return (bitD->bitContainer << (bitD->bitsConsumed & bitMask)) >> (((bitMask+1)-nbBits) & bitMask);442}443444MEM_STATIC void BIT_skipBits(BIT_DStream_t* bitD, U32 nbBits)445{446bitD->bitsConsumed += nbBits;447}448449MEM_STATIC size_t BIT_readBits(BIT_DStream_t* bitD, U32 nbBits)450{451size_t value = BIT_lookBits(bitD, nbBits);452BIT_skipBits(bitD, nbBits);453return value;454}455456/*!BIT_readBitsFast :457* unsafe version; only works only if nbBits >= 1 */458MEM_STATIC size_t BIT_readBitsFast(BIT_DStream_t* bitD, U32 nbBits)459{460size_t value = BIT_lookBitsFast(bitD, nbBits);461BIT_skipBits(bitD, nbBits);462return value;463}464465MEM_STATIC BIT_DStream_status BIT_reloadDStream(BIT_DStream_t* bitD)466{467if (bitD->bitsConsumed > (sizeof(bitD->bitContainer)*8)) /* should never happen */468return BIT_DStream_overflow;469470if (bitD->ptr >= bitD->start + sizeof(bitD->bitContainer))471{472bitD->ptr -= bitD->bitsConsumed >> 3;473bitD->bitsConsumed &= 7;474bitD->bitContainer = MEM_readLEST(bitD->ptr);475return BIT_DStream_unfinished;476}477if (bitD->ptr == bitD->start)478{479if (bitD->bitsConsumed < sizeof(bitD->bitContainer)*8) return BIT_DStream_endOfBuffer;480return BIT_DStream_completed;481}482{483U32 nbBytes = bitD->bitsConsumed >> 3;484BIT_DStream_status result = BIT_DStream_unfinished;485if (bitD->ptr - nbBytes < bitD->start)486{487nbBytes = (U32)(bitD->ptr - bitD->start); /* ptr > start */488result = BIT_DStream_endOfBuffer;489}490bitD->ptr -= nbBytes;491bitD->bitsConsumed -= nbBytes*8;492bitD->bitContainer = MEM_readLEST(bitD->ptr); /* reminder : srcSize > sizeof(bitD) */493return result;494}495}496497/*! BIT_endOfDStream498* @return Tells if DStream has reached its exact end499*/500MEM_STATIC unsigned BIT_endOfDStream(const BIT_DStream_t* DStream)501{502return ((DStream->ptr == DStream->start) && (DStream->bitsConsumed == sizeof(DStream->bitContainer)*8));503}504505#if defined (__cplusplus)506}507#endif508509#endif /* BITSTREAM_H_MODULE */510/* ******************************************************************511Error codes and messages512Copyright (C) 2013-2015, Yann Collet513514BSD 2-Clause License (http://www.opensource.org/licenses/bsd-license.php)515516Redistribution and use in source and binary forms, with or without517modification, are permitted provided that the following conditions are518met:519520* Redistributions of source code must retain the above copyright521notice, this list of conditions and the following disclaimer.522* Redistributions in binary form must reproduce the above523copyright notice, this list of conditions and the following disclaimer524in the documentation and/or other materials provided with the525distribution.526527THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS528"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT529LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR530A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT531OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,532SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT533LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,534DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY535THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT536(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE537OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.538539You can contact the author at :540- Source repository : https://github.com/Cyan4973/FiniteStateEntropy541- Public forum : https://groups.google.com/forum/#!forum/lz4c542****************************************************************** */543#ifndef ERROR_H_MODULE544#define ERROR_H_MODULE545546#if defined (__cplusplus)547extern "C" {548#endif549550551/******************************************552* Compiler-specific553******************************************/554#if defined (__cplusplus) || (defined (__STDC_VERSION__) && (__STDC_VERSION__ >= 199901L) /* C99 */)555# define ERR_STATIC static inline556#elif defined(_MSC_VER)557# define ERR_STATIC static __inline558#elif defined(__GNUC__)559# define ERR_STATIC static __attribute__((unused))560#else561# define ERR_STATIC static /* this version may generate warnings for unused static functions; disable the relevant warning */562#endif563564565/******************************************566* Error Management567******************************************/568#define PREFIX(name) ZSTD_error_##name569570#define ERROR(name) (size_t)-PREFIX(name)571572#define ERROR_LIST(ITEM) \573ITEM(PREFIX(No_Error)) ITEM(PREFIX(GENERIC)) \574ITEM(PREFIX(dstSize_tooSmall)) ITEM(PREFIX(srcSize_wrong)) \575ITEM(PREFIX(prefix_unknown)) ITEM(PREFIX(corruption_detected)) \576ITEM(PREFIX(tableLog_tooLarge)) ITEM(PREFIX(maxSymbolValue_tooLarge)) ITEM(PREFIX(maxSymbolValue_tooSmall)) \577ITEM(PREFIX(maxCode))578579#define ERROR_GENERATE_ENUM(ENUM) ENUM,580typedef enum { ERROR_LIST(ERROR_GENERATE_ENUM) } ERR_codes; /* enum is exposed, to detect & handle specific errors; compare function result to -enum value */581582#define ERROR_CONVERTTOSTRING(STRING) #STRING,583#define ERROR_GENERATE_STRING(EXPR) ERROR_CONVERTTOSTRING(EXPR)584static const char* ERR_strings[] = { ERROR_LIST(ERROR_GENERATE_STRING) };585586ERR_STATIC unsigned ERR_isError(size_t code) { return (code > ERROR(maxCode)); }587588ERR_STATIC const char* ERR_getErrorName(size_t code)589{590static const char* codeError = "Unspecified error code";591if (ERR_isError(code)) return ERR_strings[-(int)(code)];592return codeError;593}594595596#if defined (__cplusplus)597}598#endif599600#endif /* ERROR_H_MODULE */601/*602Constructor and Destructor of type FSE_CTable603Note that its size depends on 'tableLog' and 'maxSymbolValue' */604typedef unsigned FSE_CTable; /* don't allocate that. It's just a way to be more restrictive than void* */605typedef unsigned FSE_DTable; /* don't allocate that. It's just a way to be more restrictive than void* */606607608/* ******************************************************************609FSE : Finite State Entropy coder610header file for static linking (only)611Copyright (C) 2013-2015, Yann Collet612613BSD 2-Clause License (http://www.opensource.org/licenses/bsd-license.php)614615Redistribution and use in source and binary forms, with or without616modification, are permitted provided that the following conditions are617met:618619* Redistributions of source code must retain the above copyright620notice, this list of conditions and the following disclaimer.621* Redistributions in binary form must reproduce the above622copyright notice, this list of conditions and the following disclaimer623in the documentation and/or other materials provided with the624distribution.625626THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS627"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT628LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR629A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT630OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,631SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT632LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,633DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY634THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT635(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE636OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.637638You can contact the author at :639- Source repository : https://github.com/Cyan4973/FiniteStateEntropy640- Public forum : https://groups.google.com/forum/#!forum/lz4c641****************************************************************** */642#if defined (__cplusplus)643extern "C" {644#endif645646647/******************************************648* Static allocation649******************************************/650/* FSE buffer bounds */651#define FSE_NCOUNTBOUND 512652#define FSE_BLOCKBOUND(size) (size + (size>>7))653#define FSE_COMPRESSBOUND(size) (FSE_NCOUNTBOUND + FSE_BLOCKBOUND(size)) /* Macro version, useful for static allocation */654655/* You can statically allocate FSE CTable/DTable as a table of unsigned using below macro */656#define FSE_CTABLE_SIZE_U32(maxTableLog, maxSymbolValue) (1 + (1<<(maxTableLog-1)) + ((maxSymbolValue+1)*2))657#define FSE_DTABLE_SIZE_U32(maxTableLog) (1 + (1<<maxTableLog))658659660/******************************************661* FSE advanced API662******************************************/663static size_t FSE_buildDTable_raw (FSE_DTable* dt, unsigned nbBits);664/* build a fake FSE_DTable, designed to read an uncompressed bitstream where each symbol uses nbBits */665666static size_t FSE_buildDTable_rle (FSE_DTable* dt, unsigned char symbolValue);667/* build a fake FSE_DTable, designed to always generate the same symbolValue */668669670/******************************************671* FSE symbol decompression API672******************************************/673typedef struct674{675size_t state;676const void* table; /* precise table may vary, depending on U16 */677} FSE_DState_t;678679680static void FSE_initDState(FSE_DState_t* DStatePtr, BIT_DStream_t* bitD, const FSE_DTable* dt);681682static unsigned char FSE_decodeSymbol(FSE_DState_t* DStatePtr, BIT_DStream_t* bitD);683684static unsigned FSE_endOfDState(const FSE_DState_t* DStatePtr);685686687/******************************************688* FSE unsafe API689******************************************/690static unsigned char FSE_decodeSymbolFast(FSE_DState_t* DStatePtr, BIT_DStream_t* bitD);691/* faster, but works only if nbBits is always >= 1 (otherwise, result will be corrupted) */692693694/******************************************695* Implementation of inline functions696******************************************/697698/* decompression */699700typedef struct {701U16 tableLog;702U16 fastMode;703} FSE_DTableHeader; /* sizeof U32 */704705typedef struct706{707unsigned short newState;708unsigned char symbol;709unsigned char nbBits;710} FSE_decode_t; /* size == U32 */711712MEM_STATIC void FSE_initDState(FSE_DState_t* DStatePtr, BIT_DStream_t* bitD, const FSE_DTable* dt)713{714FSE_DTableHeader DTableH;715memcpy(&DTableH, dt, sizeof(DTableH));716DStatePtr->state = BIT_readBits(bitD, DTableH.tableLog);717BIT_reloadDStream(bitD);718DStatePtr->table = dt + 1;719}720721MEM_STATIC BYTE FSE_decodeSymbol(FSE_DState_t* DStatePtr, BIT_DStream_t* bitD)722{723const FSE_decode_t DInfo = ((const FSE_decode_t*)(DStatePtr->table))[DStatePtr->state];724const U32 nbBits = DInfo.nbBits;725BYTE symbol = DInfo.symbol;726size_t lowBits = BIT_readBits(bitD, nbBits);727728DStatePtr->state = DInfo.newState + lowBits;729return symbol;730}731732MEM_STATIC BYTE FSE_decodeSymbolFast(FSE_DState_t* DStatePtr, BIT_DStream_t* bitD)733{734const FSE_decode_t DInfo = ((const FSE_decode_t*)(DStatePtr->table))[DStatePtr->state];735const U32 nbBits = DInfo.nbBits;736BYTE symbol = DInfo.symbol;737size_t lowBits = BIT_readBitsFast(bitD, nbBits);738739DStatePtr->state = DInfo.newState + lowBits;740return symbol;741}742743MEM_STATIC unsigned FSE_endOfDState(const FSE_DState_t* DStatePtr)744{745return DStatePtr->state == 0;746}747748749#if defined (__cplusplus)750}751#endif752/* ******************************************************************753Huff0 : Huffman coder, part of New Generation Entropy library754header file for static linking (only)755Copyright (C) 2013-2015, Yann Collet756757BSD 2-Clause License (http://www.opensource.org/licenses/bsd-license.php)758759Redistribution and use in source and binary forms, with or without760modification, are permitted provided that the following conditions are761met:762763* Redistributions of source code must retain the above copyright764notice, this list of conditions and the following disclaimer.765* Redistributions in binary form must reproduce the above766copyright notice, this list of conditions and the following disclaimer767in the documentation and/or other materials provided with the768distribution.769770THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS771"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT772LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR773A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT774OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,775SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT776LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,777DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY778THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT779(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE780OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.781782You can contact the author at :783- Source repository : https://github.com/Cyan4973/FiniteStateEntropy784- Public forum : https://groups.google.com/forum/#!forum/lz4c785****************************************************************** */786787#if defined (__cplusplus)788extern "C" {789#endif790791/******************************************792* Static allocation macros793******************************************/794/* Huff0 buffer bounds */795#define HUF_CTABLEBOUND 129796#define HUF_BLOCKBOUND(size) (size + (size>>8) + 8) /* only true if incompressible pre-filtered with fast heuristic */797#define HUF_COMPRESSBOUND(size) (HUF_CTABLEBOUND + HUF_BLOCKBOUND(size)) /* Macro version, useful for static allocation */798799/* static allocation of Huff0's DTable */800#define HUF_DTABLE_SIZE(maxTableLog) (1 + (1<<maxTableLog)) /* nb Cells; use unsigned short for X2, unsigned int for X4 */801#define HUF_CREATE_STATIC_DTABLEX2(DTable, maxTableLog) \802unsigned short DTable[HUF_DTABLE_SIZE(maxTableLog)] = { maxTableLog }803#define HUF_CREATE_STATIC_DTABLEX4(DTable, maxTableLog) \804unsigned int DTable[HUF_DTABLE_SIZE(maxTableLog)] = { maxTableLog }805#define HUF_CREATE_STATIC_DTABLEX6(DTable, maxTableLog) \806unsigned int DTable[HUF_DTABLE_SIZE(maxTableLog) * 3 / 2] = { maxTableLog }807808809/******************************************810* Advanced functions811******************************************/812static size_t HUF_decompress4X2 (void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize); /* single-symbol decoder */813static size_t HUF_decompress4X4 (void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize); /* double-symbols decoder */814815816#if defined (__cplusplus)817}818#endif819820/*821zstd - standard compression library822Header File823Copyright (C) 2014-2015, Yann Collet.824825BSD 2-Clause License (http://www.opensource.org/licenses/bsd-license.php)826827Redistribution and use in source and binary forms, with or without828modification, are permitted provided that the following conditions are829met:830* Redistributions of source code must retain the above copyright831notice, this list of conditions and the following disclaimer.832* Redistributions in binary form must reproduce the above833copyright notice, this list of conditions and the following disclaimer834in the documentation and/or other materials provided with the835distribution.836THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS837"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT838LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR839A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT840OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,841SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT842LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,843DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY844THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT845(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE846OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.847848You can contact the author at :849- zstd source repository : https://github.com/Cyan4973/zstd850- ztsd public forum : https://groups.google.com/forum/#!forum/lz4c851*/852853#if defined (__cplusplus)854extern "C" {855#endif856857/* *************************************858* Includes859***************************************/860#include <stddef.h> /* size_t */861862863/* *************************************864* Version865***************************************/866#define ZSTD_VERSION_MAJOR 0 /* for breaking interface changes */867#define ZSTD_VERSION_MINOR 2 /* for new (non-breaking) interface capabilities */868#define ZSTD_VERSION_RELEASE 2 /* for tweaks, bug-fixes, or development */869#define ZSTD_VERSION_NUMBER (ZSTD_VERSION_MAJOR *100*100 + ZSTD_VERSION_MINOR *100 + ZSTD_VERSION_RELEASE)870871872/* *************************************873* Advanced functions874***************************************/875typedef struct ZSTD_CCtx_s ZSTD_CCtx; /* incomplete type */876877#if defined (__cplusplus)878}879#endif880/*881zstd - standard compression library882Header File for static linking only883Copyright (C) 2014-2015, Yann Collet.884885BSD 2-Clause License (http://www.opensource.org/licenses/bsd-license.php)886887Redistribution and use in source and binary forms, with or without888modification, are permitted provided that the following conditions are889met:890* Redistributions of source code must retain the above copyright891notice, this list of conditions and the following disclaimer.892* Redistributions in binary form must reproduce the above893copyright notice, this list of conditions and the following disclaimer894in the documentation and/or other materials provided with the895distribution.896THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS897"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT898LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR899A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT900OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,901SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT902LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,903DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY904THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT905(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE906OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.907908You can contact the author at :909- zstd source repository : https://github.com/Cyan4973/zstd910- ztsd public forum : https://groups.google.com/forum/#!forum/lz4c911*/912913/* The objects defined into this file should be considered experimental.914* They are not labelled stable, as their prototype may change in the future.915* You can use them for tests, provide feedback, or if you can endure risk of future changes.916*/917918#if defined (__cplusplus)919extern "C" {920#endif921922/* *************************************923* Streaming functions924***************************************/925926typedef struct ZSTD_DCtx_s ZSTD_DCtx;927928/*929Use above functions alternatively.930ZSTD_nextSrcSizeToDecompress() tells how much bytes to provide as 'srcSize' to ZSTD_decompressContinue().931ZSTD_decompressContinue() will use previous data blocks to improve compression if they are located prior to current block.932Result is the number of bytes regenerated within 'dst'.933It can be zero, which is not an error; it just means ZSTD_decompressContinue() has decoded some header.934*/935936/* *************************************937* Prefix - version detection938***************************************/939#define ZSTD_magicNumber 0xFD2FB523 /* v0.3 */940941942#if defined (__cplusplus)943}944#endif945/* ******************************************************************946FSE : Finite State Entropy coder947Copyright (C) 2013-2015, Yann Collet.948949BSD 2-Clause License (http://www.opensource.org/licenses/bsd-license.php)950951Redistribution and use in source and binary forms, with or without952modification, are permitted provided that the following conditions are953met:954955* Redistributions of source code must retain the above copyright956notice, this list of conditions and the following disclaimer.957* Redistributions in binary form must reproduce the above958copyright notice, this list of conditions and the following disclaimer959in the documentation and/or other materials provided with the960distribution.961962THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS963"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT964LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR965A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT966OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,967SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT968LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,969DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY970THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT971(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE972OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.973974You can contact the author at :975- FSE source repository : https://github.com/Cyan4973/FiniteStateEntropy976- Public forum : https://groups.google.com/forum/#!forum/lz4c977****************************************************************** */978979#ifndef FSE_COMMONDEFS_ONLY980981/****************************************************************982* Tuning parameters983****************************************************************/984/* MEMORY_USAGE :985* Memory usage formula : N->2^N Bytes (examples : 10 -> 1KB; 12 -> 4KB ; 16 -> 64KB; 20 -> 1MB; etc.)986* Increasing memory usage improves compression ratio987* Reduced memory usage can improve speed, due to cache effect988* Recommended max value is 14, for 16KB, which nicely fits into Intel x86 L1 cache */989#define FSE_MAX_MEMORY_USAGE 14990#define FSE_DEFAULT_MEMORY_USAGE 13991992/* FSE_MAX_SYMBOL_VALUE :993* Maximum symbol value authorized.994* Required for proper stack allocation */995#define FSE_MAX_SYMBOL_VALUE 255996997998/****************************************************************999* template functions type & suffix1000****************************************************************/1001#define FSE_FUNCTION_TYPE BYTE1002#define FSE_FUNCTION_EXTENSION100310041005/****************************************************************1006* Byte symbol type1007****************************************************************/1008#endif /* !FSE_COMMONDEFS_ONLY */100910101011/****************************************************************1012* Compiler specifics1013****************************************************************/1014#ifdef _MSC_VER /* Visual Studio */1015# define FORCE_INLINE static __forceinline1016# include <intrin.h> /* For Visual 2005 */1017# pragma warning(disable : 4127) /* disable: C4127: conditional expression is constant */1018# pragma warning(disable : 4214) /* disable: C4214: non-int bitfields */1019#else1020# if defined (__cplusplus) || defined (__STDC_VERSION__) && __STDC_VERSION__ >= 199901L /* C99 */1021# ifdef __GNUC__1022# define FORCE_INLINE static inline __attribute__((always_inline))1023# else1024# define FORCE_INLINE static inline1025# endif1026# else1027# define FORCE_INLINE static1028# endif /* __STDC_VERSION__ */1029#endif103010311032/****************************************************************1033* Includes1034****************************************************************/1035#include <stdlib.h> /* malloc, free, qsort */1036#include <string.h> /* memcpy, memset */1037#include <stdio.h> /* printf (debug) */10381039/****************************************************************1040* Constants1041*****************************************************************/1042#define FSE_MAX_TABLELOG (FSE_MAX_MEMORY_USAGE-2)1043#define FSE_MAX_TABLESIZE (1U<<FSE_MAX_TABLELOG)1044#define FSE_MAXTABLESIZE_MASK (FSE_MAX_TABLESIZE-1)1045#define FSE_DEFAULT_TABLELOG (FSE_DEFAULT_MEMORY_USAGE-2)1046#define FSE_MIN_TABLELOG 510471048#define FSE_TABLELOG_ABSOLUTE_MAX 151049#if FSE_MAX_TABLELOG > FSE_TABLELOG_ABSOLUTE_MAX1050#error "FSE_MAX_TABLELOG > FSE_TABLELOG_ABSOLUTE_MAX is not supported"1051#endif105210531054/****************************************************************1055* Error Management1056****************************************************************/1057#define FSE_STATIC_ASSERT(c) { enum { FSE_static_assert = 1/(int)(!!(c)) }; } /* use only *after* variable declarations */105810591060/****************************************************************1061* Complex types1062****************************************************************/1063typedef U32 DTable_max_t[FSE_DTABLE_SIZE_U32(FSE_MAX_TABLELOG)];106410651066/****************************************************************1067* Templates1068****************************************************************/1069/*1070designed to be included1071for type-specific functions (template emulation in C)1072Objective is to write these functions only once, for improved maintenance1073*/10741075/* safety checks */1076#ifndef FSE_FUNCTION_EXTENSION1077# error "FSE_FUNCTION_EXTENSION must be defined"1078#endif1079#ifndef FSE_FUNCTION_TYPE1080# error "FSE_FUNCTION_TYPE must be defined"1081#endif10821083/* Function names */1084#define FSE_CAT(X,Y) X##Y1085#define FSE_FUNCTION_NAME(X,Y) FSE_CAT(X,Y)1086#define FSE_TYPE_NAME(X,Y) FSE_CAT(X,Y)108710881089/* Function templates */10901091#define FSE_DECODE_TYPE FSE_decode_t10921093static U32 FSE_tableStep(U32 tableSize) { return (tableSize>>1) + (tableSize>>3) + 3; }10941095static size_t FSE_buildDTable1096(FSE_DTable* dt, const short* normalizedCounter, unsigned maxSymbolValue, unsigned tableLog)1097{1098void* ptr = dt+1;1099FSE_DTableHeader DTableH;1100FSE_DECODE_TYPE* const tableDecode = (FSE_DECODE_TYPE*)ptr;1101const U32 tableSize = 1 << tableLog;1102const U32 tableMask = tableSize-1;1103const U32 step = FSE_tableStep(tableSize);1104U16 symbolNext[FSE_MAX_SYMBOL_VALUE+1];1105U32 position = 0;1106U32 highThreshold = tableSize-1;1107const S16 largeLimit= (S16)(1 << (tableLog-1));1108U32 noLarge = 1;1109U32 s;11101111/* Sanity Checks */1112if (maxSymbolValue > FSE_MAX_SYMBOL_VALUE) return ERROR(maxSymbolValue_tooLarge);1113if (tableLog > FSE_MAX_TABLELOG) return ERROR(tableLog_tooLarge);11141115/* Init, lay down lowprob symbols */1116DTableH.tableLog = (U16)tableLog;1117for (s=0; s<=maxSymbolValue; s++)1118{1119if (normalizedCounter[s]==-1)1120{1121tableDecode[highThreshold--].symbol = (FSE_FUNCTION_TYPE)s;1122symbolNext[s] = 1;1123}1124else1125{1126if (normalizedCounter[s] >= largeLimit) noLarge=0;1127symbolNext[s] = normalizedCounter[s];1128}1129}11301131/* Spread symbols */1132for (s=0; s<=maxSymbolValue; s++)1133{1134int i;1135for (i=0; i<normalizedCounter[s]; i++)1136{1137tableDecode[position].symbol = (FSE_FUNCTION_TYPE)s;1138position = (position + step) & tableMask;1139while (position > highThreshold) position = (position + step) & tableMask; /* lowprob area */1140}1141}11421143if (position!=0) return ERROR(GENERIC); /* position must reach all cells once, otherwise normalizedCounter is incorrect */11441145/* Build Decoding table */1146{1147U32 i;1148for (i=0; i<tableSize; i++)1149{1150FSE_FUNCTION_TYPE symbol = (FSE_FUNCTION_TYPE)(tableDecode[i].symbol);1151U16 nextState = symbolNext[symbol]++;1152tableDecode[i].nbBits = (BYTE) (tableLog - BIT_highbit32 ((U32)nextState) );1153tableDecode[i].newState = (U16) ( (nextState << tableDecode[i].nbBits) - tableSize);1154}1155}11561157DTableH.fastMode = (U16)noLarge;1158memcpy(dt, &DTableH, sizeof(DTableH));1159return 0;1160}116111621163#ifndef FSE_COMMONDEFS_ONLY1164/******************************************1165* FSE helper functions1166******************************************/1167static unsigned FSE_isError(size_t code) { return ERR_isError(code); }116811691170/****************************************************************1171* FSE NCount encoding-decoding1172****************************************************************/1173static short FSE_abs(short a)1174{1175return a<0 ? -a : a;1176}11771178static size_t FSE_readNCount (short* normalizedCounter, unsigned* maxSVPtr, unsigned* tableLogPtr,1179const void* headerBuffer, size_t hbSize)1180{1181const BYTE* const istart = (const BYTE*) headerBuffer;1182const BYTE* const iend = istart + hbSize;1183const BYTE* ip = istart;1184int nbBits;1185int remaining;1186int threshold;1187U32 bitStream;1188int bitCount;1189unsigned charnum = 0;1190int previous0 = 0;11911192if (hbSize < 4) return ERROR(srcSize_wrong);1193bitStream = MEM_readLE32(ip);1194nbBits = (bitStream & 0xF) + FSE_MIN_TABLELOG; /* extract tableLog */1195if (nbBits > FSE_TABLELOG_ABSOLUTE_MAX) return ERROR(tableLog_tooLarge);1196bitStream >>= 4;1197bitCount = 4;1198*tableLogPtr = nbBits;1199remaining = (1<<nbBits)+1;1200threshold = 1<<nbBits;1201nbBits++;12021203while ((remaining>1) && (charnum<=*maxSVPtr))1204{1205if (previous0)1206{1207unsigned n0 = charnum;1208while ((bitStream & 0xFFFF) == 0xFFFF)1209{1210n0+=24;1211if (ip < iend-5)1212{1213ip+=2;1214bitStream = MEM_readLE32(ip) >> bitCount;1215}1216else1217{1218bitStream >>= 16;1219bitCount+=16;1220}1221}1222while ((bitStream & 3) == 3)1223{1224n0+=3;1225bitStream>>=2;1226bitCount+=2;1227}1228n0 += bitStream & 3;1229bitCount += 2;1230if (n0 > *maxSVPtr) return ERROR(maxSymbolValue_tooSmall);1231while (charnum < n0) normalizedCounter[charnum++] = 0;1232if ((ip <= iend-7) || (ip + (bitCount>>3) <= iend-4))1233{1234ip += bitCount>>3;1235bitCount &= 7;1236bitStream = MEM_readLE32(ip) >> bitCount;1237}1238else1239bitStream >>= 2;1240}1241{1242const short max = (short)((2*threshold-1)-remaining);1243short count;12441245if ((bitStream & (threshold-1)) < (U32)max)1246{1247count = (short)(bitStream & (threshold-1));1248bitCount += nbBits-1;1249}1250else1251{1252count = (short)(bitStream & (2*threshold-1));1253if (count >= threshold) count -= max;1254bitCount += nbBits;1255}12561257count--; /* extra accuracy */1258remaining -= FSE_abs(count);1259normalizedCounter[charnum++] = count;1260previous0 = !count;1261while (remaining < threshold)1262{1263nbBits--;1264threshold >>= 1;1265}12661267{1268if ((ip <= iend-7) || (ip + (bitCount>>3) <= iend-4))1269{1270ip += bitCount>>3;1271bitCount &= 7;1272}1273else1274{1275bitCount -= (int)(8 * (iend - 4 - ip));1276ip = iend - 4;1277}1278bitStream = MEM_readLE32(ip) >> (bitCount & 31);1279}1280}1281}1282if (remaining != 1) return ERROR(GENERIC);1283*maxSVPtr = charnum-1;12841285ip += (bitCount+7)>>3;1286if ((size_t)(ip-istart) > hbSize) return ERROR(srcSize_wrong);1287return ip-istart;1288}128912901291/*********************************************************1292* Decompression (Byte symbols)1293*********************************************************/1294static size_t FSE_buildDTable_rle (FSE_DTable* dt, BYTE symbolValue)1295{1296void* ptr = dt;1297FSE_DTableHeader* const DTableH = (FSE_DTableHeader*)ptr;1298FSE_decode_t* const cell = (FSE_decode_t*)(ptr) + 1;12991300DTableH->tableLog = 0;1301DTableH->fastMode = 0;13021303cell->newState = 0;1304cell->symbol = symbolValue;1305cell->nbBits = 0;13061307return 0;1308}130913101311static size_t FSE_buildDTable_raw (FSE_DTable* dt, unsigned nbBits)1312{1313void* ptr = dt;1314FSE_DTableHeader* const DTableH = (FSE_DTableHeader*)ptr;1315FSE_decode_t* const dinfo = (FSE_decode_t*)(ptr) + 1;1316const unsigned tableSize = 1 << nbBits;1317const unsigned tableMask = tableSize - 1;1318const unsigned maxSymbolValue = tableMask;1319unsigned s;13201321/* Sanity checks */1322if (nbBits < 1) return ERROR(GENERIC); /* min size */13231324/* Build Decoding Table */1325DTableH->tableLog = (U16)nbBits;1326DTableH->fastMode = 1;1327for (s=0; s<=maxSymbolValue; s++)1328{1329dinfo[s].newState = 0;1330dinfo[s].symbol = (BYTE)s;1331dinfo[s].nbBits = (BYTE)nbBits;1332}13331334return 0;1335}13361337FORCE_INLINE size_t FSE_decompress_usingDTable_generic(1338void* dst, size_t maxDstSize,1339const void* cSrc, size_t cSrcSize,1340const FSE_DTable* dt, const unsigned fast)1341{1342BYTE* const ostart = (BYTE*) dst;1343BYTE* op = ostart;1344BYTE* const omax = op + maxDstSize;1345BYTE* const olimit = omax-3;13461347BIT_DStream_t bitD;1348FSE_DState_t state1;1349FSE_DState_t state2;1350size_t errorCode;13511352/* Init */1353errorCode = BIT_initDStream(&bitD, cSrc, cSrcSize); /* replaced last arg by maxCompressed Size */1354if (FSE_isError(errorCode)) return errorCode;13551356FSE_initDState(&state1, &bitD, dt);1357FSE_initDState(&state2, &bitD, dt);13581359#define FSE_GETSYMBOL(statePtr) fast ? FSE_decodeSymbolFast(statePtr, &bitD) : FSE_decodeSymbol(statePtr, &bitD)13601361/* 4 symbols per loop */1362for ( ; (BIT_reloadDStream(&bitD)==BIT_DStream_unfinished) && (op<olimit) ; op+=4)1363{1364op[0] = FSE_GETSYMBOL(&state1);13651366if (FSE_MAX_TABLELOG*2+7 > sizeof(bitD.bitContainer)*8) /* This test must be static */1367BIT_reloadDStream(&bitD);13681369op[1] = FSE_GETSYMBOL(&state2);13701371if (FSE_MAX_TABLELOG*4+7 > sizeof(bitD.bitContainer)*8) /* This test must be static */1372{ if (BIT_reloadDStream(&bitD) > BIT_DStream_unfinished) { op+=2; break; } }13731374op[2] = FSE_GETSYMBOL(&state1);13751376if (FSE_MAX_TABLELOG*2+7 > sizeof(bitD.bitContainer)*8) /* This test must be static */1377BIT_reloadDStream(&bitD);13781379op[3] = FSE_GETSYMBOL(&state2);1380}13811382/* tail */1383/* note : BIT_reloadDStream(&bitD) >= FSE_DStream_partiallyFilled; Ends at exactly BIT_DStream_completed */1384while (1)1385{1386if ( (BIT_reloadDStream(&bitD)>BIT_DStream_completed) || (op==omax) || (BIT_endOfDStream(&bitD) && (fast || FSE_endOfDState(&state1))) )1387break;13881389*op++ = FSE_GETSYMBOL(&state1);13901391if ( (BIT_reloadDStream(&bitD)>BIT_DStream_completed) || (op==omax) || (BIT_endOfDStream(&bitD) && (fast || FSE_endOfDState(&state2))) )1392break;13931394*op++ = FSE_GETSYMBOL(&state2);1395}13961397/* end ? */1398if (BIT_endOfDStream(&bitD) && FSE_endOfDState(&state1) && FSE_endOfDState(&state2))1399return op-ostart;14001401if (op==omax) return ERROR(dstSize_tooSmall); /* dst buffer is full, but cSrc unfinished */14021403return ERROR(corruption_detected);1404}140514061407static size_t FSE_decompress_usingDTable(void* dst, size_t originalSize,1408const void* cSrc, size_t cSrcSize,1409const FSE_DTable* dt)1410{1411FSE_DTableHeader DTableH;1412memcpy(&DTableH, dt, sizeof(DTableH));14131414/* select fast mode (static) */1415if (DTableH.fastMode) return FSE_decompress_usingDTable_generic(dst, originalSize, cSrc, cSrcSize, dt, 1);1416return FSE_decompress_usingDTable_generic(dst, originalSize, cSrc, cSrcSize, dt, 0);1417}141814191420static size_t FSE_decompress(void* dst, size_t maxDstSize, const void* cSrc, size_t cSrcSize)1421{1422const BYTE* const istart = (const BYTE*)cSrc;1423const BYTE* ip = istart;1424short counting[FSE_MAX_SYMBOL_VALUE+1];1425DTable_max_t dt; /* Static analyzer seems unable to understand this table will be properly initialized later */1426unsigned tableLog;1427unsigned maxSymbolValue = FSE_MAX_SYMBOL_VALUE;1428size_t errorCode;14291430if (cSrcSize<2) return ERROR(srcSize_wrong); /* too small input size */14311432/* normal FSE decoding mode */1433errorCode = FSE_readNCount (counting, &maxSymbolValue, &tableLog, istart, cSrcSize);1434if (FSE_isError(errorCode)) return errorCode;1435if (errorCode >= cSrcSize) return ERROR(srcSize_wrong); /* too small input size */1436ip += errorCode;1437cSrcSize -= errorCode;14381439errorCode = FSE_buildDTable (dt, counting, maxSymbolValue, tableLog);1440if (FSE_isError(errorCode)) return errorCode;14411442/* always return, even if it is an error code */1443return FSE_decompress_usingDTable (dst, maxDstSize, ip, cSrcSize, dt);1444}1445144614471448#endif /* FSE_COMMONDEFS_ONLY */1449/* ******************************************************************1450Huff0 : Huffman coder, part of New Generation Entropy library1451Copyright (C) 2013-2015, Yann Collet.14521453BSD 2-Clause License (http://www.opensource.org/licenses/bsd-license.php)14541455Redistribution and use in source and binary forms, with or without1456modification, are permitted provided that the following conditions are1457met:14581459* Redistributions of source code must retain the above copyright1460notice, this list of conditions and the following disclaimer.1461* Redistributions in binary form must reproduce the above1462copyright notice, this list of conditions and the following disclaimer1463in the documentation and/or other materials provided with the1464distribution.14651466THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS1467"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT1468LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR1469A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT1470OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,1471SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT1472LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,1473DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY1474THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT1475(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE1476OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.14771478You can contact the author at :1479- FSE+Huff0 source repository : https://github.com/Cyan4973/FiniteStateEntropy1480- Public forum : https://groups.google.com/forum/#!forum/lz4c1481****************************************************************** */14821483/****************************************************************1484* Compiler specifics1485****************************************************************/1486#if defined (__cplusplus) || (defined (__STDC_VERSION__) && (__STDC_VERSION__ >= 199901L) /* C99 */)1487/* inline is defined */1488#elif defined(_MSC_VER)1489# pragma warning(disable : 4127) /* disable: C4127: conditional expression is constant */1490# define inline __inline1491#else1492# define inline /* disable inline */1493#endif149414951496/****************************************************************1497* Includes1498****************************************************************/1499#include <stdlib.h> /* malloc, free, qsort */1500#include <string.h> /* memcpy, memset */1501#include <stdio.h> /* printf (debug) */15021503/****************************************************************1504* Error Management1505****************************************************************/1506#define HUF_STATIC_ASSERT(c) { enum { HUF_static_assert = 1/(int)(!!(c)) }; } /* use only *after* variable declarations */150715081509/******************************************1510* Helper functions1511******************************************/1512static unsigned HUF_isError(size_t code) { return ERR_isError(code); }15131514#define HUF_ABSOLUTEMAX_TABLELOG 16 /* absolute limit of HUF_MAX_TABLELOG. Beyond that value, code does not work */1515#define HUF_MAX_TABLELOG 12 /* max configured tableLog (for static allocation); can be modified up to HUF_ABSOLUTEMAX_TABLELOG */1516#define HUF_DEFAULT_TABLELOG HUF_MAX_TABLELOG /* tableLog by default, when not specified */1517#define HUF_MAX_SYMBOL_VALUE 2551518#if (HUF_MAX_TABLELOG > HUF_ABSOLUTEMAX_TABLELOG)1519# error "HUF_MAX_TABLELOG is too large !"1520#endif1521152215231524/*********************************************************1525* Huff0 : Huffman block decompression1526*********************************************************/1527typedef struct { BYTE byte; BYTE nbBits; } HUF_DEltX2; /* single-symbol decoding */15281529typedef struct { U16 sequence; BYTE nbBits; BYTE length; } HUF_DEltX4; /* double-symbols decoding */15301531typedef struct { BYTE symbol; BYTE weight; } sortedSymbol_t;15321533/*! HUF_readStats1534Read compact Huffman tree, saved by HUF_writeCTable1535@huffWeight : destination buffer1536@return : size read from `src`1537*/1538static size_t HUF_readStats(BYTE* huffWeight, size_t hwSize, U32* rankStats,1539U32* nbSymbolsPtr, U32* tableLogPtr,1540const void* src, size_t srcSize)1541{1542U32 weightTotal;1543U32 tableLog;1544const BYTE* ip = (const BYTE*) src;1545size_t iSize;1546size_t oSize;1547U32 n;15481549if (!srcSize) return ERROR(srcSize_wrong);1550iSize = ip[0];1551//memset(huffWeight, 0, hwSize); /* is not necessary, even though some analyzer complain ... */15521553if (iSize >= 128) /* special header */1554{1555if (iSize >= (242)) /* RLE */1556{1557static int l[14] = { 1, 2, 3, 4, 7, 8, 15, 16, 31, 32, 63, 64, 127, 128 };1558oSize = l[iSize-242];1559memset(huffWeight, 1, hwSize);1560iSize = 0;1561}1562else /* Incompressible */1563{1564oSize = iSize - 127;1565iSize = ((oSize+1)/2);1566if (iSize+1 > srcSize) return ERROR(srcSize_wrong);1567if (oSize >= hwSize) return ERROR(corruption_detected);1568ip += 1;1569for (n=0; n<oSize; n+=2)1570{1571huffWeight[n] = ip[n/2] >> 4;1572huffWeight[n+1] = ip[n/2] & 15;1573}1574}1575}1576else /* header compressed with FSE (normal case) */1577{1578if (iSize+1 > srcSize) return ERROR(srcSize_wrong);1579oSize = FSE_decompress(huffWeight, hwSize-1, ip+1, iSize); /* max (hwSize-1) values decoded, as last one is implied */1580if (FSE_isError(oSize)) return oSize;1581}15821583/* collect weight stats */1584memset(rankStats, 0, (HUF_ABSOLUTEMAX_TABLELOG + 1) * sizeof(U32));1585weightTotal = 0;1586for (n=0; n<oSize; n++)1587{1588if (huffWeight[n] >= HUF_ABSOLUTEMAX_TABLELOG) return ERROR(corruption_detected);1589rankStats[huffWeight[n]]++;1590weightTotal += (1 << huffWeight[n]) >> 1;1591}1592if (weightTotal == 0) return ERROR(corruption_detected);15931594/* get last non-null symbol weight (implied, total must be 2^n) */1595tableLog = BIT_highbit32(weightTotal) + 1;1596if (tableLog > HUF_ABSOLUTEMAX_TABLELOG) return ERROR(corruption_detected);1597{1598U32 total = 1 << tableLog;1599U32 rest = total - weightTotal;1600U32 verif = 1 << BIT_highbit32(rest);1601U32 lastWeight = BIT_highbit32(rest) + 1;1602if (verif != rest) return ERROR(corruption_detected); /* last value must be a clean power of 2 */1603huffWeight[oSize] = (BYTE)lastWeight;1604rankStats[lastWeight]++;1605}16061607/* check tree construction validity */1608if ((rankStats[1] < 2) || (rankStats[1] & 1)) return ERROR(corruption_detected); /* by construction : at least 2 elts of rank 1, must be even */16091610/* results */1611*nbSymbolsPtr = (U32)(oSize+1);1612*tableLogPtr = tableLog;1613return iSize+1;1614}161516161617/**************************/1618/* single-symbol decoding */1619/**************************/16201621static size_t HUF_readDTableX2 (U16* DTable, const void* src, size_t srcSize)1622{1623BYTE huffWeight[HUF_MAX_SYMBOL_VALUE + 1];1624U32 rankVal[HUF_ABSOLUTEMAX_TABLELOG + 1]; /* large enough for values from 0 to 16 */1625U32 tableLog = 0;1626const BYTE* ip = (const BYTE*) src;1627size_t iSize = ip[0];1628U32 nbSymbols = 0;1629U32 n;1630U32 nextRankStart;1631void* ptr = DTable+1;1632HUF_DEltX2* const dt = (HUF_DEltX2*)(ptr);16331634HUF_STATIC_ASSERT(sizeof(HUF_DEltX2) == sizeof(U16)); /* if compilation fails here, assertion is false */1635//memset(huffWeight, 0, sizeof(huffWeight)); /* is not necessary, even though some analyzer complain ... */16361637iSize = HUF_readStats(huffWeight, HUF_MAX_SYMBOL_VALUE + 1, rankVal, &nbSymbols, &tableLog, src, srcSize);1638if (HUF_isError(iSize)) return iSize;16391640/* check result */1641if (tableLog > DTable[0]) return ERROR(tableLog_tooLarge); /* DTable is too small */1642DTable[0] = (U16)tableLog; /* maybe should separate sizeof DTable, as allocated, from used size of DTable, in case of DTable re-use */16431644/* Prepare ranks */1645nextRankStart = 0;1646for (n=1; n<=tableLog; n++)1647{1648U32 current = nextRankStart;1649nextRankStart += (rankVal[n] << (n-1));1650rankVal[n] = current;1651}16521653/* fill DTable */1654for (n=0; n<nbSymbols; n++)1655{1656const U32 w = huffWeight[n];1657const U32 length = (1 << w) >> 1;1658U32 i;1659HUF_DEltX2 D;1660D.byte = (BYTE)n; D.nbBits = (BYTE)(tableLog + 1 - w);1661for (i = rankVal[w]; i < rankVal[w] + length; i++)1662dt[i] = D;1663rankVal[w] += length;1664}16651666return iSize;1667}16681669static BYTE HUF_decodeSymbolX2(BIT_DStream_t* Dstream, const HUF_DEltX2* dt, const U32 dtLog)1670{1671const size_t val = BIT_lookBitsFast(Dstream, dtLog); /* note : dtLog >= 1 */1672const BYTE c = dt[val].byte;1673BIT_skipBits(Dstream, dt[val].nbBits);1674return c;1675}16761677#define HUF_DECODE_SYMBOLX2_0(ptr, DStreamPtr) \1678*ptr++ = HUF_decodeSymbolX2(DStreamPtr, dt, dtLog)16791680#define HUF_DECODE_SYMBOLX2_1(ptr, DStreamPtr) \1681if (MEM_64bits() || (HUF_MAX_TABLELOG<=12)) \1682HUF_DECODE_SYMBOLX2_0(ptr, DStreamPtr)16831684#define HUF_DECODE_SYMBOLX2_2(ptr, DStreamPtr) \1685if (MEM_64bits()) \1686HUF_DECODE_SYMBOLX2_0(ptr, DStreamPtr)16871688static inline size_t HUF_decodeStreamX2(BYTE* p, BIT_DStream_t* const bitDPtr, BYTE* const pEnd, const HUF_DEltX2* const dt, const U32 dtLog)1689{1690BYTE* const pStart = p;16911692/* up to 4 symbols at a time */1693while ((BIT_reloadDStream(bitDPtr) == BIT_DStream_unfinished) && (p <= pEnd-4))1694{1695HUF_DECODE_SYMBOLX2_2(p, bitDPtr);1696HUF_DECODE_SYMBOLX2_1(p, bitDPtr);1697HUF_DECODE_SYMBOLX2_2(p, bitDPtr);1698HUF_DECODE_SYMBOLX2_0(p, bitDPtr);1699}17001701/* closer to the end */1702while ((BIT_reloadDStream(bitDPtr) == BIT_DStream_unfinished) && (p < pEnd))1703HUF_DECODE_SYMBOLX2_0(p, bitDPtr);17041705/* no more data to retrieve from bitstream, hence no need to reload */1706while (p < pEnd)1707HUF_DECODE_SYMBOLX2_0(p, bitDPtr);17081709return pEnd-pStart;1710}171117121713static size_t HUF_decompress4X2_usingDTable(1714void* dst, size_t dstSize,1715const void* cSrc, size_t cSrcSize,1716const U16* DTable)1717{1718if (cSrcSize < 10) return ERROR(corruption_detected); /* strict minimum : jump table + 1 byte per stream */17191720{1721const BYTE* const istart = (const BYTE*) cSrc;1722BYTE* const ostart = (BYTE*) dst;1723BYTE* const oend = ostart + dstSize;17241725const void* ptr = DTable;1726const HUF_DEltX2* const dt = ((const HUF_DEltX2*)ptr) +1;1727const U32 dtLog = DTable[0];1728size_t errorCode;17291730/* Init */1731BIT_DStream_t bitD1;1732BIT_DStream_t bitD2;1733BIT_DStream_t bitD3;1734BIT_DStream_t bitD4;1735const size_t length1 = MEM_readLE16(istart);1736const size_t length2 = MEM_readLE16(istart+2);1737const size_t length3 = MEM_readLE16(istart+4);1738size_t length4;1739const BYTE* const istart1 = istart + 6; /* jumpTable */1740const BYTE* const istart2 = istart1 + length1;1741const BYTE* const istart3 = istart2 + length2;1742const BYTE* const istart4 = istart3 + length3;1743const size_t segmentSize = (dstSize+3) / 4;1744BYTE* const opStart2 = ostart + segmentSize;1745BYTE* const opStart3 = opStart2 + segmentSize;1746BYTE* const opStart4 = opStart3 + segmentSize;1747BYTE* op1 = ostart;1748BYTE* op2 = opStart2;1749BYTE* op3 = opStart3;1750BYTE* op4 = opStart4;1751U32 endSignal;17521753length4 = cSrcSize - (length1 + length2 + length3 + 6);1754if (length4 > cSrcSize) return ERROR(corruption_detected); /* overflow */1755errorCode = BIT_initDStream(&bitD1, istart1, length1);1756if (HUF_isError(errorCode)) return errorCode;1757errorCode = BIT_initDStream(&bitD2, istart2, length2);1758if (HUF_isError(errorCode)) return errorCode;1759errorCode = BIT_initDStream(&bitD3, istart3, length3);1760if (HUF_isError(errorCode)) return errorCode;1761errorCode = BIT_initDStream(&bitD4, istart4, length4);1762if (HUF_isError(errorCode)) return errorCode;17631764/* 16-32 symbols per loop (4-8 symbols per stream) */1765endSignal = BIT_reloadDStream(&bitD1) | BIT_reloadDStream(&bitD2) | BIT_reloadDStream(&bitD3) | BIT_reloadDStream(&bitD4);1766for ( ; (endSignal==BIT_DStream_unfinished) && (op4<(oend-7)) ; )1767{1768HUF_DECODE_SYMBOLX2_2(op1, &bitD1);1769HUF_DECODE_SYMBOLX2_2(op2, &bitD2);1770HUF_DECODE_SYMBOLX2_2(op3, &bitD3);1771HUF_DECODE_SYMBOLX2_2(op4, &bitD4);1772HUF_DECODE_SYMBOLX2_1(op1, &bitD1);1773HUF_DECODE_SYMBOLX2_1(op2, &bitD2);1774HUF_DECODE_SYMBOLX2_1(op3, &bitD3);1775HUF_DECODE_SYMBOLX2_1(op4, &bitD4);1776HUF_DECODE_SYMBOLX2_2(op1, &bitD1);1777HUF_DECODE_SYMBOLX2_2(op2, &bitD2);1778HUF_DECODE_SYMBOLX2_2(op3, &bitD3);1779HUF_DECODE_SYMBOLX2_2(op4, &bitD4);1780HUF_DECODE_SYMBOLX2_0(op1, &bitD1);1781HUF_DECODE_SYMBOLX2_0(op2, &bitD2);1782HUF_DECODE_SYMBOLX2_0(op3, &bitD3);1783HUF_DECODE_SYMBOLX2_0(op4, &bitD4);17841785endSignal = BIT_reloadDStream(&bitD1) | BIT_reloadDStream(&bitD2) | BIT_reloadDStream(&bitD3) | BIT_reloadDStream(&bitD4);1786}17871788/* check corruption */1789if (op1 > opStart2) return ERROR(corruption_detected);1790if (op2 > opStart3) return ERROR(corruption_detected);1791if (op3 > opStart4) return ERROR(corruption_detected);1792/* note : op4 supposed already verified within main loop */17931794/* finish bitStreams one by one */1795HUF_decodeStreamX2(op1, &bitD1, opStart2, dt, dtLog);1796HUF_decodeStreamX2(op2, &bitD2, opStart3, dt, dtLog);1797HUF_decodeStreamX2(op3, &bitD3, opStart4, dt, dtLog);1798HUF_decodeStreamX2(op4, &bitD4, oend, dt, dtLog);17991800/* check */1801endSignal = BIT_endOfDStream(&bitD1) & BIT_endOfDStream(&bitD2) & BIT_endOfDStream(&bitD3) & BIT_endOfDStream(&bitD4);1802if (!endSignal) return ERROR(corruption_detected);18031804/* decoded size */1805return dstSize;1806}1807}180818091810static size_t HUF_decompress4X2 (void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize)1811{1812HUF_CREATE_STATIC_DTABLEX2(DTable, HUF_MAX_TABLELOG);1813const BYTE* ip = (const BYTE*) cSrc;1814size_t errorCode;18151816errorCode = HUF_readDTableX2 (DTable, cSrc, cSrcSize);1817if (HUF_isError(errorCode)) return errorCode;1818if (errorCode >= cSrcSize) return ERROR(srcSize_wrong);1819ip += errorCode;1820cSrcSize -= errorCode;18211822return HUF_decompress4X2_usingDTable (dst, dstSize, ip, cSrcSize, DTable);1823}182418251826/***************************/1827/* double-symbols decoding */1828/***************************/18291830static void HUF_fillDTableX4Level2(HUF_DEltX4* DTable, U32 sizeLog, const U32 consumed,1831const U32* rankValOrigin, const int minWeight,1832const sortedSymbol_t* sortedSymbols, const U32 sortedListSize,1833U32 nbBitsBaseline, U16 baseSeq)1834{1835HUF_DEltX4 DElt;1836U32 rankVal[HUF_ABSOLUTEMAX_TABLELOG + 1];1837U32 s;18381839/* get pre-calculated rankVal */1840memcpy(rankVal, rankValOrigin, sizeof(rankVal));18411842/* fill skipped values */1843if (minWeight>1)1844{1845U32 i, skipSize = rankVal[minWeight];1846MEM_writeLE16(&(DElt.sequence), baseSeq);1847DElt.nbBits = (BYTE)(consumed);1848DElt.length = 1;1849for (i = 0; i < skipSize; i++)1850DTable[i] = DElt;1851}18521853/* fill DTable */1854for (s=0; s<sortedListSize; s++) /* note : sortedSymbols already skipped */1855{1856const U32 symbol = sortedSymbols[s].symbol;1857const U32 weight = sortedSymbols[s].weight;1858const U32 nbBits = nbBitsBaseline - weight;1859const U32 length = 1 << (sizeLog-nbBits);1860const U32 start = rankVal[weight];1861U32 i = start;1862const U32 end = start + length;18631864MEM_writeLE16(&(DElt.sequence), (U16)(baseSeq + (symbol << 8)));1865DElt.nbBits = (BYTE)(nbBits + consumed);1866DElt.length = 2;1867do { DTable[i++] = DElt; } while (i<end); /* since length >= 1 */18681869rankVal[weight] += length;1870}1871}18721873typedef U32 rankVal_t[HUF_ABSOLUTEMAX_TABLELOG][HUF_ABSOLUTEMAX_TABLELOG + 1];18741875static void HUF_fillDTableX4(HUF_DEltX4* DTable, const U32 targetLog,1876const sortedSymbol_t* sortedList, const U32 sortedListSize,1877const U32* rankStart, rankVal_t rankValOrigin, const U32 maxWeight,1878const U32 nbBitsBaseline)1879{1880U32 rankVal[HUF_ABSOLUTEMAX_TABLELOG + 1];1881const int scaleLog = nbBitsBaseline - targetLog; /* note : targetLog >= srcLog, hence scaleLog <= 1 */1882const U32 minBits = nbBitsBaseline - maxWeight;1883U32 s;18841885memcpy(rankVal, rankValOrigin, sizeof(rankVal));18861887/* fill DTable */1888for (s=0; s<sortedListSize; s++)1889{1890const U16 symbol = sortedList[s].symbol;1891const U32 weight = sortedList[s].weight;1892const U32 nbBits = nbBitsBaseline - weight;1893const U32 start = rankVal[weight];1894const U32 length = 1 << (targetLog-nbBits);18951896if (targetLog-nbBits >= minBits) /* enough room for a second symbol */1897{1898U32 sortedRank;1899int minWeight = nbBits + scaleLog;1900if (minWeight < 1) minWeight = 1;1901sortedRank = rankStart[minWeight];1902HUF_fillDTableX4Level2(DTable+start, targetLog-nbBits, nbBits,1903rankValOrigin[nbBits], minWeight,1904sortedList+sortedRank, sortedListSize-sortedRank,1905nbBitsBaseline, symbol);1906}1907else1908{1909U32 i;1910const U32 end = start + length;1911HUF_DEltX4 DElt;19121913MEM_writeLE16(&(DElt.sequence), symbol);1914DElt.nbBits = (BYTE)(nbBits);1915DElt.length = 1;1916for (i = start; i < end; i++)1917DTable[i] = DElt;1918}1919rankVal[weight] += length;1920}1921}19221923static size_t HUF_readDTableX4 (U32* DTable, const void* src, size_t srcSize)1924{1925BYTE weightList[HUF_MAX_SYMBOL_VALUE + 1];1926sortedSymbol_t sortedSymbol[HUF_MAX_SYMBOL_VALUE + 1];1927U32 rankStats[HUF_ABSOLUTEMAX_TABLELOG + 1] = { 0 };1928U32 rankStart0[HUF_ABSOLUTEMAX_TABLELOG + 2] = { 0 };1929U32* const rankStart = rankStart0+1;1930rankVal_t rankVal;1931U32 tableLog, maxW, sizeOfSort, nbSymbols;1932const U32 memLog = DTable[0];1933const BYTE* ip = (const BYTE*) src;1934size_t iSize = ip[0];1935void* ptr = DTable;1936HUF_DEltX4* const dt = ((HUF_DEltX4*)ptr) + 1;19371938HUF_STATIC_ASSERT(sizeof(HUF_DEltX4) == sizeof(U32)); /* if compilation fails here, assertion is false */1939if (memLog > HUF_ABSOLUTEMAX_TABLELOG) return ERROR(tableLog_tooLarge);1940//memset(weightList, 0, sizeof(weightList)); /* is not necessary, even though some analyzer complain ... */19411942iSize = HUF_readStats(weightList, HUF_MAX_SYMBOL_VALUE + 1, rankStats, &nbSymbols, &tableLog, src, srcSize);1943if (HUF_isError(iSize)) return iSize;19441945/* check result */1946if (tableLog > memLog) return ERROR(tableLog_tooLarge); /* DTable can't fit code depth */19471948/* find maxWeight */1949for (maxW = tableLog; rankStats[maxW]==0; maxW--)1950{ if (!maxW) return ERROR(GENERIC); } /* necessarily finds a solution before maxW==0 */19511952/* Get start index of each weight */1953{1954U32 w, nextRankStart = 0;1955for (w=1; w<=maxW; w++)1956{1957U32 current = nextRankStart;1958nextRankStart += rankStats[w];1959rankStart[w] = current;1960}1961rankStart[0] = nextRankStart; /* put all 0w symbols at the end of sorted list*/1962sizeOfSort = nextRankStart;1963}19641965/* sort symbols by weight */1966{1967U32 s;1968for (s=0; s<nbSymbols; s++)1969{1970U32 w = weightList[s];1971U32 r = rankStart[w]++;1972sortedSymbol[r].symbol = (BYTE)s;1973sortedSymbol[r].weight = (BYTE)w;1974}1975rankStart[0] = 0; /* forget 0w symbols; this is beginning of weight(1) */1976}19771978/* Build rankVal */1979{1980const U32 minBits = tableLog+1 - maxW;1981U32 nextRankVal = 0;1982U32 w, consumed;1983const int rescale = (memLog-tableLog) - 1; /* tableLog <= memLog */1984U32* rankVal0 = rankVal[0];1985for (w=1; w<=maxW; w++)1986{1987U32 current = nextRankVal;1988nextRankVal += rankStats[w] << (w+rescale);1989rankVal0[w] = current;1990}1991for (consumed = minBits; consumed <= memLog - minBits; consumed++)1992{1993U32* rankValPtr = rankVal[consumed];1994for (w = 1; w <= maxW; w++)1995{1996rankValPtr[w] = rankVal0[w] >> consumed;1997}1998}1999}20002001HUF_fillDTableX4(dt, memLog,2002sortedSymbol, sizeOfSort,2003rankStart0, rankVal, maxW,2004tableLog+1);20052006return iSize;2007}200820092010static U32 HUF_decodeSymbolX4(void* op, BIT_DStream_t* DStream, const HUF_DEltX4* dt, const U32 dtLog)2011{2012const size_t val = BIT_lookBitsFast(DStream, dtLog); /* note : dtLog >= 1 */2013memcpy(op, dt+val, 2);2014BIT_skipBits(DStream, dt[val].nbBits);2015return dt[val].length;2016}20172018static U32 HUF_decodeLastSymbolX4(void* op, BIT_DStream_t* DStream, const HUF_DEltX4* dt, const U32 dtLog)2019{2020const size_t val = BIT_lookBitsFast(DStream, dtLog); /* note : dtLog >= 1 */2021memcpy(op, dt+val, 1);2022if (dt[val].length==1) BIT_skipBits(DStream, dt[val].nbBits);2023else2024{2025if (DStream->bitsConsumed < (sizeof(DStream->bitContainer)*8))2026{2027BIT_skipBits(DStream, dt[val].nbBits);2028if (DStream->bitsConsumed > (sizeof(DStream->bitContainer)*8))2029DStream->bitsConsumed = (sizeof(DStream->bitContainer)*8); /* ugly hack; works only because it's the last symbol. Note : can't easily extract nbBits from just this symbol */2030}2031}2032return 1;2033}203420352036#define HUF_DECODE_SYMBOLX4_0(ptr, DStreamPtr) \2037ptr += HUF_decodeSymbolX4(ptr, DStreamPtr, dt, dtLog)20382039#define HUF_DECODE_SYMBOLX4_1(ptr, DStreamPtr) \2040if (MEM_64bits() || (HUF_MAX_TABLELOG<=12)) \2041ptr += HUF_decodeSymbolX4(ptr, DStreamPtr, dt, dtLog)20422043#define HUF_DECODE_SYMBOLX4_2(ptr, DStreamPtr) \2044if (MEM_64bits()) \2045ptr += HUF_decodeSymbolX4(ptr, DStreamPtr, dt, dtLog)20462047static inline size_t HUF_decodeStreamX4(BYTE* p, BIT_DStream_t* bitDPtr, BYTE* const pEnd, const HUF_DEltX4* const dt, const U32 dtLog)2048{2049BYTE* const pStart = p;20502051/* up to 8 symbols at a time */2052while ((BIT_reloadDStream(bitDPtr) == BIT_DStream_unfinished) && (p < pEnd-7))2053{2054HUF_DECODE_SYMBOLX4_2(p, bitDPtr);2055HUF_DECODE_SYMBOLX4_1(p, bitDPtr);2056HUF_DECODE_SYMBOLX4_2(p, bitDPtr);2057HUF_DECODE_SYMBOLX4_0(p, bitDPtr);2058}20592060/* closer to the end */2061while ((BIT_reloadDStream(bitDPtr) == BIT_DStream_unfinished) && (p <= pEnd-2))2062HUF_DECODE_SYMBOLX4_0(p, bitDPtr);20632064while (p <= pEnd-2)2065HUF_DECODE_SYMBOLX4_0(p, bitDPtr); /* no need to reload : reached the end of DStream */20662067if (p < pEnd)2068p += HUF_decodeLastSymbolX4(p, bitDPtr, dt, dtLog);20692070return p-pStart;2071}2072207320742075static size_t HUF_decompress4X4_usingDTable(2076void* dst, size_t dstSize,2077const void* cSrc, size_t cSrcSize,2078const U32* DTable)2079{2080if (cSrcSize < 10) return ERROR(corruption_detected); /* strict minimum : jump table + 1 byte per stream */20812082{2083const BYTE* const istart = (const BYTE*) cSrc;2084BYTE* const ostart = (BYTE*) dst;2085BYTE* const oend = ostart + dstSize;20862087const void* ptr = DTable;2088const HUF_DEltX4* const dt = ((const HUF_DEltX4*)ptr) +1;2089const U32 dtLog = DTable[0];2090size_t errorCode;20912092/* Init */2093BIT_DStream_t bitD1;2094BIT_DStream_t bitD2;2095BIT_DStream_t bitD3;2096BIT_DStream_t bitD4;2097const size_t length1 = MEM_readLE16(istart);2098const size_t length2 = MEM_readLE16(istart+2);2099const size_t length3 = MEM_readLE16(istart+4);2100size_t length4;2101const BYTE* const istart1 = istart + 6; /* jumpTable */2102const BYTE* const istart2 = istart1 + length1;2103const BYTE* const istart3 = istart2 + length2;2104const BYTE* const istart4 = istart3 + length3;2105const size_t segmentSize = (dstSize+3) / 4;2106BYTE* const opStart2 = ostart + segmentSize;2107BYTE* const opStart3 = opStart2 + segmentSize;2108BYTE* const opStart4 = opStart3 + segmentSize;2109BYTE* op1 = ostart;2110BYTE* op2 = opStart2;2111BYTE* op3 = opStart3;2112BYTE* op4 = opStart4;2113U32 endSignal;21142115length4 = cSrcSize - (length1 + length2 + length3 + 6);2116if (length4 > cSrcSize) return ERROR(corruption_detected); /* overflow */2117errorCode = BIT_initDStream(&bitD1, istart1, length1);2118if (HUF_isError(errorCode)) return errorCode;2119errorCode = BIT_initDStream(&bitD2, istart2, length2);2120if (HUF_isError(errorCode)) return errorCode;2121errorCode = BIT_initDStream(&bitD3, istart3, length3);2122if (HUF_isError(errorCode)) return errorCode;2123errorCode = BIT_initDStream(&bitD4, istart4, length4);2124if (HUF_isError(errorCode)) return errorCode;21252126/* 16-32 symbols per loop (4-8 symbols per stream) */2127endSignal = BIT_reloadDStream(&bitD1) | BIT_reloadDStream(&bitD2) | BIT_reloadDStream(&bitD3) | BIT_reloadDStream(&bitD4);2128for ( ; (endSignal==BIT_DStream_unfinished) && (op4<(oend-7)) ; )2129{2130HUF_DECODE_SYMBOLX4_2(op1, &bitD1);2131HUF_DECODE_SYMBOLX4_2(op2, &bitD2);2132HUF_DECODE_SYMBOLX4_2(op3, &bitD3);2133HUF_DECODE_SYMBOLX4_2(op4, &bitD4);2134HUF_DECODE_SYMBOLX4_1(op1, &bitD1);2135HUF_DECODE_SYMBOLX4_1(op2, &bitD2);2136HUF_DECODE_SYMBOLX4_1(op3, &bitD3);2137HUF_DECODE_SYMBOLX4_1(op4, &bitD4);2138HUF_DECODE_SYMBOLX4_2(op1, &bitD1);2139HUF_DECODE_SYMBOLX4_2(op2, &bitD2);2140HUF_DECODE_SYMBOLX4_2(op3, &bitD3);2141HUF_DECODE_SYMBOLX4_2(op4, &bitD4);2142HUF_DECODE_SYMBOLX4_0(op1, &bitD1);2143HUF_DECODE_SYMBOLX4_0(op2, &bitD2);2144HUF_DECODE_SYMBOLX4_0(op3, &bitD3);2145HUF_DECODE_SYMBOLX4_0(op4, &bitD4);21462147endSignal = BIT_reloadDStream(&bitD1) | BIT_reloadDStream(&bitD2) | BIT_reloadDStream(&bitD3) | BIT_reloadDStream(&bitD4);2148}21492150/* check corruption */2151if (op1 > opStart2) return ERROR(corruption_detected);2152if (op2 > opStart3) return ERROR(corruption_detected);2153if (op3 > opStart4) return ERROR(corruption_detected);2154/* note : op4 supposed already verified within main loop */21552156/* finish bitStreams one by one */2157HUF_decodeStreamX4(op1, &bitD1, opStart2, dt, dtLog);2158HUF_decodeStreamX4(op2, &bitD2, opStart3, dt, dtLog);2159HUF_decodeStreamX4(op3, &bitD3, opStart4, dt, dtLog);2160HUF_decodeStreamX4(op4, &bitD4, oend, dt, dtLog);21612162/* check */2163endSignal = BIT_endOfDStream(&bitD1) & BIT_endOfDStream(&bitD2) & BIT_endOfDStream(&bitD3) & BIT_endOfDStream(&bitD4);2164if (!endSignal) return ERROR(corruption_detected);21652166/* decoded size */2167return dstSize;2168}2169}217021712172static size_t HUF_decompress4X4 (void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize)2173{2174HUF_CREATE_STATIC_DTABLEX4(DTable, HUF_MAX_TABLELOG);2175const BYTE* ip = (const BYTE*) cSrc;21762177size_t hSize = HUF_readDTableX4 (DTable, cSrc, cSrcSize);2178if (HUF_isError(hSize)) return hSize;2179if (hSize >= cSrcSize) return ERROR(srcSize_wrong);2180ip += hSize;2181cSrcSize -= hSize;21822183return HUF_decompress4X4_usingDTable (dst, dstSize, ip, cSrcSize, DTable);2184}218521862187/**********************************/2188/* Generic decompression selector */2189/**********************************/21902191typedef struct { U32 tableTime; U32 decode256Time; } algo_time_t;2192static const algo_time_t algoTime[16 /* Quantization */][3 /* single, double, quad */] =2193{2194/* single, double, quad */2195{{0,0}, {1,1}, {2,2}}, /* Q==0 : impossible */2196{{0,0}, {1,1}, {2,2}}, /* Q==1 : impossible */2197{{ 38,130}, {1313, 74}, {2151, 38}}, /* Q == 2 : 12-18% */2198{{ 448,128}, {1353, 74}, {2238, 41}}, /* Q == 3 : 18-25% */2199{{ 556,128}, {1353, 74}, {2238, 47}}, /* Q == 4 : 25-32% */2200{{ 714,128}, {1418, 74}, {2436, 53}}, /* Q == 5 : 32-38% */2201{{ 883,128}, {1437, 74}, {2464, 61}}, /* Q == 6 : 38-44% */2202{{ 897,128}, {1515, 75}, {2622, 68}}, /* Q == 7 : 44-50% */2203{{ 926,128}, {1613, 75}, {2730, 75}}, /* Q == 8 : 50-56% */2204{{ 947,128}, {1729, 77}, {3359, 77}}, /* Q == 9 : 56-62% */2205{{1107,128}, {2083, 81}, {4006, 84}}, /* Q ==10 : 62-69% */2206{{1177,128}, {2379, 87}, {4785, 88}}, /* Q ==11 : 69-75% */2207{{1242,128}, {2415, 93}, {5155, 84}}, /* Q ==12 : 75-81% */2208{{1349,128}, {2644,106}, {5260,106}}, /* Q ==13 : 81-87% */2209{{1455,128}, {2422,124}, {4174,124}}, /* Q ==14 : 87-93% */2210{{ 722,128}, {1891,145}, {1936,146}}, /* Q ==15 : 93-99% */2211};22122213typedef size_t (*decompressionAlgo)(void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize);22142215static size_t HUF_decompress (void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize)2216{2217static const decompressionAlgo decompress[3] = { HUF_decompress4X2, HUF_decompress4X4, NULL };2218/* estimate decompression time */2219U32 Q;2220const U32 D256 = (U32)(dstSize >> 8);2221U32 Dtime[3];2222U32 algoNb = 0;2223int n;22242225/* validation checks */2226if (dstSize == 0) return ERROR(dstSize_tooSmall);2227if (cSrcSize > dstSize) return ERROR(corruption_detected); /* invalid */2228if (cSrcSize == dstSize) { memcpy(dst, cSrc, dstSize); return dstSize; } /* not compressed */2229if (cSrcSize == 1) { memset(dst, *(const BYTE*)cSrc, dstSize); return dstSize; } /* RLE */22302231/* decoder timing evaluation */2232Q = (U32)(cSrcSize * 16 / dstSize); /* Q < 16 since dstSize > cSrcSize */2233for (n=0; n<3; n++)2234Dtime[n] = algoTime[Q][n].tableTime + (algoTime[Q][n].decode256Time * D256);22352236Dtime[1] += Dtime[1] >> 4; Dtime[2] += Dtime[2] >> 3; /* advantage to algorithms using less memory, for cache eviction */22372238if (Dtime[1] < Dtime[0]) algoNb = 1;22392240return decompress[algoNb](dst, dstSize, cSrc, cSrcSize);22412242//return HUF_decompress4X2(dst, dstSize, cSrc, cSrcSize); /* multi-streams single-symbol decoding */2243//return HUF_decompress4X4(dst, dstSize, cSrc, cSrcSize); /* multi-streams double-symbols decoding */2244//return HUF_decompress4X6(dst, dstSize, cSrc, cSrcSize); /* multi-streams quad-symbols decoding */2245}2246/*2247zstd - standard compression library2248Copyright (C) 2014-2015, Yann Collet.22492250BSD 2-Clause License (http://www.opensource.org/licenses/bsd-license.php)22512252Redistribution and use in source and binary forms, with or without2253modification, are permitted provided that the following conditions are2254met:2255* Redistributions of source code must retain the above copyright2256notice, this list of conditions and the following disclaimer.2257* Redistributions in binary form must reproduce the above2258copyright notice, this list of conditions and the following disclaimer2259in the documentation and/or other materials provided with the2260distribution.2261THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS2262"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT2263LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR2264A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT2265OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,2266SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT2267LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,2268DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY2269THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT2270(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE2271OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.22722273You can contact the author at :2274- zstd source repository : https://github.com/Cyan4973/zstd2275- ztsd public forum : https://groups.google.com/forum/#!forum/lz4c2276*/22772278/* ***************************************************************2279* Tuning parameters2280*****************************************************************/2281/*!2282* MEMORY_USAGE :2283* Memory usage formula : N->2^N Bytes (examples : 10 -> 1KB; 12 -> 4KB ; 16 -> 64KB; 20 -> 1MB; etc.)2284* Increasing memory usage improves compression ratio2285* Reduced memory usage can improve speed, due to cache effect2286*/2287#define ZSTD_MEMORY_USAGE 1722882289/*!2290* HEAPMODE :2291* Select how default compression functions will allocate memory for their hash table,2292* in memory stack (0, fastest), or in memory heap (1, requires malloc())2293* Note that compression context is fairly large, as a consequence heap memory is recommended.2294*/2295#ifndef ZSTD_HEAPMODE2296# define ZSTD_HEAPMODE 12297#endif /* ZSTD_HEAPMODE */22982299/*!2300* LEGACY_SUPPORT :2301* decompressor can decode older formats (starting from Zstd 0.1+)2302*/2303#ifndef ZSTD_LEGACY_SUPPORT2304# define ZSTD_LEGACY_SUPPORT 12305#endif230623072308/* *******************************************************2309* Includes2310*********************************************************/2311#include <stdlib.h> /* calloc */2312#include <string.h> /* memcpy, memmove */2313#include <stdio.h> /* debug : printf */231423152316/* *******************************************************2317* Compiler specifics2318*********************************************************/2319#ifdef __AVX2__2320# include <immintrin.h> /* AVX2 intrinsics */2321#endif23222323#ifdef _MSC_VER /* Visual Studio */2324# include <intrin.h> /* For Visual 2005 */2325# pragma warning(disable : 4127) /* disable: C4127: conditional expression is constant */2326# pragma warning(disable : 4324) /* disable: C4324: padded structure */2327#else2328# define GCC_VERSION (__GNUC__ * 100 + __GNUC_MINOR__)2329#endif233023312332/* *******************************************************2333* Constants2334*********************************************************/2335#define HASH_LOG (ZSTD_MEMORY_USAGE - 2)2336#define HASH_TABLESIZE (1 << HASH_LOG)2337#define HASH_MASK (HASH_TABLESIZE - 1)23382339#define KNUTH 265443576123402341#define BIT7 1282342#define BIT6 642343#define BIT5 322344#define BIT4 162345#define BIT1 22346#define BIT0 123472348#define KB *(1 <<10)2349#define MB *(1 <<20)2350#define GB *(1U<<30)23512352#define BLOCKSIZE (128 KB) /* define, for static allocation */2353#define MIN_SEQUENCES_SIZE (2 /*seqNb*/ + 2 /*dumps*/ + 3 /*seqTables*/ + 1 /*bitStream*/)2354#define MIN_CBLOCK_SIZE (3 /*litCSize*/ + MIN_SEQUENCES_SIZE)2355#define IS_RAW BIT02356#define IS_RLE BIT123572358#define WORKPLACESIZE (BLOCKSIZE*3)2359#define MINMATCH 42360#define MLbits 72361#define LLbits 62362#define Offbits 52363#define MaxML ((1<<MLbits )-1)2364#define MaxLL ((1<<LLbits )-1)2365#define MaxOff 312366#define LitFSELog 112367#define MLFSELog 102368#define LLFSELog 102369#define OffFSELog 92370#define MAX(a,b) ((a)<(b)?(b):(a))2371#define MaxSeq MAX(MaxLL, MaxML)23722373#define LITERAL_NOENTROPY 632374#define COMMAND_NOENTROPY 7 /* to remove */23752376#define ZSTD_CONTENTSIZE_ERROR (0ULL - 2)23772378static const size_t ZSTD_blockHeaderSize = 3;2379static const size_t ZSTD_frameHeaderSize = 4;238023812382/* *******************************************************2383* Memory operations2384**********************************************************/2385static void ZSTD_copy4(void* dst, const void* src) { memcpy(dst, src, 4); }23862387static void ZSTD_copy8(void* dst, const void* src) { memcpy(dst, src, 8); }23882389#define COPY8(d,s) { ZSTD_copy8(d,s); d+=8; s+=8; }23902391/*! ZSTD_wildcopy : custom version of memcpy(), can copy up to 7-8 bytes too many */2392static void ZSTD_wildcopy(void* dst, const void* src, ptrdiff_t length)2393{2394const BYTE* ip = (const BYTE*)src;2395BYTE* op = (BYTE*)dst;2396BYTE* const oend = op + length;2397do COPY8(op, ip) while (op < oend);2398}239924002401/* **************************************2402* Local structures2403****************************************/2404typedef enum { bt_compressed, bt_raw, bt_rle, bt_end } blockType_t;24052406typedef struct2407{2408blockType_t blockType;2409U32 origSize;2410} blockProperties_t;24112412typedef struct {2413void* buffer;2414U32* offsetStart;2415U32* offset;2416BYTE* offCodeStart;2417BYTE* offCode;2418BYTE* litStart;2419BYTE* lit;2420BYTE* litLengthStart;2421BYTE* litLength;2422BYTE* matchLengthStart;2423BYTE* matchLength;2424BYTE* dumpsStart;2425BYTE* dumps;2426} seqStore_t;242724282429/* *************************************2430* Error Management2431***************************************/2432/*! ZSTD_isError2433* tells if a return value is an error code */2434static unsigned ZSTD_isError(size_t code) { return ERR_isError(code); }2435243624372438/* *************************************************************2439* Decompression section2440***************************************************************/2441struct ZSTD_DCtx_s2442{2443U32 LLTable[FSE_DTABLE_SIZE_U32(LLFSELog)];2444U32 OffTable[FSE_DTABLE_SIZE_U32(OffFSELog)];2445U32 MLTable[FSE_DTABLE_SIZE_U32(MLFSELog)];2446void* previousDstEnd;2447void* base;2448size_t expected;2449blockType_t bType;2450U32 phase;2451const BYTE* litPtr;2452size_t litSize;2453BYTE litBuffer[BLOCKSIZE + 8 /* margin for wildcopy */];2454}; /* typedef'd to ZSTD_Dctx within "zstd_static.h" */245524562457static size_t ZSTD_getcBlockSize(const void* src, size_t srcSize, blockProperties_t* bpPtr)2458{2459const BYTE* const in = (const BYTE* const)src;2460BYTE headerFlags;2461U32 cSize;24622463if (srcSize < 3) return ERROR(srcSize_wrong);24642465headerFlags = *in;2466cSize = in[2] + (in[1]<<8) + ((in[0] & 7)<<16);24672468bpPtr->blockType = (blockType_t)(headerFlags >> 6);2469bpPtr->origSize = (bpPtr->blockType == bt_rle) ? cSize : 0;24702471if (bpPtr->blockType == bt_end) return 0;2472if (bpPtr->blockType == bt_rle) return 1;2473return cSize;2474}24752476static size_t ZSTD_copyUncompressedBlock(void* dst, size_t maxDstSize, const void* src, size_t srcSize)2477{2478if (srcSize > maxDstSize) return ERROR(dstSize_tooSmall);2479if (srcSize > 0) {2480memcpy(dst, src, srcSize);2481}2482return srcSize;2483}248424852486/** ZSTD_decompressLiterals2487@return : nb of bytes read from src, or an error code*/2488static size_t ZSTD_decompressLiterals(void* dst, size_t* maxDstSizePtr,2489const void* src, size_t srcSize)2490{2491const BYTE* ip = (const BYTE*)src;24922493const size_t litSize = (MEM_readLE32(src) & 0x1FFFFF) >> 2; /* no buffer issue : srcSize >= MIN_CBLOCK_SIZE */2494const size_t litCSize = (MEM_readLE32(ip+2) & 0xFFFFFF) >> 5; /* no buffer issue : srcSize >= MIN_CBLOCK_SIZE */24952496if (litSize > *maxDstSizePtr) return ERROR(corruption_detected);2497if (litCSize + 5 > srcSize) return ERROR(corruption_detected);24982499if (HUF_isError(HUF_decompress(dst, litSize, ip+5, litCSize))) return ERROR(corruption_detected);25002501*maxDstSizePtr = litSize;2502return litCSize + 5;2503}250425052506/** ZSTD_decodeLiteralsBlock2507@return : nb of bytes read from src (< srcSize )*/2508static size_t ZSTD_decodeLiteralsBlock(void* ctx,2509const void* src, size_t srcSize)2510{2511ZSTD_DCtx* dctx = (ZSTD_DCtx*)ctx;2512const BYTE* const istart = (const BYTE* const)src;25132514/* any compressed block with literals segment must be at least this size */2515if (srcSize < MIN_CBLOCK_SIZE) return ERROR(corruption_detected);25162517switch(*istart & 3)2518{2519default:2520case 0:2521{2522size_t litSize = BLOCKSIZE;2523const size_t readSize = ZSTD_decompressLiterals(dctx->litBuffer, &litSize, src, srcSize);2524dctx->litPtr = dctx->litBuffer;2525dctx->litSize = litSize;2526memset(dctx->litBuffer + dctx->litSize, 0, 8);2527return readSize; /* works if it's an error too */2528}2529case IS_RAW:2530{2531const size_t litSize = (MEM_readLE32(istart) & 0xFFFFFF) >> 2; /* no buffer issue : srcSize >= MIN_CBLOCK_SIZE */2532if (litSize > srcSize-11) /* risk of reading too far with wildcopy */2533{2534if (litSize > BLOCKSIZE) return ERROR(corruption_detected);2535if (litSize > srcSize-3) return ERROR(corruption_detected);2536memcpy(dctx->litBuffer, istart, litSize);2537dctx->litPtr = dctx->litBuffer;2538dctx->litSize = litSize;2539memset(dctx->litBuffer + dctx->litSize, 0, 8);2540return litSize+3;2541}2542/* direct reference into compressed stream */2543dctx->litPtr = istart+3;2544dctx->litSize = litSize;2545return litSize+3;2546}2547case IS_RLE:2548{2549const size_t litSize = (MEM_readLE32(istart) & 0xFFFFFF) >> 2; /* no buffer issue : srcSize >= MIN_CBLOCK_SIZE */2550if (litSize > BLOCKSIZE) return ERROR(corruption_detected);2551memset(dctx->litBuffer, istart[3], litSize + 8);2552dctx->litPtr = dctx->litBuffer;2553dctx->litSize = litSize;2554return 4;2555}2556}2557}255825592560static size_t ZSTD_decodeSeqHeaders(int* nbSeq, const BYTE** dumpsPtr, size_t* dumpsLengthPtr,2561FSE_DTable* DTableLL, FSE_DTable* DTableML, FSE_DTable* DTableOffb,2562const void* src, size_t srcSize)2563{2564const BYTE* const istart = (const BYTE* const)src;2565const BYTE* ip = istart;2566const BYTE* const iend = istart + srcSize;2567U32 LLtype, Offtype, MLtype;2568U32 LLlog, Offlog, MLlog;2569size_t dumpsLength;25702571/* check */2572if (srcSize < 5) return ERROR(srcSize_wrong);25732574/* SeqHead */2575*nbSeq = MEM_readLE16(ip); ip+=2;2576LLtype = *ip >> 6;2577Offtype = (*ip >> 4) & 3;2578MLtype = (*ip >> 2) & 3;2579if (*ip & 2)2580{2581dumpsLength = ip[2];2582dumpsLength += ip[1] << 8;2583ip += 3;2584}2585else2586{2587dumpsLength = ip[1];2588dumpsLength += (ip[0] & 1) << 8;2589ip += 2;2590}2591*dumpsPtr = ip;2592ip += dumpsLength;2593*dumpsLengthPtr = dumpsLength;25942595/* check */2596if (ip > iend-3) return ERROR(srcSize_wrong); /* min : all 3 are "raw", hence no header, but at least xxLog bits per type */25972598/* sequences */2599{2600S16 norm[MaxML+1]; /* assumption : MaxML >= MaxLL and MaxOff */2601size_t headerSize;26022603/* Build DTables */2604switch(LLtype)2605{2606case bt_rle :2607LLlog = 0;2608FSE_buildDTable_rle(DTableLL, *ip++); break;2609case bt_raw :2610LLlog = LLbits;2611FSE_buildDTable_raw(DTableLL, LLbits); break;2612default :2613{ U32 max = MaxLL;2614headerSize = FSE_readNCount(norm, &max, &LLlog, ip, iend-ip);2615if (FSE_isError(headerSize)) return ERROR(GENERIC);2616if (LLlog > LLFSELog) return ERROR(corruption_detected);2617ip += headerSize;2618FSE_buildDTable(DTableLL, norm, max, LLlog);2619} }26202621switch(Offtype)2622{2623case bt_rle :2624Offlog = 0;2625if (ip > iend-2) return ERROR(srcSize_wrong); /* min : "raw", hence no header, but at least xxLog bits */2626FSE_buildDTable_rle(DTableOffb, *ip++ & MaxOff); /* if *ip > MaxOff, data is corrupted */2627break;2628case bt_raw :2629Offlog = Offbits;2630FSE_buildDTable_raw(DTableOffb, Offbits); break;2631default :2632{ U32 max = MaxOff;2633headerSize = FSE_readNCount(norm, &max, &Offlog, ip, iend-ip);2634if (FSE_isError(headerSize)) return ERROR(GENERIC);2635if (Offlog > OffFSELog) return ERROR(corruption_detected);2636ip += headerSize;2637FSE_buildDTable(DTableOffb, norm, max, Offlog);2638} }26392640switch(MLtype)2641{2642case bt_rle :2643MLlog = 0;2644if (ip > iend-2) return ERROR(srcSize_wrong); /* min : "raw", hence no header, but at least xxLog bits */2645FSE_buildDTable_rle(DTableML, *ip++); break;2646case bt_raw :2647MLlog = MLbits;2648FSE_buildDTable_raw(DTableML, MLbits); break;2649default :2650{ U32 max = MaxML;2651headerSize = FSE_readNCount(norm, &max, &MLlog, ip, iend-ip);2652if (FSE_isError(headerSize)) return ERROR(GENERIC);2653if (MLlog > MLFSELog) return ERROR(corruption_detected);2654ip += headerSize;2655FSE_buildDTable(DTableML, norm, max, MLlog);2656} } }26572658return ip-istart;2659}266026612662typedef struct {2663size_t litLength;2664size_t offset;2665size_t matchLength;2666} seq_t;26672668typedef struct {2669BIT_DStream_t DStream;2670FSE_DState_t stateLL;2671FSE_DState_t stateOffb;2672FSE_DState_t stateML;2673size_t prevOffset;2674const BYTE* dumps;2675const BYTE* dumpsEnd;2676} seqState_t;267726782679static void ZSTD_decodeSequence(seq_t* seq, seqState_t* seqState)2680{2681size_t litLength;2682size_t prevOffset;2683size_t offset;2684size_t matchLength;2685const BYTE* dumps = seqState->dumps;2686const BYTE* const de = seqState->dumpsEnd;26872688/* Literal length */2689litLength = FSE_decodeSymbol(&(seqState->stateLL), &(seqState->DStream));2690prevOffset = litLength ? seq->offset : seqState->prevOffset;2691seqState->prevOffset = seq->offset;2692if (litLength == MaxLL)2693{2694const U32 add = dumps<de ? *dumps++ : 0;2695if (add < 255) litLength += add;2696else if (dumps + 3 <= de)2697{2698litLength = MEM_readLE24(dumps);2699dumps += 3;2700}2701if (dumps >= de) dumps = de-1; /* late correction, to avoid read overflow (data is now corrupted anyway) */2702}27032704/* Offset */2705{2706static const size_t offsetPrefix[MaxOff+1] = { /* note : size_t faster than U32 */27071 /*fake*/, 1, 2, 4, 8, 16, 32, 64, 128, 256,2708512, 1024, 2048, 4096, 8192, 16384, 32768, 65536, 131072, 262144,2709524288, 1048576, 2097152, 4194304, 8388608, 16777216, 33554432, /*fake*/ 1, 1, 1, 1, 1 };2710U32 offsetCode, nbBits;2711offsetCode = FSE_decodeSymbol(&(seqState->stateOffb), &(seqState->DStream)); /* <= maxOff, by table construction */2712if (MEM_32bits()) BIT_reloadDStream(&(seqState->DStream));2713nbBits = offsetCode - 1;2714if (offsetCode==0) nbBits = 0; /* cmove */2715offset = offsetPrefix[offsetCode] + BIT_readBits(&(seqState->DStream), nbBits);2716if (MEM_32bits()) BIT_reloadDStream(&(seqState->DStream));2717if (offsetCode==0) offset = prevOffset; /* cmove */2718}27192720/* MatchLength */2721matchLength = FSE_decodeSymbol(&(seqState->stateML), &(seqState->DStream));2722if (matchLength == MaxML)2723{2724const U32 add = dumps<de ? *dumps++ : 0;2725if (add < 255) matchLength += add;2726else if (dumps + 3 <= de)2727{2728matchLength = MEM_readLE24(dumps);2729dumps += 3;2730}2731if (dumps >= de) dumps = de-1; /* late correction, to avoid read overflow (data is now corrupted anyway) */2732}2733matchLength += MINMATCH;27342735/* save result */2736seq->litLength = litLength;2737seq->offset = offset;2738seq->matchLength = matchLength;2739seqState->dumps = dumps;2740}274127422743static size_t ZSTD_execSequence(BYTE* op,2744seq_t sequence,2745const BYTE** litPtr, const BYTE* const litLimit,2746BYTE* const base, BYTE* const oend)2747{2748static const int dec32table[] = {0, 1, 2, 1, 4, 4, 4, 4}; /* added */2749static const int dec64table[] = {8, 8, 8, 7, 8, 9,10,11}; /* subtracted */2750const BYTE* const ostart = op;2751BYTE* const oLitEnd = op + sequence.litLength;2752BYTE* const oMatchEnd = op + sequence.litLength + sequence.matchLength; /* risk : address space overflow (32-bits) */2753BYTE* const oend_8 = oend-8;2754const BYTE* const litEnd = *litPtr + sequence.litLength;27552756/* checks */2757if (oLitEnd > oend_8) return ERROR(dstSize_tooSmall); /* last match must start at a minimum distance of 8 from oend */2758if (oMatchEnd > oend) return ERROR(dstSize_tooSmall); /* overwrite beyond dst buffer */2759if (litEnd > litLimit) return ERROR(corruption_detected); /* overRead beyond lit buffer */27602761/* copy Literals */2762ZSTD_wildcopy(op, *litPtr, sequence.litLength); /* note : oLitEnd <= oend-8 : no risk of overwrite beyond oend */2763op = oLitEnd;2764*litPtr = litEnd; /* update for next sequence */27652766/* copy Match */2767{2768const BYTE* match = op - sequence.offset;27692770/* check */2771if (sequence.offset > (size_t)op) return ERROR(corruption_detected); /* address space overflow test (this test seems kept by clang optimizer) */2772//if (match > op) return ERROR(corruption_detected); /* address space overflow test (is clang optimizer removing this test ?) */2773if (match < base) return ERROR(corruption_detected);27742775/* close range match, overlap */2776if (sequence.offset < 8)2777{2778const int dec64 = dec64table[sequence.offset];2779op[0] = match[0];2780op[1] = match[1];2781op[2] = match[2];2782op[3] = match[3];2783match += dec32table[sequence.offset];2784ZSTD_copy4(op+4, match);2785match -= dec64;2786}2787else2788{2789ZSTD_copy8(op, match);2790}2791op += 8; match += 8;27922793if (oMatchEnd > oend-(16-MINMATCH))2794{2795if (op < oend_8)2796{2797ZSTD_wildcopy(op, match, oend_8 - op);2798match += oend_8 - op;2799op = oend_8;2800}2801while (op < oMatchEnd) *op++ = *match++;2802}2803else2804{2805ZSTD_wildcopy(op, match, (ptrdiff_t)sequence.matchLength-8); /* works even if matchLength < 8 */2806}2807}28082809return oMatchEnd - ostart;2810}28112812static size_t ZSTD_decompressSequences(2813void* ctx,2814void* dst, size_t maxDstSize,2815const void* seqStart, size_t seqSize)2816{2817ZSTD_DCtx* dctx = (ZSTD_DCtx*)ctx;2818const BYTE* ip = (const BYTE*)seqStart;2819const BYTE* const iend = ip + seqSize;2820BYTE* const ostart = (BYTE* const)dst;2821BYTE* op = ostart;2822BYTE* const oend = ostart + maxDstSize;2823size_t errorCode, dumpsLength;2824const BYTE* litPtr = dctx->litPtr;2825const BYTE* const litEnd = litPtr + dctx->litSize;2826int nbSeq;2827const BYTE* dumps;2828U32* DTableLL = dctx->LLTable;2829U32* DTableML = dctx->MLTable;2830U32* DTableOffb = dctx->OffTable;2831BYTE* const base = (BYTE*) (dctx->base);28322833/* Build Decoding Tables */2834errorCode = ZSTD_decodeSeqHeaders(&nbSeq, &dumps, &dumpsLength,2835DTableLL, DTableML, DTableOffb,2836ip, iend-ip);2837if (ZSTD_isError(errorCode)) return errorCode;2838ip += errorCode;28392840/* Regen sequences */2841{2842seq_t sequence;2843seqState_t seqState;28442845memset(&sequence, 0, sizeof(sequence));2846seqState.dumps = dumps;2847seqState.dumpsEnd = dumps + dumpsLength;2848seqState.prevOffset = sequence.offset = 4;2849errorCode = BIT_initDStream(&(seqState.DStream), ip, iend-ip);2850if (ERR_isError(errorCode)) return ERROR(corruption_detected);2851FSE_initDState(&(seqState.stateLL), &(seqState.DStream), DTableLL);2852FSE_initDState(&(seqState.stateOffb), &(seqState.DStream), DTableOffb);2853FSE_initDState(&(seqState.stateML), &(seqState.DStream), DTableML);28542855for ( ; (BIT_reloadDStream(&(seqState.DStream)) <= BIT_DStream_completed) && (nbSeq>0) ; )2856{2857size_t oneSeqSize;2858nbSeq--;2859ZSTD_decodeSequence(&sequence, &seqState);2860oneSeqSize = ZSTD_execSequence(op, sequence, &litPtr, litEnd, base, oend);2861if (ZSTD_isError(oneSeqSize)) return oneSeqSize;2862op += oneSeqSize;2863}28642865/* check if reached exact end */2866if ( !BIT_endOfDStream(&(seqState.DStream)) ) return ERROR(corruption_detected); /* requested too much : data is corrupted */2867if (nbSeq<0) return ERROR(corruption_detected); /* requested too many sequences : data is corrupted */28682869/* last literal segment */2870{2871size_t lastLLSize = litEnd - litPtr;2872if (litPtr > litEnd) return ERROR(corruption_detected);2873if (op+lastLLSize > oend) return ERROR(dstSize_tooSmall);2874if (lastLLSize > 0) {2875if (op != litPtr) memmove(op, litPtr, lastLLSize);2876op += lastLLSize;2877}2878}2879}28802881return op-ostart;2882}288328842885static size_t ZSTD_decompressBlock(2886void* ctx,2887void* dst, size_t maxDstSize,2888const void* src, size_t srcSize)2889{2890/* blockType == blockCompressed */2891const BYTE* ip = (const BYTE*)src;28922893/* Decode literals sub-block */2894size_t litCSize = ZSTD_decodeLiteralsBlock(ctx, src, srcSize);2895if (ZSTD_isError(litCSize)) return litCSize;2896ip += litCSize;2897srcSize -= litCSize;28982899return ZSTD_decompressSequences(ctx, dst, maxDstSize, ip, srcSize);2900}290129022903static size_t ZSTD_decompressDCtx(void* ctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize)2904{2905const BYTE* ip = (const BYTE*)src;2906const BYTE* iend = ip + srcSize;2907BYTE* const ostart = (BYTE* const)dst;2908BYTE* op = ostart;2909BYTE* const oend = ostart + maxDstSize;2910size_t remainingSize = srcSize;2911U32 magicNumber;2912blockProperties_t blockProperties;29132914/* Frame Header */2915if (srcSize < ZSTD_frameHeaderSize+ZSTD_blockHeaderSize) return ERROR(srcSize_wrong);2916magicNumber = MEM_readLE32(src);2917if (magicNumber != ZSTD_magicNumber) return ERROR(prefix_unknown);2918ip += ZSTD_frameHeaderSize; remainingSize -= ZSTD_frameHeaderSize;29192920/* Loop on each block */2921while (1)2922{2923size_t decodedSize=0;2924size_t cBlockSize = ZSTD_getcBlockSize(ip, iend-ip, &blockProperties);2925if (ZSTD_isError(cBlockSize)) return cBlockSize;29262927ip += ZSTD_blockHeaderSize;2928remainingSize -= ZSTD_blockHeaderSize;2929if (cBlockSize > remainingSize) return ERROR(srcSize_wrong);29302931switch(blockProperties.blockType)2932{2933case bt_compressed:2934decodedSize = ZSTD_decompressBlock(ctx, op, oend-op, ip, cBlockSize);2935break;2936case bt_raw :2937decodedSize = ZSTD_copyUncompressedBlock(op, oend-op, ip, cBlockSize);2938break;2939case bt_rle :2940return ERROR(GENERIC); /* not yet supported */2941break;2942case bt_end :2943/* end of frame */2944if (remainingSize) return ERROR(srcSize_wrong);2945break;2946default:2947return ERROR(GENERIC); /* impossible */2948}2949if (cBlockSize == 0) break; /* bt_end */29502951if (ZSTD_isError(decodedSize)) return decodedSize;2952op += decodedSize;2953ip += cBlockSize;2954remainingSize -= cBlockSize;2955}29562957return op-ostart;2958}29592960static size_t ZSTD_decompress(void* dst, size_t maxDstSize, const void* src, size_t srcSize)2961{2962ZSTD_DCtx ctx;2963ctx.base = dst;2964return ZSTD_decompressDCtx(&ctx, dst, maxDstSize, src, srcSize);2965}29662967/* ZSTD_errorFrameSizeInfoLegacy() :2968assumes `cSize` and `dBound` are _not_ NULL */2969MEM_STATIC void ZSTD_errorFrameSizeInfoLegacy(size_t* cSize, unsigned long long* dBound, size_t ret)2970{2971*cSize = ret;2972*dBound = ZSTD_CONTENTSIZE_ERROR;2973}29742975void ZSTDv03_findFrameSizeInfoLegacy(const void *src, size_t srcSize, size_t* cSize, unsigned long long* dBound)2976{2977const BYTE* ip = (const BYTE*)src;2978size_t remainingSize = srcSize;2979size_t nbBlocks = 0;2980U32 magicNumber;2981blockProperties_t blockProperties;29822983/* Frame Header */2984if (srcSize < ZSTD_frameHeaderSize+ZSTD_blockHeaderSize) {2985ZSTD_errorFrameSizeInfoLegacy(cSize, dBound, ERROR(srcSize_wrong));2986return;2987}2988magicNumber = MEM_readLE32(src);2989if (magicNumber != ZSTD_magicNumber) {2990ZSTD_errorFrameSizeInfoLegacy(cSize, dBound, ERROR(prefix_unknown));2991return;2992}2993ip += ZSTD_frameHeaderSize; remainingSize -= ZSTD_frameHeaderSize;29942995/* Loop on each block */2996while (1)2997{2998size_t cBlockSize = ZSTD_getcBlockSize(ip, remainingSize, &blockProperties);2999if (ZSTD_isError(cBlockSize)) {3000ZSTD_errorFrameSizeInfoLegacy(cSize, dBound, cBlockSize);3001return;3002}30033004ip += ZSTD_blockHeaderSize;3005remainingSize -= ZSTD_blockHeaderSize;3006if (cBlockSize > remainingSize) {3007ZSTD_errorFrameSizeInfoLegacy(cSize, dBound, ERROR(srcSize_wrong));3008return;3009}30103011if (cBlockSize == 0) break; /* bt_end */30123013ip += cBlockSize;3014remainingSize -= cBlockSize;3015nbBlocks++;3016}30173018*cSize = ip - (const BYTE*)src;3019*dBound = nbBlocks * BLOCKSIZE;3020}302130223023/*******************************3024* Streaming Decompression API3025*******************************/30263027static size_t ZSTD_resetDCtx(ZSTD_DCtx* dctx)3028{3029dctx->expected = ZSTD_frameHeaderSize;3030dctx->phase = 0;3031dctx->previousDstEnd = NULL;3032dctx->base = NULL;3033return 0;3034}30353036static ZSTD_DCtx* ZSTD_createDCtx(void)3037{3038ZSTD_DCtx* dctx = (ZSTD_DCtx*)malloc(sizeof(ZSTD_DCtx));3039if (dctx==NULL) return NULL;3040ZSTD_resetDCtx(dctx);3041return dctx;3042}30433044static size_t ZSTD_freeDCtx(ZSTD_DCtx* dctx)3045{3046free(dctx);3047return 0;3048}30493050static size_t ZSTD_nextSrcSizeToDecompress(ZSTD_DCtx* dctx)3051{3052return dctx->expected;3053}30543055static size_t ZSTD_decompressContinue(ZSTD_DCtx* ctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize)3056{3057/* Sanity check */3058if (srcSize != ctx->expected) return ERROR(srcSize_wrong);3059if (dst != ctx->previousDstEnd) /* not contiguous */3060ctx->base = dst;30613062/* Decompress : frame header */3063if (ctx->phase == 0)3064{3065/* Check frame magic header */3066U32 magicNumber = MEM_readLE32(src);3067if (magicNumber != ZSTD_magicNumber) return ERROR(prefix_unknown);3068ctx->phase = 1;3069ctx->expected = ZSTD_blockHeaderSize;3070return 0;3071}30723073/* Decompress : block header */3074if (ctx->phase == 1)3075{3076blockProperties_t bp;3077size_t blockSize = ZSTD_getcBlockSize(src, ZSTD_blockHeaderSize, &bp);3078if (ZSTD_isError(blockSize)) return blockSize;3079if (bp.blockType == bt_end)3080{3081ctx->expected = 0;3082ctx->phase = 0;3083}3084else3085{3086ctx->expected = blockSize;3087ctx->bType = bp.blockType;3088ctx->phase = 2;3089}30903091return 0;3092}30933094/* Decompress : block content */3095{3096size_t rSize;3097switch(ctx->bType)3098{3099case bt_compressed:3100rSize = ZSTD_decompressBlock(ctx, dst, maxDstSize, src, srcSize);3101break;3102case bt_raw :3103rSize = ZSTD_copyUncompressedBlock(dst, maxDstSize, src, srcSize);3104break;3105case bt_rle :3106return ERROR(GENERIC); /* not yet handled */3107break;3108case bt_end : /* should never happen (filtered at phase 1) */3109rSize = 0;3110break;3111default:3112return ERROR(GENERIC);3113}3114ctx->phase = 1;3115ctx->expected = ZSTD_blockHeaderSize;3116ctx->previousDstEnd = (void*)( ((char*)dst) + rSize);3117return rSize;3118}31193120}312131223123/* wrapper layer */31243125unsigned ZSTDv03_isError(size_t code)3126{3127return ZSTD_isError(code);3128}31293130size_t ZSTDv03_decompress( void* dst, size_t maxOriginalSize,3131const void* src, size_t compressedSize)3132{3133return ZSTD_decompress(dst, maxOriginalSize, src, compressedSize);3134}31353136ZSTDv03_Dctx* ZSTDv03_createDCtx(void)3137{3138return (ZSTDv03_Dctx*)ZSTD_createDCtx();3139}31403141size_t ZSTDv03_freeDCtx(ZSTDv03_Dctx* dctx)3142{3143return ZSTD_freeDCtx((ZSTD_DCtx*)dctx);3144}31453146size_t ZSTDv03_resetDCtx(ZSTDv03_Dctx* dctx)3147{3148return ZSTD_resetDCtx((ZSTD_DCtx*)dctx);3149}31503151size_t ZSTDv03_nextSrcSizeToDecompress(ZSTDv03_Dctx* dctx)3152{3153return ZSTD_nextSrcSizeToDecompress((ZSTD_DCtx*)dctx);3154}31553156size_t ZSTDv03_decompressContinue(ZSTDv03_Dctx* dctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize)3157{3158return ZSTD_decompressContinue((ZSTD_DCtx*)dctx, dst, maxDstSize, src, srcSize);3159}316031613162