#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
main(int argc, char **argv)
{
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;
}
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>