编写伪装成 TTY 的程序答案

【问题标题】：Write program that pretends to be a TTY编写伪装成 TTY 的程序
【发布时间】：2017-02-25 08:16:26
【问题描述】：

我正在编写一个程序，该程序从标准输入读取输入，操作输入，并将输出写入标准输出。然而，许多程序检查标准输入是终端还是管道（通过调用像isatty 这样的函数），并以不同的方式生成输出。如何让我的程序伪装成 TTY？

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

请注意，我问的是编程问题，而不是工具。所以，像 script 或 unbuffer 这样的东西不是我想要的。

【问题讨论】：

This Unix & Linux Stack Exchange question 可能更相关。您真正要搜索的术语是 pseudo tty（或 pty）。
我知道，但如果我错了，请纠正我。 Unix Stacke Exchange 用于 unix 工具，而不是编程问题。
目的是保存从任何程序读取的颜色代码。
对 U&L SE 的看法是正确的，但已接受答案中的第一点将为您提供足够的提示以帮助您继续前进 (reading the pty manual page)。
也许这个：github.com/kr/pty?

标签： c linux macos go tty

【解决方案1】：

以下是在 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;
      }
    }
  }
}

【讨论】：