|
|
| version 1.1.1.1, 2011/11/07 08:47:16 | version 1.1.1.1.2.6, 2011/11/30 01:06:30 |
|---|---|
| Line 1 | Line 1 |
| #include "global.h" | #include "global.h" |
| #include "rtlm.h" | |
| #include "utils.h" | |
| #include "daemon.h" | |
| sl_config cfg; | |
| extern char compiled[], compiledby[], compilehost[]; | |
| int Verbose; | |
| static void | |
| Usage(void) | |
| { | |
| printf( " -= MQTT Broker =- MQTT Service from ELWIX\n" | |
| "=== %s@%s === Compiled: %s ===\n\n" | |
| "\t-c <config>\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); | |
| } | |
| int | int |
| main(int argc, char **argv) | main(int argc, char **argv) |
| { | { |
| return 0; | char ch, batch = 0, szStr[STRSIZ], szCfgName[MAXPATHLEN]; |
| register int i; | |
| sqlite3 *acc = NULL, *pub = NULL; | |
| FILE *logg = NULL; | |
| int sock = -1, ret = 0; | |
| struct passwd *pass; | |
| 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': | |
| Verbose++; | |
| break; | |
| case 'h': | |
| default: | |
| Usage(); | |
| return 1; | |
| } | |
| argc -= optind; | |
| argv += optind; | |
| if (LoadConfig(szCfgName, &cfg)) { | |
| 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); | |
| 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; | |
| } | |
| VERB(2) syslog(LOG_DEBUG, "Service is ready for start engine ..."); | |
| if ((sock = srv_Socket(&cfg)) == -1) { | |
| ret = 4; | |
| 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(); | |
| 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; | |
| } | } |