使用 os.forkpty() 创建一个伪终端以 ssh 到远程服务器并与其通信

Question

我正在尝试编写一个 python 脚本，它可以 ssh 到远程服务器，并可以从 python 客户端执行 ls,cd 等简单命令。但是，在成功 ssh 到服务器后，我无法读取伪终端的输出。任何人都可以在这里帮助我，以便我可以在服务器上执行一些命令。

这是示例代码：

#!/usr/bin/python2.6
import os,sys,time,thread
pid,fd = os.forkpty()
if pid == 0:
    os.execv('/usr/bin/ssh',['/usr/bin/ssh','user@host',])
    sys.exit(0)
else:
    output = os.read(fd,1024)
    print output
    data = output
    os.write(fd,'password\n')
    time.sleep(1)
    output = os.read(fd,1024)
    print output
    os.write(fd,'ls\n')
    output = os.read(fd,1024)
    print output

示例输出：

user@host's password: 

Last login: Wed Aug 24 03:16:57 2011 from 1x.x.x.xxxx

-bash: ulimit: open files: cannot modify limit: Operation not permitted
host: /home/user>ls

Answer 1

我建议尝试模块 pexpect，它是专门为此类事情构建的（通过伪 TTY 与其他应用程序交互），或者 Fabric，它是为此类事情更抽象地构建的（自动化远程服务器上的系统管理任务）使用 SSH）。

pexpect：http://pypi.python.org/pypi/pexpect/

结构：http://docs.fabfile.org/en/1.11/

Answer 2

如前所述，最好使用公钥。当我正常使用它们时，我已经更改了你的程序，以便它在这里工作。

#!/usr/bin/python2.6
import os,sys,time,thread
pid,fd = os.forkpty()
if pid == 0:
    os.execv('/usr/bin/ssh',['/usr/bin/ssh','localhost',])
    sys.exit(0)
else:
    output = os.read(fd,1024)
    print output
    os.write(fd,'ls\n')
    time.sleep(1) # this is new!
    output = os.read(fd,1024)
    print output

通过添加 sleep(1)，我给远程主机（或者，在我的例子中，不是那么远程的主机）有时间处理 ls 命令并生成其输出。

如果您发送ls并立即阅读，您只会阅读当前存在的内容。也许你应该循环阅读。

或者你应该这样做：

import subprocess
sp = subprocess.Popen(("ssh", "localhost", "ls"), stdout=subprocess.PIPE)
print sp.stdout.read()