--- mqtt/src/mqttd_calls.c 2012/06/29 13:46:37 1.2.2.30 +++ mqtt/src/mqttd_calls.c 2013/01/18 10:17:22 1.4.4.1 @@ -1,5 +1,51 @@ +/************************************************************************* +* (C) 2011 AITNET ltd - Sofia/Bulgaria - +* by Michael Pounov +* +* $Author: misho $ +* $Id: mqttd_calls.c,v 1.4.4.1 2013/01/18 10:17:22 misho Exp $ +* +************************************************************************** +The ELWIX and AITNET software is distributed under the following +terms: + +All of the documentation and software included in the ELWIX and AITNET +Releases is copyrighted by ELWIX - Sofia/Bulgaria + +Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013 + by Michael Pounov . All rights reserved. + +Redistribution and use in source and binary forms, with or without +modification, are permitted provided that the following conditions +are met: +1. Redistributions of source code must retain the above copyright + notice, this list of conditions and the following disclaimer. +2. Redistributions in binary form must reproduce the above copyright + notice, this list of conditions and the following disclaimer in the + documentation and/or other materials provided with the distribution. +3. All advertising materials mentioning features or use of this software + must display the following acknowledgement: +This product includes software developed by Michael Pounov +ELWIX - Embedded LightWeight unIX and its contributors. +4. Neither the name of AITNET nor the names of its contributors + may be used to endorse or promote products derived from this software + without specific prior written permission. + +THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND +ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE +IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE +ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE +FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL +DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS +OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) +HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT +LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY +OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF +SUCH DAMAGE. +*/ #include "global.h" #include "mqttd.h" +#include "utils.h" #include "rtlm.h" #include "mqttd_calls.h" @@ -9,8 +55,9 @@ mkPkt(void * __restrict data, int dlen) { ait_val_t *p = NULL; - if (!(p = io_allocVar())) { - ioDEBUG(7, "Error:: in send packet prepare #%d - %s", io_GetErrno(), io_GetError()); + if (!(p = ait_allocVar())) { + EVERBOSE(7, "Error:: in send packet prepare #%d - %s", + elwix_GetErrno(), elwix_GetError()); return NULL; } @@ -26,7 +73,7 @@ freePkt(ait_val_t ** __restrict p) if (!p) return; - io_freeVar(p); + ait_freeVar(p); } static void * @@ -37,14 +84,16 @@ sendPacket(sched_task_t *task) u_char *pos; if (!p || AIT_ISEMPTY(p)) { - ioDEBUG(9, "Error:: invalid packet or found empty content ..."); + EVERBOSE(9, "Error:: invalid packet or found empty content ..."); return NULL; } + EVERBOSE(7, "Send packet length %d for socket %d\n", AIT_LEN(p), (u_int) TASK_FD(task)); + for (slen = AIT_LEN(p), pos = AIT_GET_BUF(p); slen > 0; slen -= n, pos += n) { n = send(TASK_FD(task), pos, slen, MSG_NOSIGNAL); if (n == -1) { - ioSYSERR(0); + ESYSERR(0); break; } } @@ -53,137 +102,137 @@ sendPacket(sched_task_t *task) return NULL; } -/* --------------------------------------------------- */ - static int -pubOnce(struct tagSession *sess, char * __restrict psTopic, int datlen) +search4send(struct tagSession * __restrict sess, const char *topic, int datlen, char qos) { + regex_t re; + regmatch_t match; ait_val_t *p = NULL; struct tagSession *s = NULL; struct tagStore *st_, *st = NULL; - regex_t re; - regmatch_t match; - int ret; char szStr[STRSIZ]; + int ret; + assert(sess); TAILQ_FOREACH(s, &Sessions, sess_node) { SLIST_FOREACH_SAFE(st, &s->sess_subscr, st_node, st_) { - if ((ret = regcomp(&re, st->st_subscr.sub_topic.msg_base, REG_EXTENDED))) { - regerror(ret, &re, szStr, sizeof szStr); + /* check for QoS */ + if (st->st_subscr.sub_ret >= qos) { + if ((ret = regcomp(&re, st->st_subscr.sub_topic.msg_base, REG_EXTENDED))) { + regerror(ret, &re, szStr, sizeof szStr); + regfree(&re); + EVERBOSE(3, "Error:: regcomp(%s) %s\n", (char*) + st->st_subscr.sub_topic.msg_base, szStr); + } + if (!regexec(&re, topic, 1, &match, 0)) { + /* MATCH */ + p = mkPkt(sess->sess_buf->msg_base, datlen); + schedWrite(root, sendPacket, p, s->sess_sock, NULL, 0); + } + 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, s->sess_sock, NULL, 0); - } - - regfree(&re); } } return 0; } +/* --------------------------------------------------- */ + +void * +sendRetain(sched_task_t *task) +{ + mqtt_subscr_t *subs, *s; + struct tagSession *sess; + int siz; + + ETRACE(2); + + assert(task); + + sess = TASK_ARG(task); + assert(sess); + + if (!sess->sess_buf) { + EVERBOSE(9, "WARNING! No allocated buffer!?!\n"); + return NULL; + } + + subs = call.ReadPUB_topic(&cfg, pub, "%", "%", 1); + if (!subs) + return NULL; + + for (s = subs; s && s->sub_topic.msg_base; s++) { + siz = s->sub_value.msg_len; + memcpy(sess->sess_buf->msg_base, s->sub_value.msg_base, + MIN(sess->sess_buf->msg_len, s->sub_value.msg_len)); + EVERBOSE(7, "Sending retain message %d bytes, QoS %hhd topic '%s' data length %d\n", + siz, s->sub_ret, (char*) s->sub_topic.msg_base, s->sub_value.msg_len); + if (siz > 0) + search4send(sess, s->sub_topic.msg_base, siz, s->sub_ret); + } + + mqtt_subFree(&subs); + return NULL; +} + +int +pubWill(struct tagSession * __restrict sess) +{ + int datlen; + + ETRACE(2); + + /* prepare will packet */ + datlen = mqtt_msgPUBLISH(sess->sess_buf, sess->sess_will.topic, 0xDEAD, 0, 1, 0, + sess->sess_will.msg, sess->sess_will.msg ? strlen(sess->sess_will.msg) : 0); + if (datlen == -1) + return -1; /* error */ + + return search4send(sess, sess->sess_will.topic, datlen, MQTT_QOS_ACK); +} + static int +pubOnce(struct tagSession *sess, char * __restrict psTopic, int datlen) +{ + return search4send(sess, psTopic, datlen, MQTT_QOS_ONCE); +} + +static int pubAck(struct tagSession *sess, u_short mid, char * __restrict psTopic, int datlen) { - ait_val_t *p = NULL; - struct tagSession *s = NULL; - struct tagStore *st_, *st = NULL; - regex_t re; - regmatch_t match; - int ret; - char szStr[STRSIZ]; - struct mqtthdr *hdr; + struct mqtthdr *hdr = (struct mqtthdr*) sess->sess_buf->msg_base; - p = mkPkt(sess->sess_buf->msg_base, datlen); - hdr = (struct mqtthdr*) sess->sess_buf->msg_base; - /* write topic to database */ call.DeletePUB_topic(&cfg, pub, sess->sess_cid, mid, psTopic, sess->sess_user, sess->sess_addr, hdr->mqtt_msg.retain); - call.WritePUB_topic(&cfg, pub, sess->sess_cid, mid, psTopic, AIT_GET_BUF(p), AIT_LEN(p), - sess->sess_user, sess->sess_addr, hdr->mqtt_msg.qos, hdr->mqtt_msg.retain); + call.WritePUB_topic(&cfg, pub, sess->sess_cid, mid, psTopic, + sess->sess_buf->msg_base, datlen, sess->sess_user, sess->sess_addr, + hdr->mqtt_msg.qos, hdr->mqtt_msg.retain); - TAILQ_FOREACH(s, &Sessions, sess_node) { - SLIST_FOREACH_SAFE(st, &s->sess_subscr, st_node, st_) { - /* check for QoS */ - if (st->st_subscr.sub_ret < MQTT_QOS_ACK) - continue; + search4send(sess, psTopic, datlen, MQTT_QOS_ACK); - 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 */ - schedWrite(root, sendPacket, mkPkt(AIT_GET_BUF(p), AIT_LEN(p)), - s->sess_sock, NULL, 0); - } - - regfree(&re); - } - } - /* delete not retain message */ call.DeletePUB_topic(&cfg, pub, sess->sess_cid, mid, psTopic, sess->sess_user, sess->sess_addr, 0); - - freePkt(&p); return 0; } static int pubExactly(struct tagSession *sess, u_short mid, char * __restrict psTopic, int datlen) { - ait_val_t *p = NULL; - struct tagSession *s = NULL; - struct tagStore *st_, *st = NULL; - regex_t re; - regmatch_t match; - int ret; - char szStr[STRSIZ]; - struct mqtthdr *hdr; + struct mqtthdr *hdr = (struct mqtthdr*) sess->sess_buf->msg_base; - p = mkPkt(sess->sess_buf->msg_base, datlen); - hdr = (struct mqtthdr*) sess->sess_buf->msg_base; - /* write topic to database */ call.DeletePUB_topic(&cfg, pub, sess->sess_cid, mid, psTopic, sess->sess_user, sess->sess_addr, hdr->mqtt_msg.retain); - call.WritePUB_topic(&cfg, pub, sess->sess_cid, mid, psTopic, AIT_GET_BUF(p), AIT_LEN(p), - sess->sess_user, sess->sess_addr, hdr->mqtt_msg.qos, hdr->mqtt_msg.retain); + call.WritePUB_topic(&cfg, pub, sess->sess_cid, mid, psTopic, + sess->sess_buf->msg_base, datlen, sess->sess_user, sess->sess_addr, + hdr->mqtt_msg.qos, hdr->mqtt_msg.retain); - TAILQ_FOREACH(s, &Sessions, sess_node) { - SLIST_FOREACH_SAFE(st, &s->sess_subscr, st_node, st_) { - /* check for QoS */ - if (st->st_subscr.sub_ret < MQTT_QOS_EXACTLY) - continue; - - 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 */ - schedWrite(root, sendPacket, mkPkt(AIT_GET_BUF(p), AIT_LEN(p)), - s->sess_sock, NULL, 0); - } - - regfree(&re); - } - } - - freePkt(&p); - return 0; + return search4send(sess, psTopic, datlen, MQTT_QOS_EXACTLY); } @@ -197,15 +246,15 @@ cmdPUBLISH(void *srv, int len, void *arg) u_short mid = 0; ait_val_t *p = NULL; - ioTRACE(2); + ETRACE(2); if (!sess) return -1; - ioDEBUG(5, "Exec PUBLISH session"); + EVERBOSE(5, "Exec PUBLISH session"); siz = mqtt_readPUBLISH(sess->sess_buf, szTopic, sizeof szTopic, &mid, NULL); if (siz == -1) { - ioDEBUG(5, "Error:: in readPUBLISH #%d - %s", mqtt_GetErrno(), mqtt_GetError()); + EVERBOSE(5, "Error:: in readPUBLISH #%d - %s", mqtt_GetErrno(), mqtt_GetError()); return 0; } @@ -216,7 +265,7 @@ cmdPUBLISH(void *srv, int len, void *arg) return 0; siz = mqtt_msgPUBACK(sess->sess_buf, mid); if (siz == -1) { - ioDEBUG(5, "Error:: in msgPUBACK #%d - %s", + EVERBOSE(5, "Error:: in msgPUBACK #%d - %s", mqtt_GetErrno(), mqtt_GetError()); return 0; } @@ -226,7 +275,7 @@ cmdPUBLISH(void *srv, int len, void *arg) return 0; siz = mqtt_msgPUBREC(sess->sess_buf, mid); if (siz == -1) { - ioDEBUG(5, "Error:: in msgPUBREC #%d - %s", + EVERBOSE(5, "Error:: in msgPUBREC #%d - %s", mqtt_GetErrno(), mqtt_GetError()); return 0; } @@ -251,15 +300,15 @@ cmdPUBREL(void *srv, int len, void *arg) u_short mid = 0; ait_val_t *p = NULL; - ioTRACE(2); + ETRACE(2); if (!sess) return -1; - ioDEBUG(5, "Exec PUBREL session"); + EVERBOSE(5, "Exec PUBREL session"); mid = mqtt_readPUBREL(sess->sess_buf); if (mid == (u_short) -1) { - ioDEBUG(5, "Error:: in readPUBREL #%d - %s", mqtt_GetErrno(), mqtt_GetError()); + EVERBOSE(5, "Error:: in readPUBREL #%d - %s", mqtt_GetErrno(), mqtt_GetError()); return 0; } @@ -269,7 +318,7 @@ cmdPUBREL(void *srv, int len, void *arg) siz = mqtt_msgPUBCOMP(sess->sess_buf, mid); if (siz == -1) { - ioDEBUG(5, "Error:: in msgPUBCOMP #%d - %s", mqtt_GetErrno(), mqtt_GetError()); + EVERBOSE(5, "Error:: in msgPUBCOMP #%d - %s", mqtt_GetErrno(), mqtt_GetError()); return 0; } @@ -292,68 +341,65 @@ cmdSUBSCRIBE(void *srv, int len, void *arg) void *ptr; ait_val_t *p = NULL; - ioTRACE(2); + ETRACE(2); if (!sess) return -1; - ioDEBUG(5, "Exec SUBSCRIBE session"); + EVERBOSE(5, "Exec SUBSCRIBE session"); siz = mqtt_readSUBSCRIBE(sess->sess_buf, &mid, &subs); if (siz == -1) { - ioDEBUG(5, "Error:: in readSUBSCRIBE #%d - %s", mqtt_GetErrno(), mqtt_GetError()); + EVERBOSE(5, "Error:: in readSUBSCRIBE #%d - %s", mqtt_GetErrno(), mqtt_GetError()); return 0; } /* add to db */ - for (i = 0; i < siz; i++, subs[i].sub_ret = MQTT_QOS_DENY) { - /* convert topic to sql search statement */ - if (mqtt_sqlTopic(subs[i].sub_topic.msg_base, buf, sizeof buf) == -1) { - ioDEBUG(5, "Error:: in db #%d - %s", mqtt_GetErrno(), mqtt_GetError()); + for (i = 0; i < siz; i++) { + store = e_malloc(sizeof(struct tagStore)); + if (!store) { + ELIBERR(elwix); continue; + } else { + store->st_msgid = mid; + mqtt_subCopy(&store->st_subscr, &subs[i]); + subs[i].sub_ret = MQTT_QOS_DENY; } - if (call.WritePUB_subscribe(&cfg, pub, sess->sess_cid, mid, buf, - sess->sess_user, sess->sess_addr, subs[i].sub_ret) > 0) { - store = io_malloc(sizeof(struct tagStore)); - if (!store) { - ioSYSERR(0); + + /* add to cache */ + SLIST_INSERT_HEAD(&sess->sess_subscr, store, st_node); + + /* convert topic to regexp */ + if (mqtt_expandTopic(subs[i].sub_topic.msg_base, buf, sizeof buf, 1, 1) == -1) { + EVERBOSE(5, "Error:: in regexp #%d - %s", mqtt_GetErrno(), mqtt_GetError()); + } else { + ptr = realloc(store->st_subscr.sub_topic.msg_base, strlen(buf) + 1); + if (!ptr) { + ESYSERR(0); continue; } else { - store->st_msgid = mid; - mqtt_subCopy(&store->st_subscr, &subs[i]); + 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); } - /* add to cache */ - SLIST_INSERT_HEAD(&sess->sess_subscr, store, st_node); + /* store to db */ + call.WritePUB_subscribe(&cfg, pub, sess->sess_cid, mid, buf, + sess->sess_user, sess->sess_addr, store->st_subscr.sub_ret); + /* subscribe pass */ + subs[i].sub_ret = MQTT_QOS_PASS; - /* 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); - 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.LOG(logg, "Added [%s] SUBSCRIBE '%s'(%d) QoS=%d from %s\n", sess->sess_cid, + store->st_subscr.sub_topic.msg_base, + store->st_subscr.sub_topic.msg_len, + store->st_subscr.sub_ret, sess->sess_addr); } } /* send acknowledge */ siz = mqtt_msgSUBACK(sess->sess_buf, subs, mid); if (siz == -1) { - ioDEBUG(5, "Error:: in msgSUBACK #%d - %s", mqtt_GetErrno(), mqtt_GetError()); + EVERBOSE(5, "Error:: in msgSUBACK #%d - %s", mqtt_GetErrno(), mqtt_GetError()); goto end; } else { p = mkPkt(sess->sess_buf->msg_base, siz); @@ -377,15 +423,15 @@ cmdUNSUBSCRIBE(void *srv, int len, void *arg) struct tagStore *store, *tmp; ait_val_t *p = NULL; - ioTRACE(2); + ETRACE(2); if (!sess) return -1; - ioDEBUG(5, "Exec UNSUBSCRIBE session"); + EVERBOSE(5, "Exec UNSUBSCRIBE session"); siz = mqtt_readUNSUBSCRIBE(sess->sess_buf, &mid, &subs); if (siz == -1) { - ioDEBUG(5, "Error:: in readUNSUBSCRIBE #%d - %s", mqtt_GetErrno(), mqtt_GetError()); + EVERBOSE(5, "Error:: in readUNSUBSCRIBE #%d - %s", mqtt_GetErrno(), mqtt_GetError()); return 0; } @@ -402,7 +448,7 @@ cmdUNSUBSCRIBE(void *srv, int len, void *arg) free(store->st_subscr.sub_topic.msg_base); if (store->st_subscr.sub_value.msg_base) free(store->st_subscr.sub_value.msg_base); - io_free(store); + e_free(store); } } @@ -413,7 +459,7 @@ cmdUNSUBSCRIBE(void *srv, int len, void *arg) /* send acknowledge */ siz = mqtt_msgUNSUBACK(sess->sess_buf, mid); if (siz == -1) { - ioDEBUG(5, "Error:: in msgUNSUBACK #%d - %s", mqtt_GetErrno(), mqtt_GetError()); + EVERBOSE(5, "Error:: in msgUNSUBACK #%d - %s", mqtt_GetErrno(), mqtt_GetError()); goto end; } else { p = mkPkt(sess->sess_buf->msg_base, siz); @@ -433,15 +479,15 @@ cmdPINGREQ(void *srv, int len, void *arg) int siz = 0; ait_val_t *p = NULL; - ioTRACE(2); + ETRACE(2); if (!sess) return -1; - ioDEBUG(5, "Exec PINGREQ session"); + EVERBOSE(5, "Exec PINGREQ session"); siz = mqtt_msgPINGRESP(sess->sess_buf); if (siz == -1) { - ioDEBUG(5, "Error:: in msgPINGRESP #%d - %s", mqtt_GetErrno(), mqtt_GetError()); + EVERBOSE(5, "Error:: in msgPINGRESP #%d - %s", mqtt_GetErrno(), mqtt_GetError()); return 0; } else { p = mkPkt(sess->sess_buf->msg_base, siz); @@ -458,14 +504,16 @@ cmdCONNECT(void *srv, int len, void *arg) struct tagStore *store; struct tagSession *sess = (struct tagSession*) arg; - ioTRACE(2); + ETRACE(2); if (!sess) return -1; - ioDEBUG(5, "Exec CONNECT session"); + EVERBOSE(5, "Exec CONNECT session"); TAILQ_REMOVE(&Sessions, sess, sess_node); + schedCancelby(root, taskTIMER, CRITERIA_CALL, sendRetain, NULL); + if (sess->sess_clean) { if (call.FiniSessPUB) call.FiniSessPUB(&cfg, pub, sess->sess_cid, sess->sess_user, "%"); @@ -483,9 +531,12 @@ cmdCONNECT(void *srv, int len, void *arg) if (store->st_subscr.sub_value.msg_base) free(store->st_subscr.sub_value.msg_base); - io_free(store); + e_free(store); } + if (sess->sess_will.flag) + pubWill(sess); + if (sess->sess_will.msg) free(sess->sess_will.msg); if (sess->sess_will.topic) @@ -502,12 +553,12 @@ cmdDISCONNECT(void *srv, int len, void *arg) { struct tagSession *sess = (struct tagSession*) arg; - ioTRACE(2); + ETRACE(2); if (!sess) return -1; - ioDEBUG(5, "Exec DISCONNECT session"); + EVERBOSE(5, "Exec DISCONNECT session"); call.LOG(logg, "Session %s stopped from %s for user %s.\n", sess->sess_cid, sess->sess_addr, sess->sess_user);