Initial import, still work in progress

1 files changed, 143 insertions, 0 deletions

diff --git a/main.c b/main.c
new file mode 100644
index 0000000..ffee131
--- /dev/null
+++ b/main.c
@@ -0,0 +1,143 @@
+/*
+ * llmnrd -- LLMNR (RFC 4705) responder daemon.
+ *
+ * Copyright (C) 2014-2015 Tobias Klauser <tklauser@distanz.ch>
+ *
+ * This file is part of llmnrd.
+ *
+ * llmnrd is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, version 2 of the License.
+ *
+ * llmnrd 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 llmnrd.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <getopt.h>
+#include <signal.h>
+#include <stdbool.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+#include <sys/ioctl.h>
+
+#include "compiler.h"
+#include "log.h"
+#include "util.h"
+
+#include "iface.h"
+#include "llmnr.h"
+
+static const char *short_opts = "H:p:dh";
+static const struct option long_opts[] = {
+	{ "hostname",	required_argument,	NULL, 'H' },
+	{ "port",	required_argument,	NULL, 'p' },
+	{ "daemonize",	no_argument,		NULL, 'd' },
+	{ "help",	no_argument,		NULL, 'h' },
+	{ NULL,		0,			NULL, 0 },
+};
+
+static void __noreturn usage_and_exit(int status)
+{
+	fprintf(stdout, "Usage: llmnrd [OPTIONS]\n");
+	exit(status);
+}
+
+static void signal_handler(int sig)
+{
+	switch (sig) {
+	case SIGINT:
+	case SIGQUIT:
+	case SIGTERM:
+		log_info("Interrupt received. Stopping llmnrd.\n");
+		iface_stop();
+		llmnr_stop();
+		break;
+	case SIGHUP:
+	default:
+		/* ignore */
+		break;
+	}
+}
+
+static void register_signal(int sig, void (*handler)(int))
+{
+	sigset_t block_mask;
+	struct sigaction saction;
+
+	sigfillset(&block_mask);
+
+	saction.sa_handler = handler;
+	saction.sa_mask = block_mask;
+
+	if (sigaction(sig, &saction, NULL) != 0) {
+		log_err("Failed to register signal handler for %s (%d)\n",
+			strsignal(sig), sig);
+	}
+}
+
+int main(int argc, char **argv)
+{
+	int c, ret = EXIT_FAILURE;
+	long num_arg;
+	bool daemonize = false;
+	char *hostname = "";
+	uint16_t port = 0;
+
+	while ((c = getopt_long(argc, argv, short_opts, long_opts, NULL)) != -1) {
+		switch (c) {
+		case 'd':
+			daemonize = true;
+			break;
+		case 'H':
+			hostname = xstrdup(optarg);
+			break;
+		case 'p':
+			num_arg = strtol(optarg, NULL, 0);
+			if (num_arg < 0 || num_arg > UINT16_MAX) {
+				log_err("Invalid port number: %ld\n", num_arg);
+				return EXIT_FAILURE;
+			}
+			port = num_arg;
+		case 'h':
+			ret = EXIT_SUCCESS;
+			/* fall through */
+		default:
+			usage_and_exit(ret);
+		}
+	}
+
+	register_signal(SIGINT, signal_handler);
+	register_signal(SIGQUIT, signal_handler);
+	register_signal(SIGTERM, signal_handler);
+	register_signal(SIGHUP, signal_handler);
+
+	if (hostname[0] == '\0') {
+		/* TODO: Consider hostname changing at runtime */
+		hostname = xmalloc(255);
+		if (gethostname(hostname, 255) != 0) {
+			log_err("Failed to get hostname");
+			return EXIT_FAILURE;
+		}
+	}
+
+	if (daemonize) {
+		/* TODO */
+	}
+
+	if (iface_start_thread() < 0)
+		goto out;
+
+	ret = llmnr_run(hostname, port);
+out:
+	free(hostname);
+	return ret;
+}