|
version 1.1.2.1, 2012/01/25 10:34:14
|
version 1.2.2.13, 2012/05/08 13:04:02
|
|
Line 1
|
Line 1
|
| #include "global.h" |
#include "global.h" |
| #include "mqttd.h" |
#include "mqttd.h" |
| |
#include "rtlm.h" |
| #include "mqttd_calls.h" |
#include "mqttd_calls.h" |
| |
|
| |
|
| int |
int |
| Publish(struct tagSession *sess) | cmdPUBLISH(void *srv, int len, void *arg) |
| { |
{ |
| struct mqtthdr *hdr; |
struct mqtthdr *hdr; |
| |
struct tagSession *sess = (struct tagSession*) arg; |
| |
|
| ioTRACE(2); |
ioTRACE(2); |
| |
|
|
Line 28 Publish(struct tagSession *sess)
|
Line 30 Publish(struct tagSession *sess)
|
| } |
} |
| |
|
| return 0; |
return 0; |
| |
} |
| |
|
| |
int |
| |
cmdPUBREL(void *srv, int len, void *arg) |
| |
{ |
| |
struct mqtthdr *hdr; |
| |
struct tagSession *sess = (struct tagSession*) arg; |
| |
|
| |
ioTRACE(2); |
| |
|
| |
if (!sess) |
| |
return -1; |
| |
|
| |
hdr = (struct mqtthdr*) sess->sess_buf->msg_base; |
| |
|
| |
return 0; |
| |
} |
| |
|
| |
int |
| |
cmdSUBSCRIBE(void *srv, int len, void *arg) |
| |
{ |
| |
struct tagSession *sess = (struct tagSession*) arg; |
| |
mqtt_subscr_t *subs = NULL; |
| |
int siz = 0; |
| |
u_short mid = 0; |
| |
register int i; |
| |
struct tagStore *store; |
| |
|
| |
ioTRACE(2); |
| |
|
| |
if (!sess) |
| |
return -1; |
| |
|
| |
ioDEBUG(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()); |
| |
return 0; |
| |
} |
| |
|
| |
/* add to db */ |
| |
for (i = 0; i < siz; i++) { |
| |
if (call.WritePUB_subscribe(&cfg, pub, mid, subs[i].sub_topic.msg_base, |
| |
sess->sess_user, sess->sess_addr, subs[i].sub_ret) > 0) { |
| |
store = malloc(sizeof(struct tagStore)); |
| |
if (!store) { |
| |
ioSYSERR(0); |
| |
goto end; |
| |
} else { |
| |
store->st_msgid = mid; |
| |
mqtt_subCopy(&store->st_subscr, &subs[i]); |
| |
} |
| |
|
| |
/* add to cache */ |
| |
SLIST_INSERT_HEAD(&sess->sess_subscr, store, st_node); |
| |
|
| |
subs[i].sub_ret = MQTT_QOS_PASS; |
| |
} else |
| |
subs[i].sub_ret = MQTT_QOS_DENY; |
| |
} |
| |
|
| |
/* send acknowledge */ |
| |
siz = mqtt_msgSUBACK(sess->sess_buf, subs, mid); |
| |
if (siz == -1) { |
| |
ioDEBUG(5, "Error:: in msgSUBACK #%d - %s", mqtt_GetErrno(), mqtt_GetError()); |
| |
goto end; |
| |
} |
| |
if ((siz = send(sess->sess_sock, sess->sess_buf->msg_base, siz, MSG_NOSIGNAL)) == -1) |
| |
ioSYSERR(0); |
| |
else { |
| |
ioDEBUG(5, "Sended %d bytes.", siz); |
| |
memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len); |
| |
} |
| |
end: |
| |
mqtt_subFree(&subs); |
| |
return 0; |
| |
} |
| |
|
| |
int |
| |
cmdUNSUBSCRIBE(void *srv, int len, void *arg) |
| |
{ |
| |
struct tagSession *sess = (struct tagSession*) arg; |
| |
mqtt_subscr_t *subs = NULL; |
| |
int siz = 0; |
| |
u_short mid = 0; |
| |
register int i; |
| |
struct tagStore *store, *tmp; |
| |
|
| |
ioTRACE(2); |
| |
|
| |
if (!sess) |
| |
return -1; |
| |
|
| |
ioDEBUG(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()); |
| |
return 0; |
| |
} |
| |
|
| |
/* del from db */ |
| |
for (i = 0; i < siz; i++) { |
| |
SLIST_FOREACH_SAFE(store, &sess->sess_subscr, st_node, tmp) { |
| |
if (store->st_subscr.sub_ret == subs[i].sub_ret && |
| |
store->st_subscr.sub_topic.msg_base && |
| |
!strcmp(store->st_subscr.sub_topic.msg_base, |
| |
subs[i].sub_topic.msg_base)) { |
| |
SLIST_REMOVE(&sess->sess_subscr, store, tagStore, st_node); |
| |
free(store); |
| |
} |
| |
} |
| |
|
| |
call.DeletePUB_subscribe(&cfg, pub, subs[i].sub_topic.msg_base, |
| |
sess->sess_user, sess->sess_addr); |
| |
} |
| |
|
| |
/* send acknowledge */ |
| |
siz = mqtt_msgUNSUBACK(sess->sess_buf, mid); |
| |
if (siz == -1) { |
| |
ioDEBUG(5, "Error:: in msgUNSUBACK #%d - %s", mqtt_GetErrno(), mqtt_GetError()); |
| |
goto end; |
| |
} |
| |
if ((siz = send(sess->sess_sock, sess->sess_buf->msg_base, siz, MSG_NOSIGNAL)) == -1) |
| |
ioSYSERR(0); |
| |
else { |
| |
ioDEBUG(5, "Sended %d bytes.", siz); |
| |
memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len); |
| |
} |
| |
end: |
| |
mqtt_subFree(&subs); |
| |
return 0; |
| |
} |
| |
|
| |
int |
| |
cmdPINGREQ(void *srv, int len, void *arg) |
| |
{ |
| |
struct tagSession *sess = (struct tagSession*) arg; |
| |
int siz = 0; |
| |
|
| |
ioTRACE(2); |
| |
|
| |
if (!sess) |
| |
return -1; |
| |
|
| |
ioDEBUG(5, "Exec PINGREQ session"); |
| |
siz = mqtt_msgPINGRESP(sess->sess_buf); |
| |
if (siz == -1) { |
| |
ioDEBUG(5, "Error:: in msgPINGRESP #%d - %s", mqtt_GetErrno(), mqtt_GetError()); |
| |
return 0; |
| |
} |
| |
if ((siz = send(sess->sess_sock, sess->sess_buf->msg_base, siz, MSG_NOSIGNAL)) == -1) { |
| |
ioSYSERR(0); |
| |
return 0; |
| |
} else { |
| |
ioDEBUG(5, "Sended %d bytes.", siz); |
| |
memset(sess->sess_buf->msg_base, 0, sess->sess_buf->msg_len); |
| |
} |
| |
|
| |
return 0; |
| |
} |
| |
|
| |
int |
| |
cmdCONNECT(void *srv, int len, void *arg) |
| |
{ |
| |
struct tagStore *store; |
| |
struct tagSession *sess = (struct tagSession*) arg; |
| |
|
| |
ioTRACE(2); |
| |
|
| |
if (!sess) |
| |
return -1; |
| |
|
| |
ioDEBUG(5, "Exec CONNECT session"); |
| |
TAILQ_REMOVE(&Sessions, sess, sess_node); |
| |
|
| |
if (call.FiniSessPUB) |
| |
call.FiniSessPUB(&cfg, pub, sess->sess_cid, sess->sess_user, "%"); |
| |
|
| |
while ((store = SLIST_FIRST(&sess->sess_subscr))) { |
| |
SLIST_REMOVE_HEAD(&sess->sess_subscr, st_node); |
| |
|
| |
if (store->st_subscr.sub_topic.msg_base) |
| |
free(store->st_subscr.sub_topic.msg_base); |
| |
if (store->st_subscr.sub_value.msg_base) |
| |
free(store->st_subscr.sub_value.msg_base); |
| |
|
| |
free(store); |
| |
} |
| |
|
| |
if (sess->sess_will.msg) |
| |
free(sess->sess_will.msg); |
| |
if (sess->sess_will.topic) |
| |
free(sess->sess_will.topic); |
| |
|
| |
call.LOG(logg, "Session %s stopped from %s for user %s.\n", sess->sess_cid, |
| |
sess->sess_addr, sess->sess_user); |
| |
|
| |
return -3; /* reconnect client */ |
| |
} |
| |
|
| |
int |
| |
cmdDISCONNECT(void *srv, int len, void *arg) |
| |
{ |
| |
struct tagSession *sess = (struct tagSession*) arg; |
| |
|
| |
ioTRACE(2); |
| |
|
| |
if (!sess) |
| |
return -1; |
| |
|
| |
ioDEBUG(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); |
| |
|
| |
return -2; /* must terminate dispatcher */ |
| } |
} |