Return to mqttd.c CVS log

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

add scheduler init and finish

    1: #include "global.h"
    2: #include "rtlm.h"
    3: #include "utils.h"
    4: #include "daemon.h"
    5: 
    6: 
    7: sl_config cfg;
    8: sched_root_task_t *root;
    9: extern char compiled[], compiledby[], compilehost[];
   10: int Verbose;
   11: 
   12: 
   13: static void
   14: Usage(void)
   15: {
   16: 	printf(	" -= MQTT Broker =- MQTT Service from ELWIX\n"
   17: 		"=== %s@%s === Compiled: %s ===\n\n"
   18: 		"\t-c <config>\tService config\n"
   19: 		"\t-b\t\tBatch mode\n"
   20: 		"\t-v\t\tVerbose (more -vvv, more verbose)\n"
   21: 		"\t-h\t\tHelp! This screen\n\n", 
   22: 		compiledby, compilehost, compiled);
   23: }
   24: 
   25: 
   26: int
   27: main(int argc, char **argv)
   28: {
   29: 	char ch, batch = 0, szStr[STRSIZ], szCfgName[MAXPATHLEN];
   30: 	register int i;
   31: 	sqlite3 *acc = NULL, *pub = NULL;
   32: 	FILE *logg = NULL;
   33: 	int sock = -1, ret = 0;
   34: 	struct passwd *pass;
   35: 
   36: 	strlcpy(szCfgName, DEFAULT_CONFIG, sizeof szCfgName);
   37: 	while ((ch = getopt(argc, argv, "hvbc:")) != -1)
   38: 		switch (ch) {
   39: 			case 'c':
   40: 				strlcpy(szCfgName, optarg, sizeof szCfgName);
   41: 				break;
   42: 			case 'b':
   43: 				batch++;
   44: 				break;
   45: 			case 'v':
   46: 				Verbose++;
   47: 				break;
   48: 			case 'h':
   49: 			default:
   50: 				Usage();
   51: 				return 1;
   52: 		}
   53: 	argc -= optind;
   54: 	argv += optind;
   55: 
   56: 	if (LoadConfig(szCfgName, &cfg)) {
   57: 		printf("Error:: can't load #%d - %s\n", cfg_GetErrno(), cfg_GetError());
   58: 		return 1;
   59: 	}
   60: 	openlog("mqttd", LOG_PID | LOG_CONS, LOG_DAEMON);
   61: 	for (i = 0; i < 3; i++)
   62: 		if (!mqttLoadRTLM(&cfg, i)) {
   63: 			printf("Error:: Can't load RTL module\n");
   64: 			while (i--)
   65: 				mqttUnloadRTLM(i);
   66: 			UnloadConfig(&cfg);
   67: 			closelog();
   68: 			return 2;
   69: 		}
   70: 	acc = call.OpenACC(&cfg);
   71: 	if (!acc) {
   72: 		ret = 3;
   73: 		goto end;
   74: 	}
   75: 	pub = call.OpenPUB(&cfg);
   76: 	if (!pub) {
   77: 		ret = 3;
   78: 		goto end;
   79: 	}
   80: 	logg = call.OpenLOG(&cfg);
   81: 	if (!logg) {
   82: 		ret = 3;
   83: 		goto end;
   84: 	}
   85: 
   86: 	if (mqttMkDir(&cfg)) {
   87: 		printf("Error:: in statedir #%d - %s\n", errno, strerror(errno));
   88: 		ret = 3;
   89: 		goto end;
   90: 	}
   91: 
   92: 	VERB(2) syslog(LOG_DEBUG, "Service is ready for start engine ...");
   93: 
   94: 	if ((sock = srv_Socket(&cfg)) == -1) {
   95: 		ret = 4;
   96: 		goto end;
   97: 	}
   98: 
   99: 	if (!batch)
  100: 		switch (fork()) {
  101: 			case -1:
  102: 				printf("Error:: in fork() #%d - %s\n", errno, strerror(errno));
  103: 				ret = 5;
  104: 				goto end;
  105: 			case 0:
  106: 				setsid();
  107: 				cfg_LoadAttribute(&cfg, CFG("mqttd"), CFG("user"), CFG(szStr), 
  108: 						sizeof szStr, MQTT_USER);
  109: 				pass = getpwnam(szStr);
  110: 				if (pass) {
  111: 					setgid(pass->pw_gid);
  112: 					setuid(pass->pw_uid);
  113: 					VERB(1) syslog(LOG_WARNING, "Try to change group #%d and user #%d", 
  114: 							pass->pw_gid, pass->pw_uid);
  115: 				}
  116: 				ret = open("/dev/null", O_RDWR);
  117: 				if (ret != -1) {
  118: 					dup2(ret, STDIN_FILENO);
  119: 					dup2(ret, STDOUT_FILENO);
  120: 					dup2(ret, STDERR_FILENO);
  121: 					close(ret);
  122: 				}
  123: 				VERB(2) syslog(LOG_DEBUG, "Welcome MQTT service into shadow land!");
  124: 				break;
  125: 			default:
  126: 				VERB(2) syslog(LOG_DEBUG, "MQTT service go to shadow land ...");
  127: 				ret = 0;
  128: 				goto end;
  129: 		}
  130: 	else
  131: 		VERB(1) printf("Start service in batch mode ...\n");
  132: 
  133: 	if (!(root = schedBegin())) {
  134: 		printf("Error:: scheduler #%d - %s\n", sched_GetErrno(), sched_GetError());
  135: 		ret = 6;
  136: 		goto end;
  137: 	}
  138: 
  139: 	Run(sock);
  140: 
  141: 	schedEnd(&root);
  142: end:
  143: 	if (sock > STDERR_FILENO)
  144: 		srv_Close(sock);
  145: 	call.CloseLOG(logg);
  146: 	call.ClosePUB(pub);
  147: 	call.CloseACC(acc);
  148: 	for (i = 0; i < 3; i++)
  149: 		mqttUnloadRTLM(i);
  150: 	closelog();
  151: 	UnloadConfig(&cfg);
  152: 	return ret;
  153: }