310 lines
9.4 KiB
C
310 lines
9.4 KiB
C
|
/*
|
||
|
* Copyright (C) 2019-2023 by Sukchan Lee <acetcom@gmail.com>
|
||
|
*
|
||
|
* This file is part of Open5GS.
|
||
|
*
|
||
|
* This program is free software: you can redistribute it and/or modify
|
||
|
* it under the terms of the GNU Affero General Public License as published by
|
||
|
* the Free Software Foundation, either version 3 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.
|
||
|
*
|
||
|
* You should have received a copy of the GNU General Public License
|
||
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
|
*/
|
||
|
|
||
|
#include "ogs-app.h"
|
||
|
|
||
|
int __ogs_app_domain;
|
||
|
|
||
|
static int read_config(void);
|
||
|
static int parse_config(void);
|
||
|
|
||
|
int ogs_app_initialize(
|
||
|
const char *version, const char *default_config,
|
||
|
const char *const argv[])
|
||
|
{
|
||
|
int rv, opt;
|
||
|
ogs_getopt_t options;
|
||
|
struct {
|
||
|
char *config_file;
|
||
|
char *log_file;
|
||
|
char *log_level;
|
||
|
char *domain_mask;
|
||
|
} optarg;
|
||
|
|
||
|
ogs_core_initialize();
|
||
|
ogs_app_setup_log();
|
||
|
|
||
|
ogs_app_context_init();
|
||
|
ogs_app_config_init();
|
||
|
ogs_app()->version = version;
|
||
|
|
||
|
/**************************************************************************
|
||
|
* Stage 1 : Command Line Options
|
||
|
*/
|
||
|
memset(&optarg, 0, sizeof(optarg));
|
||
|
|
||
|
ogs_getopt_init(&options, (char**)argv);
|
||
|
while ((opt = ogs_getopt(&options, "c:l:e:m:")) != -1) {
|
||
|
switch (opt) {
|
||
|
case 'c':
|
||
|
optarg.config_file = options.optarg;
|
||
|
break;
|
||
|
case 'l':
|
||
|
optarg.log_file = options.optarg;
|
||
|
break;
|
||
|
case 'e':
|
||
|
optarg.log_level = options.optarg;
|
||
|
break;
|
||
|
case 'm':
|
||
|
optarg.domain_mask = options.optarg;
|
||
|
break;
|
||
|
case '?':
|
||
|
default:
|
||
|
ogs_assert_if_reached();
|
||
|
return OGS_ERROR;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**************************************************************************
|
||
|
* Stage 2 : Load Configuration File
|
||
|
*/
|
||
|
if (optarg.config_file)
|
||
|
ogs_app()->file = optarg.config_file;
|
||
|
else
|
||
|
ogs_app()->file = default_config;
|
||
|
|
||
|
rv = read_config();
|
||
|
if (rv != OGS_OK) return rv;
|
||
|
|
||
|
rv = parse_config();
|
||
|
if (rv != OGS_OK) return rv;
|
||
|
|
||
|
/**************************************************************************
|
||
|
* Stage 3 : Initialize Default Memory Pool
|
||
|
*/
|
||
|
ogs_pkbuf_default_create(&ogs_global_conf()->pkbuf_config);
|
||
|
|
||
|
/**************************************************************************
|
||
|
* Stage 4 : Setup LOG Module
|
||
|
*/
|
||
|
if (optarg.log_file)
|
||
|
ogs_app()->logger.file = optarg.log_file;
|
||
|
|
||
|
if (ogs_app()->logger.file) {
|
||
|
if (ogs_log_add_file(ogs_app()->logger.file) == NULL) {
|
||
|
ogs_fatal("cannot open log file : %s",
|
||
|
ogs_app()->logger.file);
|
||
|
return OGS_ERROR;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (optarg.domain_mask)
|
||
|
ogs_app()->logger.domain = optarg.domain_mask;
|
||
|
|
||
|
if (optarg.log_level)
|
||
|
ogs_app()->logger.level = optarg.log_level;
|
||
|
|
||
|
rv = ogs_log_config_domain(
|
||
|
ogs_app()->logger.domain, ogs_app()->logger.level);
|
||
|
if (rv != OGS_OK) return rv;
|
||
|
|
||
|
/**************************************************************************
|
||
|
* Stage 5 : Setup Database Module
|
||
|
*/
|
||
|
if (ogs_env_get("DB_URI"))
|
||
|
ogs_app()->db_uri = ogs_env_get("DB_URI");
|
||
|
|
||
|
/**************************************************************************
|
||
|
* Stage 6 : Print Banner
|
||
|
*/
|
||
|
if (ogs_app()->version) {
|
||
|
ogs_log_print(OGS_LOG_INFO,
|
||
|
"Open5GS daemon %s\n\n", ogs_app()->version);
|
||
|
|
||
|
ogs_info("Configuration: '%s'", ogs_app()->file);
|
||
|
|
||
|
if (ogs_app()->logger.file) {
|
||
|
ogs_info("File Logging: '%s'", ogs_app()->logger.file);
|
||
|
|
||
|
if (ogs_app()->logger.level)
|
||
|
ogs_info("LOG-LEVEL: '%s'", ogs_app()->logger.level);
|
||
|
|
||
|
if (ogs_app()->logger.domain)
|
||
|
ogs_info("LOG-DOMAIN: '%s'", ogs_app()->logger.domain);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**************************************************************************
|
||
|
* Stage 7 : Queue, Timer and Poll
|
||
|
*/
|
||
|
ogs_app()->queue = ogs_queue_create(ogs_app()->pool.event);
|
||
|
ogs_assert(ogs_app()->queue);
|
||
|
ogs_app()->timer_mgr = ogs_timer_mgr_create(ogs_app()->pool.timer);
|
||
|
ogs_assert(ogs_app()->timer_mgr);
|
||
|
ogs_app()->pollset = ogs_pollset_create(ogs_app()->pool.socket);
|
||
|
ogs_assert(ogs_app()->pollset);
|
||
|
|
||
|
return rv;
|
||
|
}
|
||
|
|
||
|
void ogs_app_terminate(void)
|
||
|
{
|
||
|
ogs_app_config_final();
|
||
|
ogs_app_context_final();
|
||
|
|
||
|
ogs_pkbuf_default_destroy();
|
||
|
|
||
|
ogs_core_terminate();
|
||
|
}
|
||
|
|
||
|
static int read_config(void)
|
||
|
{
|
||
|
FILE *file;
|
||
|
yaml_parser_t parser;
|
||
|
yaml_document_t *document = NULL;
|
||
|
|
||
|
ogs_assert(ogs_app()->file);
|
||
|
|
||
|
file = fopen(ogs_app()->file, "rb");
|
||
|
if (!file) {
|
||
|
ogs_fatal("cannot open file `%s`", ogs_app()->file);
|
||
|
return OGS_ERROR;
|
||
|
}
|
||
|
|
||
|
ogs_assert(yaml_parser_initialize(&parser));
|
||
|
yaml_parser_set_input_file(&parser, file);
|
||
|
|
||
|
document = calloc(1, sizeof(yaml_document_t));
|
||
|
if (!yaml_parser_load(&parser, document)) {
|
||
|
ogs_fatal("Failed to parse configuration file '%s'", ogs_app()->file);
|
||
|
switch (parser.error) {
|
||
|
case YAML_MEMORY_ERROR:
|
||
|
ogs_error("Memory error: Not enough memory for parsing");
|
||
|
break;
|
||
|
case YAML_READER_ERROR:
|
||
|
if (parser.problem_value != -1)
|
||
|
ogs_error("Reader error - %s: #%X at %zd", parser.problem,
|
||
|
parser.problem_value, parser.problem_offset);
|
||
|
else
|
||
|
ogs_error("Reader error - %s at %zd", parser.problem,
|
||
|
parser.problem_offset);
|
||
|
break;
|
||
|
case YAML_SCANNER_ERROR:
|
||
|
if (parser.context)
|
||
|
ogs_error("Scanner error - %s at line %zu, column %zu"
|
||
|
"%s at line %zu, column %zu", parser.context,
|
||
|
parser.context_mark.line+1,
|
||
|
parser.context_mark.column+1,
|
||
|
parser.problem, parser.problem_mark.line+1,
|
||
|
parser.problem_mark.column+1);
|
||
|
else
|
||
|
ogs_error("Scanner error - %s at line %zu, column %zu",
|
||
|
parser.problem, parser.problem_mark.line+1,
|
||
|
parser.problem_mark.column+1);
|
||
|
break;
|
||
|
case YAML_PARSER_ERROR:
|
||
|
if (parser.context)
|
||
|
ogs_error("Parser error - %s at line %zu, column %zu"
|
||
|
"%s at line %zu, column %zu", parser.context,
|
||
|
parser.context_mark.line+1,
|
||
|
parser.context_mark.column+1,
|
||
|
parser.problem, parser.problem_mark.line+1,
|
||
|
parser.problem_mark.column+1);
|
||
|
else
|
||
|
ogs_error("Parser error - %s at line %zu, column %zu",
|
||
|
parser.problem, parser.problem_mark.line+1,
|
||
|
parser.problem_mark.column+1);
|
||
|
break;
|
||
|
default:
|
||
|
/* Couldn't happen. */
|
||
|
ogs_assert_if_reached();
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
free(document);
|
||
|
yaml_parser_delete(&parser);
|
||
|
ogs_assert(!fclose(file));
|
||
|
return OGS_ERROR;
|
||
|
}
|
||
|
|
||
|
ogs_app()->document = document;
|
||
|
|
||
|
yaml_parser_delete(&parser);
|
||
|
ogs_assert(!fclose(file));
|
||
|
|
||
|
return OGS_OK;
|
||
|
}
|
||
|
|
||
|
static int context_prepare(void)
|
||
|
{
|
||
|
#define USRSCTP_LOCAL_UDP_PORT 9899
|
||
|
ogs_app()->usrsctp.udp_port = USRSCTP_LOCAL_UDP_PORT;
|
||
|
|
||
|
return OGS_OK;
|
||
|
}
|
||
|
|
||
|
static int context_validation(void)
|
||
|
{
|
||
|
return OGS_OK;
|
||
|
}
|
||
|
|
||
|
static int parse_config(void)
|
||
|
{
|
||
|
int rv;
|
||
|
yaml_document_t *document = NULL;
|
||
|
ogs_yaml_iter_t root_iter;
|
||
|
|
||
|
document = ogs_app()->document;
|
||
|
ogs_assert(document);
|
||
|
|
||
|
rv = context_prepare();
|
||
|
if (rv != OGS_OK) return rv;
|
||
|
|
||
|
ogs_yaml_iter_init(&root_iter, document);
|
||
|
while (ogs_yaml_iter_next(&root_iter)) {
|
||
|
const char *root_key = ogs_yaml_iter_key(&root_iter);
|
||
|
ogs_assert(root_key);
|
||
|
if (!strcmp(root_key, "db_uri")) {
|
||
|
ogs_app()->db_uri = ogs_yaml_iter_value(&root_iter);
|
||
|
} else if (!strcmp(root_key, "logger")) {
|
||
|
ogs_yaml_iter_t logger_iter;
|
||
|
ogs_yaml_iter_recurse(&root_iter, &logger_iter);
|
||
|
while (ogs_yaml_iter_next(&logger_iter)) {
|
||
|
const char *logger_key = ogs_yaml_iter_key(&logger_iter);
|
||
|
ogs_assert(logger_key);
|
||
|
if (!strcmp(logger_key, "file")) {
|
||
|
ogs_app()->logger.file = ogs_yaml_iter_value(&logger_iter);
|
||
|
} else if (!strcmp(logger_key, "level")) {
|
||
|
ogs_app()->logger.level =
|
||
|
ogs_yaml_iter_value(&logger_iter);
|
||
|
} else if (!strcmp(logger_key, "domain")) {
|
||
|
ogs_app()->logger.domain =
|
||
|
ogs_yaml_iter_value(&logger_iter);
|
||
|
}
|
||
|
}
|
||
|
} else if (!strcmp(root_key, "global")) {
|
||
|
rv = ogs_app_parse_global_conf(&root_iter);
|
||
|
if (rv != OGS_OK) {
|
||
|
ogs_error("ogs_global_conf_parse_config() failed");
|
||
|
return rv;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
rv = context_validation();
|
||
|
if (rv != OGS_OK) return rv;
|
||
|
|
||
|
return OGS_OK;
|
||
|
}
|
||
|
|
||
|
void ogs_app_setup_log(void)
|
||
|
{
|
||
|
ogs_log_install_domain(&__ogs_app_domain, "app", ogs_core()->log.level);
|
||
|
}
|