--- mqtt/src/client.c 2012/01/30 22:58:24 1.2.2.1 +++ mqtt/src/client.c 2017/10/08 22:49:24 1.5 @@ -1,27 +1,54 @@ +/************************************************************************* +* (C) 2011 AITNET ltd - Sofia/Bulgaria - +* by Michael Pounov +* +* $Author: misho $ +* $Id: client.c,v 1.5 2017/10/08 22:49:24 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 "mqtt.h" #include "client.h" int -InitClient(void) -{ - int sock; - - sock = socket(args->addr.sa.sa_family, SOCK_STREAM, IPPROTO_TCP); - if (sock == -1) { - printf("Error:: socket() #%d - %s\n", errno, strerror(errno)); - return -1; - } - if (connect(sock, &args->addr.sa, args->addr.sa.sa_len) == -1) { - printf("Error:: connect() #%d - %s\n", errno, strerror(errno)); - return -1; - } - - return sock; -} - -int ConnectClient(int sock) { int siz = 0; @@ -41,7 +68,7 @@ ConnectClient(int sock) printf("Error:: send() #%d - %s\n", errno, strerror(errno)); return -1; } else - ioVERBOSE(3) printf("Sended CONNECT %d bytes\n", siz); + EVERBS(3) printf("Sended CONNECT %d bytes\n", siz); pfd.fd = sock; pfd.events = POLLIN | POLLPRI; @@ -50,82 +77,23 @@ ConnectClient(int sock) printf("Error:: poll() #%d - %s\n", errno, strerror(errno)); return -1; case 0: - ioVERBOSE(3) printf("Timeout reached (%d) ...\n", args->ka * 1000); + EVERBS(3) printf("Timeout reached (%d) ...\n", args->ka * 1000); return -1; } if (pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) return -1; + memset(args->msg->msg_base, 0, args->msg->msg_len); siz = recv(sock, args->msg->msg_base, args->msg->msg_len, 0); if (siz == -1) { printf("Error:: recv() #%d - %s\n", errno, strerror(errno)); return -1; } else - ioVERBOSE(3) printf("Received %d bytes\n", siz); + EVERBS(3) printf("Received %d bytes\n", siz); return (u_char) mqtt_readCONNACK(args->msg); } -int -CloseClient(int sock) -{ - int siz = 0; - - siz = mqtt_msgDISCONNECT(args->msg); - if (siz == -1) { - printf("Error:: msgDISCONNECT #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); - return -1; - } - - siz = send(sock, args->msg->msg_base, siz, 0); - if (siz == -1) { - printf("Error:: send() #%d - %s\n", errno, strerror(errno)); - return -1; - } else - ioVERBOSE(3) printf("Sended DISCONNECT %d bytes\n", siz); - - shutdown(sock, SHUT_RDWR); - close(sock); - return siz; -} - -int -KAClient(int sock) -{ - int siz = 0; - struct pollfd pfd; - - siz = mqtt_msgPINGREQ(args->msg); - if (siz == -1) { - printf("Error:: msgPINGREQ #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); - return -1; - } - if (SendTo(sock, siz) == -1) - return -1; - - pfd.fd = sock; - pfd.events = POLLIN | POLLPRI; - switch (poll(&pfd, 1, args->ka * 1000)) { - case -1: - printf("Error:: poll() #%d - %s\n", errno, strerror(errno)); - return -1; - case 0: - ioVERBOSE(3) printf("Timeout reached (%d) ...\n", args->ka * 1000); - return -1; - } - if (pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) - return -1; - - siz = recv(sock, args->msg->msg_base, args->msg->msg_len, 0); - if (siz == -1) { - printf("Error:: recv() #%d - %s\n", errno, strerror(errno)); - return -1; - } else - ioVERBOSE(3) printf("Received %d bytes\n", siz); - - return mqtt_readPINGRESP(args->msg); -} - void * OpenFile(void) { @@ -161,25 +129,27 @@ CloseFile(void) } } -inline int +int SendTo(int sock, int siz) { - siz = send(sock, args->msg->msg_base, siz, 0); + siz = send(sock, args->msg->msg_base, siz, MSG_NOSIGNAL); if (siz == -1) { printf("Error:: send() #%d - %s\n", errno, strerror(errno)); return -1; } else - ioVERBOSE(3) printf("Sended %d bytes\n", siz); + EVERBS(3) printf("Sended %d bytes\n", siz); return siz; } -inline int +int RecvFrom(int sock) { struct pollfd pfd; int siz = 0; + memset(args->msg->msg_base, 0, args->msg->msg_len); + pfd.fd = sock; pfd.events = POLLIN | POLLPRI; do { @@ -188,8 +158,8 @@ RecvFrom(int sock) printf("Error:: poll() #%d - %s\n", errno, strerror(errno)); return -1; case 0: - ioVERBOSE(3) printf("Timeout reached (%d) ...\n", args->ka * 1000); - if (KAClient(sock) == -1) + EVERBS(3) printf("Timeout reached (%d) ...\n", args->ka * 1000); + if (mqtt_KeepAlive(sock, args->ka, 1) == -1) return -1; continue; } @@ -202,7 +172,7 @@ RecvFrom(int sock) printf("Error:: recv() #%d - %s\n", errno, strerror(errno)); return -1; } else - ioVERBOSE(3) printf("Received %d bytes\n", siz); + EVERBS(3) printf("Received %d bytes\n", siz); return siz; }