version 1.1.1.1.2.5, 2011/11/30 00:12:30
|
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 "rtlm.h" |
#include "utils.h" |
#include "utils.h" |
|
#include "daemon.h" |
|
|
|
|
sl_config cfg; |
sl_config cfg; |
Line 14 Usage(void)
|
Line 15 Usage(void)
|
printf( " -= MQTT Broker =- MQTT Service from ELWIX\n" |
printf( " -= MQTT Broker =- MQTT Service from ELWIX\n" |
"=== %s@%s === Compiled: %s ===\n\n" |
"=== %s@%s === Compiled: %s ===\n\n" |
"\t-c <config>\tService config\n" |
"\t-c <config>\tService config\n" |
|
"\t-b\t\tBatch mode\n" |
"\t-v\t\tVerbose (more -vvv, more verbose)\n" |
"\t-v\t\tVerbose (more -vvv, more verbose)\n" |
"\t-h\t\tHelp! This screen\n\n", |
"\t-h\t\tHelp! This screen\n\n", |
compiledby, compilehost, compiled); |
compiledby, compilehost, compiled); |
Line 23 Usage(void)
|
Line 25 Usage(void)
|
int |
int |
main(int argc, char **argv) |
main(int argc, char **argv) |
{ |
{ |
char ch, szCfgName[MAXPATHLEN]; | char ch, batch = 0, szStr[STRSIZ], szCfgName[MAXPATHLEN]; |
register int i; |
register int i; |
sqlite3 *acc = NULL, *pub = NULL; |
sqlite3 *acc = NULL, *pub = NULL; |
FILE *logg = NULL; |
FILE *logg = NULL; |
int sock, ret = 0; | int sock = -1, ret = 0; |
| struct passwd *pass; |
|
|
strlcpy(szCfgName, DEFAULT_CONFIG, sizeof szCfgName); |
strlcpy(szCfgName, DEFAULT_CONFIG, sizeof szCfgName); |
while ((ch = getopt(argc, argv, "hvc:")) != -1) | while ((ch = getopt(argc, argv, "hvbc:")) != -1) |
switch (ch) { |
switch (ch) { |
case 'c': |
case 'c': |
strlcpy(szCfgName, optarg, sizeof szCfgName); |
strlcpy(szCfgName, optarg, sizeof szCfgName); |
break; |
break; |
|
case 'b': |
|
batch++; |
|
break; |
case 'v': |
case 'v': |
Verbose++; |
Verbose++; |
break; |
break; |
Line 50 main(int argc, char **argv)
|
Line 56 main(int argc, char **argv)
|
printf("Error:: can't load #%d - %s\n", cfg_GetErrno(), cfg_GetError()); |
printf("Error:: can't load #%d - %s\n", cfg_GetErrno(), cfg_GetError()); |
return 1; |
return 1; |
} |
} |
|
openlog("mqttd", LOG_PID | LOG_CONS, LOG_DAEMON); |
for (i = 0; i < 3; i++) |
for (i = 0; i < 3; i++) |
if (!mqttLoadRTLM(&cfg, i)) { |
if (!mqttLoadRTLM(&cfg, i)) { |
printf("Error:: Can't load RTL module\n"); |
printf("Error:: Can't load RTL module\n"); |
while (i--) |
while (i--) |
mqttUnloadRTLM(i); |
mqttUnloadRTLM(i); |
UnloadConfig(&cfg); |
UnloadConfig(&cfg); |
|
closelog(); |
return 2; |
return 2; |
} |
} |
acc = call.OpenACC(&cfg); |
acc = call.OpenACC(&cfg); |
Line 80 main(int argc, char **argv)
|
Line 88 main(int argc, char **argv)
|
goto end; |
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) { |
if ((sock = srv_Socket(&cfg)) == -1) { |
ret = 4; |
ret = 4; |
goto end; |
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: |
end: |
|
if (sock > STDERR_FILENO) |
|
srv_Close(sock); |
call.CloseLOG(logg); |
call.CloseLOG(logg); |
call.ClosePUB(pub); |
call.ClosePUB(pub); |
call.CloseACC(acc); |
call.CloseACC(acc); |
for (i = 0; i < 3; i++) |
for (i = 0; i < 3; i++) |
mqttUnloadRTLM(i); |
mqttUnloadRTLM(i); |
|
closelog(); |
UnloadConfig(&cfg); |
UnloadConfig(&cfg); |
return ret; |
return ret; |
} |
} |