7zip — C ++ LZMA SDK: функция распаковки сжатого файла LZMA2

Я пытаюсь создать функцию, которая распаковывает LZAM2 сжатые данные. Я вдохновил себя от этот урок который прекрасно работает для LZMA и я попытался адаптировать его для LZMA2, Я успешно создал функцию сжатия для LZMA2, но у меня нет успеха для несжатого.

Вот функция сжатия:

static void Compress2Inc(std::vector<unsigned char> &outBuf,
const std::vector<unsigned char> &inBuf)
{
CLzma2EncHandle enc = Lzma2Enc_Create(&SzAllocForLzma, &SzAllocForLzma2);
assert(enc);

CLzma2EncProps props;
Lzma2EncProps_Init(&props);
props.lzmaProps.writeEndMark = 1; // 0 or 1

SRes res = Lzma2Enc_SetProps(enc, &props);
assert(res == SZ_OK);

unsigned propsSize = LZMA_PROPS_SIZE;
outBuf.resize(propsSize);

res = Lzma2Enc_WriteProperties(enc);
//cout << res;
//assert(res == SZ_OK && propsSize == LZMA_PROPS_SIZE);

VectorInStream inStream = { &VectorInStream_Read, &inBuf, 0 };
VectorOutStream outStream = { &VectorOutStream_Write, &outBuf };

res = Lzma2Enc_Encode(enc,
(ISeqOutStream*)&outStream, (ISeqInStream*)&inStream,
0);
assert(res == SZ_OK);

Lzma2Enc_Destroy(enc);
}

Куда:

static void *AllocForLzma2(void *, size_t size) { return BigAlloc(size); }
static void FreeForLzma2(void *, void *address) { BigFree(address); }
static ISzAlloc SzAllocForLzma2 = { AllocForLzma2, FreeForLzma2 };

static void *AllocForLzma(void *, size_t size) { return MyAlloc(size); }
static void FreeForLzma(void *, void *address) { MyFree(address); }
static ISzAlloc SzAllocForLzma = { AllocForLzma, FreeForLzma };

typedef struct
{
ISeqInStream SeqInStream;
const std::vector<unsigned char> *Buf;
unsigned BufPos;
} VectorInStream;

SRes VectorInStream_Read(void *p, void *buf, size_t *size)
{
VectorInStream *ctx = (VectorInStream*)p;
*size = min(*size, ctx->Buf->size() - ctx->BufPos);
if (*size)
memcpy(buf, &(*ctx->Buf)[ctx->BufPos], *size);
ctx->BufPos += *size;
return SZ_OK;
}

typedef struct
{
ISeqOutStream SeqOutStream;
std::vector<unsigned char> *Buf;
} VectorOutStream;

size_t VectorOutStream_Write(void *p, const void *buf, size_t size)
{
VectorOutStream *ctx = (VectorOutStream*)p;
if (size)
{
unsigned oldSize = ctx->Buf->size();
ctx->Buf->resize(oldSize + size);
memcpy(&(*ctx->Buf)[oldSize], buf, size);
}
return size;
}

Вот то, что я имею до сих пор с функцией распаковки, но Lzma2Dec_DecodeToBuf функция возвращает код ошибки 1(SZ_ERROR_DATA) и я просто не мог найти что-либо в Интернете по этому поводу в Интернете.

static void Uncompress2Inc(std::vector<unsigned char> &outBuf,
const std::vector<unsigned char> &inBuf)
{
CLzma2Dec dec;
Lzma2Dec_Construct(&dec);

SRes res = Lzma2Dec_Allocate(&dec, outBuf.size(), &SzAllocForLzma);
assert(res == SZ_OK);

Lzma2Dec_Init(&dec);

outBuf.resize(UNCOMPRESSED_SIZE);
unsigned outPos = 0, inPos = LZMA_PROPS_SIZE;
ELzmaStatus status;
const unsigned BUF_SIZE = 10240;
while (outPos < outBuf.size())
{
unsigned destLen = min(BUF_SIZE, outBuf.size() - outPos);
unsigned srcLen  = min(BUF_SIZE, inBuf.size() - inPos);
unsigned srcLenOld = srcLen, destLenOld = destLen;

res = Lzma2Dec_DecodeToBuf(&dec,
&outBuf[outPos], &destLen,
&inBuf[inPos], &srcLen,
(outPos + destLen == outBuf.size()) ? LZMA_FINISH_END : LZMA_FINISH_ANY,
&status);

assert(res == SZ_OK);
inPos += srcLen;
outPos += destLen;
if (status == LZMA_STATUS_FINISHED_WITH_MARK)
break;
}

Lzma2Dec_Free(&dec, &SzAllocForLzma);
outBuf.resize(outPos);
}

я использую Visual Studio 2008 а также LZMA SDK скачано с Вот. Кто то Вот была точно такая же проблема, но я не мог использовать его код …

