Diff for /mqtt/src/mqttd_calls.c between versions 1.2.2.18 and 1.2.2.22

version 1.2.2.18, 2012/05/27 10:04:05 version 1.2.2.22, 2012/06/19 08:32:17
Line 4 Line 4
 #include "mqttd_calls.h"  #include "mqttd_calls.h"
   
   
   static inline struct tagPkt *
   mkPkt(void * __restrict data, int dlen)
   {
           struct tagPkt *p = NULL;
   
           p = io_malloc(sizeof(struct tagPkt));
           if (!p) {
                   ioDEBUG(7, "Error:: in send packet prepare #%d - %s", io_GetErrno(), io_GetError());
                   return NULL;
           } else
                   memset(p, 0, sizeof(struct tagPkt));
   
           p->pkt_data = io_allocVar();
           if (!p->pkt_data) {
                   ioDEBUG(7, "Error:: in send packet prepare #%d - %s", io_GetErrno(), io_GetError());
                   io_free(p);
                   return NULL;
           }
   
           if (data && dlen > 0)
                   AIT_SET_BUF(p->pkt_data, data, dlen);
   
           return p;
   }
   
   static inline void
   freePkt(struct tagPkt * __restrict p)
   {
           if (!p)
                   return;
   
           io_freeVar(&p->pkt_data);
           io_free(p);
   }
   
   static void *
   sendPacket(sched_task_t *task)
   {
           struct tagPkt *p = TASK_ARG(task);
           register int n, slen;
           u_char *pos;
   
           if (!p || !p->pkt_data || AIT_ISEMPTY(p->pkt_data)) {
                   ioDEBUG(9, "Error:: invalid packet or found empty content ...");
                   return NULL;
           }
   
           for (slen = AIT_LEN(p->pkt_data), pos = AIT_GET_BUF(p->pkt_data); slen > 0; 
                           slen -= n, pos += n) {
                   n = send(TASK_FD(task), pos, slen, MSG_NOSIGNAL);
                   if (n == -1) {
                           ioSYSERR(0);
                           break;
                   }
           }
   
           freePkt(p);
           return NULL;
   }
   
   /* --------------------------------------------------- */
   
 static int  static int
pubOnce(struct tagSession *sess, u_short mid, char * __restrict psTopic, pubOnce(struct tagSession *sess, char * __restrict psTopic, int datlen)
                int topicLen, char * __restrict data, int datlen) 
 {  {
           struct tagPkt *p = NULL;
           struct tagSession *s = NULL;
           struct tagStore *st = NULL;
           regex_t re;
           regmatch_t match;
           int ret;
           char szStr[STRSIZ];
   
           TAILQ_FOREACH(s, &Sessions, sess_node) {
                   SLIST_FOREACH(st, &s->sess_subscr, st_node) {
                           if ((ret = regcomp(&re, st->st_subscr.sub_topic.msg_base, REG_EXTENDED))) {
                                   regerror(ret, &re, szStr, sizeof szStr);
                                   regfree(&re);
                                   ioDEBUG(3, "Error:: regcomp(%s) %s\n", (char*)
                                                   st->st_subscr.sub_topic.msg_base, szStr);
                           }
                           if (!regexec(&re, psTopic, 1, &match, 0)) {
                                   /* MATCH */
                                   p = mkPkt(sess->sess_buf->msg_base, datlen);
                                   schedWrite(root, sendPacket, p, sess->sess_sock, NULL, 0);
                           }
   
                           regfree(&re);
                   }
           }
   
         return 0;          return 0;
 }  }
   
 static int  static int
pubAck(struct tagSession *sess, u_short mid, char * __restrict psTopic, pubAck(struct tagSession *sess, char * __restrict psTopic, int datlen)
                int topicLen, char * __restrict data, int datlen) 
 {  {
         return 0;          return 0;
 }  }
   
 static int  static int
