paramiko修改本分源码

Posted 2020-10-22 Bigberg

一、获取paramiko源码

　　环境：Python3

　　下载地址：https://github.com/paramiko/paramiko

　　使用的是 demos这个文件夹

二、修改部分源码用以登入

　　2.1 windwos使用paramiko

　　切换到demos文件夹下　

G:\\python\\untitled\\study13\\demos>python demo.py


#报错
File "G:\\python\\untitled\\study13\\demos\\interactive.py", line 84, in writeall
sys.stdout.write(data)

TypeError: write() argument must be str, not bytes


# python3中socket传输是bytes类型，而这边接收的是str类型

　　修改报错的  interactive.py , line 84

    def writeall(sock):
        while True:
            data = sock.recv(256)
            if not data:
                sys.stdout.write(\'\\r\\n*** EOF ***\\r\\n\\r\\n\')
                sys.stdout.flush()
                break
            sys.stdout.write(data.decode())  # 将原本的data 解码
            sys.stdout.flush()

# data.decode() 即可

　　改完之后，输入ip 和 密码可以登入了

　　

 

 三、修改部分源码以记录输入的命令

　　从上图可以看出，通过paramiko中的demo.py 连接服务器，执行的命令是从“Here we go” 后面正式开始的，也就是在此之后的命令是通过paramiko封装的，用以发往不同的服务器。所以我们需要找到这个位置，查看它们的源码。

 chan = t.open_session()
    chan.get_pty()
    chan.invoke_shell()
    print(\'*** Here we go!\\n\')
    interactive.interactive_shell(chan)
    chan.close()
    t.close()


# 可以看到  print(\'*** Here we go!\\n\') 后面
# 运行的一个函数 interactive.interactive_shell(chan)
# 所以要去看这个函数写的什么

　　修改interactive_shell(chan) 中 posix_shell(chan)函数, 以记录所有输入的字符（命令）

        cmd = []  # 定义一个空列表，记录所有输入的字符
        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(\'\\r\\n*** EOF\\r\\n\')
                        break
                    sys.stdout.write(x)
                    sys.stdout.flush()
                except socket.timeout:
                    pass
            if sys.stdin in r:
                x = sys.stdin.read(1)
                if len(x) == 0:
                    break
                if x == "\\r":   # 回车表示一条命令输入完成
                    cmd_str = "".join(cmd)  # 拼接命令
                    print(\'\\n\' + \'-->\' + cmd_str)
                    cmd = []  # 清空cmd列表
                else:
                    cmd.append(x)  # 记录输入的字符
                chan.send(x)

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

　　在linux服务器上测试

　　

　　这样输入的命令已经可以看到了。