--- mqtt/src/mqttd.c 2011/11/30 00:12:30 1.1.1.1.2.5 +++ mqtt/src/mqttd.c 2011/11/30 01:06:30 1.1.1.1.2.6 @@ -1,6 +1,7 @@ #include "global.h" #include "rtlm.h" #include "utils.h" +#include "daemon.h" sl_config cfg; @@ -14,6 +15,7 @@ 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); @@ -23,18 +25,22 @@ Usage(void) int main(int argc, char **argv) { - char ch, szCfgName[MAXPATHLEN]; + char ch, batch = 0, szStr[STRSIZ], szCfgName[MAXPATHLEN]; register int i; sqlite3 *acc = NULL, *pub = NULL; FILE *logg = NULL; - int sock, ret = 0; + int sock = -1, ret = 0; + struct passwd *pass; strlcpy(szCfgName, DEFAULT_CONFIG, sizeof szCfgName); - while ((ch = getopt(argc, argv, "hvc:")) != -1) + while ((ch = getopt(argc, argv, "hvbc:")) != -1) switch (ch) { case 'c': strlcpy(szCfgName, optarg, sizeof szCfgName); break; + case 'b': + batch++; + break; case 'v': Verbose++; break; @@ -50,12 +56,14 @@ main(int argc, char **argv) printf("Error:: can't load #%d - %s\n", cfg_GetErrno(), cfg_GetError()); return 1; } + openlog("mqttd", LOG_PID | LOG_CONS, LOG_DAEMON); for (i = 0; i < 3; i++) if (!mqttLoadRTLM(&cfg, i)) { printf("Error:: Can't load RTL module\n"); while (i--) mqttUnloadRTLM(i); UnloadConfig(&cfg); + closelog(); return 2; } acc = call.OpenACC(&cfg); @@ -80,20 +88,58 @@ main(int argc, char **argv) goto end; } - VERB(2) printf("Service is ready for start engine ...\n"); + VERB(2) syslog(LOG_DEBUG, "Service is ready for start engine ..."); if ((sock = srv_Socket(&cfg)) == -1) { ret = 4; goto end; } - srv_Close(sock); + if (!batch) + switch (fork()) { + case -1: + printf("Error:: in fork() #%d - %s\n", errno, strerror(errno)); + ret = 5; + goto end; + case 0: + setsid(); + cfg_LoadAttribute(&cfg, CFG("mqttd"), CFG("user"), CFG(szStr), + sizeof szStr, MQTT_USER); + pass = getpwnam(szStr); + 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); + } + ret = open("/dev/null", O_RDWR); + if (ret != -1) { + dup2(ret, STDIN_FILENO); + dup2(ret, STDOUT_FILENO); + dup2(ret, STDERR_FILENO); + close(ret); + } + VERB(2) syslog(LOG_DEBUG, "Welcome MQTT service into shadow land!"); + break; + default: + VERB(2) syslog(LOG_DEBUG, "MQTT service go to shadow land ..."); + ret = 0; + goto end; + } + else + VERB(1) printf("Start service in batch mode ...\n"); + + Run(); + end: + if (sock > STDERR_FILENO) + srv_Close(sock); call.CloseLOG(logg); call.ClosePUB(pub); call.CloseACC(acc); for (i = 0; i < 3; i++) mqttUnloadRTLM(i); + closelog(); UnloadConfig(&cfg); return ret; }