Diff for /mqtt/src/mqttd_calls.c between versions 1.2.2.28 and 1.2.2.32

version 1.2.2.28, 2012/06/26 14:51:33 version 1.2.2.32, 2012/06/29 15:43:13
Line 1 Line 1
 #include "global.h"  #include "global.h"
 #include "mqttd.h"  #include "mqttd.h"
   #include "utils.h"
 #include "rtlm.h"  #include "rtlm.h"
 #include "mqttd_calls.h"  #include "mqttd_calls.h"
   
Line 60  pubOnce(struct tagSession *sess, char * __restrict psT Line 61  pubOnce(struct tagSession *sess, char * __restrict psT
 {  {
         ait_val_t *p = NULL;          ait_val_t *p = NULL;
         struct tagSession *s = NULL;          struct tagSession *s = NULL;
        struct tagStore *st = NULL;        struct tagStore *st_, *st = NULL;
         regex_t re;          regex_t re;
         regmatch_t match;          regmatch_t match;
         int ret;          int ret;
         char szStr[STRSIZ];          char szStr[STRSIZ];
   
   
         TAILQ_FOREACH(s, &Sessions, sess_node) {          TAILQ_FOREACH(s, &Sessions, sess_node) {
                SLIST_FOREACH(st, &s->sess_subscr, st_node) {                SLIST_FOREACH_SAFE(st, &s->sess_subscr, st_node, st_) {
                         if ((ret = regcomp(&re, st->st_subscr.sub_topic.msg_base, REG_EXTENDED))) {                          if ((ret = regcomp(&re, st->st_subscr.sub_topic.msg_base, REG_EXTENDED))) {
                                 regerror(ret, &re, szStr, sizeof szStr);                                  regerror(ret, &re, szStr, sizeof szStr);
                                 regfree(&re);                                  regfree(&re);
Line 92  pubAck(struct tagSession *sess, u_short mid, char * __ Line 94  pubAck(struct tagSession *sess, u_short mid, char * __
 {  {
         ait_val_t *p = NULL;          ait_val_t *p = NULL;
         struct tagSession *s = NULL;          struct tagSession *s = NULL;
        struct tagStore *st = NULL;        struct tagStore *st_, *st = NULL;
         regex_t re;          regex_t re;
         regmatch_t match;          regmatch_t match;
         int ret;          int ret;
Line 109  pubAck(struct tagSession *sess, u_short mid, char * __ Line 111  pubAck(struct tagSession *sess, u_short mid, char * __
                         sess->sess_user, sess->sess_addr, hdr->mqtt_msg.qos, hdr->mqtt_msg.retain);                          sess->sess_user, sess->sess_addr, hdr->mqtt_msg.qos, hdr->mqtt_msg.retain);
   
         TAILQ_FOREACH(s, &Sessions, sess_node) {          TAILQ_FOREACH(s, &Sessions, sess_node) {
                SLIST_FOREACH(st, &s->sess_subscr, st_node) {                SLIST_FOREACH_SAFE(st, &s->sess_subscr, st_node, st_) {
                         /* check for QoS */                          /* check for QoS */
                         if (st->st_subscr.sub_ret < MQTT_QOS_ACK)                          if (st->st_subscr.sub_ret < MQTT_QOS_ACK)
                                 continue;                                  continue;
Line 143  pubExactly(struct tagSession *sess, u_short mid, char  Line 145  pubExactly(struct tagSession *sess, u_short mid, char 
 {  {
         ait_val_t *p = NULL;          ait_val_t *p = NULL;
         struct tagSession *s = NULL;          struct tagSession *s = NULL;
        struct tagStore *st = NULL;        struct tagStore *st_, *st = NULL;
         regex_t re;          regex_t re;
         regmatch_t match;          regmatch_t match;
         int ret;          int ret;
Line 160  pubExactly(struct tagSession *sess, u_short mid, char  Line 162  pubExactly(struct tagSession *sess, u_short mid, char 
                         sess->sess_user, sess->sess_addr, hdr->mqtt_msg.qos, hdr->mqtt_msg.retain);                          sess->sess_user, sess->sess_addr, hdr->mqtt_msg.qos, hdr->mqtt_msg.retain);
   
         TAILQ_FOREACH(s, &Sessions, sess_node) {          TAILQ_FOREACH(s, &Sessions, sess_node) {
                SLIST_FOREACH(st, &s->sess_subscr, st_node) {                SLIST_FOREACH_SAFE(st, &s->sess_subscr, st_node, st_) {
                         /* check for QoS */                          /* check for QoS */
                         if (st->st_subscr.sub_ret < MQTT_QOS_EXACTLY)                          if (st->st_subscr.sub_ret < MQTT_QOS_EXACTLY)
                                 continue;                                  continue;
Line 262  cmdPUBREL(void *srv, int len, void *arg) Line 264  cmdPUBREL(void *srv, int len, void *arg)
                 return 0;                  return 0;
         }          }
   
        // TODO:: Delete from database topic        /* delete not retain message */
         call.DeletePUB_topic(&cfg, pub, sess->sess_cid, mid, "%", sess->sess_user, 
                         sess->sess_addr, 0);
   
         siz = mqtt_msgPUBCOMP(sess->sess_buf, mid);          siz = mqtt_msgPUBCOMP(sess->sess_buf, mid);
         if (siz == -1) {          if (siz == -1) {
                 ioDEBUG(5, "Error:: in msgPUBCOMP #%d - %s", mqtt_GetErrno(), mqtt_GetError());                  ioDEBUG(5, "Error:: in msgPUBCOMP #%d - %s", mqtt_GetErrno(), mqtt_GetError());
                 return 0;                  return 0;
         } else {  
                 p = mkPkt(sess->sess_buf->msg_base, siz);  
                 memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len);  
         }          }
   
           p = mkPkt(sess->sess_buf->msg_base, siz);
           memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len);
         schedWrite(root, sendPacket, p, sess->sess_sock, NULL, 0);          schedWrite(root, sendPacket, p, sess->sess_sock, NULL, 0);
         return 0;          return 0;
 }  }
Line 304  cmdSUBSCRIBE(void *srv, int len, void *arg) Line 307  cmdSUBSCRIBE(void *srv, int len, void *arg)
   
         /* add to db */          /* add to db */
         for (i = 0; i < siz; i++, subs[i].sub_ret = MQTT_QOS_DENY) {          for (i = 0; i < siz; i++, subs[i].sub_ret = MQTT_QOS_DENY) {
                /* convert topic to sql search statement */                store = io_malloc(sizeof(struct tagStore));
                if (mqtt_sqlTopic(subs[i].sub_topic.msg_base, buf, sizeof buf) == -1) {                if (!store) {
                        ioDEBUG(5, "Error:: in db #%d - %s", mqtt_GetErrno(), mqtt_GetError());                        ioSYSERR(0);
                         continue;                          continue;
                   } else {
                           store->st_msgid = mid;
                           mqtt_subCopy(&store->st_subscr, &subs[i]);
                 }                  }
                if (call.WritePUB_subscribe(&cfg, pub, sess->sess_cid, mid, buf, 
                                sess->sess_user, sess->sess_addr, subs[i].sub_ret) > 0) {                /* add to cache */
                        store = io_malloc(sizeof(struct tagStore));                SLIST_INSERT_HEAD(&sess->sess_subscr, store, st_node);
                        if (!store) {
                 /* convert topic to regexp */
                 if (mqtt_expandTopic(subs[i].sub_topic.msg_base, buf, sizeof buf, 1, 1) == -1) {
                         ioDEBUG(5, "Error:: in regexp #%d - %s", mqtt_GetErrno(), mqtt_GetError());
 
                         subs[i].sub_ret = MQTT_QOS_DENY;
                 } else {
                         ptr = realloc(store->st_subscr.sub_topic.msg_base, strlen(buf) + 1);
                         if (!ptr) {
                                 ioSYSERR(0);                                  ioSYSERR(0);
                                 continue;                                  continue;
                         } else {                          } else {
                                store->st_msgid = mid;                                store->st_subscr.sub_topic.msg_base = ptr;
                                mqtt_subCopy(&store->st_subscr, &subs[i]);                                store->st_subscr.sub_topic.msg_len = strlen(buf) + 1;
                                 memcpy(store->st_subscr.sub_topic.msg_base, buf, 
                                                 store->st_subscr.sub_topic.msg_len);
                         }                          }
   
                        /* add to cache */                        call.LOG(logg, "Added [%s] SUBSCRIBE '%s'(%d) from %s\n", sess->sess_cid, 
                        SLIST_INSERT_HEAD(&sess->sess_subscr, store, st_node);                                        store->st_subscr.sub_topic.msg_base, 
                                         store->st_subscr.sub_topic.msg_len, sess->sess_addr);
   
                        /* convert topic to regexp */                        subs[i].sub_ret = MQTT_QOS_PASS;
                        if (mqtt_expandTopic(subs[i].sub_topic.msg_base, buf, sizeof buf, 1, 1) == -1) { 
                                ioDEBUG(5, "Error:: in regexp #%d - %s", mqtt_GetErrno(), mqtt_GetError()); 
 
                                subs[i].sub_ret = MQTT_QOS_DENY; 
                        } else { 
                                ptr = realloc(store->st_subscr.sub_topic.msg_base, strlen(buf) + 1); 
                                if (!ptr) 
                                        ioSYSERR(0); 
                                else { 
                                        store->st_subscr.sub_topic.msg_base = ptr; 
                                        store->st_subscr.sub_topic.msg_len = strlen(buf) + 1; 
                                        memcpy(store->st_subscr.sub_topic.msg_base, buf,  
                                                        store->st_subscr.sub_topic.msg_len); 
                                } 
 
                                call.LOG(logg, "Added [%s] SUBSCRIBE '%s'(%d) from %s\n", sess->sess_cid,  
                                                store->st_subscr.sub_topic.msg_base,  
                                                store->st_subscr.sub_topic.msg_len, sess->sess_addr); 
 
                                subs[i].sub_ret = MQTT_QOS_PASS; 
                        } 
                 }                  }
   
                   call.WritePUB_subscribe(&cfg, pub, sess->sess_cid, mid, buf, 
                                   sess->sess_user, sess->sess_addr, subs[i].sub_ret);
         }          }
   
         /* send acknowledge */          /* send acknowledge */
Line 483  cmdCONNECT(void *srv, int len, void *arg) Line 482  cmdCONNECT(void *srv, int len, void *arg)
   
                 io_free(store);                  io_free(store);
         }          }
   
           if (sess->sess_will.flag)
                   srv_Will(sess);
   
         if (sess->sess_will.msg)          if (sess->sess_will.msg)
                 free(sess->sess_will.msg);                  free(sess->sess_will.msg);

Removed from v.1.2.2.28  
changed lines
  Added in v.1.2.2.32


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