blob: 53f998a4374b1a2de72f3dd75be2f51f9d2cc48f [file] [log] [blame]
William Kurkianea869482019-04-09 15:16:11 -04001/*
2 * Copyright (c) 2016-present, Yann Collet, Facebook, Inc.
3 * All rights reserved.
4 *
5 * This source code is licensed under both the BSD-style license (found in the
6 * LICENSE file in the root directory of this source tree) and the GPLv2 (found
7 * in the COPYING file in the root directory of this source tree).
8 * You may select, at your option, one of the above-listed licenses.
9 */
10
11#include "zstd_compress_internal.h"
12#include "zstd_lazy.h"
13
14
15/*-*************************************
16* Binary Tree search
17***************************************/
18
Abhilash S.L3b494632019-07-16 15:51:09 +053019static void
20ZSTD_updateDUBT(ZSTD_matchState_t* ms,
William Kurkianea869482019-04-09 15:16:11 -040021 const BYTE* ip, const BYTE* iend,
22 U32 mls)
23{
Abhilash S.L3b494632019-07-16 15:51:09 +053024 const ZSTD_compressionParameters* const cParams = &ms->cParams;
William Kurkianea869482019-04-09 15:16:11 -040025 U32* const hashTable = ms->hashTable;
26 U32 const hashLog = cParams->hashLog;
27
28 U32* const bt = ms->chainTable;
29 U32 const btLog = cParams->chainLog - 1;
30 U32 const btMask = (1 << btLog) - 1;
31
32 const BYTE* const base = ms->window.base;
33 U32 const target = (U32)(ip - base);
34 U32 idx = ms->nextToUpdate;
35
36 if (idx != target)
37 DEBUGLOG(7, "ZSTD_updateDUBT, from %u to %u (dictLimit:%u)",
38 idx, target, ms->window.dictLimit);
39 assert(ip + 8 <= iend); /* condition for ZSTD_hashPtr */
40 (void)iend;
41
42 assert(idx >= ms->window.dictLimit); /* condition for valid base+idx */
43 for ( ; idx < target ; idx++) {
44 size_t const h = ZSTD_hashPtr(base + idx, hashLog, mls); /* assumption : ip + 8 <= iend */
45 U32 const matchIndex = hashTable[h];
46
47 U32* const nextCandidatePtr = bt + 2*(idx&btMask);
48 U32* const sortMarkPtr = nextCandidatePtr + 1;
49
50 DEBUGLOG(8, "ZSTD_updateDUBT: insert %u", idx);
51 hashTable[h] = idx; /* Update Hash Table */
52 *nextCandidatePtr = matchIndex; /* update BT like a chain */
53 *sortMarkPtr = ZSTD_DUBT_UNSORTED_MARK;
54 }
55 ms->nextToUpdate = target;
56}
57
58
59/** ZSTD_insertDUBT1() :
60 * sort one already inserted but unsorted position
61 * assumption : current >= btlow == (current - btmask)
62 * doesn't fail */
Abhilash S.L3b494632019-07-16 15:51:09 +053063static void
64ZSTD_insertDUBT1(ZSTD_matchState_t* ms,
William Kurkianea869482019-04-09 15:16:11 -040065 U32 current, const BYTE* inputEnd,
Abhilash S.L3b494632019-07-16 15:51:09 +053066 U32 nbCompares, U32 btLow,
67 const ZSTD_dictMode_e dictMode)
William Kurkianea869482019-04-09 15:16:11 -040068{
Abhilash S.L3b494632019-07-16 15:51:09 +053069 const ZSTD_compressionParameters* const cParams = &ms->cParams;
70 U32* const bt = ms->chainTable;
71 U32 const btLog = cParams->chainLog - 1;
72 U32 const btMask = (1 << btLog) - 1;
William Kurkianea869482019-04-09 15:16:11 -040073 size_t commonLengthSmaller=0, commonLengthLarger=0;
74 const BYTE* const base = ms->window.base;
75 const BYTE* const dictBase = ms->window.dictBase;
76 const U32 dictLimit = ms->window.dictLimit;
77 const BYTE* const ip = (current>=dictLimit) ? base + current : dictBase + current;
78 const BYTE* const iend = (current>=dictLimit) ? inputEnd : dictBase + dictLimit;
79 const BYTE* const dictEnd = dictBase + dictLimit;
80 const BYTE* const prefixStart = base + dictLimit;
81 const BYTE* match;
82 U32* smallerPtr = bt + 2*(current&btMask);
83 U32* largerPtr = smallerPtr + 1;
Abhilash S.L3b494632019-07-16 15:51:09 +053084 U32 matchIndex = *smallerPtr; /* this candidate is unsorted : next sorted candidate is reached through *smallerPtr, while *largerPtr contains previous unsorted candidate (which is already saved and can be overwritten) */
William Kurkianea869482019-04-09 15:16:11 -040085 U32 dummy32; /* to be nullified at the end */
86 U32 const windowLow = ms->window.lowLimit;
87
88 DEBUGLOG(8, "ZSTD_insertDUBT1(%u) (dictLimit=%u, lowLimit=%u)",
89 current, dictLimit, windowLow);
90 assert(current >= btLow);
91 assert(ip < iend); /* condition for ZSTD_count */
92
93 while (nbCompares-- && (matchIndex > windowLow)) {
94 U32* const nextPtr = bt + 2*(matchIndex & btMask);
95 size_t matchLength = MIN(commonLengthSmaller, commonLengthLarger); /* guaranteed minimum nb of common bytes */
96 assert(matchIndex < current);
Abhilash S.L3b494632019-07-16 15:51:09 +053097 /* note : all candidates are now supposed sorted,
98 * but it's still possible to have nextPtr[1] == ZSTD_DUBT_UNSORTED_MARK
99 * when a real index has the same value as ZSTD_DUBT_UNSORTED_MARK */
William Kurkianea869482019-04-09 15:16:11 -0400100
Abhilash S.L3b494632019-07-16 15:51:09 +0530101 if ( (dictMode != ZSTD_extDict)
William Kurkianea869482019-04-09 15:16:11 -0400102 || (matchIndex+matchLength >= dictLimit) /* both in current segment*/
103 || (current < dictLimit) /* both in extDict */) {
Abhilash S.L3b494632019-07-16 15:51:09 +0530104 const BYTE* const mBase = ( (dictMode != ZSTD_extDict)
105 || (matchIndex+matchLength >= dictLimit)) ?
106 base : dictBase;
William Kurkianea869482019-04-09 15:16:11 -0400107 assert( (matchIndex+matchLength >= dictLimit) /* might be wrong if extDict is incorrectly set to 0 */
108 || (current < dictLimit) );
109 match = mBase + matchIndex;
110 matchLength += ZSTD_count(ip+matchLength, match+matchLength, iend);
111 } else {
112 match = dictBase + matchIndex;
113 matchLength += ZSTD_count_2segments(ip+matchLength, match+matchLength, iend, dictEnd, prefixStart);
114 if (matchIndex+matchLength >= dictLimit)
Abhilash S.L3b494632019-07-16 15:51:09 +0530115 match = base + matchIndex; /* preparation for next read of match[matchLength] */
William Kurkianea869482019-04-09 15:16:11 -0400116 }
117
118 DEBUGLOG(8, "ZSTD_insertDUBT1: comparing %u with %u : found %u common bytes ",
119 current, matchIndex, (U32)matchLength);
120
121 if (ip+matchLength == iend) { /* equal : no way to know if inf or sup */
122 break; /* drop , to guarantee consistency ; miss a bit of compression, but other solutions can corrupt tree */
123 }
124
125 if (match[matchLength] < ip[matchLength]) { /* necessarily within buffer */
126 /* match is smaller than current */
127 *smallerPtr = matchIndex; /* update smaller idx */
128 commonLengthSmaller = matchLength; /* all smaller will now have at least this guaranteed common length */
129 if (matchIndex <= btLow) { smallerPtr=&dummy32; break; } /* beyond tree size, stop searching */
130 DEBUGLOG(8, "ZSTD_insertDUBT1: %u (>btLow=%u) is smaller : next => %u",
131 matchIndex, btLow, nextPtr[1]);
132 smallerPtr = nextPtr+1; /* new "candidate" => larger than match, which was smaller than target */
133 matchIndex = nextPtr[1]; /* new matchIndex, larger than previous and closer to current */
134 } else {
135 /* match is larger than current */
136 *largerPtr = matchIndex;
137 commonLengthLarger = matchLength;
138 if (matchIndex <= btLow) { largerPtr=&dummy32; break; } /* beyond tree size, stop searching */
139 DEBUGLOG(8, "ZSTD_insertDUBT1: %u (>btLow=%u) is larger => %u",
140 matchIndex, btLow, nextPtr[0]);
141 largerPtr = nextPtr;
142 matchIndex = nextPtr[0];
143 } }
144
145 *smallerPtr = *largerPtr = 0;
146}
147
148
Abhilash S.L3b494632019-07-16 15:51:09 +0530149static size_t
150ZSTD_DUBT_findBetterDictMatch (
151 ZSTD_matchState_t* ms,
152 const BYTE* const ip, const BYTE* const iend,
153 size_t* offsetPtr,
154 size_t bestLength,
155 U32 nbCompares,
156 U32 const mls,
157 const ZSTD_dictMode_e dictMode)
William Kurkianea869482019-04-09 15:16:11 -0400158{
Abhilash S.L3b494632019-07-16 15:51:09 +0530159 const ZSTD_matchState_t * const dms = ms->dictMatchState;
160 const ZSTD_compressionParameters* const dmsCParams = &dms->cParams;
161 const U32 * const dictHashTable = dms->hashTable;
162 U32 const hashLog = dmsCParams->hashLog;
163 size_t const h = ZSTD_hashPtr(ip, hashLog, mls);
164 U32 dictMatchIndex = dictHashTable[h];
165
166 const BYTE* const base = ms->window.base;
167 const BYTE* const prefixStart = base + ms->window.dictLimit;
168 U32 const current = (U32)(ip-base);
169 const BYTE* const dictBase = dms->window.base;
170 const BYTE* const dictEnd = dms->window.nextSrc;
171 U32 const dictHighLimit = (U32)(dms->window.nextSrc - dms->window.base);
172 U32 const dictLowLimit = dms->window.lowLimit;
173 U32 const dictIndexDelta = ms->window.lowLimit - dictHighLimit;
174
175 U32* const dictBt = dms->chainTable;
176 U32 const btLog = dmsCParams->chainLog - 1;
177 U32 const btMask = (1 << btLog) - 1;
178 U32 const btLow = (btMask >= dictHighLimit - dictLowLimit) ? dictLowLimit : dictHighLimit - btMask;
179
180 size_t commonLengthSmaller=0, commonLengthLarger=0;
181
182 (void)dictMode;
183 assert(dictMode == ZSTD_dictMatchState);
184
185 while (nbCompares-- && (dictMatchIndex > dictLowLimit)) {
186 U32* const nextPtr = dictBt + 2*(dictMatchIndex & btMask);
187 size_t matchLength = MIN(commonLengthSmaller, commonLengthLarger); /* guaranteed minimum nb of common bytes */
188 const BYTE* match = dictBase + dictMatchIndex;
189 matchLength += ZSTD_count_2segments(ip+matchLength, match+matchLength, iend, dictEnd, prefixStart);
190 if (dictMatchIndex+matchLength >= dictHighLimit)
191 match = base + dictMatchIndex + dictIndexDelta; /* to prepare for next usage of match[matchLength] */
192
193 if (matchLength > bestLength) {
194 U32 matchIndex = dictMatchIndex + dictIndexDelta;
195 if ( (4*(int)(matchLength-bestLength)) > (int)(ZSTD_highbit32(current-matchIndex+1) - ZSTD_highbit32((U32)offsetPtr[0]+1)) ) {
196 DEBUGLOG(9, "ZSTD_DUBT_findBetterDictMatch(%u) : found better match length %u -> %u and offsetCode %u -> %u (dictMatchIndex %u, matchIndex %u)",
197 current, (U32)bestLength, (U32)matchLength, (U32)*offsetPtr, ZSTD_REP_MOVE + current - matchIndex, dictMatchIndex, matchIndex);
198 bestLength = matchLength, *offsetPtr = ZSTD_REP_MOVE + current - matchIndex;
199 }
200 if (ip+matchLength == iend) { /* reached end of input : ip[matchLength] is not valid, no way to know if it's larger or smaller than match */
201 break; /* drop, to guarantee consistency (miss a little bit of compression) */
202 }
203 }
204
205 if (match[matchLength] < ip[matchLength]) {
206 if (dictMatchIndex <= btLow) { break; } /* beyond tree size, stop the search */
207 commonLengthSmaller = matchLength; /* all smaller will now have at least this guaranteed common length */
208 dictMatchIndex = nextPtr[1]; /* new matchIndex larger than previous (closer to current) */
209 } else {
210 /* match is larger than current */
211 if (dictMatchIndex <= btLow) { break; } /* beyond tree size, stop the search */
212 commonLengthLarger = matchLength;
213 dictMatchIndex = nextPtr[0];
214 }
215 }
216
217 if (bestLength >= MINMATCH) {
218 U32 const mIndex = current - ((U32)*offsetPtr - ZSTD_REP_MOVE); (void)mIndex;
219 DEBUGLOG(8, "ZSTD_DUBT_findBetterDictMatch(%u) : found match of length %u and offsetCode %u (pos %u)",
220 current, (U32)bestLength, (U32)*offsetPtr, mIndex);
221 }
222 return bestLength;
223
224}
225
226
227static size_t
228ZSTD_DUBT_findBestMatch(ZSTD_matchState_t* ms,
229 const BYTE* const ip, const BYTE* const iend,
230 size_t* offsetPtr,
231 U32 const mls,
232 const ZSTD_dictMode_e dictMode)
233{
234 const ZSTD_compressionParameters* const cParams = &ms->cParams;
William Kurkianea869482019-04-09 15:16:11 -0400235 U32* const hashTable = ms->hashTable;
236 U32 const hashLog = cParams->hashLog;
237 size_t const h = ZSTD_hashPtr(ip, hashLog, mls);
238 U32 matchIndex = hashTable[h];
239
240 const BYTE* const base = ms->window.base;
241 U32 const current = (U32)(ip-base);
242 U32 const windowLow = ms->window.lowLimit;
243
244 U32* const bt = ms->chainTable;
245 U32 const btLog = cParams->chainLog - 1;
246 U32 const btMask = (1 << btLog) - 1;
247 U32 const btLow = (btMask >= current) ? 0 : current - btMask;
248 U32 const unsortLimit = MAX(btLow, windowLow);
249
250 U32* nextCandidate = bt + 2*(matchIndex&btMask);
251 U32* unsortedMark = bt + 2*(matchIndex&btMask) + 1;
252 U32 nbCompares = 1U << cParams->searchLog;
253 U32 nbCandidates = nbCompares;
254 U32 previousCandidate = 0;
255
256 DEBUGLOG(7, "ZSTD_DUBT_findBestMatch (%u) ", current);
257 assert(ip <= iend-8); /* required for h calculation */
258
259 /* reach end of unsorted candidates list */
260 while ( (matchIndex > unsortLimit)
261 && (*unsortedMark == ZSTD_DUBT_UNSORTED_MARK)
262 && (nbCandidates > 1) ) {
263 DEBUGLOG(8, "ZSTD_DUBT_findBestMatch: candidate %u is unsorted",
264 matchIndex);
Abhilash S.L3b494632019-07-16 15:51:09 +0530265 *unsortedMark = previousCandidate; /* the unsortedMark becomes a reversed chain, to move up back to original position */
William Kurkianea869482019-04-09 15:16:11 -0400266 previousCandidate = matchIndex;
267 matchIndex = *nextCandidate;
268 nextCandidate = bt + 2*(matchIndex&btMask);
269 unsortedMark = bt + 2*(matchIndex&btMask) + 1;
270 nbCandidates --;
271 }
272
Abhilash S.L3b494632019-07-16 15:51:09 +0530273 /* nullify last candidate if it's still unsorted
274 * simplification, detrimental to compression ratio, beneficial for speed */
William Kurkianea869482019-04-09 15:16:11 -0400275 if ( (matchIndex > unsortLimit)
276 && (*unsortedMark==ZSTD_DUBT_UNSORTED_MARK) ) {
277 DEBUGLOG(7, "ZSTD_DUBT_findBestMatch: nullify last unsorted candidate %u",
278 matchIndex);
Abhilash S.L3b494632019-07-16 15:51:09 +0530279 *nextCandidate = *unsortedMark = 0;
William Kurkianea869482019-04-09 15:16:11 -0400280 }
281
282 /* batch sort stacked candidates */
283 matchIndex = previousCandidate;
284 while (matchIndex) { /* will end on matchIndex == 0 */
285 U32* const nextCandidateIdxPtr = bt + 2*(matchIndex&btMask) + 1;
286 U32 const nextCandidateIdx = *nextCandidateIdxPtr;
Abhilash S.L3b494632019-07-16 15:51:09 +0530287 ZSTD_insertDUBT1(ms, matchIndex, iend,
288 nbCandidates, unsortLimit, dictMode);
William Kurkianea869482019-04-09 15:16:11 -0400289 matchIndex = nextCandidateIdx;
290 nbCandidates++;
291 }
292
293 /* find longest match */
Abhilash S.L3b494632019-07-16 15:51:09 +0530294 { size_t commonLengthSmaller = 0, commonLengthLarger = 0;
William Kurkianea869482019-04-09 15:16:11 -0400295 const BYTE* const dictBase = ms->window.dictBase;
296 const U32 dictLimit = ms->window.dictLimit;
297 const BYTE* const dictEnd = dictBase + dictLimit;
298 const BYTE* const prefixStart = base + dictLimit;
299 U32* smallerPtr = bt + 2*(current&btMask);
300 U32* largerPtr = bt + 2*(current&btMask) + 1;
Abhilash S.L3b494632019-07-16 15:51:09 +0530301 U32 matchEndIdx = current + 8 + 1;
William Kurkianea869482019-04-09 15:16:11 -0400302 U32 dummy32; /* to be nullified at the end */
303 size_t bestLength = 0;
304
305 matchIndex = hashTable[h];
306 hashTable[h] = current; /* Update Hash Table */
307
308 while (nbCompares-- && (matchIndex > windowLow)) {
309 U32* const nextPtr = bt + 2*(matchIndex & btMask);
310 size_t matchLength = MIN(commonLengthSmaller, commonLengthLarger); /* guaranteed minimum nb of common bytes */
311 const BYTE* match;
312
Abhilash S.L3b494632019-07-16 15:51:09 +0530313 if ((dictMode != ZSTD_extDict) || (matchIndex+matchLength >= dictLimit)) {
William Kurkianea869482019-04-09 15:16:11 -0400314 match = base + matchIndex;
315 matchLength += ZSTD_count(ip+matchLength, match+matchLength, iend);
316 } else {
317 match = dictBase + matchIndex;
318 matchLength += ZSTD_count_2segments(ip+matchLength, match+matchLength, iend, dictEnd, prefixStart);
319 if (matchIndex+matchLength >= dictLimit)
320 match = base + matchIndex; /* to prepare for next usage of match[matchLength] */
321 }
322
323 if (matchLength > bestLength) {
324 if (matchLength > matchEndIdx - matchIndex)
325 matchEndIdx = matchIndex + (U32)matchLength;
326 if ( (4*(int)(matchLength-bestLength)) > (int)(ZSTD_highbit32(current-matchIndex+1) - ZSTD_highbit32((U32)offsetPtr[0]+1)) )
327 bestLength = matchLength, *offsetPtr = ZSTD_REP_MOVE + current - matchIndex;
328 if (ip+matchLength == iend) { /* equal : no way to know if inf or sup */
Abhilash S.L3b494632019-07-16 15:51:09 +0530329 if (dictMode == ZSTD_dictMatchState) {
330 nbCompares = 0; /* in addition to avoiding checking any
331 * further in this loop, make sure we
332 * skip checking in the dictionary. */
333 }
William Kurkianea869482019-04-09 15:16:11 -0400334 break; /* drop, to guarantee consistency (miss a little bit of compression) */
335 }
336 }
337
338 if (match[matchLength] < ip[matchLength]) {
339 /* match is smaller than current */
340 *smallerPtr = matchIndex; /* update smaller idx */
341 commonLengthSmaller = matchLength; /* all smaller will now have at least this guaranteed common length */
342 if (matchIndex <= btLow) { smallerPtr=&dummy32; break; } /* beyond tree size, stop the search */
343 smallerPtr = nextPtr+1; /* new "smaller" => larger of match */
344 matchIndex = nextPtr[1]; /* new matchIndex larger than previous (closer to current) */
345 } else {
346 /* match is larger than current */
347 *largerPtr = matchIndex;
348 commonLengthLarger = matchLength;
349 if (matchIndex <= btLow) { largerPtr=&dummy32; break; } /* beyond tree size, stop the search */
350 largerPtr = nextPtr;
351 matchIndex = nextPtr[0];
352 } }
353
354 *smallerPtr = *largerPtr = 0;
355
Abhilash S.L3b494632019-07-16 15:51:09 +0530356 if (dictMode == ZSTD_dictMatchState && nbCompares) {
357 bestLength = ZSTD_DUBT_findBetterDictMatch(
358 ms, ip, iend,
359 offsetPtr, bestLength, nbCompares,
360 mls, dictMode);
361 }
362
William Kurkianea869482019-04-09 15:16:11 -0400363 assert(matchEndIdx > current+8); /* ensure nextToUpdate is increased */
364 ms->nextToUpdate = matchEndIdx - 8; /* skip repetitive patterns */
365 if (bestLength >= MINMATCH) {
366 U32 const mIndex = current - ((U32)*offsetPtr - ZSTD_REP_MOVE); (void)mIndex;
367 DEBUGLOG(8, "ZSTD_DUBT_findBestMatch(%u) : found match of length %u and offsetCode %u (pos %u)",
368 current, (U32)bestLength, (U32)*offsetPtr, mIndex);
369 }
370 return bestLength;
371 }
372}
373
374
375/** ZSTD_BtFindBestMatch() : Tree updater, providing best match */
Abhilash S.L3b494632019-07-16 15:51:09 +0530376FORCE_INLINE_TEMPLATE size_t
377ZSTD_BtFindBestMatch( ZSTD_matchState_t* ms,
378 const BYTE* const ip, const BYTE* const iLimit,
379 size_t* offsetPtr,
380 const U32 mls /* template */,
381 const ZSTD_dictMode_e dictMode)
William Kurkianea869482019-04-09 15:16:11 -0400382{
383 DEBUGLOG(7, "ZSTD_BtFindBestMatch");
384 if (ip < ms->window.base + ms->nextToUpdate) return 0; /* skipped area */
Abhilash S.L3b494632019-07-16 15:51:09 +0530385 ZSTD_updateDUBT(ms, ip, iLimit, mls);
386 return ZSTD_DUBT_findBestMatch(ms, ip, iLimit, offsetPtr, mls, dictMode);
William Kurkianea869482019-04-09 15:16:11 -0400387}
388
389
Abhilash S.L3b494632019-07-16 15:51:09 +0530390static size_t
391ZSTD_BtFindBestMatch_selectMLS ( ZSTD_matchState_t* ms,
392 const BYTE* ip, const BYTE* const iLimit,
393 size_t* offsetPtr)
William Kurkianea869482019-04-09 15:16:11 -0400394{
Abhilash S.L3b494632019-07-16 15:51:09 +0530395 switch(ms->cParams.minMatch)
William Kurkianea869482019-04-09 15:16:11 -0400396 {
397 default : /* includes case 3 */
Abhilash S.L3b494632019-07-16 15:51:09 +0530398 case 4 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 4, ZSTD_noDict);
399 case 5 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 5, ZSTD_noDict);
William Kurkianea869482019-04-09 15:16:11 -0400400 case 7 :
Abhilash S.L3b494632019-07-16 15:51:09 +0530401 case 6 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 6, ZSTD_noDict);
William Kurkianea869482019-04-09 15:16:11 -0400402 }
403}
404
405
Abhilash S.L3b494632019-07-16 15:51:09 +0530406static size_t ZSTD_BtFindBestMatch_dictMatchState_selectMLS (
407 ZSTD_matchState_t* ms,
William Kurkianea869482019-04-09 15:16:11 -0400408 const BYTE* ip, const BYTE* const iLimit,
409 size_t* offsetPtr)
410{
Abhilash S.L3b494632019-07-16 15:51:09 +0530411 switch(ms->cParams.minMatch)
William Kurkianea869482019-04-09 15:16:11 -0400412 {
413 default : /* includes case 3 */
Abhilash S.L3b494632019-07-16 15:51:09 +0530414 case 4 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 4, ZSTD_dictMatchState);
415 case 5 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 5, ZSTD_dictMatchState);
William Kurkianea869482019-04-09 15:16:11 -0400416 case 7 :
Abhilash S.L3b494632019-07-16 15:51:09 +0530417 case 6 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 6, ZSTD_dictMatchState);
418 }
419}
420
421
422static size_t ZSTD_BtFindBestMatch_extDict_selectMLS (
423 ZSTD_matchState_t* ms,
424 const BYTE* ip, const BYTE* const iLimit,
425 size_t* offsetPtr)
426{
427 switch(ms->cParams.minMatch)
428 {
429 default : /* includes case 3 */
430 case 4 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 4, ZSTD_extDict);
431 case 5 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 5, ZSTD_extDict);
432 case 7 :
433 case 6 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 6, ZSTD_extDict);
William Kurkianea869482019-04-09 15:16:11 -0400434 }
435}
436
437
438
439/* *********************************
440* Hash Chain
441***********************************/
Abhilash S.L3b494632019-07-16 15:51:09 +0530442#define NEXT_IN_CHAIN(d, mask) chainTable[(d) & (mask)]
William Kurkianea869482019-04-09 15:16:11 -0400443
444/* Update chains up to ip (excluded)
445 Assumption : always within prefix (i.e. not within extDict) */
446static U32 ZSTD_insertAndFindFirstIndex_internal(
Abhilash S.L3b494632019-07-16 15:51:09 +0530447 ZSTD_matchState_t* ms,
448 const ZSTD_compressionParameters* const cParams,
William Kurkianea869482019-04-09 15:16:11 -0400449 const BYTE* ip, U32 const mls)
450{
451 U32* const hashTable = ms->hashTable;
452 const U32 hashLog = cParams->hashLog;
453 U32* const chainTable = ms->chainTable;
454 const U32 chainMask = (1 << cParams->chainLog) - 1;
455 const BYTE* const base = ms->window.base;
456 const U32 target = (U32)(ip - base);
457 U32 idx = ms->nextToUpdate;
458
459 while(idx < target) { /* catch up */
460 size_t const h = ZSTD_hashPtr(base+idx, hashLog, mls);
461 NEXT_IN_CHAIN(idx, chainMask) = hashTable[h];
462 hashTable[h] = idx;
463 idx++;
464 }
465
466 ms->nextToUpdate = target;
467 return hashTable[ZSTD_hashPtr(ip, hashLog, mls)];
468}
469
Abhilash S.L3b494632019-07-16 15:51:09 +0530470U32 ZSTD_insertAndFindFirstIndex(ZSTD_matchState_t* ms, const BYTE* ip) {
471 const ZSTD_compressionParameters* const cParams = &ms->cParams;
472 return ZSTD_insertAndFindFirstIndex_internal(ms, cParams, ip, ms->cParams.minMatch);
William Kurkianea869482019-04-09 15:16:11 -0400473}
474
475
476/* inlining is important to hardwire a hot branch (template emulation) */
477FORCE_INLINE_TEMPLATE
478size_t ZSTD_HcFindBestMatch_generic (
Abhilash S.L3b494632019-07-16 15:51:09 +0530479 ZSTD_matchState_t* ms,
William Kurkianea869482019-04-09 15:16:11 -0400480 const BYTE* const ip, const BYTE* const iLimit,
481 size_t* offsetPtr,
Abhilash S.L3b494632019-07-16 15:51:09 +0530482 const U32 mls, const ZSTD_dictMode_e dictMode)
William Kurkianea869482019-04-09 15:16:11 -0400483{
Abhilash S.L3b494632019-07-16 15:51:09 +0530484 const ZSTD_compressionParameters* const cParams = &ms->cParams;
William Kurkianea869482019-04-09 15:16:11 -0400485 U32* const chainTable = ms->chainTable;
486 const U32 chainSize = (1 << cParams->chainLog);
487 const U32 chainMask = chainSize-1;
488 const BYTE* const base = ms->window.base;
489 const BYTE* const dictBase = ms->window.dictBase;
490 const U32 dictLimit = ms->window.dictLimit;
491 const BYTE* const prefixStart = base + dictLimit;
492 const BYTE* const dictEnd = dictBase + dictLimit;
493 const U32 lowLimit = ms->window.lowLimit;
494 const U32 current = (U32)(ip-base);
495 const U32 minChain = current > chainSize ? current - chainSize : 0;
496 U32 nbAttempts = 1U << cParams->searchLog;
497 size_t ml=4-1;
498
499 /* HC4 match finder */
500 U32 matchIndex = ZSTD_insertAndFindFirstIndex_internal(ms, cParams, ip, mls);
501
502 for ( ; (matchIndex>lowLimit) & (nbAttempts>0) ; nbAttempts--) {
503 size_t currentMl=0;
Abhilash S.L3b494632019-07-16 15:51:09 +0530504 if ((dictMode != ZSTD_extDict) || matchIndex >= dictLimit) {
William Kurkianea869482019-04-09 15:16:11 -0400505 const BYTE* const match = base + matchIndex;
Abhilash S.L3b494632019-07-16 15:51:09 +0530506 assert(matchIndex >= dictLimit); /* ensures this is true if dictMode != ZSTD_extDict */
William Kurkianea869482019-04-09 15:16:11 -0400507 if (match[ml] == ip[ml]) /* potentially better */
508 currentMl = ZSTD_count(ip, match, iLimit);
509 } else {
510 const BYTE* const match = dictBase + matchIndex;
511 assert(match+4 <= dictEnd);
512 if (MEM_read32(match) == MEM_read32(ip)) /* assumption : matchIndex <= dictLimit-4 (by table construction) */
513 currentMl = ZSTD_count_2segments(ip+4, match+4, iLimit, dictEnd, prefixStart) + 4;
514 }
515
516 /* save best solution */
517 if (currentMl > ml) {
518 ml = currentMl;
519 *offsetPtr = current - matchIndex + ZSTD_REP_MOVE;
520 if (ip+currentMl == iLimit) break; /* best possible, avoids read overflow on next attempt */
521 }
522
523 if (matchIndex <= minChain) break;
524 matchIndex = NEXT_IN_CHAIN(matchIndex, chainMask);
525 }
526
Abhilash S.L3b494632019-07-16 15:51:09 +0530527 if (dictMode == ZSTD_dictMatchState) {
528 const ZSTD_matchState_t* const dms = ms->dictMatchState;
529 const U32* const dmsChainTable = dms->chainTable;
530 const U32 dmsChainSize = (1 << dms->cParams.chainLog);
531 const U32 dmsChainMask = dmsChainSize - 1;
532 const U32 dmsLowestIndex = dms->window.dictLimit;
533 const BYTE* const dmsBase = dms->window.base;
534 const BYTE* const dmsEnd = dms->window.nextSrc;
535 const U32 dmsSize = (U32)(dmsEnd - dmsBase);
536 const U32 dmsIndexDelta = dictLimit - dmsSize;
537 const U32 dmsMinChain = dmsSize > dmsChainSize ? dmsSize - dmsChainSize : 0;
538
539 matchIndex = dms->hashTable[ZSTD_hashPtr(ip, dms->cParams.hashLog, mls)];
540
541 for ( ; (matchIndex>dmsLowestIndex) & (nbAttempts>0) ; nbAttempts--) {
542 size_t currentMl=0;
543 const BYTE* const match = dmsBase + matchIndex;
544 assert(match+4 <= dmsEnd);
545 if (MEM_read32(match) == MEM_read32(ip)) /* assumption : matchIndex <= dictLimit-4 (by table construction) */
546 currentMl = ZSTD_count_2segments(ip+4, match+4, iLimit, dmsEnd, prefixStart) + 4;
547
548 /* save best solution */
549 if (currentMl > ml) {
550 ml = currentMl;
551 *offsetPtr = current - (matchIndex + dmsIndexDelta) + ZSTD_REP_MOVE;
552 if (ip+currentMl == iLimit) break; /* best possible, avoids read overflow on next attempt */
553 }
554
555 if (matchIndex <= dmsMinChain) break;
556 matchIndex = dmsChainTable[matchIndex & dmsChainMask];
557 }
558 }
559
William Kurkianea869482019-04-09 15:16:11 -0400560 return ml;
561}
562
563
564FORCE_INLINE_TEMPLATE size_t ZSTD_HcFindBestMatch_selectMLS (
Abhilash S.L3b494632019-07-16 15:51:09 +0530565 ZSTD_matchState_t* ms,
William Kurkianea869482019-04-09 15:16:11 -0400566 const BYTE* ip, const BYTE* const iLimit,
567 size_t* offsetPtr)
568{
Abhilash S.L3b494632019-07-16 15:51:09 +0530569 switch(ms->cParams.minMatch)
William Kurkianea869482019-04-09 15:16:11 -0400570 {
571 default : /* includes case 3 */
Abhilash S.L3b494632019-07-16 15:51:09 +0530572 case 4 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 4, ZSTD_noDict);
573 case 5 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 5, ZSTD_noDict);
William Kurkianea869482019-04-09 15:16:11 -0400574 case 7 :
Abhilash S.L3b494632019-07-16 15:51:09 +0530575 case 6 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 6, ZSTD_noDict);
576 }
577}
578
579
580static size_t ZSTD_HcFindBestMatch_dictMatchState_selectMLS (
581 ZSTD_matchState_t* ms,
582 const BYTE* ip, const BYTE* const iLimit,
583 size_t* offsetPtr)
584{
585 switch(ms->cParams.minMatch)
586 {
587 default : /* includes case 3 */
588 case 4 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 4, ZSTD_dictMatchState);
589 case 5 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 5, ZSTD_dictMatchState);
590 case 7 :
591 case 6 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 6, ZSTD_dictMatchState);
William Kurkianea869482019-04-09 15:16:11 -0400592 }
593}
594
595
596FORCE_INLINE_TEMPLATE size_t ZSTD_HcFindBestMatch_extDict_selectMLS (
Abhilash S.L3b494632019-07-16 15:51:09 +0530597 ZSTD_matchState_t* ms,
William Kurkianea869482019-04-09 15:16:11 -0400598 const BYTE* ip, const BYTE* const iLimit,
Abhilash S.L3b494632019-07-16 15:51:09 +0530599 size_t* offsetPtr)
William Kurkianea869482019-04-09 15:16:11 -0400600{
Abhilash S.L3b494632019-07-16 15:51:09 +0530601 switch(ms->cParams.minMatch)
William Kurkianea869482019-04-09 15:16:11 -0400602 {
603 default : /* includes case 3 */
Abhilash S.L3b494632019-07-16 15:51:09 +0530604 case 4 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 4, ZSTD_extDict);
605 case 5 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 5, ZSTD_extDict);
William Kurkianea869482019-04-09 15:16:11 -0400606 case 7 :
Abhilash S.L3b494632019-07-16 15:51:09 +0530607 case 6 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 6, ZSTD_extDict);
William Kurkianea869482019-04-09 15:16:11 -0400608 }
609}
610
611
612/* *******************************
613* Common parser - lazy strategy
614*********************************/
615FORCE_INLINE_TEMPLATE
616size_t ZSTD_compressBlock_lazy_generic(
617 ZSTD_matchState_t* ms, seqStore_t* seqStore,
618 U32 rep[ZSTD_REP_NUM],
William Kurkianea869482019-04-09 15:16:11 -0400619 const void* src, size_t srcSize,
Abhilash S.L3b494632019-07-16 15:51:09 +0530620 const U32 searchMethod, const U32 depth,
621 ZSTD_dictMode_e const dictMode)
William Kurkianea869482019-04-09 15:16:11 -0400622{
623 const BYTE* const istart = (const BYTE*)src;
624 const BYTE* ip = istart;
625 const BYTE* anchor = istart;
626 const BYTE* const iend = istart + srcSize;
627 const BYTE* const ilimit = iend - 8;
Abhilash S.L3b494632019-07-16 15:51:09 +0530628 const BYTE* const base = ms->window.base;
629 const U32 prefixLowestIndex = ms->window.dictLimit;
630 const BYTE* const prefixLowest = base + prefixLowestIndex;
William Kurkianea869482019-04-09 15:16:11 -0400631
632 typedef size_t (*searchMax_f)(
Abhilash S.L3b494632019-07-16 15:51:09 +0530633 ZSTD_matchState_t* ms,
William Kurkianea869482019-04-09 15:16:11 -0400634 const BYTE* ip, const BYTE* iLimit, size_t* offsetPtr);
Abhilash S.L3b494632019-07-16 15:51:09 +0530635 searchMax_f const searchMax = dictMode == ZSTD_dictMatchState ?
636 (searchMethod ? ZSTD_BtFindBestMatch_dictMatchState_selectMLS : ZSTD_HcFindBestMatch_dictMatchState_selectMLS) :
637 (searchMethod ? ZSTD_BtFindBestMatch_selectMLS : ZSTD_HcFindBestMatch_selectMLS);
William Kurkianea869482019-04-09 15:16:11 -0400638 U32 offset_1 = rep[0], offset_2 = rep[1], savedOffset=0;
639
Abhilash S.L3b494632019-07-16 15:51:09 +0530640 const ZSTD_matchState_t* const dms = ms->dictMatchState;
641 const U32 dictLowestIndex = dictMode == ZSTD_dictMatchState ?
642 dms->window.dictLimit : 0;
643 const BYTE* const dictBase = dictMode == ZSTD_dictMatchState ?
644 dms->window.base : NULL;
645 const BYTE* const dictLowest = dictMode == ZSTD_dictMatchState ?
646 dictBase + dictLowestIndex : NULL;
647 const BYTE* const dictEnd = dictMode == ZSTD_dictMatchState ?
648 dms->window.nextSrc : NULL;
649 const U32 dictIndexDelta = dictMode == ZSTD_dictMatchState ?
650 prefixLowestIndex - (U32)(dictEnd - dictBase) :
651 0;
652 const U32 dictAndPrefixLength = (U32)(ip - prefixLowest + dictEnd - dictLowest);
653
William Kurkianea869482019-04-09 15:16:11 -0400654 /* init */
Abhilash S.L3b494632019-07-16 15:51:09 +0530655 ip += (dictAndPrefixLength == 0);
William Kurkianea869482019-04-09 15:16:11 -0400656 ms->nextToUpdate3 = ms->nextToUpdate;
Abhilash S.L3b494632019-07-16 15:51:09 +0530657 if (dictMode == ZSTD_noDict) {
658 U32 const maxRep = (U32)(ip - prefixLowest);
William Kurkianea869482019-04-09 15:16:11 -0400659 if (offset_2 > maxRep) savedOffset = offset_2, offset_2 = 0;
660 if (offset_1 > maxRep) savedOffset = offset_1, offset_1 = 0;
661 }
Abhilash S.L3b494632019-07-16 15:51:09 +0530662 if (dictMode == ZSTD_dictMatchState) {
663 /* dictMatchState repCode checks don't currently handle repCode == 0
664 * disabling. */
665 assert(offset_1 <= dictAndPrefixLength);
666 assert(offset_2 <= dictAndPrefixLength);
667 }
William Kurkianea869482019-04-09 15:16:11 -0400668
669 /* Match Loop */
670 while (ip < ilimit) {
671 size_t matchLength=0;
672 size_t offset=0;
673 const BYTE* start=ip+1;
674
675 /* check repCode */
Abhilash S.L3b494632019-07-16 15:51:09 +0530676 if (dictMode == ZSTD_dictMatchState) {
677 const U32 repIndex = (U32)(ip - base) + 1 - offset_1;
678 const BYTE* repMatch = (dictMode == ZSTD_dictMatchState
679 && repIndex < prefixLowestIndex) ?
680 dictBase + (repIndex - dictIndexDelta) :
681 base + repIndex;
682 if (((U32)((prefixLowestIndex-1) - repIndex) >= 3 /* intentional underflow */)
683 && (MEM_read32(repMatch) == MEM_read32(ip+1)) ) {
684 const BYTE* repMatchEnd = repIndex < prefixLowestIndex ? dictEnd : iend;
685 matchLength = ZSTD_count_2segments(ip+1+4, repMatch+4, iend, repMatchEnd, prefixLowest) + 4;
686 if (depth==0) goto _storeSequence;
687 }
688 }
689 if ( dictMode == ZSTD_noDict
690 && ((offset_1 > 0) & (MEM_read32(ip+1-offset_1) == MEM_read32(ip+1)))) {
William Kurkianea869482019-04-09 15:16:11 -0400691 matchLength = ZSTD_count(ip+1+4, ip+1+4-offset_1, iend) + 4;
692 if (depth==0) goto _storeSequence;
693 }
694
695 /* first search (depth 0) */
Abhilash S.L3b494632019-07-16 15:51:09 +0530696 { size_t offsetFound = 999999999;
697 size_t const ml2 = searchMax(ms, ip, iend, &offsetFound);
William Kurkianea869482019-04-09 15:16:11 -0400698 if (ml2 > matchLength)
699 matchLength = ml2, start = ip, offset=offsetFound;
700 }
701
702 if (matchLength < 4) {
703 ip += ((ip-anchor) >> kSearchStrength) + 1; /* jump faster over incompressible sections */
704 continue;
705 }
706
707 /* let's try to find a better solution */
708 if (depth>=1)
709 while (ip<ilimit) {
710 ip ++;
Abhilash S.L3b494632019-07-16 15:51:09 +0530711 if ( (dictMode == ZSTD_noDict)
712 && (offset) && ((offset_1>0) & (MEM_read32(ip) == MEM_read32(ip - offset_1)))) {
William Kurkianea869482019-04-09 15:16:11 -0400713 size_t const mlRep = ZSTD_count(ip+4, ip+4-offset_1, iend) + 4;
714 int const gain2 = (int)(mlRep * 3);
715 int const gain1 = (int)(matchLength*3 - ZSTD_highbit32((U32)offset+1) + 1);
716 if ((mlRep >= 4) && (gain2 > gain1))
717 matchLength = mlRep, offset = 0, start = ip;
718 }
Abhilash S.L3b494632019-07-16 15:51:09 +0530719 if (dictMode == ZSTD_dictMatchState) {
720 const U32 repIndex = (U32)(ip - base) - offset_1;
721 const BYTE* repMatch = repIndex < prefixLowestIndex ?
722 dictBase + (repIndex - dictIndexDelta) :
723 base + repIndex;
724 if (((U32)((prefixLowestIndex-1) - repIndex) >= 3 /* intentional underflow */)
725 && (MEM_read32(repMatch) == MEM_read32(ip)) ) {
726 const BYTE* repMatchEnd = repIndex < prefixLowestIndex ? dictEnd : iend;
727 size_t const mlRep = ZSTD_count_2segments(ip+4, repMatch+4, iend, repMatchEnd, prefixLowest) + 4;
728 int const gain2 = (int)(mlRep * 3);
729 int const gain1 = (int)(matchLength*3 - ZSTD_highbit32((U32)offset+1) + 1);
730 if ((mlRep >= 4) && (gain2 > gain1))
731 matchLength = mlRep, offset = 0, start = ip;
732 }
733 }
734 { size_t offset2=999999999;
735 size_t const ml2 = searchMax(ms, ip, iend, &offset2);
William Kurkianea869482019-04-09 15:16:11 -0400736 int const gain2 = (int)(ml2*4 - ZSTD_highbit32((U32)offset2+1)); /* raw approx */
737 int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offset+1) + 4);
738 if ((ml2 >= 4) && (gain2 > gain1)) {
739 matchLength = ml2, offset = offset2, start = ip;
740 continue; /* search a better one */
741 } }
742
743 /* let's find an even better one */
744 if ((depth==2) && (ip<ilimit)) {
745 ip ++;
Abhilash S.L3b494632019-07-16 15:51:09 +0530746 if ( (dictMode == ZSTD_noDict)
747 && (offset) && ((offset_1>0) & (MEM_read32(ip) == MEM_read32(ip - offset_1)))) {
748 size_t const mlRep = ZSTD_count(ip+4, ip+4-offset_1, iend) + 4;
749 int const gain2 = (int)(mlRep * 4);
William Kurkianea869482019-04-09 15:16:11 -0400750 int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offset+1) + 1);
Abhilash S.L3b494632019-07-16 15:51:09 +0530751 if ((mlRep >= 4) && (gain2 > gain1))
752 matchLength = mlRep, offset = 0, start = ip;
William Kurkianea869482019-04-09 15:16:11 -0400753 }
Abhilash S.L3b494632019-07-16 15:51:09 +0530754 if (dictMode == ZSTD_dictMatchState) {
755 const U32 repIndex = (U32)(ip - base) - offset_1;
756 const BYTE* repMatch = repIndex < prefixLowestIndex ?
757 dictBase + (repIndex - dictIndexDelta) :
758 base + repIndex;
759 if (((U32)((prefixLowestIndex-1) - repIndex) >= 3 /* intentional underflow */)
760 && (MEM_read32(repMatch) == MEM_read32(ip)) ) {
761 const BYTE* repMatchEnd = repIndex < prefixLowestIndex ? dictEnd : iend;
762 size_t const mlRep = ZSTD_count_2segments(ip+4, repMatch+4, iend, repMatchEnd, prefixLowest) + 4;
763 int const gain2 = (int)(mlRep * 4);
764 int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offset+1) + 1);
765 if ((mlRep >= 4) && (gain2 > gain1))
766 matchLength = mlRep, offset = 0, start = ip;
767 }
768 }
769 { size_t offset2=999999999;
770 size_t const ml2 = searchMax(ms, ip, iend, &offset2);
William Kurkianea869482019-04-09 15:16:11 -0400771 int const gain2 = (int)(ml2*4 - ZSTD_highbit32((U32)offset2+1)); /* raw approx */
772 int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offset+1) + 7);
773 if ((ml2 >= 4) && (gain2 > gain1)) {
774 matchLength = ml2, offset = offset2, start = ip;
775 continue;
776 } } }
777 break; /* nothing found : store previous solution */
778 }
779
780 /* NOTE:
781 * start[-offset+ZSTD_REP_MOVE-1] is undefined behavior.
782 * (-offset+ZSTD_REP_MOVE-1) is unsigned, and is added to start, which
783 * overflows the pointer, which is undefined behavior.
784 */
785 /* catch up */
786 if (offset) {
Abhilash S.L3b494632019-07-16 15:51:09 +0530787 if (dictMode == ZSTD_noDict) {
788 while ( ((start > anchor) & (start - (offset-ZSTD_REP_MOVE) > prefixLowest))
789 && (start[-1] == (start-(offset-ZSTD_REP_MOVE))[-1]) ) /* only search for offset within prefix */
790 { start--; matchLength++; }
791 }
792 if (dictMode == ZSTD_dictMatchState) {
793 U32 const matchIndex = (U32)((start-base) - (offset - ZSTD_REP_MOVE));
794 const BYTE* match = (matchIndex < prefixLowestIndex) ? dictBase + matchIndex - dictIndexDelta : base + matchIndex;
795 const BYTE* const mStart = (matchIndex < prefixLowestIndex) ? dictLowest : prefixLowest;
796 while ((start>anchor) && (match>mStart) && (start[-1] == match[-1])) { start--; match--; matchLength++; } /* catch up */
797 }
William Kurkianea869482019-04-09 15:16:11 -0400798 offset_2 = offset_1; offset_1 = (U32)(offset - ZSTD_REP_MOVE);
799 }
800 /* store sequence */
801_storeSequence:
802 { size_t const litLength = start - anchor;
803 ZSTD_storeSeq(seqStore, litLength, anchor, (U32)offset, matchLength-MINMATCH);
804 anchor = ip = start + matchLength;
805 }
806
807 /* check immediate repcode */
Abhilash S.L3b494632019-07-16 15:51:09 +0530808 if (dictMode == ZSTD_dictMatchState) {
809 while (ip <= ilimit) {
810 U32 const current2 = (U32)(ip-base);
811 U32 const repIndex = current2 - offset_2;
812 const BYTE* repMatch = dictMode == ZSTD_dictMatchState
813 && repIndex < prefixLowestIndex ?
814 dictBase - dictIndexDelta + repIndex :
815 base + repIndex;
816 if ( ((U32)((prefixLowestIndex-1) - (U32)repIndex) >= 3 /* intentional overflow */)
817 && (MEM_read32(repMatch) == MEM_read32(ip)) ) {
818 const BYTE* const repEnd2 = repIndex < prefixLowestIndex ? dictEnd : iend;
819 matchLength = ZSTD_count_2segments(ip+4, repMatch+4, iend, repEnd2, prefixLowest) + 4;
820 offset = offset_2; offset_2 = offset_1; offset_1 = (U32)offset; /* swap offset_2 <=> offset_1 */
821 ZSTD_storeSeq(seqStore, 0, anchor, 0, matchLength-MINMATCH);
822 ip += matchLength;
823 anchor = ip;
824 continue;
825 }
826 break;
827 }
828 }
829
830 if (dictMode == ZSTD_noDict) {
831 while ( ((ip <= ilimit) & (offset_2>0))
832 && (MEM_read32(ip) == MEM_read32(ip - offset_2)) ) {
833 /* store sequence */
834 matchLength = ZSTD_count(ip+4, ip+4-offset_2, iend) + 4;
835 offset = offset_2; offset_2 = offset_1; offset_1 = (U32)offset; /* swap repcodes */
836 ZSTD_storeSeq(seqStore, 0, anchor, 0, matchLength-MINMATCH);
837 ip += matchLength;
838 anchor = ip;
839 continue; /* faster when present ... (?) */
840 } } }
William Kurkianea869482019-04-09 15:16:11 -0400841
842 /* Save reps for next block */
843 rep[0] = offset_1 ? offset_1 : savedOffset;
844 rep[1] = offset_2 ? offset_2 : savedOffset;
845
846 /* Return the last literals size */
847 return iend - anchor;
848}
849
850
851size_t ZSTD_compressBlock_btlazy2(
852 ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
Abhilash S.L3b494632019-07-16 15:51:09 +0530853 void const* src, size_t srcSize)
William Kurkianea869482019-04-09 15:16:11 -0400854{
Abhilash S.L3b494632019-07-16 15:51:09 +0530855 return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, 1, 2, ZSTD_noDict);
William Kurkianea869482019-04-09 15:16:11 -0400856}
857
858size_t ZSTD_compressBlock_lazy2(
859 ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
Abhilash S.L3b494632019-07-16 15:51:09 +0530860 void const* src, size_t srcSize)
William Kurkianea869482019-04-09 15:16:11 -0400861{
Abhilash S.L3b494632019-07-16 15:51:09 +0530862 return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, 0, 2, ZSTD_noDict);
William Kurkianea869482019-04-09 15:16:11 -0400863}
864
865size_t ZSTD_compressBlock_lazy(
866 ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
Abhilash S.L3b494632019-07-16 15:51:09 +0530867 void const* src, size_t srcSize)
William Kurkianea869482019-04-09 15:16:11 -0400868{
Abhilash S.L3b494632019-07-16 15:51:09 +0530869 return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, 0, 1, ZSTD_noDict);
William Kurkianea869482019-04-09 15:16:11 -0400870}
871
872size_t ZSTD_compressBlock_greedy(
873 ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
Abhilash S.L3b494632019-07-16 15:51:09 +0530874 void const* src, size_t srcSize)
William Kurkianea869482019-04-09 15:16:11 -0400875{
Abhilash S.L3b494632019-07-16 15:51:09 +0530876 return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, 0, 0, ZSTD_noDict);
877}
878
879size_t ZSTD_compressBlock_btlazy2_dictMatchState(
880 ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
881 void const* src, size_t srcSize)
882{
883 return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, 1, 2, ZSTD_dictMatchState);
884}
885
886size_t ZSTD_compressBlock_lazy2_dictMatchState(
887 ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
888 void const* src, size_t srcSize)
889{
890 return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, 0, 2, ZSTD_dictMatchState);
891}
892
893size_t ZSTD_compressBlock_lazy_dictMatchState(
894 ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
895 void const* src, size_t srcSize)
896{
897 return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, 0, 1, ZSTD_dictMatchState);
898}
899
900size_t ZSTD_compressBlock_greedy_dictMatchState(
901 ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
902 void const* src, size_t srcSize)
903{
904 return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, 0, 0, ZSTD_dictMatchState);
William Kurkianea869482019-04-09 15:16:11 -0400905}
906
907
908FORCE_INLINE_TEMPLATE
909size_t ZSTD_compressBlock_lazy_extDict_generic(
910 ZSTD_matchState_t* ms, seqStore_t* seqStore,
911 U32 rep[ZSTD_REP_NUM],
William Kurkianea869482019-04-09 15:16:11 -0400912 const void* src, size_t srcSize,
913 const U32 searchMethod, const U32 depth)
914{
915 const BYTE* const istart = (const BYTE*)src;
916 const BYTE* ip = istart;
917 const BYTE* anchor = istart;
918 const BYTE* const iend = istart + srcSize;
919 const BYTE* const ilimit = iend - 8;
920 const BYTE* const base = ms->window.base;
921 const U32 dictLimit = ms->window.dictLimit;
922 const U32 lowestIndex = ms->window.lowLimit;
923 const BYTE* const prefixStart = base + dictLimit;
924 const BYTE* const dictBase = ms->window.dictBase;
925 const BYTE* const dictEnd = dictBase + dictLimit;
926 const BYTE* const dictStart = dictBase + lowestIndex;
927
928 typedef size_t (*searchMax_f)(
Abhilash S.L3b494632019-07-16 15:51:09 +0530929 ZSTD_matchState_t* ms,
William Kurkianea869482019-04-09 15:16:11 -0400930 const BYTE* ip, const BYTE* iLimit, size_t* offsetPtr);
Abhilash S.L3b494632019-07-16 15:51:09 +0530931 searchMax_f searchMax = searchMethod ? ZSTD_BtFindBestMatch_extDict_selectMLS : ZSTD_HcFindBestMatch_extDict_selectMLS;
William Kurkianea869482019-04-09 15:16:11 -0400932
933 U32 offset_1 = rep[0], offset_2 = rep[1];
934
935 /* init */
936 ms->nextToUpdate3 = ms->nextToUpdate;
937 ip += (ip == prefixStart);
938
939 /* Match Loop */
940 while (ip < ilimit) {
941 size_t matchLength=0;
942 size_t offset=0;
943 const BYTE* start=ip+1;
944 U32 current = (U32)(ip-base);
945
946 /* check repCode */
947 { const U32 repIndex = (U32)(current+1 - offset_1);
948 const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
949 const BYTE* const repMatch = repBase + repIndex;
950 if (((U32)((dictLimit-1) - repIndex) >= 3) & (repIndex > lowestIndex)) /* intentional overflow */
951 if (MEM_read32(ip+1) == MEM_read32(repMatch)) {
952 /* repcode detected we should take it */
953 const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
954 matchLength = ZSTD_count_2segments(ip+1+4, repMatch+4, iend, repEnd, prefixStart) + 4;
955 if (depth==0) goto _storeSequence;
956 } }
957
958 /* first search (depth 0) */
Abhilash S.L3b494632019-07-16 15:51:09 +0530959 { size_t offsetFound = 999999999;
960 size_t const ml2 = searchMax(ms, ip, iend, &offsetFound);
William Kurkianea869482019-04-09 15:16:11 -0400961 if (ml2 > matchLength)
962 matchLength = ml2, start = ip, offset=offsetFound;
963 }
964
965 if (matchLength < 4) {
966 ip += ((ip-anchor) >> kSearchStrength) + 1; /* jump faster over incompressible sections */
967 continue;
968 }
969
970 /* let's try to find a better solution */
971 if (depth>=1)
972 while (ip<ilimit) {
973 ip ++;
974 current++;
975 /* check repCode */
976 if (offset) {
977 const U32 repIndex = (U32)(current - offset_1);
978 const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
979 const BYTE* const repMatch = repBase + repIndex;
980 if (((U32)((dictLimit-1) - repIndex) >= 3) & (repIndex > lowestIndex)) /* intentional overflow */
981 if (MEM_read32(ip) == MEM_read32(repMatch)) {
982 /* repcode detected */
983 const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
984 size_t const repLength = ZSTD_count_2segments(ip+4, repMatch+4, iend, repEnd, prefixStart) + 4;
985 int const gain2 = (int)(repLength * 3);
986 int const gain1 = (int)(matchLength*3 - ZSTD_highbit32((U32)offset+1) + 1);
987 if ((repLength >= 4) && (gain2 > gain1))
988 matchLength = repLength, offset = 0, start = ip;
989 } }
990
991 /* search match, depth 1 */
Abhilash S.L3b494632019-07-16 15:51:09 +0530992 { size_t offset2=999999999;
993 size_t const ml2 = searchMax(ms, ip, iend, &offset2);
William Kurkianea869482019-04-09 15:16:11 -0400994 int const gain2 = (int)(ml2*4 - ZSTD_highbit32((U32)offset2+1)); /* raw approx */
995 int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offset+1) + 4);
996 if ((ml2 >= 4) && (gain2 > gain1)) {
997 matchLength = ml2, offset = offset2, start = ip;
998 continue; /* search a better one */
999 } }
1000
1001 /* let's find an even better one */
1002 if ((depth==2) && (ip<ilimit)) {
1003 ip ++;
1004 current++;
1005 /* check repCode */
1006 if (offset) {
1007 const U32 repIndex = (U32)(current - offset_1);
1008 const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
1009 const BYTE* const repMatch = repBase + repIndex;
1010 if (((U32)((dictLimit-1) - repIndex) >= 3) & (repIndex > lowestIndex)) /* intentional overflow */
1011 if (MEM_read32(ip) == MEM_read32(repMatch)) {
1012 /* repcode detected */
1013 const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
1014 size_t const repLength = ZSTD_count_2segments(ip+4, repMatch+4, iend, repEnd, prefixStart) + 4;
1015 int const gain2 = (int)(repLength * 4);
1016 int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offset+1) + 1);
1017 if ((repLength >= 4) && (gain2 > gain1))
1018 matchLength = repLength, offset = 0, start = ip;
1019 } }
1020
1021 /* search match, depth 2 */
Abhilash S.L3b494632019-07-16 15:51:09 +05301022 { size_t offset2=999999999;
1023 size_t const ml2 = searchMax(ms, ip, iend, &offset2);
William Kurkianea869482019-04-09 15:16:11 -04001024 int const gain2 = (int)(ml2*4 - ZSTD_highbit32((U32)offset2+1)); /* raw approx */
1025 int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offset+1) + 7);
1026 if ((ml2 >= 4) && (gain2 > gain1)) {
1027 matchLength = ml2, offset = offset2, start = ip;
1028 continue;
1029 } } }
1030 break; /* nothing found : store previous solution */
1031 }
1032
1033 /* catch up */
1034 if (offset) {
1035 U32 const matchIndex = (U32)((start-base) - (offset - ZSTD_REP_MOVE));
1036 const BYTE* match = (matchIndex < dictLimit) ? dictBase + matchIndex : base + matchIndex;
1037 const BYTE* const mStart = (matchIndex < dictLimit) ? dictStart : prefixStart;
1038 while ((start>anchor) && (match>mStart) && (start[-1] == match[-1])) { start--; match--; matchLength++; } /* catch up */
1039 offset_2 = offset_1; offset_1 = (U32)(offset - ZSTD_REP_MOVE);
1040 }
1041
1042 /* store sequence */
1043_storeSequence:
1044 { size_t const litLength = start - anchor;
1045 ZSTD_storeSeq(seqStore, litLength, anchor, (U32)offset, matchLength-MINMATCH);
1046 anchor = ip = start + matchLength;
1047 }
1048
1049 /* check immediate repcode */
1050 while (ip <= ilimit) {
1051 const U32 repIndex = (U32)((ip-base) - offset_2);
1052 const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
1053 const BYTE* const repMatch = repBase + repIndex;
1054 if (((U32)((dictLimit-1) - repIndex) >= 3) & (repIndex > lowestIndex)) /* intentional overflow */
1055 if (MEM_read32(ip) == MEM_read32(repMatch)) {
1056 /* repcode detected we should take it */
1057 const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
1058 matchLength = ZSTD_count_2segments(ip+4, repMatch+4, iend, repEnd, prefixStart) + 4;
1059 offset = offset_2; offset_2 = offset_1; offset_1 = (U32)offset; /* swap offset history */
1060 ZSTD_storeSeq(seqStore, 0, anchor, 0, matchLength-MINMATCH);
1061 ip += matchLength;
1062 anchor = ip;
1063 continue; /* faster when present ... (?) */
1064 }
1065 break;
1066 } }
1067
1068 /* Save reps for next block */
1069 rep[0] = offset_1;
1070 rep[1] = offset_2;
1071
1072 /* Return the last literals size */
1073 return iend - anchor;
1074}
1075
1076
1077size_t ZSTD_compressBlock_greedy_extDict(
1078 ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
Abhilash S.L3b494632019-07-16 15:51:09 +05301079 void const* src, size_t srcSize)
William Kurkianea869482019-04-09 15:16:11 -04001080{
Abhilash S.L3b494632019-07-16 15:51:09 +05301081 return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, 0, 0);
William Kurkianea869482019-04-09 15:16:11 -04001082}
1083
1084size_t ZSTD_compressBlock_lazy_extDict(
1085 ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
Abhilash S.L3b494632019-07-16 15:51:09 +05301086 void const* src, size_t srcSize)
William Kurkianea869482019-04-09 15:16:11 -04001087
1088{
Abhilash S.L3b494632019-07-16 15:51:09 +05301089 return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, 0, 1);
William Kurkianea869482019-04-09 15:16:11 -04001090}
1091
1092size_t ZSTD_compressBlock_lazy2_extDict(
1093 ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
Abhilash S.L3b494632019-07-16 15:51:09 +05301094 void const* src, size_t srcSize)
William Kurkianea869482019-04-09 15:16:11 -04001095
1096{
Abhilash S.L3b494632019-07-16 15:51:09 +05301097 return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, 0, 2);
William Kurkianea869482019-04-09 15:16:11 -04001098}
1099
1100size_t ZSTD_compressBlock_btlazy2_extDict(
1101 ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
Abhilash S.L3b494632019-07-16 15:51:09 +05301102 void const* src, size_t srcSize)
William Kurkianea869482019-04-09 15:16:11 -04001103
1104{
Abhilash S.L3b494632019-07-16 15:51:09 +05301105 return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, 1, 2);
William Kurkianea869482019-04-09 15:16:11 -04001106}