fix/session_mapping

ensure session is tracked properly per client, allow it to be sent in initial handshake
JarbasHiveMind · Jul 1, 2024 · 6f753af · 6f753af
1 parent e54626b
commit 6f753af
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/hivemind_core/protocol.py b/hivemind_core/protocol.py
@@ -447,8 +447,13 @@ def handle_handshake_message(
             client.socket.close()
             return
 
-        LOG.debug(f"client session_id: {client.sess.session_id}")
         LOG.debug(f"client site_id: {client.sess.site_id}")
+        if client.sess.session_id != "default":
+            LOG.debug(f"client session_id: {client.sess.session_id}")
+            self.clients[client.peer] = client
+        else:
+            LOG.warning("client did not send a session in it's handshake")
+
         msg = HiveMessage(HiveMessageType.HANDSHAKE, payload)
         client.send(msg)  # client can recreate crypto_key on his side now