summaryrefslogtreecommitdiffstats
path: root/avahi-daemon/main.c
diff options
context:
space:
mode:
Diffstat (limited to 'avahi-daemon/main.c')
-rw-r--r--avahi-daemon/main.c346
1 files changed, 316 insertions, 30 deletions
diff --git a/avahi-daemon/main.c b/avahi-daemon/main.c
index d59e9fb..4373f9a 100644
--- a/avahi-daemon/main.c
+++ b/avahi-daemon/main.c
@@ -23,19 +23,45 @@
#include <config.h>
#endif
+#include <getopt.h>
+#include <string.h>
+#include <signal.h>
+
+#include <libdaemon/dfork.h>
+#include <libdaemon/dsignal.h>
+#include <libdaemon/dlog.h>
+#include <libdaemon/dpid.h>
+
#include <avahi-core/core.h>
+#include <avahi-core/log.h>
#include "main.h"
#include "simple-protocol.h"
+#include "dbus-protocol.h"
#include "static-services.h"
AvahiServer *avahi_server = NULL;
+typedef enum {
+ DAEMON_RUN,
+ DAEMON_KILL,
+ DAEMON_VERSION,
+ DAEMON_HELP
+} DaemonCommand;
+
+typedef struct {
+ AvahiServerConfig server_config;
+ DaemonCommand command;
+ gboolean daemonize;
+ gchar *config_file;
+ gboolean enable_dbus;
+} DaemonConfig;
+
static void server_callback(AvahiServer *s, AvahiServerState state, gpointer userdata) {
g_assert(s);
if (state == AVAHI_SERVER_RUNNING) {
- g_message("Server startup complete. Host name is <%s>", avahi_server_get_host_name_fqdn(s));
+ avahi_log_info("Server startup complete. Host name is <%s>", avahi_server_get_host_name_fqdn(s));
static_service_add_to_server();
} else if (state == AVAHI_SERVER_COLLISION) {
gchar *n;
@@ -43,7 +69,7 @@ static void server_callback(AvahiServer *s, AvahiServerState state, gpointer use
static_service_remove_from_server();
n = avahi_alternative_host_name(avahi_server_get_host_name(s));
- g_message("Host name conflict, retrying with <%s>", n);
+ avahi_log_warn("Host name conflict, retrying with <%s>", n);
avahi_server_set_host_name(s, n);
g_free(n);
}
@@ -52,64 +78,227 @@ static void server_callback(AvahiServer *s, AvahiServerState state, gpointer use
static void help(FILE *f, const gchar *argv0) {
fprintf(f,
"%s [options]\n"
- " -h --help Show this help\n"
- " -D --daemon Daemonize after startup\n"
- " -k --kill Kill a running daemon\n"
- " -v --version Show version\n",
+ " -h --help Show this help\n"
+ " -D --daemonize Daemonize after startup\n"
+ " -k --kill Kill a running daemon\n"
+ " -V --version Show version\n"
+ " -f --file=FILE Load the specified configuration file instead of the default\n",
argv0);
}
-static gint parse_command_line(AvahiServerConfig *config, int argc, char *argv[]) {
+static gint parse_command_line(DaemonConfig *config, int argc, char *argv[]) {
+ gint c;
+
+ static const struct option const long_options[] = {
+ { "help", no_argument, NULL, 'h' },
+ { "daemonize", no_argument, NULL, 'D' },
+ { "kill", no_argument, NULL, 'k' },
+ { "version", no_argument, NULL, 'V' },
+ { "file", required_argument, NULL, 'f' },
+ };
- return 0;
-}
+ g_assert(config);
-static gint load_config_file(AvahiServerConfig *config) {
+ opterr = 0;
+ while ((c = getopt_long(argc, argv, "hDkVf:", long_options, NULL)) >= 0) {
+ switch(c) {
+ case 'h':
+ config->command = DAEMON_HELP;
+ break;
+ case 'D':
+ config->daemonize = TRUE;
+ break;
+ case 'k':
+ config->command = DAEMON_KILL;
+ break;
+ case 'V':
+ config->command = DAEMON_VERSION;
+ break;
+ case 'f':
+ g_free(config->config_file);
+ config->config_file = g_strdup(optarg);
+ break;
+ default:
+ fprintf(stderr, "Invalid command line argument: %c\n", c);
+ return -1;
+ }
+ }
+
+ if (optind < argc) {
+ fprintf(stderr, "Too many arguments\n");
+ return -1;
+ }
+
return 0;
}
-int main(int argc, char *argv[]) {
- GMainLoop *loop = NULL;
- gint r = 255;
- AvahiServerConfig config;
+static gboolean is_yes(const gchar *s) {
+ g_assert(s);
+
+ return *s == 'y' || *s == 'Y';
+}
- avahi_server_config_init(&config);
+static gint load_config_file(DaemonConfig *config) {
+ int r = -1;
+ GKeyFile *f = NULL;
+ GError *err = NULL;
+ gchar **groups = NULL, **g, **keys = NULL;
- if (load_config_file(&config) < 0)
+ g_assert(config);
+
+ f = g_key_file_new();
+
+ if (!g_key_file_load_from_file(f, config->config_file ? config->config_file : AVAHI_CONFIG_FILE, G_KEY_FILE_NONE, &err)) {
+ fprintf(stderr, "Unable to read config file: %s\n", err->message);
goto finish;
+ }
- if (parse_command_line(&config, argc, argv) < 0)
- goto finish;
+ groups = g_key_file_get_groups(f, NULL);
+
+ for (g = groups; *g; g++) {
+ if (g_strcasecmp(*g, "server") == 0) {
+ gchar **k;
+
+ keys = g_key_file_get_keys(f, *g, NULL, NULL);
+
+ for (k = keys; *k; k++) {
+ if (g_strcasecmp(*k, "host-name") == 0) {
+ g_free(config->server_config.host_name);
+ config->server_config.host_name = g_strdup(g_key_file_get_string(f, *g, *k, NULL));
+ } else if (g_strcasecmp(*k, "domain-name") == 0) {
+ g_free(config->server_config.domain_name);
+ config->server_config.domain_name = g_strdup(g_key_file_get_string(f, *g, *k, NULL));
+ } else if (g_strcasecmp(*k, "use-ipv4") == 0)
+ config->server_config.use_ipv4 = is_yes(g_key_file_get_string(f, *g, *k, NULL));
+ else if (g_strcasecmp(*k, "use-ipv6") == 0)
+ config->server_config.use_ipv6 = is_yes(g_key_file_get_string(f, *g, *k, NULL));
+ else if (g_strcasecmp(*k, "check-response-ttl") == 0)
+ config->server_config.check_response_ttl = is_yes(g_key_file_get_string(f, *g, *k, NULL));
+ else if (g_strcasecmp(*k, "use-iff-running") == 0)
+ config->server_config.use_iff_running = is_yes(g_key_file_get_string(f, *g, *k, NULL));
+ else if (g_strcasecmp(*k, "enable-dbus") == 0)
+ config->enable_dbus = is_yes(g_key_file_get_string(f, *g, *k, NULL));
+ else {
+ fprintf(stderr, "Invalid configuration key \"%s\" in group \"%s\"\n", *k, *g);
+ goto finish;
+ }
+ }
+
+ } else if (g_strcasecmp(*g, "publish") == 0) {
+ gchar **k;
+
+ keys = g_key_file_get_keys(f, *g, NULL, NULL);
+
+ for (k = keys; *k; k++) {
+ if (g_strcasecmp(*k, "publish-addresses") == 0)
+ config->server_config.publish_addresses = is_yes(g_key_file_get_string(f, *g, *k, NULL));
+ else if (g_strcasecmp(*k, "publish-hinfo") == 0)
+ config->server_config.publish_hinfo = is_yes(g_key_file_get_string(f, *g, *k, NULL));
+ else if (g_strcasecmp(*k, "publish-workstation") == 0)
+ config->server_config.publish_workstation = is_yes(g_key_file_get_string(f, *g, *k, NULL));
+ else if (g_strcasecmp(*k, "publish-domain") == 0)
+ config->server_config.publish_domain = is_yes(g_key_file_get_string(f, *g, *k, NULL));
+ else {
+ fprintf(stderr, "Invalid configuration key \"%s\" in group \"%s\"\n", *k, *g);
+ goto finish;
+ }
+ }
+
+ } else if (g_strcasecmp(*g, "reflector") == 0) {
+ gchar **k;
+
+ keys = g_key_file_get_keys(f, *g, NULL, NULL);
+
+ for (k = keys; *k; k++) {
+ if (g_strcasecmp(*k, "enable-reflector") == 0)
+ config->server_config.enable_reflector = is_yes(g_key_file_get_string(f, *g, *k, NULL));
+ else if (g_strcasecmp(*k, "reflect-ipv") == 0)
+ config->server_config.reflect_ipv = is_yes(g_key_file_get_string(f, *g, *k, NULL));
+ else {
+ fprintf(stderr, "Invalid configuration key \"%s\" in group \"%s\"\n", *k, *g);
+ goto finish;
+ }
+ }
- loop = g_main_loop_new(NULL, FALSE);
+ } else {
+ fprintf(stderr, "Invalid configuration file group \"%s\".\n", *g);
+ goto finish;
+ }
+ }
+
+ r = 0;
+
+finish:
+
+ g_strfreev(groups);
+ g_strfreev(keys);
+
+ if (err)
+ g_error_free (err);
+
+ if (f)
+ g_key_file_free(f);
+
+ return r;
+}
+
+static void log_function(AvahiLogLevel level, const gchar *txt) {
+
+ static const int const log_level_map[] = {
+ LOG_ERR,
+ LOG_WARNING,
+ LOG_NOTICE,
+ LOG_INFO,
+ LOG_DEBUG
+ };
+
+ g_assert(level < AVAHI_LOG_LEVEL_MAX);
+ g_assert(txt);
+ daemon_log(log_level_map[level], "%s", txt);
+}
+
+static gint run_server(DaemonConfig *config) {
+ GMainLoop *loop = NULL;
+ gint r = -1;
+
+ g_assert(config);
+
+ loop = g_main_loop_new(NULL, FALSE);
+
if (simple_protocol_setup(NULL) < 0)
goto finish;
-
+
#ifdef ENABLE_DBUS
- if (dbus_protocol_setup (loop) < 0)
- goto finish;
+ if (config->enable_dbus)
+ if (dbus_protocol_setup(loop) < 0)
+ goto finish;
#endif
-
- if (!(avahi_server = avahi_server_new(NULL, &config, server_callback, NULL)))
+
+ if (!(avahi_server = avahi_server_new(NULL, &config->server_config, server_callback, NULL)))
goto finish;
-
+
static_service_load();
+ if (config->daemonize) {
+ daemon_retval_send(0);
+ r = 0;
+ }
+
g_main_loop_run(loop);
- r = 0;
-
-finish:
+finish:
+
static_service_remove_from_server();
static_service_free_all();
simple_protocol_shutdown();
#ifdef ENABLE_DBUS
- dbus_protocol_shutdown();
+ if (config->enable_dbus)
+ dbus_protocol_shutdown();
#endif
if (avahi_server)
@@ -118,7 +307,104 @@ finish:
if (loop)
g_main_loop_unref(loop);
- avahi_server_config_free(&config);
+ if (r != 0 && config->daemonize)
+ daemon_retval_send(0);
+
+ return r;
+}
+
+int main(int argc, char *argv[]) {
+ gint r = 255;
+ DaemonConfig config;
+ const gchar *argv0;
+ gboolean wrote_pid_file = FALSE;
+
+ avahi_set_log_function(log_function);
+
+ avahi_server_config_init(&config.server_config);
+ config.command = DAEMON_RUN;
+ config.daemonize = FALSE;
+ config.config_file = NULL;
+ config.enable_dbus = TRUE;
+
+ if ((argv0 = strrchr(argv[0], '/')))
+ argv0++;
+ else
+ argv0 = argv[0];
+ daemon_pid_file_ident = daemon_log_ident = (char *) argv0;
+
+ if (parse_command_line(&config, argc, argv) < 0)
+ goto finish;
+
+ if (load_config_file(&config) < 0)
+ goto finish;
+
+ if (config.command == DAEMON_HELP) {
+ help(stdout, argv0);
+ r = 0;
+ } else if (config.command == DAEMON_VERSION) {
+ printf("%s "PACKAGE_VERSION"\n", argv0);
+
+ } else if (config.command == DAEMON_KILL) {
+ if (daemon_pid_file_kill_wait(SIGTERM, 5) < 0) {
+ avahi_log_warn("Failed to kill daemon");
+ goto finish;
+ }
+
+ r = 0;
+
+ } else if (config.command == DAEMON_RUN) {
+ pid_t pid;
+
+ if ((pid = daemon_pid_file_is_running()) >= 0) {
+ avahi_log_error("Daemon already running on PID %u", pid);
+ goto finish;
+ }
+
+ if (config.daemonize) {
+
+ daemon_retval_init();
+
+ if ((pid = daemon_fork()) < 0)
+ goto finish;
+ else if (pid != 0) {
+ int ret;
+ /** Parent **/
+
+ if ((ret = daemon_retval_wait(20)) < 0) {
+ avahi_log_error("Could not recieve return value from daemon process.");
+ goto finish;
+ }
+
+ r = ret;
+ goto finish;
+ }
+
+ /* Child */
+ }
+
+ if (daemon_pid_file_create() < 0) {
+ avahi_log_error("Failed to create PID file.");
+ daemon_retval_send(1);
+ goto finish;
+ } else
+ wrote_pid_file = TRUE;
+
+ if (run_server(&config) == 0)
+ r = 0;
+ }
+
+finish:
+
+ if (config.daemonize)
+ daemon_retval_done();
+
+ avahi_server_config_free(&config.server_config);
+ g_free(config.config_file);
+
+ if (wrote_pid_file)
+ daemon_pid_file_remove();
+
return r;
}