[BACK]Return to mqttd.c CVS log [TXT] [DIR] Up to [ELWIX - Embedded LightWeight unIX -] / mqtt / src

Diff for /mqtt/src/mqttd.c between versions 1.1.1.1.2.8 and 1.2.2.10

version 1.1.1.1.2.8, 2011/12/01 08:15:25 version 1.2.2.10, 2012/06/26 14:51:33
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;
 sched_root_task_t *root;  sched_root_task_t *root;
   sqlite3 *acc, *pub;
   FILE *logg;
 extern char compiled[], compiledby[], compilehost[];  extern char compiled[], compiledby[], compilehost[];
 static char szCfgName[MAXPATHLEN];  static char szCfgName[MAXPATHLEN];
int Verbose, Kill;volatile intptr_t Kill;
   
   
 static void  static void
Line 30  sigHand(int sig) Line 36  sigHand(int sig)
   
         switch (sig) {          switch (sig) {
                 case SIGHUP:                  case SIGHUP:
                        UnloadConfig(&cfg);                        cfgUnloadConfig(&cfg);
                        if (!LoadConfig(szCfgName, &cfg)) {                        if (!cfgLoadConfig(szCfgName, &cfg)) {
                                VERB(1) syslog(LOG_DEBUG, "Config reload OK!");                                ioDEBUG(1, "Config reload OK!");
                                 break;                                  break;
                         }                          }
   
                        syslog(LOG_ERR, "Error:: can't reload #%d - %s", cfg_GetErrno(), cfg_GetError());                        ioLIBERR(cfg);
                 case SIGINT:
                 case SIGTERM:                  case SIGTERM:
                        VERB(1) syslog(LOG_DEBUG, "Terminate MQTT service in progress");                        ioDEBUG(1, "Terminate MQTT service in progress");
                         Kill++;                          Kill++;
                         break;                          break;
                 case SIGCHLD:                  case SIGCHLD:
                         while (waitpid(-1, &stat, WNOHANG) > 0);                          while (waitpid(-1, &stat, WNOHANG) > 0);
                         break;                          break;
                   case SIGPIPE:
                           break;
         }          }
 }  }
   
Line 51  sigHand(int sig) Line 60  sigHand(int sig)
 int  int
 main(int argc, char **argv)  main(int argc, char **argv)
 {  {
        char ch, batch = 0, szStr[STRSIZ];        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;          struct sigaction sa;
           ait_val_t v;
   
           TAILQ_INIT(&Sessions);
   
         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 69  main(int argc, char **argv) Line 79  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 79  main(int argc, char **argv) Line 89  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;
         }          }
         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--)                        mqttUnloadRTLM(acc);
                                mqttUnloadRTLM(i);                        mqttUnloadRTLM(pub);
                        UnloadConfig(&cfg);                        mqttUnloadRTLM(logg);
                         cfgUnloadConfig(&cfg);
                         closelog();                          closelog();
                         return 2;                          return 2;
                 }                  }
Line 109  main(int argc, char **argv) Line 121  main(int argc, char **argv)
                 goto end;                  goto end;
         }          }
   
   
         if (mqttMkDir(&cfg)) {          if (mqttMkDir(&cfg)) {
                 printf("Error:: in statedir #%d - %s\n", errno, strerror(errno));                  printf("Error:: in statedir #%d - %s\n", errno, strerror(errno));
                 ret = 3;                  ret = 3;
                 goto end;                  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)          if (!batch)
                 switch (fork()) {                  switch (fork()) {
                         case -1:                          case -1:
Line 138  main(int argc, char **argv) Line 144  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);                                  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 ...");
   
        VERB(2) syslog(LOG_DEBUG, "Service is ready for start engine ...");        memset(&sa, 0, sizeof sa);
         sigemptyset(&sa.sa_mask);
         sa.sa_handler = sigHand;
         sigaction(SIGHUP, &sa, NULL);
         sigaction(SIGINT, &sa, NULL);
         sigaction(SIGTERM, &sa, NULL);
         sigaction(SIGCHLD, &sa, NULL);
         sigaction(SIGPIPE, &sa, NULL);
         ioDEBUG(2, "Service is ready for starting engine ...");
   
         if ((sock = srv_Socket(&cfg)) == -1) {          if ((sock = srv_Socket(&cfg)) == -1) {
                 ret = 4;                  ret = 4;
                 goto end;                  goto end;
         }          }
   
        cfg_LoadAttribute(&cfg, CFG("mqttd"), CFG("user"), CFG(szStr), sizeof szStr, MQTT_USER);        cfg_loadAttribute(&cfg, "mqttd", "user", &v, MQTT_USER);
        pass = getpwnam(szStr);        pass = getpwnam(AIT_GET_STR(&v));
         AIT_FREE_VAL(&v);
         if (pass) {          if (pass) {
                 setgid(pass->pw_gid);                  setgid(pass->pw_gid);
                 setuid(pass->pw_uid);                  setuid(pass->pw_uid);
                VERB(1) syslog(LOG_WARNING, "Try to change group #%d and user #%d", pass->pw_gid, pass->pw_uid);                ioDEBUG(2, "Try to change group #%d and user #%d", pass->pw_gid, pass->pw_uid);
         }          }
   
         if (!(root = schedBegin())) {          if (!(root = schedBegin())) {
                printf("Error:: scheduler #%d - %s\n", sched_GetErrno(), sched_GetError());                ioLIBERR(sched);
                 ret = 6;                  ret = 6;
                 goto end;                  goto end;
         }          }
   
           /* go catch the cat ... */
         Run(sock);          Run(sock);
   
         schedEnd(&root);          schedEnd(&root);
end:end:    /* free all resources */
        if (sock > STDERR_FILENO)        srv_Close(sock);
                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++)        mqttUnloadRTLM(acc);
                mqttUnloadRTLM(i);        mqttUnloadRTLM(pub);
         mqttUnloadRTLM(logg);
         cfgUnloadConfig(&cfg);
         closelog();          closelog();
         UnloadConfig(&cfg);  
         return ret;          return ret;
 }  }
Removed from v.1.1.1.1.2.8  
changed lines
  Added in v.1.2.2.10

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