1
Fork 0
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:
Tim Schumacher 2023-01-01 00:38:05 +01:00 committed by Andrew Kaster
parent d717a08003
commit d23f0a7405
4 changed files with 34 additions and 16 deletions

View file

@ -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);
} }

View file

@ -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);

View file

@ -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;
} }
} }

View file

@ -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 };
}; };