首页 > 代码库 > jumpserver_core

jumpserver_core

2024-09-01 12:26:46   213人阅读

<style>body,table { font-family: 微软雅黑; font-size: 10pt } table { border-collapse: collapse; border: solid gray; border-width: 2px 0 2px 0 } th { border: 1px solid gray; padding: 4px; background-color: #DDD } td { border: 1px solid gray; padding: 4px } tr:nth-child(2n) { background-color: #f8f8f8 }</style>

一、标准版 
#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
功能：
    实现了跳板ssh功能
缺点：
    1、用户的输入的cmd只有在回车后，才会被发送到远端 server；造成不能使用补全功能
    2、每次回显消息时候，都会附带这执行该条消息的cmd
"""
import paramiko
import sys
import select


tran = paramiko.Transport((‘192.168.10.206‘, 22,), )  # 基于创建transport 对象
tran.start_client()                                   # 基于transport 创建一个SSH session 作为 client
tran.auth_password(‘root‘, ‘redhat‘)                  # 向server 进行连接认证


chan = tran.open_session()  # 向server请求一个新的会话Chanlen，可定义窗口大小，最大包大小，超时时间
chan.get_pty()              # 基于该session 通道，向server请求一个伪终端
chan.invoke_shell()         # 开启一个交互shell：通过伪终端


while True:
    read, write, error = select.select([chan, sys.stdin, ], [], [], 1)
    # 若chan 发生变化，表示远端server发送了数据过来
    if chan in read:  
        try:
            data = http://www.mamicode.com/chan.recv(1024).decode()
            if len(data) == 0: # 链路异常，会受到空消息
                print(‘\r\n *** END *** \r\n‘)
                break  
            sys.stdout.write(data)
            sys.stdout.flush()

        except Exception as ex:
            print(ex)
    
    # 若stdin有变化，表示用户有cmd需要送给远端server
    if sys.stdin in read:
        inp = sys.stdin.readline()
        chan.sendall(inp)

chan.close()
tran.close()

二、适用版

#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
功能：
    实现了跳板ssh功能
    实现了Tab补全，实时发送cmd（每输入一个字符都会发送）
缺点：
    体验不好
"""
import paramiko
import sys
import select
import termios
import tty

tran = paramiko.Transport((‘192.168.10.206‘, 22,), )
tran.start_client()
tran.auth_password(‘root‘, ‘redhat‘)

chan = tran.open_session()
chan.get_pty()
chan.invoke_shell()


oldtty = termios.tcgetattr(sys.stdin)  # 获取源终端属性
try:
    # 为当前终端设置新属性
    tty.setraw(sys.stdin.fileno())
    tty.setcbreak(sys.stdin.fileno())
    chan.settimeout(0.0)

    while True:
        read, write, error = select.select([chan, sys.stdin, ], [], [], 1)

        # 接受消息
        if chan in read:
            try:
                data = http://www.mamicode.com/chan.recv(1024).decode()
                if len(data) == 0:
                    print(‘\r\n *** END *** \r\n‘)
                    break
                sys.stdout.write(data)
                sys.stdout.flush()

            except Exception as ex:
                print(ex)

        # 发送命令
        if sys.stdin in read:
            inp = sys.stdin.read(1)  # 注意这里一定是一个字符，若为readline则会卡主；
            chan.sendall(inp)

finally:
    # 将终端恢复为原始属性
    termios.tcsetattr(sys.stdin, termios.TCSADRAIN, oldtty)

chan.close()
tran.close()

三、豪华版 
#!/usr/bin/env python
# -*- coding:utf-8 -*-

import os
import sys
import paramiko
import select
import getpass
import termios
import tty


def pre_init():
    # 获取用户名
    default_username = getpass.getuser()
    username = input(‘username, defult is {}: ‘.format(default_username)).strip()
    if len(username) == 0:
        username = default_username

    # 获取目标服务器
    hostname = input(‘hostname: ‘).strip()
    if len(hostname) == 0:
        print(‘hostname required‘.center(40, ‘*‘))
        exit(1)
    return (username, hostname)


class JumpClient:
    def __init__(self, username, hostname):
        self.username = username
        self.hostname = hostname
        self.tran = paramiko.Transport((self.hostname, 22))
        self.tran.start_client()
        try:
            self.init_connection()
        except Exception as ex:
            print(ex)
        else:
            self.chan = self.tran.open_session()
            self.chan.get_pty()
            self.chan.invoke_shell()

    def init_connection(self):
        # 获取验证类型；默认为密码认证
        default_auth_type = ‘p‘
        auth_type = input(‘Auth type [password(p)/key(k)], defulat is [p]: ‘).strip()
        if len(auth_type) == 0:
            auth_type = default_auth_type

        # 载入私钥；判断位置、判断是否需要密码
        if auth_type == ‘k‘:
            default_key = os.path.join(os.environ.get(‘HOME‘), ‘.ssh‘, ‘id_rsa‘)
            auth_key = input(‘Auth key path, defulat is {}: ‘.format(default_key))
            if len(auth_key) == 0:
                auth_key = default_key
            # 若私钥没有密码，则直接载入；若有密码，则输入密码
            try:
                key = paramiko.RSAKey.from_private_key_file(auth_key)
            except paramiko.PasswordRequiredException:
                passwd = getpass.getpass(‘RSA key\‘s  passwd: ‘).strip()
                key = paramiko.RSAKey.from_private_key_file(auth_key, password=passwd)
            finally:
                self.tran.auth_publickey(self.username, key)  # 链接服务器

        if auth_type == ‘p‘:
            server_passwd = getpass.getpass(‘password: ‘).strip()
            self.tran.auth_password(self.username, server_passwd)

    def open_terminal(self):
        # 获取原先 tty终端属性
        oldtty = termios.tcgetattr(sys.stdin)
        try:
            # 设置新的 tty终端属性
            tty.setraw(sys.stdin.fileno())
            tty.setcbreak(sys.stdin.fileno())
            self.chan.settimeout(0.0)

            while True:
                r_lst, w_lst, e_lst = select.select([self.chan, sys.stdin, ], [], [], 1)
                if self.chan in r_lst:
                    try:
                        data = http://www.mamicode.com/self.chan.recv(1024).decode()
                        if len(data) == 0:
                            print(‘\r\n*** EOF ***\r\n‘)
                            break
                        else:
                            sys.stdout.write(data)  # 不用print，因为末尾会自动追加换行
                            sys.stdout.flush()
                    except Exception as ex:
                        print(ex)

                if sys.stdin in r_lst:
                    data = http://www.mamicode.com/sys.stdin.read(1)  # 读取一个字符就发送
                    self.chan.sendall(data)
        finally:
            # 恢复原先 tty终端属性
            termios.tcsetattr(sys.stdin, termios.TCSADRAIN, oldtty)

    def close_terminal(self):
        self.chan.close()
        self.tran.close()

if __name__ == ‘__main__‘:
    user, host = pre_init()
    obj = JumpClient(user, host)
    obj.open_terminal()
    obj.close_terminal()
四、至尊版
#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
支持的客户端：1、windows（不支持补全、字符立即发送）  2、linux（支持补全等，但vim工具有问题）
"""
import os
import sys
import paramiko
import select
import getpass
import threading
try:
    import termios
    import tty
    linux_client = True
except ImportError:
    linux_client = False


def pre_init():
    # 获取目标服务器
    hostname = input(‘hostname: ‘).strip()
    if len(hostname) == 0:
        print(‘hostname required‘.center(40, ‘*‘))
        exit(1)
    # 获取用户名
    default_username = getpass.getuser()
    username = input(‘username: ‘).strip()
    if len(username) == 0:
        username = default_username

    return (username, hostname)


class JumpClient:
    def __init__(self, username, hostname):
        self.username = username
        self.hostname = hostname
        self.tran = paramiko.Transport((self.hostname, 22))
        self.tran.start_client()
        # 判断，若为linux 则调用linux初始化方法；若为windows 则调用windows初始化方法
        try:
            if linux_client:
                self.linux_init()
            else:
                self.windows_init()
        except Exception as ex:
            print(ex)
        else:
            self.chan = self.tran.open_session()
            self.chan.get_pty()
            self.chan.invoke_shell()

    def linux_init(self):
        # 获取验证类型；默认为密码认证
        default_auth_type = ‘p‘
        auth_type = input(‘Auth type [password(p)/key(k)], defulat is [p]: ‘).strip()
        if len(auth_type) == 0:
            auth_type = default_auth_type

        # 载入私钥；判断位置、判断是否需要密码
        if auth_type == ‘k‘:
            default_key = os.path.join(os.environ.get(‘HOME‘), ‘.ssh‘, ‘id_rsa‘)
            auth_key = input(‘Auth key path, defulat is {}: ‘.format(default_key))
            if len(auth_key) == 0:
                auth_key = default_key
            # 若私钥没有密码，则直接载入；若有密码，则输入密码
            try:
                key = paramiko.RSAKey.from_private_key_file(auth_key)
            except paramiko.PasswordRequiredException:
                passwd = getpass.getpass(‘RSA key\‘s  passwd: ‘).strip()
                key = paramiko.RSAKey.from_private_key_file(auth_key, password=passwd)
            finally:
                self.tran.auth_publickey(self.username, key)

        if auth_type == ‘p‘:
            server_passwd = getpass.getpass(‘password: ‘).strip()
            self.tran.auth_password(self.username, server_passwd)

    def linux_run(self):
        # 获取原先 tty终端属性
        oldtty = termios.tcgetattr(sys.stdin)
        try:
            # 设置新的 tty终端属性
            tty.setraw(sys.stdin.fileno())
            tty.setcbreak(sys.stdin.fileno())
            self.chan.settimeout(0.0)

            while True:
                r_lst, w_lst, e_lst = select.select([self.chan, sys.stdin, ], [], [], 1)
                if self.chan in r_lst:
                    try:
                        data = http://www.mamicode.com/self.chan.recv(1024).decode()
                        if len(data) == 0:
                            print(‘\r\n*** EOF ***\r\n‘)
                            break
                        else:
                            sys.stdout.write(data)  # 不用print，因为末尾会自动追加换行
                            sys.stdout.flush()
                    except Exception as ex:
                        print(ex)

                if sys.stdin in r_lst:
                    data = http://www.mamicode.com/sys.stdin.read(1)  # 读取一个字符就发送
                    self.chan.sendall(data)
        finally:
            # 恢复原先 tty终端属性
            termios.tcsetattr(sys.stdin, termios.TCSADRAIN, oldtty)

    def windows_init(self):
        user_pass = input(‘password: ‘).strip()
        self.tran.auth_password(self.username, user_pass)

    def windows_run(self):
        # 子线程的任务：从传入的socket对象中，不断接受数据（如果有数据的话）并打印出来
        def function(sk):
            while True:
                try:
                    ret = sk.recv(1024).decode()
                    if len(ret) == 0:
                        break
                    else:
                        sys.stdout.write(ret)
                        sys.stdout.flush()
                except Exception as ex:
                    print(ex)
        # 接受数据：开启一个子线程，并指定任务
        t = threading.Thread(target=function, args=(self.chan,))
        t.start()
        # 发送数据：主线程本身去完成
        try:
            while True:
                cmd = sys.stdin.read(1)
                self.chan.send(cmd)
        except Exception as ex:
            print(ex)

    def close_terminal(self):
        self.chan.close()
        self.tran.close()


if __name__ == ‘__main__‘:
    user, host = pre_init()  # 获取要登录的目标服务器
    obj = JumpClient(user, host)
    if linux_client:
        obj.linux_run()
    else:
        obj.windows_run()
    # 关闭链接
    obj.close_terminal()











 

jumpserver_core


联系
我们