|
|
| 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", |