From f2c1cf7f04cefa57182eeb646c4f3fe246e69b0c Mon Sep 17 00:00:00 2001
From: Miguel Grinberg <miguel.grinberg@gmail.com>
Date: Sun, 4 Aug 2019 14:30:23 +0100
Subject: [PATCH] Do not allow emits on a namespace that is not connected
 (Fixes #325)

---
 socketio/asyncio_client.py           | 3 +++
 socketio/client.py                   | 3 +++
 socketio/exceptions.py               | 4 ++++
 tests/asyncio/test_asyncio_client.py | 8 ++++++++
 tests/common/test_client.py          | 8 ++++++++
 5 files changed, 26 insertions(+)

diff --git a/socketio/asyncio_client.py b/socketio/asyncio_client.py
index 40b182c..1ec3660 100644
--- a/socketio/asyncio_client.py
+++ b/socketio/asyncio_client.py
@@ -142,6 +142,9 @@ class AsyncClient(client.Client):
         Note: this method is a coroutine.
         """
         namespace = namespace or '/'
+        if namespace != '/' and namespace not in self.namespaces:
+            raise exceptions.BadNamespaceError(
+                namespace + ' is not a connected namespace.')
         self.logger.info('Emitting event "%s" [%s]', event, namespace)
         if callback is not None:
             id = self._generate_ack_id(namespace, callback)
diff --git a/socketio/client.py b/socketio/client.py
index 2c5d8b2..7ab64ed 100644
--- a/socketio/client.py
+++ b/socketio/client.py
@@ -298,6 +298,9 @@ class Client(object):
                          when addressing an individual client.
         """
         namespace = namespace or '/'
+        if namespace != '/' and namespace not in self.namespaces:
+            raise exceptions.BadNamespaceError(
+                namespace + ' is not a connected namespace.')
         self.logger.info('Emitting event "%s" [%s]', event, namespace)
         if callback is not None:
             id = self._generate_ack_id(namespace, callback)
diff --git a/socketio/exceptions.py b/socketio/exceptions.py
index 344aabb..289300c 100644
--- a/socketio/exceptions.py
+++ b/socketio/exceptions.py
@@ -24,3 +24,7 @@ class ConnectionRefusedError(ConnectionError):
 
 class TimeoutError(SocketIOError):
     pass
+
+
+class BadNamespaceError(SocketIOError):
+    pass
diff --git a/tests/asyncio/test_asyncio_client.py b/tests/asyncio/test_asyncio_client.py
index ecf88bc..6ae071b 100644
--- a/tests/asyncio/test_asyncio_client.py
+++ b/tests/asyncio/test_asyncio_client.py
@@ -194,6 +194,7 @@ class TestAsyncClient(unittest.TestCase):
 
     def test_emit_namespace(self):
         c = asyncio_client.AsyncClient()
+        c.namespaces = ['/foo']
         c._send_packet = AsyncMock()
         _run(c.emit('foo', namespace='/foo'))
         expected_packet = packet.Packet(packet.EVENT, namespace='/foo',
@@ -202,6 +203,12 @@ class TestAsyncClient(unittest.TestCase):
         self.assertEqual(c._send_packet.mock.call_args_list[0][0][0].encode(),
                          expected_packet.encode())
 
+    def test_emit_unknown_namespace(self):
+        c = asyncio_client.AsyncClient()
+        c.namespaces = ['/foo']
+        self.assertRaises(exceptions.BadNamespaceError, _run,
+                          c.emit('foo', namespace='/bar'))
+
     def test_emit_with_callback(self):
         c = asyncio_client.AsyncClient()
         c._send_packet = AsyncMock()
@@ -216,6 +223,7 @@ class TestAsyncClient(unittest.TestCase):
 
     def test_emit_namespace_with_callback(self):
         c = asyncio_client.AsyncClient()
+        c.namespaces = ['/foo']
         c._send_packet = AsyncMock()
         c._generate_ack_id = mock.MagicMock(return_value=123)
         _run(c.emit('foo', namespace='/foo', callback='cb'))
diff --git a/tests/common/test_client.py b/tests/common/test_client.py
index fea7e10..3974a8e 100644
--- a/tests/common/test_client.py
+++ b/tests/common/test_client.py
@@ -288,6 +288,7 @@ class TestClient(unittest.TestCase):
 
     def test_emit_namespace(self):
         c = client.Client()
+        c.namespaces = ['/foo']
         c._send_packet = mock.MagicMock()
         c.emit('foo', namespace='/foo')
         expected_packet = packet.Packet(packet.EVENT, namespace='/foo',
@@ -296,6 +297,12 @@ class TestClient(unittest.TestCase):
         self.assertEqual(c._send_packet.call_args_list[0][0][0].encode(),
                          expected_packet.encode())
 
+    def test_emit_unknown_namespace(self):
+        c = client.Client()
+        c.namespaces = ['/foo']
+        self.assertRaises(exceptions.BadNamespaceError, c.emit, 'foo',
+                          namespace='/bar')
+
     def test_emit_with_callback(self):
         c = client.Client()
         c._send_packet = mock.MagicMock()
@@ -310,6 +317,7 @@ class TestClient(unittest.TestCase):
 
     def test_emit_namespace_with_callback(self):
         c = client.Client()
+        c.namespaces = ['/foo']
         c._send_packet = mock.MagicMock()
         c._generate_ack_id = mock.MagicMock(return_value=123)
         c.emit('foo', namespace='/foo', callback='cb')