Question

我正在尝试使用ioctl()来检索终端的宽度，但是在管道传输或重定向到stdin时它不起作用。

我已经通过解析tput cols的结果来解决这个问题，但是使用外部命令感觉很脏。另外，我认为这会降低Windows的可移植性，因为Windows不使用兼容bourne的外壳？

main.c

// tput method
char res[10];

FILE cmd = popen("tput cols", "r");
fgets(res, 10 - 1, cmd);
pclose(cmd);

unsigned short term_cols = atoi(res);
printf("Term width (tput): %d\n", term_cols);

// ioctl method
struct winsize ws;
if (ioctl(STDIN_FILENO, TIOCGWINSZ, &ws) == 0)
{
  printf("Term width (ioctl): %d\n", ws.ws_col);
}
else
{
  printf("Failed to retrieve term width from ioctl()");
}

输出

$ bin/main  
Term width (tput): 84  
Term width (ioctl): 84

$ echo "test" | bin/main  
Term width (tput): 84  
Failed to retrieve term width from ioctl()

我已尝试在代码开始时fflush(stdin);，但没有任何区别。这是对ioctl()的限制吗？还是有办法解决？

Answer 1

您可能正在打印未初始化变量的值。您的代码不会检查ioctl是否成功，如果失败，则不会影响ws。

修复：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>

...
if (ioctl(STDIN_FILENO, TIOCGWINSZ, &ws) == -1) {
    fprintf(stderr, "can't get the window size of stdin: %s\n", strerror(errno));
    exit(EXIT_FAILURE);
}

当您将某些内容通过管道传输到程序中时，stdin不是指终端，而是管道。管道没有窗口大小。这就是TIOCGWINSZ在这里失败的原因。

便携式解决方案似乎是：

const char *term = ctermid(NULL);
if (!term[0]) {
    fprintf(stderr, "can't get the name of my controlling terminal\n");
    exit(EXIT_FAILURE);
}
int fd = open(term, O_RDONLY);
if (fd == -1) {
    fprintf(stderr, "can't open my terminal at %s: %s\n", term, strerror(errno));
    exit(EXIT_FAILURE);
}
if (ioctl(fd, TIOCGWINSZ, &ws) == -1) {
    fprintf(stderr, "can't get the window size of %s: %s\n", term, strerror(errno));
    exit(EXIT_FAILURE);
}
close(fd);

在管道输送到stdin时使用ioctl填充winsize结构

1 个答案: