重写head命令时-n选项行数输出异常的技术求助

你的-n选项代码存在几个关键错误，导致无法正确输出指定行数，以下是问题点和修正方案：

核心错误点 #

• 指针sizeof误用：sizeof(buff)获取的是指针本身的大小（通常8字节），而非你malloc分配的1024字节，这会导致扫描范围错误。
• 忽略实际读取字节数：read返回的szt是本次实际读到的字节数，循环中j应该遍历到szt而非缓冲区总大小，否则会访问未初始化的内存。
• 输出逻辑缺失：你仅在找到第n+1个换行时才输出一次，前面的行完全没有输出，这是最致命的逻辑错误。
• 语法错误：open(file.txt, O_RDONLY)中的文件名未加引号，正确写法应为open("file.txt", O_RDONLY)。
• 内存泄漏：malloc的缓冲区未执行free操作。

修正后的-n选项代码示例 #

#define BUFF_SIZE 1024

// 变量定义与初始化
int lines_read = 0;
int fd = open("file.txt", O_RDONLY);
if (fd == -1) {
    perror("open failed");
    exit(EXIT_FAILURE);
}

char *buff = malloc(BUFF_SIZE);
if (!buff) {
    perror("malloc failed");
    close(fd);
    exit(EXIT_FAILURE);
}

ssize_t szt;
while ((szt = read(fd, buff, BUFF_SIZE)) > 0) {
    int output_pos = 0;
    for (int j = 0; j < szt; ++j) {
        if (buff[j] == '\n') {
            lines_read++;
            // 输出当前行内容（从上次输出结束位置到当前换行）
            write(STDOUT_FILENO, buff + output_pos, j - output_pos + 1);
            output_pos = j + 1;
            
            if (lines_read >= n) {
                // 达到指定行数，直接终止所有循环
                goto cleanup;
            }
        }
    }
    // 若缓冲区未扫完所有行（比如最后一行无换行），输出剩余内容
    if (lines_read < n) {
        write(STDOUT_FILENO, buff + output_pos, szt - output_pos);
    }
}

cleanup:
free(buff);
close(fd);

补充说明 #

1. 用goto跳出多层循环是C语言处理这类场景的常规写法，比嵌套break更清晰。
2. 每次扫描到换行时，仅输出当前未输出的行内容，避免重复输出或遗漏。
3. 增加了open和malloc的错误处理，提升程序健壮性。
4. 标准head命令还有其他可选参数，比如-q（不显示文件名）、-v（强制显示文件名）、-z（处理NUL结尾的行），可在核心逻辑稳定后扩展实现。

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