学会使用getopt函数[通俗易懂]

大家好，又见面了，我是你们的朋友全栈君。

简介

getopt函数是命令行参数解析函数，在平时阅读源码的时候经常遇到，很有必要对其总结一下，做个记录!

命令行参数各组成部分的名称

先来了解下命令行参数各组成部分的名称。直接上图：

学会使用getopt函数[通俗易懂]

非常清楚，命令行参数由Command name，Option，Option argument 以及Operands组成。
Command name不用多说，就是程序的名称。
操作对象Operands又称nonoption argument。
重点就是这里的选项Option了，它决定着程序的行为，而选项参数Option argument是选项Option所需要的信息。

getopt()函数

#include <unistd.h>

int getopt(int argc, char * const argv[],
                  const char *optstring);

extern char *optarg;
extern int optind, opterr, optopt;

#include <getopt.h>

int getopt_long(int argc, char * const argv[],
                  const char *optstring,
                  const struct option *longopts, int *longindex);

int getopt_long_only(int argc, char * const argv[],
                  const char *optstring,
                  const struct option *longopts, int *longindex);

getopt()函数将传递给mian()函数的argc，argv作为参数，同时接受字符串参数optstring — optstring是由选项Option字母组成的字符串。关于optstring的格式规范简单总结如下：

（1）单个字符，表示该选项Option不需要参数。

（2）单个字符后接一个冒号”:“，表示该选项Option需要一个选项参数Option argument。选项参数Option argument可以紧跟在选项Option之后，或者以空格隔开。选项参数Option argument的首地址赋给optarg。

（3）单个字符后接两个冒号”::“，表示该选项Option的选项参数Option argument是可选的。当提供了Option argument时，必须紧跟Option之后，不能以空格隔开，否则getopt()会认为该选项Option没有选项参数Option argument，optarg赋值为NULL。相反，提供了选项参数Option argument，则optarg指向Option argument。

为了使用getopt()，我们需要在while循环中不断地调用直到其返回-1为止。每一次调用，当getopt()找到一个有效的Option的时候就会返回这个Option字符，并设置几个全局变量。
getopt()设置的全局变量包括：

char *optarg — 当匹配一个选项后，如果该选项带选项参数，则optarg指向选项参数字符串；若该选项不带选项参数，则optarg为NULL；若该选项的选项参数为可选时，optarg为NULL表明无选项参数，optarg不为NULL时则指向选项参数字符串。

int optind — 下一个待处理元素在argv中的索引值。即下一次调用getopt的时候，从optind存储的位置处开始扫描选项。当getopt()返回-1后，optind是argv中第一个Operands的索引值。optind的初始值为1。

int opterr — opterr的值非0时，在getopt()遇到无法识别的选项，或者某个选项丢失选项参数的时候，getopt()会打印错误信息到标准错误输出。opterr值为0时，则不打印错误信息。

int optopt — 在上述两种错误之一发生时，一般情况下getopt()会返回’?’，并且将optopt赋值为发生错误的选项。

使用getopt()时，会犯的错误无外乎有两个：无法识别的选项（Invalid option）和丢失选项参数（Missing option argument）
通常情况下，getopt()在发现这两个错误时，会打印相应的错误信息，并且返回字符”?” 。例如，遇见无法识别的选项时会打印”invalid option”，发现丢失参数时打印”option requires an argument”。但是当设置opterr为0时，则不会打印这些信息，因此为了便于发现错误，默认情况下，opterr都是非零值。

如果你想亲自处理这两种错误的话，应该怎么做呢？首先你要知道什么时候发生的错误是无法识别的选项，什么时候发生的错误是丢失选项参数。如果像上面描述的那样，都是返回字符”?”的话，肯定是无法分辨出的。有什么办法吗？有！ getopt()允许我们设置optstring的首字符为冒号”:”，在这种情况下，当发生无法识别的选项错误时getopt()返回字符”?”，当发生丢失选项参数错误时返回字符”:”。这样我们就可以很轻松地分辨出错误类型了，不过代价是getopt()不会再打印错误信息了，一切事物都由我们自己来处理了。

getopt()扫描模式

关于getopt()的扫描模式，man文档说的不是很清楚，这里通过实例来阐释，尽量做到简单明了，一看即懂。
getopt()的默认模式扫描模式是这样的：getopt()从左到右按顺序扫描argv[1]到argv[argc-1]，并将选项Option和选项参数Option argument按它们在命令行上出现的次序放到argv数组的左边，而那些Operands则按它们出现的次序放在argv的后边。也就是说，getopt()在默认扫描模式下，会重新排序argv数组。
来看个实例，假设我们调用getopt(argc, argv, “ab:c:de::”);
从前面的知识点，我们很容易得出结论：选项a，d是不需要选项参数的，选项b，c需要选项参数，而选项e的选项参数是可选的，即如果提供选项e的选项参数的话，那么选项参数必须紧跟选项e之后，不能以空格隔开。

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

static int cnt = 1;

static void print(int optc, int argc, char *argv[], int optind)
{
    int i;

    printf("%02d: optc - '%c', argv: ", cnt++, optc);
    for (i = 0; i < argc; i++) {
        printf("%s ", argv[i]);
    }
    printf("---- optind = %d\n", optind);
}

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

    print('0', argc, argv, optind);

    while ((optc = getopt(argc, argv, ":ab:c:de::")) != -1) {
        print(optc, argc, argv, optind);
        switch (optc) {
            default:
                break;
        }
    }

    print('0', argc, argv, optind);

    exit(EXIT_SUCCESS);
}

以下是实际的运行情况：

学会使用getopt函数[通俗易懂]

当然在真实情况下，一个程序很少需要这么多的Operands，这里只是为了更清楚地演示getopt()是如何扫描命令行参数的。
扫描过程中，要时刻铭记optind是下一个待处理元素在argv中的索引，当遇到Operands的时候则跳过，optind数值增加跳过的Operands个数。好，现在我们根据这些规则，详细分析下刚刚程序的扫描过程：
第一行：即getopt()扫描重排序之前，可以看到optind的值默认被初始化为1。
第二行：getopt()首先从operand1开始扫描，发现operand1是Operands，则跳过，optind增加1等于2指向-a，继续扫描。扫描到-a时发现是有效选项，则optind增加1等于3指向operand2，然后返回选项a。
第三行：在继续扫描前，getopt()重新排序argv数组，将“-a”和“operand1”的位置互换。继续扫描，发现operand2是Operands，跳过，optind增加1等于4指向-b，继续扫描。发现-b是有效选项，因为选项b需要参数，因此把barg的首地址赋给optarg，optind增加2等于6指向operand3，返回选项b。
第四行：在继续扫描前，getopt()重新排序argv数组，将“-b barg”和“operand1 operand2”的位置互换。继续扫描，发现operand3是Operands，跳过，optind增加1等于7指向-c，继续扫描。扫描到-c是发现是有效的选项，因为选项c跟选项b一样，都需要参数，因此处理过程是一样的，把carg的首地址赋给optarg，optind增加2等于9指向operand4，返回选项c。
第五行：在继续扫描前，getopt()重新排序argv数组，将“-c carg”和“operand1 operand2 operand3”的位置互换。继续扫描，发现operand4是Operands，跳过，optind增加1等于10指向-d，继续扫描。扫描到-d时发现是有效选项，因为选项d不需要参数，因此直接optind增加1等于11指向operand5，返回选项d。
第六行：在继续扫描前，getopt()重新排序argv数组，将“-d”和“operand1 operand2 operand3 operand4”的位置互换。继续扫描，发现operand5是Operands，跳过，optind增加1等于12指向operand6，继续扫描。扫描到operand6时发现依然是Operands，跳过，optind增加1等于13指向-e，继续扫描。扫描到-e时发现是有效选项，因为后面的operand7与-e之间有间隔，因此这里选项e没有参数。optind增加1等于14指向operand7，返回选项e。
第七行：在继续扫描前，getopt()重新排序argv数组，将“-e”和“operand1 operand2 operand3 operand4 operand5 operand6”的位置互换。继续扫描，发现operand7是Operands，跳过，optind增加1等于15指向argv[argc]，即NULL。至此扫描完毕，getopt()重新设置optind为8，是其指向第一个Operands，即operand1，最后返回-1停止扫描。

通常情况下（即opterr等于1时），getopt()使用这种扫描模式就很好，可以给命令行操作带来很大的灵活性，为什么这么说？让我们举个例子：

例如编译上面的程序getopt.c, 我们可以gcc -g -o getopt getopt.c，也可以把-g放到最后，这样gcc -o getopt getopt.c -g，在这种模式下都是没有问题的，因为getopt()不会在遇到第一个Operands（这里的Operands是”getopt.c“）的情况下就停止扫描选项，而是边扫描选项边重新排序，直到整个命令行参数扫描完为止，因此不会漏掉选项-g。

但是有的时候，你可能希望你的程序接收的参数中Operands必须放在最后，也就是遇到第一个Operands就立即停止参数扫描。例如常用的系统命令env就是这样处理的。env命令的作用是使指定的程序在修改过的新环境变量中运行。典型用法如下：
env – myprog arg1 清空环境变量，运行程序myprog，程序参数为arg1
env – PATH=/bin:/usr/bin myprog arg1 清空环境变量，修改PATH环境变量，运行程序myprog，程序参数为arg1
env -u IFS PATH=/bin:/usr/bin myprog arg1 删除环境变量IFS，修改PATH环境变量，运行程序myprog，程序参数为arg1
可以通过将optstring的首字符设置为‘+’来改变getopt()的扫描模式。我们可以修改前面的程序为getopt(argc, argv, “+ab:c:de::”);然后再一同样的方式调用程序，如下：

再来看一个调用实例：

可以看到，在这种扫描模式下，的确在遇到第一个Operands时就立即停止扫描，忽略后续所有命令行参数，也不会进行重排序操作。

此外，还有最后一种扫描模式是通过将optstring首字符设置为“-”来实现的。我们将程序改为getopt(argc, argv, “-ab:c:de::”);先来看下运行结果：

可以看到在这种模式下，getopt()会按顺序扫描整个命令行参数，遇到有效选项时会正常处理，而遇到Operands时却是这样处理的：返回1，optarg赋值为Operands的首地址。

至此，我们就介绍完了getopt()的所有扫描模式。需要补充的两点是：
1. 在后两种扫描模式的情况下，如果还需要自己来区分无效选项和丢失选项参数这两种错误的话，那么应该将optstring的第二个字符设置为字符‘：’。
2. 不管在哪种扫描模式下，当遇到字符串“–”时都会停止扫描，后续的命令行参数当作Operands来处理。所以如何删除文件名以‘-’起始的文件？例如当前目录下有个文件的名字是“-foo”，如何删除呢？有两种办法：1. rm ./-foo 2. rm — -foo