| version 1.1.2.31, 2012/01/24 16:28:28 | version 1.2.2.4, 2012/01/30 13:39:22 | 
| Line 2 | Line 2 | 
 | #include "rtlm.h" | #include "rtlm.h" | 
 | #include "utils.h" | #include "utils.h" | 
 | #include "mqttd.h" | #include "mqttd.h" | 
 |  | #include "mqttd_calls.h" | 
 |  |  | 
 |  |  | 
 | static void *startSession(sched_task_t *task); | static void *startSession(sched_task_t *task); | 
| Line 26  initSession(int sock, ait_val_t * __restrict v) | Line 27  initSession(int sock, ait_val_t * __restrict v) | 
 | } else | } else | 
 | memset(sess, 0, sizeof(struct tagSession)); | memset(sess, 0, sizeof(struct tagSession)); | 
 |  |  | 
 |  | pthread_mutex_init(&sess->sess_mtx, NULL); | 
 |  |  | 
 |  | TAILQ_INIT(&sess->sess_sndqueue); | 
 |  |  | 
 | str = (const char*) cfg_GetAttribute(&cfg, CFG("mqttd"), CFG("retry")); | str = (const char*) cfg_GetAttribute(&cfg, CFG("mqttd"), CFG("retry")); | 
 | if (!str) | if (!str) | 
 | sess->sess_retry = DEFAULT_RETRY; | sess->sess_retry = DEFAULT_RETRY; | 
| Line 40  initSession(int sock, ait_val_t * __restrict v) | Line 45  initSession(int sock, ait_val_t * __restrict v) | 
 | return NULL; | return NULL; | 
 | } | } | 
 |  |  | 
 |  | sess->sess_srv = mqtt_srv_Init(sock, sess->sess_buf); | 
 |  | if (!sess->sess_srv) { | 
 |  | ioDEBUG(3, "Error:: in srv_Init #%d - %s", mqtt_GetErrno(), mqtt_GetError()); | 
 |  | mqtt_msgFree(&sess->sess_buf, 42); | 
 |  | free(sess); | 
 |  | io_freeVar(v); | 
 |  | return NULL; | 
 |  | } else { | 
 |  | mqtt_srv_setCmd(sess->sess_srv, MQTT_TYPE_CONNECT, cmdCONNECT); | 
 |  | mqtt_srv_setCmd(sess->sess_srv, MQTT_TYPE_PUBLISH, cmdPUBLISH); | 
 |  | mqtt_srv_setCmd(sess->sess_srv, MQTT_TYPE_PUBREL, cmdPUBREL); | 
 |  | mqtt_srv_setCmd(sess->sess_srv, MQTT_TYPE_SUBSCRIBE, cmdSUBSCRIBE); | 
 |  | mqtt_srv_setCmd(sess->sess_srv, MQTT_TYPE_UNSUBSCRIBE, cmdUNSUBSCRIBE); | 
 |  | mqtt_srv_setCmd(sess->sess_srv, MQTT_TYPE_PINGREQ, cmdPINGREQ); | 
 |  | mqtt_srv_setCmd(sess->sess_srv, MQTT_TYPE_DISCONNECT, cmdDISCONNECT); | 
 |  | } | 
 |  |  | 
 | sess->sess_sock = sock; | sess->sess_sock = sock; | 
 | strlcpy(sess->sess_addr, (char*) AIT_GET_STR(v), sizeof sess->sess_addr); | strlcpy(sess->sess_addr, (char*) AIT_GET_STR(v), sizeof sess->sess_addr); | 
 | io_freeVar(v); | io_freeVar(v); | 
| Line 59  finiSession(struct tagSession *sess, int preservSock) | Line 81  finiSession(struct tagSession *sess, int preservSock) | 
 | if (call.FiniSessPUB) | if (call.FiniSessPUB) | 
 | call.FiniSessPUB(&cfg, pub, sess->sess_cid, sess->sess_user, "%"); | call.FiniSessPUB(&cfg, pub, sess->sess_cid, sess->sess_user, "%"); | 
 |  |  | 
| while ((store = SLIST_FIRST(&sess->sess_sndqueue))) { | pthread_mutex_lock(&sess->sess_mtx); | 
| SLIST_REMOVE_HEAD(&sess->sess_sndqueue, st_node); | while ((store = TAILQ_FIRST(&sess->sess_sndqueue))) { | 
|  | TAILQ_REMOVE(&sess->sess_sndqueue, store, 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); | free(store); | 
 | } | } | 
 |  | pthread_mutex_unlock(&sess->sess_mtx); | 
 |  | pthread_mutex_destroy(&sess->sess_mtx); | 
 |  |  | 
 | if (sess->sess_will.msg) | if (sess->sess_will.msg) | 
 | free(sess->sess_will.msg); | free(sess->sess_will.msg); | 
| Line 72  finiSession(struct tagSession *sess, int preservSock) | Line 103  finiSession(struct tagSession *sess, int preservSock) | 
 | if (sess->sess_sock > STDERR_FILENO && !preservSock) | if (sess->sess_sock > STDERR_FILENO && !preservSock) | 
 | srv_Close(sess->sess_sock); | srv_Close(sess->sess_sock); | 
 |  |  | 
 |  | mqtt_srv_Fini(&sess->sess_srv); | 
 | mqtt_msgFree(&sess->sess_buf, 42); | mqtt_msgFree(&sess->sess_buf, 42); | 
 |  |  | 
 | free(sess); | free(sess); | 
