check message length

This commit is contained in:
orignal 2020-03-23 18:09:57 -04:00
parent fe9ac10f02
commit 744e893dce
2 changed files with 12 additions and 3 deletions

View file

@ -245,10 +245,18 @@ namespace client
{ {
m_PayloadLen = bufbe32toh (m_Header + I2CP_HEADER_LENGTH_OFFSET); m_PayloadLen = bufbe32toh (m_Header + I2CP_HEADER_LENGTH_OFFSET);
if (m_PayloadLen > 0) if (m_PayloadLen > 0)
{
if (m_PayloadLen <= I2CP_MAX_MESSAGE_LENGTH)
{ {
m_Payload = new uint8_t[m_PayloadLen]; m_Payload = new uint8_t[m_PayloadLen];
ReceivePayload (); ReceivePayload ();
} }
else
{
LogPrint (eLogError, "I2CP: Unexpected payload length ", m_PayloadLen);
Terminate ();
}
}
else // no following payload else // no following payload
{ {
HandleMessage (); HandleMessage ();

View file

@ -24,6 +24,7 @@ namespace client
{ {
const uint8_t I2CP_PROTOCOL_BYTE = 0x2A; const uint8_t I2CP_PROTOCOL_BYTE = 0x2A;
const size_t I2CP_SESSION_BUFFER_SIZE = 4096; const size_t I2CP_SESSION_BUFFER_SIZE = 4096;
const size_t I2CP_MAX_MESSAGE_LENGTH = 65535;
const size_t I2CP_HEADER_LENGTH_OFFSET = 0; const size_t I2CP_HEADER_LENGTH_OFFSET = 0;
const size_t I2CP_HEADER_TYPE_OFFSET = I2CP_HEADER_LENGTH_OFFSET + 4; const size_t I2CP_HEADER_TYPE_OFFSET = I2CP_HEADER_LENGTH_OFFSET + 4;