pubExactly(struct tagSession *sess, u_short mid, char * __restrict psTopic, pubExactly(struct tagSession *sess, char * __restrict psTopic, int datlen)
                int topicLen, char * __restrict data, int datlen) 
 {  {
         return 0;          return 0;
 }  }
Line 31  cmdPUBLISH(void *srv, int len, void *arg) Line 116  cmdPUBLISH(void *srv, int len, void *arg)
 {  {
         struct mqtthdr *hdr;          struct mqtthdr *hdr;
         struct tagSession *sess = (struct tagSession*) arg;          struct tagSession *sess = (struct tagSession*) arg;
         void *data = NULL;  
         char szTopic[STRSIZ] = { 0 };          char szTopic[STRSIZ] = { 0 };
         int siz = 0;          int siz = 0;
         u_short mid = 0;          u_short mid = 0;
           struct tagPkt *p = NULL;
   
         ioTRACE(2);          ioTRACE(2);
   
Line 42  cmdPUBLISH(void *srv, int len, void *arg) Line 127  cmdPUBLISH(void *srv, int len, void *arg)
                 return -1;                  return -1;
   
         ioDEBUG(5, "Exec PUBLISH session");          ioDEBUG(5, "Exec PUBLISH session");
        siz = mqtt_readPUBLISH(sess->sess_buf, szTopic, sizeof szTopic, &mid, &data);        siz = mqtt_readPUBLISH(sess->sess_buf, szTopic, sizeof szTopic, &mid, NULL);
         if (siz == -1) {          if (siz == -1) {
                 ioDEBUG(5, "Error:: in readPUBLISH #%d - %s", mqtt_GetErrno(), mqtt_GetError());                  ioDEBUG(5, "Error:: in readPUBLISH #%d - %s", mqtt_GetErrno(), mqtt_GetError());
                 return 0;                  return 0;
         }          }
   
           /* duplicate packet for retransmit to subscribers */
           /*
           pubpkt = mqtt_msgDup(sess->sess_buf);
           if (!pubpkt) {
                   ioDEBUG(5, "Error:: in duplicate packet #%d - %s", mqtt_GetErrno(), mqtt_GetError());
                   return 0;
           } else
           */
   
         hdr = (struct mqtthdr*) sess->sess_buf->msg_base;          hdr = (struct mqtthdr*) sess->sess_buf->msg_base;
         switch (hdr->mqtt_msg.qos) {          switch (hdr->mqtt_msg.qos) {
                 case MQTT_QOS_ACK:                  case MQTT_QOS_ACK:
                        pubAck(sess, mid, szTopic, sizeof szTopic, data, siz);                        pubAck(sess, szTopic, siz);
                         siz = mqtt_msgPUBACK(sess->sess_buf, mid);                          siz = mqtt_msgPUBACK(sess->sess_buf, mid);
                         if (siz == -1) {                          if (siz == -1) {
                                 ioDEBUG(5, "Error:: in msgPUBACK #%d - %s",                                   ioDEBUG(5, "Error:: in msgPUBACK #%d - %s", 
                                                 mqtt_GetErrno(), mqtt_GetError());                                                  mqtt_GetErrno(), mqtt_GetError());
                                goto end;                                return 0;
                         }                          }
                         break;                          break;
                 case MQTT_QOS_EXACTLY:                  case MQTT_QOS_EXACTLY:
                        pubExactly(sess, mid, szTopic, sizeof szTopic, data, siz);                        pubExactly(sess, szTopic, siz);
                         siz = mqtt_msgPUBREC(sess->sess_buf, mid);                          siz = mqtt_msgPUBREC(sess->sess_buf, mid);
                         if (siz == -1) {                          if (siz == -1) {
                                 ioDEBUG(5, "Error:: in msgPUBREC #%d - %s",                                   ioDEBUG(5, "Error:: in msgPUBREC #%d - %s", 
                                                 mqtt_GetErrno(), mqtt_GetError());                                                  mqtt_GetErrno(), mqtt_GetError());
                                goto end;                                return 0;
                         }                          }
                         break;                          break;
                 case MQTT_QOS_ONCE:                  case MQTT_QOS_ONCE:
                        pubOnce(sess, mid, szTopic, sizeof szTopic, data, siz);                        pubOnce(sess, szTopic, siz);
                 default:                  default:
                        goto end;                        return 0;
         }          }
   
        if ((siz = send(sess->sess_sock, sess->sess_buf->msg_base, siz, MSG_NOSIGNAL)) == -1)        p = mkPkt(sess->sess_buf->msg_base, siz);
                ioSYSERR(0);        memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len);
        else {        schedWrite(root, sendPacket, p, sess->sess_sock, NULL, 0);
                ioDEBUG(5, "Sended %d bytes.", siz); 
                memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len); 
        } 
