1*27162e4eSAndroid Build Coastguard Worker /*
2*27162e4eSAndroid Build Coastguard Worker * LZ4 auto-framing library
3*27162e4eSAndroid Build Coastguard Worker * Copyright (C) 2011-2016, Yann Collet.
4*27162e4eSAndroid Build Coastguard Worker *
5*27162e4eSAndroid Build Coastguard Worker * BSD 2-Clause License (http://www.opensource.org/licenses/bsd-license.php)
6*27162e4eSAndroid Build Coastguard Worker *
7*27162e4eSAndroid Build Coastguard Worker * Redistribution and use in source and binary forms, with or without
8*27162e4eSAndroid Build Coastguard Worker * modification, are permitted provided that the following conditions are
9*27162e4eSAndroid Build Coastguard Worker * met:
10*27162e4eSAndroid Build Coastguard Worker *
11*27162e4eSAndroid Build Coastguard Worker * - Redistributions of source code must retain the above copyright
12*27162e4eSAndroid Build Coastguard Worker * notice, this list of conditions and the following disclaimer.
13*27162e4eSAndroid Build Coastguard Worker * - Redistributions in binary form must reproduce the above
14*27162e4eSAndroid Build Coastguard Worker * copyright notice, this list of conditions and the following disclaimer
15*27162e4eSAndroid Build Coastguard Worker * in the documentation and/or other materials provided with the
16*27162e4eSAndroid Build Coastguard Worker * distribution.
17*27162e4eSAndroid Build Coastguard Worker *
18*27162e4eSAndroid Build Coastguard Worker * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19*27162e4eSAndroid Build Coastguard Worker * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20*27162e4eSAndroid Build Coastguard Worker * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21*27162e4eSAndroid Build Coastguard Worker * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22*27162e4eSAndroid Build Coastguard Worker * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23*27162e4eSAndroid Build Coastguard Worker * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24*27162e4eSAndroid Build Coastguard Worker * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25*27162e4eSAndroid Build Coastguard Worker * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26*27162e4eSAndroid Build Coastguard Worker * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27*27162e4eSAndroid Build Coastguard Worker * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28*27162e4eSAndroid Build Coastguard Worker * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29*27162e4eSAndroid Build Coastguard Worker *
30*27162e4eSAndroid Build Coastguard Worker * You can contact the author at :
31*27162e4eSAndroid Build Coastguard Worker * - LZ4 homepage : http://www.lz4.org
32*27162e4eSAndroid Build Coastguard Worker * - LZ4 source repository : https://github.com/lz4/lz4
33*27162e4eSAndroid Build Coastguard Worker */
34*27162e4eSAndroid Build Coastguard Worker
35*27162e4eSAndroid Build Coastguard Worker /* LZ4F is a stand-alone API to create LZ4-compressed Frames
36*27162e4eSAndroid Build Coastguard Worker * in full conformance with specification v1.6.1 .
37*27162e4eSAndroid Build Coastguard Worker * This library rely upon memory management capabilities (malloc, free)
38*27162e4eSAndroid Build Coastguard Worker * provided either by <stdlib.h>,
39*27162e4eSAndroid Build Coastguard Worker * or redirected towards another library of user's choice
40*27162e4eSAndroid Build Coastguard Worker * (see Memory Routines below).
41*27162e4eSAndroid Build Coastguard Worker */
42*27162e4eSAndroid Build Coastguard Worker
43*27162e4eSAndroid Build Coastguard Worker
44*27162e4eSAndroid Build Coastguard Worker /*-************************************
45*27162e4eSAndroid Build Coastguard Worker * Compiler Options
46*27162e4eSAndroid Build Coastguard Worker **************************************/
47*27162e4eSAndroid Build Coastguard Worker #include <limits.h>
48*27162e4eSAndroid Build Coastguard Worker #ifdef _MSC_VER /* Visual Studio */
49*27162e4eSAndroid Build Coastguard Worker # pragma warning(disable : 4127) /* disable: C4127: conditional expression is constant */
50*27162e4eSAndroid Build Coastguard Worker #endif
51*27162e4eSAndroid Build Coastguard Worker
52*27162e4eSAndroid Build Coastguard Worker
53*27162e4eSAndroid Build Coastguard Worker /*-************************************
54*27162e4eSAndroid Build Coastguard Worker * Tuning parameters
55*27162e4eSAndroid Build Coastguard Worker **************************************/
56*27162e4eSAndroid Build Coastguard Worker /*
57*27162e4eSAndroid Build Coastguard Worker * LZ4F_HEAPMODE :
58*27162e4eSAndroid Build Coastguard Worker * Control how LZ4F_compressFrame allocates the Compression State,
59*27162e4eSAndroid Build Coastguard Worker * either on stack (0:default, fastest), or in memory heap (1:requires malloc()).
60*27162e4eSAndroid Build Coastguard Worker */
61*27162e4eSAndroid Build Coastguard Worker #ifndef LZ4F_HEAPMODE
62*27162e4eSAndroid Build Coastguard Worker # define LZ4F_HEAPMODE 0
63*27162e4eSAndroid Build Coastguard Worker #endif
64*27162e4eSAndroid Build Coastguard Worker
65*27162e4eSAndroid Build Coastguard Worker
66*27162e4eSAndroid Build Coastguard Worker /*-************************************
67*27162e4eSAndroid Build Coastguard Worker * Library declarations
68*27162e4eSAndroid Build Coastguard Worker **************************************/
69*27162e4eSAndroid Build Coastguard Worker #define LZ4F_STATIC_LINKING_ONLY
70*27162e4eSAndroid Build Coastguard Worker #include "lz4frame.h"
71*27162e4eSAndroid Build Coastguard Worker #define LZ4_STATIC_LINKING_ONLY
72*27162e4eSAndroid Build Coastguard Worker #include "lz4.h"
73*27162e4eSAndroid Build Coastguard Worker #define LZ4_HC_STATIC_LINKING_ONLY
74*27162e4eSAndroid Build Coastguard Worker #include "lz4hc.h"
75*27162e4eSAndroid Build Coastguard Worker #define XXH_STATIC_LINKING_ONLY
76*27162e4eSAndroid Build Coastguard Worker #include "xxhash.h"
77*27162e4eSAndroid Build Coastguard Worker
78*27162e4eSAndroid Build Coastguard Worker
79*27162e4eSAndroid Build Coastguard Worker /*-************************************
80*27162e4eSAndroid Build Coastguard Worker * Memory routines
81*27162e4eSAndroid Build Coastguard Worker **************************************/
82*27162e4eSAndroid Build Coastguard Worker /*
83*27162e4eSAndroid Build Coastguard Worker * User may redirect invocations of
84*27162e4eSAndroid Build Coastguard Worker * malloc(), calloc() and free()
85*27162e4eSAndroid Build Coastguard Worker * towards another library or solution of their choice
86*27162e4eSAndroid Build Coastguard Worker * by modifying below section.
87*27162e4eSAndroid Build Coastguard Worker **/
88*27162e4eSAndroid Build Coastguard Worker
89*27162e4eSAndroid Build Coastguard Worker #include <string.h> /* memset, memcpy, memmove */
90*27162e4eSAndroid Build Coastguard Worker #ifndef LZ4_SRC_INCLUDED /* avoid redefinition when sources are coalesced */
91*27162e4eSAndroid Build Coastguard Worker # define MEM_INIT(p,v,s) memset((p),(v),(s))
92*27162e4eSAndroid Build Coastguard Worker #endif
93*27162e4eSAndroid Build Coastguard Worker
94*27162e4eSAndroid Build Coastguard Worker #ifndef LZ4_SRC_INCLUDED /* avoid redefinition when sources are coalesced */
95*27162e4eSAndroid Build Coastguard Worker # include <stdlib.h> /* malloc, calloc, free */
96*27162e4eSAndroid Build Coastguard Worker # define ALLOC(s) malloc(s)
97*27162e4eSAndroid Build Coastguard Worker # define ALLOC_AND_ZERO(s) calloc(1,(s))
98*27162e4eSAndroid Build Coastguard Worker # define FREEMEM(p) free(p)
99*27162e4eSAndroid Build Coastguard Worker #endif
100*27162e4eSAndroid Build Coastguard Worker
LZ4F_calloc(size_t s,LZ4F_CustomMem cmem)101*27162e4eSAndroid Build Coastguard Worker static void* LZ4F_calloc(size_t s, LZ4F_CustomMem cmem)
102*27162e4eSAndroid Build Coastguard Worker {
103*27162e4eSAndroid Build Coastguard Worker /* custom calloc defined : use it */
104*27162e4eSAndroid Build Coastguard Worker if (cmem.customCalloc != NULL) {
105*27162e4eSAndroid Build Coastguard Worker return cmem.customCalloc(cmem.opaqueState, s);
106*27162e4eSAndroid Build Coastguard Worker }
107*27162e4eSAndroid Build Coastguard Worker /* nothing defined : use default <stdlib.h>'s calloc() */
108*27162e4eSAndroid Build Coastguard Worker if (cmem.customAlloc == NULL) {
109*27162e4eSAndroid Build Coastguard Worker return ALLOC_AND_ZERO(s);
110*27162e4eSAndroid Build Coastguard Worker }
111*27162e4eSAndroid Build Coastguard Worker /* only custom alloc defined : use it, and combine it with memset() */
112*27162e4eSAndroid Build Coastguard Worker { void* const p = cmem.customAlloc(cmem.opaqueState, s);
113*27162e4eSAndroid Build Coastguard Worker if (p != NULL) MEM_INIT(p, 0, s);
114*27162e4eSAndroid Build Coastguard Worker return p;
115*27162e4eSAndroid Build Coastguard Worker } }
116*27162e4eSAndroid Build Coastguard Worker
LZ4F_malloc(size_t s,LZ4F_CustomMem cmem)117*27162e4eSAndroid Build Coastguard Worker static void* LZ4F_malloc(size_t s, LZ4F_CustomMem cmem)
118*27162e4eSAndroid Build Coastguard Worker {
119*27162e4eSAndroid Build Coastguard Worker /* custom malloc defined : use it */
120*27162e4eSAndroid Build Coastguard Worker if (cmem.customAlloc != NULL) {
121*27162e4eSAndroid Build Coastguard Worker return cmem.customAlloc(cmem.opaqueState, s);
122*27162e4eSAndroid Build Coastguard Worker }
123*27162e4eSAndroid Build Coastguard Worker /* nothing defined : use default <stdlib.h>'s malloc() */
124*27162e4eSAndroid Build Coastguard Worker return ALLOC(s);
125*27162e4eSAndroid Build Coastguard Worker }
126*27162e4eSAndroid Build Coastguard Worker
LZ4F_free(void * p,LZ4F_CustomMem cmem)127*27162e4eSAndroid Build Coastguard Worker static void LZ4F_free(void* p, LZ4F_CustomMem cmem)
128*27162e4eSAndroid Build Coastguard Worker {
129*27162e4eSAndroid Build Coastguard Worker if (p == NULL) return;
130*27162e4eSAndroid Build Coastguard Worker if (cmem.customFree != NULL) {
131*27162e4eSAndroid Build Coastguard Worker /* custom allocation defined : use it */
132*27162e4eSAndroid Build Coastguard Worker cmem.customFree(cmem.opaqueState, p);
133*27162e4eSAndroid Build Coastguard Worker return;
134*27162e4eSAndroid Build Coastguard Worker }
135*27162e4eSAndroid Build Coastguard Worker /* nothing defined : use default <stdlib.h>'s free() */
136*27162e4eSAndroid Build Coastguard Worker FREEMEM(p);
137*27162e4eSAndroid Build Coastguard Worker }
138*27162e4eSAndroid Build Coastguard Worker
139*27162e4eSAndroid Build Coastguard Worker
140*27162e4eSAndroid Build Coastguard Worker /*-************************************
141*27162e4eSAndroid Build Coastguard Worker * Debug
142*27162e4eSAndroid Build Coastguard Worker **************************************/
143*27162e4eSAndroid Build Coastguard Worker #if defined(LZ4_DEBUG) && (LZ4_DEBUG>=1)
144*27162e4eSAndroid Build Coastguard Worker # include <assert.h>
145*27162e4eSAndroid Build Coastguard Worker #else
146*27162e4eSAndroid Build Coastguard Worker # ifndef assert
147*27162e4eSAndroid Build Coastguard Worker # define assert(condition) ((void)0)
148*27162e4eSAndroid Build Coastguard Worker # endif
149*27162e4eSAndroid Build Coastguard Worker #endif
150*27162e4eSAndroid Build Coastguard Worker
151*27162e4eSAndroid Build Coastguard Worker #define LZ4F_STATIC_ASSERT(c) { enum { LZ4F_static_assert = 1/(int)(!!(c)) }; } /* use only *after* variable declarations */
152*27162e4eSAndroid Build Coastguard Worker
153*27162e4eSAndroid Build Coastguard Worker #if defined(LZ4_DEBUG) && (LZ4_DEBUG>=2) && !defined(DEBUGLOG)
154*27162e4eSAndroid Build Coastguard Worker # include <stdio.h>
155*27162e4eSAndroid Build Coastguard Worker static int g_debuglog_enable = 1;
156*27162e4eSAndroid Build Coastguard Worker # define DEBUGLOG(l, ...) { \
157*27162e4eSAndroid Build Coastguard Worker if ((g_debuglog_enable) && (l<=LZ4_DEBUG)) { \
158*27162e4eSAndroid Build Coastguard Worker fprintf(stderr, __FILE__ " (%i): ", __LINE__ ); \
159*27162e4eSAndroid Build Coastguard Worker fprintf(stderr, __VA_ARGS__); \
160*27162e4eSAndroid Build Coastguard Worker fprintf(stderr, " \n"); \
161*27162e4eSAndroid Build Coastguard Worker } }
162*27162e4eSAndroid Build Coastguard Worker #else
163*27162e4eSAndroid Build Coastguard Worker # define DEBUGLOG(l, ...) {} /* disabled */
164*27162e4eSAndroid Build Coastguard Worker #endif
165*27162e4eSAndroid Build Coastguard Worker
166*27162e4eSAndroid Build Coastguard Worker
167*27162e4eSAndroid Build Coastguard Worker /*-************************************
168*27162e4eSAndroid Build Coastguard Worker * Basic Types
169*27162e4eSAndroid Build Coastguard Worker **************************************/
170*27162e4eSAndroid Build Coastguard Worker #if !defined (__VMS) && (defined (__cplusplus) || (defined (__STDC_VERSION__) && (__STDC_VERSION__ >= 199901L) /* C99 */) )
171*27162e4eSAndroid Build Coastguard Worker # include <stdint.h>
172*27162e4eSAndroid Build Coastguard Worker typedef uint8_t BYTE;
173*27162e4eSAndroid Build Coastguard Worker typedef uint16_t U16;
174*27162e4eSAndroid Build Coastguard Worker typedef uint32_t U32;
175*27162e4eSAndroid Build Coastguard Worker typedef int32_t S32;
176*27162e4eSAndroid Build Coastguard Worker typedef uint64_t U64;
177*27162e4eSAndroid Build Coastguard Worker #else
178*27162e4eSAndroid Build Coastguard Worker typedef unsigned char BYTE;
179*27162e4eSAndroid Build Coastguard Worker typedef unsigned short U16;
180*27162e4eSAndroid Build Coastguard Worker typedef unsigned int U32;
181*27162e4eSAndroid Build Coastguard Worker typedef signed int S32;
182*27162e4eSAndroid Build Coastguard Worker typedef unsigned long long U64;
183*27162e4eSAndroid Build Coastguard Worker #endif
184*27162e4eSAndroid Build Coastguard Worker
185*27162e4eSAndroid Build Coastguard Worker
186*27162e4eSAndroid Build Coastguard Worker /* unoptimized version; solves endianness & alignment issues */
LZ4F_readLE32(const void * src)187*27162e4eSAndroid Build Coastguard Worker static U32 LZ4F_readLE32 (const void* src)
188*27162e4eSAndroid Build Coastguard Worker {
189*27162e4eSAndroid Build Coastguard Worker const BYTE* const srcPtr = (const BYTE*)src;
190*27162e4eSAndroid Build Coastguard Worker U32 value32 = srcPtr[0];
191*27162e4eSAndroid Build Coastguard Worker value32 |= ((U32)srcPtr[1])<< 8;
192*27162e4eSAndroid Build Coastguard Worker value32 |= ((U32)srcPtr[2])<<16;
193*27162e4eSAndroid Build Coastguard Worker value32 |= ((U32)srcPtr[3])<<24;
194*27162e4eSAndroid Build Coastguard Worker return value32;
195*27162e4eSAndroid Build Coastguard Worker }
196*27162e4eSAndroid Build Coastguard Worker
LZ4F_writeLE32(void * dst,U32 value32)197*27162e4eSAndroid Build Coastguard Worker static void LZ4F_writeLE32 (void* dst, U32 value32)
198*27162e4eSAndroid Build Coastguard Worker {
199*27162e4eSAndroid Build Coastguard Worker BYTE* const dstPtr = (BYTE*)dst;
200*27162e4eSAndroid Build Coastguard Worker dstPtr[0] = (BYTE)value32;
201*27162e4eSAndroid Build Coastguard Worker dstPtr[1] = (BYTE)(value32 >> 8);
202*27162e4eSAndroid Build Coastguard Worker dstPtr[2] = (BYTE)(value32 >> 16);
203*27162e4eSAndroid Build Coastguard Worker dstPtr[3] = (BYTE)(value32 >> 24);
204*27162e4eSAndroid Build Coastguard Worker }
205*27162e4eSAndroid Build Coastguard Worker
LZ4F_readLE64(const void * src)206*27162e4eSAndroid Build Coastguard Worker static U64 LZ4F_readLE64 (const void* src)
207*27162e4eSAndroid Build Coastguard Worker {
208*27162e4eSAndroid Build Coastguard Worker const BYTE* const srcPtr = (const BYTE*)src;
209*27162e4eSAndroid Build Coastguard Worker U64 value64 = srcPtr[0];
210*27162e4eSAndroid Build Coastguard Worker value64 |= ((U64)srcPtr[1]<<8);
211*27162e4eSAndroid Build Coastguard Worker value64 |= ((U64)srcPtr[2]<<16);
212*27162e4eSAndroid Build Coastguard Worker value64 |= ((U64)srcPtr[3]<<24);
213*27162e4eSAndroid Build Coastguard Worker value64 |= ((U64)srcPtr[4]<<32);
214*27162e4eSAndroid Build Coastguard Worker value64 |= ((U64)srcPtr[5]<<40);
215*27162e4eSAndroid Build Coastguard Worker value64 |= ((U64)srcPtr[6]<<48);
216*27162e4eSAndroid Build Coastguard Worker value64 |= ((U64)srcPtr[7]<<56);
217*27162e4eSAndroid Build Coastguard Worker return value64;
218*27162e4eSAndroid Build Coastguard Worker }
219*27162e4eSAndroid Build Coastguard Worker
LZ4F_writeLE64(void * dst,U64 value64)220*27162e4eSAndroid Build Coastguard Worker static void LZ4F_writeLE64 (void* dst, U64 value64)
221*27162e4eSAndroid Build Coastguard Worker {
222*27162e4eSAndroid Build Coastguard Worker BYTE* const dstPtr = (BYTE*)dst;
223*27162e4eSAndroid Build Coastguard Worker dstPtr[0] = (BYTE)value64;
224*27162e4eSAndroid Build Coastguard Worker dstPtr[1] = (BYTE)(value64 >> 8);
225*27162e4eSAndroid Build Coastguard Worker dstPtr[2] = (BYTE)(value64 >> 16);
226*27162e4eSAndroid Build Coastguard Worker dstPtr[3] = (BYTE)(value64 >> 24);
227*27162e4eSAndroid Build Coastguard Worker dstPtr[4] = (BYTE)(value64 >> 32);
228*27162e4eSAndroid Build Coastguard Worker dstPtr[5] = (BYTE)(value64 >> 40);
229*27162e4eSAndroid Build Coastguard Worker dstPtr[6] = (BYTE)(value64 >> 48);
230*27162e4eSAndroid Build Coastguard Worker dstPtr[7] = (BYTE)(value64 >> 56);
231*27162e4eSAndroid Build Coastguard Worker }
232*27162e4eSAndroid Build Coastguard Worker
233*27162e4eSAndroid Build Coastguard Worker
234*27162e4eSAndroid Build Coastguard Worker /*-************************************
235*27162e4eSAndroid Build Coastguard Worker * Constants
236*27162e4eSAndroid Build Coastguard Worker **************************************/
237*27162e4eSAndroid Build Coastguard Worker #ifndef LZ4_SRC_INCLUDED /* avoid double definition */
238*27162e4eSAndroid Build Coastguard Worker # define KB *(1<<10)
239*27162e4eSAndroid Build Coastguard Worker # define MB *(1<<20)
240*27162e4eSAndroid Build Coastguard Worker # define GB *(1<<30)
241*27162e4eSAndroid Build Coastguard Worker #endif
242*27162e4eSAndroid Build Coastguard Worker
243*27162e4eSAndroid Build Coastguard Worker #define _1BIT 0x01
244*27162e4eSAndroid Build Coastguard Worker #define _2BITS 0x03
245*27162e4eSAndroid Build Coastguard Worker #define _3BITS 0x07
246*27162e4eSAndroid Build Coastguard Worker #define _4BITS 0x0F
247*27162e4eSAndroid Build Coastguard Worker #define _8BITS 0xFF
248*27162e4eSAndroid Build Coastguard Worker
249*27162e4eSAndroid Build Coastguard Worker #define LZ4F_BLOCKUNCOMPRESSED_FLAG 0x80000000U
250*27162e4eSAndroid Build Coastguard Worker #define LZ4F_BLOCKSIZEID_DEFAULT LZ4F_max64KB
251*27162e4eSAndroid Build Coastguard Worker
252*27162e4eSAndroid Build Coastguard Worker static const size_t minFHSize = LZ4F_HEADER_SIZE_MIN; /* 7 */
253*27162e4eSAndroid Build Coastguard Worker static const size_t maxFHSize = LZ4F_HEADER_SIZE_MAX; /* 19 */
254*27162e4eSAndroid Build Coastguard Worker static const size_t BHSize = LZ4F_BLOCK_HEADER_SIZE; /* block header : size, and compress flag */
255*27162e4eSAndroid Build Coastguard Worker static const size_t BFSize = LZ4F_BLOCK_CHECKSUM_SIZE; /* block footer : checksum (optional) */
256*27162e4eSAndroid Build Coastguard Worker
257*27162e4eSAndroid Build Coastguard Worker
258*27162e4eSAndroid Build Coastguard Worker /*-************************************
259*27162e4eSAndroid Build Coastguard Worker * Structures and local types
260*27162e4eSAndroid Build Coastguard Worker **************************************/
261*27162e4eSAndroid Build Coastguard Worker
262*27162e4eSAndroid Build Coastguard Worker typedef enum { LZ4B_COMPRESSED, LZ4B_UNCOMPRESSED} LZ4F_BlockCompressMode_e;
263*27162e4eSAndroid Build Coastguard Worker typedef enum { ctxNone, ctxFast, ctxHC } LZ4F_CtxType_e;
264*27162e4eSAndroid Build Coastguard Worker
265*27162e4eSAndroid Build Coastguard Worker typedef struct LZ4F_cctx_s
266*27162e4eSAndroid Build Coastguard Worker {
267*27162e4eSAndroid Build Coastguard Worker LZ4F_CustomMem cmem;
268*27162e4eSAndroid Build Coastguard Worker LZ4F_preferences_t prefs;
269*27162e4eSAndroid Build Coastguard Worker U32 version;
270*27162e4eSAndroid Build Coastguard Worker U32 cStage; /* 0 : compression uninitialized ; 1 : initialized, can compress */
271*27162e4eSAndroid Build Coastguard Worker const LZ4F_CDict* cdict;
272*27162e4eSAndroid Build Coastguard Worker size_t maxBlockSize;
273*27162e4eSAndroid Build Coastguard Worker size_t maxBufferSize;
274*27162e4eSAndroid Build Coastguard Worker BYTE* tmpBuff; /* internal buffer, for streaming */
275*27162e4eSAndroid Build Coastguard Worker BYTE* tmpIn; /* starting position of data compress within internal buffer (>= tmpBuff) */
276*27162e4eSAndroid Build Coastguard Worker size_t tmpInSize; /* amount of data to compress after tmpIn */
277*27162e4eSAndroid Build Coastguard Worker U64 totalInSize;
278*27162e4eSAndroid Build Coastguard Worker XXH32_state_t xxh;
279*27162e4eSAndroid Build Coastguard Worker void* lz4CtxPtr;
280*27162e4eSAndroid Build Coastguard Worker U16 lz4CtxAlloc; /* sized for: 0 = none, 1 = lz4 ctx, 2 = lz4hc ctx */
281*27162e4eSAndroid Build Coastguard Worker U16 lz4CtxType; /* in use as: 0 = none, 1 = lz4 ctx, 2 = lz4hc ctx */
282*27162e4eSAndroid Build Coastguard Worker LZ4F_BlockCompressMode_e blockCompressMode;
283*27162e4eSAndroid Build Coastguard Worker } LZ4F_cctx_t;
284*27162e4eSAndroid Build Coastguard Worker
285*27162e4eSAndroid Build Coastguard Worker
286*27162e4eSAndroid Build Coastguard Worker /*-************************************
287*27162e4eSAndroid Build Coastguard Worker * Error management
288*27162e4eSAndroid Build Coastguard Worker **************************************/
289*27162e4eSAndroid Build Coastguard Worker #define LZ4F_GENERATE_STRING(STRING) #STRING,
290*27162e4eSAndroid Build Coastguard Worker static const char* LZ4F_errorStrings[] = { LZ4F_LIST_ERRORS(LZ4F_GENERATE_STRING) };
291*27162e4eSAndroid Build Coastguard Worker
292*27162e4eSAndroid Build Coastguard Worker
LZ4F_isError(LZ4F_errorCode_t code)293*27162e4eSAndroid Build Coastguard Worker unsigned LZ4F_isError(LZ4F_errorCode_t code)
294*27162e4eSAndroid Build Coastguard Worker {
295*27162e4eSAndroid Build Coastguard Worker return (code > (LZ4F_errorCode_t)(-LZ4F_ERROR_maxCode));
296*27162e4eSAndroid Build Coastguard Worker }
297*27162e4eSAndroid Build Coastguard Worker
LZ4F_getErrorName(LZ4F_errorCode_t code)298*27162e4eSAndroid Build Coastguard Worker const char* LZ4F_getErrorName(LZ4F_errorCode_t code)
299*27162e4eSAndroid Build Coastguard Worker {
300*27162e4eSAndroid Build Coastguard Worker static const char* codeError = "Unspecified error code";
301*27162e4eSAndroid Build Coastguard Worker if (LZ4F_isError(code)) return LZ4F_errorStrings[-(int)(code)];
302*27162e4eSAndroid Build Coastguard Worker return codeError;
303*27162e4eSAndroid Build Coastguard Worker }
304*27162e4eSAndroid Build Coastguard Worker
LZ4F_getErrorCode(size_t functionResult)305*27162e4eSAndroid Build Coastguard Worker LZ4F_errorCodes LZ4F_getErrorCode(size_t functionResult)
306*27162e4eSAndroid Build Coastguard Worker {
307*27162e4eSAndroid Build Coastguard Worker if (!LZ4F_isError(functionResult)) return LZ4F_OK_NoError;
308*27162e4eSAndroid Build Coastguard Worker return (LZ4F_errorCodes)(-(ptrdiff_t)functionResult);
309*27162e4eSAndroid Build Coastguard Worker }
310*27162e4eSAndroid Build Coastguard Worker
LZ4F_returnErrorCode(LZ4F_errorCodes code)311*27162e4eSAndroid Build Coastguard Worker static LZ4F_errorCode_t LZ4F_returnErrorCode(LZ4F_errorCodes code)
312*27162e4eSAndroid Build Coastguard Worker {
313*27162e4eSAndroid Build Coastguard Worker /* A compilation error here means sizeof(ptrdiff_t) is not large enough */
314*27162e4eSAndroid Build Coastguard Worker LZ4F_STATIC_ASSERT(sizeof(ptrdiff_t) >= sizeof(size_t));
315*27162e4eSAndroid Build Coastguard Worker return (LZ4F_errorCode_t)-(ptrdiff_t)code;
316*27162e4eSAndroid Build Coastguard Worker }
317*27162e4eSAndroid Build Coastguard Worker
318*27162e4eSAndroid Build Coastguard Worker #define RETURN_ERROR(e) return LZ4F_returnErrorCode(LZ4F_ERROR_ ## e)
319*27162e4eSAndroid Build Coastguard Worker
320*27162e4eSAndroid Build Coastguard Worker #define RETURN_ERROR_IF(c,e) do { \
321*27162e4eSAndroid Build Coastguard Worker if (c) { \
322*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(3, "Error: " #c); \
323*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR(e); \
324*27162e4eSAndroid Build Coastguard Worker } \
325*27162e4eSAndroid Build Coastguard Worker } while (0)
326*27162e4eSAndroid Build Coastguard Worker
327*27162e4eSAndroid Build Coastguard Worker #define FORWARD_IF_ERROR(r) do { if (LZ4F_isError(r)) return (r); } while (0)
328*27162e4eSAndroid Build Coastguard Worker
LZ4F_getVersion(void)329*27162e4eSAndroid Build Coastguard Worker unsigned LZ4F_getVersion(void) { return LZ4F_VERSION; }
330*27162e4eSAndroid Build Coastguard Worker
LZ4F_compressionLevel_max(void)331*27162e4eSAndroid Build Coastguard Worker int LZ4F_compressionLevel_max(void) { return LZ4HC_CLEVEL_MAX; }
332*27162e4eSAndroid Build Coastguard Worker
LZ4F_getBlockSize(LZ4F_blockSizeID_t blockSizeID)333*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_getBlockSize(LZ4F_blockSizeID_t blockSizeID)
334*27162e4eSAndroid Build Coastguard Worker {
335*27162e4eSAndroid Build Coastguard Worker static const size_t blockSizes[4] = { 64 KB, 256 KB, 1 MB, 4 MB };
336*27162e4eSAndroid Build Coastguard Worker
337*27162e4eSAndroid Build Coastguard Worker if (blockSizeID == 0) blockSizeID = LZ4F_BLOCKSIZEID_DEFAULT;
338*27162e4eSAndroid Build Coastguard Worker if (blockSizeID < LZ4F_max64KB || blockSizeID > LZ4F_max4MB)
339*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR(maxBlockSize_invalid);
340*27162e4eSAndroid Build Coastguard Worker { int const blockSizeIdx = (int)blockSizeID - (int)LZ4F_max64KB;
341*27162e4eSAndroid Build Coastguard Worker return blockSizes[blockSizeIdx];
342*27162e4eSAndroid Build Coastguard Worker } }
343*27162e4eSAndroid Build Coastguard Worker
344*27162e4eSAndroid Build Coastguard Worker /*-************************************
345*27162e4eSAndroid Build Coastguard Worker * Private functions
346*27162e4eSAndroid Build Coastguard Worker **************************************/
347*27162e4eSAndroid Build Coastguard Worker #define MIN(a,b) ( (a) < (b) ? (a) : (b) )
348*27162e4eSAndroid Build Coastguard Worker
LZ4F_headerChecksum(const void * header,size_t length)349*27162e4eSAndroid Build Coastguard Worker static BYTE LZ4F_headerChecksum (const void* header, size_t length)
350*27162e4eSAndroid Build Coastguard Worker {
351*27162e4eSAndroid Build Coastguard Worker U32 const xxh = XXH32(header, length, 0);
352*27162e4eSAndroid Build Coastguard Worker return (BYTE)(xxh >> 8);
353*27162e4eSAndroid Build Coastguard Worker }
354*27162e4eSAndroid Build Coastguard Worker
355*27162e4eSAndroid Build Coastguard Worker
356*27162e4eSAndroid Build Coastguard Worker /*-************************************
357*27162e4eSAndroid Build Coastguard Worker * Simple-pass compression functions
358*27162e4eSAndroid Build Coastguard Worker **************************************/
LZ4F_optimalBSID(const LZ4F_blockSizeID_t requestedBSID,const size_t srcSize)359*27162e4eSAndroid Build Coastguard Worker static LZ4F_blockSizeID_t LZ4F_optimalBSID(const LZ4F_blockSizeID_t requestedBSID,
360*27162e4eSAndroid Build Coastguard Worker const size_t srcSize)
361*27162e4eSAndroid Build Coastguard Worker {
362*27162e4eSAndroid Build Coastguard Worker LZ4F_blockSizeID_t proposedBSID = LZ4F_max64KB;
363*27162e4eSAndroid Build Coastguard Worker size_t maxBlockSize = 64 KB;
364*27162e4eSAndroid Build Coastguard Worker while (requestedBSID > proposedBSID) {
365*27162e4eSAndroid Build Coastguard Worker if (srcSize <= maxBlockSize)
366*27162e4eSAndroid Build Coastguard Worker return proposedBSID;
367*27162e4eSAndroid Build Coastguard Worker proposedBSID = (LZ4F_blockSizeID_t)((int)proposedBSID + 1);
368*27162e4eSAndroid Build Coastguard Worker maxBlockSize <<= 2;
369*27162e4eSAndroid Build Coastguard Worker }
370*27162e4eSAndroid Build Coastguard Worker return requestedBSID;
371*27162e4eSAndroid Build Coastguard Worker }
372*27162e4eSAndroid Build Coastguard Worker
373*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_compressBound_internal() :
374*27162e4eSAndroid Build Coastguard Worker * Provides dstCapacity given a srcSize to guarantee operation success in worst case situations.
375*27162e4eSAndroid Build Coastguard Worker * prefsPtr is optional : if NULL is provided, preferences will be set to cover worst case scenario.
376*27162e4eSAndroid Build Coastguard Worker * @return is always the same for a srcSize and prefsPtr, so it can be relied upon to size reusable buffers.
377*27162e4eSAndroid Build Coastguard Worker * When srcSize==0, LZ4F_compressBound() provides an upper bound for LZ4F_flush() and LZ4F_compressEnd() operations.
378*27162e4eSAndroid Build Coastguard Worker */
LZ4F_compressBound_internal(size_t srcSize,const LZ4F_preferences_t * preferencesPtr,size_t alreadyBuffered)379*27162e4eSAndroid Build Coastguard Worker static size_t LZ4F_compressBound_internal(size_t srcSize,
380*27162e4eSAndroid Build Coastguard Worker const LZ4F_preferences_t* preferencesPtr,
381*27162e4eSAndroid Build Coastguard Worker size_t alreadyBuffered)
382*27162e4eSAndroid Build Coastguard Worker {
383*27162e4eSAndroid Build Coastguard Worker LZ4F_preferences_t prefsNull = LZ4F_INIT_PREFERENCES;
384*27162e4eSAndroid Build Coastguard Worker prefsNull.frameInfo.contentChecksumFlag = LZ4F_contentChecksumEnabled; /* worst case */
385*27162e4eSAndroid Build Coastguard Worker prefsNull.frameInfo.blockChecksumFlag = LZ4F_blockChecksumEnabled; /* worst case */
386*27162e4eSAndroid Build Coastguard Worker { const LZ4F_preferences_t* const prefsPtr = (preferencesPtr==NULL) ? &prefsNull : preferencesPtr;
387*27162e4eSAndroid Build Coastguard Worker U32 const flush = prefsPtr->autoFlush | (srcSize==0);
388*27162e4eSAndroid Build Coastguard Worker LZ4F_blockSizeID_t const blockID = prefsPtr->frameInfo.blockSizeID;
389*27162e4eSAndroid Build Coastguard Worker size_t const blockSize = LZ4F_getBlockSize(blockID);
390*27162e4eSAndroid Build Coastguard Worker size_t const maxBuffered = blockSize - 1;
391*27162e4eSAndroid Build Coastguard Worker size_t const bufferedSize = MIN(alreadyBuffered, maxBuffered);
392*27162e4eSAndroid Build Coastguard Worker size_t const maxSrcSize = srcSize + bufferedSize;
393*27162e4eSAndroid Build Coastguard Worker unsigned const nbFullBlocks = (unsigned)(maxSrcSize / blockSize);
394*27162e4eSAndroid Build Coastguard Worker size_t const partialBlockSize = maxSrcSize & (blockSize-1);
395*27162e4eSAndroid Build Coastguard Worker size_t const lastBlockSize = flush ? partialBlockSize : 0;
396*27162e4eSAndroid Build Coastguard Worker unsigned const nbBlocks = nbFullBlocks + (lastBlockSize>0);
397*27162e4eSAndroid Build Coastguard Worker
398*27162e4eSAndroid Build Coastguard Worker size_t const blockCRCSize = BFSize * prefsPtr->frameInfo.blockChecksumFlag;
399*27162e4eSAndroid Build Coastguard Worker size_t const frameEnd = BHSize + (prefsPtr->frameInfo.contentChecksumFlag*BFSize);
400*27162e4eSAndroid Build Coastguard Worker
401*27162e4eSAndroid Build Coastguard Worker return ((BHSize + blockCRCSize) * nbBlocks) +
402*27162e4eSAndroid Build Coastguard Worker (blockSize * nbFullBlocks) + lastBlockSize + frameEnd;
403*27162e4eSAndroid Build Coastguard Worker }
404*27162e4eSAndroid Build Coastguard Worker }
405*27162e4eSAndroid Build Coastguard Worker
LZ4F_compressFrameBound(size_t srcSize,const LZ4F_preferences_t * preferencesPtr)406*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_compressFrameBound(size_t srcSize, const LZ4F_preferences_t* preferencesPtr)
407*27162e4eSAndroid Build Coastguard Worker {
408*27162e4eSAndroid Build Coastguard Worker LZ4F_preferences_t prefs;
409*27162e4eSAndroid Build Coastguard Worker size_t const headerSize = maxFHSize; /* max header size, including optional fields */
410*27162e4eSAndroid Build Coastguard Worker
411*27162e4eSAndroid Build Coastguard Worker if (preferencesPtr!=NULL) prefs = *preferencesPtr;
412*27162e4eSAndroid Build Coastguard Worker else MEM_INIT(&prefs, 0, sizeof(prefs));
413*27162e4eSAndroid Build Coastguard Worker prefs.autoFlush = 1;
414*27162e4eSAndroid Build Coastguard Worker
415*27162e4eSAndroid Build Coastguard Worker return headerSize + LZ4F_compressBound_internal(srcSize, &prefs, 0);;
416*27162e4eSAndroid Build Coastguard Worker }
417*27162e4eSAndroid Build Coastguard Worker
418*27162e4eSAndroid Build Coastguard Worker
419*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_compressFrame_usingCDict() :
420*27162e4eSAndroid Build Coastguard Worker * Compress srcBuffer using a dictionary, in a single step.
421*27162e4eSAndroid Build Coastguard Worker * cdict can be NULL, in which case, no dictionary is used.
422*27162e4eSAndroid Build Coastguard Worker * dstBuffer MUST be >= LZ4F_compressFrameBound(srcSize, preferencesPtr).
423*27162e4eSAndroid Build Coastguard Worker * The LZ4F_preferences_t structure is optional : you may provide NULL as argument,
424*27162e4eSAndroid Build Coastguard Worker * however, it's the only way to provide a dictID, so it's not recommended.
425*27162e4eSAndroid Build Coastguard Worker * @return : number of bytes written into dstBuffer,
426*27162e4eSAndroid Build Coastguard Worker * or an error code if it fails (can be tested using LZ4F_isError())
427*27162e4eSAndroid Build Coastguard Worker */
LZ4F_compressFrame_usingCDict(LZ4F_cctx * cctx,void * dstBuffer,size_t dstCapacity,const void * srcBuffer,size_t srcSize,const LZ4F_CDict * cdict,const LZ4F_preferences_t * preferencesPtr)428*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_compressFrame_usingCDict(LZ4F_cctx* cctx,
429*27162e4eSAndroid Build Coastguard Worker void* dstBuffer, size_t dstCapacity,
430*27162e4eSAndroid Build Coastguard Worker const void* srcBuffer, size_t srcSize,
431*27162e4eSAndroid Build Coastguard Worker const LZ4F_CDict* cdict,
432*27162e4eSAndroid Build Coastguard Worker const LZ4F_preferences_t* preferencesPtr)
433*27162e4eSAndroid Build Coastguard Worker {
434*27162e4eSAndroid Build Coastguard Worker LZ4F_preferences_t prefs;
435*27162e4eSAndroid Build Coastguard Worker LZ4F_compressOptions_t options;
436*27162e4eSAndroid Build Coastguard Worker BYTE* const dstStart = (BYTE*) dstBuffer;
437*27162e4eSAndroid Build Coastguard Worker BYTE* dstPtr = dstStart;
438*27162e4eSAndroid Build Coastguard Worker BYTE* const dstEnd = dstStart + dstCapacity;
439*27162e4eSAndroid Build Coastguard Worker
440*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(4, "LZ4F_compressFrame_usingCDict (srcSize=%u)", (unsigned)srcSize);
441*27162e4eSAndroid Build Coastguard Worker if (preferencesPtr!=NULL)
442*27162e4eSAndroid Build Coastguard Worker prefs = *preferencesPtr;
443*27162e4eSAndroid Build Coastguard Worker else
444*27162e4eSAndroid Build Coastguard Worker MEM_INIT(&prefs, 0, sizeof(prefs));
445*27162e4eSAndroid Build Coastguard Worker if (prefs.frameInfo.contentSize != 0)
446*27162e4eSAndroid Build Coastguard Worker prefs.frameInfo.contentSize = (U64)srcSize; /* auto-correct content size if selected (!=0) */
447*27162e4eSAndroid Build Coastguard Worker
448*27162e4eSAndroid Build Coastguard Worker prefs.frameInfo.blockSizeID = LZ4F_optimalBSID(prefs.frameInfo.blockSizeID, srcSize);
449*27162e4eSAndroid Build Coastguard Worker prefs.autoFlush = 1;
450*27162e4eSAndroid Build Coastguard Worker if (srcSize <= LZ4F_getBlockSize(prefs.frameInfo.blockSizeID))
451*27162e4eSAndroid Build Coastguard Worker prefs.frameInfo.blockMode = LZ4F_blockIndependent; /* only one block => no need for inter-block link */
452*27162e4eSAndroid Build Coastguard Worker
453*27162e4eSAndroid Build Coastguard Worker MEM_INIT(&options, 0, sizeof(options));
454*27162e4eSAndroid Build Coastguard Worker options.stableSrc = 1;
455*27162e4eSAndroid Build Coastguard Worker
456*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(dstCapacity < LZ4F_compressFrameBound(srcSize, &prefs), dstMaxSize_tooSmall);
457*27162e4eSAndroid Build Coastguard Worker
458*27162e4eSAndroid Build Coastguard Worker { size_t const headerSize = LZ4F_compressBegin_usingCDict(cctx, dstBuffer, dstCapacity, cdict, &prefs); /* write header */
459*27162e4eSAndroid Build Coastguard Worker FORWARD_IF_ERROR(headerSize);
460*27162e4eSAndroid Build Coastguard Worker dstPtr += headerSize; /* header size */ }
461*27162e4eSAndroid Build Coastguard Worker
462*27162e4eSAndroid Build Coastguard Worker assert(dstEnd >= dstPtr);
463*27162e4eSAndroid Build Coastguard Worker { size_t const cSize = LZ4F_compressUpdate(cctx, dstPtr, (size_t)(dstEnd-dstPtr), srcBuffer, srcSize, &options);
464*27162e4eSAndroid Build Coastguard Worker FORWARD_IF_ERROR(cSize);
465*27162e4eSAndroid Build Coastguard Worker dstPtr += cSize; }
466*27162e4eSAndroid Build Coastguard Worker
467*27162e4eSAndroid Build Coastguard Worker assert(dstEnd >= dstPtr);
468*27162e4eSAndroid Build Coastguard Worker { size_t const tailSize = LZ4F_compressEnd(cctx, dstPtr, (size_t)(dstEnd-dstPtr), &options); /* flush last block, and generate suffix */
469*27162e4eSAndroid Build Coastguard Worker FORWARD_IF_ERROR(tailSize);
470*27162e4eSAndroid Build Coastguard Worker dstPtr += tailSize; }
471*27162e4eSAndroid Build Coastguard Worker
472*27162e4eSAndroid Build Coastguard Worker assert(dstEnd >= dstStart);
473*27162e4eSAndroid Build Coastguard Worker return (size_t)(dstPtr - dstStart);
474*27162e4eSAndroid Build Coastguard Worker }
475*27162e4eSAndroid Build Coastguard Worker
476*27162e4eSAndroid Build Coastguard Worker
477*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_compressFrame() :
478*27162e4eSAndroid Build Coastguard Worker * Compress an entire srcBuffer into a valid LZ4 frame, in a single step.
479*27162e4eSAndroid Build Coastguard Worker * dstBuffer MUST be >= LZ4F_compressFrameBound(srcSize, preferencesPtr).
480*27162e4eSAndroid Build Coastguard Worker * The LZ4F_preferences_t structure is optional : you can provide NULL as argument. All preferences will be set to default.
481*27162e4eSAndroid Build Coastguard Worker * @return : number of bytes written into dstBuffer.
482*27162e4eSAndroid Build Coastguard Worker * or an error code if it fails (can be tested using LZ4F_isError())
483*27162e4eSAndroid Build Coastguard Worker */
LZ4F_compressFrame(void * dstBuffer,size_t dstCapacity,const void * srcBuffer,size_t srcSize,const LZ4F_preferences_t * preferencesPtr)484*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_compressFrame(void* dstBuffer, size_t dstCapacity,
485*27162e4eSAndroid Build Coastguard Worker const void* srcBuffer, size_t srcSize,
486*27162e4eSAndroid Build Coastguard Worker const LZ4F_preferences_t* preferencesPtr)
487*27162e4eSAndroid Build Coastguard Worker {
488*27162e4eSAndroid Build Coastguard Worker size_t result;
489*27162e4eSAndroid Build Coastguard Worker #if (LZ4F_HEAPMODE)
490*27162e4eSAndroid Build Coastguard Worker LZ4F_cctx_t* cctxPtr;
491*27162e4eSAndroid Build Coastguard Worker result = LZ4F_createCompressionContext(&cctxPtr, LZ4F_VERSION);
492*27162e4eSAndroid Build Coastguard Worker FORWARD_IF_ERROR(result);
493*27162e4eSAndroid Build Coastguard Worker #else
494*27162e4eSAndroid Build Coastguard Worker LZ4F_cctx_t cctx;
495*27162e4eSAndroid Build Coastguard Worker LZ4_stream_t lz4ctx;
496*27162e4eSAndroid Build Coastguard Worker LZ4F_cctx_t* const cctxPtr = &cctx;
497*27162e4eSAndroid Build Coastguard Worker
498*27162e4eSAndroid Build Coastguard Worker MEM_INIT(&cctx, 0, sizeof(cctx));
499*27162e4eSAndroid Build Coastguard Worker cctx.version = LZ4F_VERSION;
500*27162e4eSAndroid Build Coastguard Worker cctx.maxBufferSize = 5 MB; /* mess with real buffer size to prevent dynamic allocation; works only because autoflush==1 & stableSrc==1 */
501*27162e4eSAndroid Build Coastguard Worker if ( preferencesPtr == NULL
502*27162e4eSAndroid Build Coastguard Worker || preferencesPtr->compressionLevel < LZ4HC_CLEVEL_MIN ) {
503*27162e4eSAndroid Build Coastguard Worker LZ4_initStream(&lz4ctx, sizeof(lz4ctx));
504*27162e4eSAndroid Build Coastguard Worker cctxPtr->lz4CtxPtr = &lz4ctx;
505*27162e4eSAndroid Build Coastguard Worker cctxPtr->lz4CtxAlloc = 1;
506*27162e4eSAndroid Build Coastguard Worker cctxPtr->lz4CtxType = ctxFast;
507*27162e4eSAndroid Build Coastguard Worker }
508*27162e4eSAndroid Build Coastguard Worker #endif
509*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(4, "LZ4F_compressFrame");
510*27162e4eSAndroid Build Coastguard Worker
511*27162e4eSAndroid Build Coastguard Worker result = LZ4F_compressFrame_usingCDict(cctxPtr, dstBuffer, dstCapacity,
512*27162e4eSAndroid Build Coastguard Worker srcBuffer, srcSize,
513*27162e4eSAndroid Build Coastguard Worker NULL, preferencesPtr);
514*27162e4eSAndroid Build Coastguard Worker
515*27162e4eSAndroid Build Coastguard Worker #if (LZ4F_HEAPMODE)
516*27162e4eSAndroid Build Coastguard Worker LZ4F_freeCompressionContext(cctxPtr);
517*27162e4eSAndroid Build Coastguard Worker #else
518*27162e4eSAndroid Build Coastguard Worker if ( preferencesPtr != NULL
519*27162e4eSAndroid Build Coastguard Worker && preferencesPtr->compressionLevel >= LZ4HC_CLEVEL_MIN ) {
520*27162e4eSAndroid Build Coastguard Worker LZ4F_free(cctxPtr->lz4CtxPtr, cctxPtr->cmem);
521*27162e4eSAndroid Build Coastguard Worker }
522*27162e4eSAndroid Build Coastguard Worker #endif
523*27162e4eSAndroid Build Coastguard Worker return result;
524*27162e4eSAndroid Build Coastguard Worker }
525*27162e4eSAndroid Build Coastguard Worker
526*27162e4eSAndroid Build Coastguard Worker
527*27162e4eSAndroid Build Coastguard Worker /*-***************************************************
528*27162e4eSAndroid Build Coastguard Worker * Dictionary compression
529*27162e4eSAndroid Build Coastguard Worker *****************************************************/
530*27162e4eSAndroid Build Coastguard Worker
531*27162e4eSAndroid Build Coastguard Worker struct LZ4F_CDict_s {
532*27162e4eSAndroid Build Coastguard Worker LZ4F_CustomMem cmem;
533*27162e4eSAndroid Build Coastguard Worker void* dictContent;
534*27162e4eSAndroid Build Coastguard Worker LZ4_stream_t* fastCtx;
535*27162e4eSAndroid Build Coastguard Worker LZ4_streamHC_t* HCCtx;
536*27162e4eSAndroid Build Coastguard Worker }; /* typedef'd to LZ4F_CDict within lz4frame_static.h */
537*27162e4eSAndroid Build Coastguard Worker
538*27162e4eSAndroid Build Coastguard Worker LZ4F_CDict*
LZ4F_createCDict_advanced(LZ4F_CustomMem cmem,const void * dictBuffer,size_t dictSize)539*27162e4eSAndroid Build Coastguard Worker LZ4F_createCDict_advanced(LZ4F_CustomMem cmem, const void* dictBuffer, size_t dictSize)
540*27162e4eSAndroid Build Coastguard Worker {
541*27162e4eSAndroid Build Coastguard Worker const char* dictStart = (const char*)dictBuffer;
542*27162e4eSAndroid Build Coastguard Worker LZ4F_CDict* const cdict = (LZ4F_CDict*)LZ4F_malloc(sizeof(*cdict), cmem);
543*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(4, "LZ4F_createCDict_advanced");
544*27162e4eSAndroid Build Coastguard Worker if (!cdict) return NULL;
545*27162e4eSAndroid Build Coastguard Worker cdict->cmem = cmem;
546*27162e4eSAndroid Build Coastguard Worker if (dictSize > 64 KB) {
547*27162e4eSAndroid Build Coastguard Worker dictStart += dictSize - 64 KB;
548*27162e4eSAndroid Build Coastguard Worker dictSize = 64 KB;
549*27162e4eSAndroid Build Coastguard Worker }
550*27162e4eSAndroid Build Coastguard Worker cdict->dictContent = LZ4F_malloc(dictSize, cmem);
551*27162e4eSAndroid Build Coastguard Worker /* note: using @cmem to allocate => can't use default create */
552*27162e4eSAndroid Build Coastguard Worker cdict->fastCtx = (LZ4_stream_t*)LZ4F_malloc(sizeof(LZ4_stream_t), cmem);
553*27162e4eSAndroid Build Coastguard Worker cdict->HCCtx = (LZ4_streamHC_t*)LZ4F_malloc(sizeof(LZ4_streamHC_t), cmem);
554*27162e4eSAndroid Build Coastguard Worker if (!cdict->dictContent || !cdict->fastCtx || !cdict->HCCtx) {
555*27162e4eSAndroid Build Coastguard Worker LZ4F_freeCDict(cdict);
556*27162e4eSAndroid Build Coastguard Worker return NULL;
557*27162e4eSAndroid Build Coastguard Worker }
558*27162e4eSAndroid Build Coastguard Worker memcpy(cdict->dictContent, dictStart, dictSize);
559*27162e4eSAndroid Build Coastguard Worker LZ4_initStream(cdict->fastCtx, sizeof(LZ4_stream_t));
560*27162e4eSAndroid Build Coastguard Worker LZ4_loadDictSlow(cdict->fastCtx, (const char*)cdict->dictContent, (int)dictSize);
561*27162e4eSAndroid Build Coastguard Worker LZ4_initStreamHC(cdict->HCCtx, sizeof(LZ4_streamHC_t));
562*27162e4eSAndroid Build Coastguard Worker /* note: we don't know at this point which compression level is going to be used
563*27162e4eSAndroid Build Coastguard Worker * as a consequence, HCCtx is created for the more common HC mode */
564*27162e4eSAndroid Build Coastguard Worker LZ4_setCompressionLevel(cdict->HCCtx, LZ4HC_CLEVEL_DEFAULT);
565*27162e4eSAndroid Build Coastguard Worker LZ4_loadDictHC(cdict->HCCtx, (const char*)cdict->dictContent, (int)dictSize);
566*27162e4eSAndroid Build Coastguard Worker return cdict;
567*27162e4eSAndroid Build Coastguard Worker }
568*27162e4eSAndroid Build Coastguard Worker
569*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_createCDict() :
570*27162e4eSAndroid Build Coastguard Worker * When compressing multiple messages / blocks with the same dictionary, it's recommended to load it just once.
571*27162e4eSAndroid Build Coastguard Worker * LZ4F_createCDict() will create a digested dictionary, ready to start future compression operations without startup delay.
572*27162e4eSAndroid Build Coastguard Worker * LZ4F_CDict can be created once and shared by multiple threads concurrently, since its usage is read-only.
573*27162e4eSAndroid Build Coastguard Worker * @dictBuffer can be released after LZ4F_CDict creation, since its content is copied within CDict
574*27162e4eSAndroid Build Coastguard Worker * @return : digested dictionary for compression, or NULL if failed */
LZ4F_createCDict(const void * dictBuffer,size_t dictSize)575*27162e4eSAndroid Build Coastguard Worker LZ4F_CDict* LZ4F_createCDict(const void* dictBuffer, size_t dictSize)
576*27162e4eSAndroid Build Coastguard Worker {
577*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(4, "LZ4F_createCDict");
578*27162e4eSAndroid Build Coastguard Worker return LZ4F_createCDict_advanced(LZ4F_defaultCMem, dictBuffer, dictSize);
579*27162e4eSAndroid Build Coastguard Worker }
580*27162e4eSAndroid Build Coastguard Worker
LZ4F_freeCDict(LZ4F_CDict * cdict)581*27162e4eSAndroid Build Coastguard Worker void LZ4F_freeCDict(LZ4F_CDict* cdict)
582*27162e4eSAndroid Build Coastguard Worker {
583*27162e4eSAndroid Build Coastguard Worker if (cdict==NULL) return; /* support free on NULL */
584*27162e4eSAndroid Build Coastguard Worker LZ4F_free(cdict->dictContent, cdict->cmem);
585*27162e4eSAndroid Build Coastguard Worker LZ4F_free(cdict->fastCtx, cdict->cmem);
586*27162e4eSAndroid Build Coastguard Worker LZ4F_free(cdict->HCCtx, cdict->cmem);
587*27162e4eSAndroid Build Coastguard Worker LZ4F_free(cdict, cdict->cmem);
588*27162e4eSAndroid Build Coastguard Worker }
589*27162e4eSAndroid Build Coastguard Worker
590*27162e4eSAndroid Build Coastguard Worker
591*27162e4eSAndroid Build Coastguard Worker /*-*********************************
592*27162e4eSAndroid Build Coastguard Worker * Advanced compression functions
593*27162e4eSAndroid Build Coastguard Worker ***********************************/
594*27162e4eSAndroid Build Coastguard Worker
595*27162e4eSAndroid Build Coastguard Worker LZ4F_cctx*
LZ4F_createCompressionContext_advanced(LZ4F_CustomMem customMem,unsigned version)596*27162e4eSAndroid Build Coastguard Worker LZ4F_createCompressionContext_advanced(LZ4F_CustomMem customMem, unsigned version)
597*27162e4eSAndroid Build Coastguard Worker {
598*27162e4eSAndroid Build Coastguard Worker LZ4F_cctx* const cctxPtr =
599*27162e4eSAndroid Build Coastguard Worker (LZ4F_cctx*)LZ4F_calloc(sizeof(LZ4F_cctx), customMem);
600*27162e4eSAndroid Build Coastguard Worker if (cctxPtr==NULL) return NULL;
601*27162e4eSAndroid Build Coastguard Worker
602*27162e4eSAndroid Build Coastguard Worker cctxPtr->cmem = customMem;
603*27162e4eSAndroid Build Coastguard Worker cctxPtr->version = version;
604*27162e4eSAndroid Build Coastguard Worker cctxPtr->cStage = 0; /* Uninitialized. Next stage : init cctx */
605*27162e4eSAndroid Build Coastguard Worker
606*27162e4eSAndroid Build Coastguard Worker return cctxPtr;
607*27162e4eSAndroid Build Coastguard Worker }
608*27162e4eSAndroid Build Coastguard Worker
609*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_createCompressionContext() :
610*27162e4eSAndroid Build Coastguard Worker * The first thing to do is to create a compressionContext object, which will be used in all compression operations.
611*27162e4eSAndroid Build Coastguard Worker * This is achieved using LZ4F_createCompressionContext(), which takes as argument a version and an LZ4F_preferences_t structure.
612*27162e4eSAndroid Build Coastguard Worker * The version provided MUST be LZ4F_VERSION. It is intended to track potential incompatible differences between different binaries.
613*27162e4eSAndroid Build Coastguard Worker * The function will provide a pointer to an allocated LZ4F_compressionContext_t object.
614*27162e4eSAndroid Build Coastguard Worker * If the result LZ4F_errorCode_t is not OK_NoError, there was an error during context creation.
615*27162e4eSAndroid Build Coastguard Worker * Object can release its memory using LZ4F_freeCompressionContext();
616*27162e4eSAndroid Build Coastguard Worker **/
617*27162e4eSAndroid Build Coastguard Worker LZ4F_errorCode_t
LZ4F_createCompressionContext(LZ4F_cctx ** LZ4F_compressionContextPtr,unsigned version)618*27162e4eSAndroid Build Coastguard Worker LZ4F_createCompressionContext(LZ4F_cctx** LZ4F_compressionContextPtr, unsigned version)
619*27162e4eSAndroid Build Coastguard Worker {
620*27162e4eSAndroid Build Coastguard Worker assert(LZ4F_compressionContextPtr != NULL); /* considered a violation of narrow contract */
621*27162e4eSAndroid Build Coastguard Worker /* in case it nonetheless happen in production */
622*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(LZ4F_compressionContextPtr == NULL, parameter_null);
623*27162e4eSAndroid Build Coastguard Worker
624*27162e4eSAndroid Build Coastguard Worker *LZ4F_compressionContextPtr = LZ4F_createCompressionContext_advanced(LZ4F_defaultCMem, version);
625*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(*LZ4F_compressionContextPtr==NULL, allocation_failed);
626*27162e4eSAndroid Build Coastguard Worker return LZ4F_OK_NoError;
627*27162e4eSAndroid Build Coastguard Worker }
628*27162e4eSAndroid Build Coastguard Worker
LZ4F_freeCompressionContext(LZ4F_cctx * cctxPtr)629*27162e4eSAndroid Build Coastguard Worker LZ4F_errorCode_t LZ4F_freeCompressionContext(LZ4F_cctx* cctxPtr)
630*27162e4eSAndroid Build Coastguard Worker {
631*27162e4eSAndroid Build Coastguard Worker if (cctxPtr != NULL) { /* support free on NULL */
632*27162e4eSAndroid Build Coastguard Worker LZ4F_free(cctxPtr->lz4CtxPtr, cctxPtr->cmem); /* note: LZ4_streamHC_t and LZ4_stream_t are simple POD types */
633*27162e4eSAndroid Build Coastguard Worker LZ4F_free(cctxPtr->tmpBuff, cctxPtr->cmem);
634*27162e4eSAndroid Build Coastguard Worker LZ4F_free(cctxPtr, cctxPtr->cmem);
635*27162e4eSAndroid Build Coastguard Worker }
636*27162e4eSAndroid Build Coastguard Worker return LZ4F_OK_NoError;
637*27162e4eSAndroid Build Coastguard Worker }
638*27162e4eSAndroid Build Coastguard Worker
639*27162e4eSAndroid Build Coastguard Worker
640*27162e4eSAndroid Build Coastguard Worker /**
641*27162e4eSAndroid Build Coastguard Worker * This function prepares the internal LZ4(HC) stream for a new compression,
642*27162e4eSAndroid Build Coastguard Worker * resetting the context and attaching the dictionary, if there is one.
643*27162e4eSAndroid Build Coastguard Worker *
644*27162e4eSAndroid Build Coastguard Worker * It needs to be called at the beginning of each independent compression
645*27162e4eSAndroid Build Coastguard Worker * stream (i.e., at the beginning of a frame in blockLinked mode, or at the
646*27162e4eSAndroid Build Coastguard Worker * beginning of each block in blockIndependent mode).
647*27162e4eSAndroid Build Coastguard Worker */
LZ4F_initStream(void * ctx,const LZ4F_CDict * cdict,int level,LZ4F_blockMode_t blockMode)648*27162e4eSAndroid Build Coastguard Worker static void LZ4F_initStream(void* ctx,
649*27162e4eSAndroid Build Coastguard Worker const LZ4F_CDict* cdict,
650*27162e4eSAndroid Build Coastguard Worker int level,
651*27162e4eSAndroid Build Coastguard Worker LZ4F_blockMode_t blockMode) {
652*27162e4eSAndroid Build Coastguard Worker if (level < LZ4HC_CLEVEL_MIN) {
653*27162e4eSAndroid Build Coastguard Worker if (cdict || blockMode == LZ4F_blockLinked) {
654*27162e4eSAndroid Build Coastguard Worker /* In these cases, we will call LZ4_compress_fast_continue(),
655*27162e4eSAndroid Build Coastguard Worker * which needs an already reset context. Otherwise, we'll call a
656*27162e4eSAndroid Build Coastguard Worker * one-shot API. The non-continued APIs internally perform their own
657*27162e4eSAndroid Build Coastguard Worker * resets at the beginning of their calls, where they know what
658*27162e4eSAndroid Build Coastguard Worker * tableType they need the context to be in. So in that case this
659*27162e4eSAndroid Build Coastguard Worker * would be misguided / wasted work. */
660*27162e4eSAndroid Build Coastguard Worker LZ4_resetStream_fast((LZ4_stream_t*)ctx);
661*27162e4eSAndroid Build Coastguard Worker if (cdict)
662*27162e4eSAndroid Build Coastguard Worker LZ4_attach_dictionary((LZ4_stream_t*)ctx, cdict->fastCtx);
663*27162e4eSAndroid Build Coastguard Worker }
664*27162e4eSAndroid Build Coastguard Worker /* In these cases, we'll call a one-shot API.
665*27162e4eSAndroid Build Coastguard Worker * The non-continued APIs internally perform their own resets
666*27162e4eSAndroid Build Coastguard Worker * at the beginning of their calls, where they know
667*27162e4eSAndroid Build Coastguard Worker * which tableType they need the context to be in.
668*27162e4eSAndroid Build Coastguard Worker * Therefore, a reset here would be wasted work. */
669*27162e4eSAndroid Build Coastguard Worker } else {
670*27162e4eSAndroid Build Coastguard Worker LZ4_resetStreamHC_fast((LZ4_streamHC_t*)ctx, level);
671*27162e4eSAndroid Build Coastguard Worker if (cdict)
672*27162e4eSAndroid Build Coastguard Worker LZ4_attach_HC_dictionary((LZ4_streamHC_t*)ctx, cdict->HCCtx);
673*27162e4eSAndroid Build Coastguard Worker }
674*27162e4eSAndroid Build Coastguard Worker }
675*27162e4eSAndroid Build Coastguard Worker
ctxTypeID_to_size(int ctxTypeID)676*27162e4eSAndroid Build Coastguard Worker static int ctxTypeID_to_size(int ctxTypeID) {
677*27162e4eSAndroid Build Coastguard Worker switch(ctxTypeID) {
678*27162e4eSAndroid Build Coastguard Worker case 1:
679*27162e4eSAndroid Build Coastguard Worker return LZ4_sizeofState();
680*27162e4eSAndroid Build Coastguard Worker case 2:
681*27162e4eSAndroid Build Coastguard Worker return LZ4_sizeofStateHC();
682*27162e4eSAndroid Build Coastguard Worker default:
683*27162e4eSAndroid Build Coastguard Worker return 0;
684*27162e4eSAndroid Build Coastguard Worker }
685*27162e4eSAndroid Build Coastguard Worker }
686*27162e4eSAndroid Build Coastguard Worker
687*27162e4eSAndroid Build Coastguard Worker /* LZ4F_compressBegin_internal()
688*27162e4eSAndroid Build Coastguard Worker * Note: only accepts @cdict _or_ @dictBuffer as non NULL.
689*27162e4eSAndroid Build Coastguard Worker */
LZ4F_compressBegin_internal(LZ4F_cctx * cctx,void * dstBuffer,size_t dstCapacity,const void * dictBuffer,size_t dictSize,const LZ4F_CDict * cdict,const LZ4F_preferences_t * preferencesPtr)690*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_compressBegin_internal(LZ4F_cctx* cctx,
691*27162e4eSAndroid Build Coastguard Worker void* dstBuffer, size_t dstCapacity,
692*27162e4eSAndroid Build Coastguard Worker const void* dictBuffer, size_t dictSize,
693*27162e4eSAndroid Build Coastguard Worker const LZ4F_CDict* cdict,
694*27162e4eSAndroid Build Coastguard Worker const LZ4F_preferences_t* preferencesPtr)
695*27162e4eSAndroid Build Coastguard Worker {
696*27162e4eSAndroid Build Coastguard Worker LZ4F_preferences_t const prefNull = LZ4F_INIT_PREFERENCES;
697*27162e4eSAndroid Build Coastguard Worker BYTE* const dstStart = (BYTE*)dstBuffer;
698*27162e4eSAndroid Build Coastguard Worker BYTE* dstPtr = dstStart;
699*27162e4eSAndroid Build Coastguard Worker
700*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(dstCapacity < maxFHSize, dstMaxSize_tooSmall);
701*27162e4eSAndroid Build Coastguard Worker if (preferencesPtr == NULL) preferencesPtr = &prefNull;
702*27162e4eSAndroid Build Coastguard Worker cctx->prefs = *preferencesPtr;
703*27162e4eSAndroid Build Coastguard Worker
704*27162e4eSAndroid Build Coastguard Worker /* cctx Management */
705*27162e4eSAndroid Build Coastguard Worker { U16 const ctxTypeID = (cctx->prefs.compressionLevel < LZ4HC_CLEVEL_MIN) ? 1 : 2;
706*27162e4eSAndroid Build Coastguard Worker int requiredSize = ctxTypeID_to_size(ctxTypeID);
707*27162e4eSAndroid Build Coastguard Worker int allocatedSize = ctxTypeID_to_size(cctx->lz4CtxAlloc);
708*27162e4eSAndroid Build Coastguard Worker if (allocatedSize < requiredSize) {
709*27162e4eSAndroid Build Coastguard Worker /* not enough space allocated */
710*27162e4eSAndroid Build Coastguard Worker LZ4F_free(cctx->lz4CtxPtr, cctx->cmem);
711*27162e4eSAndroid Build Coastguard Worker if (cctx->prefs.compressionLevel < LZ4HC_CLEVEL_MIN) {
712*27162e4eSAndroid Build Coastguard Worker /* must take ownership of memory allocation,
713*27162e4eSAndroid Build Coastguard Worker * in order to respect custom allocator contract */
714*27162e4eSAndroid Build Coastguard Worker cctx->lz4CtxPtr = LZ4F_malloc(sizeof(LZ4_stream_t), cctx->cmem);
715*27162e4eSAndroid Build Coastguard Worker if (cctx->lz4CtxPtr)
716*27162e4eSAndroid Build Coastguard Worker LZ4_initStream(cctx->lz4CtxPtr, sizeof(LZ4_stream_t));
717*27162e4eSAndroid Build Coastguard Worker } else {
718*27162e4eSAndroid Build Coastguard Worker cctx->lz4CtxPtr = LZ4F_malloc(sizeof(LZ4_streamHC_t), cctx->cmem);
719*27162e4eSAndroid Build Coastguard Worker if (cctx->lz4CtxPtr)
720*27162e4eSAndroid Build Coastguard Worker LZ4_initStreamHC(cctx->lz4CtxPtr, sizeof(LZ4_streamHC_t));
721*27162e4eSAndroid Build Coastguard Worker }
722*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(cctx->lz4CtxPtr == NULL, allocation_failed);
723*27162e4eSAndroid Build Coastguard Worker cctx->lz4CtxAlloc = ctxTypeID;
724*27162e4eSAndroid Build Coastguard Worker cctx->lz4CtxType = ctxTypeID;
725*27162e4eSAndroid Build Coastguard Worker } else if (cctx->lz4CtxType != ctxTypeID) {
726*27162e4eSAndroid Build Coastguard Worker /* otherwise, a sufficient buffer is already allocated,
727*27162e4eSAndroid Build Coastguard Worker * but we need to reset it to the correct context type */
728*27162e4eSAndroid Build Coastguard Worker if (cctx->prefs.compressionLevel < LZ4HC_CLEVEL_MIN) {
729*27162e4eSAndroid Build Coastguard Worker LZ4_initStream((LZ4_stream_t*)cctx->lz4CtxPtr, sizeof(LZ4_stream_t));
730*27162e4eSAndroid Build Coastguard Worker } else {
731*27162e4eSAndroid Build Coastguard Worker LZ4_initStreamHC((LZ4_streamHC_t*)cctx->lz4CtxPtr, sizeof(LZ4_streamHC_t));
732*27162e4eSAndroid Build Coastguard Worker LZ4_setCompressionLevel((LZ4_streamHC_t*)cctx->lz4CtxPtr, cctx->prefs.compressionLevel);
733*27162e4eSAndroid Build Coastguard Worker }
734*27162e4eSAndroid Build Coastguard Worker cctx->lz4CtxType = ctxTypeID;
735*27162e4eSAndroid Build Coastguard Worker } }
736*27162e4eSAndroid Build Coastguard Worker
737*27162e4eSAndroid Build Coastguard Worker /* Buffer Management */
738*27162e4eSAndroid Build Coastguard Worker if (cctx->prefs.frameInfo.blockSizeID == 0)
739*27162e4eSAndroid Build Coastguard Worker cctx->prefs.frameInfo.blockSizeID = LZ4F_BLOCKSIZEID_DEFAULT;
740*27162e4eSAndroid Build Coastguard Worker cctx->maxBlockSize = LZ4F_getBlockSize(cctx->prefs.frameInfo.blockSizeID);
741*27162e4eSAndroid Build Coastguard Worker
742*27162e4eSAndroid Build Coastguard Worker { size_t const requiredBuffSize = preferencesPtr->autoFlush ?
743*27162e4eSAndroid Build Coastguard Worker ((cctx->prefs.frameInfo.blockMode == LZ4F_blockLinked) ? 64 KB : 0) : /* only needs past data up to window size */
744*27162e4eSAndroid Build Coastguard Worker cctx->maxBlockSize + ((cctx->prefs.frameInfo.blockMode == LZ4F_blockLinked) ? 128 KB : 0);
745*27162e4eSAndroid Build Coastguard Worker
746*27162e4eSAndroid Build Coastguard Worker if (cctx->maxBufferSize < requiredBuffSize) {
747*27162e4eSAndroid Build Coastguard Worker cctx->maxBufferSize = 0;
748*27162e4eSAndroid Build Coastguard Worker LZ4F_free(cctx->tmpBuff, cctx->cmem);
749*27162e4eSAndroid Build Coastguard Worker cctx->tmpBuff = (BYTE*)LZ4F_malloc(requiredBuffSize, cctx->cmem);
750*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(cctx->tmpBuff == NULL, allocation_failed);
751*27162e4eSAndroid Build Coastguard Worker cctx->maxBufferSize = requiredBuffSize;
752*27162e4eSAndroid Build Coastguard Worker } }
753*27162e4eSAndroid Build Coastguard Worker cctx->tmpIn = cctx->tmpBuff;
754*27162e4eSAndroid Build Coastguard Worker cctx->tmpInSize = 0;
755*27162e4eSAndroid Build Coastguard Worker (void)XXH32_reset(&(cctx->xxh), 0);
756*27162e4eSAndroid Build Coastguard Worker
757*27162e4eSAndroid Build Coastguard Worker /* context init */
758*27162e4eSAndroid Build Coastguard Worker cctx->cdict = cdict;
759*27162e4eSAndroid Build Coastguard Worker if (cctx->prefs.frameInfo.blockMode == LZ4F_blockLinked) {
760*27162e4eSAndroid Build Coastguard Worker /* frame init only for blockLinked : blockIndependent will be init at each block */
761*27162e4eSAndroid Build Coastguard Worker LZ4F_initStream(cctx->lz4CtxPtr, cdict, cctx->prefs.compressionLevel, LZ4F_blockLinked);
762*27162e4eSAndroid Build Coastguard Worker }
763*27162e4eSAndroid Build Coastguard Worker if (preferencesPtr->compressionLevel >= LZ4HC_CLEVEL_MIN) {
764*27162e4eSAndroid Build Coastguard Worker LZ4_favorDecompressionSpeed((LZ4_streamHC_t*)cctx->lz4CtxPtr, (int)preferencesPtr->favorDecSpeed);
765*27162e4eSAndroid Build Coastguard Worker }
766*27162e4eSAndroid Build Coastguard Worker if (dictBuffer) {
767*27162e4eSAndroid Build Coastguard Worker assert(cdict == NULL);
768*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(dictSize > INT_MAX, parameter_invalid);
769*27162e4eSAndroid Build Coastguard Worker if (cctx->lz4CtxType == ctxFast) {
770*27162e4eSAndroid Build Coastguard Worker /* lz4 fast*/
771*27162e4eSAndroid Build Coastguard Worker LZ4_loadDict((LZ4_stream_t*)cctx->lz4CtxPtr, (const char*)dictBuffer, (int)dictSize);
772*27162e4eSAndroid Build Coastguard Worker } else {
773*27162e4eSAndroid Build Coastguard Worker /* lz4hc */
774*27162e4eSAndroid Build Coastguard Worker assert(cctx->lz4CtxType == ctxHC);
775*27162e4eSAndroid Build Coastguard Worker LZ4_loadDictHC((LZ4_streamHC_t*)cctx->lz4CtxPtr, (const char*)dictBuffer, (int)dictSize);
776*27162e4eSAndroid Build Coastguard Worker }
777*27162e4eSAndroid Build Coastguard Worker }
778*27162e4eSAndroid Build Coastguard Worker
779*27162e4eSAndroid Build Coastguard Worker /* Stage 2 : Write Frame Header */
780*27162e4eSAndroid Build Coastguard Worker
781*27162e4eSAndroid Build Coastguard Worker /* Magic Number */
782*27162e4eSAndroid Build Coastguard Worker LZ4F_writeLE32(dstPtr, LZ4F_MAGICNUMBER);
783*27162e4eSAndroid Build Coastguard Worker dstPtr += 4;
784*27162e4eSAndroid Build Coastguard Worker { BYTE* const headerStart = dstPtr;
785*27162e4eSAndroid Build Coastguard Worker
786*27162e4eSAndroid Build Coastguard Worker /* FLG Byte */
787*27162e4eSAndroid Build Coastguard Worker *dstPtr++ = (BYTE)(((1 & _2BITS) << 6) /* Version('01') */
788*27162e4eSAndroid Build Coastguard Worker + ((cctx->prefs.frameInfo.blockMode & _1BIT ) << 5)
789*27162e4eSAndroid Build Coastguard Worker + ((cctx->prefs.frameInfo.blockChecksumFlag & _1BIT ) << 4)
790*27162e4eSAndroid Build Coastguard Worker + ((unsigned)(cctx->prefs.frameInfo.contentSize > 0) << 3)
791*27162e4eSAndroid Build Coastguard Worker + ((cctx->prefs.frameInfo.contentChecksumFlag & _1BIT ) << 2)
792*27162e4eSAndroid Build Coastguard Worker + (cctx->prefs.frameInfo.dictID > 0) );
793*27162e4eSAndroid Build Coastguard Worker /* BD Byte */
794*27162e4eSAndroid Build Coastguard Worker *dstPtr++ = (BYTE)((cctx->prefs.frameInfo.blockSizeID & _3BITS) << 4);
795*27162e4eSAndroid Build Coastguard Worker /* Optional Frame content size field */
796*27162e4eSAndroid Build Coastguard Worker if (cctx->prefs.frameInfo.contentSize) {
797*27162e4eSAndroid Build Coastguard Worker LZ4F_writeLE64(dstPtr, cctx->prefs.frameInfo.contentSize);
798*27162e4eSAndroid Build Coastguard Worker dstPtr += 8;
799*27162e4eSAndroid Build Coastguard Worker cctx->totalInSize = 0;
800*27162e4eSAndroid Build Coastguard Worker }
801*27162e4eSAndroid Build Coastguard Worker /* Optional dictionary ID field */
802*27162e4eSAndroid Build Coastguard Worker if (cctx->prefs.frameInfo.dictID) {
803*27162e4eSAndroid Build Coastguard Worker LZ4F_writeLE32(dstPtr, cctx->prefs.frameInfo.dictID);
804*27162e4eSAndroid Build Coastguard Worker dstPtr += 4;
805*27162e4eSAndroid Build Coastguard Worker }
806*27162e4eSAndroid Build Coastguard Worker /* Header CRC Byte */
807*27162e4eSAndroid Build Coastguard Worker *dstPtr = LZ4F_headerChecksum(headerStart, (size_t)(dstPtr - headerStart));
808*27162e4eSAndroid Build Coastguard Worker dstPtr++;
809*27162e4eSAndroid Build Coastguard Worker }
810*27162e4eSAndroid Build Coastguard Worker
811*27162e4eSAndroid Build Coastguard Worker cctx->cStage = 1; /* header written, now request input data block */
812*27162e4eSAndroid Build Coastguard Worker return (size_t)(dstPtr - dstStart);
813*27162e4eSAndroid Build Coastguard Worker }
814*27162e4eSAndroid Build Coastguard Worker
LZ4F_compressBegin(LZ4F_cctx * cctx,void * dstBuffer,size_t dstCapacity,const LZ4F_preferences_t * preferencesPtr)815*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_compressBegin(LZ4F_cctx* cctx,
816*27162e4eSAndroid Build Coastguard Worker void* dstBuffer, size_t dstCapacity,
817*27162e4eSAndroid Build Coastguard Worker const LZ4F_preferences_t* preferencesPtr)
818*27162e4eSAndroid Build Coastguard Worker {
819*27162e4eSAndroid Build Coastguard Worker return LZ4F_compressBegin_internal(cctx, dstBuffer, dstCapacity,
820*27162e4eSAndroid Build Coastguard Worker NULL, 0,
821*27162e4eSAndroid Build Coastguard Worker NULL, preferencesPtr);
822*27162e4eSAndroid Build Coastguard Worker }
823*27162e4eSAndroid Build Coastguard Worker
824*27162e4eSAndroid Build Coastguard Worker /* LZ4F_compressBegin_usingDictOnce:
825*27162e4eSAndroid Build Coastguard Worker * Hidden implementation,
826*27162e4eSAndroid Build Coastguard Worker * employed for multi-threaded compression
827*27162e4eSAndroid Build Coastguard Worker * when frame defines linked blocks */
LZ4F_compressBegin_usingDictOnce(LZ4F_cctx * cctx,void * dstBuffer,size_t dstCapacity,const void * dict,size_t dictSize,const LZ4F_preferences_t * preferencesPtr)828*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_compressBegin_usingDictOnce(LZ4F_cctx* cctx,
829*27162e4eSAndroid Build Coastguard Worker void* dstBuffer, size_t dstCapacity,
830*27162e4eSAndroid Build Coastguard Worker const void* dict, size_t dictSize,
831*27162e4eSAndroid Build Coastguard Worker const LZ4F_preferences_t* preferencesPtr)
832*27162e4eSAndroid Build Coastguard Worker {
833*27162e4eSAndroid Build Coastguard Worker return LZ4F_compressBegin_internal(cctx, dstBuffer, dstCapacity,
834*27162e4eSAndroid Build Coastguard Worker dict, dictSize,
835*27162e4eSAndroid Build Coastguard Worker NULL, preferencesPtr);
836*27162e4eSAndroid Build Coastguard Worker }
837*27162e4eSAndroid Build Coastguard Worker
LZ4F_compressBegin_usingDict(LZ4F_cctx * cctx,void * dstBuffer,size_t dstCapacity,const void * dict,size_t dictSize,const LZ4F_preferences_t * preferencesPtr)838*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_compressBegin_usingDict(LZ4F_cctx* cctx,
839*27162e4eSAndroid Build Coastguard Worker void* dstBuffer, size_t dstCapacity,
840*27162e4eSAndroid Build Coastguard Worker const void* dict, size_t dictSize,
841*27162e4eSAndroid Build Coastguard Worker const LZ4F_preferences_t* preferencesPtr)
842*27162e4eSAndroid Build Coastguard Worker {
843*27162e4eSAndroid Build Coastguard Worker /* note : incorrect implementation :
844*27162e4eSAndroid Build Coastguard Worker * this will only use the dictionary once,
845*27162e4eSAndroid Build Coastguard Worker * instead of once *per* block when frames defines independent blocks */
846*27162e4eSAndroid Build Coastguard Worker return LZ4F_compressBegin_usingDictOnce(cctx, dstBuffer, dstCapacity,
847*27162e4eSAndroid Build Coastguard Worker dict, dictSize,
848*27162e4eSAndroid Build Coastguard Worker preferencesPtr);
849*27162e4eSAndroid Build Coastguard Worker }
850*27162e4eSAndroid Build Coastguard Worker
LZ4F_compressBegin_usingCDict(LZ4F_cctx * cctx,void * dstBuffer,size_t dstCapacity,const LZ4F_CDict * cdict,const LZ4F_preferences_t * preferencesPtr)851*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_compressBegin_usingCDict(LZ4F_cctx* cctx,
852*27162e4eSAndroid Build Coastguard Worker void* dstBuffer, size_t dstCapacity,
853*27162e4eSAndroid Build Coastguard Worker const LZ4F_CDict* cdict,
854*27162e4eSAndroid Build Coastguard Worker const LZ4F_preferences_t* preferencesPtr)
855*27162e4eSAndroid Build Coastguard Worker {
856*27162e4eSAndroid Build Coastguard Worker return LZ4F_compressBegin_internal(cctx, dstBuffer, dstCapacity,
857*27162e4eSAndroid Build Coastguard Worker NULL, 0,
858*27162e4eSAndroid Build Coastguard Worker cdict, preferencesPtr);
859*27162e4eSAndroid Build Coastguard Worker }
860*27162e4eSAndroid Build Coastguard Worker
861*27162e4eSAndroid Build Coastguard Worker
862*27162e4eSAndroid Build Coastguard Worker /* LZ4F_compressBound() :
863*27162e4eSAndroid Build Coastguard Worker * @return minimum capacity of dstBuffer for a given srcSize to handle worst case scenario.
864*27162e4eSAndroid Build Coastguard Worker * LZ4F_preferences_t structure is optional : if NULL, preferences will be set to cover worst case scenario.
865*27162e4eSAndroid Build Coastguard Worker * This function cannot fail.
866*27162e4eSAndroid Build Coastguard Worker */
LZ4F_compressBound(size_t srcSize,const LZ4F_preferences_t * preferencesPtr)867*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_compressBound(size_t srcSize, const LZ4F_preferences_t* preferencesPtr)
868*27162e4eSAndroid Build Coastguard Worker {
869*27162e4eSAndroid Build Coastguard Worker if (preferencesPtr && preferencesPtr->autoFlush) {
870*27162e4eSAndroid Build Coastguard Worker return LZ4F_compressBound_internal(srcSize, preferencesPtr, 0);
871*27162e4eSAndroid Build Coastguard Worker }
872*27162e4eSAndroid Build Coastguard Worker return LZ4F_compressBound_internal(srcSize, preferencesPtr, (size_t)-1);
873*27162e4eSAndroid Build Coastguard Worker }
874*27162e4eSAndroid Build Coastguard Worker
875*27162e4eSAndroid Build Coastguard Worker
876*27162e4eSAndroid Build Coastguard Worker typedef int (*compressFunc_t)(void* ctx, const char* src, char* dst, int srcSize, int dstSize, int level, const LZ4F_CDict* cdict);
877*27162e4eSAndroid Build Coastguard Worker
878*27162e4eSAndroid Build Coastguard Worker
879*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_makeBlock():
880*27162e4eSAndroid Build Coastguard Worker * compress a single block, add header and optional checksum.
881*27162e4eSAndroid Build Coastguard Worker * assumption : dst buffer capacity is >= BHSize + srcSize + crcSize
882*27162e4eSAndroid Build Coastguard Worker */
LZ4F_makeBlock(void * dst,const void * src,size_t srcSize,compressFunc_t compress,void * lz4ctx,int level,const LZ4F_CDict * cdict,LZ4F_blockChecksum_t crcFlag)883*27162e4eSAndroid Build Coastguard Worker static size_t LZ4F_makeBlock(void* dst,
884*27162e4eSAndroid Build Coastguard Worker const void* src, size_t srcSize,
885*27162e4eSAndroid Build Coastguard Worker compressFunc_t compress, void* lz4ctx, int level,
886*27162e4eSAndroid Build Coastguard Worker const LZ4F_CDict* cdict,
887*27162e4eSAndroid Build Coastguard Worker LZ4F_blockChecksum_t crcFlag)
888*27162e4eSAndroid Build Coastguard Worker {
889*27162e4eSAndroid Build Coastguard Worker BYTE* const cSizePtr = (BYTE*)dst;
890*27162e4eSAndroid Build Coastguard Worker U32 cSize;
891*27162e4eSAndroid Build Coastguard Worker assert(compress != NULL);
892*27162e4eSAndroid Build Coastguard Worker cSize = (U32)compress(lz4ctx, (const char*)src, (char*)(cSizePtr+BHSize),
893*27162e4eSAndroid Build Coastguard Worker (int)(srcSize), (int)(srcSize-1),
894*27162e4eSAndroid Build Coastguard Worker level, cdict);
895*27162e4eSAndroid Build Coastguard Worker
896*27162e4eSAndroid Build Coastguard Worker if (cSize == 0 || cSize >= srcSize) {
897*27162e4eSAndroid Build Coastguard Worker cSize = (U32)srcSize;
898*27162e4eSAndroid Build Coastguard Worker LZ4F_writeLE32(cSizePtr, cSize | LZ4F_BLOCKUNCOMPRESSED_FLAG);
899*27162e4eSAndroid Build Coastguard Worker memcpy(cSizePtr+BHSize, src, srcSize);
900*27162e4eSAndroid Build Coastguard Worker } else {
901*27162e4eSAndroid Build Coastguard Worker LZ4F_writeLE32(cSizePtr, cSize);
902*27162e4eSAndroid Build Coastguard Worker }
903*27162e4eSAndroid Build Coastguard Worker if (crcFlag) {
904*27162e4eSAndroid Build Coastguard Worker U32 const crc32 = XXH32(cSizePtr+BHSize, cSize, 0); /* checksum of compressed data */
905*27162e4eSAndroid Build Coastguard Worker LZ4F_writeLE32(cSizePtr+BHSize+cSize, crc32);
906*27162e4eSAndroid Build Coastguard Worker }
907*27162e4eSAndroid Build Coastguard Worker return BHSize + cSize + ((U32)crcFlag)*BFSize;
908*27162e4eSAndroid Build Coastguard Worker }
909*27162e4eSAndroid Build Coastguard Worker
910*27162e4eSAndroid Build Coastguard Worker
LZ4F_compressBlock(void * ctx,const char * src,char * dst,int srcSize,int dstCapacity,int level,const LZ4F_CDict * cdict)911*27162e4eSAndroid Build Coastguard Worker static int LZ4F_compressBlock(void* ctx, const char* src, char* dst, int srcSize, int dstCapacity, int level, const LZ4F_CDict* cdict)
912*27162e4eSAndroid Build Coastguard Worker {
913*27162e4eSAndroid Build Coastguard Worker int const acceleration = (level < 0) ? -level + 1 : 1;
914*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(5, "LZ4F_compressBlock (srcSize=%i)", srcSize);
915*27162e4eSAndroid Build Coastguard Worker LZ4F_initStream(ctx, cdict, level, LZ4F_blockIndependent);
916*27162e4eSAndroid Build Coastguard Worker if (cdict) {
917*27162e4eSAndroid Build Coastguard Worker return LZ4_compress_fast_continue((LZ4_stream_t*)ctx, src, dst, srcSize, dstCapacity, acceleration);
918*27162e4eSAndroid Build Coastguard Worker } else {
919*27162e4eSAndroid Build Coastguard Worker return LZ4_compress_fast_extState_fastReset(ctx, src, dst, srcSize, dstCapacity, acceleration);
920*27162e4eSAndroid Build Coastguard Worker }
921*27162e4eSAndroid Build Coastguard Worker }
922*27162e4eSAndroid Build Coastguard Worker
LZ4F_compressBlock_continue(void * ctx,const char * src,char * dst,int srcSize,int dstCapacity,int level,const LZ4F_CDict * cdict)923*27162e4eSAndroid Build Coastguard Worker static int LZ4F_compressBlock_continue(void* ctx, const char* src, char* dst, int srcSize, int dstCapacity, int level, const LZ4F_CDict* cdict)
924*27162e4eSAndroid Build Coastguard Worker {
925*27162e4eSAndroid Build Coastguard Worker int const acceleration = (level < 0) ? -level + 1 : 1;
926*27162e4eSAndroid Build Coastguard Worker (void)cdict; /* init once at beginning of frame */
927*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(5, "LZ4F_compressBlock_continue (srcSize=%i)", srcSize);
928*27162e4eSAndroid Build Coastguard Worker return LZ4_compress_fast_continue((LZ4_stream_t*)ctx, src, dst, srcSize, dstCapacity, acceleration);
929*27162e4eSAndroid Build Coastguard Worker }
930*27162e4eSAndroid Build Coastguard Worker
LZ4F_compressBlockHC(void * ctx,const char * src,char * dst,int srcSize,int dstCapacity,int level,const LZ4F_CDict * cdict)931*27162e4eSAndroid Build Coastguard Worker static int LZ4F_compressBlockHC(void* ctx, const char* src, char* dst, int srcSize, int dstCapacity, int level, const LZ4F_CDict* cdict)
932*27162e4eSAndroid Build Coastguard Worker {
933*27162e4eSAndroid Build Coastguard Worker LZ4F_initStream(ctx, cdict, level, LZ4F_blockIndependent);
934*27162e4eSAndroid Build Coastguard Worker if (cdict) {
935*27162e4eSAndroid Build Coastguard Worker return LZ4_compress_HC_continue((LZ4_streamHC_t*)ctx, src, dst, srcSize, dstCapacity);
936*27162e4eSAndroid Build Coastguard Worker }
937*27162e4eSAndroid Build Coastguard Worker return LZ4_compress_HC_extStateHC_fastReset(ctx, src, dst, srcSize, dstCapacity, level);
938*27162e4eSAndroid Build Coastguard Worker }
939*27162e4eSAndroid Build Coastguard Worker
LZ4F_compressBlockHC_continue(void * ctx,const char * src,char * dst,int srcSize,int dstCapacity,int level,const LZ4F_CDict * cdict)940*27162e4eSAndroid Build Coastguard Worker static int LZ4F_compressBlockHC_continue(void* ctx, const char* src, char* dst, int srcSize, int dstCapacity, int level, const LZ4F_CDict* cdict)
941*27162e4eSAndroid Build Coastguard Worker {
942*27162e4eSAndroid Build Coastguard Worker (void)level; (void)cdict; /* init once at beginning of frame */
943*27162e4eSAndroid Build Coastguard Worker return LZ4_compress_HC_continue((LZ4_streamHC_t*)ctx, src, dst, srcSize, dstCapacity);
944*27162e4eSAndroid Build Coastguard Worker }
945*27162e4eSAndroid Build Coastguard Worker
LZ4F_doNotCompressBlock(void * ctx,const char * src,char * dst,int srcSize,int dstCapacity,int level,const LZ4F_CDict * cdict)946*27162e4eSAndroid Build Coastguard Worker static int LZ4F_doNotCompressBlock(void* ctx, const char* src, char* dst, int srcSize, int dstCapacity, int level, const LZ4F_CDict* cdict)
947*27162e4eSAndroid Build Coastguard Worker {
948*27162e4eSAndroid Build Coastguard Worker (void)ctx; (void)src; (void)dst; (void)srcSize; (void)dstCapacity; (void)level; (void)cdict;
949*27162e4eSAndroid Build Coastguard Worker return 0;
950*27162e4eSAndroid Build Coastguard Worker }
951*27162e4eSAndroid Build Coastguard Worker
LZ4F_selectCompression(LZ4F_blockMode_t blockMode,int level,LZ4F_BlockCompressMode_e compressMode)952*27162e4eSAndroid Build Coastguard Worker static compressFunc_t LZ4F_selectCompression(LZ4F_blockMode_t blockMode, int level, LZ4F_BlockCompressMode_e compressMode)
953*27162e4eSAndroid Build Coastguard Worker {
954*27162e4eSAndroid Build Coastguard Worker if (compressMode == LZ4B_UNCOMPRESSED)
955*27162e4eSAndroid Build Coastguard Worker return LZ4F_doNotCompressBlock;
956*27162e4eSAndroid Build Coastguard Worker if (level < LZ4HC_CLEVEL_MIN) {
957*27162e4eSAndroid Build Coastguard Worker if (blockMode == LZ4F_blockIndependent) return LZ4F_compressBlock;
958*27162e4eSAndroid Build Coastguard Worker return LZ4F_compressBlock_continue;
959*27162e4eSAndroid Build Coastguard Worker }
960*27162e4eSAndroid Build Coastguard Worker if (blockMode == LZ4F_blockIndependent) return LZ4F_compressBlockHC;
961*27162e4eSAndroid Build Coastguard Worker return LZ4F_compressBlockHC_continue;
962*27162e4eSAndroid Build Coastguard Worker }
963*27162e4eSAndroid Build Coastguard Worker
964*27162e4eSAndroid Build Coastguard Worker /* Save history (up to 64KB) into @tmpBuff */
LZ4F_localSaveDict(LZ4F_cctx_t * cctxPtr)965*27162e4eSAndroid Build Coastguard Worker static int LZ4F_localSaveDict(LZ4F_cctx_t* cctxPtr)
966*27162e4eSAndroid Build Coastguard Worker {
967*27162e4eSAndroid Build Coastguard Worker if (cctxPtr->prefs.compressionLevel < LZ4HC_CLEVEL_MIN)
968*27162e4eSAndroid Build Coastguard Worker return LZ4_saveDict ((LZ4_stream_t*)(cctxPtr->lz4CtxPtr), (char*)(cctxPtr->tmpBuff), 64 KB);
969*27162e4eSAndroid Build Coastguard Worker return LZ4_saveDictHC ((LZ4_streamHC_t*)(cctxPtr->lz4CtxPtr), (char*)(cctxPtr->tmpBuff), 64 KB);
970*27162e4eSAndroid Build Coastguard Worker }
971*27162e4eSAndroid Build Coastguard Worker
972*27162e4eSAndroid Build Coastguard Worker typedef enum { notDone, fromTmpBuffer, fromSrcBuffer } LZ4F_lastBlockStatus;
973*27162e4eSAndroid Build Coastguard Worker
974*27162e4eSAndroid Build Coastguard Worker static const LZ4F_compressOptions_t k_cOptionsNull = { 0, { 0, 0, 0 } };
975*27162e4eSAndroid Build Coastguard Worker
976*27162e4eSAndroid Build Coastguard Worker
977*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_compressUpdateImpl() :
978*27162e4eSAndroid Build Coastguard Worker * LZ4F_compressUpdate() can be called repetitively to compress as much data as necessary.
979*27162e4eSAndroid Build Coastguard Worker * When successful, the function always entirely consumes @srcBuffer.
980*27162e4eSAndroid Build Coastguard Worker * src data is either buffered or compressed into @dstBuffer.
981*27162e4eSAndroid Build Coastguard Worker * If the block compression does not match the compression of the previous block, the old data is flushed
982*27162e4eSAndroid Build Coastguard Worker * and operations continue with the new compression mode.
983*27162e4eSAndroid Build Coastguard Worker * @dstCapacity MUST be >= LZ4F_compressBound(srcSize, preferencesPtr) when block compression is turned on.
984*27162e4eSAndroid Build Coastguard Worker * @compressOptionsPtr is optional : provide NULL to mean "default".
985*27162e4eSAndroid Build Coastguard Worker * @return : the number of bytes written into dstBuffer. It can be zero, meaning input data was just buffered.
986*27162e4eSAndroid Build Coastguard Worker * or an error code if it fails (which can be tested using LZ4F_isError())
987*27162e4eSAndroid Build Coastguard Worker * After an error, the state is left in a UB state, and must be re-initialized.
988*27162e4eSAndroid Build Coastguard Worker */
LZ4F_compressUpdateImpl(LZ4F_cctx * cctxPtr,void * dstBuffer,size_t dstCapacity,const void * srcBuffer,size_t srcSize,const LZ4F_compressOptions_t * compressOptionsPtr,LZ4F_BlockCompressMode_e blockCompression)989*27162e4eSAndroid Build Coastguard Worker static size_t LZ4F_compressUpdateImpl(LZ4F_cctx* cctxPtr,
990*27162e4eSAndroid Build Coastguard Worker void* dstBuffer, size_t dstCapacity,
991*27162e4eSAndroid Build Coastguard Worker const void* srcBuffer, size_t srcSize,
992*27162e4eSAndroid Build Coastguard Worker const LZ4F_compressOptions_t* compressOptionsPtr,
993*27162e4eSAndroid Build Coastguard Worker LZ4F_BlockCompressMode_e blockCompression)
994*27162e4eSAndroid Build Coastguard Worker {
995*27162e4eSAndroid Build Coastguard Worker size_t const blockSize = cctxPtr->maxBlockSize;
996*27162e4eSAndroid Build Coastguard Worker const BYTE* srcPtr = (const BYTE*)srcBuffer;
997*27162e4eSAndroid Build Coastguard Worker const BYTE* const srcEnd = srcPtr + srcSize;
998*27162e4eSAndroid Build Coastguard Worker BYTE* const dstStart = (BYTE*)dstBuffer;
999*27162e4eSAndroid Build Coastguard Worker BYTE* dstPtr = dstStart;
1000*27162e4eSAndroid Build Coastguard Worker LZ4F_lastBlockStatus lastBlockCompressed = notDone;
1001*27162e4eSAndroid Build Coastguard Worker compressFunc_t const compress = LZ4F_selectCompression(cctxPtr->prefs.frameInfo.blockMode, cctxPtr->prefs.compressionLevel, blockCompression);
1002*27162e4eSAndroid Build Coastguard Worker size_t bytesWritten;
1003*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(4, "LZ4F_compressUpdate (srcSize=%zu)", srcSize);
1004*27162e4eSAndroid Build Coastguard Worker
1005*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(cctxPtr->cStage != 1, compressionState_uninitialized); /* state must be initialized and waiting for next block */
1006*27162e4eSAndroid Build Coastguard Worker if (dstCapacity < LZ4F_compressBound_internal(srcSize, &(cctxPtr->prefs), cctxPtr->tmpInSize))
1007*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR(dstMaxSize_tooSmall);
1008*27162e4eSAndroid Build Coastguard Worker
1009*27162e4eSAndroid Build Coastguard Worker if (blockCompression == LZ4B_UNCOMPRESSED && dstCapacity < srcSize)
1010*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR(dstMaxSize_tooSmall);
1011*27162e4eSAndroid Build Coastguard Worker
1012*27162e4eSAndroid Build Coastguard Worker /* flush currently written block, to continue with new block compression */
1013*27162e4eSAndroid Build Coastguard Worker if (cctxPtr->blockCompressMode != blockCompression) {
1014*27162e4eSAndroid Build Coastguard Worker bytesWritten = LZ4F_flush(cctxPtr, dstBuffer, dstCapacity, compressOptionsPtr);
1015*27162e4eSAndroid Build Coastguard Worker dstPtr += bytesWritten;
1016*27162e4eSAndroid Build Coastguard Worker cctxPtr->blockCompressMode = blockCompression;
1017*27162e4eSAndroid Build Coastguard Worker }
1018*27162e4eSAndroid Build Coastguard Worker
1019*27162e4eSAndroid Build Coastguard Worker if (compressOptionsPtr == NULL) compressOptionsPtr = &k_cOptionsNull;
1020*27162e4eSAndroid Build Coastguard Worker
1021*27162e4eSAndroid Build Coastguard Worker /* complete tmp buffer */
1022*27162e4eSAndroid Build Coastguard Worker if (cctxPtr->tmpInSize > 0) { /* some data already within tmp buffer */
1023*27162e4eSAndroid Build Coastguard Worker size_t const sizeToCopy = blockSize - cctxPtr->tmpInSize;
1024*27162e4eSAndroid Build Coastguard Worker assert(blockSize > cctxPtr->tmpInSize);
1025*27162e4eSAndroid Build Coastguard Worker if (sizeToCopy > srcSize) {
1026*27162e4eSAndroid Build Coastguard Worker /* add src to tmpIn buffer */
1027*27162e4eSAndroid Build Coastguard Worker memcpy(cctxPtr->tmpIn + cctxPtr->tmpInSize, srcBuffer, srcSize);
1028*27162e4eSAndroid Build Coastguard Worker srcPtr = srcEnd;
1029*27162e4eSAndroid Build Coastguard Worker cctxPtr->tmpInSize += srcSize;
1030*27162e4eSAndroid Build Coastguard Worker /* still needs some CRC */
1031*27162e4eSAndroid Build Coastguard Worker } else {
1032*27162e4eSAndroid Build Coastguard Worker /* complete tmpIn block and then compress it */
1033*27162e4eSAndroid Build Coastguard Worker lastBlockCompressed = fromTmpBuffer;
1034*27162e4eSAndroid Build Coastguard Worker memcpy(cctxPtr->tmpIn + cctxPtr->tmpInSize, srcBuffer, sizeToCopy);
1035*27162e4eSAndroid Build Coastguard Worker srcPtr += sizeToCopy;
1036*27162e4eSAndroid Build Coastguard Worker
1037*27162e4eSAndroid Build Coastguard Worker dstPtr += LZ4F_makeBlock(dstPtr,
1038*27162e4eSAndroid Build Coastguard Worker cctxPtr->tmpIn, blockSize,
1039*27162e4eSAndroid Build Coastguard Worker compress, cctxPtr->lz4CtxPtr, cctxPtr->prefs.compressionLevel,
1040*27162e4eSAndroid Build Coastguard Worker cctxPtr->cdict,
1041*27162e4eSAndroid Build Coastguard Worker cctxPtr->prefs.frameInfo.blockChecksumFlag);
1042*27162e4eSAndroid Build Coastguard Worker if (cctxPtr->prefs.frameInfo.blockMode==LZ4F_blockLinked) cctxPtr->tmpIn += blockSize;
1043*27162e4eSAndroid Build Coastguard Worker cctxPtr->tmpInSize = 0;
1044*27162e4eSAndroid Build Coastguard Worker } }
1045*27162e4eSAndroid Build Coastguard Worker
1046*27162e4eSAndroid Build Coastguard Worker while ((size_t)(srcEnd - srcPtr) >= blockSize) {
1047*27162e4eSAndroid Build Coastguard Worker /* compress full blocks */
1048*27162e4eSAndroid Build Coastguard Worker lastBlockCompressed = fromSrcBuffer;
1049*27162e4eSAndroid Build Coastguard Worker dstPtr += LZ4F_makeBlock(dstPtr,
1050*27162e4eSAndroid Build Coastguard Worker srcPtr, blockSize,
1051*27162e4eSAndroid Build Coastguard Worker compress, cctxPtr->lz4CtxPtr, cctxPtr->prefs.compressionLevel,
1052*27162e4eSAndroid Build Coastguard Worker cctxPtr->cdict,
1053*27162e4eSAndroid Build Coastguard Worker cctxPtr->prefs.frameInfo.blockChecksumFlag);
1054*27162e4eSAndroid Build Coastguard Worker srcPtr += blockSize;
1055*27162e4eSAndroid Build Coastguard Worker }
1056*27162e4eSAndroid Build Coastguard Worker
1057*27162e4eSAndroid Build Coastguard Worker if ((cctxPtr->prefs.autoFlush) && (srcPtr < srcEnd)) {
1058*27162e4eSAndroid Build Coastguard Worker /* autoFlush : remaining input (< blockSize) is compressed */
1059*27162e4eSAndroid Build Coastguard Worker lastBlockCompressed = fromSrcBuffer;
1060*27162e4eSAndroid Build Coastguard Worker dstPtr += LZ4F_makeBlock(dstPtr,
1061*27162e4eSAndroid Build Coastguard Worker srcPtr, (size_t)(srcEnd - srcPtr),
1062*27162e4eSAndroid Build Coastguard Worker compress, cctxPtr->lz4CtxPtr, cctxPtr->prefs.compressionLevel,
1063*27162e4eSAndroid Build Coastguard Worker cctxPtr->cdict,
1064*27162e4eSAndroid Build Coastguard Worker cctxPtr->prefs.frameInfo.blockChecksumFlag);
1065*27162e4eSAndroid Build Coastguard Worker srcPtr = srcEnd;
1066*27162e4eSAndroid Build Coastguard Worker }
1067*27162e4eSAndroid Build Coastguard Worker
1068*27162e4eSAndroid Build Coastguard Worker /* preserve dictionary within @tmpBuff whenever necessary */
1069*27162e4eSAndroid Build Coastguard Worker if ((cctxPtr->prefs.frameInfo.blockMode==LZ4F_blockLinked) && (lastBlockCompressed==fromSrcBuffer)) {
1070*27162e4eSAndroid Build Coastguard Worker /* linked blocks are only supported in compressed mode, see LZ4F_uncompressedUpdate */
1071*27162e4eSAndroid Build Coastguard Worker assert(blockCompression == LZ4B_COMPRESSED);
1072*27162e4eSAndroid Build Coastguard Worker if (compressOptionsPtr->stableSrc) {
1073*27162e4eSAndroid Build Coastguard Worker cctxPtr->tmpIn = cctxPtr->tmpBuff; /* src is stable : dictionary remains in src across invocations */
1074*27162e4eSAndroid Build Coastguard Worker } else {
1075*27162e4eSAndroid Build Coastguard Worker int const realDictSize = LZ4F_localSaveDict(cctxPtr);
1076*27162e4eSAndroid Build Coastguard Worker assert(0 <= realDictSize && realDictSize <= 64 KB);
1077*27162e4eSAndroid Build Coastguard Worker cctxPtr->tmpIn = cctxPtr->tmpBuff + realDictSize;
1078*27162e4eSAndroid Build Coastguard Worker }
1079*27162e4eSAndroid Build Coastguard Worker }
1080*27162e4eSAndroid Build Coastguard Worker
1081*27162e4eSAndroid Build Coastguard Worker /* keep tmpIn within limits */
1082*27162e4eSAndroid Build Coastguard Worker if (!(cctxPtr->prefs.autoFlush) /* no autoflush : there may be some data left within internal buffer */
1083*27162e4eSAndroid Build Coastguard Worker && (cctxPtr->tmpIn + blockSize) > (cctxPtr->tmpBuff + cctxPtr->maxBufferSize) ) /* not enough room to store next block */
1084*27162e4eSAndroid Build Coastguard Worker {
1085*27162e4eSAndroid Build Coastguard Worker /* only preserve 64KB within internal buffer. Ensures there is enough room for next block.
1086*27162e4eSAndroid Build Coastguard Worker * note: this situation necessarily implies lastBlockCompressed==fromTmpBuffer */
1087*27162e4eSAndroid Build Coastguard Worker int const realDictSize = LZ4F_localSaveDict(cctxPtr);
1088*27162e4eSAndroid Build Coastguard Worker cctxPtr->tmpIn = cctxPtr->tmpBuff + realDictSize;
1089*27162e4eSAndroid Build Coastguard Worker assert((cctxPtr->tmpIn + blockSize) <= (cctxPtr->tmpBuff + cctxPtr->maxBufferSize));
1090*27162e4eSAndroid Build Coastguard Worker }
1091*27162e4eSAndroid Build Coastguard Worker
1092*27162e4eSAndroid Build Coastguard Worker /* some input data left, necessarily < blockSize */
1093*27162e4eSAndroid Build Coastguard Worker if (srcPtr < srcEnd) {
1094*27162e4eSAndroid Build Coastguard Worker /* fill tmp buffer */
1095*27162e4eSAndroid Build Coastguard Worker size_t const sizeToCopy = (size_t)(srcEnd - srcPtr);
1096*27162e4eSAndroid Build Coastguard Worker memcpy(cctxPtr->tmpIn, srcPtr, sizeToCopy);
1097*27162e4eSAndroid Build Coastguard Worker cctxPtr->tmpInSize = sizeToCopy;
1098*27162e4eSAndroid Build Coastguard Worker }
1099*27162e4eSAndroid Build Coastguard Worker
1100*27162e4eSAndroid Build Coastguard Worker if (cctxPtr->prefs.frameInfo.contentChecksumFlag == LZ4F_contentChecksumEnabled)
1101*27162e4eSAndroid Build Coastguard Worker (void)XXH32_update(&(cctxPtr->xxh), srcBuffer, srcSize);
1102*27162e4eSAndroid Build Coastguard Worker
1103*27162e4eSAndroid Build Coastguard Worker cctxPtr->totalInSize += srcSize;
1104*27162e4eSAndroid Build Coastguard Worker return (size_t)(dstPtr - dstStart);
1105*27162e4eSAndroid Build Coastguard Worker }
1106*27162e4eSAndroid Build Coastguard Worker
1107*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_compressUpdate() :
1108*27162e4eSAndroid Build Coastguard Worker * LZ4F_compressUpdate() can be called repetitively to compress as much data as necessary.
1109*27162e4eSAndroid Build Coastguard Worker * When successful, the function always entirely consumes @srcBuffer.
1110*27162e4eSAndroid Build Coastguard Worker * src data is either buffered or compressed into @dstBuffer.
1111*27162e4eSAndroid Build Coastguard Worker * If previously an uncompressed block was written, buffered data is flushed
1112*27162e4eSAndroid Build Coastguard Worker * before appending compressed data is continued.
1113*27162e4eSAndroid Build Coastguard Worker * @dstCapacity MUST be >= LZ4F_compressBound(srcSize, preferencesPtr).
1114*27162e4eSAndroid Build Coastguard Worker * @compressOptionsPtr is optional : provide NULL to mean "default".
1115*27162e4eSAndroid Build Coastguard Worker * @return : the number of bytes written into dstBuffer. It can be zero, meaning input data was just buffered.
1116*27162e4eSAndroid Build Coastguard Worker * or an error code if it fails (which can be tested using LZ4F_isError())
1117*27162e4eSAndroid Build Coastguard Worker * After an error, the state is left in a UB state, and must be re-initialized.
1118*27162e4eSAndroid Build Coastguard Worker */
LZ4F_compressUpdate(LZ4F_cctx * cctxPtr,void * dstBuffer,size_t dstCapacity,const void * srcBuffer,size_t srcSize,const LZ4F_compressOptions_t * compressOptionsPtr)1119*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_compressUpdate(LZ4F_cctx* cctxPtr,
1120*27162e4eSAndroid Build Coastguard Worker void* dstBuffer, size_t dstCapacity,
1121*27162e4eSAndroid Build Coastguard Worker const void* srcBuffer, size_t srcSize,
1122*27162e4eSAndroid Build Coastguard Worker const LZ4F_compressOptions_t* compressOptionsPtr)
1123*27162e4eSAndroid Build Coastguard Worker {
1124*27162e4eSAndroid Build Coastguard Worker return LZ4F_compressUpdateImpl(cctxPtr,
1125*27162e4eSAndroid Build Coastguard Worker dstBuffer, dstCapacity,
1126*27162e4eSAndroid Build Coastguard Worker srcBuffer, srcSize,
1127*27162e4eSAndroid Build Coastguard Worker compressOptionsPtr, LZ4B_COMPRESSED);
1128*27162e4eSAndroid Build Coastguard Worker }
1129*27162e4eSAndroid Build Coastguard Worker
1130*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_uncompressedUpdate() :
1131*27162e4eSAndroid Build Coastguard Worker * Same as LZ4F_compressUpdate(), but requests blocks to be sent uncompressed.
1132*27162e4eSAndroid Build Coastguard Worker * This symbol is only supported when LZ4F_blockIndependent is used
1133*27162e4eSAndroid Build Coastguard Worker * @dstCapacity MUST be >= LZ4F_compressBound(srcSize, preferencesPtr).
1134*27162e4eSAndroid Build Coastguard Worker * @compressOptionsPtr is optional : provide NULL to mean "default".
1135*27162e4eSAndroid Build Coastguard Worker * @return : the number of bytes written into dstBuffer. It can be zero, meaning input data was just buffered.
1136*27162e4eSAndroid Build Coastguard Worker * or an error code if it fails (which can be tested using LZ4F_isError())
1137*27162e4eSAndroid Build Coastguard Worker * After an error, the state is left in a UB state, and must be re-initialized.
1138*27162e4eSAndroid Build Coastguard Worker */
LZ4F_uncompressedUpdate(LZ4F_cctx * cctxPtr,void * dstBuffer,size_t dstCapacity,const void * srcBuffer,size_t srcSize,const LZ4F_compressOptions_t * compressOptionsPtr)1139*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_uncompressedUpdate(LZ4F_cctx* cctxPtr,
1140*27162e4eSAndroid Build Coastguard Worker void* dstBuffer, size_t dstCapacity,
1141*27162e4eSAndroid Build Coastguard Worker const void* srcBuffer, size_t srcSize,
1142*27162e4eSAndroid Build Coastguard Worker const LZ4F_compressOptions_t* compressOptionsPtr)
1143*27162e4eSAndroid Build Coastguard Worker {
1144*27162e4eSAndroid Build Coastguard Worker return LZ4F_compressUpdateImpl(cctxPtr,
1145*27162e4eSAndroid Build Coastguard Worker dstBuffer, dstCapacity,
1146*27162e4eSAndroid Build Coastguard Worker srcBuffer, srcSize,
1147*27162e4eSAndroid Build Coastguard Worker compressOptionsPtr, LZ4B_UNCOMPRESSED);
1148*27162e4eSAndroid Build Coastguard Worker }
1149*27162e4eSAndroid Build Coastguard Worker
1150*27162e4eSAndroid Build Coastguard Worker
1151*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_flush() :
1152*27162e4eSAndroid Build Coastguard Worker * When compressed data must be sent immediately, without waiting for a block to be filled,
1153*27162e4eSAndroid Build Coastguard Worker * invoke LZ4_flush(), which will immediately compress any remaining data stored within LZ4F_cctx.
1154*27162e4eSAndroid Build Coastguard Worker * The result of the function is the number of bytes written into dstBuffer.
1155*27162e4eSAndroid Build Coastguard Worker * It can be zero, this means there was no data left within LZ4F_cctx.
1156*27162e4eSAndroid Build Coastguard Worker * The function outputs an error code if it fails (can be tested using LZ4F_isError())
1157*27162e4eSAndroid Build Coastguard Worker * LZ4F_compressOptions_t* is optional. NULL is a valid argument.
1158*27162e4eSAndroid Build Coastguard Worker */
LZ4F_flush(LZ4F_cctx * cctxPtr,void * dstBuffer,size_t dstCapacity,const LZ4F_compressOptions_t * compressOptionsPtr)1159*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_flush(LZ4F_cctx* cctxPtr,
1160*27162e4eSAndroid Build Coastguard Worker void* dstBuffer, size_t dstCapacity,
1161*27162e4eSAndroid Build Coastguard Worker const LZ4F_compressOptions_t* compressOptionsPtr)
1162*27162e4eSAndroid Build Coastguard Worker {
1163*27162e4eSAndroid Build Coastguard Worker BYTE* const dstStart = (BYTE*)dstBuffer;
1164*27162e4eSAndroid Build Coastguard Worker BYTE* dstPtr = dstStart;
1165*27162e4eSAndroid Build Coastguard Worker compressFunc_t compress;
1166*27162e4eSAndroid Build Coastguard Worker
1167*27162e4eSAndroid Build Coastguard Worker if (cctxPtr->tmpInSize == 0) return 0; /* nothing to flush */
1168*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(cctxPtr->cStage != 1, compressionState_uninitialized);
1169*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(dstCapacity < (cctxPtr->tmpInSize + BHSize + BFSize), dstMaxSize_tooSmall);
1170*27162e4eSAndroid Build Coastguard Worker (void)compressOptionsPtr; /* not useful (yet) */
1171*27162e4eSAndroid Build Coastguard Worker
1172*27162e4eSAndroid Build Coastguard Worker /* select compression function */
1173*27162e4eSAndroid Build Coastguard Worker compress = LZ4F_selectCompression(cctxPtr->prefs.frameInfo.blockMode, cctxPtr->prefs.compressionLevel, cctxPtr->blockCompressMode);
1174*27162e4eSAndroid Build Coastguard Worker
1175*27162e4eSAndroid Build Coastguard Worker /* compress tmp buffer */
1176*27162e4eSAndroid Build Coastguard Worker dstPtr += LZ4F_makeBlock(dstPtr,
1177*27162e4eSAndroid Build Coastguard Worker cctxPtr->tmpIn, cctxPtr->tmpInSize,
1178*27162e4eSAndroid Build Coastguard Worker compress, cctxPtr->lz4CtxPtr, cctxPtr->prefs.compressionLevel,
1179*27162e4eSAndroid Build Coastguard Worker cctxPtr->cdict,
1180*27162e4eSAndroid Build Coastguard Worker cctxPtr->prefs.frameInfo.blockChecksumFlag);
1181*27162e4eSAndroid Build Coastguard Worker assert(((void)"flush overflows dstBuffer!", (size_t)(dstPtr - dstStart) <= dstCapacity));
1182*27162e4eSAndroid Build Coastguard Worker
1183*27162e4eSAndroid Build Coastguard Worker if (cctxPtr->prefs.frameInfo.blockMode == LZ4F_blockLinked)
1184*27162e4eSAndroid Build Coastguard Worker cctxPtr->tmpIn += cctxPtr->tmpInSize;
1185*27162e4eSAndroid Build Coastguard Worker cctxPtr->tmpInSize = 0;
1186*27162e4eSAndroid Build Coastguard Worker
1187*27162e4eSAndroid Build Coastguard Worker /* keep tmpIn within limits */
1188*27162e4eSAndroid Build Coastguard Worker if ((cctxPtr->tmpIn + cctxPtr->maxBlockSize) > (cctxPtr->tmpBuff + cctxPtr->maxBufferSize)) { /* necessarily LZ4F_blockLinked */
1189*27162e4eSAndroid Build Coastguard Worker int const realDictSize = LZ4F_localSaveDict(cctxPtr);
1190*27162e4eSAndroid Build Coastguard Worker cctxPtr->tmpIn = cctxPtr->tmpBuff + realDictSize;
1191*27162e4eSAndroid Build Coastguard Worker }
1192*27162e4eSAndroid Build Coastguard Worker
1193*27162e4eSAndroid Build Coastguard Worker return (size_t)(dstPtr - dstStart);
1194*27162e4eSAndroid Build Coastguard Worker }
1195*27162e4eSAndroid Build Coastguard Worker
1196*27162e4eSAndroid Build Coastguard Worker
1197*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_compressEnd() :
1198*27162e4eSAndroid Build Coastguard Worker * When you want to properly finish the compressed frame, just call LZ4F_compressEnd().
1199*27162e4eSAndroid Build Coastguard Worker * It will flush whatever data remained within compressionContext (like LZ4_flush())
1200*27162e4eSAndroid Build Coastguard Worker * but also properly finalize the frame, with an endMark and an (optional) checksum.
1201*27162e4eSAndroid Build Coastguard Worker * LZ4F_compressOptions_t structure is optional : you can provide NULL as argument.
1202*27162e4eSAndroid Build Coastguard Worker * @return: the number of bytes written into dstBuffer (necessarily >= 4 (endMark size))
1203*27162e4eSAndroid Build Coastguard Worker * or an error code if it fails (can be tested using LZ4F_isError())
1204*27162e4eSAndroid Build Coastguard Worker * The context can then be used again to compress a new frame, starting with LZ4F_compressBegin().
1205*27162e4eSAndroid Build Coastguard Worker */
LZ4F_compressEnd(LZ4F_cctx * cctxPtr,void * dstBuffer,size_t dstCapacity,const LZ4F_compressOptions_t * compressOptionsPtr)1206*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_compressEnd(LZ4F_cctx* cctxPtr,
1207*27162e4eSAndroid Build Coastguard Worker void* dstBuffer, size_t dstCapacity,
1208*27162e4eSAndroid Build Coastguard Worker const LZ4F_compressOptions_t* compressOptionsPtr)
1209*27162e4eSAndroid Build Coastguard Worker {
1210*27162e4eSAndroid Build Coastguard Worker BYTE* const dstStart = (BYTE*)dstBuffer;
1211*27162e4eSAndroid Build Coastguard Worker BYTE* dstPtr = dstStart;
1212*27162e4eSAndroid Build Coastguard Worker
1213*27162e4eSAndroid Build Coastguard Worker size_t const flushSize = LZ4F_flush(cctxPtr, dstBuffer, dstCapacity, compressOptionsPtr);
1214*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(5,"LZ4F_compressEnd: dstCapacity=%u", (unsigned)dstCapacity);
1215*27162e4eSAndroid Build Coastguard Worker FORWARD_IF_ERROR(flushSize);
1216*27162e4eSAndroid Build Coastguard Worker dstPtr += flushSize;
1217*27162e4eSAndroid Build Coastguard Worker
1218*27162e4eSAndroid Build Coastguard Worker assert(flushSize <= dstCapacity);
1219*27162e4eSAndroid Build Coastguard Worker dstCapacity -= flushSize;
1220*27162e4eSAndroid Build Coastguard Worker
1221*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(dstCapacity < 4, dstMaxSize_tooSmall);
1222*27162e4eSAndroid Build Coastguard Worker LZ4F_writeLE32(dstPtr, 0);
1223*27162e4eSAndroid Build Coastguard Worker dstPtr += 4; /* endMark */
1224*27162e4eSAndroid Build Coastguard Worker
1225*27162e4eSAndroid Build Coastguard Worker if (cctxPtr->prefs.frameInfo.contentChecksumFlag == LZ4F_contentChecksumEnabled) {
1226*27162e4eSAndroid Build Coastguard Worker U32 const xxh = XXH32_digest(&(cctxPtr->xxh));
1227*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(dstCapacity < 8, dstMaxSize_tooSmall);
1228*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(5,"Writing 32-bit content checksum (0x%0X)", xxh);
1229*27162e4eSAndroid Build Coastguard Worker LZ4F_writeLE32(dstPtr, xxh);
1230*27162e4eSAndroid Build Coastguard Worker dstPtr+=4; /* content Checksum */
1231*27162e4eSAndroid Build Coastguard Worker }
1232*27162e4eSAndroid Build Coastguard Worker
1233*27162e4eSAndroid Build Coastguard Worker cctxPtr->cStage = 0; /* state is now re-usable (with identical preferences) */
1234*27162e4eSAndroid Build Coastguard Worker
1235*27162e4eSAndroid Build Coastguard Worker if (cctxPtr->prefs.frameInfo.contentSize) {
1236*27162e4eSAndroid Build Coastguard Worker if (cctxPtr->prefs.frameInfo.contentSize != cctxPtr->totalInSize)
1237*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR(frameSize_wrong);
1238*27162e4eSAndroid Build Coastguard Worker }
1239*27162e4eSAndroid Build Coastguard Worker
1240*27162e4eSAndroid Build Coastguard Worker return (size_t)(dstPtr - dstStart);
1241*27162e4eSAndroid Build Coastguard Worker }
1242*27162e4eSAndroid Build Coastguard Worker
1243*27162e4eSAndroid Build Coastguard Worker
1244*27162e4eSAndroid Build Coastguard Worker /*-***************************************************
1245*27162e4eSAndroid Build Coastguard Worker * Frame Decompression
1246*27162e4eSAndroid Build Coastguard Worker *****************************************************/
1247*27162e4eSAndroid Build Coastguard Worker
1248*27162e4eSAndroid Build Coastguard Worker typedef enum {
1249*27162e4eSAndroid Build Coastguard Worker dstage_getFrameHeader=0, dstage_storeFrameHeader,
1250*27162e4eSAndroid Build Coastguard Worker dstage_init,
1251*27162e4eSAndroid Build Coastguard Worker dstage_getBlockHeader, dstage_storeBlockHeader,
1252*27162e4eSAndroid Build Coastguard Worker dstage_copyDirect, dstage_getBlockChecksum,
1253*27162e4eSAndroid Build Coastguard Worker dstage_getCBlock, dstage_storeCBlock,
1254*27162e4eSAndroid Build Coastguard Worker dstage_flushOut,
1255*27162e4eSAndroid Build Coastguard Worker dstage_getSuffix, dstage_storeSuffix,
1256*27162e4eSAndroid Build Coastguard Worker dstage_getSFrameSize, dstage_storeSFrameSize,
1257*27162e4eSAndroid Build Coastguard Worker dstage_skipSkippable
1258*27162e4eSAndroid Build Coastguard Worker } dStage_t;
1259*27162e4eSAndroid Build Coastguard Worker
1260*27162e4eSAndroid Build Coastguard Worker struct LZ4F_dctx_s {
1261*27162e4eSAndroid Build Coastguard Worker LZ4F_CustomMem cmem;
1262*27162e4eSAndroid Build Coastguard Worker LZ4F_frameInfo_t frameInfo;
1263*27162e4eSAndroid Build Coastguard Worker U32 version;
1264*27162e4eSAndroid Build Coastguard Worker dStage_t dStage;
1265*27162e4eSAndroid Build Coastguard Worker U64 frameRemainingSize;
1266*27162e4eSAndroid Build Coastguard Worker size_t maxBlockSize;
1267*27162e4eSAndroid Build Coastguard Worker size_t maxBufferSize;
1268*27162e4eSAndroid Build Coastguard Worker BYTE* tmpIn;
1269*27162e4eSAndroid Build Coastguard Worker size_t tmpInSize;
1270*27162e4eSAndroid Build Coastguard Worker size_t tmpInTarget;
1271*27162e4eSAndroid Build Coastguard Worker BYTE* tmpOutBuffer;
1272*27162e4eSAndroid Build Coastguard Worker const BYTE* dict;
1273*27162e4eSAndroid Build Coastguard Worker size_t dictSize;
1274*27162e4eSAndroid Build Coastguard Worker BYTE* tmpOut;
1275*27162e4eSAndroid Build Coastguard Worker size_t tmpOutSize;
1276*27162e4eSAndroid Build Coastguard Worker size_t tmpOutStart;
1277*27162e4eSAndroid Build Coastguard Worker XXH32_state_t xxh;
1278*27162e4eSAndroid Build Coastguard Worker XXH32_state_t blockChecksum;
1279*27162e4eSAndroid Build Coastguard Worker int skipChecksum;
1280*27162e4eSAndroid Build Coastguard Worker BYTE header[LZ4F_HEADER_SIZE_MAX];
1281*27162e4eSAndroid Build Coastguard Worker }; /* typedef'd to LZ4F_dctx in lz4frame.h */
1282*27162e4eSAndroid Build Coastguard Worker
1283*27162e4eSAndroid Build Coastguard Worker
LZ4F_createDecompressionContext_advanced(LZ4F_CustomMem customMem,unsigned version)1284*27162e4eSAndroid Build Coastguard Worker LZ4F_dctx* LZ4F_createDecompressionContext_advanced(LZ4F_CustomMem customMem, unsigned version)
1285*27162e4eSAndroid Build Coastguard Worker {
1286*27162e4eSAndroid Build Coastguard Worker LZ4F_dctx* const dctx = (LZ4F_dctx*)LZ4F_calloc(sizeof(LZ4F_dctx), customMem);
1287*27162e4eSAndroid Build Coastguard Worker if (dctx == NULL) return NULL;
1288*27162e4eSAndroid Build Coastguard Worker
1289*27162e4eSAndroid Build Coastguard Worker dctx->cmem = customMem;
1290*27162e4eSAndroid Build Coastguard Worker dctx->version = version;
1291*27162e4eSAndroid Build Coastguard Worker return dctx;
1292*27162e4eSAndroid Build Coastguard Worker }
1293*27162e4eSAndroid Build Coastguard Worker
1294*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_createDecompressionContext() :
1295*27162e4eSAndroid Build Coastguard Worker * Create a decompressionContext object, which will track all decompression operations.
1296*27162e4eSAndroid Build Coastguard Worker * Provides a pointer to a fully allocated and initialized LZ4F_decompressionContext object.
1297*27162e4eSAndroid Build Coastguard Worker * Object can later be released using LZ4F_freeDecompressionContext().
1298*27162e4eSAndroid Build Coastguard Worker * @return : if != 0, there was an error during context creation.
1299*27162e4eSAndroid Build Coastguard Worker */
1300*27162e4eSAndroid Build Coastguard Worker LZ4F_errorCode_t
LZ4F_createDecompressionContext(LZ4F_dctx ** LZ4F_decompressionContextPtr,unsigned versionNumber)1301*27162e4eSAndroid Build Coastguard Worker LZ4F_createDecompressionContext(LZ4F_dctx** LZ4F_decompressionContextPtr, unsigned versionNumber)
1302*27162e4eSAndroid Build Coastguard Worker {
1303*27162e4eSAndroid Build Coastguard Worker assert(LZ4F_decompressionContextPtr != NULL); /* violation of narrow contract */
1304*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(LZ4F_decompressionContextPtr == NULL, parameter_null); /* in case it nonetheless happen in production */
1305*27162e4eSAndroid Build Coastguard Worker
1306*27162e4eSAndroid Build Coastguard Worker *LZ4F_decompressionContextPtr = LZ4F_createDecompressionContext_advanced(LZ4F_defaultCMem, versionNumber);
1307*27162e4eSAndroid Build Coastguard Worker if (*LZ4F_decompressionContextPtr == NULL) { /* failed allocation */
1308*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR(allocation_failed);
1309*27162e4eSAndroid Build Coastguard Worker }
1310*27162e4eSAndroid Build Coastguard Worker return LZ4F_OK_NoError;
1311*27162e4eSAndroid Build Coastguard Worker }
1312*27162e4eSAndroid Build Coastguard Worker
LZ4F_freeDecompressionContext(LZ4F_dctx * dctx)1313*27162e4eSAndroid Build Coastguard Worker LZ4F_errorCode_t LZ4F_freeDecompressionContext(LZ4F_dctx* dctx)
1314*27162e4eSAndroid Build Coastguard Worker {
1315*27162e4eSAndroid Build Coastguard Worker LZ4F_errorCode_t result = LZ4F_OK_NoError;
1316*27162e4eSAndroid Build Coastguard Worker if (dctx != NULL) { /* can accept NULL input, like free() */
1317*27162e4eSAndroid Build Coastguard Worker result = (LZ4F_errorCode_t)dctx->dStage;
1318*27162e4eSAndroid Build Coastguard Worker LZ4F_free(dctx->tmpIn, dctx->cmem);
1319*27162e4eSAndroid Build Coastguard Worker LZ4F_free(dctx->tmpOutBuffer, dctx->cmem);
1320*27162e4eSAndroid Build Coastguard Worker LZ4F_free(dctx, dctx->cmem);
1321*27162e4eSAndroid Build Coastguard Worker }
1322*27162e4eSAndroid Build Coastguard Worker return result;
1323*27162e4eSAndroid Build Coastguard Worker }
1324*27162e4eSAndroid Build Coastguard Worker
1325*27162e4eSAndroid Build Coastguard Worker
1326*27162e4eSAndroid Build Coastguard Worker /*==--- Streaming Decompression operations ---==*/
LZ4F_resetDecompressionContext(LZ4F_dctx * dctx)1327*27162e4eSAndroid Build Coastguard Worker void LZ4F_resetDecompressionContext(LZ4F_dctx* dctx)
1328*27162e4eSAndroid Build Coastguard Worker {
1329*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(5, "LZ4F_resetDecompressionContext");
1330*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_getFrameHeader;
1331*27162e4eSAndroid Build Coastguard Worker dctx->dict = NULL;
1332*27162e4eSAndroid Build Coastguard Worker dctx->dictSize = 0;
1333*27162e4eSAndroid Build Coastguard Worker dctx->skipChecksum = 0;
1334*27162e4eSAndroid Build Coastguard Worker dctx->frameRemainingSize = 0;
1335*27162e4eSAndroid Build Coastguard Worker }
1336*27162e4eSAndroid Build Coastguard Worker
1337*27162e4eSAndroid Build Coastguard Worker
1338*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_decodeHeader() :
1339*27162e4eSAndroid Build Coastguard Worker * input : `src` points at the **beginning of the frame**
1340*27162e4eSAndroid Build Coastguard Worker * output : set internal values of dctx, such as
1341*27162e4eSAndroid Build Coastguard Worker * dctx->frameInfo and dctx->dStage.
1342*27162e4eSAndroid Build Coastguard Worker * Also allocates internal buffers.
1343*27162e4eSAndroid Build Coastguard Worker * @return : nb Bytes read from src (necessarily <= srcSize)
1344*27162e4eSAndroid Build Coastguard Worker * or an error code (testable with LZ4F_isError())
1345*27162e4eSAndroid Build Coastguard Worker */
LZ4F_decodeHeader(LZ4F_dctx * dctx,const void * src,size_t srcSize)1346*27162e4eSAndroid Build Coastguard Worker static size_t LZ4F_decodeHeader(LZ4F_dctx* dctx, const void* src, size_t srcSize)
1347*27162e4eSAndroid Build Coastguard Worker {
1348*27162e4eSAndroid Build Coastguard Worker unsigned blockMode, blockChecksumFlag, contentSizeFlag, contentChecksumFlag, dictIDFlag, blockSizeID;
1349*27162e4eSAndroid Build Coastguard Worker size_t frameHeaderSize;
1350*27162e4eSAndroid Build Coastguard Worker const BYTE* srcPtr = (const BYTE*)src;
1351*27162e4eSAndroid Build Coastguard Worker
1352*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(5, "LZ4F_decodeHeader");
1353*27162e4eSAndroid Build Coastguard Worker /* need to decode header to get frameInfo */
1354*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(srcSize < minFHSize, frameHeader_incomplete); /* minimal frame header size */
1355*27162e4eSAndroid Build Coastguard Worker MEM_INIT(&(dctx->frameInfo), 0, sizeof(dctx->frameInfo));
1356*27162e4eSAndroid Build Coastguard Worker
1357*27162e4eSAndroid Build Coastguard Worker /* special case : skippable frames */
1358*27162e4eSAndroid Build Coastguard Worker if ((LZ4F_readLE32(srcPtr) & 0xFFFFFFF0U) == LZ4F_MAGIC_SKIPPABLE_START) {
1359*27162e4eSAndroid Build Coastguard Worker dctx->frameInfo.frameType = LZ4F_skippableFrame;
1360*27162e4eSAndroid Build Coastguard Worker if (src == (void*)(dctx->header)) {
1361*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize = srcSize;
1362*27162e4eSAndroid Build Coastguard Worker dctx->tmpInTarget = 8;
1363*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_storeSFrameSize;
1364*27162e4eSAndroid Build Coastguard Worker return srcSize;
1365*27162e4eSAndroid Build Coastguard Worker } else {
1366*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_getSFrameSize;
1367*27162e4eSAndroid Build Coastguard Worker return 4;
1368*27162e4eSAndroid Build Coastguard Worker } }
1369*27162e4eSAndroid Build Coastguard Worker
1370*27162e4eSAndroid Build Coastguard Worker /* control magic number */
1371*27162e4eSAndroid Build Coastguard Worker #ifndef FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION
1372*27162e4eSAndroid Build Coastguard Worker if (LZ4F_readLE32(srcPtr) != LZ4F_MAGICNUMBER) {
1373*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(4, "frame header error : unknown magic number");
1374*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR(frameType_unknown);
1375*27162e4eSAndroid Build Coastguard Worker }
1376*27162e4eSAndroid Build Coastguard Worker #endif
1377*27162e4eSAndroid Build Coastguard Worker dctx->frameInfo.frameType = LZ4F_frame;
1378*27162e4eSAndroid Build Coastguard Worker
1379*27162e4eSAndroid Build Coastguard Worker /* Flags */
1380*27162e4eSAndroid Build Coastguard Worker { U32 const FLG = srcPtr[4];
1381*27162e4eSAndroid Build Coastguard Worker U32 const version = (FLG>>6) & _2BITS;
1382*27162e4eSAndroid Build Coastguard Worker blockChecksumFlag = (FLG>>4) & _1BIT;
1383*27162e4eSAndroid Build Coastguard Worker blockMode = (FLG>>5) & _1BIT;
1384*27162e4eSAndroid Build Coastguard Worker contentSizeFlag = (FLG>>3) & _1BIT;
1385*27162e4eSAndroid Build Coastguard Worker contentChecksumFlag = (FLG>>2) & _1BIT;
1386*27162e4eSAndroid Build Coastguard Worker dictIDFlag = FLG & _1BIT;
1387*27162e4eSAndroid Build Coastguard Worker /* validate */
1388*27162e4eSAndroid Build Coastguard Worker if (((FLG>>1)&_1BIT) != 0) RETURN_ERROR(reservedFlag_set); /* Reserved bit */
1389*27162e4eSAndroid Build Coastguard Worker if (version != 1) RETURN_ERROR(headerVersion_wrong); /* Version Number, only supported value */
1390*27162e4eSAndroid Build Coastguard Worker }
1391*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(6, "contentSizeFlag: %u", contentSizeFlag);
1392*27162e4eSAndroid Build Coastguard Worker
1393*27162e4eSAndroid Build Coastguard Worker /* Frame Header Size */
1394*27162e4eSAndroid Build Coastguard Worker frameHeaderSize = minFHSize + (contentSizeFlag?8:0) + (dictIDFlag?4:0);
1395*27162e4eSAndroid Build Coastguard Worker
1396*27162e4eSAndroid Build Coastguard Worker if (srcSize < frameHeaderSize) {
1397*27162e4eSAndroid Build Coastguard Worker /* not enough input to fully decode frame header */
1398*27162e4eSAndroid Build Coastguard Worker if (srcPtr != dctx->header)
1399*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->header, srcPtr, srcSize);
1400*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize = srcSize;
1401*27162e4eSAndroid Build Coastguard Worker dctx->tmpInTarget = frameHeaderSize;
1402*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_storeFrameHeader;
1403*27162e4eSAndroid Build Coastguard Worker return srcSize;
1404*27162e4eSAndroid Build Coastguard Worker }
1405*27162e4eSAndroid Build Coastguard Worker
1406*27162e4eSAndroid Build Coastguard Worker { U32 const BD = srcPtr[5];
1407*27162e4eSAndroid Build Coastguard Worker blockSizeID = (BD>>4) & _3BITS;
1408*27162e4eSAndroid Build Coastguard Worker /* validate */
1409*27162e4eSAndroid Build Coastguard Worker if (((BD>>7)&_1BIT) != 0) RETURN_ERROR(reservedFlag_set); /* Reserved bit */
1410*27162e4eSAndroid Build Coastguard Worker if (blockSizeID < 4) RETURN_ERROR(maxBlockSize_invalid); /* 4-7 only supported values for the time being */
1411*27162e4eSAndroid Build Coastguard Worker if (((BD>>0)&_4BITS) != 0) RETURN_ERROR(reservedFlag_set); /* Reserved bits */
1412*27162e4eSAndroid Build Coastguard Worker }
1413*27162e4eSAndroid Build Coastguard Worker
1414*27162e4eSAndroid Build Coastguard Worker /* check header */
1415*27162e4eSAndroid Build Coastguard Worker assert(frameHeaderSize > 5);
1416*27162e4eSAndroid Build Coastguard Worker #ifndef FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION
1417*27162e4eSAndroid Build Coastguard Worker { BYTE const HC = LZ4F_headerChecksum(srcPtr+4, frameHeaderSize-5);
1418*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(HC != srcPtr[frameHeaderSize-1], headerChecksum_invalid);
1419*27162e4eSAndroid Build Coastguard Worker }
1420*27162e4eSAndroid Build Coastguard Worker #endif
1421*27162e4eSAndroid Build Coastguard Worker
1422*27162e4eSAndroid Build Coastguard Worker /* save */
1423*27162e4eSAndroid Build Coastguard Worker dctx->frameInfo.blockMode = (LZ4F_blockMode_t)blockMode;
1424*27162e4eSAndroid Build Coastguard Worker dctx->frameInfo.blockChecksumFlag = (LZ4F_blockChecksum_t)blockChecksumFlag;
1425*27162e4eSAndroid Build Coastguard Worker dctx->frameInfo.contentChecksumFlag = (LZ4F_contentChecksum_t)contentChecksumFlag;
1426*27162e4eSAndroid Build Coastguard Worker dctx->frameInfo.blockSizeID = (LZ4F_blockSizeID_t)blockSizeID;
1427*27162e4eSAndroid Build Coastguard Worker dctx->maxBlockSize = LZ4F_getBlockSize((LZ4F_blockSizeID_t)blockSizeID);
1428*27162e4eSAndroid Build Coastguard Worker if (contentSizeFlag) {
1429*27162e4eSAndroid Build Coastguard Worker dctx->frameRemainingSize = dctx->frameInfo.contentSize = LZ4F_readLE64(srcPtr+6);
1430*27162e4eSAndroid Build Coastguard Worker }
1431*27162e4eSAndroid Build Coastguard Worker if (dictIDFlag)
1432*27162e4eSAndroid Build Coastguard Worker dctx->frameInfo.dictID = LZ4F_readLE32(srcPtr + frameHeaderSize - 5);
1433*27162e4eSAndroid Build Coastguard Worker
1434*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_init;
1435*27162e4eSAndroid Build Coastguard Worker
1436*27162e4eSAndroid Build Coastguard Worker return frameHeaderSize;
1437*27162e4eSAndroid Build Coastguard Worker }
1438*27162e4eSAndroid Build Coastguard Worker
1439*27162e4eSAndroid Build Coastguard Worker
1440*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_headerSize() :
1441*27162e4eSAndroid Build Coastguard Worker * @return : size of frame header
1442*27162e4eSAndroid Build Coastguard Worker * or an error code, which can be tested using LZ4F_isError()
1443*27162e4eSAndroid Build Coastguard Worker */
LZ4F_headerSize(const void * src,size_t srcSize)1444*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_headerSize(const void* src, size_t srcSize)
1445*27162e4eSAndroid Build Coastguard Worker {
1446*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(src == NULL, srcPtr_wrong);
1447*27162e4eSAndroid Build Coastguard Worker
1448*27162e4eSAndroid Build Coastguard Worker /* minimal srcSize to determine header size */
1449*27162e4eSAndroid Build Coastguard Worker if (srcSize < LZ4F_MIN_SIZE_TO_KNOW_HEADER_LENGTH)
1450*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR(frameHeader_incomplete);
1451*27162e4eSAndroid Build Coastguard Worker
1452*27162e4eSAndroid Build Coastguard Worker /* special case : skippable frames */
1453*27162e4eSAndroid Build Coastguard Worker if ((LZ4F_readLE32(src) & 0xFFFFFFF0U) == LZ4F_MAGIC_SKIPPABLE_START)
1454*27162e4eSAndroid Build Coastguard Worker return 8;
1455*27162e4eSAndroid Build Coastguard Worker
1456*27162e4eSAndroid Build Coastguard Worker /* control magic number */
1457*27162e4eSAndroid Build Coastguard Worker #ifndef FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION
1458*27162e4eSAndroid Build Coastguard Worker if (LZ4F_readLE32(src) != LZ4F_MAGICNUMBER)
1459*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR(frameType_unknown);
1460*27162e4eSAndroid Build Coastguard Worker #endif
1461*27162e4eSAndroid Build Coastguard Worker
1462*27162e4eSAndroid Build Coastguard Worker /* Frame Header Size */
1463*27162e4eSAndroid Build Coastguard Worker { BYTE const FLG = ((const BYTE*)src)[4];
1464*27162e4eSAndroid Build Coastguard Worker U32 const contentSizeFlag = (FLG>>3) & _1BIT;
1465*27162e4eSAndroid Build Coastguard Worker U32 const dictIDFlag = FLG & _1BIT;
1466*27162e4eSAndroid Build Coastguard Worker return minFHSize + (contentSizeFlag?8:0) + (dictIDFlag?4:0);
1467*27162e4eSAndroid Build Coastguard Worker }
1468*27162e4eSAndroid Build Coastguard Worker }
1469*27162e4eSAndroid Build Coastguard Worker
1470*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_getFrameInfo() :
1471*27162e4eSAndroid Build Coastguard Worker * This function extracts frame parameters (max blockSize, frame checksum, etc.).
1472*27162e4eSAndroid Build Coastguard Worker * Usage is optional. Objective is to provide relevant information for allocation purposes.
1473*27162e4eSAndroid Build Coastguard Worker * This function works in 2 situations :
1474*27162e4eSAndroid Build Coastguard Worker * - At the beginning of a new frame, in which case it will decode this information from `srcBuffer`, and start the decoding process.
1475*27162e4eSAndroid Build Coastguard Worker * Amount of input data provided must be large enough to successfully decode the frame header.
1476*27162e4eSAndroid Build Coastguard Worker * A header size is variable, but is guaranteed to be <= LZ4F_HEADER_SIZE_MAX bytes. It's possible to provide more input data than this minimum.
1477*27162e4eSAndroid Build Coastguard Worker * - After decoding has been started. In which case, no input is read, frame parameters are extracted from dctx.
1478*27162e4eSAndroid Build Coastguard Worker * The number of bytes consumed from srcBuffer will be updated within *srcSizePtr (necessarily <= original value).
1479*27162e4eSAndroid Build Coastguard Worker * Decompression must resume from (srcBuffer + *srcSizePtr).
1480*27162e4eSAndroid Build Coastguard Worker * @return : an hint about how many srcSize bytes LZ4F_decompress() expects for next call,
1481*27162e4eSAndroid Build Coastguard Worker * or an error code which can be tested using LZ4F_isError()
1482*27162e4eSAndroid Build Coastguard Worker * note 1 : in case of error, dctx is not modified. Decoding operations can resume from where they stopped.
1483*27162e4eSAndroid Build Coastguard Worker * note 2 : frame parameters are *copied into* an already allocated LZ4F_frameInfo_t structure.
1484*27162e4eSAndroid Build Coastguard Worker */
LZ4F_getFrameInfo(LZ4F_dctx * dctx,LZ4F_frameInfo_t * frameInfoPtr,const void * srcBuffer,size_t * srcSizePtr)1485*27162e4eSAndroid Build Coastguard Worker LZ4F_errorCode_t LZ4F_getFrameInfo(LZ4F_dctx* dctx,
1486*27162e4eSAndroid Build Coastguard Worker LZ4F_frameInfo_t* frameInfoPtr,
1487*27162e4eSAndroid Build Coastguard Worker const void* srcBuffer, size_t* srcSizePtr)
1488*27162e4eSAndroid Build Coastguard Worker {
1489*27162e4eSAndroid Build Coastguard Worker LZ4F_STATIC_ASSERT(dstage_getFrameHeader < dstage_storeFrameHeader);
1490*27162e4eSAndroid Build Coastguard Worker if (dctx->dStage > dstage_storeFrameHeader) {
1491*27162e4eSAndroid Build Coastguard Worker /* frameInfo already decoded */
1492*27162e4eSAndroid Build Coastguard Worker size_t o=0, i=0;
1493*27162e4eSAndroid Build Coastguard Worker *srcSizePtr = 0;
1494*27162e4eSAndroid Build Coastguard Worker *frameInfoPtr = dctx->frameInfo;
1495*27162e4eSAndroid Build Coastguard Worker /* returns : recommended nb of bytes for LZ4F_decompress() */
1496*27162e4eSAndroid Build Coastguard Worker return LZ4F_decompress(dctx, NULL, &o, NULL, &i, NULL);
1497*27162e4eSAndroid Build Coastguard Worker } else {
1498*27162e4eSAndroid Build Coastguard Worker if (dctx->dStage == dstage_storeFrameHeader) {
1499*27162e4eSAndroid Build Coastguard Worker /* frame decoding already started, in the middle of header => automatic fail */
1500*27162e4eSAndroid Build Coastguard Worker *srcSizePtr = 0;
1501*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR(frameDecoding_alreadyStarted);
1502*27162e4eSAndroid Build Coastguard Worker } else {
1503*27162e4eSAndroid Build Coastguard Worker size_t const hSize = LZ4F_headerSize(srcBuffer, *srcSizePtr);
1504*27162e4eSAndroid Build Coastguard Worker if (LZ4F_isError(hSize)) { *srcSizePtr=0; return hSize; }
1505*27162e4eSAndroid Build Coastguard Worker if (*srcSizePtr < hSize) {
1506*27162e4eSAndroid Build Coastguard Worker *srcSizePtr=0;
1507*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR(frameHeader_incomplete);
1508*27162e4eSAndroid Build Coastguard Worker }
1509*27162e4eSAndroid Build Coastguard Worker
1510*27162e4eSAndroid Build Coastguard Worker { size_t decodeResult = LZ4F_decodeHeader(dctx, srcBuffer, hSize);
1511*27162e4eSAndroid Build Coastguard Worker if (LZ4F_isError(decodeResult)) {
1512*27162e4eSAndroid Build Coastguard Worker *srcSizePtr = 0;
1513*27162e4eSAndroid Build Coastguard Worker } else {
1514*27162e4eSAndroid Build Coastguard Worker *srcSizePtr = decodeResult;
1515*27162e4eSAndroid Build Coastguard Worker decodeResult = BHSize; /* block header size */
1516*27162e4eSAndroid Build Coastguard Worker }
1517*27162e4eSAndroid Build Coastguard Worker *frameInfoPtr = dctx->frameInfo;
1518*27162e4eSAndroid Build Coastguard Worker return decodeResult;
1519*27162e4eSAndroid Build Coastguard Worker } } }
1520*27162e4eSAndroid Build Coastguard Worker }
1521*27162e4eSAndroid Build Coastguard Worker
1522*27162e4eSAndroid Build Coastguard Worker
1523*27162e4eSAndroid Build Coastguard Worker /* LZ4F_updateDict() :
1524*27162e4eSAndroid Build Coastguard Worker * only used for LZ4F_blockLinked mode
1525*27162e4eSAndroid Build Coastguard Worker * Condition : @dstPtr != NULL
1526*27162e4eSAndroid Build Coastguard Worker */
LZ4F_updateDict(LZ4F_dctx * dctx,const BYTE * dstPtr,size_t dstSize,const BYTE * dstBufferStart,unsigned withinTmp)1527*27162e4eSAndroid Build Coastguard Worker static void LZ4F_updateDict(LZ4F_dctx* dctx,
1528*27162e4eSAndroid Build Coastguard Worker const BYTE* dstPtr, size_t dstSize, const BYTE* dstBufferStart,
1529*27162e4eSAndroid Build Coastguard Worker unsigned withinTmp)
1530*27162e4eSAndroid Build Coastguard Worker {
1531*27162e4eSAndroid Build Coastguard Worker assert(dstPtr != NULL);
1532*27162e4eSAndroid Build Coastguard Worker if (dctx->dictSize==0) dctx->dict = (const BYTE*)dstPtr; /* will lead to prefix mode */
1533*27162e4eSAndroid Build Coastguard Worker assert(dctx->dict != NULL);
1534*27162e4eSAndroid Build Coastguard Worker
1535*27162e4eSAndroid Build Coastguard Worker if (dctx->dict + dctx->dictSize == dstPtr) { /* prefix mode, everything within dstBuffer */
1536*27162e4eSAndroid Build Coastguard Worker dctx->dictSize += dstSize;
1537*27162e4eSAndroid Build Coastguard Worker return;
1538*27162e4eSAndroid Build Coastguard Worker }
1539*27162e4eSAndroid Build Coastguard Worker
1540*27162e4eSAndroid Build Coastguard Worker assert(dstPtr >= dstBufferStart);
1541*27162e4eSAndroid Build Coastguard Worker if ((size_t)(dstPtr - dstBufferStart) + dstSize >= 64 KB) { /* history in dstBuffer becomes large enough to become dictionary */
1542*27162e4eSAndroid Build Coastguard Worker dctx->dict = (const BYTE*)dstBufferStart;
1543*27162e4eSAndroid Build Coastguard Worker dctx->dictSize = (size_t)(dstPtr - dstBufferStart) + dstSize;
1544*27162e4eSAndroid Build Coastguard Worker return;
1545*27162e4eSAndroid Build Coastguard Worker }
1546*27162e4eSAndroid Build Coastguard Worker
1547*27162e4eSAndroid Build Coastguard Worker assert(dstSize < 64 KB); /* if dstSize >= 64 KB, dictionary would be set into dstBuffer directly */
1548*27162e4eSAndroid Build Coastguard Worker
1549*27162e4eSAndroid Build Coastguard Worker /* dstBuffer does not contain whole useful history (64 KB), so it must be saved within tmpOutBuffer */
1550*27162e4eSAndroid Build Coastguard Worker assert(dctx->tmpOutBuffer != NULL);
1551*27162e4eSAndroid Build Coastguard Worker
1552*27162e4eSAndroid Build Coastguard Worker if (withinTmp && (dctx->dict == dctx->tmpOutBuffer)) { /* continue history within tmpOutBuffer */
1553*27162e4eSAndroid Build Coastguard Worker /* withinTmp expectation : content of [dstPtr,dstSize] is same as [dict+dictSize,dstSize], so we just extend it */
1554*27162e4eSAndroid Build Coastguard Worker assert(dctx->dict + dctx->dictSize == dctx->tmpOut + dctx->tmpOutStart);
1555*27162e4eSAndroid Build Coastguard Worker dctx->dictSize += dstSize;
1556*27162e4eSAndroid Build Coastguard Worker return;
1557*27162e4eSAndroid Build Coastguard Worker }
1558*27162e4eSAndroid Build Coastguard Worker
1559*27162e4eSAndroid Build Coastguard Worker if (withinTmp) { /* copy relevant dict portion in front of tmpOut within tmpOutBuffer */
1560*27162e4eSAndroid Build Coastguard Worker size_t const preserveSize = (size_t)(dctx->tmpOut - dctx->tmpOutBuffer);
1561*27162e4eSAndroid Build Coastguard Worker size_t copySize = 64 KB - dctx->tmpOutSize;
1562*27162e4eSAndroid Build Coastguard Worker const BYTE* const oldDictEnd = dctx->dict + dctx->dictSize - dctx->tmpOutStart;
1563*27162e4eSAndroid Build Coastguard Worker if (dctx->tmpOutSize > 64 KB) copySize = 0;
1564*27162e4eSAndroid Build Coastguard Worker if (copySize > preserveSize) copySize = preserveSize;
1565*27162e4eSAndroid Build Coastguard Worker
1566*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->tmpOutBuffer + preserveSize - copySize, oldDictEnd - copySize, copySize);
1567*27162e4eSAndroid Build Coastguard Worker
1568*27162e4eSAndroid Build Coastguard Worker dctx->dict = dctx->tmpOutBuffer;
1569*27162e4eSAndroid Build Coastguard Worker dctx->dictSize = preserveSize + dctx->tmpOutStart + dstSize;
1570*27162e4eSAndroid Build Coastguard Worker return;
1571*27162e4eSAndroid Build Coastguard Worker }
1572*27162e4eSAndroid Build Coastguard Worker
1573*27162e4eSAndroid Build Coastguard Worker if (dctx->dict == dctx->tmpOutBuffer) { /* copy dst into tmp to complete dict */
1574*27162e4eSAndroid Build Coastguard Worker if (dctx->dictSize + dstSize > dctx->maxBufferSize) { /* tmp buffer not large enough */
1575*27162e4eSAndroid Build Coastguard Worker size_t const preserveSize = 64 KB - dstSize;
1576*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->tmpOutBuffer, dctx->dict + dctx->dictSize - preserveSize, preserveSize);
1577*27162e4eSAndroid Build Coastguard Worker dctx->dictSize = preserveSize;
1578*27162e4eSAndroid Build Coastguard Worker }
1579*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->tmpOutBuffer + dctx->dictSize, dstPtr, dstSize);
1580*27162e4eSAndroid Build Coastguard Worker dctx->dictSize += dstSize;
1581*27162e4eSAndroid Build Coastguard Worker return;
1582*27162e4eSAndroid Build Coastguard Worker }
1583*27162e4eSAndroid Build Coastguard Worker
1584*27162e4eSAndroid Build Coastguard Worker /* join dict & dest into tmp */
1585*27162e4eSAndroid Build Coastguard Worker { size_t preserveSize = 64 KB - dstSize;
1586*27162e4eSAndroid Build Coastguard Worker if (preserveSize > dctx->dictSize) preserveSize = dctx->dictSize;
1587*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->tmpOutBuffer, dctx->dict + dctx->dictSize - preserveSize, preserveSize);
1588*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->tmpOutBuffer + preserveSize, dstPtr, dstSize);
1589*27162e4eSAndroid Build Coastguard Worker dctx->dict = dctx->tmpOutBuffer;
1590*27162e4eSAndroid Build Coastguard Worker dctx->dictSize = preserveSize + dstSize;
1591*27162e4eSAndroid Build Coastguard Worker }
1592*27162e4eSAndroid Build Coastguard Worker }
1593*27162e4eSAndroid Build Coastguard Worker
1594*27162e4eSAndroid Build Coastguard Worker
1595*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_decompress() :
1596*27162e4eSAndroid Build Coastguard Worker * Call this function repetitively to regenerate compressed data in srcBuffer.
1597*27162e4eSAndroid Build Coastguard Worker * The function will attempt to decode up to *srcSizePtr bytes from srcBuffer
1598*27162e4eSAndroid Build Coastguard Worker * into dstBuffer of capacity *dstSizePtr.
1599*27162e4eSAndroid Build Coastguard Worker *
1600*27162e4eSAndroid Build Coastguard Worker * The number of bytes regenerated into dstBuffer will be provided within *dstSizePtr (necessarily <= original value).
1601*27162e4eSAndroid Build Coastguard Worker *
1602*27162e4eSAndroid Build Coastguard Worker * The number of bytes effectively read from srcBuffer will be provided within *srcSizePtr (necessarily <= original value).
1603*27162e4eSAndroid Build Coastguard Worker * If number of bytes read is < number of bytes provided, then decompression operation is not complete.
1604*27162e4eSAndroid Build Coastguard Worker * Remaining data will have to be presented again in a subsequent invocation.
1605*27162e4eSAndroid Build Coastguard Worker *
1606*27162e4eSAndroid Build Coastguard Worker * The function result is an hint of the better srcSize to use for next call to LZ4F_decompress.
1607*27162e4eSAndroid Build Coastguard Worker * Schematically, it's the size of the current (or remaining) compressed block + header of next block.
1608*27162e4eSAndroid Build Coastguard Worker * Respecting the hint provides a small boost to performance, since it allows less buffer shuffling.
1609*27162e4eSAndroid Build Coastguard Worker * Note that this is just a hint, and it's always possible to any srcSize value.
1610*27162e4eSAndroid Build Coastguard Worker * When a frame is fully decoded, @return will be 0.
1611*27162e4eSAndroid Build Coastguard Worker * If decompression failed, @return is an error code which can be tested using LZ4F_isError().
1612*27162e4eSAndroid Build Coastguard Worker */
LZ4F_decompress(LZ4F_dctx * dctx,void * dstBuffer,size_t * dstSizePtr,const void * srcBuffer,size_t * srcSizePtr,const LZ4F_decompressOptions_t * decompressOptionsPtr)1613*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_decompress(LZ4F_dctx* dctx,
1614*27162e4eSAndroid Build Coastguard Worker void* dstBuffer, size_t* dstSizePtr,
1615*27162e4eSAndroid Build Coastguard Worker const void* srcBuffer, size_t* srcSizePtr,
1616*27162e4eSAndroid Build Coastguard Worker const LZ4F_decompressOptions_t* decompressOptionsPtr)
1617*27162e4eSAndroid Build Coastguard Worker {
1618*27162e4eSAndroid Build Coastguard Worker LZ4F_decompressOptions_t optionsNull;
1619*27162e4eSAndroid Build Coastguard Worker const BYTE* const srcStart = (const BYTE*)srcBuffer;
1620*27162e4eSAndroid Build Coastguard Worker const BYTE* const srcEnd = srcStart + *srcSizePtr;
1621*27162e4eSAndroid Build Coastguard Worker const BYTE* srcPtr = srcStart;
1622*27162e4eSAndroid Build Coastguard Worker BYTE* const dstStart = (BYTE*)dstBuffer;
1623*27162e4eSAndroid Build Coastguard Worker BYTE* const dstEnd = dstStart ? dstStart + *dstSizePtr : NULL;
1624*27162e4eSAndroid Build Coastguard Worker BYTE* dstPtr = dstStart;
1625*27162e4eSAndroid Build Coastguard Worker const BYTE* selectedIn = NULL;
1626*27162e4eSAndroid Build Coastguard Worker unsigned doAnotherStage = 1;
1627*27162e4eSAndroid Build Coastguard Worker size_t nextSrcSizeHint = 1;
1628*27162e4eSAndroid Build Coastguard Worker
1629*27162e4eSAndroid Build Coastguard Worker
1630*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(5, "LZ4F_decompress: src[%p](%u) => dst[%p](%u)",
1631*27162e4eSAndroid Build Coastguard Worker srcBuffer, (unsigned)*srcSizePtr, dstBuffer, (unsigned)*dstSizePtr);
1632*27162e4eSAndroid Build Coastguard Worker if (dstBuffer == NULL) assert(*dstSizePtr == 0);
1633*27162e4eSAndroid Build Coastguard Worker MEM_INIT(&optionsNull, 0, sizeof(optionsNull));
1634*27162e4eSAndroid Build Coastguard Worker if (decompressOptionsPtr==NULL) decompressOptionsPtr = &optionsNull;
1635*27162e4eSAndroid Build Coastguard Worker *srcSizePtr = 0;
1636*27162e4eSAndroid Build Coastguard Worker *dstSizePtr = 0;
1637*27162e4eSAndroid Build Coastguard Worker assert(dctx != NULL);
1638*27162e4eSAndroid Build Coastguard Worker dctx->skipChecksum |= (decompressOptionsPtr->skipChecksums != 0); /* once set, disable for the remainder of the frame */
1639*27162e4eSAndroid Build Coastguard Worker
1640*27162e4eSAndroid Build Coastguard Worker /* behaves as a state machine */
1641*27162e4eSAndroid Build Coastguard Worker
1642*27162e4eSAndroid Build Coastguard Worker while (doAnotherStage) {
1643*27162e4eSAndroid Build Coastguard Worker
1644*27162e4eSAndroid Build Coastguard Worker switch(dctx->dStage)
1645*27162e4eSAndroid Build Coastguard Worker {
1646*27162e4eSAndroid Build Coastguard Worker
1647*27162e4eSAndroid Build Coastguard Worker case dstage_getFrameHeader:
1648*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(6, "dstage_getFrameHeader");
1649*27162e4eSAndroid Build Coastguard Worker if ((size_t)(srcEnd-srcPtr) >= maxFHSize) { /* enough to decode - shortcut */
1650*27162e4eSAndroid Build Coastguard Worker size_t const hSize = LZ4F_decodeHeader(dctx, srcPtr, (size_t)(srcEnd-srcPtr)); /* will update dStage appropriately */
1651*27162e4eSAndroid Build Coastguard Worker FORWARD_IF_ERROR(hSize);
1652*27162e4eSAndroid Build Coastguard Worker srcPtr += hSize;
1653*27162e4eSAndroid Build Coastguard Worker break;
1654*27162e4eSAndroid Build Coastguard Worker }
1655*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize = 0;
1656*27162e4eSAndroid Build Coastguard Worker if (srcEnd-srcPtr == 0) return minFHSize; /* 0-size input */
1657*27162e4eSAndroid Build Coastguard Worker dctx->tmpInTarget = minFHSize; /* minimum size to decode header */
1658*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_storeFrameHeader;
1659*27162e4eSAndroid Build Coastguard Worker /* fall-through */
1660*27162e4eSAndroid Build Coastguard Worker
1661*27162e4eSAndroid Build Coastguard Worker case dstage_storeFrameHeader:
1662*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(6, "dstage_storeFrameHeader");
1663*27162e4eSAndroid Build Coastguard Worker { size_t const sizeToCopy = MIN(dctx->tmpInTarget - dctx->tmpInSize, (size_t)(srcEnd - srcPtr));
1664*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->header + dctx->tmpInSize, srcPtr, sizeToCopy);
1665*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize += sizeToCopy;
1666*27162e4eSAndroid Build Coastguard Worker srcPtr += sizeToCopy;
1667*27162e4eSAndroid Build Coastguard Worker }
1668*27162e4eSAndroid Build Coastguard Worker if (dctx->tmpInSize < dctx->tmpInTarget) {
1669*27162e4eSAndroid Build Coastguard Worker nextSrcSizeHint = (dctx->tmpInTarget - dctx->tmpInSize) + BHSize; /* rest of header + nextBlockHeader */
1670*27162e4eSAndroid Build Coastguard Worker doAnotherStage = 0; /* not enough src data, ask for some more */
1671*27162e4eSAndroid Build Coastguard Worker break;
1672*27162e4eSAndroid Build Coastguard Worker }
1673*27162e4eSAndroid Build Coastguard Worker FORWARD_IF_ERROR( LZ4F_decodeHeader(dctx, dctx->header, dctx->tmpInTarget) ); /* will update dStage appropriately */
1674*27162e4eSAndroid Build Coastguard Worker break;
1675*27162e4eSAndroid Build Coastguard Worker
1676*27162e4eSAndroid Build Coastguard Worker case dstage_init:
1677*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(6, "dstage_init");
1678*27162e4eSAndroid Build Coastguard Worker if (dctx->frameInfo.contentChecksumFlag) (void)XXH32_reset(&(dctx->xxh), 0);
1679*27162e4eSAndroid Build Coastguard Worker /* internal buffers allocation */
1680*27162e4eSAndroid Build Coastguard Worker { size_t const bufferNeeded = dctx->maxBlockSize
1681*27162e4eSAndroid Build Coastguard Worker + ((dctx->frameInfo.blockMode==LZ4F_blockLinked) ? 128 KB : 0);
1682*27162e4eSAndroid Build Coastguard Worker if (bufferNeeded > dctx->maxBufferSize) { /* tmp buffers too small */
1683*27162e4eSAndroid Build Coastguard Worker dctx->maxBufferSize = 0; /* ensure allocation will be re-attempted on next entry*/
1684*27162e4eSAndroid Build Coastguard Worker LZ4F_free(dctx->tmpIn, dctx->cmem);
1685*27162e4eSAndroid Build Coastguard Worker dctx->tmpIn = (BYTE*)LZ4F_malloc(dctx->maxBlockSize + BFSize /* block checksum */, dctx->cmem);
1686*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(dctx->tmpIn == NULL, allocation_failed);
1687*27162e4eSAndroid Build Coastguard Worker LZ4F_free(dctx->tmpOutBuffer, dctx->cmem);
1688*27162e4eSAndroid Build Coastguard Worker dctx->tmpOutBuffer= (BYTE*)LZ4F_malloc(bufferNeeded, dctx->cmem);
1689*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(dctx->tmpOutBuffer== NULL, allocation_failed);
1690*27162e4eSAndroid Build Coastguard Worker dctx->maxBufferSize = bufferNeeded;
1691*27162e4eSAndroid Build Coastguard Worker } }
1692*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize = 0;
1693*27162e4eSAndroid Build Coastguard Worker dctx->tmpInTarget = 0;
1694*27162e4eSAndroid Build Coastguard Worker dctx->tmpOut = dctx->tmpOutBuffer;
1695*27162e4eSAndroid Build Coastguard Worker dctx->tmpOutStart = 0;
1696*27162e4eSAndroid Build Coastguard Worker dctx->tmpOutSize = 0;
1697*27162e4eSAndroid Build Coastguard Worker
1698*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_getBlockHeader;
1699*27162e4eSAndroid Build Coastguard Worker /* fall-through */
1700*27162e4eSAndroid Build Coastguard Worker
1701*27162e4eSAndroid Build Coastguard Worker case dstage_getBlockHeader:
1702*27162e4eSAndroid Build Coastguard Worker if ((size_t)(srcEnd - srcPtr) >= BHSize) {
1703*27162e4eSAndroid Build Coastguard Worker selectedIn = srcPtr;
1704*27162e4eSAndroid Build Coastguard Worker srcPtr += BHSize;
1705*27162e4eSAndroid Build Coastguard Worker } else {
1706*27162e4eSAndroid Build Coastguard Worker /* not enough input to read cBlockSize field */
1707*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize = 0;
1708*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_storeBlockHeader;
1709*27162e4eSAndroid Build Coastguard Worker }
1710*27162e4eSAndroid Build Coastguard Worker
1711*27162e4eSAndroid Build Coastguard Worker if (dctx->dStage == dstage_storeBlockHeader) /* can be skipped */
1712*27162e4eSAndroid Build Coastguard Worker case dstage_storeBlockHeader:
1713*27162e4eSAndroid Build Coastguard Worker { size_t const remainingInput = (size_t)(srcEnd - srcPtr);
1714*27162e4eSAndroid Build Coastguard Worker size_t const wantedData = BHSize - dctx->tmpInSize;
1715*27162e4eSAndroid Build Coastguard Worker size_t const sizeToCopy = MIN(wantedData, remainingInput);
1716*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->tmpIn + dctx->tmpInSize, srcPtr, sizeToCopy);
1717*27162e4eSAndroid Build Coastguard Worker srcPtr += sizeToCopy;
1718*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize += sizeToCopy;
1719*27162e4eSAndroid Build Coastguard Worker
1720*27162e4eSAndroid Build Coastguard Worker if (dctx->tmpInSize < BHSize) { /* not enough input for cBlockSize */
1721*27162e4eSAndroid Build Coastguard Worker nextSrcSizeHint = BHSize - dctx->tmpInSize;
1722*27162e4eSAndroid Build Coastguard Worker doAnotherStage = 0;
1723*27162e4eSAndroid Build Coastguard Worker break;
1724*27162e4eSAndroid Build Coastguard Worker }
1725*27162e4eSAndroid Build Coastguard Worker selectedIn = dctx->tmpIn;
1726*27162e4eSAndroid Build Coastguard Worker } /* if (dctx->dStage == dstage_storeBlockHeader) */
1727*27162e4eSAndroid Build Coastguard Worker
1728*27162e4eSAndroid Build Coastguard Worker /* decode block header */
1729*27162e4eSAndroid Build Coastguard Worker { U32 const blockHeader = LZ4F_readLE32(selectedIn);
1730*27162e4eSAndroid Build Coastguard Worker size_t const nextCBlockSize = blockHeader & 0x7FFFFFFFU;
1731*27162e4eSAndroid Build Coastguard Worker size_t const crcSize = dctx->frameInfo.blockChecksumFlag * BFSize;
1732*27162e4eSAndroid Build Coastguard Worker if (blockHeader==0) { /* frameEnd signal, no more block */
1733*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(5, "end of frame");
1734*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_getSuffix;
1735*27162e4eSAndroid Build Coastguard Worker break;
1736*27162e4eSAndroid Build Coastguard Worker }
1737*27162e4eSAndroid Build Coastguard Worker if (nextCBlockSize > dctx->maxBlockSize) {
1738*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR(maxBlockSize_invalid);
1739*27162e4eSAndroid Build Coastguard Worker }
1740*27162e4eSAndroid Build Coastguard Worker if (blockHeader & LZ4F_BLOCKUNCOMPRESSED_FLAG) {
1741*27162e4eSAndroid Build Coastguard Worker /* next block is uncompressed */
1742*27162e4eSAndroid Build Coastguard Worker dctx->tmpInTarget = nextCBlockSize;
1743*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(5, "next block is uncompressed (size %u)", (U32)nextCBlockSize);
1744*27162e4eSAndroid Build Coastguard Worker if (dctx->frameInfo.blockChecksumFlag) {
1745*27162e4eSAndroid Build Coastguard Worker (void)XXH32_reset(&dctx->blockChecksum, 0);
1746*27162e4eSAndroid Build Coastguard Worker }
1747*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_copyDirect;
1748*27162e4eSAndroid Build Coastguard Worker break;
1749*27162e4eSAndroid Build Coastguard Worker }
1750*27162e4eSAndroid Build Coastguard Worker /* next block is a compressed block */
1751*27162e4eSAndroid Build Coastguard Worker dctx->tmpInTarget = nextCBlockSize + crcSize;
1752*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_getCBlock;
1753*27162e4eSAndroid Build Coastguard Worker if (dstPtr==dstEnd || srcPtr==srcEnd) {
1754*27162e4eSAndroid Build Coastguard Worker nextSrcSizeHint = BHSize + nextCBlockSize + crcSize;
1755*27162e4eSAndroid Build Coastguard Worker doAnotherStage = 0;
1756*27162e4eSAndroid Build Coastguard Worker }
1757*27162e4eSAndroid Build Coastguard Worker break;
1758*27162e4eSAndroid Build Coastguard Worker }
1759*27162e4eSAndroid Build Coastguard Worker
1760*27162e4eSAndroid Build Coastguard Worker case dstage_copyDirect: /* uncompressed block */
1761*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(6, "dstage_copyDirect");
1762*27162e4eSAndroid Build Coastguard Worker { size_t sizeToCopy;
1763*27162e4eSAndroid Build Coastguard Worker if (dstPtr == NULL) {
1764*27162e4eSAndroid Build Coastguard Worker sizeToCopy = 0;
1765*27162e4eSAndroid Build Coastguard Worker } else {
1766*27162e4eSAndroid Build Coastguard Worker size_t const minBuffSize = MIN((size_t)(srcEnd-srcPtr), (size_t)(dstEnd-dstPtr));
1767*27162e4eSAndroid Build Coastguard Worker sizeToCopy = MIN(dctx->tmpInTarget, minBuffSize);
1768*27162e4eSAndroid Build Coastguard Worker memcpy(dstPtr, srcPtr, sizeToCopy);
1769*27162e4eSAndroid Build Coastguard Worker if (!dctx->skipChecksum) {
1770*27162e4eSAndroid Build Coastguard Worker if (dctx->frameInfo.blockChecksumFlag) {
1771*27162e4eSAndroid Build Coastguard Worker (void)XXH32_update(&dctx->blockChecksum, srcPtr, sizeToCopy);
1772*27162e4eSAndroid Build Coastguard Worker }
1773*27162e4eSAndroid Build Coastguard Worker if (dctx->frameInfo.contentChecksumFlag)
1774*27162e4eSAndroid Build Coastguard Worker (void)XXH32_update(&dctx->xxh, srcPtr, sizeToCopy);
1775*27162e4eSAndroid Build Coastguard Worker }
1776*27162e4eSAndroid Build Coastguard Worker if (dctx->frameInfo.contentSize)
1777*27162e4eSAndroid Build Coastguard Worker dctx->frameRemainingSize -= sizeToCopy;
1778*27162e4eSAndroid Build Coastguard Worker
1779*27162e4eSAndroid Build Coastguard Worker /* history management (linked blocks only)*/
1780*27162e4eSAndroid Build Coastguard Worker if (dctx->frameInfo.blockMode == LZ4F_blockLinked) {
1781*27162e4eSAndroid Build Coastguard Worker LZ4F_updateDict(dctx, dstPtr, sizeToCopy, dstStart, 0);
1782*27162e4eSAndroid Build Coastguard Worker }
1783*27162e4eSAndroid Build Coastguard Worker srcPtr += sizeToCopy;
1784*27162e4eSAndroid Build Coastguard Worker dstPtr += sizeToCopy;
1785*27162e4eSAndroid Build Coastguard Worker }
1786*27162e4eSAndroid Build Coastguard Worker if (sizeToCopy == dctx->tmpInTarget) { /* all done */
1787*27162e4eSAndroid Build Coastguard Worker if (dctx->frameInfo.blockChecksumFlag) {
1788*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize = 0;
1789*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_getBlockChecksum;
1790*27162e4eSAndroid Build Coastguard Worker } else
1791*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_getBlockHeader; /* new block */
1792*27162e4eSAndroid Build Coastguard Worker break;
1793*27162e4eSAndroid Build Coastguard Worker }
1794*27162e4eSAndroid Build Coastguard Worker dctx->tmpInTarget -= sizeToCopy; /* need to copy more */
1795*27162e4eSAndroid Build Coastguard Worker }
1796*27162e4eSAndroid Build Coastguard Worker nextSrcSizeHint = dctx->tmpInTarget +
1797*27162e4eSAndroid Build Coastguard Worker +(dctx->frameInfo.blockChecksumFlag ? BFSize : 0)
1798*27162e4eSAndroid Build Coastguard Worker + BHSize /* next header size */;
1799*27162e4eSAndroid Build Coastguard Worker doAnotherStage = 0;
1800*27162e4eSAndroid Build Coastguard Worker break;
1801*27162e4eSAndroid Build Coastguard Worker
1802*27162e4eSAndroid Build Coastguard Worker /* check block checksum for recently transferred uncompressed block */
1803*27162e4eSAndroid Build Coastguard Worker case dstage_getBlockChecksum:
1804*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(6, "dstage_getBlockChecksum");
1805*27162e4eSAndroid Build Coastguard Worker { const void* crcSrc;
1806*27162e4eSAndroid Build Coastguard Worker if ((srcEnd-srcPtr >= 4) && (dctx->tmpInSize==0)) {
1807*27162e4eSAndroid Build Coastguard Worker crcSrc = srcPtr;
1808*27162e4eSAndroid Build Coastguard Worker srcPtr += 4;
1809*27162e4eSAndroid Build Coastguard Worker } else {
1810*27162e4eSAndroid Build Coastguard Worker size_t const stillToCopy = 4 - dctx->tmpInSize;
1811*27162e4eSAndroid Build Coastguard Worker size_t const sizeToCopy = MIN(stillToCopy, (size_t)(srcEnd-srcPtr));
1812*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->header + dctx->tmpInSize, srcPtr, sizeToCopy);
1813*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize += sizeToCopy;
1814*27162e4eSAndroid Build Coastguard Worker srcPtr += sizeToCopy;
1815*27162e4eSAndroid Build Coastguard Worker if (dctx->tmpInSize < 4) { /* all input consumed */
1816*27162e4eSAndroid Build Coastguard Worker doAnotherStage = 0;
1817*27162e4eSAndroid Build Coastguard Worker break;
1818*27162e4eSAndroid Build Coastguard Worker }
1819*27162e4eSAndroid Build Coastguard Worker crcSrc = dctx->header;
1820*27162e4eSAndroid Build Coastguard Worker }
1821*27162e4eSAndroid Build Coastguard Worker if (!dctx->skipChecksum) {
1822*27162e4eSAndroid Build Coastguard Worker U32 const readCRC = LZ4F_readLE32(crcSrc);
1823*27162e4eSAndroid Build Coastguard Worker U32 const calcCRC = XXH32_digest(&dctx->blockChecksum);
1824*27162e4eSAndroid Build Coastguard Worker #ifndef FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION
1825*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(6, "compare block checksum");
1826*27162e4eSAndroid Build Coastguard Worker if (readCRC != calcCRC) {
1827*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(4, "incorrect block checksum: %08X != %08X",
1828*27162e4eSAndroid Build Coastguard Worker readCRC, calcCRC);
1829*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR(blockChecksum_invalid);
1830*27162e4eSAndroid Build Coastguard Worker }
1831*27162e4eSAndroid Build Coastguard Worker #else
1832*27162e4eSAndroid Build Coastguard Worker (void)readCRC;
1833*27162e4eSAndroid Build Coastguard Worker (void)calcCRC;
1834*27162e4eSAndroid Build Coastguard Worker #endif
1835*27162e4eSAndroid Build Coastguard Worker } }
1836*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_getBlockHeader; /* new block */
1837*27162e4eSAndroid Build Coastguard Worker break;
1838*27162e4eSAndroid Build Coastguard Worker
1839*27162e4eSAndroid Build Coastguard Worker case dstage_getCBlock:
1840*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(6, "dstage_getCBlock");
1841*27162e4eSAndroid Build Coastguard Worker if ((size_t)(srcEnd-srcPtr) < dctx->tmpInTarget) {
1842*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize = 0;
1843*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_storeCBlock;
1844*27162e4eSAndroid Build Coastguard Worker break;
1845*27162e4eSAndroid Build Coastguard Worker }
1846*27162e4eSAndroid Build Coastguard Worker /* input large enough to read full block directly */
1847*27162e4eSAndroid Build Coastguard Worker selectedIn = srcPtr;
1848*27162e4eSAndroid Build Coastguard Worker srcPtr += dctx->tmpInTarget;
1849*27162e4eSAndroid Build Coastguard Worker
1850*27162e4eSAndroid Build Coastguard Worker if (0) /* always jump over next block */
1851*27162e4eSAndroid Build Coastguard Worker case dstage_storeCBlock:
1852*27162e4eSAndroid Build Coastguard Worker { size_t const wantedData = dctx->tmpInTarget - dctx->tmpInSize;
1853*27162e4eSAndroid Build Coastguard Worker size_t const inputLeft = (size_t)(srcEnd-srcPtr);
1854*27162e4eSAndroid Build Coastguard Worker size_t const sizeToCopy = MIN(wantedData, inputLeft);
1855*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->tmpIn + dctx->tmpInSize, srcPtr, sizeToCopy);
1856*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize += sizeToCopy;
1857*27162e4eSAndroid Build Coastguard Worker srcPtr += sizeToCopy;
1858*27162e4eSAndroid Build Coastguard Worker if (dctx->tmpInSize < dctx->tmpInTarget) { /* need more input */
1859*27162e4eSAndroid Build Coastguard Worker nextSrcSizeHint = (dctx->tmpInTarget - dctx->tmpInSize)
1860*27162e4eSAndroid Build Coastguard Worker + (dctx->frameInfo.blockChecksumFlag ? BFSize : 0)
1861*27162e4eSAndroid Build Coastguard Worker + BHSize /* next header size */;
1862*27162e4eSAndroid Build Coastguard Worker doAnotherStage = 0;
1863*27162e4eSAndroid Build Coastguard Worker break;
1864*27162e4eSAndroid Build Coastguard Worker }
1865*27162e4eSAndroid Build Coastguard Worker selectedIn = dctx->tmpIn;
1866*27162e4eSAndroid Build Coastguard Worker }
1867*27162e4eSAndroid Build Coastguard Worker
1868*27162e4eSAndroid Build Coastguard Worker /* At this stage, input is large enough to decode a block */
1869*27162e4eSAndroid Build Coastguard Worker
1870*27162e4eSAndroid Build Coastguard Worker /* First, decode and control block checksum if it exists */
1871*27162e4eSAndroid Build Coastguard Worker if (dctx->frameInfo.blockChecksumFlag) {
1872*27162e4eSAndroid Build Coastguard Worker assert(dctx->tmpInTarget >= 4);
1873*27162e4eSAndroid Build Coastguard Worker dctx->tmpInTarget -= 4;
1874*27162e4eSAndroid Build Coastguard Worker assert(selectedIn != NULL); /* selectedIn is defined at this stage (either srcPtr, or dctx->tmpIn) */
1875*27162e4eSAndroid Build Coastguard Worker { U32 const readBlockCrc = LZ4F_readLE32(selectedIn + dctx->tmpInTarget);
1876*27162e4eSAndroid Build Coastguard Worker U32 const calcBlockCrc = XXH32(selectedIn, dctx->tmpInTarget, 0);
1877*27162e4eSAndroid Build Coastguard Worker #ifndef FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION
1878*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(readBlockCrc != calcBlockCrc, blockChecksum_invalid);
1879*27162e4eSAndroid Build Coastguard Worker #else
1880*27162e4eSAndroid Build Coastguard Worker (void)readBlockCrc;
1881*27162e4eSAndroid Build Coastguard Worker (void)calcBlockCrc;
1882*27162e4eSAndroid Build Coastguard Worker #endif
1883*27162e4eSAndroid Build Coastguard Worker } }
1884*27162e4eSAndroid Build Coastguard Worker
1885*27162e4eSAndroid Build Coastguard Worker /* decode directly into destination buffer if there is enough room */
1886*27162e4eSAndroid Build Coastguard Worker if ( ((size_t)(dstEnd-dstPtr) >= dctx->maxBlockSize)
1887*27162e4eSAndroid Build Coastguard Worker /* unless the dictionary is stored in tmpOut:
1888*27162e4eSAndroid Build Coastguard Worker * in which case it's faster to decode within tmpOut
1889*27162e4eSAndroid Build Coastguard Worker * to benefit from prefix speedup */
1890*27162e4eSAndroid Build Coastguard Worker && !(dctx->dict!= NULL && (const BYTE*)dctx->dict + dctx->dictSize == dctx->tmpOut) )
1891*27162e4eSAndroid Build Coastguard Worker {
1892*27162e4eSAndroid Build Coastguard Worker const char* dict = (const char*)dctx->dict;
1893*27162e4eSAndroid Build Coastguard Worker size_t dictSize = dctx->dictSize;
1894*27162e4eSAndroid Build Coastguard Worker int decodedSize;
1895*27162e4eSAndroid Build Coastguard Worker assert(dstPtr != NULL);
1896*27162e4eSAndroid Build Coastguard Worker if (dict && dictSize > 1 GB) {
1897*27162e4eSAndroid Build Coastguard Worker /* overflow control : dctx->dictSize is an int, avoid truncation / sign issues */
1898*27162e4eSAndroid Build Coastguard Worker dict += dictSize - 64 KB;
1899*27162e4eSAndroid Build Coastguard Worker dictSize = 64 KB;
1900*27162e4eSAndroid Build Coastguard Worker }
1901*27162e4eSAndroid Build Coastguard Worker decodedSize = LZ4_decompress_safe_usingDict(
1902*27162e4eSAndroid Build Coastguard Worker (const char*)selectedIn, (char*)dstPtr,
1903*27162e4eSAndroid Build Coastguard Worker (int)dctx->tmpInTarget, (int)dctx->maxBlockSize,
1904*27162e4eSAndroid Build Coastguard Worker dict, (int)dictSize);
1905*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(decodedSize < 0, decompressionFailed);
1906*27162e4eSAndroid Build Coastguard Worker if ((dctx->frameInfo.contentChecksumFlag) && (!dctx->skipChecksum))
1907*27162e4eSAndroid Build Coastguard Worker XXH32_update(&(dctx->xxh), dstPtr, (size_t)decodedSize);
1908*27162e4eSAndroid Build Coastguard Worker if (dctx->frameInfo.contentSize)
1909*27162e4eSAndroid Build Coastguard Worker dctx->frameRemainingSize -= (size_t)decodedSize;
1910*27162e4eSAndroid Build Coastguard Worker
1911*27162e4eSAndroid Build Coastguard Worker /* dictionary management */
1912*27162e4eSAndroid Build Coastguard Worker if (dctx->frameInfo.blockMode==LZ4F_blockLinked) {
1913*27162e4eSAndroid Build Coastguard Worker LZ4F_updateDict(dctx, dstPtr, (size_t)decodedSize, dstStart, 0);
1914*27162e4eSAndroid Build Coastguard Worker }
1915*27162e4eSAndroid Build Coastguard Worker
1916*27162e4eSAndroid Build Coastguard Worker dstPtr += decodedSize;
1917*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_getBlockHeader; /* end of block, let's get another one */
1918*27162e4eSAndroid Build Coastguard Worker break;
1919*27162e4eSAndroid Build Coastguard Worker }
1920*27162e4eSAndroid Build Coastguard Worker
1921*27162e4eSAndroid Build Coastguard Worker /* not enough place into dst : decode into tmpOut */
1922*27162e4eSAndroid Build Coastguard Worker
1923*27162e4eSAndroid Build Coastguard Worker /* manage dictionary */
1924*27162e4eSAndroid Build Coastguard Worker if (dctx->frameInfo.blockMode == LZ4F_blockLinked) {
1925*27162e4eSAndroid Build Coastguard Worker if (dctx->dict == dctx->tmpOutBuffer) {
1926*27162e4eSAndroid Build Coastguard Worker /* truncate dictionary to 64 KB if too big */
1927*27162e4eSAndroid Build Coastguard Worker if (dctx->dictSize > 128 KB) {
1928*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->tmpOutBuffer, dctx->dict + dctx->dictSize - 64 KB, 64 KB);
1929*27162e4eSAndroid Build Coastguard Worker dctx->dictSize = 64 KB;
1930*27162e4eSAndroid Build Coastguard Worker }
1931*27162e4eSAndroid Build Coastguard Worker dctx->tmpOut = dctx->tmpOutBuffer + dctx->dictSize;
1932*27162e4eSAndroid Build Coastguard Worker } else { /* dict not within tmpOut */
1933*27162e4eSAndroid Build Coastguard Worker size_t const reservedDictSpace = MIN(dctx->dictSize, 64 KB);
1934*27162e4eSAndroid Build Coastguard Worker dctx->tmpOut = dctx->tmpOutBuffer + reservedDictSpace;
1935*27162e4eSAndroid Build Coastguard Worker } }
1936*27162e4eSAndroid Build Coastguard Worker
1937*27162e4eSAndroid Build Coastguard Worker /* Decode block into tmpOut */
1938*27162e4eSAndroid Build Coastguard Worker { const char* dict = (const char*)dctx->dict;
1939*27162e4eSAndroid Build Coastguard Worker size_t dictSize = dctx->dictSize;
1940*27162e4eSAndroid Build Coastguard Worker int decodedSize;
1941*27162e4eSAndroid Build Coastguard Worker if (dict && dictSize > 1 GB) {
1942*27162e4eSAndroid Build Coastguard Worker /* the dictSize param is an int, avoid truncation / sign issues */
1943*27162e4eSAndroid Build Coastguard Worker dict += dictSize - 64 KB;
1944*27162e4eSAndroid Build Coastguard Worker dictSize = 64 KB;
1945*27162e4eSAndroid Build Coastguard Worker }
1946*27162e4eSAndroid Build Coastguard Worker decodedSize = LZ4_decompress_safe_usingDict(
1947*27162e4eSAndroid Build Coastguard Worker (const char*)selectedIn, (char*)dctx->tmpOut,
1948*27162e4eSAndroid Build Coastguard Worker (int)dctx->tmpInTarget, (int)dctx->maxBlockSize,
1949*27162e4eSAndroid Build Coastguard Worker dict, (int)dictSize);
1950*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(decodedSize < 0, decompressionFailed);
1951*27162e4eSAndroid Build Coastguard Worker if (dctx->frameInfo.contentChecksumFlag && !dctx->skipChecksum)
1952*27162e4eSAndroid Build Coastguard Worker XXH32_update(&(dctx->xxh), dctx->tmpOut, (size_t)decodedSize);
1953*27162e4eSAndroid Build Coastguard Worker if (dctx->frameInfo.contentSize)
1954*27162e4eSAndroid Build Coastguard Worker dctx->frameRemainingSize -= (size_t)decodedSize;
1955*27162e4eSAndroid Build Coastguard Worker dctx->tmpOutSize = (size_t)decodedSize;
1956*27162e4eSAndroid Build Coastguard Worker dctx->tmpOutStart = 0;
1957*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_flushOut;
1958*27162e4eSAndroid Build Coastguard Worker }
1959*27162e4eSAndroid Build Coastguard Worker /* fall-through */
1960*27162e4eSAndroid Build Coastguard Worker
1961*27162e4eSAndroid Build Coastguard Worker case dstage_flushOut: /* flush decoded data from tmpOut to dstBuffer */
1962*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(6, "dstage_flushOut");
1963*27162e4eSAndroid Build Coastguard Worker if (dstPtr != NULL) {
1964*27162e4eSAndroid Build Coastguard Worker size_t const sizeToCopy = MIN(dctx->tmpOutSize - dctx->tmpOutStart, (size_t)(dstEnd-dstPtr));
1965*27162e4eSAndroid Build Coastguard Worker memcpy(dstPtr, dctx->tmpOut + dctx->tmpOutStart, sizeToCopy);
1966*27162e4eSAndroid Build Coastguard Worker
1967*27162e4eSAndroid Build Coastguard Worker /* dictionary management */
1968*27162e4eSAndroid Build Coastguard Worker if (dctx->frameInfo.blockMode == LZ4F_blockLinked)
1969*27162e4eSAndroid Build Coastguard Worker LZ4F_updateDict(dctx, dstPtr, sizeToCopy, dstStart, 1 /*withinTmp*/);
1970*27162e4eSAndroid Build Coastguard Worker
1971*27162e4eSAndroid Build Coastguard Worker dctx->tmpOutStart += sizeToCopy;
1972*27162e4eSAndroid Build Coastguard Worker dstPtr += sizeToCopy;
1973*27162e4eSAndroid Build Coastguard Worker }
1974*27162e4eSAndroid Build Coastguard Worker if (dctx->tmpOutStart == dctx->tmpOutSize) { /* all flushed */
1975*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_getBlockHeader; /* get next block */
1976*27162e4eSAndroid Build Coastguard Worker break;
1977*27162e4eSAndroid Build Coastguard Worker }
1978*27162e4eSAndroid Build Coastguard Worker /* could not flush everything : stop there, just request a block header */
1979*27162e4eSAndroid Build Coastguard Worker doAnotherStage = 0;
1980*27162e4eSAndroid Build Coastguard Worker nextSrcSizeHint = BHSize;
1981*27162e4eSAndroid Build Coastguard Worker break;
1982*27162e4eSAndroid Build Coastguard Worker
1983*27162e4eSAndroid Build Coastguard Worker case dstage_getSuffix:
1984*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(dctx->frameRemainingSize, frameSize_wrong); /* incorrect frame size decoded */
1985*27162e4eSAndroid Build Coastguard Worker if (!dctx->frameInfo.contentChecksumFlag) { /* no checksum, frame is completed */
1986*27162e4eSAndroid Build Coastguard Worker nextSrcSizeHint = 0;
1987*27162e4eSAndroid Build Coastguard Worker LZ4F_resetDecompressionContext(dctx);
1988*27162e4eSAndroid Build Coastguard Worker doAnotherStage = 0;
1989*27162e4eSAndroid Build Coastguard Worker break;
1990*27162e4eSAndroid Build Coastguard Worker }
1991*27162e4eSAndroid Build Coastguard Worker if ((srcEnd - srcPtr) < 4) { /* not enough size for entire CRC */
1992*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize = 0;
1993*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_storeSuffix;
1994*27162e4eSAndroid Build Coastguard Worker } else {
1995*27162e4eSAndroid Build Coastguard Worker selectedIn = srcPtr;
1996*27162e4eSAndroid Build Coastguard Worker srcPtr += 4;
1997*27162e4eSAndroid Build Coastguard Worker }
1998*27162e4eSAndroid Build Coastguard Worker
1999*27162e4eSAndroid Build Coastguard Worker if (dctx->dStage == dstage_storeSuffix) /* can be skipped */
2000*27162e4eSAndroid Build Coastguard Worker case dstage_storeSuffix:
2001*27162e4eSAndroid Build Coastguard Worker { size_t const remainingInput = (size_t)(srcEnd - srcPtr);
2002*27162e4eSAndroid Build Coastguard Worker size_t const wantedData = 4 - dctx->tmpInSize;
2003*27162e4eSAndroid Build Coastguard Worker size_t const sizeToCopy = MIN(wantedData, remainingInput);
2004*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->tmpIn + dctx->tmpInSize, srcPtr, sizeToCopy);
2005*27162e4eSAndroid Build Coastguard Worker srcPtr += sizeToCopy;
2006*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize += sizeToCopy;
2007*27162e4eSAndroid Build Coastguard Worker if (dctx->tmpInSize < 4) { /* not enough input to read complete suffix */
2008*27162e4eSAndroid Build Coastguard Worker nextSrcSizeHint = 4 - dctx->tmpInSize;
2009*27162e4eSAndroid Build Coastguard Worker doAnotherStage=0;
2010*27162e4eSAndroid Build Coastguard Worker break;
2011*27162e4eSAndroid Build Coastguard Worker }
2012*27162e4eSAndroid Build Coastguard Worker selectedIn = dctx->tmpIn;
2013*27162e4eSAndroid Build Coastguard Worker } /* if (dctx->dStage == dstage_storeSuffix) */
2014*27162e4eSAndroid Build Coastguard Worker
2015*27162e4eSAndroid Build Coastguard Worker /* case dstage_checkSuffix: */ /* no direct entry, avoid initialization risks */
2016*27162e4eSAndroid Build Coastguard Worker if (!dctx->skipChecksum) {
2017*27162e4eSAndroid Build Coastguard Worker U32 const readCRC = LZ4F_readLE32(selectedIn);
2018*27162e4eSAndroid Build Coastguard Worker U32 const resultCRC = XXH32_digest(&(dctx->xxh));
2019*27162e4eSAndroid Build Coastguard Worker DEBUGLOG(4, "frame checksum: stored 0x%0X vs 0x%0X processed", readCRC, resultCRC);
2020*27162e4eSAndroid Build Coastguard Worker #ifndef FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION
2021*27162e4eSAndroid Build Coastguard Worker RETURN_ERROR_IF(readCRC != resultCRC, contentChecksum_invalid);
2022*27162e4eSAndroid Build Coastguard Worker #else
2023*27162e4eSAndroid Build Coastguard Worker (void)readCRC;
2024*27162e4eSAndroid Build Coastguard Worker (void)resultCRC;
2025*27162e4eSAndroid Build Coastguard Worker #endif
2026*27162e4eSAndroid Build Coastguard Worker }
2027*27162e4eSAndroid Build Coastguard Worker nextSrcSizeHint = 0;
2028*27162e4eSAndroid Build Coastguard Worker LZ4F_resetDecompressionContext(dctx);
2029*27162e4eSAndroid Build Coastguard Worker doAnotherStage = 0;
2030*27162e4eSAndroid Build Coastguard Worker break;
2031*27162e4eSAndroid Build Coastguard Worker
2032*27162e4eSAndroid Build Coastguard Worker case dstage_getSFrameSize:
2033*27162e4eSAndroid Build Coastguard Worker if ((srcEnd - srcPtr) >= 4) {
2034*27162e4eSAndroid Build Coastguard Worker selectedIn = srcPtr;
2035*27162e4eSAndroid Build Coastguard Worker srcPtr += 4;
2036*27162e4eSAndroid Build Coastguard Worker } else {
2037*27162e4eSAndroid Build Coastguard Worker /* not enough input to read cBlockSize field */
2038*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize = 4;
2039*27162e4eSAndroid Build Coastguard Worker dctx->tmpInTarget = 8;
2040*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_storeSFrameSize;
2041*27162e4eSAndroid Build Coastguard Worker }
2042*27162e4eSAndroid Build Coastguard Worker
2043*27162e4eSAndroid Build Coastguard Worker if (dctx->dStage == dstage_storeSFrameSize)
2044*27162e4eSAndroid Build Coastguard Worker case dstage_storeSFrameSize:
2045*27162e4eSAndroid Build Coastguard Worker { size_t const sizeToCopy = MIN(dctx->tmpInTarget - dctx->tmpInSize,
2046*27162e4eSAndroid Build Coastguard Worker (size_t)(srcEnd - srcPtr) );
2047*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->header + dctx->tmpInSize, srcPtr, sizeToCopy);
2048*27162e4eSAndroid Build Coastguard Worker srcPtr += sizeToCopy;
2049*27162e4eSAndroid Build Coastguard Worker dctx->tmpInSize += sizeToCopy;
2050*27162e4eSAndroid Build Coastguard Worker if (dctx->tmpInSize < dctx->tmpInTarget) {
2051*27162e4eSAndroid Build Coastguard Worker /* not enough input to get full sBlockSize; wait for more */
2052*27162e4eSAndroid Build Coastguard Worker nextSrcSizeHint = dctx->tmpInTarget - dctx->tmpInSize;
2053*27162e4eSAndroid Build Coastguard Worker doAnotherStage = 0;
2054*27162e4eSAndroid Build Coastguard Worker break;
2055*27162e4eSAndroid Build Coastguard Worker }
2056*27162e4eSAndroid Build Coastguard Worker selectedIn = dctx->header + 4;
2057*27162e4eSAndroid Build Coastguard Worker } /* if (dctx->dStage == dstage_storeSFrameSize) */
2058*27162e4eSAndroid Build Coastguard Worker
2059*27162e4eSAndroid Build Coastguard Worker /* case dstage_decodeSFrameSize: */ /* no direct entry */
2060*27162e4eSAndroid Build Coastguard Worker { size_t const SFrameSize = LZ4F_readLE32(selectedIn);
2061*27162e4eSAndroid Build Coastguard Worker dctx->frameInfo.contentSize = SFrameSize;
2062*27162e4eSAndroid Build Coastguard Worker dctx->tmpInTarget = SFrameSize;
2063*27162e4eSAndroid Build Coastguard Worker dctx->dStage = dstage_skipSkippable;
2064*27162e4eSAndroid Build Coastguard Worker break;
2065*27162e4eSAndroid Build Coastguard Worker }
2066*27162e4eSAndroid Build Coastguard Worker
2067*27162e4eSAndroid Build Coastguard Worker case dstage_skipSkippable:
2068*27162e4eSAndroid Build Coastguard Worker { size_t const skipSize = MIN(dctx->tmpInTarget, (size_t)(srcEnd-srcPtr));
2069*27162e4eSAndroid Build Coastguard Worker srcPtr += skipSize;
2070*27162e4eSAndroid Build Coastguard Worker dctx->tmpInTarget -= skipSize;
2071*27162e4eSAndroid Build Coastguard Worker doAnotherStage = 0;
2072*27162e4eSAndroid Build Coastguard Worker nextSrcSizeHint = dctx->tmpInTarget;
2073*27162e4eSAndroid Build Coastguard Worker if (nextSrcSizeHint) break; /* still more to skip */
2074*27162e4eSAndroid Build Coastguard Worker /* frame fully skipped : prepare context for a new frame */
2075*27162e4eSAndroid Build Coastguard Worker LZ4F_resetDecompressionContext(dctx);
2076*27162e4eSAndroid Build Coastguard Worker break;
2077*27162e4eSAndroid Build Coastguard Worker }
2078*27162e4eSAndroid Build Coastguard Worker } /* switch (dctx->dStage) */
2079*27162e4eSAndroid Build Coastguard Worker } /* while (doAnotherStage) */
2080*27162e4eSAndroid Build Coastguard Worker
2081*27162e4eSAndroid Build Coastguard Worker /* preserve history within tmpOut whenever necessary */
2082*27162e4eSAndroid Build Coastguard Worker LZ4F_STATIC_ASSERT((unsigned)dstage_init == 2);
2083*27162e4eSAndroid Build Coastguard Worker if ( (dctx->frameInfo.blockMode==LZ4F_blockLinked) /* next block will use up to 64KB from previous ones */
2084*27162e4eSAndroid Build Coastguard Worker && (dctx->dict != dctx->tmpOutBuffer) /* dictionary is not already within tmp */
2085*27162e4eSAndroid Build Coastguard Worker && (dctx->dict != NULL) /* dictionary exists */
2086*27162e4eSAndroid Build Coastguard Worker && (!decompressOptionsPtr->stableDst) /* cannot rely on dst data to remain there for next call */
2087*27162e4eSAndroid Build Coastguard Worker && ((unsigned)(dctx->dStage)-2 < (unsigned)(dstage_getSuffix)-2) ) /* valid stages : [init ... getSuffix[ */
2088*27162e4eSAndroid Build Coastguard Worker {
2089*27162e4eSAndroid Build Coastguard Worker if (dctx->dStage == dstage_flushOut) {
2090*27162e4eSAndroid Build Coastguard Worker size_t const preserveSize = (size_t)(dctx->tmpOut - dctx->tmpOutBuffer);
2091*27162e4eSAndroid Build Coastguard Worker size_t copySize = 64 KB - dctx->tmpOutSize;
2092*27162e4eSAndroid Build Coastguard Worker const BYTE* oldDictEnd = dctx->dict + dctx->dictSize - dctx->tmpOutStart;
2093*27162e4eSAndroid Build Coastguard Worker if (dctx->tmpOutSize > 64 KB) copySize = 0;
2094*27162e4eSAndroid Build Coastguard Worker if (copySize > preserveSize) copySize = preserveSize;
2095*27162e4eSAndroid Build Coastguard Worker assert(dctx->tmpOutBuffer != NULL);
2096*27162e4eSAndroid Build Coastguard Worker
2097*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->tmpOutBuffer + preserveSize - copySize, oldDictEnd - copySize, copySize);
2098*27162e4eSAndroid Build Coastguard Worker
2099*27162e4eSAndroid Build Coastguard Worker dctx->dict = dctx->tmpOutBuffer;
2100*27162e4eSAndroid Build Coastguard Worker dctx->dictSize = preserveSize + dctx->tmpOutStart;
2101*27162e4eSAndroid Build Coastguard Worker } else {
2102*27162e4eSAndroid Build Coastguard Worker const BYTE* const oldDictEnd = dctx->dict + dctx->dictSize;
2103*27162e4eSAndroid Build Coastguard Worker size_t const newDictSize = MIN(dctx->dictSize, 64 KB);
2104*27162e4eSAndroid Build Coastguard Worker
2105*27162e4eSAndroid Build Coastguard Worker memcpy(dctx->tmpOutBuffer, oldDictEnd - newDictSize, newDictSize);
2106*27162e4eSAndroid Build Coastguard Worker
2107*27162e4eSAndroid Build Coastguard Worker dctx->dict = dctx->tmpOutBuffer;
2108*27162e4eSAndroid Build Coastguard Worker dctx->dictSize = newDictSize;
2109*27162e4eSAndroid Build Coastguard Worker dctx->tmpOut = dctx->tmpOutBuffer + newDictSize;
2110*27162e4eSAndroid Build Coastguard Worker }
2111*27162e4eSAndroid Build Coastguard Worker }
2112*27162e4eSAndroid Build Coastguard Worker
2113*27162e4eSAndroid Build Coastguard Worker *srcSizePtr = (size_t)(srcPtr - srcStart);
2114*27162e4eSAndroid Build Coastguard Worker *dstSizePtr = (size_t)(dstPtr - dstStart);
2115*27162e4eSAndroid Build Coastguard Worker return nextSrcSizeHint;
2116*27162e4eSAndroid Build Coastguard Worker }
2117*27162e4eSAndroid Build Coastguard Worker
2118*27162e4eSAndroid Build Coastguard Worker /*! LZ4F_decompress_usingDict() :
2119*27162e4eSAndroid Build Coastguard Worker * Same as LZ4F_decompress(), using a predefined dictionary.
2120*27162e4eSAndroid Build Coastguard Worker * Dictionary is used "in place", without any preprocessing.
2121*27162e4eSAndroid Build Coastguard Worker * It must remain accessible throughout the entire frame decoding.
2122*27162e4eSAndroid Build Coastguard Worker */
LZ4F_decompress_usingDict(LZ4F_dctx * dctx,void * dstBuffer,size_t * dstSizePtr,const void * srcBuffer,size_t * srcSizePtr,const void * dict,size_t dictSize,const LZ4F_decompressOptions_t * decompressOptionsPtr)2123*27162e4eSAndroid Build Coastguard Worker size_t LZ4F_decompress_usingDict(LZ4F_dctx* dctx,
2124*27162e4eSAndroid Build Coastguard Worker void* dstBuffer, size_t* dstSizePtr,
2125*27162e4eSAndroid Build Coastguard Worker const void* srcBuffer, size_t* srcSizePtr,
2126*27162e4eSAndroid Build Coastguard Worker const void* dict, size_t dictSize,
2127*27162e4eSAndroid Build Coastguard Worker const LZ4F_decompressOptions_t* decompressOptionsPtr)
2128*27162e4eSAndroid Build Coastguard Worker {
2129*27162e4eSAndroid Build Coastguard Worker if (dctx->dStage <= dstage_init) {
2130*27162e4eSAndroid Build Coastguard Worker dctx->dict = (const BYTE*)dict;
2131*27162e4eSAndroid Build Coastguard Worker dctx->dictSize = dictSize;
2132*27162e4eSAndroid Build Coastguard Worker }
2133*27162e4eSAndroid Build Coastguard Worker return LZ4F_decompress(dctx, dstBuffer, dstSizePtr,
2134*27162e4eSAndroid Build Coastguard Worker srcBuffer, srcSizePtr,
2135*27162e4eSAndroid Build Coastguard Worker decompressOptionsPtr);
2136*27162e4eSAndroid Build Coastguard Worker }
2137