--- mqtt/src/mqtt_subs.c 2012/06/20 15:01:30 1.2.2.16 +++ mqtt/src/mqtt_subs.c 2017/10/08 22:49:25 1.5 @@ -1,11 +1,54 @@ +/************************************************************************* +* (C) 2011 AITNET ltd - Sofia/Bulgaria - +* by Michael Pounov +* +* $Author: misho $ +* $Id: mqtt_subs.c,v 1.5 2017/10/08 22:49:25 misho Exp $ +* +************************************************************************** +The ELWIX and AITNET software is distributed under the following +terms: + +All of the documentation and software included in the ELWIX and AITNET +Releases is copyrighted by ELWIX - Sofia/Bulgaria + +Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013 + by Michael Pounov . All rights reserved. + +Redistribution and use in source and binary forms, with or without +modification, are permitted provided that the following conditions +are met: +1. Redistributions of source code must retain the above copyright + notice, this list of conditions and the following disclaimer. +2. Redistributions in binary form must reproduce the above copyright + notice, this list of conditions and the following disclaimer in the + documentation and/or other materials provided with the distribution. +3. All advertising materials mentioning features or use of this software + must display the following acknowledgement: +This product includes software developed by Michael Pounov +ELWIX - Embedded LightWeight unIX and its contributors. +4. Neither the name of AITNET nor the names of its contributors + may be used to endorse or promote products derived from this software + without specific prior written permission. + +THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND +ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE +IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE +ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE +FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL +DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS +OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) +HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT +LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY +OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF +SUCH DAMAGE. +*/ #include "global.h" #include "rtlm.h" #include "mqtt.h" #include "client.h" -io_enableDEBUG; - extern char compiled[], compiledby[], compilehost[]; volatile intptr_t Kill; sched_root_task_t *root; @@ -126,13 +169,13 @@ execProc(sched_task_t *task) char szLine[MAXPATHLEN + BUFSIZ] = { 0 }; snprintf(szLine, sizeof szLine, "%s '%s' %hu %u", AIT_GET_STR(&args->Value), - (char*) TASK_ARG(task), (u_short) TASK_VAL(task), TASK_DATLEN(task)); + (char*) TASK_ARG(task), (u_short) TASK_VAL(task), (u_int) TASK_DATLEN(task)); if (TASK_ARG(task)) - io_free(TASK_ARG(task)); + e_free(TASK_ARG(task)); f = popen(szLine, "w"); if (!f) { - ioSYSERR(0); + ESYSERR(0); return NULL; } else fputs(TASK_DATA(task), f); @@ -148,6 +191,7 @@ pubRX(sched_task_t *task) void *data = NULL; u_short mid; time_t tim; + struct mqtthdr *hdr; rlen = RecvFrom(TASK_FD(task)); if (rlen == -1) @@ -156,27 +200,60 @@ pubRX(sched_task_t *task) Kill++; return NULL; } - siz = mqtt_readPUBLISH(args->msg, szTopic, sizeof szTopic, &mid, &data); - if (siz == -1) - goto end; - /* send to output */ - tim = time(NULL); - strftime(szTime, sizeof szTime, "%Y-%m-%d %H:%M:%S", localtime(&tim)); - fprintf(TASK_ARG(task), "\n[%s] Message ID: %04hu, Length: %u, Topic: %s\n", - szTime, mid, siz, szTopic); + while (rlen > 0) { + hdr = (struct mqtthdr*) args->msg->msg_base; - if (data) { - fputs((const char*) data, TASK_ARG(task)); - free(data); - } + switch (hdr->mqtt_msg.type) { + case MQTT_TYPE_PUBLISH: + siz = mqtt_readPUBLISH(args->msg, szTopic, sizeof szTopic, &mid, &data); + if (siz == -1) + goto end; + else { + siz = mqtt_pktLen(hdr); + rlen -= siz; + EVERBS(4) printf("Remains %d bytes, packet %d bytes\n", rlen, siz); + } - fprintf(TASK_ARG(task), "\n"); - fflush(TASK_ARG(task)); + /* send to output */ + tim = time(NULL); + strftime(szTime, sizeof szTime, "%Y-%m-%d %H:%M:%S", localtime(&tim)); + fprintf(TASK_ARG(task), "\n[%s] Message ID: %04hu, QoS: %hhu, " + "Length: %u, Topic: %s\n", szTime, mid, hdr->mqtt_msg.qos, + siz, szTopic); - /* if exists exec script */ - if (!AIT_ISEMPTY(&args->Value)) - schedEvent(root, execProc, io_strdup(szTopic), mid, data, siz); + if (data) { + fputs((const char*) data, TASK_ARG(task)); + free(data); + } + + fprintf(TASK_ARG(task), "\n"); + fflush(TASK_ARG(task)); + + /* if exists exec script */ + if (!AIT_ISEMPTY(&args->Value)) + schedEvent(root, execProc, e_strdup(szTopic), mid, data, siz); + + memmove(args->msg->msg_base, args->msg->msg_base + siz, rlen); + break; + case MQTT_TYPE_PINGREQ: + siz = mqtt_msgPINGRESP(args->msg); + if (siz == -1) + goto end; + else + rlen -= siz; + + /* send ping reply */ + if (SendTo(TASK_FD(task), siz) == -1) + goto end; + + memmove(args->msg->msg_base, args->msg->msg_base + siz, rlen); + break; + default: + EVERBS(1) printf("Unwanted message type #%d ...\n", hdr->mqtt_msg.type); + goto end; + } + } end: schedReadSelf(task); return NULL; @@ -195,14 +272,14 @@ main(int argc, char **argv) FILE *lf; struct sigaction sa; - if (!(args = io_malloc(sizeof(struct tagArgs)))) { + if (!(args = e_malloc(sizeof(struct tagArgs)))) { printf("Error:: in arguments #%d - %s\n", errno, strerror(errno)); return 1; } else memset(args, 0, sizeof(struct tagArgs)); if (!(args->subscr = mqtt_subAlloc(idx))) { printf("Error:: in subscribes array #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); - io_free(args); + e_free(args); return 1; } else args->free = cleanArgs; @@ -210,7 +287,7 @@ main(int argc, char **argv) if (!(args->msg = mqtt_msgAlloc(USHRT_MAX))) { printf("Error:: in mqtt buffer #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); args->free(args); - io_free(args); + e_free(args); return 1; } @@ -248,7 +325,7 @@ main(int argc, char **argv) if (!sub) { printf("Error:: #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); args->free(args); - io_free(args); + e_free(args); return 1; } else sub += idx++; @@ -271,7 +348,7 @@ main(int argc, char **argv) if (args->QoS > MQTT_QOS_EXACTLY) { printf("Error:: invalid QoS level %d\n", args->QoS); args->free(args); - io_free(args); + e_free(args); return 1; } break; @@ -288,7 +365,7 @@ main(int argc, char **argv) batch = 0; break; case 'v': - io_incDebug; + e_incVerbose; break; case 'u': un = 1; @@ -296,7 +373,7 @@ main(int argc, char **argv) case 'h': default: args->free(args); - io_free(args); + e_free(args); Usage(); return 1; } @@ -305,7 +382,7 @@ main(int argc, char **argv) if (argc < 2) { printf("Error:: host for connect not found, connection id or topic not supplied!\n\n"); args->free(args); - io_free(args); + e_free(args); Usage(); return 1; } else { @@ -316,16 +393,17 @@ main(int argc, char **argv) AIT_FREE_VAL(&args->Value); AIT_SET_STR(&args->Value, argv[2]); } - if (!io_gethostbyname(*argv, port, &args->addr)) { - printf("Error:: host not valid #%d - %s\n", io_GetErrno(), io_GetError()); + if (!e_gethostbyname(*argv, port, &args->addr)) { + printf("Error:: host not valid #%d - %s\n", elwix_GetErrno(), elwix_GetError()); args->free(args); - io_free(args); + e_free(args); Usage(); return 1; } - printf("Connecting to %s:%d ... ", io_n2addr(&args->addr, &val), io_n2port(&args->addr)); + printf("Connecting to %s:%d ... ", e_n2addr(&args->addr, &val), e_n2port(&args->addr)); AIT_FREE_VAL(&val); + memset(&sa, 0, sizeof sa); sa.sa_handler = sigz; sigemptyset(&sa.sa_mask); sigaction(SIGTERM, &sa, NULL); @@ -390,18 +468,18 @@ main(int argc, char **argv) else lf = stdout; if (lf) { - root = schedBegin(); - ret = Subscribe(args->cli->sock, lf); + root = schedBegin(); + schedRead(root, pubRX, lf, args->cli->sock, NULL, 0); schedRun(root, &Kill); + schedEnd(&root); + if (un) Unsubscribe(args->cli->sock); fclose(lf); - - schedEnd(&root); } else printf("Error:: in output file #%d - %s\n", errno, strerror(errno)); } else @@ -410,6 +488,6 @@ main(int argc, char **argv) mqtt_cli_Close(&args->cli); end: args->free(args); - io_free(args); + e_free(args); return ret; }