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.5 and 1.1.1.1.2.10

version 1.1.1.1.2.5, 2011/11/30 00:12:30	version 1.1.1.1.2.10, 2011/12/09 09:53:54
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"


sl_config cfg;	sl_config cfg;
	sessions_t Sessions;
	sched_root_task_t *root;
extern char compiled[], compiledby[], compilehost[];	extern char compiled[], compiledby[], compilehost[];
int Verbose;	static char szCfgName[MAXPATHLEN];
	int Verbose, Kill;


static void	static void
Line 14 Usage(void)	Line 19 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);
}	}

	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, szCfgName[MAXPATHLEN];	char ch, batch = 0, szStr[STRSIZ];
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;
	struct sigaction sa;

	TAILQ_INIT(&Sessions);

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 87 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 119 main(int argc, char **argv)
goto end;	goto end;
}	}

VERB(2) printf("Service is ready for start engine ...\n");	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 (!batch)
	switch (fork()) {
	case -1:
	printf("Error:: in fork() #%d - %s\n", errno, strerror(errno));
	ret = 5;
	goto end;
	case 0:
	setsid();

	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 ...");
	sleep(1);
	ret = 0;
	goto end;
	}
	else
	VERB(1) printf("Start service in batch mode ...\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);	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)
	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;
}	}

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

Removed from v.1.1.1.1.2.5
changed lines
	Added in v.1.1.1.1.2.10