Subversion Repositories HomeAutomation

Rev

Rev 1939 | Rev 1945 | Go to most recent revision | Only display areas with differences | Regard whitespace | Details | Blame | Last modification | View Log | SVN | RSS feed

Rev 1939 Rev 1940
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 "can/CanDaemon.h"
39
#include "can/CanDaemon.h"
40
#include "vm/Manager.h"
40
#include "vm/Manager.h"
41
#include "storage/Manager.h"
41
#include "storage/Manager.h"
42
#include "mbb/Manager.h"
42
#include "mbb/Manager.h"
43
 
43
 
44
#include "vm/plugin/System.h"
44
#include "vm/plugin/System.h"
45
#include "vm/plugin/Timer.h"
45
#include "vm/plugin/Timer.h"
46
#include "vm/plugin/Module.h"
46
#include "vm/plugin/Module.h"
47
#include "vm/plugin/Node.h"
47
#include "vm/plugin/Node.h"
48
#include "vm/plugin/Console.h"
48
#include "vm/plugin/Console.h"
49
#include "vm/plugin/Storage.h"
49
#include "vm/plugin/Storage.h"
50
#include "vm/plugin/Socket.h"
50
#include "vm/plugin/Socket.h"
51
 
51
 
52
#ifdef USE_PLUGIN_XORG
52
#ifdef USE_PLUGIN_XORG
53
#include "vm/plugin/Xorg.h"
53
#include "vm/plugin/Xorg.h"
54
#endif // USE_PLUGIN_XORG
54
#endif // USE_PLUGIN_XORG
55
 
55
 
56
using namespace atom;
56
using namespace atom;
57
 
57
 
58
logging::Logger LOG ("Main");
58
logging::Logger LOG ("Main");
59
std::vector<broker::Subscriber::Pointer> subscribers;
59
std::vector<broker::Subscriber::Pointer> subscribers;
60
boost::condition on_message_condition;
60
boost::condition on_message_condition;
61
boost::mutex guard_mutex;
61
boost::mutex guard_mutex;
62
 
62
 
63
void Handler (int status);
63
void Handler (int status);
64
void CleanUp();
64
void CleanUp();
65
 
65
 
