--- libaitmqtt/src/conn.c	2012/06/20 08:55:50	1.1.1.1.2.3
+++ libaitmqtt/src/conn.c	2012/06/20 09:17:28	1.1.1.1.2.5
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@openbsd-bg.org>
 *
 * $Author: misho $
-* $Id: conn.c,v 1.1.1.1.2.3 2012/06/20 08:55:50 misho Exp $
+* $Id: conn.c,v 1.1.1.1.2.5 2012/06/20 09:17:28 misho Exp $
 *
 **************************************************************************
 The ELWIX and AITNET software is distributed under the following
@@ -77,8 +77,6 @@ mqtt_msgCONNECT(mqtt_msg_t * __restrict buf, const cha
 
 	if (!buf || !csConnID)
 		return -1;
-	else
-		data = buf->msg_base;
 	if (strlen(csConnID) > 23) {
 		mqtt_SetErr(EINVAL, "Invalid argument ConnID is too long (max 23 bytes)");
 		return -1;
@@ -106,7 +104,7 @@ mqtt_msgCONNECT(mqtt_msg_t * __restrict buf, const cha
 	}
 	if (csWillTopic && *csWillTopic) {		/* will messages */
 		len += sizeof(mqtt_len_t) + strlen(csWillTopic);
-		len += sizeof(mqtt_len_t) + strlen(csWillMessage);
+		len += sizeof(mqtt_len_t) + (csWillMessage ? strlen(csWillMessage) : 0);
 	}
 
 	/* calculate header size */
@@ -116,10 +114,13 @@ mqtt_msgCONNECT(mqtt_msg_t * __restrict buf, const cha
 
 	if (mqtt_msgRealloc(buf, siz + len) == -1)
 		return -1;
-	else
+	else {
+		data = buf->msg_base;
 		hdr = (struct mqtthdr *) data;
+	}
 
 	/* fixed header */
+	printf("%p - %p\n", hdr, data);
 	MQTTHDR_MSGINIT(hdr);
 	hdr->mqtt_msg.type = MQTT_TYPE_CONNECT;
 	*(u_int*) hdr->mqtt_len = n;