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.3.12.1 2016/09/14 15:52:36 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 - 2016
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: #ifndef __linux__
79: if (connect(cli->sock, addr, addr->sa_len) == -1) {
80: #else
81: if (connect(cli->sock, addr, addr->sa_family == AF_INET6 ?
82: sizeof(struct sockaddr_in6) :
83: sizeof(struct sockaddr_in)) == -1) {
84: #endif
85: LOGERR;
86: close(cli->sock);
87: free(cli);
88: return NULL;
89: }
90:
91: cli->buf = mqtt_msgAlloc(USHRT_MAX);
92: if (!cli->buf) {
93: close(cli->sock);
94: free(cli);
95: return NULL;
96: }
97:
98: return cli;
99: }
100:
101: /*
102: * mqtt_cli_Close() - Close client connection
103: *
104: * @cli = connected client
105: * return: -1 error or 0 disconnected client and freed all resources
106: */
107: int
108: mqtt_cli_Close(mqtt_cli_t ** __restrict cli)
109: {
110: int siz = 0;
111:
112: if (!cli || !*cli)
113: return -1;
114:
115: /* send disconnect */
116: siz = mqtt_msgDISCONNECT((*cli)->buf);
117: if (siz > -1) {
118: siz = send((*cli)->sock, (*cli)->buf->msg_base, siz, MSG_NOSIGNAL);
119: if (siz > -1)
120: shutdown((*cli)->sock, SHUT_RDWR);
121: }
122: close((*cli)->sock);
123:
124: mqtt_msgFree(&(*cli)->buf, 42);
125:
126: free(*cli);
127: *cli = NULL;
128: return 0;
129: }
130:
131: /*
132: * mqtt_cli_Subscribe() - Subscribe to broker
133: *
134: * @cli = connected client
135: * @Topics = Topics for subscribes
136: * @msgID = Message ID
137: * @Dup = Duplicated request
138: * @QoS = Message QoS
139: * return: NULL error or !=NULL allocated array with subscribed QoS responses,
140: * must be free() result!
141: */
142: u_char *
143: mqtt_cli_Subscribe(mqtt_cli_t * __restrict cli, mqtt_subscr_t * __restrict Topics,
144: u_short msgID, u_char Dup, u_char QoS)
145: {
146: int siz = 0;
147: u_short mid = 0;
148: u_char *qoses = NULL;
149:
150: if (!cli || !Topics)
151: return NULL;
152:
153: /* send subscribe */
154: siz = mqtt_msgSUBSCRIBE(cli->buf, Topics, msgID, Dup, QoS);
155: if (siz == -1)
156: return NULL;
157: siz = send(cli->sock, cli->buf->msg_base, siz, MSG_NOSIGNAL);
158: if (siz == -1) {
159: LOGERR;
160: return NULL;
161: } else
162: memset(cli->buf->msg_base, 0, cli->buf->msg_len);
163:
164: if ((siz = mqtt_wait4data(cli->sock, cli->timeout, POLLIN | POLLPRI)) == -1) {
165: return NULL;
166: } else if (siz && mqtt_KeepAlive(cli->sock, cli->timeout, 1))
167: return NULL;
168:
169: /* receive suback */
170: siz = recv(cli->sock, cli->buf->msg_base, cli->buf->msg_len, 0);
171: if (siz == -1) {
172: LOGERR;
173: return NULL;
174: }
175: siz = mqtt_readSUBACK(cli->buf, &mid, &qoses);
176: if (siz == -1)
177: return NULL;
178: if (msgID != mid) {
179: free(qoses);
180: mqtt_SetErr(ECANCELED, "Receive different message ID %hu != %hu", msgID, mid);
181: return NULL;
182: }
183:
184: return qoses;
185: }
186:
187: /*
188: * mqtt_cli_Unsubscribe() - Unsubscribe from broker
189: *
190: * @cli = connected client
191: * @Topics = Topics for unsubscribes
192: * @msgID = Message ID
193: * @Dup = Duplicated request
194: * @QoS = Message QoS
195: * return: -1 error or 0 ok
196: */
197: int
198: mqtt_cli_Unsubscribe(mqtt_cli_t * __restrict cli, mqtt_subscr_t * __restrict Topics,
199: u_short msgID, u_char Dup, u_char QoS)
200: {
201: int siz = 0;
202:
203: if (!cli || !Topics)
204: return -1;
205:
206: /* send unsubscribe */
207: siz = mqtt_msgUNSUBSCRIBE(cli->buf, Topics, msgID, Dup, QoS);
208: if (siz == -1)
209: return -1;
210: siz = send(cli->sock, cli->buf->msg_base, siz, MSG_NOSIGNAL);
211: if (siz == -1) {
212: LOGERR;
213: return -1;
214: } else
215: memset(cli->buf->msg_base, 0, cli->buf->msg_len);
216:
217: if ((siz = mqtt_wait4data(cli->sock, cli->timeout, POLLIN | POLLPRI)) == -1) {
218: return -1;
219: } else if (siz && mqtt_KeepAlive(cli->sock, cli->timeout, 1))
220: return -1;
221:
222: /* receive unsuback */
223: siz = recv(cli->sock, cli->buf->msg_base, cli->buf->msg_len, 0);
224: if (siz == -1) {
225: LOGERR;
226: return -1;
227: }
228: siz = mqtt_readUNSUBACK(cli->buf);
229: if (siz == -1)
230: return -1;
231: if (msgID != siz) {
232: mqtt_SetErr(ECANCELED, "Receive different message ID %hu != %hu", msgID, siz);
233: return -1;
234: }
235:
236: return 0;
237: }
238:
239: /*
240: * mqtt_cli_Publish() - Publish message to broker
241: *
242: * @cli = connected client
243: * @msgID = Message ID
244: * @Dup = Duplicated request
245: * @QoS = Message QoS
246: * @Retain = Retain message
247: * @csTopic = Topic
248: * @pData = Data
249: * @datLen = Data length
250: * return: -1 error or > -1 sended bytes
251: */
252: int
253: mqtt_cli_Publish(mqtt_cli_t * __restrict cli, u_short msgID, u_char Dup, u_char QoS, u_char Retain,
254: const char *csTopic, const void *pData, int datLen)
255: {
256: int wlen = 0, siz = 0;
257:
258: if (!cli || !csTopic)
259: return -1;
260:
261: /* send publish */
262: siz = mqtt_msgPUBLISH(cli->buf, csTopic, msgID, Dup, QoS, Retain, pData, datLen);
263: if (siz == -1)
264: return -1;
265: siz = send(cli->sock, cli->buf->msg_base, siz, MSG_NOSIGNAL);
266: if (siz == -1) {
267: LOGERR;
268: return -1;
269: } else {
270: wlen = siz;
271: memset(cli->buf->msg_base, 0, cli->buf->msg_len);
272: }
273:
274: if (QoS == MQTT_QOS_ONCE) /* no reply */
275: goto end;
276:
277: if ((siz = mqtt_wait4data(cli->sock, cli->timeout, POLLIN | POLLPRI)) == -1) {
278: return -1;
279: } else if (siz && mqtt_KeepAlive(cli->sock, cli->timeout, 1))
280: return -1;
281:
282: /* receive PUBxxx */
283: siz = recv(cli->sock, cli->buf->msg_base, cli->buf->msg_len, 0);
284: if (siz == -1) {
285: LOGERR;
286: return -1;
287: }
288:
289: if (QoS == MQTT_QOS_ACK) { /* reply with PUBACK */
290: siz = mqtt_readPUBACK(cli->buf);
291: if (siz == -1)
292: return -1;
293: if (msgID != siz) {
294: mqtt_SetErr(ECANCELED, "Receive different message ID %hu != %hu", msgID, siz);
295: return -1;
296: }
297: goto end;
298: } else { /* reply with PUBREC */
299: siz = mqtt_readPUBREC(cli->buf);
300: if (siz == -1)
301: return -1;
302: if (msgID != siz) {
303: mqtt_SetErr(ECANCELED, "Receive different message ID %hu != %hu", msgID, siz);
304: return -1;
305: }
306: }
307:
308: do {
309: /* send publish release QoS == 2 */
310: siz = mqtt_msgPUBREL(cli->buf, msgID);
311: if (siz == -1)
312: return -1;
313: siz = send(cli->sock, cli->buf->msg_base, siz, MSG_NOSIGNAL);
314: if (siz == -1) {
315: LOGERR;
316: return -1;
317: } else
318: memset(cli->buf->msg_base, 0, cli->buf->msg_len);
319:
320: if ((siz = mqtt_wait4data(cli->sock, cli->timeout, POLLIN | POLLPRI)) == -1) {
321: return -1;
322: } else if (siz && mqtt_KeepAlive(cli->sock, cli->timeout, 1)) {
323: if (Dup++ > 1)
324: return -1;
325: else
326: continue;
327: }
328:
329: /* receive PUBCOMP */
330: siz = recv(cli->sock, cli->buf->msg_base, cli->buf->msg_len, 0);
331: if (siz == -1) {
332: LOGERR;
333: return -1;
334: }
335:
336: siz = mqtt_readPUBCOMP(cli->buf);
337: if (siz == -1)
338: return -1;
339: if (msgID != siz) {
340: mqtt_SetErr(ECANCELED, "Receive different message ID %hu != %hu", msgID, siz);
341: if (Dup++ > 1)
342: return -1;
343: else
344: continue;
345: }
346: } while (0);
347:
348: end:
349: return wlen;
350: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to cliside.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / libaitmqtt / src