oci/zeronet
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Catch message type errors on connection parsing

Browse source
This commit is contained in:
shortcutme 2017-06-19 16:16:13 +02:00
parent 9f72fdeb41
commit 7d9cb65ba1
No known key found for this signature in database
GPG key ID: 5B63BAE6CB9613AE
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/Connection/Connection.py
View file

@ -157,11 +157,14 @@ class Connection(object):
self.unpacker.feed(buff) self.unpacker.feed(buff)
buff = None buff = None
for message in self.unpacker: for message in self.unpacker:
try:
self.incomplete_buff_recv = 0 self.incomplete_buff_recv = 0
if "stream_bytes" in message: if "stream_bytes" in message:
self.handleStream(message) self.handleStream(message)
else: else:
self.handleMessage(message) self.handleMessage(message)
except TypeError:
raise Exception("Invalid message type: %s" % type(message))
message = None message = None
except Exception, err: except Exception, err: