mqtt/src/mqttd.c - diff

Return to mqttd.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / mqtt / src

Diff for /mqtt/src/mqttd.c between versions 1.1.1.1.2.6 and 1.2.2.6

version 1.1.1.1.2.6, 2011/11/30 01:06:30	version 1.2.2.6, 2012/04/24 13:54:50
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;	io_enableDEBUG;

	cfg_root_t cfg;
	sessions_t Sessions;
	pubs_t Pubs;
	sched_root_task_t *root;
	sqlite3 acc, pub;
	pthread_mutex_t mtx_sess, mtx_pub;
	FILE *logg;
extern char compiled[], compiledby[], compilehost[];	extern char compiled[], compiledby[], compilehost[];
int Verbose;	static char szCfgName[MAXPATHLEN];
	volatile intptr_t Kill;


static void	static void
Line 21 Usage(void)	Line 31 Usage(void)
compiledby, compilehost, compiled);	compiledby, compilehost, compiled);
}	}

	static void
	sigHand(int sig)
	{
	int stat;
	struct tagSession *sess;

	switch (sig) {
	case SIGHUP:
	cfgUnloadConfig(&cfg);
	if (!cfgLoadConfig(szCfgName, &cfg)) {
	ioDEBUG(1, "Config reload OK!");
	break;
	}

	ioLIBERR(cfg);
	case SIGTERM:
	ioDEBUG(1, "Terminate MQTT service in progress");
	Kill++;

	SESS_LOCK;
	TAILQ_FOREACH(sess, &Sessions, sess_node)
	if (sess->sess_tid)
	pthread_cancel(sess->sess_tid);
	SESS_UNLOCK;
	break;
	case SIGCHLD:
	while (waitpid(-1, &stat, WNOHANG) > 0);
	break;
	case SIGPIPE:
	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;
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;
	ait_val_t v;

	TAILQ_INIT(&Sessions);
	TAILQ_INIT(&Pubs);

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 42 main(int argc, char **argv)	Line 88 main(int argc, char **argv)
batch++;	batch++;
break;	break;
case 'v':	case 'v':
Verbose++;	io_incDebug;
break;	break;
case 'h':	case 'h':
default:	default:
Line 52 main(int argc, char **argv)	Line 98 main(int argc, char **argv)
argc -= optind;	argc -= optind;
argv += optind;	argv += optind;

if (LoadConfig(szCfgName, &cfg)) {	if (cfgLoadConfig(szCfgName, &cfg)) {
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);
	pthread_mutex_init(&mtx_pub, NULL);
openlog("mqttd", LOG_PID \| LOG_CONS, LOG_DAEMON);	openlog("mqttd", LOG_PID \| LOG_CONS, LOG_DAEMON);
	/* load 3 plugins */
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);	cfgUnloadConfig(&cfg);
closelog();	closelog();
	pthread_mutex_destroy(&mtx_pub);
	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 139 main(int argc, char **argv)
goto end;	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)	if (!batch)
switch (fork()) {	switch (fork()) {
case -1:	case -1:
Line 103 main(int argc, char **argv)	Line 147 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 119 main(int argc, char **argv)	Line 155 main(int argc, char **argv)
dup2(ret, STDERR_FILENO);	dup2(ret, STDERR_FILENO);
close(ret);	close(ret);
}	}
VERB(2) syslog(LOG_DEBUG, "Welcome MQTT service into shadow land!");	ioDEBUG(2, "Welcome MQTT service into shadow land!");
break;	break;
default:	default:
VERB(2) syslog(LOG_DEBUG, "MQTT service go to shadow land ...");	ioDEBUG(2, "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");	ioDEBUG(1, "Start service in batch mode ...");

Run();	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);
	sigaction(SIGPIPE, &sa, NULL);
	ioDEBUG(2, "Service is ready for starting engine ...");

end:	if ((sock = srv_Socket(&cfg)) == -1) {
if (sock > STDERR_FILENO)	ret = 4;
srv_Close(sock);	goto end;
	}

	cfg_loadAttribute(&cfg, "mqttd", "user", &v, MQTT_USER);
	pass = getpwnam(AIT_GET_STR(&v));
	AIT_FREE_VAL(&v);
	if (pass) {
	setgid(pass->pw_gid);
	setuid(pass->pw_uid);
	ioDEBUG(2, "Try to change group #%d and user #%d", pass->pw_gid, pass->pw_uid);
	}

	if (!(root = schedBegin())) {
	ioLIBERR(sched);
	ret = 6;
	goto end;
	}

	/* go catch the cat ... */
	Run(sock);

	schedEnd(&root);
	end: /* free all resources */
	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();	closelog();
UnloadConfig(&cfg);	cfgUnloadConfig(&cfg);
	pthread_mutex_destroy(&mtx_pub);
	pthread_mutex_destroy(&mtx_sess);
return ret;	return ret;
}	}

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>

Removed from v.1.1.1.1.2.6
changed lines
	Added in v.1.2.2.6