version 1.1.1.1.2.6, 2011/11/30 01:06:30
|
version 1.1.1.1.2.12, 2011/12/14 12:45:30
|
Line 1
|
Line 1
|
#include "global.h" |
#include "global.h" |
|
#include "mqttd.h" |
#include "rtlm.h" |
#include "rtlm.h" |
#include "utils.h" |
#include "utils.h" |
#include "daemon.h" |
#include "daemon.h" |
|
|
|
|
sl_config cfg; |
sl_config cfg; |
|
sessions_t Sessions; |
|
sched_root_task_t *root; |
|
sqlite3 *acc, *pub; |
|
pthread_mutex_t mtx_sess; |
|
FILE *logg; |
extern char compiled[], compiledby[], compilehost[]; |
extern char compiled[], compiledby[], compilehost[]; |
int Verbose; | static char szCfgName[MAXPATHLEN]; |
| int Verbose, Kill; |
|
|
|
|
static void |
static void |
Line 21 Usage(void)
|
Line 28 Usage(void)
|
compiledby, compilehost, compiled); |
compiledby, compilehost, compiled); |
} |
} |
|
|
|
static void |
|
sigHand(int sig) |
|
{ |
|
int stat; |
|
|
|
switch (sig) { |
|
case SIGHUP: |
|
UnloadConfig(&cfg); |
|
if (!LoadConfig(szCfgName, &cfg)) { |
|
VERB(1) syslog(LOG_DEBUG, "Config reload OK!"); |
|
break; |
|
} |
|
|
|
syslog(LOG_ERR, "Error:: can't reload #%d - %s", cfg_GetErrno(), cfg_GetError()); |
|
case SIGTERM: |
|
VERB(1) syslog(LOG_DEBUG, "Terminate MQTT service in progress"); |
|
Kill++; |
|
break; |
|
case SIGCHLD: |
|
while (waitpid(-1, &stat, WNOHANG) > 0); |
|
break; |
|
} |
|
} |
|
|
|
|
int |
int |
main(int argc, char **argv) |
main(int argc, char **argv) |
{ |
{ |
char ch, batch = 0, szStr[STRSIZ], szCfgName[MAXPATHLEN]; | char ch, batch = 0, szStr[STRSIZ]; |
register int i; |
register int i; |
sqlite3 *acc = NULL, *pub = NULL; |
|
FILE *logg = NULL; |
|
int sock = -1, ret = 0; |
int sock = -1, ret = 0; |
struct passwd *pass; |
struct passwd *pass; |
|
struct sigaction sa; |
|
|
|
TAILQ_INIT(&Sessions); |
|
|
strlcpy(szCfgName, DEFAULT_CONFIG, sizeof szCfgName); |
strlcpy(szCfgName, DEFAULT_CONFIG, sizeof szCfgName); |
while ((ch = getopt(argc, argv, "hvbc:")) != -1) |
while ((ch = getopt(argc, argv, "hvbc:")) != -1) |
switch (ch) { |
switch (ch) { |
Line 56 main(int argc, char **argv)
|
Line 88 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; |
} |
} |
|
pthread_mutex_init(&mtx_sess, NULL); |
openlog("mqttd", LOG_PID | LOG_CONS, LOG_DAEMON); |
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)) { |
Line 64 main(int argc, char **argv)
|
Line 97 main(int argc, char **argv)
|
mqttUnloadRTLM(i); |
mqttUnloadRTLM(i); |
UnloadConfig(&cfg); |
UnloadConfig(&cfg); |
closelog(); |
closelog(); |
|
pthread_mutex_destroy(&mtx_sess); |
return 2; |
return 2; |
} |
} |
acc = call.OpenACC(&cfg); |
acc = call.OpenACC(&cfg); |
Line 88 main(int argc, char **argv)
|
Line 122 main(int argc, char **argv)
|
goto end; |
goto end; |
} |
} |
|
|
VERB(2) syslog(LOG_DEBUG, "Service is ready for start engine ..."); | memset(&sa, 0, sizeof sa); |
| sigemptyset(&sa.sa_mask); |
| sa.sa_handler = sigHand; |
| sigaction(SIGHUP, &sa, NULL); |
| sigaction(SIGTERM, &sa, NULL); |
| sigaction(SIGCHLD, &sa, NULL); |
|
|
if ((sock = srv_Socket(&cfg)) == -1) { |
|
ret = 4; |
|
goto end; |
|
} |
|
|
|
if (!batch) |
if (!batch) |
switch (fork()) { |
switch (fork()) { |
case -1: |
case -1: |
Line 103 main(int argc, char **argv)
|
Line 137 main(int argc, char **argv)
|
goto end; |
goto end; |
case 0: |
case 0: |
setsid(); |
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); |
ret = open("/dev/null", O_RDWR); |
if (ret != -1) { |
if (ret != -1) { |
dup2(ret, STDIN_FILENO); |
dup2(ret, STDIN_FILENO); |
Line 123 main(int argc, char **argv)
|
Line 149 main(int argc, char **argv)
|
break; |
break; |
default: |
default: |
VERB(2) syslog(LOG_DEBUG, "MQTT service go to shadow land ..."); |
VERB(2) syslog(LOG_DEBUG, "MQTT service go to shadow land ..."); |
|
sleep(1); |
ret = 0; |
ret = 0; |
goto end; |
goto end; |
} |
} |
else |
else |
VERB(1) printf("Start service in batch mode ...\n"); |
VERB(1) printf("Start service in batch mode ...\n"); |
|
|
Run(); | VERB(2) syslog(LOG_DEBUG, "Service is ready for start engine ..."); |
|
|
|
if ((sock = srv_Socket(&cfg)) == -1) { |
|
ret = 4; |
|
goto end; |
|
} |
|
|
|
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); |
|
} |
|
|
|
if (!(root = schedBegin())) { |
|
printf("Error:: scheduler #%d - %s\n", sched_GetErrno(), sched_GetError()); |
|
ret = 6; |
|
goto end; |
|
} |
|
|
|
Run(sock); |
|
|
|
schedEnd(&root); |
end: |
end: |
if (sock > STDERR_FILENO) |
if (sock > STDERR_FILENO) |
srv_Close(sock); |
srv_Close(sock); |
Line 141 end:
|
Line 190 end:
|
mqttUnloadRTLM(i); |
mqttUnloadRTLM(i); |
closelog(); |
closelog(); |
UnloadConfig(&cfg); |
UnloadConfig(&cfg); |
|
pthread_mutex_destroy(&mtx_sess); |
return ret; |
return ret; |
} |
} |