Diff for /mqtt/src/daemon.c between versions 1.1.2.14 and 1.1.2.16

version 1.1.2.14, 2011/12/09 13:43:55 version 1.1.2.16, 2011/12/12 09:24:25
Line 1 Line 1
 #include "global.h"  #include "global.h"
   #include "rtlm.h"
 #include "mqttd.h"  #include "mqttd.h"
   
   
 extern char cliCmd[], *cliStr[];  
   
   
 static inline struct tagSession *  static inline struct tagSession *
 initSession(int sock, char * __restrict addr)  initSession(int sock, char * __restrict addr)
 {  {
         struct tagSession *sess = NULL;          struct tagSession *sess = NULL;
         const char *str;          const char *str;
   
           FTRACE(5);
   
         if (!addr)          if (!addr)
                 return NULL;                  return NULL;
   
         sess = malloc(sizeof(struct tagSession));          sess = malloc(sizeof(struct tagSession));
         if (!sess) {          if (!sess) {
                 VERB(3) syslog(LOG_ERR, "Error:: in malloc #%d - %s", errno, strerror(errno));                  VERB(3) syslog(LOG_ERR, "Error:: in malloc #%d - %s", errno, strerror(errno));
                 free(addr);  
                 return NULL;                  return NULL;
         } else          } else
                 memset(sess, 0, sizeof(struct tagSession));                  memset(sess, 0, sizeof(struct tagSession));
Line 28  initSession(int sock, char * __restrict addr) Line 27  initSession(int sock, char * __restrict addr)
         else          else
                 sess->sess_retry = strtol(str, NULL, 0);                  sess->sess_retry = strtol(str, NULL, 0);
   
         sess->sess_sock = sock;  
         strlcpy(sess->sess_addr, addr, sizeof sess->sess_addr);  
         free(addr);  
   
         sess->sess_buf = mqtt_msgAlloc(0);          sess->sess_buf = mqtt_msgAlloc(0);
         if (!sess->sess_buf) {          if (!sess->sess_buf) {
                 VERB(3) syslog(LOG_ERR, "Error:: in msgAlloc #%d - %s", mqtt_GetErrno(), mqtt_GetError());                  VERB(3) syslog(LOG_ERR, "Error:: in msgAlloc #%d - %s", mqtt_GetErrno(), mqtt_GetError());
Line 39  initSession(int sock, char * __restrict addr) Line 34  initSession(int sock, char * __restrict addr)
                 return NULL;                  return NULL;
         }          }
   
           sess->sess_sock = sock;
           strlcpy(sess->sess_addr, addr, sizeof sess->sess_addr);
           free(addr);
         return sess;          return sess;
 }  }
   
Line 48  finiSession(struct tagSession *sess) Line 46  finiSession(struct tagSession *sess)
         register int i;          register int i;
         struct tagStore *store;          struct tagStore *store;
   
           FTRACE(5);
   
         if (!sess)          if (!sess)
                 return;                  return;
   
Line 75  finiSession(struct tagSession *sess) Line 75  finiSession(struct tagSession *sess)
 }  }
   
 static void *  static void *
   thrSession(struct tagSession *sess)
   {
           FTRACE(2);
   
           TAILQ_REMOVE(&Sessions, sess, sess_node);
           pthread_exit(NULL);
           return NULL;
   }
   
   static void *
 startSession(sched_task_t *task)  startSession(sched_task_t *task)
 {  {
         u_char basebuf[USHRT_MAX];          u_char basebuf[USHRT_MAX];
         mqtt_cb_t cbs[MQTT_TYPE_MAX + 1] = { 0 };  
         mqtt_msg_t buf = { basebuf, sizeof basebuf };          mqtt_msg_t buf = { basebuf, sizeof basebuf };
         mqtthdr_connflgs_t flg;          mqtthdr_connflgs_t flg;
         mqtthdr_connack_t cack;          mqtthdr_connack_t cack;
         int ret = 0;  
         struct timeval tv = { 0 };  
         struct tagSession *sess = NULL;          struct tagSession *sess = NULL;
   
         FTRACE(4);          FTRACE(4);
   
         sess = initSession(TASK_FD(task), TASK_ARG(task));          sess = initSession(TASK_FD(task), TASK_ARG(task));
        if (!sess)        if (!sess) {
                 if (TASK_ARG(task))
                         free(TASK_ARG(task));
                 close(TASK_FD(task));
                 return NULL;                  return NULL;
           }
   
         if (recv(sess->sess_sock, buf.msg_base, buf.msg_len, 0) == -1) {          if (recv(sess->sess_sock, buf.msg_base, buf.msg_len, 0) == -1) {
                 VERB(3) syslog(LOG_ERR, "Error:: recv(%d) #%d - %s", sess->sess_sock, errno, strerror(errno));                  VERB(3) syslog(LOG_ERR, "Error:: recv(%d) #%d - %s", sess->sess_sock, errno, strerror(errno));
Line 111  startSession(sched_task_t *task) Line 122  startSession(sched_task_t *task)
         }          }
   
         /* check online table for user */          /* check online table for user */
           if (call.LoginACC(&cfg, acc, sess->sess_user, sess->sess_pass)) {
                   VERB(1) syslog(LOG_WARNING, "Login:: DENIED for username %s and password %s", 
                                   sess->sess_user, sess->sess_pass);
                   goto end;
           } else
                   VERB(1) syslog(LOG_WARNING, "Login:: ALLOWED for username %s ...", sess->sess_user);
   
           /* Start session thread OK ... */
           TAILQ_INSERT_TAIL(&Sessions, sess, sess_node);
           pthread_create(&sess->sess_tid, NULL, (void*(*)(void*)) thrSession, sess);
           pthread_detach(sess->sess_tid);
           VERB(1) syslog(LOG_DEBUG, "ConnID=%s(%s) for %s login OK!", 
                           sess->sess_cid, sess->sess_user, sess->sess_addr);
           return NULL;
 end:    /* close client connection */  end:    /* close client connection */
         if (sess)  
                 free(sess);  
         close(TASK_FD(task));  
         VERB(1) syslog(LOG_DEBUG, "Close client %s with socket=%d",           VERB(1) syslog(LOG_DEBUG, "Close client %s with socket=%d", 
                         (char*) TASK_ARG(task), (int) TASK_FD(task));                          (char*) TASK_ARG(task), (int) TASK_FD(task));
        if (TASK_ARG(task))        finiSession(sess);
                free(TASK_ARG(task)); 
         return NULL;          return NULL;
 }  }
   

Removed from v.1.1.2.14  
changed lines
  Added in v.1.1.2.16


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