Catch message type errors on connection parsing
This commit is contained in:
parent
9f72fdeb41
commit
7d9cb65ba1
1 changed files with 8 additions and 5 deletions
|
@ -157,11 +157,14 @@ class Connection(object):
|
|||
self.unpacker.feed(buff)
|
||||
buff = None
|
||||
for message in self.unpacker:
|
||||
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:
|
||||
|
|
Loading…
Reference in a new issue