xref: /dflybsd-src/contrib/zstd/lib/decompress/zstd_decompress_block.c (revision a28cd43d19e8b720a6c852a4bbc5ae147a26165a)
1*a28cd43dSSascha Wildner /*
2*a28cd43dSSascha Wildner  * Copyright (c) 2016-2020, Yann Collet, Facebook, Inc.
3*a28cd43dSSascha Wildner  * All rights reserved.
4*a28cd43dSSascha Wildner  *
5*a28cd43dSSascha Wildner  * This source code is licensed under both the BSD-style license (found in the
6*a28cd43dSSascha Wildner  * LICENSE file in the root directory of this source tree) and the GPLv2 (found
7*a28cd43dSSascha Wildner  * in the COPYING file in the root directory of this source tree).
8*a28cd43dSSascha Wildner  * You may select, at your option, one of the above-listed licenses.
9*a28cd43dSSascha Wildner  */
10*a28cd43dSSascha Wildner 
11*a28cd43dSSascha Wildner /* zstd_decompress_block :
12*a28cd43dSSascha Wildner  * this module takes care of decompressing _compressed_ block */
13*a28cd43dSSascha Wildner 
14*a28cd43dSSascha Wildner /*-*******************************************************
15*a28cd43dSSascha Wildner *  Dependencies
16*a28cd43dSSascha Wildner *********************************************************/
17*a28cd43dSSascha Wildner #include "../common/zstd_deps.h"   /* ZSTD_memcpy, ZSTD_memmove, ZSTD_memset */
18*a28cd43dSSascha Wildner #include "../common/compiler.h"    /* prefetch */
19*a28cd43dSSascha Wildner #include "../common/cpu.h"         /* bmi2 */
20*a28cd43dSSascha Wildner #include "../common/mem.h"         /* low level memory routines */
21*a28cd43dSSascha Wildner #define FSE_STATIC_LINKING_ONLY
22*a28cd43dSSascha Wildner #include "../common/fse.h"
23*a28cd43dSSascha Wildner #define HUF_STATIC_LINKING_ONLY
24*a28cd43dSSascha Wildner #include "../common/huf.h"
25*a28cd43dSSascha Wildner #include "../common/zstd_internal.h"
26*a28cd43dSSascha Wildner #include "zstd_decompress_internal.h"   /* ZSTD_DCtx */
27*a28cd43dSSascha Wildner #include "zstd_ddict.h"  /* ZSTD_DDictDictContent */
28*a28cd43dSSascha Wildner #include "zstd_decompress_block.h"
29*a28cd43dSSascha Wildner 
30*a28cd43dSSascha Wildner /*_*******************************************************
31*a28cd43dSSascha Wildner *  Macros
32*a28cd43dSSascha Wildner **********************************************************/
33*a28cd43dSSascha Wildner 
34*a28cd43dSSascha Wildner /* These two optional macros force the use one way or another of the two
35*a28cd43dSSascha Wildner  * ZSTD_decompressSequences implementations. You can't force in both directions
36*a28cd43dSSascha Wildner  * at the same time.
37*a28cd43dSSascha Wildner  */
38*a28cd43dSSascha Wildner #if defined(ZSTD_FORCE_DECOMPRESS_SEQUENCES_SHORT) && \
39*a28cd43dSSascha Wildner     defined(ZSTD_FORCE_DECOMPRESS_SEQUENCES_LONG)
40*a28cd43dSSascha Wildner #error "Cannot force the use of the short and the long ZSTD_decompressSequences variants!"
41*a28cd43dSSascha Wildner #endif
42*a28cd43dSSascha Wildner 
43*a28cd43dSSascha Wildner 
44*a28cd43dSSascha Wildner /*_*******************************************************
45*a28cd43dSSascha Wildner *  Memory operations
46*a28cd43dSSascha Wildner **********************************************************/
ZSTD_copy4(void * dst,const void * src)47*a28cd43dSSascha Wildner static void ZSTD_copy4(void* dst, const void* src) { ZSTD_memcpy(dst, src, 4); }
48*a28cd43dSSascha Wildner 
49*a28cd43dSSascha Wildner 
50*a28cd43dSSascha Wildner /*-*************************************************************
51*a28cd43dSSascha Wildner  *   Block decoding
52*a28cd43dSSascha Wildner  ***************************************************************/
53*a28cd43dSSascha Wildner 
54*a28cd43dSSascha Wildner /*! ZSTD_getcBlockSize() :
55*a28cd43dSSascha Wildner  *  Provides the size of compressed block from block header `src` */
ZSTD_getcBlockSize(const void * src,size_t srcSize,blockProperties_t * bpPtr)56*a28cd43dSSascha Wildner size_t ZSTD_getcBlockSize(const void* src, size_t srcSize,
57*a28cd43dSSascha Wildner                           blockProperties_t* bpPtr)
58*a28cd43dSSascha Wildner {
59*a28cd43dSSascha Wildner     RETURN_ERROR_IF(srcSize < ZSTD_blockHeaderSize, srcSize_wrong, "");
60*a28cd43dSSascha Wildner 
61*a28cd43dSSascha Wildner     {   U32 const cBlockHeader = MEM_readLE24(src);
62*a28cd43dSSascha Wildner         U32 const cSize = cBlockHeader >> 3;
63*a28cd43dSSascha Wildner         bpPtr->lastBlock = cBlockHeader & 1;
64*a28cd43dSSascha Wildner         bpPtr->blockType = (blockType_e)((cBlockHeader >> 1) & 3);
65*a28cd43dSSascha Wildner         bpPtr->origSize = cSize;   /* only useful for RLE */
66*a28cd43dSSascha Wildner         if (bpPtr->blockType == bt_rle) return 1;
67*a28cd43dSSascha Wildner         RETURN_ERROR_IF(bpPtr->blockType == bt_reserved, corruption_detected, "");
68*a28cd43dSSascha Wildner         return cSize;
69*a28cd43dSSascha Wildner     }
70*a28cd43dSSascha Wildner }
71*a28cd43dSSascha Wildner 
72*a28cd43dSSascha Wildner 
73*a28cd43dSSascha Wildner /* Hidden declaration for fullbench */
74*a28cd43dSSascha Wildner size_t ZSTD_decodeLiteralsBlock(ZSTD_DCtx* dctx,
75*a28cd43dSSascha Wildner                           const void* src, size_t srcSize);
76*a28cd43dSSascha Wildner /*! ZSTD_decodeLiteralsBlock() :
77*a28cd43dSSascha Wildner  * @return : nb of bytes read from src (< srcSize )
78*a28cd43dSSascha Wildner  *  note : symbol not declared but exposed for fullbench */
ZSTD_decodeLiteralsBlock(ZSTD_DCtx * dctx,const void * src,size_t srcSize)79*a28cd43dSSascha Wildner size_t ZSTD_decodeLiteralsBlock(ZSTD_DCtx* dctx,
80*a28cd43dSSascha Wildner                           const void* src, size_t srcSize)   /* note : srcSize < BLOCKSIZE */
81*a28cd43dSSascha Wildner {
82*a28cd43dSSascha Wildner     DEBUGLOG(5, "ZSTD_decodeLiteralsBlock");
83*a28cd43dSSascha Wildner     RETURN_ERROR_IF(srcSize < MIN_CBLOCK_SIZE, corruption_detected, "");
84*a28cd43dSSascha Wildner 
85*a28cd43dSSascha Wildner     {   const BYTE* const istart = (const BYTE*) src;
86*a28cd43dSSascha Wildner         symbolEncodingType_e const litEncType = (symbolEncodingType_e)(istart[0] & 3);
87*a28cd43dSSascha Wildner 
88*a28cd43dSSascha Wildner         switch(litEncType)
89*a28cd43dSSascha Wildner         {
90*a28cd43dSSascha Wildner         case set_repeat:
91*a28cd43dSSascha Wildner             DEBUGLOG(5, "set_repeat flag : re-using stats from previous compressed literals block");
92*a28cd43dSSascha Wildner             RETURN_ERROR_IF(dctx->litEntropy==0, dictionary_corrupted, "");
93*a28cd43dSSascha Wildner             /* fall-through */
94*a28cd43dSSascha Wildner 
95*a28cd43dSSascha Wildner         case set_compressed:
96*a28cd43dSSascha Wildner             RETURN_ERROR_IF(srcSize < 5, corruption_detected, "srcSize >= MIN_CBLOCK_SIZE == 3; here we need up to 5 for case 3");
97*a28cd43dSSascha Wildner             {   size_t lhSize, litSize, litCSize;
98*a28cd43dSSascha Wildner                 U32 singleStream=0;
99*a28cd43dSSascha Wildner                 U32 const lhlCode = (istart[0] >> 2) & 3;
100*a28cd43dSSascha Wildner                 U32 const lhc = MEM_readLE32(istart);
101*a28cd43dSSascha Wildner                 size_t hufSuccess;
102*a28cd43dSSascha Wildner                 switch(lhlCode)
103*a28cd43dSSascha Wildner                 {
104*a28cd43dSSascha Wildner                 case 0: case 1: default:   /* note : default is impossible, since lhlCode into [0..3] */
105*a28cd43dSSascha Wildner                     /* 2 - 2 - 10 - 10 */
106*a28cd43dSSascha Wildner                     singleStream = !lhlCode;
107*a28cd43dSSascha Wildner                     lhSize = 3;
108*a28cd43dSSascha Wildner                     litSize  = (lhc >> 4) & 0x3FF;
109*a28cd43dSSascha Wildner                     litCSize = (lhc >> 14) & 0x3FF;
110*a28cd43dSSascha Wildner                     break;
111*a28cd43dSSascha Wildner                 case 2:
112*a28cd43dSSascha Wildner                     /* 2 - 2 - 14 - 14 */
113*a28cd43dSSascha Wildner                     lhSize = 4;
114*a28cd43dSSascha Wildner                     litSize  = (lhc >> 4) & 0x3FFF;
115*a28cd43dSSascha Wildner                     litCSize = lhc >> 18;
116*a28cd43dSSascha Wildner                     break;
117*a28cd43dSSascha Wildner                 case 3:
118*a28cd43dSSascha Wildner                     /* 2 - 2 - 18 - 18 */
119*a28cd43dSSascha Wildner                     lhSize = 5;
120*a28cd43dSSascha Wildner                     litSize  = (lhc >> 4) & 0x3FFFF;
121*a28cd43dSSascha Wildner                     litCSize = (lhc >> 22) + ((size_t)istart[4] << 10);
122*a28cd43dSSascha Wildner                     break;
123*a28cd43dSSascha Wildner                 }
124*a28cd43dSSascha Wildner                 RETURN_ERROR_IF(litSize > ZSTD_BLOCKSIZE_MAX, corruption_detected, "");
125*a28cd43dSSascha Wildner                 RETURN_ERROR_IF(litCSize + lhSize > srcSize, corruption_detected, "");
126*a28cd43dSSascha Wildner 
127*a28cd43dSSascha Wildner                 /* prefetch huffman table if cold */
128*a28cd43dSSascha Wildner                 if (dctx->ddictIsCold && (litSize > 768 /* heuristic */)) {
129*a28cd43dSSascha Wildner                     PREFETCH_AREA(dctx->HUFptr, sizeof(dctx->entropy.hufTable));
130*a28cd43dSSascha Wildner                 }
131*a28cd43dSSascha Wildner 
132*a28cd43dSSascha Wildner                 if (litEncType==set_repeat) {
133*a28cd43dSSascha Wildner                     if (singleStream) {
134*a28cd43dSSascha Wildner                         hufSuccess = HUF_decompress1X_usingDTable_bmi2(
135*a28cd43dSSascha Wildner                             dctx->litBuffer, litSize, istart+lhSize, litCSize,
136*a28cd43dSSascha Wildner                             dctx->HUFptr, dctx->bmi2);
137*a28cd43dSSascha Wildner                     } else {
138*a28cd43dSSascha Wildner                         hufSuccess = HUF_decompress4X_usingDTable_bmi2(
139*a28cd43dSSascha Wildner                             dctx->litBuffer, litSize, istart+lhSize, litCSize,
140*a28cd43dSSascha Wildner                             dctx->HUFptr, dctx->bmi2);
141*a28cd43dSSascha Wildner                     }
142*a28cd43dSSascha Wildner                 } else {
143*a28cd43dSSascha Wildner                     if (singleStream) {
144*a28cd43dSSascha Wildner #if defined(HUF_FORCE_DECOMPRESS_X2)
145*a28cd43dSSascha Wildner                         hufSuccess = HUF_decompress1X_DCtx_wksp(
146*a28cd43dSSascha Wildner                             dctx->entropy.hufTable, dctx->litBuffer, litSize,
147*a28cd43dSSascha Wildner                             istart+lhSize, litCSize, dctx->workspace,
148*a28cd43dSSascha Wildner                             sizeof(dctx->workspace));
149*a28cd43dSSascha Wildner #else
150*a28cd43dSSascha Wildner                         hufSuccess = HUF_decompress1X1_DCtx_wksp_bmi2(
151*a28cd43dSSascha Wildner                             dctx->entropy.hufTable, dctx->litBuffer, litSize,
152*a28cd43dSSascha Wildner                             istart+lhSize, litCSize, dctx->workspace,
153*a28cd43dSSascha Wildner                             sizeof(dctx->workspace), dctx->bmi2);
154*a28cd43dSSascha Wildner #endif
155*a28cd43dSSascha Wildner                     } else {
156*a28cd43dSSascha Wildner                         hufSuccess = HUF_decompress4X_hufOnly_wksp_bmi2(
157*a28cd43dSSascha Wildner                             dctx->entropy.hufTable, dctx->litBuffer, litSize,
158*a28cd43dSSascha Wildner                             istart+lhSize, litCSize, dctx->workspace,
159*a28cd43dSSascha Wildner                             sizeof(dctx->workspace), dctx->bmi2);
160*a28cd43dSSascha Wildner                     }
161*a28cd43dSSascha Wildner                 }
162*a28cd43dSSascha Wildner 
163*a28cd43dSSascha Wildner                 RETURN_ERROR_IF(HUF_isError(hufSuccess), corruption_detected, "");
164*a28cd43dSSascha Wildner 
165*a28cd43dSSascha Wildner                 dctx->litPtr = dctx->litBuffer;
166*a28cd43dSSascha Wildner                 dctx->litSize = litSize;
167*a28cd43dSSascha Wildner                 dctx->litEntropy = 1;
168*a28cd43dSSascha Wildner                 if (litEncType==set_compressed) dctx->HUFptr = dctx->entropy.hufTable;
169*a28cd43dSSascha Wildner                 ZSTD_memset(dctx->litBuffer + dctx->litSize, 0, WILDCOPY_OVERLENGTH);
170*a28cd43dSSascha Wildner                 return litCSize + lhSize;
171*a28cd43dSSascha Wildner             }
172*a28cd43dSSascha Wildner 
173*a28cd43dSSascha Wildner         case set_basic:
174*a28cd43dSSascha Wildner             {   size_t litSize, lhSize;
175*a28cd43dSSascha Wildner                 U32 const lhlCode = ((istart[0]) >> 2) & 3;
176*a28cd43dSSascha Wildner                 switch(lhlCode)
177*a28cd43dSSascha Wildner                 {
178*a28cd43dSSascha Wildner                 case 0: case 2: default:   /* note : default is impossible, since lhlCode into [0..3] */
179*a28cd43dSSascha Wildner                     lhSize = 1;
180*a28cd43dSSascha Wildner                     litSize = istart[0] >> 3;
181*a28cd43dSSascha Wildner                     break;
182*a28cd43dSSascha Wildner                 case 1:
183*a28cd43dSSascha Wildner                     lhSize = 2;
184*a28cd43dSSascha Wildner                     litSize = MEM_readLE16(istart) >> 4;
185*a28cd43dSSascha Wildner                     break;
186*a28cd43dSSascha Wildner                 case 3:
187*a28cd43dSSascha Wildner                     lhSize = 3;
188*a28cd43dSSascha Wildner                     litSize = MEM_readLE24(istart) >> 4;
189*a28cd43dSSascha Wildner                     break;
190*a28cd43dSSascha Wildner                 }
191*a28cd43dSSascha Wildner 
192*a28cd43dSSascha Wildner                 if (lhSize+litSize+WILDCOPY_OVERLENGTH > srcSize) {  /* risk reading beyond src buffer with wildcopy */
193*a28cd43dSSascha Wildner                     RETURN_ERROR_IF(litSize+lhSize > srcSize, corruption_detected, "");
194*a28cd43dSSascha Wildner                     ZSTD_memcpy(dctx->litBuffer, istart+lhSize, litSize);
195*a28cd43dSSascha Wildner                     dctx->litPtr = dctx->litBuffer;
196*a28cd43dSSascha Wildner                     dctx->litSize = litSize;
197*a28cd43dSSascha Wildner                     ZSTD_memset(dctx->litBuffer + dctx->litSize, 0, WILDCOPY_OVERLENGTH);
198*a28cd43dSSascha Wildner                     return lhSize+litSize;
199*a28cd43dSSascha Wildner                 }
200*a28cd43dSSascha Wildner                 /* direct reference into compressed stream */
201*a28cd43dSSascha Wildner                 dctx->litPtr = istart+lhSize;
202*a28cd43dSSascha Wildner                 dctx->litSize = litSize;
203*a28cd43dSSascha Wildner                 return lhSize+litSize;
204*a28cd43dSSascha Wildner             }
205*a28cd43dSSascha Wildner 
206*a28cd43dSSascha Wildner         case set_rle:
207*a28cd43dSSascha Wildner             {   U32 const lhlCode = ((istart[0]) >> 2) & 3;
208*a28cd43dSSascha Wildner                 size_t litSize, lhSize;
209*a28cd43dSSascha Wildner                 switch(lhlCode)
210*a28cd43dSSascha Wildner                 {
211*a28cd43dSSascha Wildner                 case 0: case 2: default:   /* note : default is impossible, since lhlCode into [0..3] */
212*a28cd43dSSascha Wildner                     lhSize = 1;
213*a28cd43dSSascha Wildner                     litSize = istart[0] >> 3;
214*a28cd43dSSascha Wildner                     break;
215*a28cd43dSSascha Wildner                 case 1:
216*a28cd43dSSascha Wildner                     lhSize = 2;
217*a28cd43dSSascha Wildner                     litSize = MEM_readLE16(istart) >> 4;
218*a28cd43dSSascha Wildner                     break;
219*a28cd43dSSascha Wildner                 case 3:
220*a28cd43dSSascha Wildner                     lhSize = 3;
221*a28cd43dSSascha Wildner                     litSize = MEM_readLE24(istart) >> 4;
222*a28cd43dSSascha Wildner                     RETURN_ERROR_IF(srcSize<4, corruption_detected, "srcSize >= MIN_CBLOCK_SIZE == 3; here we need lhSize+1 = 4");
223*a28cd43dSSascha Wildner                     break;
224*a28cd43dSSascha Wildner                 }
225*a28cd43dSSascha Wildner                 RETURN_ERROR_IF(litSize > ZSTD_BLOCKSIZE_MAX, corruption_detected, "");
226*a28cd43dSSascha Wildner                 ZSTD_memset(dctx->litBuffer, istart[lhSize], litSize + WILDCOPY_OVERLENGTH);
227*a28cd43dSSascha Wildner                 dctx->litPtr = dctx->litBuffer;
228*a28cd43dSSascha Wildner                 dctx->litSize = litSize;
229*a28cd43dSSascha Wildner                 return lhSize+1;
230*a28cd43dSSascha Wildner             }
231*a28cd43dSSascha Wildner         default:
232*a28cd43dSSascha Wildner             RETURN_ERROR(corruption_detected, "impossible");
233*a28cd43dSSascha Wildner         }
234*a28cd43dSSascha Wildner     }
235*a28cd43dSSascha Wildner }
236*a28cd43dSSascha Wildner 
237*a28cd43dSSascha Wildner /* Default FSE distribution tables.
238*a28cd43dSSascha Wildner  * These are pre-calculated FSE decoding tables using default distributions as defined in specification :
239*a28cd43dSSascha Wildner  * https://github.com/facebook/zstd/blob/release/doc/zstd_compression_format.md#default-distributions
240*a28cd43dSSascha Wildner  * They were generated programmatically with following method :
241*a28cd43dSSascha Wildner  * - start from default distributions, present in /lib/common/zstd_internal.h
242*a28cd43dSSascha Wildner  * - generate tables normally, using ZSTD_buildFSETable()
243*a28cd43dSSascha Wildner  * - printout the content of tables
244*a28cd43dSSascha Wildner  * - pretify output, report below, test with fuzzer to ensure it's correct */
245*a28cd43dSSascha Wildner 
246*a28cd43dSSascha Wildner /* Default FSE distribution table for Literal Lengths */
247*a28cd43dSSascha Wildner static const ZSTD_seqSymbol LL_defaultDTable[(1<<LL_DEFAULTNORMLOG)+1] = {
248*a28cd43dSSascha Wildner      {  1,  1,  1, LL_DEFAULTNORMLOG},  /* header : fastMode, tableLog */
249*a28cd43dSSascha Wildner      /* nextState, nbAddBits, nbBits, baseVal */
250*a28cd43dSSascha Wildner      {  0,  0,  4,    0},  { 16,  0,  4,    0},
251*a28cd43dSSascha Wildner      { 32,  0,  5,    1},  {  0,  0,  5,    3},
252*a28cd43dSSascha Wildner      {  0,  0,  5,    4},  {  0,  0,  5,    6},
253*a28cd43dSSascha Wildner      {  0,  0,  5,    7},  {  0,  0,  5,    9},
254*a28cd43dSSascha Wildner      {  0,  0,  5,   10},  {  0,  0,  5,   12},
255*a28cd43dSSascha Wildner      {  0,  0,  6,   14},  {  0,  1,  5,   16},
256*a28cd43dSSascha Wildner      {  0,  1,  5,   20},  {  0,  1,  5,   22},
257*a28cd43dSSascha Wildner      {  0,  2,  5,   28},  {  0,  3,  5,   32},
258*a28cd43dSSascha Wildner      {  0,  4,  5,   48},  { 32,  6,  5,   64},
259*a28cd43dSSascha Wildner      {  0,  7,  5,  128},  {  0,  8,  6,  256},
260*a28cd43dSSascha Wildner      {  0, 10,  6, 1024},  {  0, 12,  6, 4096},
261*a28cd43dSSascha Wildner      { 32,  0,  4,    0},  {  0,  0,  4,    1},
262*a28cd43dSSascha Wildner      {  0,  0,  5,    2},  { 32,  0,  5,    4},
263*a28cd43dSSascha Wildner      {  0,  0,  5,    5},  { 32,  0,  5,    7},
264*a28cd43dSSascha Wildner      {  0,  0,  5,    8},  { 32,  0,  5,   10},
265*a28cd43dSSascha Wildner      {  0,  0,  5,   11},  {  0,  0,  6,   13},
266*a28cd43dSSascha Wildner      { 32,  1,  5,   16},  {  0,  1,  5,   18},
267*a28cd43dSSascha Wildner      { 32,  1,  5,   22},  {  0,  2,  5,   24},
268*a28cd43dSSascha Wildner      { 32,  3,  5,   32},  {  0,  3,  5,   40},
269*a28cd43dSSascha Wildner      {  0,  6,  4,   64},  { 16,  6,  4,   64},
270*a28cd43dSSascha Wildner      { 32,  7,  5,  128},  {  0,  9,  6,  512},
271*a28cd43dSSascha Wildner      {  0, 11,  6, 2048},  { 48,  0,  4,    0},
272*a28cd43dSSascha Wildner      { 16,  0,  4,    1},  { 32,  0,  5,    2},
273*a28cd43dSSascha Wildner      { 32,  0,  5,    3},  { 32,  0,  5,    5},
274*a28cd43dSSascha Wildner      { 32,  0,  5,    6},  { 32,  0,  5,    8},
275*a28cd43dSSascha Wildner      { 32,  0,  5,    9},  { 32,  0,  5,   11},
276*a28cd43dSSascha Wildner      { 32,  0,  5,   12},  {  0,  0,  6,   15},
277*a28cd43dSSascha Wildner      { 32,  1,  5,   18},  { 32,  1,  5,   20},
278*a28cd43dSSascha Wildner      { 32,  2,  5,   24},  { 32,  2,  5,   28},
279*a28cd43dSSascha Wildner      { 32,  3,  5,   40},  { 32,  4,  5,   48},
280*a28cd43dSSascha Wildner      {  0, 16,  6,65536},  {  0, 15,  6,32768},
281*a28cd43dSSascha Wildner      {  0, 14,  6,16384},  {  0, 13,  6, 8192},
282*a28cd43dSSascha Wildner };   /* LL_defaultDTable */
283*a28cd43dSSascha Wildner 
284*a28cd43dSSascha Wildner /* Default FSE distribution table for Offset Codes */
285*a28cd43dSSascha Wildner static const ZSTD_seqSymbol OF_defaultDTable[(1<<OF_DEFAULTNORMLOG)+1] = {
286*a28cd43dSSascha Wildner     {  1,  1,  1, OF_DEFAULTNORMLOG},  /* header : fastMode, tableLog */
287*a28cd43dSSascha Wildner     /* nextState, nbAddBits, nbBits, baseVal */
288*a28cd43dSSascha Wildner     {  0,  0,  5,    0},     {  0,  6,  4,   61},
289*a28cd43dSSascha Wildner     {  0,  9,  5,  509},     {  0, 15,  5,32765},
290*a28cd43dSSascha Wildner     {  0, 21,  5,2097149},   {  0,  3,  5,    5},
291*a28cd43dSSascha Wildner     {  0,  7,  4,  125},     {  0, 12,  5, 4093},
292*a28cd43dSSascha Wildner     {  0, 18,  5,262141},    {  0, 23,  5,8388605},
293*a28cd43dSSascha Wildner     {  0,  5,  5,   29},     {  0,  8,  4,  253},
294*a28cd43dSSascha Wildner     {  0, 14,  5,16381},     {  0, 20,  5,1048573},
295*a28cd43dSSascha Wildner     {  0,  2,  5,    1},     { 16,  7,  4,  125},
296*a28cd43dSSascha Wildner     {  0, 11,  5, 2045},     {  0, 17,  5,131069},
297*a28cd43dSSascha Wildner     {  0, 22,  5,4194301},   {  0,  4,  5,   13},
298*a28cd43dSSascha Wildner     { 16,  8,  4,  253},     {  0, 13,  5, 8189},
299*a28cd43dSSascha Wildner     {  0, 19,  5,524285},    {  0,  1,  5,    1},
300*a28cd43dSSascha Wildner     { 16,  6,  4,   61},     {  0, 10,  5, 1021},
301*a28cd43dSSascha Wildner     {  0, 16,  5,65533},     {  0, 28,  5,268435453},
302*a28cd43dSSascha Wildner     {  0, 27,  5,134217725}, {  0, 26,  5,67108861},
303*a28cd43dSSascha Wildner     {  0, 25,  5,33554429},  {  0, 24,  5,16777213},
304*a28cd43dSSascha Wildner };   /* OF_defaultDTable */
305*a28cd43dSSascha Wildner 
306*a28cd43dSSascha Wildner 
307*a28cd43dSSascha Wildner /* Default FSE distribution table for Match Lengths */
308*a28cd43dSSascha Wildner static const ZSTD_seqSymbol ML_defaultDTable[(1<<ML_DEFAULTNORMLOG)+1] = {
309*a28cd43dSSascha Wildner     {  1,  1,  1, ML_DEFAULTNORMLOG},  /* header : fastMode, tableLog */
310*a28cd43dSSascha Wildner     /* nextState, nbAddBits, nbBits, baseVal */
311*a28cd43dSSascha Wildner     {  0,  0,  6,    3},  {  0,  0,  4,    4},
312*a28cd43dSSascha Wildner     { 32,  0,  5,    5},  {  0,  0,  5,    6},
313*a28cd43dSSascha Wildner     {  0,  0,  5,    8},  {  0,  0,  5,    9},
314*a28cd43dSSascha Wildner     {  0,  0,  5,   11},  {  0,  0,  6,   13},
315*a28cd43dSSascha Wildner     {  0,  0,  6,   16},  {  0,  0,  6,   19},
316*a28cd43dSSascha Wildner     {  0,  0,  6,   22},  {  0,  0,  6,   25},
317*a28cd43dSSascha Wildner     {  0,  0,  6,   28},  {  0,  0,  6,   31},
318*a28cd43dSSascha Wildner     {  0,  0,  6,   34},  {  0,  1,  6,   37},
319*a28cd43dSSascha Wildner     {  0,  1,  6,   41},  {  0,  2,  6,   47},
320*a28cd43dSSascha Wildner     {  0,  3,  6,   59},  {  0,  4,  6,   83},
321*a28cd43dSSascha Wildner     {  0,  7,  6,  131},  {  0,  9,  6,  515},
322*a28cd43dSSascha Wildner     { 16,  0,  4,    4},  {  0,  0,  4,    5},
323*a28cd43dSSascha Wildner     { 32,  0,  5,    6},  {  0,  0,  5,    7},
324*a28cd43dSSascha Wildner     { 32,  0,  5,    9},  {  0,  0,  5,   10},
325*a28cd43dSSascha Wildner     {  0,  0,  6,   12},  {  0,  0,  6,   15},
326*a28cd43dSSascha Wildner     {  0,  0,  6,   18},  {  0,  0,  6,   21},
327*a28cd43dSSascha Wildner     {  0,  0,  6,   24},  {  0,  0,  6,   27},
328*a28cd43dSSascha Wildner     {  0,  0,  6,   30},  {  0,  0,  6,   33},
329*a28cd43dSSascha Wildner     {  0,  1,  6,   35},  {  0,  1,  6,   39},
330*a28cd43dSSascha Wildner     {  0,  2,  6,   43},  {  0,  3,  6,   51},
331*a28cd43dSSascha Wildner     {  0,  4,  6,   67},  {  0,  5,  6,   99},
332*a28cd43dSSascha Wildner     {  0,  8,  6,  259},  { 32,  0,  4,    4},
333*a28cd43dSSascha Wildner     { 48,  0,  4,    4},  { 16,  0,  4,    5},
334*a28cd43dSSascha Wildner     { 32,  0,  5,    7},  { 32,  0,  5,    8},
335*a28cd43dSSascha Wildner     { 32,  0,  5,   10},  { 32,  0,  5,   11},
336*a28cd43dSSascha Wildner     {  0,  0,  6,   14},  {  0,  0,  6,   17},
337*a28cd43dSSascha Wildner     {  0,  0,  6,   20},  {  0,  0,  6,   23},
338*a28cd43dSSascha Wildner     {  0,  0,  6,   26},  {  0,  0,  6,   29},
339*a28cd43dSSascha Wildner     {  0,  0,  6,   32},  {  0, 16,  6,65539},
340*a28cd43dSSascha Wildner     {  0, 15,  6,32771},  {  0, 14,  6,16387},
341*a28cd43dSSascha Wildner     {  0, 13,  6, 8195},  {  0, 12,  6, 4099},
342*a28cd43dSSascha Wildner     {  0, 11,  6, 2051},  {  0, 10,  6, 1027},
343*a28cd43dSSascha Wildner };   /* ML_defaultDTable */
344*a28cd43dSSascha Wildner 
345*a28cd43dSSascha Wildner 
ZSTD_buildSeqTable_rle(ZSTD_seqSymbol * dt,U32 baseValue,U32 nbAddBits)346*a28cd43dSSascha Wildner static void ZSTD_buildSeqTable_rle(ZSTD_seqSymbol* dt, U32 baseValue, U32 nbAddBits)
347*a28cd43dSSascha Wildner {
348*a28cd43dSSascha Wildner     void* ptr = dt;
349*a28cd43dSSascha Wildner     ZSTD_seqSymbol_header* const DTableH = (ZSTD_seqSymbol_header*)ptr;
350*a28cd43dSSascha Wildner     ZSTD_seqSymbol* const cell = dt + 1;
351*a28cd43dSSascha Wildner 
352*a28cd43dSSascha Wildner     DTableH->tableLog = 0;
353*a28cd43dSSascha Wildner     DTableH->fastMode = 0;
354*a28cd43dSSascha Wildner 
355*a28cd43dSSascha Wildner     cell->nbBits = 0;
356*a28cd43dSSascha Wildner     cell->nextState = 0;
357*a28cd43dSSascha Wildner     assert(nbAddBits < 255);
358*a28cd43dSSascha Wildner     cell->nbAdditionalBits = (BYTE)nbAddBits;
359*a28cd43dSSascha Wildner     cell->baseValue = baseValue;
360*a28cd43dSSascha Wildner }
361*a28cd43dSSascha Wildner 
362*a28cd43dSSascha Wildner 
363*a28cd43dSSascha Wildner /* ZSTD_buildFSETable() :
364*a28cd43dSSascha Wildner  * generate FSE decoding table for one symbol (ll, ml or off)
365*a28cd43dSSascha Wildner  * cannot fail if input is valid =>
366*a28cd43dSSascha Wildner  * all inputs are presumed validated at this stage */
367*a28cd43dSSascha Wildner FORCE_INLINE_TEMPLATE
ZSTD_buildFSETable_body(ZSTD_seqSymbol * dt,const short * normalizedCounter,unsigned maxSymbolValue,const U32 * baseValue,const U32 * nbAdditionalBits,unsigned tableLog,void * wksp,size_t wkspSize)368*a28cd43dSSascha Wildner void ZSTD_buildFSETable_body(ZSTD_seqSymbol* dt,
369*a28cd43dSSascha Wildner             const short* normalizedCounter, unsigned maxSymbolValue,
370*a28cd43dSSascha Wildner             const U32* baseValue, const U32* nbAdditionalBits,
371*a28cd43dSSascha Wildner             unsigned tableLog, void* wksp, size_t wkspSize)
372*a28cd43dSSascha Wildner {
373*a28cd43dSSascha Wildner     ZSTD_seqSymbol* const tableDecode = dt+1;
374*a28cd43dSSascha Wildner     U32 const maxSV1 = maxSymbolValue + 1;
375*a28cd43dSSascha Wildner     U32 const tableSize = 1 << tableLog;
376*a28cd43dSSascha Wildner 
377*a28cd43dSSascha Wildner     U16* symbolNext = (U16*)wksp;
378*a28cd43dSSascha Wildner     BYTE* spread = (BYTE*)(symbolNext + MaxSeq + 1);
379*a28cd43dSSascha Wildner     U32 highThreshold = tableSize - 1;
380*a28cd43dSSascha Wildner 
381*a28cd43dSSascha Wildner 
382*a28cd43dSSascha Wildner     /* Sanity Checks */
383*a28cd43dSSascha Wildner     assert(maxSymbolValue <= MaxSeq);
384*a28cd43dSSascha Wildner     assert(tableLog <= MaxFSELog);
385*a28cd43dSSascha Wildner     assert(wkspSize >= ZSTD_BUILD_FSE_TABLE_WKSP_SIZE);
386*a28cd43dSSascha Wildner     (void)wkspSize;
387*a28cd43dSSascha Wildner     /* Init, lay down lowprob symbols */
388*a28cd43dSSascha Wildner     {   ZSTD_seqSymbol_header DTableH;
389*a28cd43dSSascha Wildner         DTableH.tableLog = tableLog;
390*a28cd43dSSascha Wildner         DTableH.fastMode = 1;
391*a28cd43dSSascha Wildner         {   S16 const largeLimit= (S16)(1 << (tableLog-1));
392*a28cd43dSSascha Wildner             U32 s;
393*a28cd43dSSascha Wildner             for (s=0; s<maxSV1; s++) {
394*a28cd43dSSascha Wildner                 if (normalizedCounter[s]==-1) {
395*a28cd43dSSascha Wildner                     tableDecode[highThreshold--].baseValue = s;
396*a28cd43dSSascha Wildner                     symbolNext[s] = 1;
397*a28cd43dSSascha Wildner                 } else {
398*a28cd43dSSascha Wildner                     if (normalizedCounter[s] >= largeLimit) DTableH.fastMode=0;
399*a28cd43dSSascha Wildner                     assert(normalizedCounter[s]>=0);
400*a28cd43dSSascha Wildner                     symbolNext[s] = (U16)normalizedCounter[s];
401*a28cd43dSSascha Wildner         }   }   }
402*a28cd43dSSascha Wildner         ZSTD_memcpy(dt, &DTableH, sizeof(DTableH));
403*a28cd43dSSascha Wildner     }
404*a28cd43dSSascha Wildner 
405*a28cd43dSSascha Wildner     /* Spread symbols */
406*a28cd43dSSascha Wildner     assert(tableSize <= 512);
407*a28cd43dSSascha Wildner     /* Specialized symbol spreading for the case when there are
408*a28cd43dSSascha Wildner      * no low probability (-1 count) symbols. When compressing
409*a28cd43dSSascha Wildner      * small blocks we avoid low probability symbols to hit this
410*a28cd43dSSascha Wildner      * case, since header decoding speed matters more.
411*a28cd43dSSascha Wildner      */
412*a28cd43dSSascha Wildner     if (highThreshold == tableSize - 1) {
413*a28cd43dSSascha Wildner         size_t const tableMask = tableSize-1;
414*a28cd43dSSascha Wildner         size_t const step = FSE_TABLESTEP(tableSize);
415*a28cd43dSSascha Wildner         /* First lay down the symbols in order.
416*a28cd43dSSascha Wildner          * We use a uint64_t to lay down 8 bytes at a time. This reduces branch
417*a28cd43dSSascha Wildner          * misses since small blocks generally have small table logs, so nearly
418*a28cd43dSSascha Wildner          * all symbols have counts <= 8. We ensure we have 8 bytes at the end of
419*a28cd43dSSascha Wildner          * our buffer to handle the over-write.
420*a28cd43dSSascha Wildner          */
421*a28cd43dSSascha Wildner         {
422*a28cd43dSSascha Wildner             U64 const add = 0x0101010101010101ull;
423*a28cd43dSSascha Wildner             size_t pos = 0;
424*a28cd43dSSascha Wildner             U64 sv = 0;
425*a28cd43dSSascha Wildner             U32 s;
426*a28cd43dSSascha Wildner             for (s=0; s<maxSV1; ++s, sv += add) {
427*a28cd43dSSascha Wildner                 int i;
428*a28cd43dSSascha Wildner                 int const n = normalizedCounter[s];
429*a28cd43dSSascha Wildner                 MEM_write64(spread + pos, sv);
430*a28cd43dSSascha Wildner                 for (i = 8; i < n; i += 8) {
431*a28cd43dSSascha Wildner                     MEM_write64(spread + pos + i, sv);
432*a28cd43dSSascha Wildner                 }
433*a28cd43dSSascha Wildner                 pos += n;
434*a28cd43dSSascha Wildner             }
435*a28cd43dSSascha Wildner         }
436*a28cd43dSSascha Wildner         /* Now we spread those positions across the table.
437*a28cd43dSSascha Wildner          * The benefit of doing it in two stages is that we avoid the the
438*a28cd43dSSascha Wildner          * variable size inner loop, which caused lots of branch misses.
439*a28cd43dSSascha Wildner          * Now we can run through all the positions without any branch misses.
440*a28cd43dSSascha Wildner          * We unroll the loop twice, since that is what emperically worked best.
441*a28cd43dSSascha Wildner          */
442*a28cd43dSSascha Wildner         {
443*a28cd43dSSascha Wildner             size_t position = 0;
444*a28cd43dSSascha Wildner             size_t s;
445*a28cd43dSSascha Wildner             size_t const unroll = 2;
446*a28cd43dSSascha Wildner             assert(tableSize % unroll == 0); /* FSE_MIN_TABLELOG is 5 */
447*a28cd43dSSascha Wildner             for (s = 0; s < (size_t)tableSize; s += unroll) {
448*a28cd43dSSascha Wildner                 size_t u;
449*a28cd43dSSascha Wildner                 for (u = 0; u < unroll; ++u) {
450*a28cd43dSSascha Wildner                     size_t const uPosition = (position + (u * step)) & tableMask;
451*a28cd43dSSascha Wildner                     tableDecode[uPosition].baseValue = spread[s + u];
452*a28cd43dSSascha Wildner                 }
453*a28cd43dSSascha Wildner                 position = (position + (unroll * step)) & tableMask;
454*a28cd43dSSascha Wildner             }
455*a28cd43dSSascha Wildner             assert(position == 0);
456*a28cd43dSSascha Wildner         }
457*a28cd43dSSascha Wildner     } else {
458*a28cd43dSSascha Wildner         U32 const tableMask = tableSize-1;
459*a28cd43dSSascha Wildner         U32 const step = FSE_TABLESTEP(tableSize);
460*a28cd43dSSascha Wildner         U32 s, position = 0;
461*a28cd43dSSascha Wildner         for (s=0; s<maxSV1; s++) {
462*a28cd43dSSascha Wildner             int i;
463*a28cd43dSSascha Wildner             int const n = normalizedCounter[s];
464*a28cd43dSSascha Wildner             for (i=0; i<n; i++) {
465*a28cd43dSSascha Wildner                 tableDecode[position].baseValue = s;
466*a28cd43dSSascha Wildner                 position = (position + step) & tableMask;
467*a28cd43dSSascha Wildner                 while (position > highThreshold) position = (position + step) & tableMask;   /* lowprob area */
468*a28cd43dSSascha Wildner         }   }
469*a28cd43dSSascha Wildner         assert(position == 0); /* position must reach all cells once, otherwise normalizedCounter is incorrect */
470*a28cd43dSSascha Wildner     }
471*a28cd43dSSascha Wildner 
472*a28cd43dSSascha Wildner     /* Build Decoding table */
473*a28cd43dSSascha Wildner     {
474*a28cd43dSSascha Wildner         U32 u;
475*a28cd43dSSascha Wildner         for (u=0; u<tableSize; u++) {
476*a28cd43dSSascha Wildner             U32 const symbol = tableDecode[u].baseValue;
477*a28cd43dSSascha Wildner             U32 const nextState = symbolNext[symbol]++;
478*a28cd43dSSascha Wildner             tableDecode[u].nbBits = (BYTE) (tableLog - BIT_highbit32(nextState) );
479*a28cd43dSSascha Wildner             tableDecode[u].nextState = (U16) ( (nextState << tableDecode[u].nbBits) - tableSize);
480*a28cd43dSSascha Wildner             assert(nbAdditionalBits[symbol] < 255);
481*a28cd43dSSascha Wildner             tableDecode[u].nbAdditionalBits = (BYTE)nbAdditionalBits[symbol];
482*a28cd43dSSascha Wildner             tableDecode[u].baseValue = baseValue[symbol];
483*a28cd43dSSascha Wildner         }
484*a28cd43dSSascha Wildner     }
485*a28cd43dSSascha Wildner }
486*a28cd43dSSascha Wildner 
487*a28cd43dSSascha Wildner /* Avoids the FORCE_INLINE of the _body() function. */
ZSTD_buildFSETable_body_default(ZSTD_seqSymbol * dt,const short * normalizedCounter,unsigned maxSymbolValue,const U32 * baseValue,const U32 * nbAdditionalBits,unsigned tableLog,void * wksp,size_t wkspSize)488*a28cd43dSSascha Wildner static void ZSTD_buildFSETable_body_default(ZSTD_seqSymbol* dt,
489*a28cd43dSSascha Wildner             const short* normalizedCounter, unsigned maxSymbolValue,
490*a28cd43dSSascha Wildner             const U32* baseValue, const U32* nbAdditionalBits,
491*a28cd43dSSascha Wildner             unsigned tableLog, void* wksp, size_t wkspSize)
492*a28cd43dSSascha Wildner {
493*a28cd43dSSascha Wildner     ZSTD_buildFSETable_body(dt, normalizedCounter, maxSymbolValue,
494*a28cd43dSSascha Wildner             baseValue, nbAdditionalBits, tableLog, wksp, wkspSize);
495*a28cd43dSSascha Wildner }
496*a28cd43dSSascha Wildner 
497*a28cd43dSSascha Wildner #if DYNAMIC_BMI2
ZSTD_buildFSETable_body_bmi2(ZSTD_seqSymbol * dt,const short * normalizedCounter,unsigned maxSymbolValue,const U32 * baseValue,const U32 * nbAdditionalBits,unsigned tableLog,void * wksp,size_t wkspSize)498*a28cd43dSSascha Wildner TARGET_ATTRIBUTE("bmi2") static void ZSTD_buildFSETable_body_bmi2(ZSTD_seqSymbol* dt,
499*a28cd43dSSascha Wildner             const short* normalizedCounter, unsigned maxSymbolValue,
500*a28cd43dSSascha Wildner             const U32* baseValue, const U32* nbAdditionalBits,
501*a28cd43dSSascha Wildner             unsigned tableLog, void* wksp, size_t wkspSize)
502*a28cd43dSSascha Wildner {
503*a28cd43dSSascha Wildner     ZSTD_buildFSETable_body(dt, normalizedCounter, maxSymbolValue,
504*a28cd43dSSascha Wildner             baseValue, nbAdditionalBits, tableLog, wksp, wkspSize);
505*a28cd43dSSascha Wildner }
506*a28cd43dSSascha Wildner #endif
507*a28cd43dSSascha Wildner 
ZSTD_buildFSETable(ZSTD_seqSymbol * dt,const short * normalizedCounter,unsigned maxSymbolValue,const U32 * baseValue,const U32 * nbAdditionalBits,unsigned tableLog,void * wksp,size_t wkspSize,int bmi2)508*a28cd43dSSascha Wildner void ZSTD_buildFSETable(ZSTD_seqSymbol* dt,
509*a28cd43dSSascha Wildner             const short* normalizedCounter, unsigned maxSymbolValue,
510*a28cd43dSSascha Wildner             const U32* baseValue, const U32* nbAdditionalBits,
511*a28cd43dSSascha Wildner             unsigned tableLog, void* wksp, size_t wkspSize, int bmi2)
512*a28cd43dSSascha Wildner {
513*a28cd43dSSascha Wildner #if DYNAMIC_BMI2
514*a28cd43dSSascha Wildner     if (bmi2) {
515*a28cd43dSSascha Wildner         ZSTD_buildFSETable_body_bmi2(dt, normalizedCounter, maxSymbolValue,
516*a28cd43dSSascha Wildner                 baseValue, nbAdditionalBits, tableLog, wksp, wkspSize);
517*a28cd43dSSascha Wildner         return;
518*a28cd43dSSascha Wildner     }
519*a28cd43dSSascha Wildner #endif
520*a28cd43dSSascha Wildner     (void)bmi2;
521*a28cd43dSSascha Wildner     ZSTD_buildFSETable_body_default(dt, normalizedCounter, maxSymbolValue,
522*a28cd43dSSascha Wildner             baseValue, nbAdditionalBits, tableLog, wksp, wkspSize);
523*a28cd43dSSascha Wildner }
524*a28cd43dSSascha Wildner 
525*a28cd43dSSascha Wildner 
526*a28cd43dSSascha Wildner /*! ZSTD_buildSeqTable() :
527*a28cd43dSSascha Wildner  * @return : nb bytes read from src,
528*a28cd43dSSascha Wildner  *           or an error code if it fails */
ZSTD_buildSeqTable(ZSTD_seqSymbol * DTableSpace,const ZSTD_seqSymbol ** DTablePtr,symbolEncodingType_e type,unsigned max,U32 maxLog,const void * src,size_t srcSize,const U32 * baseValue,const U32 * nbAdditionalBits,const ZSTD_seqSymbol * defaultTable,U32 flagRepeatTable,int ddictIsCold,int nbSeq,U32 * wksp,size_t wkspSize,int bmi2)529*a28cd43dSSascha Wildner static size_t ZSTD_buildSeqTable(ZSTD_seqSymbol* DTableSpace, const ZSTD_seqSymbol** DTablePtr,
530*a28cd43dSSascha Wildner                                  symbolEncodingType_e type, unsigned max, U32 maxLog,
531*a28cd43dSSascha Wildner                                  const void* src, size_t srcSize,
532*a28cd43dSSascha Wildner                                  const U32* baseValue, const U32* nbAdditionalBits,
533*a28cd43dSSascha Wildner                                  const ZSTD_seqSymbol* defaultTable, U32 flagRepeatTable,
534*a28cd43dSSascha Wildner                                  int ddictIsCold, int nbSeq, U32* wksp, size_t wkspSize,
535*a28cd43dSSascha Wildner                                  int bmi2)
536*a28cd43dSSascha Wildner {
537*a28cd43dSSascha Wildner     switch(type)
538*a28cd43dSSascha Wildner     {
539*a28cd43dSSascha Wildner     case set_rle :
540*a28cd43dSSascha Wildner         RETURN_ERROR_IF(!srcSize, srcSize_wrong, "");
541*a28cd43dSSascha Wildner         RETURN_ERROR_IF((*(const BYTE*)src) > max, corruption_detected, "");
542*a28cd43dSSascha Wildner         {   U32 const symbol = *(const BYTE*)src;
543*a28cd43dSSascha Wildner             U32 const baseline = baseValue[symbol];
544*a28cd43dSSascha Wildner             U32 const nbBits = nbAdditionalBits[symbol];
545*a28cd43dSSascha Wildner             ZSTD_buildSeqTable_rle(DTableSpace, baseline, nbBits);
546*a28cd43dSSascha Wildner         }
547*a28cd43dSSascha Wildner         *DTablePtr = DTableSpace;
548*a28cd43dSSascha Wildner         return 1;
549*a28cd43dSSascha Wildner     case set_basic :
550*a28cd43dSSascha Wildner         *DTablePtr = defaultTable;
551*a28cd43dSSascha Wildner         return 0;
552*a28cd43dSSascha Wildner     case set_repeat:
553*a28cd43dSSascha Wildner         RETURN_ERROR_IF(!flagRepeatTable, corruption_detected, "");
554*a28cd43dSSascha Wildner         /* prefetch FSE table if used */
555*a28cd43dSSascha Wildner         if (ddictIsCold && (nbSeq > 24 /* heuristic */)) {
556*a28cd43dSSascha Wildner             const void* const pStart = *DTablePtr;
557*a28cd43dSSascha Wildner             size_t const pSize = sizeof(ZSTD_seqSymbol) * (SEQSYMBOL_TABLE_SIZE(maxLog));
558*a28cd43dSSascha Wildner             PREFETCH_AREA(pStart, pSize);
559*a28cd43dSSascha Wildner         }
560*a28cd43dSSascha Wildner         return 0;
561*a28cd43dSSascha Wildner     case set_compressed :
562*a28cd43dSSascha Wildner         {   unsigned tableLog;
563*a28cd43dSSascha Wildner             S16 norm[MaxSeq+1];
564*a28cd43dSSascha Wildner             size_t const headerSize = FSE_readNCount(norm, &max, &tableLog, src, srcSize);
565*a28cd43dSSascha Wildner             RETURN_ERROR_IF(FSE_isError(headerSize), corruption_detected, "");
566*a28cd43dSSascha Wildner             RETURN_ERROR_IF(tableLog > maxLog, corruption_detected, "");
567*a28cd43dSSascha Wildner             ZSTD_buildFSETable(DTableSpace, norm, max, baseValue, nbAdditionalBits, tableLog, wksp, wkspSize, bmi2);
568*a28cd43dSSascha Wildner             *DTablePtr = DTableSpace;
569*a28cd43dSSascha Wildner             return headerSize;
570*a28cd43dSSascha Wildner         }
571*a28cd43dSSascha Wildner     default :
572*a28cd43dSSascha Wildner         assert(0);
573*a28cd43dSSascha Wildner         RETURN_ERROR(GENERIC, "impossible");
574*a28cd43dSSascha Wildner     }
575*a28cd43dSSascha Wildner }
576*a28cd43dSSascha Wildner 
ZSTD_decodeSeqHeaders(ZSTD_DCtx * dctx,int * nbSeqPtr,const void * src,size_t srcSize)577*a28cd43dSSascha Wildner size_t ZSTD_decodeSeqHeaders(ZSTD_DCtx* dctx, int* nbSeqPtr,
578*a28cd43dSSascha Wildner                              const void* src, size_t srcSize)
579*a28cd43dSSascha Wildner {
580*a28cd43dSSascha Wildner     const BYTE* const istart = (const BYTE* const)src;
581*a28cd43dSSascha Wildner     const BYTE* const iend = istart + srcSize;
582*a28cd43dSSascha Wildner     const BYTE* ip = istart;
583*a28cd43dSSascha Wildner     int nbSeq;
584*a28cd43dSSascha Wildner     DEBUGLOG(5, "ZSTD_decodeSeqHeaders");
585*a28cd43dSSascha Wildner 
586*a28cd43dSSascha Wildner     /* check */
587*a28cd43dSSascha Wildner     RETURN_ERROR_IF(srcSize < MIN_SEQUENCES_SIZE, srcSize_wrong, "");
588*a28cd43dSSascha Wildner 
589*a28cd43dSSascha Wildner     /* SeqHead */
590*a28cd43dSSascha Wildner     nbSeq = *ip++;
591*a28cd43dSSascha Wildner     if (!nbSeq) {
592*a28cd43dSSascha Wildner         *nbSeqPtr=0;
593*a28cd43dSSascha Wildner         RETURN_ERROR_IF(srcSize != 1, srcSize_wrong, "");
594*a28cd43dSSascha Wildner         return 1;
595*a28cd43dSSascha Wildner     }
596*a28cd43dSSascha Wildner     if (nbSeq > 0x7F) {
597*a28cd43dSSascha Wildner         if (nbSeq == 0xFF) {
598*a28cd43dSSascha Wildner             RETURN_ERROR_IF(ip+2 > iend, srcSize_wrong, "");
599*a28cd43dSSascha Wildner             nbSeq = MEM_readLE16(ip) + LONGNBSEQ;
600*a28cd43dSSascha Wildner             ip+=2;
601*a28cd43dSSascha Wildner         } else {
602*a28cd43dSSascha Wildner             RETURN_ERROR_IF(ip >= iend, srcSize_wrong, "");
603*a28cd43dSSascha Wildner             nbSeq = ((nbSeq-0x80)<<8) + *ip++;
604*a28cd43dSSascha Wildner         }
605*a28cd43dSSascha Wildner     }
606*a28cd43dSSascha Wildner     *nbSeqPtr = nbSeq;
607*a28cd43dSSascha Wildner 
608*a28cd43dSSascha Wildner     /* FSE table descriptors */
609*a28cd43dSSascha Wildner     RETURN_ERROR_IF(ip+1 > iend, srcSize_wrong, ""); /* minimum possible size: 1 byte for symbol encoding types */
610*a28cd43dSSascha Wildner     {   symbolEncodingType_e const LLtype = (symbolEncodingType_e)(*ip >> 6);
611*a28cd43dSSascha Wildner         symbolEncodingType_e const OFtype = (symbolEncodingType_e)((*ip >> 4) & 3);
612*a28cd43dSSascha Wildner         symbolEncodingType_e const MLtype = (symbolEncodingType_e)((*ip >> 2) & 3);
613*a28cd43dSSascha Wildner         ip++;
614*a28cd43dSSascha Wildner 
615*a28cd43dSSascha Wildner         /* Build DTables */
616*a28cd43dSSascha Wildner         {   size_t const llhSize = ZSTD_buildSeqTable(dctx->entropy.LLTable, &dctx->LLTptr,
617*a28cd43dSSascha Wildner                                                       LLtype, MaxLL, LLFSELog,
618*a28cd43dSSascha Wildner                                                       ip, iend-ip,
619*a28cd43dSSascha Wildner                                                       LL_base, LL_bits,
620*a28cd43dSSascha Wildner                                                       LL_defaultDTable, dctx->fseEntropy,
621*a28cd43dSSascha Wildner                                                       dctx->ddictIsCold, nbSeq,
622*a28cd43dSSascha Wildner                                                       dctx->workspace, sizeof(dctx->workspace),
623*a28cd43dSSascha Wildner                                                       dctx->bmi2);
624*a28cd43dSSascha Wildner             RETURN_ERROR_IF(ZSTD_isError(llhSize), corruption_detected, "ZSTD_buildSeqTable failed");
625*a28cd43dSSascha Wildner             ip += llhSize;
626*a28cd43dSSascha Wildner         }
627*a28cd43dSSascha Wildner 
628*a28cd43dSSascha Wildner         {   size_t const ofhSize = ZSTD_buildSeqTable(dctx->entropy.OFTable, &dctx->OFTptr,
629*a28cd43dSSascha Wildner                                                       OFtype, MaxOff, OffFSELog,
630*a28cd43dSSascha Wildner                                                       ip, iend-ip,
631*a28cd43dSSascha Wildner                                                       OF_base, OF_bits,
632*a28cd43dSSascha Wildner                                                       OF_defaultDTable, dctx->fseEntropy,
633*a28cd43dSSascha Wildner                                                       dctx->ddictIsCold, nbSeq,
634*a28cd43dSSascha Wildner                                                       dctx->workspace, sizeof(dctx->workspace),
635*a28cd43dSSascha Wildner                                                       dctx->bmi2);
636*a28cd43dSSascha Wildner             RETURN_ERROR_IF(ZSTD_isError(ofhSize), corruption_detected, "ZSTD_buildSeqTable failed");
637*a28cd43dSSascha Wildner             ip += ofhSize;
638*a28cd43dSSascha Wildner         }
639*a28cd43dSSascha Wildner 
640*a28cd43dSSascha Wildner         {   size_t const mlhSize = ZSTD_buildSeqTable(dctx->entropy.MLTable, &dctx->MLTptr,
641*a28cd43dSSascha Wildner                                                       MLtype, MaxML, MLFSELog,
642*a28cd43dSSascha Wildner                                                       ip, iend-ip,
643*a28cd43dSSascha Wildner                                                       ML_base, ML_bits,
644*a28cd43dSSascha Wildner                                                       ML_defaultDTable, dctx->fseEntropy,
645*a28cd43dSSascha Wildner                                                       dctx->ddictIsCold, nbSeq,
646*a28cd43dSSascha Wildner                                                       dctx->workspace, sizeof(dctx->workspace),
647*a28cd43dSSascha Wildner                                                       dctx->bmi2);
648*a28cd43dSSascha Wildner             RETURN_ERROR_IF(ZSTD_isError(mlhSize), corruption_detected, "ZSTD_buildSeqTable failed");
649*a28cd43dSSascha Wildner             ip += mlhSize;
650*a28cd43dSSascha Wildner         }
651*a28cd43dSSascha Wildner     }
652*a28cd43dSSascha Wildner 
653*a28cd43dSSascha Wildner     return ip-istart;
654*a28cd43dSSascha Wildner }
655*a28cd43dSSascha Wildner 
656*a28cd43dSSascha Wildner 
657*a28cd43dSSascha Wildner typedef struct {
658*a28cd43dSSascha Wildner     size_t litLength;
659*a28cd43dSSascha Wildner     size_t matchLength;
660*a28cd43dSSascha Wildner     size_t offset;
661*a28cd43dSSascha Wildner     const BYTE* match;
662*a28cd43dSSascha Wildner } seq_t;
663*a28cd43dSSascha Wildner 
664*a28cd43dSSascha Wildner typedef struct {
665*a28cd43dSSascha Wildner     size_t state;
666*a28cd43dSSascha Wildner     const ZSTD_seqSymbol* table;
667*a28cd43dSSascha Wildner } ZSTD_fseState;
668*a28cd43dSSascha Wildner 
669*a28cd43dSSascha Wildner typedef struct {
670*a28cd43dSSascha Wildner     BIT_DStream_t DStream;
671*a28cd43dSSascha Wildner     ZSTD_fseState stateLL;
672*a28cd43dSSascha Wildner     ZSTD_fseState stateOffb;
673*a28cd43dSSascha Wildner     ZSTD_fseState stateML;
674*a28cd43dSSascha Wildner     size_t prevOffset[ZSTD_REP_NUM];
675*a28cd43dSSascha Wildner     const BYTE* prefixStart;
676*a28cd43dSSascha Wildner     const BYTE* dictEnd;
677*a28cd43dSSascha Wildner     size_t pos;
678*a28cd43dSSascha Wildner } seqState_t;
679*a28cd43dSSascha Wildner 
680*a28cd43dSSascha Wildner /*! ZSTD_overlapCopy8() :
681*a28cd43dSSascha Wildner  *  Copies 8 bytes from ip to op and updates op and ip where ip <= op.
682*a28cd43dSSascha Wildner  *  If the offset is < 8 then the offset is spread to at least 8 bytes.
683*a28cd43dSSascha Wildner  *
684*a28cd43dSSascha Wildner  *  Precondition: *ip <= *op
685*a28cd43dSSascha Wildner  *  Postcondition: *op - *op >= 8
686*a28cd43dSSascha Wildner  */
ZSTD_overlapCopy8(BYTE ** op,BYTE const ** ip,size_t offset)687*a28cd43dSSascha Wildner HINT_INLINE void ZSTD_overlapCopy8(BYTE** op, BYTE const** ip, size_t offset) {
688*a28cd43dSSascha Wildner     assert(*ip <= *op);
689*a28cd43dSSascha Wildner     if (offset < 8) {
690*a28cd43dSSascha Wildner         /* close range match, overlap */
691*a28cd43dSSascha Wildner         static const U32 dec32table[] = { 0, 1, 2, 1, 4, 4, 4, 4 };   /* added */
692*a28cd43dSSascha Wildner         static const int dec64table[] = { 8, 8, 8, 7, 8, 9,10,11 };   /* subtracted */
693*a28cd43dSSascha Wildner         int const sub2 = dec64table[offset];
694*a28cd43dSSascha Wildner         (*op)[0] = (*ip)[0];
695*a28cd43dSSascha Wildner         (*op)[1] = (*ip)[1];
696*a28cd43dSSascha Wildner         (*op)[2] = (*ip)[2];
697*a28cd43dSSascha Wildner         (*op)[3] = (*ip)[3];
698*a28cd43dSSascha Wildner         *ip += dec32table[offset];
699*a28cd43dSSascha Wildner         ZSTD_copy4(*op+4, *ip);
700*a28cd43dSSascha Wildner         *ip -= sub2;
701*a28cd43dSSascha Wildner     } else {
702*a28cd43dSSascha Wildner         ZSTD_copy8(*op, *ip);
703*a28cd43dSSascha Wildner     }
704*a28cd43dSSascha Wildner     *ip += 8;
705*a28cd43dSSascha Wildner     *op += 8;
706*a28cd43dSSascha Wildner     assert(*op - *ip >= 8);
707*a28cd43dSSascha Wildner }
708*a28cd43dSSascha Wildner 
709*a28cd43dSSascha Wildner /*! ZSTD_safecopy() :
710*a28cd43dSSascha Wildner  *  Specialized version of memcpy() that is allowed to READ up to WILDCOPY_OVERLENGTH past the input buffer
711*a28cd43dSSascha Wildner  *  and write up to 16 bytes past oend_w (op >= oend_w is allowed).
712*a28cd43dSSascha Wildner  *  This function is only called in the uncommon case where the sequence is near the end of the block. It
713*a28cd43dSSascha Wildner  *  should be fast for a single long sequence, but can be slow for several short sequences.
714*a28cd43dSSascha Wildner  *
715*a28cd43dSSascha Wildner  *  @param ovtype controls the overlap detection
716*a28cd43dSSascha Wildner  *         - ZSTD_no_overlap: The source and destination are guaranteed to be at least WILDCOPY_VECLEN bytes apart.
717*a28cd43dSSascha Wildner  *         - ZSTD_overlap_src_before_dst: The src and dst may overlap and may be any distance apart.
718*a28cd43dSSascha Wildner  *           The src buffer must be before the dst buffer.
719*a28cd43dSSascha Wildner  */
ZSTD_safecopy(BYTE * op,BYTE * const oend_w,BYTE const * ip,ptrdiff_t length,ZSTD_overlap_e ovtype)720*a28cd43dSSascha Wildner static void ZSTD_safecopy(BYTE* op, BYTE* const oend_w, BYTE const* ip, ptrdiff_t length, ZSTD_overlap_e ovtype) {
721*a28cd43dSSascha Wildner     ptrdiff_t const diff = op - ip;
722*a28cd43dSSascha Wildner     BYTE* const oend = op + length;
723*a28cd43dSSascha Wildner 
724*a28cd43dSSascha Wildner     assert((ovtype == ZSTD_no_overlap && (diff <= -8 || diff >= 8 || op >= oend_w)) ||
725*a28cd43dSSascha Wildner            (ovtype == ZSTD_overlap_src_before_dst && diff >= 0));
726*a28cd43dSSascha Wildner 
727*a28cd43dSSascha Wildner     if (length < 8) {
728*a28cd43dSSascha Wildner         /* Handle short lengths. */
729*a28cd43dSSascha Wildner         while (op < oend) *op++ = *ip++;
730*a28cd43dSSascha Wildner         return;
731*a28cd43dSSascha Wildner     }
732*a28cd43dSSascha Wildner     if (ovtype == ZSTD_overlap_src_before_dst) {
733*a28cd43dSSascha Wildner         /* Copy 8 bytes and ensure the offset >= 8 when there can be overlap. */
734*a28cd43dSSascha Wildner         assert(length >= 8);
735*a28cd43dSSascha Wildner         ZSTD_overlapCopy8(&op, &ip, diff);
736*a28cd43dSSascha Wildner         assert(op - ip >= 8);
737*a28cd43dSSascha Wildner         assert(op <= oend);
738*a28cd43dSSascha Wildner     }
739*a28cd43dSSascha Wildner 
740*a28cd43dSSascha Wildner     if (oend <= oend_w) {
741*a28cd43dSSascha Wildner         /* No risk of overwrite. */
742*a28cd43dSSascha Wildner         ZSTD_wildcopy(op, ip, length, ovtype);
743*a28cd43dSSascha Wildner         return;
744*a28cd43dSSascha Wildner     }
745*a28cd43dSSascha Wildner     if (op <= oend_w) {
746*a28cd43dSSascha Wildner         /* Wildcopy until we get close to the end. */
747*a28cd43dSSascha Wildner         assert(oend > oend_w);
748*a28cd43dSSascha Wildner         ZSTD_wildcopy(op, ip, oend_w - op, ovtype);
749*a28cd43dSSascha Wildner         ip += oend_w - op;
750*a28cd43dSSascha Wildner         op = oend_w;
751*a28cd43dSSascha Wildner     }
752*a28cd43dSSascha Wildner     /* Handle the leftovers. */
753*a28cd43dSSascha Wildner     while (op < oend) *op++ = *ip++;
754*a28cd43dSSascha Wildner }
755*a28cd43dSSascha Wildner 
756*a28cd43dSSascha Wildner /* ZSTD_execSequenceEnd():
757*a28cd43dSSascha Wildner  * This version handles cases that are near the end of the output buffer. It requires
758*a28cd43dSSascha Wildner  * more careful checks to make sure there is no overflow. By separating out these hard
759*a28cd43dSSascha Wildner  * and unlikely cases, we can speed up the common cases.
760*a28cd43dSSascha Wildner  *
761*a28cd43dSSascha Wildner  * NOTE: This function needs to be fast for a single long sequence, but doesn't need
762*a28cd43dSSascha Wildner  * to be optimized for many small sequences, since those fall into ZSTD_execSequence().
763*a28cd43dSSascha Wildner  */
764*a28cd43dSSascha Wildner FORCE_NOINLINE
ZSTD_execSequenceEnd(BYTE * op,BYTE * const oend,seq_t sequence,const BYTE ** litPtr,const BYTE * const litLimit,const BYTE * const prefixStart,const BYTE * const virtualStart,const BYTE * const dictEnd)765*a28cd43dSSascha Wildner size_t ZSTD_execSequenceEnd(BYTE* op,
766*a28cd43dSSascha Wildner                             BYTE* const oend, seq_t sequence,
767*a28cd43dSSascha Wildner                             const BYTE** litPtr, const BYTE* const litLimit,
768*a28cd43dSSascha Wildner                             const BYTE* const prefixStart, const BYTE* const virtualStart, const BYTE* const dictEnd)
769*a28cd43dSSascha Wildner {
770*a28cd43dSSascha Wildner     BYTE* const oLitEnd = op + sequence.litLength;
771*a28cd43dSSascha Wildner     size_t const sequenceLength = sequence.litLength + sequence.matchLength;
772*a28cd43dSSascha Wildner     const BYTE* const iLitEnd = *litPtr + sequence.litLength;
773*a28cd43dSSascha Wildner     const BYTE* match = oLitEnd - sequence.offset;
774*a28cd43dSSascha Wildner     BYTE* const oend_w = oend - WILDCOPY_OVERLENGTH;
775*a28cd43dSSascha Wildner 
776*a28cd43dSSascha Wildner     /* bounds checks : careful of address space overflow in 32-bit mode */
777*a28cd43dSSascha Wildner     RETURN_ERROR_IF(sequenceLength > (size_t)(oend - op), dstSize_tooSmall, "last match must fit within dstBuffer");
778*a28cd43dSSascha Wildner     RETURN_ERROR_IF(sequence.litLength > (size_t)(litLimit - *litPtr), corruption_detected, "try to read beyond literal buffer");
779*a28cd43dSSascha Wildner     assert(op < op + sequenceLength);
780*a28cd43dSSascha Wildner     assert(oLitEnd < op + sequenceLength);
781*a28cd43dSSascha Wildner 
782*a28cd43dSSascha Wildner     /* copy literals */
783*a28cd43dSSascha Wildner     ZSTD_safecopy(op, oend_w, *litPtr, sequence.litLength, ZSTD_no_overlap);
784*a28cd43dSSascha Wildner     op = oLitEnd;
785*a28cd43dSSascha Wildner     *litPtr = iLitEnd;
786*a28cd43dSSascha Wildner 
787*a28cd43dSSascha Wildner     /* copy Match */
788*a28cd43dSSascha Wildner     if (sequence.offset > (size_t)(oLitEnd - prefixStart)) {
789*a28cd43dSSascha Wildner         /* offset beyond prefix */
790*a28cd43dSSascha Wildner         RETURN_ERROR_IF(sequence.offset > (size_t)(oLitEnd - virtualStart), corruption_detected, "");
791*a28cd43dSSascha Wildner         match = dictEnd - (prefixStart-match);
792*a28cd43dSSascha Wildner         if (match + sequence.matchLength <= dictEnd) {
793*a28cd43dSSascha Wildner             ZSTD_memmove(oLitEnd, match, sequence.matchLength);
794*a28cd43dSSascha Wildner             return sequenceLength;
795*a28cd43dSSascha Wildner         }
796*a28cd43dSSascha Wildner         /* span extDict & currentPrefixSegment */
797*a28cd43dSSascha Wildner         {   size_t const length1 = dictEnd - match;
798*a28cd43dSSascha Wildner             ZSTD_memmove(oLitEnd, match, length1);
799*a28cd43dSSascha Wildner             op = oLitEnd + length1;
800*a28cd43dSSascha Wildner             sequence.matchLength -= length1;
801*a28cd43dSSascha Wildner             match = prefixStart;
802*a28cd43dSSascha Wildner     }   }
803*a28cd43dSSascha Wildner     ZSTD_safecopy(op, oend_w, match, sequence.matchLength, ZSTD_overlap_src_before_dst);
804*a28cd43dSSascha Wildner     return sequenceLength;
805*a28cd43dSSascha Wildner }
806*a28cd43dSSascha Wildner 
807*a28cd43dSSascha Wildner HINT_INLINE
ZSTD_execSequence(BYTE * op,BYTE * const oend,seq_t sequence,const BYTE ** litPtr,const BYTE * const litLimit,const BYTE * const prefixStart,const BYTE * const virtualStart,const BYTE * const dictEnd)808*a28cd43dSSascha Wildner size_t ZSTD_execSequence(BYTE* op,
809*a28cd43dSSascha Wildner                          BYTE* const oend, seq_t sequence,
810*a28cd43dSSascha Wildner                          const BYTE** litPtr, const BYTE* const litLimit,
811*a28cd43dSSascha Wildner                          const BYTE* const prefixStart, const BYTE* const virtualStart, const BYTE* const dictEnd)
812*a28cd43dSSascha Wildner {
813*a28cd43dSSascha Wildner     BYTE* const oLitEnd = op + sequence.litLength;
814*a28cd43dSSascha Wildner     size_t const sequenceLength = sequence.litLength + sequence.matchLength;
815*a28cd43dSSascha Wildner     BYTE* const oMatchEnd = op + sequenceLength;   /* risk : address space overflow (32-bits) */
816*a28cd43dSSascha Wildner     BYTE* const oend_w = oend - WILDCOPY_OVERLENGTH;   /* risk : address space underflow on oend=NULL */
817*a28cd43dSSascha Wildner     const BYTE* const iLitEnd = *litPtr + sequence.litLength;
818*a28cd43dSSascha Wildner     const BYTE* match = oLitEnd - sequence.offset;
819*a28cd43dSSascha Wildner 
820*a28cd43dSSascha Wildner     assert(op != NULL /* Precondition */);
821*a28cd43dSSascha Wildner     assert(oend_w < oend /* No underflow */);
822*a28cd43dSSascha Wildner     /* Handle edge cases in a slow path:
823*a28cd43dSSascha Wildner      *   - Read beyond end of literals
824*a28cd43dSSascha Wildner      *   - Match end is within WILDCOPY_OVERLIMIT of oend
825*a28cd43dSSascha Wildner      *   - 32-bit mode and the match length overflows
826*a28cd43dSSascha Wildner      */
827*a28cd43dSSascha Wildner     if (UNLIKELY(
828*a28cd43dSSascha Wildner             iLitEnd > litLimit ||
829*a28cd43dSSascha Wildner             oMatchEnd > oend_w ||
830*a28cd43dSSascha Wildner             (MEM_32bits() && (size_t)(oend - op) < sequenceLength + WILDCOPY_OVERLENGTH)))
831*a28cd43dSSascha Wildner         return ZSTD_execSequenceEnd(op, oend, sequence, litPtr, litLimit, prefixStart, virtualStart, dictEnd);
832*a28cd43dSSascha Wildner 
833*a28cd43dSSascha Wildner     /* Assumptions (everything else goes into ZSTD_execSequenceEnd()) */
834*a28cd43dSSascha Wildner     assert(op <= oLitEnd /* No overflow */);
835*a28cd43dSSascha Wildner     assert(oLitEnd < oMatchEnd /* Non-zero match & no overflow */);
836*a28cd43dSSascha Wildner     assert(oMatchEnd <= oend /* No underflow */);
837*a28cd43dSSascha Wildner     assert(iLitEnd <= litLimit /* Literal length is in bounds */);
838*a28cd43dSSascha Wildner     assert(oLitEnd <= oend_w /* Can wildcopy literals */);
839*a28cd43dSSascha Wildner     assert(oMatchEnd <= oend_w /* Can wildcopy matches */);
840*a28cd43dSSascha Wildner 
841*a28cd43dSSascha Wildner     /* Copy Literals:
842*a28cd43dSSascha Wildner      * Split out litLength <= 16 since it is nearly always true. +1.6% on gcc-9.
843*a28cd43dSSascha Wildner      * We likely don't need the full 32-byte wildcopy.
844*a28cd43dSSascha Wildner      */
845*a28cd43dSSascha Wildner     assert(WILDCOPY_OVERLENGTH >= 16);
846*a28cd43dSSascha Wildner     ZSTD_copy16(op, (*litPtr));
847*a28cd43dSSascha Wildner     if (UNLIKELY(sequence.litLength > 16)) {
848*a28cd43dSSascha Wildner         ZSTD_wildcopy(op+16, (*litPtr)+16, sequence.litLength-16, ZSTD_no_overlap);
849*a28cd43dSSascha Wildner     }
850*a28cd43dSSascha Wildner     op = oLitEnd;
851*a28cd43dSSascha Wildner     *litPtr = iLitEnd;   /* update for next sequence */
852*a28cd43dSSascha Wildner 
853*a28cd43dSSascha Wildner     /* Copy Match */
854*a28cd43dSSascha Wildner     if (sequence.offset > (size_t)(oLitEnd - prefixStart)) {
855*a28cd43dSSascha Wildner         /* offset beyond prefix -> go into extDict */
856*a28cd43dSSascha Wildner         RETURN_ERROR_IF(UNLIKELY(sequence.offset > (size_t)(oLitEnd - virtualStart)), corruption_detected, "");
857*a28cd43dSSascha Wildner         match = dictEnd + (match - prefixStart);
858*a28cd43dSSascha Wildner         if (match + sequence.matchLength <= dictEnd) {
859*a28cd43dSSascha Wildner             ZSTD_memmove(oLitEnd, match, sequence.matchLength);
860*a28cd43dSSascha Wildner             return sequenceLength;
861*a28cd43dSSascha Wildner         }
862*a28cd43dSSascha Wildner         /* span extDict & currentPrefixSegment */
863*a28cd43dSSascha Wildner         {   size_t const length1 = dictEnd - match;
864*a28cd43dSSascha Wildner             ZSTD_memmove(oLitEnd, match, length1);
865*a28cd43dSSascha Wildner             op = oLitEnd + length1;
866*a28cd43dSSascha Wildner             sequence.matchLength -= length1;
867*a28cd43dSSascha Wildner             match = prefixStart;
868*a28cd43dSSascha Wildner     }   }
869*a28cd43dSSascha Wildner     /* Match within prefix of 1 or more bytes */
870*a28cd43dSSascha Wildner     assert(op <= oMatchEnd);
871*a28cd43dSSascha Wildner     assert(oMatchEnd <= oend_w);
872*a28cd43dSSascha Wildner     assert(match >= prefixStart);
873*a28cd43dSSascha Wildner     assert(sequence.matchLength >= 1);
874*a28cd43dSSascha Wildner 
875*a28cd43dSSascha Wildner     /* Nearly all offsets are >= WILDCOPY_VECLEN bytes, which means we can use wildcopy
876*a28cd43dSSascha Wildner      * without overlap checking.
877*a28cd43dSSascha Wildner      */
878*a28cd43dSSascha Wildner     if (LIKELY(sequence.offset >= WILDCOPY_VECLEN)) {
879*a28cd43dSSascha Wildner         /* We bet on a full wildcopy for matches, since we expect matches to be
880*a28cd43dSSascha Wildner          * longer than literals (in general). In silesia, ~10% of matches are longer
881*a28cd43dSSascha Wildner          * than 16 bytes.
882*a28cd43dSSascha Wildner          */
883*a28cd43dSSascha Wildner         ZSTD_wildcopy(op, match, (ptrdiff_t)sequence.matchLength, ZSTD_no_overlap);
884*a28cd43dSSascha Wildner         return sequenceLength;
885*a28cd43dSSascha Wildner     }
886*a28cd43dSSascha Wildner     assert(sequence.offset < WILDCOPY_VECLEN);
887*a28cd43dSSascha Wildner 
888*a28cd43dSSascha Wildner     /* Copy 8 bytes and spread the offset to be >= 8. */
889*a28cd43dSSascha Wildner     ZSTD_overlapCopy8(&op, &match, sequence.offset);
890*a28cd43dSSascha Wildner 
891*a28cd43dSSascha Wildner     /* If the match length is > 8 bytes, then continue with the wildcopy. */
892*a28cd43dSSascha Wildner     if (sequence.matchLength > 8) {
893*a28cd43dSSascha Wildner         assert(op < oMatchEnd);
894*a28cd43dSSascha Wildner         ZSTD_wildcopy(op, match, (ptrdiff_t)sequence.matchLength-8, ZSTD_overlap_src_before_dst);
895*a28cd43dSSascha Wildner     }
896*a28cd43dSSascha Wildner     return sequenceLength;
897*a28cd43dSSascha Wildner }
898*a28cd43dSSascha Wildner 
899*a28cd43dSSascha Wildner static void
ZSTD_initFseState(ZSTD_fseState * DStatePtr,BIT_DStream_t * bitD,const ZSTD_seqSymbol * dt)900*a28cd43dSSascha Wildner ZSTD_initFseState(ZSTD_fseState* DStatePtr, BIT_DStream_t* bitD, const ZSTD_seqSymbol* dt)
901*a28cd43dSSascha Wildner {
902*a28cd43dSSascha Wildner     const void* ptr = dt;
903*a28cd43dSSascha Wildner     const ZSTD_seqSymbol_header* const DTableH = (const ZSTD_seqSymbol_header*)ptr;
904*a28cd43dSSascha Wildner     DStatePtr->state = BIT_readBits(bitD, DTableH->tableLog);
905*a28cd43dSSascha Wildner     DEBUGLOG(6, "ZSTD_initFseState : val=%u using %u bits",
906*a28cd43dSSascha Wildner                 (U32)DStatePtr->state, DTableH->tableLog);
907*a28cd43dSSascha Wildner     BIT_reloadDStream(bitD);
908*a28cd43dSSascha Wildner     DStatePtr->table = dt + 1;
909*a28cd43dSSascha Wildner }
910*a28cd43dSSascha Wildner 
911*a28cd43dSSascha Wildner FORCE_INLINE_TEMPLATE void
ZSTD_updateFseState(ZSTD_fseState * DStatePtr,BIT_DStream_t * bitD)912*a28cd43dSSascha Wildner ZSTD_updateFseState(ZSTD_fseState* DStatePtr, BIT_DStream_t* bitD)
913*a28cd43dSSascha Wildner {
914*a28cd43dSSascha Wildner     ZSTD_seqSymbol const DInfo = DStatePtr->table[DStatePtr->state];
915*a28cd43dSSascha Wildner     U32 const nbBits = DInfo.nbBits;
916*a28cd43dSSascha Wildner     size_t const lowBits = BIT_readBits(bitD, nbBits);
917*a28cd43dSSascha Wildner     DStatePtr->state = DInfo.nextState + lowBits;
918*a28cd43dSSascha Wildner }
919*a28cd43dSSascha Wildner 
920*a28cd43dSSascha Wildner FORCE_INLINE_TEMPLATE void
ZSTD_updateFseStateWithDInfo(ZSTD_fseState * DStatePtr,BIT_DStream_t * bitD,ZSTD_seqSymbol const DInfo)921*a28cd43dSSascha Wildner ZSTD_updateFseStateWithDInfo(ZSTD_fseState* DStatePtr, BIT_DStream_t* bitD, ZSTD_seqSymbol const DInfo)
922*a28cd43dSSascha Wildner {
923*a28cd43dSSascha Wildner     U32 const nbBits = DInfo.nbBits;
924*a28cd43dSSascha Wildner     size_t const lowBits = BIT_readBits(bitD, nbBits);
925*a28cd43dSSascha Wildner     DStatePtr->state = DInfo.nextState + lowBits;
926*a28cd43dSSascha Wildner }
927*a28cd43dSSascha Wildner 
928*a28cd43dSSascha Wildner /* We need to add at most (ZSTD_WINDOWLOG_MAX_32 - 1) bits to read the maximum
929*a28cd43dSSascha Wildner  * offset bits. But we can only read at most (STREAM_ACCUMULATOR_MIN_32 - 1)
930*a28cd43dSSascha Wildner  * bits before reloading. This value is the maximum number of bytes we read
931*a28cd43dSSascha Wildner  * after reloading when we are decoding long offsets.
932*a28cd43dSSascha Wildner  */
933*a28cd43dSSascha Wildner #define LONG_OFFSETS_MAX_EXTRA_BITS_32                       \
934*a28cd43dSSascha Wildner     (ZSTD_WINDOWLOG_MAX_32 > STREAM_ACCUMULATOR_MIN_32       \
935*a28cd43dSSascha Wildner         ? ZSTD_WINDOWLOG_MAX_32 - STREAM_ACCUMULATOR_MIN_32  \
936*a28cd43dSSascha Wildner         : 0)
937*a28cd43dSSascha Wildner 
938*a28cd43dSSascha Wildner typedef enum { ZSTD_lo_isRegularOffset, ZSTD_lo_isLongOffset=1 } ZSTD_longOffset_e;
939*a28cd43dSSascha Wildner typedef enum { ZSTD_p_noPrefetch=0, ZSTD_p_prefetch=1 } ZSTD_prefetch_e;
940*a28cd43dSSascha Wildner 
941*a28cd43dSSascha Wildner FORCE_INLINE_TEMPLATE seq_t
ZSTD_decodeSequence(seqState_t * seqState,const ZSTD_longOffset_e longOffsets,const ZSTD_prefetch_e prefetch)942*a28cd43dSSascha Wildner ZSTD_decodeSequence(seqState_t* seqState, const ZSTD_longOffset_e longOffsets, const ZSTD_prefetch_e prefetch)
943*a28cd43dSSascha Wildner {
944*a28cd43dSSascha Wildner     seq_t seq;
945*a28cd43dSSascha Wildner     ZSTD_seqSymbol const llDInfo = seqState->stateLL.table[seqState->stateLL.state];
946*a28cd43dSSascha Wildner     ZSTD_seqSymbol const mlDInfo = seqState->stateML.table[seqState->stateML.state];
947*a28cd43dSSascha Wildner     ZSTD_seqSymbol const ofDInfo = seqState->stateOffb.table[seqState->stateOffb.state];
948*a28cd43dSSascha Wildner     U32 const llBase = llDInfo.baseValue;
949*a28cd43dSSascha Wildner     U32 const mlBase = mlDInfo.baseValue;
950*a28cd43dSSascha Wildner     U32 const ofBase = ofDInfo.baseValue;
951*a28cd43dSSascha Wildner     BYTE const llBits = llDInfo.nbAdditionalBits;
952*a28cd43dSSascha Wildner     BYTE const mlBits = mlDInfo.nbAdditionalBits;
953*a28cd43dSSascha Wildner     BYTE const ofBits = ofDInfo.nbAdditionalBits;
954*a28cd43dSSascha Wildner     BYTE const totalBits = llBits+mlBits+ofBits;
955*a28cd43dSSascha Wildner 
956*a28cd43dSSascha Wildner     /* sequence */
957*a28cd43dSSascha Wildner     {   size_t offset;
958*a28cd43dSSascha Wildner         if (ofBits > 1) {
959*a28cd43dSSascha Wildner             ZSTD_STATIC_ASSERT(ZSTD_lo_isLongOffset == 1);
960*a28cd43dSSascha Wildner             ZSTD_STATIC_ASSERT(LONG_OFFSETS_MAX_EXTRA_BITS_32 == 5);
961*a28cd43dSSascha Wildner             assert(ofBits <= MaxOff);
962*a28cd43dSSascha Wildner             if (MEM_32bits() && longOffsets && (ofBits >= STREAM_ACCUMULATOR_MIN_32)) {
963*a28cd43dSSascha Wildner                 U32 const extraBits = ofBits - MIN(ofBits, 32 - seqState->DStream.bitsConsumed);
964*a28cd43dSSascha Wildner                 offset = ofBase + (BIT_readBitsFast(&seqState->DStream, ofBits - extraBits) << extraBits);
965*a28cd43dSSascha Wildner                 BIT_reloadDStream(&seqState->DStream);
966*a28cd43dSSascha Wildner                 if (extraBits) offset += BIT_readBitsFast(&seqState->DStream, extraBits);
967*a28cd43dSSascha Wildner                 assert(extraBits <= LONG_OFFSETS_MAX_EXTRA_BITS_32);   /* to avoid another reload */
968*a28cd43dSSascha Wildner             } else {
969*a28cd43dSSascha Wildner                 offset = ofBase + BIT_readBitsFast(&seqState->DStream, ofBits/*>0*/);   /* <=  (ZSTD_WINDOWLOG_MAX-1) bits */
970*a28cd43dSSascha Wildner                 if (MEM_32bits()) BIT_reloadDStream(&seqState->DStream);
971*a28cd43dSSascha Wildner             }
972*a28cd43dSSascha Wildner             seqState->prevOffset[2] = seqState->prevOffset[1];
973*a28cd43dSSascha Wildner             seqState->prevOffset[1] = seqState->prevOffset[0];
974*a28cd43dSSascha Wildner             seqState->prevOffset[0] = offset;
975*a28cd43dSSascha Wildner         } else {
976*a28cd43dSSascha Wildner             U32 const ll0 = (llBase == 0);
977*a28cd43dSSascha Wildner             if (LIKELY((ofBits == 0))) {
978*a28cd43dSSascha Wildner                 if (LIKELY(!ll0))
979*a28cd43dSSascha Wildner                     offset = seqState->prevOffset[0];
980*a28cd43dSSascha Wildner                 else {
981*a28cd43dSSascha Wildner                     offset = seqState->prevOffset[1];
982*a28cd43dSSascha Wildner                     seqState->prevOffset[1] = seqState->prevOffset[0];
983*a28cd43dSSascha Wildner                     seqState->prevOffset[0] = offset;
984*a28cd43dSSascha Wildner                 }
985*a28cd43dSSascha Wildner             } else {
986*a28cd43dSSascha Wildner                 offset = ofBase + ll0 + BIT_readBitsFast(&seqState->DStream, 1);
987*a28cd43dSSascha Wildner                 {   size_t temp = (offset==3) ? seqState->prevOffset[0] - 1 : seqState->prevOffset[offset];
988*a28cd43dSSascha Wildner                     temp += !temp;   /* 0 is not valid; input is corrupted; force offset to 1 */
989*a28cd43dSSascha Wildner                     if (offset != 1) seqState->prevOffset[2] = seqState->prevOffset[1];
990*a28cd43dSSascha Wildner                     seqState->prevOffset[1] = seqState->prevOffset[0];
991*a28cd43dSSascha Wildner                     seqState->prevOffset[0] = offset = temp;
992*a28cd43dSSascha Wildner         }   }   }
993*a28cd43dSSascha Wildner         seq.offset = offset;
994*a28cd43dSSascha Wildner     }
995*a28cd43dSSascha Wildner 
996*a28cd43dSSascha Wildner     seq.matchLength = mlBase;
997*a28cd43dSSascha Wildner     if (mlBits > 0)
998*a28cd43dSSascha Wildner         seq.matchLength += BIT_readBitsFast(&seqState->DStream, mlBits/*>0*/);
999*a28cd43dSSascha Wildner 
1000*a28cd43dSSascha Wildner     if (MEM_32bits() && (mlBits+llBits >= STREAM_ACCUMULATOR_MIN_32-LONG_OFFSETS_MAX_EXTRA_BITS_32))
1001*a28cd43dSSascha Wildner         BIT_reloadDStream(&seqState->DStream);
1002*a28cd43dSSascha Wildner     if (MEM_64bits() && UNLIKELY(totalBits >= STREAM_ACCUMULATOR_MIN_64-(LLFSELog+MLFSELog+OffFSELog)))
1003*a28cd43dSSascha Wildner         BIT_reloadDStream(&seqState->DStream);
1004*a28cd43dSSascha Wildner     /* Ensure there are enough bits to read the rest of data in 64-bit mode. */
1005*a28cd43dSSascha Wildner     ZSTD_STATIC_ASSERT(16+LLFSELog+MLFSELog+OffFSELog < STREAM_ACCUMULATOR_MIN_64);
1006*a28cd43dSSascha Wildner 
1007*a28cd43dSSascha Wildner     seq.litLength = llBase;
1008*a28cd43dSSascha Wildner     if (llBits > 0)
1009*a28cd43dSSascha Wildner         seq.litLength += BIT_readBitsFast(&seqState->DStream, llBits/*>0*/);
1010*a28cd43dSSascha Wildner 
1011*a28cd43dSSascha Wildner     if (MEM_32bits())
1012*a28cd43dSSascha Wildner         BIT_reloadDStream(&seqState->DStream);
1013*a28cd43dSSascha Wildner 
1014*a28cd43dSSascha Wildner     DEBUGLOG(6, "seq: litL=%u, matchL=%u, offset=%u",
1015*a28cd43dSSascha Wildner                 (U32)seq.litLength, (U32)seq.matchLength, (U32)seq.offset);
1016*a28cd43dSSascha Wildner 
1017*a28cd43dSSascha Wildner     if (prefetch == ZSTD_p_prefetch) {
1018*a28cd43dSSascha Wildner         size_t const pos = seqState->pos + seq.litLength;
1019*a28cd43dSSascha Wildner         const BYTE* const matchBase = (seq.offset > pos) ? seqState->dictEnd : seqState->prefixStart;
1020*a28cd43dSSascha Wildner         seq.match = matchBase + pos - seq.offset;  /* note : this operation can overflow when seq.offset is really too large, which can only happen when input is corrupted.
1021*a28cd43dSSascha Wildner                                                     * No consequence though : no memory access will occur, offset is only used for prefetching */
1022*a28cd43dSSascha Wildner         seqState->pos = pos + seq.matchLength;
1023*a28cd43dSSascha Wildner     }
1024*a28cd43dSSascha Wildner 
1025*a28cd43dSSascha Wildner     /* ANS state update
1026*a28cd43dSSascha Wildner      * gcc-9.0.0 does 2.5% worse with ZSTD_updateFseStateWithDInfo().
1027*a28cd43dSSascha Wildner      * clang-9.2.0 does 7% worse with ZSTD_updateFseState().
1028*a28cd43dSSascha Wildner      * Naturally it seems like ZSTD_updateFseStateWithDInfo() should be the
1029*a28cd43dSSascha Wildner      * better option, so it is the default for other compilers. But, if you
1030*a28cd43dSSascha Wildner      * measure that it is worse, please put up a pull request.
1031*a28cd43dSSascha Wildner      */
1032*a28cd43dSSascha Wildner     {
1033*a28cd43dSSascha Wildner #if defined(__GNUC__) && !defined(__clang__)
1034*a28cd43dSSascha Wildner         const int kUseUpdateFseState = 1;
1035*a28cd43dSSascha Wildner #else
1036*a28cd43dSSascha Wildner         const int kUseUpdateFseState = 0;
1037*a28cd43dSSascha Wildner #endif
1038*a28cd43dSSascha Wildner         if (kUseUpdateFseState) {
1039*a28cd43dSSascha Wildner             ZSTD_updateFseState(&seqState->stateLL, &seqState->DStream);    /* <=  9 bits */
1040*a28cd43dSSascha Wildner             ZSTD_updateFseState(&seqState->stateML, &seqState->DStream);    /* <=  9 bits */
1041*a28cd43dSSascha Wildner             if (MEM_32bits()) BIT_reloadDStream(&seqState->DStream);    /* <= 18 bits */
1042*a28cd43dSSascha Wildner             ZSTD_updateFseState(&seqState->stateOffb, &seqState->DStream);  /* <=  8 bits */
1043*a28cd43dSSascha Wildner         } else {
1044*a28cd43dSSascha Wildner             ZSTD_updateFseStateWithDInfo(&seqState->stateLL, &seqState->DStream, llDInfo);    /* <=  9 bits */
1045*a28cd43dSSascha Wildner             ZSTD_updateFseStateWithDInfo(&seqState->stateML, &seqState->DStream, mlDInfo);    /* <=  9 bits */
1046*a28cd43dSSascha Wildner             if (MEM_32bits()) BIT_reloadDStream(&seqState->DStream);    /* <= 18 bits */
1047*a28cd43dSSascha Wildner             ZSTD_updateFseStateWithDInfo(&seqState->stateOffb, &seqState->DStream, ofDInfo);  /* <=  8 bits */
1048*a28cd43dSSascha Wildner         }
1049*a28cd43dSSascha Wildner     }
1050*a28cd43dSSascha Wildner 
1051*a28cd43dSSascha Wildner     return seq;
1052*a28cd43dSSascha Wildner }
1053*a28cd43dSSascha Wildner 
1054*a28cd43dSSascha Wildner #ifdef FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION
ZSTD_dictionaryIsActive(ZSTD_DCtx const * dctx,BYTE const * prefixStart,BYTE const * oLitEnd)1055*a28cd43dSSascha Wildner MEM_STATIC int ZSTD_dictionaryIsActive(ZSTD_DCtx const* dctx, BYTE const* prefixStart, BYTE const* oLitEnd)
1056*a28cd43dSSascha Wildner {
1057*a28cd43dSSascha Wildner     size_t const windowSize = dctx->fParams.windowSize;
1058*a28cd43dSSascha Wildner     /* No dictionary used. */
1059*a28cd43dSSascha Wildner     if (dctx->dictContentEndForFuzzing == NULL) return 0;
1060*a28cd43dSSascha Wildner     /* Dictionary is our prefix. */
1061*a28cd43dSSascha Wildner     if (prefixStart == dctx->dictContentBeginForFuzzing) return 1;
1062*a28cd43dSSascha Wildner     /* Dictionary is not our ext-dict. */
1063*a28cd43dSSascha Wildner     if (dctx->dictEnd != dctx->dictContentEndForFuzzing) return 0;
1064*a28cd43dSSascha Wildner     /* Dictionary is not within our window size. */
1065*a28cd43dSSascha Wildner     if ((size_t)(oLitEnd - prefixStart) >= windowSize) return 0;
1066*a28cd43dSSascha Wildner     /* Dictionary is active. */
1067*a28cd43dSSascha Wildner     return 1;
1068*a28cd43dSSascha Wildner }
1069*a28cd43dSSascha Wildner 
ZSTD_assertValidSequence(ZSTD_DCtx const * dctx,BYTE const * op,BYTE const * oend,seq_t const seq,BYTE const * prefixStart,BYTE const * virtualStart)1070*a28cd43dSSascha Wildner MEM_STATIC void ZSTD_assertValidSequence(
1071*a28cd43dSSascha Wildner         ZSTD_DCtx const* dctx,
1072*a28cd43dSSascha Wildner         BYTE const* op, BYTE const* oend,
1073*a28cd43dSSascha Wildner         seq_t const seq,
1074*a28cd43dSSascha Wildner         BYTE const* prefixStart, BYTE const* virtualStart)
1075*a28cd43dSSascha Wildner {
1076*a28cd43dSSascha Wildner #if DEBUGLEVEL >= 1
1077*a28cd43dSSascha Wildner     size_t const windowSize = dctx->fParams.windowSize;
1078*a28cd43dSSascha Wildner     size_t const sequenceSize = seq.litLength + seq.matchLength;
1079*a28cd43dSSascha Wildner     BYTE const* const oLitEnd = op + seq.litLength;
1080*a28cd43dSSascha Wildner     DEBUGLOG(6, "Checking sequence: litL=%u matchL=%u offset=%u",
1081*a28cd43dSSascha Wildner             (U32)seq.litLength, (U32)seq.matchLength, (U32)seq.offset);
1082*a28cd43dSSascha Wildner     assert(op <= oend);
1083*a28cd43dSSascha Wildner     assert((size_t)(oend - op) >= sequenceSize);
1084*a28cd43dSSascha Wildner     assert(sequenceSize <= ZSTD_BLOCKSIZE_MAX);
1085*a28cd43dSSascha Wildner     if (ZSTD_dictionaryIsActive(dctx, prefixStart, oLitEnd)) {
1086*a28cd43dSSascha Wildner         size_t const dictSize = (size_t)((char const*)dctx->dictContentEndForFuzzing - (char const*)dctx->dictContentBeginForFuzzing);
1087*a28cd43dSSascha Wildner         /* Offset must be within the dictionary. */
1088*a28cd43dSSascha Wildner         assert(seq.offset <= (size_t)(oLitEnd - virtualStart));
1089*a28cd43dSSascha Wildner         assert(seq.offset <= windowSize + dictSize);
1090*a28cd43dSSascha Wildner     } else {
1091*a28cd43dSSascha Wildner         /* Offset must be within our window. */
1092*a28cd43dSSascha Wildner         assert(seq.offset <= windowSize);
1093*a28cd43dSSascha Wildner     }
1094*a28cd43dSSascha Wildner #else
1095*a28cd43dSSascha Wildner     (void)dctx, (void)op, (void)oend, (void)seq, (void)prefixStart, (void)virtualStart;
1096*a28cd43dSSascha Wildner #endif
1097*a28cd43dSSascha Wildner }
1098*a28cd43dSSascha Wildner #endif
1099*a28cd43dSSascha Wildner 
1100*a28cd43dSSascha Wildner #ifndef ZSTD_FORCE_DECOMPRESS_SEQUENCES_LONG
1101*a28cd43dSSascha Wildner FORCE_INLINE_TEMPLATE size_t
1102*a28cd43dSSascha Wildner DONT_VECTORIZE
ZSTD_decompressSequences_body(ZSTD_DCtx * dctx,void * dst,size_t maxDstSize,const void * seqStart,size_t seqSize,int nbSeq,const ZSTD_longOffset_e isLongOffset,const int frame)1103*a28cd43dSSascha Wildner ZSTD_decompressSequences_body( ZSTD_DCtx* dctx,
1104*a28cd43dSSascha Wildner                                void* dst, size_t maxDstSize,
1105*a28cd43dSSascha Wildner                          const void* seqStart, size_t seqSize, int nbSeq,
1106*a28cd43dSSascha Wildner                          const ZSTD_longOffset_e isLongOffset,
1107*a28cd43dSSascha Wildner                          const int frame)
1108*a28cd43dSSascha Wildner {
1109*a28cd43dSSascha Wildner     const BYTE* ip = (const BYTE*)seqStart;
1110*a28cd43dSSascha Wildner     const BYTE* const iend = ip + seqSize;
1111*a28cd43dSSascha Wildner     BYTE* const ostart = (BYTE* const)dst;
1112*a28cd43dSSascha Wildner     BYTE* const oend = ostart + maxDstSize;
1113*a28cd43dSSascha Wildner     BYTE* op = ostart;
1114*a28cd43dSSascha Wildner     const BYTE* litPtr = dctx->litPtr;
1115*a28cd43dSSascha Wildner     const BYTE* const litEnd = litPtr + dctx->litSize;
1116*a28cd43dSSascha Wildner     const BYTE* const prefixStart = (const BYTE*) (dctx->prefixStart);
1117*a28cd43dSSascha Wildner     const BYTE* const vBase = (const BYTE*) (dctx->virtualStart);
1118*a28cd43dSSascha Wildner     const BYTE* const dictEnd = (const BYTE*) (dctx->dictEnd);
1119*a28cd43dSSascha Wildner     DEBUGLOG(5, "ZSTD_decompressSequences_body");
1120*a28cd43dSSascha Wildner     (void)frame;
1121*a28cd43dSSascha Wildner 
1122*a28cd43dSSascha Wildner     /* Regen sequences */
1123*a28cd43dSSascha Wildner     if (nbSeq) {
1124*a28cd43dSSascha Wildner         seqState_t seqState;
1125*a28cd43dSSascha Wildner         size_t error = 0;
1126*a28cd43dSSascha Wildner         dctx->fseEntropy = 1;
1127*a28cd43dSSascha Wildner         { U32 i; for (i=0; i<ZSTD_REP_NUM; i++) seqState.prevOffset[i] = dctx->entropy.rep[i]; }
1128*a28cd43dSSascha Wildner         RETURN_ERROR_IF(
1129*a28cd43dSSascha Wildner             ERR_isError(BIT_initDStream(&seqState.DStream, ip, iend-ip)),
1130*a28cd43dSSascha Wildner             corruption_detected, "");
1131*a28cd43dSSascha Wildner         ZSTD_initFseState(&seqState.stateLL, &seqState.DStream, dctx->LLTptr);
1132*a28cd43dSSascha Wildner         ZSTD_initFseState(&seqState.stateOffb, &seqState.DStream, dctx->OFTptr);
1133*a28cd43dSSascha Wildner         ZSTD_initFseState(&seqState.stateML, &seqState.DStream, dctx->MLTptr);
1134*a28cd43dSSascha Wildner         assert(dst != NULL);
1135*a28cd43dSSascha Wildner 
1136*a28cd43dSSascha Wildner         ZSTD_STATIC_ASSERT(
1137*a28cd43dSSascha Wildner                 BIT_DStream_unfinished < BIT_DStream_completed &&
1138*a28cd43dSSascha Wildner                 BIT_DStream_endOfBuffer < BIT_DStream_completed &&
1139*a28cd43dSSascha Wildner                 BIT_DStream_completed < BIT_DStream_overflow);
1140*a28cd43dSSascha Wildner 
1141*a28cd43dSSascha Wildner #if defined(__GNUC__) && defined(__x86_64__)
1142*a28cd43dSSascha Wildner         /* Align the decompression loop to 32 + 16 bytes.
1143*a28cd43dSSascha Wildner          *
1144*a28cd43dSSascha Wildner          * zstd compiled with gcc-9 on an Intel i9-9900k shows 10% decompression
1145*a28cd43dSSascha Wildner          * speed swings based on the alignment of the decompression loop. This
1146*a28cd43dSSascha Wildner          * performance swing is caused by parts of the decompression loop falling
1147*a28cd43dSSascha Wildner          * out of the DSB. The entire decompression loop should fit in the DSB,
1148*a28cd43dSSascha Wildner          * when it can't we get much worse performance. You can measure if you've
1149*a28cd43dSSascha Wildner          * hit the good case or the bad case with this perf command for some
1150*a28cd43dSSascha Wildner          * compressed file test.zst:
1151*a28cd43dSSascha Wildner          *
1152*a28cd43dSSascha Wildner          *   perf stat -e cycles -e instructions -e idq.all_dsb_cycles_any_uops \
1153*a28cd43dSSascha Wildner          *             -e idq.all_mite_cycles_any_uops -- ./zstd -tq test.zst
1154*a28cd43dSSascha Wildner          *
1155*a28cd43dSSascha Wildner          * If you see most cycles served out of the MITE you've hit the bad case.
1156*a28cd43dSSascha Wildner          * If you see most cycles served out of the DSB you've hit the good case.
1157*a28cd43dSSascha Wildner          * If it is pretty even then you may be in an okay case.
1158*a28cd43dSSascha Wildner          *
1159*a28cd43dSSascha Wildner          * I've been able to reproduce this issue on the following CPUs:
1160*a28cd43dSSascha Wildner          *   - Kabylake: Macbook Pro (15-inch, 2019) 2.4 GHz Intel Core i9
1161*a28cd43dSSascha Wildner          *               Use Instruments->Counters to get DSB/MITE cycles.
1162*a28cd43dSSascha Wildner          *               I never got performance swings, but I was able to
1163*a28cd43dSSascha Wildner          *               go from the good case of mostly DSB to half of the
1164*a28cd43dSSascha Wildner          *               cycles served from MITE.
1165*a28cd43dSSascha Wildner          *   - Coffeelake: Intel i9-9900k
1166*a28cd43dSSascha Wildner          *
1167*a28cd43dSSascha Wildner          * I haven't been able to reproduce the instability or DSB misses on any
1168*a28cd43dSSascha Wildner          * of the following CPUS:
1169*a28cd43dSSascha Wildner          *   - Haswell
1170*a28cd43dSSascha Wildner          *   - Broadwell: Intel(R) Xeon(R) CPU E5-2680 v4 @ 2.40GH
1171*a28cd43dSSascha Wildner          *   - Skylake
1172*a28cd43dSSascha Wildner          *
1173*a28cd43dSSascha Wildner          * If you are seeing performance stability this script can help test.
1174*a28cd43dSSascha Wildner          * It tests on 4 commits in zstd where I saw performance change.
1175*a28cd43dSSascha Wildner          *
1176*a28cd43dSSascha Wildner          *   https://gist.github.com/terrelln/9889fc06a423fd5ca6e99351564473f4
1177*a28cd43dSSascha Wildner          */
1178*a28cd43dSSascha Wildner         __asm__(".p2align 5");
1179*a28cd43dSSascha Wildner         __asm__("nop");
1180*a28cd43dSSascha Wildner         __asm__(".p2align 4");
1181*a28cd43dSSascha Wildner #endif
1182*a28cd43dSSascha Wildner         for ( ; ; ) {
1183*a28cd43dSSascha Wildner             seq_t const sequence = ZSTD_decodeSequence(&seqState, isLongOffset, ZSTD_p_noPrefetch);
1184*a28cd43dSSascha Wildner             size_t const oneSeqSize = ZSTD_execSequence(op, oend, sequence, &litPtr, litEnd, prefixStart, vBase, dictEnd);
1185*a28cd43dSSascha Wildner #if defined(FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION) && defined(FUZZING_ASSERT_VALID_SEQUENCE)
1186*a28cd43dSSascha Wildner             assert(!ZSTD_isError(oneSeqSize));
1187*a28cd43dSSascha Wildner             if (frame) ZSTD_assertValidSequence(dctx, op, oend, sequence, prefixStart, vBase);
1188*a28cd43dSSascha Wildner #endif
1189*a28cd43dSSascha Wildner             DEBUGLOG(6, "regenerated sequence size : %u", (U32)oneSeqSize);
1190*a28cd43dSSascha Wildner             BIT_reloadDStream(&(seqState.DStream));
1191*a28cd43dSSascha Wildner             op += oneSeqSize;
1192*a28cd43dSSascha Wildner             /* gcc and clang both don't like early returns in this loop.
1193*a28cd43dSSascha Wildner              * Instead break and check for an error at the end of the loop.
1194*a28cd43dSSascha Wildner              */
1195*a28cd43dSSascha Wildner             if (UNLIKELY(ZSTD_isError(oneSeqSize))) {
1196*a28cd43dSSascha Wildner                 error = oneSeqSize;
1197*a28cd43dSSascha Wildner                 break;
1198*a28cd43dSSascha Wildner             }
1199*a28cd43dSSascha Wildner             if (UNLIKELY(!--nbSeq)) break;
1200*a28cd43dSSascha Wildner         }
1201*a28cd43dSSascha Wildner 
1202*a28cd43dSSascha Wildner         /* check if reached exact end */
1203*a28cd43dSSascha Wildner         DEBUGLOG(5, "ZSTD_decompressSequences_body: after decode loop, remaining nbSeq : %i", nbSeq);
1204*a28cd43dSSascha Wildner         if (ZSTD_isError(error)) return error;
1205*a28cd43dSSascha Wildner         RETURN_ERROR_IF(nbSeq, corruption_detected, "");
1206*a28cd43dSSascha Wildner         RETURN_ERROR_IF(BIT_reloadDStream(&seqState.DStream) < BIT_DStream_completed, corruption_detected, "");
1207*a28cd43dSSascha Wildner         /* save reps for next block */
1208*a28cd43dSSascha Wildner         { U32 i; for (i=0; i<ZSTD_REP_NUM; i++) dctx->entropy.rep[i] = (U32)(seqState.prevOffset[i]); }
1209*a28cd43dSSascha Wildner     }
1210*a28cd43dSSascha Wildner 
1211*a28cd43dSSascha Wildner     /* last literal segment */
1212*a28cd43dSSascha Wildner     {   size_t const lastLLSize = litEnd - litPtr;
1213*a28cd43dSSascha Wildner         RETURN_ERROR_IF(lastLLSize > (size_t)(oend-op), dstSize_tooSmall, "");
1214*a28cd43dSSascha Wildner         if (op != NULL) {
1215*a28cd43dSSascha Wildner             ZSTD_memcpy(op, litPtr, lastLLSize);
1216*a28cd43dSSascha Wildner             op += lastLLSize;
1217*a28cd43dSSascha Wildner         }
1218*a28cd43dSSascha Wildner     }
1219*a28cd43dSSascha Wildner 
1220*a28cd43dSSascha Wildner     return op-ostart;
1221*a28cd43dSSascha Wildner }
1222*a28cd43dSSascha Wildner 
1223*a28cd43dSSascha Wildner static size_t
ZSTD_decompressSequences_default(ZSTD_DCtx * dctx,void * dst,size_t maxDstSize,const void * seqStart,size_t seqSize,int nbSeq,const ZSTD_longOffset_e isLongOffset,const int frame)1224*a28cd43dSSascha Wildner ZSTD_decompressSequences_default(ZSTD_DCtx* dctx,
1225*a28cd43dSSascha Wildner                                  void* dst, size_t maxDstSize,
1226*a28cd43dSSascha Wildner                            const void* seqStart, size_t seqSize, int nbSeq,
1227*a28cd43dSSascha Wildner                            const ZSTD_longOffset_e isLongOffset,
1228*a28cd43dSSascha Wildner                            const int frame)
1229*a28cd43dSSascha Wildner {
1230*a28cd43dSSascha Wildner     return ZSTD_decompressSequences_body(dctx, dst, maxDstSize, seqStart, seqSize, nbSeq, isLongOffset, frame);
1231*a28cd43dSSascha Wildner }
1232*a28cd43dSSascha Wildner #endif /* ZSTD_FORCE_DECOMPRESS_SEQUENCES_LONG */
1233*a28cd43dSSascha Wildner 
1234*a28cd43dSSascha Wildner #ifndef ZSTD_FORCE_DECOMPRESS_SEQUENCES_SHORT
1235*a28cd43dSSascha Wildner FORCE_INLINE_TEMPLATE size_t
ZSTD_decompressSequencesLong_body(ZSTD_DCtx * dctx,void * dst,size_t maxDstSize,const void * seqStart,size_t seqSize,int nbSeq,const ZSTD_longOffset_e isLongOffset,const int frame)1236*a28cd43dSSascha Wildner ZSTD_decompressSequencesLong_body(
1237*a28cd43dSSascha Wildner                                ZSTD_DCtx* dctx,
1238*a28cd43dSSascha Wildner                                void* dst, size_t maxDstSize,
1239*a28cd43dSSascha Wildner                          const void* seqStart, size_t seqSize, int nbSeq,
1240*a28cd43dSSascha Wildner                          const ZSTD_longOffset_e isLongOffset,
1241*a28cd43dSSascha Wildner                          const int frame)
1242*a28cd43dSSascha Wildner {
1243*a28cd43dSSascha Wildner     const BYTE* ip = (const BYTE*)seqStart;
1244*a28cd43dSSascha Wildner     const BYTE* const iend = ip + seqSize;
1245*a28cd43dSSascha Wildner     BYTE* const ostart = (BYTE* const)dst;
1246*a28cd43dSSascha Wildner     BYTE* const oend = ostart + maxDstSize;
1247*a28cd43dSSascha Wildner     BYTE* op = ostart;
1248*a28cd43dSSascha Wildner     const BYTE* litPtr = dctx->litPtr;
1249*a28cd43dSSascha Wildner     const BYTE* const litEnd = litPtr + dctx->litSize;
1250*a28cd43dSSascha Wildner     const BYTE* const prefixStart = (const BYTE*) (dctx->prefixStart);
1251*a28cd43dSSascha Wildner     const BYTE* const dictStart = (const BYTE*) (dctx->virtualStart);
1252*a28cd43dSSascha Wildner     const BYTE* const dictEnd = (const BYTE*) (dctx->dictEnd);
1253*a28cd43dSSascha Wildner     (void)frame;
1254*a28cd43dSSascha Wildner 
1255*a28cd43dSSascha Wildner     /* Regen sequences */
1256*a28cd43dSSascha Wildner     if (nbSeq) {
1257*a28cd43dSSascha Wildner #define STORED_SEQS 4
1258*a28cd43dSSascha Wildner #define STORED_SEQS_MASK (STORED_SEQS-1)
1259*a28cd43dSSascha Wildner #define ADVANCED_SEQS 4
1260*a28cd43dSSascha Wildner         seq_t sequences[STORED_SEQS];
1261*a28cd43dSSascha Wildner         int const seqAdvance = MIN(nbSeq, ADVANCED_SEQS);
1262*a28cd43dSSascha Wildner         seqState_t seqState;
1263*a28cd43dSSascha Wildner         int seqNb;
1264*a28cd43dSSascha Wildner         dctx->fseEntropy = 1;
1265*a28cd43dSSascha Wildner         { int i; for (i=0; i<ZSTD_REP_NUM; i++) seqState.prevOffset[i] = dctx->entropy.rep[i]; }
1266*a28cd43dSSascha Wildner         seqState.prefixStart = prefixStart;
1267*a28cd43dSSascha Wildner         seqState.pos = (size_t)(op-prefixStart);
1268*a28cd43dSSascha Wildner         seqState.dictEnd = dictEnd;
1269*a28cd43dSSascha Wildner         assert(dst != NULL);
1270*a28cd43dSSascha Wildner         assert(iend >= ip);
1271*a28cd43dSSascha Wildner         RETURN_ERROR_IF(
1272*a28cd43dSSascha Wildner             ERR_isError(BIT_initDStream(&seqState.DStream, ip, iend-ip)),
1273*a28cd43dSSascha Wildner             corruption_detected, "");
1274*a28cd43dSSascha Wildner         ZSTD_initFseState(&seqState.stateLL, &seqState.DStream, dctx->LLTptr);
1275*a28cd43dSSascha Wildner         ZSTD_initFseState(&seqState.stateOffb, &seqState.DStream, dctx->OFTptr);
1276*a28cd43dSSascha Wildner         ZSTD_initFseState(&seqState.stateML, &seqState.DStream, dctx->MLTptr);
1277*a28cd43dSSascha Wildner 
1278*a28cd43dSSascha Wildner         /* prepare in advance */
1279*a28cd43dSSascha Wildner         for (seqNb=0; (BIT_reloadDStream(&seqState.DStream) <= BIT_DStream_completed) && (seqNb<seqAdvance); seqNb++) {
1280*a28cd43dSSascha Wildner             sequences[seqNb] = ZSTD_decodeSequence(&seqState, isLongOffset, ZSTD_p_prefetch);
1281*a28cd43dSSascha Wildner             PREFETCH_L1(sequences[seqNb].match); PREFETCH_L1(sequences[seqNb].match + sequences[seqNb].matchLength - 1); /* note : it's safe to invoke PREFETCH() on any memory address, including invalid ones */
1282*a28cd43dSSascha Wildner         }
1283*a28cd43dSSascha Wildner         RETURN_ERROR_IF(seqNb<seqAdvance, corruption_detected, "");
1284*a28cd43dSSascha Wildner 
1285*a28cd43dSSascha Wildner         /* decode and decompress */
1286*a28cd43dSSascha Wildner         for ( ; (BIT_reloadDStream(&(seqState.DStream)) <= BIT_DStream_completed) && (seqNb<nbSeq) ; seqNb++) {
1287*a28cd43dSSascha Wildner             seq_t const sequence = ZSTD_decodeSequence(&seqState, isLongOffset, ZSTD_p_prefetch);
1288*a28cd43dSSascha Wildner             size_t const oneSeqSize = ZSTD_execSequence(op, oend, sequences[(seqNb-ADVANCED_SEQS) & STORED_SEQS_MASK], &litPtr, litEnd, prefixStart, dictStart, dictEnd);
1289*a28cd43dSSascha Wildner #if defined(FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION) && defined(FUZZING_ASSERT_VALID_SEQUENCE)
1290*a28cd43dSSascha Wildner             assert(!ZSTD_isError(oneSeqSize));
1291*a28cd43dSSascha Wildner             if (frame) ZSTD_assertValidSequence(dctx, op, oend, sequences[(seqNb-ADVANCED_SEQS) & STORED_SEQS_MASK], prefixStart, dictStart);
1292*a28cd43dSSascha Wildner #endif
1293*a28cd43dSSascha Wildner             if (ZSTD_isError(oneSeqSize)) return oneSeqSize;
1294*a28cd43dSSascha Wildner             PREFETCH_L1(sequence.match); PREFETCH_L1(sequence.match + sequence.matchLength - 1); /* note : it's safe to invoke PREFETCH() on any memory address, including invalid ones */
1295*a28cd43dSSascha Wildner             sequences[seqNb & STORED_SEQS_MASK] = sequence;
1296*a28cd43dSSascha Wildner             op += oneSeqSize;
1297*a28cd43dSSascha Wildner         }
1298*a28cd43dSSascha Wildner         RETURN_ERROR_IF(seqNb<nbSeq, corruption_detected, "");
1299*a28cd43dSSascha Wildner 
1300*a28cd43dSSascha Wildner         /* finish queue */
1301*a28cd43dSSascha Wildner         seqNb -= seqAdvance;
1302*a28cd43dSSascha Wildner         for ( ; seqNb<nbSeq ; seqNb++) {
1303*a28cd43dSSascha Wildner             size_t const oneSeqSize = ZSTD_execSequence(op, oend, sequences[seqNb&STORED_SEQS_MASK], &litPtr, litEnd, prefixStart, dictStart, dictEnd);
1304*a28cd43dSSascha Wildner #if defined(FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION) && defined(FUZZING_ASSERT_VALID_SEQUENCE)
1305*a28cd43dSSascha Wildner             assert(!ZSTD_isError(oneSeqSize));
1306*a28cd43dSSascha Wildner             if (frame) ZSTD_assertValidSequence(dctx, op, oend, sequences[seqNb&STORED_SEQS_MASK], prefixStart, dictStart);
1307*a28cd43dSSascha Wildner #endif
1308*a28cd43dSSascha Wildner             if (ZSTD_isError(oneSeqSize)) return oneSeqSize;
1309*a28cd43dSSascha Wildner             op += oneSeqSize;
1310*a28cd43dSSascha Wildner         }
1311*a28cd43dSSascha Wildner 
1312*a28cd43dSSascha Wildner         /* save reps for next block */
1313*a28cd43dSSascha Wildner         { U32 i; for (i=0; i<ZSTD_REP_NUM; i++) dctx->entropy.rep[i] = (U32)(seqState.prevOffset[i]); }
1314*a28cd43dSSascha Wildner     }
1315*a28cd43dSSascha Wildner 
1316*a28cd43dSSascha Wildner     /* last literal segment */
1317*a28cd43dSSascha Wildner     {   size_t const lastLLSize = litEnd - litPtr;
1318*a28cd43dSSascha Wildner         RETURN_ERROR_IF(lastLLSize > (size_t)(oend-op), dstSize_tooSmall, "");
1319*a28cd43dSSascha Wildner         if (op != NULL) {
1320*a28cd43dSSascha Wildner             ZSTD_memcpy(op, litPtr, lastLLSize);
1321*a28cd43dSSascha Wildner             op += lastLLSize;
1322*a28cd43dSSascha Wildner         }
1323*a28cd43dSSascha Wildner     }
1324*a28cd43dSSascha Wildner 
1325*a28cd43dSSascha Wildner     return op-ostart;
1326*a28cd43dSSascha Wildner }
1327*a28cd43dSSascha Wildner 
1328*a28cd43dSSascha Wildner static size_t
ZSTD_decompressSequencesLong_default(ZSTD_DCtx * dctx,void * dst,size_t maxDstSize,const void * seqStart,size_t seqSize,int nbSeq,const ZSTD_longOffset_e isLongOffset,const int frame)1329*a28cd43dSSascha Wildner ZSTD_decompressSequencesLong_default(ZSTD_DCtx* dctx,
1330*a28cd43dSSascha Wildner                                  void* dst, size_t maxDstSize,
1331*a28cd43dSSascha Wildner                            const void* seqStart, size_t seqSize, int nbSeq,
1332*a28cd43dSSascha Wildner                            const ZSTD_longOffset_e isLongOffset,
1333*a28cd43dSSascha Wildner                            const int frame)
1334*a28cd43dSSascha Wildner {
1335*a28cd43dSSascha Wildner     return ZSTD_decompressSequencesLong_body(dctx, dst, maxDstSize, seqStart, seqSize, nbSeq, isLongOffset, frame);
1336*a28cd43dSSascha Wildner }
1337*a28cd43dSSascha Wildner #endif /* ZSTD_FORCE_DECOMPRESS_SEQUENCES_SHORT */
1338*a28cd43dSSascha Wildner 
1339*a28cd43dSSascha Wildner 
1340*a28cd43dSSascha Wildner 
1341*a28cd43dSSascha Wildner #if DYNAMIC_BMI2
1342*a28cd43dSSascha Wildner 
1343*a28cd43dSSascha Wildner #ifndef ZSTD_FORCE_DECOMPRESS_SEQUENCES_LONG
1344*a28cd43dSSascha Wildner static TARGET_ATTRIBUTE("bmi2") size_t
1345*a28cd43dSSascha Wildner DONT_VECTORIZE
ZSTD_decompressSequences_bmi2(ZSTD_DCtx * dctx,void * dst,size_t maxDstSize,const void * seqStart,size_t seqSize,int nbSeq,const ZSTD_longOffset_e isLongOffset,const int frame)1346*a28cd43dSSascha Wildner ZSTD_decompressSequences_bmi2(ZSTD_DCtx* dctx,
1347*a28cd43dSSascha Wildner                                  void* dst, size_t maxDstSize,
1348*a28cd43dSSascha Wildner                            const void* seqStart, size_t seqSize, int nbSeq,
1349*a28cd43dSSascha Wildner                            const ZSTD_longOffset_e isLongOffset,
1350*a28cd43dSSascha Wildner                            const int frame)
1351*a28cd43dSSascha Wildner {
1352*a28cd43dSSascha Wildner     return ZSTD_decompressSequences_body(dctx, dst, maxDstSize, seqStart, seqSize, nbSeq, isLongOffset, frame);
1353*a28cd43dSSascha Wildner }
1354*a28cd43dSSascha Wildner #endif /* ZSTD_FORCE_DECOMPRESS_SEQUENCES_LONG */
1355*a28cd43dSSascha Wildner 
1356*a28cd43dSSascha Wildner #ifndef ZSTD_FORCE_DECOMPRESS_SEQUENCES_SHORT
1357*a28cd43dSSascha Wildner static TARGET_ATTRIBUTE("bmi2") size_t
ZSTD_decompressSequencesLong_bmi2(ZSTD_DCtx * dctx,void * dst,size_t maxDstSize,const void * seqStart,size_t seqSize,int nbSeq,const ZSTD_longOffset_e isLongOffset,const int frame)1358*a28cd43dSSascha Wildner ZSTD_decompressSequencesLong_bmi2(ZSTD_DCtx* dctx,
1359*a28cd43dSSascha Wildner                                  void* dst, size_t maxDstSize,
1360*a28cd43dSSascha Wildner                            const void* seqStart, size_t seqSize, int nbSeq,
1361*a28cd43dSSascha Wildner                            const ZSTD_longOffset_e isLongOffset,
1362*a28cd43dSSascha Wildner                            const int frame)
1363*a28cd43dSSascha Wildner {
1364*a28cd43dSSascha Wildner     return ZSTD_decompressSequencesLong_body(dctx, dst, maxDstSize, seqStart, seqSize, nbSeq, isLongOffset, frame);
1365*a28cd43dSSascha Wildner }
1366*a28cd43dSSascha Wildner #endif /* ZSTD_FORCE_DECOMPRESS_SEQUENCES_SHORT */
1367*a28cd43dSSascha Wildner 
1368*a28cd43dSSascha Wildner #endif /* DYNAMIC_BMI2 */
1369*a28cd43dSSascha Wildner 
1370*a28cd43dSSascha Wildner typedef size_t (*ZSTD_decompressSequences_t)(
1371*a28cd43dSSascha Wildner                             ZSTD_DCtx* dctx,
1372*a28cd43dSSascha Wildner                             void* dst, size_t maxDstSize,
1373*a28cd43dSSascha Wildner                             const void* seqStart, size_t seqSize, int nbSeq,
1374*a28cd43dSSascha Wildner                             const ZSTD_longOffset_e isLongOffset,
1375*a28cd43dSSascha Wildner                             const int frame);
1376*a28cd43dSSascha Wildner 
1377*a28cd43dSSascha Wildner #ifndef ZSTD_FORCE_DECOMPRESS_SEQUENCES_LONG
1378*a28cd43dSSascha Wildner static size_t
ZSTD_decompressSequences(ZSTD_DCtx * dctx,void * dst,size_t maxDstSize,const void * seqStart,size_t seqSize,int nbSeq,const ZSTD_longOffset_e isLongOffset,const int frame)1379*a28cd43dSSascha Wildner ZSTD_decompressSequences(ZSTD_DCtx* dctx, void* dst, size_t maxDstSize,
1380*a28cd43dSSascha Wildner                    const void* seqStart, size_t seqSize, int nbSeq,
1381*a28cd43dSSascha Wildner                    const ZSTD_longOffset_e isLongOffset,
1382*a28cd43dSSascha Wildner                    const int frame)
1383*a28cd43dSSascha Wildner {
1384*a28cd43dSSascha Wildner     DEBUGLOG(5, "ZSTD_decompressSequences");
1385*a28cd43dSSascha Wildner #if DYNAMIC_BMI2
1386*a28cd43dSSascha Wildner     if (dctx->bmi2) {
1387*a28cd43dSSascha Wildner         return ZSTD_decompressSequences_bmi2(dctx, dst, maxDstSize, seqStart, seqSize, nbSeq, isLongOffset, frame);
1388*a28cd43dSSascha Wildner     }
1389*a28cd43dSSascha Wildner #endif
1390*a28cd43dSSascha Wildner   return ZSTD_decompressSequences_default(dctx, dst, maxDstSize, seqStart, seqSize, nbSeq, isLongOffset, frame);
1391*a28cd43dSSascha Wildner }
1392*a28cd43dSSascha Wildner #endif /* ZSTD_FORCE_DECOMPRESS_SEQUENCES_LONG */
1393*a28cd43dSSascha Wildner 
1394*a28cd43dSSascha Wildner 
1395*a28cd43dSSascha Wildner #ifndef ZSTD_FORCE_DECOMPRESS_SEQUENCES_SHORT
1396*a28cd43dSSascha Wildner /* ZSTD_decompressSequencesLong() :
1397*a28cd43dSSascha Wildner  * decompression function triggered when a minimum share of offsets is considered "long",
1398*a28cd43dSSascha Wildner  * aka out of cache.
1399*a28cd43dSSascha Wildner  * note : "long" definition seems overloaded here, sometimes meaning "wider than bitstream register", and sometimes meaning "farther than memory cache distance".
1400*a28cd43dSSascha Wildner  * This function will try to mitigate main memory latency through the use of prefetching */
1401*a28cd43dSSascha Wildner static size_t
ZSTD_decompressSequencesLong(ZSTD_DCtx * dctx,void * dst,size_t maxDstSize,const void * seqStart,size_t seqSize,int nbSeq,const ZSTD_longOffset_e isLongOffset,const int frame)1402*a28cd43dSSascha Wildner ZSTD_decompressSequencesLong(ZSTD_DCtx* dctx,
1403*a28cd43dSSascha Wildner                              void* dst, size_t maxDstSize,
1404*a28cd43dSSascha Wildner                              const void* seqStart, size_t seqSize, int nbSeq,
1405*a28cd43dSSascha Wildner                              const ZSTD_longOffset_e isLongOffset,
1406*a28cd43dSSascha Wildner                              const int frame)
1407*a28cd43dSSascha Wildner {
1408*a28cd43dSSascha Wildner     DEBUGLOG(5, "ZSTD_decompressSequencesLong");
1409*a28cd43dSSascha Wildner #if DYNAMIC_BMI2
1410*a28cd43dSSascha Wildner     if (dctx->bmi2) {
1411*a28cd43dSSascha Wildner         return ZSTD_decompressSequencesLong_bmi2(dctx, dst, maxDstSize, seqStart, seqSize, nbSeq, isLongOffset, frame);
1412*a28cd43dSSascha Wildner     }
1413*a28cd43dSSascha Wildner #endif
1414*a28cd43dSSascha Wildner   return ZSTD_decompressSequencesLong_default(dctx, dst, maxDstSize, seqStart, seqSize, nbSeq, isLongOffset, frame);
1415*a28cd43dSSascha Wildner }
1416*a28cd43dSSascha Wildner #endif /* ZSTD_FORCE_DECOMPRESS_SEQUENCES_SHORT */
1417*a28cd43dSSascha Wildner 
1418*a28cd43dSSascha Wildner 
1419*a28cd43dSSascha Wildner 
1420*a28cd43dSSascha Wildner #if !defined(ZSTD_FORCE_DECOMPRESS_SEQUENCES_SHORT) && \
1421*a28cd43dSSascha Wildner     !defined(ZSTD_FORCE_DECOMPRESS_SEQUENCES_LONG)
1422*a28cd43dSSascha Wildner /* ZSTD_getLongOffsetsShare() :
1423*a28cd43dSSascha Wildner  * condition : offTable must be valid
1424*a28cd43dSSascha Wildner  * @return : "share" of long offsets (arbitrarily defined as > (1<<23))
1425*a28cd43dSSascha Wildner  *           compared to maximum possible of (1<<OffFSELog) */
1426*a28cd43dSSascha Wildner static unsigned
ZSTD_getLongOffsetsShare(const ZSTD_seqSymbol * offTable)1427*a28cd43dSSascha Wildner ZSTD_getLongOffsetsShare(const ZSTD_seqSymbol* offTable)
1428*a28cd43dSSascha Wildner {
1429*a28cd43dSSascha Wildner     const void* ptr = offTable;
1430*a28cd43dSSascha Wildner     U32 const tableLog = ((const ZSTD_seqSymbol_header*)ptr)[0].tableLog;
1431*a28cd43dSSascha Wildner     const ZSTD_seqSymbol* table = offTable + 1;
1432*a28cd43dSSascha Wildner     U32 const max = 1 << tableLog;
1433*a28cd43dSSascha Wildner     U32 u, total = 0;
1434*a28cd43dSSascha Wildner     DEBUGLOG(5, "ZSTD_getLongOffsetsShare: (tableLog=%u)", tableLog);
1435*a28cd43dSSascha Wildner 
1436*a28cd43dSSascha Wildner     assert(max <= (1 << OffFSELog));  /* max not too large */
1437*a28cd43dSSascha Wildner     for (u=0; u<max; u++) {
1438*a28cd43dSSascha Wildner         if (table[u].nbAdditionalBits > 22) total += 1;
1439*a28cd43dSSascha Wildner     }
1440*a28cd43dSSascha Wildner 
1441*a28cd43dSSascha Wildner     assert(tableLog <= OffFSELog);
1442*a28cd43dSSascha Wildner     total <<= (OffFSELog - tableLog);  /* scale to OffFSELog */
1443*a28cd43dSSascha Wildner 
1444*a28cd43dSSascha Wildner     return total;
1445*a28cd43dSSascha Wildner }
1446*a28cd43dSSascha Wildner #endif
1447*a28cd43dSSascha Wildner 
1448*a28cd43dSSascha Wildner size_t
ZSTD_decompressBlock_internal(ZSTD_DCtx * dctx,void * dst,size_t dstCapacity,const void * src,size_t srcSize,const int frame)1449*a28cd43dSSascha Wildner ZSTD_decompressBlock_internal(ZSTD_DCtx* dctx,
1450*a28cd43dSSascha Wildner                               void* dst, size_t dstCapacity,
1451*a28cd43dSSascha Wildner                         const void* src, size_t srcSize, const int frame)
1452*a28cd43dSSascha Wildner {   /* blockType == blockCompressed */
1453*a28cd43dSSascha Wildner     const BYTE* ip = (const BYTE*)src;
1454*a28cd43dSSascha Wildner     /* isLongOffset must be true if there are long offsets.
1455*a28cd43dSSascha Wildner      * Offsets are long if they are larger than 2^STREAM_ACCUMULATOR_MIN.
1456*a28cd43dSSascha Wildner      * We don't expect that to be the case in 64-bit mode.
1457*a28cd43dSSascha Wildner      * In block mode, window size is not known, so we have to be conservative.
1458*a28cd43dSSascha Wildner      * (note: but it could be evaluated from current-lowLimit)
1459*a28cd43dSSascha Wildner      */
1460*a28cd43dSSascha Wildner     ZSTD_longOffset_e const isLongOffset = (ZSTD_longOffset_e)(MEM_32bits() && (!frame || (dctx->fParams.windowSize > (1ULL << STREAM_ACCUMULATOR_MIN))));
1461*a28cd43dSSascha Wildner     DEBUGLOG(5, "ZSTD_decompressBlock_internal (size : %u)", (U32)srcSize);
1462*a28cd43dSSascha Wildner 
1463*a28cd43dSSascha Wildner     RETURN_ERROR_IF(srcSize >= ZSTD_BLOCKSIZE_MAX, srcSize_wrong, "");
1464*a28cd43dSSascha Wildner 
1465*a28cd43dSSascha Wildner     /* Decode literals section */
1466*a28cd43dSSascha Wildner     {   size_t const litCSize = ZSTD_decodeLiteralsBlock(dctx, src, srcSize);
1467*a28cd43dSSascha Wildner         DEBUGLOG(5, "ZSTD_decodeLiteralsBlock : %u", (U32)litCSize);
1468*a28cd43dSSascha Wildner         if (ZSTD_isError(litCSize)) return litCSize;
1469*a28cd43dSSascha Wildner         ip += litCSize;
1470*a28cd43dSSascha Wildner         srcSize -= litCSize;
1471*a28cd43dSSascha Wildner     }
1472*a28cd43dSSascha Wildner 
1473*a28cd43dSSascha Wildner     /* Build Decoding Tables */
1474*a28cd43dSSascha Wildner     {
1475*a28cd43dSSascha Wildner         /* These macros control at build-time which decompressor implementation
1476*a28cd43dSSascha Wildner          * we use. If neither is defined, we do some inspection and dispatch at
1477*a28cd43dSSascha Wildner          * runtime.
1478*a28cd43dSSascha Wildner          */
1479*a28cd43dSSascha Wildner #if !defined(ZSTD_FORCE_DECOMPRESS_SEQUENCES_SHORT) && \
1480*a28cd43dSSascha Wildner     !defined(ZSTD_FORCE_DECOMPRESS_SEQUENCES_LONG)
1481*a28cd43dSSascha Wildner         int usePrefetchDecoder = dctx->ddictIsCold;
1482*a28cd43dSSascha Wildner #endif
1483*a28cd43dSSascha Wildner         int nbSeq;
1484*a28cd43dSSascha Wildner         size_t const seqHSize = ZSTD_decodeSeqHeaders(dctx, &nbSeq, ip, srcSize);
1485*a28cd43dSSascha Wildner         if (ZSTD_isError(seqHSize)) return seqHSize;
1486*a28cd43dSSascha Wildner         ip += seqHSize;
1487*a28cd43dSSascha Wildner         srcSize -= seqHSize;
1488*a28cd43dSSascha Wildner 
1489*a28cd43dSSascha Wildner         RETURN_ERROR_IF(dst == NULL && nbSeq > 0, dstSize_tooSmall, "NULL not handled");
1490*a28cd43dSSascha Wildner 
1491*a28cd43dSSascha Wildner #if !defined(ZSTD_FORCE_DECOMPRESS_SEQUENCES_SHORT) && \
1492*a28cd43dSSascha Wildner     !defined(ZSTD_FORCE_DECOMPRESS_SEQUENCES_LONG)
1493*a28cd43dSSascha Wildner         if ( !usePrefetchDecoder
1494*a28cd43dSSascha Wildner           && (!frame || (dctx->fParams.windowSize > (1<<24)))
1495*a28cd43dSSascha Wildner           && (nbSeq>ADVANCED_SEQS) ) {  /* could probably use a larger nbSeq limit */
1496*a28cd43dSSascha Wildner             U32 const shareLongOffsets = ZSTD_getLongOffsetsShare(dctx->OFTptr);
1497*a28cd43dSSascha Wildner             U32 const minShare = MEM_64bits() ? 7 : 20; /* heuristic values, correspond to 2.73% and 7.81% */
1498*a28cd43dSSascha Wildner             usePrefetchDecoder = (shareLongOffsets >= minShare);
1499*a28cd43dSSascha Wildner         }
1500*a28cd43dSSascha Wildner #endif
1501*a28cd43dSSascha Wildner 
1502*a28cd43dSSascha Wildner         dctx->ddictIsCold = 0;
1503*a28cd43dSSascha Wildner 
1504*a28cd43dSSascha Wildner #if !defined(ZSTD_FORCE_DECOMPRESS_SEQUENCES_SHORT) && \
1505*a28cd43dSSascha Wildner     !defined(ZSTD_FORCE_DECOMPRESS_SEQUENCES_LONG)
1506*a28cd43dSSascha Wildner         if (usePrefetchDecoder)
1507*a28cd43dSSascha Wildner #endif
1508*a28cd43dSSascha Wildner #ifndef ZSTD_FORCE_DECOMPRESS_SEQUENCES_SHORT
1509*a28cd43dSSascha Wildner             return ZSTD_decompressSequencesLong(dctx, dst, dstCapacity, ip, srcSize, nbSeq, isLongOffset, frame);
1510*a28cd43dSSascha Wildner #endif
1511*a28cd43dSSascha Wildner 
1512*a28cd43dSSascha Wildner #ifndef ZSTD_FORCE_DECOMPRESS_SEQUENCES_LONG
1513*a28cd43dSSascha Wildner         /* else */
1514*a28cd43dSSascha Wildner         return ZSTD_decompressSequences(dctx, dst, dstCapacity, ip, srcSize, nbSeq, isLongOffset, frame);
1515*a28cd43dSSascha Wildner #endif
1516*a28cd43dSSascha Wildner     }
1517*a28cd43dSSascha Wildner }
1518*a28cd43dSSascha Wildner 
1519*a28cd43dSSascha Wildner 
ZSTD_checkContinuity(ZSTD_DCtx * dctx,const void * dst)1520*a28cd43dSSascha Wildner void ZSTD_checkContinuity(ZSTD_DCtx* dctx, const void* dst)
1521*a28cd43dSSascha Wildner {
1522*a28cd43dSSascha Wildner     if (dst != dctx->previousDstEnd) {   /* not contiguous */
1523*a28cd43dSSascha Wildner         dctx->dictEnd = dctx->previousDstEnd;
1524*a28cd43dSSascha Wildner         dctx->virtualStart = (const char*)dst - ((const char*)(dctx->previousDstEnd) - (const char*)(dctx->prefixStart));
1525*a28cd43dSSascha Wildner         dctx->prefixStart = dst;
1526*a28cd43dSSascha Wildner         dctx->previousDstEnd = dst;
1527*a28cd43dSSascha Wildner     }
1528*a28cd43dSSascha Wildner }
1529*a28cd43dSSascha Wildner 
1530*a28cd43dSSascha Wildner 
ZSTD_decompressBlock(ZSTD_DCtx * dctx,void * dst,size_t dstCapacity,const void * src,size_t srcSize)1531*a28cd43dSSascha Wildner size_t ZSTD_decompressBlock(ZSTD_DCtx* dctx,
1532*a28cd43dSSascha Wildner                             void* dst, size_t dstCapacity,
1533*a28cd43dSSascha Wildner                       const void* src, size_t srcSize)
1534*a28cd43dSSascha Wildner {
1535*a28cd43dSSascha Wildner     size_t dSize;
1536*a28cd43dSSascha Wildner     ZSTD_checkContinuity(dctx, dst);
1537*a28cd43dSSascha Wildner     dSize = ZSTD_decompressBlock_internal(dctx, dst, dstCapacity, src, srcSize, /* frame */ 0);
1538*a28cd43dSSascha Wildner     dctx->previousDstEnd = (char*)dst + dSize;
1539*a28cd43dSSascha Wildner     return dSize;
1540*a28cd43dSSascha Wildner }
1541