Annotation of libaitmqtt/src/cliside.c, revision 1.1.2.5

1.1.2.2   misho       1: /*************************************************************************
                      2: * (C) 2011 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
                      3: *  by Michael Pounov <misho@openbsd-bg.org>
                      4: *
                      5: * $Author: misho $
1.1.2.5 ! misho       6: * $Id: cliside.c,v 1.1.2.4 2012/05/08 09:07:31 misho Exp $
1.1.2.2   misho       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.1.2.1   misho      46: #include "global.h"
                     47: 
                     48: 
1.1.2.3   misho      49: /*
                     50:  * mqtt_cli_Open() - Open client connection to MQTT broker
                     51:  *
                     52:  * @addr = brokers address
1.1.2.4   misho      53:  * @timeout = timeout
1.1.2.3   misho      54:  * return: NULL error or !=NULL connected to broker
                     55:  */
                     56: mqtt_cli_t *
1.1.2.4   misho      57: mqtt_cli_Open(struct sockaddr *addr, u_short timeout)
1.1.2.3   misho      58: {
                     59:        mqtt_cli_t *cli;
                     60: 
                     61:        if (!addr)
                     62:                return NULL;
                     63: 
                     64:        cli = malloc(sizeof(mqtt_cli_t));
                     65:        if (!cli) {
                     66:                LOGERR;
                     67:                return NULL;
                     68:        } else
                     69:                memset(cli, 0, sizeof(mqtt_cli_t));
                     70: 
1.1.2.4   misho      71:        cli->timeout = timeout;
1.1.2.3   misho      72:        cli->sock = socket(addr->sa_family, SOCK_STREAM, IPPROTO_TCP);
                     73:        if (cli->sock == -1) {
                     74:                LOGERR;
                     75:                free(cli);
                     76:                return NULL;
                     77:        }
                     78:        if (connect(cli->sock, addr, addr->sa_len) == -1) {
                     79:                LOGERR;
                     80:                close(cli->sock);
                     81:                free(cli);
                     82:                return NULL;
                     83:        }
                     84: 
                     85:        cli->buf = mqtt_msgAlloc(USHRT_MAX);
                     86:        if (!cli->buf) {
                     87:                close(cli->sock);
                     88:                free(cli);
                     89:                return NULL;
                     90:        }
                     91: 
                     92:        return cli;
                     93: }
                     94: 
                     95: /*
                     96:  * mqtt_cli_Close() - Close client connection
                     97:  *
                     98:  * @cli = connected client
                     99:  * return: -1 error or 0 disconnected client and freed all resources
                    100:  */
                    101: int
                    102: mqtt_cli_Close(mqtt_cli_t ** __restrict cli)
                    103: {
                    104:        int siz = 0;
                    105: 
                    106:        if (!cli || !*cli)
                    107:                return -1;
                    108: 
                    109:        /* send disconnect */
                    110:        siz = mqtt_msgDISCONNECT((*cli)->buf);
                    111:        if (siz > -1) {
                    112:                siz = send((*cli)->sock, (*cli)->buf->msg_base, siz, MSG_NOSIGNAL);
                    113:                if (siz > -1)
                    114:                        shutdown((*cli)->sock, SHUT_RDWR);
                    115:        }
                    116:        close((*cli)->sock);
                    117: 
                    118:        mqtt_msgFree(&(*cli)->buf, 42);
                    119: 
                    120:        free(*cli);
                    121:        *cli = NULL;
                    122:        return 0;
                    123: }
