zstdlib 0.13.0-x86_64-linux

Sign up to get free protection for your applications and to get access to all the features.
Files changed (129) hide show
  1. checksums.yaml +7 -0
  2. data/.yardopts +6 -0
  3. data/CHANGES.md +107 -0
  4. data/Gemfile +3 -0
  5. data/README.md +107 -0
  6. data/Rakefile +59 -0
  7. data/ext/zstdlib_c/extconf.rb +59 -0
  8. data/ext/zstdlib_c/ruby/zlib-2.2/zstdlib.c +4675 -0
  9. data/ext/zstdlib_c/ruby/zlib-2.3/zstdlib.c +4702 -0
  10. data/ext/zstdlib_c/ruby/zlib-2.4/zstdlib.c +4859 -0
  11. data/ext/zstdlib_c/ruby/zlib-2.5/zstdlib.c +4864 -0
  12. data/ext/zstdlib_c/ruby/zlib-2.6/zstdlib.c +4906 -0
  13. data/ext/zstdlib_c/ruby/zlib-2.7/zstdlib.c +4895 -0
  14. data/ext/zstdlib_c/ruby/zlib-3.0/zstdlib.c +4994 -0
  15. data/ext/zstdlib_c/ruby/zlib-3.1/zstdlib.c +5076 -0
  16. data/ext/zstdlib_c/ruby/zlib-3.2/zstdlib.c +5090 -0
  17. data/ext/zstdlib_c/ruby/zlib-3.3/zstdlib.c +5090 -0
  18. data/ext/zstdlib_c/zlib-1.3.1/adler32.c +164 -0
  19. data/ext/zstdlib_c/zlib-1.3.1/compress.c +75 -0
  20. data/ext/zstdlib_c/zlib-1.3.1/crc32.c +1049 -0
  21. data/ext/zstdlib_c/zlib-1.3.1/crc32.h +9446 -0
  22. data/ext/zstdlib_c/zlib-1.3.1/deflate.c +2139 -0
  23. data/ext/zstdlib_c/zlib-1.3.1/deflate.h +377 -0
  24. data/ext/zstdlib_c/zlib-1.3.1/gzclose.c +23 -0
  25. data/ext/zstdlib_c/zlib-1.3.1/gzguts.h +214 -0
  26. data/ext/zstdlib_c/zlib-1.3.1/gzlib.c +582 -0
  27. data/ext/zstdlib_c/zlib-1.3.1/gzread.c +602 -0
  28. data/ext/zstdlib_c/zlib-1.3.1/gzwrite.c +631 -0
  29. data/ext/zstdlib_c/zlib-1.3.1/infback.c +628 -0
  30. data/ext/zstdlib_c/zlib-1.3.1/inffast.c +320 -0
  31. data/ext/zstdlib_c/zlib-1.3.1/inffast.h +11 -0
  32. data/ext/zstdlib_c/zlib-1.3.1/inffixed.h +94 -0
  33. data/ext/zstdlib_c/zlib-1.3.1/inflate.c +1526 -0
  34. data/ext/zstdlib_c/zlib-1.3.1/inflate.h +126 -0
  35. data/ext/zstdlib_c/zlib-1.3.1/inftrees.c +299 -0
  36. data/ext/zstdlib_c/zlib-1.3.1/inftrees.h +62 -0
  37. data/ext/zstdlib_c/zlib-1.3.1/trees.c +1117 -0
  38. data/ext/zstdlib_c/zlib-1.3.1/trees.h +128 -0
  39. data/ext/zstdlib_c/zlib-1.3.1/uncompr.c +85 -0
  40. data/ext/zstdlib_c/zlib-1.3.1/zconf.h +543 -0
  41. data/ext/zstdlib_c/zlib-1.3.1/zlib.h +1938 -0
  42. data/ext/zstdlib_c/zlib-1.3.1/zutil.c +299 -0
  43. data/ext/zstdlib_c/zlib-1.3.1/zutil.h +254 -0
  44. data/ext/zstdlib_c/zlib.mk +14 -0
  45. data/ext/zstdlib_c/zlibwrapper/zlibwrapper.c +10 -0
  46. data/ext/zstdlib_c/zlibwrapper.mk +14 -0
  47. data/ext/zstdlib_c/zstd-1.5.6/lib/common/allocations.h +55 -0
  48. data/ext/zstdlib_c/zstd-1.5.6/lib/common/bits.h +200 -0
  49. data/ext/zstdlib_c/zstd-1.5.6/lib/common/bitstream.h +457 -0
  50. data/ext/zstdlib_c/zstd-1.5.6/lib/common/compiler.h +450 -0
  51. data/ext/zstdlib_c/zstd-1.5.6/lib/common/cpu.h +249 -0
  52. data/ext/zstdlib_c/zstd-1.5.6/lib/common/debug.c +30 -0
  53. data/ext/zstdlib_c/zstd-1.5.6/lib/common/debug.h +116 -0
  54. data/ext/zstdlib_c/zstd-1.5.6/lib/common/entropy_common.c +340 -0
  55. data/ext/zstdlib_c/zstd-1.5.6/lib/common/error_private.c +63 -0
  56. data/ext/zstdlib_c/zstd-1.5.6/lib/common/error_private.h +168 -0
  57. data/ext/zstdlib_c/zstd-1.5.6/lib/common/fse.h +640 -0
  58. data/ext/zstdlib_c/zstd-1.5.6/lib/common/fse_decompress.c +313 -0
  59. data/ext/zstdlib_c/zstd-1.5.6/lib/common/huf.h +286 -0
  60. data/ext/zstdlib_c/zstd-1.5.6/lib/common/mem.h +426 -0
  61. data/ext/zstdlib_c/zstd-1.5.6/lib/common/pool.c +371 -0
  62. data/ext/zstdlib_c/zstd-1.5.6/lib/common/pool.h +90 -0
  63. data/ext/zstdlib_c/zstd-1.5.6/lib/common/portability_macros.h +158 -0
  64. data/ext/zstdlib_c/zstd-1.5.6/lib/common/threading.c +182 -0
  65. data/ext/zstdlib_c/zstd-1.5.6/lib/common/threading.h +150 -0
  66. data/ext/zstdlib_c/zstd-1.5.6/lib/common/xxhash.c +18 -0
  67. data/ext/zstdlib_c/zstd-1.5.6/lib/common/xxhash.h +7020 -0
  68. data/ext/zstdlib_c/zstd-1.5.6/lib/common/zstd_common.c +48 -0
  69. data/ext/zstdlib_c/zstd-1.5.6/lib/common/zstd_deps.h +111 -0
  70. data/ext/zstdlib_c/zstd-1.5.6/lib/common/zstd_internal.h +392 -0
  71. data/ext/zstdlib_c/zstd-1.5.6/lib/common/zstd_trace.h +163 -0
  72. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/clevels.h +134 -0
  73. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/fse_compress.c +625 -0
  74. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/hist.c +181 -0
  75. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/hist.h +75 -0
  76. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/huf_compress.c +1464 -0
  77. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_compress.c +7153 -0
  78. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_compress_internal.h +1534 -0
  79. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_compress_literals.c +235 -0
  80. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_compress_literals.h +39 -0
  81. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_compress_sequences.c +442 -0
  82. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_compress_sequences.h +54 -0
  83. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_compress_superblock.c +688 -0
  84. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_compress_superblock.h +32 -0
  85. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_cwksp.h +748 -0
  86. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_double_fast.c +770 -0
  87. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_double_fast.h +50 -0
  88. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_fast.c +968 -0
  89. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_fast.h +38 -0
  90. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_lazy.c +2199 -0
  91. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_lazy.h +202 -0
  92. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_ldm.c +730 -0
  93. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_ldm.h +117 -0
  94. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_ldm_geartab.h +106 -0
  95. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_opt.c +1576 -0
  96. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstd_opt.h +80 -0
  97. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstdmt_compress.c +1882 -0
  98. data/ext/zstdlib_c/zstd-1.5.6/lib/compress/zstdmt_compress.h +113 -0
  99. data/ext/zstdlib_c/zstd-1.5.6/lib/decompress/huf_decompress.c +1944 -0
  100. data/ext/zstdlib_c/zstd-1.5.6/lib/decompress/huf_decompress_amd64.S +595 -0
  101. data/ext/zstdlib_c/zstd-1.5.6/lib/decompress/zstd_ddict.c +244 -0
  102. data/ext/zstdlib_c/zstd-1.5.6/lib/decompress/zstd_ddict.h +44 -0
  103. data/ext/zstdlib_c/zstd-1.5.6/lib/decompress/zstd_decompress.c +2407 -0
  104. data/ext/zstdlib_c/zstd-1.5.6/lib/decompress/zstd_decompress_block.c +2215 -0
  105. data/ext/zstdlib_c/zstd-1.5.6/lib/decompress/zstd_decompress_block.h +73 -0
  106. data/ext/zstdlib_c/zstd-1.5.6/lib/decompress/zstd_decompress_internal.h +240 -0
  107. data/ext/zstdlib_c/zstd-1.5.6/lib/zdict.h +474 -0
  108. data/ext/zstdlib_c/zstd-1.5.6/lib/zstd.h +3089 -0
  109. data/ext/zstdlib_c/zstd-1.5.6/lib/zstd_errors.h +114 -0
  110. data/ext/zstdlib_c/zstd-1.5.6/zlibWrapper/gzclose.c +26 -0
  111. data/ext/zstdlib_c/zstd-1.5.6/zlibWrapper/gzcompatibility.h +68 -0
  112. data/ext/zstdlib_c/zstd-1.5.6/zlibWrapper/gzguts.h +229 -0
  113. data/ext/zstdlib_c/zstd-1.5.6/zlibWrapper/gzlib.c +587 -0
  114. data/ext/zstdlib_c/zstd-1.5.6/zlibWrapper/gzread.c +637 -0
  115. data/ext/zstdlib_c/zstd-1.5.6/zlibWrapper/gzwrite.c +631 -0
  116. data/ext/zstdlib_c/zstd-1.5.6/zlibWrapper/zstd_zlibwrapper.c +1200 -0
  117. data/ext/zstdlib_c/zstd-1.5.6/zlibWrapper/zstd_zlibwrapper.h +91 -0
  118. data/ext/zstdlib_c/zstd.mk +15 -0
  119. data/lib/2.4/zstdlib_c.so +0 -0
  120. data/lib/2.5/zstdlib_c.so +0 -0
  121. data/lib/2.6/zstdlib_c.so +0 -0
  122. data/lib/2.7/zstdlib_c.so +0 -0
  123. data/lib/3.0/zstdlib_c.so +0 -0
  124. data/lib/3.1/zstdlib_c.so +0 -0
  125. data/lib/3.2/zstdlib_c.so +0 -0
  126. data/lib/3.3/zstdlib_c.so +0 -0
  127. data/lib/zstdlib.rb +6 -0
  128. data/test/zstdlib_test.rb +21 -0
  129. metadata +243 -0
@@ -0,0 +1,2199 @@
1
+ /*
2
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
3
+ * All rights reserved.
4
+ *
5
+ * This source code is licensed under both the BSD-style license (found in the
6
+ * LICENSE file in the root directory of this source tree) and the GPLv2 (found
7
+ * in the COPYING file in the root directory of this source tree).
8
+ * You may select, at your option, one of the above-listed licenses.
9
+ */
10
+
11
+ #include "zstd_compress_internal.h"
12
+ #include "zstd_lazy.h"
13
+ #include "../common/bits.h" /* ZSTD_countTrailingZeros64 */
14
+
15
+ #if !defined(ZSTD_EXCLUDE_GREEDY_BLOCK_COMPRESSOR) \
16
+ || !defined(ZSTD_EXCLUDE_LAZY_BLOCK_COMPRESSOR) \
17
+ || !defined(ZSTD_EXCLUDE_LAZY2_BLOCK_COMPRESSOR) \
18
+ || !defined(ZSTD_EXCLUDE_BTLAZY2_BLOCK_COMPRESSOR)
19
+
20
+ #define kLazySkippingStep 8
21
+
22
+
23
+ /*-*************************************
24
+ * Binary Tree search
25
+ ***************************************/
26
+
27
+ static
28
+ ZSTD_ALLOW_POINTER_OVERFLOW_ATTR
29
+ void ZSTD_updateDUBT(ZSTD_matchState_t* ms,
30
+ const BYTE* ip, const BYTE* iend,
31
+ U32 mls)
32
+ {
33
+ const ZSTD_compressionParameters* const cParams = &ms->cParams;
34
+ U32* const hashTable = ms->hashTable;
35
+ U32 const hashLog = cParams->hashLog;
36
+
37
+ U32* const bt = ms->chainTable;
38
+ U32 const btLog = cParams->chainLog - 1;
39
+ U32 const btMask = (1 << btLog) - 1;
40
+
41
+ const BYTE* const base = ms->window.base;
42
+ U32 const target = (U32)(ip - base);
43
+ U32 idx = ms->nextToUpdate;
44
+
45
+ if (idx != target)
46
+ DEBUGLOG(7, "ZSTD_updateDUBT, from %u to %u (dictLimit:%u)",
47
+ idx, target, ms->window.dictLimit);
48
+ assert(ip + 8 <= iend); /* condition for ZSTD_hashPtr */
49
+ (void)iend;
50
+
51
+ assert(idx >= ms->window.dictLimit); /* condition for valid base+idx */
52
+ for ( ; idx < target ; idx++) {
53
+ size_t const h = ZSTD_hashPtr(base + idx, hashLog, mls); /* assumption : ip + 8 <= iend */
54
+ U32 const matchIndex = hashTable[h];
55
+
56
+ U32* const nextCandidatePtr = bt + 2*(idx&btMask);
57
+ U32* const sortMarkPtr = nextCandidatePtr + 1;
58
+
59
+ DEBUGLOG(8, "ZSTD_updateDUBT: insert %u", idx);
60
+ hashTable[h] = idx; /* Update Hash Table */
61
+ *nextCandidatePtr = matchIndex; /* update BT like a chain */
62
+ *sortMarkPtr = ZSTD_DUBT_UNSORTED_MARK;
63
+ }
64
+ ms->nextToUpdate = target;
65
+ }
66
+
67
+
68
+ /** ZSTD_insertDUBT1() :
69
+ * sort one already inserted but unsorted position
70
+ * assumption : curr >= btlow == (curr - btmask)
71
+ * doesn't fail */
72
+ static
73
+ ZSTD_ALLOW_POINTER_OVERFLOW_ATTR
74
+ void ZSTD_insertDUBT1(const ZSTD_matchState_t* ms,
75
+ U32 curr, const BYTE* inputEnd,
76
+ U32 nbCompares, U32 btLow,
77
+ const ZSTD_dictMode_e dictMode)
78
+ {
79
+ const ZSTD_compressionParameters* const cParams = &ms->cParams;
80
+ U32* const bt = ms->chainTable;
81
+ U32 const btLog = cParams->chainLog - 1;
82
+ U32 const btMask = (1 << btLog) - 1;
83
+ size_t commonLengthSmaller=0, commonLengthLarger=0;
84
+ const BYTE* const base = ms->window.base;
85
+ const BYTE* const dictBase = ms->window.dictBase;
86
+ const U32 dictLimit = ms->window.dictLimit;
87
+ const BYTE* const ip = (curr>=dictLimit) ? base + curr : dictBase + curr;
88
+ const BYTE* const iend = (curr>=dictLimit) ? inputEnd : dictBase + dictLimit;
89
+ const BYTE* const dictEnd = dictBase + dictLimit;
90
+ const BYTE* const prefixStart = base + dictLimit;
91
+ const BYTE* match;
92
+ U32* smallerPtr = bt + 2*(curr&btMask);
93
+ U32* largerPtr = smallerPtr + 1;
94
+ 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) */
95
+ U32 dummy32; /* to be nullified at the end */
96
+ U32 const windowValid = ms->window.lowLimit;
97
+ U32 const maxDistance = 1U << cParams->windowLog;
98
+ U32 const windowLow = (curr - windowValid > maxDistance) ? curr - maxDistance : windowValid;
99
+
100
+
101
+ DEBUGLOG(8, "ZSTD_insertDUBT1(%u) (dictLimit=%u, lowLimit=%u)",
102
+ curr, dictLimit, windowLow);
103
+ assert(curr >= btLow);
104
+ assert(ip < iend); /* condition for ZSTD_count */
105
+
106
+ for (; nbCompares && (matchIndex > windowLow); --nbCompares) {
107
+ U32* const nextPtr = bt + 2*(matchIndex & btMask);
108
+ size_t matchLength = MIN(commonLengthSmaller, commonLengthLarger); /* guaranteed minimum nb of common bytes */
109
+ assert(matchIndex < curr);
110
+ /* note : all candidates are now supposed sorted,
111
+ * but it's still possible to have nextPtr[1] == ZSTD_DUBT_UNSORTED_MARK
112
+ * when a real index has the same value as ZSTD_DUBT_UNSORTED_MARK */
113
+
114
+ if ( (dictMode != ZSTD_extDict)
115
+ || (matchIndex+matchLength >= dictLimit) /* both in current segment*/
116
+ || (curr < dictLimit) /* both in extDict */) {
117
+ const BYTE* const mBase = ( (dictMode != ZSTD_extDict)
118
+ || (matchIndex+matchLength >= dictLimit)) ?
119
+ base : dictBase;
120
+ assert( (matchIndex+matchLength >= dictLimit) /* might be wrong if extDict is incorrectly set to 0 */
121
+ || (curr < dictLimit) );
122
+ match = mBase + matchIndex;
123
+ matchLength += ZSTD_count(ip+matchLength, match+matchLength, iend);
124
+ } else {
125
+ match = dictBase + matchIndex;
126
+ matchLength += ZSTD_count_2segments(ip+matchLength, match+matchLength, iend, dictEnd, prefixStart);
127
+ if (matchIndex+matchLength >= dictLimit)
128
+ match = base + matchIndex; /* preparation for next read of match[matchLength] */
129
+ }
130
+
131
+ DEBUGLOG(8, "ZSTD_insertDUBT1: comparing %u with %u : found %u common bytes ",
132
+ curr, matchIndex, (U32)matchLength);
133
+
134
+ if (ip+matchLength == iend) { /* equal : no way to know if inf or sup */
135
+ break; /* drop , to guarantee consistency ; miss a bit of compression, but other solutions can corrupt tree */
136
+ }
137
+
138
+ if (match[matchLength] < ip[matchLength]) { /* necessarily within buffer */
139
+ /* match is smaller than current */
140
+ *smallerPtr = matchIndex; /* update smaller idx */
141
+ commonLengthSmaller = matchLength; /* all smaller will now have at least this guaranteed common length */
142
+ if (matchIndex <= btLow) { smallerPtr=&dummy32; break; } /* beyond tree size, stop searching */
143
+ DEBUGLOG(8, "ZSTD_insertDUBT1: %u (>btLow=%u) is smaller : next => %u",
144
+ matchIndex, btLow, nextPtr[1]);
145
+ smallerPtr = nextPtr+1; /* new "candidate" => larger than match, which was smaller than target */
146
+ matchIndex = nextPtr[1]; /* new matchIndex, larger than previous and closer to current */
147
+ } else {
148
+ /* match is larger than current */
149
+ *largerPtr = matchIndex;
150
+ commonLengthLarger = matchLength;
151
+ if (matchIndex <= btLow) { largerPtr=&dummy32; break; } /* beyond tree size, stop searching */
152
+ DEBUGLOG(8, "ZSTD_insertDUBT1: %u (>btLow=%u) is larger => %u",
153
+ matchIndex, btLow, nextPtr[0]);
154
+ largerPtr = nextPtr;
155
+ matchIndex = nextPtr[0];
156
+ } }
157
+
158
+ *smallerPtr = *largerPtr = 0;
159
+ }
160
+
161
+
162
+ static
163
+ ZSTD_ALLOW_POINTER_OVERFLOW_ATTR
164
+ size_t ZSTD_DUBT_findBetterDictMatch (
165
+ const ZSTD_matchState_t* ms,
166
+ const BYTE* const ip, const BYTE* const iend,
167
+ size_t* offsetPtr,
168
+ size_t bestLength,
169
+ U32 nbCompares,
170
+ U32 const mls,
171
+ const ZSTD_dictMode_e dictMode)
172
+ {
173
+ const ZSTD_matchState_t * const dms = ms->dictMatchState;
174
+ const ZSTD_compressionParameters* const dmsCParams = &dms->cParams;
175
+ const U32 * const dictHashTable = dms->hashTable;
176
+ U32 const hashLog = dmsCParams->hashLog;
177
+ size_t const h = ZSTD_hashPtr(ip, hashLog, mls);
178
+ U32 dictMatchIndex = dictHashTable[h];
179
+
180
+ const BYTE* const base = ms->window.base;
181
+ const BYTE* const prefixStart = base + ms->window.dictLimit;
182
+ U32 const curr = (U32)(ip-base);
183
+ const BYTE* const dictBase = dms->window.base;
184
+ const BYTE* const dictEnd = dms->window.nextSrc;
185
+ U32 const dictHighLimit = (U32)(dms->window.nextSrc - dms->window.base);
186
+ U32 const dictLowLimit = dms->window.lowLimit;
187
+ U32 const dictIndexDelta = ms->window.lowLimit - dictHighLimit;
188
+
189
+ U32* const dictBt = dms->chainTable;
190
+ U32 const btLog = dmsCParams->chainLog - 1;
191
+ U32 const btMask = (1 << btLog) - 1;
192
+ U32 const btLow = (btMask >= dictHighLimit - dictLowLimit) ? dictLowLimit : dictHighLimit - btMask;
193
+
194
+ size_t commonLengthSmaller=0, commonLengthLarger=0;
195
+
196
+ (void)dictMode;
197
+ assert(dictMode == ZSTD_dictMatchState);
198
+
199
+ for (; nbCompares && (dictMatchIndex > dictLowLimit); --nbCompares) {
200
+ U32* const nextPtr = dictBt + 2*(dictMatchIndex & btMask);
201
+ size_t matchLength = MIN(commonLengthSmaller, commonLengthLarger); /* guaranteed minimum nb of common bytes */
202
+ const BYTE* match = dictBase + dictMatchIndex;
203
+ matchLength += ZSTD_count_2segments(ip+matchLength, match+matchLength, iend, dictEnd, prefixStart);
204
+ if (dictMatchIndex+matchLength >= dictHighLimit)
205
+ match = base + dictMatchIndex + dictIndexDelta; /* to prepare for next usage of match[matchLength] */
206
+
207
+ if (matchLength > bestLength) {
208
+ U32 matchIndex = dictMatchIndex + dictIndexDelta;
209
+ if ( (4*(int)(matchLength-bestLength)) > (int)(ZSTD_highbit32(curr-matchIndex+1) - ZSTD_highbit32((U32)offsetPtr[0]+1)) ) {
210
+ DEBUGLOG(9, "ZSTD_DUBT_findBetterDictMatch(%u) : found better match length %u -> %u and offsetCode %u -> %u (dictMatchIndex %u, matchIndex %u)",
211
+ curr, (U32)bestLength, (U32)matchLength, (U32)*offsetPtr, OFFSET_TO_OFFBASE(curr - matchIndex), dictMatchIndex, matchIndex);
212
+ bestLength = matchLength, *offsetPtr = OFFSET_TO_OFFBASE(curr - matchIndex);
213
+ }
214
+ 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 */
215
+ break; /* drop, to guarantee consistency (miss a little bit of compression) */
216
+ }
217
+ }
218
+
219
+ if (match[matchLength] < ip[matchLength]) {
220
+ if (dictMatchIndex <= btLow) { break; } /* beyond tree size, stop the search */
221
+ commonLengthSmaller = matchLength; /* all smaller will now have at least this guaranteed common length */
222
+ dictMatchIndex = nextPtr[1]; /* new matchIndex larger than previous (closer to current) */
223
+ } else {
224
+ /* match is larger than current */
225
+ if (dictMatchIndex <= btLow) { break; } /* beyond tree size, stop the search */
226
+ commonLengthLarger = matchLength;
227
+ dictMatchIndex = nextPtr[0];
228
+ }
229
+ }
230
+
231
+ if (bestLength >= MINMATCH) {
232
+ U32 const mIndex = curr - (U32)OFFBASE_TO_OFFSET(*offsetPtr); (void)mIndex;
233
+ DEBUGLOG(8, "ZSTD_DUBT_findBetterDictMatch(%u) : found match of length %u and offsetCode %u (pos %u)",
234
+ curr, (U32)bestLength, (U32)*offsetPtr, mIndex);
235
+ }
236
+ return bestLength;
237
+
238
+ }
239
+
240
+
241
+ static
242
+ ZSTD_ALLOW_POINTER_OVERFLOW_ATTR
243
+ size_t ZSTD_DUBT_findBestMatch(ZSTD_matchState_t* ms,
244
+ const BYTE* const ip, const BYTE* const iend,
245
+ size_t* offBasePtr,
246
+ U32 const mls,
247
+ const ZSTD_dictMode_e dictMode)
248
+ {
249
+ const ZSTD_compressionParameters* const cParams = &ms->cParams;
250
+ U32* const hashTable = ms->hashTable;
251
+ U32 const hashLog = cParams->hashLog;
252
+ size_t const h = ZSTD_hashPtr(ip, hashLog, mls);
253
+ U32 matchIndex = hashTable[h];
254
+
255
+ const BYTE* const base = ms->window.base;
256
+ U32 const curr = (U32)(ip-base);
257
+ U32 const windowLow = ZSTD_getLowestMatchIndex(ms, curr, cParams->windowLog);
258
+
259
+ U32* const bt = ms->chainTable;
260
+ U32 const btLog = cParams->chainLog - 1;
261
+ U32 const btMask = (1 << btLog) - 1;
262
+ U32 const btLow = (btMask >= curr) ? 0 : curr - btMask;
263
+ U32 const unsortLimit = MAX(btLow, windowLow);
264
+
265
+ U32* nextCandidate = bt + 2*(matchIndex&btMask);
266
+ U32* unsortedMark = bt + 2*(matchIndex&btMask) + 1;
267
+ U32 nbCompares = 1U << cParams->searchLog;
268
+ U32 nbCandidates = nbCompares;
269
+ U32 previousCandidate = 0;
270
+
271
+ DEBUGLOG(7, "ZSTD_DUBT_findBestMatch (%u) ", curr);
272
+ assert(ip <= iend-8); /* required for h calculation */
273
+ assert(dictMode != ZSTD_dedicatedDictSearch);
274
+
275
+ /* reach end of unsorted candidates list */
276
+ while ( (matchIndex > unsortLimit)
277
+ && (*unsortedMark == ZSTD_DUBT_UNSORTED_MARK)
278
+ && (nbCandidates > 1) ) {
279
+ DEBUGLOG(8, "ZSTD_DUBT_findBestMatch: candidate %u is unsorted",
280
+ matchIndex);
281
+ *unsortedMark = previousCandidate; /* the unsortedMark becomes a reversed chain, to move up back to original position */
282
+ previousCandidate = matchIndex;
283
+ matchIndex = *nextCandidate;
284
+ nextCandidate = bt + 2*(matchIndex&btMask);
285
+ unsortedMark = bt + 2*(matchIndex&btMask) + 1;
286
+ nbCandidates --;
287
+ }
288
+
289
+ /* nullify last candidate if it's still unsorted
290
+ * simplification, detrimental to compression ratio, beneficial for speed */
291
+ if ( (matchIndex > unsortLimit)
292
+ && (*unsortedMark==ZSTD_DUBT_UNSORTED_MARK) ) {
293
+ DEBUGLOG(7, "ZSTD_DUBT_findBestMatch: nullify last unsorted candidate %u",
294
+ matchIndex);
295
+ *nextCandidate = *unsortedMark = 0;
296
+ }
297
+
298
+ /* batch sort stacked candidates */
299
+ matchIndex = previousCandidate;
300
+ while (matchIndex) { /* will end on matchIndex == 0 */
301
+ U32* const nextCandidateIdxPtr = bt + 2*(matchIndex&btMask) + 1;
302
+ U32 const nextCandidateIdx = *nextCandidateIdxPtr;
303
+ ZSTD_insertDUBT1(ms, matchIndex, iend,
304
+ nbCandidates, unsortLimit, dictMode);
305
+ matchIndex = nextCandidateIdx;
306
+ nbCandidates++;
307
+ }
308
+
309
+ /* find longest match */
310
+ { size_t commonLengthSmaller = 0, commonLengthLarger = 0;
311
+ const BYTE* const dictBase = ms->window.dictBase;
312
+ const U32 dictLimit = ms->window.dictLimit;
313
+ const BYTE* const dictEnd = dictBase + dictLimit;
314
+ const BYTE* const prefixStart = base + dictLimit;
315
+ U32* smallerPtr = bt + 2*(curr&btMask);
316
+ U32* largerPtr = bt + 2*(curr&btMask) + 1;
317
+ U32 matchEndIdx = curr + 8 + 1;
318
+ U32 dummy32; /* to be nullified at the end */
319
+ size_t bestLength = 0;
320
+
321
+ matchIndex = hashTable[h];
322
+ hashTable[h] = curr; /* Update Hash Table */
323
+
324
+ for (; nbCompares && (matchIndex > windowLow); --nbCompares) {
325
+ U32* const nextPtr = bt + 2*(matchIndex & btMask);
326
+ size_t matchLength = MIN(commonLengthSmaller, commonLengthLarger); /* guaranteed minimum nb of common bytes */
327
+ const BYTE* match;
328
+
329
+ if ((dictMode != ZSTD_extDict) || (matchIndex+matchLength >= dictLimit)) {
330
+ match = base + matchIndex;
331
+ matchLength += ZSTD_count(ip+matchLength, match+matchLength, iend);
332
+ } else {
333
+ match = dictBase + matchIndex;
334
+ matchLength += ZSTD_count_2segments(ip+matchLength, match+matchLength, iend, dictEnd, prefixStart);
335
+ if (matchIndex+matchLength >= dictLimit)
336
+ match = base + matchIndex; /* to prepare for next usage of match[matchLength] */
337
+ }
338
+
339
+ if (matchLength > bestLength) {
340
+ if (matchLength > matchEndIdx - matchIndex)
341
+ matchEndIdx = matchIndex + (U32)matchLength;
342
+ if ( (4*(int)(matchLength-bestLength)) > (int)(ZSTD_highbit32(curr - matchIndex + 1) - ZSTD_highbit32((U32)*offBasePtr)) )
343
+ bestLength = matchLength, *offBasePtr = OFFSET_TO_OFFBASE(curr - matchIndex);
344
+ if (ip+matchLength == iend) { /* equal : no way to know if inf or sup */
345
+ if (dictMode == ZSTD_dictMatchState) {
346
+ nbCompares = 0; /* in addition to avoiding checking any
347
+ * further in this loop, make sure we
348
+ * skip checking in the dictionary. */
349
+ }
350
+ break; /* drop, to guarantee consistency (miss a little bit of compression) */
351
+ }
352
+ }
353
+
354
+ if (match[matchLength] < ip[matchLength]) {
355
+ /* match is smaller than current */
356
+ *smallerPtr = matchIndex; /* update smaller idx */
357
+ commonLengthSmaller = matchLength; /* all smaller will now have at least this guaranteed common length */
358
+ if (matchIndex <= btLow) { smallerPtr=&dummy32; break; } /* beyond tree size, stop the search */
359
+ smallerPtr = nextPtr+1; /* new "smaller" => larger of match */
360
+ matchIndex = nextPtr[1]; /* new matchIndex larger than previous (closer to current) */
361
+ } else {
362
+ /* match is larger than current */
363
+ *largerPtr = matchIndex;
364
+ commonLengthLarger = matchLength;
365
+ if (matchIndex <= btLow) { largerPtr=&dummy32; break; } /* beyond tree size, stop the search */
366
+ largerPtr = nextPtr;
367
+ matchIndex = nextPtr[0];
368
+ } }
369
+
370
+ *smallerPtr = *largerPtr = 0;
371
+
372
+ assert(nbCompares <= (1U << ZSTD_SEARCHLOG_MAX)); /* Check we haven't underflowed. */
373
+ if (dictMode == ZSTD_dictMatchState && nbCompares) {
374
+ bestLength = ZSTD_DUBT_findBetterDictMatch(
375
+ ms, ip, iend,
376
+ offBasePtr, bestLength, nbCompares,
377
+ mls, dictMode);
378
+ }
379
+
380
+ assert(matchEndIdx > curr+8); /* ensure nextToUpdate is increased */
381
+ ms->nextToUpdate = matchEndIdx - 8; /* skip repetitive patterns */
382
+ if (bestLength >= MINMATCH) {
383
+ U32 const mIndex = curr - (U32)OFFBASE_TO_OFFSET(*offBasePtr); (void)mIndex;
384
+ DEBUGLOG(8, "ZSTD_DUBT_findBestMatch(%u) : found match of length %u and offsetCode %u (pos %u)",
385
+ curr, (U32)bestLength, (U32)*offBasePtr, mIndex);
386
+ }
387
+ return bestLength;
388
+ }
389
+ }
390
+
391
+
392
+ /** ZSTD_BtFindBestMatch() : Tree updater, providing best match */
393
+ FORCE_INLINE_TEMPLATE
394
+ ZSTD_ALLOW_POINTER_OVERFLOW_ATTR
395
+ size_t ZSTD_BtFindBestMatch( ZSTD_matchState_t* ms,
396
+ const BYTE* const ip, const BYTE* const iLimit,
397
+ size_t* offBasePtr,
398
+ const U32 mls /* template */,
399
+ const ZSTD_dictMode_e dictMode)
400
+ {
401
+ DEBUGLOG(7, "ZSTD_BtFindBestMatch");
402
+ if (ip < ms->window.base + ms->nextToUpdate) return 0; /* skipped area */
403
+ ZSTD_updateDUBT(ms, ip, iLimit, mls);
404
+ return ZSTD_DUBT_findBestMatch(ms, ip, iLimit, offBasePtr, mls, dictMode);
405
+ }
406
+
407
+ /***********************************
408
+ * Dedicated dict search
409
+ ***********************************/
410
+
411
+ void ZSTD_dedicatedDictSearch_lazy_loadDictionary(ZSTD_matchState_t* ms, const BYTE* const ip)
412
+ {
413
+ const BYTE* const base = ms->window.base;
414
+ U32 const target = (U32)(ip - base);
415
+ U32* const hashTable = ms->hashTable;
416
+ U32* const chainTable = ms->chainTable;
417
+ U32 const chainSize = 1 << ms->cParams.chainLog;
418
+ U32 idx = ms->nextToUpdate;
419
+ U32 const minChain = chainSize < target - idx ? target - chainSize : idx;
420
+ U32 const bucketSize = 1 << ZSTD_LAZY_DDSS_BUCKET_LOG;
421
+ U32 const cacheSize = bucketSize - 1;
422
+ U32 const chainAttempts = (1 << ms->cParams.searchLog) - cacheSize;
423
+ U32 const chainLimit = chainAttempts > 255 ? 255 : chainAttempts;
424
+
425
+ /* We know the hashtable is oversized by a factor of `bucketSize`.
426
+ * We are going to temporarily pretend `bucketSize == 1`, keeping only a
427
+ * single entry. We will use the rest of the space to construct a temporary
428
+ * chaintable.
429
+ */
430
+ U32 const hashLog = ms->cParams.hashLog - ZSTD_LAZY_DDSS_BUCKET_LOG;
431
+ U32* const tmpHashTable = hashTable;
432
+ U32* const tmpChainTable = hashTable + ((size_t)1 << hashLog);
433
+ U32 const tmpChainSize = (U32)((1 << ZSTD_LAZY_DDSS_BUCKET_LOG) - 1) << hashLog;
434
+ U32 const tmpMinChain = tmpChainSize < target ? target - tmpChainSize : idx;
435
+ U32 hashIdx;
436
+
437
+ assert(ms->cParams.chainLog <= 24);
438
+ assert(ms->cParams.hashLog > ms->cParams.chainLog);
439
+ assert(idx != 0);
440
+ assert(tmpMinChain <= minChain);
441
+
442
+ /* fill conventional hash table and conventional chain table */
443
+ for ( ; idx < target; idx++) {
444
+ U32 const h = (U32)ZSTD_hashPtr(base + idx, hashLog, ms->cParams.minMatch);
445
+ if (idx >= tmpMinChain) {
446
+ tmpChainTable[idx - tmpMinChain] = hashTable[h];
447
+ }
448
+ tmpHashTable[h] = idx;
449
+ }
450
+
451
+ /* sort chains into ddss chain table */
452
+ {
453
+ U32 chainPos = 0;
454
+ for (hashIdx = 0; hashIdx < (1U << hashLog); hashIdx++) {
455
+ U32 count;
456
+ U32 countBeyondMinChain = 0;
457
+ U32 i = tmpHashTable[hashIdx];
458
+ for (count = 0; i >= tmpMinChain && count < cacheSize; count++) {
459
+ /* skip through the chain to the first position that won't be
460
+ * in the hash cache bucket */
461
+ if (i < minChain) {
462
+ countBeyondMinChain++;
463
+ }
464
+ i = tmpChainTable[i - tmpMinChain];
465
+ }
466
+ if (count == cacheSize) {
467
+ for (count = 0; count < chainLimit;) {
468
+ if (i < minChain) {
469
+ if (!i || ++countBeyondMinChain > cacheSize) {
470
+ /* only allow pulling `cacheSize` number of entries
471
+ * into the cache or chainTable beyond `minChain`,
472
+ * to replace the entries pulled out of the
473
+ * chainTable into the cache. This lets us reach
474
+ * back further without increasing the total number
475
+ * of entries in the chainTable, guaranteeing the
476
+ * DDSS chain table will fit into the space
477
+ * allocated for the regular one. */
478
+ break;
479
+ }
480
+ }
481
+ chainTable[chainPos++] = i;
482
+ count++;
483
+ if (i < tmpMinChain) {
484
+ break;
485
+ }
486
+ i = tmpChainTable[i - tmpMinChain];
487
+ }
488
+ } else {
489
+ count = 0;
490
+ }
491
+ if (count) {
492
+ tmpHashTable[hashIdx] = ((chainPos - count) << 8) + count;
493
+ } else {
494
+ tmpHashTable[hashIdx] = 0;
495
+ }
496
+ }
497
+ assert(chainPos <= chainSize); /* I believe this is guaranteed... */
498
+ }
499
+
500
+ /* move chain pointers into the last entry of each hash bucket */
501
+ for (hashIdx = (1 << hashLog); hashIdx; ) {
502
+ U32 const bucketIdx = --hashIdx << ZSTD_LAZY_DDSS_BUCKET_LOG;
503
+ U32 const chainPackedPointer = tmpHashTable[hashIdx];
504
+ U32 i;
505
+ for (i = 0; i < cacheSize; i++) {
506
+ hashTable[bucketIdx + i] = 0;
507
+ }
508
+ hashTable[bucketIdx + bucketSize - 1] = chainPackedPointer;
509
+ }
510
+
511
+ /* fill the buckets of the hash table */
512
+ for (idx = ms->nextToUpdate; idx < target; idx++) {
513
+ U32 const h = (U32)ZSTD_hashPtr(base + idx, hashLog, ms->cParams.minMatch)
514
+ << ZSTD_LAZY_DDSS_BUCKET_LOG;
515
+ U32 i;
516
+ /* Shift hash cache down 1. */
517
+ for (i = cacheSize - 1; i; i--)
518
+ hashTable[h + i] = hashTable[h + i - 1];
519
+ hashTable[h] = idx;
520
+ }
521
+
522
+ ms->nextToUpdate = target;
523
+ }
524
+
525
+ /* Returns the longest match length found in the dedicated dict search structure.
526
+ * If none are longer than the argument ml, then ml will be returned.
527
+ */
528
+ FORCE_INLINE_TEMPLATE
529
+ size_t ZSTD_dedicatedDictSearch_lazy_search(size_t* offsetPtr, size_t ml, U32 nbAttempts,
530
+ const ZSTD_matchState_t* const dms,
531
+ const BYTE* const ip, const BYTE* const iLimit,
532
+ const BYTE* const prefixStart, const U32 curr,
533
+ const U32 dictLimit, const size_t ddsIdx) {
534
+ const U32 ddsLowestIndex = dms->window.dictLimit;
535
+ const BYTE* const ddsBase = dms->window.base;
536
+ const BYTE* const ddsEnd = dms->window.nextSrc;
537
+ const U32 ddsSize = (U32)(ddsEnd - ddsBase);
538
+ const U32 ddsIndexDelta = dictLimit - ddsSize;
539
+ const U32 bucketSize = (1 << ZSTD_LAZY_DDSS_BUCKET_LOG);
540
+ const U32 bucketLimit = nbAttempts < bucketSize - 1 ? nbAttempts : bucketSize - 1;
541
+ U32 ddsAttempt;
542
+ U32 matchIndex;
543
+
544
+ for (ddsAttempt = 0; ddsAttempt < bucketSize - 1; ddsAttempt++) {
545
+ PREFETCH_L1(ddsBase + dms->hashTable[ddsIdx + ddsAttempt]);
546
+ }
547
+
548
+ {
549
+ U32 const chainPackedPointer = dms->hashTable[ddsIdx + bucketSize - 1];
550
+ U32 const chainIndex = chainPackedPointer >> 8;
551
+
552
+ PREFETCH_L1(&dms->chainTable[chainIndex]);
553
+ }
554
+
555
+ for (ddsAttempt = 0; ddsAttempt < bucketLimit; ddsAttempt++) {
556
+ size_t currentMl=0;
557
+ const BYTE* match;
558
+ matchIndex = dms->hashTable[ddsIdx + ddsAttempt];
559
+ match = ddsBase + matchIndex;
560
+
561
+ if (!matchIndex) {
562
+ return ml;
563
+ }
564
+
565
+ /* guaranteed by table construction */
566
+ (void)ddsLowestIndex;
567
+ assert(matchIndex >= ddsLowestIndex);
568
+ assert(match+4 <= ddsEnd);
569
+ if (MEM_read32(match) == MEM_read32(ip)) {
570
+ /* assumption : matchIndex <= dictLimit-4 (by table construction) */
571
+ currentMl = ZSTD_count_2segments(ip+4, match+4, iLimit, ddsEnd, prefixStart) + 4;
572
+ }
573
+
574
+ /* save best solution */
575
+ if (currentMl > ml) {
576
+ ml = currentMl;
577
+ *offsetPtr = OFFSET_TO_OFFBASE(curr - (matchIndex + ddsIndexDelta));
578
+ if (ip+currentMl == iLimit) {
579
+ /* best possible, avoids read overflow on next attempt */
580
+ return ml;
581
+ }
582
+ }
583
+ }
584
+
585
+ {
586
+ U32 const chainPackedPointer = dms->hashTable[ddsIdx + bucketSize - 1];
587
+ U32 chainIndex = chainPackedPointer >> 8;
588
+ U32 const chainLength = chainPackedPointer & 0xFF;
589
+ U32 const chainAttempts = nbAttempts - ddsAttempt;
590
+ U32 const chainLimit = chainAttempts > chainLength ? chainLength : chainAttempts;
591
+ U32 chainAttempt;
592
+
593
+ for (chainAttempt = 0 ; chainAttempt < chainLimit; chainAttempt++) {
594
+ PREFETCH_L1(ddsBase + dms->chainTable[chainIndex + chainAttempt]);
595
+ }
596
+
597
+ for (chainAttempt = 0 ; chainAttempt < chainLimit; chainAttempt++, chainIndex++) {
598
+ size_t currentMl=0;
599
+ const BYTE* match;
600
+ matchIndex = dms->chainTable[chainIndex];
601
+ match = ddsBase + matchIndex;
602
+
603
+ /* guaranteed by table construction */
604
+ assert(matchIndex >= ddsLowestIndex);
605
+ assert(match+4 <= ddsEnd);
606
+ if (MEM_read32(match) == MEM_read32(ip)) {
607
+ /* assumption : matchIndex <= dictLimit-4 (by table construction) */
608
+ currentMl = ZSTD_count_2segments(ip+4, match+4, iLimit, ddsEnd, prefixStart) + 4;
609
+ }
610
+
611
+ /* save best solution */
612
+ if (currentMl > ml) {
613
+ ml = currentMl;
614
+ *offsetPtr = OFFSET_TO_OFFBASE(curr - (matchIndex + ddsIndexDelta));
615
+ if (ip+currentMl == iLimit) break; /* best possible, avoids read overflow on next attempt */
616
+ }
617
+ }
618
+ }
619
+ return ml;
620
+ }
621
+
622
+
623
+ /* *********************************
624
+ * Hash Chain
625
+ ***********************************/
626
+ #define NEXT_IN_CHAIN(d, mask) chainTable[(d) & (mask)]
627
+
628
+ /* Update chains up to ip (excluded)
629
+ Assumption : always within prefix (i.e. not within extDict) */
630
+ FORCE_INLINE_TEMPLATE
631
+ ZSTD_ALLOW_POINTER_OVERFLOW_ATTR
632
+ U32 ZSTD_insertAndFindFirstIndex_internal(
633
+ ZSTD_matchState_t* ms,
634
+ const ZSTD_compressionParameters* const cParams,
635
+ const BYTE* ip, U32 const mls, U32 const lazySkipping)
636
+ {
637
+ U32* const hashTable = ms->hashTable;
638
+ const U32 hashLog = cParams->hashLog;
639
+ U32* const chainTable = ms->chainTable;
640
+ const U32 chainMask = (1 << cParams->chainLog) - 1;
641
+ const BYTE* const base = ms->window.base;
642
+ const U32 target = (U32)(ip - base);
643
+ U32 idx = ms->nextToUpdate;
644
+
645
+ while(idx < target) { /* catch up */
646
+ size_t const h = ZSTD_hashPtr(base+idx, hashLog, mls);
647
+ NEXT_IN_CHAIN(idx, chainMask) = hashTable[h];
648
+ hashTable[h] = idx;
649
+ idx++;
650
+ /* Stop inserting every position when in the lazy skipping mode. */
651
+ if (lazySkipping)
652
+ break;
653
+ }
654
+
655
+ ms->nextToUpdate = target;
656
+ return hashTable[ZSTD_hashPtr(ip, hashLog, mls)];
657
+ }
658
+
659
+ U32 ZSTD_insertAndFindFirstIndex(ZSTD_matchState_t* ms, const BYTE* ip) {
660
+ const ZSTD_compressionParameters* const cParams = &ms->cParams;
661
+ return ZSTD_insertAndFindFirstIndex_internal(ms, cParams, ip, ms->cParams.minMatch, /* lazySkipping*/ 0);
662
+ }
663
+
664
+ /* inlining is important to hardwire a hot branch (template emulation) */
665
+ FORCE_INLINE_TEMPLATE
666
+ ZSTD_ALLOW_POINTER_OVERFLOW_ATTR
667
+ size_t ZSTD_HcFindBestMatch(
668
+ ZSTD_matchState_t* ms,
669
+ const BYTE* const ip, const BYTE* const iLimit,
670
+ size_t* offsetPtr,
671
+ const U32 mls, const ZSTD_dictMode_e dictMode)
672
+ {
673
+ const ZSTD_compressionParameters* const cParams = &ms->cParams;
674
+ U32* const chainTable = ms->chainTable;
675
+ const U32 chainSize = (1 << cParams->chainLog);
676
+ const U32 chainMask = chainSize-1;
677
+ const BYTE* const base = ms->window.base;
678
+ const BYTE* const dictBase = ms->window.dictBase;
679
+ const U32 dictLimit = ms->window.dictLimit;
680
+ const BYTE* const prefixStart = base + dictLimit;
681
+ const BYTE* const dictEnd = dictBase + dictLimit;
682
+ const U32 curr = (U32)(ip-base);
683
+ const U32 maxDistance = 1U << cParams->windowLog;
684
+ const U32 lowestValid = ms->window.lowLimit;
685
+ const U32 withinMaxDistance = (curr - lowestValid > maxDistance) ? curr - maxDistance : lowestValid;
686
+ const U32 isDictionary = (ms->loadedDictEnd != 0);
687
+ const U32 lowLimit = isDictionary ? lowestValid : withinMaxDistance;
688
+ const U32 minChain = curr > chainSize ? curr - chainSize : 0;
689
+ U32 nbAttempts = 1U << cParams->searchLog;
690
+ size_t ml=4-1;
691
+
692
+ const ZSTD_matchState_t* const dms = ms->dictMatchState;
693
+ const U32 ddsHashLog = dictMode == ZSTD_dedicatedDictSearch
694
+ ? dms->cParams.hashLog - ZSTD_LAZY_DDSS_BUCKET_LOG : 0;
695
+ const size_t ddsIdx = dictMode == ZSTD_dedicatedDictSearch
696
+ ? ZSTD_hashPtr(ip, ddsHashLog, mls) << ZSTD_LAZY_DDSS_BUCKET_LOG : 0;
697
+
698
+ U32 matchIndex;
699
+
700
+ if (dictMode == ZSTD_dedicatedDictSearch) {
701
+ const U32* entry = &dms->hashTable[ddsIdx];
702
+ PREFETCH_L1(entry);
703
+ }
704
+
705
+ /* HC4 match finder */
706
+ matchIndex = ZSTD_insertAndFindFirstIndex_internal(ms, cParams, ip, mls, ms->lazySkipping);
707
+
708
+ for ( ; (matchIndex>=lowLimit) & (nbAttempts>0) ; nbAttempts--) {
709
+ size_t currentMl=0;
710
+ if ((dictMode != ZSTD_extDict) || matchIndex >= dictLimit) {
711
+ const BYTE* const match = base + matchIndex;
712
+ assert(matchIndex >= dictLimit); /* ensures this is true if dictMode != ZSTD_extDict */
713
+ /* read 4B starting from (match + ml + 1 - sizeof(U32)) */
714
+ if (MEM_read32(match + ml - 3) == MEM_read32(ip + ml - 3)) /* potentially better */
715
+ currentMl = ZSTD_count(ip, match, iLimit);
716
+ } else {
717
+ const BYTE* const match = dictBase + matchIndex;
718
+ assert(match+4 <= dictEnd);
719
+ if (MEM_read32(match) == MEM_read32(ip)) /* assumption : matchIndex <= dictLimit-4 (by table construction) */
720
+ currentMl = ZSTD_count_2segments(ip+4, match+4, iLimit, dictEnd, prefixStart) + 4;
721
+ }
722
+
723
+ /* save best solution */
724
+ if (currentMl > ml) {
725
+ ml = currentMl;
726
+ *offsetPtr = OFFSET_TO_OFFBASE(curr - matchIndex);
727
+ if (ip+currentMl == iLimit) break; /* best possible, avoids read overflow on next attempt */
728
+ }
729
+
730
+ if (matchIndex <= minChain) break;
731
+ matchIndex = NEXT_IN_CHAIN(matchIndex, chainMask);
732
+ }
733
+
734
+ assert(nbAttempts <= (1U << ZSTD_SEARCHLOG_MAX)); /* Check we haven't underflowed. */
735
+ if (dictMode == ZSTD_dedicatedDictSearch) {
736
+ ml = ZSTD_dedicatedDictSearch_lazy_search(offsetPtr, ml, nbAttempts, dms,
737
+ ip, iLimit, prefixStart, curr, dictLimit, ddsIdx);
738
+ } else if (dictMode == ZSTD_dictMatchState) {
739
+ const U32* const dmsChainTable = dms->chainTable;
740
+ const U32 dmsChainSize = (1 << dms->cParams.chainLog);
741
+ const U32 dmsChainMask = dmsChainSize - 1;
742
+ const U32 dmsLowestIndex = dms->window.dictLimit;
743
+ const BYTE* const dmsBase = dms->window.base;
744
+ const BYTE* const dmsEnd = dms->window.nextSrc;
745
+ const U32 dmsSize = (U32)(dmsEnd - dmsBase);
746
+ const U32 dmsIndexDelta = dictLimit - dmsSize;
747
+ const U32 dmsMinChain = dmsSize > dmsChainSize ? dmsSize - dmsChainSize : 0;
748
+
749
+ matchIndex = dms->hashTable[ZSTD_hashPtr(ip, dms->cParams.hashLog, mls)];
750
+
751
+ for ( ; (matchIndex>=dmsLowestIndex) & (nbAttempts>0) ; nbAttempts--) {
752
+ size_t currentMl=0;
753
+ const BYTE* const match = dmsBase + matchIndex;
754
+ assert(match+4 <= dmsEnd);
755
+ if (MEM_read32(match) == MEM_read32(ip)) /* assumption : matchIndex <= dictLimit-4 (by table construction) */
756
+ currentMl = ZSTD_count_2segments(ip+4, match+4, iLimit, dmsEnd, prefixStart) + 4;
757
+
758
+ /* save best solution */
759
+ if (currentMl > ml) {
760
+ ml = currentMl;
761
+ assert(curr > matchIndex + dmsIndexDelta);
762
+ *offsetPtr = OFFSET_TO_OFFBASE(curr - (matchIndex + dmsIndexDelta));
763
+ if (ip+currentMl == iLimit) break; /* best possible, avoids read overflow on next attempt */
764
+ }
765
+
766
+ if (matchIndex <= dmsMinChain) break;
767
+
768
+ matchIndex = dmsChainTable[matchIndex & dmsChainMask];
769
+ }
770
+ }
771
+
772
+ return ml;
773
+ }
774
+
775
+ /* *********************************
776
+ * (SIMD) Row-based matchfinder
777
+ ***********************************/
778
+ /* Constants for row-based hash */
779
+ #define ZSTD_ROW_HASH_TAG_MASK ((1u << ZSTD_ROW_HASH_TAG_BITS) - 1)
780
+ #define ZSTD_ROW_HASH_MAX_ENTRIES 64 /* absolute maximum number of entries per row, for all configurations */
781
+
782
+ #define ZSTD_ROW_HASH_CACHE_MASK (ZSTD_ROW_HASH_CACHE_SIZE - 1)
783
+
784
+ typedef U64 ZSTD_VecMask; /* Clarifies when we are interacting with a U64 representing a mask of matches */
785
+
786
+ /* ZSTD_VecMask_next():
787
+ * Starting from the LSB, returns the idx of the next non-zero bit.
788
+ * Basically counting the nb of trailing zeroes.
789
+ */
790
+ MEM_STATIC U32 ZSTD_VecMask_next(ZSTD_VecMask val) {
791
+ return ZSTD_countTrailingZeros64(val);
792
+ }
793
+
794
+ /* ZSTD_row_nextIndex():
795
+ * Returns the next index to insert at within a tagTable row, and updates the "head"
796
+ * value to reflect the update. Essentially cycles backwards from [1, {entries per row})
797
+ */
798
+ FORCE_INLINE_TEMPLATE U32 ZSTD_row_nextIndex(BYTE* const tagRow, U32 const rowMask) {
799
+ U32 next = (*tagRow-1) & rowMask;
800
+ next += (next == 0) ? rowMask : 0; /* skip first position */
801
+ *tagRow = (BYTE)next;
802
+ return next;
803
+ }
804
+
805
+ /* ZSTD_isAligned():
806
+ * Checks that a pointer is aligned to "align" bytes which must be a power of 2.
807
+ */
808
+ MEM_STATIC int ZSTD_isAligned(void const* ptr, size_t align) {
809
+ assert((align & (align - 1)) == 0);
810
+ return (((size_t)ptr) & (align - 1)) == 0;
811
+ }
812
+
813
+ /* ZSTD_row_prefetch():
814
+ * Performs prefetching for the hashTable and tagTable at a given row.
815
+ */
816
+ FORCE_INLINE_TEMPLATE void ZSTD_row_prefetch(U32 const* hashTable, BYTE const* tagTable, U32 const relRow, U32 const rowLog) {
817
+ PREFETCH_L1(hashTable + relRow);
818
+ if (rowLog >= 5) {
819
+ PREFETCH_L1(hashTable + relRow + 16);
820
+ /* Note: prefetching more of the hash table does not appear to be beneficial for 128-entry rows */
821
+ }
822
+ PREFETCH_L1(tagTable + relRow);
823
+ if (rowLog == 6) {
824
+ PREFETCH_L1(tagTable + relRow + 32);
825
+ }
826
+ assert(rowLog == 4 || rowLog == 5 || rowLog == 6);
827
+ assert(ZSTD_isAligned(hashTable + relRow, 64)); /* prefetched hash row always 64-byte aligned */
828
+ assert(ZSTD_isAligned(tagTable + relRow, (size_t)1 << rowLog)); /* prefetched tagRow sits on correct multiple of bytes (32,64,128) */
829
+ }
830
+
831
+ /* ZSTD_row_fillHashCache():
832
+ * Fill up the hash cache starting at idx, prefetching up to ZSTD_ROW_HASH_CACHE_SIZE entries,
833
+ * but not beyond iLimit.
834
+ */
835
+ FORCE_INLINE_TEMPLATE
836
+ ZSTD_ALLOW_POINTER_OVERFLOW_ATTR
837
+ void ZSTD_row_fillHashCache(ZSTD_matchState_t* ms, const BYTE* base,
838
+ U32 const rowLog, U32 const mls,
839
+ U32 idx, const BYTE* const iLimit)
840
+ {
841
+ U32 const* const hashTable = ms->hashTable;
842
+ BYTE const* const tagTable = ms->tagTable;
843
+ U32 const hashLog = ms->rowHashLog;
844
+ U32 const maxElemsToPrefetch = (base + idx) > iLimit ? 0 : (U32)(iLimit - (base + idx) + 1);
845
+ U32 const lim = idx + MIN(ZSTD_ROW_HASH_CACHE_SIZE, maxElemsToPrefetch);
846
+
847
+ for (; idx < lim; ++idx) {
848
+ U32 const hash = (U32)ZSTD_hashPtrSalted(base + idx, hashLog + ZSTD_ROW_HASH_TAG_BITS, mls, ms->hashSalt);
849
+ U32 const row = (hash >> ZSTD_ROW_HASH_TAG_BITS) << rowLog;
850
+ ZSTD_row_prefetch(hashTable, tagTable, row, rowLog);
851
+ ms->hashCache[idx & ZSTD_ROW_HASH_CACHE_MASK] = hash;
852
+ }
853
+
854
+ DEBUGLOG(6, "ZSTD_row_fillHashCache(): [%u %u %u %u %u %u %u %u]", ms->hashCache[0], ms->hashCache[1],
855
+ ms->hashCache[2], ms->hashCache[3], ms->hashCache[4],
856
+ ms->hashCache[5], ms->hashCache[6], ms->hashCache[7]);
857
+ }
858
+
859
+ /* ZSTD_row_nextCachedHash():
860
+ * Returns the hash of base + idx, and replaces the hash in the hash cache with the byte at
861
+ * base + idx + ZSTD_ROW_HASH_CACHE_SIZE. Also prefetches the appropriate rows from hashTable and tagTable.
862
+ */
863
+ FORCE_INLINE_TEMPLATE
864
+ ZSTD_ALLOW_POINTER_OVERFLOW_ATTR
865
+ U32 ZSTD_row_nextCachedHash(U32* cache, U32 const* hashTable,
866
+ BYTE const* tagTable, BYTE const* base,
867
+ U32 idx, U32 const hashLog,
868
+ U32 const rowLog, U32 const mls,
869
+ U64 const hashSalt)
870
+ {
871
+ U32 const newHash = (U32)ZSTD_hashPtrSalted(base+idx+ZSTD_ROW_HASH_CACHE_SIZE, hashLog + ZSTD_ROW_HASH_TAG_BITS, mls, hashSalt);
872
+ U32 const row = (newHash >> ZSTD_ROW_HASH_TAG_BITS) << rowLog;
873
+ ZSTD_row_prefetch(hashTable, tagTable, row, rowLog);
874
+ { U32 const hash = cache[idx & ZSTD_ROW_HASH_CACHE_MASK];
875
+ cache[idx & ZSTD_ROW_HASH_CACHE_MASK] = newHash;
876
+ return hash;
877
+ }
878
+ }
879
+
880
+ /* ZSTD_row_update_internalImpl():
881
+ * Updates the hash table with positions starting from updateStartIdx until updateEndIdx.
882
+ */
883
+ FORCE_INLINE_TEMPLATE
884
+ ZSTD_ALLOW_POINTER_OVERFLOW_ATTR
885
+ void ZSTD_row_update_internalImpl(ZSTD_matchState_t* ms,
886
+ U32 updateStartIdx, U32 const updateEndIdx,
887
+ U32 const mls, U32 const rowLog,
888
+ U32 const rowMask, U32 const useCache)
889
+ {
890
+ U32* const hashTable = ms->hashTable;
891
+ BYTE* const tagTable = ms->tagTable;
892
+ U32 const hashLog = ms->rowHashLog;
893
+ const BYTE* const base = ms->window.base;
894
+
895
+ DEBUGLOG(6, "ZSTD_row_update_internalImpl(): updateStartIdx=%u, updateEndIdx=%u", updateStartIdx, updateEndIdx);
896
+ for (; updateStartIdx < updateEndIdx; ++updateStartIdx) {
897
+ U32 const hash = useCache ? ZSTD_row_nextCachedHash(ms->hashCache, hashTable, tagTable, base, updateStartIdx, hashLog, rowLog, mls, ms->hashSalt)
898
+ : (U32)ZSTD_hashPtrSalted(base + updateStartIdx, hashLog + ZSTD_ROW_HASH_TAG_BITS, mls, ms->hashSalt);
899
+ U32 const relRow = (hash >> ZSTD_ROW_HASH_TAG_BITS) << rowLog;
900
+ U32* const row = hashTable + relRow;
901
+ BYTE* tagRow = tagTable + relRow;
902
+ U32 const pos = ZSTD_row_nextIndex(tagRow, rowMask);
903
+
904
+ assert(hash == ZSTD_hashPtrSalted(base + updateStartIdx, hashLog + ZSTD_ROW_HASH_TAG_BITS, mls, ms->hashSalt));
905
+ tagRow[pos] = hash & ZSTD_ROW_HASH_TAG_MASK;
906
+ row[pos] = updateStartIdx;
907
+ }
908
+ }
909
+
910
+ /* ZSTD_row_update_internal():
911
+ * Inserts the byte at ip into the appropriate position in the hash table, and updates ms->nextToUpdate.
912
+ * Skips sections of long matches as is necessary.
913
+ */
914
+ FORCE_INLINE_TEMPLATE
915
+ ZSTD_ALLOW_POINTER_OVERFLOW_ATTR
916
+ void ZSTD_row_update_internal(ZSTD_matchState_t* ms, const BYTE* ip,
917
+ U32 const mls, U32 const rowLog,
918
+ U32 const rowMask, U32 const useCache)
919
+ {
920
+ U32 idx = ms->nextToUpdate;
921
+ const BYTE* const base = ms->window.base;
922
+ const U32 target = (U32)(ip - base);
923
+ const U32 kSkipThreshold = 384;
924
+ const U32 kMaxMatchStartPositionsToUpdate = 96;
925
+ const U32 kMaxMatchEndPositionsToUpdate = 32;
926
+
927
+ if (useCache) {
928
+ /* Only skip positions when using hash cache, i.e.
929
+ * if we are loading a dict, don't skip anything.
930
+ * If we decide to skip, then we only update a set number
931
+ * of positions at the beginning and end of the match.
932
+ */
933
+ if (UNLIKELY(target - idx > kSkipThreshold)) {
934
+ U32 const bound = idx + kMaxMatchStartPositionsToUpdate;
935
+ ZSTD_row_update_internalImpl(ms, idx, bound, mls, rowLog, rowMask, useCache);
936
+ idx = target - kMaxMatchEndPositionsToUpdate;
937
+ ZSTD_row_fillHashCache(ms, base, rowLog, mls, idx, ip+1);
938
+ }
939
+ }
940
+ assert(target >= idx);
941
+ ZSTD_row_update_internalImpl(ms, idx, target, mls, rowLog, rowMask, useCache);
942
+ ms->nextToUpdate = target;
943
+ }
944
+
945
+ /* ZSTD_row_update():
946
+ * External wrapper for ZSTD_row_update_internal(). Used for filling the hashtable during dictionary
947
+ * processing.
948
+ */
949
+ void ZSTD_row_update(ZSTD_matchState_t* const ms, const BYTE* ip) {
950
+ const U32 rowLog = BOUNDED(4, ms->cParams.searchLog, 6);
951
+ const U32 rowMask = (1u << rowLog) - 1;
952
+ const U32 mls = MIN(ms->cParams.minMatch, 6 /* mls caps out at 6 */);
953
+
954
+ DEBUGLOG(5, "ZSTD_row_update(), rowLog=%u", rowLog);
955
+ ZSTD_row_update_internal(ms, ip, mls, rowLog, rowMask, 0 /* don't use cache */);
956
+ }
957
+
958
+ /* Returns the mask width of bits group of which will be set to 1. Given not all
959
+ * architectures have easy movemask instruction, this helps to iterate over
960
+ * groups of bits easier and faster.
961
+ */
962
+ FORCE_INLINE_TEMPLATE U32
963
+ ZSTD_row_matchMaskGroupWidth(const U32 rowEntries)
964
+ {
965
+ assert((rowEntries == 16) || (rowEntries == 32) || rowEntries == 64);
966
+ assert(rowEntries <= ZSTD_ROW_HASH_MAX_ENTRIES);
967
+ (void)rowEntries;
968
+ #if defined(ZSTD_ARCH_ARM_NEON)
969
+ /* NEON path only works for little endian */
970
+ if (!MEM_isLittleEndian()) {
971
+ return 1;
972
+ }
973
+ if (rowEntries == 16) {
974
+ return 4;
975
+ }
976
+ if (rowEntries == 32) {
977
+ return 2;
978
+ }
979
+ if (rowEntries == 64) {
980
+ return 1;
981
+ }
982
+ #endif
983
+ return 1;
984
+ }
985
+
986
+ #if defined(ZSTD_ARCH_X86_SSE2)
987
+ FORCE_INLINE_TEMPLATE ZSTD_VecMask
988
+ ZSTD_row_getSSEMask(int nbChunks, const BYTE* const src, const BYTE tag, const U32 head)
989
+ {
990
+ const __m128i comparisonMask = _mm_set1_epi8((char)tag);
991
+ int matches[4] = {0};
992
+ int i;
993
+ assert(nbChunks == 1 || nbChunks == 2 || nbChunks == 4);
994
+ for (i=0; i<nbChunks; i++) {
995
+ const __m128i chunk = _mm_loadu_si128((const __m128i*)(const void*)(src + 16*i));
996
+ const __m128i equalMask = _mm_cmpeq_epi8(chunk, comparisonMask);
997
+ matches[i] = _mm_movemask_epi8(equalMask);
998
+ }
999
+ if (nbChunks == 1) return ZSTD_rotateRight_U16((U16)matches[0], head);
1000
+ if (nbChunks == 2) return ZSTD_rotateRight_U32((U32)matches[1] << 16 | (U32)matches[0], head);
1001
+ assert(nbChunks == 4);
1002
+ return ZSTD_rotateRight_U64((U64)matches[3] << 48 | (U64)matches[2] << 32 | (U64)matches[1] << 16 | (U64)matches[0], head);
1003
+ }
1004
+ #endif
1005
+
1006
+ #if defined(ZSTD_ARCH_ARM_NEON)
1007
+ FORCE_INLINE_TEMPLATE ZSTD_VecMask
1008
+ ZSTD_row_getNEONMask(const U32 rowEntries, const BYTE* const src, const BYTE tag, const U32 headGrouped)
1009
+ {
1010
+ assert((rowEntries == 16) || (rowEntries == 32) || rowEntries == 64);
1011
+ if (rowEntries == 16) {
1012
+ /* vshrn_n_u16 shifts by 4 every u16 and narrows to 8 lower bits.
1013
+ * After that groups of 4 bits represent the equalMask. We lower
1014
+ * all bits except the highest in these groups by doing AND with
1015
+ * 0x88 = 0b10001000.
1016
+ */
1017
+ const uint8x16_t chunk = vld1q_u8(src);
1018
+ const uint16x8_t equalMask = vreinterpretq_u16_u8(vceqq_u8(chunk, vdupq_n_u8(tag)));
1019
+ const uint8x8_t res = vshrn_n_u16(equalMask, 4);
1020
+ const U64 matches = vget_lane_u64(vreinterpret_u64_u8(res), 0);
1021
+ return ZSTD_rotateRight_U64(matches, headGrouped) & 0x8888888888888888ull;
1022
+ } else if (rowEntries == 32) {
1023
+ /* Same idea as with rowEntries == 16 but doing AND with
1024
+ * 0x55 = 0b01010101.
1025
+ */
1026
+ const uint16x8x2_t chunk = vld2q_u16((const uint16_t*)(const void*)src);
1027
+ const uint8x16_t chunk0 = vreinterpretq_u8_u16(chunk.val[0]);
1028
+ const uint8x16_t chunk1 = vreinterpretq_u8_u16(chunk.val[1]);
1029
+ const uint8x16_t dup = vdupq_n_u8(tag);
1030
+ const uint8x8_t t0 = vshrn_n_u16(vreinterpretq_u16_u8(vceqq_u8(chunk0, dup)), 6);
1031
+ const uint8x8_t t1 = vshrn_n_u16(vreinterpretq_u16_u8(vceqq_u8(chunk1, dup)), 6);
1032
+ const uint8x8_t res = vsli_n_u8(t0, t1, 4);
1033
+ const U64 matches = vget_lane_u64(vreinterpret_u64_u8(res), 0) ;
1034
+ return ZSTD_rotateRight_U64(matches, headGrouped) & 0x5555555555555555ull;
1035
+ } else { /* rowEntries == 64 */
1036
+ const uint8x16x4_t chunk = vld4q_u8(src);
1037
+ const uint8x16_t dup = vdupq_n_u8(tag);
1038
+ const uint8x16_t cmp0 = vceqq_u8(chunk.val[0], dup);
1039
+ const uint8x16_t cmp1 = vceqq_u8(chunk.val[1], dup);
1040
+ const uint8x16_t cmp2 = vceqq_u8(chunk.val[2], dup);
1041
+ const uint8x16_t cmp3 = vceqq_u8(chunk.val[3], dup);
1042
+
1043
+ const uint8x16_t t0 = vsriq_n_u8(cmp1, cmp0, 1);
1044
+ const uint8x16_t t1 = vsriq_n_u8(cmp3, cmp2, 1);
1045
+ const uint8x16_t t2 = vsriq_n_u8(t1, t0, 2);
1046
+ const uint8x16_t t3 = vsriq_n_u8(t2, t2, 4);
1047
+ const uint8x8_t t4 = vshrn_n_u16(vreinterpretq_u16_u8(t3), 4);
1048
+ const U64 matches = vget_lane_u64(vreinterpret_u64_u8(t4), 0);
1049
+ return ZSTD_rotateRight_U64(matches, headGrouped);
1050
+ }
1051
+ }
1052
+ #endif
1053
+
1054
+ /* Returns a ZSTD_VecMask (U64) that has the nth group (determined by
1055
+ * ZSTD_row_matchMaskGroupWidth) of bits set to 1 if the newly-computed "tag"
1056
+ * matches the hash at the nth position in a row of the tagTable.
1057
+ * Each row is a circular buffer beginning at the value of "headGrouped". So we
1058
+ * must rotate the "matches" bitfield to match up with the actual layout of the
1059
+ * entries within the hashTable */
1060
+ FORCE_INLINE_TEMPLATE ZSTD_VecMask
1061
+ ZSTD_row_getMatchMask(const BYTE* const tagRow, const BYTE tag, const U32 headGrouped, const U32 rowEntries)
1062
+ {
1063
+ const BYTE* const src = tagRow;
1064
+ assert((rowEntries == 16) || (rowEntries == 32) || rowEntries == 64);
1065
+ assert(rowEntries <= ZSTD_ROW_HASH_MAX_ENTRIES);
1066
+ assert(ZSTD_row_matchMaskGroupWidth(rowEntries) * rowEntries <= sizeof(ZSTD_VecMask) * 8);
1067
+
1068
+ #if defined(ZSTD_ARCH_X86_SSE2)
1069
+
1070
+ return ZSTD_row_getSSEMask(rowEntries / 16, src, tag, headGrouped);
1071
+
1072
+ #else /* SW or NEON-LE */
1073
+
1074
+ # if defined(ZSTD_ARCH_ARM_NEON)
1075
+ /* This NEON path only works for little endian - otherwise use SWAR below */
1076
+ if (MEM_isLittleEndian()) {
1077
+ return ZSTD_row_getNEONMask(rowEntries, src, tag, headGrouped);
1078
+ }
1079
+ # endif /* ZSTD_ARCH_ARM_NEON */
1080
+ /* SWAR */
1081
+ { const int chunkSize = sizeof(size_t);
1082
+ const size_t shiftAmount = ((chunkSize * 8) - chunkSize);
1083
+ const size_t xFF = ~((size_t)0);
1084
+ const size_t x01 = xFF / 0xFF;
1085
+ const size_t x80 = x01 << 7;
1086
+ const size_t splatChar = tag * x01;
1087
+ ZSTD_VecMask matches = 0;
1088
+ int i = rowEntries - chunkSize;
1089
+ assert((sizeof(size_t) == 4) || (sizeof(size_t) == 8));
1090
+ if (MEM_isLittleEndian()) { /* runtime check so have two loops */
1091
+ const size_t extractMagic = (xFF / 0x7F) >> chunkSize;
1092
+ do {
1093
+ size_t chunk = MEM_readST(&src[i]);
1094
+ chunk ^= splatChar;
1095
+ chunk = (((chunk | x80) - x01) | chunk) & x80;
1096
+ matches <<= chunkSize;
1097
+ matches |= (chunk * extractMagic) >> shiftAmount;
1098
+ i -= chunkSize;
1099
+ } while (i >= 0);
1100
+ } else { /* big endian: reverse bits during extraction */
1101
+ const size_t msb = xFF ^ (xFF >> 1);
1102
+ const size_t extractMagic = (msb / 0x1FF) | msb;
1103
+ do {
1104
+ size_t chunk = MEM_readST(&src[i]);
1105
+ chunk ^= splatChar;
1106
+ chunk = (((chunk | x80) - x01) | chunk) & x80;
1107
+ matches <<= chunkSize;
1108
+ matches |= ((chunk >> 7) * extractMagic) >> shiftAmount;
1109
+ i -= chunkSize;
1110
+ } while (i >= 0);
1111
+ }
1112
+ matches = ~matches;
1113
+ if (rowEntries == 16) {
1114
+ return ZSTD_rotateRight_U16((U16)matches, headGrouped);
1115
+ } else if (rowEntries == 32) {
1116
+ return ZSTD_rotateRight_U32((U32)matches, headGrouped);
1117
+ } else {
1118
+ return ZSTD_rotateRight_U64((U64)matches, headGrouped);
1119
+ }
1120
+ }
1121
+ #endif
1122
+ }
1123
+
1124
+ /* The high-level approach of the SIMD row based match finder is as follows:
1125
+ * - Figure out where to insert the new entry:
1126
+ * - Generate a hash for current input posistion and split it into a one byte of tag and `rowHashLog` bits of index.
1127
+ * - The hash is salted by a value that changes on every contex reset, so when the same table is used
1128
+ * we will avoid collisions that would otherwise slow us down by intorducing phantom matches.
1129
+ * - The hashTable is effectively split into groups or "rows" of 15 or 31 entries of U32, and the index determines
1130
+ * which row to insert into.
1131
+ * - Determine the correct position within the row to insert the entry into. Each row of 15 or 31 can
1132
+ * be considered as a circular buffer with a "head" index that resides in the tagTable (overall 16 or 32 bytes
1133
+ * per row).
1134
+ * - Use SIMD to efficiently compare the tags in the tagTable to the 1-byte tag calculated for the position and
1135
+ * generate a bitfield that we can cycle through to check the collisions in the hash table.
1136
+ * - Pick the longest match.
1137
+ * - Insert the tag into the equivalent row and position in the tagTable.
1138
+ */
1139
+ FORCE_INLINE_TEMPLATE
1140
+ ZSTD_ALLOW_POINTER_OVERFLOW_ATTR
1141
+ size_t ZSTD_RowFindBestMatch(
1142
+ ZSTD_matchState_t* ms,
1143
+ const BYTE* const ip, const BYTE* const iLimit,
1144
+ size_t* offsetPtr,
1145
+ const U32 mls, const ZSTD_dictMode_e dictMode,
1146
+ const U32 rowLog)
1147
+ {
1148
+ U32* const hashTable = ms->hashTable;
1149
+ BYTE* const tagTable = ms->tagTable;
1150
+ U32* const hashCache = ms->hashCache;
1151
+ const U32 hashLog = ms->rowHashLog;
1152
+ const ZSTD_compressionParameters* const cParams = &ms->cParams;
1153
+ const BYTE* const base = ms->window.base;
1154
+ const BYTE* const dictBase = ms->window.dictBase;
1155
+ const U32 dictLimit = ms->window.dictLimit;
1156
+ const BYTE* const prefixStart = base + dictLimit;
1157
+ const BYTE* const dictEnd = dictBase + dictLimit;
1158
+ const U32 curr = (U32)(ip-base);
1159
+ const U32 maxDistance = 1U << cParams->windowLog;
1160
+ const U32 lowestValid = ms->window.lowLimit;
1161
+ const U32 withinMaxDistance = (curr - lowestValid > maxDistance) ? curr - maxDistance : lowestValid;
1162
+ const U32 isDictionary = (ms->loadedDictEnd != 0);
1163
+ const U32 lowLimit = isDictionary ? lowestValid : withinMaxDistance;
1164
+ const U32 rowEntries = (1U << rowLog);
1165
+ const U32 rowMask = rowEntries - 1;
1166
+ const U32 cappedSearchLog = MIN(cParams->searchLog, rowLog); /* nb of searches is capped at nb entries per row */
1167
+ const U32 groupWidth = ZSTD_row_matchMaskGroupWidth(rowEntries);
1168
+ const U64 hashSalt = ms->hashSalt;
1169
+ U32 nbAttempts = 1U << cappedSearchLog;
1170
+ size_t ml=4-1;
1171
+ U32 hash;
1172
+
1173
+ /* DMS/DDS variables that may be referenced laster */
1174
+ const ZSTD_matchState_t* const dms = ms->dictMatchState;
1175
+
1176
+ /* Initialize the following variables to satisfy static analyzer */
1177
+ size_t ddsIdx = 0;
1178
+ U32 ddsExtraAttempts = 0; /* cctx hash tables are limited in searches, but allow extra searches into DDS */
1179
+ U32 dmsTag = 0;
1180
+ U32* dmsRow = NULL;
1181
+ BYTE* dmsTagRow = NULL;
1182
+
1183
+ if (dictMode == ZSTD_dedicatedDictSearch) {
1184
+ const U32 ddsHashLog = dms->cParams.hashLog - ZSTD_LAZY_DDSS_BUCKET_LOG;
1185
+ { /* Prefetch DDS hashtable entry */
1186
+ ddsIdx = ZSTD_hashPtr(ip, ddsHashLog, mls) << ZSTD_LAZY_DDSS_BUCKET_LOG;
1187
+ PREFETCH_L1(&dms->hashTable[ddsIdx]);
1188
+ }
1189
+ ddsExtraAttempts = cParams->searchLog > rowLog ? 1U << (cParams->searchLog - rowLog) : 0;
1190
+ }
1191
+
1192
+ if (dictMode == ZSTD_dictMatchState) {
1193
+ /* Prefetch DMS rows */
1194
+ U32* const dmsHashTable = dms->hashTable;
1195
+ BYTE* const dmsTagTable = dms->tagTable;
1196
+ U32 const dmsHash = (U32)ZSTD_hashPtr(ip, dms->rowHashLog + ZSTD_ROW_HASH_TAG_BITS, mls);
1197
+ U32 const dmsRelRow = (dmsHash >> ZSTD_ROW_HASH_TAG_BITS) << rowLog;
1198
+ dmsTag = dmsHash & ZSTD_ROW_HASH_TAG_MASK;
1199
+ dmsTagRow = (BYTE*)(dmsTagTable + dmsRelRow);
1200
+ dmsRow = dmsHashTable + dmsRelRow;
1201
+ ZSTD_row_prefetch(dmsHashTable, dmsTagTable, dmsRelRow, rowLog);
1202
+ }
1203
+
1204
+ /* Update the hashTable and tagTable up to (but not including) ip */
1205
+ if (!ms->lazySkipping) {
1206
+ ZSTD_row_update_internal(ms, ip, mls, rowLog, rowMask, 1 /* useCache */);
1207
+ hash = ZSTD_row_nextCachedHash(hashCache, hashTable, tagTable, base, curr, hashLog, rowLog, mls, hashSalt);
1208
+ } else {
1209
+ /* Stop inserting every position when in the lazy skipping mode.
1210
+ * The hash cache is also not kept up to date in this mode.
1211
+ */
1212
+ hash = (U32)ZSTD_hashPtrSalted(ip, hashLog + ZSTD_ROW_HASH_TAG_BITS, mls, hashSalt);
1213
+ ms->nextToUpdate = curr;
1214
+ }
1215
+ ms->hashSaltEntropy += hash; /* collect salt entropy */
1216
+
1217
+ { /* Get the hash for ip, compute the appropriate row */
1218
+ U32 const relRow = (hash >> ZSTD_ROW_HASH_TAG_BITS) << rowLog;
1219
+ U32 const tag = hash & ZSTD_ROW_HASH_TAG_MASK;
1220
+ U32* const row = hashTable + relRow;
1221
+ BYTE* tagRow = (BYTE*)(tagTable + relRow);
1222
+ U32 const headGrouped = (*tagRow & rowMask) * groupWidth;
1223
+ U32 matchBuffer[ZSTD_ROW_HASH_MAX_ENTRIES];
1224
+ size_t numMatches = 0;
1225
+ size_t currMatch = 0;
1226
+ ZSTD_VecMask matches = ZSTD_row_getMatchMask(tagRow, (BYTE)tag, headGrouped, rowEntries);
1227
+
1228
+ /* Cycle through the matches and prefetch */
1229
+ for (; (matches > 0) && (nbAttempts > 0); matches &= (matches - 1)) {
1230
+ U32 const matchPos = ((headGrouped + ZSTD_VecMask_next(matches)) / groupWidth) & rowMask;
1231
+ U32 const matchIndex = row[matchPos];
1232
+ if(matchPos == 0) continue;
1233
+ assert(numMatches < rowEntries);
1234
+ if (matchIndex < lowLimit)
1235
+ break;
1236
+ if ((dictMode != ZSTD_extDict) || matchIndex >= dictLimit) {
1237
+ PREFETCH_L1(base + matchIndex);
1238
+ } else {
1239
+ PREFETCH_L1(dictBase + matchIndex);
1240
+ }
1241
+ matchBuffer[numMatches++] = matchIndex;
1242
+ --nbAttempts;
1243
+ }
1244
+
1245
+ /* Speed opt: insert current byte into hashtable too. This allows us to avoid one iteration of the loop
1246
+ in ZSTD_row_update_internal() at the next search. */
1247
+ {
1248
+ U32 const pos = ZSTD_row_nextIndex(tagRow, rowMask);
1249
+ tagRow[pos] = (BYTE)tag;
1250
+ row[pos] = ms->nextToUpdate++;
1251
+ }
1252
+
1253
+ /* Return the longest match */
1254
+ for (; currMatch < numMatches; ++currMatch) {
1255
+ U32 const matchIndex = matchBuffer[currMatch];
1256
+ size_t currentMl=0;
1257
+ assert(matchIndex < curr);
1258
+ assert(matchIndex >= lowLimit);
1259
+
1260
+ if ((dictMode != ZSTD_extDict) || matchIndex >= dictLimit) {
1261
+ const BYTE* const match = base + matchIndex;
1262
+ assert(matchIndex >= dictLimit); /* ensures this is true if dictMode != ZSTD_extDict */
1263
+ /* read 4B starting from (match + ml + 1 - sizeof(U32)) */
1264
+ if (MEM_read32(match + ml - 3) == MEM_read32(ip + ml - 3)) /* potentially better */
1265
+ currentMl = ZSTD_count(ip, match, iLimit);
1266
+ } else {
1267
+ const BYTE* const match = dictBase + matchIndex;
1268
+ assert(match+4 <= dictEnd);
1269
+ if (MEM_read32(match) == MEM_read32(ip)) /* assumption : matchIndex <= dictLimit-4 (by table construction) */
1270
+ currentMl = ZSTD_count_2segments(ip+4, match+4, iLimit, dictEnd, prefixStart) + 4;
1271
+ }
1272
+
1273
+ /* Save best solution */
1274
+ if (currentMl > ml) {
1275
+ ml = currentMl;
1276
+ *offsetPtr = OFFSET_TO_OFFBASE(curr - matchIndex);
1277
+ if (ip+currentMl == iLimit) break; /* best possible, avoids read overflow on next attempt */
1278
+ }
1279
+ }
1280
+ }
1281
+
1282
+ assert(nbAttempts <= (1U << ZSTD_SEARCHLOG_MAX)); /* Check we haven't underflowed. */
1283
+ if (dictMode == ZSTD_dedicatedDictSearch) {
1284
+ ml = ZSTD_dedicatedDictSearch_lazy_search(offsetPtr, ml, nbAttempts + ddsExtraAttempts, dms,
1285
+ ip, iLimit, prefixStart, curr, dictLimit, ddsIdx);
1286
+ } else if (dictMode == ZSTD_dictMatchState) {
1287
+ /* TODO: Measure and potentially add prefetching to DMS */
1288
+ const U32 dmsLowestIndex = dms->window.dictLimit;
1289
+ const BYTE* const dmsBase = dms->window.base;
1290
+ const BYTE* const dmsEnd = dms->window.nextSrc;
1291
+ const U32 dmsSize = (U32)(dmsEnd - dmsBase);
1292
+ const U32 dmsIndexDelta = dictLimit - dmsSize;
1293
+
1294
+ { U32 const headGrouped = (*dmsTagRow & rowMask) * groupWidth;
1295
+ U32 matchBuffer[ZSTD_ROW_HASH_MAX_ENTRIES];
1296
+ size_t numMatches = 0;
1297
+ size_t currMatch = 0;
1298
+ ZSTD_VecMask matches = ZSTD_row_getMatchMask(dmsTagRow, (BYTE)dmsTag, headGrouped, rowEntries);
1299
+
1300
+ for (; (matches > 0) && (nbAttempts > 0); matches &= (matches - 1)) {
1301
+ U32 const matchPos = ((headGrouped + ZSTD_VecMask_next(matches)) / groupWidth) & rowMask;
1302
+ U32 const matchIndex = dmsRow[matchPos];
1303
+ if(matchPos == 0) continue;
1304
+ if (matchIndex < dmsLowestIndex)
1305
+ break;
1306
+ PREFETCH_L1(dmsBase + matchIndex);
1307
+ matchBuffer[numMatches++] = matchIndex;
1308
+ --nbAttempts;
1309
+ }
1310
+
1311
+ /* Return the longest match */
1312
+ for (; currMatch < numMatches; ++currMatch) {
1313
+ U32 const matchIndex = matchBuffer[currMatch];
1314
+ size_t currentMl=0;
1315
+ assert(matchIndex >= dmsLowestIndex);
1316
+ assert(matchIndex < curr);
1317
+
1318
+ { const BYTE* const match = dmsBase + matchIndex;
1319
+ assert(match+4 <= dmsEnd);
1320
+ if (MEM_read32(match) == MEM_read32(ip))
1321
+ currentMl = ZSTD_count_2segments(ip+4, match+4, iLimit, dmsEnd, prefixStart) + 4;
1322
+ }
1323
+
1324
+ if (currentMl > ml) {
1325
+ ml = currentMl;
1326
+ assert(curr > matchIndex + dmsIndexDelta);
1327
+ *offsetPtr = OFFSET_TO_OFFBASE(curr - (matchIndex + dmsIndexDelta));
1328
+ if (ip+currentMl == iLimit) break;
1329
+ }
1330
+ }
1331
+ }
1332
+ }
1333
+ return ml;
1334
+ }
1335
+
1336
+
1337
+ /**
1338
+ * Generate search functions templated on (dictMode, mls, rowLog).
1339
+ * These functions are outlined for code size & compilation time.
1340
+ * ZSTD_searchMax() dispatches to the correct implementation function.
1341
+ *
1342
+ * TODO: The start of the search function involves loading and calculating a
1343
+ * bunch of constants from the ZSTD_matchState_t. These computations could be
1344
+ * done in an initialization function, and saved somewhere in the match state.
1345
+ * Then we could pass a pointer to the saved state instead of the match state,
1346
+ * and avoid duplicate computations.
1347
+ *
1348
+ * TODO: Move the match re-winding into searchMax. This improves compression
1349
+ * ratio, and unlocks further simplifications with the next TODO.
1350
+ *
1351
+ * TODO: Try moving the repcode search into searchMax. After the re-winding
1352
+ * and repcode search are in searchMax, there is no more logic in the match
1353
+ * finder loop that requires knowledge about the dictMode. So we should be
1354
+ * able to avoid force inlining it, and we can join the extDict loop with
1355
+ * the single segment loop. It should go in searchMax instead of its own
1356
+ * function to avoid having multiple virtual function calls per search.
1357
+ */
1358
+
1359
+ #define ZSTD_BT_SEARCH_FN(dictMode, mls) ZSTD_BtFindBestMatch_##dictMode##_##mls
1360
+ #define ZSTD_HC_SEARCH_FN(dictMode, mls) ZSTD_HcFindBestMatch_##dictMode##_##mls
1361
+ #define ZSTD_ROW_SEARCH_FN(dictMode, mls, rowLog) ZSTD_RowFindBestMatch_##dictMode##_##mls##_##rowLog
1362
+
1363
+ #define ZSTD_SEARCH_FN_ATTRS FORCE_NOINLINE
1364
+
1365
+ #define GEN_ZSTD_BT_SEARCH_FN(dictMode, mls) \
1366
+ ZSTD_SEARCH_FN_ATTRS size_t ZSTD_BT_SEARCH_FN(dictMode, mls)( \
1367
+ ZSTD_matchState_t* ms, \
1368
+ const BYTE* ip, const BYTE* const iLimit, \
1369
+ size_t* offBasePtr) \
1370
+ { \
1371
+ assert(MAX(4, MIN(6, ms->cParams.minMatch)) == mls); \
1372
+ return ZSTD_BtFindBestMatch(ms, ip, iLimit, offBasePtr, mls, ZSTD_##dictMode); \
1373
+ } \
1374
+
1375
+ #define GEN_ZSTD_HC_SEARCH_FN(dictMode, mls) \
1376
+ ZSTD_SEARCH_FN_ATTRS size_t ZSTD_HC_SEARCH_FN(dictMode, mls)( \
1377
+ ZSTD_matchState_t* ms, \
1378
+ const BYTE* ip, const BYTE* const iLimit, \
1379
+ size_t* offsetPtr) \
1380
+ { \
1381
+ assert(MAX(4, MIN(6, ms->cParams.minMatch)) == mls); \
1382
+ return ZSTD_HcFindBestMatch(ms, ip, iLimit, offsetPtr, mls, ZSTD_##dictMode); \
1383
+ } \
1384
+
1385
+ #define GEN_ZSTD_ROW_SEARCH_FN(dictMode, mls, rowLog) \
1386
+ ZSTD_SEARCH_FN_ATTRS size_t ZSTD_ROW_SEARCH_FN(dictMode, mls, rowLog)( \
1387
+ ZSTD_matchState_t* ms, \
1388
+ const BYTE* ip, const BYTE* const iLimit, \
1389
+ size_t* offsetPtr) \
1390
+ { \
1391
+ assert(MAX(4, MIN(6, ms->cParams.minMatch)) == mls); \
1392
+ assert(MAX(4, MIN(6, ms->cParams.searchLog)) == rowLog); \
1393
+ return ZSTD_RowFindBestMatch(ms, ip, iLimit, offsetPtr, mls, ZSTD_##dictMode, rowLog); \
1394
+ } \
1395
+
1396
+ #define ZSTD_FOR_EACH_ROWLOG(X, dictMode, mls) \
1397
+ X(dictMode, mls, 4) \
1398
+ X(dictMode, mls, 5) \
1399
+ X(dictMode, mls, 6)
1400
+
1401
+ #define ZSTD_FOR_EACH_MLS_ROWLOG(X, dictMode) \
1402
+ ZSTD_FOR_EACH_ROWLOG(X, dictMode, 4) \
1403
+ ZSTD_FOR_EACH_ROWLOG(X, dictMode, 5) \
1404
+ ZSTD_FOR_EACH_ROWLOG(X, dictMode, 6)
1405
+
1406
+ #define ZSTD_FOR_EACH_MLS(X, dictMode) \
1407
+ X(dictMode, 4) \
1408
+ X(dictMode, 5) \
1409
+ X(dictMode, 6)
1410
+
1411
+ #define ZSTD_FOR_EACH_DICT_MODE(X, ...) \
1412
+ X(__VA_ARGS__, noDict) \
1413
+ X(__VA_ARGS__, extDict) \
1414
+ X(__VA_ARGS__, dictMatchState) \
1415
+ X(__VA_ARGS__, dedicatedDictSearch)
1416
+
1417
+ /* Generate row search fns for each combination of (dictMode, mls, rowLog) */
1418
+ ZSTD_FOR_EACH_DICT_MODE(ZSTD_FOR_EACH_MLS_ROWLOG, GEN_ZSTD_ROW_SEARCH_FN)
1419
+ /* Generate binary Tree search fns for each combination of (dictMode, mls) */
1420
+ ZSTD_FOR_EACH_DICT_MODE(ZSTD_FOR_EACH_MLS, GEN_ZSTD_BT_SEARCH_FN)
1421
+ /* Generate hash chain search fns for each combination of (dictMode, mls) */
1422
+ ZSTD_FOR_EACH_DICT_MODE(ZSTD_FOR_EACH_MLS, GEN_ZSTD_HC_SEARCH_FN)
1423
+
1424
+ typedef enum { search_hashChain=0, search_binaryTree=1, search_rowHash=2 } searchMethod_e;
1425
+
1426
+ #define GEN_ZSTD_CALL_BT_SEARCH_FN(dictMode, mls) \
1427
+ case mls: \
1428
+ return ZSTD_BT_SEARCH_FN(dictMode, mls)(ms, ip, iend, offsetPtr);
1429
+ #define GEN_ZSTD_CALL_HC_SEARCH_FN(dictMode, mls) \
1430
+ case mls: \
1431
+ return ZSTD_HC_SEARCH_FN(dictMode, mls)(ms, ip, iend, offsetPtr);
1432
+ #define GEN_ZSTD_CALL_ROW_SEARCH_FN(dictMode, mls, rowLog) \
1433
+ case rowLog: \
1434
+ return ZSTD_ROW_SEARCH_FN(dictMode, mls, rowLog)(ms, ip, iend, offsetPtr);
1435
+
1436
+ #define ZSTD_SWITCH_MLS(X, dictMode) \
1437
+ switch (mls) { \
1438
+ ZSTD_FOR_EACH_MLS(X, dictMode) \
1439
+ }
1440
+
1441
+ #define ZSTD_SWITCH_ROWLOG(dictMode, mls) \
1442
+ case mls: \
1443
+ switch (rowLog) { \
1444
+ ZSTD_FOR_EACH_ROWLOG(GEN_ZSTD_CALL_ROW_SEARCH_FN, dictMode, mls) \
1445
+ } \
1446
+ ZSTD_UNREACHABLE; \
1447
+ break;
1448
+
1449
+ #define ZSTD_SWITCH_SEARCH_METHOD(dictMode) \
1450
+ switch (searchMethod) { \
1451
+ case search_hashChain: \
1452
+ ZSTD_SWITCH_MLS(GEN_ZSTD_CALL_HC_SEARCH_FN, dictMode) \
1453
+ break; \
1454
+ case search_binaryTree: \
1455
+ ZSTD_SWITCH_MLS(GEN_ZSTD_CALL_BT_SEARCH_FN, dictMode) \
1456
+ break; \
1457
+ case search_rowHash: \
1458
+ ZSTD_SWITCH_MLS(ZSTD_SWITCH_ROWLOG, dictMode) \
1459
+ break; \
1460
+ } \
1461
+ ZSTD_UNREACHABLE;
1462
+
1463
+ /**
1464
+ * Searches for the longest match at @p ip.
1465
+ * Dispatches to the correct implementation function based on the
1466
+ * (searchMethod, dictMode, mls, rowLog). We use switch statements
1467
+ * here instead of using an indirect function call through a function
1468
+ * pointer because after Spectre and Meltdown mitigations, indirect
1469
+ * function calls can be very costly, especially in the kernel.
1470
+ *
1471
+ * NOTE: dictMode and searchMethod should be templated, so those switch
1472
+ * statements should be optimized out. Only the mls & rowLog switches
1473
+ * should be left.
1474
+ *
1475
+ * @param ms The match state.
1476
+ * @param ip The position to search at.
1477
+ * @param iend The end of the input data.
1478
+ * @param[out] offsetPtr Stores the match offset into this pointer.
1479
+ * @param mls The minimum search length, in the range [4, 6].
1480
+ * @param rowLog The row log (if applicable), in the range [4, 6].
1481
+ * @param searchMethod The search method to use (templated).
1482
+ * @param dictMode The dictMode (templated).
1483
+ *
1484
+ * @returns The length of the longest match found, or < mls if no match is found.
1485
+ * If a match is found its offset is stored in @p offsetPtr.
1486
+ */
1487
+ FORCE_INLINE_TEMPLATE size_t ZSTD_searchMax(
1488
+ ZSTD_matchState_t* ms,
1489
+ const BYTE* ip,
1490
+ const BYTE* iend,
1491
+ size_t* offsetPtr,
1492
+ U32 const mls,
1493
+ U32 const rowLog,
1494
+ searchMethod_e const searchMethod,
1495
+ ZSTD_dictMode_e const dictMode)
1496
+ {
1497
+ if (dictMode == ZSTD_noDict) {
1498
+ ZSTD_SWITCH_SEARCH_METHOD(noDict)
1499
+ } else if (dictMode == ZSTD_extDict) {
1500
+ ZSTD_SWITCH_SEARCH_METHOD(extDict)
1501
+ } else if (dictMode == ZSTD_dictMatchState) {
1502
+ ZSTD_SWITCH_SEARCH_METHOD(dictMatchState)
1503
+ } else if (dictMode == ZSTD_dedicatedDictSearch) {
1504
+ ZSTD_SWITCH_SEARCH_METHOD(dedicatedDictSearch)
1505
+ }
1506
+ ZSTD_UNREACHABLE;
1507
+ return 0;
1508
+ }
1509
+
1510
+ /* *******************************
1511
+ * Common parser - lazy strategy
1512
+ *********************************/
1513
+
1514
+ FORCE_INLINE_TEMPLATE
1515
+ ZSTD_ALLOW_POINTER_OVERFLOW_ATTR
1516
+ size_t ZSTD_compressBlock_lazy_generic(
1517
+ ZSTD_matchState_t* ms, seqStore_t* seqStore,
1518
+ U32 rep[ZSTD_REP_NUM],
1519
+ const void* src, size_t srcSize,
1520
+ const searchMethod_e searchMethod, const U32 depth,
1521
+ ZSTD_dictMode_e const dictMode)
1522
+ {
1523
+ const BYTE* const istart = (const BYTE*)src;
1524
+ const BYTE* ip = istart;
1525
+ const BYTE* anchor = istart;
1526
+ const BYTE* const iend = istart + srcSize;
1527
+ const BYTE* const ilimit = (searchMethod == search_rowHash) ? iend - 8 - ZSTD_ROW_HASH_CACHE_SIZE : iend - 8;
1528
+ const BYTE* const base = ms->window.base;
1529
+ const U32 prefixLowestIndex = ms->window.dictLimit;
1530
+ const BYTE* const prefixLowest = base + prefixLowestIndex;
1531
+ const U32 mls = BOUNDED(4, ms->cParams.minMatch, 6);
1532
+ const U32 rowLog = BOUNDED(4, ms->cParams.searchLog, 6);
1533
+
1534
+ U32 offset_1 = rep[0], offset_2 = rep[1];
1535
+ U32 offsetSaved1 = 0, offsetSaved2 = 0;
1536
+
1537
+ const int isDMS = dictMode == ZSTD_dictMatchState;
1538
+ const int isDDS = dictMode == ZSTD_dedicatedDictSearch;
1539
+ const int isDxS = isDMS || isDDS;
1540
+ const ZSTD_matchState_t* const dms = ms->dictMatchState;
1541
+ const U32 dictLowestIndex = isDxS ? dms->window.dictLimit : 0;
1542
+ const BYTE* const dictBase = isDxS ? dms->window.base : NULL;
1543
+ const BYTE* const dictLowest = isDxS ? dictBase + dictLowestIndex : NULL;
1544
+ const BYTE* const dictEnd = isDxS ? dms->window.nextSrc : NULL;
1545
+ const U32 dictIndexDelta = isDxS ?
1546
+ prefixLowestIndex - (U32)(dictEnd - dictBase) :
1547
+ 0;
1548
+ const U32 dictAndPrefixLength = (U32)((ip - prefixLowest) + (dictEnd - dictLowest));
1549
+
1550
+ DEBUGLOG(5, "ZSTD_compressBlock_lazy_generic (dictMode=%u) (searchFunc=%u)", (U32)dictMode, (U32)searchMethod);
1551
+ ip += (dictAndPrefixLength == 0);
1552
+ if (dictMode == ZSTD_noDict) {
1553
+ U32 const curr = (U32)(ip - base);
1554
+ U32 const windowLow = ZSTD_getLowestPrefixIndex(ms, curr, ms->cParams.windowLog);
1555
+ U32 const maxRep = curr - windowLow;
1556
+ if (offset_2 > maxRep) offsetSaved2 = offset_2, offset_2 = 0;
1557
+ if (offset_1 > maxRep) offsetSaved1 = offset_1, offset_1 = 0;
1558
+ }
1559
+ if (isDxS) {
1560
+ /* dictMatchState repCode checks don't currently handle repCode == 0
1561
+ * disabling. */
1562
+ assert(offset_1 <= dictAndPrefixLength);
1563
+ assert(offset_2 <= dictAndPrefixLength);
1564
+ }
1565
+
1566
+ /* Reset the lazy skipping state */
1567
+ ms->lazySkipping = 0;
1568
+
1569
+ if (searchMethod == search_rowHash) {
1570
+ ZSTD_row_fillHashCache(ms, base, rowLog, mls, ms->nextToUpdate, ilimit);
1571
+ }
1572
+
1573
+ /* Match Loop */
1574
+ #if defined(__GNUC__) && defined(__x86_64__)
1575
+ /* I've measured random a 5% speed loss on levels 5 & 6 (greedy) when the
1576
+ * code alignment is perturbed. To fix the instability align the loop on 32-bytes.
1577
+ */
1578
+ __asm__(".p2align 5");
1579
+ #endif
1580
+ while (ip < ilimit) {
1581
+ size_t matchLength=0;
1582
+ size_t offBase = REPCODE1_TO_OFFBASE;
1583
+ const BYTE* start=ip+1;
1584
+ DEBUGLOG(7, "search baseline (depth 0)");
1585
+
1586
+ /* check repCode */
1587
+ if (isDxS) {
1588
+ const U32 repIndex = (U32)(ip - base) + 1 - offset_1;
1589
+ const BYTE* repMatch = ((dictMode == ZSTD_dictMatchState || dictMode == ZSTD_dedicatedDictSearch)
1590
+ && repIndex < prefixLowestIndex) ?
1591
+ dictBase + (repIndex - dictIndexDelta) :
1592
+ base + repIndex;
1593
+ if (((U32)((prefixLowestIndex-1) - repIndex) >= 3 /* intentional underflow */)
1594
+ && (MEM_read32(repMatch) == MEM_read32(ip+1)) ) {
1595
+ const BYTE* repMatchEnd = repIndex < prefixLowestIndex ? dictEnd : iend;
1596
+ matchLength = ZSTD_count_2segments(ip+1+4, repMatch+4, iend, repMatchEnd, prefixLowest) + 4;
1597
+ if (depth==0) goto _storeSequence;
1598
+ }
1599
+ }
1600
+ if ( dictMode == ZSTD_noDict
1601
+ && ((offset_1 > 0) & (MEM_read32(ip+1-offset_1) == MEM_read32(ip+1)))) {
1602
+ matchLength = ZSTD_count(ip+1+4, ip+1+4-offset_1, iend) + 4;
1603
+ if (depth==0) goto _storeSequence;
1604
+ }
1605
+
1606
+ /* first search (depth 0) */
1607
+ { size_t offbaseFound = 999999999;
1608
+ size_t const ml2 = ZSTD_searchMax(ms, ip, iend, &offbaseFound, mls, rowLog, searchMethod, dictMode);
1609
+ if (ml2 > matchLength)
1610
+ matchLength = ml2, start = ip, offBase = offbaseFound;
1611
+ }
1612
+
1613
+ if (matchLength < 4) {
1614
+ size_t const step = ((size_t)(ip-anchor) >> kSearchStrength) + 1; /* jump faster over incompressible sections */;
1615
+ ip += step;
1616
+ /* Enter the lazy skipping mode once we are skipping more than 8 bytes at a time.
1617
+ * In this mode we stop inserting every position into our tables, and only insert
1618
+ * positions that we search, which is one in step positions.
1619
+ * The exact cutoff is flexible, I've just chosen a number that is reasonably high,
1620
+ * so we minimize the compression ratio loss in "normal" scenarios. This mode gets
1621
+ * triggered once we've gone 2KB without finding any matches.
1622
+ */
1623
+ ms->lazySkipping = step > kLazySkippingStep;
1624
+ continue;
1625
+ }
1626
+
1627
+ /* let's try to find a better solution */
1628
+ if (depth>=1)
1629
+ while (ip<ilimit) {
1630
+ DEBUGLOG(7, "search depth 1");
1631
+ ip ++;
1632
+ if ( (dictMode == ZSTD_noDict)
1633
+ && (offBase) && ((offset_1>0) & (MEM_read32(ip) == MEM_read32(ip - offset_1)))) {
1634
+ size_t const mlRep = ZSTD_count(ip+4, ip+4-offset_1, iend) + 4;
1635
+ int const gain2 = (int)(mlRep * 3);
1636
+ int const gain1 = (int)(matchLength*3 - ZSTD_highbit32((U32)offBase) + 1);
1637
+ if ((mlRep >= 4) && (gain2 > gain1))
1638
+ matchLength = mlRep, offBase = REPCODE1_TO_OFFBASE, start = ip;
1639
+ }
1640
+ if (isDxS) {
1641
+ const U32 repIndex = (U32)(ip - base) - offset_1;
1642
+ const BYTE* repMatch = repIndex < prefixLowestIndex ?
1643
+ dictBase + (repIndex - dictIndexDelta) :
1644
+ base + repIndex;
1645
+ if (((U32)((prefixLowestIndex-1) - repIndex) >= 3 /* intentional underflow */)
1646
+ && (MEM_read32(repMatch) == MEM_read32(ip)) ) {
1647
+ const BYTE* repMatchEnd = repIndex < prefixLowestIndex ? dictEnd : iend;
1648
+ size_t const mlRep = ZSTD_count_2segments(ip+4, repMatch+4, iend, repMatchEnd, prefixLowest) + 4;
1649
+ int const gain2 = (int)(mlRep * 3);
1650
+ int const gain1 = (int)(matchLength*3 - ZSTD_highbit32((U32)offBase) + 1);
1651
+ if ((mlRep >= 4) && (gain2 > gain1))
1652
+ matchLength = mlRep, offBase = REPCODE1_TO_OFFBASE, start = ip;
1653
+ }
1654
+ }
1655
+ { size_t ofbCandidate=999999999;
1656
+ size_t const ml2 = ZSTD_searchMax(ms, ip, iend, &ofbCandidate, mls, rowLog, searchMethod, dictMode);
1657
+ int const gain2 = (int)(ml2*4 - ZSTD_highbit32((U32)ofbCandidate)); /* raw approx */
1658
+ int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offBase) + 4);
1659
+ if ((ml2 >= 4) && (gain2 > gain1)) {
1660
+ matchLength = ml2, offBase = ofbCandidate, start = ip;
1661
+ continue; /* search a better one */
1662
+ } }
1663
+
1664
+ /* let's find an even better one */
1665
+ if ((depth==2) && (ip<ilimit)) {
1666
+ DEBUGLOG(7, "search depth 2");
1667
+ ip ++;
1668
+ if ( (dictMode == ZSTD_noDict)
1669
+ && (offBase) && ((offset_1>0) & (MEM_read32(ip) == MEM_read32(ip - offset_1)))) {
1670
+ size_t const mlRep = ZSTD_count(ip+4, ip+4-offset_1, iend) + 4;
1671
+ int const gain2 = (int)(mlRep * 4);
1672
+ int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offBase) + 1);
1673
+ if ((mlRep >= 4) && (gain2 > gain1))
1674
+ matchLength = mlRep, offBase = REPCODE1_TO_OFFBASE, start = ip;
1675
+ }
1676
+ if (isDxS) {
1677
+ const U32 repIndex = (U32)(ip - base) - offset_1;
1678
+ const BYTE* repMatch = repIndex < prefixLowestIndex ?
1679
+ dictBase + (repIndex - dictIndexDelta) :
1680
+ base + repIndex;
1681
+ if (((U32)((prefixLowestIndex-1) - repIndex) >= 3 /* intentional underflow */)
1682
+ && (MEM_read32(repMatch) == MEM_read32(ip)) ) {
1683
+ const BYTE* repMatchEnd = repIndex < prefixLowestIndex ? dictEnd : iend;
1684
+ size_t const mlRep = ZSTD_count_2segments(ip+4, repMatch+4, iend, repMatchEnd, prefixLowest) + 4;
1685
+ int const gain2 = (int)(mlRep * 4);
1686
+ int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offBase) + 1);
1687
+ if ((mlRep >= 4) && (gain2 > gain1))
1688
+ matchLength = mlRep, offBase = REPCODE1_TO_OFFBASE, start = ip;
1689
+ }
1690
+ }
1691
+ { size_t ofbCandidate=999999999;
1692
+ size_t const ml2 = ZSTD_searchMax(ms, ip, iend, &ofbCandidate, mls, rowLog, searchMethod, dictMode);
1693
+ int const gain2 = (int)(ml2*4 - ZSTD_highbit32((U32)ofbCandidate)); /* raw approx */
1694
+ int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offBase) + 7);
1695
+ if ((ml2 >= 4) && (gain2 > gain1)) {
1696
+ matchLength = ml2, offBase = ofbCandidate, start = ip;
1697
+ continue;
1698
+ } } }
1699
+ break; /* nothing found : store previous solution */
1700
+ }
1701
+
1702
+ /* NOTE:
1703
+ * Pay attention that `start[-value]` can lead to strange undefined behavior
1704
+ * notably if `value` is unsigned, resulting in a large positive `-value`.
1705
+ */
1706
+ /* catch up */
1707
+ if (OFFBASE_IS_OFFSET(offBase)) {
1708
+ if (dictMode == ZSTD_noDict) {
1709
+ while ( ((start > anchor) & (start - OFFBASE_TO_OFFSET(offBase) > prefixLowest))
1710
+ && (start[-1] == (start-OFFBASE_TO_OFFSET(offBase))[-1]) ) /* only search for offset within prefix */
1711
+ { start--; matchLength++; }
1712
+ }
1713
+ if (isDxS) {
1714
+ U32 const matchIndex = (U32)((size_t)(start-base) - OFFBASE_TO_OFFSET(offBase));
1715
+ const BYTE* match = (matchIndex < prefixLowestIndex) ? dictBase + matchIndex - dictIndexDelta : base + matchIndex;
1716
+ const BYTE* const mStart = (matchIndex < prefixLowestIndex) ? dictLowest : prefixLowest;
1717
+ while ((start>anchor) && (match>mStart) && (start[-1] == match[-1])) { start--; match--; matchLength++; } /* catch up */
1718
+ }
1719
+ offset_2 = offset_1; offset_1 = (U32)OFFBASE_TO_OFFSET(offBase);
1720
+ }
1721
+ /* store sequence */
1722
+ _storeSequence:
1723
+ { size_t const litLength = (size_t)(start - anchor);
1724
+ ZSTD_storeSeq(seqStore, litLength, anchor, iend, (U32)offBase, matchLength);
1725
+ anchor = ip = start + matchLength;
1726
+ }
1727
+ if (ms->lazySkipping) {
1728
+ /* We've found a match, disable lazy skipping mode, and refill the hash cache. */
1729
+ if (searchMethod == search_rowHash) {
1730
+ ZSTD_row_fillHashCache(ms, base, rowLog, mls, ms->nextToUpdate, ilimit);
1731
+ }
1732
+ ms->lazySkipping = 0;
1733
+ }
1734
+
1735
+ /* check immediate repcode */
1736
+ if (isDxS) {
1737
+ while (ip <= ilimit) {
1738
+ U32 const current2 = (U32)(ip-base);
1739
+ U32 const repIndex = current2 - offset_2;
1740
+ const BYTE* repMatch = repIndex < prefixLowestIndex ?
1741
+ dictBase - dictIndexDelta + repIndex :
1742
+ base + repIndex;
1743
+ if ( ((U32)((prefixLowestIndex-1) - (U32)repIndex) >= 3 /* intentional overflow */)
1744
+ && (MEM_read32(repMatch) == MEM_read32(ip)) ) {
1745
+ const BYTE* const repEnd2 = repIndex < prefixLowestIndex ? dictEnd : iend;
1746
+ matchLength = ZSTD_count_2segments(ip+4, repMatch+4, iend, repEnd2, prefixLowest) + 4;
1747
+ offBase = offset_2; offset_2 = offset_1; offset_1 = (U32)offBase; /* swap offset_2 <=> offset_1 */
1748
+ ZSTD_storeSeq(seqStore, 0, anchor, iend, REPCODE1_TO_OFFBASE, matchLength);
1749
+ ip += matchLength;
1750
+ anchor = ip;
1751
+ continue;
1752
+ }
1753
+ break;
1754
+ }
1755
+ }
1756
+
1757
+ if (dictMode == ZSTD_noDict) {
1758
+ while ( ((ip <= ilimit) & (offset_2>0))
1759
+ && (MEM_read32(ip) == MEM_read32(ip - offset_2)) ) {
1760
+ /* store sequence */
1761
+ matchLength = ZSTD_count(ip+4, ip+4-offset_2, iend) + 4;
1762
+ offBase = offset_2; offset_2 = offset_1; offset_1 = (U32)offBase; /* swap repcodes */
1763
+ ZSTD_storeSeq(seqStore, 0, anchor, iend, REPCODE1_TO_OFFBASE, matchLength);
1764
+ ip += matchLength;
1765
+ anchor = ip;
1766
+ continue; /* faster when present ... (?) */
1767
+ } } }
1768
+
1769
+ /* If offset_1 started invalid (offsetSaved1 != 0) and became valid (offset_1 != 0),
1770
+ * rotate saved offsets. See comment in ZSTD_compressBlock_fast_noDict for more context. */
1771
+ offsetSaved2 = ((offsetSaved1 != 0) && (offset_1 != 0)) ? offsetSaved1 : offsetSaved2;
1772
+
1773
+ /* save reps for next block */
1774
+ rep[0] = offset_1 ? offset_1 : offsetSaved1;
1775
+ rep[1] = offset_2 ? offset_2 : offsetSaved2;
1776
+
1777
+ /* Return the last literals size */
1778
+ return (size_t)(iend - anchor);
1779
+ }
1780
+ #endif /* build exclusions */
1781
+
1782
+
1783
+ #ifndef ZSTD_EXCLUDE_GREEDY_BLOCK_COMPRESSOR
1784
+ size_t ZSTD_compressBlock_greedy(
1785
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1786
+ void const* src, size_t srcSize)
1787
+ {
1788
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 0, ZSTD_noDict);
1789
+ }
1790
+
1791
+ size_t ZSTD_compressBlock_greedy_dictMatchState(
1792
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1793
+ void const* src, size_t srcSize)
1794
+ {
1795
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 0, ZSTD_dictMatchState);
1796
+ }
1797
+
1798
+ size_t ZSTD_compressBlock_greedy_dedicatedDictSearch(
1799
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1800
+ void const* src, size_t srcSize)
1801
+ {
1802
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 0, ZSTD_dedicatedDictSearch);
1803
+ }
1804
+
1805
+ size_t ZSTD_compressBlock_greedy_row(
1806
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1807
+ void const* src, size_t srcSize)
1808
+ {
1809
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_rowHash, 0, ZSTD_noDict);
1810
+ }
1811
+
1812
+ size_t ZSTD_compressBlock_greedy_dictMatchState_row(
1813
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1814
+ void const* src, size_t srcSize)
1815
+ {
1816
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_rowHash, 0, ZSTD_dictMatchState);
1817
+ }
1818
+
1819
+ size_t ZSTD_compressBlock_greedy_dedicatedDictSearch_row(
1820
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1821
+ void const* src, size_t srcSize)
1822
+ {
1823
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_rowHash, 0, ZSTD_dedicatedDictSearch);
1824
+ }
1825
+ #endif
1826
+
1827
+ #ifndef ZSTD_EXCLUDE_LAZY_BLOCK_COMPRESSOR
1828
+ size_t ZSTD_compressBlock_lazy(
1829
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1830
+ void const* src, size_t srcSize)
1831
+ {
1832
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 1, ZSTD_noDict);
1833
+ }
1834
+
1835
+ size_t ZSTD_compressBlock_lazy_dictMatchState(
1836
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1837
+ void const* src, size_t srcSize)
1838
+ {
1839
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 1, ZSTD_dictMatchState);
1840
+ }
1841
+
1842
+ size_t ZSTD_compressBlock_lazy_dedicatedDictSearch(
1843
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1844
+ void const* src, size_t srcSize)
1845
+ {
1846
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 1, ZSTD_dedicatedDictSearch);
1847
+ }
1848
+
1849
+ size_t ZSTD_compressBlock_lazy_row(
1850
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1851
+ void const* src, size_t srcSize)
1852
+ {
1853
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_rowHash, 1, ZSTD_noDict);
1854
+ }
1855
+
1856
+ size_t ZSTD_compressBlock_lazy_dictMatchState_row(
1857
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1858
+ void const* src, size_t srcSize)
1859
+ {
1860
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_rowHash, 1, ZSTD_dictMatchState);
1861
+ }
1862
+
1863
+ size_t ZSTD_compressBlock_lazy_dedicatedDictSearch_row(
1864
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1865
+ void const* src, size_t srcSize)
1866
+ {
1867
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_rowHash, 1, ZSTD_dedicatedDictSearch);
1868
+ }
1869
+ #endif
1870
+
1871
+ #ifndef ZSTD_EXCLUDE_LAZY2_BLOCK_COMPRESSOR
1872
+ size_t ZSTD_compressBlock_lazy2(
1873
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1874
+ void const* src, size_t srcSize)
1875
+ {
1876
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 2, ZSTD_noDict);
1877
+ }
1878
+
1879
+ size_t ZSTD_compressBlock_lazy2_dictMatchState(
1880
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1881
+ void const* src, size_t srcSize)
1882
+ {
1883
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 2, ZSTD_dictMatchState);
1884
+ }
1885
+
1886
+ size_t ZSTD_compressBlock_lazy2_dedicatedDictSearch(
1887
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1888
+ void const* src, size_t srcSize)
1889
+ {
1890
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 2, ZSTD_dedicatedDictSearch);
1891
+ }
1892
+
1893
+ size_t ZSTD_compressBlock_lazy2_row(
1894
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1895
+ void const* src, size_t srcSize)
1896
+ {
1897
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_rowHash, 2, ZSTD_noDict);
1898
+ }
1899
+
1900
+ size_t ZSTD_compressBlock_lazy2_dictMatchState_row(
1901
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1902
+ void const* src, size_t srcSize)
1903
+ {
1904
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_rowHash, 2, ZSTD_dictMatchState);
1905
+ }
1906
+
1907
+ size_t ZSTD_compressBlock_lazy2_dedicatedDictSearch_row(
1908
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1909
+ void const* src, size_t srcSize)
1910
+ {
1911
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_rowHash, 2, ZSTD_dedicatedDictSearch);
1912
+ }
1913
+ #endif
1914
+
1915
+ #ifndef ZSTD_EXCLUDE_BTLAZY2_BLOCK_COMPRESSOR
1916
+ size_t ZSTD_compressBlock_btlazy2(
1917
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1918
+ void const* src, size_t srcSize)
1919
+ {
1920
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_binaryTree, 2, ZSTD_noDict);
1921
+ }
1922
+
1923
+ size_t ZSTD_compressBlock_btlazy2_dictMatchState(
1924
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
1925
+ void const* src, size_t srcSize)
1926
+ {
1927
+ return ZSTD_compressBlock_lazy_generic(ms, seqStore, rep, src, srcSize, search_binaryTree, 2, ZSTD_dictMatchState);
1928
+ }
1929
+ #endif
1930
+
1931
+ #if !defined(ZSTD_EXCLUDE_GREEDY_BLOCK_COMPRESSOR) \
1932
+ || !defined(ZSTD_EXCLUDE_LAZY_BLOCK_COMPRESSOR) \
1933
+ || !defined(ZSTD_EXCLUDE_LAZY2_BLOCK_COMPRESSOR) \
1934
+ || !defined(ZSTD_EXCLUDE_BTLAZY2_BLOCK_COMPRESSOR)
1935
+ FORCE_INLINE_TEMPLATE
1936
+ ZSTD_ALLOW_POINTER_OVERFLOW_ATTR
1937
+ size_t ZSTD_compressBlock_lazy_extDict_generic(
1938
+ ZSTD_matchState_t* ms, seqStore_t* seqStore,
1939
+ U32 rep[ZSTD_REP_NUM],
1940
+ const void* src, size_t srcSize,
1941
+ const searchMethod_e searchMethod, const U32 depth)
1942
+ {
1943
+ const BYTE* const istart = (const BYTE*)src;
1944
+ const BYTE* ip = istart;
1945
+ const BYTE* anchor = istart;
1946
+ const BYTE* const iend = istart + srcSize;
1947
+ const BYTE* const ilimit = searchMethod == search_rowHash ? iend - 8 - ZSTD_ROW_HASH_CACHE_SIZE : iend - 8;
1948
+ const BYTE* const base = ms->window.base;
1949
+ const U32 dictLimit = ms->window.dictLimit;
1950
+ const BYTE* const prefixStart = base + dictLimit;
1951
+ const BYTE* const dictBase = ms->window.dictBase;
1952
+ const BYTE* const dictEnd = dictBase + dictLimit;
1953
+ const BYTE* const dictStart = dictBase + ms->window.lowLimit;
1954
+ const U32 windowLog = ms->cParams.windowLog;
1955
+ const U32 mls = BOUNDED(4, ms->cParams.minMatch, 6);
1956
+ const U32 rowLog = BOUNDED(4, ms->cParams.searchLog, 6);
1957
+
1958
+ U32 offset_1 = rep[0], offset_2 = rep[1];
1959
+
1960
+ DEBUGLOG(5, "ZSTD_compressBlock_lazy_extDict_generic (searchFunc=%u)", (U32)searchMethod);
1961
+
1962
+ /* Reset the lazy skipping state */
1963
+ ms->lazySkipping = 0;
1964
+
1965
+ /* init */
1966
+ ip += (ip == prefixStart);
1967
+ if (searchMethod == search_rowHash) {
1968
+ ZSTD_row_fillHashCache(ms, base, rowLog, mls, ms->nextToUpdate, ilimit);
1969
+ }
1970
+
1971
+ /* Match Loop */
1972
+ #if defined(__GNUC__) && defined(__x86_64__)
1973
+ /* I've measured random a 5% speed loss on levels 5 & 6 (greedy) when the
1974
+ * code alignment is perturbed. To fix the instability align the loop on 32-bytes.
1975
+ */
1976
+ __asm__(".p2align 5");
1977
+ #endif
1978
+ while (ip < ilimit) {
1979
+ size_t matchLength=0;
1980
+ size_t offBase = REPCODE1_TO_OFFBASE;
1981
+ const BYTE* start=ip+1;
1982
+ U32 curr = (U32)(ip-base);
1983
+
1984
+ /* check repCode */
1985
+ { const U32 windowLow = ZSTD_getLowestMatchIndex(ms, curr+1, windowLog);
1986
+ const U32 repIndex = (U32)(curr+1 - offset_1);
1987
+ const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
1988
+ const BYTE* const repMatch = repBase + repIndex;
1989
+ if ( ((U32)((dictLimit-1) - repIndex) >= 3) /* intentional overflow */
1990
+ & (offset_1 <= curr+1 - windowLow) ) /* note: we are searching at curr+1 */
1991
+ if (MEM_read32(ip+1) == MEM_read32(repMatch)) {
1992
+ /* repcode detected we should take it */
1993
+ const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
1994
+ matchLength = ZSTD_count_2segments(ip+1+4, repMatch+4, iend, repEnd, prefixStart) + 4;
1995
+ if (depth==0) goto _storeSequence;
1996
+ } }
1997
+
1998
+ /* first search (depth 0) */
1999
+ { size_t ofbCandidate = 999999999;
2000
+ size_t const ml2 = ZSTD_searchMax(ms, ip, iend, &ofbCandidate, mls, rowLog, searchMethod, ZSTD_extDict);
2001
+ if (ml2 > matchLength)
2002
+ matchLength = ml2, start = ip, offBase = ofbCandidate;
2003
+ }
2004
+
2005
+ if (matchLength < 4) {
2006
+ size_t const step = ((size_t)(ip-anchor) >> kSearchStrength);
2007
+ ip += step + 1; /* jump faster over incompressible sections */
2008
+ /* Enter the lazy skipping mode once we are skipping more than 8 bytes at a time.
2009
+ * In this mode we stop inserting every position into our tables, and only insert
2010
+ * positions that we search, which is one in step positions.
2011
+ * The exact cutoff is flexible, I've just chosen a number that is reasonably high,
2012
+ * so we minimize the compression ratio loss in "normal" scenarios. This mode gets
2013
+ * triggered once we've gone 2KB without finding any matches.
2014
+ */
2015
+ ms->lazySkipping = step > kLazySkippingStep;
2016
+ continue;
2017
+ }
2018
+
2019
+ /* let's try to find a better solution */
2020
+ if (depth>=1)
2021
+ while (ip<ilimit) {
2022
+ ip ++;
2023
+ curr++;
2024
+ /* check repCode */
2025
+ if (offBase) {
2026
+ const U32 windowLow = ZSTD_getLowestMatchIndex(ms, curr, windowLog);
2027
+ const U32 repIndex = (U32)(curr - offset_1);
2028
+ const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
2029
+ const BYTE* const repMatch = repBase + repIndex;
2030
+ if ( ((U32)((dictLimit-1) - repIndex) >= 3) /* intentional overflow : do not test positions overlapping 2 memory segments */
2031
+ & (offset_1 <= curr - windowLow) ) /* equivalent to `curr > repIndex >= windowLow` */
2032
+ if (MEM_read32(ip) == MEM_read32(repMatch)) {
2033
+ /* repcode detected */
2034
+ const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
2035
+ size_t const repLength = ZSTD_count_2segments(ip+4, repMatch+4, iend, repEnd, prefixStart) + 4;
2036
+ int const gain2 = (int)(repLength * 3);
2037
+ int const gain1 = (int)(matchLength*3 - ZSTD_highbit32((U32)offBase) + 1);
2038
+ if ((repLength >= 4) && (gain2 > gain1))
2039
+ matchLength = repLength, offBase = REPCODE1_TO_OFFBASE, start = ip;
2040
+ } }
2041
+
2042
+ /* search match, depth 1 */
2043
+ { size_t ofbCandidate = 999999999;
2044
+ size_t const ml2 = ZSTD_searchMax(ms, ip, iend, &ofbCandidate, mls, rowLog, searchMethod, ZSTD_extDict);
2045
+ int const gain2 = (int)(ml2*4 - ZSTD_highbit32((U32)ofbCandidate)); /* raw approx */
2046
+ int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offBase) + 4);
2047
+ if ((ml2 >= 4) && (gain2 > gain1)) {
2048
+ matchLength = ml2, offBase = ofbCandidate, start = ip;
2049
+ continue; /* search a better one */
2050
+ } }
2051
+
2052
+ /* let's find an even better one */
2053
+ if ((depth==2) && (ip<ilimit)) {
2054
+ ip ++;
2055
+ curr++;
2056
+ /* check repCode */
2057
+ if (offBase) {
2058
+ const U32 windowLow = ZSTD_getLowestMatchIndex(ms, curr, windowLog);
2059
+ const U32 repIndex = (U32)(curr - offset_1);
2060
+ const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
2061
+ const BYTE* const repMatch = repBase + repIndex;
2062
+ if ( ((U32)((dictLimit-1) - repIndex) >= 3) /* intentional overflow : do not test positions overlapping 2 memory segments */
2063
+ & (offset_1 <= curr - windowLow) ) /* equivalent to `curr > repIndex >= windowLow` */
2064
+ if (MEM_read32(ip) == MEM_read32(repMatch)) {
2065
+ /* repcode detected */
2066
+ const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
2067
+ size_t const repLength = ZSTD_count_2segments(ip+4, repMatch+4, iend, repEnd, prefixStart) + 4;
2068
+ int const gain2 = (int)(repLength * 4);
2069
+ int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offBase) + 1);
2070
+ if ((repLength >= 4) && (gain2 > gain1))
2071
+ matchLength = repLength, offBase = REPCODE1_TO_OFFBASE, start = ip;
2072
+ } }
2073
+
2074
+ /* search match, depth 2 */
2075
+ { size_t ofbCandidate = 999999999;
2076
+ size_t const ml2 = ZSTD_searchMax(ms, ip, iend, &ofbCandidate, mls, rowLog, searchMethod, ZSTD_extDict);
2077
+ int const gain2 = (int)(ml2*4 - ZSTD_highbit32((U32)ofbCandidate)); /* raw approx */
2078
+ int const gain1 = (int)(matchLength*4 - ZSTD_highbit32((U32)offBase) + 7);
2079
+ if ((ml2 >= 4) && (gain2 > gain1)) {
2080
+ matchLength = ml2, offBase = ofbCandidate, start = ip;
2081
+ continue;
2082
+ } } }
2083
+ break; /* nothing found : store previous solution */
2084
+ }
2085
+
2086
+ /* catch up */
2087
+ if (OFFBASE_IS_OFFSET(offBase)) {
2088
+ U32 const matchIndex = (U32)((size_t)(start-base) - OFFBASE_TO_OFFSET(offBase));
2089
+ const BYTE* match = (matchIndex < dictLimit) ? dictBase + matchIndex : base + matchIndex;
2090
+ const BYTE* const mStart = (matchIndex < dictLimit) ? dictStart : prefixStart;
2091
+ while ((start>anchor) && (match>mStart) && (start[-1] == match[-1])) { start--; match--; matchLength++; } /* catch up */
2092
+ offset_2 = offset_1; offset_1 = (U32)OFFBASE_TO_OFFSET(offBase);
2093
+ }
2094
+
2095
+ /* store sequence */
2096
+ _storeSequence:
2097
+ { size_t const litLength = (size_t)(start - anchor);
2098
+ ZSTD_storeSeq(seqStore, litLength, anchor, iend, (U32)offBase, matchLength);
2099
+ anchor = ip = start + matchLength;
2100
+ }
2101
+ if (ms->lazySkipping) {
2102
+ /* We've found a match, disable lazy skipping mode, and refill the hash cache. */
2103
+ if (searchMethod == search_rowHash) {
2104
+ ZSTD_row_fillHashCache(ms, base, rowLog, mls, ms->nextToUpdate, ilimit);
2105
+ }
2106
+ ms->lazySkipping = 0;
2107
+ }
2108
+
2109
+ /* check immediate repcode */
2110
+ while (ip <= ilimit) {
2111
+ const U32 repCurrent = (U32)(ip-base);
2112
+ const U32 windowLow = ZSTD_getLowestMatchIndex(ms, repCurrent, windowLog);
2113
+ const U32 repIndex = repCurrent - offset_2;
2114
+ const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
2115
+ const BYTE* const repMatch = repBase + repIndex;
2116
+ if ( ((U32)((dictLimit-1) - repIndex) >= 3) /* intentional overflow : do not test positions overlapping 2 memory segments */
2117
+ & (offset_2 <= repCurrent - windowLow) ) /* equivalent to `curr > repIndex >= windowLow` */
2118
+ if (MEM_read32(ip) == MEM_read32(repMatch)) {
2119
+ /* repcode detected we should take it */
2120
+ const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
2121
+ matchLength = ZSTD_count_2segments(ip+4, repMatch+4, iend, repEnd, prefixStart) + 4;
2122
+ offBase = offset_2; offset_2 = offset_1; offset_1 = (U32)offBase; /* swap offset history */
2123
+ ZSTD_storeSeq(seqStore, 0, anchor, iend, REPCODE1_TO_OFFBASE, matchLength);
2124
+ ip += matchLength;
2125
+ anchor = ip;
2126
+ continue; /* faster when present ... (?) */
2127
+ }
2128
+ break;
2129
+ } }
2130
+
2131
+ /* Save reps for next block */
2132
+ rep[0] = offset_1;
2133
+ rep[1] = offset_2;
2134
+
2135
+ /* Return the last literals size */
2136
+ return (size_t)(iend - anchor);
2137
+ }
2138
+ #endif /* build exclusions */
2139
+
2140
+ #ifndef ZSTD_EXCLUDE_GREEDY_BLOCK_COMPRESSOR
2141
+ size_t ZSTD_compressBlock_greedy_extDict(
2142
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
2143
+ void const* src, size_t srcSize)
2144
+ {
2145
+ return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 0);
2146
+ }
2147
+
2148
+ size_t ZSTD_compressBlock_greedy_extDict_row(
2149
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
2150
+ void const* src, size_t srcSize)
2151
+ {
2152
+ return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, search_rowHash, 0);
2153
+ }
2154
+ #endif
2155
+
2156
+ #ifndef ZSTD_EXCLUDE_LAZY_BLOCK_COMPRESSOR
2157
+ size_t ZSTD_compressBlock_lazy_extDict(
2158
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
2159
+ void const* src, size_t srcSize)
2160
+
2161
+ {
2162
+ return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 1);
2163
+ }
2164
+
2165
+ size_t ZSTD_compressBlock_lazy_extDict_row(
2166
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
2167
+ void const* src, size_t srcSize)
2168
+
2169
+ {
2170
+ return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, search_rowHash, 1);
2171
+ }
2172
+ #endif
2173
+
2174
+ #ifndef ZSTD_EXCLUDE_LAZY2_BLOCK_COMPRESSOR
2175
+ size_t ZSTD_compressBlock_lazy2_extDict(
2176
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
2177
+ void const* src, size_t srcSize)
2178
+
2179
+ {
2180
+ return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, search_hashChain, 2);
2181
+ }
2182
+
2183
+ size_t ZSTD_compressBlock_lazy2_extDict_row(
2184
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
2185
+ void const* src, size_t srcSize)
2186
+ {
2187
+ return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, search_rowHash, 2);
2188
+ }
2189
+ #endif
2190
+
2191
+ #ifndef ZSTD_EXCLUDE_BTLAZY2_BLOCK_COMPRESSOR
2192
+ size_t ZSTD_compressBlock_btlazy2_extDict(
2193
+ ZSTD_matchState_t* ms, seqStore_t* seqStore, U32 rep[ZSTD_REP_NUM],
2194
+ void const* src, size_t srcSize)
2195
+
2196
+ {
2197
+ return ZSTD_compressBlock_lazy_extDict_generic(ms, seqStore, rep, src, srcSize, search_binaryTree, 2);
2198
+ }
2199
+ #endif