使用 atoi 将命令行参数转换为 int，仅返回输入的第一个数字

Question

我必须使用命令行参数为我的Uni分配设置蛇游戏的地图。我们没有特别告诉我们使用atoi来帮助将命令行参数从字符串转换为int，但是我认为atoi的简单性质会做到这一点。在测试时，我发现它只是占据了第一个数字。

int main(int argc, char *argv[])
{
    int isUserInput;
    char arg1, arg2, arg3;

    arg1 = argv[1][0];
    arg2 = argv[2][0];
    arg3 = argv[3][0];

    isUserInput = checkUserArg(arg1, arg2, arg3);

int checkUserArg(char arg1, char arg2, char arg3)
{
    int validation;
    int rowMap, colMap, snakeLength;

    rowMap = atoi(&arg1);
    colMap = atoi(&arg2);
    snakeLength = atoi(&arg3);

    if ((rowMap < 5) || (colMap < 5) || (snakeLength < 3))
    {
        validation = FALSE;
    }
    else if ((rowMap >= 5) || (colMap >= 5) || (snakeLength >= 3))
    {
        if (snakeLength < colMap)
        {
            validation = TRUE;
        }
        else
        {
            validation = FALSE;
        }
    }
    else
    {
        validation = FALSE;
    }
    
    return validation;
}

用户必须输入3个命令行参数（./文件num1 num2 num2 num3）。我使用atoi将字符串命令行参数转换为int，但是在测试时，我将数字打印回了，它不会仅转换第二个数字，例如1 -9起作用，但是从10个开始的任何内容都只显示了第一个数字。

对为什么这是什么想法？ 干杯。

Answer 1

代码中有多个问题：

• atoi仅使用第一个数字，因为您明确提取第一个数字并将其传递给char。该函数调用实际上具有＆amp; arg1是单个char> char的地址，而不是null终结者C字符串的地址。

  。

• checkuserarg使用atoi转换参数，如果转换的值超过类型int，则具有未定义的行为。建议使用strtol，并允许进行更精细的检查。

• checkuserarg应通过指针参数将转换值返回到呼叫者。

• checkuserarg的第二个测试是多余的：如果第一个测试为false，则第二个测试中的所有3个比较都是正确的。

• 而不是true和false，您应该使用＆lt; stdbool.h＆gt;的定义。。

这是修改版本：

#include <errno.h>
#include <limits.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>

bool convertArg(const char *arg, int *vp) {
    char *p;
    long num;

    errno = 0;
    num = strtol(arg, &p, 10);
    if (errno || p == arg || *p != '\0' || num < INT_MIN || num > INT_MAX) {
        *vp = 0;
        return false;
    } else {
        *vp = (int)num;
        return true;
    }
}

int main(int argc, char *argv[]) {
    int rowMap, colMap, snakeLength;

    if (argc != 4) {
        fprintf(stderr, "program needs 3 arguments\n");
        return 1;
    }
    if (!converArg(argv[1], &rowMap) || rowMap < 5) {
        fprintf(stderr, "invalid rowMap argument\n");
        return 1;
    }
    if (!converArg(argv[2], &colMap) || colMap < 5) {
        fprintf(stderr, "invalid colMap argument\n");
        return 1;
    }
    if (!converArg(argv[3], &snakeLength) || snakeLength < 3 || snakeLength >= colMap) {
        fprintf(stderr, "invalid snakeLength argument\n");
        return 1;
    }
    [...]
}

Answer 2

单个字符不是字符串。字符串是一个字符数组，末尾带有空终止符。

你应该这样做：

bool checkUserArg (const char* arg1, const char* arg2, const char* arg3);

const 因为我们不应该修改参数。现在这个函数可以直接使用参数调用atoi。

但是 atoi 在设计上是一个损坏的函数，永远不应该使用，因为它没有明确定义的错误处理。直接在 argv 上使用它是危险的。您应该始终使用strtol而不是atoi，它是一个更安全、更强大的函数。

例子：

#include <stdlib.h>
#include <stdbool.h>

int main (int argc, char *argv[])
{
    if(argc != 4)
    {
      // error handling!
    }

    if(checkUserArg(argv[1], argv[2], argv[3]) == false)
    {
      /* error handling */
    }  
...

bool checkUserArg (const char* arg1, const char* arg2, const char* arg3)
{
   const char* endptr;
   ...
   rowMap = strtol(arg1, &endptr, 10); // 10 for decimal base
   if(endptr == arg1) // if these compare equal, the conversion failed
   {
     return false;
   }
   ...

   return true;
}