C语言实现head/tail命令时atoi引发段错误问题求助

你的程序崩溃的核心原因是getopt选项字符串定义错误，以及对应选项的参数处理逻辑不匹配：

1. 你写的选项字符串"cn:"含义是：

   • c：该选项不需要跟随参数（后面没有冒号:）
   • n:：该选项必须跟随参数（后面有冒号:）
     但你在处理case 'c'时直接访问了optarg，而当选项无参数时，optarg的值是未定义的，访问它会触发内存非法访问，直接导致程序崩溃。

2. 从head/tail命令的实际需求来看，-c选项是需要指定字节数的，所以你需要给选项字符串里的c后面加上冒号，明确它需要参数。

修复后的代码 #

#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <unistd.h>

const int BUFFER_SIZE = 100;

int main(int argc, char *argv[]) {
    int c = 10;  // 默认显示10字节
    int n = 10;  // 默认显示10行，原代码设为0不符合head默认行为，这里调整
    int opt;
    // 修改选项字符串为"c:n:"，表示-c和-n都需要参数
    while ((opt = getopt(argc, argv, "c:n:")) != -1) {
        switch(opt) {
        case 'c':
            c = atoi(optarg);
            break;
        case 'n':
            n = atoi(optarg);
            break;
        case '?':
            // 处理无效选项或缺少参数的情况，比如用户只输入-c没给参数
            fprintf(stderr, "用法: %s [-c 字节数] [-n 行数] [文件名]\n", argv[0]);
            exit(EXIT_FAILURE);
        }
    }
    printf("c = %d & n = %d\n", c, n);
    
    // 后续可添加文件读取、内容输出的逻辑
    return 0;
}

额外注意事项 #

• 必须处理getopt返回'?'的情况：当用户输入无效选项，或者需要参数的选项（比如-c）未提供参数时，getopt会返回'?'，此时应打印用法提示并退出。
• 原代码中n的初始值设为0不符合head命令的默认行为，默认应该显示10行，所以调整为10。
• 后续添加文件处理逻辑时，要考虑用户未指定文件名的情况（此时读取标准输入）。

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