--- libaitmqtt/src/pub.c 2012/01/26 13:07:33 1.1.1.1 +++ libaitmqtt/src/pub.c 2012/04/27 15:15:12 1.1.1.1.2.3 @@ -3,7 +3,7 @@ * by Michael Pounov * * $Author: misho $ -* $Id: pub.c,v 1.1.1.1 2012/01/26 13:07:33 misho Exp $ +* $Id: pub.c,v 1.1.1.1.2.3 2012/04/27 15:15:12 misho Exp $ * ************************************************************************** The ELWIX and AITNET software is distributed under the following @@ -12,7 +12,7 @@ terms: All of the documentation and software included in the ELWIX and AITNET Releases is copyrighted by ELWIX - Sofia/Bulgaria -Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011 +Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012 by Michael Pounov . All rights reserved. Redistribution and use in source and binary forms, with or without @@ -46,8 +46,6 @@ SUCH DAMAGE. #include "global.h" -/* ------------------------------------------------------------------- */ - /* * mqtt_msgPUBLISH() Create PUBLISH message * @@ -74,11 +72,11 @@ mqtt_msgPUBLISH(mqtt_msg_t * __restrict buf, const cha if (!buf || !csTopic) return -1; if (QOS > MQTT_QOS_EXACTLY) { - mqtt_SetErr(EINVAL, "Error:: invalid QoS parameter"); + mqtt_SetErr(EINVAL, "Invalid QoS parameter"); return -1; } if (!msgID && QOS != MQTT_QOS_ONCE) { - mqtt_SetErr(EINVAL, "Error:: invalid MessageID parameter must be >0"); + mqtt_SetErr(EINVAL, "Invalid MessageID parameter must be >0"); return -1; } @@ -114,7 +112,6 @@ mqtt_msgPUBLISH(mqtt_msg_t * __restrict buf, const cha hdr->mqtt_msg.retain = Retain ? 1 : 0; *hdr->mqtt_len = mqtt_encodeLen(siz - sizeof(struct mqtthdr)); - mqtt_msgRealloc(buf, siz); return siz; } @@ -212,9 +209,9 @@ mqtt_msgPUBCOMP(mqtt_msg_t * __restrict buf, u_short m * @msgID = MessageID * @pData = Data buffer * @datLen = Data buffer length, if *datLen == 0 allocate memory for pData - * return: NULL error or !=NULL MQTT fixed header + * return: NULL error or !=NULL MQTT data buffer */ -struct mqtthdr * +void * mqtt_readPUBLISH(mqtt_msg_t * __restrict buf, char * __restrict psTopic, int topicLen, u_short *msgID, void * __restrict pData, int *datLen) { @@ -224,7 +221,7 @@ mqtt_readPUBLISH(mqtt_msg_t * __restrict buf, char * _ mqtt_len_t *v; caddr_t pos; - if (!buf || !psTopic || !msgID || !pData) + if (!buf || !psTopic || !msgID) return NULL; hdr = _mqtt_readHEADER(buf, MQTT_TYPE_PUBLISH, &ret, &len); @@ -236,7 +233,7 @@ mqtt_readPUBLISH(mqtt_msg_t * __restrict buf, char * _ /* topic */ len -= MQTTHDR_VAR_SIZEOF(var); if (len < 0) { - mqtt_SetErr(EINVAL, "Error:: short message length %d", len); + mqtt_SetErr(EINVAL, "Short message length %d", len); return NULL; } else { memset(psTopic, 0, topicLen--); @@ -248,7 +245,7 @@ mqtt_readPUBLISH(mqtt_msg_t * __restrict buf, char * _ len -= sizeof(mqtt_len_t); if (len < 0) { - mqtt_SetErr(EINVAL, "Error:: short message length %d", len); + mqtt_SetErr(EINVAL, "Short message length %d", len); return NULL; } else { *msgID = ntohs(v->val); @@ -257,15 +254,17 @@ mqtt_readPUBLISH(mqtt_msg_t * __restrict buf, char * _ /* data */ if (len < 0) { - mqtt_SetErr(EINVAL, "Error:: short message length %d", len); + mqtt_SetErr(EINVAL, "Short message length %d", len); return NULL; } else { - if (!*datLen) { - if (!(pData = malloc(len))) { + if (!*datLen || !pData) { + if (!(pData = malloc(len + 1))) { LOGERR; return NULL; - } else + } else { *datLen = len; + ((char*) pData)[len] = 0; + } } memset(pData, 0, *datLen); @@ -274,7 +273,7 @@ mqtt_readPUBLISH(mqtt_msg_t * __restrict buf, char * _ memcpy(pData, pos, *datLen); } - return hdr; + return pData; } /* @@ -295,7 +294,7 @@ mqtt_readPUBACK(mqtt_msg_t * __restrict buf) if (!hdr) return (u_short) -1; if (len < sizeof(mqtt_len_t)) { - mqtt_SetErr(EINVAL, "Error:: short message length %d", len); + mqtt_SetErr(EINVAL, "Short message length %d", len); return (u_short) -1; } else { pos = buf->msg_base + ret + 1; @@ -323,7 +322,7 @@ mqtt_readPUBREC(mqtt_msg_t * __restrict buf) if (!hdr) return (u_short) -1; if (len < sizeof(mqtt_len_t)) { - mqtt_SetErr(EINVAL, "Error:: short message length %d", len); + mqtt_SetErr(EINVAL, "Short message length %d", len); return (u_short) -1; } else { pos = buf->msg_base + ret + 1; @@ -351,7 +350,7 @@ mqtt_readPUBREL(mqtt_msg_t * __restrict buf) if (!hdr) return (u_short) -1; if (len < sizeof(mqtt_len_t)) { - mqtt_SetErr(EINVAL, "Error:: short message length %d", len); + mqtt_SetErr(EINVAL, "Short message length %d", len); return (u_short) -1; } else { pos = buf->msg_base + ret + 1; @@ -379,7 +378,7 @@ mqtt_readPUBCOMP(mqtt_msg_t * __restrict buf) if (!hdr) return (u_short) -1; if (len < sizeof(mqtt_len_t)) { - mqtt_SetErr(EINVAL, "Error:: short message length %d", len); + mqtt_SetErr(EINVAL, "Short message length %d", len); return (u_short) -1; } else { pos = buf->msg_base + ret + 1;