--- mqtt/src/mqttd.c 2011/12/12 09:24:25 1.1.1.1.2.11 +++ mqtt/src/mqttd.c 2012/07/03 12:46:00 1.4 @@ -1,3 +1,48 @@ +/************************************************************************* +* (C) 2011 AITNET ltd - Sofia/Bulgaria - +* by Michael Pounov +* +* $Author: misho $ +* $Id: mqttd.c,v 1.4 2012/07/03 12:46:00 misho Exp $ +* +************************************************************************** +The ELWIX and AITNET software is distributed under the following +terms: + +All of the documentation and software included in the ELWIX and AITNET +Releases is copyrighted by ELWIX - Sofia/Bulgaria + +Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012 + by Michael Pounov . All rights reserved. + +Redistribution and use in source and binary forms, with or without +modification, are permitted provided that the following conditions +are met: +1. Redistributions of source code must retain the above copyright + notice, this list of conditions and the following disclaimer. +2. Redistributions in binary form must reproduce the above copyright + notice, this list of conditions and the following disclaimer in the + documentation and/or other materials provided with the distribution. +3. All advertising materials mentioning features or use of this software + must display the following acknowledgement: +This product includes software developed by Michael Pounov +ELWIX - Embedded LightWeight unIX and its contributors. +4. Neither the name of AITNET nor the names of its contributors + may be used to endorse or promote products derived from this software + without specific prior written permission. + +THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND +ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE +IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE +ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE +FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL +DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS +OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) +HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT +LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY +OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF +SUCH DAMAGE. +*/ #include "global.h" #include "mqttd.h" #include "rtlm.h" @@ -5,14 +50,16 @@ #include "daemon.h" -sl_config cfg; +io_enableDEBUG; + +cfg_root_t cfg; sessions_t Sessions; sched_root_task_t *root; sqlite3 *acc, *pub; FILE *logg; extern char compiled[], compiledby[], compilehost[]; static char szCfgName[MAXPATHLEN]; -int Verbose, Kill; +volatile intptr_t Kill; static void @@ -34,20 +81,23 @@ sigHand(int sig) switch (sig) { case SIGHUP: - UnloadConfig(&cfg); - if (!LoadConfig(szCfgName, &cfg)) { - VERB(1) syslog(LOG_DEBUG, "Config reload OK!"); + cfgUnloadConfig(&cfg); + if (!cfgLoadConfig(szCfgName, &cfg)) { + ioDEBUG(1, "Config reload OK!"); break; } - syslog(LOG_ERR, "Error:: can't reload #%d - %s", cfg_GetErrno(), cfg_GetError()); + ioLIBERR(cfg); + case SIGINT: case SIGTERM: - VERB(1) syslog(LOG_DEBUG, "Terminate MQTT service in progress"); + ioDEBUG(1, "Terminate MQTT service in progress"); Kill++; break; case SIGCHLD: while (waitpid(-1, &stat, WNOHANG) > 0); break; + case SIGPIPE: + break; } } @@ -55,11 +105,12 @@ sigHand(int sig) int main(int argc, char **argv) { - char ch, batch = 0, szStr[STRSIZ]; + char ch, batch = 0; register int i; int sock = -1, ret = 0; struct passwd *pass; struct sigaction sa; + ait_val_t v; TAILQ_INIT(&Sessions); @@ -73,7 +124,7 @@ main(int argc, char **argv) batch++; break; case 'v': - Verbose++; + io_incDebug; break; case 'h': default: @@ -83,17 +134,19 @@ main(int argc, char **argv) argc -= optind; argv += optind; - if (LoadConfig(szCfgName, &cfg)) { + if (cfgLoadConfig(szCfgName, &cfg)) { printf("Error:: can't load #%d - %s\n", cfg_GetErrno(), cfg_GetError()); return 1; } openlog("mqttd", LOG_PID | LOG_CONS, LOG_DAEMON); + /* load 3 plugins */ for (i = 0; i < 3; i++) if (!mqttLoadRTLM(&cfg, i)) { printf("Error:: Can't load RTL module\n"); - while (i--) - mqttUnloadRTLM(i); - UnloadConfig(&cfg); + mqttUnloadRTLM(acc); + mqttUnloadRTLM(pub); + mqttUnloadRTLM(logg); + cfgUnloadConfig(&cfg); closelog(); return 2; } @@ -113,19 +166,13 @@ main(int argc, char **argv) goto end; } + if (mqttMkDir(&cfg)) { printf("Error:: in statedir #%d - %s\n", errno, strerror(errno)); ret = 3; goto end; } - memset(&sa, 0, sizeof sa); - sigemptyset(&sa.sa_mask); - sa.sa_handler = sigHand; - sigaction(SIGHUP, &sa, NULL); - sigaction(SIGTERM, &sa, NULL); - sigaction(SIGCHLD, &sa, NULL); - if (!batch) switch (fork()) { case -1: @@ -142,50 +189,60 @@ main(int argc, char **argv) dup2(ret, STDERR_FILENO); close(ret); } - VERB(2) syslog(LOG_DEBUG, "Welcome MQTT service into shadow land!"); + ioDEBUG(2, "Welcome MQTT service into shadow land!"); break; default: - VERB(2) syslog(LOG_DEBUG, "MQTT service go to shadow land ..."); + ioDEBUG(2, "MQTT service go to shadow land ..."); sleep(1); ret = 0; goto end; } else - VERB(1) printf("Start service in batch mode ...\n"); + ioDEBUG(1, "Start service in batch mode ..."); - VERB(2) syslog(LOG_DEBUG, "Service is ready for start engine ..."); + memset(&sa, 0, sizeof sa); + sigemptyset(&sa.sa_mask); + sa.sa_handler = sigHand; + sigaction(SIGHUP, &sa, NULL); + sigaction(SIGINT, &sa, NULL); + sigaction(SIGTERM, &sa, NULL); + sigaction(SIGCHLD, &sa, NULL); + sigaction(SIGPIPE, &sa, NULL); + ioDEBUG(2, "Service is ready for starting engine ..."); if ((sock = srv_Socket(&cfg)) == -1) { ret = 4; goto end; } - cfg_LoadAttribute(&cfg, CFG("mqttd"), CFG("user"), CFG(szStr), sizeof szStr, MQTT_USER); - pass = getpwnam(szStr); + cfg_loadAttribute(&cfg, "mqttd", "user", &v, MQTT_USER); + pass = getpwnam(AIT_GET_STR(&v)); + AIT_FREE_VAL(&v); if (pass) { setgid(pass->pw_gid); setuid(pass->pw_uid); - VERB(1) syslog(LOG_WARNING, "Try to change group #%d and user #%d", pass->pw_gid, pass->pw_uid); + ioDEBUG(2, "Try to change group #%d and user #%d", pass->pw_gid, pass->pw_uid); } if (!(root = schedBegin())) { - printf("Error:: scheduler #%d - %s\n", sched_GetErrno(), sched_GetError()); + ioLIBERR(sched); ret = 6; goto end; } + /* go catch the cat ... */ Run(sock); schedEnd(&root); -end: - if (sock > STDERR_FILENO) - srv_Close(sock); +end: /* free all resources */ + srv_Close(sock); call.CloseLOG(logg); call.ClosePUB(pub); call.CloseACC(acc); - for (i = 0; i < 3; i++) - mqttUnloadRTLM(i); + mqttUnloadRTLM(acc); + mqttUnloadRTLM(pub); + mqttUnloadRTLM(logg); + cfgUnloadConfig(&cfg); closelog(); - UnloadConfig(&cfg); return ret; }