openairinterface5g/openair-cn/OPENAIRHSS/utils/hss_config.c

/*******************************************************************************

  Eurecom OpenAirInterface
  Copyright(c) 1999 - 2013 Eurecom

  This program is free software; you can redistribute it and/or modify it
  under the terms and conditions of the GNU General Public License,
  version 2, as published by the Free Software Foundation.

  This program is distributed in the hope 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.

  You should have received a copy of the GNU General Public License along with
  this program; if not, write to the Free Software Foundation, Inc.,
  51 Franklin St - Fifth Floor, Boston, MA 02110-1301 USA.

  The full GNU General Public License is included in this distribution in
  the file called "COPYING".

  Contact Information
  Openair Admin: openair_admin@eurecom.fr
  Openair Tech : openair_tech@eurecom.fr
  Forums       : http://forums.eurecom.fr/openairinterface
  Address      : EURECOM, Campus SophiaTech, 450 Route des Chappes
                 06410 Biot FRANCE

*******************************************************************************/

/*! \file hss_config.c
 * \brief Base configuration for the HSS. Parse command line and configuration file
 * \author Sebastien ROUX <sebastien.roux@eurecom.fr>
 * \date 2013
 * \version 0.1
 */

#if HAVE_CONFIG_H
# include "config.h"
#endif

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <getopt.h>

#include "hss_config.h"
#include "hss_parser.h"

/* YACC forward declarations */
extern int  yyparse (struct hss_config_s *hss_config_p);

static int config_parse_command_line(int argc, char *argv[],
                                     hss_config_t *hss_config_p);
static int config_parse_file(hss_config_t *hss_config_p);
static void display_banner(void);
static void config_display(hss_config_t *hss_config_p);

static struct option long_options[] = {
    { "config",  1, 0, 'c' },
    { "help",    0, 0, 'h' },
    { "version", 0, 0, 'v' },
    { 0, 0, 0, 0 },
};

static const char option_string[] = "c:vh";

int config_init(int argc, char *argv[], hss_config_t *hss_config_p)
{
    int ret = 0;

    if (hss_config_p == NULL) {
        return EINVAL;
    }
    if ((ret = config_parse_command_line(argc, argv, hss_config_p)) != 0) {
        return ret;
    }
    display_banner();
    if ((ret = config_parse_file(hss_config_p)) != 0) {
        /* Parsing of the file failed. -> abort */
        abort();
    }
    config_display(hss_config_p);
    return ret;
}

static void display_banner(void)
{
    fprintf(stdout, "==== EURECOM %s v%s ====\n", PACKAGE_NAME, PACKAGE_VERSION);
    fprintf(stdout, "Please report any bug to: %s\n\n", PACKAGE_BUGREPORT);
}

static void usage(void)
{
    display_banner();
    fprintf(stdout, "Usage: openair_hss [options]\n\n");
    fprintf(stdout, "Available options:\n");
    fprintf(stdout, "\t--help\n\t-h\n");
    fprintf(stdout, "\t\tPrint this help and return\n\n");
    fprintf(stdout, "\t--config=<path>\n\t-c<path>\n");
    fprintf(stdout, "\t\tSet the configuration file for hss\n");
    fprintf(stdout, "\t\tSee template in conf dir\n\n");
    fprintf(stdout, "\t--version\n\t-v\n");
    fprintf(stdout, "\t\tPrint %s version and return\n", PACKAGE_NAME);
}

static void config_display(hss_config_t *hss_config_p)
{
    fprintf(stdout, "Configuration\n");
    fprintf(stdout, "* Global:\n");
    fprintf(stdout, "\t- File .............: %s\n", hss_config_p->config);
    fprintf(stdout, "* MYSQL:\n");
    fprintf(stdout, "\t- Server ...........: %s\n", hss_config_p->mysql_server);
    fprintf(stdout, "\t- Database .........: %s\n", hss_config_p->mysql_database);
    fprintf(stdout, "\t- User .............: %s\n", hss_config_p->mysql_user);
    fprintf(stdout, "\t- Password .........: %s\n",
            (hss_config_p->mysql_password == NULL) ? "None" : "*****");
    fprintf(stdout, "* FreeDiameter:\n");
    fprintf(stdout, "\t- Conf file ........: %s\n",
            hss_config_p->freediameter_config);
}

static int config_parse_command_line(int argc, char *argv[],
                                     hss_config_t *hss_config_p)
{
    int c;
    int option_index = 0;

    while ((c = getopt_long (argc, argv, option_string, long_options,
                             &option_index)) != -1) {
        switch(c) {
            case 'c': {
                hss_config_p->config = strdup(optarg);
            }
            break;
            case 'v': {
                /* We display version and return immediately */
                display_banner();
                exit(0);
            }
            break;
            default:
            case 'h': {
                usage();
                exit(0);
            }
            break;
        }
    }
    return 0;
}

static int config_parse_file(hss_config_t *hss_config_p)
{
    extern FILE *yyin;
    int ret = -1;

    if (hss_config_p == NULL) {
        return ret;
    }
    if (hss_config_p->config == NULL) {
        return ret;
    }

    printf("Parsing configuration file: %s\n", hss_config_p->config);

    yyin = fopen(hss_config_p->config, "r");
    if (!yyin) {
        /* We failed to open the file */
        fprintf(stderr, "Unable to open the configuration file: %s (%d:%s)\n",
                hss_config_p->config, errno, strerror(errno));
        return errno;
    }

    /* Call the yacc parser */
    ret = yyparse(hss_config_p);

    /* Close the file descriptor */
    if (fclose(yyin) != 0) {
        fprintf(stderr, "Unable to close the configuration file: %s (%d:%s)\n",
                hss_config_p->config, errno, strerror(errno));
        return errno;
    }
    return ret;
}