ebusd/src/main.cpp

/*
 * Copyright (C) Roland Jax 2014 <roland.jax@liwest.at>
 *
 * This file is part of ebus-daemon.
 *
 * ebus-daemon 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, either version 3 of the License, or
 * (at your option) any later version.
 *
 * ebus-daemon 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 ebus-daemon. If not, see http://www.gnu.org/licenses/.
 */

#include "libebus.h"
#include "logger.h"
#include "daemon.h"
#include "appl.h"
#include "network.h"
#include "ebusloop.h"
#include "cycdata.h"
#include "baseloop.h"
#include <iostream>
#include <memory>
#include <csignal>
#include <cstring>
#include <cstdio>
#include <sstream>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

using namespace libebus;

Appl& A = Appl::Instance();
Daemon& D = Daemon::Instance();
LogInstance& L = LogInstance::Instance();

Network* network;
Commands* commands;
EBusLoop* ebusloop;
CYCData* cycdata;

void define_args()
{
	A.addItem("p_address", Appl::Param("FF"), "a", "address",
		  "\tebus device address (default: FF)",
		  Appl::type_string, Appl::opt_mandatory);

	A.addItem("p_device", Appl::Param("/dev/ttyUSB0"), "d", "device",
		  "\tebus device (serial or network) (default: /dev/ttyUSB0)",
		  Appl::type_string, Appl::opt_mandatory);

	A.addItem("p_nodevicecheck", Appl::Param(false), "n", "nodevicecheck",
		  "disable valid ebus device test\n",
		  Appl::type_bool, Appl::opt_none);

	A.addItem("p_ebusconfdir", Appl::Param("/etc/ebus-daemon"), "e", "ebusconfdir",
		  "directory for ebus configuration (default: /etc/ebus-daemon)\n",
		  Appl::type_string, Appl::opt_mandatory);

	A.addItem("p_foreground", Appl::Param(false), "f", "foreground",
		  "run in foreground\n",
		  Appl::type_bool, Appl::opt_none);

	A.addItem("p_port", Appl::Param(8888), "p", "port",
		  "\tlisten port (default: 8888)",
		  Appl::type_int, Appl::opt_mandatory);

	A.addItem("p_localhost", Appl::Param(false), "", "localhost",
		  "listen localhost only\n",
		  Appl::type_bool, Appl::opt_none);

	A.addItem("p_logfile", Appl::Param("/var/log/ebus-daemon.log"), "l", "logfile",
		  "\tlog file name (default: /var/log/ebus-daemon.log)",
		  Appl::type_string, Appl::opt_mandatory);

	A.addItem("p_area", Appl::Param(all), "", "logarea",
		  "\tlogging area (bas=1, net=2, bus=4, cyc=8, default: all=15)",
		  Appl::type_int, Appl::opt_mandatory);

	A.addItem("p_level", Appl::Param(trace), "", "loglevel",
		  "\tlogging level (error=0, event=1, default: trace=2, debug=3)\n",
		  Appl::type_int, Appl::opt_mandatory);

	A.addItem("p_dump", Appl::Param(false), "D", "dump",
		  "\tenable dump",
		  Appl::type_bool, Appl::opt_none);

	A.addItem("p_dumpfile", Appl::Param("/tmp/ebus_dump.bin"), "", "dumpfile",
		  "\tdump file name (default: /tmp/ebus_dump.bin)",
		  Appl::type_string, Appl::opt_mandatory);

	A.addItem("p_dumpsize", Appl::Param(100), "", "dumpsize",
		  "\tmax size for dump file in kB (default: 100)\n",
		  Appl::type_long, Appl::opt_mandatory);

	A.addItem("p_settings", Appl::Param(false), "", "settings",
		  "\tprint daemon settings\n",
		  Appl::type_bool, Appl::opt_none);

	A.addItem("p_help", Appl::Param(false), "h", "help",
		  "\tprint this message",
		  Appl::type_bool, Appl::opt_none);
}

void shutdown()
{
	// free Network
	if (network != NULL)
		delete network;

	// free CYCData
	if (cycdata != NULL) {
		cycdata->stop();
		delete cycdata;
	}

	// free EBusLoop
	if (ebusloop != NULL) {
		ebusloop->stop();
		ebusloop->join();
		delete ebusloop;
	}

	// free Commands DB
	if (commands != NULL)
		delete commands;

	// reset all signal handlers to default
	signal(SIGHUP, SIG_DFL);
	signal(SIGINT, SIG_DFL);
	signal(SIGTERM, SIG_DFL);

	// delete Daemon pid file
	if (D.status() == true)
		D.stop();

	// stop Logger
	L.log(bas, event, "ebus-daemon stopped");
	L.stop();
	L.join();

	exit(EXIT_SUCCESS);
}

void signal_handler(int sig)
{
	switch (sig) {
	case SIGHUP:
		L.log(bas, event, "SIGHUP received");
		break;
	case SIGINT:
		L.log(bas, event, "SIGINT received");
		shutdown();
		break;
	case SIGTERM:
		L.log(bas, event, "SIGTERM received");
		shutdown();
		break;
	default:
		L.log(bas, event, "undefined signal %s", strsignal(sig));
		break;
	}
}

int main(int argc, char* argv[])
{
	// define Arguments and Application variables
	define_args();

	// parse Arguments
	if (A.parseArgs(argc, argv) == false) {
		A.printArgs();
		exit(EXIT_FAILURE);
	}

	// print Help
	if (A.getParam<bool>("p_help") == true) {
		A.printArgs();
		exit(EXIT_SUCCESS);
	}

	// print Daemon settings
	if (A.getParam<bool>("p_settings") == true)
		A.printSettings();

	// make me Daemon
	if (A.getParam<bool>("p_foreground") == true) {
		L += new LogConsole(A.getParam<int>("p_area"),
				    static_cast<const Level>(A.getParam<int>("p_level")),
				    "logConsole");
	} else {
		D.run("/var/run/ebus-daemon.pid");
		L += new LogFile(A.getParam<int>("p_area"),
				 static_cast<const Level>(A.getParam<int>("p_level")),
				 "logFile", A.getParam<const char*>("p_logfile"));
	}

	// trap Signals that we expect to receive
	signal(SIGHUP, signal_handler);
	signal(SIGINT, signal_handler);
	signal(SIGTERM, signal_handler);

	// start Logger
	L.start("logInstance");
	// wait for Logger be ready
	usleep(100000);
	L.log(bas, event, "ebus-daemon started");

	// create Commands DB
	commands = ConfigCommands(A.getParam<const char*>("p_ebusconfdir"), CSV).getCommands();
	L.log(bas, debug, "ebus configuration dir: %s", A.getParam<const char*>("p_ebusconfdir"));
	L.log(bas, event, "commands DB with %d entries created", commands->size());

	// create EBusLoop
	ebusloop = new EBusLoop();
	ebusloop->start("ebusloop");

	// create CYCData
	cycdata = new CYCData(ebusloop, commands);
	cycdata->start("cycdata");

	// create Network
	network = new Network(A.getParam<bool>("p_localhost"));

	// create BaseLoop
	BaseLoop baseloop(ebusloop, cycdata, commands);

	// start Network
	network->addQueue(baseloop.getQueue());
	network->start("netListener");

	// start Baseloop
	baseloop.start();

	shutdown();
}