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

version 1.1.1.1.2.5, 2011/11/30 00:12:30 version 1.1.1.1.2.6, 2011/11/30 01:06:30
Line 1 Line 1
 #include "global.h"  #include "global.h"
 #include "rtlm.h"  #include "rtlm.h"
 #include "utils.h"  #include "utils.h"
   #include "daemon.h"
   
   
 sl_config cfg;  sl_config cfg;
Line 14  Usage(void) Line 15  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);
Line 23  Usage(void) Line 25  Usage(void)
 int  int
 main(int argc, char **argv)  main(int argc, char **argv)
 {  {
        char ch, szCfgName[MAXPATHLEN];        char ch, batch = 0, szStr[STRSIZ], szCfgName[MAXPATHLEN];
         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;
   
         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 56  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 88  main(int argc, char **argv)
                 goto end;                  goto end;
         }          }
   
        VERB(2) printf("Service is ready for start engine ...\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);        if (!batch)
                 switch (fork()) {
                         case -1:
                                 printf("Error:: in fork() #%d - %s\n", errno, strerror(errno));
                                 ret = 5;
                                 goto end;
                         case 0:
                                 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);
                                 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 ...");
                                 ret = 0;
                                 goto end;
                 }
         else
                 VERB(1) printf("Start service in batch mode ...\n");
 
         Run();
 
 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;
 }  }

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


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