Annotation of mqtt/src/client.c, revision 1.4

1.4     ! misho       1: /*************************************************************************
        !             2: * (C) 2011 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
        !             3: *  by Michael Pounov <misho@openbsd-bg.org>
        !             4: *
        !             5: * $Author: misho $
        !             6: * $Id: client.c,v 1.3.2.1 2012/07/03 12:22:56 misho Exp $
        !             7: *
        !             8: **************************************************************************
        !             9: The ELWIX and AITNET software is distributed under the following
        !            10: terms:
        !            11: 
        !            12: All of the documentation and software included in the ELWIX and AITNET
        !            13: Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
        !            14: 
        !            15: Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012
        !            16:        by Michael Pounov <misho@elwix.org>.  All rights reserved.
        !            17: 
        !            18: Redistribution and use in source and binary forms, with or without
        !            19: modification, are permitted provided that the following conditions
        !            20: are met:
        !            21: 1. Redistributions of source code must retain the above copyright
        !            22:    notice, this list of conditions and the following disclaimer.
        !            23: 2. Redistributions in binary form must reproduce the above copyright
        !            24:    notice, this list of conditions and the following disclaimer in the
        !            25:    documentation and/or other materials provided with the distribution.
        !            26: 3. All advertising materials mentioning features or use of this software
        !            27:    must display the following acknowledgement:
        !            28: This product includes software developed by Michael Pounov <misho@elwix.org>
        !            29: ELWIX - Embedded LightWeight unIX and its contributors.
        !            30: 4. Neither the name of AITNET nor the names of its contributors
        !            31:    may be used to endorse or promote products derived from this software
        !            32:    without specific prior written permission.
        !            33: 
        !            34: THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND
        !            35: ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
        !            36: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
        !            37: ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
        !            38: FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
        !            39: DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
        !            40: OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
        !            41: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
        !            42: LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
        !            43: OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
        !            44: SUCH DAMAGE.
        !            45: */
