Initial Commit

2018-12-18 12:07:12 +01:00
commit 7ad040c6c1
31 changed files with 1974 additions and 0 deletions
@@ -0,0 +1,177 @@
+#include <iostream>
+#include <string>
+#include <signal.h>
+#include <syslog.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <boost/lexical_cast.hpp>
+#include <boost/program_options.hpp>
+#include <boost/property_tree/ptree.hpp>
+#include <boost/property_tree/ini_parser.hpp>
+#include "UniFi/Device.h"
+#include "Bluetooth/Device.h"
+
+
+int main(int argc, char** argv)
+{
+	try
+	{
+		setlogmask(LOG_UPTO(LOG_INFO));
+		openlog("PresenceDetection", LOG_CONS | LOG_NDELAY | LOG_PERROR | LOG_PID, LOG_USER);
+
+		boost::program_options::options_description description("Options");
+		description.add_options()
+			("help,h", "Provide Help Message")
+			("daemon,d", "Run the process as Daemon");
+
+		boost::program_options::variables_map vm;
+		boost::program_options::store(boost::program_options::parse_command_line(argc, argv, description), vm);
+		boost::program_options::notify(vm);
+
+		if (vm.count("help"))
+		{
+			std::cerr << description << std::endl;
+			return 1;
+		}
+
+		bool daemon = false;
+		if (vm.count("daemon"))
+			daemon = true;
+
+		boost::property_tree::ptree pt;
+		boost::property_tree::ini_parser::read_ini("PresenceDetection.ini", pt);
+
+		if (daemon)
+		{
+			syslog(LOG_INFO, "Starting Daemon");
+			pid_t pid, sid;
+			pid = fork();
+
+			if (pid < 0)
+				exit(EXIT_FAILURE);
+			if (pid > 0)
+				exit(EXIT_SUCCESS);
+
+			umask(0);
+			sid = setsid();
+			if (sid < 0)
+				exit(EXIT_FAILURE);
+
+			if ((chdir("/")) < 0)
+				exit(EXIT_FAILURE);
+
+			close(STDIN_FILENO);
+			close(STDOUT_FILENO);
+			close(STDERR_FILENO);
+		}
+
+		int port;
+		boost::optional<boost::property_tree::ptree&> presenceDetectionPort = pt.get_child_optional("PresenceDetection.Port");
+		if (presenceDetectionPort)
+			port = pt.get<int>("PresenceDetection.Port");
+		else
+			throw std::runtime_error("Port directive missing in ini file.");
+
+		std::string target;
+		boost::optional<boost::property_tree::ptree&> presenceDetectionTarget = pt.get_child_optional("PresenceDetection.Target");
+		if (presenceDetectionTarget)
+			target = pt.get<std::string>("PresenceDetection.Target");
+		else
+			throw std::runtime_error("Target directive missing in ini file.");
+
+		bool unifi = false;
+		boost::optional<boost::property_tree::ptree&> presenceDetectionUniFi = pt.get_child_optional("PresenceDetection.UniFi");
+		if (presenceDetectionUniFi)
+			unifi = (pt.get<std::string>("PresenceDetection.UniFi") == "On");
+
+		bool bluetooth = false;
+		boost::optional<boost::property_tree::ptree&> presenceDetectionBluetooth = pt.get_child_optional("PresenceDetection.Bluetooth");
+		if (presenceDetectionBluetooth)
+			bluetooth = (pt.get<std::string>("PresenceDetection.Bluetooth") == "On");
+
+		std::shared_ptr<PresenceDetection::UniFi::Device> pUniFiDevice;
+		if (unifi)
+		{
+			std::string hostname;
+			boost::optional<boost::property_tree::ptree&> unifiHostname = pt.get_child_optional("UniFi.Hostname");
+			if (unifiHostname)
+				hostname = pt.get<std::string>("UniFi.Hostname");
+			else
+				throw std::runtime_error("UniFi Hostname directive missing in ini file.");
+
+			std::string username;
+			boost::optional<boost::property_tree::ptree&> unifiUsername = pt.get_child_optional("UniFi.Username");
+			if (unifiUsername)
+				username = pt.get<std::string>("UniFi.Username");
+			else
+				throw std::runtime_error("UniFi Username directive missing in ini file.");
+
+			std::string password;
+			boost::optional<boost::property_tree::ptree&> unifiPassword = pt.get_child_optional("UniFi.Password");
+			if (unifiPassword)
+				password = pt.get<std::string>("UniFi.Password");
+			else
+				throw std::runtime_error("UniFi Password directive missing in ini file.");
+
+			std::string cookiefile;
+			boost::optional<boost::property_tree::ptree&> unifiCookieFile = pt.get_child_optional("UniFi.CookieFile");
+			if (unifiCookieFile)
+				cookiefile = pt.get<std::string>("UniFi.CookieFile");
+			else
+				throw std::runtime_error("UniFi CookieFile directive missing in ini file.");
+
+			pUniFiDevice = std::make_shared<PresenceDetection::UniFi::Device>(hostname, username, password, cookiefile, target);
+		}
+
+		std::shared_ptr<PresenceDetection::Bluetooth::Device> pBluetoothDevice;
+		if (bluetooth)
+		{
+			std::string devices;
+			boost::optional<boost::property_tree::ptree&> bluetoothDevices = pt.get_child_optional("Bluetooth.Devices");
+			if (bluetoothDevices)
+				devices = pt.get<std::string>("Bluetooth.Devices");
+			else
+				throw std::runtime_error("Bluetooth Devices directive missing in ini file.");
+
+			pBluetoothDevice = std::make_shared<PresenceDetection::Bluetooth::Device>(devices, target);
+		}
+
+		sigset_t wset;
+		sigemptyset(&wset);
+		sigaddset(&wset,SIGHUP);
+		sigaddset(&wset,SIGINT);
+		sigaddset(&wset,SIGTERM);
+		int sig;
+		sigwait(&wset,&sig);
+
+		syslog(LOG_INFO, "Stopping...");
+
+		if (pUniFiDevice)
+			pUniFiDevice->Stop();
+
+		if (pBluetoothDevice)
+			pBluetoothDevice->Stop();
+
+		if (pUniFiDevice)
+			pUniFiDevice->Wait();
+
+		if (pBluetoothDevice)
+			pBluetoothDevice->Wait();
+
+		closelog();
+	}
+	catch (const std::exception& e)
+	{
+		std::stringstream ss;
+		ss << "ERROR: " << e.what() << std::endl;
+
+		std::cerr << ss.str() << std::endl;
+		syslog(LOG_ERR, "%s", ss.str().c_str());
+
+		closelog();
+
+		return -1;
+	}
+
+	return 0;
+}