66
int main (int argc, char **argv)
66
int main (int argc, char **argv)
67
{
67
{
68
  LOG.Info ("\033[29;1mAtom Daemon, version " + std::string (VERSION) + " starting...\033[0m");
68
  LOG.Info ("\033[29;1mAtom Daemon, version " + std::string (VERSION) + " starting...\033[0m");
69
  LOG.Info ("\033[29;1mReleased under " + std::string (LICENSE) + ".\033[0m");
69
  LOG.Info ("\033[29;1mReleased under " + std::string (LICENSE) + ".\033[0m");
70
  LOG.Info ("Written by Mattias Runge 2010-2012.");
70
  LOG.Info ("Written by Mattias Runge 2010-2012.");
71
 
71
 
72
  signal (SIGTERM, Handler);
72
  signal (SIGTERM, Handler);
73
  signal (SIGINT, Handler);
73
  signal (SIGINT, Handler);
74
  signal (SIGQUIT, Handler);
74
  signal (SIGQUIT, Handler);
75
  signal (SIGABRT, Handler);
75
  signal (SIGABRT, Handler);
76
  signal (SIGPIPE, Handler);
76
  signal (SIGPIPE, Handler);
77
 
77
 
78
  config::Manager::Create();
78
  config::Manager::Create();
79
 
79
 
80
  if (!config::Manager::Instance()->Set (argc, argv))
80
  if (!config::Manager::Instance()->Set (argc, argv))
81
  {
81
  {
82
    CleanUp();
82
    CleanUp();
83
    return EXIT_SUCCESS;
83
    return EXIT_SUCCESS;
84
  }
84
  }
85
 
85
 
86
  if (config::Manager::Instance()->Exist ("LogFile"))
86
  if (config::Manager::Instance()->Exist ("LogFile"))
87
  {
87
  {
88
    logging::Logger::OpenFile (config::Manager::Instance()->GetAsString ("LogFile"));
88
    logging::Logger::OpenFile (config::Manager::Instance()->GetAsString ("LogFile"));
89
  }
89
  }
90
 
90
 
91
  if (config::Manager::Instance()->Exist ("LogLevel"))
91
  if (config::Manager::Instance()->Exist ("LogLevel"))
92
  {
92
  {
93
    logging::Logger::SetLevel ( (logging::Logger::Level) config::Manager::Instance()->GetAsInt ("LogLevel"));
93
    logging::Logger::SetLevel ( (logging::Logger::Level) config::Manager::Instance()->GetAsInt ("LogLevel"));
94
  }
94
  }
95
 
95
 
96
  if (config::Manager::Instance()->Exist ("daemon"))
96
  if (config::Manager::Instance()->Exist ("daemon"))
97
  {
97
  {
98
    LOG.Info ("Entering daemon mode...");
98
    LOG.Info ("Entering daemon mode...");
99
 
99
 
100
    if (daemon (0, 0) == -1)
100
    if (daemon (0, 0) == -1)
101
    {
101
    {
102
      LOG.Error ("Could not enter daemon mode. Exiting...");
102
      LOG.Error ("Could not enter daemon mode. Exiting...");
103
      CleanUp();
103
      CleanUp();
104
      return EXIT_FAILURE;
104
      return EXIT_FAILURE;
105
    }
105
    }
106
 
106
 
107
    LOG.Info ("Deamon mode entered successfully!");
107
    LOG.Info ("Deamon mode entered successfully!");
108
  }
108
  }
109
 
109
 
110
  storage::Manager::Create();
110
  storage::Manager::Create();
111
 
111
 
112
  timer::Manager::Create();
112
  timer::Manager::Create();
113
  broker::Manager::Create();
113
  broker::Manager::Create();
114
  net::Manager::Create();
114
  net::Manager::Create();
115
  can::Protocol::Create();
115
  can::Protocol::Create();
116
  control::Manager::Create();
116
  control::Manager::Create();
117
  vm::Manager::Create();
117
  vm::Manager::Create();
118
 
118
 
119
  if (config::Manager::Instance()->Exist ("MbbPort"))
119
  if (config::Manager::Instance()->Exist ("MbbPort"))
120
  {
120
  {
121
    mbb::Manager::Create (config::Manager::Instance()->GetAsInt ("MbbPort"));
121
    mbb::Manager::Create (config::Manager::Instance()->GetAsInt ("MbbPort"));
122
  }
122
  }
123
 
123
 
124
  storage::Manager::Instance()->SetRootPath (config::Manager::Instance()->GetAsString ("StoragePath"));
124
  storage::Manager::Instance()->SetRootPath (config::Manager::Instance()->GetAsString ("StoragePath"));
125
 
125
 
126
  can::Protocol::Instance()->Load (config::Manager::Instance()->GetAsString ("ProtocolFile"));
126
  can::Protocol::Instance()->Load (config::Manager::Instance()->GetAsString ("ProtocolFile"));
127
 
127
 
128
  if (config::Manager::Instance()->Exist ("MonitorPort"))
128
  if (config::Manager::Instance()->Exist ("MonitorPort"))
129
  {
129
  {
130
    subscribers.push_back (can::Monitor::Pointer (new can::Monitor (config::Manager::Instance()->GetAsInt ("MonitorPort"))));
130
    subscribers.push_back (can::Monitor::Pointer (new can::Monitor (config::Manager::Instance()->GetAsInt ("MonitorPort"))));
131
  }
131
  }
132
 
132
 
133
  if (config::Manager::Instance()->Exist ("DaemonPort"))
133
  if (config::Manager::Instance()->Exist ("DaemonPort"))
134
  {
134
  {
135
    subscribers.push_back (can::CanDaemon::Pointer (new can::CanDaemon (config::Manager::Instance()->GetAsInt ("DaemonPort"))));
135
    subscribers.push_back (can::CanDaemon::Pointer (new can::CanDaemon (config::Manager::Instance()->GetAsInt ("DaemonPort"))));
136
  }
136
  }
137
 
137
 
138
  subscribers.push_back (control::Manager::Instance());
138
  subscribers.push_back (control::Manager::Instance());
139
 
139
 
140
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::System (vm::Manager::Instance()->GetIoService())));
140
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::System (vm::Manager::Instance()->GetIoService())));
141
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Console (vm::Manager::Instance()->GetIoService(), config::Manager::Instance()->GetAsInt ("CommandPort"))));
141
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Console (vm::Manager::Instance()->GetIoService(), config::Manager::Instance()->GetAsInt ("CommandPort"))));
142
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Storage (vm::Manager::Instance()->GetIoService())));
142
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Storage (vm::Manager::Instance()->GetIoService())));
143
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Timer (vm::Manager::Instance()->GetIoService())));
143
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Timer (vm::Manager::Instance()->GetIoService())));
144
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Module (vm::Manager::Instance()->GetIoService())));
144
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Module (vm::Manager::Instance()->GetIoService())));
145
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Node (vm::Manager::Instance()->GetIoService())));
145
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Node (vm::Manager::Instance()->GetIoService())));
146
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Socket (vm::Manager::Instance()->GetIoService())));
146
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Socket (vm::Manager::Instance()->GetIoService())));
147
 
147
 
148
#ifdef USE_PLUGIN_XORG
148
#ifdef USE_PLUGIN_XORG
149
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Xorg (vm::Manager::Instance()->GetIoService())));
149
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Xorg (vm::Manager::Instance()->GetIoService())));
150
#endif // USE_PLUGIN_XORG
150
#endif // USE_PLUGIN_XORG
151
 
151
 
152
  vm::Manager::Instance()->Start (config::Manager::Instance()->GetAsString ("ScriptPath"), config::Manager::Instance()->GetAsString ("UserScriptPath"));
