1baa3858d3f5d128a5c8466b700098109edcad5f2repo sync// Lzma2Decoder.cpp 2baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 3baa3858d3f5d128a5c8466b700098109edcad5f2repo sync#include "StdAfx.h" 4baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 5baa3858d3f5d128a5c8466b700098109edcad5f2repo sync#include "../../../C/Alloc.h" 6baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 7baa3858d3f5d128a5c8466b700098109edcad5f2repo sync#include "../Common/StreamUtils.h" 8baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 9baa3858d3f5d128a5c8466b700098109edcad5f2repo sync#include "Lzma2Decoder.h" 10baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 11baa3858d3f5d128a5c8466b700098109edcad5f2repo syncstatic HRESULT SResToHRESULT(SRes res) 12baa3858d3f5d128a5c8466b700098109edcad5f2repo sync{ 13baa3858d3f5d128a5c8466b700098109edcad5f2repo sync switch(res) 14baa3858d3f5d128a5c8466b700098109edcad5f2repo sync { 15baa3858d3f5d128a5c8466b700098109edcad5f2repo sync case SZ_OK: return S_OK; 16baa3858d3f5d128a5c8466b700098109edcad5f2repo sync case SZ_ERROR_MEM: return E_OUTOFMEMORY; 17baa3858d3f5d128a5c8466b700098109edcad5f2repo sync case SZ_ERROR_PARAM: return E_INVALIDARG; 18baa3858d3f5d128a5c8466b700098109edcad5f2repo sync // case SZ_ERROR_PROGRESS: return E_ABORT; 19baa3858d3f5d128a5c8466b700098109edcad5f2repo sync case SZ_ERROR_DATA: return S_FALSE; 20baa3858d3f5d128a5c8466b700098109edcad5f2repo sync } 21baa3858d3f5d128a5c8466b700098109edcad5f2repo sync return E_FAIL; 22baa3858d3f5d128a5c8466b700098109edcad5f2repo sync} 23baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 24baa3858d3f5d128a5c8466b700098109edcad5f2repo syncnamespace NCompress { 25baa3858d3f5d128a5c8466b700098109edcad5f2repo syncnamespace NLzma2 { 26baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 27baa3858d3f5d128a5c8466b700098109edcad5f2repo syncstatic const UInt32 kInBufSize = 1 << 20; 28baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 29baa3858d3f5d128a5c8466b700098109edcad5f2repo syncCDecoder::CDecoder(): _inBuf(0), _outSizeDefined(false) 30baa3858d3f5d128a5c8466b700098109edcad5f2repo sync{ 31baa3858d3f5d128a5c8466b700098109edcad5f2repo sync Lzma2Dec_Construct(&_state); 32baa3858d3f5d128a5c8466b700098109edcad5f2repo sync} 33baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 34baa3858d3f5d128a5c8466b700098109edcad5f2repo syncstatic void *SzAlloc(void *p, size_t size) { p = p; return MyAlloc(size); } 35baa3858d3f5d128a5c8466b700098109edcad5f2repo syncstatic void SzFree(void *p, void *address) { p = p; MyFree(address); } 36baa3858d3f5d128a5c8466b700098109edcad5f2repo syncstatic ISzAlloc g_Alloc = { SzAlloc, SzFree }; 37baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 38baa3858d3f5d128a5c8466b700098109edcad5f2repo syncCDecoder::~CDecoder() 39baa3858d3f5d128a5c8466b700098109edcad5f2repo sync{ 40baa3858d3f5d128a5c8466b700098109edcad5f2repo sync Lzma2Dec_Free(&_state, &g_Alloc); 41baa3858d3f5d128a5c8466b700098109edcad5f2repo sync MyFree(_inBuf); 42baa3858d3f5d128a5c8466b700098109edcad5f2repo sync} 43baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 44baa3858d3f5d128a5c8466b700098109edcad5f2repo syncSTDMETHODIMP CDecoder::SetDecoderProperties2(const Byte *prop, UInt32 size) 45baa3858d3f5d128a5c8466b700098109edcad5f2repo sync{ 46baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (size != 1) return SZ_ERROR_UNSUPPORTED; 47baa3858d3f5d128a5c8466b700098109edcad5f2repo sync RINOK(SResToHRESULT(Lzma2Dec_Allocate(&_state, prop[0], &g_Alloc))); 48baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (_inBuf == 0) 49baa3858d3f5d128a5c8466b700098109edcad5f2repo sync { 50baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _inBuf = (Byte *)MyAlloc(kInBufSize); 51baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (_inBuf == 0) 52baa3858d3f5d128a5c8466b700098109edcad5f2repo sync return E_OUTOFMEMORY; 53baa3858d3f5d128a5c8466b700098109edcad5f2repo sync } 54baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 55baa3858d3f5d128a5c8466b700098109edcad5f2repo sync return S_OK; 56baa3858d3f5d128a5c8466b700098109edcad5f2repo sync} 57baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 58baa3858d3f5d128a5c8466b700098109edcad5f2repo syncSTDMETHODIMP CDecoder::GetInStreamProcessedSize(UInt64 *value) { *value = _inSizeProcessed; return S_OK; } 59baa3858d3f5d128a5c8466b700098109edcad5f2repo syncSTDMETHODIMP CDecoder::SetInStream(ISequentialInStream *inStream) { _inStream = inStream; return S_OK; } 60baa3858d3f5d128a5c8466b700098109edcad5f2repo syncSTDMETHODIMP CDecoder::ReleaseInStream() { _inStream.Release(); return S_OK; } 61baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 62baa3858d3f5d128a5c8466b700098109edcad5f2repo syncSTDMETHODIMP CDecoder::SetOutStreamSize(const UInt64 *outSize) 63baa3858d3f5d128a5c8466b700098109edcad5f2repo sync{ 64baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _outSizeDefined = (outSize != NULL); 65baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (_outSizeDefined) 66baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _outSize = *outSize; 67baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 68baa3858d3f5d128a5c8466b700098109edcad5f2repo sync Lzma2Dec_Init(&_state); 69baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 70baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _inPos = _inSize = 0; 71baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _inSizeProcessed = _outSizeProcessed = 0; 72baa3858d3f5d128a5c8466b700098109edcad5f2repo sync return S_OK; 73baa3858d3f5d128a5c8466b700098109edcad5f2repo sync} 74baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 75baa3858d3f5d128a5c8466b700098109edcad5f2repo syncSTDMETHODIMP CDecoder::Code(ISequentialInStream *inStream, 76baa3858d3f5d128a5c8466b700098109edcad5f2repo sync ISequentialOutStream *outStream, const UInt64 * /* inSize */, 77baa3858d3f5d128a5c8466b700098109edcad5f2repo sync const UInt64 *outSize, ICompressProgressInfo *progress) 78baa3858d3f5d128a5c8466b700098109edcad5f2repo sync{ 79baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (_inBuf == 0) 80baa3858d3f5d128a5c8466b700098109edcad5f2repo sync return S_FALSE; 81baa3858d3f5d128a5c8466b700098109edcad5f2repo sync SetOutStreamSize(outSize); 82baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 83baa3858d3f5d128a5c8466b700098109edcad5f2repo sync for (;;) 84baa3858d3f5d128a5c8466b700098109edcad5f2repo sync { 85baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (_inPos == _inSize) 86baa3858d3f5d128a5c8466b700098109edcad5f2repo sync { 87baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _inPos = _inSize = 0; 88baa3858d3f5d128a5c8466b700098109edcad5f2repo sync RINOK(inStream->Read(_inBuf, kInBufSize, &_inSize)); 89baa3858d3f5d128a5c8466b700098109edcad5f2repo sync } 90baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 91baa3858d3f5d128a5c8466b700098109edcad5f2repo sync SizeT dicPos = _state.decoder.dicPos; 92baa3858d3f5d128a5c8466b700098109edcad5f2repo sync SizeT curSize = _state.decoder.dicBufSize - dicPos; 93baa3858d3f5d128a5c8466b700098109edcad5f2repo sync const UInt32 kStepSize = ((UInt32)1 << 22); 94baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (curSize > kStepSize) 95baa3858d3f5d128a5c8466b700098109edcad5f2repo sync curSize = (SizeT)kStepSize; 96baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 97baa3858d3f5d128a5c8466b700098109edcad5f2repo sync ELzmaFinishMode finishMode = LZMA_FINISH_ANY; 98baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (_outSizeDefined) 99baa3858d3f5d128a5c8466b700098109edcad5f2repo sync { 100baa3858d3f5d128a5c8466b700098109edcad5f2repo sync const UInt64 rem = _outSize - _outSizeProcessed; 101baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (rem < curSize) 102baa3858d3f5d128a5c8466b700098109edcad5f2repo sync { 103baa3858d3f5d128a5c8466b700098109edcad5f2repo sync curSize = (SizeT)rem; 104baa3858d3f5d128a5c8466b700098109edcad5f2repo sync /* 105baa3858d3f5d128a5c8466b700098109edcad5f2repo sync // finishMode = LZMA_FINISH_END; 106baa3858d3f5d128a5c8466b700098109edcad5f2repo sync we can't use LZMA_FINISH_END here to allow partial decoding 107baa3858d3f5d128a5c8466b700098109edcad5f2repo sync */ 108baa3858d3f5d128a5c8466b700098109edcad5f2repo sync } 109baa3858d3f5d128a5c8466b700098109edcad5f2repo sync } 110baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 111baa3858d3f5d128a5c8466b700098109edcad5f2repo sync SizeT inSizeProcessed = _inSize - _inPos; 112baa3858d3f5d128a5c8466b700098109edcad5f2repo sync ELzmaStatus status; 113baa3858d3f5d128a5c8466b700098109edcad5f2repo sync SRes res = Lzma2Dec_DecodeToDic(&_state, dicPos + curSize, _inBuf + _inPos, &inSizeProcessed, finishMode, &status); 114baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 115baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _inPos += (UInt32)inSizeProcessed; 116baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _inSizeProcessed += inSizeProcessed; 117baa3858d3f5d128a5c8466b700098109edcad5f2repo sync SizeT outSizeProcessed = _state.decoder.dicPos - dicPos; 118baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _outSizeProcessed += outSizeProcessed; 119baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 120baa3858d3f5d128a5c8466b700098109edcad5f2repo sync bool finished = (inSizeProcessed == 0 && outSizeProcessed == 0); 121baa3858d3f5d128a5c8466b700098109edcad5f2repo sync bool stopDecoding = (_outSizeDefined && _outSizeProcessed >= _outSize); 122baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 123baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (res != 0 || _state.decoder.dicPos == _state.decoder.dicBufSize || finished || stopDecoding) 124baa3858d3f5d128a5c8466b700098109edcad5f2repo sync { 125baa3858d3f5d128a5c8466b700098109edcad5f2repo sync HRESULT res2 = WriteStream(outStream, _state.decoder.dic, _state.decoder.dicPos); 126baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (res != 0) 127baa3858d3f5d128a5c8466b700098109edcad5f2repo sync return S_FALSE; 128baa3858d3f5d128a5c8466b700098109edcad5f2repo sync RINOK(res2); 129baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (stopDecoding) 130baa3858d3f5d128a5c8466b700098109edcad5f2repo sync return S_OK; 131baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (finished) 132baa3858d3f5d128a5c8466b700098109edcad5f2repo sync return (status == LZMA_STATUS_FINISHED_WITH_MARK ? S_OK : S_FALSE); 133baa3858d3f5d128a5c8466b700098109edcad5f2repo sync } 134baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (_state.decoder.dicPos == _state.decoder.dicBufSize) 135baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _state.decoder.dicPos = 0; 136baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 137baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (progress != NULL) 138baa3858d3f5d128a5c8466b700098109edcad5f2repo sync { 139baa3858d3f5d128a5c8466b700098109edcad5f2repo sync RINOK(progress->SetRatioInfo(&_inSizeProcessed, &_outSizeProcessed)); 140baa3858d3f5d128a5c8466b700098109edcad5f2repo sync } 141baa3858d3f5d128a5c8466b700098109edcad5f2repo sync } 142baa3858d3f5d128a5c8466b700098109edcad5f2repo sync} 143baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 144baa3858d3f5d128a5c8466b700098109edcad5f2repo sync#ifndef NO_READ_FROM_CODER 145baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 146baa3858d3f5d128a5c8466b700098109edcad5f2repo syncSTDMETHODIMP CDecoder::Read(void *data, UInt32 size, UInt32 *processedSize) 147baa3858d3f5d128a5c8466b700098109edcad5f2repo sync{ 148baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (processedSize) 149baa3858d3f5d128a5c8466b700098109edcad5f2repo sync *processedSize = 0; 150baa3858d3f5d128a5c8466b700098109edcad5f2repo sync do 151baa3858d3f5d128a5c8466b700098109edcad5f2repo sync { 152baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (_inPos == _inSize) 153baa3858d3f5d128a5c8466b700098109edcad5f2repo sync { 154baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _inPos = _inSize = 0; 155baa3858d3f5d128a5c8466b700098109edcad5f2repo sync RINOK(_inStream->Read(_inBuf, kInBufSize, &_inSize)); 156baa3858d3f5d128a5c8466b700098109edcad5f2repo sync } 157baa3858d3f5d128a5c8466b700098109edcad5f2repo sync { 158baa3858d3f5d128a5c8466b700098109edcad5f2repo sync SizeT inProcessed = _inSize - _inPos; 159baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 160baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (_outSizeDefined) 161baa3858d3f5d128a5c8466b700098109edcad5f2repo sync { 162baa3858d3f5d128a5c8466b700098109edcad5f2repo sync const UInt64 rem = _outSize - _outSizeProcessed; 163baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (rem < size) 164baa3858d3f5d128a5c8466b700098109edcad5f2repo sync size = (UInt32)rem; 165baa3858d3f5d128a5c8466b700098109edcad5f2repo sync } 166baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 167baa3858d3f5d128a5c8466b700098109edcad5f2repo sync SizeT outProcessed = size; 168baa3858d3f5d128a5c8466b700098109edcad5f2repo sync ELzmaStatus status; 169baa3858d3f5d128a5c8466b700098109edcad5f2repo sync SRes res = Lzma2Dec_DecodeToBuf(&_state, (Byte *)data, &outProcessed, 170baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _inBuf + _inPos, &inProcessed, LZMA_FINISH_ANY, &status); 171baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _inPos += (UInt32)inProcessed; 172baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _inSizeProcessed += inProcessed; 173baa3858d3f5d128a5c8466b700098109edcad5f2repo sync _outSizeProcessed += outProcessed; 174baa3858d3f5d128a5c8466b700098109edcad5f2repo sync size -= (UInt32)outProcessed; 175baa3858d3f5d128a5c8466b700098109edcad5f2repo sync data = (Byte *)data + outProcessed; 176baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (processedSize) 177baa3858d3f5d128a5c8466b700098109edcad5f2repo sync *processedSize += (UInt32)outProcessed; 178baa3858d3f5d128a5c8466b700098109edcad5f2repo sync RINOK(SResToHRESULT(res)); 179baa3858d3f5d128a5c8466b700098109edcad5f2repo sync if (inProcessed == 0 && outProcessed == 0) 180baa3858d3f5d128a5c8466b700098109edcad5f2repo sync return S_OK; 181baa3858d3f5d128a5c8466b700098109edcad5f2repo sync } 182baa3858d3f5d128a5c8466b700098109edcad5f2repo sync } 183baa3858d3f5d128a5c8466b700098109edcad5f2repo sync while (size != 0); 184baa3858d3f5d128a5c8466b700098109edcad5f2repo sync return S_OK; 185baa3858d3f5d128a5c8466b700098109edcad5f2repo sync} 186baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 187baa3858d3f5d128a5c8466b700098109edcad5f2repo sync#endif 188baa3858d3f5d128a5c8466b700098109edcad5f2repo sync 189baa3858d3f5d128a5c8466b700098109edcad5f2repo sync}} 190