WebSVN – HomeAutomation – Diff – /trunk/PcSoftware/Atom1.5/src/atomd.cpp


/*
 *
 *  Copyright (C) 2010  Mattias Runge
 *
 *  This program 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 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program 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 this program; if not, write to the Free Software Foundation, Inc.,
 *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 */
 
#include <signal.h>
 
#include <boost/program_options.hpp>
#include <boost/thread/mutex.hpp>
#include <boost/thread/condition.hpp>
#include <boost/thread/locks.hpp>
 
#include "config.h"
 
#include "logging/Logger.h"
#include "net/Manager.h"
#include "timer/Manager.h"
#include "config/Manager.h"
#include "can/Protocol.h"
#include "broker/Manager.h"
#include "control/Manager.h"
#include "can/Network.h"
#include "can/Monitor.h"
#include "vm/Manager.h"
#include "storage/Manager.h"
 
#include "vm/plugin/System.h"
#include "vm/plugin/Timer.h"
#include "vm/plugin/Module.h"
#include "vm/plugin/Node.h"
#include "vm/plugin/Console.h"
#include "vm/plugin/Storage.h"
#include "vm/plugin/Socket.h"
 
#ifdef USE_PLUGIN_XORG
#include "vm/plugin/Xorg.h"
#endif // USE_PLUGIN_XORG
 
using namespace atom;
 
logging::Logger LOG("Main");
std::vector<broker::Subscriber::Pointer> subscribers;
boost::condition on_message_condition;
boost::mutex guard_mutex;
 
void Handler(int status);
void CleanUp();
 
int main(int argc, char **argv)
{
    LOG.Info("\033[29;1mAtom Daemon, version " + std::string(VERSION) + " starting...\033[0m");
    LOG.Info("\033[29;1mReleased under " + std::string(LICENSE) + ".\033[0m");
    LOG.Info("Written by Mattias Runge 2010.");
   
    signal(SIGTERM, Handler);
    signal(SIGINT, Handler);
    signal(SIGQUIT, Handler);
    signal(SIGABRT, Handler);
    signal(SIGPIPE, Handler);
   
    config::Manager::Create();
   
    if (!config::Manager::Instance()->Set(argc, argv))
    {
        CleanUp();
        return EXIT_SUCCESS;
    }
   
    if (config::Manager::Instance()->Exist("LogFile"))
    {
        logging::Logger::OpenFile(config::Manager::Instance()->GetAsString("LogFile"));
    }
   
    if (config::Manager::Instance()->Exist("LogLevel"))
    {
        logging::Logger::SetLevel((logging::Logger::Level)config::Manager::Instance()->GetAsInt("LogLevel"));
    }
   
    if (config::Manager::Instance()->Exist("daemon"))
    {
        LOG.Info("Entering daemon mode...");
       
        if (daemon(0, 0) == -1)
        {
            LOG.Error("Could not enter daemon mode. Exiting...");
            CleanUp();
            return EXIT_FAILURE;
        }
       
        LOG.Info("Deamon mode entered successfully!");
    }
 
    storage::Manager::Create();
    timer::Manager::Create();
    broker::Manager::Create();
    net::Manager::Create();
    can::Protocol::Create();
    control::Manager::Create();
    vm::Manager::Create();
   
    storage::Manager::Instance()->SetRootPath(config::Manager::Instance()->GetAsString("StoragePath"));
   
    can::Protocol::Instance()->Load(config::Manager::Instance()->GetAsString("ProtocolFile"));
   
    if (config::Manager::Instance()->Exist("MonitorPort"))
    {
        subscribers.push_back(can::Monitor::Pointer(new can::Monitor(config::Manager::Instance()->GetAsInt("MonitorPort"))));  
    }
   
    subscribers.push_back(control::Manager::Instance());  
   
    vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::System(vm::Manager::Instance()->GetIoService())));
    vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Console(vm::Manager::Instance()->GetIoService(), config::Manager::Instance()->GetAsInt("CommandPort"))));
    vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Storage(vm::Manager::Instance()->GetIoService())));
    vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Timer(vm::Manager::Instance()->GetIoService())));
    vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Module(vm::Manager::Instance()->GetIoService())));
    vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Node(vm::Manager::Instance()->GetIoService())));
    vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Socket(vm::Manager::Instance()->GetIoService())));
   
