从异步子进程获取实时输出

2022-03-25 00:00:00 python python-asyncio

问题描述

我正在尝试使用Python asyncio子进程启动交互式SSH会话并自动输入密码。实际用例并不重要,但它有助于说明我的问题。这是我的代码:

    proc = await asyncio.create_subprocess_exec(
        'ssh', 'user@127.0.0.1',
        stdout=asyncio.subprocess.PIPE,
        stderr=asyncio.subprocess.STDOUT,
        stdin=asyncio.subprocess.PIPE,
    )

    # This loop could be replaced by async for, I imagine
    while True:
        buf = await proc.stdout.read()
        if not buf:
            break
        print(f'stdout: { buf }')

我预计它的工作方式类似于异步流,其中我可以创建两个任务/子例程/未来,一个用于侦听StreamReader(在本例中由proc.stdout给出),另一个用于写入StreamWriter(proc.stdin)。

但是,它并没有像预期的那样工作。ssh命令的前几行输出被直接打印到终端,直到它到达密码提示符(或主机密钥提示符,视情况而定)并等待手动输入。我希望能够读取前几行,检查它是要求输入密码还是主机提示,并相应地写入StreamReader。

它唯一一次运行print(f'stdout: { buf }')行是在我按Enter之后,当它打印时,很明显";:B‘主机密钥验证失败。 ‘&Q;。

我也尝试了推荐的proc.communicate(),虽然不如使用StreamReader/Writer整洁,但有相同的问题:等待手动输入时执行冻结。

这究竟应该如何工作?如果不是我想象的那样,何乐而不为呢?有没有什么方法可以不用求助于线程中的某种繁忙循环来实现这一点呢?

PS:我解释使用ssh只是为了清楚起见。我最终使用plink来实现我想要的功能,但是我想了解如何使用python来运行任意命令。


解决方案

这不是异步CIO特有的问题。ssh进程不与stdinstdout流交互,而是直接访问TTY device,以确保密码输入得到正确保护。

您有三个选项可以解决此问题:

  • 不要使用ssh,而是使用其他SSH客户端,一个不需要TTY控制的客户端。对于Ayncio,您可以使用asyncssh library。该库直接实现SSH协议,因此不需要单独的进程,并且它直接接受用户名和密码凭据。

  • 为SSH提供pseudo-tty对话,这是您的Python程序控制之一。pexpect library为您提供了一个高级API,可用于完全控制ssh命令。

  • 设置另一个密码提示符供ssh使用。如果没有TTY,ssh程序可以通过SSH_ASKPASS环境变量让其他处理密码输入。大多数ssh版本都非常picky about when they'll accept SSH_ASKPASS however,您也需要设置DISPLAY,使用-n命令行开关ssh和使用setsid command在新会话中运行ssh,并断开任何TTY。

    我以前介绍过如何在answer to a question about git and ssh中将SSH_ASKPASS与Asyncio一起使用。

阻力最小的方法是使用pExpect,因为它本地支持异步(任何接受async_=True的方法都可以用作协程):

import pexpect

proc = pexpect.spawn('ssh user@127.0.0.1')
await child.expect('password:', timeout=120, async_=True) 
child.sendline(password_for_user)

相关文章