parse_conf.c

#include "parse_conf.h"
#include "common.h"
#include "strutil.h"
#include "configure.h"

/*
 * parseconf_load_setting()函数只是
 * 由parseconf_load_file()函数调用，
 * 并不希望让用户调用，
 * 所以这里parseconf_load_settting()
 * 函数的声明和实现都在源文件中
 */
static void parseconf_load_setting(const char *setting);

//bool类型的配置变量
//这种struct a b;的形式值得借鉴
static struct parseconf_bool_setting
{
    const char *p_setting_name;
    int *p_variable;
}
parseconf_bool_array[] =
{
    { "pasv_enable", &tunable_pasv_enable },
    { "port_enable", &tunable_port_enable },
    { NULL, NULL }
};

static struct parseconf_uint_setting
{
    const char *p_setting_name;
    unsigned int *p_variable;
}
parseconf_uint_array[] =
{
    { "listen_port", &tunable_listen_port },
    { "max_clients", &tunable_max_clients },
    { "max_per_ip", &tunable_max_per_ip },
    { "accept_timeout", &tunable_accept_timeout },
    { "connect_timeout", &tunable_connect_timeout },
    { "idle_session_timeout", &tunable_idle_session_timeout },
    { "data_connection_timeout", &tunable_data_connection_timeout },
    { "local_umask", &tunable_local_umask },
    { "upload_max_rate", &tunable_upload_max_rate },
    { "download_max_rate", &tunable_download_max_rate },
    { NULL, NULL }
};

static struct parseconf_str_setting
{
    const char *p_setting_name;
    const char **p_variable;
}
parseconf_str_array[] =
{
    { "listen_address", &tunable_listen_address },
    { NULL, NULL }
};

//parseconf_load_file()
void parseconf_load_file(const char *path)
{
    FILE *fp = fopen(path, "r");
    if (fp == NULL)
        ERR_EXIT("fopen");

    char setting_line[1024] = {0};
    while (fgets(setting_line, sizeof(setting_line), fp) != NULL)
    {
        if (strlen(setting_line) == 0
                || setting_line[0] == '#'
                || str_all_space(setting_line))
            continue;

        str_trim_crlf(setting_line);
        parseconf_load_setting(setting_line);
        memset(setting_line, 0, sizeof(setting_line));
    }

    fclose(fp);
}

static void parseconf_load_setting(const char *setting)
{
    // 去除左空格
    while (isspace(*setting))
        ++setting;

    char key[128] ={0};
    char value[128] = {0};
    str_split(setting, key, value, '=');
    if (strlen(value) == 0)
    {
        fprintf(stderr, "mising value in config file for: %s\n", key);
        exit(EXIT_FAILURE);
    }


    {
        const struct parseconf_str_setting *p_str_setting = parseconf_str_array;
        while (p_str_setting->p_setting_name != NULL)
        {
            if (strcmp(key, p_str_setting->p_setting_name) == 0)
            {
                const char **p_cur_setting = p_str_setting->p_variable;
                if (*p_cur_setting)
                    free((char*)*p_cur_setting);

                *p_cur_setting = strdup(value);
                return;
            }

            ++p_str_setting;
        }
    }

    {
        const struct parseconf_bool_setting *p_bool_setting = parseconf_bool_array;
        while (p_bool_setting->p_setting_name != NULL)
        {
            if (strcmp(key, p_bool_setting->p_setting_name) == 0)
            {
                str_upper(value);
                if (strcmp(value, "YES") == 0
                        || strcmp(value, "TRUE") == 0
                        || strcmp(value, "1") == 0)
                    *(p_bool_setting->p_variable) = 1;
                else if (strcmp(value, "NO") == 0
                        || strcmp(value, "FALSE") == 0
                        || strcmp(value, "0") == 0)
                    *(p_bool_setting->p_variable) = 0;
                else
                {
                    fprintf(stderr, "bad bool value in config file for: %s\n", key);
                    exit(EXIT_FAILURE);
                }

                return;
            }

            ++p_bool_setting;
        }
    }

    {
        const struct parseconf_uint_setting *p_uint_setting = parseconf_uint_array;
        while (p_uint_setting->p_setting_name != NULL)
        {
            if (strcmp(key, p_uint_setting->p_setting_name) == 0)
            {
                if (value[0] == '0')
                    *(p_uint_setting->p_variable) = str_octal_to_uint(value);
                else
                    *(p_uint_setting->p_variable) = atoi(value);

                return;
            }

            ++p_uint_setting;
        }
    }
}