linux 我如何将初始输入导入交互式流程?

bwntbbo3  于 2023-02-21  发布在  Linux
关注(0)|答案(7)|浏览(169)

我希望能够在交互式进程的启动中注入一个初始命令,这样我就可以做这样的事情:

echo "initial command" | INSERT_MAGIC_HERE some_tool

tool> initial command 

[result of initial command] 

tool> [now I type an interactive command]

什么不起作用:

  • 仅仅通过管道输入初始命令是不起作用的,因为这会导致stdin没有连接到终端
  • 写入/dev/pts/[number]会将输出发送到终端,而不是像从终端发送一样将输入发送到进程

什么会,但有缺点:

  • 创建一个命令来派生一个子节点,写入它的stdin,然后转发它自己的stdin中的所有内容。缺点-终端控制的东西(比如行模式和字符模式)不起作用。也许我可以做一些伪终端的代理?
  • 修改xterm的版本(无论如何,我会为这个任务启动一个),其中包含一个命令行选项,可以在遇到所需的提示字符串后注入额外的命令。
  • 创建我试图运行的工具的修改版本,使其接受命令行上的初始命令。中断标准安装。

(The顺便说一句,当前感兴趣的工具是android的adb shell --我想在手机上打开一个交互式shell,自动运行一个命令,然后进行交互式会话)

70gysomp

70gysomp1#

你不需要写一个新的工具来转发stdin--已经写了一个(cat):

(echo "initial command" && cat) | some_tool

这样做的缺点是将管道连接到some_tool,而不是连接到终端。

ve7v8dk2

ve7v8dk22#

公认的答案很简单,而且大多数都是好的。
但它有一个缺点:程序得到一个管道作为它的输入,而不是一个终端。这意味着自动完成将不起作用。在很多情况下,这也禁用了漂亮的输出,我听说一些程序只是拒绝工作,如果stdin不是一个终端。
下面的程序解决了这个问题。它创建了一个伪终端,派生了一个连接到这个伪终端的程序。它首先通过命令行提供额外的输入,然后通过stdin提供用户提供的输入。
例如,ptypipe "import this" python3让Python首先执行“import this”,然后它会把你带到交互式命令提示符,并提供工作完成和其他功能。
类似地,ptypipe "date" bash运行Bash,它执行date,然后给你一个shell,同样,还有工作完成、彩色提示符等等。

#!/usr/bin/env python3

import sys
import os
import pty
import tty
import select
import subprocess

STDIN_FILENO = 0
STDOUT_FILENO = 1
STDERR_FILENO = 2

def _writen(fd, data):
    while data:
        n = os.write(fd, data)
        data = data[n:]

def main_loop(master_fd, extra_input):
    fds = [master_fd, STDIN_FILENO]

    _writen(master_fd, extra_input)

    while True:
        rfds, _, _ = select.select(fds, [], [])
        if master_fd in rfds:
            data = os.read(master_fd, 1024)
            if not data:
                fds.remove(master_fd)
            else:
                os.write(STDOUT_FILENO, data)
        if STDIN_FILENO in rfds:
            data = os.read(STDIN_FILENO, 1024)
            if not data:
                fds.remove(STDIN_FILENO)
            else:
                _writen(master_fd, data)

def main():
    extra_input = sys.argv[1]
    interactive_command = sys.argv[2]

    if hasattr(os, "fsencode"):
        # convert them back to bytes
        # http://bugs.python.org/issue8776
        interactive_command = os.fsencode(interactive_command)
        extra_input = os.fsencode(extra_input)

    # add implicit newline
    if extra_input and extra_input[-1] != b'\n':
        extra_input += b'\n'

    # replace LF with CR (shells like CR for some reason)
    extra_input = extra_input.replace(b'\n', b'\r')

    pid, master_fd = pty.fork()

    if pid == 0:
        os.execlp("sh", "/bin/sh", "-c", interactive_command)

    try:
        mode = tty.tcgetattr(STDIN_FILENO)
        tty.setraw(STDIN_FILENO)
        restore = True
    except tty.error:    # This is the same as termios.error
        restore = False

    try:
        main_loop(master_fd, extra_input)
    except OSError:
        if restore:
            tty.tcsetattr(0, tty.TCSAFLUSH, mode)

    os.close(master_fd)
    return os.waitpid(pid, 0)[1]

if __name__ == "__main__":
    main()

(Note:我担心这个解决方案可能包含死锁。您可能希望以小块的形式提供extra_input以避免死锁)

roqulrg3

roqulrg33#

使用程序“expect”很容易做到这一点,该程序旨在让您编写脚本来与程序交互。
我编写了一个expect脚本bc.exp来启动计算器“bc”,并向它发送命令“obase=16”,使它进入十六进制输出模式,然后将控制权交给我。
脚本(位于名为bc.exp的文件中)为

spawn bc
send "obase=16\n"
interact {
 \003 exit
}

一个人用

expect bc.exp
f45qwnt8

f45qwnt84#

@caf的回答是一个坚实的答案。
我想我会在这里用一个适合我的相关选项来扩展它。
我的<initial command>实际上是一个命令列表,在一个文本文件中找到,因此,我想将它通过管道传输到<some command>stdin的连接。
cat处理得非常好,它接受-stdin读取:

cat init_file - | some_command

这与cafs的回答中讨论的限制相同。

oipij1gg

oipij1gg5#

你也可以(在某些情况下,当从终端运行时)使用tee直接写终端输出,名称tee指的是像T一样,通过stdin到stdout,同时也写文件(/dev/tty,在本例中是终端)。
echo "initial command" | tee /dev/tty | some_tool
https://unix.stackexchange.com/a/178754/89546

3j86kqsm

3j86kqsm6#

如果你在tmux内部运行,你可以告诉它发送密钥。
例如

$ stty -echo; tmux send-keys test; stty echo

将把test放入终端输入。(stty防止在终端上看到键)
或者,如果使用vi模式:

stty -echo
tmux send-keys escape 0
stty echo
read -p "rename: " -e -i 'old name' new_name

这会将escape和0放入输入缓冲区,以将下面的读取行置于命令模式,并将光标移动到0。

a8jjtwal

a8jjtwal7#

也许你可以用a here document把你的输入传递给abd,比如这样(用bc做一个简单的计算)。

[axe@gromp ~]$ bc <<END
> 3 + 4
> END
7

bc会话之后保持打开状态,因此在开始和结束标记之间(“〈〈END”和“END”之间)提供的内容将传递给命令。

相关问题