--- mqtt/src/mqttd.c 2011/11/23 00:42:17 1.1.1.1.2.1 +++ mqtt/src/mqttd.c 2013/01/18 10:17:22 1.4.4.1 @@ -1,12 +1,246 @@ +/************************************************************************* +* (C) 2011 AITNET ltd - Sofia/Bulgaria - +* by Michael Pounov +* +* $Author: misho $ +* $Id: mqttd.c,v 1.4.4.1 2013/01/18 10:17:22 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, 2013 + 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" +#include "utils.h" +#include "daemon.h" -sl_config cfg; +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]; +volatile intptr_t Kill; +static void +Usage(void) +{ + printf( " -= MQTT Broker =- MQTT Service from ELWIX\n" + "=== %s@%s === Compiled: %s ===\n\n" + "\t-c \tService config\n" + "\t-b\t\tBatch mode\n" + "\t-v\t\tVerbose (more -vvv, more verbose)\n" + "\t-h\t\tHelp! This screen\n\n", + compiledby, compilehost, compiled); +} + +static void +sigHand(int sig) +{ + int stat; + + switch (sig) { + case SIGHUP: + cfgUnloadConfig(&cfg); + if (!cfgLoadConfig(szCfgName, &cfg)) { + EVERBOSE(1, "Config reload OK!"); + break; + } + + ELIBERR(cfg); + case SIGINT: + case SIGTERM: + EVERBOSE(1, "Terminate MQTT service in progress"); + Kill++; + break; + case SIGCHLD: + while (waitpid(-1, &stat, WNOHANG) > 0); + break; + case SIGPIPE: + break; + } +} + + int main(int argc, char **argv) { - return 0; + 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); + + strlcpy(szCfgName, DEFAULT_CONFIG, sizeof szCfgName); + while ((ch = getopt(argc, argv, "hvbc:")) != -1) + switch (ch) { + case 'c': + strlcpy(szCfgName, optarg, sizeof szCfgName); + break; + case 'b': + batch++; + break; + case 'v': + e_incVerbose; + break; + case 'h': + default: + Usage(); + return 1; + } + argc -= optind; + argv += optind; + + 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"); + mqttUnloadRTLM(acc); + mqttUnloadRTLM(pub); + mqttUnloadRTLM(logg); + cfgUnloadConfig(&cfg); + closelog(); + return 2; + } + acc = call.OpenACC(&cfg); + if (!acc) { + ret = 3; + goto end; + } + pub = call.OpenPUB(&cfg); + if (!pub) { + ret = 3; + goto end; + } + logg = call.OpenLOG(&cfg); + if (!logg) { + ret = 3; + goto end; + } + + + if (mqttMkDir(&cfg)) { + printf("Error:: in statedir #%d - %s\n", errno, strerror(errno)); + ret = 3; + goto end; + } + + if (!batch) + switch (fork()) { + case -1: + printf("Error:: in fork() #%d - %s\n", errno, strerror(errno)); + ret = 5; + goto end; + case 0: + setsid(); + + ret = open("/dev/null", O_RDWR); + if (ret != -1) { + dup2(ret, STDIN_FILENO); + dup2(ret, STDOUT_FILENO); + dup2(ret, STDERR_FILENO); + close(ret); + } + EVERBOSE(2, "Welcome MQTT service into shadow land!"); + break; + default: + EVERBOSE(2, "MQTT service go to shadow land ..."); + sleep(1); + ret = 0; + goto end; + } + else + EVERBOSE(1, "Start service in batch mode ..."); + + 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); + EVERBOSE(2, "Service is ready for starting engine ..."); + + if ((sock = srv_Socket(&cfg)) == -1) { + ret = 4; + goto end; + } + + 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); + EVERBOSE(2, "Try to change group #%d and user #%d", pass->pw_gid, pass->pw_uid); + } + + if (!(root = schedBegin())) { + ELIBERR(sched); + ret = 6; + goto end; + } + + /* go catch the cat ... */ + Run(sock); + + schedEnd(&root); +end: /* free all resources */ + srv_Close(sock); + call.CloseLOG(logg); + call.ClosePUB(pub); + call.CloseACC(acc); + mqttUnloadRTLM(acc); + mqttUnloadRTLM(pub); + mqttUnloadRTLM(logg); + cfgUnloadConfig(&cfg); + closelog(); + return ret; }