Add error hanlding for messages.

This commit is contained in:
Yuxin Wang 2018-09-26 21:01:11 -04:00
parent f56bc1c33d
commit d3901e740e
2 changed files with 4 additions and 0 deletions

View file

@ -120,6 +120,8 @@ class Peer(MessageServer):
elif message['type'] == MessageType.REPLY_FILE_LIST:
self._file_list = message['file_list']
self._list_file_lock.release()
else:
logger.error('Undefined message with type {}, full message: {}'.format(message['type'], message))
logger.debug(self._peers.values())

View file

@ -72,6 +72,8 @@ class Tracker(MessageServer):
'type': MessageType.REPLY_FILE_LIST,
'file_list': self._file_list
})
else:
logger.error('Undefined message with {} type, full packet: {}'.format(message['type'], message))
def _client_closed(self, client):
assert isinstance(client, socket.socket)