--- mqtt/src/mqttd.c	2011/11/07 08:47:16	1.1.1.1
+++ mqtt/src/mqttd.c	2011/12/01 08:15:25	1.1.1.1.2.8
@@ -1,8 +1,187 @@
 #include "global.h"
+#include "rtlm.h"
+#include "utils.h"
+#include "daemon.h"
 
 
+sl_config cfg;
+sched_root_task_t *root;
+extern char compiled[], compiledby[], compilehost[];
+static char szCfgName[MAXPATHLEN];
+int Verbose, Kill;
+
+
+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);
+}
+
+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
 main(int argc, char **argv)
 {
-	return 0;
+	char ch, batch = 0, szStr[STRSIZ];
+	register int i;
+	sqlite3 *acc = NULL, *pub = NULL;
+	FILE *logg = NULL;
+	int sock = -1, ret = 0;
+	struct passwd *pass;
+	struct sigaction sa;
+
+	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;
+	}
+
+	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) {
+		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:
+	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;
 }