Annotation of libaitmqtt/src/srvside.c, revision 1.3

1.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.3     ! misho       6: * $Id: srvside.c,v 1.2.4.2 2012/06/29 08:44:34 misho Exp $
1.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: */
                     46: #include "global.h"
                     47: 
                     48: 
                     49: /*
1.3     ! misho      50:  * mqtt_srv_cliInit() Init MQTT server side support for clients
1.2       misho      51:  *
                     52:  * @sock = Client socket from accept()
                     53:  * @buf = Message buffer
                     54:  * @timeout = timeout
1.3     ! misho      55:  * @nb = Non block socket
1.2       misho      56:  * return: NULL error or !=NULL allocated server handle
                     57:  */
                     58: mqtt_srv_t *
1.3     ! misho      59: mqtt_srv_cliInit(int sock, mqtt_msg_t * __restrict buf, u_short timeout, int nb)
1.2       misho      60: {
                     61:        mqtt_srv_t *srv = NULL;
1.3     ! misho      62:        struct timeval tv = { timeout, 0 };
        !            63:        int n;
1.2       misho      64: 
                     65:        if (!buf || !buf->msg_base || !buf->msg_len)
                     66:                return srv;
1.3     ! misho      67:        else
        !            68:                n = buf->msg_len;
1.2       misho      69: 
                     70:        srv = malloc(sizeof(mqtt_srv_t));
                     71:        if (!srv) {
                     72:                LOGERR;
                     73:                return srv;
                     74:        } else
                     75:                memset(srv, 0, sizeof(mqtt_srv_t));
                     76: 
1.3     ! misho      77:        /* set options of client socket */
        !            78:        setsockopt(sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n);
        !            79:        setsockopt(sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n);
        !            80:        setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO, &tv, sizeof tv);
        !            81:        setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof tv);
        !            82:        ioctl(sock, FIONBIO, nb);
        !            83: 
1.2       misho      84:        srv->sock = sock;
                     85:        srv->buf = buf;
                     86:        srv->timeout = timeout;
                     87: 
                     88:        return srv;
                     89: }
                     90: 
                     91: /*
1.3     ! misho      92:  * mqtt_srv_cliFini() Finish MQTT server side support for clients
1.2       misho      93:  *
                     94:  * @psrv = Server handle
                     95:  * return: none
                     96:  */
                     97: void
1.3     ! misho      98: mqtt_srv_cliFini(mqtt_srv_t ** __restrict psrv)
1.2       misho      99: {
                    100:        if (psrv && *psrv) {
                    101:                free(*psrv);
                    102:                *psrv = NULL;
                    103:        }
                    104: }
                    105: 
                    106: /*
1.3     ! misho     107:  * mqtt_srv_cliDispatch() MQTT server dispatcher for clients
1.2       misho     108:  *
                    109:  * @srv = Server handle
                    110:  * @rlen = Ready bytes to parse
                    111:  * @arg = Argument pass to command callback
                    112:  * return: -1 error or 0 ok
                    113:  */
                    114: int
1.3     ! misho     115: mqtt_srv_cliDispatch(mqtt_srv_t * __restrict srv, int rlen, void *arg)
1.2       misho     116: {
                    117:        struct mqtthdr *hdr;
                    118: 
                    119:        if (!srv || !srv->buf->msg_base || !srv->buf->msg_len)
                    120:                return -1;
                    121:        if (srv->buf->msg_len < sizeof(struct mqtthdr) || rlen < sizeof(struct mqtthdr)) {
                    122:                mqtt_SetErr(EINVAL, "Message is too short ...");
                    123:                return -1;
                    124:        }
                    125:        hdr = (struct mqtthdr*) srv->buf->msg_base;
                    126:        if (hdr->mqtt_msg.type > MQTT_TYPE_MAX) {
                    127:                mqtt_SetErr(ENOSYS, "Message type %d not implemented", hdr->mqtt_msg.type);
                    128:                return -1;
                    129:        }
                    130:        if (srv->cmds[hdr->mqtt_msg.type])
                    131:                return srv->cmds[hdr->mqtt_msg.type]((void*) srv, rlen, arg);
                    132: 
                    133:        mqtt_SetErr(ENOTSUP, "Unsupported message type %d", hdr->mqtt_msg.type);
                    134:        return -1;
                    135: }
1.3     ! misho     136: 
        !           137: 
        !           138: /*
        !           139:  * mqtt_srv_Create() - Create server socket
        !           140:  *
        !           141:  * @sa = Server bind address
        !           142:  * @salen = Server struct sockaddr size
        !           143:  * return: -1 error or >-1 server socket
        !           144:  */
        !           145: int
        !           146: mqtt_srv_Create(struct sockaddr * __restrict sa, int salen)
        !           147: {
        !           148:        int s = -1, n = 1;
        !           149: 
        !           150:        if (!sa)
        !           151:                return -1;
        !           152: 
        !           153:        s = socket(sa->sa_family, SOCK_STREAM, 0);
        !           154:        if (s == -1) {
        !           155:                LOGERR;
        !           156:                return -1;
        !           157:        }
        !           158:        if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) {
        !           159:                LOGERR;
        !           160:                close(s);
        !           161:                return -1;
        !           162:        }
        !           163:        if (bind(s, sa, salen) == -1) {
        !           164:                LOGERR;
        !           165:                close(s);
        !           166:                return -1;
        !           167:        }
        !           168: 
        !           169:        return s;
        !           170: }
        !           171: 
        !           172: /*
        !           173:  * mqtt_srv_Destroy() - Close server socket
        !           174:  *
        !           175:  * @sock = Server socket
        !           176:  * return: -1 error or 0 ok
        !           177:  */
        !           178: int
        !           179: mqtt_srv_Destroy(int sock)
        !           180: {
        !           181:        shutdown(sock, SHUT_RDWR);
        !           182:        return close(sock);
        !           183: }
        !           184: 
        !           185: /*
        !           186:  * mqtt_srv_Listen() - Listen server socket
        !           187:  *
        !           188:  * @sock = Server socket
        !           189:  * @maxconn = max number of pending connections
        !           190:  * @nb = Non block socket
        !           191:  * return: -1 error or 0 ok
        !           192:  */
        !           193: int
        !           194: mqtt_srv_Listen(int sock, int maxconn, int nb)
        !           195: {
        !           196:        if (!maxconn)
        !           197:                maxconn = SOMAXCONN;
        !           198: 
        !           199:        if (listen(sock, maxconn) == -1) {
        !           200:                LOGERR;
        !           201:                return -1;
        !           202:        } else
        !           203:                ioctl(sock, FIONBIO, nb);
        !           204: 
        !           205:        return 0;
        !           206: }

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