Return to srvside.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / libaitmqtt / src

ver 1.3

    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: srvside.c,v 1.3 2012/06/29 13:54:48 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: */
   46: #include "global.h"
   47: 
   48: 
   49: /*
   50:  * mqtt_srv_cliInit() Init MQTT server side support for clients
   51:  *
   52:  * @sock = Client socket from accept()
   53:  * @buf = Message buffer
   54:  * @timeout = timeout
   55:  * @nb = Non block socket
   56:  * return: NULL error or !=NULL allocated server handle
   57:  */
   58: mqtt_srv_t *
   59: mqtt_srv_cliInit(int sock, mqtt_msg_t * __restrict buf, u_short timeout, int nb)
   60: {
   61: 	mqtt_srv_t *srv = NULL;
   62: 	struct timeval tv = { timeout, 0 };
   63: 	int n;
   64: 
   65: 	if (!buf || !buf->msg_base || !buf->msg_len)
   66: 		return srv;
   67: 	else
   68: 		n = buf->msg_len;
   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: 
   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: 
   84: 	srv->sock = sock;
   85: 	srv->buf = buf;
   86: 	srv->timeout = timeout;
   87: 
   88: 	return srv;
   89: }
   90: 
   91: /*
   92:  * mqtt_srv_cliFini() Finish MQTT server side support for clients
   93:  *
   94:  * @psrv = Server handle
   95:  * return: none
   96:  */
   97: void
   98: mqtt_srv_cliFini(mqtt_srv_t ** __restrict psrv)
   99: {
  100: 	if (psrv && *psrv) {
  101: 		free(*psrv);
  102: 		*psrv = NULL;
  103: 	}
  104: }
  105: 
  106: /*
  107:  * mqtt_srv_cliDispatch() MQTT server dispatcher for clients
  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
  115: mqtt_srv_cliDispatch(mqtt_srv_t * __restrict srv, int rlen, void *arg)
  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: }
  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: }