version 1.1.2.11, 2011/12/08 17:02:25
|
version 1.1.2.15, 2011/12/09 14:01:37
|
Line 5
|
Line 5
|
extern char cliCmd[], *cliStr[]; |
extern char cliCmd[], *cliStr[]; |
|
|
|
|
|
static inline struct tagSession * |
|
initSession(int sock, char * __restrict addr) |
|
{ |
|
struct tagSession *sess = NULL; |
|
const char *str; |
|
|
|
FTRACE(5); |
|
|
|
if (!addr) |
|
return NULL; |
|
|
|
sess = malloc(sizeof(struct tagSession)); |
|
if (!sess) { |
|
VERB(3) syslog(LOG_ERR, "Error:: in malloc #%d - %s", errno, strerror(errno)); |
|
return NULL; |
|
} else |
|
memset(sess, 0, sizeof(struct tagSession)); |
|
|
|
str = cfg_GetAttribute(&cfg, CFG("mqttd"), CFG("retry")); |
|
if (!str) |
|
sess->sess_retry = DEFAULT_RETRY; |
|
else |
|
sess->sess_retry = strtol(str, NULL, 0); |
|
|
|
sess->sess_buf = mqtt_msgAlloc(0); |
|
if (!sess->sess_buf) { |
|
VERB(3) syslog(LOG_ERR, "Error:: in msgAlloc #%d - %s", mqtt_GetErrno(), mqtt_GetError()); |
|
free(sess); |
|
return NULL; |
|
} |
|
|
|
sess->sess_sock = sock; |
|
strlcpy(sess->sess_addr, addr, sizeof sess->sess_addr); |
|
free(addr); |
|
return sess; |
|
} |
|
|
|
static void |
|
finiSession(struct tagSession *sess) |
|
{ |
|
register int i; |
|
struct tagStore *store; |
|
|
|
FTRACE(5); |
|
|
|
if (!sess) |
|
return; |
|
|
|
for (i = 0; i < MQTT_QOS_RESERVED; i++) |
|
while ((store = SLIST_FIRST(&sess->sess_store[i]))) { |
|
SLIST_REMOVE_HEAD(&sess->sess_store[i], 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 (sess->sess_sock > STDERR_FILENO) { |
|
shutdown(sess->sess_sock, SHUT_RDWR); |
|
close(sess->sess_sock); |
|
} |
|
|
|
mqtt_msgFree(&sess->sess_buf, 42); |
|
|
|
if (sess->sess_tid) |
|
pthread_cancel(sess->sess_tid); |
|
free(sess); |
|
} |
|
|
static void * |
static void * |
|
thrSession(struct tagSession *sess) |
|
{ |
|
FTRACE(2); |
|
|
|
TAILQ_REMOVE(&Sessions, sess, sess_node); |
|
pthread_exit(NULL); |
|
return NULL; |
|
} |
|
|
|
static void * |
startSession(sched_task_t *task) |
startSession(sched_task_t *task) |
{ |
{ |
u_char basebuf[USHRT_MAX]; |
u_char basebuf[USHRT_MAX]; |
mqtt_cb_t cbs[MQTT_TYPE_MAX + 1] = { 0 }; | mqtt_msg_t buf = { basebuf, sizeof basebuf }; |
// mqtt_msg_t *buf = { basebuf, sizeof basebuf }; | |
mqtthdr_connflgs_t flg; |
mqtthdr_connflgs_t flg; |
int ret = 0; | mqtthdr_connack_t cack; |
struct timeval tv = { 0 }; | struct tagSession *sess = NULL; |
u_short ka; | |
|
|
FTRACE(4); |
FTRACE(4); |
|
|
/* | sess = initSession(TASK_FD(task), TASK_ARG(task)); |
buf = mqtt_msgAlloc(USHRT_MAX); | if (!sess) { |
if (!buf) { | if (TASK_ARG(task)) |
syslog(LOG_ERR, "Error:: allocate message buf (%d) #%d - %s", (int) TASK_FD(task), | free(TASK_ARG(task)); |
mqtt_GetErrno(), mqtt_GetError()); | close(TASK_FD(task)); |
goto end; | return NULL; |
} |
} |
*/ |
|
|
|
if (recv(TASK_FD(task), basebuf, sizeof basebuf, 0) == -1) { | if (recv(sess->sess_sock, buf.msg_base, buf.msg_len, 0) == -1) { |
VERB(3) syslog(LOG_ERR, "Error:: recv(%d) #%d - %s", (int) TASK_FD(task), | VERB(3) syslog(LOG_ERR, "Error:: recv(%d) #%d - %s", sess->sess_sock, errno, strerror(errno)); |
errno, strerror(errno)); | |
goto end; |
goto end; |
} |
} |
| cack = mqtt_readCONNECT(&buf, &sess->sess_ka, sess->sess_cid, sizeof sess->sess_cid, |
/* | sess->sess_user, sizeof sess->sess_user, sess->sess_pass, sizeof sess->sess_pass, |
flg = mqtt_readCONNECT(buf, &ka, ); | &sess->sess_will.topic, &sess->sess_will.msg); |
| flg.flags = cack.reserved; |
if (flg.reserved) { |
if (flg.reserved) { |
VERB(3) syslog(LOG_ERR, "Error:: in MQTT protocol #%d - %s", | VERB(3) syslog(LOG_ERR, "Error:: in MQTT protocol #%d - %s", mqtt_GetErrno(), mqtt_GetError()); |
mqtt_GetErrno(), mqtt_GetError()); | |
goto end; |
goto end; |
} */ | } else { |
| sess->sess_clean = flg.clean_sess; |
| sess->sess_will.qos = flg.will_qos; |
| sess->sess_will.retain = flg.will_retain; |
| sess->sess_will.flag = flg.will_flg; |
| } |
|
|
/* |
|
for (ret = i = 0; cliCmd[i] && !(ret = (hdr->mqtt_msg.type == cliCmd[i])); i++); |
|
if (!ret) { |
|
VERB(2) syslog(LOG_ERR, "Error:: wrong command type #%d %s", |
|
hdr->mqtt_msg.type, cliStr[i]); |
|
goto end; |
|
} |
|
*/ |
|
|
|
/* check online table for user */ |
/* check online table for user */ |
// ChkSessPUB(&cfg, ); |
|
|
|
// ret = mqttDispatcher(cbs, buf); | /* Start session thread OK ... */ |
// mqtt_msgFree(&buf, 42); | TAILQ_INSERT_TAIL(&Sessions, sess, sess_node); |
| pthread_create(&sess->sess_tid, NULL, (void*(*)(void*)) thrSession, sess); |
/* | pthread_detach(sess->sess_tid); |
switch (ret) { | VERB(1) syslog(LOG_DEBUG, "ConnID=%s(%s) for %s login OK!", |
case -1: | sess->sess_cid, sess->sess_user, sess->sess_addr); |
syslog(LOG_ERR, "Error:: in MQTT protocol #%d - %s", mqtt_GetErrno(), mqtt_GetError()); | return NULL; |
break; | |
default: | |
return NULL; | |
} | |
*/ | |
end: /* close client connection */ |
end: /* close client connection */ |
close(TASK_FD(task)); |
|
VERB(1) syslog(LOG_DEBUG, "Close client %s with socket=%d", |
VERB(1) syslog(LOG_DEBUG, "Close client %s with socket=%d", |
(char*) TASK_ARG(task), (int) TASK_FD(task)); |
(char*) TASK_ARG(task), (int) TASK_FD(task)); |
if (TASK_ARG(task)) | finiSession(sess); |
free(TASK_ARG(task)); | |
return NULL; |
return NULL; |
} |
} |
|
|