python subprocess.Popen 和后台 ssh 端口转发

Question

我需要具有本质上模拟用户使用 ssh 创建端口转发的功能。所以，本质上应该是这样的： - 执行 ssh -f -N -L 10000:gateway:11000 localhost - 如果该命令有输出，则向用户显示并泵送用户的输入作为响应 - 结束 我在下面想出的代码几乎满足了我的需要：

    ssh_proc = Popen(['ssh', '-f', '-N', '-L', '10000:gateway:11000', 'localhost'], stdin=PIPE, stdout=PIPE)
stdoutdata, stderrdata = ssh_proc.communicate()

但问题是它永远不会完成。我看到命令已执行并创建了转发隧道，但 communications() 仍然卡住。我可以使用 Ctrl+C 打破这种情况并得到以下结果：

    ^CTraceback (most recent call last):
  File "sshman.py", line 278, in <module>
    add(args.remote_host, args.remote_port, args.local_port, args.local_host)
  File "sshman.py", line 125, in add
    stdoutdata, stderrdata = ssh_proc.communicate()
  File "/usr/lib64/python2.7/subprocess.py", line 740, in communicate
    return self._communicate(input)
  File "/usr/lib64/python2.7/subprocess.py", line 1256, in _communicate
    stdout, stderr = self._communicate_with_poll(input)
  File "/usr/lib64/python2.7/subprocess.py", line 1310, in _communicate_with_poll
    ready = poller.poll()
KeyboardInterrupt

由于我将 -f 与 ssh 命令一起使用，因此它应该分叉连接。有没有办法在 communications() 完成时中断它，或者是否有更优雅的解决方案？

预先感谢您的任何意见/建议/评论。

Answer 1

我想解决方案很简单

ssh_proc = Popen(['ssh', '-f', '-N', '-L', '10000:gateway:11000', 'localhost'], stdin=PIPE, stdout=PIPE)
stat = ssh_proc.poll()
while stat == None:
    stat = ssh_proc.poll()

Answer 2

当尝试设置 ssh 隧道时，我遇到了一个相对问题，communicate() 将永远挂起，我发现它从 stderr 和 stdout 读取，因此，如果您的命令没有向 stderr 写入任何内容，那么它会挂起，因为它正在等待从 stderr (fd = 2) 读取某些内容。
因此，作为解决方案，您可以使用 os.set_blocking(ssh_proc.stderr.fileno(), False)：

# Start the process.
ssh_proc = Popen(...)
# Wait till it's done.
ssh_proc.wait()
# Disable block for read.
os.set_blocking(ssh_proc.stderr.fileno(), False)
os.set_blocking(ssh_proc.stdout.fileno(), False)
# Read stdout and stderr.
err = ssh_proc.stderr.readlines()
out = ssh_proc.stdout.readlines()
# Print stderr and stdout.
if out:
    out = out[0]
    print(out)
if err:
    err = err[0]
    print(err)
if cmd.returncode == 0:
    return 0

这样，即使没有从 stdout 或 stderr 读取任何内容，communicate() 函数也不会阻止进程。