如何让getopt()同时支持合并参数与单字符独立参数

我需要程序支持以下两种命令格式：

• ./myProgram -f required_arg1 -v required_arg2
• ./myProgram -fv required_arg1 required_arg2

两种格式中，required_arg1都属于-f，required_arg2都属于-v。但运行./myProgram -fv required_arg1 required_arg2时，v会被当作-f的参数，无法让-f和-v各自获取对应的optarg。当前代码如下：

int opt;
int fflag = 0;
int iflag = 0;

while ((opt = getopt(argc, argv, "f:v:i:")) != -1) {        
    switch (opt) {
        case 'f':

            if (fflag) {
                cerr << argv[0] << ": cannot have more than one -f in same command.\n";
                return false;
            }
            else {
                fflag = 1;
            }
            
            if (fflag && iflag) {
                cerr << argv[0] << ": cannot have -i and -f in the same command.\n";
                return false;
            }
            
            cout << "arg for -f: " << optarg << "\n";
            break;
        case 'i':
            iflag = 1;

            if (fflag && iflag) {
                cerr << argv[0] << ": cannot have -i and -f in the same command.\n";
                return false;
            }

            cout << "arg for -i: " << optarg << "\n";
            break;
        case 'v':
            cout << "arg for -v: " << optarg << "\n";
            break;
        default:
            break;
    }
}

解决方案 #

问题根源在于getopt的默认行为：当选项字符串中f:表示-f需要参数时，它会优先把当前选项后的紧跟字符当作参数（比如-fv中的v），而非将v视为独立选项。要修正这个问题，我们可以在解析-f时手动检查参数是否为其他选项字符，若为则调整解析逻辑：

int opt;
int fflag = 0;
int iflag = 0;

while ((opt = getopt(argc, argv, "f:v:i:")) != -1) {        
    switch (opt) {
        case 'f': {
            // 检查是否是合并选项的情况（参数是单个选项字符）
            bool merged_opt = false;
            char next_opt = 0;
            if (optarg != nullptr && strlen(optarg) == 1 && strchr("vi", optarg[0]) != nullptr) {
                merged_opt = true;
                next_opt = optarg[0];
                // 把下一个argv元素作为-f的参数
                if (optind >= argc) {
                    cerr << argv[0] << ": option -f requires an argument.\n";
                    return false;
                }
                optarg = argv[optind];
                optind++;
            }

            if (fflag) {
                cerr << argv[0] << ": cannot have more than one -f in same command.\n";
                return false;
            }
            fflag = 1;
            
            if (fflag && iflag) {
                cerr << argv[0] << ": cannot have -i and -f in the same command.\n";
                return false;
            }
            
            cout << "arg for -f: " << optarg << "\n";

            // 如果是合并选项，手动触发下一个选项的解析
            if (merged_opt) {
                opt = next_opt;
                continue; // 回到switch处理下一个选项
            }
            break;
        }
        case 'i':
            iflag = 1;

            if (fflag && iflag) {
                cerr << argv[0] << ": cannot have -i and -f in the same command.\n";
                return false;
            }

            cout << "arg for -i: " << optarg << "\n";
            break;
        case 'v':
            cout << "arg for -v: " << optarg << "\n";
            break;
        default:
            cerr << argv[0] << ": invalid option -" << (char)optopt << "\n";
            return false;
    }
}

逻辑说明 #

1. 解析-f时，检查optarg是否为单个字符且是其他有效选项（v或i），判断是否为合并选项场景。
2. 若是合并选项，则将下一个argv元素作为-f的参数，并记录后续要解析的选项字符。
3. 完成-f的处理后，手动将opt设置为记录的选项字符，让循环继续解析该选项（比如v），从而实现-fv等价于-f -v的效果。

内容的提问来源于stack exchange，提问作者Tyler