From 3470493a6d60243027abd2bb53058a184bcfbe9b Mon Sep 17 00:00:00 2001
From: Roman Trapeznikov <roma190505@yandex.ru>
Date: Wed, 12 Jul 2023 16:28:31 +0300
Subject: [PATCH] multipart message fix

---
 src/ballistica/scene_v1/connection/connection.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/ballistica/scene_v1/connection/connection.cc b/src/ballistica/scene_v1/connection/connection.cc
index 02e81301..c6a21701 100644
--- a/src/ballistica/scene_v1/connection/connection.cc
+++ b/src/ballistica/scene_v1/connection/connection.cc
@@ -439,6 +439,10 @@ void Connection::HandleMessagePacket(const std::vector<uint8_t>& buffer) {
         Log(LogLevel::kError, "got invalid BA_MESSAGE_MULTIPART");
       }
       if (buffer[0] == BA_MESSAGE_MULTIPART_END) {
+        if (multipart_buffer_[0] == BA_MESSAGE_MULTIPART) {
+          BA_LOG_ONCE(LogLevel::kError, "nested multipart message detected; kicking");
+          Error("");
+        }
         HandleMessagePacket(multipart_buffer_);
         multipart_buffer_.clear();
       }