输出重定向至日志文件时终端宽度获取失败的修复方案

当标准输出（stdout）被重定向到管道或文件时，它不再关联终端设备，ioctl(STDOUT_FILENO, TIOCGWINSZ)会调用失败——因为该命令仅对终端设备有效。我们可以通过多渠道尝试获取宽度+默认 fallback 的方式解决这个问题：

修改后的代码 #

#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/ioctl.h>
#include <unistd.h>
#include <string.h>

size_t get_terminal_width(const size_t fixed_size) {
    struct winsize w;
    // 1. 尝试从标准输出获取
    if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &w) != -1 && w.ws_col > 0) {
        return (w.ws_col > fixed_size ? w.ws_col - fixed_size : 1);
    }
    // 2. 尝试从标准输入获取（重定向时stdin可能仍关联终端）
    if (ioctl(STDIN_FILENO, TIOCGWINSZ, &w) != -1 && w.ws_col > 0) {
        return (w.ws_col > fixed_size ? w.ws_col - fixed_size : 1);
    }
    // 3. 尝试从标准错误获取（多数重定向场景下stderr仍关联终端）
    if (ioctl(STDERR_FILENO, TIOCGWINSZ, &w) != -1 && w.ws_col > 0) {
        return (w.ws_col > fixed_size ? w.ws_col - fixed_size : 1);
    }
    // 4. 读取shell导出的COLUMNS环境变量
    char* cols_env = getenv("COLUMNS");
    if (cols_env != NULL) {
        long cols = strtol(cols_env, NULL, 10);
        if (cols > 0) {
            return (cols > fixed_size ? cols - fixed_size : 1);
        }
    }
    // 5. 所有方法失败，返回默认宽度（80列是终端常见默认值）
    size_t default_width = 80;
    return (default_width > fixed_size ? default_width - fixed_size : 1);
}

int main() {
    printf("%zu\n", get_terminal_width(50));
    return 0;
}

关键优化点 #

• 多渠道尝试：依次检查stdout、stdin、stderr，覆盖不同重定向场景下的终端关联情况
• 环境变量兼容：读取COLUMNS变量，适配shell主动导出的终端宽度设置
• 优雅降级：移除原代码中直接终止程序的逻辑，所有方法失败时返回合理默认值，保证程序正常运行
• 有效性校验：对获取到的终端宽度值做合法性检查，避免无效值导致错误

测试场景验证 #

• 直接运行./a.out：仍能正常获取当前终端宽度
• 输出重定向到文件./a.out >> b.txt：会从stderr或环境变量获取宽度，失败则返回默认值（如80-50=30）
• 管道输出./a.out |& tee a.txt：若stderr仍关联终端，会从stderr获取宽度；否则 fallback 到环境变量或默认值

内容的提问来源于stack exchange，提问作者Harry