/src/suricata7/libhtp/htp/lzma/LzmaDec.c
Line | Count | Source (jump to first uncovered line) |
1 | | /* LzmaDec.c -- LZMA Decoder |
2 | | 2018-07-04 : Igor Pavlov : Public domain */ |
3 | | |
4 | | #include "Precomp.h" |
5 | | |
6 | | #include <string.h> |
7 | | #include <stdlib.h> |
8 | | #include <stdint.h> |
9 | | |
10 | | /* #include "CpuArch.h" */ |
11 | | #include "LzmaDec.h" |
12 | | |
13 | 0 | #define kNumTopBits 24 |
14 | 0 | #define kTopValue ((UInt32)1 << kNumTopBits) |
15 | | |
16 | 0 | #define kNumBitModelTotalBits 11 |
17 | 0 | #define kBitModelTotal (1 << kNumBitModelTotalBits) |
18 | 0 | #define kNumMoveBits 5 |
19 | | |
20 | 0 | #define RC_INIT_SIZE 5 |
21 | | |
22 | 0 | #define NORMALIZE if (range < kTopValue) { range <<= 8; code = (code << 8) | (*buf++); } |
23 | | |
24 | 0 | #define IF_BIT_0(p) ttt = *(p); NORMALIZE; bound = (range >> kNumBitModelTotalBits) * (UInt32)ttt; if (code < bound) |
25 | 0 | #define UPDATE_0(p) range = bound; *(p) = (CLzmaProb)(ttt + ((kBitModelTotal - ttt) >> kNumMoveBits)); |
26 | 0 | #define UPDATE_1(p) range -= bound; code -= bound; *(p) = (CLzmaProb)(ttt - (ttt >> kNumMoveBits)); |
27 | 0 | #define GET_BIT2(p, i, A0, A1) IF_BIT_0(p) \ |
28 | 0 | { UPDATE_0(p); i = (i + i); A0; } else \ |
29 | 0 | { UPDATE_1(p); i = (i + i) + 1; A1; } |
30 | | |
31 | 0 | #define TREE_GET_BIT(probs, i) { GET_BIT2(probs + i, i, ;, ;); } |
32 | | |
33 | 0 | #define REV_BIT(p, i, A0, A1) IF_BIT_0(p + i) \ |
34 | 0 | { UPDATE_0(p + i); A0; } else \ |
35 | 0 | { UPDATE_1(p + i); A1; } |
36 | 0 | #define REV_BIT_VAR( p, i, m) REV_BIT(p, i, i += m; m += m, m += m; i += m; ) |
37 | 0 | #define REV_BIT_CONST(p, i, m) REV_BIT(p, i, i += m; , i += m * 2; ) |
38 | 0 | #define REV_BIT_LAST( p, i, m) REV_BIT(p, i, i -= m , ; ) |
39 | | |
40 | | #define TREE_DECODE(probs, limit, i) \ |
41 | 0 | { i = 1; do { TREE_GET_BIT(probs, i); } while (i < limit); i -= limit; } |
42 | | |
43 | | /* #define _LZMA_SIZE_OPT */ |
44 | | |
45 | | #ifdef _LZMA_SIZE_OPT |
46 | | #define TREE_6_DECODE(probs, i) TREE_DECODE(probs, (1 << 6), i) |
47 | | #else |
48 | | #define TREE_6_DECODE(probs, i) \ |
49 | 0 | { i = 1; \ |
50 | 0 | TREE_GET_BIT(probs, i); \ |
51 | 0 | TREE_GET_BIT(probs, i); \ |
52 | 0 | TREE_GET_BIT(probs, i); \ |
53 | 0 | TREE_GET_BIT(probs, i); \ |
54 | 0 | TREE_GET_BIT(probs, i); \ |
55 | 0 | TREE_GET_BIT(probs, i); \ |
56 | 0 | i -= 0x40; } |
57 | | #endif |
58 | | |
59 | 0 | #define NORMAL_LITER_DEC TREE_GET_BIT(prob, symbol) |
60 | | #define MATCHED_LITER_DEC \ |
61 | 0 | matchByte += matchByte; \ |
62 | 0 | bit = offs; \ |
63 | 0 | offs &= matchByte; \ |
64 | 0 | probLit = prob + (offs + bit + symbol); \ |
65 | 0 | GET_BIT2(probLit, symbol, offs ^= bit; , ;) |
66 | | |
67 | | |
68 | | |
69 | 0 | #define NORMALIZE_CHECK if (range < kTopValue) { if (buf >= bufLimit) return DUMMY_ERROR; range <<= 8; code = (code << 8) | (*buf++); } |
70 | | |
71 | 0 | #define IF_BIT_0_CHECK(p) ttt = *(p); NORMALIZE_CHECK; bound = (range >> kNumBitModelTotalBits) * (UInt32)ttt; if (code < bound) |
72 | 0 | #define UPDATE_0_CHECK range = bound; |
73 | 0 | #define UPDATE_1_CHECK range -= bound; code -= bound; |
74 | 0 | #define GET_BIT2_CHECK(p, i, A0, A1) IF_BIT_0_CHECK(p) \ |
75 | 0 | { UPDATE_0_CHECK; i = (i + i); A0; } else \ |
76 | 0 | { UPDATE_1_CHECK; i = (i + i) + 1; A1; } |
77 | 0 | #define GET_BIT_CHECK(p, i) GET_BIT2_CHECK(p, i, ; , ;) |
78 | | #define TREE_DECODE_CHECK(probs, limit, i) \ |
79 | 0 | { i = 1; do { GET_BIT_CHECK(probs + i, i) } while (i < limit); i -= limit; } |
80 | | |
81 | | |
82 | 0 | #define REV_BIT_CHECK(p, i, m) IF_BIT_0_CHECK(p + i) \ |
83 | 0 | { UPDATE_0_CHECK; i += m; m += m; } else \ |
84 | 0 | { UPDATE_1_CHECK; m += m; i += m; } |
85 | | |
86 | | |
87 | 0 | #define kNumPosBitsMax 4 |
88 | 0 | #define kNumPosStatesMax (1 << kNumPosBitsMax) |
89 | | |
90 | 0 | #define kLenNumLowBits 3 |
91 | 0 | #define kLenNumLowSymbols (1 << kLenNumLowBits) |
92 | 0 | #define kLenNumHighBits 8 |
93 | 0 | #define kLenNumHighSymbols (1 << kLenNumHighBits) |
94 | | |
95 | 0 | #define LenLow 0 |
96 | 0 | #define LenHigh (LenLow + 2 * (kNumPosStatesMax << kLenNumLowBits)) |
97 | 0 | #define kNumLenProbs (LenHigh + kLenNumHighSymbols) |
98 | | |
99 | 0 | #define LenChoice LenLow |
100 | 0 | #define LenChoice2 (LenLow + (1 << kLenNumLowBits)) |
101 | | |
102 | 0 | #define kNumStates 12 |
103 | 0 | #define kNumStates2 16 |
104 | 0 | #define kNumLitStates 7 |
105 | | |
106 | 0 | #define kStartPosModelIndex 4 |
107 | 0 | #define kEndPosModelIndex 14 |
108 | 0 | #define kNumFullDistances (1 << (kEndPosModelIndex >> 1)) |
109 | | |
110 | 0 | #define kNumPosSlotBits 6 |
111 | 0 | #define kNumLenToPosStates 4 |
112 | | |
113 | 0 | #define kNumAlignBits 4 |
114 | 0 | #define kAlignTableSize (1 << kNumAlignBits) |
115 | | |
116 | 0 | #define kMatchMinLen 2 |
117 | 0 | #define kMatchSpecLenStart (kMatchMinLen + kLenNumLowSymbols * 2 + kLenNumHighSymbols) |
118 | | |
119 | | /* External ASM code needs same CLzmaProb array layout. So don't change it. */ |
120 | | |
121 | | /* (probs_1664) is faster and better for code size at some platforms */ |
122 | | /* |
123 | | #ifdef MY_CPU_X86_OR_AMD64 |
124 | | */ |
125 | 0 | #define kStartOffset 1664 |
126 | 0 | #define GET_PROBS p->probs_1664 |
127 | | /* |
128 | | #define GET_PROBS p->probs + kStartOffset |
129 | | #else |
130 | | #define kStartOffset 0 |
131 | | #define GET_PROBS p->probs |
132 | | #endif |
133 | | */ |
134 | | |
135 | 0 | #define SpecPos (-kStartOffset) |
136 | 0 | #define IsRep0Long (SpecPos + kNumFullDistances) |
137 | 0 | #define RepLenCoder (IsRep0Long + (kNumStates2 << kNumPosBitsMax)) |
138 | 0 | #define LenCoder (RepLenCoder + kNumLenProbs) |
139 | 0 | #define IsMatch (LenCoder + kNumLenProbs) |
140 | 0 | #define Align (IsMatch + (kNumStates2 << kNumPosBitsMax)) |
141 | 0 | #define IsRep (Align + kAlignTableSize) |
142 | 0 | #define IsRepG0 (IsRep + kNumStates) |
143 | 0 | #define IsRepG1 (IsRepG0 + kNumStates) |
144 | 0 | #define IsRepG2 (IsRepG1 + kNumStates) |
145 | 0 | #define PosSlot (IsRepG2 + kNumStates) |
146 | 0 | #define Literal (PosSlot + (kNumLenToPosStates << kNumPosSlotBits)) |
147 | 0 | #define NUM_BASE_PROBS (Literal + kStartOffset) |
148 | | |
149 | | #if Align != 0 && kStartOffset != 0 |
150 | | #error Stop_Compiling_Bad_LZMA_kAlign |
151 | | #endif |
152 | | |
153 | | #if NUM_BASE_PROBS != 1984 |
154 | | #error Stop_Compiling_Bad_LZMA_PROBS |
155 | | #endif |
156 | | |
157 | | |
158 | 0 | #define LZMA_LIT_SIZE 0x300 |
159 | | |
160 | 0 | #define LzmaProps_GetNumProbs(p) (NUM_BASE_PROBS + ((UInt32)LZMA_LIT_SIZE << ((p)->lc + (p)->lp))) |
161 | | |
162 | | |
163 | 0 | #define CALC_POS_STATE(processedPos, pbMask) (((processedPos) & (pbMask)) << 4) |
164 | 0 | #define COMBINED_PS_STATE (posState + state) |
165 | 0 | #define GET_LEN_STATE (posState) |
166 | | |
167 | 0 | #define LZMA_DIC_MIN (1 << 12) |
168 | | |
169 | | /* |
170 | | p->remainLen : shows status of LZMA decoder: |
171 | | < kMatchSpecLenStart : normal remain |
172 | | = kMatchSpecLenStart : finished |
173 | | = kMatchSpecLenStart + 1 : need init range coder |
174 | | = kMatchSpecLenStart + 2 : need init range coder and state |
175 | | */ |
176 | | |
177 | | /* ---------- LZMA_DECODE_REAL ---------- */ |
178 | | /* |
179 | | LzmaDec_DecodeReal_3() can be implemented in external ASM file. |
180 | | 3 - is the code compatibility version of that function for check at link time. |
181 | | */ |
182 | | |
183 | | #define LZMA_DECODE_REAL LzmaDec_DecodeReal_3 |
184 | | |
185 | | /* |
186 | | LZMA_DECODE_REAL() |
187 | | In: |
188 | | RangeCoder is normalized |
189 | | if (p->dicPos == limit) |
190 | | { |
191 | | LzmaDec_TryDummy() was called before to exclude LITERAL and MATCH-REP cases. |
192 | | So first symbol can be only MATCH-NON-REP. And if that MATCH-NON-REP symbol |
193 | | is not END_OF_PAYALOAD_MARKER, then function returns error code. |
194 | | } |
195 | | |
196 | | Processing: |
197 | | first LZMA symbol will be decoded in any case |
198 | | All checks for limits are at the end of main loop, |
199 | | It will decode new LZMA-symbols while (p->buf < bufLimit && dicPos < limit), |
200 | | RangeCoder is still without last normalization when (p->buf < bufLimit) is being checked. |
201 | | |
202 | | Out: |
203 | | RangeCoder is normalized |
204 | | Result: |
205 | | SZ_OK - OK |
206 | | SZ_ERROR_DATA - Error |
207 | | p->remainLen: |
208 | | < kMatchSpecLenStart : normal remain |
209 | | = kMatchSpecLenStart : finished |
210 | | */ |
211 | | |
212 | | |
213 | | #ifdef _LZMA_DEC_OPT |
214 | | |
215 | | int MY_FAST_CALL LZMA_DECODE_REAL(CLzmaDec *p, SizeT limit, const Byte *bufLimit); |
216 | | |
217 | | #else |
218 | | |
219 | | static |
220 | | int MY_FAST_CALL LZMA_DECODE_REAL(CLzmaDec *p, SizeT limit, const Byte *bufLimit) |
221 | 0 | { |
222 | 0 | CLzmaProb *probs = GET_PROBS; |
223 | 0 | unsigned state = (unsigned)p->state; |
224 | 0 | UInt32 rep0 = p->reps[0], rep1 = p->reps[1], rep2 = p->reps[2], rep3 = p->reps[3]; |
225 | 0 | unsigned pbMask = ((unsigned)1 << (p->prop.pb)) - 1; |
226 | 0 | unsigned lc = p->prop.lc; |
227 | 0 | unsigned lpMask = ((unsigned)0x100 << p->prop.lp) - ((unsigned)0x100 >> lc); |
228 | |
|
229 | 0 | Byte *dic = p->dic; |
230 | 0 | SizeT dicBufSize = p->dicBufSize; |
231 | 0 | SizeT dicPos = p->dicPos; |
232 | | |
233 | 0 | UInt32 processedPos = p->processedPos; |
234 | 0 | UInt32 checkDicSize = p->checkDicSize; |
235 | 0 | unsigned len = 0; |
236 | |
|
237 | 0 | const Byte *buf = p->buf; |
238 | 0 | UInt32 range = p->range; |
239 | 0 | UInt32 code = p->code; |
240 | |
|
241 | 0 | do |
242 | 0 | { |
243 | 0 | CLzmaProb *prob; |
244 | 0 | UInt32 bound; |
245 | 0 | unsigned ttt; |
246 | 0 | unsigned posState = CALC_POS_STATE(processedPos, pbMask); |
247 | |
|
248 | 0 | prob = probs + IsMatch + COMBINED_PS_STATE; |
249 | 0 | IF_BIT_0(prob) |
250 | 0 | { |
251 | 0 | unsigned symbol; |
252 | 0 | UPDATE_0(prob); |
253 | 0 | prob = probs + Literal; |
254 | 0 | if (processedPos != 0 || checkDicSize != 0) |
255 | 0 | prob += (UInt32)3 * ((((processedPos << 8) + dic[(dicPos == 0 ? dicBufSize : dicPos) - 1]) & lpMask) << lc); |
256 | 0 | processedPos++; |
257 | |
|
258 | 0 | if (state < kNumLitStates) |
259 | 0 | { |
260 | 0 | state -= (state < 4) ? state : 3; |
261 | 0 | symbol = 1; |
262 | | #ifdef _LZMA_SIZE_OPT |
263 | | do { NORMAL_LITER_DEC } while (symbol < 0x100); |
264 | | #else |
265 | 0 | NORMAL_LITER_DEC |
266 | 0 | NORMAL_LITER_DEC |
267 | 0 | NORMAL_LITER_DEC |
268 | 0 | NORMAL_LITER_DEC |
269 | 0 | NORMAL_LITER_DEC |
270 | 0 | NORMAL_LITER_DEC |
271 | 0 | NORMAL_LITER_DEC |
272 | 0 | NORMAL_LITER_DEC |
273 | 0 | #endif |
274 | 0 | } |
275 | 0 | else |
276 | 0 | { |
277 | 0 | unsigned matchByte = dic[dicPos - rep0 + (dicPos < rep0 ? dicBufSize : 0)]; |
278 | 0 | unsigned offs = 0x100; |
279 | 0 | state -= (state < 10) ? 3 : 6; |
280 | 0 | symbol = 1; |
281 | | #ifdef _LZMA_SIZE_OPT |
282 | | do |
283 | | { |
284 | | unsigned bit; |
285 | | CLzmaProb *probLit; |
286 | | MATCHED_LITER_DEC |
287 | | } |
288 | | while (symbol < 0x100); |
289 | | #else |
290 | 0 | { |
291 | 0 | unsigned bit; |
292 | 0 | CLzmaProb *probLit; |
293 | 0 | MATCHED_LITER_DEC |
294 | 0 | MATCHED_LITER_DEC |
295 | 0 | MATCHED_LITER_DEC |
296 | 0 | MATCHED_LITER_DEC |
297 | 0 | MATCHED_LITER_DEC |
298 | 0 | MATCHED_LITER_DEC |
299 | 0 | MATCHED_LITER_DEC |
300 | 0 | MATCHED_LITER_DEC |
301 | 0 | } |
302 | 0 | #endif |
303 | 0 | } |
304 | |
|
305 | 0 | dic[dicPos++] = (Byte)symbol; |
306 | 0 | continue; |
307 | 0 | } |
308 | | |
309 | 0 | { |
310 | 0 | UPDATE_1(prob); |
311 | 0 | prob = probs + IsRep + state; |
312 | 0 | IF_BIT_0(prob) |
313 | 0 | { |
314 | 0 | UPDATE_0(prob); |
315 | 0 | state += kNumStates; |
316 | 0 | prob = probs + LenCoder; |
317 | 0 | } |
318 | 0 | else |
319 | 0 | { |
320 | 0 | UPDATE_1(prob); |
321 | | /* |
322 | | // that case was checked before with kBadRepCode |
323 | | if (checkDicSize == 0 && processedPos == 0) |
324 | | return SZ_ERROR_DATA; |
325 | | */ |
326 | 0 | prob = probs + IsRepG0 + state; |
327 | 0 | IF_BIT_0(prob) |
328 | 0 | { |
329 | 0 | UPDATE_0(prob); |
330 | 0 | prob = probs + IsRep0Long + COMBINED_PS_STATE; |
331 | 0 | IF_BIT_0(prob) |
332 | 0 | { |
333 | 0 | UPDATE_0(prob); |
334 | 0 | dic[dicPos] = dic[dicPos - rep0 + (dicPos < rep0 ? dicBufSize : 0)]; |
335 | 0 | dicPos++; |
336 | 0 | processedPos++; |
337 | 0 | state = state < kNumLitStates ? 9 : 11; |
338 | 0 | continue; |
339 | 0 | } |
340 | 0 | UPDATE_1(prob); |
341 | 0 | } |
342 | 0 | else |
343 | 0 | { |
344 | 0 | UInt32 distance; |
345 | 0 | UPDATE_1(prob); |
346 | 0 | prob = probs + IsRepG1 + state; |
347 | 0 | IF_BIT_0(prob) |
348 | 0 | { |
349 | 0 | UPDATE_0(prob); |
350 | 0 | distance = rep1; |
351 | 0 | } |
352 | 0 | else |
353 | 0 | { |
354 | 0 | UPDATE_1(prob); |
355 | 0 | prob = probs + IsRepG2 + state; |
356 | 0 | IF_BIT_0(prob) |
357 | 0 | { |
358 | 0 | UPDATE_0(prob); |
359 | 0 | distance = rep2; |
360 | 0 | } |
361 | 0 | else |
362 | 0 | { |
363 | 0 | UPDATE_1(prob); |
364 | 0 | distance = rep3; |
365 | 0 | rep3 = rep2; |
366 | 0 | } |
367 | 0 | rep2 = rep1; |
368 | 0 | } |
369 | 0 | rep1 = rep0; |
370 | 0 | rep0 = distance; |
371 | 0 | } |
372 | 0 | state = state < kNumLitStates ? 8 : 11; |
373 | 0 | prob = probs + RepLenCoder; |
374 | 0 | } |
375 | | |
376 | | #ifdef _LZMA_SIZE_OPT |
377 | | { |
378 | | unsigned lim, offset; |
379 | | CLzmaProb *probLen = prob + LenChoice; |
380 | | IF_BIT_0(probLen) |
381 | | { |
382 | | UPDATE_0(probLen); |
383 | | probLen = prob + LenLow + GET_LEN_STATE; |
384 | | offset = 0; |
385 | | lim = (1 << kLenNumLowBits); |
386 | | } |
387 | | else |
388 | | { |
389 | | UPDATE_1(probLen); |
390 | | probLen = prob + LenChoice2; |
391 | | IF_BIT_0(probLen) |
392 | | { |
393 | | UPDATE_0(probLen); |
394 | | probLen = prob + LenLow + GET_LEN_STATE + (1 << kLenNumLowBits); |
395 | | offset = kLenNumLowSymbols; |
396 | | lim = (1 << kLenNumLowBits); |
397 | | } |
398 | | else |
399 | | { |
400 | | UPDATE_1(probLen); |
401 | | probLen = prob + LenHigh; |
402 | | offset = kLenNumLowSymbols * 2; |
403 | | lim = (1 << kLenNumHighBits); |
404 | | } |
405 | | } |
406 | | TREE_DECODE(probLen, lim, len); |
407 | | len += offset; |
408 | | } |
409 | | #else |
410 | 0 | { |
411 | 0 | CLzmaProb *probLen = prob + LenChoice; |
412 | 0 | IF_BIT_0(probLen) |
413 | 0 | { |
414 | 0 | UPDATE_0(probLen); |
415 | 0 | probLen = prob + LenLow + GET_LEN_STATE; |
416 | 0 | len = 1; |
417 | 0 | TREE_GET_BIT(probLen, len); |
418 | 0 | TREE_GET_BIT(probLen, len); |
419 | 0 | TREE_GET_BIT(probLen, len); |
420 | 0 | len -= 8; |
421 | 0 | } |
422 | 0 | else |
423 | 0 | { |
424 | 0 | UPDATE_1(probLen); |
425 | 0 | probLen = prob + LenChoice2; |
426 | 0 | IF_BIT_0(probLen) |
427 | 0 | { |
428 | 0 | UPDATE_0(probLen); |
429 | 0 | probLen = prob + LenLow + GET_LEN_STATE + (1 << kLenNumLowBits); |
430 | 0 | len = 1; |
431 | 0 | TREE_GET_BIT(probLen, len); |
432 | 0 | TREE_GET_BIT(probLen, len); |
433 | 0 | TREE_GET_BIT(probLen, len); |
434 | 0 | } |
435 | 0 | else |
436 | 0 | { |
437 | 0 | UPDATE_1(probLen); |
438 | 0 | probLen = prob + LenHigh; |
439 | 0 | TREE_DECODE(probLen, (1 << kLenNumHighBits), len); |
440 | 0 | len += kLenNumLowSymbols * 2; |
441 | 0 | } |
442 | 0 | } |
443 | 0 | } |
444 | 0 | #endif |
445 | |
|
446 | 0 | if (state >= kNumStates) |
447 | 0 | { |
448 | 0 | UInt32 distance; |
449 | 0 | prob = probs + PosSlot + |
450 | 0 | ((len < kNumLenToPosStates ? len : kNumLenToPosStates - 1) << kNumPosSlotBits); |
451 | 0 | TREE_6_DECODE(prob, distance); |
452 | 0 | if (distance >= kStartPosModelIndex) |
453 | 0 | { |
454 | 0 | unsigned posSlot = (unsigned)distance; |
455 | 0 | unsigned numDirectBits = (unsigned)(((distance >> 1) - 1)); |
456 | 0 | distance = (2 | (distance & 1)); |
457 | 0 | if (posSlot < kEndPosModelIndex) |
458 | 0 | { |
459 | 0 | distance <<= numDirectBits; |
460 | 0 | prob = probs + SpecPos; |
461 | 0 | { |
462 | 0 | UInt32 m = 1; |
463 | 0 | distance++; |
464 | 0 | do |
465 | 0 | { |
466 | 0 | REV_BIT_VAR(prob, distance, m); |
467 | 0 | } |
468 | 0 | while (--numDirectBits); |
469 | 0 | distance -= m; |
470 | 0 | } |
471 | 0 | } |
472 | 0 | else |
473 | 0 | { |
474 | 0 | numDirectBits -= kNumAlignBits; |
475 | 0 | do |
476 | 0 | { |
477 | 0 | NORMALIZE |
478 | 0 | range >>= 1; |
479 | | |
480 | 0 | { |
481 | 0 | UInt32 t; |
482 | 0 | code -= range; |
483 | 0 | t = (0 - ((UInt32)code >> 31)); /* (UInt32)((Int32)code >> 31) */ |
484 | 0 | distance = (distance << 1) + (t + 1); |
485 | 0 | code += range & t; |
486 | 0 | } |
487 | | /* |
488 | | distance <<= 1; |
489 | | if (code >= range) |
490 | | { |
491 | | code -= range; |
492 | | distance |= 1; |
493 | | } |
494 | | */ |
495 | 0 | } |
496 | 0 | while (--numDirectBits); |
497 | 0 | prob = probs + Align; |
498 | 0 | distance <<= kNumAlignBits; |
499 | 0 | { |
500 | 0 | unsigned i = 1; |
501 | 0 | REV_BIT_CONST(prob, i, 1); |
502 | 0 | REV_BIT_CONST(prob, i, 2); |
503 | 0 | REV_BIT_CONST(prob, i, 4); |
504 | 0 | REV_BIT_LAST (prob, i, 8); |
505 | 0 | distance |= i; |
506 | 0 | } |
507 | 0 | if (distance == (UInt32)0xFFFFFFFF) |
508 | 0 | { |
509 | 0 | len = kMatchSpecLenStart; |
510 | 0 | state -= kNumStates; |
511 | 0 | break; |
512 | 0 | } |
513 | 0 | } |
514 | 0 | } |
515 | | |
516 | 0 | rep3 = rep2; |
517 | 0 | rep2 = rep1; |
518 | 0 | rep1 = rep0; |
519 | 0 | rep0 = distance + 1; |
520 | 0 | state = (state < kNumStates + kNumLitStates) ? kNumLitStates : kNumLitStates + 3; |
521 | 0 | if (distance >= (checkDicSize == 0 ? processedPos: checkDicSize)) |
522 | 0 | { |
523 | 0 | p->dicPos = dicPos; |
524 | 0 | return SZ_ERROR_DATA; |
525 | 0 | } |
526 | 0 | } |
527 | | |
528 | 0 | len += kMatchMinLen; |
529 | |
|
530 | 0 | { |
531 | 0 | SizeT rem; |
532 | 0 | unsigned curLen; |
533 | 0 | SizeT pos; |
534 | | |
535 | 0 | if ((rem = limit - dicPos) == 0) |
536 | 0 | { |
537 | 0 | p->dicPos = dicPos; |
538 | 0 | return SZ_ERROR_DATA; |
539 | 0 | } |
540 | | |
541 | 0 | curLen = ((rem < len) ? (unsigned)rem : len); |
542 | 0 | pos = dicPos - rep0 + (dicPos < rep0 ? dicBufSize : 0); |
543 | |
|
544 | 0 | processedPos += (UInt32)curLen; |
545 | |
|
546 | 0 | len -= curLen; |
547 | 0 | if (curLen <= dicBufSize - pos) |
548 | 0 | { |
549 | 0 | Byte *dest = dic + dicPos; |
550 | 0 | ptrdiff_t src = (ptrdiff_t)pos - (ptrdiff_t)dicPos; |
551 | 0 | const Byte *lim = dest + curLen; |
552 | 0 | dicPos += (SizeT)curLen; |
553 | 0 | do |
554 | 0 | *(dest) = (Byte)*(dest + src); |
555 | 0 | while (++dest != lim); |
556 | 0 | } |
557 | 0 | else |
558 | 0 | { |
559 | 0 | do |
560 | 0 | { |
561 | 0 | dic[dicPos++] = dic[pos]; |
562 | 0 | if (++pos == dicBufSize) |
563 | 0 | pos = 0; |
564 | 0 | } |
565 | 0 | while (--curLen != 0); |
566 | 0 | } |
567 | 0 | } |
568 | 0 | } |
569 | 0 | } |
570 | 0 | while (dicPos < limit && buf < bufLimit); |
571 | | |
572 | 0 | NORMALIZE; |
573 | | |
574 | 0 | p->buf = buf; |
575 | 0 | p->range = range; |
576 | 0 | p->code = code; |
577 | 0 | p->remainLen = (UInt32)len; |
578 | 0 | p->dicPos = dicPos; |
579 | 0 | p->processedPos = processedPos; |
580 | 0 | p->reps[0] = rep0; |
581 | 0 | p->reps[1] = rep1; |
582 | 0 | p->reps[2] = rep2; |
583 | 0 | p->reps[3] = rep3; |
584 | 0 | p->state = (UInt32)state; |
585 | |
|
586 | 0 | return SZ_OK; |
587 | 0 | } |
588 | | #endif |
589 | | |
590 | | static void MY_FAST_CALL LzmaDec_WriteRem(CLzmaDec *p, SizeT limit) |
591 | 0 | { |
592 | 0 | if (p->remainLen != 0 && p->remainLen < kMatchSpecLenStart) |
593 | 0 | { |
594 | 0 | Byte *dic = p->dic; |
595 | 0 | SizeT dicPos = p->dicPos; |
596 | 0 | SizeT dicBufSize = p->dicBufSize; |
597 | 0 | unsigned len = (unsigned)p->remainLen; |
598 | 0 | SizeT rep0 = p->reps[0]; /* we use SizeT to avoid the BUG of VC14 for AMD64 */ |
599 | 0 | SizeT rem = limit - dicPos; |
600 | 0 | if (rem < len) |
601 | 0 | len = (unsigned)(rem); |
602 | |
|
603 | 0 | if (p->checkDicSize == 0 && p->prop.dicSize - p->processedPos <= len) |
604 | 0 | p->checkDicSize = p->prop.dicSize; |
605 | |
|
606 | 0 | p->processedPos += (UInt32)len; |
607 | 0 | p->remainLen -= (UInt32)len; |
608 | 0 | while (len != 0) |
609 | 0 | { |
610 | 0 | len--; |
611 | 0 | dic[dicPos] = dic[dicPos - rep0 + (dicPos < rep0 ? dicBufSize : 0)]; |
612 | 0 | dicPos++; |
613 | 0 | } |
614 | 0 | p->dicPos = dicPos; |
615 | 0 | } |
616 | 0 | } |
617 | | |
618 | | |
619 | 0 | #define kRange0 0xFFFFFFFF |
620 | 0 | #define kBound0 ((kRange0 >> kNumBitModelTotalBits) << (kNumBitModelTotalBits - 1)) |
621 | 0 | #define kBadRepCode (kBound0 + (((kRange0 - kBound0) >> kNumBitModelTotalBits) << (kNumBitModelTotalBits - 1))) |
622 | | #if kBadRepCode != (0xC0000000 - 0x400) |
623 | | #error Stop_Compiling_Bad_LZMA_Check |
624 | | #endif |
625 | | |
626 | | static int MY_FAST_CALL LzmaDec_DecodeReal2(CLzmaDec *p, SizeT limit, const Byte *bufLimit, SizeT memlimit) |
627 | 0 | { |
628 | 0 | do |
629 | 0 | { |
630 | 0 | SizeT limit2 = limit; |
631 | 0 | if (p->checkDicSize == 0) |
632 | 0 | { |
633 | 0 | UInt32 rem = p->prop.dicSize - p->processedPos; |
634 | 0 | if (limit - p->dicPos > rem) { |
635 | 0 | if (p->dicBufSize < p->prop.dicSize) { |
636 | 0 | p->dicBufSize = p->prop.dicSize; |
637 | 0 | if (p->dicBufSize > memlimit) { |
638 | 0 | return SZ_ERROR_MEM; |
639 | 0 | } |
640 | 0 | Byte *tmp = realloc(p->dic, p->dicBufSize); |
641 | 0 | if (!tmp) { |
642 | 0 | return SZ_ERROR_MEM; |
643 | 0 | } |
644 | 0 | p->dic = tmp; |
645 | 0 | } |
646 | 0 | limit2 = p->dicPos + rem; |
647 | 0 | } |
648 | | |
649 | 0 | if (p->processedPos == 0) |
650 | 0 | if (p->code >= kBadRepCode) |
651 | 0 | return SZ_ERROR_DATA; |
652 | 0 | } |
653 | | |
654 | 0 | RINOK(LZMA_DECODE_REAL(p, limit2, bufLimit)); |
655 | | |
656 | 0 | if (p->checkDicSize == 0 && p->processedPos >= p->prop.dicSize) |
657 | 0 | p->checkDicSize = p->prop.dicSize; |
658 | | |
659 | 0 | LzmaDec_WriteRem(p, limit); |
660 | 0 | } |
661 | 0 | while (p->dicPos < limit && p->buf < bufLimit && p->remainLen < kMatchSpecLenStart); |
662 | | |
663 | 0 | return 0; |
664 | 0 | } |
665 | | |
666 | | typedef enum |
667 | | { |
668 | | DUMMY_ERROR, /* unexpected end of input stream */ |
669 | | DUMMY_LIT, |
670 | | DUMMY_MATCH, |
671 | | DUMMY_REP |
672 | | } ELzmaDummy; |
673 | | |
674 | | static ELzmaDummy LzmaDec_TryDummy(const CLzmaDec *p, const Byte *buf, SizeT inSize) |
675 | 0 | { |
676 | 0 | UInt32 range = p->range; |
677 | 0 | UInt32 code = p->code; |
678 | 0 | const Byte *bufLimit = buf + inSize; |
679 | 0 | const CLzmaProb *probs = GET_PROBS; |
680 | 0 | unsigned state = (unsigned)p->state; |
681 | 0 | ELzmaDummy res; |
682 | |
|
683 | 0 | { |
684 | 0 | const CLzmaProb *prob; |
685 | 0 | UInt32 bound; |
686 | 0 | unsigned ttt; |
687 | 0 | unsigned posState = CALC_POS_STATE(p->processedPos, (1 << p->prop.pb) - 1); |
688 | |
|
689 | 0 | prob = probs + IsMatch + COMBINED_PS_STATE; |
690 | 0 | IF_BIT_0_CHECK(prob) |
691 | 0 | { |
692 | 0 | UPDATE_0_CHECK |
693 | | |
694 | | /* if (bufLimit - buf >= 7) return DUMMY_LIT; */ |
695 | |
|
696 | 0 | prob = probs + Literal; |
697 | 0 | if (p->checkDicSize != 0 || p->processedPos != 0) |
698 | 0 | prob += ((UInt32)LZMA_LIT_SIZE * |
699 | 0 | ((((p->processedPos) & ((1 << (p->prop.lp)) - 1)) << p->prop.lc) + |
700 | 0 | (p->dic[(p->dicPos == 0 ? p->dicBufSize : p->dicPos) - 1] >> (8 - p->prop.lc)))); |
701 | |
|
702 | 0 | if (state < kNumLitStates) |
703 | 0 | { |
704 | 0 | unsigned symbol = 1; |
705 | 0 | do { GET_BIT_CHECK(prob + symbol, symbol) } while (symbol < 0x100); |
706 | 0 | } |
707 | 0 | else |
708 | 0 | { |
709 | 0 | unsigned matchByte = p->dic[p->dicPos - p->reps[0] + |
710 | 0 | (p->dicPos < p->reps[0] ? p->dicBufSize : 0)]; |
711 | 0 | unsigned offs = 0x100; |
712 | 0 | unsigned symbol = 1; |
713 | 0 | do |
714 | 0 | { |
715 | 0 | unsigned bit; |
716 | 0 | const CLzmaProb *probLit; |
717 | 0 | matchByte += matchByte; |
718 | 0 | bit = offs; |
719 | 0 | offs &= matchByte; |
720 | 0 | probLit = prob + (offs + bit + symbol); |
721 | 0 | GET_BIT2_CHECK(probLit, symbol, offs ^= bit; , ; ) |
722 | 0 | } |
723 | 0 | while (symbol < 0x100); |
724 | 0 | } |
725 | 0 | res = DUMMY_LIT; |
726 | 0 | } |
727 | 0 | else |
728 | 0 | { |
729 | 0 | unsigned len; |
730 | 0 | UPDATE_1_CHECK; |
731 | |
|
732 | 0 | prob = probs + IsRep + state; |
733 | 0 | IF_BIT_0_CHECK(prob) |
734 | 0 | { |
735 | 0 | UPDATE_0_CHECK; |
736 | 0 | state = 0; |
737 | 0 | prob = probs + LenCoder; |
738 | 0 | res = DUMMY_MATCH; |
739 | 0 | } |
740 | 0 | else |
741 | 0 | { |
742 | 0 | UPDATE_1_CHECK; |
743 | 0 | res = DUMMY_REP; |
744 | 0 | prob = probs + IsRepG0 + state; |
745 | 0 | IF_BIT_0_CHECK(prob) |
746 | 0 | { |
747 | 0 | UPDATE_0_CHECK; |
748 | 0 | prob = probs + IsRep0Long + COMBINED_PS_STATE; |
749 | 0 | IF_BIT_0_CHECK(prob) |
750 | 0 | { |
751 | 0 | UPDATE_0_CHECK; |
752 | 0 | NORMALIZE_CHECK; |
753 | 0 | return DUMMY_REP; |
754 | 0 | } |
755 | 0 | else |
756 | 0 | { |
757 | 0 | UPDATE_1_CHECK; |
758 | 0 | } |
759 | 0 | } |
760 | 0 | else |
761 | 0 | { |
762 | 0 | UPDATE_1_CHECK; |
763 | 0 | prob = probs + IsRepG1 + state; |
764 | 0 | IF_BIT_0_CHECK(prob) |
765 | 0 | { |
766 | 0 | UPDATE_0_CHECK; |
767 | 0 | } |
768 | 0 | else |
769 | 0 | { |
770 | 0 | UPDATE_1_CHECK; |
771 | 0 | prob = probs + IsRepG2 + state; |
772 | 0 | IF_BIT_0_CHECK(prob) |
773 | 0 | { |
774 | 0 | UPDATE_0_CHECK; |
775 | 0 | } |
776 | 0 | else |
777 | 0 | { |
778 | 0 | UPDATE_1_CHECK; |
779 | 0 | } |
780 | 0 | } |
781 | 0 | } |
782 | 0 | state = kNumStates; |
783 | 0 | prob = probs + RepLenCoder; |
784 | 0 | } |
785 | 0 | { |
786 | 0 | unsigned limit, offset; |
787 | 0 | const CLzmaProb *probLen = prob + LenChoice; |
788 | 0 | IF_BIT_0_CHECK(probLen) |
789 | 0 | { |
790 | 0 | UPDATE_0_CHECK; |
791 | 0 | probLen = prob + LenLow + GET_LEN_STATE; |
792 | 0 | offset = 0; |
793 | 0 | limit = 1 << kLenNumLowBits; |
794 | 0 | } |
795 | 0 | else |
796 | 0 | { |
797 | 0 | UPDATE_1_CHECK; |
798 | 0 | probLen = prob + LenChoice2; |
799 | 0 | IF_BIT_0_CHECK(probLen) |
800 | 0 | { |
801 | 0 | UPDATE_0_CHECK; |
802 | 0 | probLen = prob + LenLow + GET_LEN_STATE + (1 << kLenNumLowBits); |
803 | 0 | offset = kLenNumLowSymbols; |
804 | 0 | limit = 1 << kLenNumLowBits; |
805 | 0 | } |
806 | 0 | else |
807 | 0 | { |
808 | 0 | UPDATE_1_CHECK; |
809 | 0 | probLen = prob + LenHigh; |
810 | 0 | offset = kLenNumLowSymbols * 2; |
811 | 0 | limit = 1 << kLenNumHighBits; |
812 | 0 | } |
813 | 0 | } |
814 | 0 | TREE_DECODE_CHECK(probLen, limit, len); |
815 | 0 | len += offset; |
816 | 0 | } |
817 | | |
818 | 0 | if (state < 4) |
819 | 0 | { |
820 | 0 | unsigned posSlot; |
821 | 0 | prob = probs + PosSlot + |
822 | 0 | ((len < kNumLenToPosStates - 1 ? len : kNumLenToPosStates - 1) << |
823 | 0 | kNumPosSlotBits); |
824 | 0 | TREE_DECODE_CHECK(prob, 1 << kNumPosSlotBits, posSlot); |
825 | 0 | if (posSlot >= kStartPosModelIndex) |
826 | 0 | { |
827 | 0 | unsigned numDirectBits = ((posSlot >> 1) - 1); |
828 | | |
829 | | /* if (bufLimit - buf >= 8) return DUMMY_MATCH; */ |
830 | |
|
831 | 0 | if (posSlot < kEndPosModelIndex) |
832 | 0 | { |
833 | 0 | prob = probs + SpecPos + ((2 | (posSlot & 1)) << numDirectBits); |
834 | 0 | } |
835 | 0 | else |
836 | 0 | { |
837 | 0 | numDirectBits -= kNumAlignBits; |
838 | 0 | do |
839 | 0 | { |
840 | 0 | NORMALIZE_CHECK |
841 | 0 | range >>= 1; |
842 | 0 | code -= range & (((code - range) >> 31) - 1); |
843 | | /* if (code >= range) code -= range; */ |
844 | 0 | } |
845 | 0 | while (--numDirectBits); |
846 | 0 | prob = probs + Align; |
847 | 0 | numDirectBits = kNumAlignBits; |
848 | 0 | } |
849 | 0 | { |
850 | 0 | unsigned i = 1; |
851 | 0 | unsigned m = 1; |
852 | 0 | do |
853 | 0 | { |
854 | 0 | REV_BIT_CHECK(prob, i, m); |
855 | 0 | } |
856 | 0 | while (--numDirectBits); |
857 | 0 | } |
858 | 0 | } |
859 | 0 | } |
860 | 0 | } |
861 | 0 | } |
862 | 0 | NORMALIZE_CHECK; |
863 | 0 | return res; |
864 | 0 | } |
865 | | |
866 | | |
867 | | static void LzmaDec_InitDicAndState(CLzmaDec *p, BoolInt initDic, BoolInt initState) |
868 | 0 | { |
869 | 0 | p->remainLen = kMatchSpecLenStart + 1; |
870 | 0 | p->tempBufSize = 0; |
871 | |
|
872 | 0 | if (initDic) |
873 | 0 | { |
874 | 0 | p->processedPos = 0; |
875 | 0 | p->checkDicSize = 0; |
876 | 0 | p->remainLen = kMatchSpecLenStart + 2; |
877 | 0 | } |
878 | 0 | if (initState) |
879 | 0 | p->remainLen = kMatchSpecLenStart + 2; |
880 | 0 | } |
881 | | |
882 | | void LzmaDec_Init(CLzmaDec *p) |
883 | 0 | { |
884 | 0 | p->dicPos = 0; |
885 | 0 | LzmaDec_InitDicAndState(p, True, True); |
886 | 0 | } |
887 | | |
888 | | |
889 | | SRes LzmaDec_DecodeToDic(CLzmaDec *p, SizeT dicLimit, const Byte *src, SizeT *srcLen, |
890 | | ELzmaFinishMode finishMode, ELzmaStatus *status, SizeT memlimit) |
891 | 0 | { |
892 | 0 | SizeT inSize = *srcLen; |
893 | 0 | (*srcLen) = 0; |
894 | | |
895 | 0 | *status = LZMA_STATUS_NOT_SPECIFIED; |
896 | |
|
897 | 0 | if (p->remainLen > kMatchSpecLenStart) |
898 | 0 | { |
899 | 0 | for (; inSize > 0 && p->tempBufSize < RC_INIT_SIZE; (*srcLen)++, inSize--) |
900 | 0 | p->tempBuf[p->tempBufSize++] = *src++; |
901 | 0 | if (p->tempBufSize != 0 && p->tempBuf[0] != 0) |
902 | 0 | return SZ_ERROR_DATA; |
903 | 0 | if (p->tempBufSize < RC_INIT_SIZE) |
904 | 0 | { |
905 | 0 | *status = LZMA_STATUS_NEEDS_MORE_INPUT; |
906 | 0 | return SZ_OK; |
907 | 0 | } |
908 | 0 | p->code = |
909 | 0 | ((UInt32)p->tempBuf[1] << 24) |
910 | 0 | | ((UInt32)p->tempBuf[2] << 16) |
911 | 0 | | ((UInt32)p->tempBuf[3] << 8) |
912 | 0 | | ((UInt32)p->tempBuf[4]); |
913 | 0 | p->range = 0xFFFFFFFF; |
914 | 0 | p->tempBufSize = 0; |
915 | |
|
916 | 0 | if (p->remainLen > kMatchSpecLenStart + 1) |
917 | 0 | { |
918 | 0 | SizeT numProbs = LzmaProps_GetNumProbs(&p->prop); |
919 | 0 | SizeT i; |
920 | 0 | CLzmaProb *probs = p->probs; |
921 | 0 | for (i = 0; i < numProbs; i++) |
922 | 0 | probs[i] = kBitModelTotal >> 1; |
923 | 0 | p->reps[0] = p->reps[1] = p->reps[2] = p->reps[3] = 1; |
924 | 0 | p->state = 0; |
925 | 0 | } |
926 | |
|
927 | 0 | p->remainLen = 0; |
928 | 0 | } |
929 | | |
930 | 0 | LzmaDec_WriteRem(p, dicLimit); |
931 | |
|
932 | 0 | while (p->remainLen != kMatchSpecLenStart) |
933 | 0 | { |
934 | 0 | int checkEndMarkNow = 0; |
935 | |
|
936 | 0 | if (p->dicPos >= dicLimit) |
937 | 0 | { |
938 | 0 | if (p->remainLen == 0 && p->code == 0) |
939 | 0 | { |
940 | 0 | *status = LZMA_STATUS_MAYBE_FINISHED_WITHOUT_MARK; |
941 | 0 | return SZ_OK; |
942 | 0 | } |
943 | 0 | if (finishMode == LZMA_FINISH_ANY) |
944 | 0 | { |
945 | 0 | *status = LZMA_STATUS_NOT_FINISHED; |
946 | 0 | return SZ_OK; |
947 | 0 | } |
948 | 0 | if (p->remainLen != 0) |
949 | 0 | { |
950 | 0 | *status = LZMA_STATUS_NOT_FINISHED; |
951 | 0 | return SZ_ERROR_DATA; |
952 | 0 | } |
953 | 0 | checkEndMarkNow = 1; |
954 | 0 | } |
955 | | |
956 | 0 | if (p->tempBufSize == 0) |
957 | 0 | { |
958 | 0 | SizeT processed; |
959 | 0 | const Byte *bufLimit; |
960 | 0 | if (inSize < LZMA_REQUIRED_INPUT_MAX || checkEndMarkNow) |
961 | 0 | { |
962 | 0 | int dummyRes = LzmaDec_TryDummy(p, src, inSize); |
963 | 0 | if (dummyRes == DUMMY_ERROR) |
964 | 0 | { |
965 | 0 | memcpy(p->tempBuf, src, inSize); |
966 | 0 | p->tempBufSize = (unsigned)inSize; |
967 | 0 | (*srcLen) += inSize; |
968 | 0 | *status = LZMA_STATUS_NEEDS_MORE_INPUT; |
969 | 0 | return SZ_OK; |
970 | 0 | } |
971 | 0 | if (checkEndMarkNow && dummyRes != DUMMY_MATCH) |
972 | 0 | { |
973 | 0 | *status = LZMA_STATUS_NOT_FINISHED; |
974 | 0 | return SZ_ERROR_DATA; |
975 | 0 | } |
976 | 0 | bufLimit = src; |
977 | 0 | } |
978 | 0 | else |
979 | 0 | bufLimit = src + inSize - LZMA_REQUIRED_INPUT_MAX; |
980 | 0 | p->buf = src; |
981 | 0 | if (LzmaDec_DecodeReal2(p, dicLimit, bufLimit, memlimit) != 0) |
982 | 0 | return SZ_ERROR_DATA; |
983 | 0 | processed = (SizeT)(p->buf - src); |
984 | 0 | (*srcLen) += processed; |
985 | 0 | src += processed; |
986 | 0 | inSize -= processed; |
987 | 0 | } |
988 | 0 | else |
989 | 0 | { |
990 | 0 | unsigned rem = p->tempBufSize, lookAhead = 0; |
991 | 0 | while (rem < LZMA_REQUIRED_INPUT_MAX && lookAhead < inSize) |
992 | 0 | p->tempBuf[rem++] = src[lookAhead++]; |
993 | 0 | p->tempBufSize = rem; |
994 | 0 | if (rem < LZMA_REQUIRED_INPUT_MAX || checkEndMarkNow) |
995 | 0 | { |
996 | 0 | int dummyRes = LzmaDec_TryDummy(p, p->tempBuf, (SizeT)rem); |
997 | 0 | if (dummyRes == DUMMY_ERROR) |
998 | 0 | { |
999 | 0 | (*srcLen) += (SizeT)lookAhead; |
1000 | 0 | *status = LZMA_STATUS_NEEDS_MORE_INPUT; |
1001 | 0 | return SZ_OK; |
1002 | 0 | } |
1003 | 0 | if (checkEndMarkNow && dummyRes != DUMMY_MATCH) |
1004 | 0 | { |
1005 | 0 | *status = LZMA_STATUS_NOT_FINISHED; |
1006 | 0 | return SZ_ERROR_DATA; |
1007 | 0 | } |
1008 | 0 | } |
1009 | 0 | p->buf = p->tempBuf; |
1010 | 0 | if (LzmaDec_DecodeReal2(p, dicLimit, p->buf, memlimit) != 0) |
1011 | 0 | return SZ_ERROR_DATA; |
1012 | | |
1013 | 0 | { |
1014 | 0 | unsigned kkk = (unsigned)(p->buf - p->tempBuf); |
1015 | 0 | if (rem < kkk) |
1016 | 0 | return SZ_ERROR_FAIL; /* some internal error */ |
1017 | 0 | rem -= kkk; |
1018 | 0 | if (lookAhead < rem) |
1019 | 0 | return SZ_ERROR_FAIL; /* some internal error */ |
1020 | 0 | lookAhead -= rem; |
1021 | 0 | } |
1022 | 0 | (*srcLen) += (SizeT)lookAhead; |
1023 | 0 | src += lookAhead; |
1024 | 0 | inSize -= (SizeT)lookAhead; |
1025 | 0 | p->tempBufSize = 0; |
1026 | 0 | } |
1027 | 0 | } |
1028 | | |
1029 | 0 | if (p->code != 0) |
1030 | 0 | return SZ_ERROR_DATA; |
1031 | 0 | *status = LZMA_STATUS_FINISHED_WITH_MARK; |
1032 | 0 | return SZ_OK; |
1033 | 0 | } |
1034 | | |
1035 | | |
1036 | | SRes LzmaDec_DecodeToBuf(CLzmaDec *p, Byte *dest, SizeT *destLen, const Byte *src, SizeT *srcLen, ELzmaFinishMode finishMode, ELzmaStatus *status, SizeT memlimit) |
1037 | 0 | { |
1038 | 0 | SizeT outSize = *destLen; |
1039 | 0 | SizeT inSize = *srcLen; |
1040 | 0 | *srcLen = *destLen = 0; |
1041 | 0 | for (;;) |
1042 | 0 | { |
1043 | 0 | SizeT inSizeCur = inSize, outSizeCur, dicPos; |
1044 | 0 | ELzmaFinishMode curFinishMode; |
1045 | 0 | SRes res; |
1046 | 0 | if (p->dicPos == p->dicBufSize) { |
1047 | 0 | if (p->dicBufSize < p->prop.dicSize) { |
1048 | 0 | if (p->dicBufSize < memlimit) { |
1049 | 0 | p->dicBufSize = p->dicBufSize << 2; |
1050 | 0 | if (p->dicBufSize > memlimit) { |
1051 | 0 | p->dicBufSize = memlimit; |
1052 | 0 | } |
1053 | 0 | if (p->dicBufSize > p->prop.dicSize) { |
1054 | 0 | p->dicBufSize = p->prop.dicSize; |
1055 | 0 | } |
1056 | 0 | Byte *tmp = realloc(p->dic, p->dicBufSize); |
1057 | 0 | if (!tmp) { |
1058 | 0 | return SZ_ERROR_MEM; |
1059 | 0 | } |
1060 | 0 | p->dic = tmp; |
1061 | 0 | } else { |
1062 | 0 | return SZ_ERROR_MEM; |
1063 | 0 | } |
1064 | 0 | } else { |
1065 | 0 | p->dicPos = 0; |
1066 | 0 | } |
1067 | 0 | } |
1068 | 0 | dicPos = p->dicPos; |
1069 | 0 | if (outSize > p->dicBufSize - dicPos) |
1070 | 0 | { |
1071 | 0 | outSizeCur = p->dicBufSize; |
1072 | 0 | curFinishMode = LZMA_FINISH_ANY; |
1073 | 0 | } |
1074 | 0 | else |
1075 | 0 | { |
1076 | 0 | outSizeCur = dicPos + outSize; |
1077 | 0 | curFinishMode = finishMode; |
1078 | 0 | } |
1079 | |
|
1080 | 0 | res = LzmaDec_DecodeToDic(p, outSizeCur, src, &inSizeCur, curFinishMode, status, memlimit); |
1081 | 0 | src += inSizeCur; |
1082 | 0 | inSize -= inSizeCur; |
1083 | 0 | *srcLen += inSizeCur; |
1084 | 0 | outSizeCur = p->dicPos - dicPos; |
1085 | 0 | memcpy(dest, p->dic + dicPos, outSizeCur); |
1086 | 0 | dest += outSizeCur; |
1087 | 0 | outSize -= outSizeCur; |
1088 | 0 | *destLen += outSizeCur; |
1089 | 0 | if (res != 0) |
1090 | 0 | return res; |
1091 | 0 | if (outSizeCur == 0 || outSize == 0) |
1092 | 0 | return SZ_OK; |
1093 | 0 | } |
1094 | 0 | } |
1095 | | |
1096 | | void LzmaDec_FreeProbs(CLzmaDec *p, ISzAllocPtr alloc) |
1097 | 0 | { |
1098 | 0 | ISzAlloc_Free(alloc, p->probs); |
1099 | 0 | p->probs = NULL; |
1100 | 0 | } |
1101 | | |
1102 | | static void LzmaDec_FreeDict(CLzmaDec *p, ISzAllocPtr alloc) |
1103 | 0 | { |
1104 | 0 | ISzAlloc_Free(alloc, p->dic); |
1105 | 0 | p->dic = NULL; |
1106 | 0 | } |
1107 | | |
1108 | | void LzmaDec_Free(CLzmaDec *p, ISzAllocPtr alloc) |
1109 | 0 | { |
1110 | 0 | LzmaDec_FreeProbs(p, alloc); |
1111 | 0 | LzmaDec_FreeDict(p, alloc); |
1112 | 0 | } |
1113 | | |
1114 | | SRes LzmaProps_Decode(CLzmaProps *p, const Byte *data, unsigned size) |
1115 | 0 | { |
1116 | 0 | UInt32 dicSize; |
1117 | 0 | Byte d; |
1118 | | |
1119 | 0 | if (size < LZMA_PROPS_SIZE) |
1120 | 0 | return SZ_ERROR_UNSUPPORTED; |
1121 | 0 | else |
1122 | 0 | dicSize = data[1] | ((UInt32)data[2] << 8) | ((UInt32)data[3] << 16) | ((UInt32)data[4] << 24); |
1123 | | |
1124 | 0 | if (dicSize < LZMA_DIC_MIN) |
1125 | 0 | dicSize = LZMA_DIC_MIN; |
1126 | 0 | p->dicSize = dicSize; |
1127 | |
|
1128 | 0 | d = data[0]; |
1129 | 0 | if (d >= (9 * 5 * 5)) |
1130 | 0 | return SZ_ERROR_UNSUPPORTED; |
1131 | | |
1132 | 0 | p->lc = (Byte)(d % 9); |
1133 | 0 | d /= 9; |
1134 | 0 | p->pb = (Byte)(d / 5); |
1135 | 0 | p->lp = (Byte)(d % 5); |
1136 | 0 | p->_pad_ = 0; |
1137 | |
|
1138 | 0 | return SZ_OK; |
1139 | 0 | } |
1140 | | |
1141 | | static SRes LzmaDec_AllocateProbs2(CLzmaDec *p, const CLzmaProps *propNew, ISzAllocPtr alloc) |
1142 | 0 | { |
1143 | 0 | UInt32 numProbs = LzmaProps_GetNumProbs(propNew); |
1144 | 0 | if (!p->probs || numProbs != p->numProbs) |
1145 | 0 | { |
1146 | 0 | LzmaDec_FreeProbs(p, alloc); |
1147 | 0 | p->probs = (CLzmaProb *)ISzAlloc_Alloc(alloc, numProbs * sizeof(CLzmaProb)); |
1148 | 0 | if (!p->probs) |
1149 | 0 | return SZ_ERROR_MEM; |
1150 | 0 | p->probs_1664 = p->probs + 1664; |
1151 | 0 | p->numProbs = numProbs; |
1152 | 0 | } |
1153 | 0 | return SZ_OK; |
1154 | 0 | } |
1155 | | |
1156 | | SRes LzmaDec_AllocateProbs(CLzmaDec *p, const Byte *props, unsigned propsSize, ISzAllocPtr alloc) |
1157 | 0 | { |
1158 | 0 | CLzmaProps propNew; |
1159 | 0 | RINOK(LzmaProps_Decode(&propNew, props, propsSize)); |
1160 | 0 | RINOK(LzmaDec_AllocateProbs2(p, &propNew, alloc)); |
1161 | 0 | p->prop = propNew; |
1162 | 0 | return SZ_OK; |
1163 | 0 | } |
1164 | | |
1165 | | SRes LzmaDec_Allocate(CLzmaDec *p, const Byte *props, unsigned propsSize, ISzAllocPtr alloc) |
1166 | 0 | { |
1167 | 0 | CLzmaProps propNew; |
1168 | 0 | SizeT dicBufSize; |
1169 | 0 | RINOK(LzmaProps_Decode(&propNew, props, propsSize)); |
1170 | 0 | RINOK(LzmaDec_AllocateProbs2(p, &propNew, alloc)); |
1171 | |
|
1172 | 0 | { |
1173 | 0 | UInt32 dictSize = propNew.dicSize; |
1174 | 0 | SizeT mask = ((UInt32)1 << 12) - 1; |
1175 | 0 | if (dictSize >= ((UInt32)1 << 30)) mask = ((UInt32)1 << 22) - 1; |
1176 | 0 | else if (dictSize >= ((UInt32)1 << 22)) mask = ((UInt32)1 << 20) - 1;; |
1177 | 0 | dicBufSize = ((SizeT)dictSize + mask) & ~mask; |
1178 | 0 | if (dicBufSize < dictSize) |
1179 | 0 | dicBufSize = dictSize; |
1180 | 0 | } |
1181 | 0 | if (dicBufSize > LZMA_DIC_MIN) { |
1182 | 0 | dicBufSize = LZMA_DIC_MIN; |
1183 | 0 | } |
1184 | |
|
1185 | 0 | if (!p->dic || dicBufSize != p->dicBufSize) |
1186 | 0 | { |
1187 | 0 | LzmaDec_FreeDict(p, alloc); |
1188 | 0 | p->dic = (Byte *)ISzAlloc_Alloc(alloc, dicBufSize); |
1189 | 0 | if (!p->dic) |
1190 | 0 | { |
1191 | 0 | LzmaDec_FreeProbs(p, alloc); |
1192 | 0 | return SZ_ERROR_MEM; |
1193 | 0 | } |
1194 | 0 | } |
1195 | 0 | p->dicBufSize = dicBufSize; |
1196 | 0 | p->prop = propNew; |
1197 | 0 | return SZ_OK; |
1198 | 0 | } |
1199 | | |
1200 | | SRes LzmaDecode(Byte *dest, SizeT *destLen, const Byte *src, SizeT *srcLen, |
1201 | | const Byte *propData, unsigned propSize, ELzmaFinishMode finishMode, |
1202 | | ELzmaStatus *status, ISzAllocPtr alloc) |
1203 | 0 | { |
1204 | 0 | CLzmaDec p; |
1205 | 0 | SRes res; |
1206 | 0 | SizeT outSize = *destLen, inSize = *srcLen; |
1207 | 0 | *destLen = *srcLen = 0; |
1208 | 0 | *status = LZMA_STATUS_NOT_SPECIFIED; |
1209 | 0 | if (inSize < RC_INIT_SIZE) |
1210 | 0 | return SZ_ERROR_INPUT_EOF; |
1211 | 0 | LzmaDec_Construct(&p); |
1212 | 0 | RINOK(LzmaDec_AllocateProbs(&p, propData, propSize, alloc)); |
1213 | 0 | p.dic = dest; |
1214 | 0 | p.dicBufSize = outSize; |
1215 | 0 | LzmaDec_Init(&p); |
1216 | 0 | *srcLen = inSize; |
1217 | 0 | res = LzmaDec_DecodeToDic(&p, outSize, src, srcLen, finishMode, status, SIZE_MAX); |
1218 | 0 | *destLen = p.dicPos; |
1219 | 0 | if (res == SZ_OK && *status == LZMA_STATUS_NEEDS_MORE_INPUT) |
1220 | 0 | res = SZ_ERROR_INPUT_EOF; |
1221 | 0 | LzmaDec_FreeProbs(&p, alloc); |
1222 | 0 | return res; |
1223 | 0 | } |