/*
 *
 *  Wireless daemon for Linux
 *
 *  Copyright (C) 2013-2014  Intel Corporation. All rights reserved.
 *
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public
 *  License as published by the Free Software Foundation; either
 *  version 2.1 of the License, or (at your option) any later version.
 *
 *  This library 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
 *  Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public
 *  License along with this library; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <stdio.h>
#include <stdlib.h>
#include <getopt.h>
#include <ell/ell.h>

#include "linux/nl80211.h"

#include "src/iwd.h"
#include "src/netdev.h"
#include "src/device.h"
#include "src/wiphy.h"
#include "src/dbus.h"
#include "src/network.h"
#include "src/eap.h"
#include "src/eapol.h"
#include "src/scan.h"
#include "src/wsc.h"
#include "src/knownnetworks.h"
#include "src/rfkill.h"
#include "src/ap.h"
#include "src/plugin.h"

#include "src/backtrace.h"

static struct l_settings *iwd_config;
static struct l_timeout *timeout;
static const char *interfaces;
static const char *nointerfaces;
static const char *phys;
static const char *nophys;
static const char *config_dir;
static const char *plugins;
static const char *noplugins;
static bool terminating;

static void main_loop_quit(struct l_timeout *timeout, void *user_data)
{
	l_main_quit();
}

void iwd_shutdown(void)
{
	if (terminating)
		return;

	terminating = true;

	dbus_shutdown();
	netdev_shutdown();

	timeout = l_timeout_create(1, main_loop_quit, NULL, NULL);
}

static void signal_handler(struct l_signal *signal, uint32_t signo,
							void *user_data)
{
	switch (signo) {
	case SIGINT:
	case SIGTERM:
		l_info("Terminate");
		iwd_shutdown();
		break;
	}
}

const struct l_settings *iwd_get_config(void)
{
	return iwd_config;
}

static void usage(void)
{
	printf("iwd - Wireless daemon\n"
		"Usage:\n");
	printf("\tiwd [options]\n");
	printf("Options:\n"
		"\t-B, --dbus-debug       Enable D-Bus debugging\n"
		"\t-i, --interfaces       Interfaces to manage\n"
		"\t-I, --nointerfaces     Interfaces to ignore\n"
		"\t-p, --phys             Phys to manage\n"
		"\t-P, --nophys           Phys to ignore\n"
		"\t-c, --config           Configuration directory to use\n"
		"\t-l, --plugin           Plugins to include\n"
		"\t-L, --noplugin         Plugins to exclude\n"
		"\t-h, --help             Show help options\n");
}

static const struct option main_options[] = {
	{ "dbus-debug",   no_argument,       NULL, 'B' },
	{ "version",      no_argument,       NULL, 'v' },
	{ "interfaces",   required_argument, NULL, 'i' },
	{ "nointerfaces", required_argument, NULL, 'I' },
	{ "phys",         required_argument, NULL, 'p' },
	{ "nophys",       required_argument, NULL, 'P' },
	{ "config",       required_argument, NULL, 'c' },
	{ "plugin",       required_argument, NULL, 'l' },
	{ "noplugin",     required_argument, NULL, 'L' },
	{ "help",         no_argument,       NULL, 'h' },
	{ }
};

static void do_debug(const char *str, void *user_data)
{
	const char *prefix = user_data;

	l_info("%s%s", prefix, str);
}

static void nl80211_appeared(void *user_data)
{
	struct l_genl_family *nl80211 = user_data;

	if (terminating)
		return;

	l_debug("Found nl80211 interface");

	if (!wiphy_init(nl80211, phys, nophys))
		l_error("Unable to init wiphy functionality");

	if (!netdev_init(nl80211, interfaces, nointerfaces))
		l_error("Unable to init netdev functionality");

	if (!scan_init(nl80211))
		l_error("Unable to init scan functionality");

	if (!wsc_init(nl80211))
		l_error("Unable to init WSC functionality");

	ap_init(nl80211);
}

static void nl80211_vanished(void *user_data)
{
	l_debug("Lost nl80211 interface");

	ap_exit();
	wsc_exit();
	scan_exit();
	netdev_exit();
	wiphy_exit();
}

int main(int argc, char *argv[])
{
	bool enable_dbus_debug = false;
	struct l_signal *signal;
	sigset_t mask;
	int exit_status;
	struct l_genl *genl;
	struct l_genl_family *nl80211;
	char *config_path;
	uint32_t eap_mtu;

	for (;;) {
		int opt;

		opt = getopt_long(argc, argv, "Bi:I:p:P:c:vh",
							main_options, NULL);
		if (opt < 0)
			break;

		switch (opt) {
		case 'B':
			enable_dbus_debug = true;
			break;
		case 'i':
			interfaces = optarg;
			break;
		case 'I':
			nointerfaces = optarg;
			break;
		case 'p':
			phys = optarg;
			break;
		case 'P':
			nophys = optarg;
			break;
		case 'v':
			printf("%s\n", VERSION);
			return EXIT_SUCCESS;
		case 'c':
			config_dir = optarg;
			break;
		case 'l':
			plugins = optarg;
			break;
		case 'L':
			noplugins = optarg;
			break;
		case 'h':
			usage();
			return EXIT_SUCCESS;
		default:
			return EXIT_FAILURE;
		}
	}

	if (argc - optind > 0) {
		fprintf(stderr, "Invalid command line parameters\n");
		return EXIT_FAILURE;
	}

	if (!l_main_init())
		return EXIT_FAILURE;

	sigemptyset(&mask);
	sigaddset(&mask, SIGINT);
	sigaddset(&mask, SIGTERM);

	signal = l_signal_create(&mask, signal_handler, NULL, NULL);

	l_log_set_stderr();
	l_debug_enable("*");

#ifdef __GLIBC__
	__iwd_backtrace_init();
#endif

	l_info("Wireless daemon version %s", VERSION);

	if (!config_dir)
		config_dir = CONFIGDIR;

	config_path = l_strdup_printf("/%s/%s", config_dir, "iwd.conf");

	iwd_config = l_settings_new();

	if (!l_settings_load_from_file(iwd_config, config_path))
		l_warn("Failed to load configuration file %s", config_path);

	l_free(config_path);

	if (!dbus_init(enable_dbus_debug)) {
		exit_status = EXIT_FAILURE;
		goto done;
	}

	plugin_init(plugins, noplugins);

	genl = l_genl_new_default();
	if (!genl) {
		l_error("Failed to open generic netlink socket");
		exit_status = EXIT_FAILURE;
		goto fail_genl;
	}

	if (getenv("IWD_GENL_DEBUG"))
		l_genl_set_debug(genl, do_debug, "[GENL] ", NULL);

	if (!device_init()) {
		exit_status = EXIT_FAILURE;
		goto fail_device;
	}

	l_debug("Opening nl80211 interface");

	nl80211 = l_genl_family_new(genl, NL80211_GENL_NAME);
	if (!nl80211) {
		l_error("Failed to open nl80211 interface");
		exit_status = EXIT_FAILURE;
		goto fail_nl80211;
	}

	l_genl_family_set_watches(nl80211, nl80211_appeared, nl80211_vanished,
								nl80211, NULL);

	if (!l_settings_get_uint(iwd_config, "EAP", "mtu", &eap_mtu))
		eap_mtu = 1400; /* on WiFi the real MTU is around 2304 */

	eap_init(eap_mtu);
	eapol_init();
	network_init();
	known_networks_init();
	rfkill_init();

	exit_status = EXIT_SUCCESS;
	l_main_run();

	rfkill_exit();
	known_networks_exit();
	network_exit();
	eapol_exit();
	eap_exit();
	plugin_exit();

	l_genl_family_unref(nl80211);

fail_nl80211:
	device_exit();

fail_device:
	l_genl_unref(genl);

fail_genl:
	dbus_exit();

done:
	l_settings_free(iwd_config);

	l_signal_remove(signal);
	l_timeout_remove(timeout);

	l_main_exit();

	return exit_status;
}