Кто-нибудь когда-либо успешно распаковывал LZMA2 сжатые файлы с использованием LZMA SDK?

Пожалуйста помоги!

4

Решение

Временный обходной путь будет заменить SRes res = Lzma2Dec_Allocate(&dec, outBuf.size(), &SzAllocForLzma); с SRes res = Lzma2Dec_Allocate(&dec, 8, &SzAllocForLzma); в Uncompress2Inc функция где 8 магическое число …

Однако это не верный способ решить проблему …

Первая ошибка в том, что Lzma2Enc_WriteProperties не возвращает результат, но байт свойства, который должен использоваться как второй параметр вызова Lzma2Dec_Allocate в Uncompress2Inc функция. В результате мы заменим магическое число 8 со свойством byte и все работает как положено.

Для достижения этого к кодированным данным должен быть добавлен 5-байтовый заголовок, который будет извлечен в функции декодирования. Вот пример, который работает в VS2008 (не самый совершенный код, но он работает … Я вернусь позже, когда у меня будет время, с лучшим примером):

void Lzma2Benchmark::compressChunk(std::vector<unsigned char> &outBuf, const std::vector<unsigned char> &inBuf)
{
//! \todo This is a temporary workaround, size needs to be added to the
m_uncompressedSize = inBuf.size();

std::cout << "Uncompressed size is: " << inBuf.size() << std::endl;

DWORD tickCountBeforeCompression = GetTickCount();

CLzma2EncHandle enc = Lzma2Enc_Create(&m_szAllocForLzma, &m_szAllocForLzma2);
assert(enc);

CLzma2EncProps props;
Lzma2EncProps_Init(&props);
props.lzmaProps.writeEndMark = 1; // 0 or 1
props.lzmaProps.level = 9;
props.lzmaProps.numThreads = 3;
//props.numTotalThreads = 2;

SRes res = Lzma2Enc_SetProps(enc, &props);
assert(res == SZ_OK);

// LZMA_PROPS_SIZE == 5 bytes
unsigned propsSize = LZMA_PROPS_SIZE;
outBuf.resize(propsSize);

// I think Lzma2Enc_WriteProperties returns the encoding properties in 1 Byte
Byte properties = Lzma2Enc_WriteProperties(enc);

//! \todo This is a temporary workaround
m_propByte = properties;

//! \todo Here m_propByte and m_uncompressedSize need to be added to outBuf's 5 byte header so simply add those 2 values to outBuf and start the encoding from there.

BenchmarkUtils::VectorInStream inStream = { &BenchmarkUtils::VectorInStream_Read, &inBuf, 0 };
BenchmarkUtils::VectorOutStream outStream = { &BenchmarkUtils::VectorOutStream_Write, &outBuf };

res = Lzma2Enc_Encode(enc,
(ISeqOutStream*)&outStream,
(ISeqInStream*)&inStream,
0);

std::cout << "Compress time is: " << GetTickCount() - tickCountBeforeCompression << " milliseconds.\n";

assert(res == SZ_OK);

Lzma2Enc_Destroy(enc);

std::cout << "Compressed size is: " << outBuf.size() << std::endl;
}

void Lzma2Benchmark::unCompressChunk(std::vector<unsigned char> &outBuf, const std::vector<unsigned char> &inBuf)
{
DWORD tickCountBeforeUncompression = GetTickCount();

CLzma2Dec dec;
Lzma2Dec_Construct(&dec);

//! \todo Heere the property size and the uncompressed size need to be extracted from inBuf, which is the compressed data.

// The second parameter is a temporary workaround.
SRes res = Lzma2Dec_Allocate(&dec, m_propByte/*8*/, &m_szAllocForLzma);
assert(res == SZ_OK);

Lzma2Dec_Init(&dec);

outBuf.resize(m_uncompressedSize);
unsigned outPos = 0, inPos = LZMA_PROPS_SIZE;
ELzmaStatus status;
const unsigned BUF_SIZE = 10240;

while(outPos < outBuf.size())
{
SizeT destLen = std::min(BUF_SIZE, outBuf.size() - outPos);
SizeT srcLen  = std::min(BUF_SIZE, inBuf.size() - inPos);
SizeT srcLenOld = srcLen, destLenOld = destLen;

res = Lzma2Dec_DecodeToBuf(&dec,
&outBuf[outPos],
&destLen,
&inBuf[inPos],
&srcLen,
(outPos + destLen == outBuf.size()) ? LZMA_FINISH_END : LZMA_FINISH_ANY,
&status);

assert(res == SZ_OK);
inPos += srcLen;
outPos += destLen;

if(status == LZMA_STATUS_FINISHED_WITH_MARK)
{
break;
}
}

Lzma2Dec_Free(&dec, &m_szAllocForLzma);

outBuf.resize(outPos);

std::cout << "Uncompress time is: " << GetTickCount() - tickCountBeforeUncompression << " milliseconds.\n";
}
1

Другие решения

Других решений пока нет …