global_conf.c

/*
 * DPVS is a software load balancer (Virtual Server) based on DPDK.
 *
 * Copyright (C) 2017 iQIYI (www.iqiyi.com).
 * All Rights Reserved.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 */
#include <fcntl.h>
#include <sys/stat.h>
#include "global_conf.h"

static void log_current_time(void)
{
    time_t t = time(0);
    char buf[256];
    strftime(buf, sizeof(buf), "%Y/%m/%d %X %A",localtime(&t));
    RTE_LOG(INFO, CFG_FILE, "load dpvs configuation file at %s\n", buf);
}

static int set_log_level(char *log_level)
{
    if (!log_level)
        rte_log_set_global_level(RTE_LOG_DEBUG);
    else if (!strncmp(log_level, "EMERG", strlen("EMERG")))
        rte_log_set_global_level(RTE_LOG_EMERG);
    else if (!strncmp(log_level, "ALERT", strlen("ALERT")))
        rte_log_set_global_level(RTE_LOG_ALERT);
    else if (!strncmp(log_level, "CRIT", strlen("CRIT")))
        rte_log_set_global_level(RTE_LOG_CRIT);
    else if (!strncmp(log_level, "ERR", strlen("ERR")))
        rte_log_set_global_level(RTE_LOG_ERR);
    else if (!strncmp(log_level, "WARNING", strlen("WARNING")))
        rte_log_set_global_level(RTE_LOG_WARNING);
    else if (!strncmp(log_level, "NOTICE", strlen("NOTICE")))
        rte_log_set_global_level(RTE_LOG_NOTICE);
    else if (!strncmp(log_level, "INFO", strlen("INFO")))
        rte_log_set_global_level(RTE_LOG_INFO);
    else if (!strncmp(log_level, "DEBUG", strlen("DEBUG")))
        rte_log_set_global_level(RTE_LOG_DEBUG);
    else {
        RTE_LOG(WARNING, CFG_FILE, "%s: illegal log level: %s\n",
                __func__, log_level);
        return EDPVS_INVAL;
    }
    return EDPVS_OK;
}

static FILE *g_log_stream;
static int set_log_file(const char *log_file)
{
    if (log_file == NULL)
        return EDPVS_NOTEXIST;

    if (g_log_stream) {
        fclose(g_log_stream);
        g_log_stream = NULL;
    }

    g_log_stream = fopen(log_file, "a+");
    if (g_log_stream == NULL) {
        RTE_LOG(WARNING, CFG_FILE, "%s: illegal log file: %s -- %s\n",
                __func__, log_file, strerror(errno));
        return EDPVS_INVAL;
    }

    if (rte_openlog_stream(g_log_stream)) {
        RTE_LOG(WARNING, CFG_FILE, "%s: fail to set log stream to %s\n",
                __func__, log_file);
        return EDPVS_DPDKAPIFAIL;
    }
    
    log_current_time();
    return EDPVS_OK;
}

static void log_level_handler(vector_t tokens)
{
    char *log_level = set_value(tokens);
    assert(log_level);
    RTE_LOG(INFO, CFG_FILE, "log_level = %s\n", log_level);
    set_log_level(log_level);
    FREE_PTR(log_level);
}

static void log_file_handler(vector_t tokens)
{
    char *log_file = set_value(tokens);
    assert(log_file);
    RTE_LOG(INFO, CFG_FILE, "log_file = %s\n", log_file);
    set_log_file(log_file);
    FREE_PTR(log_file);
}

void install_global_keywords(void)
{
    install_keyword_root("global_defs", NULL);
    install_keyword("log_level", log_level_handler, KW_TYPE_NORMAL);
    install_keyword("log_file", log_file_handler, KW_TYPE_NORMAL);

}

int global_conf_init(void)
{
    return EDPVS_OK;
}

int global_conf_term(void)
{
    if (g_log_stream) {
        fclose(g_log_stream);
        g_log_stream = NULL;
    }
    return EDPVS_OK;
}