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

Added get_addr to get real client addr

pull/4/head
Sheng 2017-11-20 17:58:32 +08:00
parent e64ffe978b
commit 5d0fad48ec
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
main.py
View File

@ -206,8 +206,16 @@ class WsockHandler(tornado.websocket.WebSocketHandler):
def check_origin(self, origin): def check_origin(self, origin):
return True return True
def get_addr(self):
ip = self.request.headers.get_list('X-Real-Ip')
port = self.request.headers.get_list('X-Real-Port')
addr = ':'.join(ip + port)
if not addr:
addr = '{}:{}'.format(*self.stream.socket.getpeername())
return addr
def open(self): def open(self):
self.src_addr = '{}:{}'.format(*self.stream.socket.getpeername()) 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.pop(self.get_argument('id'), None)
if not worker: if not worker: