summaryrefslogtreecommitdiff
path: root/freeDiameterd/main.c
diff options
context:
space:
mode:
Diffstat (limited to 'freeDiameterd/main.c')
-rw-r--r--freeDiameterd/main.c217
1 files changed, 167 insertions, 50 deletions
diff --git a/freeDiameterd/main.c b/freeDiameterd/main.c
index c4ecbb4..47037a5 100644
--- a/freeDiameterd/main.c
+++ b/freeDiameterd/main.c
@@ -2,7 +2,7 @@
* Software License Agreement (BSD License) *
* Author: Sebastien Decugis <sdecugis@freediameter.net> *
* *
-* Copyright (c) 2015, WIDE Project and NICT *
+* Copyright (c) 2020, WIDE Project and NICT *
* All rights reserved. *
* *
* Redistribution and use of this software in source and binary forms, with or without modification, are *
@@ -33,13 +33,22 @@
* ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. *
*********************************************************************************************************/
+#if defined(__GLIBC__)
+#define _BSD_SOURCE /* for vsyslog */
+#endif
+
#include <freeDiameter/freeDiameter-host.h>
#include <freeDiameter/libfdcore.h>
#include <signal.h>
#include <getopt.h>
#include <locale.h>
-
+#include <syslog.h>
+#include <stdarg.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
/* forward declarations */
static int main_cmdline(int argc, char *argv[]);
@@ -47,68 +56,151 @@ static void * catch_signals(void * arg);
static pthread_t signals_thr;
static char *conffile = NULL;
+static int daemon_mode = 0;
static int gnutls_debug = 0;
+static char *pidfile = NULL;
/* gnutls debug */
static void fd_gnutls_debug(int level, const char * str) {
fd_log_debug(" [gnutls:%d] %s", level, str);
}
+static void pidfile_cleanup(void)
+{
+ if (pidfile != NULL) {
+ LOG_I("Removing pidfile '%s'", pidfile);
+ CHECK_SYS_DO( unlink(pidfile), /* ignore */ );
+ pidfile = NULL;
+ }
+}
+
+static int pidfile_create(void)
+{
+ if (pidfile == NULL) {
+ return 0;
+ }
+
+ /* Create pidfile */
+ FILE * fp = fopen(pidfile, "w");
+ if (fp == NULL) {
+ int ret = errno;
+ LOG_F("Unable to write pidfile '%s'; Error: %s",
+ pidfile, strerror(ret));
+ pidfile = NULL; /* disable pidfile_cleanup() */
+ return ret;
+ }
+
+ /* Cleanup pidfile on exit */
+ if (atexit(pidfile_cleanup) != 0) {
+ LOG_F("Unable to setup pidfile cleanup");
+ CHECK_SYS( fclose(fp) );
+ pidfile_cleanup();
+ return EINVAL;
+ }
+
+ /* Write the pid and close pidfile */
+ fprintf(fp, "%d\n", getpid());
+ CHECK_SYS_DO( fclose(fp), { pidfile_cleanup(); return __ret__; } );
+
+ LOG_I("Created pidfile '%s'", pidfile);
+ return 0;
+}
+
+
+static void syslog_logger(int loglevel, const char * format, va_list args)
+{
+ if (loglevel < fd_g_debug_lvl) {
+ return;
+ }
+
+ int level;
+
+ switch (loglevel) {
+ case FD_LOG_INFO:
+ level = LOG_INFO;
+ break;
+ case FD_LOG_NOTICE:
+ level = LOG_NOTICE;
+ break;
+ case FD_LOG_ERROR:
+ level = LOG_ERR;
+ break;
+ case FD_LOG_FATAL:
+ level = LOG_CRIT;
+ break;
+ default:
+ /* fallthrough */
+ case FD_LOG_ANNOYING:
+ case FD_LOG_DEBUG:
+ level = LOG_DEBUG;
+ break;
+ }
+
+ vsyslog(level | LOG_DAEMON, format, args);
+}
+
/* freeDiameter starting point */
int main(int argc, char * argv[])
{
int ret;
sigset_t sig_all;
-
+
/* Block all signals from the current thread and all its future children -- we will catch everything in catch_signals */
sigfillset(&sig_all);
ret = pthread_sigmask(SIG_BLOCK, &sig_all, NULL);
ASSERT(ret == 0);
-
+
/* Parse the command-line */
ret = main_cmdline(argc, argv);
if (ret != 0) {
- return ret;
+ return EXIT_FAILURE;
}
-
+
+ if (daemon_mode) {
+ TRACE_DEBUG(INFO, "entering background mode");
+ CHECK_SYS_DO( daemon(1, 0), return EXIT_FAILURE );
+ }
+
+ CHECK_FCT_DO( pidfile_create(), return EXIT_FAILURE );
+
/* Initialize the core library */
ret = fd_core_initialize();
if (ret != 0) {
fprintf(stderr, "An error occurred during freeDiameter core library initialization.\n");
- return ret;
+ return EXIT_FAILURE;
}
-
+
/* Set gnutls debug level ? */
if (gnutls_debug) {
gnutls_global_set_log_function((gnutls_log_func)fd_gnutls_debug);
gnutls_global_set_log_level (gnutls_debug);
TRACE_DEBUG(INFO, "Enabled GNUTLS debug at level %d", gnutls_debug);
}
-
+
/* Parse the configuration file */
CHECK_FCT_DO( fd_core_parseconf(conffile), goto error );
-
+
/* Start the servers */
CHECK_FCT_DO( fd_core_start(), goto error );
-
+
/* Allow SIGINT and SIGTERM from this point to terminate the application */
CHECK_POSIX_DO( pthread_create(&signals_thr, NULL, catch_signals, NULL), goto error );
-
+
TRACE_DEBUG(INFO, FD_PROJECT_BINARY " daemon initialized.");
/* Now, just wait for termination */
- CHECK_FCT( fd_core_wait_shutdown_complete() );
-
+ CHECK_FCT_DO( fd_core_wait_shutdown_complete(), return EXIT_FAILURE );
+
/* Just in case it was not the result of a signal, we cancel signals_thr */
fd_thr_term(&signals_thr);
-
- return 0;
-error:
+
+ return EXIT_SUCCESS;
+error:
CHECK_FCT_DO( fd_core_shutdown(), );
- CHECK_FCT( fd_core_wait_shutdown_complete() );
+ CHECK_FCT_DO( fd_core_wait_shutdown_complete(), return EXIT_FAILURE );
fd_thr_term(&signals_thr);
- return -1;
+ return EXIT_FAILURE;
}
@@ -134,18 +226,21 @@ static void main_help( void )
printf( " This daemon is an implementation of the Diameter protocol\n"
" used for Authentication, Authorization, and Accounting (AAA).\n");
printf("\nUsage: " FD_PROJECT_BINARY " [OPTIONS]...\n");
- printf( " -h, --help Print help and exit\n"
- " -V, --version Print version and exit\n"
- " -c, --config=filename Read configuration from this file instead of the \n"
- " default location (" DEFAULT_CONF_PATH "/" FD_DEFAULT_CONF_FILENAME ").\n");
+ printf( " -h, --help Print help and exit\n"
+ " -V, --version Print version and exit\n"
+ " -c, --config=filename Read configuration from this file instead of the \n"
+ " default location (" DEFAULT_CONF_PATH "/" FD_DEFAULT_CONF_FILENAME ")\n"
+ " -D, --daemon Start program in background\n"
+ " -p, --pidfile=filename Write PID to filename\n"
+ " -s, --syslog Write log output to syslog (instead of stdout)\n");
printf( "\nDebug:\n"
" These options are mostly useful for developers\n"
- " -l, --dbglocale Set the locale for error messages\n"
- " -d, --debug Increase verbosity of debug messages if default logger is used\n"
- " -q, --quiet Decrease verbosity if default logger is used\n"
+ " -d, --debug Increase verbosity of log messages\n"
" -f, --dbg_func <func> Enable all traces within the function <func>\n"
" -F, --dbg_file <file.c> Enable all traces within the file <file.c> (basename match)\n"
- " --dbg_gnutls <int> Enable GNU TLS debug at level <int>\n"
+ " -g, --dbg_gnutls <int> Enable GNU TLS debug at level <int>\n"
+ " -l, --dbglocale Set the locale for error messages\n"
+ " -q, --quiet Decrease verbosity of log messages\n"
);
}
@@ -155,11 +250,14 @@ static int main_cmdline(int argc, char *argv[])
int c;
int option_index = 0;
char * locale;
-
+
struct option long_options[] = {
{ "help", no_argument, NULL, 'h' },
{ "version", no_argument, NULL, 'V' },
{ "config", required_argument, NULL, 'c' },
+ { "syslog", no_argument, NULL, 's' },
+ { "daemon", no_argument, NULL, 'D' },
+ { "pidfile", required_argument, NULL, 'p' },
{ "debug", no_argument, NULL, 'd' },
{ "quiet", no_argument, NULL, 'q' },
{ "dbglocale", optional_argument, NULL, 'l' },
@@ -168,21 +266,21 @@ static int main_cmdline(int argc, char *argv[])
{ "dbg_gnutls", required_argument, NULL, 'g' },
{ NULL, 0, NULL, 0 }
};
-
+
/* Loop on arguments */
while (1) {
- c = getopt_long (argc, argv, "hVc:dql:f:F:g:", long_options, &option_index);
- if (c == -1)
+ c = getopt_long (argc, argv, "hVc:Dp:dql:f:F:g:s", long_options, &option_index);
+ if (c == -1)
break; /* Exit from the loop. */
-
+
switch (c) {
case 'h': /* Print help and exit. */
main_help();
- exit(0);
+ exit(EXIT_SUCCESS);
case 'V': /* Print version and exit. */
main_version();
- exit(0);
+ exit(EXIT_SUCCESS);
case 'c': /* Read configuration from this file instead of the default location.. */
if (optarg == NULL ) {
@@ -192,6 +290,18 @@ static int main_cmdline(int argc, char *argv[])
conffile = optarg;
break;
+ case 'D':
+ daemon_mode = 1;
+ break;
+
+ case 'p': /* Write pidfile */
+ if (optarg == NULL ) {
+ fprintf(stderr, "Missing argument with --pidfile directive\n");
+ return EINVAL;
+ }
+ pidfile = optarg;
+ break;
+
case 'l': /* Change the locale. */
locale = setlocale(LC_ALL, optarg?:"");
if (!locale) {
@@ -199,11 +309,11 @@ static int main_cmdline(int argc, char *argv[])
return EINVAL;
}
break;
-
+
case 'd': /* Increase verbosity of debug messages. */
fd_g_debug_lvl--;
break;
-
+
case 'f': /* Full debug for the function with this name. */
#ifdef DEBUG
fd_debug_one_function = optarg;
@@ -213,7 +323,7 @@ static int main_cmdline(int argc, char *argv[])
return EINVAL;
#endif /* DEBUG */
break;
-
+
case 'F': /* Full debug for the file with this name. */
#ifdef DEBUG
fd_debug_one_file = basename(optarg);
@@ -223,15 +333,22 @@ static int main_cmdline(int argc, char *argv[])
return EINVAL;
#endif /* DEBUG */
break;
-
+
case 'g': /* Set a debug level and function for GNU TLS calls. */
gnutls_debug = (int)atoi(optarg);
break;
-
+
case 'q': /* Decrease verbosity then remove debug messages. */
fd_g_debug_lvl++;
break;
+ case 's': /* Write log data using syslog(3) */
+ if (fd_log_handler_register(syslog_logger) != 0) {
+ fprintf(stderr, "Cannot initialize syslog logger\n");
+ return EINVAL;
+ }
+ break;
+
case '?': /* Invalid option. */
/* `getopt_long' already printed an error message. */
fprintf(stderr, "getopt_long found an invalid character\n");
@@ -243,7 +360,7 @@ static int main_cmdline(int argc, char *argv[])
return EINVAL;
}
}
-
+
return 0;
}
@@ -252,29 +369,29 @@ static void * catch_signals(void * arg)
{
sigset_t ss;
fd_log_threadname ( "signals catcher" );
-
+
sigemptyset(&ss);
-
+
/* Signals that terminate the daemon */
sigaddset(&ss, SIGTERM);
sigaddset(&ss, SIGINT);
-
+
/* Signals that send an event */
sigaddset(&ss, SIGUSR1);
sigaddset(&ss, SIGUSR2);
-
+
/* We unblock all other signals, so that their default handler is used (such as SIGTSTP) */
CHECK_SYS_DO( pthread_sigmask( SIG_SETMASK, &ss, NULL ), goto out );
-
+
/* Now loop on the reception of the signal */
while (1) {
int sig, *ps;
-
+
/* Wait to receive the next signal */
CHECK_POSIX_DO( sigwait(&ss, &sig), break );
-
+
TRACE_DEBUG(FULL, "Signal %d caught", sig);
-
+
switch (sig) {
case SIGUSR1:
case SIGUSR2:
@@ -282,14 +399,14 @@ static void * catch_signals(void * arg)
*ps = sig;
CHECK_FCT_DO( fd_event_send(fd_g_config->cnf_main_ev, FDEV_TRIGGER, sizeof(int), ps), goto out );
break;
-
+
case SIGINT:
case SIGTERM:
CHECK_FCT_DO( fd_core_shutdown(), goto out );
}
}
-out:
+out:
/* Better way to handle this ? */
ASSERT(0);
return NULL;
generated by cgit v1.2.3 (git 2.39.1) at 2026-09-11 13:09:20 +0000