File:  [ELWIX - Embedded LightWeight unIX -] / libaitmqtt / src / cliside.c
Revision 1.1.2.4: download - view: text, annotated - select for diffs - revision graph
Tue May 8 09:07:31 2012 UTC (12 years, 2 months ago) by misho
Branches: mqtt1_1
added new field in api structures for timeout
added new argument timeout

    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: cliside.c,v 1.1.2.4 2012/05/08 09:07:31 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_cli_Open() - Open client connection to MQTT broker
   51:  *
   52:  * @addr = brokers address
   53:  * @timeout = timeout
   54:  * return: NULL error or !=NULL connected to broker
   55:  */
   56: mqtt_cli_t *
   57: mqtt_cli_Open(struct sockaddr *addr, u_short timeout)
   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: 
   71: 	cli->timeout = timeout;
   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: }
  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: 
  144: 	if (!cli)
  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: }

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