更改paramiko 源码 记录命令实现堡垒机功能

利用paramiko 下的demo可以很容易的实现记录客户在操作客户机时的命令，修改demosinteractive.py

def posix_shell(chan):
    import select
    oldtty = termios.tcgetattr(sys.stdin)
    try:
        tty.setraw(sys.stdin.fileno())
        tty.setcbreak(sys.stdin.fileno())
        chan.settimeout(0.0)


        #这里就是主要改造的地方了
        cmd_list=[]
           #定义日志输出的位置
        log_file='/home/myshare/demos/test.log'
        f=open(log_file,'rb+')
        while True:
            r, w, e = select.select([chan, sys.stdin], [], [])

            if chan in r:
                try:
                    x = u(chan.recv(1024))
                    if len(x) == 0:
                        sys.stdout.write('
*** EOF
')
                        break
                    sys.stdout.write(x)
                    sys.stdout.flush()
                except socket.timeout:
                    pass
            if sys.stdin in r:
                x = sys.stdin.read(1)
                   #这个x每次敲击键盘就会记录一次，然后就将输入的内容交给输出的地方，这样才完成1个字符1个字符的显示过程。这个地方仔细想一下或者打印结X果就明白什么意思了
                   #这里将字符串放入列表中，然后当用户敲回车的时候在将列表中的字符串拼接起来
                cmd_list.append(x)
                if x == '
':
                    cmd=''.join(cmd_list).replace('
','
')
                    f.write(cmd)
                    f.flush()
                    cmd_list=[]
                if len(x) == 0:
                    break
                chan.send(x)

    finally:
        termios.tcsetattr(sys.stdin, termios.TCSADRAIN, oldtty)

此时方法

[root@controller demos]# ./demo.py root@10.64.8.10