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: pub.c,v 1.1 2012/01/26 13:07:33 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
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:
51: /*
52: * mqtt_msgPUBLISH() Create PUBLISH message
53: *
54: * @buf = Message buffer
55: * @csTopic = Publish topic
56: * @msgID = MessageID >0, if QOS != MQTT_QOS_ONCE
57: * @Dup = Duplicate message
58: * @QOS = QoS
59: * @Retain = Retain message
60: * @pData = Publish data into topic
61: * @datlen = Publish data length
62: * return: -1 error or >-1 message size for send
63: */
64: int
65: mqtt_msgPUBLISH(mqtt_msg_t * __restrict buf, const char *csTopic, u_short msgID,
66: u_char Dup, u_char QOS, u_char Retain, const void *pData, int datlen)
67: {
68: int siz = 0;
69: struct mqtthdr *hdr;
70: mqtthdr_var_t *topic;
71: mqtt_len_t *mid;
72: void *data;
73:
74: if (!buf || !csTopic)
75: return -1;
76: if (QOS > MQTT_QOS_EXACTLY) {
77: mqtt_SetErr(EINVAL, "Error:: invalid QoS parameter");
78: return -1;
79: }
80: if (!msgID && QOS != MQTT_QOS_ONCE) {
81: mqtt_SetErr(EINVAL, "Error:: invalid MessageID parameter must be >0");
82: return -1;
83: }
84:
85: if (mqtt_msgRealloc(buf, MQTTMSG_MAX) == -1)
86: return -1;
87: else {
88: hdr = (struct mqtthdr *) (buf->msg_base + siz);
89: siz += sizeof(struct mqtthdr);
90: }
91:
92: /* variable header */
93: topic = (mqtthdr_var_t*) (buf->msg_base + siz);
94: topic->var_sb.val = htons(strlen(csTopic));
95: memcpy(topic->var_data, csTopic, ntohs(topic->var_sb.val));
96: siz += MQTTHDR_VAR_SIZEOF(topic);
97:
98: mid = (mqtt_len_t*) (buf->msg_base + siz);
99: mid->val = htons(msgID);
100: siz += sizeof(mqtt_len_t);
101:
102: /* load with data */
103: if (pData && datlen) {
104: data = buf->msg_base + siz;
105: memcpy(data, pData, datlen);
106: siz += datlen;
107: }
108:
109: /* fixed header */
110: MQTTHDR_MSGINIT(hdr);
111: hdr->mqtt_msg.type = MQTT_TYPE_PUBLISH;
112: hdr->mqtt_msg.qos = QOS;
113: hdr->mqtt_msg.dup = Dup ? 1 : 0;
114: hdr->mqtt_msg.retain = Retain ? 1 : 0;
115: *hdr->mqtt_len = mqtt_encodeLen(siz - sizeof(struct mqtthdr));
116:
117: mqtt_msgRealloc(buf, siz);
118: return siz;
119: }
120:
121: static int
122: _mqtt_msgPUB_(mqtt_msg_t * __restrict buf, u_char cmd, u_short msgID)
123: {
124: int siz = 0;
125: struct mqtthdr *hdr;
126: mqtt_len_t *v;
127:
128: if (!buf)
129: return -1;
130:
131: if (mqtt_msgRealloc(buf, sizeof(struct mqtthdr) + sizeof(mqtt_len_t)) == -1)
132: return -1;
133: else {
134: hdr = (struct mqtthdr *) (buf->msg_base + siz);
135: siz += sizeof(struct mqtthdr);
136: v = (mqtt_len_t*) (buf->msg_base + siz);
137: siz += sizeof(mqtt_len_t);
138: }
139:
140: /* fixed header */
141: MQTTHDR_MSGINIT(hdr);
142: hdr->mqtt_msg.type = cmd;
143: *hdr->mqtt_len = sizeof(mqtt_len_t);
144:
145: /* MessageID */
146: v->val = htons(msgID);
147:
148: return siz;
149: }
150:
151: /*
152: * mqtt_msgPUBACK() Create PUBACK message
153: *
154: * @buf = Message buffer
155: * @msgID = MessageID
156: * return: -1 error or >-1 message size for send
157: */
158: inline int
159: mqtt_msgPUBACK(mqtt_msg_t * __restrict buf, u_short msgID)
160: {
161: return _mqtt_msgPUB_(buf, MQTT_TYPE_PUBACK, msgID);
162: }
163:
164: /*
165: * mqtt_msgPUBREC() Create PUBREC message
166: *
167: * @buf = Message buffer
168: * @msgID = MessageID
169: * return: -1 error or >-1 message size for send
170: */
171: inline int
172: mqtt_msgPUBREC(mqtt_msg_t * __restrict buf, u_short msgID)
173: {
174: return _mqtt_msgPUB_(buf, MQTT_TYPE_PUBREC, msgID);
175: }
176:
177: /*
178: * mqtt_msgPUBREL() Create PUBREL message
179: *
180: * @buf = Message buffer
181: * @msgID = MessageID
182: * return: -1 error or >-1 message size for send
183: */
184: inline int
185: mqtt_msgPUBREL(mqtt_msg_t * __restrict buf, u_short msgID)
186: {
187: return _mqtt_msgPUB_(buf, MQTT_TYPE_PUBREL, msgID);
188: }
189:
190: /*
191: * mqtt_msgPUBCOMP() Create PUBCOMP message
192: *
193: * @buf = Message buffer
194: * @msgID = MessageID
195: * return: -1 error or >-1 message size for send
196: */
197: inline int
198: mqtt_msgPUBCOMP(mqtt_msg_t * __restrict buf, u_short msgID)
199: {
200: return _mqtt_msgPUB_(buf, MQTT_TYPE_PUBCOMP, msgID);
201: }
202:
203:
204: /* ============= decode ============ */
205:
206: /*
207: * mqtt_readPUBLISH() Read PUBLISH message
208: *
209: * @buf = Message buffer
210: * @psTopic = Topic
211: * @topicLen = Topic length
212: * @msgID = MessageID
213: * @pData = Data buffer
214: * @datLen = Data buffer length, if *datLen == 0 allocate memory for pData
215: * return: NULL error or !=NULL MQTT fixed header
216: */
217: struct mqtthdr *
218: mqtt_readPUBLISH(mqtt_msg_t * __restrict buf, char * __restrict psTopic, int topicLen,
219: u_short *msgID, void * __restrict pData, int *datLen)
220: {
221: int len, ret;
222: struct mqtthdr *hdr;
223: mqtthdr_var_t *var;
224: mqtt_len_t *v;
225: caddr_t pos;
226:
227: if (!buf || !psTopic || !msgID || !pData)
228: return NULL;
229:
230: hdr = _mqtt_readHEADER(buf, MQTT_TYPE_PUBLISH, &ret, &len);
231: if (!hdr)
232: return NULL;
233: pos = buf->msg_base + ret + 1;
234: var = (mqtthdr_var_t*) pos;
235:
236: /* topic */
237: len -= MQTTHDR_VAR_SIZEOF(var);
238: if (len < 0) {
239: mqtt_SetErr(EINVAL, "Error:: short message length %d", len);
240: return NULL;
241: } else {
242: memset(psTopic, 0, topicLen--);
243: memcpy(psTopic, var->var_data, ntohs(var->var_sb.val) > topicLen ?
244: topicLen : ntohs(var->var_sb.val));
245: pos += MQTTHDR_VAR_SIZEOF(var);
246: v = (mqtt_len_t*) pos;
247: }
248:
249: len -= sizeof(mqtt_len_t);
250: if (len < 0) {
251: mqtt_SetErr(EINVAL, "Error:: short message length %d", len);
252: return NULL;
253: } else {
254: *msgID = ntohs(v->val);
255: pos += sizeof(mqtt_len_t);
256: }
257:
258: /* data */
259: if (len < 0) {
260: mqtt_SetErr(EINVAL, "Error:: short message length %d", len);
261: return NULL;
262: } else {
263: if (!*datLen) {
264: if (!(pData = malloc(len))) {
265: LOGERR;
266: return NULL;
267: } else
268: *datLen = len;
269: }
270:
271: memset(pData, 0, *datLen);
272: if (len < *datLen)
273: *datLen = len;
274: memcpy(pData, pos, *datLen);
275: }
276:
277: return hdr;
278: }
279:
280: /*
281: * mqtt_readPUBACK() Read PUBACK message
282: *
283: * @buf = Message buffer
284: * return: -1 error or MessageID
285: */
286: u_short
287: mqtt_readPUBACK(mqtt_msg_t * __restrict buf)
288: {
289: int len, ret;
290: struct mqtthdr *hdr;
291: mqtt_len_t *v;
292: caddr_t pos;
293:
294: hdr = _mqtt_readHEADER(buf, MQTT_TYPE_PUBACK, &ret, &len);
295: if (!hdr)
296: return (u_short) -1;
297: if (len < sizeof(mqtt_len_t)) {
298: mqtt_SetErr(EINVAL, "Error:: short message length %d", len);
299: return (u_short) -1;
300: } else {
301: pos = buf->msg_base + ret + 1;
302: v = (mqtt_len_t*) pos;
303: }
304:
305: return ntohs(v->val);
306: }
307:
308: /*
309: * mqtt_readPUBREC() Read PUBREC message
310: *
311: * @buf = Message buffer
312: * return: -1 error or MessageID
313: */
314: u_short
315: mqtt_readPUBREC(mqtt_msg_t * __restrict buf)
316: {
317: int len, ret;
318: struct mqtthdr *hdr;
319: mqtt_len_t *v;
320: caddr_t pos;
321:
322: hdr = _mqtt_readHEADER(buf, MQTT_TYPE_PUBREC, &ret, &len);
323: if (!hdr)
324: return (u_short) -1;
325: if (len < sizeof(mqtt_len_t)) {
326: mqtt_SetErr(EINVAL, "Error:: short message length %d", len);
327: return (u_short) -1;
328: } else {
329: pos = buf->msg_base + ret + 1;
330: v = (mqtt_len_t*) pos;
331: }
332:
333: return ntohs(v->val);
334: }
335:
336: /*
337: * mqtt_readPUBREL() Read PUBREL message
338: *
339: * @buf = Message buffer
340: * return: -1 error or MessageID
341: */
342: u_short
343: mqtt_readPUBREL(mqtt_msg_t * __restrict buf)
344: {
345: int len, ret;
346: struct mqtthdr *hdr;
347: mqtt_len_t *v;
348: caddr_t pos;
349:
350: hdr = _mqtt_readHEADER(buf, MQTT_TYPE_PUBREL, &ret, &len);
351: if (!hdr)
352: return (u_short) -1;
353: if (len < sizeof(mqtt_len_t)) {
354: mqtt_SetErr(EINVAL, "Error:: short message length %d", len);
355: return (u_short) -1;
356: } else {
357: pos = buf->msg_base + ret + 1;
358: v = (mqtt_len_t*) pos;
359: }
360:
361: return ntohs(v->val);
362: }
363:
364: /*
365: * mqtt_readPUBCOMP() Read PUBCOMP message
366: *
367: * @buf = Message buffer
368: * return: -1 error or MessageID
369: */
370: u_short
371: mqtt_readPUBCOMP(mqtt_msg_t * __restrict buf)
372: {
373: int len, ret;
374: struct mqtthdr *hdr;
375: mqtt_len_t *v;
376: caddr_t pos;
377:
378: hdr = _mqtt_readHEADER(buf, MQTT_TYPE_PUBCOMP, &ret, &len);
379: if (!hdr)
380: return (u_short) -1;
381: if (len < sizeof(mqtt_len_t)) {
382: mqtt_SetErr(EINVAL, "Error:: short message length %d", len);
383: return (u_short) -1;
384: } else {
385: pos = buf->msg_base + ret + 1;
386: v = (mqtt_len_t*) pos;
387: }
388:
389: return ntohs(v->val);
390: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to pub.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