c - 在 C 中分割命令行参数

Question

我想将第一个命令行参数拆分为两个不同的数字。以这种方式运行程序时出现段错误错误:

gcc -ansi main.c -o main
./main 6000V7000

这是源代码:

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

int main(int argc, char *argv[]) {
    char *token;
    char arr[200];
    strcpy(arr, argv[1]);

    token = strtok(arr, "v,V");
    int firstNumber = atoi(token);
    token = strtok(NULL, "v,V");
    int secondNumber = atoi(token);

    return 0;
}

如何解决这个问题？

Answer 1

您不测试是否至少有一个命令行参数，也不测试该参数的长度是否少于 200 个字符，也不测试 strtok 的返回值:您将出现未定义的行为如果命令没有给出任何参数，或者参数不包含任何字符 v、V 或 ,。

如果您使用 gcc -ansi main.c -o main 有效编译程序并使用发布的参数作为 ./main 6000V7000 运行它，您不应该得到段错误...有些事情您没有告诉我们;)

最好避免疯狂的假设:测试意外情况，以便在所有情况下为程序提供定义的行为。

这是使用 sscanf() 解决您的问题的更简单方法:

#include <stdio.h>

int main(int argc, char *argv[]) {
    int a, b;
    if (argc > 1 && sscanf(argv[1], "%d%*1[vV,]%d", &a, &b) == 2)
        printf("a=%d, b=%d\n", a, b);
    return 0;
}