close connection if message size exceeds 1meg

This commit is contained in:
badaix 2016-04-19 07:48:45 +02:00
parent bb413e516d
commit 40ea6e9dad
2 changed files with 8 additions and 0 deletions

View file

@ -157,6 +157,12 @@ void StreamSession::getNextMessage()
vector<char> buffer(baseMsgSize);
socketRead(&buffer[0], baseMsgSize);
baseMessage.deserialize(&buffer[0]);
if (baseMessage.size > msg::max_size)
{
logS(kLogErr) << "received message of type " << baseMessage.type << " to large: " << baseMessage.size << "\n";
stop();
return;
}
// logO << "getNextMessage: " << baseMessage.type << ", size: " << baseMessage.size << ", id: " << baseMessage.id << ", refers: " << baseMessage.refersTo << "\n";
if (baseMessage.size > buffer.size())
buffer.resize(baseMessage.size);