#ifdef USE_PLUGIN_XORG
    vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Xorg(vm::Manager::Instance()->GetIoService())));
#endif // USE_PLUGIN_XORG
   
    vm::Manager::Instance()->Start(config::Manager::Instance()->GetAsString("ScriptPath"), config::Manager::Instance()->GetAsString("UserScriptPath"));
   
    if (config::Manager::Instance()->Exist("CanNet"))
    {
        common::StringList cannetworks = config::Manager::Instance()->GetAsStringVector("CanNet");
       
        for (int n = 0; n < cannetworks.size(); n++)
        {
            subscribers.push_back(can::Network::Pointer(new can::Network(cannetworks[n])));            
        }
    }
   
    boost::mutex::scoped_lock guard(guard_mutex);
    on_message_condition.wait(guard);
   
    LOG.Info("Cleaning up...");
   
    CleanUp();
   
    LOG.Info("Thank you for using Atom. Goodbye!");
   
    return EXIT_SUCCESS;
}
 
void CleanUp()
{
    subscribers.clear();
    config::Manager::Delete();
    timer::Manager::Delete();
    control::Manager::Delete();
    vm::Manager::Delete();
    can::Protocol::Delete();
    broker::Manager::Delete();
    net::Manager::Delete();
    storage::Manager::Delete();
}
 
void Handler(int status)
{
    std::string signal_name = "Unknown";
   
    switch (status)
    {
        case SIGTERM:
        {
            signal_name = "Terminate";
            break;
        }  
        case SIGINT:
        {
            signal_name = "Interupt";
            break;
        }  
        case SIGQUIT:
        {
            signal_name = "Quit";
            break;
        }  
        case SIGABRT:
        {
            signal_name = "Abort";
            break;
        }  
        case SIGIO:
        {
            signal_name = "I/O";
            break;
        }  
        case SIGPIPE:
        {
            signal_name = "Pipe";
            break;
        }
    }
   
    LOG.Debug("Received signal " + signal_name + "(" + boost::lexical_cast<std::string>(status) + ").");
   
    if (status != SIGPIPE)
    {
        on_message_condition.notify_all();
    }
}
 
 

