RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-26 05:57:44 +00:00
Code Issues Activity Actions

LibCompress: Allow partial header reads in GzipDecompressor

Browse source 
We now read the header into a temporary header byte array that is used
as the header once its filled up by the input stream, instead of just
ending the stream if we are out of bytes mid header.
This commit is contained in:
Idan Horowitz 2021-03-16 17:39:50 +02:00 • committed by Andreas Kling
parent 3dfa3d2d9d
commit 1d8ab74cbf
2 changed files with 11 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

12
Userland/Libraries/LibCompress/Gzip.cpp
View file

@ -106,14 +106,20 @@ size_t GzipDecompressor::read(Bytes bytes)
return nread;
} else {
BlockHeader header;
m_input_stream >> Bytes { &header, sizeof(header) };
m_partial_header_offset += m_input_stream.read(Bytes { m_partial_header, sizeof(BlockHeader) }.slice(m_partial_header_offset));
if (m_input_stream.handle_any_error()) {
if (m_input_stream.handle_any_error() || m_input_stream.unreliable_eof()) {
m_eof = true;
return 0;
}
if (m_partial_header_offset < sizeof(BlockHeader)) {
return 0; // partial header read
}
m_partial_header_offset = 0;
BlockHeader header = *(reinterpret_cast<BlockHeader*>(m_partial_header));
if (!header.valid_magic_number() || !header.supported_by_implementation()) {
set_fatal_error();
return 0;