浏览代码

Add protocol validation

Ivan Savenko 1 年之前
父节点
当前提交
f620a07389
共有 1 个文件被更改,包括 6 次插入0 次删除
  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),