Rev 1675	Rev 1679
1	/*	1	/*
2	*	2	*
3	* Copyright (C) 2010 Mattias Runge	3	* Copyright (C) 2010 Mattias Runge
4	*	4	*
5	* This program is free software; you can redistribute it and/or modify	5	* This program is free software; you can redistribute it and/or modify
6	* it under the terms of the GNU General Public License as published by	6	* it under the terms of the GNU General Public License as published by
7	* the Free Software Foundation; either version 2 of the License, or	7	* the Free Software Foundation; either version 2 of the License, or
8	* (at your option) any later version.	8	* (at your option) any later version.
9	*	9	*
10	* This program is distributed in the hope that it will be useful,	10	* This program is distributed in the hope that it will be useful,
11	* but WITHOUT ANY WARRANTY; without even the implied warranty of	11	* but WITHOUT ANY WARRANTY; without even the implied warranty of
12	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the	12	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13	* GNU General Public License for more details.	13	* GNU General Public License for more details.
14	*	14	*
15	* You should have received a copy of the GNU General Public License along	15	* You should have received a copy of the GNU General Public License along
16	* with this program; if not, write to the Free Software Foundation, Inc.,	16	* with this program; if not, write to the Free Software Foundation, Inc.,
17	* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.	17	* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18	*	18	*
19	*/	19	*/
20		20
21	#include <signal.h>	21	#include <signal.h>
22		22
23	#include <boost/program_options.hpp>	23	#include <boost/program_options.hpp>
24	#include <boost/thread/mutex.hpp>	24	#include <boost/thread/mutex.hpp>
25	#include <boost/thread/condition.hpp>	25	#include <boost/thread/condition.hpp>
26	#include <boost/thread/locks.hpp>	26	#include <boost/thread/locks.hpp>
27		27
28	#include "config.h"	28	#include "config.h"
29		29
30	#include "logging/Logger.h"	30	#include "logging/Logger.h"
31	#include "net/Manager.h"	31	#include "net/Manager.h"
32	#include "timer/Manager.h"	32	#include "timer/Manager.h"
33	#include "config/Manager.h"	33	#include "config/Manager.h"
34	#include "can/Protocol.h"	34	#include "can/Protocol.h"
35	#include "broker/Manager.h"	35	#include "broker/Manager.h"
36	#include "control/Manager.h"	36	#include "control/Manager.h"
37	#include "can/Network.h"	37	#include "can/Network.h"
38	#include "can/Monitor.h"	38	#include "can/Monitor.h"
39	#include "vm/Manager.h"	39	#include "vm/Manager.h"
40	#include "storage/Manager.h"	40	#include "storage/Manager.h"
41		41
42	#include "vm/plugin/System.h"	42	#include "vm/plugin/System.h"
43	#include "vm/plugin/Timer.h"	43	#include "vm/plugin/Timer.h"
44	#include "vm/plugin/Module.h"	44	#include "vm/plugin/Module.h"
45	#include "vm/plugin/Node.h"	45	#include "vm/plugin/Node.h"
46	#include "vm/plugin/Console.h"	46	#include "vm/plugin/Console.h"
47	#include "vm/plugin/Storage.h"	47	#include "vm/plugin/Storage.h"
48	#include "vm/plugin/Socket.h"	48	#include "vm/plugin/Socket.h"
49		49
50	#ifdef USE_PLUGIN_XORG	50	#ifdef USE_PLUGIN_XORG
51	#include "vm/plugin/Xorg.h"	51	#include "vm/plugin/Xorg.h"
52	#endif // USE_PLUGIN_XORG	52	#endif // USE_PLUGIN_XORG
53		53
54	using namespace atom;	54	using namespace atom;
55		55
56	logging::Logger LOG("Main");	56	logging::Logger LOG("Main");
57	std::vector<broker::Subscriber::Pointer> subscribers;	57	std::vector<broker::Subscriber::Pointer> subscribers;
58	boost::condition on_message_condition;	58	boost::condition on_message_condition;
59	boost::mutex guard_mutex;	59	boost::mutex guard_mutex;
60		60
61	void Handler(int status);	61	void Handler(int status);
62	void CleanUp();	62	void CleanUp();
63		63
64	int main(int argc, char **argv)	64	int main(int argc, char **argv)
65	{	65	{
66	LOG.Info("\033[29;1mAtom Daemon, version " + std::string(VERSION) + " starting...\033[0m");	66	LOG.Info("\033[29;1mAtom Daemon, version " + std::string(VERSION) + " starting...\033[0m");
67	LOG.Info("\033[29;1mReleased under ~~GPL~~ ~~version~~ 2.\033[0m");	67	LOG.Info("\033[29;1mReleased under " + std::string(LICENSE) + ".\033[0m");
68	LOG.Info("Written by Mattias Runge 2010.");	68	LOG.Info("Written by Mattias Runge 2010.");
69		69
70	signal(SIGTERM, Handler);	70	signal(SIGTERM, Handler);
71	signal(SIGINT, Handler);	71	signal(SIGINT, Handler);
72	signal(SIGQUIT, Handler);	72	signal(SIGQUIT, Handler);
73	signal(SIGABRT, Handler);	73	signal(SIGABRT, Handler);
74	signal(SIGPIPE, Handler);	74	signal(SIGPIPE, Handler);
75		75
76	config::Manager::Create();	76	config::Manager::Create();
77		77
78	if (!config::Manager::Instance()->Set(argc, argv))	78	if (!config::Manager::Instance()->Set(argc, argv))
79	{	79	{
80	CleanUp();	80	CleanUp();
81	return EXIT_SUCCESS;	81	return EXIT_SUCCESS;
82	}	82	}
83		83
84	if (config::Manager::Instance()->Exist("LogFile"))	84	if (config::Manager::Instance()->Exist("LogFile"))
85	{	85	{
86	logging::Logger::OpenFile(config::Manager::Instance()->GetAsString("LogFile"));	86	logging::Logger::OpenFile(config::Manager::Instance()->GetAsString("LogFile"));
87	}	87	}
88		88
89	if (config::Manager::Instance()->Exist("LogLevel"))	89	if (config::Manager::Instance()->Exist("LogLevel"))
90	{	90	{
91	logging::Logger::SetLevel((logging::Logger::Level)config::Manager::Instance()->GetAsInt("LogLevel"));	91	logging::Logger::SetLevel((logging::Logger::Level)config::Manager::Instance()->GetAsInt("LogLevel"));
92	}	92	}
93		93
94	if (config::Manager::Instance()->Exist("daemon"))	94	if (config::Manager::Instance()->Exist("daemon"))
95	{	95	{
96	LOG.Info("Entering daemon mode...");	96	LOG.Info("Entering daemon mode...");
97		97
98	if (daemon(0, 0) == -1)	98	if (daemon(0, 0) == -1)
99	{	99	{
100	LOG.Error("Could not enter daemon mode. Exiting...");	100	LOG.Error("Could not enter daemon mode. Exiting...");
101	CleanUp();	101	CleanUp();
102	return EXIT_FAILURE;	102	return EXIT_FAILURE;
103	}	103	}
104		104
105	LOG.Info("Deamon mode entered successfully!");	105	LOG.Info("Deamon mode entered successfully!");
106	}	106	}
107		107
108	storage::Manager::Create();	108	storage::Manager::Create();
109	timer::Manager::Create();	109	timer::Manager::Create();
110	broker::Manager::Create();	110	broker::Manager::Create();
111	net::Manager::Create();	111	net::Manager::Create();
112	can::Protocol::Create();	112	can::Protocol::Create();
113	control::Manager::Create();	113	control::Manager::Create();
114	vm::Manager::Create();	114	vm::Manager::Create();
115		115
116	storage::Manager::Instance()->SetRootPath(config::Manager::Instance()->GetAsString("StoragePath"));	116	storage::Manager::Instance()->SetRootPath(config::Manager::Instance()->GetAsString("StoragePath"));
117		117
118	can::Protocol::Instance()->Load(config::Manager::Instance()->GetAsString("ProtocolFile"));	118	can::Protocol::Instance()->Load(config::Manager::Instance()->GetAsString("ProtocolFile"));
119		119
120	if (config::Manager::Instance()->Exist("MonitorPort"))	120	if (config::Manager::Instance()->Exist("MonitorPort"))
121	{	121	{
122	subscribers.push_back(can::Monitor::Pointer(new can::Monitor(config::Manager::Instance()->GetAsInt("MonitorPort"))));	122	subscribers.push_back(can::Monitor::Pointer(new can::Monitor(config::Manager::Instance()->GetAsInt("MonitorPort"))));
123	}	123	}
124		124
125	subscribers.push_back(control::Manager::Instance());	125	subscribers.push_back(control::Manager::Instance());
126		126
127	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::System(vm::Manager::Instance()->GetIoService())));	127	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::System(vm::Manager::Instance()->GetIoService())));
128	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Console(vm::Manager::Instance()->GetIoService(), config::Manager::Instance()->GetAsInt("CommandPort"))));	128	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Console(vm::Manager::Instance()->GetIoService(), config::Manager::Instance()->GetAsInt("CommandPort"))));
129	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Storage(vm::Manager::Instance()->GetIoService())));	129	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Storage(vm::Manager::Instance()->GetIoService())));
130	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Timer(vm::Manager::Instance()->GetIoService())));	130	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Timer(vm::Manager::Instance()->GetIoService())));
131	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Module(vm::Manager::Instance()->GetIoService())));	131	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Module(vm::Manager::Instance()->GetIoService())));
132	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Node(vm::Manager::Instance()->GetIoService())));	132	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Node(vm::Manager::Instance()->GetIoService())));
133	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Socket(vm::Manager::Instance()->GetIoService())));	133	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Socket(vm::Manager::Instance()->GetIoService())));
134		134
135	#ifdef USE_PLUGIN_XORG	135	#ifdef USE_PLUGIN_XORG
136	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Xorg(vm::Manager::Instance()->GetIoService())));	136	vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Xorg(vm::Manager::Instance()->GetIoService())));
137	#endif // USE_PLUGIN_XORG	137	#endif // USE_PLUGIN_XORG
138		138
139	vm::Manager::Instance()->Start(config::Manager::Instance()->GetAsString("ScriptPath"), config::Manager::Instance()->GetAsString("UserScriptPath"));	139	vm::Manager::Instance()->Start(config::Manager::Instance()->GetAsString("ScriptPath"), config::Manager::Instance()->GetAsString("UserScriptPath"));
140		140
141	if (config::Manager::Instance()->Exist("CanNet"))	141	if (config::Manager::Instance()->Exist("CanNet"))
142	{	142	{
143	common::StringList cannetworks = config::Manager::Instance()->GetAsStringVector("CanNet");	143	common::StringList cannetworks = config::Manager::Instance()->GetAsStringVector("CanNet");
144		144
145	for (int n = 0; n < cannetworks.size(); n++)	145	for (int n = 0; n < cannetworks.size(); n++)
146	{	146	{
147	subscribers.push_back(can::Network::Pointer(new can::Network(cannetworks[n])));	147	subscribers.push_back(can::Network::Pointer(new can::Network(cannetworks[n])));
148	}	148	}
149	}	149	}
150		150
151	boost::mutex::scoped_lock guard(guard_mutex);	151	boost::mutex::scoped_lock guard(guard_mutex);
152	on_message_condition.wait(guard);	152	on_message_condition.wait(guard);
153		153
154	LOG.Info("Cleaning up...");	154	LOG.Info("Cleaning up...");
155		155
156	CleanUp();	156	CleanUp();
157		157
158	LOG.Info("Thank you for using Atom. Goodbye!");	158	LOG.Info("Thank you for using Atom. Goodbye!");
159		159
160	return EXIT_SUCCESS;	160	return EXIT_SUCCESS;
161	}	161	}
162		162
163	void CleanUp()	163	void CleanUp()
164	{	164	{
165	subscribers.clear();	165	subscribers.clear();
166	config::Manager::Delete();	166	config::Manager::Delete();
167	timer::Manager::Delete();	167	timer::Manager::Delete();
168	control::Manager::Delete();	168	control::Manager::Delete();
169	vm::Manager::Delete();	169	vm::Manager::Delete();
170	can::Protocol::Delete();	170	can::Protocol::Delete();
171	broker::Manager::Delete();	171	broker::Manager::Delete();
172	net::Manager::Delete();	172	net::Manager::Delete();
173	storage::Manager::Delete();	173	storage::Manager::Delete();
174	}	174	}
175		175
176	void Handler(int status)	176	void Handler(int status)
177	{	177	{
178	std::string signal_name = "Unknown";	178	std::string signal_name = "Unknown";
179		179
180	switch (status)	180	switch (status)
181	{	181	{
182	case SIGTERM:	182	case SIGTERM:
183	{	183	{
184	signal_name = "Terminate";	184	signal_name = "Terminate";
185	break;	185	break;
186	}	186	}
187	case SIGINT:	187	case SIGINT:
188	{	188	{
189	signal_name = "Interupt";	189	signal_name = "Interupt";
190	break;	190	break;
191	}	191	}
192	case SIGQUIT:	192	case SIGQUIT:
193	{	193	{
194	signal_name = "Quit";	194	signal_name = "Quit";
195	break;	195	break;
196	}	196	}
197	case SIGABRT:	197	case SIGABRT:
198	{	198	{
199	signal_name = "Abort";	199	signal_name = "Abort";
200	break;	200	break;
201	}	201	}
202	case SIGIO:	202	case SIGIO:
203	{	203	{
204	signal_name = "I/O";	204	signal_name = "I/O";
205	break;	205	break;
206	}	206	}
207	case SIGPIPE:	207	case SIGPIPE:
208	{	208	{
209	signal_name = "Pipe";	209	signal_name = "Pipe";
210	break;	210	break;
211	}	211	}
212	}	212	}
213		213
214	LOG.Debug("Received signal " + signal_name + "(" + boost::lexical_cast<std::string>(status) + ").");	214	LOG.Debug("Received signal " + signal_name + "(" + boost::lexical_cast<std::string>(status) + ").");
215		215
216	if (status != SIGPIPE)	216	if (status != SIGPIPE)
217	{	217	{
218	on_message_condition.notify_all();	218	on_message_condition.notify_all();
219	}	219	}
220	}	220	}
221		221
222		222

Subversion Repositories HomeAutomation

(root)/trunk/PcSoftware/Atom1.5/src/atomd.cpp – Rev 1675 → 1679