152
  vm::Manager::Instance()->Start (config::Manager::Instance()->GetAsString ("ScriptPath"), config::Manager::Instance()->GetAsString ("UserScriptPath"));
153
 
153
 
154
  if (config::Manager::Instance()->Exist ("CanNet"))
154
  if (config::Manager::Instance()->Exist ("CanNet"))
155
  {
155
  {
156
    common::StringList cannetworks = config::Manager::Instance()->GetAsStringVector ("CanNet");
156
    common::StringList cannetworks = config::Manager::Instance()->GetAsStringVector ("CanNet");
157
 
157
 
158
    for (uint n = 0; n < cannetworks.size(); n++)
158
    for (uint n = 0; n < cannetworks.size(); n++)
159
    {
159
    {
160
      subscribers.push_back (can::Network::Pointer (new can::Network (cannetworks[n])));
160
      subscribers.push_back (can::Network::Pointer (new can::Network (cannetworks[n])));
161
    }
161
    }
162
  }
162
  }
163
 
163
 
164
  boost::mutex::scoped_lock guard (guard_mutex);
164
  boost::mutex::scoped_lock guard (guard_mutex);
165
 
165
 
166
  on_message_condition.wait (guard);
166
  on_message_condition.wait (guard);
167
 
167
 
168
  LOG.Info ("Cleaning up...");
168
  LOG.Info ("Cleaning up...");
169
 
169
 
170
  CleanUp();
170
  CleanUp();
171
 
171
 
172
  LOG.Info ("Thank you for using Atom. Goodbye!");
172
  LOG.Info ("Thank you for using Atom. Goodbye!");
173
 
173
 
174
  return EXIT_SUCCESS;
174
  return EXIT_SUCCESS;
175
}
175
}
176
 
176
 
177
void CleanUp()
177
void CleanUp()
178
{
178
{
179
  net::Manager::Instance()->Stop();
179
  net::Manager::Instance()->Stop();
180
 
180
 
181
  subscribers.clear();
181
  subscribers.clear();
182
  config::Manager::Delete();
182
  config::Manager::Delete();
183
  timer::Manager::Delete();
183
  timer::Manager::Delete();
184
  control::Manager::Delete();
184
  control::Manager::Delete();
185
  mbb::Manager::Delete();
185
  mbb::Manager::Delete();
186
  can::Protocol::Delete();
186
  can::Protocol::Delete();
187
  broker::Manager::Delete();
187
  broker::Manager::Delete();
188
  vm::Manager::Delete();
188
  vm::Manager::Delete();
189
  net::Manager::Delete();
189
  net::Manager::Delete();
190
  storage::Manager::Delete();
190
  storage::Manager::Delete();
191
}
191
}
192
 
192
 
193
void Handler (int status)
193
void Handler (int status)
194
{
194
{
195
  std::string signal_name = "Unknown";
195
  std::string signal_name = "Unknown";
196
 
196
 
197
  switch (status)
197
  switch (status)
198
  {
198
  {
199
    case SIGTERM:
199
    case SIGTERM:
200
    {
200
    {
201
      signal_name = "Terminate";
201
      signal_name = "Terminate";
202
      break;
202
      break;
203
    }
203
    }
204
 
204
 
205
    case SIGINT:
205
    case SIGINT:
206
    {
206
    {
207
      signal_name = "Interupt";
207
      signal_name = "Interrupt";
208
      break;
208
      break;
209
    }
209
    }
210
 
210
 
211
    case SIGQUIT:
211
    case SIGQUIT:
212
    {
212
    {
213
      signal_name = "Quit";
213
      signal_name = "Quit";
214
      break;
214
      break;
215
    }
215
    }
216
 
216
 
217
    case SIGABRT:
217
    case SIGABRT:
218
    {
218
    {
219
      signal_name = "Abort";
219
      signal_name = "Abort";
220
      break;
220
      break;
221
    }
221
    }
222
 
222
 
223
    case SIGIO:
223
    case SIGIO:
224
    {
224
    {
225
      signal_name = "I/O";
225
      signal_name = "I/O";
226
      break;
226
      break;
227
    }
227
    }
228
 
228
 
229
    case SIGPIPE:
229
    case SIGPIPE:
230
    {
230
    {
231
      signal_name = "Pipe";
231
      signal_name = "Pipe";
232
      break;
232
      break;
233
    }
233
    }
234
  }
234
  }
235
 
235
 
236
  LOG.Debug ("Received signal " + signal_name + "(" + boost::lexical_cast<std::string> (status) + ").");
236
  LOG.Debug ("Received signal " + signal_name + "(" + boost::lexical_cast<std::string> (status) + ").");
237
 
237
 
238
  if (status != SIGPIPE)
238
  if (status != SIGPIPE)
239
  {
239
  {
240
    on_message_condition.notify_all();
240
    on_message_condition.notify_all();
241
  }
241
  }
242
}
242
}
243
 
243