在 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;
}

Answer 2

给出的代码和命令行参数没有段错误。但是，如果命令行参数省略分隔符、在第二个数字之前包含一个空格或完全省略任何参数，那么它将失败。

以下将防止错误输入导致运行时错误：

if( argc > 1 )
{
    strcpy( arr, argv[1]);

    int firstNumber = 0 ;
    int secondNumber = 0 ;
    token = strtok(arr, "v,V");
    if( token != NULL )
    {
        firstNumber = atoi(token) ;
        token = strtok(NULL, "v,V") ;
        if( token != NULL )
        {
            secondNumber= atoi(token); 
        }
    }
}

Answer 3

你必须在函数的开头定义 var ：

关注code 可以工作：

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

int main(int argc, char *argv[]){

    char *token;
    char arr[200];
    int firstNumber;
    int secondNumber;

    strcpy( arr, argv[1]);

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

    return 0;

}

Answer 4

您的输入中只有一个"V"。对 strtok() 的第二次调用没有找到 V 并返回 NULL。