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

LibCompress: Convert GzipDecompressor from recursive to iterative

Browse source 
This way a gzip compressed file that contains a large amount of small
blocks wont cause a stack overflow.
This commit is contained in:
Idan Horowitz 2021-03-21 15:13:19 +02:00 • committed by Andreas Kling
parent 00bbe76c94
commit b3b8c01ebf
1 changed files with 77 additions and 69 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -70,20 +70,24 @@ GzipDecompressor::~GzipDecompressor()
// FIXME: Again, there are surely a ton of bugs because the code doesn't check for read errors.
size_t GzipDecompressor::read(Bytes bytes)
{
size_t total_read = 0;
while (total_read < bytes.size()) {
if (has_any_error() || m_eof)
return 0;
break;
auto slice = bytes.slice(total_read);
if (m_current_member.has_value()) {
size_t nread = current_member().m_stream.read(bytes);
current_member().m_checksum.update(bytes.trim(nread));
size_t nread = current_member().m_stream.read(slice);
current_member().m_checksum.update(slice.trim(nread));
current_member().m_nread += nread;
if (current_member().m_stream.handle_any_error()) {
set_fatal_error();
return 0;
break;
}
if (nread < bytes.size()) {
if (nread < slice.size()) {
LittleEndian<u32> crc32, input_size;
m_input_stream >> crc32 >> input_size;
@ -91,30 +95,32 @@ size_t GzipDecompressor::read(Bytes bytes)
// FIXME: Somehow the checksum is incorrect?
set_fatal_error();
return 0;
break;
}
if (input_size != current_member().m_nread) {
set_fatal_error();
return 0;
break;
}
m_current_member.clear();
return nread + read(bytes.slice(nread));
total_read += nread;
continue;
}
return nread;
total_read += nread;
continue;
} else {
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() || m_input_stream.unreliable_eof()) {
m_eof = true;
return 0;
break;
}
if (m_partial_header_offset < sizeof(BlockHeader)) {
return 0; // partial header read
break; // partial header read
}
m_partial_header_offset = 0;
@ -122,7 +128,7 @@ size_t GzipDecompressor::read(Bytes bytes)
if (!header.valid_magic_number() || !header.supported_by_implementation()) {
set_fatal_error();
return 0;
break;
}
if (header.flags & Flags::FEXTRA) {
@ -148,8 +154,10 @@ size_t GzipDecompressor::read(Bytes bytes)
}
m_current_member.emplace(header, m_input_stream);
return read(bytes);
continue;
}
}
return total_read;
}
bool GzipDecompressor::read_or_error(Bytes bytes)