From c952be4e4518b8ac93bcec6b926eacd2293dc839 Mon Sep 17 00:00:00 2001 From: midoks Date: Thu, 26 Jan 2023 22:49:55 +0800 Subject: [PATCH] up --- class/core/mw.py | 1 + class/core/ssh_terminal.py | 29 +++++++------- plugins/webssh/index.py | 27 +++++++++++++ plugins/webssh/js/webssh.js | 75 ++++++++++++++++++++++++++++--------- 4 files changed, 99 insertions(+), 33 deletions(-) diff --git a/class/core/mw.py b/class/core/mw.py index e81f99e0b..65140e902 100755 --- a/class/core/mw.py +++ b/class/core/mw.py @@ -1503,6 +1503,7 @@ def getSshDir(): def createRsa(): + # ssh-keygen -t rsa -P "" -C "midoks@163.com" ssh_dir = getSshDir() # mw.execShell("rm -f /root/.ssh/*") if not os.path.exists(ssh_dir + '/authorized_keys'): diff --git a/class/core/ssh_terminal.py b/class/core/ssh_terminal.py index 9246ff3ee..3baad80cc 100644 --- a/class/core/ssh_terminal.py +++ b/class/core/ssh_terminal.py @@ -37,6 +37,7 @@ class ssh_terminal: __sid = '' __host = None + __type = '0' __port = 22 __user = None __pass = None @@ -243,28 +244,24 @@ class ssh_terminal: if not self.__pass and not self.__pkey: return self.returnMsg(False, '密码或私钥不能都为空: {}:{}'.format(self.__host, self.__port)) - if self.__pkey: + if self.__pkey != '' and self.__type != '0': self.debug('正在认证私钥') - if sys.version_info[0] == 2: - try: - self.__pkey = self.__pkey.encode('utf-8') - except: - pass - p_file = BytesIO(self.__pkey) - else: - p_file = StringIO(self.__pkey) - + p_file = StringIO(str(self.__pkey.replace('\\n', '\n'))) + # p_file = "/tmp/t_ssh_pkey.txt" + # mw.writeFile(p_file, self.__pkey.replace('\\n', '\n')) + # mw.execShell('chmod 600 ' + p_file) try: + p_file.seek(0) pkey = paramiko.RSAKey.from_private_key(p_file) except: try: p_file.seek(0) # 重置游标 - pkey = paramiko.Ed25519Key.from_private_key(p_file) + pkey = paramiko.Ed25519Key.from_private_key( + p_file) except: try: p_file.seek(0) - pkey = paramiko.ECDSAKey.from_private_key( - p_file) + pkey = paramiko.ECDSAKey.from_private_key(p_file) except: p_file.seek(0) pkey = paramiko.DSSKey.from_private_key(p_file) @@ -285,8 +282,7 @@ class ssh_terminal: self.setSshdConfig(True) self.__tp.close() e = str(e) - if e.find('websocket error!') != -1: - return self.returnMsg(True, '连接成功') + # print(e) if e.find('Authentication timeout') != -1: self.debug("认证超时{}".format(e)) return self.returnMsg(False, '认证超时,请按回车重试!{}'.format(e)) @@ -332,6 +328,9 @@ class ssh_terminal: _t = mw.readFile(dst_info) info = json.loads(_t) + if 'type' in info: + self.__type = info['type'] + if 'port' in info: self.__port = int(info['port']) if 'username' in info: diff --git a/plugins/webssh/index.py b/plugins/webssh/index.py index 8e410b5ae..02619f2ef 100755 --- a/plugins/webssh/index.py +++ b/plugins/webssh/index.py @@ -112,6 +112,33 @@ class App(): rdata = mw.readFile(file) return json.loads(rdata) + def get_server_by_host(self): + args = self.getArgs() + check = self.checkArgs(args, ['host']) + if not check[0]: + return check[1] + + info_file = self.__host_dir + '/' + args['host'] + '/info.json' + if os.path.exists(info_file): + try: + info_tmp = self.getSshInfo(info_file) + host_info = {} + host_info['host'] = args['host'] + host_info['port'] = info_tmp['port'] + host_info['ps'] = info_tmp['ps'] + host_info['type'] = info_tmp['type'] + if 'password' in info_tmp: + host_info['password'] = info_tmp['password'] + if 'pkey' in info_tmp: + host_info['pkey'] = info_tmp['pkey'] + if 'pkey_passwd' in info_tmp: + host_info['pkey_passwd'] = info_tmp['pkey_passwd'] + except Exception as e: + return mw.returnJson(False, '错误:' + str(e)) + + return mw.returnJson(True, 'ok!', host_info) + return mw.returnJson(False, '不存在此配置') + def get_server_list(self): host_list = [] for name in os.listdir(self.__host_dir): diff --git a/plugins/webssh/js/webssh.js b/plugins/webssh/js/webssh.js index 4e94786d2..2441af1d2 100755 --- a/plugins/webssh/js/webssh.js +++ b/plugins/webssh/js/webssh.js @@ -354,7 +354,7 @@ function webShell_getHostList(info){ for (var i = 0; i < alist.length; i++) { tli+='
  • \ \ - '+alist[i]['host']+'\ + '+alist[i]['host']+'\ \ \ \ @@ -364,13 +364,13 @@ function webShell_getHostList(info){ $('.tootls_host_list').html(tli); - $('.data-host-list .glyphicon-edit').click(function(){ + $('.data-host-list .glyphicon-edit').click(function(e){ var index = $(this).parent().parent().attr('data-index'); var info = alist[index]; webShell_addServer(info); }); - $('.data-host-list .glyphicon-trash').click(function(){ + $('.data-host-list .glyphicon-trash').click(function(e){ var index = $(this).parent().parent().attr('data-index'); var t = alist[index]; appPost('del_server', {host:t['host']}, function(rdata){ @@ -382,16 +382,17 @@ function webShell_getHostList(info){ }); - $('.tootls_host_list').on('click', 'li', function (e) { - var index = $(this).index(); - var host = $(this).data('host'); - $(this).find('i').addClass('active'); + $('.tootls_host_list .host').on('click', function (e) { + var _this = $(this).parent(); + var index = $(_this).index(); + var host = $(_this).data('host'); + $(_this).find('i').addClass('active'); if ($('.item[data-host="' + host + '"]').length > 0) { - layer.msg('已经打开!', { icon: 0, time: 3000 }) - // layer.msg('如需多会话窗口，请右键终端标题复制会话!', { icon: 0, time: 3000 }); + layer.msg('已经打开!', { icon: 0, time: 3000 }); } else { webShell_openTermView(alist[index]); } + e.preventDefault(); }); }); } @@ -420,7 +421,7 @@ function webShell_addServer(info=[]){ 验证方式\
    \
    \ - \ + \ \
    \ @@ -452,6 +453,38 @@ function webShell_addServer(info=[]){
    \ ', success:function(){ + if (typeof(info['host'])!='undefined'){ + $('input[name="host"]').val(info['host']); + appPost('get_server_by_host',{host:info['host']},function(rdata){ + var rdata = $.parseJSON(rdata.data); + var jdata = rdata.data; + if (jdata['type'] == 0){ + $('input[name="password"]').val(jdata['password']); + } else{ + $('textarea[name="pkey"]').val(jdata['pkey']); + $('input[name="pkey_passwd"]').val(jdata['pkey_passwd']); + } + $('input[name="ps"]').val(jdata['ps']); + $('.auth_type_checkbox').each(function(){ + if ($(this).data('ctype') == jdata['type']){ + $(this).addClass('btn-success'); + } else { + $(this).removeClass('btn-success'); + } + }); + if (jdata['type'] == 0){ + $('.c_password_view').removeClass('show').addClass('show'); + $('.c_pkey_view').removeClass('show').addClass('hide'); + $('.key_pwd_line').removeClass('show').addClass('hide'); + + }else{ + $('.c_password_view').removeClass('show').addClass('hide'); + $('.c_pkey_view').removeClass('show').addClass('show'); + $('.key_pwd_line').removeClass('show').addClass('show'); + } + }); + } + $('.auth_type_checkbox').click(function(){ var ctype = $(this).attr('data-ctype'); $('.auth_type_checkbox').removeClass('btn-success'); @@ -473,18 +506,20 @@ function webShell_addServer(info=[]){ var host = $('input[name="host"]').val(); var port = $('input[name="port"]').val(); var username = $('input[name="username"]').val(); - - var type = $('.auth_type_checkbox').attr('data-ctype'); + var password = ''; + var pkey = ''; + var pkey_passwd = ''; + var type = $('.btn-group .btn-success').attr('data-ctype'); if (type == "0"){ - var password = $('input[name="password"]').val(); + password = $('input[name="password"]').val(); } else{ - var pkey = $('input[name="pkey"]').val(); - var pkey_passwd = $('input[name="pkey_passwd"]').val(); + pkey = $('textarea[name="pkey"]').val(); + pkey_passwd = $('input[name="pkey_passwd"]').val(); } var ps = $('input[name="ps"]').val(); - appPost('add_server',{ + var req_data = { type:type, host:host, port:port, @@ -492,8 +527,12 @@ function webShell_addServer(info=[]){ password:password, pkey:pkey, pkey_passwd:pkey_passwd, - ps - },function(rdata){ + ps:ps, + }; + + console.log(req_data); + + appPost('add_server',req_data,function(rdata){ layer.close(l); var rdata = $.parseJSON(rdata.data); showMsg(rdata.msg, function(){