jinql
/
webssh
mirror of https://github.com/huashengdun/webssh
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Close websocket with no reason when invalid worker id provided or addr unmatched

pull/12/head
Sheng 2018-03-06 10:59:12 +08:00
parent 311fcfedc9
commit 9ba2512be4
1 changed files with 9 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
main.py
View File

@ -246,18 +246,15 @@ class WsockHandler(MixinHandler, tornado.websocket.WebSocketHandler):
def open(self): def open(self):
self.src_addr = self.get_addr() self.src_addr = self.get_addr()
logging.info('Connected from {}:{}'.format(*self.src_addr)) logging.info('Connected from {}:{}'.format(*self.src_addr))
worker = workers.pop(self.get_argument('id'), None) worker = workers.get(self.get_argument('id'), None)
if not worker: if worker and worker.src_addr[0] == self.src_addr[0]:
self.close(reason='Invalid worker id.') workers.pop(worker.id)
return self.set_nodelay(True)
if self.src_addr[0] != worker.src_addr[0]: worker.set_handler(self)
self.close(reason='Invalid client addr.') self.worker_ref = weakref.ref(worker)
return self.loop.add_handler(worker.fd, worker, IOLoop.READ)
else:
self.set_nodelay(True) self.close()
worker.set_handler(self)
self.worker_ref = weakref.ref(worker)
self.loop.add_handler(worker.fd, worker, IOLoop.READ)
def on_message(self, message): def on_message(self, message):
logging.debug('"{}" from {}:{}'.format(message, *self.src_addr)) logging.debug('"{}" from {}:{}'.format(message, *self.src_addr))