end: 
        if (data) 
                free(data); 
         return 0;          return 0;
 }  }
   
Line 92  cmdPUBREL(void *srv, int len, void *arg) Line 180  cmdPUBREL(void *srv, int len, void *arg)
         struct tagSession *sess = (struct tagSession*) arg;          struct tagSession *sess = (struct tagSession*) arg;
         int siz = 0;          int siz = 0;
         u_short mid = 0;          u_short mid = 0;
           struct tagPkt *p = NULL;
   
         ioTRACE(2);          ioTRACE(2);
   
Line 111  cmdPUBREL(void *srv, int len, void *arg) Line 200  cmdPUBREL(void *srv, int len, void *arg)
         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 {
        if ((siz = send(sess->sess_sock, sess->sess_buf->msg_base, siz, MSG_NOSIGNAL)) == -1)                p = mkPkt(sess->sess_buf->msg_base, siz);
                ioSYSERR(0); 
        else { 
                ioDEBUG(5, "Sended %d bytes.", siz); 
                 memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len);                  memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len);
         }          }
   
           schedWrite(root, sendPacket, p, sess->sess_sock, NULL, 0);
         return 0;          return 0;
 }  }
   
Line 133  cmdSUBSCRIBE(void *srv, int len, void *arg) Line 220  cmdSUBSCRIBE(void *srv, int len, void *arg)
         struct tagStore *store;          struct tagStore *store;
         char buf[BUFSIZ];          char buf[BUFSIZ];
         void *ptr;          void *ptr;
           struct tagPkt *p = NULL;
   
         ioTRACE(2);          ioTRACE(2);
   
Line 198  cmdSUBSCRIBE(void *srv, int len, void *arg) Line 286  cmdSUBSCRIBE(void *srv, int len, void *arg)
         if (siz == -1) {          if (siz == -1) {
                 ioDEBUG(5, "Error:: in msgSUBACK #%d - %s", mqtt_GetErrno(), mqtt_GetError());                  ioDEBUG(5, "Error:: in msgSUBACK #%d - %s", mqtt_GetErrno(), mqtt_GetError());
                 goto end;                  goto end;
        }        } else {
        if ((siz = send(sess->sess_sock, sess->sess_buf->msg_base, siz, MSG_NOSIGNAL)) == -1)                p = mkPkt(sess->sess_buf->msg_base, siz);
                ioSYSERR(0); 
        else { 
                ioDEBUG(5, "Sended %d bytes.", siz); 
                 memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len);                  memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len);
         }          }
   
           schedWrite(root, sendPacket, p, sess->sess_sock, NULL, 0);
 end:  end:
         mqtt_subFree(&subs);          mqtt_subFree(&subs);
         return 0;          return 0;
Line 219  cmdUNSUBSCRIBE(void *srv, int len, void *arg) Line 306  cmdUNSUBSCRIBE(void *srv, int len, void *arg)
         u_short mid = 0;          u_short mid = 0;
         register int i;          register int i;
         struct tagStore *store, *tmp;          struct tagStore *store, *tmp;
           struct tagPkt *p = NULL;
   
         ioTRACE(2);          ioTRACE(2);
   
