Rev 2107 | Show entire file | Ignore whitespace | Details | Blame | Last modification | View Log | SVN | RSS feed
| Rev 2107 | Rev 2216 | ||
|---|---|---|---|
| Line 37... | Line 37... | ||
| 37 | #include "broker/Manager.h" |
37 | #include "broker/Manager.h" |
| 38 | #include "control/Manager.h" |
38 | #include "control/Manager.h" |
| 39 | #include "can/Network.h" |
39 | #include "can/Network.h" |
| 40 | #include "can/Monitor.h" |
40 | #include "can/Monitor.h" |
| 41 | #include "can/CanDaemon.h" |
41 | #include "can/CanDaemon.h" |
| - | 42 | #include "can/Forward.h" |
|
| 42 | #include "vm/Manager.h" |
43 | #include "vm/Manager.h" |
| 43 | #include "storage/Manager.h" |
44 | #include "storage/Manager.h" |
| 44 | #include "net/GetFile.h" |
45 | #include "net/GetFile.h" |
| 45 | #include "net/Url.h" |
46 | #include "net/Url.h" |
| 46 | 47 | ||
| Line 93... | Line 94... | ||
| 93 | int main(int argc, char **argv) |
94 | int main(int argc, char **argv) |
| 94 | { |
95 | { |
| 95 | /* Set core dump parameters */ |
96 | /* Set core dump parameters */ |
| 96 | rlimit core_limit = { RLIM_INFINITY, RLIM_INFINITY }; |
97 | rlimit core_limit = { RLIM_INFINITY, RLIM_INFINITY }; |
| 97 | setrlimit( RLIMIT_CORE, &core_limit ); // enable core dumps |
98 | setrlimit( RLIMIT_CORE, &core_limit ); // enable core dumps |
| 98 | 99 | ||
| 99 | 100 | ||
| 100 | /* Print startup information */ |
101 | /* Print startup information */ |
| 101 | log::Info(log_module_, "Atom daemon, version %s starting...", VERSION); |
102 | log::Info(log_module_, "Atom daemon, version %s starting...", VERSION); |
| 102 | log::Info(log_module_, "Released under %s.", LICENSE); |
103 | log::Info(log_module_, "Released under %s.", LICENSE); |
| 103 | log::Info(log_module_, "Written by Mattias Runge 2010-2012."); |
104 | log::Info(log_module_, "Written by Mattias Runge 2010-2012."); |
| 104 | 105 | ||
| 105 | 106 | ||
| 106 | /* Register signal handlers */ |
107 | /* Register signal handlers */ |
| 107 | signal(SIGTERM, Handler); |
108 | signal(SIGTERM, Handler); |
| 108 | signal(SIGINT, Handler); |
109 | signal(SIGINT, Handler); |
| 109 | signal(SIGQUIT, Handler); |
110 | signal(SIGQUIT, Handler); |
| 110 | signal(SIGABRT, Handler); |
111 | signal(SIGABRT, Handler); |
| 111 | signal(SIGPIPE, Handler); |
112 | signal(SIGPIPE, Handler); |
| 112 | 113 | ||
| 113 | 114 | ||
| 114 | /* Check configuration */ |
115 | /* Check configuration */ |
| 115 | config::Manager::Create(); |
116 | config::Manager::Create(); |
| 116 | 117 | ||
| 117 | if (!config::Manager::Instance()->Set(argc, argv)) |
118 | if (!config::Manager::Instance()->Set(argc, argv)) |
| 118 | { |
119 | { |
| 119 | CleanUp(); |
120 | CleanUp(); |
| 120 | return EXIT_SUCCESS; |
121 | return EXIT_SUCCESS; |
| 121 | } |
122 | } |
| 122 | 123 | ||
| 123 | 124 | ||
| 124 | /* Set up logging */ |
125 | /* Set up logging */ |
| 125 | if (config::Manager::Instance()->Exist("LogFile")) |
126 | if (config::Manager::Instance()->Exist("LogFile")) |
| 126 | { |
127 | { |
| 127 | log::OpenFile(config::Manager::Instance()->GetAsString("LogFile")); |
128 | log::OpenFile(config::Manager::Instance()->GetAsString("LogFile")); |
| 128 | } |
129 | } |
| 129 | 130 | ||
| 130 | if (config::Manager::Instance()->Exist("LogLevelMask")) |
131 | if (config::Manager::Instance()->Exist("LogLevelMask")) |
| 131 | { |
132 | { |
| 132 | log::SetLevelByString(config::Manager::Instance()->GetAsString("LogLevelMask")); |
133 | log::SetLevelByString(config::Manager::Instance()->GetAsString("LogLevelMask")); |
| 133 | } |
134 | } |
| 134 | 135 | ||
| 135 | 136 | ||
| 136 | /* Enter daemon mode if requested */ |
137 | /* Enter daemon mode if requested */ |
| 137 | if (config::Manager::Instance()->Exist("daemon")) |
138 | if (config::Manager::Instance()->Exist("daemon")) |
| 138 | { |
139 | { |
| 139 | LOG.Info("Entering daemon mode..."); |
140 | LOG.Info("Entering daemon mode..."); |
| 140 | 141 | ||
| Line 146... | Line 147... | ||
| 146 | } |
147 | } |
| 147 | 148 | ||
| 148 | log::Info(log_module_, "Deamon mode entered successfully!"); |
149 | log::Info(log_module_, "Deamon mode entered successfully!"); |
| 149 | } |
150 | } |
| 150 | 151 | ||
| 151 | 152 | ||
| 152 | /* Create singletons */ |
153 | /* Create singletons */ |
| 153 | storage::Manager::Create(); |
154 | storage::Manager::Create(); |
| 154 | timer::Manager::Create(); |
155 | timer::Manager::Create(); |
| 155 | broker::Manager::Create(); |
156 | broker::Manager::Create(); |
| 156 | net::Manager::Create(); |
157 | net::Manager::Create(); |
| 157 | can::Protocol::Create(); |
158 | can::Protocol::Create(); |
| 158 | control::Manager::Create(); |
159 | control::Manager::Create(); |
| 159 | vm::Manager::Create(); |
160 | vm::Manager::Create(); |
| 160 | 161 | ||
| 161 | 162 | ||
| 162 | 163 | ||
| 163 | std::string protocol_filename = config::Manager::Instance()->GetAsString("ProtocolFile"); |
164 | std::string protocol_filename = config::Manager::Instance()->GetAsString("ProtocolFile"); |
| 164 | 165 | ||
| 165 | try |
166 | try |
| 166 | { |
167 | { |
| 167 | net::Url url(protocol_filename); |
168 | net::Url url(protocol_filename); |
| 168 | 169 | ||
| 169 | downloader = net::GetFile::Pointer(new net::GetFile()); |
170 | downloader = net::GetFile::Pointer(new net::GetFile()); |
| 170 | 171 | ||
| 171 | log::Info(log_module_, "Will try to download protocol file from \"%s\"...", protocol_filename.data()); |
172 | log::Info(log_module_, "Will try to download protocol file from \"%s\"...", protocol_filename.data()); |
| 172 | 173 | ||
| 173 | downloader->Start(url, HandleProtocolFileDownload); |
174 | downloader->Start(url, HandleProtocolFileDownload); |
| 174 | } |
175 | } |
| 175 | catch (std::exception& exception) |
176 | catch (std::exception& exception) |
| 176 | { |
177 | { |
| 177 | log::Debug(log_module_, "Could not parse ProtocolFile as a URL, \"%s\".", protocol_filename.data()); |
178 | log::Debug(log_module_, "Could not parse ProtocolFile as a URL, \"%s\".", protocol_filename.data()); |
| 178 | ContinueInitialization(protocol_filename); |
179 | ContinueInitialization(protocol_filename); |
| 179 | } |
180 | } |
| 180 | 181 | ||
| 181 | 182 | ||
| 182 | /* Lock main thread and wait for exit */ |
183 | /* Lock main thread and wait for exit */ |
| 183 | boost::mutex::scoped_lock guard(guard_mutex); |
184 | boost::mutex::scoped_lock guard(guard_mutex); |
| 184 | 185 | ||
| 185 | on_message_condition.wait(guard); |
186 | on_message_condition.wait(guard); |
| 186 | 187 | ||
| 187 | 188 | ||
| 188 | /* Cleanup */ |
189 | /* Cleanup */ |
| 189 | log::Info(log_module_, "Cleaning up..."); |
190 | log::Info(log_module_, "Cleaning up..."); |
| 190 | 191 | ||
| 191 | CleanUp(); |
192 | CleanUp(); |
| 192 | 193 | ||
| 193 | log::Info(log_module_, "Thank you for using Atom. Goodbye!"); |
194 | log::Info(log_module_, "Thank you for using Atom. Goodbye!"); |
| 194 | 195 | ||
| 195 | return EXIT_SUCCESS; |
196 | return EXIT_SUCCESS; |
| 196 | } |
197 | } |
| 197 | 198 | ||
| 198 | void ContinueInitialization(std::string protocol_filename) |
199 | void ContinueInitialization(std::string protocol_filename) |
| 199 | { |
200 | { |
| 200 | /* Load protocol file */ |
201 | /* Load protocol file */ |
| 201 | if (!can::Protocol::Instance()->Load(protocol_filename)) |
202 | if (!can::Protocol::Instance()->Load(protocol_filename)) |
| 202 | { |
203 | { |
| 203 | log::Error(log_module_, "Failed to load %s!", protocol_filename.data()); |
204 | log::Error(log_module_, "Failed to load %s!", protocol_filename.data()); |
| 204 | 205 | ||
| 205 | on_message_condition.notify_all(); |
206 | on_message_condition.notify_all(); |
| 206 | } |
207 | } |
| 207 | 208 | ||
| 208 | 209 | ||
| 209 | /* Set up storage path */ |
210 | /* Set up storage path */ |
| 210 | storage::Manager::Instance()->SetRootPath(config::Manager::Instance()->GetAsString("StoragePath")); |
211 | storage::Manager::Instance()->SetRootPath(config::Manager::Instance()->GetAsString("StoragePath")); |
| 211 | 212 | ||
| 212 | 213 | ||
| 213 | /* Start monitor server */ |
214 | /* Start monitor server */ |
| 214 | if (config::Manager::Instance()->Exist("MonitorPort")) |
215 | if (config::Manager::Instance()->Exist("MonitorPort")) |
| 215 | { |
216 | { |
| 216 | subscribers.push_back(can::Monitor::Pointer(new can::Monitor(config::Manager::Instance()->GetAsInt("MonitorPort")))); |
217 | subscribers.push_back(can::Monitor::Pointer(new can::Monitor(config::Manager::Instance()->GetAsInt("MonitorPort")))); |
| 217 | } |
218 | } |
| 218 | 219 | ||
| 219 | 220 | ||
| 220 | /* Start CanDaemon server */ |
221 | /* Start CanDaemon server */ |
| 221 | if (config::Manager::Instance()->Exist("DaemonPort")) |
222 | if (config::Manager::Instance()->Exist("DaemonPort")) |
| 222 | { |
223 | { |
| 223 | subscribers.push_back(can::CanDaemon::Pointer(new can::CanDaemon(config::Manager::Instance()->GetAsInt("DaemonPort")))); |
224 | subscribers.push_back(can::CanDaemon::Pointer(new can::CanDaemon(config::Manager::Instance()->GetAsInt("DaemonPort")))); |
| 224 | } |
225 | } |
| 225 | 226 | ||
| - | 227 | /* Start CAN forward server */ |
|
| - | 228 | if (config::Manager::Instance()->Exist("CanForwardPort")) |
|
| - | 229 | { |
|
| - | 230 | subscribers.push_back(can::Forward::Pointer(new can::Forward(config::Manager::Instance()->GetAsInt("CanForwardPort")))); |
|
| 226 | 231 | } |
|
| - | 232 | ||
| - | 233 | ||
| - | 234 | ||
| 227 | /* Subscribe Can control manager to data */ |
235 | /* Subscribe Can control manager to data */ |
| 228 | subscribers.push_back(control::Manager::Instance()); |
236 | subscribers.push_back(control::Manager::Instance()); |
| 229 | 237 | ||
| 230 | 238 | ||
| 231 | /* Start VM plugins */ |
239 | /* Start VM plugins */ |
| 232 | vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::System(vm::Manager::Instance()->GetIoService()))); |
240 | vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::System(vm::Manager::Instance()->GetIoService()))); |
| 233 | vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Console(vm::Manager::Instance()->GetIoService(), config::Manager::Instance()->GetAsInt("CommandPort")))); |
241 | vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Console(vm::Manager::Instance()->GetIoService(), config::Manager::Instance()->GetAsInt("CommandPort")))); |
| 234 | vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::JsPipe(vm::Manager::Instance()->GetIoService(), config::Manager::Instance()->GetAsInt("JsPipePort")))); |
242 | vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::JsPipe(vm::Manager::Instance()->GetIoService(), config::Manager::Instance()->GetAsInt("JsPipePort")))); |
| 235 | vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Storage(vm::Manager::Instance()->GetIoService()))); |
243 | vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Storage(vm::Manager::Instance()->GetIoService()))); |
| Line 239... | Line 247... | ||
| 239 | vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Socket(vm::Manager::Instance()->GetIoService()))); |
247 | vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Socket(vm::Manager::Instance()->GetIoService()))); |
| 240 | 248 | ||
| 241 | #ifdef USE_PLUGIN_XORG |
249 | #ifdef USE_PLUGIN_XORG |
| 242 | vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Xorg(vm::Manager::Instance()->GetIoService()))); |
250 | vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::Xorg(vm::Manager::Instance()->GetIoService()))); |
| 243 | #endif // USE_PLUGIN_XORG |
251 | #endif // USE_PLUGIN_XORG |
| 244 | 252 | ||
| 245 | #ifdef USE_PLUGIN_MYSQL |
253 | #ifdef USE_PLUGIN_MYSQL |
| 246 | vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::MySql(vm::Manager::Instance()->GetIoService()))); |
254 | vm::Manager::Instance()->AddPlugin(vm::Plugin::Pointer(new vm::plugin::MySql(vm::Manager::Instance()->GetIoService()))); |
| 247 | #endif // USE_PLUGIN_MYSQL |
255 | #endif // USE_PLUGIN_MYSQL |
| 248 | 256 | ||
| 249 | 257 | ||
| 250 | /* Start VM */ |
258 | /* Start VM */ |
| 251 | vm::Manager::Instance()->Start(config::Manager::Instance()->GetAsString("ScriptPath"), config::Manager::Instance()->GetAsString("UserScriptPath")); |
259 | vm::Manager::Instance()->Start(config::Manager::Instance()->GetAsString("ScriptPath"), config::Manager::Instance()->GetAsString("UserScriptPath")); |
| 252 | 260 | ||
| 253 | 261 | ||
| 254 | /* Connect to Can network */ |
262 | /* Connect to Can network */ |
| 255 | if (config::Manager::Instance()->Exist("CanNet")) |
263 | if (config::Manager::Instance()->Exist("CanNet")) |
| 256 | { |
264 | { |
| 257 | common::StringList cannetworks = config::Manager::Instance()->GetAsStringVector("CanNet"); |
265 | common::StringList cannetworks = config::Manager::Instance()->GetAsStringVector("CanNet"); |
| 258 | 266 | ||
| Line 264... | Line 272... | ||
| 264 | } |
272 | } |
| 265 | 273 | ||
| 266 | void CleanUp() |
274 | void CleanUp() |
| 267 | { |
275 | { |
| 268 | downloader.reset(); |
276 | downloader.reset(); |
| 269 | 277 | ||
| 270 | if (net::Manager::Instance().use_count() > 0) |
278 | if (net::Manager::Instance().use_count() > 0) |
| 271 | { |
279 | { |
| 272 | net::Manager::Instance()->Stop(); |
280 | net::Manager::Instance()->Stop(); |
| 273 | } |
281 | } |
| 274 | 282 | ||
| Line 279... | Line 287... | ||
| 279 | can::Protocol::Delete(); |
287 | can::Protocol::Delete(); |
| 280 | broker::Manager::Delete(); |
288 | broker::Manager::Delete(); |
| 281 | vm::Manager::Delete(); |
289 | vm::Manager::Delete(); |
| 282 | net::Manager::Delete(); |
290 | net::Manager::Delete(); |
| 283 | storage::Manager::Delete(); |
291 | storage::Manager::Delete(); |
| 284 | 292 | ||
| 285 | log::CloseFile(); |
293 | log::CloseFile(); |
| 286 | } |
294 | } |
| 287 | 295 | ||
| 288 | void Handler(int status) |
296 | void Handler(int status) |
| 289 | { |
297 | { |