xref: /OK3568_Linux_fs/kernel/lib/lz4/lz4_compress.c (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun  * LZ4 - Fast LZ compression algorithm
3*4882a593Smuzhiyun  * Copyright (C) 2011 - 2016, Yann Collet.
4*4882a593Smuzhiyun  * BSD 2 - Clause License (http://www.opensource.org/licenses/bsd - license.php)
5*4882a593Smuzhiyun  * Redistribution and use in source and binary forms, with or without
6*4882a593Smuzhiyun  * modification, are permitted provided that the following conditions are
7*4882a593Smuzhiyun  * met:
8*4882a593Smuzhiyun  *	* Redistributions of source code must retain the above copyright
9*4882a593Smuzhiyun  *	  notice, this list of conditions and the following disclaimer.
10*4882a593Smuzhiyun  *	* Redistributions in binary form must reproduce the above
11*4882a593Smuzhiyun  * copyright notice, this list of conditions and the following disclaimer
12*4882a593Smuzhiyun  * in the documentation and/or other materials provided with the
13*4882a593Smuzhiyun  * distribution.
14*4882a593Smuzhiyun  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
15*4882a593Smuzhiyun  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
16*4882a593Smuzhiyun  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
17*4882a593Smuzhiyun  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
18*4882a593Smuzhiyun  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
19*4882a593Smuzhiyun  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
20*4882a593Smuzhiyun  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21*4882a593Smuzhiyun  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22*4882a593Smuzhiyun  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23*4882a593Smuzhiyun  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
24*4882a593Smuzhiyun  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25*4882a593Smuzhiyun  * You can contact the author at :
26*4882a593Smuzhiyun  *	- LZ4 homepage : http://www.lz4.org
27*4882a593Smuzhiyun  *	- LZ4 source repository : https://github.com/lz4/lz4
28*4882a593Smuzhiyun  *
29*4882a593Smuzhiyun  *	Changed for kernel usage by:
30*4882a593Smuzhiyun  *	Sven Schmidt <4sschmid@informatik.uni-hamburg.de>
31*4882a593Smuzhiyun  */
32*4882a593Smuzhiyun 
33*4882a593Smuzhiyun /*-************************************
34*4882a593Smuzhiyun  *	Dependencies
35*4882a593Smuzhiyun  **************************************/
36*4882a593Smuzhiyun #include <linux/lz4.h>
37*4882a593Smuzhiyun #include "lz4defs.h"
38*4882a593Smuzhiyun #include <linux/module.h>
39*4882a593Smuzhiyun #include <linux/kernel.h>
40*4882a593Smuzhiyun #include <asm/unaligned.h>
41*4882a593Smuzhiyun 
42*4882a593Smuzhiyun static const int LZ4_minLength = (MFLIMIT + 1);
43*4882a593Smuzhiyun static const int LZ4_64Klimit = ((64 * KB) + (MFLIMIT - 1));
44*4882a593Smuzhiyun 
45*4882a593Smuzhiyun /*-******************************
46*4882a593Smuzhiyun  *	Compression functions
47*4882a593Smuzhiyun  ********************************/
LZ4_hash4(U32 sequence,tableType_t const tableType)48*4882a593Smuzhiyun static FORCE_INLINE U32 LZ4_hash4(
49*4882a593Smuzhiyun 	U32 sequence,
50*4882a593Smuzhiyun 	tableType_t const tableType)
51*4882a593Smuzhiyun {
52*4882a593Smuzhiyun 	if (tableType == byU16)
53*4882a593Smuzhiyun 		return ((sequence * 2654435761U)
54*4882a593Smuzhiyun 			>> ((MINMATCH * 8) - (LZ4_HASHLOG + 1)));
55*4882a593Smuzhiyun 	else
56*4882a593Smuzhiyun 		return ((sequence * 2654435761U)
57*4882a593Smuzhiyun 			>> ((MINMATCH * 8) - LZ4_HASHLOG));
58*4882a593Smuzhiyun }
59*4882a593Smuzhiyun 
LZ4_hash5(U64 sequence,tableType_t const tableType)60*4882a593Smuzhiyun static FORCE_INLINE U32 LZ4_hash5(
61*4882a593Smuzhiyun 	U64 sequence,
62*4882a593Smuzhiyun 	tableType_t const tableType)
63*4882a593Smuzhiyun {
64*4882a593Smuzhiyun 	const U32 hashLog = (tableType == byU16)
65*4882a593Smuzhiyun 		? LZ4_HASHLOG + 1
66*4882a593Smuzhiyun 		: LZ4_HASHLOG;
67*4882a593Smuzhiyun 
68*4882a593Smuzhiyun #if LZ4_LITTLE_ENDIAN
69*4882a593Smuzhiyun 	static const U64 prime5bytes = 889523592379ULL;
70*4882a593Smuzhiyun 
71*4882a593Smuzhiyun 	return (U32)(((sequence << 24) * prime5bytes) >> (64 - hashLog));
72*4882a593Smuzhiyun #else
73*4882a593Smuzhiyun 	static const U64 prime8bytes = 11400714785074694791ULL;
74*4882a593Smuzhiyun 
75*4882a593Smuzhiyun 	return (U32)(((sequence >> 24) * prime8bytes) >> (64 - hashLog));
76*4882a593Smuzhiyun #endif
77*4882a593Smuzhiyun }
78*4882a593Smuzhiyun 
LZ4_hashPosition(const void * p,tableType_t const tableType)79*4882a593Smuzhiyun static FORCE_INLINE U32 LZ4_hashPosition(
80*4882a593Smuzhiyun 	const void *p,
81*4882a593Smuzhiyun 	tableType_t const tableType)
82*4882a593Smuzhiyun {
83*4882a593Smuzhiyun #if LZ4_ARCH64
84*4882a593Smuzhiyun 	if (tableType == byU32)
85*4882a593Smuzhiyun 		return LZ4_hash5(LZ4_read_ARCH(p), tableType);
86*4882a593Smuzhiyun #endif
87*4882a593Smuzhiyun 
88*4882a593Smuzhiyun 	return LZ4_hash4(LZ4_read32(p), tableType);
89*4882a593Smuzhiyun }
90*4882a593Smuzhiyun 
LZ4_putPositionOnHash(const BYTE * p,U32 h,void * tableBase,tableType_t const tableType,const BYTE * srcBase)91*4882a593Smuzhiyun static void LZ4_putPositionOnHash(
92*4882a593Smuzhiyun 	const BYTE *p,
93*4882a593Smuzhiyun 	U32 h,
94*4882a593Smuzhiyun 	void *tableBase,
95*4882a593Smuzhiyun 	tableType_t const tableType,
96*4882a593Smuzhiyun 	const BYTE *srcBase)
97*4882a593Smuzhiyun {
98*4882a593Smuzhiyun 	switch (tableType) {
99*4882a593Smuzhiyun 	case byPtr:
100*4882a593Smuzhiyun 	{
101*4882a593Smuzhiyun 		const BYTE **hashTable = (const BYTE **)tableBase;
102*4882a593Smuzhiyun 
103*4882a593Smuzhiyun 		hashTable[h] = p;
104*4882a593Smuzhiyun 		return;
105*4882a593Smuzhiyun 	}
106*4882a593Smuzhiyun 	case byU32:
107*4882a593Smuzhiyun 	{
108*4882a593Smuzhiyun 		U32 *hashTable = (U32 *) tableBase;
109*4882a593Smuzhiyun 
110*4882a593Smuzhiyun 		hashTable[h] = (U32)(p - srcBase);
111*4882a593Smuzhiyun 		return;
112*4882a593Smuzhiyun 	}
113*4882a593Smuzhiyun 	case byU16:
114*4882a593Smuzhiyun 	{
115*4882a593Smuzhiyun 		U16 *hashTable = (U16 *) tableBase;
116*4882a593Smuzhiyun 
117*4882a593Smuzhiyun 		hashTable[h] = (U16)(p - srcBase);
118*4882a593Smuzhiyun 		return;
119*4882a593Smuzhiyun 	}
120*4882a593Smuzhiyun 	}
121*4882a593Smuzhiyun }
122*4882a593Smuzhiyun 
LZ4_putPosition(const BYTE * p,void * tableBase,tableType_t tableType,const BYTE * srcBase)123*4882a593Smuzhiyun static FORCE_INLINE void LZ4_putPosition(
124*4882a593Smuzhiyun 	const BYTE *p,
125*4882a593Smuzhiyun 	void *tableBase,
126*4882a593Smuzhiyun 	tableType_t tableType,
127*4882a593Smuzhiyun 	const BYTE *srcBase)
128*4882a593Smuzhiyun {
129*4882a593Smuzhiyun 	U32 const h = LZ4_hashPosition(p, tableType);
130*4882a593Smuzhiyun 
131*4882a593Smuzhiyun 	LZ4_putPositionOnHash(p, h, tableBase, tableType, srcBase);
132*4882a593Smuzhiyun }
133*4882a593Smuzhiyun 
LZ4_getPositionOnHash(U32 h,void * tableBase,tableType_t tableType,const BYTE * srcBase)134*4882a593Smuzhiyun static const BYTE *LZ4_getPositionOnHash(
135*4882a593Smuzhiyun 	U32 h,
136*4882a593Smuzhiyun 	void *tableBase,
137*4882a593Smuzhiyun 	tableType_t tableType,
138*4882a593Smuzhiyun 	const BYTE *srcBase)
139*4882a593Smuzhiyun {
140*4882a593Smuzhiyun 	if (tableType == byPtr) {
141*4882a593Smuzhiyun 		const BYTE **hashTable = (const BYTE **) tableBase;
142*4882a593Smuzhiyun 
143*4882a593Smuzhiyun 		return hashTable[h];
144*4882a593Smuzhiyun 	}
145*4882a593Smuzhiyun 
146*4882a593Smuzhiyun 	if (tableType == byU32) {
147*4882a593Smuzhiyun 		const U32 * const hashTable = (U32 *) tableBase;
148*4882a593Smuzhiyun 
149*4882a593Smuzhiyun 		return hashTable[h] + srcBase;
150*4882a593Smuzhiyun 	}
151*4882a593Smuzhiyun 
152*4882a593Smuzhiyun 	{
153*4882a593Smuzhiyun 		/* default, to ensure a return */
154*4882a593Smuzhiyun 		const U16 * const hashTable = (U16 *) tableBase;
155*4882a593Smuzhiyun 
156*4882a593Smuzhiyun 		return hashTable[h] + srcBase;
157*4882a593Smuzhiyun 	}
158*4882a593Smuzhiyun }
159*4882a593Smuzhiyun 
LZ4_getPosition(const BYTE * p,void * tableBase,tableType_t tableType,const BYTE * srcBase)160*4882a593Smuzhiyun static FORCE_INLINE const BYTE *LZ4_getPosition(
161*4882a593Smuzhiyun 	const BYTE *p,
162*4882a593Smuzhiyun 	void *tableBase,
163*4882a593Smuzhiyun 	tableType_t tableType,
164*4882a593Smuzhiyun 	const BYTE *srcBase)
165*4882a593Smuzhiyun {
166*4882a593Smuzhiyun 	U32 const h = LZ4_hashPosition(p, tableType);
167*4882a593Smuzhiyun 
168*4882a593Smuzhiyun 	return LZ4_getPositionOnHash(h, tableBase, tableType, srcBase);
169*4882a593Smuzhiyun }
170*4882a593Smuzhiyun 
171*4882a593Smuzhiyun 
172*4882a593Smuzhiyun /*
173*4882a593Smuzhiyun  * LZ4_compress_generic() :
174*4882a593Smuzhiyun  * inlined, to ensure branches are decided at compilation time
175*4882a593Smuzhiyun  */
LZ4_compress_generic(LZ4_stream_t_internal * const dictPtr,const char * const source,char * const dest,const int inputSize,const int maxOutputSize,const limitedOutput_directive outputLimited,const tableType_t tableType,const dict_directive dict,const dictIssue_directive dictIssue,const U32 acceleration)176*4882a593Smuzhiyun static FORCE_INLINE int LZ4_compress_generic(
177*4882a593Smuzhiyun 	LZ4_stream_t_internal * const dictPtr,
178*4882a593Smuzhiyun 	const char * const source,
179*4882a593Smuzhiyun 	char * const dest,
180*4882a593Smuzhiyun 	const int inputSize,
181*4882a593Smuzhiyun 	const int maxOutputSize,
182*4882a593Smuzhiyun 	const limitedOutput_directive outputLimited,
183*4882a593Smuzhiyun 	const tableType_t tableType,
184*4882a593Smuzhiyun 	const dict_directive dict,
185*4882a593Smuzhiyun 	const dictIssue_directive dictIssue,
186*4882a593Smuzhiyun 	const U32 acceleration)
187*4882a593Smuzhiyun {
188*4882a593Smuzhiyun 	const BYTE *ip = (const BYTE *) source;
189*4882a593Smuzhiyun 	const BYTE *base;
190*4882a593Smuzhiyun 	const BYTE *lowLimit;
191*4882a593Smuzhiyun 	const BYTE * const lowRefLimit = ip - dictPtr->dictSize;
192*4882a593Smuzhiyun 	const BYTE * const dictionary = dictPtr->dictionary;
193*4882a593Smuzhiyun 	const BYTE * const dictEnd = dictionary + dictPtr->dictSize;
194*4882a593Smuzhiyun 	const size_t dictDelta = dictEnd - (const BYTE *)source;
195*4882a593Smuzhiyun 	const BYTE *anchor = (const BYTE *) source;
196*4882a593Smuzhiyun 	const BYTE * const iend = ip + inputSize;
197*4882a593Smuzhiyun 	const BYTE * const mflimit = iend - MFLIMIT;
198*4882a593Smuzhiyun 	const BYTE * const matchlimit = iend - LASTLITERALS;
199*4882a593Smuzhiyun 
200*4882a593Smuzhiyun 	BYTE *op = (BYTE *) dest;
201*4882a593Smuzhiyun 	BYTE * const olimit = op + maxOutputSize;
202*4882a593Smuzhiyun 
203*4882a593Smuzhiyun 	U32 forwardH;
204*4882a593Smuzhiyun 	size_t refDelta = 0;
205*4882a593Smuzhiyun 
206*4882a593Smuzhiyun 	/* Init conditions */
207*4882a593Smuzhiyun 	if ((U32)inputSize > (U32)LZ4_MAX_INPUT_SIZE) {
208*4882a593Smuzhiyun 		/* Unsupported inputSize, too large (or negative) */
209*4882a593Smuzhiyun 		return 0;
210*4882a593Smuzhiyun 	}
211*4882a593Smuzhiyun 
212*4882a593Smuzhiyun 	switch (dict) {
213*4882a593Smuzhiyun 	case noDict:
214*4882a593Smuzhiyun 	default:
215*4882a593Smuzhiyun 		base = (const BYTE *)source;
216*4882a593Smuzhiyun 		lowLimit = (const BYTE *)source;
217*4882a593Smuzhiyun 		break;
218*4882a593Smuzhiyun 	case withPrefix64k:
219*4882a593Smuzhiyun 		base = (const BYTE *)source - dictPtr->currentOffset;
220*4882a593Smuzhiyun 		lowLimit = (const BYTE *)source - dictPtr->dictSize;
221*4882a593Smuzhiyun 		break;
222*4882a593Smuzhiyun 	case usingExtDict:
223*4882a593Smuzhiyun 		base = (const BYTE *)source - dictPtr->currentOffset;
224*4882a593Smuzhiyun 		lowLimit = (const BYTE *)source;
225*4882a593Smuzhiyun 		break;
226*4882a593Smuzhiyun 	}
227*4882a593Smuzhiyun 
228*4882a593Smuzhiyun 	if ((tableType == byU16)
229*4882a593Smuzhiyun 		&& (inputSize >= LZ4_64Klimit)) {
230*4882a593Smuzhiyun 		/* Size too large (not within 64K limit) */
231*4882a593Smuzhiyun 		return 0;
232*4882a593Smuzhiyun 	}
233*4882a593Smuzhiyun 
234*4882a593Smuzhiyun 	if (inputSize < LZ4_minLength) {
235*4882a593Smuzhiyun 		/* Input too small, no compression (all literals) */
236*4882a593Smuzhiyun 		goto _last_literals;
237*4882a593Smuzhiyun 	}
238*4882a593Smuzhiyun 
239*4882a593Smuzhiyun 	/* First Byte */
240*4882a593Smuzhiyun 	LZ4_putPosition(ip, dictPtr->hashTable, tableType, base);
241*4882a593Smuzhiyun 	ip++;
242*4882a593Smuzhiyun 	forwardH = LZ4_hashPosition(ip, tableType);
243*4882a593Smuzhiyun 
244*4882a593Smuzhiyun 	/* Main Loop */
245*4882a593Smuzhiyun 	for ( ; ; ) {
246*4882a593Smuzhiyun 		const BYTE *match;
247*4882a593Smuzhiyun 		BYTE *token;
248*4882a593Smuzhiyun 
249*4882a593Smuzhiyun 		/* Find a match */
250*4882a593Smuzhiyun 		{
251*4882a593Smuzhiyun 			const BYTE *forwardIp = ip;
252*4882a593Smuzhiyun 			unsigned int step = 1;
253*4882a593Smuzhiyun 			unsigned int searchMatchNb = acceleration << LZ4_SKIPTRIGGER;
254*4882a593Smuzhiyun 
255*4882a593Smuzhiyun 			do {
256*4882a593Smuzhiyun 				U32 const h = forwardH;
257*4882a593Smuzhiyun 
258*4882a593Smuzhiyun 				ip = forwardIp;
259*4882a593Smuzhiyun 				forwardIp += step;
260*4882a593Smuzhiyun 				step = (searchMatchNb++ >> LZ4_SKIPTRIGGER);
261*4882a593Smuzhiyun 
262*4882a593Smuzhiyun 				if (unlikely(forwardIp > mflimit))
263*4882a593Smuzhiyun 					goto _last_literals;
264*4882a593Smuzhiyun 
265*4882a593Smuzhiyun 				match = LZ4_getPositionOnHash(h,
266*4882a593Smuzhiyun 					dictPtr->hashTable,
267*4882a593Smuzhiyun 					tableType, base);
268*4882a593Smuzhiyun 
269*4882a593Smuzhiyun 				if (dict == usingExtDict) {
270*4882a593Smuzhiyun 					if (match < (const BYTE *)source) {
271*4882a593Smuzhiyun 						refDelta = dictDelta;
272*4882a593Smuzhiyun 						lowLimit = dictionary;
273*4882a593Smuzhiyun 					} else {
274*4882a593Smuzhiyun 						refDelta = 0;
275*4882a593Smuzhiyun 						lowLimit = (const BYTE *)source;
276*4882a593Smuzhiyun 				}	 }
277*4882a593Smuzhiyun 
278*4882a593Smuzhiyun 				forwardH = LZ4_hashPosition(forwardIp,
279*4882a593Smuzhiyun 					tableType);
280*4882a593Smuzhiyun 
281*4882a593Smuzhiyun 				LZ4_putPositionOnHash(ip, h, dictPtr->hashTable,
282*4882a593Smuzhiyun 					tableType, base);
283*4882a593Smuzhiyun 			} while (((dictIssue == dictSmall)
284*4882a593Smuzhiyun 					? (match < lowRefLimit)
285*4882a593Smuzhiyun 					: 0)
286*4882a593Smuzhiyun 				|| ((tableType == byU16)
287*4882a593Smuzhiyun 					? 0
288*4882a593Smuzhiyun 					: (match + MAX_DISTANCE < ip))
289*4882a593Smuzhiyun 				|| (LZ4_read32(match + refDelta)
290*4882a593Smuzhiyun 					!= LZ4_read32(ip)));
291*4882a593Smuzhiyun 		}
292*4882a593Smuzhiyun 
293*4882a593Smuzhiyun 		/* Catch up */
294*4882a593Smuzhiyun 		while (((ip > anchor) & (match + refDelta > lowLimit))
295*4882a593Smuzhiyun 				&& (unlikely(ip[-1] == match[refDelta - 1]))) {
296*4882a593Smuzhiyun 			ip--;
297*4882a593Smuzhiyun 			match--;
298*4882a593Smuzhiyun 		}
299*4882a593Smuzhiyun 
300*4882a593Smuzhiyun 		/* Encode Literals */
301*4882a593Smuzhiyun 		{
302*4882a593Smuzhiyun 			unsigned const int litLength = (unsigned int)(ip - anchor);
303*4882a593Smuzhiyun 
304*4882a593Smuzhiyun 			token = op++;
305*4882a593Smuzhiyun 
306*4882a593Smuzhiyun 			if ((outputLimited) &&
307*4882a593Smuzhiyun 				/* Check output buffer overflow */
308*4882a593Smuzhiyun 				(unlikely(op + litLength +
309*4882a593Smuzhiyun 					(2 + 1 + LASTLITERALS) +
310*4882a593Smuzhiyun 					(litLength / 255) > olimit)))
311*4882a593Smuzhiyun 				return 0;
312*4882a593Smuzhiyun 
313*4882a593Smuzhiyun 			if (litLength >= RUN_MASK) {
314*4882a593Smuzhiyun 				int len = (int)litLength - RUN_MASK;
315*4882a593Smuzhiyun 
316*4882a593Smuzhiyun 				*token = (RUN_MASK << ML_BITS);
317*4882a593Smuzhiyun 
318*4882a593Smuzhiyun 				for (; len >= 255; len -= 255)
319*4882a593Smuzhiyun 					*op++ = 255;
320*4882a593Smuzhiyun 				*op++ = (BYTE)len;
321*4882a593Smuzhiyun 			} else
322*4882a593Smuzhiyun 				*token = (BYTE)(litLength << ML_BITS);
323*4882a593Smuzhiyun 
324*4882a593Smuzhiyun 			/* Copy Literals */
325*4882a593Smuzhiyun 			LZ4_wildCopy(op, anchor, op + litLength);
326*4882a593Smuzhiyun 			op += litLength;
327*4882a593Smuzhiyun 		}
328*4882a593Smuzhiyun 
329*4882a593Smuzhiyun _next_match:
330*4882a593Smuzhiyun 		/* Encode Offset */
331*4882a593Smuzhiyun 		LZ4_writeLE16(op, (U16)(ip - match));
332*4882a593Smuzhiyun 		op += 2;
333*4882a593Smuzhiyun 
334*4882a593Smuzhiyun 		/* Encode MatchLength */
335*4882a593Smuzhiyun 		{
336*4882a593Smuzhiyun 			unsigned int matchCode;
337*4882a593Smuzhiyun 
338*4882a593Smuzhiyun 			if ((dict == usingExtDict)
339*4882a593Smuzhiyun 				&& (lowLimit == dictionary)) {
340*4882a593Smuzhiyun 				const BYTE *limit;
341*4882a593Smuzhiyun 
342*4882a593Smuzhiyun 				match += refDelta;
343*4882a593Smuzhiyun 				limit = ip + (dictEnd - match);
344*4882a593Smuzhiyun 
345*4882a593Smuzhiyun 				if (limit > matchlimit)
346*4882a593Smuzhiyun 					limit = matchlimit;
347*4882a593Smuzhiyun 
348*4882a593Smuzhiyun 				matchCode = LZ4_count(ip + MINMATCH,
349*4882a593Smuzhiyun 					match + MINMATCH, limit);
350*4882a593Smuzhiyun 
351*4882a593Smuzhiyun 				ip += MINMATCH + matchCode;
352*4882a593Smuzhiyun 
353*4882a593Smuzhiyun 				if (ip == limit) {
354*4882a593Smuzhiyun 					unsigned const int more = LZ4_count(ip,
355*4882a593Smuzhiyun 						(const BYTE *)source,
356*4882a593Smuzhiyun 						matchlimit);
357*4882a593Smuzhiyun 
358*4882a593Smuzhiyun 					matchCode += more;
359*4882a593Smuzhiyun 					ip += more;
360*4882a593Smuzhiyun 				}
361*4882a593Smuzhiyun 			} else {
362*4882a593Smuzhiyun 				matchCode = LZ4_count(ip + MINMATCH,
363*4882a593Smuzhiyun 					match + MINMATCH, matchlimit);
364*4882a593Smuzhiyun 				ip += MINMATCH + matchCode;
365*4882a593Smuzhiyun 			}
366*4882a593Smuzhiyun 
367*4882a593Smuzhiyun 			if (outputLimited &&
368*4882a593Smuzhiyun 				/* Check output buffer overflow */
369*4882a593Smuzhiyun 				(unlikely(op +
370*4882a593Smuzhiyun 					(1 + LASTLITERALS) +
371*4882a593Smuzhiyun 					(matchCode >> 8) > olimit)))
372*4882a593Smuzhiyun 				return 0;
373*4882a593Smuzhiyun 
374*4882a593Smuzhiyun 			if (matchCode >= ML_MASK) {
375*4882a593Smuzhiyun 				*token += ML_MASK;
376*4882a593Smuzhiyun 				matchCode -= ML_MASK;
377*4882a593Smuzhiyun 				LZ4_write32(op, 0xFFFFFFFF);
378*4882a593Smuzhiyun 
379*4882a593Smuzhiyun 				while (matchCode >= 4 * 255) {
380*4882a593Smuzhiyun 					op += 4;
381*4882a593Smuzhiyun 					LZ4_write32(op, 0xFFFFFFFF);
382*4882a593Smuzhiyun 					matchCode -= 4 * 255;
383*4882a593Smuzhiyun 				}
384*4882a593Smuzhiyun 
385*4882a593Smuzhiyun 				op += matchCode / 255;
386*4882a593Smuzhiyun 				*op++ = (BYTE)(matchCode % 255);
387*4882a593Smuzhiyun 			} else
388*4882a593Smuzhiyun 				*token += (BYTE)(matchCode);
389*4882a593Smuzhiyun 		}
390*4882a593Smuzhiyun 
391*4882a593Smuzhiyun 		anchor = ip;
392*4882a593Smuzhiyun 
393*4882a593Smuzhiyun 		/* Test end of chunk */
394*4882a593Smuzhiyun 		if (ip > mflimit)
395*4882a593Smuzhiyun 			break;
396*4882a593Smuzhiyun 
397*4882a593Smuzhiyun 		/* Fill table */
398*4882a593Smuzhiyun 		LZ4_putPosition(ip - 2, dictPtr->hashTable, tableType, base);
399*4882a593Smuzhiyun 
400*4882a593Smuzhiyun 		/* Test next position */
401*4882a593Smuzhiyun 		match = LZ4_getPosition(ip, dictPtr->hashTable,
402*4882a593Smuzhiyun 			tableType, base);
403*4882a593Smuzhiyun 
404*4882a593Smuzhiyun 		if (dict == usingExtDict) {
405*4882a593Smuzhiyun 			if (match < (const BYTE *)source) {
406*4882a593Smuzhiyun 				refDelta = dictDelta;
407*4882a593Smuzhiyun 				lowLimit = dictionary;
408*4882a593Smuzhiyun 			} else {
409*4882a593Smuzhiyun 				refDelta = 0;
410*4882a593Smuzhiyun 				lowLimit = (const BYTE *)source;
411*4882a593Smuzhiyun 			}
412*4882a593Smuzhiyun 		}
413*4882a593Smuzhiyun 
414*4882a593Smuzhiyun 		LZ4_putPosition(ip, dictPtr->hashTable, tableType, base);
415*4882a593Smuzhiyun 
416*4882a593Smuzhiyun 		if (((dictIssue == dictSmall) ? (match >= lowRefLimit) : 1)
417*4882a593Smuzhiyun 			&& (match + MAX_DISTANCE >= ip)
418*4882a593Smuzhiyun 			&& (LZ4_read32(match + refDelta) == LZ4_read32(ip))) {
419*4882a593Smuzhiyun 			token = op++;
420*4882a593Smuzhiyun 			*token = 0;
421*4882a593Smuzhiyun 			goto _next_match;
422*4882a593Smuzhiyun 		}
423*4882a593Smuzhiyun 
424*4882a593Smuzhiyun 		/* Prepare next loop */
425*4882a593Smuzhiyun 		forwardH = LZ4_hashPosition(++ip, tableType);
426*4882a593Smuzhiyun 	}
427*4882a593Smuzhiyun 
428*4882a593Smuzhiyun _last_literals:
429*4882a593Smuzhiyun 	/* Encode Last Literals */
430*4882a593Smuzhiyun 	{
431*4882a593Smuzhiyun 		size_t const lastRun = (size_t)(iend - anchor);
432*4882a593Smuzhiyun 
433*4882a593Smuzhiyun 		if ((outputLimited) &&
434*4882a593Smuzhiyun 			/* Check output buffer overflow */
435*4882a593Smuzhiyun 			((op - (BYTE *)dest) + lastRun + 1 +
436*4882a593Smuzhiyun 			((lastRun + 255 - RUN_MASK) / 255) > (U32)maxOutputSize))
437*4882a593Smuzhiyun 			return 0;
438*4882a593Smuzhiyun 
439*4882a593Smuzhiyun 		if (lastRun >= RUN_MASK) {
440*4882a593Smuzhiyun 			size_t accumulator = lastRun - RUN_MASK;
441*4882a593Smuzhiyun 			*op++ = RUN_MASK << ML_BITS;
442*4882a593Smuzhiyun 			for (; accumulator >= 255; accumulator -= 255)
443*4882a593Smuzhiyun 				*op++ = 255;
444*4882a593Smuzhiyun 			*op++ = (BYTE) accumulator;
445*4882a593Smuzhiyun 		} else {
446*4882a593Smuzhiyun 			*op++ = (BYTE)(lastRun << ML_BITS);
447*4882a593Smuzhiyun 		}
448*4882a593Smuzhiyun 
449*4882a593Smuzhiyun 		LZ4_memcpy(op, anchor, lastRun);
450*4882a593Smuzhiyun 
451*4882a593Smuzhiyun 		op += lastRun;
452*4882a593Smuzhiyun 	}
453*4882a593Smuzhiyun 
454*4882a593Smuzhiyun 	/* End */
455*4882a593Smuzhiyun 	return (int) (((char *)op) - dest);
456*4882a593Smuzhiyun }
457*4882a593Smuzhiyun 
LZ4_compress_fast_extState(void * state,const char * source,char * dest,int inputSize,int maxOutputSize,int acceleration)458*4882a593Smuzhiyun static int LZ4_compress_fast_extState(
459*4882a593Smuzhiyun 	void *state,
460*4882a593Smuzhiyun 	const char *source,
461*4882a593Smuzhiyun 	char *dest,
462*4882a593Smuzhiyun 	int inputSize,
463*4882a593Smuzhiyun 	int maxOutputSize,
464*4882a593Smuzhiyun 	int acceleration)
465*4882a593Smuzhiyun {
466*4882a593Smuzhiyun 	LZ4_stream_t_internal *ctx = &((LZ4_stream_t *)state)->internal_donotuse;
467*4882a593Smuzhiyun #if LZ4_ARCH64
468*4882a593Smuzhiyun 	const tableType_t tableType = byU32;
469*4882a593Smuzhiyun #else
470*4882a593Smuzhiyun 	const tableType_t tableType = byPtr;
471*4882a593Smuzhiyun #endif
472*4882a593Smuzhiyun 
473*4882a593Smuzhiyun 	LZ4_resetStream((LZ4_stream_t *)state);
474*4882a593Smuzhiyun 
475*4882a593Smuzhiyun 	if (acceleration < 1)
476*4882a593Smuzhiyun 		acceleration = LZ4_ACCELERATION_DEFAULT;
477*4882a593Smuzhiyun 
478*4882a593Smuzhiyun 	if (maxOutputSize >= LZ4_COMPRESSBOUND(inputSize)) {
479*4882a593Smuzhiyun 		if (inputSize < LZ4_64Klimit)
480*4882a593Smuzhiyun 			return LZ4_compress_generic(ctx, source,
481*4882a593Smuzhiyun 				dest, inputSize, 0,
482*4882a593Smuzhiyun 				noLimit, byU16, noDict,
483*4882a593Smuzhiyun 				noDictIssue, acceleration);
484*4882a593Smuzhiyun 		else
485*4882a593Smuzhiyun 			return LZ4_compress_generic(ctx, source,
486*4882a593Smuzhiyun 				dest, inputSize, 0,
487*4882a593Smuzhiyun 				noLimit, tableType, noDict,
488*4882a593Smuzhiyun 				noDictIssue, acceleration);
489*4882a593Smuzhiyun 	} else {
490*4882a593Smuzhiyun 		if (inputSize < LZ4_64Klimit)
491*4882a593Smuzhiyun 			return LZ4_compress_generic(ctx, source,
492*4882a593Smuzhiyun 				dest, inputSize,
493*4882a593Smuzhiyun 				maxOutputSize, limitedOutput, byU16, noDict,
494*4882a593Smuzhiyun 				noDictIssue, acceleration);
495*4882a593Smuzhiyun 		else
496*4882a593Smuzhiyun 			return LZ4_compress_generic(ctx, source,
497*4882a593Smuzhiyun 				dest, inputSize,
498*4882a593Smuzhiyun 				maxOutputSize, limitedOutput, tableType, noDict,
499*4882a593Smuzhiyun 				noDictIssue, acceleration);
500*4882a593Smuzhiyun 	}
501*4882a593Smuzhiyun }
502*4882a593Smuzhiyun 
LZ4_compress_fast(const char * source,char * dest,int inputSize,int maxOutputSize,int acceleration,void * wrkmem)503*4882a593Smuzhiyun int LZ4_compress_fast(const char *source, char *dest, int inputSize,
504*4882a593Smuzhiyun 	int maxOutputSize, int acceleration, void *wrkmem)
505*4882a593Smuzhiyun {
506*4882a593Smuzhiyun 	return LZ4_compress_fast_extState(wrkmem, source, dest, inputSize,
507*4882a593Smuzhiyun 		maxOutputSize, acceleration);
508*4882a593Smuzhiyun }
509*4882a593Smuzhiyun EXPORT_SYMBOL(LZ4_compress_fast);
510*4882a593Smuzhiyun 
LZ4_compress_default(const char * source,char * dest,int inputSize,int maxOutputSize,void * wrkmem)511*4882a593Smuzhiyun int LZ4_compress_default(const char *source, char *dest, int inputSize,
512*4882a593Smuzhiyun 	int maxOutputSize, void *wrkmem)
513*4882a593Smuzhiyun {
514*4882a593Smuzhiyun 	return LZ4_compress_fast(source, dest, inputSize,
515*4882a593Smuzhiyun 		maxOutputSize, LZ4_ACCELERATION_DEFAULT, wrkmem);
516*4882a593Smuzhiyun }
517*4882a593Smuzhiyun EXPORT_SYMBOL(LZ4_compress_default);
518*4882a593Smuzhiyun 
519*4882a593Smuzhiyun /*-******************************
520*4882a593Smuzhiyun  *	*_destSize() variant
521*4882a593Smuzhiyun  ********************************/
LZ4_compress_destSize_generic(LZ4_stream_t_internal * const ctx,const char * const src,char * const dst,int * const srcSizePtr,const int targetDstSize,const tableType_t tableType)522*4882a593Smuzhiyun static int LZ4_compress_destSize_generic(
523*4882a593Smuzhiyun 	LZ4_stream_t_internal * const ctx,
524*4882a593Smuzhiyun 	const char * const src,
525*4882a593Smuzhiyun 	char * const dst,
526*4882a593Smuzhiyun 	int * const srcSizePtr,
527*4882a593Smuzhiyun 	const int targetDstSize,
528*4882a593Smuzhiyun 	const tableType_t tableType)
529*4882a593Smuzhiyun {
530*4882a593Smuzhiyun 	const BYTE *ip = (const BYTE *) src;
531*4882a593Smuzhiyun 	const BYTE *base = (const BYTE *) src;
532*4882a593Smuzhiyun 	const BYTE *lowLimit = (const BYTE *) src;
533*4882a593Smuzhiyun 	const BYTE *anchor = ip;
534*4882a593Smuzhiyun 	const BYTE * const iend = ip + *srcSizePtr;
535*4882a593Smuzhiyun 	const BYTE * const mflimit = iend - MFLIMIT;
536*4882a593Smuzhiyun 	const BYTE * const matchlimit = iend - LASTLITERALS;
537*4882a593Smuzhiyun 
538*4882a593Smuzhiyun 	BYTE *op = (BYTE *) dst;
539*4882a593Smuzhiyun 	BYTE * const oend = op + targetDstSize;
540*4882a593Smuzhiyun 	BYTE * const oMaxLit = op + targetDstSize - 2 /* offset */
541*4882a593Smuzhiyun 		- 8 /* because 8 + MINMATCH == MFLIMIT */ - 1 /* token */;
542*4882a593Smuzhiyun 	BYTE * const oMaxMatch = op + targetDstSize
543*4882a593Smuzhiyun 		- (LASTLITERALS + 1 /* token */);
544*4882a593Smuzhiyun 	BYTE * const oMaxSeq = oMaxLit - 1 /* token */;
545*4882a593Smuzhiyun 
546*4882a593Smuzhiyun 	U32 forwardH;
547*4882a593Smuzhiyun 
548*4882a593Smuzhiyun 	/* Init conditions */
549*4882a593Smuzhiyun 	/* Impossible to store anything */
550*4882a593Smuzhiyun 	if (targetDstSize < 1)
551*4882a593Smuzhiyun 		return 0;
552*4882a593Smuzhiyun 	/* Unsupported input size, too large (or negative) */
553*4882a593Smuzhiyun 	if ((U32)*srcSizePtr > (U32)LZ4_MAX_INPUT_SIZE)
554*4882a593Smuzhiyun 		return 0;
555*4882a593Smuzhiyun 	/* Size too large (not within 64K limit) */
556*4882a593Smuzhiyun 	if ((tableType == byU16) && (*srcSizePtr >= LZ4_64Klimit))
557*4882a593Smuzhiyun 		return 0;
558*4882a593Smuzhiyun 	/* Input too small, no compression (all literals) */
559*4882a593Smuzhiyun 	if (*srcSizePtr < LZ4_minLength)
560*4882a593Smuzhiyun 		goto _last_literals;
561*4882a593Smuzhiyun 
562*4882a593Smuzhiyun 	/* First Byte */
563*4882a593Smuzhiyun 	*srcSizePtr = 0;
564*4882a593Smuzhiyun 	LZ4_putPosition(ip, ctx->hashTable, tableType, base);
565*4882a593Smuzhiyun 	ip++; forwardH = LZ4_hashPosition(ip, tableType);
566*4882a593Smuzhiyun 
567*4882a593Smuzhiyun 	/* Main Loop */
568*4882a593Smuzhiyun 	for ( ; ; ) {
569*4882a593Smuzhiyun 		const BYTE *match;
570*4882a593Smuzhiyun 		BYTE *token;
571*4882a593Smuzhiyun 
572*4882a593Smuzhiyun 		/* Find a match */
573*4882a593Smuzhiyun 		{
574*4882a593Smuzhiyun 			const BYTE *forwardIp = ip;
575*4882a593Smuzhiyun 			unsigned int step = 1;
576*4882a593Smuzhiyun 			unsigned int searchMatchNb = 1 << LZ4_SKIPTRIGGER;
577*4882a593Smuzhiyun 
578*4882a593Smuzhiyun 			do {
579*4882a593Smuzhiyun 				U32 h = forwardH;
580*4882a593Smuzhiyun 
581*4882a593Smuzhiyun 				ip = forwardIp;
582*4882a593Smuzhiyun 				forwardIp += step;
583*4882a593Smuzhiyun 				step = (searchMatchNb++ >> LZ4_SKIPTRIGGER);
584*4882a593Smuzhiyun 
585*4882a593Smuzhiyun 				if (unlikely(forwardIp > mflimit))
586*4882a593Smuzhiyun 					goto _last_literals;
587*4882a593Smuzhiyun 
588*4882a593Smuzhiyun 				match = LZ4_getPositionOnHash(h, ctx->hashTable,
589*4882a593Smuzhiyun 					tableType, base);
590*4882a593Smuzhiyun 				forwardH = LZ4_hashPosition(forwardIp,
591*4882a593Smuzhiyun 					tableType);
592*4882a593Smuzhiyun 				LZ4_putPositionOnHash(ip, h,
593*4882a593Smuzhiyun 					ctx->hashTable, tableType,
594*4882a593Smuzhiyun 					base);
595*4882a593Smuzhiyun 
596*4882a593Smuzhiyun 			} while (((tableType == byU16)
597*4882a593Smuzhiyun 				? 0
598*4882a593Smuzhiyun 				: (match + MAX_DISTANCE < ip))
599*4882a593Smuzhiyun 				|| (LZ4_read32(match) != LZ4_read32(ip)));
600*4882a593Smuzhiyun 		}
601*4882a593Smuzhiyun 
602*4882a593Smuzhiyun 		/* Catch up */
603*4882a593Smuzhiyun 		while ((ip > anchor)
604*4882a593Smuzhiyun 			&& (match > lowLimit)
605*4882a593Smuzhiyun 			&& (unlikely(ip[-1] == match[-1]))) {
606*4882a593Smuzhiyun 			ip--;
607*4882a593Smuzhiyun 			match--;
608*4882a593Smuzhiyun 		}
609*4882a593Smuzhiyun 
610*4882a593Smuzhiyun 		/* Encode Literal length */
611*4882a593Smuzhiyun 		{
612*4882a593Smuzhiyun 			unsigned int litLength = (unsigned int)(ip - anchor);
613*4882a593Smuzhiyun 
614*4882a593Smuzhiyun 			token = op++;
615*4882a593Smuzhiyun 			if (op + ((litLength + 240) / 255)
616*4882a593Smuzhiyun 				+ litLength > oMaxLit) {
617*4882a593Smuzhiyun 				/* Not enough space for a last match */
618*4882a593Smuzhiyun 				op--;
619*4882a593Smuzhiyun 				goto _last_literals;
620*4882a593Smuzhiyun 			}
621*4882a593Smuzhiyun 			if (litLength >= RUN_MASK) {
622*4882a593Smuzhiyun 				unsigned int len = litLength - RUN_MASK;
623*4882a593Smuzhiyun 				*token = (RUN_MASK<<ML_BITS);
624*4882a593Smuzhiyun 				for (; len >= 255; len -= 255)
625*4882a593Smuzhiyun 					*op++ = 255;
626*4882a593Smuzhiyun 				*op++ = (BYTE)len;
627*4882a593Smuzhiyun 			} else
628*4882a593Smuzhiyun 				*token = (BYTE)(litLength << ML_BITS);
629*4882a593Smuzhiyun 
630*4882a593Smuzhiyun 			/* Copy Literals */
631*4882a593Smuzhiyun 			LZ4_wildCopy(op, anchor, op + litLength);
632*4882a593Smuzhiyun 			op += litLength;
633*4882a593Smuzhiyun 		}
634*4882a593Smuzhiyun 
635*4882a593Smuzhiyun _next_match:
636*4882a593Smuzhiyun 		/* Encode Offset */
637*4882a593Smuzhiyun 		LZ4_writeLE16(op, (U16)(ip - match)); op += 2;
638*4882a593Smuzhiyun 
639*4882a593Smuzhiyun 		/* Encode MatchLength */
640*4882a593Smuzhiyun 		{
641*4882a593Smuzhiyun 			size_t matchLength = LZ4_count(ip + MINMATCH,
642*4882a593Smuzhiyun 			match + MINMATCH, matchlimit);
643*4882a593Smuzhiyun 
644*4882a593Smuzhiyun 			if (op + ((matchLength + 240)/255) > oMaxMatch) {
645*4882a593Smuzhiyun 				/* Match description too long : reduce it */
646*4882a593Smuzhiyun 				matchLength = (15 - 1) + (oMaxMatch - op) * 255;
647*4882a593Smuzhiyun 			}
648*4882a593Smuzhiyun 			ip += MINMATCH + matchLength;
649*4882a593Smuzhiyun 
650*4882a593Smuzhiyun 			if (matchLength >= ML_MASK) {
651*4882a593Smuzhiyun 				*token += ML_MASK;
652*4882a593Smuzhiyun 				matchLength -= ML_MASK;
653*4882a593Smuzhiyun 				while (matchLength >= 255) {
654*4882a593Smuzhiyun 					matchLength -= 255;
655*4882a593Smuzhiyun 					*op++ = 255;
656*4882a593Smuzhiyun 				}
657*4882a593Smuzhiyun 				*op++ = (BYTE)matchLength;
658*4882a593Smuzhiyun 			} else
659*4882a593Smuzhiyun 				*token += (BYTE)(matchLength);
660*4882a593Smuzhiyun 		}
661*4882a593Smuzhiyun 
662*4882a593Smuzhiyun 		anchor = ip;
663*4882a593Smuzhiyun 
664*4882a593Smuzhiyun 		/* Test end of block */
665*4882a593Smuzhiyun 		if (ip > mflimit)
666*4882a593Smuzhiyun 			break;
667*4882a593Smuzhiyun 		if (op > oMaxSeq)
668*4882a593Smuzhiyun 			break;
669*4882a593Smuzhiyun 
670*4882a593Smuzhiyun 		/* Fill table */
671*4882a593Smuzhiyun 		LZ4_putPosition(ip - 2, ctx->hashTable, tableType, base);
672*4882a593Smuzhiyun 
673*4882a593Smuzhiyun 		/* Test next position */
674*4882a593Smuzhiyun 		match = LZ4_getPosition(ip, ctx->hashTable, tableType, base);
675*4882a593Smuzhiyun 		LZ4_putPosition(ip, ctx->hashTable, tableType, base);
676*4882a593Smuzhiyun 
677*4882a593Smuzhiyun 		if ((match + MAX_DISTANCE >= ip)
678*4882a593Smuzhiyun 			&& (LZ4_read32(match) == LZ4_read32(ip))) {
679*4882a593Smuzhiyun 			token = op++; *token = 0;
680*4882a593Smuzhiyun 			goto _next_match;
681*4882a593Smuzhiyun 		}
682*4882a593Smuzhiyun 
683*4882a593Smuzhiyun 		/* Prepare next loop */
684*4882a593Smuzhiyun 		forwardH = LZ4_hashPosition(++ip, tableType);
685*4882a593Smuzhiyun 	}
686*4882a593Smuzhiyun 
687*4882a593Smuzhiyun _last_literals:
688*4882a593Smuzhiyun 	/* Encode Last Literals */
689*4882a593Smuzhiyun 	{
690*4882a593Smuzhiyun 		size_t lastRunSize = (size_t)(iend - anchor);
691*4882a593Smuzhiyun 
692*4882a593Smuzhiyun 		if (op + 1 /* token */
693*4882a593Smuzhiyun 			+ ((lastRunSize + 240) / 255) /* litLength */
694*4882a593Smuzhiyun 			+ lastRunSize /* literals */ > oend) {
695*4882a593Smuzhiyun 			/* adapt lastRunSize to fill 'dst' */
696*4882a593Smuzhiyun 			lastRunSize	= (oend - op) - 1;
697*4882a593Smuzhiyun 			lastRunSize -= (lastRunSize + 240) / 255;
698*4882a593Smuzhiyun 		}
699*4882a593Smuzhiyun 		ip = anchor + lastRunSize;
700*4882a593Smuzhiyun 
701*4882a593Smuzhiyun 		if (lastRunSize >= RUN_MASK) {
702*4882a593Smuzhiyun 			size_t accumulator = lastRunSize - RUN_MASK;
703*4882a593Smuzhiyun 
704*4882a593Smuzhiyun 			*op++ = RUN_MASK << ML_BITS;
705*4882a593Smuzhiyun 			for (; accumulator >= 255; accumulator -= 255)
706*4882a593Smuzhiyun 				*op++ = 255;
707*4882a593Smuzhiyun 			*op++ = (BYTE) accumulator;
708*4882a593Smuzhiyun 		} else {
709*4882a593Smuzhiyun 			*op++ = (BYTE)(lastRunSize<<ML_BITS);
710*4882a593Smuzhiyun 		}
711*4882a593Smuzhiyun 		LZ4_memcpy(op, anchor, lastRunSize);
712*4882a593Smuzhiyun 		op += lastRunSize;
713*4882a593Smuzhiyun 	}
714*4882a593Smuzhiyun 
715*4882a593Smuzhiyun 	/* End */
716*4882a593Smuzhiyun 	*srcSizePtr = (int) (((const char *)ip) - src);
717*4882a593Smuzhiyun 	return (int) (((char *)op) - dst);
718*4882a593Smuzhiyun }
719*4882a593Smuzhiyun 
LZ4_compress_destSize_extState(LZ4_stream_t * state,const char * src,char * dst,int * srcSizePtr,int targetDstSize)720*4882a593Smuzhiyun static int LZ4_compress_destSize_extState(
721*4882a593Smuzhiyun 	LZ4_stream_t *state,
722*4882a593Smuzhiyun 	const char *src,
723*4882a593Smuzhiyun 	char *dst,
724*4882a593Smuzhiyun 	int *srcSizePtr,
725*4882a593Smuzhiyun 	int targetDstSize)
726*4882a593Smuzhiyun {
727*4882a593Smuzhiyun #if LZ4_ARCH64
728*4882a593Smuzhiyun 	const tableType_t tableType = byU32;
729*4882a593Smuzhiyun #else
730*4882a593Smuzhiyun 	const tableType_t tableType = byPtr;
731*4882a593Smuzhiyun #endif
732*4882a593Smuzhiyun 
733*4882a593Smuzhiyun 	LZ4_resetStream(state);
734*4882a593Smuzhiyun 
735*4882a593Smuzhiyun 	if (targetDstSize >= LZ4_COMPRESSBOUND(*srcSizePtr)) {
736*4882a593Smuzhiyun 		/* compression success is guaranteed */
737*4882a593Smuzhiyun 		return LZ4_compress_fast_extState(
738*4882a593Smuzhiyun 			state, src, dst, *srcSizePtr,
739*4882a593Smuzhiyun 			targetDstSize, 1);
740*4882a593Smuzhiyun 	} else {
741*4882a593Smuzhiyun 		if (*srcSizePtr < LZ4_64Klimit)
742*4882a593Smuzhiyun 			return LZ4_compress_destSize_generic(
743*4882a593Smuzhiyun 				&state->internal_donotuse,
744*4882a593Smuzhiyun 				src, dst, srcSizePtr,
745*4882a593Smuzhiyun 				targetDstSize, byU16);
746*4882a593Smuzhiyun 		else
747*4882a593Smuzhiyun 			return LZ4_compress_destSize_generic(
748*4882a593Smuzhiyun 				&state->internal_donotuse,
749*4882a593Smuzhiyun 				src, dst, srcSizePtr,
750*4882a593Smuzhiyun 				targetDstSize, tableType);
751*4882a593Smuzhiyun 	}
752*4882a593Smuzhiyun }
753*4882a593Smuzhiyun 
754*4882a593Smuzhiyun 
LZ4_compress_destSize(const char * src,char * dst,int * srcSizePtr,int targetDstSize,void * wrkmem)755*4882a593Smuzhiyun int LZ4_compress_destSize(
756*4882a593Smuzhiyun 	const char *src,
757*4882a593Smuzhiyun 	char *dst,
758*4882a593Smuzhiyun 	int *srcSizePtr,
759*4882a593Smuzhiyun 	int targetDstSize,
760*4882a593Smuzhiyun 	void *wrkmem)
761*4882a593Smuzhiyun {
762*4882a593Smuzhiyun 	return LZ4_compress_destSize_extState(wrkmem, src, dst, srcSizePtr,
763*4882a593Smuzhiyun 		targetDstSize);
764*4882a593Smuzhiyun }
765*4882a593Smuzhiyun EXPORT_SYMBOL(LZ4_compress_destSize);
766*4882a593Smuzhiyun 
767*4882a593Smuzhiyun /*-******************************
768*4882a593Smuzhiyun  *	Streaming functions
769*4882a593Smuzhiyun  ********************************/
LZ4_resetStream(LZ4_stream_t * LZ4_stream)770*4882a593Smuzhiyun void LZ4_resetStream(LZ4_stream_t *LZ4_stream)
771*4882a593Smuzhiyun {
772*4882a593Smuzhiyun 	memset(LZ4_stream, 0, sizeof(LZ4_stream_t));
773*4882a593Smuzhiyun }
774*4882a593Smuzhiyun 
LZ4_loadDict(LZ4_stream_t * LZ4_dict,const char * dictionary,int dictSize)775*4882a593Smuzhiyun int LZ4_loadDict(LZ4_stream_t *LZ4_dict,
776*4882a593Smuzhiyun 	const char *dictionary, int dictSize)
777*4882a593Smuzhiyun {
778*4882a593Smuzhiyun 	LZ4_stream_t_internal *dict = &LZ4_dict->internal_donotuse;
779*4882a593Smuzhiyun 	const BYTE *p = (const BYTE *)dictionary;
780*4882a593Smuzhiyun 	const BYTE * const dictEnd = p + dictSize;
781*4882a593Smuzhiyun 	const BYTE *base;
782*4882a593Smuzhiyun 
783*4882a593Smuzhiyun 	if ((dict->initCheck)
784*4882a593Smuzhiyun 		|| (dict->currentOffset > 1 * GB)) {
785*4882a593Smuzhiyun 		/* Uninitialized structure, or reuse overflow */
786*4882a593Smuzhiyun 		LZ4_resetStream(LZ4_dict);
787*4882a593Smuzhiyun 	}
788*4882a593Smuzhiyun 
789*4882a593Smuzhiyun 	if (dictSize < (int)HASH_UNIT) {
790*4882a593Smuzhiyun 		dict->dictionary = NULL;
791*4882a593Smuzhiyun 		dict->dictSize = 0;
792*4882a593Smuzhiyun 		return 0;
793*4882a593Smuzhiyun 	}
794*4882a593Smuzhiyun 
795*4882a593Smuzhiyun 	if ((dictEnd - p) > 64 * KB)
796*4882a593Smuzhiyun 		p = dictEnd - 64 * KB;
797*4882a593Smuzhiyun 	dict->currentOffset += 64 * KB;
798*4882a593Smuzhiyun 	base = p - dict->currentOffset;
799*4882a593Smuzhiyun 	dict->dictionary = p;
800*4882a593Smuzhiyun 	dict->dictSize = (U32)(dictEnd - p);
801*4882a593Smuzhiyun 	dict->currentOffset += dict->dictSize;
802*4882a593Smuzhiyun 
803*4882a593Smuzhiyun 	while (p <= dictEnd - HASH_UNIT) {
804*4882a593Smuzhiyun 		LZ4_putPosition(p, dict->hashTable, byU32, base);
805*4882a593Smuzhiyun 		p += 3;
806*4882a593Smuzhiyun 	}
807*4882a593Smuzhiyun 
808*4882a593Smuzhiyun 	return dict->dictSize;
809*4882a593Smuzhiyun }
810*4882a593Smuzhiyun EXPORT_SYMBOL(LZ4_loadDict);
811*4882a593Smuzhiyun 
LZ4_renormDictT(LZ4_stream_t_internal * LZ4_dict,const BYTE * src)812*4882a593Smuzhiyun static void LZ4_renormDictT(LZ4_stream_t_internal *LZ4_dict,
813*4882a593Smuzhiyun 	const BYTE *src)
814*4882a593Smuzhiyun {
815*4882a593Smuzhiyun 	if ((LZ4_dict->currentOffset > 0x80000000) ||
816*4882a593Smuzhiyun 		((uptrval)LZ4_dict->currentOffset > (uptrval)src)) {
817*4882a593Smuzhiyun 		/* address space overflow */
818*4882a593Smuzhiyun 		/* rescale hash table */
819*4882a593Smuzhiyun 		U32 const delta = LZ4_dict->currentOffset - 64 * KB;
820*4882a593Smuzhiyun 		const BYTE *dictEnd = LZ4_dict->dictionary + LZ4_dict->dictSize;
821*4882a593Smuzhiyun 		int i;
822*4882a593Smuzhiyun 
823*4882a593Smuzhiyun 		for (i = 0; i < LZ4_HASH_SIZE_U32; i++) {
824*4882a593Smuzhiyun 			if (LZ4_dict->hashTable[i] < delta)
825*4882a593Smuzhiyun 				LZ4_dict->hashTable[i] = 0;
826*4882a593Smuzhiyun 			else
827*4882a593Smuzhiyun 				LZ4_dict->hashTable[i] -= delta;
828*4882a593Smuzhiyun 		}
829*4882a593Smuzhiyun 		LZ4_dict->currentOffset = 64 * KB;
830*4882a593Smuzhiyun 		if (LZ4_dict->dictSize > 64 * KB)
831*4882a593Smuzhiyun 			LZ4_dict->dictSize = 64 * KB;
832*4882a593Smuzhiyun 		LZ4_dict->dictionary = dictEnd - LZ4_dict->dictSize;
833*4882a593Smuzhiyun 	}
834*4882a593Smuzhiyun }
835*4882a593Smuzhiyun 
LZ4_saveDict(LZ4_stream_t * LZ4_dict,char * safeBuffer,int dictSize)836*4882a593Smuzhiyun int LZ4_saveDict(LZ4_stream_t *LZ4_dict, char *safeBuffer, int dictSize)
837*4882a593Smuzhiyun {
838*4882a593Smuzhiyun 	LZ4_stream_t_internal * const dict = &LZ4_dict->internal_donotuse;
839*4882a593Smuzhiyun 	const BYTE * const previousDictEnd = dict->dictionary + dict->dictSize;
840*4882a593Smuzhiyun 
841*4882a593Smuzhiyun 	if ((U32)dictSize > 64 * KB) {
842*4882a593Smuzhiyun 		/* useless to define a dictionary > 64 * KB */
843*4882a593Smuzhiyun 		dictSize = 64 * KB;
844*4882a593Smuzhiyun 	}
845*4882a593Smuzhiyun 	if ((U32)dictSize > dict->dictSize)
846*4882a593Smuzhiyun 		dictSize = dict->dictSize;
847*4882a593Smuzhiyun 
848*4882a593Smuzhiyun 	memmove(safeBuffer, previousDictEnd - dictSize, dictSize);
849*4882a593Smuzhiyun 
850*4882a593Smuzhiyun 	dict->dictionary = (const BYTE *)safeBuffer;
851*4882a593Smuzhiyun 	dict->dictSize = (U32)dictSize;
852*4882a593Smuzhiyun 
853*4882a593Smuzhiyun 	return dictSize;
854*4882a593Smuzhiyun }
855*4882a593Smuzhiyun EXPORT_SYMBOL(LZ4_saveDict);
856*4882a593Smuzhiyun 
LZ4_compress_fast_continue(LZ4_stream_t * LZ4_stream,const char * source,char * dest,int inputSize,int maxOutputSize,int acceleration)857*4882a593Smuzhiyun int LZ4_compress_fast_continue(LZ4_stream_t *LZ4_stream, const char *source,
858*4882a593Smuzhiyun 	char *dest, int inputSize, int maxOutputSize, int acceleration)
859*4882a593Smuzhiyun {
860*4882a593Smuzhiyun 	LZ4_stream_t_internal *streamPtr = &LZ4_stream->internal_donotuse;
861*4882a593Smuzhiyun 	const BYTE * const dictEnd = streamPtr->dictionary
862*4882a593Smuzhiyun 		+ streamPtr->dictSize;
863*4882a593Smuzhiyun 
864*4882a593Smuzhiyun 	const BYTE *smallest = (const BYTE *) source;
865*4882a593Smuzhiyun 
866*4882a593Smuzhiyun 	if (streamPtr->initCheck) {
867*4882a593Smuzhiyun 		/* Uninitialized structure detected */
868*4882a593Smuzhiyun 		return 0;
869*4882a593Smuzhiyun 	}
870*4882a593Smuzhiyun 
871*4882a593Smuzhiyun 	if ((streamPtr->dictSize > 0) && (smallest > dictEnd))
872*4882a593Smuzhiyun 		smallest = dictEnd;
873*4882a593Smuzhiyun 
874*4882a593Smuzhiyun 	LZ4_renormDictT(streamPtr, smallest);
875*4882a593Smuzhiyun 
876*4882a593Smuzhiyun 	if (acceleration < 1)
877*4882a593Smuzhiyun 		acceleration = LZ4_ACCELERATION_DEFAULT;
878*4882a593Smuzhiyun 
879*4882a593Smuzhiyun 	/* Check overlapping input/dictionary space */
880*4882a593Smuzhiyun 	{
881*4882a593Smuzhiyun 		const BYTE *sourceEnd = (const BYTE *) source + inputSize;
882*4882a593Smuzhiyun 
883*4882a593Smuzhiyun 		if ((sourceEnd > streamPtr->dictionary)
884*4882a593Smuzhiyun 			&& (sourceEnd < dictEnd)) {
885*4882a593Smuzhiyun 			streamPtr->dictSize = (U32)(dictEnd - sourceEnd);
886*4882a593Smuzhiyun 			if (streamPtr->dictSize > 64 * KB)
887*4882a593Smuzhiyun 				streamPtr->dictSize = 64 * KB;
888*4882a593Smuzhiyun 			if (streamPtr->dictSize < 4)
889*4882a593Smuzhiyun 				streamPtr->dictSize = 0;
890*4882a593Smuzhiyun 			streamPtr->dictionary = dictEnd - streamPtr->dictSize;
891*4882a593Smuzhiyun 		}
892*4882a593Smuzhiyun 	}
893*4882a593Smuzhiyun 
894*4882a593Smuzhiyun 	/* prefix mode : source data follows dictionary */
895*4882a593Smuzhiyun 	if (dictEnd == (const BYTE *)source) {
896*4882a593Smuzhiyun 		int result;
897*4882a593Smuzhiyun 
898*4882a593Smuzhiyun 		if ((streamPtr->dictSize < 64 * KB) &&
899*4882a593Smuzhiyun 			(streamPtr->dictSize < streamPtr->currentOffset)) {
900*4882a593Smuzhiyun 			result = LZ4_compress_generic(
901*4882a593Smuzhiyun 				streamPtr, source, dest, inputSize,
902*4882a593Smuzhiyun 				maxOutputSize, limitedOutput, byU32,
903*4882a593Smuzhiyun 				withPrefix64k, dictSmall, acceleration);
904*4882a593Smuzhiyun 		} else {
905*4882a593Smuzhiyun 			result = LZ4_compress_generic(
906*4882a593Smuzhiyun 				streamPtr, source, dest, inputSize,
907*4882a593Smuzhiyun 				maxOutputSize, limitedOutput, byU32,
908*4882a593Smuzhiyun 				withPrefix64k, noDictIssue, acceleration);
909*4882a593Smuzhiyun 		}
910*4882a593Smuzhiyun 		streamPtr->dictSize += (U32)inputSize;
911*4882a593Smuzhiyun 		streamPtr->currentOffset += (U32)inputSize;
912*4882a593Smuzhiyun 		return result;
913*4882a593Smuzhiyun 	}
914*4882a593Smuzhiyun 
915*4882a593Smuzhiyun 	/* external dictionary mode */
916*4882a593Smuzhiyun 	{
917*4882a593Smuzhiyun 		int result;
918*4882a593Smuzhiyun 
919*4882a593Smuzhiyun 		if ((streamPtr->dictSize < 64 * KB) &&
920*4882a593Smuzhiyun 			(streamPtr->dictSize < streamPtr->currentOffset)) {
921*4882a593Smuzhiyun 			result = LZ4_compress_generic(
922*4882a593Smuzhiyun 				streamPtr, source, dest, inputSize,
923*4882a593Smuzhiyun 				maxOutputSize, limitedOutput, byU32,
924*4882a593Smuzhiyun 				usingExtDict, dictSmall, acceleration);
925*4882a593Smuzhiyun 		} else {
926*4882a593Smuzhiyun 			result = LZ4_compress_generic(
927*4882a593Smuzhiyun 				streamPtr, source, dest, inputSize,
928*4882a593Smuzhiyun 				maxOutputSize, limitedOutput, byU32,
929*4882a593Smuzhiyun 				usingExtDict, noDictIssue, acceleration);
930*4882a593Smuzhiyun 		}
931*4882a593Smuzhiyun 		streamPtr->dictionary = (const BYTE *)source;
932*4882a593Smuzhiyun 		streamPtr->dictSize = (U32)inputSize;
933*4882a593Smuzhiyun 		streamPtr->currentOffset += (U32)inputSize;
934*4882a593Smuzhiyun 		return result;
935*4882a593Smuzhiyun 	}
936*4882a593Smuzhiyun }
937*4882a593Smuzhiyun EXPORT_SYMBOL(LZ4_compress_fast_continue);
938*4882a593Smuzhiyun 
939*4882a593Smuzhiyun MODULE_LICENSE("Dual BSD/GPL");
940*4882a593Smuzhiyun MODULE_DESCRIPTION("LZ4 compressor");
941