system/serenity
system/serenity
1
0
Fork 0
mirror of https://github.com/SerenityOS/serenity synced 2024-07-21 18:15:58 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

AK: Fix unsigned integer underflow in DuplexMemoryStream::write.

Browse source
This commit is contained in:
asynts 2020-12-09 15:10:33 +01:00 committed by Andreas Kling
parent a34e023a33
commit ce15c9a04c
3 changed files with 18 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
AK/Forward.h
View file

@ -133,6 +133,15 @@ class WeakPtr;
template<typename T, size_t inline_capacity = 0>
class Vector;
template<typename... Parameters>
void dbgln(const char* fmtstr, const Parameters&...);
template<typename... Parameters>
void warnln(const char* fmtstr, const Parameters&...);
template<typename... Parameters>
void outln(const char* fmtstr, const Parameters&...);
}
using AK::Array;

2
AK/MemoryStream.h
View file

@ -289,7 +289,7 @@ public:
if ((m_write_offset + nwritten) % chunk_size == 0)
m_chunks.append(ByteBuffer::create_uninitialized(chunk_size));
nwritten += bytes.copy_trimmed_to(m_chunks.last().bytes().slice(m_write_offset % chunk_size));
nwritten += bytes.slice(nwritten).copy_trimmed_to(m_chunks.last().bytes().slice(m_write_offset % chunk_size));
}
m_write_offset += nwritten;

8
AK/Tests/TestMemoryStream.cpp
View file

@ -208,4 +208,12 @@ TEST_CASE(offset_of_out_of_bounds)
EXPECT(!stream.offset_of(target).has_value());
}
TEST_CASE(unsigned_integer_underflow_regression)
{
Array<u8, DuplexMemoryStream::chunk_size + 1> buffer;
DuplexMemoryStream stream;
stream << buffer;
}
TEST_MAIN(MemoryStream)