AuroraRuntime/Source/Compression/Compressors/DeflateDecompressor.hpp

125 lines
3.3 KiB
C++

/***
Copyright (C) 2022 J Reece Wilson (a/k/a "Reece"). All rights reserved.
File: DeflateDecompressor.hpp
Date: 2022-2-15
Author: Reece
***/
#pragma once
#include "zlib.h"
namespace Aurora::Compression
{
struct ZIPInflate : BaseStream
{
DecompressInfo meta;
ZIPInflate(const DecompressInfo &meta, bool hasBits, AuInt8 bits) : meta(meta), BaseStream(meta.uInternalStreamSize), cBits_(bits), bHasBits_(hasBits)
{}
~ZIPInflate()
{
if (this->init_)
{
inflateEnd(&this->ctx_);
}
}
bool Init(const AuSPtr<IO::IStreamReader> &reader) override
{
int ret;
this->pReader_ = reader;
if (!this->IsValid())
{
SysPushErrorMem();
return false;
}
if (this->bHasBits_)
{
ret = inflateInit2(&this->ctx_, this->cBits_);
}
else
{
ret = inflateInit(&this->ctx_);
}
if (ret < Z_OK)
{
SysPushErrorMem("Error: {}", zError(ret));
return false;
}
this->init_ = true;
this->SetArray(this->din_);
this->SetOutArray(this->dout_);
return true;
}
AuStreamReadWrittenPair_t Ingest_s(AuUInt32 input) override
{
AuUInt32 done {}, read {};
if (!this->pReader_)
{
return {};
}
while (read < input)
{
read += IngestForInPointer<Bytef, uInt>(this->pReader_, this->ctx_.next_in, this->ctx_.avail_in, input - read, this);
if (!this->ctx_.avail_in)
{
return {read, done};
}
do
{
auto [pMainDOut, uMainDOutLength] = this->GetDOutPair();
this->ctx_.avail_out = uMainDOutLength;
this->ctx_.next_out = (Bytef *)pMainDOut;
if (!this->ctx_.avail_out)
{
break;
}
auto ret = inflate(&this->ctx_, Z_NO_FLUSH);
if (ret < Z_OK)
{
this->SetLastError((int)ret, zError(ret));
this->pReader_.reset();
return AuMakePair(read, 0);
}
auto have = uMainDOutLength - this->ctx_.avail_out;
done += have;
if (!Write2(pMainDOut,
have))
{
this->pReader_.reset();
this->SetLastError(0x69, "OOM");
return AuMakePair(read, 0);
}
}
while (this->ctx_.avail_out == 0);
}
return {read, done};
}
private:
AuSPtr<IO::IStreamReader> pReader_;
z_stream ctx_ {};
bool init_ {};
bool bHasBits_;
AuInt8 cBits_;
unsigned char din_[kChunkSize];
unsigned char dout_[kChunkSize];
};
}