xref: /freebsd-src/sys/contrib/openzfs/module/zstd/lib/decompress/huf_decompress.c (revision c03c5b1c80914ec656fbee84539355d1fad68bf9)
1*c03c5b1cSMartin Matuska /* ******************************************************************
2*c03c5b1cSMartin Matuska  * huff0 huffman decoder,
3*c03c5b1cSMartin Matuska  * part of Finite State Entropy library
4*c03c5b1cSMartin Matuska  * Copyright (c) 2013-2020, Yann Collet, Facebook, Inc.
5*c03c5b1cSMartin Matuska  *
6*c03c5b1cSMartin Matuska  *  You can contact the author at :
7*c03c5b1cSMartin Matuska  *  - FSE+HUF source repository : https://github.com/Cyan4973/FiniteStateEntropy
8*c03c5b1cSMartin Matuska  *
9*c03c5b1cSMartin Matuska  * This source code is licensed under both the BSD-style license (found in the
10*c03c5b1cSMartin Matuska  * LICENSE file in the root directory of this source tree) and the GPLv2 (found
11*c03c5b1cSMartin Matuska  * in the COPYING file in the root directory of this source tree).
12*c03c5b1cSMartin Matuska  * You may select, at your option, one of the above-listed licenses.
13*c03c5b1cSMartin Matuska ****************************************************************** */
14*c03c5b1cSMartin Matuska 
15*c03c5b1cSMartin Matuska /* **************************************************************
16*c03c5b1cSMartin Matuska *  Dependencies
17*c03c5b1cSMartin Matuska ****************************************************************/
18*c03c5b1cSMartin Matuska #include <string.h>     /* memcpy, memset */
19*c03c5b1cSMartin Matuska #include "../common/compiler.h"
20*c03c5b1cSMartin Matuska #include "../common/bitstream.h"  /* BIT_* */
21*c03c5b1cSMartin Matuska #include "../common/fse.h"        /* to compress headers */
22*c03c5b1cSMartin Matuska #define HUF_STATIC_LINKING_ONLY
23*c03c5b1cSMartin Matuska #include "../common/huf.h"
24*c03c5b1cSMartin Matuska #include "../common/error_private.h"
25*c03c5b1cSMartin Matuska 
26*c03c5b1cSMartin Matuska /* **************************************************************
27*c03c5b1cSMartin Matuska *  Macros
28*c03c5b1cSMartin Matuska ****************************************************************/
29*c03c5b1cSMartin Matuska 
30*c03c5b1cSMartin Matuska /* These two optional macros force the use one way or another of the two
31*c03c5b1cSMartin Matuska  * Huffman decompression implementations. You can't force in both directions
32*c03c5b1cSMartin Matuska  * at the same time.
33*c03c5b1cSMartin Matuska  */
34*c03c5b1cSMartin Matuska #if defined(HUF_FORCE_DECOMPRESS_X1) && \
35*c03c5b1cSMartin Matuska     defined(HUF_FORCE_DECOMPRESS_X2)
36*c03c5b1cSMartin Matuska #error "Cannot force the use of the X1 and X2 decoders at the same time!"
37*c03c5b1cSMartin Matuska #endif
38*c03c5b1cSMartin Matuska 
39*c03c5b1cSMartin Matuska 
40*c03c5b1cSMartin Matuska /* **************************************************************
41*c03c5b1cSMartin Matuska *  Error Management
42*c03c5b1cSMartin Matuska ****************************************************************/
43*c03c5b1cSMartin Matuska #define HUF_isError ERR_isError
44*c03c5b1cSMartin Matuska 
45*c03c5b1cSMartin Matuska 
46*c03c5b1cSMartin Matuska /* **************************************************************
47*c03c5b1cSMartin Matuska *  Byte alignment for workSpace management
48*c03c5b1cSMartin Matuska ****************************************************************/
49*c03c5b1cSMartin Matuska #define HUF_ALIGN(x, a)         HUF_ALIGN_MASK((x), (a) - 1)
50*c03c5b1cSMartin Matuska #define HUF_ALIGN_MASK(x, mask) (((x) + (mask)) & ~(mask))
51*c03c5b1cSMartin Matuska 
52*c03c5b1cSMartin Matuska 
53*c03c5b1cSMartin Matuska /* **************************************************************
54*c03c5b1cSMartin Matuska *  BMI2 Variant Wrappers
55*c03c5b1cSMartin Matuska ****************************************************************/
56*c03c5b1cSMartin Matuska #if DYNAMIC_BMI2
57*c03c5b1cSMartin Matuska 
58*c03c5b1cSMartin Matuska #define HUF_DGEN(fn)                                                        \
59*c03c5b1cSMartin Matuska                                                                             \
60*c03c5b1cSMartin Matuska     static size_t fn##_default(                                             \
61*c03c5b1cSMartin Matuska                   void* dst,  size_t dstSize,                               \
62*c03c5b1cSMartin Matuska             const void* cSrc, size_t cSrcSize,                              \
63*c03c5b1cSMartin Matuska             const HUF_DTable* DTable)                                       \
64*c03c5b1cSMartin Matuska     {                                                                       \
65*c03c5b1cSMartin Matuska         return fn##_body(dst, dstSize, cSrc, cSrcSize, DTable);             \
66*c03c5b1cSMartin Matuska     }                                                                       \
67*c03c5b1cSMartin Matuska                                                                             \
68*c03c5b1cSMartin Matuska     static TARGET_ATTRIBUTE("bmi2") size_t fn##_bmi2(                       \
69*c03c5b1cSMartin Matuska                   void* dst,  size_t dstSize,                               \
70*c03c5b1cSMartin Matuska             const void* cSrc, size_t cSrcSize,                              \
71*c03c5b1cSMartin Matuska             const HUF_DTable* DTable)                                       \
72*c03c5b1cSMartin Matuska     {                                                                       \
73*c03c5b1cSMartin Matuska         return fn##_body(dst, dstSize, cSrc, cSrcSize, DTable);             \
74*c03c5b1cSMartin Matuska     }                                                                       \
75*c03c5b1cSMartin Matuska                                                                             \
76*c03c5b1cSMartin Matuska     static size_t fn(void* dst, size_t dstSize, void const* cSrc,           \
77*c03c5b1cSMartin Matuska                      size_t cSrcSize, HUF_DTable const* DTable, int bmi2)   \
78*c03c5b1cSMartin Matuska     {                                                                       \
79*c03c5b1cSMartin Matuska         if (bmi2) {                                                         \
80*c03c5b1cSMartin Matuska             return fn##_bmi2(dst, dstSize, cSrc, cSrcSize, DTable);         \
81*c03c5b1cSMartin Matuska         }                                                                   \
82*c03c5b1cSMartin Matuska         return fn##_default(dst, dstSize, cSrc, cSrcSize, DTable);          \
83*c03c5b1cSMartin Matuska     }
84*c03c5b1cSMartin Matuska 
85*c03c5b1cSMartin Matuska #else
86*c03c5b1cSMartin Matuska 
87*c03c5b1cSMartin Matuska #define HUF_DGEN(fn)                                                        \
88*c03c5b1cSMartin Matuska     static size_t fn(void* dst, size_t dstSize, void const* cSrc,           \
89*c03c5b1cSMartin Matuska                      size_t cSrcSize, HUF_DTable const* DTable, int bmi2)   \
90*c03c5b1cSMartin Matuska     {                                                                       \
91*c03c5b1cSMartin Matuska         (void)bmi2;                                                         \
92*c03c5b1cSMartin Matuska         return fn##_body(dst, dstSize, cSrc, cSrcSize, DTable);             \
93*c03c5b1cSMartin Matuska     }
94*c03c5b1cSMartin Matuska 
95*c03c5b1cSMartin Matuska #endif
96*c03c5b1cSMartin Matuska 
97*c03c5b1cSMartin Matuska 
98*c03c5b1cSMartin Matuska /*-***************************/
99*c03c5b1cSMartin Matuska /*  generic DTableDesc       */
100*c03c5b1cSMartin Matuska /*-***************************/
101*c03c5b1cSMartin Matuska typedef struct { BYTE maxTableLog; BYTE tableType; BYTE tableLog; BYTE reserved; } DTableDesc;
102*c03c5b1cSMartin Matuska 
HUF_getDTableDesc(const HUF_DTable * table)103*c03c5b1cSMartin Matuska static DTableDesc HUF_getDTableDesc(const HUF_DTable* table)
104*c03c5b1cSMartin Matuska {
105*c03c5b1cSMartin Matuska     DTableDesc dtd;
106*c03c5b1cSMartin Matuska     memcpy(&dtd, table, sizeof(dtd));
107*c03c5b1cSMartin Matuska     return dtd;
108*c03c5b1cSMartin Matuska }
109*c03c5b1cSMartin Matuska 
110*c03c5b1cSMartin Matuska 
111*c03c5b1cSMartin Matuska #ifndef HUF_FORCE_DECOMPRESS_X2
112*c03c5b1cSMartin Matuska 
113*c03c5b1cSMartin Matuska /*-***************************/
114*c03c5b1cSMartin Matuska /*  single-symbol decoding   */
115*c03c5b1cSMartin Matuska /*-***************************/
116*c03c5b1cSMartin Matuska typedef struct { BYTE byte; BYTE nbBits; } HUF_DEltX1;   /* single-symbol decoding */
117*c03c5b1cSMartin Matuska 
HUF_readDTableX1_wksp(HUF_DTable * DTable,const void * src,size_t srcSize,void * workSpace,size_t wkspSize)118*c03c5b1cSMartin Matuska size_t HUF_readDTableX1_wksp(HUF_DTable* DTable, const void* src, size_t srcSize, void* workSpace, size_t wkspSize)
119*c03c5b1cSMartin Matuska {
120*c03c5b1cSMartin Matuska     U32 tableLog = 0;
121*c03c5b1cSMartin Matuska     U32 nbSymbols = 0;
122*c03c5b1cSMartin Matuska     size_t iSize;
123*c03c5b1cSMartin Matuska     void* const dtPtr = DTable + 1;
124*c03c5b1cSMartin Matuska     HUF_DEltX1* const dt = (HUF_DEltX1*)dtPtr;
125*c03c5b1cSMartin Matuska 
126*c03c5b1cSMartin Matuska     U32* rankVal;
127*c03c5b1cSMartin Matuska     BYTE* huffWeight;
128*c03c5b1cSMartin Matuska     size_t spaceUsed32 = 0;
129*c03c5b1cSMartin Matuska 
130*c03c5b1cSMartin Matuska     rankVal = (U32 *)workSpace + spaceUsed32;
131*c03c5b1cSMartin Matuska     spaceUsed32 += HUF_TABLELOG_ABSOLUTEMAX + 1;
132*c03c5b1cSMartin Matuska     huffWeight = (BYTE *)((U32 *)workSpace + spaceUsed32);
133*c03c5b1cSMartin Matuska     spaceUsed32 += HUF_ALIGN(HUF_SYMBOLVALUE_MAX + 1, sizeof(U32)) >> 2;
134*c03c5b1cSMartin Matuska 
135*c03c5b1cSMartin Matuska     if ((spaceUsed32 << 2) > wkspSize) return ERROR(tableLog_tooLarge);
136*c03c5b1cSMartin Matuska 
137*c03c5b1cSMartin Matuska     DEBUG_STATIC_ASSERT(sizeof(DTableDesc) == sizeof(HUF_DTable));
138*c03c5b1cSMartin Matuska     /* memset(huffWeight, 0, sizeof(huffWeight)); */   /* is not necessary, even though some analyzer complain ... */
139*c03c5b1cSMartin Matuska 
140*c03c5b1cSMartin Matuska     iSize = HUF_readStats(huffWeight, HUF_SYMBOLVALUE_MAX + 1, rankVal, &nbSymbols, &tableLog, src, srcSize);
141*c03c5b1cSMartin Matuska     if (HUF_isError(iSize)) return iSize;
142*c03c5b1cSMartin Matuska 
143*c03c5b1cSMartin Matuska     /* Table header */
144*c03c5b1cSMartin Matuska     {   DTableDesc dtd = HUF_getDTableDesc(DTable);
145*c03c5b1cSMartin Matuska         if (tableLog > (U32)(dtd.maxTableLog+1)) return ERROR(tableLog_tooLarge);   /* DTable too small, Huffman tree cannot fit in */
146*c03c5b1cSMartin Matuska         dtd.tableType = 0;
147*c03c5b1cSMartin Matuska         dtd.tableLog = (BYTE)tableLog;
148*c03c5b1cSMartin Matuska         memcpy(DTable, &dtd, sizeof(dtd));
149*c03c5b1cSMartin Matuska     }
150*c03c5b1cSMartin Matuska 
151*c03c5b1cSMartin Matuska     /* Calculate starting value for each rank */
152*c03c5b1cSMartin Matuska     {   U32 n, nextRankStart = 0;
153*c03c5b1cSMartin Matuska         for (n=1; n<tableLog+1; n++) {
154*c03c5b1cSMartin Matuska             U32 const current = nextRankStart;
155*c03c5b1cSMartin Matuska             nextRankStart += (rankVal[n] << (n-1));
156*c03c5b1cSMartin Matuska             rankVal[n] = current;
157*c03c5b1cSMartin Matuska     }   }
158*c03c5b1cSMartin Matuska 
159*c03c5b1cSMartin Matuska     /* fill DTable */
160*c03c5b1cSMartin Matuska     {   U32 n;
161*c03c5b1cSMartin Matuska         size_t const nEnd = nbSymbols;
162*c03c5b1cSMartin Matuska         for (n=0; n<nEnd; n++) {
163*c03c5b1cSMartin Matuska             size_t const w = huffWeight[n];
164*c03c5b1cSMartin Matuska             size_t const length = (1 << w) >> 1;
165*c03c5b1cSMartin Matuska             size_t const uStart = rankVal[w];
166*c03c5b1cSMartin Matuska             size_t const uEnd = uStart + length;
167*c03c5b1cSMartin Matuska             size_t u;
168*c03c5b1cSMartin Matuska             HUF_DEltX1 D;
169*c03c5b1cSMartin Matuska             D.byte = (BYTE)n;
170*c03c5b1cSMartin Matuska             D.nbBits = (BYTE)(tableLog + 1 - w);
171*c03c5b1cSMartin Matuska             rankVal[w] = (U32)uEnd;
172*c03c5b1cSMartin Matuska             if (length < 4) {
173*c03c5b1cSMartin Matuska                 /* Use length in the loop bound so the compiler knows it is short. */
174*c03c5b1cSMartin Matuska                 for (u = 0; u < length; ++u)
175*c03c5b1cSMartin Matuska                     dt[uStart + u] = D;
176*c03c5b1cSMartin Matuska             } else {
177*c03c5b1cSMartin Matuska                 /* Unroll the loop 4 times, we know it is a power of 2. */
178*c03c5b1cSMartin Matuska                 for (u = uStart; u < uEnd; u += 4) {
179*c03c5b1cSMartin Matuska                     dt[u + 0] = D;
180*c03c5b1cSMartin Matuska                     dt[u + 1] = D;
181*c03c5b1cSMartin Matuska                     dt[u + 2] = D;
182*c03c5b1cSMartin Matuska                     dt[u + 3] = D;
183*c03c5b1cSMartin Matuska     }   }   }   }
184*c03c5b1cSMartin Matuska     return iSize;
185*c03c5b1cSMartin Matuska }
186*c03c5b1cSMartin Matuska 
HUF_readDTableX1(HUF_DTable * DTable,const void * src,size_t srcSize)187*c03c5b1cSMartin Matuska size_t HUF_readDTableX1(HUF_DTable* DTable, const void* src, size_t srcSize)
188*c03c5b1cSMartin Matuska {
189*c03c5b1cSMartin Matuska     U32 workSpace[HUF_DECOMPRESS_WORKSPACE_SIZE_U32];
190*c03c5b1cSMartin Matuska     return HUF_readDTableX1_wksp(DTable, src, srcSize,
191*c03c5b1cSMartin Matuska                                  workSpace, sizeof(workSpace));
192*c03c5b1cSMartin Matuska }
193*c03c5b1cSMartin Matuska 
194*c03c5b1cSMartin Matuska FORCE_INLINE_TEMPLATE BYTE
HUF_decodeSymbolX1(BIT_DStream_t * Dstream,const HUF_DEltX1 * dt,const U32 dtLog)195*c03c5b1cSMartin Matuska HUF_decodeSymbolX1(BIT_DStream_t* Dstream, const HUF_DEltX1* dt, const U32 dtLog)
196*c03c5b1cSMartin Matuska {
197*c03c5b1cSMartin Matuska     size_t const val = BIT_lookBitsFast(Dstream, dtLog); /* note : dtLog >= 1 */
198*c03c5b1cSMartin Matuska     BYTE const c = dt[val].byte;
199*c03c5b1cSMartin Matuska     BIT_skipBits(Dstream, dt[val].nbBits);
200*c03c5b1cSMartin Matuska     return c;
201*c03c5b1cSMartin Matuska }
202*c03c5b1cSMartin Matuska 
203*c03c5b1cSMartin Matuska #define HUF_DECODE_SYMBOLX1_0(ptr, DStreamPtr) \
204*c03c5b1cSMartin Matuska     *ptr++ = HUF_decodeSymbolX1(DStreamPtr, dt, dtLog)
205*c03c5b1cSMartin Matuska 
206*c03c5b1cSMartin Matuska #define HUF_DECODE_SYMBOLX1_1(ptr, DStreamPtr)  \
207*c03c5b1cSMartin Matuska     if (MEM_64bits() || (HUF_TABLELOG_MAX<=12)) \
208*c03c5b1cSMartin Matuska         HUF_DECODE_SYMBOLX1_0(ptr, DStreamPtr)
209*c03c5b1cSMartin Matuska 
210*c03c5b1cSMartin Matuska #define HUF_DECODE_SYMBOLX1_2(ptr, DStreamPtr) \
211*c03c5b1cSMartin Matuska     if (MEM_64bits()) \
212*c03c5b1cSMartin Matuska         HUF_DECODE_SYMBOLX1_0(ptr, DStreamPtr)
213*c03c5b1cSMartin Matuska 
214*c03c5b1cSMartin Matuska HINT_INLINE size_t
HUF_decodeStreamX1(BYTE * p,BIT_DStream_t * const bitDPtr,BYTE * const pEnd,const HUF_DEltX1 * const dt,const U32 dtLog)215*c03c5b1cSMartin Matuska HUF_decodeStreamX1(BYTE* p, BIT_DStream_t* const bitDPtr, BYTE* const pEnd, const HUF_DEltX1* const dt, const U32 dtLog)
216*c03c5b1cSMartin Matuska {
217*c03c5b1cSMartin Matuska     BYTE* const pStart = p;
218*c03c5b1cSMartin Matuska 
219*c03c5b1cSMartin Matuska     /* up to 4 symbols at a time */
220*c03c5b1cSMartin Matuska     while ((BIT_reloadDStream(bitDPtr) == BIT_DStream_unfinished) & (p < pEnd-3)) {
221*c03c5b1cSMartin Matuska         HUF_DECODE_SYMBOLX1_2(p, bitDPtr);
222*c03c5b1cSMartin Matuska         HUF_DECODE_SYMBOLX1_1(p, bitDPtr);
223*c03c5b1cSMartin Matuska         HUF_DECODE_SYMBOLX1_2(p, bitDPtr);
224*c03c5b1cSMartin Matuska         HUF_DECODE_SYMBOLX1_0(p, bitDPtr);
225*c03c5b1cSMartin Matuska     }
226*c03c5b1cSMartin Matuska 
227*c03c5b1cSMartin Matuska     /* [0-3] symbols remaining */
228*c03c5b1cSMartin Matuska     if (MEM_32bits())
229*c03c5b1cSMartin Matuska         while ((BIT_reloadDStream(bitDPtr) == BIT_DStream_unfinished) & (p < pEnd))
230*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_0(p, bitDPtr);
231*c03c5b1cSMartin Matuska 
232*c03c5b1cSMartin Matuska     /* no more data to retrieve from bitstream, no need to reload */
233*c03c5b1cSMartin Matuska     while (p < pEnd)
234*c03c5b1cSMartin Matuska         HUF_DECODE_SYMBOLX1_0(p, bitDPtr);
235*c03c5b1cSMartin Matuska 
236*c03c5b1cSMartin Matuska     return pEnd-pStart;
237*c03c5b1cSMartin Matuska }
238*c03c5b1cSMartin Matuska 
239*c03c5b1cSMartin Matuska FORCE_INLINE_TEMPLATE size_t
HUF_decompress1X1_usingDTable_internal_body(void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,const HUF_DTable * DTable)240*c03c5b1cSMartin Matuska HUF_decompress1X1_usingDTable_internal_body(
241*c03c5b1cSMartin Matuska           void* dst,  size_t dstSize,
242*c03c5b1cSMartin Matuska     const void* cSrc, size_t cSrcSize,
243*c03c5b1cSMartin Matuska     const HUF_DTable* DTable)
244*c03c5b1cSMartin Matuska {
245*c03c5b1cSMartin Matuska     BYTE* op = (BYTE*)dst;
246*c03c5b1cSMartin Matuska     BYTE* const oend = op + dstSize;
247*c03c5b1cSMartin Matuska     const void* dtPtr = DTable + 1;
248*c03c5b1cSMartin Matuska     const HUF_DEltX1* const dt = (const HUF_DEltX1*)dtPtr;
249*c03c5b1cSMartin Matuska     BIT_DStream_t bitD;
250*c03c5b1cSMartin Matuska     DTableDesc const dtd = HUF_getDTableDesc(DTable);
251*c03c5b1cSMartin Matuska     U32 const dtLog = dtd.tableLog;
252*c03c5b1cSMartin Matuska 
253*c03c5b1cSMartin Matuska     CHECK_F( BIT_initDStream(&bitD, cSrc, cSrcSize) );
254*c03c5b1cSMartin Matuska 
255*c03c5b1cSMartin Matuska     HUF_decodeStreamX1(op, &bitD, oend, dt, dtLog);
256*c03c5b1cSMartin Matuska 
257*c03c5b1cSMartin Matuska     if (!BIT_endOfDStream(&bitD)) return ERROR(corruption_detected);
258*c03c5b1cSMartin Matuska 
259*c03c5b1cSMartin Matuska     return dstSize;
260*c03c5b1cSMartin Matuska }
261*c03c5b1cSMartin Matuska 
262*c03c5b1cSMartin Matuska FORCE_INLINE_TEMPLATE size_t
HUF_decompress4X1_usingDTable_internal_body(void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,const HUF_DTable * DTable)263*c03c5b1cSMartin Matuska HUF_decompress4X1_usingDTable_internal_body(
264*c03c5b1cSMartin Matuska           void* dst,  size_t dstSize,
265*c03c5b1cSMartin Matuska     const void* cSrc, size_t cSrcSize,
266*c03c5b1cSMartin Matuska     const HUF_DTable* DTable)
267*c03c5b1cSMartin Matuska {
268*c03c5b1cSMartin Matuska     /* Check */
269*c03c5b1cSMartin Matuska     if (cSrcSize < 10) return ERROR(corruption_detected);  /* strict minimum : jump table + 1 byte per stream */
270*c03c5b1cSMartin Matuska 
271*c03c5b1cSMartin Matuska     {   const BYTE* const istart = (const BYTE*) cSrc;
272*c03c5b1cSMartin Matuska         BYTE* const ostart = (BYTE*) dst;
273*c03c5b1cSMartin Matuska         BYTE* const oend = ostart + dstSize;
274*c03c5b1cSMartin Matuska         BYTE* const olimit = oend - 3;
275*c03c5b1cSMartin Matuska         const void* const dtPtr = DTable + 1;
276*c03c5b1cSMartin Matuska         const HUF_DEltX1* const dt = (const HUF_DEltX1*)dtPtr;
277*c03c5b1cSMartin Matuska 
278*c03c5b1cSMartin Matuska         /* Init */
279*c03c5b1cSMartin Matuska         BIT_DStream_t bitD1;
280*c03c5b1cSMartin Matuska         BIT_DStream_t bitD2;
281*c03c5b1cSMartin Matuska         BIT_DStream_t bitD3;
282*c03c5b1cSMartin Matuska         BIT_DStream_t bitD4;
283*c03c5b1cSMartin Matuska         size_t const length1 = MEM_readLE16(istart);
284*c03c5b1cSMartin Matuska         size_t const length2 = MEM_readLE16(istart+2);
285*c03c5b1cSMartin Matuska         size_t const length3 = MEM_readLE16(istart+4);
286*c03c5b1cSMartin Matuska         size_t const length4 = cSrcSize - (length1 + length2 + length3 + 6);
287*c03c5b1cSMartin Matuska         const BYTE* const istart1 = istart + 6;  /* jumpTable */
288*c03c5b1cSMartin Matuska         const BYTE* const istart2 = istart1 + length1;
289*c03c5b1cSMartin Matuska         const BYTE* const istart3 = istart2 + length2;
290*c03c5b1cSMartin Matuska         const BYTE* const istart4 = istart3 + length3;
291*c03c5b1cSMartin Matuska         const size_t segmentSize = (dstSize+3) / 4;
292*c03c5b1cSMartin Matuska         BYTE* const opStart2 = ostart + segmentSize;
293*c03c5b1cSMartin Matuska         BYTE* const opStart3 = opStart2 + segmentSize;
294*c03c5b1cSMartin Matuska         BYTE* const opStart4 = opStart3 + segmentSize;
295*c03c5b1cSMartin Matuska         BYTE* op1 = ostart;
296*c03c5b1cSMartin Matuska         BYTE* op2 = opStart2;
297*c03c5b1cSMartin Matuska         BYTE* op3 = opStart3;
298*c03c5b1cSMartin Matuska         BYTE* op4 = opStart4;
299*c03c5b1cSMartin Matuska         DTableDesc const dtd = HUF_getDTableDesc(DTable);
300*c03c5b1cSMartin Matuska         U32 const dtLog = dtd.tableLog;
301*c03c5b1cSMartin Matuska         U32 endSignal = 1;
302*c03c5b1cSMartin Matuska 
303*c03c5b1cSMartin Matuska         if (length4 > cSrcSize) return ERROR(corruption_detected);   /* overflow */
304*c03c5b1cSMartin Matuska         CHECK_F( BIT_initDStream(&bitD1, istart1, length1) );
305*c03c5b1cSMartin Matuska         CHECK_F( BIT_initDStream(&bitD2, istart2, length2) );
306*c03c5b1cSMartin Matuska         CHECK_F( BIT_initDStream(&bitD3, istart3, length3) );
307*c03c5b1cSMartin Matuska         CHECK_F( BIT_initDStream(&bitD4, istart4, length4) );
308*c03c5b1cSMartin Matuska 
309*c03c5b1cSMartin Matuska         /* up to 16 symbols per loop (4 symbols per stream) in 64-bit mode */
310*c03c5b1cSMartin Matuska         for ( ; (endSignal) & (op4 < olimit) ; ) {
311*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_2(op1, &bitD1);
312*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_2(op2, &bitD2);
313*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_2(op3, &bitD3);
314*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_2(op4, &bitD4);
315*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_1(op1, &bitD1);
316*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_1(op2, &bitD2);
317*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_1(op3, &bitD3);
318*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_1(op4, &bitD4);
319*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_2(op1, &bitD1);
320*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_2(op2, &bitD2);
321*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_2(op3, &bitD3);
322*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_2(op4, &bitD4);
323*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_0(op1, &bitD1);
324*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_0(op2, &bitD2);
325*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_0(op3, &bitD3);
326*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX1_0(op4, &bitD4);
327*c03c5b1cSMartin Matuska             endSignal &= BIT_reloadDStreamFast(&bitD1) == BIT_DStream_unfinished;
328*c03c5b1cSMartin Matuska             endSignal &= BIT_reloadDStreamFast(&bitD2) == BIT_DStream_unfinished;
329*c03c5b1cSMartin Matuska             endSignal &= BIT_reloadDStreamFast(&bitD3) == BIT_DStream_unfinished;
330*c03c5b1cSMartin Matuska             endSignal &= BIT_reloadDStreamFast(&bitD4) == BIT_DStream_unfinished;
331*c03c5b1cSMartin Matuska         }
332*c03c5b1cSMartin Matuska 
333*c03c5b1cSMartin Matuska         /* check corruption */
334*c03c5b1cSMartin Matuska         /* note : should not be necessary : op# advance in lock step, and we control op4.
335*c03c5b1cSMartin Matuska          *        but curiously, binary generated by gcc 7.2 & 7.3 with -mbmi2 runs faster when >=1 test is present */
336*c03c5b1cSMartin Matuska         if (op1 > opStart2) return ERROR(corruption_detected);
337*c03c5b1cSMartin Matuska         if (op2 > opStart3) return ERROR(corruption_detected);
338*c03c5b1cSMartin Matuska         if (op3 > opStart4) return ERROR(corruption_detected);
339*c03c5b1cSMartin Matuska         /* note : op4 supposed already verified within main loop */
340*c03c5b1cSMartin Matuska 
341*c03c5b1cSMartin Matuska         /* finish bitStreams one by one */
342*c03c5b1cSMartin Matuska         HUF_decodeStreamX1(op1, &bitD1, opStart2, dt, dtLog);
343*c03c5b1cSMartin Matuska         HUF_decodeStreamX1(op2, &bitD2, opStart3, dt, dtLog);
344*c03c5b1cSMartin Matuska         HUF_decodeStreamX1(op3, &bitD3, opStart4, dt, dtLog);
345*c03c5b1cSMartin Matuska         HUF_decodeStreamX1(op4, &bitD4, oend,     dt, dtLog);
346*c03c5b1cSMartin Matuska 
347*c03c5b1cSMartin Matuska         /* check */
348*c03c5b1cSMartin Matuska         { U32 const endCheck = BIT_endOfDStream(&bitD1) & BIT_endOfDStream(&bitD2) & BIT_endOfDStream(&bitD3) & BIT_endOfDStream(&bitD4);
349*c03c5b1cSMartin Matuska           if (!endCheck) return ERROR(corruption_detected); }
350*c03c5b1cSMartin Matuska 
351*c03c5b1cSMartin Matuska         /* decoded size */
352*c03c5b1cSMartin Matuska         return dstSize;
353*c03c5b1cSMartin Matuska     }
354*c03c5b1cSMartin Matuska }
355*c03c5b1cSMartin Matuska 
356*c03c5b1cSMartin Matuska 
357*c03c5b1cSMartin Matuska typedef size_t (*HUF_decompress_usingDTable_t)(void *dst, size_t dstSize,
358*c03c5b1cSMartin Matuska                                                const void *cSrc,
359*c03c5b1cSMartin Matuska                                                size_t cSrcSize,
360*c03c5b1cSMartin Matuska                                                const HUF_DTable *DTable);
361*c03c5b1cSMartin Matuska 
362*c03c5b1cSMartin Matuska HUF_DGEN(HUF_decompress1X1_usingDTable_internal)
HUF_DGEN(HUF_decompress4X1_usingDTable_internal)363*c03c5b1cSMartin Matuska HUF_DGEN(HUF_decompress4X1_usingDTable_internal)
364*c03c5b1cSMartin Matuska 
365*c03c5b1cSMartin Matuska 
366*c03c5b1cSMartin Matuska 
367*c03c5b1cSMartin Matuska size_t HUF_decompress1X1_usingDTable(
368*c03c5b1cSMartin Matuska           void* dst,  size_t dstSize,
369*c03c5b1cSMartin Matuska     const void* cSrc, size_t cSrcSize,
370*c03c5b1cSMartin Matuska     const HUF_DTable* DTable)
371*c03c5b1cSMartin Matuska {
372*c03c5b1cSMartin Matuska     DTableDesc dtd = HUF_getDTableDesc(DTable);
373*c03c5b1cSMartin Matuska     if (dtd.tableType != 0) return ERROR(GENERIC);
374*c03c5b1cSMartin Matuska     return HUF_decompress1X1_usingDTable_internal(dst, dstSize, cSrc, cSrcSize, DTable, /* bmi2 */ 0);
375*c03c5b1cSMartin Matuska }
376*c03c5b1cSMartin Matuska 
HUF_decompress1X1_DCtx_wksp(HUF_DTable * DCtx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,void * workSpace,size_t wkspSize)377*c03c5b1cSMartin Matuska size_t HUF_decompress1X1_DCtx_wksp(HUF_DTable* DCtx, void* dst, size_t dstSize,
378*c03c5b1cSMartin Matuska                                    const void* cSrc, size_t cSrcSize,
379*c03c5b1cSMartin Matuska                                    void* workSpace, size_t wkspSize)
380*c03c5b1cSMartin Matuska {
381*c03c5b1cSMartin Matuska     const BYTE* ip = (const BYTE*) cSrc;
382*c03c5b1cSMartin Matuska 
383*c03c5b1cSMartin Matuska     size_t const hSize = HUF_readDTableX1_wksp(DCtx, cSrc, cSrcSize, workSpace, wkspSize);
384*c03c5b1cSMartin Matuska     if (HUF_isError(hSize)) return hSize;
385*c03c5b1cSMartin Matuska     if (hSize >= cSrcSize) return ERROR(srcSize_wrong);
386*c03c5b1cSMartin Matuska     ip += hSize; cSrcSize -= hSize;
387*c03c5b1cSMartin Matuska 
388*c03c5b1cSMartin Matuska     return HUF_decompress1X1_usingDTable_internal(dst, dstSize, ip, cSrcSize, DCtx, /* bmi2 */ 0);
389*c03c5b1cSMartin Matuska }
390*c03c5b1cSMartin Matuska 
391*c03c5b1cSMartin Matuska 
HUF_decompress1X1_DCtx(HUF_DTable * DCtx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize)392*c03c5b1cSMartin Matuska size_t HUF_decompress1X1_DCtx(HUF_DTable* DCtx, void* dst, size_t dstSize,
393*c03c5b1cSMartin Matuska                               const void* cSrc, size_t cSrcSize)
394*c03c5b1cSMartin Matuska {
395*c03c5b1cSMartin Matuska     U32 workSpace[HUF_DECOMPRESS_WORKSPACE_SIZE_U32];
396*c03c5b1cSMartin Matuska     return HUF_decompress1X1_DCtx_wksp(DCtx, dst, dstSize, cSrc, cSrcSize,
397*c03c5b1cSMartin Matuska                                        workSpace, sizeof(workSpace));
398*c03c5b1cSMartin Matuska }
399*c03c5b1cSMartin Matuska 
HUF_decompress1X1(void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize)400*c03c5b1cSMartin Matuska size_t HUF_decompress1X1 (void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize)
401*c03c5b1cSMartin Matuska {
402*c03c5b1cSMartin Matuska     HUF_CREATE_STATIC_DTABLEX1(DTable, HUF_TABLELOG_MAX);
403*c03c5b1cSMartin Matuska     return HUF_decompress1X1_DCtx (DTable, dst, dstSize, cSrc, cSrcSize);
404*c03c5b1cSMartin Matuska }
405*c03c5b1cSMartin Matuska 
HUF_decompress4X1_usingDTable(void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,const HUF_DTable * DTable)406*c03c5b1cSMartin Matuska size_t HUF_decompress4X1_usingDTable(
407*c03c5b1cSMartin Matuska           void* dst,  size_t dstSize,
408*c03c5b1cSMartin Matuska     const void* cSrc, size_t cSrcSize,
409*c03c5b1cSMartin Matuska     const HUF_DTable* DTable)
410*c03c5b1cSMartin Matuska {
411*c03c5b1cSMartin Matuska     DTableDesc dtd = HUF_getDTableDesc(DTable);
412*c03c5b1cSMartin Matuska     if (dtd.tableType != 0) return ERROR(GENERIC);
413*c03c5b1cSMartin Matuska     return HUF_decompress4X1_usingDTable_internal(dst, dstSize, cSrc, cSrcSize, DTable, /* bmi2 */ 0);
414*c03c5b1cSMartin Matuska }
415*c03c5b1cSMartin Matuska 
HUF_decompress4X1_DCtx_wksp_bmi2(HUF_DTable * dctx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,void * workSpace,size_t wkspSize,int bmi2)416*c03c5b1cSMartin Matuska static size_t HUF_decompress4X1_DCtx_wksp_bmi2(HUF_DTable* dctx, void* dst, size_t dstSize,
417*c03c5b1cSMartin Matuska                                    const void* cSrc, size_t cSrcSize,
418*c03c5b1cSMartin Matuska                                    void* workSpace, size_t wkspSize, int bmi2)
419*c03c5b1cSMartin Matuska {
420*c03c5b1cSMartin Matuska     const BYTE* ip = (const BYTE*) cSrc;
421*c03c5b1cSMartin Matuska 
422*c03c5b1cSMartin Matuska     size_t const hSize = HUF_readDTableX1_wksp (dctx, cSrc, cSrcSize,
423*c03c5b1cSMartin Matuska                                                 workSpace, wkspSize);
424*c03c5b1cSMartin Matuska     if (HUF_isError(hSize)) return hSize;
425*c03c5b1cSMartin Matuska     if (hSize >= cSrcSize) return ERROR(srcSize_wrong);
426*c03c5b1cSMartin Matuska     ip += hSize; cSrcSize -= hSize;
427*c03c5b1cSMartin Matuska 
428*c03c5b1cSMartin Matuska     return HUF_decompress4X1_usingDTable_internal(dst, dstSize, ip, cSrcSize, dctx, bmi2);
429*c03c5b1cSMartin Matuska }
430*c03c5b1cSMartin Matuska 
HUF_decompress4X1_DCtx_wksp(HUF_DTable * dctx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,void * workSpace,size_t wkspSize)431*c03c5b1cSMartin Matuska size_t HUF_decompress4X1_DCtx_wksp(HUF_DTable* dctx, void* dst, size_t dstSize,
432*c03c5b1cSMartin Matuska                                    const void* cSrc, size_t cSrcSize,
433*c03c5b1cSMartin Matuska                                    void* workSpace, size_t wkspSize)
434*c03c5b1cSMartin Matuska {
435*c03c5b1cSMartin Matuska     return HUF_decompress4X1_DCtx_wksp_bmi2(dctx, dst, dstSize, cSrc, cSrcSize, workSpace, wkspSize, 0);
436*c03c5b1cSMartin Matuska }
437*c03c5b1cSMartin Matuska 
438*c03c5b1cSMartin Matuska 
HUF_decompress4X1_DCtx(HUF_DTable * dctx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize)439*c03c5b1cSMartin Matuska size_t HUF_decompress4X1_DCtx (HUF_DTable* dctx, void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize)
440*c03c5b1cSMartin Matuska {
441*c03c5b1cSMartin Matuska     U32 workSpace[HUF_DECOMPRESS_WORKSPACE_SIZE_U32];
442*c03c5b1cSMartin Matuska     return HUF_decompress4X1_DCtx_wksp(dctx, dst, dstSize, cSrc, cSrcSize,
443*c03c5b1cSMartin Matuska                                        workSpace, sizeof(workSpace));
444*c03c5b1cSMartin Matuska }
HUF_decompress4X1(void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize)445*c03c5b1cSMartin Matuska size_t HUF_decompress4X1 (void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize)
446*c03c5b1cSMartin Matuska {
447*c03c5b1cSMartin Matuska     HUF_CREATE_STATIC_DTABLEX1(DTable, HUF_TABLELOG_MAX);
448*c03c5b1cSMartin Matuska     return HUF_decompress4X1_DCtx(DTable, dst, dstSize, cSrc, cSrcSize);
449*c03c5b1cSMartin Matuska }
450*c03c5b1cSMartin Matuska 
451*c03c5b1cSMartin Matuska #endif /* HUF_FORCE_DECOMPRESS_X2 */
452*c03c5b1cSMartin Matuska 
453*c03c5b1cSMartin Matuska 
454*c03c5b1cSMartin Matuska #ifndef HUF_FORCE_DECOMPRESS_X1
455*c03c5b1cSMartin Matuska 
456*c03c5b1cSMartin Matuska /* *************************/
457*c03c5b1cSMartin Matuska /* double-symbols decoding */
458*c03c5b1cSMartin Matuska /* *************************/
459*c03c5b1cSMartin Matuska 
460*c03c5b1cSMartin Matuska typedef struct { U16 sequence; BYTE nbBits; BYTE length; } HUF_DEltX2;  /* double-symbols decoding */
461*c03c5b1cSMartin Matuska typedef struct { BYTE symbol; BYTE weight; } sortedSymbol_t;
462*c03c5b1cSMartin Matuska typedef U32 rankValCol_t[HUF_TABLELOG_MAX + 1];
463*c03c5b1cSMartin Matuska typedef rankValCol_t rankVal_t[HUF_TABLELOG_MAX];
464*c03c5b1cSMartin Matuska 
465*c03c5b1cSMartin Matuska 
466*c03c5b1cSMartin Matuska /* HUF_fillDTableX2Level2() :
467*c03c5b1cSMartin Matuska  * `rankValOrigin` must be a table of at least (HUF_TABLELOG_MAX + 1) U32 */
HUF_fillDTableX2Level2(HUF_DEltX2 * DTable,U32 sizeLog,const U32 consumed,const U32 * rankValOrigin,const int minWeight,const sortedSymbol_t * sortedSymbols,const U32 sortedListSize,U32 nbBitsBaseline,U16 baseSeq)468*c03c5b1cSMartin Matuska static void HUF_fillDTableX2Level2(HUF_DEltX2* DTable, U32 sizeLog, const U32 consumed,
469*c03c5b1cSMartin Matuska                            const U32* rankValOrigin, const int minWeight,
470*c03c5b1cSMartin Matuska                            const sortedSymbol_t* sortedSymbols, const U32 sortedListSize,
471*c03c5b1cSMartin Matuska                            U32 nbBitsBaseline, U16 baseSeq)
472*c03c5b1cSMartin Matuska {
473*c03c5b1cSMartin Matuska     HUF_DEltX2 DElt;
474*c03c5b1cSMartin Matuska     U32 rankVal[HUF_TABLELOG_MAX + 1];
475*c03c5b1cSMartin Matuska 
476*c03c5b1cSMartin Matuska     /* get pre-calculated rankVal */
477*c03c5b1cSMartin Matuska     memcpy(rankVal, rankValOrigin, sizeof(rankVal));
478*c03c5b1cSMartin Matuska 
479*c03c5b1cSMartin Matuska     /* fill skipped values */
480*c03c5b1cSMartin Matuska     if (minWeight>1) {
481*c03c5b1cSMartin Matuska         U32 i, skipSize = rankVal[minWeight];
482*c03c5b1cSMartin Matuska         MEM_writeLE16(&(DElt.sequence), baseSeq);
483*c03c5b1cSMartin Matuska         DElt.nbBits   = (BYTE)(consumed);
484*c03c5b1cSMartin Matuska         DElt.length   = 1;
485*c03c5b1cSMartin Matuska         for (i = 0; i < skipSize; i++)
486*c03c5b1cSMartin Matuska             DTable[i] = DElt;
487*c03c5b1cSMartin Matuska     }
488*c03c5b1cSMartin Matuska 
489*c03c5b1cSMartin Matuska     /* fill DTable */
490*c03c5b1cSMartin Matuska     {   U32 s; for (s=0; s<sortedListSize; s++) {   /* note : sortedSymbols already skipped */
491*c03c5b1cSMartin Matuska             const U32 symbol = sortedSymbols[s].symbol;
492*c03c5b1cSMartin Matuska             const U32 weight = sortedSymbols[s].weight;
493*c03c5b1cSMartin Matuska             const U32 nbBits = nbBitsBaseline - weight;
494*c03c5b1cSMartin Matuska             const U32 length = 1 << (sizeLog-nbBits);
495*c03c5b1cSMartin Matuska             const U32 start = rankVal[weight];
496*c03c5b1cSMartin Matuska             U32 i = start;
497*c03c5b1cSMartin Matuska             const U32 end = start + length;
498*c03c5b1cSMartin Matuska 
499*c03c5b1cSMartin Matuska             MEM_writeLE16(&(DElt.sequence), (U16)(baseSeq + (symbol << 8)));
500*c03c5b1cSMartin Matuska             DElt.nbBits = (BYTE)(nbBits + consumed);
501*c03c5b1cSMartin Matuska             DElt.length = 2;
502*c03c5b1cSMartin Matuska             do { DTable[i++] = DElt; } while (i<end);   /* since length >= 1 */
503*c03c5b1cSMartin Matuska 
504*c03c5b1cSMartin Matuska             rankVal[weight] += length;
505*c03c5b1cSMartin Matuska     }   }
506*c03c5b1cSMartin Matuska }
507*c03c5b1cSMartin Matuska 
508*c03c5b1cSMartin Matuska 
HUF_fillDTableX2(HUF_DEltX2 * DTable,const U32 targetLog,const sortedSymbol_t * sortedList,const U32 sortedListSize,const U32 * rankStart,rankVal_t rankValOrigin,const U32 maxWeight,const U32 nbBitsBaseline)509*c03c5b1cSMartin Matuska static void HUF_fillDTableX2(HUF_DEltX2* DTable, const U32 targetLog,
510*c03c5b1cSMartin Matuska                            const sortedSymbol_t* sortedList, const U32 sortedListSize,
511*c03c5b1cSMartin Matuska                            const U32* rankStart, rankVal_t rankValOrigin, const U32 maxWeight,
512*c03c5b1cSMartin Matuska                            const U32 nbBitsBaseline)
513*c03c5b1cSMartin Matuska {
514*c03c5b1cSMartin Matuska     U32 rankVal[HUF_TABLELOG_MAX + 1];
515*c03c5b1cSMartin Matuska     const int scaleLog = nbBitsBaseline - targetLog;   /* note : targetLog >= srcLog, hence scaleLog <= 1 */
516*c03c5b1cSMartin Matuska     const U32 minBits  = nbBitsBaseline - maxWeight;
517*c03c5b1cSMartin Matuska     U32 s;
518*c03c5b1cSMartin Matuska 
519*c03c5b1cSMartin Matuska     memcpy(rankVal, rankValOrigin, sizeof(rankVal));
520*c03c5b1cSMartin Matuska 
521*c03c5b1cSMartin Matuska     /* fill DTable */
522*c03c5b1cSMartin Matuska     for (s=0; s<sortedListSize; s++) {
523*c03c5b1cSMartin Matuska         const U16 symbol = sortedList[s].symbol;
524*c03c5b1cSMartin Matuska         const U32 weight = sortedList[s].weight;
525*c03c5b1cSMartin Matuska         const U32 nbBits = nbBitsBaseline - weight;
526*c03c5b1cSMartin Matuska         const U32 start = rankVal[weight];
527*c03c5b1cSMartin Matuska         const U32 length = 1 << (targetLog-nbBits);
528*c03c5b1cSMartin Matuska 
529*c03c5b1cSMartin Matuska         if (targetLog-nbBits >= minBits) {   /* enough room for a second symbol */
530*c03c5b1cSMartin Matuska             U32 sortedRank;
531*c03c5b1cSMartin Matuska             int minWeight = nbBits + scaleLog;
532*c03c5b1cSMartin Matuska             if (minWeight < 1) minWeight = 1;
533*c03c5b1cSMartin Matuska             sortedRank = rankStart[minWeight];
534*c03c5b1cSMartin Matuska             HUF_fillDTableX2Level2(DTable+start, targetLog-nbBits, nbBits,
535*c03c5b1cSMartin Matuska                            rankValOrigin[nbBits], minWeight,
536*c03c5b1cSMartin Matuska                            sortedList+sortedRank, sortedListSize-sortedRank,
537*c03c5b1cSMartin Matuska                            nbBitsBaseline, symbol);
538*c03c5b1cSMartin Matuska         } else {
539*c03c5b1cSMartin Matuska             HUF_DEltX2 DElt;
540*c03c5b1cSMartin Matuska             MEM_writeLE16(&(DElt.sequence), symbol);
541*c03c5b1cSMartin Matuska             DElt.nbBits = (BYTE)(nbBits);
542*c03c5b1cSMartin Matuska             DElt.length = 1;
543*c03c5b1cSMartin Matuska             {   U32 const end = start + length;
544*c03c5b1cSMartin Matuska                 U32 u;
545*c03c5b1cSMartin Matuska                 for (u = start; u < end; u++) DTable[u] = DElt;
546*c03c5b1cSMartin Matuska         }   }
547*c03c5b1cSMartin Matuska         rankVal[weight] += length;
548*c03c5b1cSMartin Matuska     }
549*c03c5b1cSMartin Matuska }
550*c03c5b1cSMartin Matuska 
HUF_readDTableX2_wksp(HUF_DTable * DTable,const void * src,size_t srcSize,void * workSpace,size_t wkspSize)551*c03c5b1cSMartin Matuska size_t HUF_readDTableX2_wksp(HUF_DTable* DTable,
552*c03c5b1cSMartin Matuska                        const void* src, size_t srcSize,
553*c03c5b1cSMartin Matuska                              void* workSpace, size_t wkspSize)
554*c03c5b1cSMartin Matuska {
555*c03c5b1cSMartin Matuska     U32 tableLog, maxW, sizeOfSort, nbSymbols;
556*c03c5b1cSMartin Matuska     DTableDesc dtd = HUF_getDTableDesc(DTable);
557*c03c5b1cSMartin Matuska     U32 const maxTableLog = dtd.maxTableLog;
558*c03c5b1cSMartin Matuska     size_t iSize;
559*c03c5b1cSMartin Matuska     void* dtPtr = DTable+1;   /* force compiler to avoid strict-aliasing */
560*c03c5b1cSMartin Matuska     HUF_DEltX2* const dt = (HUF_DEltX2*)dtPtr;
561*c03c5b1cSMartin Matuska     U32 *rankStart;
562*c03c5b1cSMartin Matuska 
563*c03c5b1cSMartin Matuska     rankValCol_t* rankVal;
564*c03c5b1cSMartin Matuska     U32* rankStats;
565*c03c5b1cSMartin Matuska     U32* rankStart0;
566*c03c5b1cSMartin Matuska     sortedSymbol_t* sortedSymbol;
567*c03c5b1cSMartin Matuska     BYTE* weightList;
568*c03c5b1cSMartin Matuska     size_t spaceUsed32 = 0;
569*c03c5b1cSMartin Matuska 
570*c03c5b1cSMartin Matuska     rankVal = (rankValCol_t *)((U32 *)workSpace + spaceUsed32);
571*c03c5b1cSMartin Matuska     spaceUsed32 += (sizeof(rankValCol_t) * HUF_TABLELOG_MAX) >> 2;
572*c03c5b1cSMartin Matuska     rankStats = (U32 *)workSpace + spaceUsed32;
573*c03c5b1cSMartin Matuska     spaceUsed32 += HUF_TABLELOG_MAX + 1;
574*c03c5b1cSMartin Matuska     rankStart0 = (U32 *)workSpace + spaceUsed32;
575*c03c5b1cSMartin Matuska     spaceUsed32 += HUF_TABLELOG_MAX + 2;
576*c03c5b1cSMartin Matuska     sortedSymbol = (sortedSymbol_t *)workSpace + (spaceUsed32 * sizeof(U32)) / sizeof(sortedSymbol_t);
577*c03c5b1cSMartin Matuska     spaceUsed32 += HUF_ALIGN(sizeof(sortedSymbol_t) * (HUF_SYMBOLVALUE_MAX + 1), sizeof(U32)) >> 2;
578*c03c5b1cSMartin Matuska     weightList = (BYTE *)((U32 *)workSpace + spaceUsed32);
579*c03c5b1cSMartin Matuska     spaceUsed32 += HUF_ALIGN(HUF_SYMBOLVALUE_MAX + 1, sizeof(U32)) >> 2;
580*c03c5b1cSMartin Matuska 
581*c03c5b1cSMartin Matuska     if ((spaceUsed32 << 2) > wkspSize) return ERROR(tableLog_tooLarge);
582*c03c5b1cSMartin Matuska 
583*c03c5b1cSMartin Matuska     rankStart = rankStart0 + 1;
584*c03c5b1cSMartin Matuska     memset(rankStats, 0, sizeof(U32) * (2 * HUF_TABLELOG_MAX + 2 + 1));
585*c03c5b1cSMartin Matuska 
586*c03c5b1cSMartin Matuska     DEBUG_STATIC_ASSERT(sizeof(HUF_DEltX2) == sizeof(HUF_DTable));   /* if compiler fails here, assertion is wrong */
587*c03c5b1cSMartin Matuska     if (maxTableLog > HUF_TABLELOG_MAX) return ERROR(tableLog_tooLarge);
588*c03c5b1cSMartin Matuska     /* memset(weightList, 0, sizeof(weightList)); */  /* is not necessary, even though some analyzer complain ... */
589*c03c5b1cSMartin Matuska 
590*c03c5b1cSMartin Matuska     iSize = HUF_readStats(weightList, HUF_SYMBOLVALUE_MAX + 1, rankStats, &nbSymbols, &tableLog, src, srcSize);
591*c03c5b1cSMartin Matuska     if (HUF_isError(iSize)) return iSize;
592*c03c5b1cSMartin Matuska 
593*c03c5b1cSMartin Matuska     /* check result */
594*c03c5b1cSMartin Matuska     if (tableLog > maxTableLog) return ERROR(tableLog_tooLarge);   /* DTable can't fit code depth */
595*c03c5b1cSMartin Matuska 
596*c03c5b1cSMartin Matuska     /* find maxWeight */
597*c03c5b1cSMartin Matuska     for (maxW = tableLog; rankStats[maxW]==0; maxW--) {}  /* necessarily finds a solution before 0 */
598*c03c5b1cSMartin Matuska 
599*c03c5b1cSMartin Matuska     /* Get start index of each weight */
600*c03c5b1cSMartin Matuska     {   U32 w, nextRankStart = 0;
601*c03c5b1cSMartin Matuska         for (w=1; w<maxW+1; w++) {
602*c03c5b1cSMartin Matuska             U32 current = nextRankStart;
603*c03c5b1cSMartin Matuska             nextRankStart += rankStats[w];
604*c03c5b1cSMartin Matuska             rankStart[w] = current;
605*c03c5b1cSMartin Matuska         }
606*c03c5b1cSMartin Matuska         rankStart[0] = nextRankStart;   /* put all 0w symbols at the end of sorted list*/
607*c03c5b1cSMartin Matuska         sizeOfSort = nextRankStart;
608*c03c5b1cSMartin Matuska     }
609*c03c5b1cSMartin Matuska 
610*c03c5b1cSMartin Matuska     /* sort symbols by weight */
611*c03c5b1cSMartin Matuska     {   U32 s;
612*c03c5b1cSMartin Matuska         for (s=0; s<nbSymbols; s++) {
613*c03c5b1cSMartin Matuska             U32 const w = weightList[s];
614*c03c5b1cSMartin Matuska             U32 const r = rankStart[w]++;
615*c03c5b1cSMartin Matuska             sortedSymbol[r].symbol = (BYTE)s;
616*c03c5b1cSMartin Matuska             sortedSymbol[r].weight = (BYTE)w;
617*c03c5b1cSMartin Matuska         }
618*c03c5b1cSMartin Matuska         rankStart[0] = 0;   /* forget 0w symbols; this is beginning of weight(1) */
619*c03c5b1cSMartin Matuska     }
620*c03c5b1cSMartin Matuska 
621*c03c5b1cSMartin Matuska     /* Build rankVal */
622*c03c5b1cSMartin Matuska     {   U32* const rankVal0 = rankVal[0];
623*c03c5b1cSMartin Matuska         {   int const rescale = (maxTableLog-tableLog) - 1;   /* tableLog <= maxTableLog */
624*c03c5b1cSMartin Matuska             U32 nextRankVal = 0;
625*c03c5b1cSMartin Matuska             U32 w;
626*c03c5b1cSMartin Matuska             for (w=1; w<maxW+1; w++) {
627*c03c5b1cSMartin Matuska                 U32 current = nextRankVal;
628*c03c5b1cSMartin Matuska                 nextRankVal += rankStats[w] << (w+rescale);
629*c03c5b1cSMartin Matuska                 rankVal0[w] = current;
630*c03c5b1cSMartin Matuska         }   }
631*c03c5b1cSMartin Matuska         {   U32 const minBits = tableLog+1 - maxW;
632*c03c5b1cSMartin Matuska             U32 consumed;
633*c03c5b1cSMartin Matuska             for (consumed = minBits; consumed < maxTableLog - minBits + 1; consumed++) {
634*c03c5b1cSMartin Matuska                 U32* const rankValPtr = rankVal[consumed];
635*c03c5b1cSMartin Matuska                 U32 w;
636*c03c5b1cSMartin Matuska                 for (w = 1; w < maxW+1; w++) {
637*c03c5b1cSMartin Matuska                     rankValPtr[w] = rankVal0[w] >> consumed;
638*c03c5b1cSMartin Matuska     }   }   }   }
639*c03c5b1cSMartin Matuska 
640*c03c5b1cSMartin Matuska     HUF_fillDTableX2(dt, maxTableLog,
641*c03c5b1cSMartin Matuska                    sortedSymbol, sizeOfSort,
642*c03c5b1cSMartin Matuska                    rankStart0, rankVal, maxW,
643*c03c5b1cSMartin Matuska                    tableLog+1);
644*c03c5b1cSMartin Matuska 
645*c03c5b1cSMartin Matuska     dtd.tableLog = (BYTE)maxTableLog;
646*c03c5b1cSMartin Matuska     dtd.tableType = 1;
647*c03c5b1cSMartin Matuska     memcpy(DTable, &dtd, sizeof(dtd));
648*c03c5b1cSMartin Matuska     return iSize;
649*c03c5b1cSMartin Matuska }
650*c03c5b1cSMartin Matuska 
HUF_readDTableX2(HUF_DTable * DTable,const void * src,size_t srcSize)651*c03c5b1cSMartin Matuska size_t HUF_readDTableX2(HUF_DTable* DTable, const void* src, size_t srcSize)
652*c03c5b1cSMartin Matuska {
653*c03c5b1cSMartin Matuska   U32 workSpace[HUF_DECOMPRESS_WORKSPACE_SIZE_U32];
654*c03c5b1cSMartin Matuska   return HUF_readDTableX2_wksp(DTable, src, srcSize,
655*c03c5b1cSMartin Matuska                                workSpace, sizeof(workSpace));
656*c03c5b1cSMartin Matuska }
657*c03c5b1cSMartin Matuska 
658*c03c5b1cSMartin Matuska 
659*c03c5b1cSMartin Matuska FORCE_INLINE_TEMPLATE U32
HUF_decodeSymbolX2(void * op,BIT_DStream_t * DStream,const HUF_DEltX2 * dt,const U32 dtLog)660*c03c5b1cSMartin Matuska HUF_decodeSymbolX2(void* op, BIT_DStream_t* DStream, const HUF_DEltX2* dt, const U32 dtLog)
661*c03c5b1cSMartin Matuska {
662*c03c5b1cSMartin Matuska     size_t const val = BIT_lookBitsFast(DStream, dtLog);   /* note : dtLog >= 1 */
663*c03c5b1cSMartin Matuska     memcpy(op, dt+val, 2);
664*c03c5b1cSMartin Matuska     BIT_skipBits(DStream, dt[val].nbBits);
665*c03c5b1cSMartin Matuska     return dt[val].length;
666*c03c5b1cSMartin Matuska }
667*c03c5b1cSMartin Matuska 
668*c03c5b1cSMartin Matuska FORCE_INLINE_TEMPLATE U32
HUF_decodeLastSymbolX2(void * op,BIT_DStream_t * DStream,const HUF_DEltX2 * dt,const U32 dtLog)669*c03c5b1cSMartin Matuska HUF_decodeLastSymbolX2(void* op, BIT_DStream_t* DStream, const HUF_DEltX2* dt, const U32 dtLog)
670*c03c5b1cSMartin Matuska {
671*c03c5b1cSMartin Matuska     size_t const val = BIT_lookBitsFast(DStream, dtLog);   /* note : dtLog >= 1 */
672*c03c5b1cSMartin Matuska     memcpy(op, dt+val, 1);
673*c03c5b1cSMartin Matuska     if (dt[val].length==1) BIT_skipBits(DStream, dt[val].nbBits);
674*c03c5b1cSMartin Matuska     else {
675*c03c5b1cSMartin Matuska         if (DStream->bitsConsumed < (sizeof(DStream->bitContainer)*8)) {
676*c03c5b1cSMartin Matuska             BIT_skipBits(DStream, dt[val].nbBits);
677*c03c5b1cSMartin Matuska             if (DStream->bitsConsumed > (sizeof(DStream->bitContainer)*8))
678*c03c5b1cSMartin Matuska                 /* ugly hack; works only because it's the last symbol. Note : can't easily extract nbBits from just this symbol */
679*c03c5b1cSMartin Matuska                 DStream->bitsConsumed = (sizeof(DStream->bitContainer)*8);
680*c03c5b1cSMartin Matuska     }   }
681*c03c5b1cSMartin Matuska     return 1;
682*c03c5b1cSMartin Matuska }
683*c03c5b1cSMartin Matuska 
684*c03c5b1cSMartin Matuska #define HUF_DECODE_SYMBOLX2_0(ptr, DStreamPtr) \
685*c03c5b1cSMartin Matuska     ptr += HUF_decodeSymbolX2(ptr, DStreamPtr, dt, dtLog)
686*c03c5b1cSMartin Matuska 
687*c03c5b1cSMartin Matuska #define HUF_DECODE_SYMBOLX2_1(ptr, DStreamPtr) \
688*c03c5b1cSMartin Matuska     if (MEM_64bits() || (HUF_TABLELOG_MAX<=12)) \
689*c03c5b1cSMartin Matuska         ptr += HUF_decodeSymbolX2(ptr, DStreamPtr, dt, dtLog)
690*c03c5b1cSMartin Matuska 
691*c03c5b1cSMartin Matuska #define HUF_DECODE_SYMBOLX2_2(ptr, DStreamPtr) \
692*c03c5b1cSMartin Matuska     if (MEM_64bits()) \
693*c03c5b1cSMartin Matuska         ptr += HUF_decodeSymbolX2(ptr, DStreamPtr, dt, dtLog)
694*c03c5b1cSMartin Matuska 
695*c03c5b1cSMartin Matuska HINT_INLINE size_t
HUF_decodeStreamX2(BYTE * p,BIT_DStream_t * bitDPtr,BYTE * const pEnd,const HUF_DEltX2 * const dt,const U32 dtLog)696*c03c5b1cSMartin Matuska HUF_decodeStreamX2(BYTE* p, BIT_DStream_t* bitDPtr, BYTE* const pEnd,
697*c03c5b1cSMartin Matuska                 const HUF_DEltX2* const dt, const U32 dtLog)
698*c03c5b1cSMartin Matuska {
699*c03c5b1cSMartin Matuska     BYTE* const pStart = p;
700*c03c5b1cSMartin Matuska 
701*c03c5b1cSMartin Matuska     /* up to 8 symbols at a time */
702*c03c5b1cSMartin Matuska     while ((BIT_reloadDStream(bitDPtr) == BIT_DStream_unfinished) & (p < pEnd-(sizeof(bitDPtr->bitContainer)-1))) {
703*c03c5b1cSMartin Matuska         HUF_DECODE_SYMBOLX2_2(p, bitDPtr);
704*c03c5b1cSMartin Matuska         HUF_DECODE_SYMBOLX2_1(p, bitDPtr);
705*c03c5b1cSMartin Matuska         HUF_DECODE_SYMBOLX2_2(p, bitDPtr);
706*c03c5b1cSMartin Matuska         HUF_DECODE_SYMBOLX2_0(p, bitDPtr);
707*c03c5b1cSMartin Matuska     }
708*c03c5b1cSMartin Matuska 
709*c03c5b1cSMartin Matuska     /* closer to end : up to 2 symbols at a time */
710*c03c5b1cSMartin Matuska     while ((BIT_reloadDStream(bitDPtr) == BIT_DStream_unfinished) & (p <= pEnd-2))
711*c03c5b1cSMartin Matuska         HUF_DECODE_SYMBOLX2_0(p, bitDPtr);
712*c03c5b1cSMartin Matuska 
713*c03c5b1cSMartin Matuska     while (p <= pEnd-2)
714*c03c5b1cSMartin Matuska         HUF_DECODE_SYMBOLX2_0(p, bitDPtr);   /* no need to reload : reached the end of DStream */
715*c03c5b1cSMartin Matuska 
716*c03c5b1cSMartin Matuska     if (p < pEnd)
717*c03c5b1cSMartin Matuska         p += HUF_decodeLastSymbolX2(p, bitDPtr, dt, dtLog);
718*c03c5b1cSMartin Matuska 
719*c03c5b1cSMartin Matuska     return p-pStart;
720*c03c5b1cSMartin Matuska }
721*c03c5b1cSMartin Matuska 
722*c03c5b1cSMartin Matuska FORCE_INLINE_TEMPLATE size_t
HUF_decompress1X2_usingDTable_internal_body(void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,const HUF_DTable * DTable)723*c03c5b1cSMartin Matuska HUF_decompress1X2_usingDTable_internal_body(
724*c03c5b1cSMartin Matuska           void* dst,  size_t dstSize,
725*c03c5b1cSMartin Matuska     const void* cSrc, size_t cSrcSize,
726*c03c5b1cSMartin Matuska     const HUF_DTable* DTable)
727*c03c5b1cSMartin Matuska {
728*c03c5b1cSMartin Matuska     BIT_DStream_t bitD;
729*c03c5b1cSMartin Matuska 
730*c03c5b1cSMartin Matuska     /* Init */
731*c03c5b1cSMartin Matuska     CHECK_F( BIT_initDStream(&bitD, cSrc, cSrcSize) );
732*c03c5b1cSMartin Matuska 
733*c03c5b1cSMartin Matuska     /* decode */
734*c03c5b1cSMartin Matuska     {   BYTE* const ostart = (BYTE*) dst;
735*c03c5b1cSMartin Matuska         BYTE* const oend = ostart + dstSize;
736*c03c5b1cSMartin Matuska         const void* const dtPtr = DTable+1;   /* force compiler to not use strict-aliasing */
737*c03c5b1cSMartin Matuska         const HUF_DEltX2* const dt = (const HUF_DEltX2*)dtPtr;
738*c03c5b1cSMartin Matuska         DTableDesc const dtd = HUF_getDTableDesc(DTable);
739*c03c5b1cSMartin Matuska         HUF_decodeStreamX2(ostart, &bitD, oend, dt, dtd.tableLog);
740*c03c5b1cSMartin Matuska     }
741*c03c5b1cSMartin Matuska 
742*c03c5b1cSMartin Matuska     /* check */
743*c03c5b1cSMartin Matuska     if (!BIT_endOfDStream(&bitD)) return ERROR(corruption_detected);
744*c03c5b1cSMartin Matuska 
745*c03c5b1cSMartin Matuska     /* decoded size */
746*c03c5b1cSMartin Matuska     return dstSize;
747*c03c5b1cSMartin Matuska }
748*c03c5b1cSMartin Matuska 
749*c03c5b1cSMartin Matuska FORCE_INLINE_TEMPLATE size_t
HUF_decompress4X2_usingDTable_internal_body(void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,const HUF_DTable * DTable)750*c03c5b1cSMartin Matuska HUF_decompress4X2_usingDTable_internal_body(
751*c03c5b1cSMartin Matuska           void* dst,  size_t dstSize,
752*c03c5b1cSMartin Matuska     const void* cSrc, size_t cSrcSize,
753*c03c5b1cSMartin Matuska     const HUF_DTable* DTable)
754*c03c5b1cSMartin Matuska {
755*c03c5b1cSMartin Matuska     if (cSrcSize < 10) return ERROR(corruption_detected);   /* strict minimum : jump table + 1 byte per stream */
756*c03c5b1cSMartin Matuska 
757*c03c5b1cSMartin Matuska     {   const BYTE* const istart = (const BYTE*) cSrc;
758*c03c5b1cSMartin Matuska         BYTE* const ostart = (BYTE*) dst;
759*c03c5b1cSMartin Matuska         BYTE* const oend = ostart + dstSize;
760*c03c5b1cSMartin Matuska         BYTE* const olimit = oend - (sizeof(size_t)-1);
761*c03c5b1cSMartin Matuska         const void* const dtPtr = DTable+1;
762*c03c5b1cSMartin Matuska         const HUF_DEltX2* const dt = (const HUF_DEltX2*)dtPtr;
763*c03c5b1cSMartin Matuska 
764*c03c5b1cSMartin Matuska         /* Init */
765*c03c5b1cSMartin Matuska         BIT_DStream_t bitD1;
766*c03c5b1cSMartin Matuska         BIT_DStream_t bitD2;
767*c03c5b1cSMartin Matuska         BIT_DStream_t bitD3;
768*c03c5b1cSMartin Matuska         BIT_DStream_t bitD4;
769*c03c5b1cSMartin Matuska         size_t const length1 = MEM_readLE16(istart);
770*c03c5b1cSMartin Matuska         size_t const length2 = MEM_readLE16(istart+2);
771*c03c5b1cSMartin Matuska         size_t const length3 = MEM_readLE16(istart+4);
772*c03c5b1cSMartin Matuska         size_t const length4 = cSrcSize - (length1 + length2 + length3 + 6);
773*c03c5b1cSMartin Matuska         const BYTE* const istart1 = istart + 6;  /* jumpTable */
774*c03c5b1cSMartin Matuska         const BYTE* const istart2 = istart1 + length1;
775*c03c5b1cSMartin Matuska         const BYTE* const istart3 = istart2 + length2;
776*c03c5b1cSMartin Matuska         const BYTE* const istart4 = istart3 + length3;
777*c03c5b1cSMartin Matuska         size_t const segmentSize = (dstSize+3) / 4;
778*c03c5b1cSMartin Matuska         BYTE* const opStart2 = ostart + segmentSize;
779*c03c5b1cSMartin Matuska         BYTE* const opStart3 = opStart2 + segmentSize;
780*c03c5b1cSMartin Matuska         BYTE* const opStart4 = opStart3 + segmentSize;
781*c03c5b1cSMartin Matuska         BYTE* op1 = ostart;
782*c03c5b1cSMartin Matuska         BYTE* op2 = opStart2;
783*c03c5b1cSMartin Matuska         BYTE* op3 = opStart3;
784*c03c5b1cSMartin Matuska         BYTE* op4 = opStart4;
785*c03c5b1cSMartin Matuska         U32 endSignal = 1;
786*c03c5b1cSMartin Matuska         DTableDesc const dtd = HUF_getDTableDesc(DTable);
787*c03c5b1cSMartin Matuska         U32 const dtLog = dtd.tableLog;
788*c03c5b1cSMartin Matuska 
789*c03c5b1cSMartin Matuska         if (length4 > cSrcSize) return ERROR(corruption_detected);   /* overflow */
790*c03c5b1cSMartin Matuska         CHECK_F( BIT_initDStream(&bitD1, istart1, length1) );
791*c03c5b1cSMartin Matuska         CHECK_F( BIT_initDStream(&bitD2, istart2, length2) );
792*c03c5b1cSMartin Matuska         CHECK_F( BIT_initDStream(&bitD3, istart3, length3) );
793*c03c5b1cSMartin Matuska         CHECK_F( BIT_initDStream(&bitD4, istart4, length4) );
794*c03c5b1cSMartin Matuska 
795*c03c5b1cSMartin Matuska         /* 16-32 symbols per loop (4-8 symbols per stream) */
796*c03c5b1cSMartin Matuska         for ( ; (endSignal) & (op4 < olimit); ) {
797*c03c5b1cSMartin Matuska #if defined(__clang__) && (defined(__x86_64__) || defined(__i386__))
798*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op1, &bitD1);
799*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_1(op1, &bitD1);
800*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op1, &bitD1);
801*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_0(op1, &bitD1);
802*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op2, &bitD2);
803*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_1(op2, &bitD2);
804*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op2, &bitD2);
805*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_0(op2, &bitD2);
806*c03c5b1cSMartin Matuska             endSignal &= BIT_reloadDStreamFast(&bitD1) == BIT_DStream_unfinished;
807*c03c5b1cSMartin Matuska             endSignal &= BIT_reloadDStreamFast(&bitD2) == BIT_DStream_unfinished;
808*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op3, &bitD3);
809*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_1(op3, &bitD3);
810*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op3, &bitD3);
811*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_0(op3, &bitD3);
812*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op4, &bitD4);
813*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_1(op4, &bitD4);
814*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op4, &bitD4);
815*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_0(op4, &bitD4);
816*c03c5b1cSMartin Matuska             endSignal &= BIT_reloadDStreamFast(&bitD3) == BIT_DStream_unfinished;
817*c03c5b1cSMartin Matuska             endSignal &= BIT_reloadDStreamFast(&bitD4) == BIT_DStream_unfinished;
818*c03c5b1cSMartin Matuska #else
819*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op1, &bitD1);
820*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op2, &bitD2);
821*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op3, &bitD3);
822*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op4, &bitD4);
823*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_1(op1, &bitD1);
824*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_1(op2, &bitD2);
825*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_1(op3, &bitD3);
826*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_1(op4, &bitD4);
827*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op1, &bitD1);
828*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op2, &bitD2);
829*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op3, &bitD3);
830*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_2(op4, &bitD4);
831*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_0(op1, &bitD1);
832*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_0(op2, &bitD2);
833*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_0(op3, &bitD3);
834*c03c5b1cSMartin Matuska             HUF_DECODE_SYMBOLX2_0(op4, &bitD4);
835*c03c5b1cSMartin Matuska             endSignal = (U32)LIKELY(
836*c03c5b1cSMartin Matuska                         (BIT_reloadDStreamFast(&bitD1) == BIT_DStream_unfinished)
837*c03c5b1cSMartin Matuska                       & (BIT_reloadDStreamFast(&bitD2) == BIT_DStream_unfinished)
838*c03c5b1cSMartin Matuska                       & (BIT_reloadDStreamFast(&bitD3) == BIT_DStream_unfinished)
839*c03c5b1cSMartin Matuska                       & (BIT_reloadDStreamFast(&bitD4) == BIT_DStream_unfinished));
840*c03c5b1cSMartin Matuska #endif
841*c03c5b1cSMartin Matuska         }
842*c03c5b1cSMartin Matuska 
843*c03c5b1cSMartin Matuska         /* check corruption */
844*c03c5b1cSMartin Matuska         if (op1 > opStart2) return ERROR(corruption_detected);
845*c03c5b1cSMartin Matuska         if (op2 > opStart3) return ERROR(corruption_detected);
846*c03c5b1cSMartin Matuska         if (op3 > opStart4) return ERROR(corruption_detected);
847*c03c5b1cSMartin Matuska         /* note : op4 already verified within main loop */
848*c03c5b1cSMartin Matuska 
849*c03c5b1cSMartin Matuska         /* finish bitStreams one by one */
850*c03c5b1cSMartin Matuska         HUF_decodeStreamX2(op1, &bitD1, opStart2, dt, dtLog);
851*c03c5b1cSMartin Matuska         HUF_decodeStreamX2(op2, &bitD2, opStart3, dt, dtLog);
852*c03c5b1cSMartin Matuska         HUF_decodeStreamX2(op3, &bitD3, opStart4, dt, dtLog);
853*c03c5b1cSMartin Matuska         HUF_decodeStreamX2(op4, &bitD4, oend,     dt, dtLog);
854*c03c5b1cSMartin Matuska 
855*c03c5b1cSMartin Matuska         /* check */
856*c03c5b1cSMartin Matuska         { U32 const endCheck = BIT_endOfDStream(&bitD1) & BIT_endOfDStream(&bitD2) & BIT_endOfDStream(&bitD3) & BIT_endOfDStream(&bitD4);
857*c03c5b1cSMartin Matuska           if (!endCheck) return ERROR(corruption_detected); }
858*c03c5b1cSMartin Matuska 
859*c03c5b1cSMartin Matuska         /* decoded size */
860*c03c5b1cSMartin Matuska         return dstSize;
861*c03c5b1cSMartin Matuska     }
862*c03c5b1cSMartin Matuska }
863*c03c5b1cSMartin Matuska 
864*c03c5b1cSMartin Matuska HUF_DGEN(HUF_decompress1X2_usingDTable_internal)
HUF_DGEN(HUF_decompress4X2_usingDTable_internal)865*c03c5b1cSMartin Matuska HUF_DGEN(HUF_decompress4X2_usingDTable_internal)
866*c03c5b1cSMartin Matuska 
867*c03c5b1cSMartin Matuska size_t HUF_decompress1X2_usingDTable(
868*c03c5b1cSMartin Matuska           void* dst,  size_t dstSize,
869*c03c5b1cSMartin Matuska     const void* cSrc, size_t cSrcSize,
870*c03c5b1cSMartin Matuska     const HUF_DTable* DTable)
871*c03c5b1cSMartin Matuska {
872*c03c5b1cSMartin Matuska     DTableDesc dtd = HUF_getDTableDesc(DTable);
873*c03c5b1cSMartin Matuska     if (dtd.tableType != 1) return ERROR(GENERIC);
874*c03c5b1cSMartin Matuska     return HUF_decompress1X2_usingDTable_internal(dst, dstSize, cSrc, cSrcSize, DTable, /* bmi2 */ 0);
875*c03c5b1cSMartin Matuska }
876*c03c5b1cSMartin Matuska 
HUF_decompress1X2_DCtx_wksp(HUF_DTable * DCtx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,void * workSpace,size_t wkspSize)877*c03c5b1cSMartin Matuska size_t HUF_decompress1X2_DCtx_wksp(HUF_DTable* DCtx, void* dst, size_t dstSize,
878*c03c5b1cSMartin Matuska                                    const void* cSrc, size_t cSrcSize,
879*c03c5b1cSMartin Matuska                                    void* workSpace, size_t wkspSize)
880*c03c5b1cSMartin Matuska {
881*c03c5b1cSMartin Matuska     const BYTE* ip = (const BYTE*) cSrc;
882*c03c5b1cSMartin Matuska 
883*c03c5b1cSMartin Matuska     size_t const hSize = HUF_readDTableX2_wksp(DCtx, cSrc, cSrcSize,
884*c03c5b1cSMartin Matuska                                                workSpace, wkspSize);
885*c03c5b1cSMartin Matuska     if (HUF_isError(hSize)) return hSize;
886*c03c5b1cSMartin Matuska     if (hSize >= cSrcSize) return ERROR(srcSize_wrong);
887*c03c5b1cSMartin Matuska     ip += hSize; cSrcSize -= hSize;
888*c03c5b1cSMartin Matuska 
889*c03c5b1cSMartin Matuska     return HUF_decompress1X2_usingDTable_internal(dst, dstSize, ip, cSrcSize, DCtx, /* bmi2 */ 0);
890*c03c5b1cSMartin Matuska }
891*c03c5b1cSMartin Matuska 
892*c03c5b1cSMartin Matuska 
HUF_decompress1X2_DCtx(HUF_DTable * DCtx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize)893*c03c5b1cSMartin Matuska size_t HUF_decompress1X2_DCtx(HUF_DTable* DCtx, void* dst, size_t dstSize,
894*c03c5b1cSMartin Matuska                               const void* cSrc, size_t cSrcSize)
895*c03c5b1cSMartin Matuska {
896*c03c5b1cSMartin Matuska     U32 workSpace[HUF_DECOMPRESS_WORKSPACE_SIZE_U32];
897*c03c5b1cSMartin Matuska     return HUF_decompress1X2_DCtx_wksp(DCtx, dst, dstSize, cSrc, cSrcSize,
898*c03c5b1cSMartin Matuska                                        workSpace, sizeof(workSpace));
899*c03c5b1cSMartin Matuska }
900*c03c5b1cSMartin Matuska 
HUF_decompress1X2(void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize)901*c03c5b1cSMartin Matuska size_t HUF_decompress1X2 (void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize)
902*c03c5b1cSMartin Matuska {
903*c03c5b1cSMartin Matuska     HUF_CREATE_STATIC_DTABLEX2(DTable, HUF_TABLELOG_MAX);
904*c03c5b1cSMartin Matuska     return HUF_decompress1X2_DCtx(DTable, dst, dstSize, cSrc, cSrcSize);
905*c03c5b1cSMartin Matuska }
906*c03c5b1cSMartin Matuska 
HUF_decompress4X2_usingDTable(void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,const HUF_DTable * DTable)907*c03c5b1cSMartin Matuska size_t HUF_decompress4X2_usingDTable(
908*c03c5b1cSMartin Matuska           void* dst,  size_t dstSize,
909*c03c5b1cSMartin Matuska     const void* cSrc, size_t cSrcSize,
910*c03c5b1cSMartin Matuska     const HUF_DTable* DTable)
911*c03c5b1cSMartin Matuska {
912*c03c5b1cSMartin Matuska     DTableDesc dtd = HUF_getDTableDesc(DTable);
913*c03c5b1cSMartin Matuska     if (dtd.tableType != 1) return ERROR(GENERIC);
914*c03c5b1cSMartin Matuska     return HUF_decompress4X2_usingDTable_internal(dst, dstSize, cSrc, cSrcSize, DTable, /* bmi2 */ 0);
915*c03c5b1cSMartin Matuska }
916*c03c5b1cSMartin Matuska 
HUF_decompress4X2_DCtx_wksp_bmi2(HUF_DTable * dctx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,void * workSpace,size_t wkspSize,int bmi2)917*c03c5b1cSMartin Matuska static size_t HUF_decompress4X2_DCtx_wksp_bmi2(HUF_DTable* dctx, void* dst, size_t dstSize,
918*c03c5b1cSMartin Matuska                                    const void* cSrc, size_t cSrcSize,
919*c03c5b1cSMartin Matuska                                    void* workSpace, size_t wkspSize, int bmi2)
920*c03c5b1cSMartin Matuska {
921*c03c5b1cSMartin Matuska     const BYTE* ip = (const BYTE*) cSrc;
922*c03c5b1cSMartin Matuska 
923*c03c5b1cSMartin Matuska     size_t hSize = HUF_readDTableX2_wksp(dctx, cSrc, cSrcSize,
924*c03c5b1cSMartin Matuska                                          workSpace, wkspSize);
925*c03c5b1cSMartin Matuska     if (HUF_isError(hSize)) return hSize;
926*c03c5b1cSMartin Matuska     if (hSize >= cSrcSize) return ERROR(srcSize_wrong);
927*c03c5b1cSMartin Matuska     ip += hSize; cSrcSize -= hSize;
928*c03c5b1cSMartin Matuska 
929*c03c5b1cSMartin Matuska     return HUF_decompress4X2_usingDTable_internal(dst, dstSize, ip, cSrcSize, dctx, bmi2);
930*c03c5b1cSMartin Matuska }
931*c03c5b1cSMartin Matuska 
HUF_decompress4X2_DCtx_wksp(HUF_DTable * dctx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,void * workSpace,size_t wkspSize)932*c03c5b1cSMartin Matuska size_t HUF_decompress4X2_DCtx_wksp(HUF_DTable* dctx, void* dst, size_t dstSize,
933*c03c5b1cSMartin Matuska                                    const void* cSrc, size_t cSrcSize,
934*c03c5b1cSMartin Matuska                                    void* workSpace, size_t wkspSize)
935*c03c5b1cSMartin Matuska {
936*c03c5b1cSMartin Matuska     return HUF_decompress4X2_DCtx_wksp_bmi2(dctx, dst, dstSize, cSrc, cSrcSize, workSpace, wkspSize, /* bmi2 */ 0);
937*c03c5b1cSMartin Matuska }
938*c03c5b1cSMartin Matuska 
939*c03c5b1cSMartin Matuska 
HUF_decompress4X2_DCtx(HUF_DTable * dctx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize)940*c03c5b1cSMartin Matuska size_t HUF_decompress4X2_DCtx(HUF_DTable* dctx, void* dst, size_t dstSize,
941*c03c5b1cSMartin Matuska                               const void* cSrc, size_t cSrcSize)
942*c03c5b1cSMartin Matuska {
943*c03c5b1cSMartin Matuska     U32 workSpace[HUF_DECOMPRESS_WORKSPACE_SIZE_U32];
944*c03c5b1cSMartin Matuska     return HUF_decompress4X2_DCtx_wksp(dctx, dst, dstSize, cSrc, cSrcSize,
945*c03c5b1cSMartin Matuska                                        workSpace, sizeof(workSpace));
946*c03c5b1cSMartin Matuska }
947*c03c5b1cSMartin Matuska 
HUF_decompress4X2(void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize)948*c03c5b1cSMartin Matuska size_t HUF_decompress4X2 (void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize)
949*c03c5b1cSMartin Matuska {
950*c03c5b1cSMartin Matuska     HUF_CREATE_STATIC_DTABLEX2(DTable, HUF_TABLELOG_MAX);
951*c03c5b1cSMartin Matuska     return HUF_decompress4X2_DCtx(DTable, dst, dstSize, cSrc, cSrcSize);
952*c03c5b1cSMartin Matuska }
953*c03c5b1cSMartin Matuska 
954*c03c5b1cSMartin Matuska #endif /* HUF_FORCE_DECOMPRESS_X1 */
955*c03c5b1cSMartin Matuska 
956*c03c5b1cSMartin Matuska 
957*c03c5b1cSMartin Matuska /* ***********************************/
958*c03c5b1cSMartin Matuska /* Universal decompression selectors */
959*c03c5b1cSMartin Matuska /* ***********************************/
960*c03c5b1cSMartin Matuska 
HUF_decompress1X_usingDTable(void * dst,size_t maxDstSize,const void * cSrc,size_t cSrcSize,const HUF_DTable * DTable)961*c03c5b1cSMartin Matuska size_t HUF_decompress1X_usingDTable(void* dst, size_t maxDstSize,
962*c03c5b1cSMartin Matuska                                     const void* cSrc, size_t cSrcSize,
963*c03c5b1cSMartin Matuska                                     const HUF_DTable* DTable)
964*c03c5b1cSMartin Matuska {
965*c03c5b1cSMartin Matuska     DTableDesc const dtd = HUF_getDTableDesc(DTable);
966*c03c5b1cSMartin Matuska #if defined(HUF_FORCE_DECOMPRESS_X1)
967*c03c5b1cSMartin Matuska     (void)dtd;
968*c03c5b1cSMartin Matuska     assert(dtd.tableType == 0);
969*c03c5b1cSMartin Matuska     return HUF_decompress1X1_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, /* bmi2 */ 0);
970*c03c5b1cSMartin Matuska #elif defined(HUF_FORCE_DECOMPRESS_X2)
971*c03c5b1cSMartin Matuska     (void)dtd;
972*c03c5b1cSMartin Matuska     assert(dtd.tableType == 1);
973*c03c5b1cSMartin Matuska     return HUF_decompress1X2_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, /* bmi2 */ 0);
974*c03c5b1cSMartin Matuska #else
975*c03c5b1cSMartin Matuska     return dtd.tableType ? HUF_decompress1X2_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, /* bmi2 */ 0) :
976*c03c5b1cSMartin Matuska                            HUF_decompress1X1_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, /* bmi2 */ 0);
977*c03c5b1cSMartin Matuska #endif
978*c03c5b1cSMartin Matuska }
979*c03c5b1cSMartin Matuska 
HUF_decompress4X_usingDTable(void * dst,size_t maxDstSize,const void * cSrc,size_t cSrcSize,const HUF_DTable * DTable)980*c03c5b1cSMartin Matuska size_t HUF_decompress4X_usingDTable(void* dst, size_t maxDstSize,
981*c03c5b1cSMartin Matuska                                     const void* cSrc, size_t cSrcSize,
982*c03c5b1cSMartin Matuska                                     const HUF_DTable* DTable)
983*c03c5b1cSMartin Matuska {
984*c03c5b1cSMartin Matuska     DTableDesc const dtd = HUF_getDTableDesc(DTable);
985*c03c5b1cSMartin Matuska #if defined(HUF_FORCE_DECOMPRESS_X1)
986*c03c5b1cSMartin Matuska     (void)dtd;
987*c03c5b1cSMartin Matuska     assert(dtd.tableType == 0);
988*c03c5b1cSMartin Matuska     return HUF_decompress4X1_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, /* bmi2 */ 0);
989*c03c5b1cSMartin Matuska #elif defined(HUF_FORCE_DECOMPRESS_X2)
990*c03c5b1cSMartin Matuska     (void)dtd;
991*c03c5b1cSMartin Matuska     assert(dtd.tableType == 1);
992*c03c5b1cSMartin Matuska     return HUF_decompress4X2_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, /* bmi2 */ 0);
993*c03c5b1cSMartin Matuska #else
994*c03c5b1cSMartin Matuska     return dtd.tableType ? HUF_decompress4X2_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, /* bmi2 */ 0) :
995*c03c5b1cSMartin Matuska                            HUF_decompress4X1_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, /* bmi2 */ 0);
996*c03c5b1cSMartin Matuska #endif
997*c03c5b1cSMartin Matuska }
998*c03c5b1cSMartin Matuska 
999*c03c5b1cSMartin Matuska 
1000*c03c5b1cSMartin Matuska #if !defined(HUF_FORCE_DECOMPRESS_X1) && !defined(HUF_FORCE_DECOMPRESS_X2)
1001*c03c5b1cSMartin Matuska typedef struct { U32 tableTime; U32 decode256Time; } algo_time_t;
1002*c03c5b1cSMartin Matuska static const algo_time_t algoTime[16 /* Quantization */][3 /* single, double, quad */] =
1003*c03c5b1cSMartin Matuska {
1004*c03c5b1cSMartin Matuska     /* single, double, quad */
1005*c03c5b1cSMartin Matuska     {{0,0}, {1,1}, {2,2}},  /* Q==0 : impossible */
1006*c03c5b1cSMartin Matuska     {{0,0}, {1,1}, {2,2}},  /* Q==1 : impossible */
1007*c03c5b1cSMartin Matuska     {{  38,130}, {1313, 74}, {2151, 38}},   /* Q == 2 : 12-18% */
1008*c03c5b1cSMartin Matuska     {{ 448,128}, {1353, 74}, {2238, 41}},   /* Q == 3 : 18-25% */
1009*c03c5b1cSMartin Matuska     {{ 556,128}, {1353, 74}, {2238, 47}},   /* Q == 4 : 25-32% */
1010*c03c5b1cSMartin Matuska     {{ 714,128}, {1418, 74}, {2436, 53}},   /* Q == 5 : 32-38% */
1011*c03c5b1cSMartin Matuska     {{ 883,128}, {1437, 74}, {2464, 61}},   /* Q == 6 : 38-44% */
1012*c03c5b1cSMartin Matuska     {{ 897,128}, {1515, 75}, {2622, 68}},   /* Q == 7 : 44-50% */
1013*c03c5b1cSMartin Matuska     {{ 926,128}, {1613, 75}, {2730, 75}},   /* Q == 8 : 50-56% */
1014*c03c5b1cSMartin Matuska     {{ 947,128}, {1729, 77}, {3359, 77}},   /* Q == 9 : 56-62% */
1015*c03c5b1cSMartin Matuska     {{1107,128}, {2083, 81}, {4006, 84}},   /* Q ==10 : 62-69% */
1016*c03c5b1cSMartin Matuska     {{1177,128}, {2379, 87}, {4785, 88}},   /* Q ==11 : 69-75% */
1017*c03c5b1cSMartin Matuska     {{1242,128}, {2415, 93}, {5155, 84}},   /* Q ==12 : 75-81% */
1018*c03c5b1cSMartin Matuska     {{1349,128}, {2644,106}, {5260,106}},   /* Q ==13 : 81-87% */
1019*c03c5b1cSMartin Matuska     {{1455,128}, {2422,124}, {4174,124}},   /* Q ==14 : 87-93% */
1020*c03c5b1cSMartin Matuska     {{ 722,128}, {1891,145}, {1936,146}},   /* Q ==15 : 93-99% */
1021*c03c5b1cSMartin Matuska };
1022*c03c5b1cSMartin Matuska #endif
1023*c03c5b1cSMartin Matuska 
1024*c03c5b1cSMartin Matuska /** HUF_selectDecoder() :
1025*c03c5b1cSMartin Matuska  *  Tells which decoder is likely to decode faster,
1026*c03c5b1cSMartin Matuska  *  based on a set of pre-computed metrics.
1027*c03c5b1cSMartin Matuska  * @return : 0==HUF_decompress4X1, 1==HUF_decompress4X2 .
1028*c03c5b1cSMartin Matuska  *  Assumption : 0 < dstSize <= 128 KB */
HUF_selectDecoder(size_t dstSize,size_t cSrcSize)1029*c03c5b1cSMartin Matuska U32 HUF_selectDecoder (size_t dstSize, size_t cSrcSize)
1030*c03c5b1cSMartin Matuska {
1031*c03c5b1cSMartin Matuska     assert(dstSize > 0);
1032*c03c5b1cSMartin Matuska     assert(dstSize <= 128*1024);
1033*c03c5b1cSMartin Matuska #if defined(HUF_FORCE_DECOMPRESS_X1)
1034*c03c5b1cSMartin Matuska     (void)dstSize;
1035*c03c5b1cSMartin Matuska     (void)cSrcSize;
1036*c03c5b1cSMartin Matuska     return 0;
1037*c03c5b1cSMartin Matuska #elif defined(HUF_FORCE_DECOMPRESS_X2)
1038*c03c5b1cSMartin Matuska     (void)dstSize;
1039*c03c5b1cSMartin Matuska     (void)cSrcSize;
1040*c03c5b1cSMartin Matuska     return 1;
1041*c03c5b1cSMartin Matuska #else
1042*c03c5b1cSMartin Matuska     /* decoder timing evaluation */
1043*c03c5b1cSMartin Matuska     {   U32 const Q = (cSrcSize >= dstSize) ? 15 : (U32)(cSrcSize * 16 / dstSize);   /* Q < 16 */
1044*c03c5b1cSMartin Matuska         U32 const D256 = (U32)(dstSize >> 8);
1045*c03c5b1cSMartin Matuska         U32 const DTime0 = algoTime[Q][0].tableTime + (algoTime[Q][0].decode256Time * D256);
1046*c03c5b1cSMartin Matuska         U32 DTime1 = algoTime[Q][1].tableTime + (algoTime[Q][1].decode256Time * D256);
1047*c03c5b1cSMartin Matuska         DTime1 += DTime1 >> 3;  /* advantage to algorithm using less memory, to reduce cache eviction */
1048*c03c5b1cSMartin Matuska         return DTime1 < DTime0;
1049*c03c5b1cSMartin Matuska     }
1050*c03c5b1cSMartin Matuska #endif
1051*c03c5b1cSMartin Matuska }
1052*c03c5b1cSMartin Matuska 
1053*c03c5b1cSMartin Matuska 
1054*c03c5b1cSMartin Matuska typedef size_t (*decompressionAlgo)(void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize);
1055*c03c5b1cSMartin Matuska 
HUF_decompress(void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize)1056*c03c5b1cSMartin Matuska size_t HUF_decompress (void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize)
1057*c03c5b1cSMartin Matuska {
1058*c03c5b1cSMartin Matuska #if !defined(HUF_FORCE_DECOMPRESS_X1) && !defined(HUF_FORCE_DECOMPRESS_X2)
1059*c03c5b1cSMartin Matuska     static const decompressionAlgo decompress[2] = { HUF_decompress4X1, HUF_decompress4X2 };
1060*c03c5b1cSMartin Matuska #endif
1061*c03c5b1cSMartin Matuska 
1062*c03c5b1cSMartin Matuska     /* validation checks */
1063*c03c5b1cSMartin Matuska     if (dstSize == 0) return ERROR(dstSize_tooSmall);
1064*c03c5b1cSMartin Matuska     if (cSrcSize > dstSize) return ERROR(corruption_detected);   /* invalid */
1065*c03c5b1cSMartin Matuska     if (cSrcSize == dstSize) { memcpy(dst, cSrc, dstSize); return dstSize; }   /* not compressed */
1066*c03c5b1cSMartin Matuska     if (cSrcSize == 1) { memset(dst, *(const BYTE*)cSrc, dstSize); return dstSize; }   /* RLE */
1067*c03c5b1cSMartin Matuska 
1068*c03c5b1cSMartin Matuska     {   U32 const algoNb = HUF_selectDecoder(dstSize, cSrcSize);
1069*c03c5b1cSMartin Matuska #if defined(HUF_FORCE_DECOMPRESS_X1)
1070*c03c5b1cSMartin Matuska         (void)algoNb;
1071*c03c5b1cSMartin Matuska         assert(algoNb == 0);
1072*c03c5b1cSMartin Matuska         return HUF_decompress4X1(dst, dstSize, cSrc, cSrcSize);
1073*c03c5b1cSMartin Matuska #elif defined(HUF_FORCE_DECOMPRESS_X2)
1074*c03c5b1cSMartin Matuska         (void)algoNb;
1075*c03c5b1cSMartin Matuska         assert(algoNb == 1);
1076*c03c5b1cSMartin Matuska         return HUF_decompress4X2(dst, dstSize, cSrc, cSrcSize);
1077*c03c5b1cSMartin Matuska #else
1078*c03c5b1cSMartin Matuska         return decompress[algoNb](dst, dstSize, cSrc, cSrcSize);
1079*c03c5b1cSMartin Matuska #endif
1080*c03c5b1cSMartin Matuska     }
1081*c03c5b1cSMartin Matuska }
1082*c03c5b1cSMartin Matuska 
HUF_decompress4X_DCtx(HUF_DTable * dctx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize)1083*c03c5b1cSMartin Matuska size_t HUF_decompress4X_DCtx (HUF_DTable* dctx, void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize)
1084*c03c5b1cSMartin Matuska {
1085*c03c5b1cSMartin Matuska     /* validation checks */
1086*c03c5b1cSMartin Matuska     if (dstSize == 0) return ERROR(dstSize_tooSmall);
1087*c03c5b1cSMartin Matuska     if (cSrcSize > dstSize) return ERROR(corruption_detected);   /* invalid */
1088*c03c5b1cSMartin Matuska     if (cSrcSize == dstSize) { memcpy(dst, cSrc, dstSize); return dstSize; }   /* not compressed */
1089*c03c5b1cSMartin Matuska     if (cSrcSize == 1) { memset(dst, *(const BYTE*)cSrc, dstSize); return dstSize; }   /* RLE */
1090*c03c5b1cSMartin Matuska 
1091*c03c5b1cSMartin Matuska     {   U32 const algoNb = HUF_selectDecoder(dstSize, cSrcSize);
1092*c03c5b1cSMartin Matuska #if defined(HUF_FORCE_DECOMPRESS_X1)
1093*c03c5b1cSMartin Matuska         (void)algoNb;
1094*c03c5b1cSMartin Matuska         assert(algoNb == 0);
1095*c03c5b1cSMartin Matuska         return HUF_decompress4X1_DCtx(dctx, dst, dstSize, cSrc, cSrcSize);
1096*c03c5b1cSMartin Matuska #elif defined(HUF_FORCE_DECOMPRESS_X2)
1097*c03c5b1cSMartin Matuska         (void)algoNb;
1098*c03c5b1cSMartin Matuska         assert(algoNb == 1);
1099*c03c5b1cSMartin Matuska         return HUF_decompress4X2_DCtx(dctx, dst, dstSize, cSrc, cSrcSize);
1100*c03c5b1cSMartin Matuska #else
1101*c03c5b1cSMartin Matuska         return algoNb ? HUF_decompress4X2_DCtx(dctx, dst, dstSize, cSrc, cSrcSize) :
1102*c03c5b1cSMartin Matuska                         HUF_decompress4X1_DCtx(dctx, dst, dstSize, cSrc, cSrcSize) ;
1103*c03c5b1cSMartin Matuska #endif
1104*c03c5b1cSMartin Matuska     }
1105*c03c5b1cSMartin Matuska }
1106*c03c5b1cSMartin Matuska 
HUF_decompress4X_hufOnly(HUF_DTable * dctx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize)1107*c03c5b1cSMartin Matuska size_t HUF_decompress4X_hufOnly(HUF_DTable* dctx, void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize)
1108*c03c5b1cSMartin Matuska {
1109*c03c5b1cSMartin Matuska     U32 workSpace[HUF_DECOMPRESS_WORKSPACE_SIZE_U32];
1110*c03c5b1cSMartin Matuska     return HUF_decompress4X_hufOnly_wksp(dctx, dst, dstSize, cSrc, cSrcSize,
1111*c03c5b1cSMartin Matuska                                          workSpace, sizeof(workSpace));
1112*c03c5b1cSMartin Matuska }
1113*c03c5b1cSMartin Matuska 
1114*c03c5b1cSMartin Matuska 
HUF_decompress4X_hufOnly_wksp(HUF_DTable * dctx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,void * workSpace,size_t wkspSize)1115*c03c5b1cSMartin Matuska size_t HUF_decompress4X_hufOnly_wksp(HUF_DTable* dctx, void* dst,
1116*c03c5b1cSMartin Matuska                                      size_t dstSize, const void* cSrc,
1117*c03c5b1cSMartin Matuska                                      size_t cSrcSize, void* workSpace,
1118*c03c5b1cSMartin Matuska                                      size_t wkspSize)
1119*c03c5b1cSMartin Matuska {
1120*c03c5b1cSMartin Matuska     /* validation checks */
1121*c03c5b1cSMartin Matuska     if (dstSize == 0) return ERROR(dstSize_tooSmall);
1122*c03c5b1cSMartin Matuska     if (cSrcSize == 0) return ERROR(corruption_detected);
1123*c03c5b1cSMartin Matuska 
1124*c03c5b1cSMartin Matuska     {   U32 const algoNb = HUF_selectDecoder(dstSize, cSrcSize);
1125*c03c5b1cSMartin Matuska #if defined(HUF_FORCE_DECOMPRESS_X1)
1126*c03c5b1cSMartin Matuska         (void)algoNb;
1127*c03c5b1cSMartin Matuska         assert(algoNb == 0);
1128*c03c5b1cSMartin Matuska         return HUF_decompress4X1_DCtx_wksp(dctx, dst, dstSize, cSrc, cSrcSize, workSpace, wkspSize);
1129*c03c5b1cSMartin Matuska #elif defined(HUF_FORCE_DECOMPRESS_X2)
1130*c03c5b1cSMartin Matuska         (void)algoNb;
1131*c03c5b1cSMartin Matuska         assert(algoNb == 1);
1132*c03c5b1cSMartin Matuska         return HUF_decompress4X2_DCtx_wksp(dctx, dst, dstSize, cSrc, cSrcSize, workSpace, wkspSize);
1133*c03c5b1cSMartin Matuska #else
1134*c03c5b1cSMartin Matuska         return algoNb ? HUF_decompress4X2_DCtx_wksp(dctx, dst, dstSize, cSrc,
1135*c03c5b1cSMartin Matuska                             cSrcSize, workSpace, wkspSize):
1136*c03c5b1cSMartin Matuska                         HUF_decompress4X1_DCtx_wksp(dctx, dst, dstSize, cSrc, cSrcSize, workSpace, wkspSize);
1137*c03c5b1cSMartin Matuska #endif
1138*c03c5b1cSMartin Matuska     }
1139*c03c5b1cSMartin Matuska }
1140*c03c5b1cSMartin Matuska 
HUF_decompress1X_DCtx_wksp(HUF_DTable * dctx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,void * workSpace,size_t wkspSize)1141*c03c5b1cSMartin Matuska size_t HUF_decompress1X_DCtx_wksp(HUF_DTable* dctx, void* dst, size_t dstSize,
1142*c03c5b1cSMartin Matuska                                   const void* cSrc, size_t cSrcSize,
1143*c03c5b1cSMartin Matuska                                   void* workSpace, size_t wkspSize)
1144*c03c5b1cSMartin Matuska {
1145*c03c5b1cSMartin Matuska     /* validation checks */
1146*c03c5b1cSMartin Matuska     if (dstSize == 0) return ERROR(dstSize_tooSmall);
1147*c03c5b1cSMartin Matuska     if (cSrcSize > dstSize) return ERROR(corruption_detected);   /* invalid */
1148*c03c5b1cSMartin Matuska     if (cSrcSize == dstSize) { memcpy(dst, cSrc, dstSize); return dstSize; }   /* not compressed */
1149*c03c5b1cSMartin Matuska     if (cSrcSize == 1) { memset(dst, *(const BYTE*)cSrc, dstSize); return dstSize; }   /* RLE */
1150*c03c5b1cSMartin Matuska 
1151*c03c5b1cSMartin Matuska     {   U32 const algoNb = HUF_selectDecoder(dstSize, cSrcSize);
1152*c03c5b1cSMartin Matuska #if defined(HUF_FORCE_DECOMPRESS_X1)
1153*c03c5b1cSMartin Matuska         (void)algoNb;
1154*c03c5b1cSMartin Matuska         assert(algoNb == 0);
1155*c03c5b1cSMartin Matuska         return HUF_decompress1X1_DCtx_wksp(dctx, dst, dstSize, cSrc,
1156*c03c5b1cSMartin Matuska                                 cSrcSize, workSpace, wkspSize);
1157*c03c5b1cSMartin Matuska #elif defined(HUF_FORCE_DECOMPRESS_X2)
1158*c03c5b1cSMartin Matuska         (void)algoNb;
1159*c03c5b1cSMartin Matuska         assert(algoNb == 1);
1160*c03c5b1cSMartin Matuska         return HUF_decompress1X2_DCtx_wksp(dctx, dst, dstSize, cSrc,
1161*c03c5b1cSMartin Matuska                                 cSrcSize, workSpace, wkspSize);
1162*c03c5b1cSMartin Matuska #else
1163*c03c5b1cSMartin Matuska         return algoNb ? HUF_decompress1X2_DCtx_wksp(dctx, dst, dstSize, cSrc,
1164*c03c5b1cSMartin Matuska                                 cSrcSize, workSpace, wkspSize):
1165*c03c5b1cSMartin Matuska                         HUF_decompress1X1_DCtx_wksp(dctx, dst, dstSize, cSrc,
1166*c03c5b1cSMartin Matuska                                 cSrcSize, workSpace, wkspSize);
1167*c03c5b1cSMartin Matuska #endif
1168*c03c5b1cSMartin Matuska     }
1169*c03c5b1cSMartin Matuska }
1170*c03c5b1cSMartin Matuska 
HUF_decompress1X_DCtx(HUF_DTable * dctx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize)1171*c03c5b1cSMartin Matuska size_t HUF_decompress1X_DCtx(HUF_DTable* dctx, void* dst, size_t dstSize,
1172*c03c5b1cSMartin Matuska                              const void* cSrc, size_t cSrcSize)
1173*c03c5b1cSMartin Matuska {
1174*c03c5b1cSMartin Matuska     U32 workSpace[HUF_DECOMPRESS_WORKSPACE_SIZE_U32];
1175*c03c5b1cSMartin Matuska     return HUF_decompress1X_DCtx_wksp(dctx, dst, dstSize, cSrc, cSrcSize,
1176*c03c5b1cSMartin Matuska                                       workSpace, sizeof(workSpace));
1177*c03c5b1cSMartin Matuska }
1178*c03c5b1cSMartin Matuska 
1179*c03c5b1cSMartin Matuska 
HUF_decompress1X_usingDTable_bmi2(void * dst,size_t maxDstSize,const void * cSrc,size_t cSrcSize,const HUF_DTable * DTable,int bmi2)1180*c03c5b1cSMartin Matuska size_t HUF_decompress1X_usingDTable_bmi2(void* dst, size_t maxDstSize, const void* cSrc, size_t cSrcSize, const HUF_DTable* DTable, int bmi2)
1181*c03c5b1cSMartin Matuska {
1182*c03c5b1cSMartin Matuska     DTableDesc const dtd = HUF_getDTableDesc(DTable);
1183*c03c5b1cSMartin Matuska #if defined(HUF_FORCE_DECOMPRESS_X1)
1184*c03c5b1cSMartin Matuska     (void)dtd;
1185*c03c5b1cSMartin Matuska     assert(dtd.tableType == 0);
1186*c03c5b1cSMartin Matuska     return HUF_decompress1X1_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, bmi2);
1187*c03c5b1cSMartin Matuska #elif defined(HUF_FORCE_DECOMPRESS_X2)
1188*c03c5b1cSMartin Matuska     (void)dtd;
1189*c03c5b1cSMartin Matuska     assert(dtd.tableType == 1);
1190*c03c5b1cSMartin Matuska     return HUF_decompress1X2_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, bmi2);
1191*c03c5b1cSMartin Matuska #else
1192*c03c5b1cSMartin Matuska     return dtd.tableType ? HUF_decompress1X2_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, bmi2) :
1193*c03c5b1cSMartin Matuska                            HUF_decompress1X1_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, bmi2);
1194*c03c5b1cSMartin Matuska #endif
1195*c03c5b1cSMartin Matuska }
1196*c03c5b1cSMartin Matuska 
1197*c03c5b1cSMartin Matuska #ifndef HUF_FORCE_DECOMPRESS_X2
HUF_decompress1X1_DCtx_wksp_bmi2(HUF_DTable * dctx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,void * workSpace,size_t wkspSize,int bmi2)1198*c03c5b1cSMartin Matuska size_t HUF_decompress1X1_DCtx_wksp_bmi2(HUF_DTable* dctx, void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize, void* workSpace, size_t wkspSize, int bmi2)
1199*c03c5b1cSMartin Matuska {
1200*c03c5b1cSMartin Matuska     const BYTE* ip = (const BYTE*) cSrc;
1201*c03c5b1cSMartin Matuska 
1202*c03c5b1cSMartin Matuska     size_t const hSize = HUF_readDTableX1_wksp(dctx, cSrc, cSrcSize, workSpace, wkspSize);
1203*c03c5b1cSMartin Matuska     if (HUF_isError(hSize)) return hSize;
1204*c03c5b1cSMartin Matuska     if (hSize >= cSrcSize) return ERROR(srcSize_wrong);
1205*c03c5b1cSMartin Matuska     ip += hSize; cSrcSize -= hSize;
1206*c03c5b1cSMartin Matuska 
1207*c03c5b1cSMartin Matuska     return HUF_decompress1X1_usingDTable_internal(dst, dstSize, ip, cSrcSize, dctx, bmi2);
1208*c03c5b1cSMartin Matuska }
1209*c03c5b1cSMartin Matuska #endif
1210*c03c5b1cSMartin Matuska 
HUF_decompress4X_usingDTable_bmi2(void * dst,size_t maxDstSize,const void * cSrc,size_t cSrcSize,const HUF_DTable * DTable,int bmi2)1211*c03c5b1cSMartin Matuska size_t HUF_decompress4X_usingDTable_bmi2(void* dst, size_t maxDstSize, const void* cSrc, size_t cSrcSize, const HUF_DTable* DTable, int bmi2)
1212*c03c5b1cSMartin Matuska {
1213*c03c5b1cSMartin Matuska     DTableDesc const dtd = HUF_getDTableDesc(DTable);
1214*c03c5b1cSMartin Matuska #if defined(HUF_FORCE_DECOMPRESS_X1)
1215*c03c5b1cSMartin Matuska     (void)dtd;
1216*c03c5b1cSMartin Matuska     assert(dtd.tableType == 0);
1217*c03c5b1cSMartin Matuska     return HUF_decompress4X1_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, bmi2);
1218*c03c5b1cSMartin Matuska #elif defined(HUF_FORCE_DECOMPRESS_X2)
1219*c03c5b1cSMartin Matuska     (void)dtd;
1220*c03c5b1cSMartin Matuska     assert(dtd.tableType == 1);
1221*c03c5b1cSMartin Matuska     return HUF_decompress4X2_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, bmi2);
1222*c03c5b1cSMartin Matuska #else
1223*c03c5b1cSMartin Matuska     return dtd.tableType ? HUF_decompress4X2_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, bmi2) :
1224*c03c5b1cSMartin Matuska                            HUF_decompress4X1_usingDTable_internal(dst, maxDstSize, cSrc, cSrcSize, DTable, bmi2);
1225*c03c5b1cSMartin Matuska #endif
1226*c03c5b1cSMartin Matuska }
1227*c03c5b1cSMartin Matuska 
HUF_decompress4X_hufOnly_wksp_bmi2(HUF_DTable * dctx,void * dst,size_t dstSize,const void * cSrc,size_t cSrcSize,void * workSpace,size_t wkspSize,int bmi2)1228*c03c5b1cSMartin Matuska size_t HUF_decompress4X_hufOnly_wksp_bmi2(HUF_DTable* dctx, void* dst, size_t dstSize, const void* cSrc, size_t cSrcSize, void* workSpace, size_t wkspSize, int bmi2)
1229*c03c5b1cSMartin Matuska {
1230*c03c5b1cSMartin Matuska     /* validation checks */
1231*c03c5b1cSMartin Matuska     if (dstSize == 0) return ERROR(dstSize_tooSmall);
1232*c03c5b1cSMartin Matuska     if (cSrcSize == 0) return ERROR(corruption_detected);
1233*c03c5b1cSMartin Matuska 
1234*c03c5b1cSMartin Matuska     {   U32 const algoNb = HUF_selectDecoder(dstSize, cSrcSize);
1235*c03c5b1cSMartin Matuska #if defined(HUF_FORCE_DECOMPRESS_X1)
1236*c03c5b1cSMartin Matuska         (void)algoNb;
1237*c03c5b1cSMartin Matuska         assert(algoNb == 0);
1238*c03c5b1cSMartin Matuska         return HUF_decompress4X1_DCtx_wksp_bmi2(dctx, dst, dstSize, cSrc, cSrcSize, workSpace, wkspSize, bmi2);
1239*c03c5b1cSMartin Matuska #elif defined(HUF_FORCE_DECOMPRESS_X2)
1240*c03c5b1cSMartin Matuska         (void)algoNb;
1241*c03c5b1cSMartin Matuska         assert(algoNb == 1);
1242*c03c5b1cSMartin Matuska         return HUF_decompress4X2_DCtx_wksp_bmi2(dctx, dst, dstSize, cSrc, cSrcSize, workSpace, wkspSize, bmi2);
1243*c03c5b1cSMartin Matuska #else
1244*c03c5b1cSMartin Matuska         return algoNb ? HUF_decompress4X2_DCtx_wksp_bmi2(dctx, dst, dstSize, cSrc, cSrcSize, workSpace, wkspSize, bmi2) :
1245*c03c5b1cSMartin Matuska                         HUF_decompress4X1_DCtx_wksp_bmi2(dctx, dst, dstSize, cSrc, cSrcSize, workSpace, wkspSize, bmi2);
1246*c03c5b1cSMartin Matuska #endif
1247*c03c5b1cSMartin Matuska     }
1248*c03c5b1cSMartin Matuska }
1249