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

version 1.1.1.1.2.6, 2011/11/30 01:06:30 version 1.1.1.1.2.11, 2011/12/12 09:24:25
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;  sl_config cfg;
   sessions_t Sessions;
   sched_root_task_t *root;
   sqlite3 *acc, *pub;
   FILE *logg;
 extern char compiled[], compiledby[], compilehost[];  extern char compiled[], compiledby[], compilehost[];
int Verbose;static char szCfgName[MAXPATHLEN];
 int Verbose, Kill;
   
   
 static void  static void
Line 21  Usage(void) Line 27  Usage(void)
                 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, batch = 0, szStr[STRSIZ], szCfgName[MAXPATHLEN];        char ch, batch = 0, szStr[STRSIZ];
         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;
   
           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 88  main(int argc, char **argv) Line 119  main(int argc, char **argv)
                 goto end;                  goto end;
         }          }
   
        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(SIGTERM, &sa, NULL);
         sigaction(SIGCHLD, &sa, NULL);
   
         if ((sock = srv_Socket(&cfg)) == -1) {  
                 ret = 4;  
                 goto end;  
         }  
   
         if (!batch)          if (!batch)
                 switch (fork()) {                  switch (fork()) {
                         case -1:                          case -1:
Line 103  main(int argc, char **argv) Line 134  main(int argc, char **argv)
                                 goto end;                                  goto end;
                         case 0:                          case 0:
                                 setsid();                                  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);                                  ret = open("/dev/null", O_RDWR);
                                 if (ret != -1) {                                  if (ret != -1) {
                                         dup2(ret, STDIN_FILENO);                                          dup2(ret, STDIN_FILENO);
Line 123  main(int argc, char **argv) Line 146  main(int argc, char **argv)
                                 break;                                  break;
                         default:                          default:
                                 VERB(2) syslog(LOG_DEBUG, "MQTT service go to shadow land ...");                                  VERB(2) syslog(LOG_DEBUG, "MQTT service go to shadow land ...");
                                   sleep(1);
                                 ret = 0;                                  ret = 0;
                                 goto end;                                  goto end;
                 }                  }
         else          else
                 VERB(1) printf("Start service in batch mode ...\n");                  VERB(1) printf("Start service in batch mode ...\n");
   
        Run();        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:  end:
         if (sock > STDERR_FILENO)          if (sock > STDERR_FILENO)
                 srv_Close(sock);                  srv_Close(sock);

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


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