More work on main.c to add osmocom functionalitites
Change-Id: I61a5ee2e100f79cef7e42dbd981e308597c30900
This commit is contained in:
parent
50502a77b0
commit
5e3098d56a
|
@ -37,6 +37,9 @@
|
||||||
#include <osmocom/vty/telnet_interface.h>
|
#include <osmocom/vty/telnet_interface.h>
|
||||||
#include <osmocom/vty/logging.h>
|
#include <osmocom/vty/logging.h>
|
||||||
#include <osmocom/vty/command.h>
|
#include <osmocom/vty/command.h>
|
||||||
|
#include <osmocom/vty/misc.h>
|
||||||
|
#include <osmocom/vty/cpu_sched_vty.h>
|
||||||
|
#include <osmocom/vty/ports.h>
|
||||||
|
|
||||||
#include <osmocom/ranap/ranap_common.h> /* ranap_set_log_area() */
|
#include <osmocom/ranap/ranap_common.h> /* ranap_set_log_area() */
|
||||||
|
|
||||||
|
@ -46,44 +49,191 @@
|
||||||
#include <osmocom/hnodeb/vty.h>
|
#include <osmocom/hnodeb/vty.h>
|
||||||
#include <osmocom/hnodeb/hnodeb.h>
|
#include <osmocom/hnodeb/hnodeb.h>
|
||||||
|
|
||||||
|
static const char * const osmohnodeb_copyright =
|
||||||
|
"OsmoHNodeB - Osmocom 3G Home NodeB implementation\r\n"
|
||||||
|
"Copyright (C) 2015 by Daniel Willmann <dwillmann@sysmocom.de\r\n"
|
||||||
|
"Copyright (C) 2021 by sysmocom s.f.m.c. GmbH <info@sysmocom.de>\r\n"
|
||||||
|
"Based on initial work (hnb-test) by Daniel Willmann\r\n"
|
||||||
|
"Contributions by Pau Espin Pedrol\r\n\r\n"
|
||||||
|
"License AGPLv3+: GNU AGPL version 3 or later <http://gnu.org/licenses/agpl-3.0.html>\r\n"
|
||||||
|
"This is free software: you are free to change and redistribute it.\r\n"
|
||||||
|
"There is NO WARRANTY, to the extent permitted by law.\r\n";
|
||||||
|
|
||||||
void *tall_hnb_ctx;
|
void *tall_hnb_ctx;
|
||||||
struct hnb *g_hnb;
|
struct hnb *g_hnb;
|
||||||
|
|
||||||
|
static struct {
|
||||||
|
const char *database_name;
|
||||||
|
const char *config_file;
|
||||||
|
int daemonize;
|
||||||
|
const char *mncc_sock_path;
|
||||||
|
} hnb_cmdline_config = {
|
||||||
|
.database_name = NULL,
|
||||||
|
.config_file = "osmo-hnodeb.cfg",
|
||||||
|
};
|
||||||
|
|
||||||
static struct vty_app_info vty_info = {
|
static struct vty_app_info vty_info = {
|
||||||
.name = "OsmoHNodeB",
|
.name = "OsmoHNodeB",
|
||||||
.version = "0",
|
.version = PACKAGE_VERSION,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
static void print_usage()
|
||||||
|
{
|
||||||
|
printf("Usage: osmo-hnodeb\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
static void print_help()
|
||||||
|
{
|
||||||
|
printf("Some useful options:\n");
|
||||||
|
printf(" -h --help This text.\n");
|
||||||
|
printf(" -d option --debug=DCC:DMM:DRR: Enable debugging.\n");
|
||||||
|
printf(" -D --daemonize Fork the process into a background daemon.\n");
|
||||||
|
printf(" -c --config-file filename The config file to use.\n");
|
||||||
|
printf(" -s --disable-color\n");
|
||||||
|
printf(" -T --timestamp Prefix every log line with a timestamp.\n");
|
||||||
|
printf(" -V --version Print the version of OsmoHNodeB.\n");
|
||||||
|
printf(" -e --log-level number Set a global loglevel.\n");
|
||||||
|
|
||||||
|
printf("\nVTY reference generation:\n");
|
||||||
|
printf(" --vty-ref-mode MODE VTY reference generation mode (e.g. 'expert').\n");
|
||||||
|
printf(" --vty-ref-xml Generate the VTY reference XML output and exit.\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
static void handle_long_options(const char *prog_name, const int long_option)
|
||||||
|
{
|
||||||
|
static int vty_ref_mode = VTY_REF_GEN_MODE_DEFAULT;
|
||||||
|
|
||||||
|
switch (long_option) {
|
||||||
|
case 1:
|
||||||
|
vty_ref_mode = get_string_value(vty_ref_gen_mode_names, optarg);
|
||||||
|
if (vty_ref_mode < 0) {
|
||||||
|
fprintf(stderr, "%s: Unknown VTY reference generation "
|
||||||
|
"mode '%s'\n", prog_name, optarg);
|
||||||
|
exit(2);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case 2:
|
||||||
|
fprintf(stderr, "Generating the VTY reference in mode '%s' (%s)\n",
|
||||||
|
get_value_string(vty_ref_gen_mode_names, vty_ref_mode),
|
||||||
|
get_value_string(vty_ref_gen_mode_desc, vty_ref_mode));
|
||||||
|
vty_dump_xml_ref_mode(stdout, (enum vty_ref_gen_mode) vty_ref_mode);
|
||||||
|
exit(0);
|
||||||
|
default:
|
||||||
|
fprintf(stderr, "%s: error parsing cmdline options\n", prog_name);
|
||||||
|
exit(2);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
static void handle_options(int argc, char **argv)
|
static void handle_options(int argc, char **argv)
|
||||||
{
|
{
|
||||||
while (1) {
|
while (1) {
|
||||||
int idx = 0, c;
|
int idx = 0, c;
|
||||||
|
static int long_option = 0;
|
||||||
static const struct option long_options[] = {
|
static const struct option long_options[] = {
|
||||||
|
{"help", 0, 0, 'h'},
|
||||||
|
{"debug", 1, 0, 'd'},
|
||||||
|
{"daemonize", 0, 0, 'D'},
|
||||||
|
{"config-file", 1, 0, 'c'},
|
||||||
|
{"disable-color", 0, 0, 's'},
|
||||||
|
{"timestamp", 0, 0, 'T'},
|
||||||
|
{"version", 0, 0, 'V' },
|
||||||
|
{"log-level", 1, 0, 'e'},
|
||||||
|
{"vty-ref-mode", 1, &long_option, 1},
|
||||||
|
{"vty-ref-xml", 0, &long_option, 2},
|
||||||
{ "ues", 1, 0, 'u' },
|
{ "ues", 1, 0, 'u' },
|
||||||
{ "gw-addr", 1, 0, 'g' },
|
{ "gw-addr", 1, 0, 'g' },
|
||||||
{ 0, 0, 0, 0 },
|
{ 0, 0, 0, 0 },
|
||||||
};
|
};
|
||||||
|
|
||||||
c = getopt_long(argc, argv, "u:g:", long_options, &idx);
|
c = getopt_long(argc, argv, "hd:Dc:sTVe:u:g:", long_options, &idx);
|
||||||
|
|
||||||
if (c == -1)
|
if (c == -1)
|
||||||
break;
|
break;
|
||||||
|
|
||||||
switch (c) {
|
switch (c) {
|
||||||
|
case 'h':
|
||||||
|
print_usage();
|
||||||
|
print_help();
|
||||||
|
exit(0);
|
||||||
|
case 0:
|
||||||
|
handle_long_options(argv[0], long_option);
|
||||||
|
break;
|
||||||
|
case 'd':
|
||||||
|
log_parse_category_mask(osmo_stderr_target, optarg);
|
||||||
|
break;
|
||||||
|
case 'D':
|
||||||
|
hnb_cmdline_config.daemonize = 1;
|
||||||
|
break;
|
||||||
|
case 'c':
|
||||||
|
hnb_cmdline_config.config_file = optarg;
|
||||||
|
break;
|
||||||
|
case 's':
|
||||||
|
log_set_use_color(osmo_stderr_target, 0);
|
||||||
|
break;
|
||||||
|
case 'T':
|
||||||
|
log_set_print_timestamp(osmo_stderr_target, 1);
|
||||||
|
break;
|
||||||
|
case 'V':
|
||||||
|
print_version(1);
|
||||||
|
exit(0);
|
||||||
|
break;
|
||||||
|
case 'e':
|
||||||
|
log_set_log_level(osmo_stderr_target, atoi(optarg));
|
||||||
|
break;
|
||||||
case 'u':
|
case 'u':
|
||||||
g_hnb->ues = atoi(optarg);
|
g_hnb->ues = atoi(optarg);
|
||||||
break;
|
break;
|
||||||
case 'g':
|
case 'g':
|
||||||
g_hnb->gw_addr = optarg;
|
g_hnb->gw_addr = optarg;
|
||||||
break;
|
break;
|
||||||
|
default:
|
||||||
|
/* catch unknown options *as well as* missing arguments. */
|
||||||
|
fprintf(stderr, "Error in command line options. Exiting.\n");
|
||||||
|
exit(-1);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static void signal_handler(int signum)
|
||||||
|
{
|
||||||
|
fprintf(stdout, "signal %u received\n", signum);
|
||||||
|
|
||||||
|
switch (signum) {
|
||||||
|
case SIGINT:
|
||||||
|
case SIGTERM:
|
||||||
|
/* If SIGTERM was already sent before, just terminate immediately. */
|
||||||
|
if (osmo_select_shutdown_requested())
|
||||||
|
exit(-1);
|
||||||
|
osmo_select_shutdown_request();
|
||||||
|
break;
|
||||||
|
case SIGABRT:
|
||||||
|
/* in case of abort, we want to obtain a talloc report and
|
||||||
|
* then run default SIGABRT handler, who will generate coredump
|
||||||
|
* and abort the process. abort() should do this for us after we
|
||||||
|
* return, but program wouldn't exit if an external SIGABRT is
|
||||||
|
* received.
|
||||||
|
*/
|
||||||
|
talloc_report(tall_vty_ctx, stderr);
|
||||||
|
talloc_report_full(tall_hnb_ctx, stderr);
|
||||||
|
signal(SIGABRT, SIG_DFL);
|
||||||
|
raise(SIGABRT);
|
||||||
|
break;
|
||||||
|
case SIGUSR1:
|
||||||
|
talloc_report(tall_vty_ctx, stderr);
|
||||||
|
talloc_report_full(tall_hnb_ctx, stderr);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
int main(int argc, char **argv)
|
int main(int argc, char **argv)
|
||||||
{
|
{
|
||||||
int rc;
|
int rc;
|
||||||
|
|
||||||
|
/* Track the use of talloc NULL memory contexts */
|
||||||
|
talloc_enable_null_tracking();
|
||||||
|
|
||||||
tall_hnb_ctx = talloc_named_const(NULL, 0, "hnb_context");
|
tall_hnb_ctx = talloc_named_const(NULL, 0, "hnb_context");
|
||||||
|
|
||||||
msgb_talloc_ctx_init(tall_hnb_ctx, 0);
|
msgb_talloc_ctx_init(tall_hnb_ctx, 0);
|
||||||
|
@ -95,36 +245,76 @@ int main(int argc, char **argv)
|
||||||
|
|
||||||
ranap_set_log_area(DRANAP);
|
ranap_set_log_area(DRANAP);
|
||||||
|
|
||||||
log_set_print_filename2(osmo_stderr_target, LOG_FILENAME_NONE);
|
vty_info.copyright = osmohnodeb_copyright;
|
||||||
log_set_use_color(osmo_stderr_target, 0);
|
vty_info.tall_ctx = tall_hnb_ctx;
|
||||||
log_set_print_category(osmo_stderr_target, 0);
|
vty_init(&vty_info);
|
||||||
log_set_print_category_hex(osmo_stderr_target, 0);
|
logging_vty_add_cmds();
|
||||||
|
osmo_talloc_vty_add_cmds();
|
||||||
|
osmo_cpu_sched_vty_init(tall_hnb_ctx);
|
||||||
|
|
||||||
g_hnb = hnb_alloc(tall_hnb_ctx);
|
g_hnb = hnb_alloc(tall_hnb_ctx);
|
||||||
|
|
||||||
vty_init(&vty_info);
|
|
||||||
hnb_vty_init();
|
hnb_vty_init();
|
||||||
|
|
||||||
rc = telnet_init_dynif(NULL, NULL, vty_get_bind_addr(), 2324);
|
handle_options(argc, argv);
|
||||||
|
|
||||||
|
|
||||||
|
rc = vty_read_config_file(hnb_cmdline_config.config_file, NULL);
|
||||||
|
if (rc < 0) {
|
||||||
|
fprintf(stderr, "Failed to parse the config file: '%s'\n",
|
||||||
|
hnb_cmdline_config.config_file);
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
rc = telnet_init_dynif(tall_hnb_ctx, g_hnb, vty_get_bind_addr(), OSMO_VTY_PORT_HNODEB);
|
||||||
if (rc < 0) {
|
if (rc < 0) {
|
||||||
perror("Error binding VTY port");
|
perror("Error binding VTY port");
|
||||||
exit(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
handle_options(argc, argv);
|
|
||||||
|
|
||||||
rc = hnb_connect(g_hnb);
|
rc = hnb_connect(g_hnb);
|
||||||
if (rc < 0) {
|
if (rc < 0) {
|
||||||
perror("Error connecting to Iuh port");
|
perror("Error connecting to Iuh port");
|
||||||
exit(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
while (1) {
|
signal(SIGINT, &signal_handler);
|
||||||
rc = osmo_select_main(0);
|
signal(SIGTERM, &signal_handler);
|
||||||
if (rc < 0)
|
signal(SIGABRT, &signal_handler);
|
||||||
exit(3);
|
signal(SIGUSR1, &signal_handler);
|
||||||
|
signal(SIGUSR2, &signal_handler);
|
||||||
|
osmo_init_ignore_signals();
|
||||||
|
|
||||||
|
if (hnb_cmdline_config.daemonize) {
|
||||||
|
rc = osmo_daemonize();
|
||||||
|
if (rc < 0) {
|
||||||
|
perror("Error during daemonize");
|
||||||
|
return 6;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
while (!osmo_select_shutdown_done()) {
|
||||||
|
osmo_select_main_ctx(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
log_fini();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Report the heap state of root context, then free,
|
||||||
|
* so both ASAN and Valgrind are happy...
|
||||||
|
*/
|
||||||
|
talloc_report_full(tall_hnb_ctx, stderr);
|
||||||
|
talloc_free(tall_hnb_ctx);
|
||||||
|
|
||||||
|
/* FIXME: VTY code still uses NULL-context */
|
||||||
|
talloc_free(tall_vty_ctx);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Report the heap state of NULL context, then free,
|
||||||
|
* so both ASAN and Valgrind are happy...
|
||||||
|
*/
|
||||||
|
talloc_report_full(NULL, stderr);
|
||||||
|
talloc_disable_null_tracking();
|
||||||
|
|
||||||
/* not reached */
|
/* not reached */
|
||||||
exit(0);
|
exit(0);
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue