paramiko修改本分源码
一、获取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服务器上测试
这样输入的命令已经可以看到了。