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

Renamed get_client_addr to get_real_client_addr

pull/12/head
Sheng 2018-04-11 20:25:54 +08:00
parent f157be5b6c
commit 24de647738
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
handler.py
View File

@ -21,7 +21,7 @@ DELAY = 3
class MixinHandler(object): class MixinHandler(object):
def get_client_addr(self): def get_real_client_addr(self):
ip = self.request.headers.get('X-Real-Ip') ip = self.request.headers.get('X-Real-Ip')
port = self.request.headers.get('X-Real-Port') port = self.request.headers.get('X-Real-Port')
addr = None addr = None
@ -103,8 +103,8 @@ class IndexHandler(MixinHandler, tornado.web.RequestHandler):
return args return args
def get_client_addr(self): def get_client_addr(self):
return super(IndexHandler, self).get_client_addr() or self.request.\ return self.get_real_client_addr() or self.request.connection.stream.\
connection.stream.socket.getpeername() socket.getpeername()
def ssh_connect(self): def ssh_connect(self):
ssh = paramiko.SSHClient() ssh = paramiko.SSHClient()
@ -173,8 +173,7 @@ class WsockHandler(MixinHandler, tornado.websocket.WebSocketHandler):
self.worker_ref = None self.worker_ref = None
def get_client_addr(self): def get_client_addr(self):
return super(WsockHandler, self).get_client_addr() or self.stream.\ return self.get_real_client_addr() or self.stream.socket.getpeername()
socket.getpeername()
def open(self): def open(self):
self.src_addr = self.get_client_addr() self.src_addr = self.get_client_addr()