|
version 1.1.2.21, 2011/12/15 12:57:40
|
version 1.1.2.24, 2011/12/16 12:16:17
|
|
Line 1
|
Line 1
|
| #include "global.h" |
#include "global.h" |
| #include "rtlm.h" |
#include "rtlm.h" |
| |
#include "utils.h" |
| #include "mqttd.h" |
#include "mqttd.h" |
| |
|
| |
|
|
Line 64 finiSession(struct tagSession *sess)
|
Line 65 finiSession(struct tagSession *sess)
|
| if (sess->sess_will.topic) |
if (sess->sess_will.topic) |
| free(sess->sess_will.topic); |
free(sess->sess_will.topic); |
| |
|
| if (sess->sess_sock > STDERR_FILENO) { | if (sess->sess_sock > STDERR_FILENO) |
| shutdown(sess->sess_sock, SHUT_RDWR); | srv_Close(sess->sess_sock); |
| close(sess->sess_sock); | |
| } | |
| |
|
| mqtt_msgFree(&sess->sess_buf, 42); |
mqtt_msgFree(&sess->sess_buf, 42); |
| |
|
| free(sess); |
free(sess); |
| } |
} |
| |
|
| static void * | static void |
| thrSession(struct tagSession *sess) | stopSession(struct tagSession *sess) |
| { |
{ |
| void thrClean(struct tagSession *sess) | mqtt_msg_t msg = { NULL, 0 }; |
| { | int ret; |
| pthread_mutex_lock(&mtx_sess); | |
| TAILQ_REMOVE(&Sessions, sess, sess_node); | pthread_mutex_lock(&mtx_sess); |
| pthread_mutex_unlock(&mtx_sess); | TAILQ_REMOVE(&Sessions, sess, sess_node); |
| ioDEBUG(1, "Close socket=%d", sess->sess_sock); | pthread_mutex_unlock(&mtx_sess); |
| finiSession(sess); | |
| | ret = mqtt_msgDISCONNECT(&msg); |
| | if ((ret = send(sess->sess_sock, msg.msg_base, ret, 0)) == -1) |
| | ioDEBUG(3, "Error:: send(%d) #%d - %s", sess->sess_sock, errno, strerror(errno)); |
| | else { |
| | ioDEBUG(5, "Sended %d bytes for disconnect", ret); |
| | free(msg.msg_base); |
| | msg.msg_len = 0; |
| } |
} |
| |
|
| pthread_cleanup_push((void(*)(void*)) thrClean, sess); | ioDEBUG(1, "Close socket=%d", sess->sess_sock); |
| | finiSession(sess); |
| | } |
| |
|
| |
static void * |
| |
thrSession(struct tagSession *sess) |
| |
{ |
| |
pthread_cleanup_push((void(*)(void*)) stopSession, sess); |
| |
|
| ioTRACE(2); |
ioTRACE(2); |
| |
|
| pthread_cleanup_pop(42); |
pthread_cleanup_pop(42); |
|
Line 101 startSession(sched_task_t *task)
|
Line 114 startSession(sched_task_t *task)
|
| mqtt_msg_t msg = { NULL, 0 }, buf = { basebuf, sizeof basebuf }; |
mqtt_msg_t msg = { NULL, 0 }, buf = { basebuf, sizeof basebuf }; |
| mqtthdr_connflgs_t flg; |
mqtthdr_connflgs_t flg; |
| mqtthdr_connack_t cack; |
mqtthdr_connack_t cack; |
| struct tagSession *sess = NULL; | struct tagSession *s, *sess = NULL; |
| int ret; |
int ret; |
| pthread_attr_t attr; |
pthread_attr_t attr; |
| |
|
|
Line 142 startSession(sched_task_t *task)
|
Line 155 startSession(sched_task_t *task)
|
| ret = MQTT_RETCODE_DENIED; |
ret = MQTT_RETCODE_DENIED; |
| goto end; |
goto end; |
| } else { |
} else { |
| ioDEBUG(0, "Login:: ALLOWED for username %s ...", sess->sess_user); | ioDEBUG(1, "Login:: ALLOWED for username %s ...", sess->sess_user); |
| ret = MQTT_RETCODE_ACCEPTED; |
ret = MQTT_RETCODE_ACCEPTED; |
| } |
} |
| |
if (call.FiniSessPUB(&cfg, pub, sess->sess_cid, sess->sess_user, "%") > 0) { |
| |
ioDEBUG(2, "Old session %s should be disconnect!", sess->sess_cid); |
| |
TAILQ_FOREACH(s, &Sessions, sess_node) |
| |
if (!strcmp(s->sess_cid, sess->sess_cid)) { |
| |
stopSession(s); |
| |
break; |
| |
} |
| |
} |
| |
if (call.InitSessPUB(&cfg, pub, sess->sess_cid, sess->sess_user, sess->sess_addr, |
| |
sess->sess_will.flag, sess->sess_will.topic, sess->sess_will.msg, |
| |
sess->sess_will.qos, sess->sess_will.retain) == -1) { |
| |
ioDEBUG(0, "Session %s DENIED for username %s", sess->sess_cid, sess->sess_user); |
| |
ret = MQTT_RETCODE_DENIED; |
| |
goto end; |
| |
} else { |
| |
ioDEBUG(0, "Session %s from %s and username %s is started", |
| |
sess->sess_cid, sess->sess_addr, sess->sess_user); |
| |
ret = MQTT_RETCODE_ACCEPTED; |
| |
} |
| |
|
| ret = mqtt_msgCONNACK(&msg, ret); |
ret = mqtt_msgCONNACK(&msg, ret); |
| if ((ret = send(sess->sess_sock, msg.msg_base, ret, 0)) == -1) { |
if ((ret = send(sess->sess_sock, msg.msg_base, ret, 0)) == -1) { |
|
Line 166 startSession(sched_task_t *task)
|
Line 198 startSession(sched_task_t *task)
|
| pthread_create(&sess->sess_tid, &attr, (void*(*)(void*)) thrSession, sess); |
pthread_create(&sess->sess_tid, &attr, (void*(*)(void*)) thrSession, sess); |
| pthread_mutex_unlock(&mtx_sess); |
pthread_mutex_unlock(&mtx_sess); |
| |
|
| ioDEBUG(1, "ConnID=%s(%s) for %s login OK!", sess->sess_cid, sess->sess_user, sess->sess_addr); | call.LOG(logg, "Session %s started from %s for user %s OK!\n", sess->sess_cid, |
| | sess->sess_addr, sess->sess_user); |
| |
|
| pthread_attr_destroy(&attr); |
pthread_attr_destroy(&attr); |
| return NULL; |
return NULL; |
|
Line 210 Run(int sock)
|
Line 243 Run(int sock)
|
| int cli; |
int cli; |
| pthread_t tid; |
pthread_t tid; |
| ait_val_t *v; |
ait_val_t *v; |
| |
char str[STRSIZ]; |
| |
|
| ioTRACE(1); |
ioTRACE(1); |
| |
|
|
Line 220 Run(int sock)
|
Line 254 Run(int sock)
|
| pthread_detach(tid); |
pthread_detach(tid); |
| ioDEBUG(2, "Run scheduler management thread"); |
ioDEBUG(2, "Run scheduler management thread"); |
| |
|
| if (listen(sock, 5 /*SOMAXCONN*/) == -1) { | if (listen(sock, SOMAXCONN) == -1) { |
| ioLOGERR(0); |
ioLOGERR(0); |
| return -1; |
return -1; |
| } |
} |
| |
|
| while (!Kill) { |
while (!Kill) { |
| printf("%%%%%%%%%%rrrrrrrrrrrrrrrrrrrrr errno=%d\n", errno); |
|
| if ((cli = accept(sock, &sa.sa, &sslen)) == -1) { |
if ((cli = accept(sock, &sa.sa, &sslen)) == -1) { |
| printf("rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr errno=%d\n", errno); |
|
| if (!Kill) |
if (!Kill) |
| continue; |
continue; |
| ioLOGERR(0); |
ioLOGERR(0); |
|
Line 238 Run(int sock)
|
Line 270 Run(int sock)
|
| if (!v) { |
if (!v) { |
| ioLIBERR(mqtt); |
ioLIBERR(mqtt); |
| break; |
break; |
| } else | } else { |
| io_n2addr(&sa, v); | memset(str, 0, sizeof str); |
| ioDEBUG(1, "Connected client with socket=%d from %s:%d", cli, | snprintf(str, sizeof str, "%s:%hu", io_n2addr(&sa, v), io_n2port(&sa)); |
| AIT_GET_STR(v), io_n2port(&sa)); | AIT_SET_STR(v, str); |
| | } |
| | ioDEBUG(1, "Connected client with socket=%d from %s", cli, AIT_GET_STR(v)); |
| |
|
| if (!schedRead(root, startSession, v, cli)) { |
if (!schedRead(root, startSession, v, cli)) { |
| io_freeVar(v); |
io_freeVar(v); |