Commit aa4d989d authored by 广宏伟's avatar 广宏伟

Merged in dev (pull request #76)

Dev
parents 8a416ea6 27c67ee0
......@@ -104,7 +104,8 @@ class SSHConnection:
ssh.connect(gateway.ip, port=gateway.port,
username=gateway.username,
password=gateway.password,
pkey=gateway.private_key_obj)
pkey=gateway.private_key_obj,
timeout=TIMEOUT)
except(paramiko.AuthenticationException,
paramiko.BadAuthenticationType,
SSHException):
......
......@@ -34,6 +34,10 @@ class BaseNamespace(Namespace):
user = app_service.check_user_cookie(session_id, csrf_token)
if token:
user = app_service.check_user_with_token(token)
msg = "Get current user: session_id<{}> token<{}> => {}".format(
session_id, token, user
)
logger.debug(msg)
return user
......@@ -149,27 +153,30 @@ class ProxyNamespace(BaseNamespace):
token = message.get('token', None)
secret = message.get('secret', None)
win_size = message.get('size', (80, 24))
room = self.new_room()
self.emit('room', {'room': room["id"], 'secret': secret})
join_room(room['id'])
if not token or not secret:
logger.debug("Token or secret is None: {}".format(token, secret))
self.emit('data', {'data': "\nOperation not permitted!",
'room': room["id"]})
msg = "Token or secret is None: {} {}".format(token, secret)
logger.error(msg)
self.emit('data', {'data': msg, 'room': room['id']}, room=room['id'])
self.emit('disconnect')
return None
return
info = app_service.get_token_asset(token)
logger.debug(info)
if not info:
logger.debug("Token info is None")
self.emit('data', {'data': "\nOperation not permitted!",
'room': room["id"]})
msg = "Token info is none, maybe token expired"
logger.error(msg)
self.emit('data', {'data': msg, 'room': room['id']}, room=room['id'])
self.emit('disconnect')
return None
user_id = info.get('user', None)
self.current_user = app_service.get_user_profile(user_id)
room["request"].user = self.current_user
# room["request"].user = self.current_user
self.on_host({
'secret': secret,
'uuid': info['asset'],
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment