C：将命令行arg映射到编译时常量

Question

我正在编写一个接受系统资源名称（例如RLIMIT_NOFILE ）并为其打印一些资源限制信息的C程序。

资源常量在<sys/resource.h>中定义，例如

#define RLIMIT_NOFILE   5

我正在寻找一种将命令行参数（例如RLIMIT_NOFILE ）映射到相应数值（例如5）的好方法。

我最初打算做类似的事情：

int resource = -1;
char *resource_names[] = {
    "RLIMIT_NOFILE", "RLIMIT_NPROC", "RLIMIT_RSS"
};

for (i = 0; i < sizeof(resource_names)/sizeof(char *); i++) {
    if (strcmp(argv[1], resource_names[i]) == 0) {
        resource = eval(resource_names[i]);
        break;
    }
}

但是C似乎没有eval东西，即使这样做，编译时常量在运行时也不可用。

现在，我正在做以下事情，但是我很好奇是否有更好的方法。

#include <stdio.h>
#include <string.h>
#include <sys/resource.h>

int main(int argc, char *argv[])
{
    if (argc != 2) {
        printf("Usage: %s <resource>\n", argv[0]);
        return 1;
    }

    char *resource_names[] = {
        "RLIMIT_NOFILE", "RLIMIT_NPROC", "RLIMIT_RSS"
    };
    int resources[] = {
        RLIMIT_NOFILE, RLIMIT_NPROC, RLIMIT_RSS
    };
    int i, resource = -1;

    for (i = 0; i < sizeof(resources)/sizeof(int); i++) {
        if (strcmp(argv[1], resource_names[i]) == 0) {
            resource = resources[i];
            break;
        }
    }
    if (resource == -1) {
        printf("Invalid resource.\n");
        return 1;
    }

    struct rlimit rlim;
    getrlimit(resource, &rlim);
    printf("%s: %ld / %ld\n", argv[1], rlim.rlim_cur, rlim.rlim_max);

    return 0;
}

Answer 1

RLIMIT_x常量都是可以用作数组索引的所有低值整数， 或者 （针对您的问题）使用数组来查找索引，并且该索引将与所需的值相对应。

或者，您可以有一个结构数组，同时包含值和字符串。 就像是

static const struct
{
    int limit;
    char *name;
} rlimits[] = {
    { RLIMIT_NOFILE, "RLIMIT_NOFILE" },
    { RLIMIT_NPROC, "RLIMIT_NPROC" },
    // Etc.
};

然后，很容易遍历数组并将字符串“映射”到值（或相反）。