Line 258  cmdUNSUBSCRIBE(void *srv, int len, void *arg) Line 346  cmdUNSUBSCRIBE(void *srv, int len, void *arg)
         if (siz == -1) {          if (siz == -1) {
                 ioDEBUG(5, "Error:: in msgUNSUBACK #%d - %s", mqtt_GetErrno(), mqtt_GetError());                  ioDEBUG(5, "Error:: in msgUNSUBACK #%d - %s", mqtt_GetErrno(), mqtt_GetError());
                 goto end;                  goto end;
        }        } else {
        if ((siz = send(sess->sess_sock, sess->sess_buf->msg_base, siz, MSG_NOSIGNAL)) == -1)                p = mkPkt(sess->sess_buf->msg_base, siz);
                ioSYSERR(0); 
        else { 
                ioDEBUG(5, "Sended %d bytes.", siz); 
                 memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len);                  memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len);
         }          }
   
           schedWrite(root, sendPacket, p, sess->sess_sock, NULL, 0);
 end:  end:
         mqtt_subFree(&subs);          mqtt_subFree(&subs);
         return 0;          return 0;
Line 275  cmdPINGREQ(void *srv, int len, void *arg) Line 362  cmdPINGREQ(void *srv, int len, void *arg)
 {  {
         struct tagSession *sess = (struct tagSession*) arg;          struct tagSession *sess = (struct tagSession*) arg;
         int siz = 0;          int siz = 0;
           struct tagPkt *p = NULL;
   
         ioTRACE(2);          ioTRACE(2);
   
Line 286  cmdPINGREQ(void *srv, int len, void *arg) Line 374  cmdPINGREQ(void *srv, int len, void *arg)
         if (siz == -1) {          if (siz == -1) {
                 ioDEBUG(5, "Error:: in msgPINGRESP #%d - %s", mqtt_GetErrno(), mqtt_GetError());                  ioDEBUG(5, "Error:: in msgPINGRESP #%d - %s", mqtt_GetErrno(), mqtt_GetError());
                 return 0;                  return 0;
         }  
         if ((siz = send(sess->sess_sock, sess->sess_buf->msg_base, siz, MSG_NOSIGNAL)) == -1) {  
                 ioSYSERR(0);  
                 return 0;  
         } else {          } else {
                ioDEBUG(5, "Sended %d bytes.", siz);                p = mkPkt(sess->sess_buf->msg_base, siz);
                 memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len);                  memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len);
         }          }
   
           schedWrite(root, sendPacket, p, sess->sess_sock, NULL, 0);
         return 0;          return 0;
 }  }
   
Line 302  int Line 387  int
 cmdCONNECT(void *srv, int len, void *arg)  cmdCONNECT(void *srv, int len, void *arg)
 {  {
         struct tagStore *store;          struct tagStore *store;
           struct tagPkt *p;
         struct tagSession *sess = (struct tagSession*) arg;          struct tagSession *sess = (struct tagSession*) arg;
   
         ioTRACE(2);          ioTRACE(2);
Line 330  cmdCONNECT(void *srv, int len, void *arg) Line 416  cmdCONNECT(void *srv, int len, void *arg)
                         free(store->st_subscr.sub_value.msg_base);                          free(store->st_subscr.sub_value.msg_base);
   
                 io_free(store);                  io_free(store);
           }
   
           while ((p = SLIST_FIRST(&sess->sess_sndpkt))) {
                   SLIST_REMOVE_HEAD(&sess->sess_sndpkt, pkt_node);
   
                   io_freeVar(&p->pkt_data);
                   io_free(p);
         }          }
   
         if (sess->sess_will.msg)          if (sess->sess_will.msg)

Removed from v.1.2.2.18  
changed lines
  Added in v.1.2.2.22


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