gpt4 book ai didi

c - 编写伪装为TTY的程序

转载 作者:行者123 更新时间:2023-12-01 14:58:11 26 4
gpt4 key购买 nike

我正在编写一个程序,该程序从stdin读取输入,操纵输入,并将输出写入stdout。但是,许多程序检查stdin是终端还是管道(通过调用 isatty 之类的函数),并生成不同的输出。如何让我的程序假装为TTY?

该解决方案应可在Linux和macOS上使用。生成独立二进制文件的任何编程语言都是可以接受的,但是Go是首选。

请注意,我是在询问编程问题,而不是在寻求工具。因此,像scriptunbuffer这样的东西不是我想要的东西。

最佳答案

以下是完全有效的代码,用于在pty中运行命令并捕获其输出。 (没有您想的那么多行。)

#include <signal.h>
#include <stdlib.h>
#include <sysexits.h>
#include <unistd.h>
#include <util.h>

pid_t child = 0;

void sighandler(int signum) {
if (child > 0) {
killpg(child, signum);
exit(signum);
}
}

// Run a command in a pty.
// Usage: /path/to/this/binary command to run
int main(int argc, char *argv[]) {
if (argc < 2) {
return EX_USAGE;
}

int master;
child = forkpty(&master, NULL, NULL, NULL);

if (child == -1) {
perror("failed to fork pty");
return EX_OSERR;
}

if (child == 0) {
// we're in the child process, so replace it with the command
execvp(argv[1], argv + 1);
perror("failed to execute command");
return EX_OSERR;
}

// trap kill signals and forward them to child process
signal(SIGHUP, sighandler);
signal(SIGINT, sighandler);
signal(SIGTERM, sighandler);

const int buf_size = 1024;
char buf[buf_size];
fd_set fds;
ssize_t bytes_read;

// forward the output continuously
while (1) {
FD_ZERO(&fds);
FD_SET(master, &fds);

if (select(master + 1, &fds, NULL, NULL, NULL) > 0 && FD_ISSET(master, &fds)) {
bytes_read = read(master, buf, buf_size);
if (bytes_read <= 0) {
return EXIT_SUCCESS;
}

if (write(STDOUT_FILENO, buf, bytes_read) != bytes_read) {
perror("failed to write to stdout");
return EX_OSERR;
}
}
}
}

关于c - 编写伪装为TTY的程序,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59796747/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com