| Line 156  KASession(struct tagSession *sess) | Line 188  KASession(struct tagSession *sess) | 
 | static void * | static void * | 
 | thrSession(struct tagSession *sess) | thrSession(struct tagSession *sess) | 
 | { | { | 
 | mqtt_msg_t msg = { NULL, 0 }; |  | 
 | int ret, locKill = 42; | int ret, locKill = 42; | 
 | struct pollfd pfd; | struct pollfd pfd; | 
 | struct mqtthdr *hdr; | struct mqtthdr *hdr; | 
 | ait_val_t *v; | ait_val_t *v; | 
 | struct tagStore *store; |  | 
 |  |  | 
 | pthread_cleanup_push((void(*)(void*)) stopSession, sess); | pthread_cleanup_push((void(*)(void*)) stopSession, sess); | 
 | ioTRACE(2); | ioTRACE(2); | 
| Line 189  thrSession(struct tagSession *sess) | Line 219  thrSession(struct tagSession *sess) | 
 | hdr = (struct mqtthdr*) sess->sess_buf->msg_base; | hdr = (struct mqtthdr*) sess->sess_buf->msg_base; | 
 |  |  | 
 | /* dispatch message type */ | /* dispatch message type */ | 
 |  | if (mqtt_srv_Dispatch(sess->sess_srv, sess)) | 
 |  | ioLIBERR(mqtt); | 
 | switch (hdr->mqtt_msg.type) { | switch (hdr->mqtt_msg.type) { | 
 | case MQTT_TYPE_CONNECT: | case MQTT_TYPE_CONNECT: | 
 | ioDEBUG(5, "Exec CONNECT session"); | ioDEBUG(5, "Exec CONNECT session"); | 
 | pthread_mutex_lock(&mtx_sess); |  | 
 | TAILQ_REMOVE(&Sessions, sess, sess_node); |  | 
 | pthread_mutex_unlock(&mtx_sess); |  | 
 |  |  | 
 | if (call.FiniSessPUB) |  | 
 | call.FiniSessPUB(&cfg, pub, sess->sess_cid, sess->sess_user, "%"); |  | 
 |  |  | 
 | while ((store = SLIST_FIRST(&sess->sess_sndqueue))) { |  | 
 | SLIST_REMOVE_HEAD(&sess->sess_sndqueue, st_node); |  | 
 | free(store); |  | 
 | } |  | 
 |  |  | 
 | if (sess->sess_will.msg) |  | 
 | free(sess->sess_will.msg); |  | 
 | if (sess->sess_will.topic) |  | 
 | free(sess->sess_will.topic); |  | 
 |  |  | 
 | if ((v = io_allocVar())) { | if ((v = io_allocVar())) { | 
 | AIT_SET_STR(v, sess->sess_addr); | AIT_SET_STR(v, sess->sess_addr); | 
 | if (!schedEvent(root, startSession, v, (u_long) sess->sess_sock, sess, ret)) | if (!schedEvent(root, startSession, v, (u_long) sess->sess_sock, sess, ret)) | 
 | io_freeVar(v); | io_freeVar(v); | 
 | } else | } else | 
 | ioLIBERR(mqtt); | ioLIBERR(mqtt); | 
 |  |  | 
 | locKill ^= locKill; | locKill ^= locKill; | 
 |  |  | 
 | call.LOG(logg, "Session %s stopped from %s for user %s.\n", sess->sess_cid, |  | 
 | sess->sess_addr, sess->sess_user); |  | 
 | continue; | continue; | 
 | case MQTT_TYPE_DISCONNECT: | case MQTT_TYPE_DISCONNECT: | 
 | ioDEBUG(5, "Exec DISCONNECT session"); | ioDEBUG(5, "Exec DISCONNECT session"); | 
 | pthread_mutex_lock(&mtx_sess); |  | 
 | TAILQ_REMOVE(&Sessions, sess, sess_node); |  | 
 | pthread_mutex_unlock(&mtx_sess); |  | 
 |  |  | 
 | finiSession(sess, 0); | finiSession(sess, 0); | 
 | locKill ^= locKill; | locKill ^= locKill; | 
 |  |  | 
 | call.LOG(logg, "Session %s stopped from %s for user %s.\n", sess->sess_cid, |  | 
 | sess->sess_addr, sess->sess_user); |  | 
 | continue; | continue; | 
 | case MQTT_TYPE_PUBLISH: | case MQTT_TYPE_PUBLISH: | 
| ioDEBUG(5, "Work in progress ..."); | ioDEBUG(5, "Exec PUBLISH topic QoS=%d", hdr->mqtt_msg.qos); | 
|  | /* | 
|  | if (cmdPUBLISH(sess)) | 
|  | locKill ^= locKill; | 
|  | */ | 
 | break; | break; | 
 | case MQTT_TYPE_PUBREL: | case MQTT_TYPE_PUBREL: | 
 | break; | break; | 
| Line 243  thrSession(struct tagSession *sess) | Line 251  thrSession(struct tagSession *sess) | 
 | case MQTT_TYPE_UNSUBSCRIBE: | case MQTT_TYPE_UNSUBSCRIBE: | 
 | break; | break; | 
 | case MQTT_TYPE_PINGREQ: | case MQTT_TYPE_PINGREQ: | 
 |  | ioDEBUG(5, "Exec PINGREQ session"); | 
 | break; | break; | 
 | default: | default: | 
 | ioDEBUG(5, "Error:: Session %s, wrong command %d - DISCARDED", | ioDEBUG(5, "Error:: Session %s, wrong command %d - DISCARDED", |