version 1.2.2.4, 2012/01/30 13:39:22
|
version 1.2.2.24, 2012/05/05 13:39:27
|
Line 5
|
Line 5
|
#include "mqttd_calls.h" |
#include "mqttd_calls.h" |
|
|
|
|
static void *startSession(sched_task_t *task); |
|
|
|
|
|
static inline struct tagSession * |
static inline struct tagSession * |
initSession(int sock, ait_val_t * __restrict v) |
initSession(int sock, ait_val_t * __restrict v) |
{ |
{ |
Line 21 initSession(int sock, ait_val_t * __restrict v)
|
Line 18 initSession(int sock, ait_val_t * __restrict v)
|
|
|
sess = malloc(sizeof(struct tagSession)); |
sess = malloc(sizeof(struct tagSession)); |
if (!sess) { |
if (!sess) { |
ioDEBUG(3, "Error:: in malloc #%d - %s", errno, strerror(errno)); | ioSYSERR(0); |
io_freeVar(v); | |
return NULL; |
return NULL; |
} else |
} else |
memset(sess, 0, sizeof(struct tagSession)); |
memset(sess, 0, sizeof(struct tagSession)); |
|
|
pthread_mutex_init(&sess->sess_mtx, NULL); |
pthread_mutex_init(&sess->sess_mtx, NULL); |
|
|
TAILQ_INIT(&sess->sess_sndqueue); | SLIST_INIT(&sess->sess_subscr); |
|
|
str = (const char*) cfg_GetAttribute(&cfg, CFG("mqttd"), CFG("retry")); | str = cfg_getAttribute(&cfg, "mqttd", "retry"); |
if (!str) |
if (!str) |
sess->sess_retry = DEFAULT_RETRY; |
sess->sess_retry = DEFAULT_RETRY; |
else |
else |
Line 39 initSession(int sock, ait_val_t * __restrict v)
|
Line 35 initSession(int sock, ait_val_t * __restrict v)
|
|
|
sess->sess_buf = mqtt_msgAlloc(USHRT_MAX); |
sess->sess_buf = mqtt_msgAlloc(USHRT_MAX); |
if (!sess->sess_buf) { |
if (!sess->sess_buf) { |
ioDEBUG(3, "Error:: in msgAlloc #%d - %s", mqtt_GetErrno(), mqtt_GetError()); | ioLIBERR(mqtt); |
free(sess); |
free(sess); |
io_freeVar(v); |
|
return NULL; |
return NULL; |
} |
} |
|
|
|
/* init server actor */ |
sess->sess_srv = mqtt_srv_Init(sock, sess->sess_buf); |
sess->sess_srv = mqtt_srv_Init(sock, sess->sess_buf); |
if (!sess->sess_srv) { |
if (!sess->sess_srv) { |
ioDEBUG(3, "Error:: in srv_Init #%d - %s", mqtt_GetErrno(), mqtt_GetError()); |
ioDEBUG(3, "Error:: in srv_Init #%d - %s", mqtt_GetErrno(), mqtt_GetError()); |
mqtt_msgFree(&sess->sess_buf, 42); |
mqtt_msgFree(&sess->sess_buf, 42); |
free(sess); |
free(sess); |
io_freeVar(v); |
|
return NULL; |
return NULL; |
} else { |
} else { |
mqtt_srv_setCmd(sess->sess_srv, MQTT_TYPE_CONNECT, cmdCONNECT); |
mqtt_srv_setCmd(sess->sess_srv, MQTT_TYPE_CONNECT, cmdCONNECT); |
Line 64 initSession(int sock, ait_val_t * __restrict v)
|
Line 59 initSession(int sock, ait_val_t * __restrict v)
|
|
|
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); |
|
return sess; |
return sess; |
} |
} |
|
|
static void | void |
finiSession(struct tagSession *sess, int preservSock) | finiSession(struct tagSession *sess) |
{ |
{ |
struct tagStore *store; |
struct tagStore *store; |
|
|
Line 81 finiSession(struct tagSession *sess, int preservSock)
|
Line 75 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, "%"); |
|
|
pthread_mutex_lock(&sess->sess_mtx); | SESS_ELEM_LOCK(sess); |
while ((store = TAILQ_FIRST(&sess->sess_sndqueue))) { | while ((store = SLIST_FIRST(&sess->sess_subscr))) { |
TAILQ_REMOVE(&sess->sess_sndqueue, store, st_node); | SLIST_REMOVE_HEAD(&sess->sess_subscr, st_node); |
|
|
if (store->st_subscr.sub_topic.msg_base) |
if (store->st_subscr.sub_topic.msg_base) |
free(store->st_subscr.sub_topic.msg_base); |
free(store->st_subscr.sub_topic.msg_base); |
Line 92 finiSession(struct tagSession *sess, int preservSock)
|
Line 86 finiSession(struct tagSession *sess, int preservSock)
|
|
|
free(store); |
free(store); |
} |
} |
pthread_mutex_unlock(&sess->sess_mtx); | SESS_ELEM_UNLOCK(sess); |
pthread_mutex_destroy(&sess->sess_mtx); |
pthread_mutex_destroy(&sess->sess_mtx); |
|
|
if (sess->sess_will.msg) |
if (sess->sess_will.msg) |
Line 100 finiSession(struct tagSession *sess, int preservSock)
|
Line 94 finiSession(struct tagSession *sess, int preservSock)
|
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 && !preservSock) | if (sess->sess_sock > STDERR_FILENO) |
srv_Close(sess->sess_sock); |
srv_Close(sess->sess_sock); |
|
|
mqtt_srv_Fini(&sess->sess_srv); |
mqtt_srv_Fini(&sess->sess_srv); |
Line 109 finiSession(struct tagSession *sess, int preservSock)
|
Line 103 finiSession(struct tagSession *sess, int preservSock)
|
free(sess); |
free(sess); |
} |
} |
|
|
static void | static void * |
stopSession(struct tagSession *sess) | leaveClient(sched_task_t *task) |
{ |
{ |
mqtt_msg_t msg = { NULL, 0 }; | struct tagSession *sess; |
int ret; |
int ret; |
|
|
ioTRACE(4); |
ioTRACE(4); |
|
|
|
assert(task); |
|
|
|
sess = TASK_ARG(task); |
assert(sess); |
assert(sess); |
|
|
pthread_mutex_lock(&mtx_sess); | SESS_LOCK; |
TAILQ_REMOVE(&Sessions, sess, sess_node); |
TAILQ_REMOVE(&Sessions, sess, sess_node); |
pthread_mutex_unlock(&mtx_sess); | SESS_UNLOCK; |
|
|
ret = mqtt_msgDISCONNECT(&msg); | ret = mqtt_msgDISCONNECT(sess->sess_buf); |
if ((ret = send(sess->sess_sock, msg.msg_base, ret, 0)) == -1) | send(TASK_FD(task), sess->sess_buf->msg_base, ret, MSG_NOSIGNAL); |
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); | |
memset(&msg, 0, sizeof msg); | |
} | |
|
|
ioDEBUG(1, "Close socket=%d", sess->sess_sock); | ioDEBUG(1, "Close socket=%d", TASK_FD(task)); |
finiSession(sess, 0); | |
| |
call.LOG(logg, "Session %s stopped from %s for user %s.\n", sess->sess_cid, |
call.LOG(logg, "Session %s stopped from %s for user %s.\n", sess->sess_cid, |
sess->sess_addr, sess->sess_user); |
sess->sess_addr, sess->sess_user); |
|
|
|
finiSession(sess); |
|
return NULL; |
} |
} |
|
|
static int | static void * |
KASession(struct tagSession *sess) | dispatchSession(sched_task_t *task) |
{ |
{ |
mqtt_msg_t msg = { NULL, 0 }; | int ret, len = 0; |
int ret; | struct tagSession *sess; |
struct pollfd pfd; | |
|
|
ioTRACE(4); | ioTRACE(2); |
|
|
|
assert(task); |
|
|
|
sess = TASK_ARG(task); |
assert(sess); |
assert(sess); |
|
|
/* ping request */ |
|
ret = mqtt_msgPINGREQ(&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)); |
|
return -1; |
|
} else { |
|
ioDEBUG(5, "Sended %d bytes for ping request", ret); |
|
free(msg.msg_base); |
|
memset(&msg, 0, sizeof msg); |
|
} |
|
|
|
pfd.fd = sess->sess_sock; |
|
pfd.events = POLLIN | POLLPRI; |
|
if ((ret = poll(&pfd, 1, sess->sess_ka * 1000)) == -1 || |
|
pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) { |
|
ioDEBUG(3, "Error:: poll(%d) #%d - %s", sess->sess_sock, errno, strerror(errno)); |
|
return -1; |
|
} else if (!ret) { |
|
ioDEBUG(5, "Warning:: Session is abandoned ... must be disconnect!"); |
|
return 1; |
|
} |
|
/* receive & decode packet */ |
/* receive & decode packet */ |
if (recv(sess->sess_sock, sess->sess_buf->msg_base, sess->sess_buf->msg_len, 0) == -1) { | if ((ret = recv(TASK_FD(task), sess->sess_buf->msg_base, sess->sess_buf->msg_len, 0)) == -1) { |
ioDEBUG(3, "Error:: recv(%d) #%d - %s", sess->sess_sock, errno, strerror(errno)); |
ioDEBUG(3, "Error:: recv(%d) #%d - %s", sess->sess_sock, errno, strerror(errno)); |
return -1; | finiSession(sess); |
| return NULL; |
| } else if (!ret) { |
| ioDEBUG(4, "Session %s EOF received.", sess->sess_cid); |
| finiSession(sess); |
| return NULL; |
} |
} |
if (mqtt_readPINGRESP(sess->sess_buf)) { |
|
ioDEBUG(5, "Warning:: Session is broken, not hear ping response ... must be disconnect!"); |
|
return 2; |
|
} |
|
|
|
/* Keep Alive is OK! */ | do { |
return 0; | |
} | |
| |
static void * | |
thrSession(struct tagSession *sess) | |
{ | |
int ret, locKill = 42; | |
struct pollfd pfd; | |
struct mqtthdr *hdr; | |
ait_val_t *v; | |
| |
pthread_cleanup_push((void(*)(void*)) stopSession, sess); | |
ioTRACE(2); | |
| |
pfd.fd = sess->sess_sock; | |
pfd.events = POLLIN | POLLPRI; | |
while (!Kill && locKill) { | |
if ((ret = poll(&pfd, 1, sess->sess_ka * 1000)) == -1 || | |
pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) { | |
ioDEBUG(3, "Error:: poll(%d) #%d - %s", sess->sess_sock, errno, strerror(errno)); | |
break; | |
} else if (!ret && (ret = KASession(sess))) { | |
call.LOG(logg, "Session %s keep-alive missing from %s for user %s ...\n", | |
sess->sess_cid, sess->sess_addr, sess->sess_user); | |
break; | |
} | |
/* receive & decode packet */ | |
if ((ret = recv(sess->sess_sock, sess->sess_buf->msg_base, sess->sess_buf->msg_len, 0)) == -1) { | |
ioDEBUG(3, "Error:: recv(%d) #%d - %s", sess->sess_sock, errno, strerror(errno)); | |
break; | |
} else if (!ret) { | |
ioDEBUG(4, "Session %s EOF received.", sess->sess_cid); | |
break; | |
} else | |
hdr = (struct mqtthdr*) sess->sess_buf->msg_base; | |
| |
/* dispatch message type */ |
/* dispatch message type */ |
if (mqtt_srv_Dispatch(sess->sess_srv, sess)) | if ((len = mqtt_srv_Dispatch(sess->sess_srv, ret, sess)) == -1) { |
ioLIBERR(mqtt); |
ioLIBERR(mqtt); |
switch (hdr->mqtt_msg.type) { | ret = 0; |
case MQTT_TYPE_CONNECT: | } else |
ioDEBUG(5, "Exec CONNECT session"); | ret -= len; |
if ((v = io_allocVar())) { | } while (len && ret > 0); |
AIT_SET_STR(v, sess->sess_addr); | |
if (!schedEvent(root, startSession, v, (u_long) sess->sess_sock, sess, ret)) | |
io_freeVar(v); | |
} else | |
ioLIBERR(mqtt); | |
locKill ^= locKill; | |
continue; | |
case MQTT_TYPE_DISCONNECT: | |
ioDEBUG(5, "Exec DISCONNECT session"); | |
finiSession(sess, 0); | |
locKill ^= locKill; | |
continue; | |
case MQTT_TYPE_PUBLISH: | |
ioDEBUG(5, "Exec PUBLISH topic QoS=%d", hdr->mqtt_msg.qos); | |
/* | |
if (cmdPUBLISH(sess)) | |
locKill ^= locKill; | |
*/ | |
break; | |
case MQTT_TYPE_PUBREL: | |
break; | |
case MQTT_TYPE_SUBSCRIBE: | |
break; | |
case MQTT_TYPE_UNSUBSCRIBE: | |
break; | |
case MQTT_TYPE_PINGREQ: | |
ioDEBUG(5, "Exec PINGREQ session"); | |
break; | |
default: | |
ioDEBUG(5, "Error:: Session %s, wrong command %d - DISCARDED", | |
sess->sess_cid, hdr->mqtt_msg.type); | |
break; | |
} | |
} | |
|
|
pthread_cleanup_pop(locKill); | if (!schedRead(root, dispatchSession, TASK_ARG(task), TASK_FD(task), NULL, 0)) |
pthread_exit(NULL); | ioLIBERR(sched); |
| return NULL; |
} |
} |
|
|
static void * | void * |
startSession(sched_task_t *task) |
startSession(sched_task_t *task) |
{ |
{ |
u_char basebuf[USHRT_MAX]; |
u_char basebuf[USHRT_MAX]; |
mqtt_msg_t msg = { NULL, 0 }, buf = { basebuf, sizeof basebuf }; | mqtt_msg_t buf = { basebuf, sizeof basebuf }; |
mqtthdr_connflgs_t flg; |
mqtthdr_connflgs_t flg; |
mqtthdr_connack_t cack; |
mqtthdr_connack_t cack; |
struct tagSession *s, *sess = NULL; |
struct tagSession *s, *sess = NULL; |
int ret; | int ret, wlen; |
pthread_attr_t attr; | |
|
|
ioTRACE(4); |
ioTRACE(4); |
|
|
assert(task); |
assert(task); |
|
|
if (!TASK_DATA(task)) { |
if (!TASK_DATA(task)) { |
|
/* flow from accept new clients */ |
sess = initSession(TASK_FD(task), TASK_ARG(task)); |
sess = initSession(TASK_FD(task), TASK_ARG(task)); |
|
io_freeVar(TASK_ARG(task)); |
if (!sess) { |
if (!sess) { |
io_freeVar(TASK_ARG(task)); |
|
close(TASK_FD(task)); |
close(TASK_FD(task)); |
return NULL; |
return NULL; |
} |
} |
Line 290 startSession(sched_task_t *task)
|
Line 195 startSession(sched_task_t *task)
|
/* receive & decode packet */ |
/* receive & decode packet */ |
if (recv(sess->sess_sock, buf.msg_base, buf.msg_len, 0) == -1) { |
if (recv(sess->sess_sock, buf.msg_base, buf.msg_len, 0) == -1) { |
ioDEBUG(3, "Error:: recv(%d) #%d - %s", sess->sess_sock, errno, strerror(errno)); |
ioDEBUG(3, "Error:: recv(%d) #%d - %s", sess->sess_sock, errno, strerror(errno)); |
finiSession(sess, 0); | finiSession(sess); |
return NULL; |
return NULL; |
} |
} |
} else { |
} else { |
sess = TASK_DATA(task); |
sess = TASK_DATA(task); |
|
assert(sess); |
|
|
buf.msg_len = TASK_DATLEN(task) > sizeof basebuf ? sizeof basebuf : TASK_DATLEN(task); |
buf.msg_len = TASK_DATLEN(task) > sizeof basebuf ? sizeof basebuf : TASK_DATLEN(task); |
memcpy(buf.msg_base, sess->sess_buf->msg_base, buf.msg_len); |
memcpy(buf.msg_base, sess->sess_buf->msg_base, buf.msg_len); |
} |
} |
Line 324 startSession(sched_task_t *task)
|
Line 231 startSession(sched_task_t *task)
|
ioDEBUG(1, "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) { |
if (call.FiniSessPUB(&cfg, pub, sess->sess_cid, sess->sess_user, "%") > 0) { |
ioDEBUG(2, "Old session %s should be disconnect!", sess->sess_cid); |
ioDEBUG(2, "Old session %s should be disconnect!", sess->sess_cid); |
TAILQ_FOREACH(s, &Sessions, sess_node) |
TAILQ_FOREACH(s, &Sessions, sess_node) |
if (!strcmp(s->sess_cid, sess->sess_cid)) { |
if (!strcmp(s->sess_cid, sess->sess_cid)) { |
/* found stale session & disconnect it! */ |
/* found stale session & disconnect it! */ |
stopSession(s); | schedWrite(root, leaveClient, sess, TASK_FD(task), NULL, 0); |
break; |
break; |
} |
} |
} |
} |
|
|
if (call.InitSessPUB(&cfg, pub, sess->sess_cid, sess->sess_user, sess->sess_addr, |
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.flag, sess->sess_will.topic, sess->sess_will.msg, |
sess->sess_will.qos, sess->sess_will.retain) == -1) { |
sess->sess_will.qos, sess->sess_will.retain) == -1) { |
Line 345 startSession(sched_task_t *task)
|
Line 254 startSession(sched_task_t *task)
|
ret = MQTT_RETCODE_ACCEPTED; |
ret = MQTT_RETCODE_ACCEPTED; |
} |
} |
|
|
ret = mqtt_msgCONNACK(&msg, ret); |
|
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)); |
|
finiSession(sess, 0); |
|
return NULL; |
|
} else { |
|
ioDEBUG(5, "Sended %d bytes", ret); |
|
free(msg.msg_base); |
|
memset(&msg, 0, sizeof msg); |
|
} |
|
|
|
/* Start session thread OK ... */ |
/* Start session thread OK ... */ |
pthread_attr_init(&attr); | SESS_LOCK; |
pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED); | |
| |
pthread_mutex_lock(&mtx_sess); | |
TAILQ_INSERT_TAIL(&Sessions, sess, sess_node); |
TAILQ_INSERT_TAIL(&Sessions, sess, sess_node); |
pthread_create(&sess->sess_tid, &attr, (void*(*)(void*)) thrSession, sess); | SESS_UNLOCK; |
pthread_mutex_unlock(&mtx_sess); | |
|
|
|
if (!schedRead(root, dispatchSession, sess, TASK_FD(task), NULL, 0)) |
|
ioLIBERR(sched); |
|
|
call.LOG(logg, "Session %s started from %s for user %s (timeout=%d) OK!\n", sess->sess_cid, |
call.LOG(logg, "Session %s started from %s for user %s (timeout=%d) OK!\n", sess->sess_cid, |
sess->sess_addr, sess->sess_user, sess->sess_ka); |
sess->sess_addr, sess->sess_user, sess->sess_ka); |
|
|
pthread_attr_destroy(&attr); |
|
return NULL; |
|
end: /* close client connection */ |
end: /* close client connection */ |
ret = mqtt_msgCONNACK(&msg, ret); | wlen = mqtt_msgCONNACK(sess->sess_buf, ret); |
if ((ret = send(sess->sess_sock, msg.msg_base, ret, 0)) == -1) { | if ((wlen = send(TASK_FD(task), sess->sess_buf->msg_base, wlen, 0)) == -1) { |
ioDEBUG(3, "Error:: send(%d) #%d - %s", sess->sess_sock, errno, strerror(errno)); | ioDEBUG(3, "Error:: send(%d) #%d - %s", TASK_FD(task), errno, strerror(errno)); |
} else { | } else |
ioDEBUG(5, "Sended %d bytes", ret); | ioDEBUG(5, "Sended %d bytes", wlen); |
free(msg.msg_base); | |
memset(&msg, 0, sizeof msg); | |
} | |
|
|
ioDEBUG(1, "Close client %s with socket=%d", sess->sess_addr, sess->sess_sock); | if (ret != MQTT_RETCODE_ACCEPTED) { |
finiSession(sess, 0); | ioDEBUG(1, "Close client %s with socket=%d", sess->sess_addr, TASK_FD(task)); |
| finiSession(sess); |
| } |
return NULL; |
return NULL; |
} |
} |
|
|
/* ----------------------------------------------------------------------- */ |
|
|
|
static void * |
static void * |
thrSched(void *arg __unused) | acceptClient(sched_task_t *task) |
{ |
{ |
struct tagSession *sess; | int cli; |
struct timespec pl = { 0, 10000000 }; | io_sockaddr_t sa; |
| socklen_t sslen = sizeof sa.ss; |
| ait_val_t *v; |
| char str[STRSIZ]; |
|
|
ioTRACE(1); | ioTRACE(4); |
|
|
schedPolling(root, &pl, NULL); | assert(task); |
schedRun(root, &Kill); | |
|
|
TAILQ_FOREACH(sess, &Sessions, sess_node) | if ((cli = accept(TASK_FD(task), &sa.sa, &sslen)) == -1) |
if (sess->sess_tid) | goto end; |
pthread_cancel(sess->sess_tid); | else |
ioDEBUG(5, "EXIT from Scheduler thread !!!"); | fcntl(cli, F_SETFL, fcntl(cli, F_GETFL, 0) | O_NONBLOCK); |
pthread_exit(NULL); | |
| v = io_allocVar(); |
| if (!v) { |
| ioLIBERR(io); |
| close(cli); |
| goto end; |
| } else { |
| memset(str, 0, sizeof str); |
| snprintf(str, sizeof str, "%s:%hu", io_n2addr(&sa, v), io_n2port(&sa)); |
| AIT_FREE_VAL(v); |
| 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, NULL, 0)) { |
| io_freeVar(v); |
| close(cli); |
| ioDEBUG(1, "Terminated client with socket=%d", cli); |
| } |
| end: |
| if (!schedRead(root, acceptClient, NULL, TASK_FD(task), NULL, 0)) |
| ioLIBERR(sched); |
| return NULL; |
} |
} |
|
|
|
/* ----------------------------------------------------------------------- */ |
|
|
int |
int |
Run(int sock) |
Run(int sock) |
{ |
{ |
io_sockaddr_t sa; | struct tagPub *pub; |
socklen_t sslen = sizeof sa.ss; | struct timespec pl = { 0, 100000000 }; |
int cli; | |
pthread_t tid; | |
ait_val_t *v; | |
char str[STRSIZ]; | |
|
|
ioTRACE(1); |
ioTRACE(1); |
|
|
if (pthread_create(&tid, NULL, thrSched, NULL)) { | if (listen(sock, SOMAXCONN) == -1) { |
ioSYSERR(0); |
ioSYSERR(0); |
return -1; |
return -1; |
} else |
} else |
pthread_detach(tid); | fcntl(sock, F_SETFL, fcntl(sock, F_GETFL, 0) | O_NONBLOCK); |
ioDEBUG(2, "Run scheduler management thread"); | |
|
|
if (listen(sock, SOMAXCONN) == -1) { | /* state machine - accept new connections */ |
ioSYSERR(0); | if (!schedRead(root, acceptClient, NULL, sock, NULL, 0)) { |
| ioLIBERR(sched); |
return -1; |
return -1; |
} |
} |
|
|
while (!Kill) { | schedPolling(root, &pl, NULL); |
if ((cli = accept(sock, &sa.sa, &sslen)) == -1) { | schedRun(root, &Kill); |
if (!Kill) | |
continue; | |
ioSYSERR(0); | |
break; | |
} | |
v = io_allocVar(); | |
if (!v) { | |
ioLIBERR(mqtt); | |
break; | |
} else { | |
memset(str, 0, sizeof str); | |
snprintf(str, sizeof str, "%s:%hu", io_n2addr(&sa, 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, NULL, 0)) { | /* free all undeleted elements into lists */ |
io_freeVar(v); | PUBS_LOCK; |
close(cli); | TAILQ_FOREACH(pub, &Pubs, pub_node) { |
ioDEBUG(1, "Terminated client with socket=%d", cli); | TAILQ_REMOVE(&Pubs, pub, pub_node); |
} | |
} | |
|
|
|
AIT_FREE_VAL(&pub->pub_name); |
|
if (pub->pub_packet.msg_base) |
|
free(pub->pub_packet.msg_base); |
|
} |
|
PUBS_UNLOCK; |
return 0; |
return 0; |
} |
} |