1.2       misho      46: #include "global.h"
                     47: #include "mqtt.h"
                     48: #include "client.h"
                     49: 
                     50: 
                     51: int
                     52: ConnectClient(int sock)
                     53: {
                     54:        int siz = 0;
                     55:        struct pollfd pfd;
                     56: 
                     57:        siz = mqtt_msgCONNECT(args->msg, (char*) AIT_GET_STR(&args->ConnID), args->ka, 
                     58:                        (char*) AIT_GET_STR(&args->User), (char*) AIT_GET_STR(&args->Pass), 
                     59:                        (char*) args->Will.Topic.val.string, (char*) args->Will.Msg.val.string, 
                     60:                        !args->notClear, args->QoS, args->Retain);
                     61:        if (siz == -1) {
                     62:                printf("Error:: msgCONNECT #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());
                     63:                return -1;
                     64:        }
                     65: 
                     66:        siz = send(sock, args->msg->msg_base, siz, 0);
                     67:        if (siz == -1) {
                     68:                printf("Error:: send() #%d - %s\n", errno, strerror(errno));
                     69:                return -1;
                     70:        } else
                     71:                ioVERBOSE(3) printf("Sended CONNECT %d bytes\n", siz);
                     72: 
                     73:        pfd.fd = sock;
                     74:        pfd.events = POLLIN | POLLPRI;
                     75:        switch (poll(&pfd, 1, args->ka * 1000)) {
                     76:                case -1:
                     77:                        printf("Error:: poll() #%d - %s\n", errno, strerror(errno));
                     78:                        return -1;
                     79:                case 0:
                     80:                        ioVERBOSE(3) printf("Timeout reached (%d) ...\n", args->ka * 1000);
                     81:                        return -1;
                     82:        }
                     83:        if (pfd.revents & (POLLERR | POLLHUP | POLLNVAL))
                     84:                return -1;
                     85: 
1.3       misho      86:        memset(args->msg->msg_base, 0, args->msg->msg_len);
1.2       misho      87:        siz = recv(sock, args->msg->msg_base, args->msg->msg_len, 0);
                     88:        if (siz == -1) {
                     89:                printf("Error:: recv() #%d - %s\n", errno, strerror(errno));
                     90:                return -1;
                     91:        } else
                     92:                ioVERBOSE(3) printf("Received %d bytes\n", siz);
                     93: 
                     94:        return (u_char) mqtt_readCONNACK(args->msg);
                     95: }
                     96: 
                     97: void *
                     98: OpenFile(void)
                     99: {
                    100:        int f, siz = 0;
                    101:        void *mem;
                    102: 
                    103:        if (!args->isFile)
                    104:                return NULL;
                    105: 
                    106:        f = open(AIT_GET_STR(&args->Value), O_RDONLY);
                    107:        if (f == -1) {
                    108:                printf("Error:: in open file #%d - %s\n", errno, strerror(errno));
                    109:                return NULL;
                    110:        }
                    111:        mem = mmap(NULL, siz, PROT_READ, MAP_PRIVATE, f, 0);
                    112:        if (mem == MAP_FAILED) {
                    113:                printf("Error:: in map file #%d - %s\n", errno, strerror(errno));
                    114:                close(f);
                    115:                return NULL;
                    116:        } else
                    117:                close(f);
                    118: 
                    119:        AIT_SET_PTR(&args->Value, mem, siz);
                    120:        return mem;
                    121: }
                    122: 
                    123: void
                    124: CloseFile(void)
                    125: {
                    126:        if (args->isFile) {
                    127:                munmap(AIT_GET_PTR(&args->Value), AIT_LEN(&args->Value));
                    128:                AIT_FREE_VAL(&args->Value);
                    129:        }
                    130: }
                    131: 
                    132: inline int
                    133: SendTo(int sock, int siz)
                    134: {
1.3       misho     135:        siz = send(sock, args->msg->msg_base, siz, MSG_NOSIGNAL);
1.2       misho     136:        if (siz == -1) {
                    137:                printf("Error:: send() #%d - %s\n", errno, strerror(errno));
                    138:                return -1;
                    139:        } else
1.3       misho     140:                ioVERBOSE(3) printf("Sended %d bytes\n", siz);
1.2       misho     141: 
                    142:        return siz;
                    143: }
                    144: 
                    145: inline int
                    146: RecvFrom(int sock)
                    147: {
                    148:        struct pollfd pfd;
                    149:        int siz = 0;
                    150: 
1.3       misho     151:        memset(args->msg->msg_base, 0, args->msg->msg_len);
                    152: 
1.2       misho     153:        pfd.fd = sock;
                    154:        pfd.events = POLLIN | POLLPRI;
                    155:        do {
                    156:                switch (poll(&pfd, 1, args->ka * 1000)) {
                    157:                        case -1:
                    158:                                printf("Error:: poll() #%d - %s\n", errno, strerror(errno));
                    159:                                return -1;
                    160:                        case 0:
                    161:                                ioVERBOSE(3) printf("Timeout reached (%d) ...\n", args->ka * 1000);
1.3       misho     162:                                if (mqtt_KeepAlive(sock, args->ka, 1) == -1)
1.2       misho     163:                                        return -1;
                    164:                                continue;
                    165:                }
                    166:                if (pfd.revents & (POLLERR | POLLHUP | POLLNVAL))
                    167:                        return -1;
                    168:        } while (0);
                    169: 
                    170:        siz = recv(sock, args->msg->msg_base, args->msg->msg_len, 0);
                    171:        if (siz == -1) {
                    172:                printf("Error:: recv() #%d - %s\n", errno, strerror(errno));
                    173:                return -1;
                    174:        } else
                    175:                ioVERBOSE(3) printf("Received %d bytes\n", siz);
                    176: 
                    177:        return siz;
                    178: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>