Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
NetworkPacket: reading outside packet is now clearer.
Use a common function to check the reading offset
  • Loading branch information
nerzhul committed Apr 1, 2015
1 parent ddf96c7 commit 113bdd3
Show file tree
Hide file tree
Showing 2 changed files with 31 additions and 38 deletions.
67 changes: 29 additions & 38 deletions src/network/networkpacket.cpp
Expand Up @@ -39,6 +39,16 @@ NetworkPacket::~NetworkPacket()
m_data.clear();
}

void NetworkPacket::checkReadOffset(u32 from_offset)
{
if (from_offset >= m_datasize) {
std::stringstream ss;
ss << "Reading outside packet (offset: " <<
from_offset << ", packet size: " << getSize() << ")";
throw SerializationError(ss.str());
}
}

void NetworkPacket::putRawPacket(u8 *data, u32 datasize, u16 peer_id)
{
// If a m_command is already set, we are rewriting on same packet
Expand All @@ -55,8 +65,7 @@ void NetworkPacket::putRawPacket(u8 *data, u32 datasize, u16 peer_id)

char* NetworkPacket::getString(u32 from_offset)
{
if (from_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(from_offset);

return (char*)&m_data[from_offset];
}
Expand Down Expand Up @@ -187,8 +196,7 @@ std::string NetworkPacket::readLongString()

NetworkPacket& NetworkPacket::operator>>(char& dst)
{
if (m_read_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(m_read_offset);

dst = readU8(&m_data[m_read_offset]);

Expand All @@ -198,8 +206,7 @@ NetworkPacket& NetworkPacket::operator>>(char& dst)

char NetworkPacket::getChar(u32 offset)
{
if (offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(offset);

return readU8(&m_data[offset]);
}
Expand Down Expand Up @@ -276,8 +283,7 @@ NetworkPacket& NetworkPacket::operator<<(float src)

NetworkPacket& NetworkPacket::operator>>(bool& dst)
{
if (m_read_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(m_read_offset);

dst = readU8(&m_data[m_read_offset]);

Expand All @@ -287,8 +293,7 @@ NetworkPacket& NetworkPacket::operator>>(bool& dst)

NetworkPacket& NetworkPacket::operator>>(u8& dst)
{
if (m_read_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(m_read_offset);

dst = readU8(&m_data[m_read_offset]);

Expand All @@ -298,8 +303,7 @@ NetworkPacket& NetworkPacket::operator>>(u8& dst)

u8 NetworkPacket::getU8(u32 offset)
{
if (offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(offset);

return readU8(&m_data[offset]);
}
Expand All @@ -310,16 +314,14 @@ u8* NetworkPacket::getU8Ptr(u32 from_offset)
return NULL;
}

if (from_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(from_offset);

return (u8*)&m_data[from_offset];
}

NetworkPacket& NetworkPacket::operator>>(u16& dst)
{
if (m_read_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(m_read_offset);

dst = readU16(&m_data[m_read_offset]);

Expand All @@ -329,16 +331,14 @@ NetworkPacket& NetworkPacket::operator>>(u16& dst)

u16 NetworkPacket::getU16(u32 from_offset)
{
if (from_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(from_offset);

return readU16(&m_data[from_offset]);
}

NetworkPacket& NetworkPacket::operator>>(u32& dst)
{
if (m_read_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(m_read_offset);

dst = readU32(&m_data[m_read_offset]);

Expand All @@ -348,8 +348,7 @@ NetworkPacket& NetworkPacket::operator>>(u32& dst)

NetworkPacket& NetworkPacket::operator>>(u64& dst)
{
if (m_read_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(m_read_offset);

dst = readU64(&m_data[m_read_offset]);

Expand All @@ -359,8 +358,7 @@ NetworkPacket& NetworkPacket::operator>>(u64& dst)

NetworkPacket& NetworkPacket::operator>>(float& dst)
{
if (m_read_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(m_read_offset);

dst = readF1000(&m_data[m_read_offset]);

Expand All @@ -370,8 +368,7 @@ NetworkPacket& NetworkPacket::operator>>(float& dst)

NetworkPacket& NetworkPacket::operator>>(v2f& dst)
{
if (m_read_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(m_read_offset);

dst = readV2F1000(&m_data[m_read_offset]);

Expand All @@ -381,8 +378,7 @@ NetworkPacket& NetworkPacket::operator>>(v2f& dst)

NetworkPacket& NetworkPacket::operator>>(v3f& dst)
{
if (m_read_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(m_read_offset);

dst = readV3F1000(&m_data[m_read_offset]);

Expand All @@ -392,8 +388,7 @@ NetworkPacket& NetworkPacket::operator>>(v3f& dst)

NetworkPacket& NetworkPacket::operator>>(s16& dst)
{
if (m_read_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(m_read_offset);

dst = readS16(&m_data[m_read_offset]);

Expand All @@ -409,8 +404,7 @@ NetworkPacket& NetworkPacket::operator<<(s16 src)

NetworkPacket& NetworkPacket::operator>>(s32& dst)
{
if (m_read_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(m_read_offset);

dst = readS32(&m_data[m_read_offset]);

Expand All @@ -426,8 +420,7 @@ NetworkPacket& NetworkPacket::operator<<(s32 src)

NetworkPacket& NetworkPacket::operator>>(v3s16& dst)
{
if (m_read_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(m_read_offset);

dst = readV3S16(&m_data[m_read_offset]);

Expand All @@ -445,8 +438,7 @@ NetworkPacket& NetworkPacket::operator>>(v2s32& dst)

NetworkPacket& NetworkPacket::operator>>(v3s32& dst)
{
if (m_read_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(m_read_offset);

dst = readV3S32(&m_data[m_read_offset]);

Expand Down Expand Up @@ -494,8 +486,7 @@ NetworkPacket& NetworkPacket::operator<<(v3s32 src)

NetworkPacket& NetworkPacket::operator>>(video::SColor& dst)
{
if (m_read_offset >= m_datasize)
throw SerializationError("Malformed packet read");
checkReadOffset(m_read_offset);

dst = readARGB8(&m_data[m_read_offset]);

Expand Down
2 changes: 2 additions & 0 deletions src/network/networkpacket.h
Expand Up @@ -109,6 +109,8 @@ class NetworkPacket
// Temp, we remove SharedBuffer when migration finished
Buffer<u8> oldForgePacket();
private:
void checkReadOffset(u32 from_offset);

template<typename T> void checkDataSize()
{
if (m_read_offset + sizeof(T) > m_datasize) {
Expand Down

0 comments on commit 113bdd3

Please sign in to comment.