Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(941)

Side by Side Diff: third_party/lzma_sdk/LzmaDec.cc

Issue 10025017: [OTS] Add lzma_sdk (Closed) Base URL: http://ots.googlecode.com/svn/trunk/
Patch Set: Created 8 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
« no previous file with comments | « third_party/lzma_sdk/LzmaDec.h ('k') | third_party/lzma_sdk/LzmaEnc.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Property Changes:
Added: svn:eol-style
+ LF
OLDNEW
(Empty)
1 /* LzmaDec.c -- LZMA Decoder
2 2009-09-20 : Igor Pavlov : Public domain */
3
4 #include "LzmaDec.h"
5
6 #include <string.h>
7
8 namespace ots {
9 namespace lzma {
10
11 #define kNumTopBits 24
12 #define kTopValue ((UInt32)1 << kNumTopBits)
13
14 #define kNumBitModelTotalBits 11
15 #define kBitModelTotal (1 << kNumBitModelTotalBits)
16 #define kNumMoveBits 5
17
18 #define RC_INIT_SIZE 5
19
20 #define NORMALIZE if (range < kTopValue) { range <<= 8; code = (code << 8) | (*b uf++); }
21
22 #define IF_BIT_0(p) ttt = *(p); NORMALIZE; bound = (range >> kNumBitModelTotalBi ts) * ttt; if (code < bound)
23 #define UPDATE_0(p) range = bound; *(p) = (CLzmaProb)(ttt + ((kBitModelTotal - t tt) >> kNumMoveBits));
24 #define UPDATE_1(p) range -= bound; code -= bound; *(p) = (CLzmaProb)(ttt - (ttt >> kNumMoveBits));
25 #define GET_BIT2(p, i, A0, A1) IF_BIT_0(p) \
26 { UPDATE_0(p); i = (i + i); A0; } else \
27 { UPDATE_1(p); i = (i + i) + 1; A1; }
28 #define GET_BIT(p, i) GET_BIT2(p, i, ; , ;)
29
30 #define TREE_GET_BIT(probs, i) { GET_BIT((probs + i), i); }
31 #define TREE_DECODE(probs, limit, i) \
32 { i = 1; do { TREE_GET_BIT(probs, i); } while (i < limit); i -= limit; }
33
34 /* #define _LZMA_SIZE_OPT */
35
36 #ifdef _LZMA_SIZE_OPT
37 #define TREE_6_DECODE(probs, i) TREE_DECODE(probs, (1 << 6), i)
38 #else
39 #define TREE_6_DECODE(probs, i) \
40 { i = 1; \
41 TREE_GET_BIT(probs, i); \
42 TREE_GET_BIT(probs, i); \
43 TREE_GET_BIT(probs, i); \
44 TREE_GET_BIT(probs, i); \
45 TREE_GET_BIT(probs, i); \
46 TREE_GET_BIT(probs, i); \
47 i -= 0x40; }
48 #endif
49
50 #define NORMALIZE_CHECK if (range < kTopValue) { if (buf >= bufLimit) return DUM MY_ERROR; range <<= 8; code = (code << 8) | (*buf++); }
51
52 #define IF_BIT_0_CHECK(p) ttt = *(p); NORMALIZE_CHECK; bound = (range >> kNumBit ModelTotalBits) * ttt; if (code < bound)
53 #define UPDATE_0_CHECK range = bound;
54 #define UPDATE_1_CHECK range -= bound; code -= bound;
55 #define GET_BIT2_CHECK(p, i, A0, A1) IF_BIT_0_CHECK(p) \
56 { UPDATE_0_CHECK; i = (i + i); A0; } else \
57 { UPDATE_1_CHECK; i = (i + i) + 1; A1; }
58 #define GET_BIT_CHECK(p, i) GET_BIT2_CHECK(p, i, ; , ;)
59 #define TREE_DECODE_CHECK(probs, limit, i) \
60 { i = 1; do { GET_BIT_CHECK(probs + i, i) } while (i < limit); i -= limit; }
61
62
63 #define kNumPosBitsMax 4
64 #define kNumPosStatesMax (1 << kNumPosBitsMax)
65
66 #define kLenNumLowBits 3
67 #define kLenNumLowSymbols (1 << kLenNumLowBits)
68 #define kLenNumMidBits 3
69 #define kLenNumMidSymbols (1 << kLenNumMidBits)
70 #define kLenNumHighBits 8
71 #define kLenNumHighSymbols (1 << kLenNumHighBits)
72
73 #define LenChoice 0
74 #define LenChoice2 (LenChoice + 1)
75 #define LenLow (LenChoice2 + 1)
76 #define LenMid (LenLow + (kNumPosStatesMax << kLenNumLowBits))
77 #define LenHigh (LenMid + (kNumPosStatesMax << kLenNumMidBits))
78 #define kNumLenProbs (LenHigh + kLenNumHighSymbols)
79
80
81 #define kNumStates 12
82 #define kNumLitStates 7
83
84 #define kStartPosModelIndex 4
85 #define kEndPosModelIndex 14
86 #define kNumFullDistances (1 << (kEndPosModelIndex >> 1))
87
88 #define kNumPosSlotBits 6
89 #define kNumLenToPosStates 4
90
91 #define kNumAlignBits 4
92 #define kAlignTableSize (1 << kNumAlignBits)
93
94 #define kMatchMinLen 2
95 #define kMatchSpecLenStart (kMatchMinLen + kLenNumLowSymbols + kLenNumMidSymbols + kLenNumHighSymbols)
96
97 #define IsMatch 0
98 #define IsRep (IsMatch + (kNumStates << kNumPosBitsMax))
99 #define IsRepG0 (IsRep + kNumStates)
100 #define IsRepG1 (IsRepG0 + kNumStates)
101 #define IsRepG2 (IsRepG1 + kNumStates)
102 #define IsRep0Long (IsRepG2 + kNumStates)
103 #define PosSlot (IsRep0Long + (kNumStates << kNumPosBitsMax))
104 #define SpecPos (PosSlot + (kNumLenToPosStates << kNumPosSlotBits))
105 #define Align (SpecPos + kNumFullDistances - kEndPosModelIndex)
106 #define LenCoder (Align + kAlignTableSize)
107 #define RepLenCoder (LenCoder + kNumLenProbs)
108 #define Literal (RepLenCoder + kNumLenProbs)
109
110 #define LZMA_BASE_SIZE 1846
111 #define LZMA_LIT_SIZE 768
112
113 #define LzmaProps_GetNumProbs(p) ((UInt32)LZMA_BASE_SIZE + (LZMA_LIT_SIZE << ((p )->lc + (p)->lp)))
114
115 #if Literal != LZMA_BASE_SIZE
116 StopCompilingDueBUG
117 #endif
118
119 #define LZMA_DIC_MIN (1 << 12)
120
121 /* First LZMA-symbol is always decoded.
122 And it decodes new LZMA-symbols while (buf < bufLimit), but "buf" is without las t normalization
123 Out:
124 Result:
125 SZ_OK - OK
126 SZ_ERROR_DATA - Error
127 p->remainLen:
128 < kMatchSpecLenStart : normal remain
129 = kMatchSpecLenStart : finished
130 = kMatchSpecLenStart + 1 : Flush marker
131 = kMatchSpecLenStart + 2 : State Init Marker
132 */
133
134 static int MY_FAST_CALL LzmaDec_DecodeReal(CLzmaDec *p, SizeT limit, const Byte *bufLimit)
135 {
136 CLzmaProb *probs = p->probs;
137
138 unsigned state = p->state;
139 UInt32 rep0 = p->reps[0], rep1 = p->reps[1], rep2 = p->reps[2], rep3 = p->reps [3];
140 unsigned pbMask = ((unsigned)1 << (p->prop.pb)) - 1;
141 unsigned lpMask = ((unsigned)1 << (p->prop.lp)) - 1;
142 unsigned lc = p->prop.lc;
143
144 Byte *dic = p->dic;
145 SizeT dicBufSize = p->dicBufSize;
146 SizeT dicPos = p->dicPos;
147
148 UInt32 processedPos = p->processedPos;
149 UInt32 checkDicSize = p->checkDicSize;
150 unsigned len = 0;
151
152 const Byte *buf = p->buf;
153 UInt32 range = p->range;
154 UInt32 code = p->code;
155
156 do
157 {
158 CLzmaProb *prob;
159 UInt32 bound;
160 unsigned ttt;
161 unsigned posState = processedPos & pbMask;
162
163 prob = probs + IsMatch + (state << kNumPosBitsMax) + posState;
164 IF_BIT_0(prob)
165 {
166 unsigned symbol;
167 UPDATE_0(prob);
168 prob = probs + Literal;
169 if (checkDicSize != 0 || processedPos != 0)
170 prob += (LZMA_LIT_SIZE * (((processedPos & lpMask) << lc) +
171 (dic[(dicPos == 0 ? dicBufSize : dicPos) - 1] >> (8 - lc))));
172
173 if (state < kNumLitStates)
174 {
175 state -= (state < 4) ? state : 3;
176 symbol = 1;
177 do { GET_BIT(prob + symbol, symbol) } while (symbol < 0x100);
178 }
179 else
180 {
181 unsigned matchByte = p->dic[(dicPos - rep0) + ((dicPos < rep0) ? dicBufS ize : 0)];
182 unsigned offs = 0x100;
183 state -= (state < 10) ? 3 : 6;
184 symbol = 1;
185 do
186 {
187 unsigned bit;
188 CLzmaProb *probLit;
189 matchByte <<= 1;
190 bit = (matchByte & offs);
191 probLit = prob + offs + bit + symbol;
192 GET_BIT2(probLit, symbol, offs &= ~bit, offs &= bit)
193 }
194 while (symbol < 0x100);
195 }
196 dic[dicPos++] = (Byte)symbol;
197 processedPos++;
198 continue;
199 }
200 else
201 {
202 UPDATE_1(prob);
203 prob = probs + IsRep + state;
204 IF_BIT_0(prob)
205 {
206 UPDATE_0(prob);
207 state += kNumStates;
208 prob = probs + LenCoder;
209 }
210 else
211 {
212 UPDATE_1(prob);
213 if (checkDicSize == 0 && processedPos == 0)
214 return SZ_ERROR_DATA;
215 prob = probs + IsRepG0 + state;
216 IF_BIT_0(prob)
217 {
218 UPDATE_0(prob);
219 prob = probs + IsRep0Long + (state << kNumPosBitsMax) + posState;
220 IF_BIT_0(prob)
221 {
222 UPDATE_0(prob);
223 dic[dicPos] = dic[(dicPos - rep0) + ((dicPos < rep0) ? dicBufSize : 0)];
224 dicPos++;
225 processedPos++;
226 state = state < kNumLitStates ? 9 : 11;
227 continue;
228 }
229 UPDATE_1(prob);
230 }
231 else
232 {
233 UInt32 distance;
234 UPDATE_1(prob);
235 prob = probs + IsRepG1 + state;
236 IF_BIT_0(prob)
237 {
238 UPDATE_0(prob);
239 distance = rep1;
240 }
241 else
242 {
243 UPDATE_1(prob);
244 prob = probs + IsRepG2 + state;
245 IF_BIT_0(prob)
246 {
247 UPDATE_0(prob);
248 distance = rep2;
249 }
250 else
251 {
252 UPDATE_1(prob);
253 distance = rep3;
254 rep3 = rep2;
255 }
256 rep2 = rep1;
257 }
258 rep1 = rep0;
259 rep0 = distance;
260 }
261 state = state < kNumLitStates ? 8 : 11;
262 prob = probs + RepLenCoder;
263 }
264 {
265 unsigned limit, offset;
266 CLzmaProb *probLen = prob + LenChoice;
267 IF_BIT_0(probLen)
268 {
269 UPDATE_0(probLen);
270 probLen = prob + LenLow + (posState << kLenNumLowBits);
271 offset = 0;
272 limit = (1 << kLenNumLowBits);
273 }
274 else
275 {
276 UPDATE_1(probLen);
277 probLen = prob + LenChoice2;
278 IF_BIT_0(probLen)
279 {
280 UPDATE_0(probLen);
281 probLen = prob + LenMid + (posState << kLenNumMidBits);
282 offset = kLenNumLowSymbols;
283 limit = (1 << kLenNumMidBits);
284 }
285 else
286 {
287 UPDATE_1(probLen);
288 probLen = prob + LenHigh;
289 offset = kLenNumLowSymbols + kLenNumMidSymbols;
290 limit = (1 << kLenNumHighBits);
291 }
292 }
293 TREE_DECODE(probLen, limit, len);
294 len += offset;
295 }
296
297 if (state >= kNumStates)
298 {
299 UInt32 distance;
300 prob = probs + PosSlot +
301 ((len < kNumLenToPosStates ? len : kNumLenToPosStates - 1) << kNumPo sSlotBits);
302 TREE_6_DECODE(prob, distance);
303 if (distance >= kStartPosModelIndex)
304 {
305 unsigned posSlot = (unsigned)distance;
306 int numDirectBits = (int)(((distance >> 1) - 1));
307 distance = (2 | (distance & 1));
308 if (posSlot < kEndPosModelIndex)
309 {
310 distance <<= numDirectBits;
311 prob = probs + SpecPos + distance - posSlot - 1;
312 {
313 UInt32 mask = 1;
314 unsigned i = 1;
315 do
316 {
317 GET_BIT2(prob + i, i, ; , distance |= mask);
318 mask <<= 1;
319 }
320 while (--numDirectBits != 0);
321 }
322 }
323 else
324 {
325 numDirectBits -= kNumAlignBits;
326 do
327 {
328 NORMALIZE
329 range >>= 1;
330
331 {
332 UInt32 t;
333 code -= range;
334 t = (0 - ((UInt32)code >> 31)); /* (UInt32)((Int32)code >> 31) * /
335 distance = (distance << 1) + (t + 1);
336 code += range & t;
337 }
338 /*
339 distance <<= 1;
340 if (code >= range)
341 {
342 code -= range;
343 distance |= 1;
344 }
345 */
346 }
347 while (--numDirectBits != 0);
348 prob = probs + Align;
349 distance <<= kNumAlignBits;
350 {
351 unsigned i = 1;
352 GET_BIT2(prob + i, i, ; , distance |= 1);
353 GET_BIT2(prob + i, i, ; , distance |= 2);
354 GET_BIT2(prob + i, i, ; , distance |= 4);
355 GET_BIT2(prob + i, i, ; , distance |= 8);
356 }
357 if (distance == (UInt32)0xFFFFFFFF)
358 {
359 len += kMatchSpecLenStart;
360 state -= kNumStates;
361 break;
362 }
363 }
364 }
365 rep3 = rep2;
366 rep2 = rep1;
367 rep1 = rep0;
368 rep0 = distance + 1;
369 if (checkDicSize == 0)
370 {
371 if (distance >= processedPos)
372 return SZ_ERROR_DATA;
373 }
374 else if (distance >= checkDicSize)
375 return SZ_ERROR_DATA;
376 state = (state < kNumStates + kNumLitStates) ? kNumLitStates : kNumLitSt ates + 3;
377 }
378
379 len += kMatchMinLen;
380
381 if (limit == dicPos)
382 return SZ_ERROR_DATA;
383 {
384 SizeT rem = limit - dicPos;
385 unsigned curLen = ((rem < len) ? (unsigned)rem : len);
386 SizeT pos = (dicPos - rep0) + ((dicPos < rep0) ? dicBufSize : 0);
387
388 processedPos += curLen;
389
390 len -= curLen;
391 if (pos + curLen <= dicBufSize)
392 {
393 Byte *dest = dic + dicPos;
394 ptrdiff_t src = (ptrdiff_t)pos - (ptrdiff_t)dicPos;
395 const Byte *lim = dest + curLen;
396 dicPos += curLen;
397 do
398 *(dest) = (Byte)*(dest + src);
399 while (++dest != lim);
400 }
401 else
402 {
403 do
404 {
405 dic[dicPos++] = dic[pos];
406 if (++pos == dicBufSize)
407 pos = 0;
408 }
409 while (--curLen != 0);
410 }
411 }
412 }
413 }
414 while (dicPos < limit && buf < bufLimit);
415 NORMALIZE;
416 p->buf = buf;
417 p->range = range;
418 p->code = code;
419 p->remainLen = len;
420 p->dicPos = dicPos;
421 p->processedPos = processedPos;
422 p->reps[0] = rep0;
423 p->reps[1] = rep1;
424 p->reps[2] = rep2;
425 p->reps[3] = rep3;
426 p->state = state;
427
428 return SZ_OK;
429 }
430
431 static void MY_FAST_CALL LzmaDec_WriteRem(CLzmaDec *p, SizeT limit)
432 {
433 if (p->remainLen != 0 && p->remainLen < kMatchSpecLenStart)
434 {
435 Byte *dic = p->dic;
436 SizeT dicPos = p->dicPos;
437 SizeT dicBufSize = p->dicBufSize;
438 unsigned len = p->remainLen;
439 UInt32 rep0 = p->reps[0];
440 if (limit - dicPos < len)
441 len = (unsigned)(limit - dicPos);
442
443 if (p->checkDicSize == 0 && p->prop.dicSize - p->processedPos <= len)
444 p->checkDicSize = p->prop.dicSize;
445
446 p->processedPos += len;
447 p->remainLen -= len;
448 while (len-- != 0)
449 {
450 dic[dicPos] = dic[(dicPos - rep0) + ((dicPos < rep0) ? dicBufSize : 0)];
451 dicPos++;
452 }
453 p->dicPos = dicPos;
454 }
455 }
456
457 static int MY_FAST_CALL LzmaDec_DecodeReal2(CLzmaDec *p, SizeT limit, const Byte *bufLimit)
458 {
459 do
460 {
461 SizeT limit2 = limit;
462 if (p->checkDicSize == 0)
463 {
464 UInt32 rem = p->prop.dicSize - p->processedPos;
465 if (limit - p->dicPos > rem)
466 limit2 = p->dicPos + rem;
467 }
468 RINOK(LzmaDec_DecodeReal(p, limit2, bufLimit));
469 if (p->processedPos >= p->prop.dicSize)
470 p->checkDicSize = p->prop.dicSize;
471 LzmaDec_WriteRem(p, limit);
472 }
473 while (p->dicPos < limit && p->buf < bufLimit && p->remainLen < kMatchSpecLenS tart);
474
475 if (p->remainLen > kMatchSpecLenStart)
476 {
477 p->remainLen = kMatchSpecLenStart;
478 }
479 return 0;
480 }
481
482 typedef enum
483 {
484 DUMMY_ERROR, /* unexpected end of input stream */
485 DUMMY_LIT,
486 DUMMY_MATCH,
487 DUMMY_REP
488 } ELzmaDummy;
489
490 static ELzmaDummy LzmaDec_TryDummy(const CLzmaDec *p, const Byte *buf, SizeT inS ize)
491 {
492 UInt32 range = p->range;
493 UInt32 code = p->code;
494 const Byte *bufLimit = buf + inSize;
495 CLzmaProb *probs = p->probs;
496 unsigned state = p->state;
497 ELzmaDummy res;
498
499 {
500 CLzmaProb *prob;
501 UInt32 bound;
502 unsigned ttt;
503 unsigned posState = (p->processedPos) & ((1 << p->prop.pb) - 1);
504
505 prob = probs + IsMatch + (state << kNumPosBitsMax) + posState;
506 IF_BIT_0_CHECK(prob)
507 {
508 UPDATE_0_CHECK
509
510 /* if (bufLimit - buf >= 7) return DUMMY_LIT; */
511
512 prob = probs + Literal;
513 if (p->checkDicSize != 0 || p->processedPos != 0)
514 prob += (LZMA_LIT_SIZE *
515 ((((p->processedPos) & ((1 << (p->prop.lp)) - 1)) << p->prop.lc) +
516 (p->dic[(p->dicPos == 0 ? p->dicBufSize : p->dicPos) - 1] >> (8 - p->p rop.lc))));
517
518 if (state < kNumLitStates)
519 {
520 unsigned symbol = 1;
521 do { GET_BIT_CHECK(prob + symbol, symbol) } while (symbol < 0x100);
522 }
523 else
524 {
525 unsigned matchByte = p->dic[p->dicPos - p->reps[0] +
526 ((p->dicPos < p->reps[0]) ? p->dicBufSize : 0)];
527 unsigned offs = 0x100;
528 unsigned symbol = 1;
529 do
530 {
531 unsigned bit;
532 CLzmaProb *probLit;
533 matchByte <<= 1;
534 bit = (matchByte & offs);
535 probLit = prob + offs + bit + symbol;
536 GET_BIT2_CHECK(probLit, symbol, offs &= ~bit, offs &= bit)
537 }
538 while (symbol < 0x100);
539 }
540 res = DUMMY_LIT;
541 }
542 else
543 {
544 unsigned len;
545 UPDATE_1_CHECK;
546
547 prob = probs + IsRep + state;
548 IF_BIT_0_CHECK(prob)
549 {
550 UPDATE_0_CHECK;
551 state = 0;
552 prob = probs + LenCoder;
553 res = DUMMY_MATCH;
554 }
555 else
556 {
557 UPDATE_1_CHECK;
558 res = DUMMY_REP;
559 prob = probs + IsRepG0 + state;
560 IF_BIT_0_CHECK(prob)
561 {
562 UPDATE_0_CHECK;
563 prob = probs + IsRep0Long + (state << kNumPosBitsMax) + posState;
564 IF_BIT_0_CHECK(prob)
565 {
566 UPDATE_0_CHECK;
567 NORMALIZE_CHECK;
568 return DUMMY_REP;
569 }
570 else
571 {
572 UPDATE_1_CHECK;
573 }
574 }
575 else
576 {
577 UPDATE_1_CHECK;
578 prob = probs + IsRepG1 + state;
579 IF_BIT_0_CHECK(prob)
580 {
581 UPDATE_0_CHECK;
582 }
583 else
584 {
585 UPDATE_1_CHECK;
586 prob = probs + IsRepG2 + state;
587 IF_BIT_0_CHECK(prob)
588 {
589 UPDATE_0_CHECK;
590 }
591 else
592 {
593 UPDATE_1_CHECK;
594 }
595 }
596 }
597 state = kNumStates;
598 prob = probs + RepLenCoder;
599 }
600 {
601 unsigned limit, offset;
602 CLzmaProb *probLen = prob + LenChoice;
603 IF_BIT_0_CHECK(probLen)
604 {
605 UPDATE_0_CHECK;
606 probLen = prob + LenLow + (posState << kLenNumLowBits);
607 offset = 0;
608 limit = 1 << kLenNumLowBits;
609 }
610 else
611 {
612 UPDATE_1_CHECK;
613 probLen = prob + LenChoice2;
614 IF_BIT_0_CHECK(probLen)
615 {
616 UPDATE_0_CHECK;
617 probLen = prob + LenMid + (posState << kLenNumMidBits);
618 offset = kLenNumLowSymbols;
619 limit = 1 << kLenNumMidBits;
620 }
621 else
622 {
623 UPDATE_1_CHECK;
624 probLen = prob + LenHigh;
625 offset = kLenNumLowSymbols + kLenNumMidSymbols;
626 limit = 1 << kLenNumHighBits;
627 }
628 }
629 TREE_DECODE_CHECK(probLen, limit, len);
630 len += offset;
631 }
632
633 if (state < 4)
634 {
635 unsigned posSlot;
636 prob = probs + PosSlot +
637 ((len < kNumLenToPosStates ? len : kNumLenToPosStates - 1) <<
638 kNumPosSlotBits);
639 TREE_DECODE_CHECK(prob, 1 << kNumPosSlotBits, posSlot);
640 if (posSlot >= kStartPosModelIndex)
641 {
642 int numDirectBits = ((posSlot >> 1) - 1);
643
644 /* if (bufLimit - buf >= 8) return DUMMY_MATCH; */
645
646 if (posSlot < kEndPosModelIndex)
647 {
648 prob = probs + SpecPos + ((2 | (posSlot & 1)) << numDirectBits) - po sSlot - 1;
649 }
650 else
651 {
652 numDirectBits -= kNumAlignBits;
653 do
654 {
655 NORMALIZE_CHECK
656 range >>= 1;
657 code -= range & (((code - range) >> 31) - 1);
658 /* if (code >= range) code -= range; */
659 }
660 while (--numDirectBits != 0);
661 prob = probs + Align;
662 numDirectBits = kNumAlignBits;
663 }
664 {
665 unsigned i = 1;
666 do
667 {
668 GET_BIT_CHECK(prob + i, i);
669 }
670 while (--numDirectBits != 0);
671 }
672 }
673 }
674 }
675 }
676 NORMALIZE_CHECK;
677 return res;
678 }
679
680
681 static void LzmaDec_InitRc(CLzmaDec *p, const Byte *data)
682 {
683 p->code = ((UInt32)data[1] << 24) | ((UInt32)data[2] << 16) | ((UInt32)data[3] << 8) | ((UInt32)data[4]);
684 p->range = 0xFFFFFFFF;
685 p->needFlush = 0;
686 }
687
688 void LzmaDec_InitDicAndState(CLzmaDec *p, Bool initDic, Bool initState)
689 {
690 p->needFlush = 1;
691 p->remainLen = 0;
692 p->tempBufSize = 0;
693
694 if (initDic)
695 {
696 p->processedPos = 0;
697 p->checkDicSize = 0;
698 p->needInitState = 1;
699 }
700 if (initState)
701 p->needInitState = 1;
702 }
703
704 void LzmaDec_Init(CLzmaDec *p)
705 {
706 p->dicPos = 0;
707 LzmaDec_InitDicAndState(p, True, True);
708 }
709
710 static void LzmaDec_InitStateReal(CLzmaDec *p)
711 {
712 UInt32 numProbs = Literal + ((UInt32)LZMA_LIT_SIZE << (p->prop.lc + p->prop.lp ));
713 UInt32 i;
714 CLzmaProb *probs = p->probs;
715 for (i = 0; i < numProbs; i++)
716 probs[i] = kBitModelTotal >> 1;
717 p->reps[0] = p->reps[1] = p->reps[2] = p->reps[3] = 1;
718 p->state = 0;
719 p->needInitState = 0;
720 }
721
722 SRes LzmaDec_DecodeToDic(CLzmaDec *p, SizeT dicLimit, const Byte *src, SizeT *sr cLen,
723 ELzmaFinishMode finishMode, ELzmaStatus *status)
724 {
725 SizeT inSize = *srcLen;
726 (*srcLen) = 0;
727 LzmaDec_WriteRem(p, dicLimit);
728
729 *status = LZMA_STATUS_NOT_SPECIFIED;
730
731 while (p->remainLen != kMatchSpecLenStart)
732 {
733 int checkEndMarkNow;
734
735 if (p->needFlush != 0)
736 {
737 for (; inSize > 0 && p->tempBufSize < RC_INIT_SIZE; (*srcLen)++, inSize- -)
738 p->tempBuf[p->tempBufSize++] = *src++;
739 if (p->tempBufSize < RC_INIT_SIZE)
740 {
741 *status = LZMA_STATUS_NEEDS_MORE_INPUT;
742 return SZ_OK;
743 }
744 if (p->tempBuf[0] != 0)
745 return SZ_ERROR_DATA;
746
747 LzmaDec_InitRc(p, p->tempBuf);
748 p->tempBufSize = 0;
749 }
750
751 checkEndMarkNow = 0;
752 if (p->dicPos >= dicLimit)
753 {
754 if (p->remainLen == 0 && p->code == 0)
755 {
756 *status = LZMA_STATUS_MAYBE_FINISHED_WITHOUT_MARK;
757 return SZ_OK;
758 }
759 if (finishMode == LZMA_FINISH_ANY)
760 {
761 *status = LZMA_STATUS_NOT_FINISHED;
762 return SZ_OK;
763 }
764 if (p->remainLen != 0)
765 {
766 *status = LZMA_STATUS_NOT_FINISHED;
767 return SZ_ERROR_DATA;
768 }
769 checkEndMarkNow = 1;
770 }
771
772 if (p->needInitState)
773 LzmaDec_InitStateReal(p);
774
775 if (p->tempBufSize == 0)
776 {
777 SizeT processed;
778 const Byte *bufLimit;
779 if (inSize < LZMA_REQUIRED_INPUT_MAX || checkEndMarkNow)
780 {
781 int dummyRes = LzmaDec_TryDummy(p, src, inSize);
782 if (dummyRes == DUMMY_ERROR)
783 {
784 memcpy(p->tempBuf, src, inSize);
785 p->tempBufSize = (unsigned)inSize;
786 (*srcLen) += inSize;
787 *status = LZMA_STATUS_NEEDS_MORE_INPUT;
788 return SZ_OK;
789 }
790 if (checkEndMarkNow && dummyRes != DUMMY_MATCH)
791 {
792 *status = LZMA_STATUS_NOT_FINISHED;
793 return SZ_ERROR_DATA;
794 }
795 bufLimit = src;
796 }
797 else
798 bufLimit = src + inSize - LZMA_REQUIRED_INPUT_MAX;
799 p->buf = src;
800 if (LzmaDec_DecodeReal2(p, dicLimit, bufLimit) != 0)
801 return SZ_ERROR_DATA;
802 processed = (SizeT)(p->buf - src);
803 (*srcLen) += processed;
804 src += processed;
805 inSize -= processed;
806 }
807 else
808 {
809 unsigned rem = p->tempBufSize, lookAhead = 0;
810 while (rem < LZMA_REQUIRED_INPUT_MAX && lookAhead < inSize)
811 p->tempBuf[rem++] = src[lookAhead++];
812 p->tempBufSize = rem;
813 if (rem < LZMA_REQUIRED_INPUT_MAX || checkEndMarkNow)
814 {
815 int dummyRes = LzmaDec_TryDummy(p, p->tempBuf, rem);
816 if (dummyRes == DUMMY_ERROR)
817 {
818 (*srcLen) += lookAhead;
819 *status = LZMA_STATUS_NEEDS_MORE_INPUT;
820 return SZ_OK;
821 }
822 if (checkEndMarkNow && dummyRes != DUMMY_MATCH)
823 {
824 *status = LZMA_STATUS_NOT_FINISHED;
825 return SZ_ERROR_DATA;
826 }
827 }
828 p->buf = p->tempBuf;
829 if (LzmaDec_DecodeReal2(p, dicLimit, p->buf) != 0)
830 return SZ_ERROR_DATA;
831 lookAhead -= (rem - (unsigned)(p->buf - p->tempBuf));
832 (*srcLen) += lookAhead;
833 src += lookAhead;
834 inSize -= lookAhead;
835 p->tempBufSize = 0;
836 }
837 }
838 if (p->code == 0)
839 *status = LZMA_STATUS_FINISHED_WITH_MARK;
840 return (p->code == 0) ? SZ_OK : SZ_ERROR_DATA;
841 }
842
843 SRes LzmaDec_DecodeToBuf(CLzmaDec *p, Byte *dest, SizeT *destLen, const Byte *sr c, SizeT *srcLen, ELzmaFinishMode finishMode, ELzmaStatus *status)
844 {
845 SizeT outSize = *destLen;
846 SizeT inSize = *srcLen;
847 *srcLen = *destLen = 0;
848 for (;;)
849 {
850 SizeT inSizeCur = inSize, outSizeCur, dicPos;
851 ELzmaFinishMode curFinishMode;
852 SRes res;
853 if (p->dicPos == p->dicBufSize)
854 p->dicPos = 0;
855 dicPos = p->dicPos;
856 if (outSize > p->dicBufSize - dicPos)
857 {
858 outSizeCur = p->dicBufSize;
859 curFinishMode = LZMA_FINISH_ANY;
860 }
861 else
862 {
863 outSizeCur = dicPos + outSize;
864 curFinishMode = finishMode;
865 }
866
867 res = LzmaDec_DecodeToDic(p, outSizeCur, src, &inSizeCur, curFinishMode, sta tus);
868 src += inSizeCur;
869 inSize -= inSizeCur;
870 *srcLen += inSizeCur;
871 outSizeCur = p->dicPos - dicPos;
872 memcpy(dest, p->dic + dicPos, outSizeCur);
873 dest += outSizeCur;
874 outSize -= outSizeCur;
875 *destLen += outSizeCur;
876 if (res != 0)
877 return res;
878 if (outSizeCur == 0 || outSize == 0)
879 return SZ_OK;
880 }
881 }
882
883 void LzmaDec_FreeProbs(CLzmaDec *p, ISzAlloc *alloc)
884 {
885 alloc->Free(alloc, p->probs);
886 p->probs = 0;
887 }
888
889 static void LzmaDec_FreeDict(CLzmaDec *p, ISzAlloc *alloc)
890 {
891 alloc->Free(alloc, p->dic);
892 p->dic = 0;
893 }
894
895 void LzmaDec_Free(CLzmaDec *p, ISzAlloc *alloc)
896 {
897 LzmaDec_FreeProbs(p, alloc);
898 LzmaDec_FreeDict(p, alloc);
899 }
900
901 SRes LzmaProps_Decode(CLzmaProps *p, const Byte *data, unsigned size)
902 {
903 UInt32 dicSize;
904 Byte d;
905
906 if (size < LZMA_PROPS_SIZE)
907 return SZ_ERROR_UNSUPPORTED;
908 else
909 dicSize = data[1] | ((UInt32)data[2] << 8) | ((UInt32)data[3] << 16) | ((UIn t32)data[4] << 24);
910
911 if (dicSize < LZMA_DIC_MIN)
912 dicSize = LZMA_DIC_MIN;
913 p->dicSize = dicSize;
914
915 d = data[0];
916 if (d >= (9 * 5 * 5))
917 return SZ_ERROR_UNSUPPORTED;
918
919 p->lc = d % 9;
920 d /= 9;
921 p->pb = d / 5;
922 p->lp = d % 5;
923
924 return SZ_OK;
925 }
926
927 static SRes LzmaDec_AllocateProbs2(CLzmaDec *p, const CLzmaProps *propNew, ISzAl loc *alloc)
928 {
929 UInt32 numProbs = LzmaProps_GetNumProbs(propNew);
930 if (p->probs == 0 || numProbs != p->numProbs)
931 {
932 LzmaDec_FreeProbs(p, alloc);
933 p->probs = (CLzmaProb *)alloc->Alloc(alloc, numProbs * sizeof(CLzmaProb));
934 p->numProbs = numProbs;
935 if (p->probs == 0)
936 return SZ_ERROR_MEM;
937 }
938 return SZ_OK;
939 }
940
941 SRes LzmaDec_AllocateProbs(CLzmaDec *p, const Byte *props, unsigned propsSize, I SzAlloc *alloc)
942 {
943 CLzmaProps propNew;
944 RINOK(LzmaProps_Decode(&propNew, props, propsSize));
945 RINOK(LzmaDec_AllocateProbs2(p, &propNew, alloc));
946 p->prop = propNew;
947 return SZ_OK;
948 }
949
950 SRes LzmaDec_Allocate(CLzmaDec *p, const Byte *props, unsigned propsSize, ISzAll oc *alloc)
951 {
952 CLzmaProps propNew;
953 SizeT dicBufSize;
954 RINOK(LzmaProps_Decode(&propNew, props, propsSize));
955 RINOK(LzmaDec_AllocateProbs2(p, &propNew, alloc));
956 dicBufSize = propNew.dicSize;
957 if (p->dic == 0 || dicBufSize != p->dicBufSize)
958 {
959 LzmaDec_FreeDict(p, alloc);
960 p->dic = (Byte *)alloc->Alloc(alloc, dicBufSize);
961 if (p->dic == 0)
962 {
963 LzmaDec_FreeProbs(p, alloc);
964 return SZ_ERROR_MEM;
965 }
966 }
967 p->dicBufSize = dicBufSize;
968 p->prop = propNew;
969 return SZ_OK;
970 }
971
972 SRes LzmaDecode(Byte *dest, SizeT *destLen, const Byte *src, SizeT *srcLen,
973 const Byte *propData, unsigned propSize, ELzmaFinishMode finishMode,
974 ELzmaStatus *status, ISzAlloc *alloc)
975 {
976 CLzmaDec p;
977 SRes res;
978 SizeT inSize = *srcLen;
979 SizeT outSize = *destLen;
980 *srcLen = *destLen = 0;
981 if (inSize < RC_INIT_SIZE)
982 return SZ_ERROR_INPUT_EOF;
983
984 LzmaDec_Construct(&p);
985 res = LzmaDec_AllocateProbs(&p, propData, propSize, alloc);
986 if (res != 0)
987 return res;
988 p.dic = dest;
989 p.dicBufSize = outSize;
990
991 LzmaDec_Init(&p);
992
993 *srcLen = inSize;
994 res = LzmaDec_DecodeToDic(&p, outSize, src, srcLen, finishMode, status);
995
996 if (res == SZ_OK && *status == LZMA_STATUS_NEEDS_MORE_INPUT)
997 res = SZ_ERROR_INPUT_EOF;
998
999 (*destLen) = p.dicPos;
1000 LzmaDec_FreeProbs(&p, alloc);
1001 return res;
1002 }
1003
1004 } // namespace lzma
1005 } // namespace ots
OLDNEW
« no previous file with comments | « third_party/lzma_sdk/LzmaDec.h ('k') | third_party/lzma_sdk/LzmaEnc.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698