Coverage Report

Created: 2025-07-23 07:29

/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
}