From 7d9cb65ba16d76cbfd9b682b646b27b130bf02bf Mon Sep 17 00:00:00 2001
From: shortcutme <tamas@zeronet.io>
Date: Mon, 19 Jun 2017 16:16:13 +0200
Subject: [PATCH] Catch message type errors on connection parsing

---
 src/Connection/Connection.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/Connection/Connection.py b/src/Connection/Connection.py
index 234053db..7134ded7 100644
--- a/src/Connection/Connection.py
+++ b/src/Connection/Connection.py
@@ -157,11 +157,14 @@ class Connection(object):
                 self.unpacker.feed(buff)
                 buff = None
                 for message in self.unpacker:
-                    self.incomplete_buff_recv = 0
-                    if "stream_bytes" in message:
-                        self.handleStream(message)
-                    else:
-                        self.handleMessage(message)
+                    try:
+                        self.incomplete_buff_recv = 0
+                        if "stream_bytes" in message:
+                            self.handleStream(message)
+                        else:
+                            self.handleMessage(message)
+                    except TypeError:
+                        raise Exception("Invalid message type: %s" % type(message))
 
                 message = None
         except Exception, err: