--- mqtt/src/mqttd_calls.c	2012/06/26 13:43:59	1.2.2.27
+++ mqtt/src/mqttd_calls.c	2012/06/29 15:43:13	1.2.2.32
@@ -1,5 +1,6 @@
 #include "global.h"
 #include "mqttd.h"
+#include "utils.h"
 #include "rtlm.h"
 #include "mqttd_calls.h"
 
@@ -60,14 +61,15 @@ pubOnce(struct tagSession *sess, char * __restrict psT
 {
 	ait_val_t *p = NULL;
 	struct tagSession *s = NULL;
-	struct tagStore *st = NULL;
+	struct tagStore *st_, *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) {
+		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);
 				regfree(&re);
@@ -92,10 +94,10 @@ pubAck(struct tagSession *sess, u_short mid, char * __
 {
 	ait_val_t *p = NULL;
 	struct tagSession *s = NULL;
-	struct tagStore *st = NULL;
+	struct tagStore *st_, *st = NULL;
 	regex_t re;
 	regmatch_t match;
-	int ret, flg = 0;
+	int ret;
 	char szStr[STRSIZ];
 	struct mqtthdr *hdr;
 
@@ -109,7 +111,11 @@ pubAck(struct tagSession *sess, u_short mid, char * __
 			sess->sess_user, sess->sess_addr, hdr->mqtt_msg.qos, hdr->mqtt_msg.retain);
 
 	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 */
+			if (st->st_subscr.sub_ret < MQTT_QOS_ACK)
+				continue;
+
 			if ((ret = regcomp(&re, st->st_subscr.sub_topic.msg_base, REG_EXTENDED))) {
 				regerror(ret, &re, szStr, sizeof szStr);
 				regfree(&re);
@@ -117,26 +123,67 @@ pubAck(struct tagSession *sess, u_short mid, char * __
 						st->st_subscr.sub_topic.msg_base, szStr);
 			}
 			if (!regexec(&re, psTopic, 1, &match, 0)) {
-				flg = 1;	/* MATCH */
-				schedWrite(root, sendPacket, p, s->sess_sock, NULL, 0);
+				/* MATCH */
+				schedWrite(root, sendPacket, mkPkt(AIT_GET_BUF(p), AIT_LEN(p)), 
+						s->sess_sock, NULL, 0);
 			}
 
 			regfree(&re);
 		}
 	}
 
-	/* if subscriber not found then free packet */
-	if (!flg)
-		freePkt(&p);
-
+	/* 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;
+
+	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);
+
+	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;
 }
 
@@ -217,17 +264,18 @@ cmdPUBREL(void *srv, int len, void *arg)
 		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);
 	if (siz == -1) {
 		ioDEBUG(5, "Error:: in msgPUBCOMP #%d - %s", mqtt_GetErrno(), mqtt_GetError());
 		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);
 	return 0;
 }
@@ -259,48 +307,44 @@ cmdSUBSCRIBE(void *srv, int len, void *arg)
 
 	/* 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());
+		store = io_malloc(sizeof(struct tagStore));
+		if (!store) {
+			ioSYSERR(0);
 			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) {
-			store = io_malloc(sizeof(struct tagStore));
-			if (!store) {
+
+		/* 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) {
+			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);
 				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);
+			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);
 
-			/* 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;
-			}
+			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 */
@@ -438,6 +482,9 @@ cmdCONNECT(void *srv, int len, void *arg)
 
 		io_free(store);
 	}
+
+	if (sess->sess_will.flag)
+		srv_Will(sess);
 
 	if (sess->sess_will.msg)
 		free(sess->sess_will.msg);