我们可以在 getopt 中添加更多选项吗？

Question

我暂时限制了这段代码，只是为了做一些基本的计算，比如加法和减法，以便了解getopt 的工作原理。

我想要实现的是：./a.out -a 20 20 -s 40 40 [result = 40 和 0]

我是 C 新手，请告诉我代码中的错误。

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

int add(int a, int b) {
    return a + b;
}

int subtract(int a, int b) {
    return a - b;
}

int main(int argc, char *argv[]) {
    FILE *file1 = fopen("Results.txt", "a");
    char ch;
    int res;
    while ((ch = getopt(argc, argv, "a:s:")) != EOF)
        switch (ch) {
          case 'a':
            res = add(atoi(optarg), atoi(argv[3]));
            fprintf(file1, "%i\n", res);
            break;
          case 's':
            res = subtract(atoi(optarg), atoi(argv[3]));
            printf("%i \n", res);
            fprintf(file1, "%i\n", res);
            break;
          default:
            fprintf(stderr, "No such option");
            return 1;
        }
    argc -= optind;
    argv += optind;
    printf("Opind = %i, argc = %i, argv = %i \n", optind, argc, argv);
    fprintf(file1, "\nWritten to file\n");
    fclose(file1);

    return 0;
}   

Answer 1

您的代码中存在多个问题：

• 您应该将ch 定义为int 以适应getopt 的可能返回值。如果argv 数组中没有更多选项，getopt 返回一个 int，它是匹配的选项字符或值 -1。 char 类型在某些平台上默认是无符号的（这是一个明智的选择），因此在这些平台上 ch != EOF 将始终为真。

• 没有更多选项时getopt的返回值是-1，而不是EOF，EOF通常定义为-1但只指定为负数。

  李>

• 您不检查fopen() 是否成功，如果无法创建或打开文件以追加模式写入，则会产生未定义的行为。

• 您没有检查-a 和-s 选项是否有足够的参数。

• add 和subtract 的第二个参数始终是argv[3]。它应该是argv数组中的下一个参数argv[optind]，使用后应该跳过它。

• argv 不能传递给 printf 用于 %i 转换说明符。目前尚不清楚您打算这样做。

这是修改后的版本：

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

int add(int a, int b) {
    return a + b;
}

int subtract(int a, int b) {
    return a - b;
}

int main(int argc, char *argv[]) {
    FILE *file1 = fopen("Results.txt", "a");
    int ch, res;

    if (file1 == NULL) {
        fprintf(stderr, "cannot open Results.txt for appending: %s\n",
                strerror(errno));
        return 1;
    }

    while ((ch = getopt(argc, argv, "a:s:")) != -1) {
        switch (ch) {
          case 'a':
            if (optind >= argc) {
                fprintf(stderr, "not enough arguments\n");
                return 1;
            }
            res = add(atoi(optarg), atoi(argv[optind]));
            optind++;
            //printf("%i\n", res);
            fprintf(file1, "%i\n", res);
            break;
          case 's':
            if (optind >= argc) {
                fprintf(stderr, "not enough arguments\n");
                return 1;
            }
            res = subtract(atoi(optarg), atoi(argv[optind]));
            optind++;
            //printf("%i\n", res);
            fprintf(file1, "%i\n", res);
            break;
          default:
            fprintf(stderr, "No such option");
            return 1;
        }
    }
    argc -= optind;
    argv += optind;
    printf("Opind = %i, argc = %i\n", optind, argc);
    fprintf(file1, "\nWritten to file\n");
    fclose(file1);
    return 0;
}