mirror of
https://github.com/RGBCube/serenity
synced 2025-07-25 05:47:35 +00:00
LibCompress: Switch DeflateDecompressor
to a fallible constructor
We don't have anything fallible in there yet, but we will soon switch the seekback buffer to the new `CircularBuffer`, which has a fallible constructor. We have to do the same for the internal `GzipDecompressor::Member` class, as it needs to construct a `DeflateCompressor` from its received stream.
This commit is contained in:
parent
d717a08003
commit
d23f0a7405
4 changed files with 34 additions and 16 deletions
|
@ -189,6 +189,11 @@ ErrorOr<bool> DeflateDecompressor::UncompressedBlock::try_read_more()
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
ErrorOr<NonnullOwnPtr<DeflateDecompressor>> DeflateDecompressor::construct(Core::Stream::Handle<Core::Stream::Stream> stream)
|
||||||
|
{
|
||||||
|
return TRY(adopt_nonnull_own_or_enomem(new (nothrow) DeflateDecompressor(move(stream))));
|
||||||
|
}
|
||||||
|
|
||||||
DeflateDecompressor::DeflateDecompressor(Core::Stream::Handle<Core::Stream::Stream> stream)
|
DeflateDecompressor::DeflateDecompressor(Core::Stream::Handle<Core::Stream::Stream> stream)
|
||||||
: m_input_stream(make<Core::Stream::LittleEndianInputBitStream>(move(stream)))
|
: m_input_stream(make<Core::Stream::LittleEndianInputBitStream>(move(stream)))
|
||||||
{
|
{
|
||||||
|
@ -311,12 +316,12 @@ void DeflateDecompressor::close()
|
||||||
ErrorOr<ByteBuffer> DeflateDecompressor::decompress_all(ReadonlyBytes bytes)
|
ErrorOr<ByteBuffer> DeflateDecompressor::decompress_all(ReadonlyBytes bytes)
|
||||||
{
|
{
|
||||||
auto memory_stream = TRY(Core::Stream::FixedMemoryStream::construct(bytes));
|
auto memory_stream = TRY(Core::Stream::FixedMemoryStream::construct(bytes));
|
||||||
DeflateDecompressor deflate_stream { move(memory_stream) };
|
auto deflate_stream = TRY(DeflateDecompressor::construct(move(memory_stream)));
|
||||||
DuplexMemoryStream output_stream;
|
DuplexMemoryStream output_stream;
|
||||||
|
|
||||||
auto buffer = TRY(ByteBuffer::create_uninitialized(4096));
|
auto buffer = TRY(ByteBuffer::create_uninitialized(4096));
|
||||||
while (!deflate_stream.is_eof()) {
|
while (!deflate_stream->is_eof()) {
|
||||||
auto const slice = TRY(deflate_stream.read(buffer));
|
auto const slice = TRY(deflate_stream->read(buffer));
|
||||||
output_stream.write_or_error(slice);
|
output_stream.write_or_error(slice);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -76,7 +76,7 @@ public:
|
||||||
friend CompressedBlock;
|
friend CompressedBlock;
|
||||||
friend UncompressedBlock;
|
friend UncompressedBlock;
|
||||||
|
|
||||||
DeflateDecompressor(Core::Stream::Handle<Core::Stream::Stream> stream);
|
static ErrorOr<NonnullOwnPtr<DeflateDecompressor>> construct(Core::Stream::Handle<Core::Stream::Stream> stream);
|
||||||
~DeflateDecompressor();
|
~DeflateDecompressor();
|
||||||
|
|
||||||
virtual ErrorOr<Bytes> read(Bytes) override;
|
virtual ErrorOr<Bytes> read(Bytes) override;
|
||||||
|
@ -88,6 +88,8 @@ public:
|
||||||
static ErrorOr<ByteBuffer> decompress_all(ReadonlyBytes);
|
static ErrorOr<ByteBuffer> decompress_all(ReadonlyBytes);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
DeflateDecompressor(Core::Stream::Handle<Core::Stream::Stream> stream);
|
||||||
|
|
||||||
ErrorOr<u32> decode_length(u32);
|
ErrorOr<u32> decode_length(u32);
|
||||||
ErrorOr<u32> decode_distance(u32);
|
ErrorOr<u32> decode_distance(u32);
|
||||||
ErrorOr<void> decode_codes(CanonicalCode& literal_code, Optional<CanonicalCode>& distance_code);
|
ErrorOr<void> decode_codes(CanonicalCode& literal_code, Optional<CanonicalCode>& distance_code);
|
||||||
|
|
|
@ -39,6 +39,18 @@ bool BlockHeader::supported_by_implementation() const
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
ErrorOr<NonnullOwnPtr<GzipDecompressor::Member>> GzipDecompressor::Member::construct(BlockHeader header, Core::Stream::Stream& stream)
|
||||||
|
{
|
||||||
|
auto deflate_stream = TRY(DeflateDecompressor::construct(Core::Stream::Handle<Core::Stream::Stream>(stream)));
|
||||||
|
return TRY(adopt_nonnull_own_or_enomem(new (nothrow) Member(header, move(deflate_stream))));
|
||||||
|
}
|
||||||
|
|
||||||
|
GzipDecompressor::Member::Member(BlockHeader header, NonnullOwnPtr<DeflateDecompressor> stream)
|
||||||
|
: m_header(header)
|
||||||
|
, m_stream(move(stream))
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
GzipDecompressor::GzipDecompressor(NonnullOwnPtr<Core::Stream::Stream> stream)
|
GzipDecompressor::GzipDecompressor(NonnullOwnPtr<Core::Stream::Stream> stream)
|
||||||
: m_input_stream(move(stream))
|
: m_input_stream(move(stream))
|
||||||
{
|
{
|
||||||
|
@ -58,8 +70,8 @@ ErrorOr<Bytes> GzipDecompressor::read(Bytes bytes)
|
||||||
|
|
||||||
auto slice = bytes.slice(total_read);
|
auto slice = bytes.slice(total_read);
|
||||||
|
|
||||||
if (m_current_member.has_value()) {
|
if (m_current_member) {
|
||||||
auto current_slice = TRY(current_member().m_stream.read(slice));
|
auto current_slice = TRY(current_member().m_stream->read(slice));
|
||||||
current_member().m_checksum.update(current_slice);
|
current_member().m_checksum.update(current_slice);
|
||||||
current_member().m_nread += current_slice.size();
|
current_member().m_nread += current_slice.size();
|
||||||
|
|
||||||
|
@ -131,7 +143,7 @@ ErrorOr<Bytes> GzipDecompressor::read(Bytes bytes)
|
||||||
// FIXME: we should probably verify this instead of just assuming it matches
|
// FIXME: we should probably verify this instead of just assuming it matches
|
||||||
}
|
}
|
||||||
|
|
||||||
m_current_member.emplace(header, *m_input_stream);
|
m_current_member = TRY(Member::construct(header, *m_input_stream));
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -56,25 +56,24 @@ public:
|
||||||
private:
|
private:
|
||||||
class Member {
|
class Member {
|
||||||
public:
|
public:
|
||||||
Member(BlockHeader header, Core::Stream::Stream& stream)
|
static ErrorOr<NonnullOwnPtr<Member>> construct(BlockHeader header, Core::Stream::Stream&);
|
||||||
: m_header(header)
|
|
||||||
, m_stream(Core::Stream::Handle<Core::Stream::Stream>(stream))
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
BlockHeader m_header;
|
BlockHeader m_header;
|
||||||
DeflateDecompressor m_stream;
|
NonnullOwnPtr<DeflateDecompressor> m_stream;
|
||||||
Crypto::Checksum::CRC32 m_checksum;
|
Crypto::Checksum::CRC32 m_checksum;
|
||||||
size_t m_nread { 0 };
|
size_t m_nread { 0 };
|
||||||
|
|
||||||
|
private:
|
||||||
|
Member(BlockHeader, NonnullOwnPtr<DeflateDecompressor>);
|
||||||
};
|
};
|
||||||
|
|
||||||
Member const& current_member() const { return m_current_member.value(); }
|
Member const& current_member() const { return *m_current_member; }
|
||||||
Member& current_member() { return m_current_member.value(); }
|
Member& current_member() { return *m_current_member; }
|
||||||
|
|
||||||
NonnullOwnPtr<Core::Stream::Stream> m_input_stream;
|
NonnullOwnPtr<Core::Stream::Stream> m_input_stream;
|
||||||
u8 m_partial_header[sizeof(BlockHeader)];
|
u8 m_partial_header[sizeof(BlockHeader)];
|
||||||
size_t m_partial_header_offset { 0 };
|
size_t m_partial_header_offset { 0 };
|
||||||
Optional<Member> m_current_member;
|
OwnPtr<Member> m_current_member {};
|
||||||
|
|
||||||
bool m_eof { false };
|
bool m_eof { false };
|
||||||
};
|
};
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue