Browse Source

Add protocol validation

Ivan Savenko 1 year ago
parent
commit
f620a07389
1 changed files with 6 additions and 0 deletions
  1. 6 0
      lib/network/NetworkConnection.cpp

+ 6 - 0
lib/network/NetworkConnection.cpp

@@ -49,6 +49,12 @@ void NetworkConnection::onHeaderReceived(const boost::system::error_code & ecHea
 		return;
 	}
 
+	if (messageSize == 0)
+	{
+		listener.onDisconnected(shared_from_this(), "Zero-sized packet!");
+		return;
+	}
+
 	boost::asio::async_read(*socket,
 							readBuffer,
 							boost::asio::transfer_exactly(messageSize),