1.1.2.4   misho     124: 
                    125: /*
                    126:  * mqtt_cli_Subscribe() - Subscribe to broker
                    127:  *
                    128:  * @cli = connected client
                    129:  * @Topics = Topics for subscribes
                    130:  * @msgID = Message ID
                    131:  * @Dup = Duplicated request
                    132:  * @QoS = Message QoS
                    133:  * return: NULL error or !=NULL allocated array with subscribed QoS responses, 
                    134:  *     must be free() result!
                    135:  */
                    136: u_char *
                    137: mqtt_cli_Subscribe(mqtt_cli_t * __restrict cli, mqtt_subscr_t * __restrict Topics, 
                    138:                u_short msgID, u_char Dup, u_char QoS)
                    139: {
                    140:        int siz = 0;
                    141:        u_short mid = 0;
                    142:        u_char *qoses = NULL;
                    143: 
1.1.2.5 ! misho     144:        if (!cli || !Topics)
1.1.2.4   misho     145:                return NULL;
                    146: 
                    147:        /* send subscribe */
                    148:        siz = mqtt_msgSUBSCRIBE(cli->buf, Topics, msgID, Dup, QoS);
                    149:        if (siz == -1)
                    150:                return NULL;
                    151:        siz = send(cli->sock, cli->buf->msg_base, siz, MSG_NOSIGNAL);
                    152:        if (siz == -1) {
                    153:                LOGERR;
                    154:                return NULL;
                    155:        }
                    156: 
                    157:        if ((siz = mqtt_wait4data(cli->sock, cli->timeout, POLLIN | POLLPRI)) == -1) {
                    158:                return NULL;
                    159:        } else if (siz && mqtt_KeepAlive(cli->sock, cli->timeout, 1))
                    160:                return NULL;
                    161: 
                    162:        /* receive suback */
                    163:        siz = recv(cli->sock, cli->buf->msg_base, cli->buf->msg_len, 0);
                    164:        if (siz == -1) {
                    165:                LOGERR;
                    166:                return NULL;
                    167:        }
                    168:        siz = mqtt_readSUBACK(cli->buf, &mid, &qoses);
                    169:        if (siz == -1)
                    170:                return NULL;
                    171:        if (msgID != mid) {
                    172:                free(qoses);
                    173:                mqtt_SetErr(EBADMSG, "Receive different message ID %hu != %hu", msgID, mid);
                    174:                return NULL;
                    175:        }
                    176: 
                    177:        return qoses;
                    178: }
1.1.2.5 ! misho     179: 
        !           180: /*
        !           181:  * mqtt_cli_Unsubscribe() - Unsubscribe from broker
        !           182:  *
        !           183:  * @cli = connected client
        !           184:  * @Topics = Topics for unsubscribes
        !           185:  * @msgID = Message ID
        !           186:  * @Dup = Duplicated request
        !           187:  * @QoS = Message QoS
        !           188:  * return: -1 error or 0 ok
        !           189:  */
        !           190: int
        !           191: mqtt_cli_Unsubscribe(mqtt_cli_t * __restrict cli, mqtt_subscr_t * __restrict Topics, 
        !           192:                u_short msgID, u_char Dup, u_char QoS)
        !           193: {
        !           194:        int siz = 0;
        !           195:        u_short mid = 0;
        !           196: 
        !           197:        if (!cli || !Topics)
        !           198:                return -1;
        !           199: 
        !           200:        /* send unsubscribe */
        !           201:        siz = mqtt_msgUNSUBSCRIBE(cli->buf, Topics, msgID, Dup, QoS);
        !           202:        if (siz == -1)
        !           203:                return -1;
        !           204:        siz = send(cli->sock, cli->buf->msg_base, siz, MSG_NOSIGNAL);
        !           205:        if (siz == -1) {
        !           206:                LOGERR;
        !           207:                return -1;
        !           208:        }
        !           209: 
        !           210:        if ((siz = mqtt_wait4data(cli->sock, cli->timeout, POLLIN | POLLPRI)) == -1) {
        !           211:                return -1;
        !           212:        } else if (siz && mqtt_KeepAlive(cli->sock, cli->timeout, 1))
        !           213:                return -1;
        !           214: 
        !           215:        /* receive unsuback */
        !           216:        siz = recv(cli->sock, cli->buf->msg_base, cli->buf->msg_len, 0);
        !           217:        if (siz == -1) {
        !           218:                LOGERR;
        !           219:                return -1;
        !           220:        }
        !           221:        siz = mqtt_readUNSUBACK(cli->buf);
        !           222:        if (siz == -1)
        !           223:                return -1;
        !           224:        if (msgID != siz) {
        !           225:                mqtt_SetErr(EBADMSG, "Receive different message ID %hu != %hu", msgID, mid);
        !           226:                return -1;
        !           227:        }
        !           228: 
        !           229:        return 0;
        !           230: }

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