forked from ShuriZma/suyu
1
0
Fork 0

Parcel: Ensure we don't read past the end of the parcels in Vi.

This commit is contained in:
Subv 2018-02-17 13:54:59 -05:00
parent 2662de6e52
commit d758332425
1 changed files with 5 additions and 0 deletions

View File

@ -39,6 +39,7 @@ public:
template <typename T>
T Read() {
ASSERT(read_index + sizeof(T) <= buffer.size());
T val;
std::memcpy(&val, buffer.data() + read_index, sizeof(T));
read_index += sizeof(T);
@ -48,6 +49,7 @@ public:
template <typename T>
T ReadUnaligned() {
ASSERT(read_index + sizeof(T) <= buffer.size());
T val;
std::memcpy(&val, buffer.data() + read_index, sizeof(T));
read_index += sizeof(T);
@ -55,6 +57,7 @@ public:
}
std::vector<u8> ReadBlock(size_t length) {
ASSERT(read_index + length <= buffer.size());
const u8* const begin = buffer.data() + read_index;
const u8* const end = begin + length;
std::vector<u8> data(begin, end);
@ -97,6 +100,8 @@ public:
}
void Deserialize() {
ASSERT(buffer.size() > sizeof(Header));
Header header{};
std::memcpy(&header, buffer.data(), sizeof(Header));