|
|
@@ -1,8 +1,8 @@
|
|
|
/*
|
|
|
LzmaDecode.c
|
|
|
- LZMA Decoder
|
|
|
+ LZMA Decoder (optimized for Speed version)
|
|
|
|
|
|
- LZMA SDK 4.05 Copyright (c) 1999-2004 Igor Pavlov (2004-08-25)
|
|
|
+ LZMA SDK 4.40 Copyright (c) 1999-2006 Igor Pavlov (2006-05-01)
|
|
|
http://www.7-zip.org/
|
|
|
|
|
|
LZMA SDK is licensed under two licenses:
|
|
|
@@ -12,19 +12,15 @@
|
|
|
follow rules of that license.
|
|
|
|
|
|
SPECIAL EXCEPTION:
|
|
|
- Igor Pavlov, as the author of this code, expressly permits you to
|
|
|
- statically or dynamically link your code (or bind by name) to the
|
|
|
- interfaces of this file without subjecting your linked code to the
|
|
|
+ Igor Pavlov, as the author of this Code, expressly permits you to
|
|
|
+ statically or dynamically link your Code (or bind by name) to the
|
|
|
+ interfaces of this file without subjecting your linked Code to the
|
|
|
terms of the CPL or GNU LGPL. Any modifications or additions
|
|
|
to this file, however, are subject to the LGPL or CPL terms.
|
|
|
*/
|
|
|
|
|
|
#include "LzmaDecode.h"
|
|
|
|
|
|
-#ifndef Byte
|
|
|
-#define Byte unsigned char
|
|
|
-#endif
|
|
|
-
|
|
|
#define kNumTopBits 24
|
|
|
#define kTopValue ((UInt32)1 << kNumTopBits)
|
|
|
|
|
|
@@ -32,242 +28,44 @@
|
|
|
#define kBitModelTotal (1 << kNumBitModelTotalBits)
|
|
|
#define kNumMoveBits 5
|
|
|
|
|
|
-typedef struct _CRangeDecoder
|
|
|
-{
|
|
|
- Byte *Buffer;
|
|
|
- Byte *BufferLim;
|
|
|
- UInt32 Range;
|
|
|
- UInt32 Code;
|
|
|
- #ifdef _LZMA_IN_CB
|
|
|
- ILzmaInCallback *InCallback;
|
|
|
- int Result;
|
|
|
- #endif
|
|
|
- int ExtraBytes;
|
|
|
-} CRangeDecoder;
|
|
|
-
|
|
|
-Byte RangeDecoderReadByte(CRangeDecoder *rd)
|
|
|
-{
|
|
|
- if (rd->Buffer == rd->BufferLim)
|
|
|
- {
|
|
|
- #ifdef _LZMA_IN_CB
|
|
|
- UInt32 size;
|
|
|
- rd->Result = rd->InCallback->Read(rd->InCallback, &rd->Buffer, &size);
|
|
|
- rd->BufferLim = rd->Buffer + size;
|
|
|
- if (size == 0)
|
|
|
- #endif
|
|
|
- {
|
|
|
- rd->ExtraBytes = 1;
|
|
|
- return 0xFF;
|
|
|
- }
|
|
|
- }
|
|
|
- return (*rd->Buffer++);
|
|
|
-}
|
|
|
+#define RC_READ_BYTE (*Buffer++)
|
|
|
|
|
|
-/* #define ReadByte (*rd->Buffer++) */
|
|
|
-#define ReadByte (RangeDecoderReadByte(rd))
|
|
|
+#define RC_INIT2 Code = 0; Range = 0xFFFFFFFF; \
|
|
|
+ { int i; for(i = 0; i < 5; i++) { RC_TEST; Code = (Code << 8) | RC_READ_BYTE; }}
|
|
|
|
|
|
-void RangeDecoderInit(CRangeDecoder *rd,
|
|
|
- #ifdef _LZMA_IN_CB
|
|
|
- ILzmaInCallback *inCallback
|
|
|
- #else
|
|
|
- Byte *stream, UInt32 bufferSize
|
|
|
- #endif
|
|
|
- )
|
|
|
-{
|
|
|
- int i;
|
|
|
- #ifdef _LZMA_IN_CB
|
|
|
- rd->InCallback = inCallback;
|
|
|
- rd->Buffer = rd->BufferLim = 0;
|
|
|
- #else
|
|
|
- rd->Buffer = stream;
|
|
|
- rd->BufferLim = stream + bufferSize;
|
|
|
- #endif
|
|
|
- rd->ExtraBytes = 0;
|
|
|
- rd->Code = 0;
|
|
|
- rd->Range = (0xFFFFFFFF);
|
|
|
- for(i = 0; i < 5; i++)
|
|
|
- rd->Code = (rd->Code << 8) | ReadByte;
|
|
|
-}
|
|
|
+#ifdef _LZMA_IN_CB
|
|
|
|
|
|
-#define RC_INIT_VAR UInt32 range = rd->Range; UInt32 code = rd->Code;
|
|
|
-#define RC_FLUSH_VAR rd->Range = range; rd->Code = code;
|
|
|
-#define RC_NORMALIZE if (range < kTopValue) { range <<= 8; code = (code << 8) | ReadByte; }
|
|
|
+#define RC_TEST { if (Buffer == BufferLim) \
|
|
|
+ { SizeT size; int result = InCallback->Read(InCallback, &Buffer, &size); if (result != LZMA_RESULT_OK) return result; \
|
|
|
+ BufferLim = Buffer + size; if (size == 0) return LZMA_RESULT_DATA_ERROR; }}
|
|
|
|
|
|
-UInt32 RangeDecoderDecodeDirectBits(CRangeDecoder *rd, int numTotalBits)
|
|
|
-{
|
|
|
- RC_INIT_VAR
|
|
|
- UInt32 result = 0;
|
|
|
- int i;
|
|
|
- for (i = numTotalBits; i > 0; i--)
|
|
|
- {
|
|
|
- /* UInt32 t; */
|
|
|
- range >>= 1;
|
|
|
+#define RC_INIT Buffer = BufferLim = 0; RC_INIT2
|
|
|
|
|
|
- result <<= 1;
|
|
|
- if (code >= range)
|
|
|
- {
|
|
|
- code -= range;
|
|
|
- result |= 1;
|
|
|
- }
|
|
|
- /*
|
|
|
- t = (code - range) >> 31;
|
|
|
- t &= 1;
|
|
|
- code -= range & (t - 1);
|
|
|
- result = (result + result) | (1 - t);
|
|
|
- */
|
|
|
- RC_NORMALIZE
|
|
|
- }
|
|
|
- RC_FLUSH_VAR
|
|
|
- return result;
|
|
|
-}
|
|
|
+#else
|
|
|
|
|
|
-int RangeDecoderBitDecode(CProb *prob, CRangeDecoder *rd)
|
|
|
-{
|
|
|
- UInt32 bound = (rd->Range >> kNumBitModelTotalBits) * *prob;
|
|
|
- if (rd->Code < bound)
|
|
|
- {
|
|
|
- rd->Range = bound;
|
|
|
- *prob += (kBitModelTotal - *prob) >> kNumMoveBits;
|
|
|
- if (rd->Range < kTopValue)
|
|
|
- {
|
|
|
- rd->Code = (rd->Code << 8) | ReadByte;
|
|
|
- rd->Range <<= 8;
|
|
|
- }
|
|
|
- return 0;
|
|
|
- }
|
|
|
- else
|
|
|
- {
|
|
|
- rd->Range -= bound;
|
|
|
- rd->Code -= bound;
|
|
|
- *prob -= (*prob) >> kNumMoveBits;
|
|
|
- if (rd->Range < kTopValue)
|
|
|
- {
|
|
|
- rd->Code = (rd->Code << 8) | ReadByte;
|
|
|
- rd->Range <<= 8;
|
|
|
- }
|
|
|
- return 1;
|
|
|
- }
|
|
|
-}
|
|
|
+#define RC_TEST { if (Buffer == BufferLim) return LZMA_RESULT_DATA_ERROR; }
|
|
|
|
|
|
-#define RC_GET_BIT2(prob, mi, A0, A1) \
|
|
|
- UInt32 bound = (range >> kNumBitModelTotalBits) * *prob; \
|
|
|
- if (code < bound) \
|
|
|
- { A0; range = bound; *prob += (kBitModelTotal - *prob) >> kNumMoveBits; mi <<= 1; } \
|
|
|
- else \
|
|
|
- { A1; range -= bound; code -= bound; *prob -= (*prob) >> kNumMoveBits; mi = (mi + mi) + 1; } \
|
|
|
- RC_NORMALIZE
|
|
|
+#define RC_INIT(buffer, bufferSize) Buffer = buffer; BufferLim = buffer + bufferSize; RC_INIT2
|
|
|
+
|
|
|
+#endif
|
|
|
|
|
|
-#define RC_GET_BIT(prob, mi) RC_GET_BIT2(prob, mi, ; , ;)
|
|
|
+#define RC_NORMALIZE if (Range < kTopValue) { RC_TEST; Range <<= 8; Code = (Code << 8) | RC_READ_BYTE; }
|
|
|
|
|
|
-int RangeDecoderBitTreeDecode(CProb *probs, int numLevels, CRangeDecoder *rd)
|
|
|
-{
|
|
|
- int mi = 1;
|
|
|
- int i;
|
|
|
- #ifdef _LZMA_LOC_OPT
|
|
|
- RC_INIT_VAR
|
|
|
- #endif
|
|
|
- for(i = numLevels; i > 0; i--)
|
|
|
- {
|
|
|
- #ifdef _LZMA_LOC_OPT
|
|
|
- CProb *prob = probs + mi;
|
|
|
- RC_GET_BIT(prob, mi)
|
|
|
- #else
|
|
|
- mi = (mi + mi) + RangeDecoderBitDecode(probs + mi, rd);
|
|
|
- #endif
|
|
|
- }
|
|
|
- #ifdef _LZMA_LOC_OPT
|
|
|
- RC_FLUSH_VAR
|
|
|
- #endif
|
|
|
- return mi - (1 << numLevels);
|
|
|
-}
|
|
|
+#define IfBit0(p) RC_NORMALIZE; bound = (Range >> kNumBitModelTotalBits) * *(p); if (Code < bound)
|
|
|
+#define UpdateBit0(p) Range = bound; *(p) += (kBitModelTotal - *(p)) >> kNumMoveBits;
|
|
|
+#define UpdateBit1(p) Range -= bound; Code -= bound; *(p) -= (*(p)) >> kNumMoveBits;
|
|
|
|
|
|
-int RangeDecoderReverseBitTreeDecode(CProb *probs, int numLevels, CRangeDecoder *rd)
|
|
|
-{
|
|
|
- int mi = 1;
|
|
|
- int i;
|
|
|
- int symbol = 0;
|
|
|
- #ifdef _LZMA_LOC_OPT
|
|
|
- RC_INIT_VAR
|
|
|
- #endif
|
|
|
- for(i = 0; i < numLevels; i++)
|
|
|
- {
|
|
|
- #ifdef _LZMA_LOC_OPT
|
|
|
- CProb *prob = probs + mi;
|
|
|
- RC_GET_BIT2(prob, mi, ; , symbol |= (1 << i))
|
|
|
- #else
|
|
|
- int bit = RangeDecoderBitDecode(probs + mi, rd);
|
|
|
- mi = mi + mi + bit;
|
|
|
- symbol |= (bit << i);
|
|
|
- #endif
|
|
|
- }
|
|
|
- #ifdef _LZMA_LOC_OPT
|
|
|
- RC_FLUSH_VAR
|
|
|
- #endif
|
|
|
- return symbol;
|
|
|
-}
|
|
|
+#define RC_GET_BIT2(p, mi, A0, A1) IfBit0(p) \
|
|
|
+ { UpdateBit0(p); mi <<= 1; A0; } else \
|
|
|
+ { UpdateBit1(p); mi = (mi + mi) + 1; A1; }
|
|
|
+
|
|
|
+#define RC_GET_BIT(p, mi) RC_GET_BIT2(p, mi, ; , ;)
|
|
|
|
|
|
-Byte LzmaLiteralDecode(CProb *probs, CRangeDecoder *rd)
|
|
|
-{
|
|
|
- int symbol = 1;
|
|
|
- #ifdef _LZMA_LOC_OPT
|
|
|
- RC_INIT_VAR
|
|
|
- #endif
|
|
|
- do
|
|
|
- {
|
|
|
- #ifdef _LZMA_LOC_OPT
|
|
|
- CProb *prob = probs + symbol;
|
|
|
- RC_GET_BIT(prob, symbol)
|
|
|
- #else
|
|
|
- symbol = (symbol + symbol) | RangeDecoderBitDecode(probs + symbol, rd);
|
|
|
- #endif
|
|
|
- }
|
|
|
- while (symbol < 0x100);
|
|
|
- #ifdef _LZMA_LOC_OPT
|
|
|
- RC_FLUSH_VAR
|
|
|
- #endif
|
|
|
- return symbol;
|
|
|
-}
|
|
|
+#define RangeDecoderBitTreeDecode(probs, numLevels, res) \
|
|
|
+ { int i = numLevels; res = 1; \
|
|
|
+ do { CProb *p = probs + res; RC_GET_BIT(p, res) } while(--i != 0); \
|
|
|
+ res -= (1 << numLevels); }
|
|
|
|
|
|
-Byte LzmaLiteralDecodeMatch(CProb *probs, CRangeDecoder *rd, Byte matchByte)
|
|
|
-{
|
|
|
- int symbol = 1;
|
|
|
- #ifdef _LZMA_LOC_OPT
|
|
|
- RC_INIT_VAR
|
|
|
- #endif
|
|
|
- do
|
|
|
- {
|
|
|
- int bit;
|
|
|
- int matchBit = (matchByte >> 7) & 1;
|
|
|
- matchByte <<= 1;
|
|
|
- #ifdef _LZMA_LOC_OPT
|
|
|
- {
|
|
|
- CProb *prob = probs + ((1 + matchBit) << 8) + symbol;
|
|
|
- RC_GET_BIT2(prob, symbol, bit = 0, bit = 1)
|
|
|
- }
|
|
|
- #else
|
|
|
- bit = RangeDecoderBitDecode(probs + ((1 + matchBit) << 8) + symbol, rd);
|
|
|
- symbol = (symbol << 1) | bit;
|
|
|
- #endif
|
|
|
- if (matchBit != bit)
|
|
|
- {
|
|
|
- while (symbol < 0x100)
|
|
|
- {
|
|
|
- #ifdef _LZMA_LOC_OPT
|
|
|
- CProb *prob = probs + symbol;
|
|
|
- RC_GET_BIT(prob, symbol)
|
|
|
- #else
|
|
|
- symbol = (symbol + symbol) | RangeDecoderBitDecode(probs + symbol, rd);
|
|
|
- #endif
|
|
|
- }
|
|
|
- break;
|
|
|
- }
|
|
|
- }
|
|
|
- while (symbol < 0x100);
|
|
|
- #ifdef _LZMA_LOC_OPT
|
|
|
- RC_FLUSH_VAR
|
|
|
- #endif
|
|
|
- return symbol;
|
|
|
-}
|
|
|
|
|
|
#define kNumPosBitsMax 4
|
|
|
#define kNumPosStatesMax (1 << kNumPosBitsMax)
|
|
|
@@ -286,19 +84,9 @@ Byte LzmaLiteralDecodeMatch(CProb *probs, CRangeDecoder *rd, Byte matchByte)
|
|
|
#define LenHigh (LenMid + (kNumPosStatesMax << kLenNumMidBits))
|
|
|
#define kNumLenProbs (LenHigh + kLenNumHighSymbols)
|
|
|
|
|
|
-int LzmaLenDecode(CProb *p, CRangeDecoder *rd, int posState)
|
|
|
-{
|
|
|
- if(RangeDecoderBitDecode(p + LenChoice, rd) == 0)
|
|
|
- return RangeDecoderBitTreeDecode(p + LenLow +
|
|
|
- (posState << kLenNumLowBits), kLenNumLowBits, rd);
|
|
|
- if(RangeDecoderBitDecode(p + LenChoice2, rd) == 0)
|
|
|
- return kLenNumLowSymbols + RangeDecoderBitTreeDecode(p + LenMid +
|
|
|
- (posState << kLenNumMidBits), kLenNumMidBits, rd);
|
|
|
- return kLenNumLowSymbols + kLenNumMidSymbols +
|
|
|
- RangeDecoderBitTreeDecode(p + LenHigh, kLenNumHighBits, rd);
|
|
|
-}
|
|
|
|
|
|
#define kNumStates 12
|
|
|
+#define kNumLitStates 7
|
|
|
|
|
|
#define kStartPosModelIndex 4
|
|
|
#define kEndPosModelIndex 14
|
|
|
@@ -329,94 +117,115 @@ int LzmaLenDecode(CProb *p, CRangeDecoder *rd, int posState)
|
|
|
StopCompilingDueBUG
|
|
|
#endif
|
|
|
|
|
|
-#ifdef _LZMA_OUT_READ
|
|
|
-
|
|
|
-typedef struct _LzmaVarState
|
|
|
+int LzmaDecodeProperties(CLzmaProperties *propsRes, const unsigned char *propsData, int size)
|
|
|
{
|
|
|
- CRangeDecoder RangeDecoder;
|
|
|
- Byte *Dictionary;
|
|
|
- UInt32 DictionarySize;
|
|
|
- UInt32 DictionaryPos;
|
|
|
- UInt32 GlobalPos;
|
|
|
- UInt32 Reps[4];
|
|
|
- int lc;
|
|
|
- int lp;
|
|
|
- int pb;
|
|
|
- int State;
|
|
|
- int PreviousIsMatch;
|
|
|
- int RemainLen;
|
|
|
-} LzmaVarState;
|
|
|
-
|
|
|
-int LzmaDecoderInit(
|
|
|
- unsigned char *buffer, UInt32 bufferSize,
|
|
|
- int lc, int lp, int pb,
|
|
|
- unsigned char *dictionary, UInt32 dictionarySize,
|
|
|
+ unsigned char prop0;
|
|
|
+ if (size < LZMA_PROPERTIES_SIZE)
|
|
|
+ return LZMA_RESULT_DATA_ERROR;
|
|
|
+ prop0 = propsData[0];
|
|
|
+ if (prop0 >= (9 * 5 * 5))
|
|
|
+ return LZMA_RESULT_DATA_ERROR;
|
|
|
+ {
|
|
|
+ for (propsRes->pb = 0; prop0 >= (9 * 5); propsRes->pb++, prop0 -= (9 * 5));
|
|
|
+ for (propsRes->lp = 0; prop0 >= 9; propsRes->lp++, prop0 -= 9);
|
|
|
+ propsRes->lc = prop0;
|
|
|
+ /*
|
|
|
+ unsigned char remainder = (unsigned char)(prop0 / 9);
|
|
|
+ propsRes->lc = prop0 % 9;
|
|
|
+ propsRes->pb = remainder / 5;
|
|
|
+ propsRes->lp = remainder % 5;
|
|
|
+ */
|
|
|
+ }
|
|
|
+
|
|
|
+ #ifdef _LZMA_OUT_READ
|
|
|
+ {
|
|
|
+ int i;
|
|
|
+ propsRes->DictionarySize = 0;
|
|
|
+ for (i = 0; i < 4; i++)
|
|
|
+ propsRes->DictionarySize += (UInt32)(propsData[1 + i]) << (i * 8);
|
|
|
+ if (propsRes->DictionarySize == 0)
|
|
|
+ propsRes->DictionarySize = 1;
|
|
|
+ }
|
|
|
+ #endif
|
|
|
+ return LZMA_RESULT_OK;
|
|
|
+}
|
|
|
+
|
|
|
+#define kLzmaStreamWasFinishedId (-1)
|
|
|
+
|
|
|
+int LzmaDecode(CLzmaDecoderState *vs,
|
|
|
#ifdef _LZMA_IN_CB
|
|
|
- ILzmaInCallback *inCallback
|
|
|
+ ILzmaInCallback *InCallback,
|
|
|
#else
|
|
|
- unsigned char *inStream, UInt32 inSize
|
|
|
+ const unsigned char *inStream, SizeT inSize, SizeT *inSizeProcessed,
|
|
|
#endif
|
|
|
- )
|
|
|
+ unsigned char *outStream, SizeT outSize, SizeT *outSizeProcessed)
|
|
|
{
|
|
|
- LzmaVarState *vs = (LzmaVarState *)buffer;
|
|
|
- CProb *p = (CProb *)(buffer + sizeof(LzmaVarState));
|
|
|
- UInt32 numProbs = Literal + ((UInt32)LZMA_LIT_SIZE << (lc + lp));
|
|
|
- UInt32 i;
|
|
|
- if (bufferSize < numProbs * sizeof(CProb) + sizeof(LzmaVarState))
|
|
|
- return LZMA_RESULT_NOT_ENOUGH_MEM;
|
|
|
- vs->Dictionary = dictionary;
|
|
|
- vs->DictionarySize = dictionarySize;
|
|
|
- vs->DictionaryPos = 0;
|
|
|
- vs->GlobalPos = 0;
|
|
|
- vs->Reps[0] = vs->Reps[1] = vs->Reps[2] = vs->Reps[3] = 1;
|
|
|
- vs->lc = lc;
|
|
|
- vs->lp = lp;
|
|
|
- vs->pb = pb;
|
|
|
- vs->State = 0;
|
|
|
- vs->PreviousIsMatch = 0;
|
|
|
- vs->RemainLen = 0;
|
|
|
- dictionary[dictionarySize - 1] = 0;
|
|
|
- for (i = 0; i < numProbs; i++)
|
|
|
- p[i] = kBitModelTotal >> 1;
|
|
|
- RangeDecoderInit(&vs->RangeDecoder,
|
|
|
- #ifdef _LZMA_IN_CB
|
|
|
- inCallback
|
|
|
- #else
|
|
|
- inStream, inSize
|
|
|
- #endif
|
|
|
- );
|
|
|
- return LZMA_RESULT_OK;
|
|
|
-}
|
|
|
+ CProb *p = vs->Probs;
|
|
|
+ SizeT nowPos = 0;
|
|
|
+ Byte previousByte = 0;
|
|
|
+ UInt32 posStateMask = (1 << (vs->Properties.pb)) - 1;
|
|
|
+ UInt32 literalPosMask = (1 << (vs->Properties.lp)) - 1;
|
|
|
+ int lc = vs->Properties.lc;
|
|
|
|
|
|
-int LzmaDecode(unsigned char *buffer,
|
|
|
- unsigned char *outStream, UInt32 outSize,
|
|
|
- UInt32 *outSizeProcessed)
|
|
|
-{
|
|
|
- LzmaVarState *vs = (LzmaVarState *)buffer;
|
|
|
- CProb *p = (CProb *)(buffer + sizeof(LzmaVarState));
|
|
|
- CRangeDecoder rd = vs->RangeDecoder;
|
|
|
+ #ifdef _LZMA_OUT_READ
|
|
|
+
|
|
|
+ UInt32 Range = vs->Range;
|
|
|
+ UInt32 Code = vs->Code;
|
|
|
+ #ifdef _LZMA_IN_CB
|
|
|
+ const Byte *Buffer = vs->Buffer;
|
|
|
+ const Byte *BufferLim = vs->BufferLim;
|
|
|
+ #else
|
|
|
+ const Byte *Buffer = inStream;
|
|
|
+ const Byte *BufferLim = inStream + inSize;
|
|
|
+ #endif
|
|
|
int state = vs->State;
|
|
|
- int previousIsMatch = vs->PreviousIsMatch;
|
|
|
- Byte previousByte;
|
|
|
UInt32 rep0 = vs->Reps[0], rep1 = vs->Reps[1], rep2 = vs->Reps[2], rep3 = vs->Reps[3];
|
|
|
- UInt32 nowPos = 0;
|
|
|
- UInt32 posStateMask = (1 << (vs->pb)) - 1;
|
|
|
- UInt32 literalPosMask = (1 << (vs->lp)) - 1;
|
|
|
- int lc = vs->lc;
|
|
|
int len = vs->RemainLen;
|
|
|
UInt32 globalPos = vs->GlobalPos;
|
|
|
+ UInt32 distanceLimit = vs->DistanceLimit;
|
|
|
|
|
|
Byte *dictionary = vs->Dictionary;
|
|
|
- UInt32 dictionarySize = vs->DictionarySize;
|
|
|
+ UInt32 dictionarySize = vs->Properties.DictionarySize;
|
|
|
UInt32 dictionaryPos = vs->DictionaryPos;
|
|
|
|
|
|
- if (len == -1)
|
|
|
- {
|
|
|
- *outSizeProcessed = 0;
|
|
|
+ Byte tempDictionary[4];
|
|
|
+
|
|
|
+ #ifndef _LZMA_IN_CB
|
|
|
+ *inSizeProcessed = 0;
|
|
|
+ #endif
|
|
|
+ *outSizeProcessed = 0;
|
|
|
+ if (len == kLzmaStreamWasFinishedId)
|
|
|
return LZMA_RESULT_OK;
|
|
|
+
|
|
|
+ if (dictionarySize == 0)
|
|
|
+ {
|
|
|
+ dictionary = tempDictionary;
|
|
|
+ dictionarySize = 1;
|
|
|
+ tempDictionary[0] = vs->TempDictionary[0];
|
|
|
}
|
|
|
|
|
|
- while(len > 0 && nowPos < outSize)
|
|
|
+ if (len == kLzmaNeedInitId)
|
|
|
+ {
|
|
|
+ {
|
|
|
+ UInt32 numProbs = Literal + ((UInt32)LZMA_LIT_SIZE << (lc + vs->Properties.lp));
|
|
|
+ UInt32 i;
|
|
|
+ for (i = 0; i < numProbs; i++)
|
|
|
+ p[i] = kBitModelTotal >> 1;
|
|
|
+ rep0 = rep1 = rep2 = rep3 = 1;
|
|
|
+ state = 0;
|
|
|
+ globalPos = 0;
|
|
|
+ distanceLimit = 0;
|
|
|
+ dictionaryPos = 0;
|
|
|
+ dictionary[dictionarySize - 1] = 0;
|
|
|
+ #ifdef _LZMA_IN_CB
|
|
|
+ RC_INIT;
|
|
|
+ #else
|
|
|
+ RC_INIT(inStream, inSize);
|
|
|
+ #endif
|
|
|
+ }
|
|
|
+ len = 0;
|
|
|
+ }
|
|
|
+ while(len != 0 && nowPos < outSize)
|
|
|
{
|
|
|
UInt32 pos = dictionaryPos - rep0;
|
|
|
if (pos >= dictionarySize)
|
|
|
@@ -430,47 +239,41 @@ int LzmaDecode(unsigned char *buffer,
|
|
|
previousByte = dictionary[dictionarySize - 1];
|
|
|
else
|
|
|
previousByte = dictionary[dictionaryPos - 1];
|
|
|
-#else
|
|
|
|
|
|
-int LzmaDecode(
|
|
|
- Byte *buffer, UInt32 bufferSize,
|
|
|
- int lc, int lp, int pb,
|
|
|
- #ifdef _LZMA_IN_CB
|
|
|
- ILzmaInCallback *inCallback,
|
|
|
- #else
|
|
|
- unsigned char *inStream, UInt32 inSize,
|
|
|
- #endif
|
|
|
- unsigned char *outStream, UInt32 outSize,
|
|
|
- UInt32 *outSizeProcessed)
|
|
|
-{
|
|
|
- UInt32 numProbs = Literal + ((UInt32)LZMA_LIT_SIZE << (lc + lp));
|
|
|
- CProb *p = (CProb *)buffer;
|
|
|
- CRangeDecoder rd;
|
|
|
- UInt32 i;
|
|
|
+ #else /* if !_LZMA_OUT_READ */
|
|
|
+
|
|
|
int state = 0;
|
|
|
- int previousIsMatch = 0;
|
|
|
- Byte previousByte = 0;
|
|
|
UInt32 rep0 = 1, rep1 = 1, rep2 = 1, rep3 = 1;
|
|
|
- UInt32 nowPos = 0;
|
|
|
- UInt32 posStateMask = (1 << pb) - 1;
|
|
|
- UInt32 literalPosMask = (1 << lp) - 1;
|
|
|
int len = 0;
|
|
|
- if (bufferSize < numProbs * sizeof(CProb))
|
|
|
- return LZMA_RESULT_NOT_ENOUGH_MEM;
|
|
|
- for (i = 0; i < numProbs; i++)
|
|
|
- p[i] = kBitModelTotal >> 1;
|
|
|
- RangeDecoderInit(&rd,
|
|
|
- #ifdef _LZMA_IN_CB
|
|
|
- inCallback
|
|
|
- #else
|
|
|
- inStream, inSize
|
|
|
- #endif
|
|
|
- );
|
|
|
-#endif
|
|
|
+ const Byte *Buffer;
|
|
|
+ const Byte *BufferLim;
|
|
|
+ UInt32 Range;
|
|
|
+ UInt32 Code;
|
|
|
|
|
|
+ #ifndef _LZMA_IN_CB
|
|
|
+ *inSizeProcessed = 0;
|
|
|
+ #endif
|
|
|
*outSizeProcessed = 0;
|
|
|
+
|
|
|
+ {
|
|
|
+ UInt32 i;
|
|
|
+ UInt32 numProbs = Literal + ((UInt32)LZMA_LIT_SIZE << (lc + vs->Properties.lp));
|
|
|
+ for (i = 0; i < numProbs; i++)
|
|
|
+ p[i] = kBitModelTotal >> 1;
|
|
|
+ }
|
|
|
+
|
|
|
+ #ifdef _LZMA_IN_CB
|
|
|
+ RC_INIT;
|
|
|
+ #else
|
|
|
+ RC_INIT(inStream, inSize);
|
|
|
+ #endif
|
|
|
+
|
|
|
+ #endif /* _LZMA_OUT_READ */
|
|
|
+
|
|
|
while(nowPos < outSize)
|
|
|
{
|
|
|
+ CProb *prob;
|
|
|
+ UInt32 bound;
|
|
|
int posState = (int)(
|
|
|
(nowPos
|
|
|
#ifdef _LZMA_OUT_READ
|
|
|
@@ -478,15 +281,13 @@ int LzmaDecode(
|
|
|
#endif
|
|
|
)
|
|
|
& posStateMask);
|
|
|
- #ifdef _LZMA_IN_CB
|
|
|
- if (rd.Result != LZMA_RESULT_OK)
|
|
|
- return rd.Result;
|
|
|
- #endif
|
|
|
- if (rd.ExtraBytes != 0)
|
|
|
- return LZMA_RESULT_DATA_ERROR;
|
|
|
- if (RangeDecoderBitDecode(p + IsMatch + (state << kNumPosBitsMax) + posState, &rd) == 0)
|
|
|
+
|
|
|
+ prob = p + IsMatch + (state << kNumPosBitsMax) + posState;
|
|
|
+ IfBit0(prob)
|
|
|
{
|
|
|
- CProb *probs = p + Literal + (LZMA_LIT_SIZE *
|
|
|
+ int symbol = 1;
|
|
|
+ UpdateBit0(prob)
|
|
|
+ prob = p + Literal + (LZMA_LIT_SIZE *
|
|
|
(((
|
|
|
(nowPos
|
|
|
#ifdef _LZMA_OUT_READ
|
|
|
@@ -495,12 +296,9 @@ int LzmaDecode(
|
|
|
)
|
|
|
& literalPosMask) << lc) + (previousByte >> (8 - lc))));
|
|
|
|
|
|
- if (state < 4) state = 0;
|
|
|
- else if (state < 10) state -= 3;
|
|
|
- else state -= 6;
|
|
|
- if (previousIsMatch)
|
|
|
+ if (state >= kNumLitStates)
|
|
|
{
|
|
|
- Byte matchByte;
|
|
|
+ int matchByte;
|
|
|
#ifdef _LZMA_OUT_READ
|
|
|
UInt32 pos = dictionaryPos - rep0;
|
|
|
if (pos >= dictionarySize)
|
|
|
@@ -509,39 +307,73 @@ int LzmaDecode(
|
|
|
#else
|
|
|
matchByte = outStream[nowPos - rep0];
|
|
|
#endif
|
|
|
- previousByte = LzmaLiteralDecodeMatch(probs, &rd, matchByte);
|
|
|
- previousIsMatch = 0;
|
|
|
+ do
|
|
|
+ {
|
|
|
+ int bit;
|
|
|
+ CProb *probLit;
|
|
|
+ matchByte <<= 1;
|
|
|
+ bit = (matchByte & 0x100);
|
|
|
+ probLit = prob + 0x100 + bit + symbol;
|
|
|
+ RC_GET_BIT2(probLit, symbol, if (bit != 0) break, if (bit == 0) break)
|
|
|
+ }
|
|
|
+ while (symbol < 0x100);
|
|
|
}
|
|
|
- else
|
|
|
- previousByte = LzmaLiteralDecode(probs, &rd);
|
|
|
+ while (symbol < 0x100)
|
|
|
+ {
|
|
|
+ CProb *probLit = prob + symbol;
|
|
|
+ RC_GET_BIT(probLit, symbol)
|
|
|
+ }
|
|
|
+ previousByte = (Byte)symbol;
|
|
|
+
|
|
|
outStream[nowPos++] = previousByte;
|
|
|
#ifdef _LZMA_OUT_READ
|
|
|
+ if (distanceLimit < dictionarySize)
|
|
|
+ distanceLimit++;
|
|
|
+
|
|
|
dictionary[dictionaryPos] = previousByte;
|
|
|
if (++dictionaryPos == dictionarySize)
|
|
|
dictionaryPos = 0;
|
|
|
#endif
|
|
|
+ if (state < 4) state = 0;
|
|
|
+ else if (state < 10) state -= 3;
|
|
|
+ else state -= 6;
|
|
|
}
|
|
|
else
|
|
|
{
|
|
|
- previousIsMatch = 1;
|
|
|
- if (RangeDecoderBitDecode(p + IsRep + state, &rd) == 1)
|
|
|
+ UpdateBit1(prob);
|
|
|
+ prob = p + IsRep + state;
|
|
|
+ IfBit0(prob)
|
|
|
+ {
|
|
|
+ UpdateBit0(prob);
|
|
|
+ rep3 = rep2;
|
|
|
+ rep2 = rep1;
|
|
|
+ rep1 = rep0;
|
|
|
+ state = state < kNumLitStates ? 0 : 3;
|
|
|
+ prob = p + LenCoder;
|
|
|
+ }
|
|
|
+ else
|
|
|
{
|
|
|
- if (RangeDecoderBitDecode(p + IsRepG0 + state, &rd) == 0)
|
|
|
+ UpdateBit1(prob);
|
|
|
+ prob = p + IsRepG0 + state;
|
|
|
+ IfBit0(prob)
|
|
|
{
|
|
|
- if (RangeDecoderBitDecode(p + IsRep0Long + (state << kNumPosBitsMax) + posState, &rd) == 0)
|
|
|
+ UpdateBit0(prob);
|
|
|
+ prob = p + IsRep0Long + (state << kNumPosBitsMax) + posState;
|
|
|
+ IfBit0(prob)
|
|
|
{
|
|
|
#ifdef _LZMA_OUT_READ
|
|
|
UInt32 pos;
|
|
|
#endif
|
|
|
- if (
|
|
|
- (nowPos
|
|
|
- #ifdef _LZMA_OUT_READ
|
|
|
- + globalPos
|
|
|
- #endif
|
|
|
- )
|
|
|
- == 0)
|
|
|
+ UpdateBit0(prob);
|
|
|
+
|
|
|
+ #ifdef _LZMA_OUT_READ
|
|
|
+ if (distanceLimit == 0)
|
|
|
+ #else
|
|
|
+ if (nowPos == 0)
|
|
|
+ #endif
|
|
|
return LZMA_RESULT_DATA_ERROR;
|
|
|
- state = state < 7 ? 9 : 11;
|
|
|
+
|
|
|
+ state = state < kNumLitStates ? 9 : 11;
|
|
|
#ifdef _LZMA_OUT_READ
|
|
|
pos = dictionaryPos - rep0;
|
|
|
if (pos >= dictionarySize)
|
|
|
@@ -554,20 +386,40 @@ int LzmaDecode(
|
|
|
previousByte = outStream[nowPos - rep0];
|
|
|
#endif
|
|
|
outStream[nowPos++] = previousByte;
|
|
|
+ #ifdef _LZMA_OUT_READ
|
|
|
+ if (distanceLimit < dictionarySize)
|
|
|
+ distanceLimit++;
|
|
|
+ #endif
|
|
|
+
|
|
|
continue;
|
|
|
}
|
|
|
+ else
|
|
|
+ {
|
|
|
+ UpdateBit1(prob);
|
|
|
+ }
|
|
|
}
|
|
|
else
|
|
|
{
|
|
|
UInt32 distance;
|
|
|
- if(RangeDecoderBitDecode(p + IsRepG1 + state, &rd) == 0)
|
|
|
+ UpdateBit1(prob);
|
|
|
+ prob = p + IsRepG1 + state;
|
|
|
+ IfBit0(prob)
|
|
|
+ {
|
|
|
+ UpdateBit0(prob);
|
|
|
distance = rep1;
|
|
|
+ }
|
|
|
else
|
|
|
{
|
|
|
- if(RangeDecoderBitDecode(p + IsRepG2 + state, &rd) == 0)
|
|
|
+ UpdateBit1(prob);
|
|
|
+ prob = p + IsRepG2 + state;
|
|
|
+ IfBit0(prob)
|
|
|
+ {
|
|
|
+ UpdateBit0(prob);
|
|
|
distance = rep2;
|
|
|
+ }
|
|
|
else
|
|
|
{
|
|
|
+ UpdateBit1(prob);
|
|
|
distance = rep3;
|
|
|
rep3 = rep2;
|
|
|
}
|
|
|
@@ -576,55 +428,115 @@ int LzmaDecode(
|
|
|
rep1 = rep0;
|
|
|
rep0 = distance;
|
|
|
}
|
|
|
- len = LzmaLenDecode(p + RepLenCoder, &rd, posState);
|
|
|
- state = state < 7 ? 8 : 11;
|
|
|
+ state = state < kNumLitStates ? 8 : 11;
|
|
|
+ prob = p + RepLenCoder;
|
|
|
}
|
|
|
- else
|
|
|
+ {
|
|
|
+ int numBits, offset;
|
|
|
+ CProb *probLen = prob + LenChoice;
|
|
|
+ IfBit0(probLen)
|
|
|
+ {
|
|
|
+ UpdateBit0(probLen);
|
|
|
+ probLen = prob + LenLow + (posState << kLenNumLowBits);
|
|
|
+ offset = 0;
|
|
|
+ numBits = kLenNumLowBits;
|
|
|
+ }
|
|
|
+ else
|
|
|
+ {
|
|
|
+ UpdateBit1(probLen);
|
|
|
+ probLen = prob + LenChoice2;
|
|
|
+ IfBit0(probLen)
|
|
|
+ {
|
|
|
+ UpdateBit0(probLen);
|
|
|
+ probLen = prob + LenMid + (posState << kLenNumMidBits);
|
|
|
+ offset = kLenNumLowSymbols;
|
|
|
+ numBits = kLenNumMidBits;
|
|
|
+ }
|
|
|
+ else
|
|
|
+ {
|
|
|
+ UpdateBit1(probLen);
|
|
|
+ probLen = prob + LenHigh;
|
|
|
+ offset = kLenNumLowSymbols + kLenNumMidSymbols;
|
|
|
+ numBits = kLenNumHighBits;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ RangeDecoderBitTreeDecode(probLen, numBits, len);
|
|
|
+ len += offset;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (state < 4)
|
|
|
{
|
|
|
int posSlot;
|
|
|
- rep3 = rep2;
|
|
|
- rep2 = rep1;
|
|
|
- rep1 = rep0;
|
|
|
- state = state < 7 ? 7 : 10;
|
|
|
- len = LzmaLenDecode(p + LenCoder, &rd, posState);
|
|
|
- posSlot = RangeDecoderBitTreeDecode(p + PosSlot +
|
|
|
+ state += kNumLitStates;
|
|
|
+ prob = p + PosSlot +
|
|
|
((len < kNumLenToPosStates ? len : kNumLenToPosStates - 1) <<
|
|
|
- kNumPosSlotBits), kNumPosSlotBits, &rd);
|
|
|
+ kNumPosSlotBits);
|
|
|
+ RangeDecoderBitTreeDecode(prob, kNumPosSlotBits, posSlot);
|
|
|
if (posSlot >= kStartPosModelIndex)
|
|
|
{
|
|
|
int numDirectBits = ((posSlot >> 1) - 1);
|
|
|
- rep0 = ((2 | ((UInt32)posSlot & 1)) << numDirectBits);
|
|
|
+ rep0 = (2 | ((UInt32)posSlot & 1));
|
|
|
if (posSlot < kEndPosModelIndex)
|
|
|
{
|
|
|
- rep0 += RangeDecoderReverseBitTreeDecode(
|
|
|
- p + SpecPos + rep0 - posSlot - 1, numDirectBits, &rd);
|
|
|
+ rep0 <<= numDirectBits;
|
|
|
+ prob = p + SpecPos + rep0 - posSlot - 1;
|
|
|
}
|
|
|
else
|
|
|
{
|
|
|
- rep0 += RangeDecoderDecodeDirectBits(&rd,
|
|
|
- numDirectBits - kNumAlignBits) << kNumAlignBits;
|
|
|
- rep0 += RangeDecoderReverseBitTreeDecode(p + Align, kNumAlignBits, &rd);
|
|
|
+ numDirectBits -= kNumAlignBits;
|
|
|
+ do
|
|
|
+ {
|
|
|
+ RC_NORMALIZE
|
|
|
+ Range >>= 1;
|
|
|
+ rep0 <<= 1;
|
|
|
+ if (Code >= Range)
|
|
|
+ {
|
|
|
+ Code -= Range;
|
|
|
+ rep0 |= 1;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ while (--numDirectBits != 0);
|
|
|
+ prob = p + Align;
|
|
|
+ rep0 <<= kNumAlignBits;
|
|
|
+ numDirectBits = kNumAlignBits;
|
|
|
+ }
|
|
|
+ {
|
|
|
+ int i = 1;
|
|
|
+ int mi = 1;
|
|
|
+ do
|
|
|
+ {
|
|
|
+ CProb *prob3 = prob + mi;
|
|
|
+ RC_GET_BIT2(prob3, mi, ; , rep0 |= i);
|
|
|
+ i <<= 1;
|
|
|
+ }
|
|
|
+ while(--numDirectBits != 0);
|
|
|
}
|
|
|
}
|
|
|
else
|
|
|
rep0 = posSlot;
|
|
|
- rep0++;
|
|
|
- }
|
|
|
- if (rep0 == (UInt32)(0))
|
|
|
- {
|
|
|
- /* it's for stream version */
|
|
|
- len = -1;
|
|
|
- break;
|
|
|
- }
|
|
|
- if (rep0 > nowPos
|
|
|
- #ifdef _LZMA_OUT_READ
|
|
|
- + globalPos
|
|
|
- #endif
|
|
|
- )
|
|
|
- {
|
|
|
- return LZMA_RESULT_DATA_ERROR;
|
|
|
+ if (++rep0 == (UInt32)(0))
|
|
|
+ {
|
|
|
+ /* it's for stream version */
|
|
|
+ len = kLzmaStreamWasFinishedId;
|
|
|
+ break;
|
|
|
+ }
|
|
|
}
|
|
|
+
|
|
|
len += kMatchMinLen;
|
|
|
+ #ifdef _LZMA_OUT_READ
|
|
|
+ if (rep0 > distanceLimit)
|
|
|
+ #else
|
|
|
+ if (rep0 > nowPos)
|
|
|
+ #endif
|
|
|
+ return LZMA_RESULT_DATA_ERROR;
|
|
|
+
|
|
|
+ #ifdef _LZMA_OUT_READ
|
|
|
+ if (dictionarySize - distanceLimit > (UInt32)len)
|
|
|
+ distanceLimit += len;
|
|
|
+ else
|
|
|
+ distanceLimit = dictionarySize;
|
|
|
+ #endif
|
|
|
+
|
|
|
do
|
|
|
{
|
|
|
#ifdef _LZMA_OUT_READ
|
|
|
@@ -638,26 +550,35 @@ int LzmaDecode(
|
|
|
#else
|
|
|
previousByte = outStream[nowPos - rep0];
|
|
|
#endif
|
|
|
- outStream[nowPos++] = previousByte;
|
|
|
len--;
|
|
|
+ outStream[nowPos++] = previousByte;
|
|
|
}
|
|
|
- while(len > 0 && nowPos < outSize);
|
|
|
+ while(len != 0 && nowPos < outSize);
|
|
|
}
|
|
|
}
|
|
|
+ RC_NORMALIZE;
|
|
|
|
|
|
#ifdef _LZMA_OUT_READ
|
|
|
- vs->RangeDecoder = rd;
|
|
|
+ vs->Range = Range;
|
|
|
+ vs->Code = Code;
|
|
|
vs->DictionaryPos = dictionaryPos;
|
|
|
- vs->GlobalPos = globalPos + nowPos;
|
|
|
+ vs->GlobalPos = globalPos + (UInt32)nowPos;
|
|
|
+ vs->DistanceLimit = distanceLimit;
|
|
|
vs->Reps[0] = rep0;
|
|
|
vs->Reps[1] = rep1;
|
|
|
vs->Reps[2] = rep2;
|
|
|
vs->Reps[3] = rep3;
|
|
|
vs->State = state;
|
|
|
- vs->PreviousIsMatch = previousIsMatch;
|
|
|
vs->RemainLen = len;
|
|
|
+ vs->TempDictionary[0] = tempDictionary[0];
|
|
|
#endif
|
|
|
|
|
|
+ #ifdef _LZMA_IN_CB
|
|
|
+ vs->Buffer = Buffer;
|
|
|
+ vs->BufferLim = BufferLim;
|
|
|
+ #else
|
|
|
+ *inSizeProcessed = (SizeT)(Buffer - inStream);
|
|
|
+ #endif
|
|
|
*outSizeProcessed = nowPos;
|
|
|
return LZMA_RESULT_OK;
|
|
|
}
|