在 C++ 中使用 GNU getopt 来读取带有和不带参数的命令行

Question

我正在使用带有 :: 函数的 GNU 库，因此它应该接受我来自命令行 -p 和 -pc 的输入。但是当我通过 -p 时，它会给我一个分段错误。我必须在不带参数的情况下传递 -p，然后从命令行传递参数“c”。我的代码如下。请提出建议。

int main(int argc, char **argv) {
    char op= '\0';
    char arg = '\0';
    while( (arg = getopt(argc, argv, "p::")) != -1 ){
        switch (arg){
            case 'p': op = *optarg;
                if (op=='c')
                    flagp = true;
                else if (op=='\0')
                    flago = true;
                else
                    cout<<"Invalid parameter"<<endl;
        }
    }
}

Answer 1

不带选项处理-p时，getopt()设置optarg为NULL，op = *optarg指令导致段错误。您需要检查 optarg 是否为空。

#include <stdio.h>
#include <getopt.h>

int main(int argc, char **argv) {
    int flagp = 0;
    int flago = 0;
    char op= '\0';
    char arg = '\0';
    while( (arg = getopt(argc, argv, "p::")) != -1 ){
        printf("arg=%c\n", arg);
        switch (arg){
            case 'p': 
                if (optarg != NULL)
                {
                    op = *optarg;
                    if (op=='c')
                        flagp = 1;
                    else if (op=='\0')
                        flago = 1;
                    else
                        fprintf(stderr,"Invalid parameter\n");
                }
                else
                {
                    printf("optarg is NULL\n");
                    flago = 1;
                }
        }
    }

    printf("ending...\n");

    printf("flagp=%d\nflago=%d\n", flagp, flago);

    return 0;
}

输出：

>small -p -pc
arg=p
optarg is NULL
arg=p
ending...
flagp=1
flago=1