在 Go 中获取终端尺寸：使用 cgo 与 ioctl 的实践指南

#include <sys/ioctl.h>
#include <termios.h> // For struct winsize

struct winsize ws; // 或者 struct ttysize ts; 根据系统定义
ioctl(0, TIOCGWINSZ, &ws); // 0 代表标准输入文件描述符

package main

// #include <sys/ioctl.h>
// typedef struct ttysize ttysize; // 假设 ttysize 已定义或等同于 winsize
import "C"

import "fmt"

func GetTerminalSizeProblematic() {
    var ts C.ttysize // 假设 ttysize 是一个有效的 C 结构体
    // C.ioctl(0, C.TIOCGWINSZ, &ts) // 这行代码会因为上述两个限制而编译失败
    fmt.Println("无法直接调用 C.ioctl")
}

#include <stdio.h>
#include <sys/ioctl.h>

int main() {
    printf("TIOCGWINSZ value: 0x%lx\n", TIOCGWINSZ);
    return 0;
}

package main

// #include <sys/ioctl.h>
// typedef struct ttysize ttysize; // 假设 ttysize 已定义
import "C"

// TIOCGWINSZ 是一个宏，我们需要其常量值。
// 此值可能因操作系统而异，例如 Linux 上通常为 0x5413。
const TIOCGWINSZ C.ulong = 0x5413

// 定义 struct ttysize，其结构通常与 struct winsize 相同
// 注意：实际系统中 TIOCGWINSZ 通常使用 struct winsize。
// 这里为了遵循原始问题，使用 ttysize。
struct ttysize {
    unsigned short ts_lines;    // 行数
    unsigned short ts_cols;     // 列数
    unsigned short ts_xpixel;   // X 方向像素（可选）
    unsigned short ts_ypixel;   // Y 方向像素（可选）
};

// 封装 ioctl 函数，使其具有固定参数列表
void myioctl(int fd, unsigned long request, struct ttysize *ts) {
    ioctl(fd, request, ts);
}

package main

/*
#include <sys/ioctl.h>
#include <termios.h> // 通常 struct winsize 在此头文件中

// 为了遵循原始问题，定义 struct ttysize。
// 在大多数现代 Unix 系统上，TIOCGWINSZ 通常与 struct winsize 配合使用。
// struct winsize 的定义通常如下：
// struct winsize {
//     unsigned short ws_row;
//     unsigned short ws_col;
//     unsigned short ws_xpixel;
//     unsigned short ws_ypixel;
// };
// 我们在这里将其命名为 ttysize，以匹配原始问题。
struct ttysize {
    unsigned short ts_lines;
    unsigned short ts_cols;
    unsigned short ts_xpixel;
    unsigned short ts_ypixel;
};

// 封装 ioctl 函数，使其具有固定参数列表，以便 cgo 可以调用。
void myioctl(int fd, unsigned long request, struct ttysize *ts) {
    ioctl(fd, request, ts);
}
*/
import "C"

import (
    "fmt"
    "os"
    "unsafe"
)

// TIOCGWINSZ 是一个宏，我们需要其常量值。
// 此值可能因操作系统而异，例如 Linux 上通常为 0x5413。
// 对于 macOS/BSD 系统，可能需要查找其对应值。
const TIOCGWINSZ C.ulong = 0x5413

// GetTerminalSize 获取当前终端的行数和列数。
// 它通过 cgo 调用 C 语言的 ioctl 系统调用实现。
func GetTerminalSize() (rows, cols int, err error) {
    // 定义 Go 语言中与 C 语言 struct ttysize 对应的结构体。
    // C.struct_ttysize 是 cgo 自动生成的 C 结构体 Go 对应类型。
    var ts C.struct_ttysize

    // 获取标准输入的文件描述符。
    // 在 Unix-like 系统中，文件描述符 0 通常是 stdin。
    fd := C.int(os.Stdin.Fd())

    // 调用封装后的 C 函数 myioctl。
    // unsafe.Pointer(&ts) 将 Go 结构体的地址转换为 C 指针。
    C.myioctl(fd, TIOCGWINSZ, (*C.struct_ttysize)(unsafe.Pointer(&ts)))

    // 从 C 结构体中提取行数和列数。
    rows = int(ts.ts_lines)
    cols = int(ts.ts_cols)
    return rows, cols, nil
}

func main() {
    rows, cols, err := GetTerminalSize()
    if err != nil {
        fmt.Fprintf(os.Stderr, "获取终端尺寸失败: %v\n", err)
        return
    }
    fmt.Printf("终端尺寸: %d 行, %d 列\n", rows, cols)
}