| /* |
| LzmaDecode.c |
| LZMA Decoder (optimized for Speed version) |
| |
| 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: |
| 1) GNU Lesser General Public License (GNU LGPL) |
| 2) Common Public License (CPL) |
| It means that you can select one of these two licenses and |
| 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 |
| terms of the CPL or GNU LGPL. Any modifications or additions |
| to this file, however, are subject to the LGPL or CPL terms. |
| */ |
| |
| #include <stdint.h> |
| |
| #include "base/lzma/priv.h" |
| |
| #define kNumTopBits 24 |
| #define kTopValue ((uint32_t)1 << kNumTopBits) |
| |
| #define kNumBitModelTotalBits 11 |
| #define kBitModelTotal (1 << kNumBitModelTotalBits) |
| #define kNumMoveBits 5 |
| |
| #define RC_READ_BYTE (*Buffer++) |
| |
| #define RC_INIT2 \ |
| Code = 0; \ |
| Range = 0xFFFFFFFF; \ |
| { \ |
| int i; \ |
| for (i = 0; i < 5; i++) { \ |
| RC_TEST; \ |
| Code = (Code << 8) | RC_READ_BYTE; \ |
| } \ |
| } |
| |
| |
| #define RC_TEST { if (Buffer == BufferLim) return LZMA_RESULT_DATA_ERROR; } |
| |
| #define RC_INIT(buffer, bufferSize) \ |
| Buffer = buffer; \ |
| BufferLim = buffer + bufferSize; \ |
| RC_INIT2 |
| |
| |
| #define RC_NORMALIZE \ |
| if (Range < kTopValue) { \ |
| RC_TEST; \ |
| Range <<= 8; \ |
| Code = (Code << 8) | RC_READ_BYTE; \ |
| } |
| |
| #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; |
| |
| #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, ; , ;) |
| |
| #define RangeDecoderBitTreeDecode(probs, numLevels, res) \ |
| { \ |
| int i = numLevels; \ |
| res = 1; \ |
| do { \ |
| CProb *cp = probs + res; \ |
| RC_GET_BIT(cp, res) \ |
| } while(--i != 0); \ |
| res -= (1 << numLevels); \ |
| } |
| |
| |
| #define kNumPosBitsMax 4 |
| #define kNumPosStatesMax (1 << kNumPosBitsMax) |
| |
| #define kLenNumLowBits 3 |
| #define kLenNumLowSymbols (1 << kLenNumLowBits) |
| #define kLenNumMidBits 3 |
| #define kLenNumMidSymbols (1 << kLenNumMidBits) |
| #define kLenNumHighBits 8 |
| #define kLenNumHighSymbols (1 << kLenNumHighBits) |
| |
| #define LenChoice 0 |
| #define LenChoice2 (LenChoice + 1) |
| #define LenLow (LenChoice2 + 1) |
| #define LenMid (LenLow + (kNumPosStatesMax << kLenNumLowBits)) |
| #define LenHigh (LenMid + (kNumPosStatesMax << kLenNumMidBits)) |
| #define kNumLenProbs (LenHigh + kLenNumHighSymbols) |
| |
| |
| #define kNumStates 12 |
| #define kNumLitStates 7 |
| |
| #define kStartPosModelIndex 4 |
| #define kEndPosModelIndex 14 |
| #define kNumFullDistances (1 << (kEndPosModelIndex >> 1)) |
| |
| #define kNumPosSlotBits 6 |
| #define kNumLenToPosStates 4 |
| |
| #define kNumAlignBits 4 |
| #define kAlignTableSize (1 << kNumAlignBits) |
| |
| #define kMatchMinLen 2 |
| |
| #define IsMatch 0 |
| #define IsRep (IsMatch + (kNumStates << kNumPosBitsMax)) |
| #define IsRepG0 (IsRep + kNumStates) |
| #define IsRepG1 (IsRepG0 + kNumStates) |
| #define IsRepG2 (IsRepG1 + kNumStates) |
| #define IsRep0Long (IsRepG2 + kNumStates) |
| #define PosSlot (IsRep0Long + (kNumStates << kNumPosBitsMax)) |
| #define SpecPos (PosSlot + (kNumLenToPosStates << kNumPosSlotBits)) |
| #define Align (SpecPos + kNumFullDistances - kEndPosModelIndex) |
| #define LenCoder (Align + kAlignTableSize) |
| #define RepLenCoder (LenCoder + kNumLenProbs) |
| #define Literal (RepLenCoder + kNumLenProbs) |
| |
| #if Literal != LZMA_BASE_SIZE |
| StopCompilingDueBUG |
| #endif |
| |
| int LzmaDecodeProperties(CLzmaProperties *propsRes, const uint8_t *propsData, |
| int size) |
| { |
| uint8_t 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; |
| } |
| |
| return LZMA_RESULT_OK; |
| } |
| |
| #define kLzmaStreamWasFinishedId (-1) |
| |
| int LzmaDecode(CLzmaDecoderState *vs, |
| const uint8_t *inStream, size_t inSize, size_t *inSizeProcessed, |
| uint8_t *outStream, size_t outSize, size_t *outSizeProcessed) |
| { |
| CProb *p = vs->Probs; |
| size_t nowPos = 0; |
| uint8_t previousByte = 0; |
| uint32_t posStateMask = (1 << (vs->Properties.pb)) - 1; |
| uint32_t literalPosMask = (1 << (vs->Properties.lp)) - 1; |
| int lc = vs->Properties.lc; |
| |
| |
| int state = 0; |
| uint32_t rep0 = 1, rep1 = 1, rep2 = 1, rep3 = 1; |
| int len = 0; |
| const uint8_t *Buffer; |
| const uint8_t *BufferLim; |
| uint32_t Range; |
| uint32_t Code; |
| |
| *inSizeProcessed = 0; |
| *outSizeProcessed = 0; |
| |
| { |
| uint32_t i; |
| uint32_t numProbs = Literal + |
| ((uint32_t)LZMA_LIT_SIZE << (lc + vs->Properties.lp)); |
| for (i = 0; i < numProbs; i++) |
| p[i] = kBitModelTotal >> 1; |
| } |
| |
| RC_INIT(inStream, inSize); |
| |
| |
| while (nowPos < outSize) { |
| CProb *prob; |
| uint32_t bound; |
| int posState = (int)(nowPos & posStateMask); |
| |
| prob = p + IsMatch + (state << kNumPosBitsMax) + posState; |
| IfBit0(prob) { |
| int symbol = 1; |
| UpdateBit0(prob) |
| prob = p + Literal + |
| (LZMA_LIT_SIZE * |
| (((nowPos & literalPosMask) << lc) + |
| (previousByte >> (8 - lc)))); |
| |
| if (state >= kNumLitStates) { |
| int matchByte; |
| matchByte = outStream[nowPos - rep0]; |
| 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); |
| } |
| while (symbol < 0x100) { |
| CProb *probLit = prob + symbol; |
| RC_GET_BIT(probLit, symbol) |
| } |
| previousByte = (uint8_t)symbol; |
| |
| outStream[nowPos++] = previousByte; |
| if (state < 4) |
| state = 0; |
| else if (state < 10) |
| state -= 3; |
| else |
| state -= 6; |
| } else { |
| 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 { |
| UpdateBit1(prob); |
| prob = p + IsRepG0 + state; |
| IfBit0(prob) { |
| UpdateBit0(prob); |
| prob = p + IsRep0Long + |
| (state << kNumPosBitsMax) + |
| posState; |
| IfBit0(prob) { |
| UpdateBit0(prob); |
| |
| if (nowPos == 0) |
| return LZMA_RESULT_DATA_ERROR; |
| |
| state = state < kNumLitStates ? |
| 9 : 11; |
| previousByte = outStream[nowPos - rep0]; |
| outStream[nowPos++] = |
| previousByte; |
| |
| continue; |
| } else { |
| UpdateBit1(prob); |
| } |
| } else { |
| uint32_t distance; |
| UpdateBit1(prob); |
| prob = p + IsRepG1 + state; |
| IfBit0(prob) { |
| UpdateBit0(prob); |
| distance = rep1; |
| } else { |
| UpdateBit1(prob); |
| prob = p + IsRepG2 + state; |
| IfBit0(prob) { |
| UpdateBit0(prob); |
| distance = rep2; |
| } else { |
| UpdateBit1(prob); |
| distance = rep3; |
| rep3 = rep2; |
| } |
| rep2 = rep1; |
| } |
| rep1 = rep0; |
| rep0 = distance; |
| } |
| state = state < kNumLitStates ? 8 : 11; |
| prob = p + RepLenCoder; |
| } |
| { |
| 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; |
| state += kNumLitStates; |
| prob = p + PosSlot + |
| ((len < kNumLenToPosStates ? |
| len : kNumLenToPosStates - 1) << |
| kNumPosSlotBits); |
| RangeDecoderBitTreeDecode( |
| prob, kNumPosSlotBits, posSlot); |
| if (posSlot >= kStartPosModelIndex) { |
| int numDirectBits = |
| ((posSlot >> 1) - 1); |
| rep0 = (2 | ((uint32_t)posSlot & 1)); |
| if (posSlot < kEndPosModelIndex) { |
| rep0 <<= numDirectBits; |
| prob = p + SpecPos + rep0 - |
| posSlot - 1; |
| } else { |
| 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; |
| if (++rep0 == (uint32_t)(0)) { |
| /* it's for stream version */ |
| len = kLzmaStreamWasFinishedId; |
| break; |
| } |
| } |
| |
| len += kMatchMinLen; |
| if (rep0 > nowPos) |
| return LZMA_RESULT_DATA_ERROR; |
| |
| |
| do { |
| previousByte = outStream[nowPos - rep0]; |
| len--; |
| outStream[nowPos++] = previousByte; |
| } while(len != 0 && nowPos < outSize); |
| } |
| } |
| RC_NORMALIZE; |
| |
| |
| *inSizeProcessed = (size_t)(Buffer - inStream); |
| *outSizeProcessed = nowPos; |
| return LZMA_RESULT_OK; |
| } |