1a28cd43dSSascha Wildner /*
2a28cd43dSSascha Wildner * Copyright (c) 2016-2020, Yann Collet, Facebook, Inc.
3a28cd43dSSascha Wildner * All rights reserved.
4a28cd43dSSascha Wildner *
5a28cd43dSSascha Wildner * This source code is licensed under both the BSD-style license (found in the
6a28cd43dSSascha Wildner * LICENSE file in the root directory of this source tree) and the GPLv2 (found
7a28cd43dSSascha Wildner * in the COPYING file in the root directory of this source tree).
8a28cd43dSSascha Wildner * You may select, at your option, one of the above-listed licenses.
9a28cd43dSSascha Wildner */
10a28cd43dSSascha Wildner
11a28cd43dSSascha Wildner #include "zstd_compress_internal.h"
12a28cd43dSSascha Wildner #include "zstd_lazy.h"
13a28cd43dSSascha Wildner
14a28cd43dSSascha Wildner
15a28cd43dSSascha Wildner /*-*************************************
16a28cd43dSSascha Wildner * Binary Tree search
17a28cd43dSSascha Wildner ***************************************/
18a28cd43dSSascha Wildner
19a28cd43dSSascha Wildner static void
ZSTD_updateDUBT(ZSTD_matchState_t * ms,const BYTE * ip,const BYTE * iend,U32 mls)20a28cd43dSSascha Wildner ZSTD_updateDUBT(ZSTD_matchState_t* ms,
21a28cd43dSSascha Wildner const BYTE* ip, const BYTE* iend,
22a28cd43dSSascha Wildner U32 mls)
23a28cd43dSSascha Wildner {
24a28cd43dSSascha Wildner const ZSTD_compressionParameters* const cParams = &ms->cParams;
25a28cd43dSSascha Wildner U32* const hashTable = ms->hashTable;
26a28cd43dSSascha Wildner U32 const hashLog = cParams->hashLog;
27a28cd43dSSascha Wildner
28a28cd43dSSascha Wildner U32* const bt = ms->chainTable;
29a28cd43dSSascha Wildner U32 const btLog = cParams->chainLog - 1;
30a28cd43dSSascha Wildner U32 const btMask = (1 << btLog) - 1;
31a28cd43dSSascha Wildner
32a28cd43dSSascha Wildner const BYTE* const base = ms->window.base;
33a28cd43dSSascha Wildner U32 const target = (U32)(ip - base);
34a28cd43dSSascha Wildner U32 idx = ms->nextToUpdate;
35a28cd43dSSascha Wildner
36a28cd43dSSascha Wildner if (idx != target)
37a28cd43dSSascha Wildner DEBUGLOG(7, "ZSTD_updateDUBT, from %u to %u (dictLimit:%u)",
38a28cd43dSSascha Wildner idx, target, ms->window.dictLimit);
39a28cd43dSSascha Wildner assert(ip + 8 <= iend); /* condition for ZSTD_hashPtr */
40a28cd43dSSascha Wildner (void)iend;
41a28cd43dSSascha Wildner
42a28cd43dSSascha Wildner assert(idx >= ms->window.dictLimit); /* condition for valid base+idx */
43a28cd43dSSascha Wildner for ( ; idx < target ; idx++) {
44a28cd43dSSascha Wildner size_t const h = ZSTD_hashPtr(base + idx, hashLog, mls); /* assumption : ip + 8 <= iend */
45a28cd43dSSascha Wildner U32 const matchIndex = hashTable[h];
46a28cd43dSSascha Wildner
47a28cd43dSSascha Wildner U32* const nextCandidatePtr = bt + 2*(idx&btMask);
48a28cd43dSSascha Wildner U32* const sortMarkPtr = nextCandidatePtr + 1;
49a28cd43dSSascha Wildner
50a28cd43dSSascha Wildner DEBUGLOG(8, "ZSTD_updateDUBT: insert %u", idx);
51a28cd43dSSascha Wildner hashTable[h] = idx; /* Update Hash Table */
52a28cd43dSSascha Wildner *nextCandidatePtr = matchIndex; /* update BT like a chain */
53a28cd43dSSascha Wildner *sortMarkPtr = ZSTD_DUBT_UNSORTED_MARK;
54a28cd43dSSascha Wildner }
55a28cd43dSSascha Wildner ms->nextToUpdate = target;
56a28cd43dSSascha Wildner }
57a28cd43dSSascha Wildner
58a28cd43dSSascha Wildner
59a28cd43dSSascha Wildner /** ZSTD_insertDUBT1() :
60a28cd43dSSascha Wildner * sort one already inserted but unsorted position
61a28cd43dSSascha Wildner * assumption : curr >= btlow == (curr - btmask)
62a28cd43dSSascha Wildner * doesn't fail */
63a28cd43dSSascha Wildner static void
ZSTD_insertDUBT1(ZSTD_matchState_t * ms,U32 curr,const BYTE * inputEnd,U32 nbCompares,U32 btLow,const ZSTD_dictMode_e dictMode)64a28cd43dSSascha Wildner ZSTD_insertDUBT1(ZSTD_matchState_t* ms,
65a28cd43dSSascha Wildner U32 curr, const BYTE* inputEnd,
66a28cd43dSSascha Wildner U32 nbCompares, U32 btLow,
67a28cd43dSSascha Wildner const ZSTD_dictMode_e dictMode)
68a28cd43dSSascha Wildner {
69a28cd43dSSascha Wildner const ZSTD_compressionParameters* const cParams = &ms->cParams;
70a28cd43dSSascha Wildner U32* const bt = ms->chainTable;
71a28cd43dSSascha Wildner U32 const btLog = cParams->chainLog - 1;
72a28cd43dSSascha Wildner U32 const btMask = (1 << btLog) - 1;
73a28cd43dSSascha Wildner size_t commonLengthSmaller=0, commonLengthLarger=0;
74a28cd43dSSascha Wildner const BYTE* const base = ms->window.base;
75a28cd43dSSascha Wildner const BYTE* const dictBase = ms->window.dictBase;
76a28cd43dSSascha Wildner const U32 dictLimit = ms->window.dictLimit;
77a28cd43dSSascha Wildner const BYTE* const ip = (curr>=dictLimit) ? base + curr : dictBase + curr;
78a28cd43dSSascha Wildner const BYTE* const iend = (curr>=dictLimit) ? inputEnd : dictBase + dictLimit;
79a28cd43dSSascha Wildner const BYTE* const dictEnd = dictBase + dictLimit;
80a28cd43dSSascha Wildner const BYTE* const prefixStart = base + dictLimit;
81a28cd43dSSascha Wildner const BYTE* match;
82a28cd43dSSascha Wildner U32* smallerPtr = bt + 2*(curr&btMask);
83a28cd43dSSascha Wildner U32* largerPtr = smallerPtr + 1;
84a28cd43dSSascha Wildner 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) */
85a28cd43dSSascha Wildner U32 dummy32; /* to be nullified at the end */
86a28cd43dSSascha Wildner U32 const windowValid = ms->window.lowLimit;
87a28cd43dSSascha Wildner U32 const maxDistance = 1U << cParams->windowLog;
88a28cd43dSSascha Wildner U32 const windowLow = (curr - windowValid > maxDistance) ? curr - maxDistance : windowValid;
89a28cd43dSSascha Wildner
90a28cd43dSSascha Wildner
91a28cd43dSSascha Wildner DEBUGLOG(8, "ZSTD_insertDUBT1(%u) (dictLimit=%u, lowLimit=%u)",
92a28cd43dSSascha Wildner curr, dictLimit, windowLow);
93a28cd43dSSascha Wildner assert(curr >= btLow);
94a28cd43dSSascha Wildner assert(ip < iend); /* condition for ZSTD_count */
95a28cd43dSSascha Wildner
96a28cd43dSSascha Wildner while (nbCompares-- && (matchIndex > windowLow)) {
97a28cd43dSSascha Wildner U32* const nextPtr = bt + 2*(matchIndex & btMask);
98a28cd43dSSascha Wildner size_t matchLength = MIN(commonLengthSmaller, commonLengthLarger); /* guaranteed minimum nb of common bytes */
99a28cd43dSSascha Wildner assert(matchIndex < curr);
100a28cd43dSSascha Wildner /* note : all candidates are now supposed sorted,
101a28cd43dSSascha Wildner * but it's still possible to have nextPtr[1] == ZSTD_DUBT_UNSORTED_MARK
102a28cd43dSSascha Wildner * when a real index has the same value as ZSTD_DUBT_UNSORTED_MARK */
103a28cd43dSSascha Wildner
104a28cd43dSSascha Wildner if ( (dictMode != ZSTD_extDict)
105a28cd43dSSascha Wildner || (matchIndex+matchLength >= dictLimit) /* both in current segment*/
106a28cd43dSSascha Wildner || (curr < dictLimit) /* both in extDict */) {
107a28cd43dSSascha Wildner const BYTE* const mBase = ( (dictMode != ZSTD_extDict)
108a28cd43dSSascha Wildner || (matchIndex+matchLength >= dictLimit)) ?
109a28cd43dSSascha Wildner base : dictBase;
110a28cd43dSSascha Wildner assert( (matchIndex+matchLength >= dictLimit) /* might be wrong if extDict is incorrectly set to 0 */
111a28cd43dSSascha Wildner || (curr < dictLimit) );
112a28cd43dSSascha Wildner match = mBase + matchIndex;
113a28cd43dSSascha Wildner matchLength += ZSTD_count(ip+matchLength, match+matchLength, iend);
114a28cd43dSSascha Wildner } else {
115a28cd43dSSascha Wildner match = dictBase + matchIndex;
116a28cd43dSSascha Wildner matchLength += ZSTD_count_2segments(ip+matchLength, match+matchLength, iend, dictEnd, prefixStart);
117a28cd43dSSascha Wildner if (matchIndex+matchLength >= dictLimit)
118a28cd43dSSascha Wildner match = base + matchIndex; /* preparation for next read of match[matchLength] */
119a28cd43dSSascha Wildner }
120a28cd43dSSascha Wildner
121a28cd43dSSascha Wildner DEBUGLOG(8, "ZSTD_insertDUBT1: comparing %u with %u : found %u common bytes ",
122a28cd43dSSascha Wildner curr, matchIndex, (U32)matchLength);
123a28cd43dSSascha Wildner
124a28cd43dSSascha Wildner if (ip+matchLength == iend) { /* equal : no way to know if inf or sup */
125a28cd43dSSascha Wildner break; /* drop , to guarantee consistency ; miss a bit of compression, but other solutions can corrupt tree */
126a28cd43dSSascha Wildner }
127a28cd43dSSascha Wildner
128a28cd43dSSascha Wildner if (match[matchLength] < ip[matchLength]) { /* necessarily within buffer */
129a28cd43dSSascha Wildner /* match is smaller than current */
130a28cd43dSSascha Wildner *smallerPtr = matchIndex; /* update smaller idx */
131a28cd43dSSascha Wildner commonLengthSmaller = matchLength; /* all smaller will now have at least this guaranteed common length */
132a28cd43dSSascha Wildner if (matchIndex <= btLow) { smallerPtr=&dummy32; break; } /* beyond tree size, stop searching */
133a28cd43dSSascha Wildner DEBUGLOG(8, "ZSTD_insertDUBT1: %u (>btLow=%u) is smaller : next => %u",
134a28cd43dSSascha Wildner matchIndex, btLow, nextPtr[1]);
135a28cd43dSSascha Wildner smallerPtr = nextPtr+1; /* new "candidate" => larger than match, which was smaller than target */
136a28cd43dSSascha Wildner matchIndex = nextPtr[1]; /* new matchIndex, larger than previous and closer to current */
137a28cd43dSSascha Wildner } else {
138a28cd43dSSascha Wildner /* match is larger than current */
139a28cd43dSSascha Wildner *largerPtr = matchIndex;
140a28cd43dSSascha Wildner commonLengthLarger = matchLength;
141a28cd43dSSascha Wildner if (matchIndex <= btLow) { largerPtr=&dummy32; break; } /* beyond tree size, stop searching */
142a28cd43dSSascha Wildner DEBUGLOG(8, "ZSTD_insertDUBT1: %u (>btLow=%u) is larger => %u",
143a28cd43dSSascha Wildner matchIndex, btLow, nextPtr[0]);
144a28cd43dSSascha Wildner largerPtr = nextPtr;
145a28cd43dSSascha Wildner matchIndex = nextPtr[0];
146a28cd43dSSascha Wildner } }
147a28cd43dSSascha Wildner
148a28cd43dSSascha Wildner *smallerPtr = *largerPtr = 0;
149a28cd43dSSascha Wildner }
150a28cd43dSSascha Wildner
151a28cd43dSSascha Wildner
152a28cd43dSSascha Wildner static size_t
ZSTD_DUBT_findBetterDictMatch(ZSTD_matchState_t * ms,const BYTE * const ip,const BYTE * const iend,size_t * offsetPtr,size_t bestLength,U32 nbCompares,U32 const mls,const ZSTD_dictMode_e dictMode)153a28cd43dSSascha Wildner ZSTD_DUBT_findBetterDictMatch (
154a28cd43dSSascha Wildner ZSTD_matchState_t* ms,
155a28cd43dSSascha Wildner const BYTE* const ip, const BYTE* const iend,
156a28cd43dSSascha Wildner size_t* offsetPtr,
157a28cd43dSSascha Wildner size_t bestLength,
158a28cd43dSSascha Wildner U32 nbCompares,
159a28cd43dSSascha Wildner U32 const mls,
160a28cd43dSSascha Wildner const ZSTD_dictMode_e dictMode)
161a28cd43dSSascha Wildner {
162a28cd43dSSascha Wildner const ZSTD_matchState_t * const dms = ms->dictMatchState;
163a28cd43dSSascha Wildner const ZSTD_compressionParameters* const dmsCParams = &dms->cParams;
164a28cd43dSSascha Wildner const U32 * const dictHashTable = dms->hashTable;
165a28cd43dSSascha Wildner U32 const hashLog = dmsCParams->hashLog;
166a28cd43dSSascha Wildner size_t const h = ZSTD_hashPtr(ip, hashLog, mls);
167a28cd43dSSascha Wildner U32 dictMatchIndex = dictHashTable[h];
168a28cd43dSSascha Wildner
169a28cd43dSSascha Wildner const BYTE* const base = ms->window.base;
170a28cd43dSSascha Wildner const BYTE* const prefixStart = base + ms->window.dictLimit;
171a28cd43dSSascha Wildner U32 const curr = (U32)(ip-base);
172a28cd43dSSascha Wildner const BYTE* const dictBase = dms->window.base;
173a28cd43dSSascha Wildner const BYTE* const dictEnd = dms->window.nextSrc;
174a28cd43dSSascha Wildner U32 const dictHighLimit = (U32)(dms->window.nextSrc - dms->window.base);
175a28cd43dSSascha Wildner U32 const dictLowLimit = dms->window.lowLimit;
176a28cd43dSSascha Wildner U32 const dictIndexDelta = ms->window.lowLimit - dictHighLimit;
177a28cd43dSSascha Wildner
178a28cd43dSSascha Wildner U32* const dictBt = dms->chainTable;
179a28cd43dSSascha Wildner U32 const btLog = dmsCParams->chainLog - 1;
180a28cd43dSSascha Wildner U32 const btMask = (1 << btLog) - 1;
181a28cd43dSSascha Wildner U32 const btLow = (btMask >= dictHighLimit - dictLowLimit) ? dictLowLimit : dictHighLimit - btMask;
182a28cd43dSSascha Wildner
183a28cd43dSSascha Wildner size_t commonLengthSmaller=0, commonLengthLarger=0;
184a28cd43dSSascha Wildner
185a28cd43dSSascha Wildner (void)dictMode;
186a28cd43dSSascha Wildner assert(dictMode == ZSTD_dictMatchState);
187a28cd43dSSascha Wildner
188a28cd43dSSascha Wildner while (nbCompares-- && (dictMatchIndex > dictLowLimit)) {
189a28cd43dSSascha Wildner U32* const nextPtr = dictBt + 2*(dictMatchIndex & btMask);
190a28cd43dSSascha Wildner size_t matchLength = MIN(commonLengthSmaller, commonLengthLarger); /* guaranteed minimum nb of common bytes */
191a28cd43dSSascha Wildner const BYTE* match = dictBase + dictMatchIndex;
192a28cd43dSSascha Wildner matchLength += ZSTD_count_2segments(ip+matchLength, match+matchLength, iend, dictEnd, prefixStart);
193a28cd43dSSascha Wildner if (dictMatchIndex+matchLength >= dictHighLimit)
194a28cd43dSSascha Wildner match = base + dictMatchIndex + dictIndexDelta; /* to prepare for next usage of match[matchLength] */
195a28cd43dSSascha Wildner
196a28cd43dSSascha Wildner if (matchLength > bestLength) {
197a28cd43dSSascha Wildner U32 matchIndex = dictMatchIndex + dictIndexDelta;
198a28cd43dSSascha Wildner if ( (4*(int)(matchLength-bestLength)) > (int)(ZSTD_highbit32(curr-matchIndex+1) - ZSTD_highbit32((U32)offsetPtr[0]+1)) ) {
199a28cd43dSSascha Wildner DEBUGLOG(9, "ZSTD_DUBT_findBetterDictMatch(%u) : found better match length %u -> %u and offsetCode %u -> %u (dictMatchIndex %u, matchIndex %u)",
200a28cd43dSSascha Wildner curr, (U32)bestLength, (U32)matchLength, (U32)*offsetPtr, ZSTD_REP_MOVE + curr - matchIndex, dictMatchIndex, matchIndex);
201a28cd43dSSascha Wildner bestLength = matchLength, *offsetPtr = ZSTD_REP_MOVE + curr - matchIndex;
202a28cd43dSSascha Wildner }
203a28cd43dSSascha Wildner 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 */
204a28cd43dSSascha Wildner break; /* drop, to guarantee consistency (miss a little bit of compression) */
205a28cd43dSSascha Wildner }
206a28cd43dSSascha Wildner }
207a28cd43dSSascha Wildner
208a28cd43dSSascha Wildner if (match[matchLength] < ip[matchLength]) {
209a28cd43dSSascha Wildner if (dictMatchIndex <= btLow) { break; } /* beyond tree size, stop the search */
210a28cd43dSSascha Wildner commonLengthSmaller = matchLength; /* all smaller will now have at least this guaranteed common length */
211a28cd43dSSascha Wildner dictMatchIndex = nextPtr[1]; /* new matchIndex larger than previous (closer to current) */
212a28cd43dSSascha Wildner } else {
213a28cd43dSSascha Wildner /* match is larger than current */
214a28cd43dSSascha Wildner if (dictMatchIndex <= btLow) { break; } /* beyond tree size, stop the search */
215a28cd43dSSascha Wildner commonLengthLarger = matchLength;
216a28cd43dSSascha Wildner dictMatchIndex = nextPtr[0];
217a28cd43dSSascha Wildner }
218a28cd43dSSascha Wildner }
219a28cd43dSSascha Wildner
220a28cd43dSSascha Wildner if (bestLength >= MINMATCH) {
221a28cd43dSSascha Wildner U32 const mIndex = curr - ((U32)*offsetPtr - ZSTD_REP_MOVE); (void)mIndex;
222a28cd43dSSascha Wildner DEBUGLOG(8, "ZSTD_DUBT_findBetterDictMatch(%u) : found match of length %u and offsetCode %u (pos %u)",
223a28cd43dSSascha Wildner curr, (U32)bestLength, (U32)*offsetPtr, mIndex);
224a28cd43dSSascha Wildner }
225a28cd43dSSascha Wildner return bestLength;
226a28cd43dSSascha Wildner
227a28cd43dSSascha Wildner }
228a28cd43dSSascha Wildner
229a28cd43dSSascha Wildner
230a28cd43dSSascha Wildner static size_t
ZSTD_DUBT_findBestMatch(ZSTD_matchState_t * ms,const BYTE * const ip,const BYTE * const iend,size_t * offsetPtr,U32 const mls,const ZSTD_dictMode_e dictMode)231a28cd43dSSascha Wildner ZSTD_DUBT_findBestMatch(ZSTD_matchState_t* ms,
232a28cd43dSSascha Wildner const BYTE* const ip, const BYTE* const iend,
233a28cd43dSSascha Wildner size_t* offsetPtr,
234a28cd43dSSascha Wildner U32 const mls,
235a28cd43dSSascha Wildner const ZSTD_dictMode_e dictMode)
236a28cd43dSSascha Wildner {
237a28cd43dSSascha Wildner const ZSTD_compressionParameters* const cParams = &ms->cParams;
238a28cd43dSSascha Wildner U32* const hashTable = ms->hashTable;
239a28cd43dSSascha Wildner U32 const hashLog = cParams->hashLog;
240a28cd43dSSascha Wildner size_t const h = ZSTD_hashPtr(ip, hashLog, mls);
241a28cd43dSSascha Wildner U32 matchIndex = hashTable[h];
242a28cd43dSSascha Wildner
243a28cd43dSSascha Wildner const BYTE* const base = ms->window.base;
244a28cd43dSSascha Wildner U32 const curr = (U32)(ip-base);
245a28cd43dSSascha Wildner U32 const windowLow = ZSTD_getLowestMatchIndex(ms, curr, cParams->windowLog);
246a28cd43dSSascha Wildner
247a28cd43dSSascha Wildner U32* const bt = ms->chainTable;
248a28cd43dSSascha Wildner U32 const btLog = cParams->chainLog - 1;
249a28cd43dSSascha Wildner U32 const btMask = (1 << btLog) - 1;
250a28cd43dSSascha Wildner U32 const btLow = (btMask >= curr) ? 0 : curr - btMask;
251a28cd43dSSascha Wildner U32 const unsortLimit = MAX(btLow, windowLow);
252a28cd43dSSascha Wildner
253a28cd43dSSascha Wildner U32* nextCandidate = bt + 2*(matchIndex&btMask);
254a28cd43dSSascha Wildner U32* unsortedMark = bt + 2*(matchIndex&btMask) + 1;
255a28cd43dSSascha Wildner U32 nbCompares = 1U << cParams->searchLog;
256a28cd43dSSascha Wildner U32 nbCandidates = nbCompares;
257a28cd43dSSascha Wildner U32 previousCandidate = 0;
258a28cd43dSSascha Wildner
259a28cd43dSSascha Wildner DEBUGLOG(7, "ZSTD_DUBT_findBestMatch (%u) ", curr);
260a28cd43dSSascha Wildner assert(ip <= iend-8); /* required for h calculation */
261a28cd43dSSascha Wildner assert(dictMode != ZSTD_dedicatedDictSearch);
262a28cd43dSSascha Wildner
263a28cd43dSSascha Wildner /* reach end of unsorted candidates list */
264a28cd43dSSascha Wildner while ( (matchIndex > unsortLimit)
265a28cd43dSSascha Wildner && (*unsortedMark == ZSTD_DUBT_UNSORTED_MARK)
266a28cd43dSSascha Wildner && (nbCandidates > 1) ) {
267a28cd43dSSascha Wildner DEBUGLOG(8, "ZSTD_DUBT_findBestMatch: candidate %u is unsorted",
268a28cd43dSSascha Wildner matchIndex);
269a28cd43dSSascha Wildner *unsortedMark = previousCandidate; /* the unsortedMark becomes a reversed chain, to move up back to original position */
270a28cd43dSSascha Wildner previousCandidate = matchIndex;
271a28cd43dSSascha Wildner matchIndex = *nextCandidate;
272a28cd43dSSascha Wildner nextCandidate = bt + 2*(matchIndex&btMask);
273a28cd43dSSascha Wildner unsortedMark = bt + 2*(matchIndex&btMask) + 1;
274a28cd43dSSascha Wildner nbCandidates --;
275a28cd43dSSascha Wildner }
276a28cd43dSSascha Wildner
277a28cd43dSSascha Wildner /* nullify last candidate if it's still unsorted
278a28cd43dSSascha Wildner * simplification, detrimental to compression ratio, beneficial for speed */
279a28cd43dSSascha Wildner if ( (matchIndex > unsortLimit)
280a28cd43dSSascha Wildner && (*unsortedMark==ZSTD_DUBT_UNSORTED_MARK) ) {
281a28cd43dSSascha Wildner DEBUGLOG(7, "ZSTD_DUBT_findBestMatch: nullify last unsorted candidate %u",
282a28cd43dSSascha Wildner matchIndex);
283a28cd43dSSascha Wildner *nextCandidate = *unsortedMark = 0;
284a28cd43dSSascha Wildner }
285a28cd43dSSascha Wildner
286a28cd43dSSascha Wildner /* batch sort stacked candidates */
287a28cd43dSSascha Wildner matchIndex = previousCandidate;
288a28cd43dSSascha Wildner while (matchIndex) { /* will end on matchIndex == 0 */
289a28cd43dSSascha Wildner U32* const nextCandidateIdxPtr = bt + 2*(matchIndex&btMask) + 1;
290a28cd43dSSascha Wildner U32 const nextCandidateIdx = *nextCandidateIdxPtr;
291a28cd43dSSascha Wildner ZSTD_insertDUBT1(ms, matchIndex, iend,
292a28cd43dSSascha Wildner nbCandidates, unsortLimit, dictMode);
293a28cd43dSSascha Wildner matchIndex = nextCandidateIdx;
294a28cd43dSSascha Wildner nbCandidates++;
295a28cd43dSSascha Wildner }
296a28cd43dSSascha Wildner
297a28cd43dSSascha Wildner /* find longest match */
298a28cd43dSSascha Wildner { size_t commonLengthSmaller = 0, commonLengthLarger = 0;
299a28cd43dSSascha Wildner const BYTE* const dictBase = ms->window.dictBase;
300a28cd43dSSascha Wildner const U32 dictLimit = ms->window.dictLimit;
301a28cd43dSSascha Wildner const BYTE* const dictEnd = dictBase + dictLimit;
302a28cd43dSSascha Wildner const BYTE* const prefixStart = base + dictLimit;
303a28cd43dSSascha Wildner U32* smallerPtr = bt + 2*(curr&btMask);
304a28cd43dSSascha Wildner U32* largerPtr = bt + 2*(curr&btMask) + 1;
305a28cd43dSSascha Wildner U32 matchEndIdx = curr + 8 + 1;
306a28cd43dSSascha Wildner U32 dummy32; /* to be nullified at the end */
307a28cd43dSSascha Wildner size_t bestLength = 0;
308a28cd43dSSascha Wildner
309a28cd43dSSascha Wildner matchIndex = hashTable[h];
310a28cd43dSSascha Wildner hashTable[h] = curr; /* Update Hash Table */
311a28cd43dSSascha Wildner
312a28cd43dSSascha Wildner while (nbCompares-- && (matchIndex > windowLow)) {
313a28cd43dSSascha Wildner U32* const nextPtr = bt + 2*(matchIndex & btMask);
314a28cd43dSSascha Wildner size_t matchLength = MIN(commonLengthSmaller, commonLengthLarger); /* guaranteed minimum nb of common bytes */
315a28cd43dSSascha Wildner const BYTE* match;
316a28cd43dSSascha Wildner
317a28cd43dSSascha Wildner if ((dictMode != ZSTD_extDict) || (matchIndex+matchLength >= dictLimit)) {
318a28cd43dSSascha Wildner match = base + matchIndex;
319a28cd43dSSascha Wildner matchLength += ZSTD_count(ip+matchLength, match+matchLength, iend);
320a28cd43dSSascha Wildner } else {
321a28cd43dSSascha Wildner match = dictBase + matchIndex;
322a28cd43dSSascha Wildner matchLength += ZSTD_count_2segments(ip+matchLength, match+matchLength, iend, dictEnd, prefixStart);
323a28cd43dSSascha Wildner if (matchIndex+matchLength >= dictLimit)
324a28cd43dSSascha Wildner match = base + matchIndex; /* to prepare for next usage of match[matchLength] */
325a28cd43dSSascha Wildner }
326a28cd43dSSascha Wildner
327a28cd43dSSascha Wildner if (matchLength > bestLength) {
328a28cd43dSSascha Wildner if (matchLength > matchEndIdx - matchIndex)
329a28cd43dSSascha Wildner matchEndIdx = matchIndex + (U32)matchLength;
330a28cd43dSSascha Wildner if ( (4*(int)(matchLength-bestLength)) > (int)(ZSTD_highbit32(curr-matchIndex+1) - ZSTD_highbit32((U32)offsetPtr[0]+1)) )
331a28cd43dSSascha Wildner bestLength = matchLength, *offsetPtr = ZSTD_REP_MOVE + curr - matchIndex;
332a28cd43dSSascha Wildner if (ip+matchLength == iend) { /* equal : no way to know if inf or sup */
333a28cd43dSSascha Wildner if (dictMode == ZSTD_dictMatchState) {
334a28cd43dSSascha Wildner nbCompares = 0; /* in addition to avoiding checking any
335a28cd43dSSascha Wildner * further in this loop, make sure we
336a28cd43dSSascha Wildner * skip checking in the dictionary. */
337a28cd43dSSascha Wildner }
338a28cd43dSSascha Wildner break; /* drop, to guarantee consistency (miss a little bit of compression) */
339a28cd43dSSascha Wildner }
340a28cd43dSSascha Wildner }
341a28cd43dSSascha Wildner
342a28cd43dSSascha Wildner if (match[matchLength] < ip[matchLength]) {
343a28cd43dSSascha Wildner /* match is smaller than current */
344a28cd43dSSascha Wildner *smallerPtr = matchIndex; /* update smaller idx */
345a28cd43dSSascha Wildner commonLengthSmaller = matchLength; /* all smaller will now have at least this guaranteed common length */
346a28cd43dSSascha Wildner if (matchIndex <= btLow) { smallerPtr=&dummy32; break; } /* beyond tree size, stop the search */
347a28cd43dSSascha Wildner smallerPtr = nextPtr+1; /* new "smaller" => larger of match */
348a28cd43dSSascha Wildner matchIndex = nextPtr[1]; /* new matchIndex larger than previous (closer to current) */
349a28cd43dSSascha Wildner } else {
350a28cd43dSSascha Wildner /* match is larger than current */
351a28cd43dSSascha Wildner *largerPtr = matchIndex;
352a28cd43dSSascha Wildner commonLengthLarger = matchLength;
353a28cd43dSSascha Wildner if (matchIndex <= btLow) { largerPtr=&dummy32; break; } /* beyond tree size, stop the search */
354a28cd43dSSascha Wildner largerPtr = nextPtr;
355a28cd43dSSascha Wildner matchIndex = nextPtr[0];
356a28cd43dSSascha Wildner } }
357a28cd43dSSascha Wildner
358a28cd43dSSascha Wildner *smallerPtr = *largerPtr = 0;
359a28cd43dSSascha Wildner
360a28cd43dSSascha Wildner if (dictMode == ZSTD_dictMatchState && nbCompares) {
361a28cd43dSSascha Wildner bestLength = ZSTD_DUBT_findBetterDictMatch(
362a28cd43dSSascha Wildner ms, ip, iend,
363a28cd43dSSascha Wildner offsetPtr, bestLength, nbCompares,
364a28cd43dSSascha Wildner mls, dictMode);
365a28cd43dSSascha Wildner }
366a28cd43dSSascha Wildner
367a28cd43dSSascha Wildner assert(matchEndIdx > curr+8); /* ensure nextToUpdate is increased */
368a28cd43dSSascha Wildner ms->nextToUpdate = matchEndIdx - 8; /* skip repetitive patterns */
369a28cd43dSSascha Wildner if (bestLength >= MINMATCH) {
370a28cd43dSSascha Wildner U32 const mIndex = curr - ((U32)*offsetPtr - ZSTD_REP_MOVE); (void)mIndex;
371a28cd43dSSascha Wildner DEBUGLOG(8, "ZSTD_DUBT_findBestMatch(%u) : found match of length %u and offsetCode %u (pos %u)",
372a28cd43dSSascha Wildner curr, (U32)bestLength, (U32)*offsetPtr, mIndex);
373a28cd43dSSascha Wildner }
374a28cd43dSSascha Wildner return bestLength;
375a28cd43dSSascha Wildner }
376a28cd43dSSascha Wildner }
377a28cd43dSSascha Wildner
378a28cd43dSSascha Wildner
379a28cd43dSSascha Wildner /** ZSTD_BtFindBestMatch() : Tree updater, providing best match */
380a28cd43dSSascha Wildner FORCE_INLINE_TEMPLATE size_t
ZSTD_BtFindBestMatch(ZSTD_matchState_t * ms,const BYTE * const ip,const BYTE * const iLimit,size_t * offsetPtr,const U32 mls,const ZSTD_dictMode_e dictMode)381a28cd43dSSascha Wildner ZSTD_BtFindBestMatch( ZSTD_matchState_t* ms,
382a28cd43dSSascha Wildner const BYTE* const ip, const BYTE* const iLimit,
383a28cd43dSSascha Wildner size_t* offsetPtr,
384a28cd43dSSascha Wildner const U32 mls /* template */,
385a28cd43dSSascha Wildner const ZSTD_dictMode_e dictMode)
386a28cd43dSSascha Wildner {
387a28cd43dSSascha Wildner DEBUGLOG(7, "ZSTD_BtFindBestMatch");
388a28cd43dSSascha Wildner if (ip < ms->window.base + ms->nextToUpdate) return 0; /* skipped area */
389a28cd43dSSascha Wildner ZSTD_updateDUBT(ms, ip, iLimit, mls);
390a28cd43dSSascha Wildner return ZSTD_DUBT_findBestMatch(ms, ip, iLimit, offsetPtr, mls, dictMode);
391a28cd43dSSascha Wildner }
392a28cd43dSSascha Wildner
393a28cd43dSSascha Wildner
394a28cd43dSSascha Wildner static size_t
ZSTD_BtFindBestMatch_selectMLS(ZSTD_matchState_t * ms,const BYTE * ip,const BYTE * const iLimit,size_t * offsetPtr)395a28cd43dSSascha Wildner ZSTD_BtFindBestMatch_selectMLS ( ZSTD_matchState_t* ms,
396a28cd43dSSascha Wildner const BYTE* ip, const BYTE* const iLimit,
397a28cd43dSSascha Wildner size_t* offsetPtr)
398a28cd43dSSascha Wildner {
399a28cd43dSSascha Wildner switch(ms->cParams.minMatch)
400a28cd43dSSascha Wildner {
401a28cd43dSSascha Wildner default : /* includes case 3 */
402a28cd43dSSascha Wildner case 4 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 4, ZSTD_noDict);
403a28cd43dSSascha Wildner case 5 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 5, ZSTD_noDict);
404a28cd43dSSascha Wildner case 7 :
405a28cd43dSSascha Wildner case 6 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 6, ZSTD_noDict);
406a28cd43dSSascha Wildner }
407a28cd43dSSascha Wildner }
408a28cd43dSSascha Wildner
409a28cd43dSSascha Wildner
ZSTD_BtFindBestMatch_dictMatchState_selectMLS(ZSTD_matchState_t * ms,const BYTE * ip,const BYTE * const iLimit,size_t * offsetPtr)410a28cd43dSSascha Wildner static size_t ZSTD_BtFindBestMatch_dictMatchState_selectMLS (
411a28cd43dSSascha Wildner ZSTD_matchState_t* ms,
412a28cd43dSSascha Wildner const BYTE* ip, const BYTE* const iLimit,
413a28cd43dSSascha Wildner size_t* offsetPtr)
414a28cd43dSSascha Wildner {
415a28cd43dSSascha Wildner switch(ms->cParams.minMatch)
416a28cd43dSSascha Wildner {
417a28cd43dSSascha Wildner default : /* includes case 3 */
418a28cd43dSSascha Wildner case 4 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 4, ZSTD_dictMatchState);
419a28cd43dSSascha Wildner case 5 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 5, ZSTD_dictMatchState);
420a28cd43dSSascha Wildner case 7 :
421a28cd43dSSascha Wildner case 6 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 6, ZSTD_dictMatchState);
422a28cd43dSSascha Wildner }
423a28cd43dSSascha Wildner }
424a28cd43dSSascha Wildner
425a28cd43dSSascha Wildner
ZSTD_BtFindBestMatch_extDict_selectMLS(ZSTD_matchState_t * ms,const BYTE * ip,const BYTE * const iLimit,size_t * offsetPtr)426a28cd43dSSascha Wildner static size_t ZSTD_BtFindBestMatch_extDict_selectMLS (
427a28cd43dSSascha Wildner ZSTD_matchState_t* ms,
428a28cd43dSSascha Wildner const BYTE* ip, const BYTE* const iLimit,
429a28cd43dSSascha Wildner size_t* offsetPtr)
430a28cd43dSSascha Wildner {
431a28cd43dSSascha Wildner switch(ms->cParams.minMatch)
432a28cd43dSSascha Wildner {
433a28cd43dSSascha Wildner default : /* includes case 3 */
434a28cd43dSSascha Wildner case 4 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 4, ZSTD_extDict);
435a28cd43dSSascha Wildner case 5 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 5, ZSTD_extDict);
436a28cd43dSSascha Wildner case 7 :
437a28cd43dSSascha Wildner case 6 : return ZSTD_BtFindBestMatch(ms, ip, iLimit, offsetPtr, 6, ZSTD_extDict);
438a28cd43dSSascha Wildner }
439a28cd43dSSascha Wildner }
440a28cd43dSSascha Wildner
441a28cd43dSSascha Wildner
442a28cd43dSSascha Wildner
443a28cd43dSSascha Wildner /* *********************************
444a28cd43dSSascha Wildner * Hash Chain
445a28cd43dSSascha Wildner ***********************************/
446a28cd43dSSascha Wildner #define NEXT_IN_CHAIN(d, mask) chainTable[(d) & (mask)]
447a28cd43dSSascha Wildner
448a28cd43dSSascha Wildner /* Update chains up to ip (excluded)
449a28cd43dSSascha Wildner Assumption : always within prefix (i.e. not within extDict) */
ZSTD_insertAndFindFirstIndex_internal(ZSTD_matchState_t * ms,const ZSTD_compressionParameters * const cParams,const BYTE * ip,U32 const mls)450a28cd43dSSascha Wildner FORCE_INLINE_TEMPLATE U32 ZSTD_insertAndFindFirstIndex_internal(
451a28cd43dSSascha Wildner ZSTD_matchState_t* ms,
452a28cd43dSSascha Wildner const ZSTD_compressionParameters* const cParams,
453a28cd43dSSascha Wildner const BYTE* ip, U32 const mls)
454a28cd43dSSascha Wildner {
455a28cd43dSSascha Wildner U32* const hashTable = ms->hashTable;
456a28cd43dSSascha Wildner const U32 hashLog = cParams->hashLog;
457a28cd43dSSascha Wildner U32* const chainTable = ms->chainTable;
458a28cd43dSSascha Wildner const U32 chainMask = (1 << cParams->chainLog) - 1;
459a28cd43dSSascha Wildner const BYTE* const base = ms->window.base;
460a28cd43dSSascha Wildner const U32 target = (U32)(ip - base);
461a28cd43dSSascha Wildner U32 idx = ms->nextToUpdate;
462a28cd43dSSascha Wildner
463a28cd43dSSascha Wildner while(idx < target) { /* catch up */
464a28cd43dSSascha Wildner size_t const h = ZSTD_hashPtr(base+idx, hashLog, mls);
465a28cd43dSSascha Wildner NEXT_IN_CHAIN(idx, chainMask) = hashTable[h];
466a28cd43dSSascha Wildner hashTable[h] = idx;
467a28cd43dSSascha Wildner idx++;
468a28cd43dSSascha Wildner }
469a28cd43dSSascha Wildner
470a28cd43dSSascha Wildner ms->nextToUpdate = target;
471a28cd43dSSascha Wildner return hashTable[ZSTD_hashPtr(ip, hashLog, mls)];
472a28cd43dSSascha Wildner }
473a28cd43dSSascha Wildner
ZSTD_insertAndFindFirstIndex(ZSTD_matchState_t * ms,const BYTE * ip)474a28cd43dSSascha Wildner U32 ZSTD_insertAndFindFirstIndex(ZSTD_matchState_t* ms, const BYTE* ip) {
475a28cd43dSSascha Wildner const ZSTD_compressionParameters* const cParams = &ms->cParams;
476a28cd43dSSascha Wildner return ZSTD_insertAndFindFirstIndex_internal(ms, cParams, ip, ms->cParams.minMatch);
477a28cd43dSSascha Wildner }
478a28cd43dSSascha Wildner
ZSTD_dedicatedDictSearch_lazy_loadDictionary(ZSTD_matchState_t * ms,const BYTE * const ip)479a28cd43dSSascha Wildner void ZSTD_dedicatedDictSearch_lazy_loadDictionary(ZSTD_matchState_t* ms, const BYTE* const ip)
480a28cd43dSSascha Wildner {
481a28cd43dSSascha Wildner const BYTE* const base = ms->window.base;
482a28cd43dSSascha Wildner U32 const target = (U32)(ip - base);
483a28cd43dSSascha Wildner U32* const hashTable = ms->hashTable;
484a28cd43dSSascha Wildner U32* const chainTable = ms->chainTable;
485a28cd43dSSascha Wildner U32 const chainSize = 1 << ms->cParams.chainLog;
486a28cd43dSSascha Wildner U32 idx = ms->nextToUpdate;
487a28cd43dSSascha Wildner U32 const minChain = chainSize < target ? target - chainSize : idx;
488a28cd43dSSascha Wildner U32 const bucketSize = 1 << ZSTD_LAZY_DDSS_BUCKET_LOG;
489a28cd43dSSascha Wildner U32 const cacheSize = bucketSize - 1;
490a28cd43dSSascha Wildner U32 const chainAttempts = (1 << ms->cParams.searchLog) - cacheSize;
491a28cd43dSSascha Wildner U32 const chainLimit = chainAttempts > 255 ? 255 : chainAttempts;
492a28cd43dSSascha Wildner
493a28cd43dSSascha Wildner /* We know the hashtable is oversized by a factor of `bucketSize`.
494a28cd43dSSascha Wildner * We are going to temporarily pretend `bucketSize == 1`, keeping only a
495a28cd43dSSascha Wildner * single entry. We will use the rest of the space to construct a temporary
496a28cd43dSSascha Wildner * chaintable.
497a28cd43dSSascha Wildner */
498a28cd43dSSascha Wildner U32 const hashLog = ms->cParams.hashLog - ZSTD_LAZY_DDSS_BUCKET_LOG;
499a28cd43dSSascha Wildner U32* const tmpHashTable = hashTable;
500a28cd43dSSascha Wildner U32* const tmpChainTable = hashTable + ((size_t)1 << hashLog);
501a28cd43dSSascha Wildner U32 const tmpChainSize = ((1 << ZSTD_LAZY_DDSS_BUCKET_LOG) - 1) << hashLog;
502a28cd43dSSascha Wildner U32 const tmpMinChain = tmpChainSize < target ? target - tmpChainSize : idx;
503a28cd43dSSascha Wildner
504a28cd43dSSascha Wildner U32 hashIdx;
505a28cd43dSSascha Wildner
506a28cd43dSSascha Wildner assert(ms->cParams.chainLog <= 24);
507a28cd43dSSascha Wildner assert(ms->cParams.hashLog >= ms->cParams.chainLog);
508a28cd43dSSascha Wildner assert(idx != 0);
509a28cd43dSSascha Wildner assert(tmpMinChain <= minChain);
510a28cd43dSSascha Wildner
511a28cd43dSSascha Wildner /* fill conventional hash table and conventional chain table */
512a28cd43dSSascha Wildner for ( ; idx < target; idx++) {
513a28cd43dSSascha Wildner U32 const h = (U32)ZSTD_hashPtr(base + idx, hashLog, ms->cParams.minMatch);
514a28cd43dSSascha Wildner if (idx >= tmpMinChain) {
515a28cd43dSSascha Wildner tmpChainTable[idx - tmpMinChain] = hashTable[h];
516a28cd43dSSascha Wildner }
517a28cd43dSSascha Wildner tmpHashTable[h] = idx;
518a28cd43dSSascha Wildner }
519a28cd43dSSascha Wildner
520a28cd43dSSascha Wildner /* sort chains into ddss chain table */
521a28cd43dSSascha Wildner {
522a28cd43dSSascha Wildner U32 chainPos = 0;
523a28cd43dSSascha Wildner for (hashIdx = 0; hashIdx < (1U << hashLog); hashIdx++) {
524a28cd43dSSascha Wildner U32 count;
525a28cd43dSSascha Wildner U32 countBeyondMinChain = 0;
526a28cd43dSSascha Wildner U32 i = tmpHashTable[hashIdx];
527a28cd43dSSascha Wildner for (count = 0; i >= tmpMinChain && count < cacheSize; count++) {
528a28cd43dSSascha Wildner /* skip through the chain to the first position that won't be
529a28cd43dSSascha Wildner * in the hash cache bucket */
530a28cd43dSSascha Wildner if (i < minChain) {
531a28cd43dSSascha Wildner countBeyondMinChain++;
532a28cd43dSSascha Wildner }
533a28cd43dSSascha Wildner i = tmpChainTable[i - tmpMinChain];
534a28cd43dSSascha Wildner }
535a28cd43dSSascha Wildner if (count == cacheSize) {
536a28cd43dSSascha Wildner for (count = 0; count < chainLimit;) {
537a28cd43dSSascha Wildner if (i < minChain) {
538a28cd43dSSascha Wildner if (!i || countBeyondMinChain++ > cacheSize) {
539a28cd43dSSascha Wildner /* only allow pulling `cacheSize` number of entries
540a28cd43dSSascha Wildner * into the cache or chainTable beyond `minChain`,
541a28cd43dSSascha Wildner * to replace the entries pulled out of the
542a28cd43dSSascha Wildner * chainTable into the cache. This lets us reach
543a28cd43dSSascha Wildner * back further without increasing the total number
544a28cd43dSSascha Wildner * of entries in the chainTable, guaranteeing the
545a28cd43dSSascha Wildner * DDSS chain table will fit into the space
546a28cd43dSSascha Wildner * allocated for the regular one. */
547a28cd43dSSascha Wildner break;
548a28cd43dSSascha Wildner }
549a28cd43dSSascha Wildner }
550a28cd43dSSascha Wildner chainTable[chainPos++] = i;
551a28cd43dSSascha Wildner count++;
552a28cd43dSSascha Wildner if (i < tmpMinChain) {
553a28cd43dSSascha Wildner break;
554a28cd43dSSascha Wildner }
555a28cd43dSSascha Wildner i = tmpChainTable[i - tmpMinChain];
556a28cd43dSSascha Wildner }
557a28cd43dSSascha Wildner } else {
558a28cd43dSSascha Wildner count = 0;
559a28cd43dSSascha Wildner }
560a28cd43dSSascha Wildner if (count) {
561a28cd43dSSascha Wildner tmpHashTable[hashIdx] = ((chainPos - count) << 8) + count;
562a28cd43dSSascha Wildner } else {
563a28cd43dSSascha Wildner tmpHashTable[hashIdx] = 0;
564a28cd43dSSascha Wildner }
565a28cd43dSSascha Wildner }
566a28cd43dSSascha Wildner assert(chainPos <= chainSize); /* I believe this is guaranteed... */
567a28cd43dSSascha Wildner }
568a28cd43dSSascha Wildner
569a28cd43dSSascha Wildner /* move chain pointers into the last entry of each hash bucket */
570a28cd43dSSascha Wildner for (hashIdx = (1 << hashLog); hashIdx; ) {
571a28cd43dSSascha Wildner U32 const bucketIdx = --hashIdx << ZSTD_LAZY_DDSS_BUCKET_LOG;
572a28cd43dSSascha Wildner U32 const chainPackedPointer = tmpHashTable[hashIdx];
573a28cd43dSSascha Wildner U32 i;
574a28cd43dSSascha Wildner for (i = 0; i < cacheSize; i++) {
575a28cd43dSSascha Wildner hashTable[bucketIdx + i] = 0;
576a28cd43dSSascha Wildner }
577a28cd43dSSascha Wildner hashTable[bucketIdx + bucketSize - 1] = chainPackedPointer;
578a28cd43dSSascha Wildner }
579a28cd43dSSascha Wildner
580a28cd43dSSascha Wildner /* fill the buckets of the hash table */
581a28cd43dSSascha Wildner for (idx = ms->nextToUpdate; idx < target; idx++) {
582a28cd43dSSascha Wildner U32 const h = (U32)ZSTD_hashPtr(base + idx, hashLog, ms->cParams.minMatch)
583a28cd43dSSascha Wildner << ZSTD_LAZY_DDSS_BUCKET_LOG;
584a28cd43dSSascha Wildner U32 i;
585a28cd43dSSascha Wildner /* Shift hash cache down 1. */
586a28cd43dSSascha Wildner for (i = cacheSize - 1; i; i--)
587a28cd43dSSascha Wildner hashTable[h + i] = hashTable[h + i - 1];
588a28cd43dSSascha Wildner hashTable[h] = idx;
589a28cd43dSSascha Wildner }
590a28cd43dSSascha Wildner
591a28cd43dSSascha Wildner ms->nextToUpdate = target;
592a28cd43dSSascha Wildner }
593a28cd43dSSascha Wildner
594a28cd43dSSascha Wildner
595a28cd43dSSascha Wildner /* inlining is important to hardwire a hot branch (template emulation) */
596a28cd43dSSascha Wildner FORCE_INLINE_TEMPLATE
ZSTD_HcFindBestMatch_generic(ZSTD_matchState_t * ms,const BYTE * const ip,const BYTE * const iLimit,size_t * offsetPtr,const U32 mls,const ZSTD_dictMode_e dictMode)597a28cd43dSSascha Wildner size_t ZSTD_HcFindBestMatch_generic (
598a28cd43dSSascha Wildner ZSTD_matchState_t* ms,
599a28cd43dSSascha Wildner const BYTE* const ip, const BYTE* const iLimit,
600a28cd43dSSascha Wildner size_t* offsetPtr,
601a28cd43dSSascha Wildner const U32 mls, const ZSTD_dictMode_e dictMode)
602a28cd43dSSascha Wildner {
603a28cd43dSSascha Wildner const ZSTD_compressionParameters* const cParams = &ms->cParams;
604a28cd43dSSascha Wildner U32* const chainTable = ms->chainTable;
605a28cd43dSSascha Wildner const U32 chainSize = (1 << cParams->chainLog);
606a28cd43dSSascha Wildner const U32 chainMask = chainSize-1;
607a28cd43dSSascha Wildner const BYTE* const base = ms->window.base;
608a28cd43dSSascha Wildner const BYTE* const dictBase = ms->window.dictBase;
609a28cd43dSSascha Wildner const U32 dictLimit = ms->window.dictLimit;
610a28cd43dSSascha Wildner const BYTE* const prefixStart = base + dictLimit;
611a28cd43dSSascha Wildner const BYTE* const dictEnd = dictBase + dictLimit;
612a28cd43dSSascha Wildner const U32 curr = (U32)(ip-base);
613a28cd43dSSascha Wildner const U32 maxDistance = 1U << cParams->windowLog;
614a28cd43dSSascha Wildner const U32 lowestValid = ms->window.lowLimit;
615a28cd43dSSascha Wildner const U32 withinMaxDistance = (curr - lowestValid > maxDistance) ? curr - maxDistance : lowestValid;
616a28cd43dSSascha Wildner const U32 isDictionary = (ms->loadedDictEnd != 0);
617a28cd43dSSascha Wildner const U32 lowLimit = isDictionary ? lowestValid : withinMaxDistance;
618a28cd43dSSascha Wildner const U32 minChain = curr > chainSize ? curr - chainSize : 0;
619a28cd43dSSascha Wildner U32 nbAttempts = 1U << cParams->searchLog;
620a28cd43dSSascha Wildner size_t ml=4-1;
621a28cd43dSSascha Wildner
622a28cd43dSSascha Wildner const ZSTD_matchState_t* const dms = ms->dictMatchState;
623a28cd43dSSascha Wildner const U32 ddsHashLog = dictMode == ZSTD_dedicatedDictSearch
624a28cd43dSSascha Wildner ? dms->cParams.hashLog - ZSTD_LAZY_DDSS_BUCKET_LOG : 0;
625a28cd43dSSascha Wildner const size_t ddsIdx = dictMode == ZSTD_dedicatedDictSearch
626a28cd43dSSascha Wildner ? ZSTD_hashPtr(ip, ddsHashLog, mls) << ZSTD_LAZY_DDSS_BUCKET_LOG : 0;
627a28cd43dSSascha Wildner
628a28cd43dSSascha Wildner U32 matchIndex;
629a28cd43dSSascha Wildner
630a28cd43dSSascha Wildner if (dictMode == ZSTD_dedicatedDictSearch) {
631a28cd43dSSascha Wildner const U32* entry = &dms->hashTable[ddsIdx];
632a28cd43dSSascha Wildner PREFETCH_L1(entry);
633a28cd43dSSascha Wildner }
634a28cd43dSSascha Wildner
635a28cd43dSSascha Wildner /* HC4 match finder */
636a28cd43dSSascha Wildner matchIndex = ZSTD_insertAndFindFirstIndex_internal(ms, cParams, ip, mls);
637a28cd43dSSascha Wildner
638a28cd43dSSascha Wildner for ( ; (matchIndex>=lowLimit) & (nbAttempts>0) ; nbAttempts--) {
639a28cd43dSSascha Wildner size_t currentMl=0;
640a28cd43dSSascha Wildner if ((dictMode != ZSTD_extDict) || matchIndex >= dictLimit) {
641a28cd43dSSascha Wildner const BYTE* const match = base + matchIndex;
642a28cd43dSSascha Wildner assert(matchIndex >= dictLimit); /* ensures this is true if dictMode != ZSTD_extDict */
643a28cd43dSSascha Wildner if (match[ml] == ip[ml]) /* potentially better */
644a28cd43dSSascha Wildner currentMl = ZSTD_count(ip, match, iLimit);
645a28cd43dSSascha Wildner } else {
646a28cd43dSSascha Wildner const BYTE* const match = dictBase + matchIndex;
647a28cd43dSSascha Wildner assert(match+4 <= dictEnd);
648a28cd43dSSascha Wildner if (MEM_read32(match) == MEM_read32(ip)) /* assumption : matchIndex <= dictLimit-4 (by table construction) */
649a28cd43dSSascha Wildner currentMl = ZSTD_count_2segments(ip+4, match+4, iLimit, dictEnd, prefixStart) + 4;
650a28cd43dSSascha Wildner }
651a28cd43dSSascha Wildner
652a28cd43dSSascha Wildner /* save best solution */
653a28cd43dSSascha Wildner if (currentMl > ml) {
654a28cd43dSSascha Wildner ml = currentMl;
655a28cd43dSSascha Wildner *offsetPtr = curr - matchIndex + ZSTD_REP_MOVE;
656a28cd43dSSascha Wildner if (ip+currentMl == iLimit) break; /* best possible, avoids read overflow on next attempt */
657a28cd43dSSascha Wildner }
658a28cd43dSSascha Wildner
659a28cd43dSSascha Wildner if (matchIndex <= minChain) break;
660a28cd43dSSascha Wildner matchIndex = NEXT_IN_CHAIN(matchIndex, chainMask);
661a28cd43dSSascha Wildner }
662a28cd43dSSascha Wildner
663a28cd43dSSascha Wildner if (dictMode == ZSTD_dedicatedDictSearch) {
664a28cd43dSSascha Wildner const U32 ddsLowestIndex = dms->window.dictLimit;
665a28cd43dSSascha Wildner const BYTE* const ddsBase = dms->window.base;
666a28cd43dSSascha Wildner const BYTE* const ddsEnd = dms->window.nextSrc;
667a28cd43dSSascha Wildner const U32 ddsSize = (U32)(ddsEnd - ddsBase);
668a28cd43dSSascha Wildner const U32 ddsIndexDelta = dictLimit - ddsSize;
669a28cd43dSSascha Wildner const U32 bucketSize = (1 << ZSTD_LAZY_DDSS_BUCKET_LOG);
670a28cd43dSSascha Wildner const U32 bucketLimit = nbAttempts < bucketSize - 1 ? nbAttempts : bucketSize - 1;
671a28cd43dSSascha Wildner U32 ddsAttempt;
672a28cd43dSSascha Wildner
673a28cd43dSSascha Wildner for (ddsAttempt = 0; ddsAttempt < bucketSize - 1; ddsAttempt++) {
674a28cd43dSSascha Wildner PREFETCH_L1(ddsBase + dms->hashTable[ddsIdx + ddsAttempt]);
675a28cd43dSSascha Wildner }
676a28cd43dSSascha Wildner
677a28cd43dSSascha Wildner {
678a28cd43dSSascha Wildner U32 const chainPackedPointer = dms->hashTable[ddsIdx + bucketSize - 1];
679a28cd43dSSascha Wildner U32 const chainIndex = chainPackedPointer >> 8;
680a28cd43dSSascha Wildner
681a28cd43dSSascha Wildner PREFETCH_L1(&dms->chainTable[chainIndex]);
682a28cd43dSSascha Wildner }
683a28cd43dSSascha Wildner
684a28cd43dSSascha Wildner for (ddsAttempt = 0; ddsAttempt < bucketLimit; ddsAttempt++) {
685a28cd43dSSascha Wildner size_t currentMl=0;
686a28cd43dSSascha Wildner const BYTE* match;
687a28cd43dSSascha Wildner matchIndex = dms->hashTable[ddsIdx + ddsAttempt];
688a28cd43dSSascha Wildner match = ddsBase + matchIndex;
689a28cd43dSSascha Wildner
690a28cd43dSSascha Wildner if (!matchIndex) {
691a28cd43dSSascha Wildner return ml;
692a28cd43dSSascha Wildner }
693a28cd43dSSascha Wildner
694a28cd43dSSascha Wildner /* guaranteed by table construction */
695a28cd43dSSascha Wildner (void)ddsLowestIndex;
696a28cd43dSSascha Wildner assert(matchIndex >= ddsLowestIndex);
697a28cd43dSSascha Wildner assert(match+4 <= ddsEnd);
698a28cd43dSSascha Wildner if (MEM_read32(match) == MEM_read32(ip)) {
699a28cd43dSSascha Wildner /* assumption : matchIndex <= dictLimit-4 (by table construction) */
700a28cd43dSSascha Wildner currentMl = ZSTD_count_2segments(ip+4, match+4, iLimit, ddsEnd, prefixStart) + 4;
701a28cd43dSSascha Wildner }
702a28cd43dSSascha Wildner
703a28cd43dSSascha Wildner /* save best solution */
704a28cd43dSSascha Wildner if (currentMl > ml) {
705a28cd43dSSascha Wildner ml = currentMl;
706a28cd43dSSascha Wildner *offsetPtr = curr - (matchIndex + ddsIndexDelta) + ZSTD_REP_MOVE;
707a28cd43dSSascha Wildner if (ip+currentMl == iLimit) {
708a28cd43dSSascha Wildner /* best possible, avoids read overflow on next attempt */
709a28cd43dSSascha Wildner return ml;
710a28cd43dSSascha Wildner }
711a28cd43dSSascha Wildner }
712a28cd43dSSascha Wildner }
713a28cd43dSSascha Wildner
714a28cd43dSSascha Wildner {
715a28cd43dSSascha Wildner U32 const chainPackedPointer = dms->hashTable[ddsIdx + bucketSize - 1];
716a28cd43dSSascha Wildner U32 chainIndex = chainPackedPointer >> 8;
717a28cd43dSSascha Wildner U32 const chainLength = chainPackedPointer & 0xFF;
718a28cd43dSSascha Wildner U32 const chainAttempts = nbAttempts - ddsAttempt;
719a28cd43dSSascha Wildner U32 const chainLimit = chainAttempts > chainLength ? chainLength : chainAttempts;
720a28cd43dSSascha Wildner U32 chainAttempt;
721a28cd43dSSascha Wildner
722a28cd43dSSascha Wildner for (chainAttempt = 0 ; chainAttempt < chainLimit; chainAttempt++) {
723a28cd43dSSascha Wildner PREFETCH_L1(ddsBase + dms->chainTable[chainIndex + chainAttempt]);
724a28cd43dSSascha Wildner }
725a28cd43dSSascha Wildner
726a28cd43dSSascha Wildner for (chainAttempt = 0 ; chainAttempt < chainLimit; chainAttempt++, chainIndex++) {
727a28cd43dSSascha Wildner size_t currentMl=0;
728a28cd43dSSascha Wildner const BYTE* match;
729a28cd43dSSascha Wildner matchIndex = dms->chainTable[chainIndex];
730a28cd43dSSascha Wildner match = ddsBase + matchIndex;
731a28cd43dSSascha Wildner
732a28cd43dSSascha Wildner /* guaranteed by table construction */
733a28cd43dSSascha Wildner assert(matchIndex >= ddsLowestIndex);
734a28cd43dSSascha Wildner assert(match+4 <= ddsEnd);
735a28cd43dSSascha Wildner if (MEM_read32(match) == MEM_read32(ip)) {
736a28cd43dSSascha Wildner /* assumption : matchIndex <= dictLimit-4 (by table construction) */
737a28cd43dSSascha Wildner currentMl = ZSTD_count_2segments(ip+4, match+4, iLimit, ddsEnd, prefixStart) + 4;
738a28cd43dSSascha Wildner }
739a28cd43dSSascha Wildner
740a28cd43dSSascha Wildner /* save best solution */
741a28cd43dSSascha Wildner if (currentMl > ml) {
742a28cd43dSSascha Wildner ml = currentMl;
743a28cd43dSSascha Wildner *offsetPtr = curr - (matchIndex + ddsIndexDelta) + ZSTD_REP_MOVE;
744a28cd43dSSascha Wildner if (ip+currentMl == iLimit) break; /* best possible, avoids read overflow on next attempt */
745a28cd43dSSascha Wildner }
746a28cd43dSSascha Wildner }
747a28cd43dSSascha Wildner }
748a28cd43dSSascha Wildner } else if (dictMode == ZSTD_dictMatchState) {
749a28cd43dSSascha Wildner const U32* const dmsChainTable = dms->chainTable;
750a28cd43dSSascha Wildner const U32 dmsChainSize = (1 << dms->cParams.chainLog);
751a28cd43dSSascha Wildner const U32 dmsChainMask = dmsChainSize - 1;
752a28cd43dSSascha Wildner const U32 dmsLowestIndex = dms->window.dictLimit;
753a28cd43dSSascha Wildner const BYTE* const dmsBase = dms->window.base;
754a28cd43dSSascha Wildner const BYTE* const dmsEnd = dms->window.nextSrc;
755a28cd43dSSascha Wildner const U32 dmsSize = (U32)(dmsEnd - dmsBase);
756a28cd43dSSascha Wildner const U32 dmsIndexDelta = dictLimit - dmsSize;
757a28cd43dSSascha Wildner const U32 dmsMinChain = dmsSize > dmsChainSize ? dmsSize - dmsChainSize : 0;
758a28cd43dSSascha Wildner
759a28cd43dSSascha Wildner matchIndex = dms->hashTable[ZSTD_hashPtr(ip, dms->cParams.hashLog, mls)];
760a28cd43dSSascha Wildner
761a28cd43dSSascha Wildner for ( ; (matchIndex>=dmsLowestIndex) & (nbAttempts>0) ; nbAttempts--) {
762a28cd43dSSascha Wildner size_t currentMl=0;
763a28cd43dSSascha Wildner const BYTE* const match = dmsBase + matchIndex;
764a28cd43dSSascha Wildner assert(match+4 <= dmsEnd);
765a28cd43dSSascha Wildner if (MEM_read32(match) == MEM_read32(ip)) /* assumption : matchIndex <= dictLimit-4 (by table construction) */
766a28cd43dSSascha Wildner currentMl = ZSTD_count_2segments(ip+4, match+4, iLimit, dmsEnd, prefixStart) + 4;
767a28cd43dSSascha Wildner
768a28cd43dSSascha Wildner /* save best solution */
769a28cd43dSSascha Wildner if (currentMl > ml) {
770a28cd43dSSascha Wildner ml = currentMl;
771a28cd43dSSascha Wildner *offsetPtr = curr - (matchIndex + dmsIndexDelta) + ZSTD_REP_MOVE;
772a28cd43dSSascha Wildner if (ip+currentMl == iLimit) break; /* best possible, avoids read overflow on next attempt */
773a28cd43dSSascha Wildner }
774a28cd43dSSascha Wildner
775a28cd43dSSascha Wildner if (matchIndex <= dmsMinChain) break;
776a28cd43dSSascha Wildner
777a28cd43dSSascha Wildner matchIndex = dmsChainTable[matchIndex & dmsChainMask];
778a28cd43dSSascha Wildner }
779a28cd43dSSascha Wildner }
780a28cd43dSSascha Wildner
781a28cd43dSSascha Wildner return ml;
782a28cd43dSSascha Wildner }
783a28cd43dSSascha Wildner
784a28cd43dSSascha Wildner
ZSTD_HcFindBestMatch_selectMLS(ZSTD_matchState_t * ms,const BYTE * ip,const BYTE * const iLimit,size_t * offsetPtr)785a28cd43dSSascha Wildner FORCE_INLINE_TEMPLATE size_t ZSTD_HcFindBestMatch_selectMLS (
786a28cd43dSSascha Wildner ZSTD_matchState_t* ms,
787a28cd43dSSascha Wildner const BYTE* ip, const BYTE* const iLimit,
788a28cd43dSSascha Wildner size_t* offsetPtr)
789a28cd43dSSascha Wildner {
790a28cd43dSSascha Wildner switch(ms->cParams.minMatch)
791a28cd43dSSascha Wildner {
792a28cd43dSSascha Wildner default : /* includes case 3 */
793a28cd43dSSascha Wildner case 4 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 4, ZSTD_noDict);
794a28cd43dSSascha Wildner case 5 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 5, ZSTD_noDict);
795a28cd43dSSascha Wildner case 7 :
796a28cd43dSSascha Wildner case 6 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 6, ZSTD_noDict);
797a28cd43dSSascha Wildner }
798a28cd43dSSascha Wildner }
799a28cd43dSSascha Wildner
800a28cd43dSSascha Wildner
ZSTD_HcFindBestMatch_dictMatchState_selectMLS(ZSTD_matchState_t * ms,const BYTE * ip,const BYTE * const iLimit,size_t * offsetPtr)801a28cd43dSSascha Wildner static size_t ZSTD_HcFindBestMatch_dictMatchState_selectMLS (
802a28cd43dSSascha Wildner ZSTD_matchState_t* ms,
803a28cd43dSSascha Wildner const BYTE* ip, const BYTE* const iLimit,
804a28cd43dSSascha Wildner size_t* offsetPtr)
805a28cd43dSSascha Wildner {
806a28cd43dSSascha Wildner switch(ms->cParams.minMatch)
807a28cd43dSSascha Wildner {
808a28cd43dSSascha Wildner default : /* includes case 3 */
809a28cd43dSSascha Wildner case 4 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 4, ZSTD_dictMatchState);
810a28cd43dSSascha Wildner case 5 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 5, ZSTD_dictMatchState);
811a28cd43dSSascha Wildner case 7 :
812a28cd43dSSascha Wildner case 6 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 6, ZSTD_dictMatchState);
813a28cd43dSSascha Wildner }
814a28cd43dSSascha Wildner }
815a28cd43dSSascha Wildner
816a28cd43dSSascha Wildner
ZSTD_HcFindBestMatch_dedicatedDictSearch_selectMLS(ZSTD_matchState_t * ms,const BYTE * ip,const BYTE * const iLimit,size_t * offsetPtr)817a28cd43dSSascha Wildner static size_t ZSTD_HcFindBestMatch_dedicatedDictSearch_selectMLS (
818a28cd43dSSascha Wildner ZSTD_matchState_t* ms,
819a28cd43dSSascha Wildner const BYTE* ip, const BYTE* const iLimit,
820a28cd43dSSascha Wildner size_t* offsetPtr)
821a28cd43dSSascha Wildner {
822a28cd43dSSascha Wildner switch(ms->cParams.minMatch)
823a28cd43dSSascha Wildner {
824a28cd43dSSascha Wildner default : /* includes case 3 */
825a28cd43dSSascha Wildner case 4 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 4, ZSTD_dedicatedDictSearch);
826a28cd43dSSascha Wildner case 5 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 5, ZSTD_dedicatedDictSearch);
827a28cd43dSSascha Wildner case 7 :
828a28cd43dSSascha Wildner case 6 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 6, ZSTD_dedicatedDictSearch);
829a28cd43dSSascha Wildner }
830a28cd43dSSascha Wildner }
831a28cd43dSSascha Wildner
832a28cd43dSSascha Wildner
ZSTD_HcFindBestMatch_extDict_selectMLS(ZSTD_matchState_t * ms,const BYTE * ip,const BYTE * const iLimit,size_t * offsetPtr)833a28cd43dSSascha Wildner FORCE_INLINE_TEMPLATE size_t ZSTD_HcFindBestMatch_extDict_selectMLS (
834a28cd43dSSascha Wildner ZSTD_matchState_t* ms,
835a28cd43dSSascha Wildner const BYTE* ip, const BYTE* const iLimit,
836a28cd43dSSascha Wildner size_t* offsetPtr)
837a28cd43dSSascha Wildner {
838a28cd43dSSascha Wildner switch(ms->cParams.minMatch)
839a28cd43dSSascha Wildner {
840a28cd43dSSascha Wildner default : /* includes case 3 */
841a28cd43dSSascha Wildner case 4 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 4, ZSTD_extDict);
842a28cd43dSSascha Wildner case 5 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 5, ZSTD_extDict);
843a28cd43dSSascha Wildner case 7 :
844a28cd43dSSascha Wildner case 6 : return ZSTD_HcFindBestMatch_generic(ms, ip, iLimit, offsetPtr, 6, ZSTD_extDict);
845a28cd43dSSascha Wildner }
846a28cd43dSSascha Wildner }
847a28cd43dSSascha Wildner
848a28cd43dSSascha Wildner
849a28cd43dSSascha Wildner /* *******************************
850a28cd43dSSascha Wildner * Common parser - lazy strategy
851a28cd43dSSascha Wildner *********************************/
852a28cd43dSSascha Wildner typedef enum { search_hashChain, search_binaryTree } searchMethod_e;
853a28cd43dSSascha Wildner
854a28cd43dSSascha Wildner FORCE_INLINE_TEMPLATE size_t
ZSTD_compressBlock_lazy_generic(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],const void * src,size_t srcSize,const searchMethod_e searchMethod,const U32 depth,ZSTD_dictMode_e const dictMode)855a28cd43dSSascha Wildner ZSTD_compressBlock_lazy_generic(
856a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore,
857a28cd43dSSascha Wildner U32 rep[ZSTD_REP_NUM],
858a28cd43dSSascha Wildner const void* src, size_t srcSize,
859a28cd43dSSascha Wildner const searchMethod_e searchMethod, const U32 depth,
860a28cd43dSSascha Wildner ZSTD_dictMode_e const dictMode)
861a28cd43dSSascha Wildner {
862a28cd43dSSascha Wildner const BYTE* const istart = (const BYTE*)src;
863a28cd43dSSascha Wildner const BYTE* ip = istart;
864a28cd43dSSascha Wildner const BYTE* anchor = istart;
865a28cd43dSSascha Wildner const BYTE* const iend = istart + srcSize;
866a28cd43dSSascha Wildner const BYTE* const ilimit = iend - 8;
867a28cd43dSSascha Wildner const BYTE* const base = ms->window.base;
868a28cd43dSSascha Wildner const U32 prefixLowestIndex = ms->window.dictLimit;
869a28cd43dSSascha Wildner const BYTE* const prefixLowest = base + prefixLowestIndex;
870a28cd43dSSascha Wildner
871a28cd43dSSascha Wildner typedef size_t (*searchMax_f)(
872a28cd43dSSascha Wildner ZSTD_matchState_t* ms,
873a28cd43dSSascha Wildner const BYTE* ip, const BYTE* iLimit, size_t* offsetPtr);
874a28cd43dSSascha Wildner
875a28cd43dSSascha Wildner /**
876a28cd43dSSascha Wildner * This table is indexed first by the four ZSTD_dictMode_e values, and then
877a28cd43dSSascha Wildner * by the two searchMethod_e values. NULLs are placed for configurations
878a28cd43dSSascha Wildner * that should never occur (extDict modes go to the other implementation
879a28cd43dSSascha Wildner * below and there is no DDSS for binary tree search yet).
880a28cd43dSSascha Wildner */
881a28cd43dSSascha Wildner const searchMax_f searchFuncs[4][2] = {
882a28cd43dSSascha Wildner {
883a28cd43dSSascha Wildner ZSTD_HcFindBestMatch_selectMLS,
884a28cd43dSSascha Wildner ZSTD_BtFindBestMatch_selectMLS
885a28cd43dSSascha Wildner },
886a28cd43dSSascha Wildner {
887a28cd43dSSascha Wildner NULL,
888a28cd43dSSascha Wildner NULL
889a28cd43dSSascha Wildner },
890a28cd43dSSascha Wildner {
891a28cd43dSSascha Wildner ZSTD_HcFindBestMatch_dictMatchState_selectMLS,
892a28cd43dSSascha Wildner ZSTD_BtFindBestMatch_dictMatchState_selectMLS
893a28cd43dSSascha Wildner },
894a28cd43dSSascha Wildner {
895a28cd43dSSascha Wildner ZSTD_HcFindBestMatch_dedicatedDictSearch_selectMLS,
896a28cd43dSSascha Wildner NULL
897a28cd43dSSascha Wildner }
898a28cd43dSSascha Wildner };
899a28cd43dSSascha Wildner
900a28cd43dSSascha Wildner searchMax_f const searchMax = searchFuncs[dictMode][searchMethod == search_binaryTree];
901a28cd43dSSascha Wildner U32 offset_1 = rep[0], offset_2 = rep[1], savedOffset=0;
902a28cd43dSSascha Wildner
903a28cd43dSSascha Wildner const int isDMS = dictMode == ZSTD_dictMatchState;
904a28cd43dSSascha Wildner const int isDDS = dictMode == ZSTD_dedicatedDictSearch;
905a28cd43dSSascha Wildner const int isDxS = isDMS || isDDS;
906a28cd43dSSascha Wildner const ZSTD_matchState_t* const dms = ms->dictMatchState;
907a28cd43dSSascha Wildner const U32 dictLowestIndex = isDxS ? dms->window.dictLimit : 0;
908a28cd43dSSascha Wildner const BYTE* const dictBase = isDxS ? dms->window.base : NULL;
909a28cd43dSSascha Wildner const BYTE* const dictLowest = isDxS ? dictBase + dictLowestIndex : NULL;
910a28cd43dSSascha Wildner const BYTE* const dictEnd = isDxS ? dms->window.nextSrc : NULL;
911a28cd43dSSascha Wildner const U32 dictIndexDelta = isDxS ?
912a28cd43dSSascha Wildner prefixLowestIndex - (U32)(dictEnd - dictBase) :
913a28cd43dSSascha Wildner 0;
914a28cd43dSSascha Wildner const U32 dictAndPrefixLength = (U32)((ip - prefixLowest) + (dictEnd - dictLowest));
915a28cd43dSSascha Wildner
916a28cd43dSSascha Wildner assert(searchMax != NULL);
917a28cd43dSSascha Wildner
918a28cd43dSSascha Wildner DEBUGLOG(5, "ZSTD_compressBlock_lazy_generic (dictMode=%u)", (U32)dictMode);
919a28cd43dSSascha Wildner
920a28cd43dSSascha Wildner /* init */
921a28cd43dSSascha Wildner ip += (dictAndPrefixLength == 0);
922a28cd43dSSascha Wildner if (dictMode == ZSTD_noDict) {
923a28cd43dSSascha Wildner U32 const curr = (U32)(ip - base);
924a28cd43dSSascha Wildner U32 const windowLow = ZSTD_getLowestPrefixIndex(ms, curr, ms->cParams.windowLog);
925a28cd43dSSascha Wildner U32 const maxRep = curr - windowLow;
926a28cd43dSSascha Wildner if (offset_2 > maxRep) savedOffset = offset_2, offset_2 = 0;
927a28cd43dSSascha Wildner if (offset_1 > maxRep) savedOffset = offset_1, offset_1 = 0;
928a28cd43dSSascha Wildner }
929a28cd43dSSascha Wildner if (isDxS) {
930a28cd43dSSascha Wildner /* dictMatchState repCode checks don't currently handle repCode == 0
931a28cd43dSSascha Wildner * disabling. */
932a28cd43dSSascha Wildner assert(offset_1 <= dictAndPrefixLength);
933a28cd43dSSascha Wildner assert(offset_2 <= dictAndPrefixLength);
934a28cd43dSSascha Wildner }
935a28cd43dSSascha Wildner
936a28cd43dSSascha Wildner /* Match Loop */
937a28cd43dSSascha Wildner #if defined(__GNUC__) && defined(__x86_64__)
938a28cd43dSSascha Wildner /* I've measured random a 5% speed loss on levels 5 & 6 (greedy) when the
939a28cd43dSSascha Wildner * code alignment is perturbed. To fix the instability align the loop on 32-bytes.
940a28cd43dSSascha Wildner */
941a28cd43dSSascha Wildner __asm__(".p2align 5");
942a28cd43dSSascha Wildner #endif
943a28cd43dSSascha Wildner while (ip < ilimit) {
944a28cd43dSSascha Wildner size_t matchLength=0;
945a28cd43dSSascha Wildner size_t offset=0;
946a28cd43dSSascha Wildner const BYTE* start=ip+1;
947a28cd43dSSascha Wildner
948a28cd43dSSascha Wildner /* check repCode */
949a28cd43dSSascha Wildner if (isDxS) {
950a28cd43dSSascha Wildner const U32 repIndex = (U32)(ip - base) + 1 - offset_1;
951a28cd43dSSascha Wildner const BYTE* repMatch = ((dictMode == ZSTD_dictMatchState || dictMode == ZSTD_dedicatedDictSearch)
952a28cd43dSSascha Wildner && repIndex < prefixLowestIndex) ?
953a28cd43dSSascha Wildner dictBase + (repIndex - dictIndexDelta) :
954a28cd43dSSascha Wildner base + repIndex;
955a28cd43dSSascha Wildner if (((U32)((prefixLowestIndex-1) - repIndex) >= 3 /* intentional underflow */)
956a28cd43dSSascha Wildner && (MEM_read32(repMatch) == MEM_read32(ip+1)) ) {
957a28cd43dSSascha Wildner const BYTE* repMatchEnd = repIndex < prefixLowestIndex ? dictEnd : iend;
958a28cd43dSSascha Wildner matchLength = ZSTD_count_2segments(ip+1+4, repMatch+4, iend, repMatchEnd, prefixLowest) + 4;
959a28cd43dSSascha Wildner if (depth==0) goto _storeSequence;
960a28cd43dSSascha Wildner }
961a28cd43dSSascha Wildner }
962a28cd43dSSascha Wildner if ( dictMode == ZSTD_noDict
963a28cd43dSSascha Wildner && ((offset_1 > 0) & (MEM_read32(ip+1-offset_1) == MEM_read32(ip+1)))) {
964a28cd43dSSascha Wildner matchLength = ZSTD_count(ip+1+4, ip+1+4-offset_1, iend) + 4;
965a28cd43dSSascha Wildner if (depth==0) goto _storeSequence;
966a28cd43dSSascha Wildner }
967a28cd43dSSascha Wildner
968a28cd43dSSascha Wildner /* first search (depth 0) */
969a28cd43dSSascha Wildner { size_t offsetFound = 999999999;
970a28cd43dSSascha Wildner size_t const ml2 = searchMax(ms, ip, iend, &offsetFound);
971a28cd43dSSascha Wildner if (ml2 > matchLength)
972a28cd43dSSascha Wildner matchLength = ml2, start = ip, offset=offsetFound;
973a28cd43dSSascha Wildner }
974a28cd43dSSascha Wildner
975a28cd43dSSascha Wildner if (matchLength < 4) {
976a28cd43dSSascha Wildner ip += ((ip-anchor) >> kSearchStrength) + 1; /* jump faster over incompressible sections */
977a28cd43dSSascha Wildner continue;
978a28cd43dSSascha Wildner }
979a28cd43dSSascha Wildner
980a28cd43dSSascha Wildner /* let's try to find a better solution */
981a28cd43dSSascha Wildner if (depth>=1)
982a28cd43dSSascha Wildner while (ip<ilimit) {
983a28cd43dSSascha Wildner ip ++;
984a28cd43dSSascha Wildner if ( (dictMode == ZSTD_noDict)
985a28cd43dSSascha Wildner && (offset) && ((offset_1>0) & (MEM_read32(ip) == MEM_read32(ip - offset_1)))) {
986a28cd43dSSascha Wildner size_t const mlRep = ZSTD_count(ip+4, ip+4-offset_1, iend) + 4;
987a28cd43dSSascha Wildner int const gain2 = (int)(mlRep * 3);
988a28cd43dSSascha Wildner int const gain1 = (int)(matchLength*3 - ZSTD_highbit32((U32)offset+1) + 1);
989a28cd43dSSascha Wildner if ((mlRep >= 4) && (gain2 > gain1))
990a28cd43dSSascha Wildner matchLength = mlRep, offset = 0, start = ip;
991a28cd43dSSascha Wildner }
992a28cd43dSSascha Wildner if (isDxS) {
993a28cd43dSSascha Wildner const U32 repIndex = (U32)(ip - base) - offset_1;
994a28cd43dSSascha Wildner const BYTE* repMatch = repIndex < prefixLowestIndex ?
995a28cd43dSSascha Wildner dictBase + (repIndex - dictIndexDelta) :
996a28cd43dSSascha Wildner base + repIndex;
997a28cd43dSSascha Wildner if (((U32)((prefixLowestIndex-1) - repIndex) >= 3 /* intentional underflow */)
998a28cd43dSSascha Wildner && (MEM_read32(repMatch) == MEM_read32(ip)) ) {
999a28cd43dSSascha Wildner const BYTE* repMatchEnd = repIndex < prefixLowestIndex ? dictEnd : iend;
1000a28cd43dSSascha Wildner size_t const mlRep = ZSTD_count_2segments(ip+4, repMatch+4, iend, repMatchEnd, prefixLowest) + 4;
1001a28cd43dSSascha Wildner int const gain2 = (int)(mlRep * 3);
1002a28cd43dSSascha Wildner int const gain1 = (int)(matchLength*3 - ZSTD_highbit32((U32)offset+1) + 1);
1003a28cd43dSSascha Wildner if ((mlRep >= 4) && (gain2 > gain1))
1004a28cd43dSSascha Wildner matchLength = mlRep, offset = 0, start = ip;
1005a28cd43dSSascha Wildner }
1006a28cd43dSSascha Wildner }
1007a28cd43dSSascha Wildner { size_t offset2=999999999;
1008a28cd43dSSascha Wildner size_t const ml2 = searchMax(ms, ip, iend, &offset2);
1009a28cd43dSSascha Wildner int const gain2 = (int)(ml2*4 - ZSTD_highbit32((U32)offset2+1)); /* raw approx */
1010a28cd43dSSascha Wildner int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offset+1) + 4);
1011a28cd43dSSascha Wildner if ((ml2 >= 4) && (gain2 > gain1)) {
1012a28cd43dSSascha Wildner matchLength = ml2, offset = offset2, start = ip;
1013a28cd43dSSascha Wildner continue; /* search a better one */
1014a28cd43dSSascha Wildner } }
1015a28cd43dSSascha Wildner
1016a28cd43dSSascha Wildner /* let's find an even better one */
1017a28cd43dSSascha Wildner if ((depth==2) && (ip<ilimit)) {
1018a28cd43dSSascha Wildner ip ++;
1019a28cd43dSSascha Wildner if ( (dictMode == ZSTD_noDict)
1020a28cd43dSSascha Wildner && (offset) && ((offset_1>0) & (MEM_read32(ip) == MEM_read32(ip - offset_1)))) {
1021a28cd43dSSascha Wildner size_t const mlRep = ZSTD_count(ip+4, ip+4-offset_1, iend) + 4;
1022a28cd43dSSascha Wildner int const gain2 = (int)(mlRep * 4);
1023a28cd43dSSascha Wildner int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offset+1) + 1);
1024a28cd43dSSascha Wildner if ((mlRep >= 4) && (gain2 > gain1))
1025a28cd43dSSascha Wildner matchLength = mlRep, offset = 0, start = ip;
1026a28cd43dSSascha Wildner }
1027a28cd43dSSascha Wildner if (isDxS) {
1028a28cd43dSSascha Wildner const U32 repIndex = (U32)(ip - base) - offset_1;
1029a28cd43dSSascha Wildner const BYTE* repMatch = repIndex < prefixLowestIndex ?
1030a28cd43dSSascha Wildner dictBase + (repIndex - dictIndexDelta) :
1031a28cd43dSSascha Wildner base + repIndex;
1032a28cd43dSSascha Wildner if (((U32)((prefixLowestIndex-1) - repIndex) >= 3 /* intentional underflow */)
1033a28cd43dSSascha Wildner && (MEM_read32(repMatch) == MEM_read32(ip)) ) {
1034a28cd43dSSascha Wildner const BYTE* repMatchEnd = repIndex < prefixLowestIndex ? dictEnd : iend;
1035a28cd43dSSascha Wildner size_t const mlRep = ZSTD_count_2segments(ip+4, repMatch+4, iend, repMatchEnd, prefixLowest) + 4;
1036a28cd43dSSascha Wildner int const gain2 = (int)(mlRep * 4);
1037a28cd43dSSascha Wildner int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offset+1) + 1);
1038a28cd43dSSascha Wildner if ((mlRep >= 4) && (gain2 > gain1))
1039a28cd43dSSascha Wildner matchLength = mlRep, offset = 0, start = ip;
1040a28cd43dSSascha Wildner }
1041a28cd43dSSascha Wildner }
1042a28cd43dSSascha Wildner { size_t offset2=999999999;
1043a28cd43dSSascha Wildner size_t const ml2 = searchMax(ms, ip, iend, &offset2);
1044a28cd43dSSascha Wildner int const gain2 = (int)(ml2*4 - ZSTD_highbit32((U32)offset2+1)); /* raw approx */
1045a28cd43dSSascha Wildner int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offset+1) + 7);
1046a28cd43dSSascha Wildner if ((ml2 >= 4) && (gain2 > gain1)) {
1047a28cd43dSSascha Wildner matchLength = ml2, offset = offset2, start = ip;
1048a28cd43dSSascha Wildner continue;
1049a28cd43dSSascha Wildner } } }
1050a28cd43dSSascha Wildner break; /* nothing found : store previous solution */
1051a28cd43dSSascha Wildner }
1052a28cd43dSSascha Wildner
1053a28cd43dSSascha Wildner /* NOTE:
1054a28cd43dSSascha Wildner * start[-offset+ZSTD_REP_MOVE-1] is undefined behavior.
1055a28cd43dSSascha Wildner * (-offset+ZSTD_REP_MOVE-1) is unsigned, and is added to start, which
1056a28cd43dSSascha Wildner * overflows the pointer, which is undefined behavior.
1057a28cd43dSSascha Wildner */
1058a28cd43dSSascha Wildner /* catch up */
1059a28cd43dSSascha Wildner if (offset) {
1060a28cd43dSSascha Wildner if (dictMode == ZSTD_noDict) {
1061a28cd43dSSascha Wildner while ( ((start > anchor) & (start - (offset-ZSTD_REP_MOVE) > prefixLowest))
1062a28cd43dSSascha Wildner && (start[-1] == (start-(offset-ZSTD_REP_MOVE))[-1]) ) /* only search for offset within prefix */
1063a28cd43dSSascha Wildner { start--; matchLength++; }
1064a28cd43dSSascha Wildner }
1065a28cd43dSSascha Wildner if (isDxS) {
1066a28cd43dSSascha Wildner U32 const matchIndex = (U32)((start-base) - (offset - ZSTD_REP_MOVE));
1067a28cd43dSSascha Wildner const BYTE* match = (matchIndex < prefixLowestIndex) ? dictBase + matchIndex - dictIndexDelta : base + matchIndex;
1068a28cd43dSSascha Wildner const BYTE* const mStart = (matchIndex < prefixLowestIndex) ? dictLowest : prefixLowest;
1069a28cd43dSSascha Wildner while ((start>anchor) && (match>mStart) && (start[-1] == match[-1])) { start--; match--; matchLength++; } /* catch up */
1070a28cd43dSSascha Wildner }
1071a28cd43dSSascha Wildner offset_2 = offset_1; offset_1 = (U32)(offset - ZSTD_REP_MOVE);
1072a28cd43dSSascha Wildner }
1073a28cd43dSSascha Wildner /* store sequence */
1074a28cd43dSSascha Wildner _storeSequence:
1075a28cd43dSSascha Wildner { size_t const litLength = start - anchor;
1076a28cd43dSSascha Wildner ZSTD_storeSeq(seqStore, litLength, anchor, iend, (U32)offset, matchLength-MINMATCH);
1077a28cd43dSSascha Wildner anchor = ip = start + matchLength;
1078a28cd43dSSascha Wildner }
1079a28cd43dSSascha Wildner
1080a28cd43dSSascha Wildner /* check immediate repcode */
1081a28cd43dSSascha Wildner if (isDxS) {
1082a28cd43dSSascha Wildner while (ip <= ilimit) {
1083a28cd43dSSascha Wildner U32 const current2 = (U32)(ip-base);
1084a28cd43dSSascha Wildner U32 const repIndex = current2 - offset_2;
1085a28cd43dSSascha Wildner const BYTE* repMatch = repIndex < prefixLowestIndex ?
1086a28cd43dSSascha Wildner dictBase - dictIndexDelta + repIndex :
1087a28cd43dSSascha Wildner base + repIndex;
1088a28cd43dSSascha Wildner if ( ((U32)((prefixLowestIndex-1) - (U32)repIndex) >= 3 /* intentional overflow */)
1089a28cd43dSSascha Wildner && (MEM_read32(repMatch) == MEM_read32(ip)) ) {
1090a28cd43dSSascha Wildner const BYTE* const repEnd2 = repIndex < prefixLowestIndex ? dictEnd : iend;
1091a28cd43dSSascha Wildner matchLength = ZSTD_count_2segments(ip+4, repMatch+4, iend, repEnd2, prefixLowest) + 4;
1092a28cd43dSSascha Wildner offset = offset_2; offset_2 = offset_1; offset_1 = (U32)offset; /* swap offset_2 <=> offset_1 */
1093a28cd43dSSascha Wildner ZSTD_storeSeq(seqStore, 0, anchor, iend, 0, matchLength-MINMATCH);
1094a28cd43dSSascha Wildner ip += matchLength;
1095a28cd43dSSascha Wildner anchor = ip;
1096a28cd43dSSascha Wildner continue;
1097a28cd43dSSascha Wildner }
1098a28cd43dSSascha Wildner break;
1099a28cd43dSSascha Wildner }
1100a28cd43dSSascha Wildner }
1101a28cd43dSSascha Wildner
1102a28cd43dSSascha Wildner if (dictMode == ZSTD_noDict) {
1103a28cd43dSSascha Wildner while ( ((ip <= ilimit) & (offset_2>0))
1104a28cd43dSSascha Wildner && (MEM_read32(ip) == MEM_read32(ip - offset_2)) ) {
1105a28cd43dSSascha Wildner /* store sequence */
1106a28cd43dSSascha Wildner matchLength = ZSTD_count(ip+4, ip+4-offset_2, iend) + 4;
1107a28cd43dSSascha Wildner offset = offset_2; offset_2 = offset_1; offset_1 = (U32)offset; /* swap repcodes */
1108a28cd43dSSascha Wildner ZSTD_storeSeq(seqStore, 0, anchor, iend, 0, matchLength-MINMATCH);
1109a28cd43dSSascha Wildner ip += matchLength;
1110a28cd43dSSascha Wildner anchor = ip;
1111a28cd43dSSascha Wildner continue; /* faster when present ... (?) */
1112a28cd43dSSascha Wildner } } }
1113a28cd43dSSascha Wildner
1114a28cd43dSSascha Wildner /* Save reps for next block */
1115a28cd43dSSascha Wildner rep[0] = offset_1 ? offset_1 : savedOffset;
1116a28cd43dSSascha Wildner rep[1] = offset_2 ? offset_2 : savedOffset;
1117a28cd43dSSascha Wildner
1118a28cd43dSSascha Wildner /* Return the last literals size */
1119a28cd43dSSascha Wildner return (size_t)(iend - anchor);
1120a28cd43dSSascha Wildner }
1121a28cd43dSSascha Wildner
1122a28cd43dSSascha Wildner
ZSTD_compressBlock_btlazy2(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1123a28cd43dSSascha Wildner size_t ZSTD_compressBlock_btlazy2(
1124a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1125a28cd43dSSascha Wildner void const* src, size_t srcSize)
1126a28cd43dSSascha Wildner {
1127a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_binaryTree, 2, ZSTD_noDict);
1128a28cd43dSSascha Wildner }
1129a28cd43dSSascha Wildner
ZSTD_compressBlock_lazy2(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1130a28cd43dSSascha Wildner size_t ZSTD_compressBlock_lazy2(
1131a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1132a28cd43dSSascha Wildner void const* src, size_t srcSize)
1133a28cd43dSSascha Wildner {
1134a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 2, ZSTD_noDict);
1135a28cd43dSSascha Wildner }
1136a28cd43dSSascha Wildner
ZSTD_compressBlock_lazy(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1137a28cd43dSSascha Wildner size_t ZSTD_compressBlock_lazy(
1138a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1139a28cd43dSSascha Wildner void const* src, size_t srcSize)
1140a28cd43dSSascha Wildner {
1141a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 1, ZSTD_noDict);
1142a28cd43dSSascha Wildner }
1143a28cd43dSSascha Wildner
ZSTD_compressBlock_greedy(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1144a28cd43dSSascha Wildner size_t ZSTD_compressBlock_greedy(
1145a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1146a28cd43dSSascha Wildner void const* src, size_t srcSize)
1147a28cd43dSSascha Wildner {
1148a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 0, ZSTD_noDict);
1149a28cd43dSSascha Wildner }
1150a28cd43dSSascha Wildner
ZSTD_compressBlock_btlazy2_dictMatchState(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1151a28cd43dSSascha Wildner size_t ZSTD_compressBlock_btlazy2_dictMatchState(
1152a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1153a28cd43dSSascha Wildner void const* src, size_t srcSize)
1154a28cd43dSSascha Wildner {
1155a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_binaryTree, 2, ZSTD_dictMatchState);
1156a28cd43dSSascha Wildner }
1157a28cd43dSSascha Wildner
ZSTD_compressBlock_lazy2_dictMatchState(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1158a28cd43dSSascha Wildner size_t ZSTD_compressBlock_lazy2_dictMatchState(
1159a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1160a28cd43dSSascha Wildner void const* src, size_t srcSize)
1161a28cd43dSSascha Wildner {
1162a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 2, ZSTD_dictMatchState);
1163a28cd43dSSascha Wildner }
1164a28cd43dSSascha Wildner
ZSTD_compressBlock_lazy_dictMatchState(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1165a28cd43dSSascha Wildner size_t ZSTD_compressBlock_lazy_dictMatchState(
1166a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1167a28cd43dSSascha Wildner void const* src, size_t srcSize)
1168a28cd43dSSascha Wildner {
1169a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 1, ZSTD_dictMatchState);
1170a28cd43dSSascha Wildner }
1171a28cd43dSSascha Wildner
ZSTD_compressBlock_greedy_dictMatchState(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1172a28cd43dSSascha Wildner size_t ZSTD_compressBlock_greedy_dictMatchState(
1173a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1174a28cd43dSSascha Wildner void const* src, size_t srcSize)
1175a28cd43dSSascha Wildner {
1176a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 0, ZSTD_dictMatchState);
1177a28cd43dSSascha Wildner }
1178a28cd43dSSascha Wildner
1179a28cd43dSSascha Wildner
ZSTD_compressBlock_lazy2_dedicatedDictSearch(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1180a28cd43dSSascha Wildner size_t ZSTD_compressBlock_lazy2_dedicatedDictSearch(
1181a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1182a28cd43dSSascha Wildner void const* src, size_t srcSize)
1183a28cd43dSSascha Wildner {
1184a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 2, ZSTD_dedicatedDictSearch);
1185a28cd43dSSascha Wildner }
1186a28cd43dSSascha Wildner
ZSTD_compressBlock_lazy_dedicatedDictSearch(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1187a28cd43dSSascha Wildner size_t ZSTD_compressBlock_lazy_dedicatedDictSearch(
1188a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1189a28cd43dSSascha Wildner void const* src, size_t srcSize)
1190a28cd43dSSascha Wildner {
1191a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 1, ZSTD_dedicatedDictSearch);
1192a28cd43dSSascha Wildner }
1193a28cd43dSSascha Wildner
ZSTD_compressBlock_greedy_dedicatedDictSearch(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1194a28cd43dSSascha Wildner size_t ZSTD_compressBlock_greedy_dedicatedDictSearch(
1195a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1196a28cd43dSSascha Wildner void const* src, size_t srcSize)
1197a28cd43dSSascha Wildner {
1198a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 0, ZSTD_dedicatedDictSearch);
1199a28cd43dSSascha Wildner }
1200a28cd43dSSascha Wildner
1201a28cd43dSSascha Wildner
1202a28cd43dSSascha Wildner FORCE_INLINE_TEMPLATE
ZSTD_compressBlock_lazy_extDict_generic(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],const void * src,size_t srcSize,const searchMethod_e searchMethod,const U32 depth)1203a28cd43dSSascha Wildner size_t ZSTD_compressBlock_lazy_extDict_generic(
1204a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore,
1205a28cd43dSSascha Wildner U32 rep[ZSTD_REP_NUM],
1206a28cd43dSSascha Wildner const void* src, size_t srcSize,
1207a28cd43dSSascha Wildner const searchMethod_e searchMethod, const U32 depth)
1208a28cd43dSSascha Wildner {
1209a28cd43dSSascha Wildner const BYTE* const istart = (const BYTE*)src;
1210a28cd43dSSascha Wildner const BYTE* ip = istart;
1211a28cd43dSSascha Wildner const BYTE* anchor = istart;
1212a28cd43dSSascha Wildner const BYTE* const iend = istart + srcSize;
1213a28cd43dSSascha Wildner const BYTE* const ilimit = iend - 8;
1214a28cd43dSSascha Wildner const BYTE* const base = ms->window.base;
1215a28cd43dSSascha Wildner const U32 dictLimit = ms->window.dictLimit;
1216a28cd43dSSascha Wildner const BYTE* const prefixStart = base + dictLimit;
1217a28cd43dSSascha Wildner const BYTE* const dictBase = ms->window.dictBase;
1218a28cd43dSSascha Wildner const BYTE* const dictEnd = dictBase + dictLimit;
1219a28cd43dSSascha Wildner const BYTE* const dictStart = dictBase + ms->window.lowLimit;
1220a28cd43dSSascha Wildner const U32 windowLog = ms->cParams.windowLog;
1221a28cd43dSSascha Wildner
1222a28cd43dSSascha Wildner typedef size_t (*searchMax_f)(
1223a28cd43dSSascha Wildner ZSTD_matchState_t* ms,
1224a28cd43dSSascha Wildner const BYTE* ip, const BYTE* iLimit, size_t* offsetPtr);
1225a28cd43dSSascha Wildner searchMax_f searchMax = searchMethod==search_binaryTree ? ZSTD_BtFindBestMatch_extDict_selectMLS : ZSTD_HcFindBestMatch_extDict_selectMLS;
1226a28cd43dSSascha Wildner
1227a28cd43dSSascha Wildner U32 offset_1 = rep[0], offset_2 = rep[1];
1228a28cd43dSSascha Wildner
1229a28cd43dSSascha Wildner DEBUGLOG(5, "ZSTD_compressBlock_lazy_extDict_generic");
1230a28cd43dSSascha Wildner
1231a28cd43dSSascha Wildner /* init */
1232a28cd43dSSascha Wildner ip += (ip == prefixStart);
1233a28cd43dSSascha Wildner
1234a28cd43dSSascha Wildner /* Match Loop */
1235a28cd43dSSascha Wildner #if defined(__GNUC__) && defined(__x86_64__)
1236a28cd43dSSascha Wildner /* I've measured random a 5% speed loss on levels 5 & 6 (greedy) when the
1237a28cd43dSSascha Wildner * code alignment is perturbed. To fix the instability align the loop on 32-bytes.
1238a28cd43dSSascha Wildner */
1239a28cd43dSSascha Wildner __asm__(".p2align 5");
1240a28cd43dSSascha Wildner #endif
1241a28cd43dSSascha Wildner while (ip < ilimit) {
1242a28cd43dSSascha Wildner size_t matchLength=0;
1243a28cd43dSSascha Wildner size_t offset=0;
1244a28cd43dSSascha Wildner const BYTE* start=ip+1;
1245a28cd43dSSascha Wildner U32 curr = (U32)(ip-base);
1246a28cd43dSSascha Wildner
1247a28cd43dSSascha Wildner /* check repCode */
1248a28cd43dSSascha Wildner { const U32 windowLow = ZSTD_getLowestMatchIndex(ms, curr+1, windowLog);
1249a28cd43dSSascha Wildner const U32 repIndex = (U32)(curr+1 - offset_1);
1250a28cd43dSSascha Wildner const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
1251a28cd43dSSascha Wildner const BYTE* const repMatch = repBase + repIndex;
1252a28cd43dSSascha Wildner if (((U32)((dictLimit-1) - repIndex) >= 3) & (repIndex > windowLow)) /* intentional overflow */
1253a28cd43dSSascha Wildner if (MEM_read32(ip+1) == MEM_read32(repMatch)) {
1254a28cd43dSSascha Wildner /* repcode detected we should take it */
1255a28cd43dSSascha Wildner const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
1256a28cd43dSSascha Wildner matchLength = ZSTD_count_2segments(ip+1+4, repMatch+4, iend, repEnd, prefixStart) + 4;
1257a28cd43dSSascha Wildner if (depth==0) goto _storeSequence;
1258a28cd43dSSascha Wildner } }
1259a28cd43dSSascha Wildner
1260a28cd43dSSascha Wildner /* first search (depth 0) */
1261a28cd43dSSascha Wildner { size_t offsetFound = 999999999;
1262a28cd43dSSascha Wildner size_t const ml2 = searchMax(ms, ip, iend, &offsetFound);
1263a28cd43dSSascha Wildner if (ml2 > matchLength)
1264a28cd43dSSascha Wildner matchLength = ml2, start = ip, offset=offsetFound;
1265a28cd43dSSascha Wildner }
1266a28cd43dSSascha Wildner
1267a28cd43dSSascha Wildner if (matchLength < 4) {
1268a28cd43dSSascha Wildner ip += ((ip-anchor) >> kSearchStrength) + 1; /* jump faster over incompressible sections */
1269a28cd43dSSascha Wildner continue;
1270a28cd43dSSascha Wildner }
1271a28cd43dSSascha Wildner
1272a28cd43dSSascha Wildner /* let's try to find a better solution */
1273a28cd43dSSascha Wildner if (depth>=1)
1274a28cd43dSSascha Wildner while (ip<ilimit) {
1275a28cd43dSSascha Wildner ip ++;
1276a28cd43dSSascha Wildner curr++;
1277a28cd43dSSascha Wildner /* check repCode */
1278a28cd43dSSascha Wildner if (offset) {
1279a28cd43dSSascha Wildner const U32 windowLow = ZSTD_getLowestMatchIndex(ms, curr, windowLog);
1280a28cd43dSSascha Wildner const U32 repIndex = (U32)(curr - offset_1);
1281a28cd43dSSascha Wildner const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
1282a28cd43dSSascha Wildner const BYTE* const repMatch = repBase + repIndex;
1283a28cd43dSSascha Wildner if (((U32)((dictLimit-1) - repIndex) >= 3) & (repIndex > windowLow)) /* intentional overflow */
1284a28cd43dSSascha Wildner if (MEM_read32(ip) == MEM_read32(repMatch)) {
1285a28cd43dSSascha Wildner /* repcode detected */
1286a28cd43dSSascha Wildner const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
1287a28cd43dSSascha Wildner size_t const repLength = ZSTD_count_2segments(ip+4, repMatch+4, iend, repEnd, prefixStart) + 4;
1288a28cd43dSSascha Wildner int const gain2 = (int)(repLength * 3);
1289a28cd43dSSascha Wildner int const gain1 = (int)(matchLength*3 - ZSTD_highbit32((U32)offset+1) + 1);
1290a28cd43dSSascha Wildner if ((repLength >= 4) && (gain2 > gain1))
1291a28cd43dSSascha Wildner matchLength = repLength, offset = 0, start = ip;
1292a28cd43dSSascha Wildner } }
1293a28cd43dSSascha Wildner
1294a28cd43dSSascha Wildner /* search match, depth 1 */
1295a28cd43dSSascha Wildner { size_t offset2=999999999;
1296a28cd43dSSascha Wildner size_t const ml2 = searchMax(ms, ip, iend, &offset2);
1297a28cd43dSSascha Wildner int const gain2 = (int)(ml2*4 - ZSTD_highbit32((U32)offset2+1)); /* raw approx */
1298a28cd43dSSascha Wildner int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offset+1) + 4);
1299a28cd43dSSascha Wildner if ((ml2 >= 4) && (gain2 > gain1)) {
1300a28cd43dSSascha Wildner matchLength = ml2, offset = offset2, start = ip;
1301a28cd43dSSascha Wildner continue; /* search a better one */
1302a28cd43dSSascha Wildner } }
1303a28cd43dSSascha Wildner
1304a28cd43dSSascha Wildner /* let's find an even better one */
1305a28cd43dSSascha Wildner if ((depth==2) && (ip<ilimit)) {
1306a28cd43dSSascha Wildner ip ++;
1307a28cd43dSSascha Wildner curr++;
1308a28cd43dSSascha Wildner /* check repCode */
1309a28cd43dSSascha Wildner if (offset) {
1310a28cd43dSSascha Wildner const U32 windowLow = ZSTD_getLowestMatchIndex(ms, curr, windowLog);
1311a28cd43dSSascha Wildner const U32 repIndex = (U32)(curr - offset_1);
1312a28cd43dSSascha Wildner const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
1313a28cd43dSSascha Wildner const BYTE* const repMatch = repBase + repIndex;
1314a28cd43dSSascha Wildner if (((U32)((dictLimit-1) - repIndex) >= 3) & (repIndex > windowLow)) /* intentional overflow */
1315a28cd43dSSascha Wildner if (MEM_read32(ip) == MEM_read32(repMatch)) {
1316a28cd43dSSascha Wildner /* repcode detected */
1317a28cd43dSSascha Wildner const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
1318a28cd43dSSascha Wildner size_t const repLength = ZSTD_count_2segments(ip+4, repMatch+4, iend, repEnd, prefixStart) + 4;
1319a28cd43dSSascha Wildner int const gain2 = (int)(repLength * 4);
1320a28cd43dSSascha Wildner int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offset+1) + 1);
1321a28cd43dSSascha Wildner if ((repLength >= 4) && (gain2 > gain1))
1322a28cd43dSSascha Wildner matchLength = repLength, offset = 0, start = ip;
1323a28cd43dSSascha Wildner } }
1324a28cd43dSSascha Wildner
1325a28cd43dSSascha Wildner /* search match, depth 2 */
1326a28cd43dSSascha Wildner { size_t offset2=999999999;
1327a28cd43dSSascha Wildner size_t const ml2 = searchMax(ms, ip, iend, &offset2);
1328a28cd43dSSascha Wildner int const gain2 = (int)(ml2*4 - ZSTD_highbit32((U32)offset2+1)); /* raw approx */
1329a28cd43dSSascha Wildner int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offset+1) + 7);
1330a28cd43dSSascha Wildner if ((ml2 >= 4) && (gain2 > gain1)) {
1331a28cd43dSSascha Wildner matchLength = ml2, offset = offset2, start = ip;
1332a28cd43dSSascha Wildner continue;
1333a28cd43dSSascha Wildner } } }
1334a28cd43dSSascha Wildner break; /* nothing found : store previous solution */
1335a28cd43dSSascha Wildner }
1336a28cd43dSSascha Wildner
1337a28cd43dSSascha Wildner /* catch up */
1338a28cd43dSSascha Wildner if (offset) {
1339a28cd43dSSascha Wildner U32 const matchIndex = (U32)((start-base) - (offset - ZSTD_REP_MOVE));
1340a28cd43dSSascha Wildner const BYTE* match = (matchIndex < dictLimit) ? dictBase + matchIndex : base + matchIndex;
1341a28cd43dSSascha Wildner const BYTE* const mStart = (matchIndex < dictLimit) ? dictStart : prefixStart;
1342a28cd43dSSascha Wildner while ((start>anchor) && (match>mStart) && (start[-1] == match[-1])) { start--; match--; matchLength++; } /* catch up */
1343a28cd43dSSascha Wildner offset_2 = offset_1; offset_1 = (U32)(offset - ZSTD_REP_MOVE);
1344a28cd43dSSascha Wildner }
1345a28cd43dSSascha Wildner
1346a28cd43dSSascha Wildner /* store sequence */
1347a28cd43dSSascha Wildner _storeSequence:
1348a28cd43dSSascha Wildner { size_t const litLength = start - anchor;
1349a28cd43dSSascha Wildner ZSTD_storeSeq(seqStore, litLength, anchor, iend, (U32)offset, matchLength-MINMATCH);
1350a28cd43dSSascha Wildner anchor = ip = start + matchLength;
1351a28cd43dSSascha Wildner }
1352a28cd43dSSascha Wildner
1353a28cd43dSSascha Wildner /* check immediate repcode */
1354a28cd43dSSascha Wildner while (ip <= ilimit) {
1355a28cd43dSSascha Wildner const U32 repCurrent = (U32)(ip-base);
1356a28cd43dSSascha Wildner const U32 windowLow = ZSTD_getLowestMatchIndex(ms, repCurrent, windowLog);
1357a28cd43dSSascha Wildner const U32 repIndex = repCurrent - offset_2;
1358a28cd43dSSascha Wildner const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
1359a28cd43dSSascha Wildner const BYTE* const repMatch = repBase + repIndex;
1360a28cd43dSSascha Wildner if (((U32)((dictLimit-1) - repIndex) >= 3) & (repIndex > windowLow)) /* intentional overflow */
1361a28cd43dSSascha Wildner if (MEM_read32(ip) == MEM_read32(repMatch)) {
1362a28cd43dSSascha Wildner /* repcode detected we should take it */
1363a28cd43dSSascha Wildner const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
1364a28cd43dSSascha Wildner matchLength = ZSTD_count_2segments(ip+4, repMatch+4, iend, repEnd, prefixStart) + 4;
1365a28cd43dSSascha Wildner offset = offset_2; offset_2 = offset_1; offset_1 = (U32)offset; /* swap offset history */
1366a28cd43dSSascha Wildner ZSTD_storeSeq(seqStore, 0, anchor, iend, 0, matchLength-MINMATCH);
1367a28cd43dSSascha Wildner ip += matchLength;
1368a28cd43dSSascha Wildner anchor = ip;
1369a28cd43dSSascha Wildner continue; /* faster when present ... (?) */
1370a28cd43dSSascha Wildner }
1371a28cd43dSSascha Wildner break;
1372a28cd43dSSascha Wildner } }
1373a28cd43dSSascha Wildner
1374a28cd43dSSascha Wildner /* Save reps for next block */
1375a28cd43dSSascha Wildner rep[0] = offset_1;
1376a28cd43dSSascha Wildner rep[1] = offset_2;
1377a28cd43dSSascha Wildner
1378a28cd43dSSascha Wildner /* Return the last literals size */
1379a28cd43dSSascha Wildner return (size_t)(iend - anchor);
1380a28cd43dSSascha Wildner }
1381a28cd43dSSascha Wildner
1382a28cd43dSSascha Wildner
ZSTD_compressBlock_greedy_extDict(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1383a28cd43dSSascha Wildner size_t ZSTD_compressBlock_greedy_extDict(
1384a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1385a28cd43dSSascha Wildner void const* src, size_t srcSize)
1386a28cd43dSSascha Wildner {
1387a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 0);
1388a28cd43dSSascha Wildner }
1389a28cd43dSSascha Wildner
ZSTD_compressBlock_lazy_extDict(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1390a28cd43dSSascha Wildner size_t ZSTD_compressBlock_lazy_extDict(
1391a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1392a28cd43dSSascha Wildner void const* src, size_t srcSize)
1393a28cd43dSSascha Wildner
1394a28cd43dSSascha Wildner {
1395a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 1);
1396a28cd43dSSascha Wildner }
1397a28cd43dSSascha Wildner
ZSTD_compressBlock_lazy2_extDict(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1398a28cd43dSSascha Wildner size_t ZSTD_compressBlock_lazy2_extDict(
1399a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1400a28cd43dSSascha Wildner void const* src, size_t srcSize)
1401a28cd43dSSascha Wildner
1402a28cd43dSSascha Wildner {
1403a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 2);
1404a28cd43dSSascha Wildner }
1405a28cd43dSSascha Wildner
ZSTD_compressBlock_btlazy2_extDict(ZSTD_matchState_t * ms,seqStore_t * seqStore,U32 rep[ZSTD_REP_NUM],void const * src,size_t srcSize)1406a28cd43dSSascha Wildner size_t ZSTD_compressBlock_btlazy2_extDict(
1407a28cd43dSSascha Wildner ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1408a28cd43dSSascha Wildner void const* src, size_t srcSize)
1409a28cd43dSSascha Wildner
1410a28cd43dSSascha Wildner {
1411a28cd43dSSascha Wildner return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, search_binaryTree, 2);
1412a28cd43dSSascha Wildner }
1413