MQTT 3.1.1

src/MQTTAsync.c

@@ -248,6 +248,7 @@ typedef struct
 			int serverURIcount;
 			char** serverURIs;
 			int currentURI;
+      int MQTTVersion;
 		} conn;
 	} details;
 } MQTTAsync_command;
@@ -334,6 +335,18 @@ void MQTTAsync_unlock_mutex(mutex_type amutex)
 }
 
 
+int MQTTAsync_checkConn(MQTTAsync_command* command)
+{
+  int rc;
+
+  FUNC_ENTRY;
+  rc = command->details.conn.currentURI < command->details.conn.serverURIcount ||
+      command->details.conn.MQTTVersion == 4;
+  FUNC_EXIT_RC(rc);
+  return rc;
+}
+
+
 int MQTTAsync_create(MQTTAsync* handle, char* serverURI, char* clientId,
 		int persistence_type, void* persistence_context)
 {
@@ -960,10 +973,16 @@ void MQTTAsync_processCommand()
 		else
 		{
 			char* serverURI = command->client->serverURI;
-			
+
 			if (command->command.details.conn.serverURIcount > 0)
 			{
-				serverURI = command->command.details.conn.serverURIs[command->command.details.conn.currentURI++];
+        if (command->command.details.conn.MQTTVersion == 3)
+        {
+          command->command.details.conn.currentURI++;
+          command->command.details.conn.MQTTVersion = 4;
+        }
+          
+				serverURI = command->command.details.conn.serverURIs[command->command.details.conn.currentURI];
 					
 				if (strncmp(URI_TCP, serverURI, strlen(URI_TCP)) == 0)
 					serverURI += strlen(URI_TCP);
@@ -976,11 +995,16 @@ void MQTTAsync_processCommand()
 #endif
 			}
 
-			Log(TRACE_MIN, -1, "Connecting to serverURI %s", serverURI);
+      if (command->command.details.conn.MQTTVersion == 0)
+        command->command.details.conn.MQTTVersion = 4;
+      else if (command->command.details.conn.MQTTVersion == 4)
+        command->command.details.conn.MQTTVersion = 3;
+
+			Log(TRACE_MIN, -1, "Connecting to serverURI %s with MQTT version %d", serverURI, command->command.details.conn.MQTTVersion);
 #if defined(OPENSSL)
-			rc = MQTTProtocol_connect(serverURI, command->client->c, command->client->ssl);
+			rc = MQTTProtocol_connect(serverURI, command->client->c, command->client->ssl, command->command.details.conn.MQTTVersion);
 #else
-			rc = MQTTProtocol_connect(serverURI, command->client->c);
+			rc = MQTTProtocol_connect(serverURI, command->client->c, command->command.details.conn.MQTTVersion);
 #endif
 			if (command->client->c->connect_state == 0)
 				rc = SOCKET_ERROR;
@@ -1097,12 +1121,10 @@ void MQTTAsync_processCommand()
 		else
 			MQTTAsync_disconnect_internal(command->client, 0);
 			
-		if (command->command.type == CONNECT &&
-			command->command.details.conn.currentURI < command->command.details.conn.serverURIcount)
+		if (command->command.type == CONNECT && MQTTAsync_checkConn(&command->command))
 		{
+			Log(TRACE_MIN, -1, "Connect failed, more to try");
 			/* put the connect command back to the head of the command queue, using the next serverURI */
-			Log(TRACE_MIN, -1, "Connect failed, now trying %s", 
-				command->command.details.conn.serverURIs[command->command.details.conn.currentURI]);
 			rc = MQTTAsync_addCommand(command, sizeof(command->command.details.conn));
 		}
 		else
@@ -1153,7 +1175,7 @@ void MQTTAsync_checkTimeouts()
 		/* check connect timeout */
 		if (m->c->connect_state != 0 && MQTTAsync_elapsed(m->connect.start_time) > (m->connect.details.conn.timeout * 1000))
 		{
-			if (m->connect.details.conn.currentURI < m->connect.details.conn.serverURIcount)
+      if (MQTTAsync_checkConn(&m->connect))
 			{
 				MQTTAsync_queuedCommand* conn;
 				
@@ -1162,9 +1184,8 @@ void MQTTAsync_checkTimeouts()
 				conn = malloc(sizeof(MQTTAsync_queuedCommand));
 				memset(conn, '\0', sizeof(MQTTAsync_queuedCommand));
 				conn->client = m;
-				conn->command = m->connect; 
-				Log(TRACE_MIN, -1, "Connect failed, now trying %s", 
-					m->connect.details.conn.serverURIs[m->connect.details.conn.currentURI]);
+				conn->command = m->connect;
+  			Log(TRACE_MIN, -1, "Connect failed, more to try");
 				MQTTAsync_addCommand(conn, sizeof(m->connect));
 			}
 			else
@@ -1477,7 +1498,7 @@ thread_return_type WINAPI MQTTAsync_receiveThread(void* n)
 					{
 						if (m->connect.details.conn.serverURIcount > 0)
 							Log(TRACE_MIN, -1, "Connect succeeded to %s", 
-								m->connect.details.conn.serverURIs[m->connect.details.conn.currentURI - 1]);
+								m->connect.details.conn.serverURIs[m->connect.details.conn.currentURI]);
 						MQTTAsync_freeConnect(m->connect);
 						if (m->connect.onSuccess)
 						{
@@ -1487,7 +1508,7 @@ thread_return_type WINAPI MQTTAsync_receiveThread(void* n)
 					}
 					else
 					{
-						if (m->connect.details.conn.currentURI < m->connect.details.conn.serverURIcount)
+            if (MQTTAsync_checkConn(&m->connect))
 						{
 							MQTTAsync_queuedCommand* conn;
 							
@@ -1497,8 +1518,7 @@ thread_return_type WINAPI MQTTAsync_receiveThread(void* n)
 							memset(conn, '\0', sizeof(MQTTAsync_queuedCommand));
 							conn->client = m;
 							conn->command = m->connect; 
-							Log(TRACE_MIN, -1, "Connect failed, now trying %s", 
-								m->connect.details.conn.serverURIs[m->connect.details.conn.currentURI]);
+							Log(TRACE_MIN, -1, "Connect failed, more to try");
 							MQTTAsync_addCommand(conn, sizeof(m->connect));
 						}
 						else
@@ -2506,10 +2526,11 @@ int MQTTAsync_connecting(MQTTAsyncs* m)
 					rc = SOCKET_ERROR;
 					goto exit;
 				}
-				else if (rc == 1) {
+				else if (rc == 1) 
+        {
 					rc = MQTTCLIENT_SUCCESS;
 					m->c->connect_state = 3;
-					if (MQTTPacket_send_connect(m->c) == SOCKET_ERROR)
+					if (MQTTPacket_send_connect(m->c, m->connect.details.conn.MQTTVersion) == SOCKET_ERROR)
 					{
 						rc = SOCKET_ERROR;
 						goto exit;
@@ -2528,7 +2549,7 @@ int MQTTAsync_connecting(MQTTAsyncs* m)
 		{
 #endif
 			m->c->connect_state = 3; /* TCP/SSL connect completed, in which case send the MQTT connect packet */
-			if ((rc = MQTTPacket_send_connect(m->c)) == SOCKET_ERROR)
+			if ((rc = MQTTPacket_send_connect(m->c, m->connect.details.conn.MQTTVersion)) == SOCKET_ERROR)
 				goto exit;
 #if defined(OPENSSL)
 		}
@@ -2543,7 +2564,7 @@ int MQTTAsync_connecting(MQTTAsyncs* m)
 		if(!m->c->cleansession && m->c->session == NULL)
 			m->c->session = SSL_get1_session(m->c->net.ssl);
 		m->c->connect_state = 3; /* SSL connect completed, in which case send the MQTT connect packet */
-		if ((rc = MQTTPacket_send_connect(m->c)) == SOCKET_ERROR)
+		if ((rc = MQTTPacket_send_connect(m->c, m->connect.details.conn.MQTTVersion)) == SOCKET_ERROR)
 			goto exit;
 	}
 #endif
@@ -2551,7 +2572,7 @@ int MQTTAsync_connecting(MQTTAsyncs* m)
 exit:
 	if ((rc != 0 && m->c->connect_state != 2) || (rc == SSL_FATAL))
 	{
-		if (m->connect.details.conn.currentURI < m->connect.details.conn.serverURIcount)
+    if (MQTTAsync_checkConn(&m->connect))
 		{
 			MQTTAsync_queuedCommand* conn;
 				
@@ -2561,8 +2582,7 @@ exit:
 			memset(conn, '\0', sizeof(MQTTAsync_queuedCommand));
 			conn->client = m;
 			conn->command = m->connect; 
-			Log(TRACE_MIN, -1, "Connect failed, now trying %s", 
-				m->connect.details.conn.serverURIs[m->connect.details.conn.currentURI]);
+			Log(TRACE_MIN, -1, "Connect failed, more to try");
 			MQTTAsync_addCommand(conn, sizeof(m->connect));
 		}
 		else
@@ -2630,7 +2650,7 @@ MQTTPacket* MQTTAsync_cycle(int* sock, unsigned long timeout, int* rc)
 			if ((m->c->connect_state == 3) && (*rc == SOCKET_ERROR))
 			{
 				Log(TRACE_MINIMUM, -1, "CONNECT sent but MQTTPacket_Factory has returned SOCKET_ERROR");
-				if (m->connect.details.conn.currentURI < m->connect.details.conn.serverURIcount)
+        if (MQTTAsync_checkConn(&m->connect))
 				{
 					MQTTAsync_queuedCommand* conn;
 				
@@ -2640,8 +2660,7 @@ MQTTPacket* MQTTAsync_cycle(int* sock, unsigned long timeout, int* rc)
 					memset(conn, '\0', sizeof(MQTTAsync_queuedCommand));
 					conn->client = m;
 					conn->command = m->connect; 
-					Log(TRACE_MIN, -1, "Connect failed, now trying %s", 
-						m->connect.details.conn.serverURIs[m->connect.details.conn.currentURI]);
+    			Log(TRACE_MIN, -1, "Connect failed, more to try");
 					MQTTAsync_addCommand(conn, sizeof(m->connect));
 				}
 				else

src/MQTTClient.c

@@ -109,7 +109,7 @@ void MQTTClient_init()
 	pthread_mutexattr_init(&attr);
 	pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_ERRORCHECK);
 	if ((rc = pthread_mutex_init(mqttclient_mutex, &attr)) != 0)
-		printf("MQTTAsync: error %d initializing client_mutex\n", rc);
+		printf("MQTTClient: error %d initializing client_mutex\n", rc);
 }
 
 #define WINAPI
@@ -127,6 +127,7 @@ MQTTPacket* MQTTClient_cycle(int* sock, unsigned long timeout, int* rc);
 int MQTTClient_cleanSession(Clients* client);
 void MQTTClient_stop();
 int MQTTClient_disconnect_internal(MQTTClient handle, int timeout);
+int MQTTClient_disconnect1(MQTTClient handle, int timeout, int internal, int stop);
 void MQTTClient_writeComplete(int socket);
 
 typedef struct
@@ -726,16 +727,13 @@ void Protocol_processPublication(Publish* publish, Clients* client)
 }
 
 
-int MQTTClient_connectURI(MQTTClient handle, MQTTClient_connectOptions* options, char* serverURI)
+int MQTTClient_connectURIVersion(MQTTClient handle, MQTTClient_connectOptions* options, char* serverURI, int MQTTVersion,
+	START_TIME_TYPE start, long millisecsTimeout)
 {
 	MQTTClients* m = handle;
-	START_TIME_TYPE start;
-	long millisecsTimeout = 30000L;
 	int rc = SOCKET_ERROR;
 
-	FUNC_ENTRY;
-	millisecsTimeout = options->connectTimeout * 1000;
-	start = MQTTClient_start_clock();
+  FUNC_ENTRY;
 	if (m->ma && !running)
 	{
 		Thread_start(MQTTClient_run, handle);
@@ -747,88 +745,11 @@ int MQTTClient_connectURI(MQTTClient handle, MQTTClient_connectOptions* options,
 		MQTTClient_sleep(100L);
 	}
 
-	m->c->keepAliveInterval = options->keepAliveInterval;
-	m->c->cleansession = options->cleansession;
-	m->c->maxInflightMessages = (options->reliable) ? 1 : 10;
-
-	if (m->c->will)
-	{
-		free(m->c->will->msg);
-		free(m->c->will->topic);
-		free(m->c->will);
-		m->c->will = NULL;
-	}
-
-	if (options->will && options->will->struct_version == 0)
-	{
-		m->c->will = malloc(sizeof(willMessages));
-		m->c->will->msg = malloc(strlen(options->will->message) + 1); 
-		strcpy(m->c->will->msg, options->will->message);
-		m->c->will->qos = options->will->qos;
-		m->c->will->retained = options->will->retained;
-		m->c->will->topic = malloc(strlen(options->will->topicName) + 1);
-		strcpy(m->c->will->topic, options->will->topicName);
-	}
-	
+	Log(TRACE_MIN, -1, "Connecting to serverURI %s with MQTT version %d", serverURI, MQTTVersion);
 #if defined(OPENSSL)
-	if (m->c->sslopts)
-	{
-		if (m->c->sslopts->trustStore)
-			free(m->c->sslopts->trustStore);
-		if (m->c->sslopts->keyStore)
-			free(m->c->sslopts->keyStore);
-		if (m->c->sslopts->privateKey)
-			free(m->c->sslopts->privateKey);
-		if (m->c->sslopts->privateKeyPassword)
-			free(m->c->sslopts->privateKeyPassword);
-		if (m->c->sslopts->enabledCipherSuites)
-			free(m->c->sslopts->enabledCipherSuites);
-		free(m->c->sslopts);
-		m->c->sslopts = NULL;
-	}
-
-	if (options->struct_version != 0 && options->ssl)
-	{
-		m->c->sslopts = malloc(sizeof(MQTTClient_SSLOptions));
-		memset(m->c->sslopts, '\0', sizeof(MQTTClient_SSLOptions));
-		if (options->ssl->trustStore)
-		{
-			m->c->sslopts->trustStore = malloc(strlen(options->ssl->trustStore) + 1);
-			strcpy(m->c->sslopts->trustStore, options->ssl->trustStore); 
-		}
-		if (options->ssl->keyStore)
-		{
-			m->c->sslopts->keyStore = malloc(strlen(options->ssl->keyStore) + 1);
-			strcpy(m->c->sslopts->keyStore, options->ssl->keyStore);
-		}
-		if (options->ssl->privateKey)
-		{
-			m->c->sslopts->privateKey = malloc(strlen(options->ssl->privateKey) + 1);
-			strcpy(m->c->sslopts->privateKey, options->ssl->privateKey);
-		}
-		if (options->ssl->privateKeyPassword)
-		{
-			m->c->sslopts->privateKeyPassword = malloc(strlen(options->ssl->privateKeyPassword) + 1);
-			strcpy(m->c->sslopts->privateKeyPassword, options->ssl->privateKeyPassword);
-		}
-		if (options->ssl->enabledCipherSuites)
-		{
-			m->c->sslopts->enabledCipherSuites = malloc(strlen(options->ssl->enabledCipherSuites) + 1);
-			strcpy(m->c->sslopts->enabledCipherSuites, options->ssl->enabledCipherSuites);
-		}
-		m->c->sslopts->enableServerCertAuth = options->ssl->enableServerCertAuth;
-	}
-#endif
-
-	m->c->username = options->username;
-	m->c->password = options->password;
-	m->c->retryInterval = options->retryInterval;
-
-	Log(TRACE_MIN, -1, "Connecting to serverURI %s", serverURI);
-#if defined(OPENSSL)
-	rc = MQTTProtocol_connect(serverURI, m->c, m->ssl);
+	rc = MQTTProtocol_connect(serverURI, m->c, m->ssl, MQTTVersion);
 #else
-	rc = MQTTProtocol_connect(serverURI, m->c);
+	rc = MQTTProtocol_connect(serverURI, m->c, MQTTVersion);
 #endif
 	if (rc == SOCKET_ERROR)
 		goto exit;
@@ -866,10 +787,11 @@ int MQTTClient_connectURI(MQTTClient handle, MQTTClient_connectOptions* options,
 					rc = SOCKET_ERROR;
 					goto exit;
 				}
-				else if (rc == 1) {
+				else if (rc == 1) 
+        {
 					rc = MQTTCLIENT_SUCCESS;
 					m->c->connect_state = 3;
-					if (MQTTPacket_send_connect(m->c) == SOCKET_ERROR)
+					if (MQTTPacket_send_connect(m->c, MQTTVersion) == SOCKET_ERROR)
 					{
 						rc = SOCKET_ERROR;
 						goto exit;
@@ -888,7 +810,7 @@ int MQTTClient_connectURI(MQTTClient handle, MQTTClient_connectOptions* options,
 		{
 #endif
 			m->c->connect_state = 3; /* TCP connect completed, in which case send the MQTT connect packet */
-			if (MQTTPacket_send_connect(m->c) == SOCKET_ERROR)
+			if (MQTTPacket_send_connect(m->c, MQTTVersion) == SOCKET_ERROR)
 			{
 				rc = SOCKET_ERROR;
 				goto exit;
@@ -912,7 +834,7 @@ int MQTTClient_connectURI(MQTTClient handle, MQTTClient_connectOptions* options,
 		if(!m->c->cleansession && m->c->session == NULL)
 			m->c->session = SSL_get1_session(m->c->net.ssl);
 		m->c->connect_state = 3; /* TCP connect completed, in which case send the MQTT connect packet */
-		if (MQTTPacket_send_connect(m->c) == SOCKET_ERROR)
+		if (MQTTPacket_send_connect(m->c, MQTTVersion) == SOCKET_ERROR)
 		{
 			rc = SOCKET_ERROR;
 			goto exit;
@@ -963,10 +885,106 @@ exit:
 	if (rc != MQTTCLIENT_SUCCESS)
 	{
 		Thread_unlock_mutex(mqttclient_mutex);
-		MQTTClient_disconnect(handle, 0); /* not "internal" because we don't want to call connection lost */
+    MQTTClient_disconnect1(handle, 0, 0, (MQTTVersion == 3)); /* not "internal" because we don't want to call connection lost */
 		Thread_lock_mutex(mqttclient_mutex);
 	}
 	FUNC_EXIT_RC(rc);
+  return rc;
+}
+
+
+int MQTTClient_connectURI(MQTTClient handle, MQTTClient_connectOptions* options, char* serverURI)
+{
+	MQTTClients* m = handle;
+	START_TIME_TYPE start;
+	long millisecsTimeout = 30000L;
+	int rc = SOCKET_ERROR;
+
+	FUNC_ENTRY;
+	millisecsTimeout = options->connectTimeout * 1000;
+	start = MQTTClient_start_clock();
+
+	m->c->keepAliveInterval = options->keepAliveInterval;
+	m->c->cleansession = options->cleansession;
+	m->c->maxInflightMessages = (options->reliable) ? 1 : 10;
+
+	if (m->c->will)
+	{
+		free(m->c->will->msg);
+		free(m->c->will->topic);
+		free(m->c->will);
+		m->c->will = NULL;
+	}
+
+	if (options->will && options->will->struct_version == 0)
+	{
+		m->c->will = malloc(sizeof(willMessages));
+		m->c->will->msg = malloc(strlen(options->will->message) + 1); 
+		strcpy(m->c->will->msg, options->will->message);
+		m->c->will->qos = options->will->qos;
+		m->c->will->retained = options->will->retained;
+		m->c->will->topic = malloc(strlen(options->will->topicName) + 1);
+		strcpy(m->c->will->topic, options->will->topicName);
+	}
+	
+#if defined(OPENSSL)
+	if (m->c->sslopts)
+	{
+		if (m->c->sslopts->trustStore)
+			free(m->c->sslopts->trustStore);
+		if (m->c->sslopts->keyStore)
+			free(m->c->sslopts->keyStore);
+		if (m->c->sslopts->privateKey)
+			free(m->c->sslopts->privateKey);
+		if (m->c->sslopts->privateKeyPassword)
+			free(m->c->sslopts->privateKeyPassword);
+		if (m->c->sslopts->enabledCipherSuites)
+			free(m->c->sslopts->enabledCipherSuites);
+		free(m->c->sslopts);
+		m->c->sslopts = NULL;
+	}
+
+	if (options->struct_version != 0 && options->ssl)
+	{
+		m->c->sslopts = malloc(sizeof(MQTTClient_SSLOptions));
+		memset(m->c->sslopts, '\0', sizeof(MQTTClient_SSLOptions));
+		if (options->ssl->trustStore)
+		{
+			m->c->sslopts->trustStore = malloc(strlen(options->ssl->trustStore) + 1);
+			strcpy(m->c->sslopts->trustStore, options->ssl->trustStore); 
+		}
+		if (options->ssl->keyStore)
+		{
+			m->c->sslopts->keyStore = malloc(strlen(options->ssl->keyStore) + 1);
+			strcpy(m->c->sslopts->keyStore, options->ssl->keyStore);
+		}
+		if (options->ssl->privateKey)
+		{
+			m->c->sslopts->privateKey = malloc(strlen(options->ssl->privateKey) + 1);
+			strcpy(m->c->sslopts->privateKey, options->ssl->privateKey);
+		}
+		if (options->ssl->privateKeyPassword)
+		{
+			m->c->sslopts->privateKeyPassword = malloc(strlen(options->ssl->privateKeyPassword) + 1);
+			strcpy(m->c->sslopts->privateKeyPassword, options->ssl->privateKeyPassword);
+		}
+		if (options->ssl->enabledCipherSuites)
+		{
+			m->c->sslopts->enabledCipherSuites = malloc(strlen(options->ssl->enabledCipherSuites) + 1);
+			strcpy(m->c->sslopts->enabledCipherSuites, options->ssl->enabledCipherSuites);
+		}
+		m->c->sslopts->enableServerCertAuth = options->ssl->enableServerCertAuth;
+	}
+#endif
+
+	m->c->username = options->username;
+	m->c->password = options->password;
+	m->c->retryInterval = options->retryInterval;
+
+  if ((rc = MQTTClient_connectURIVersion(handle, options, serverURI, 4	, start, millisecsTimeout)) != MQTTCLIENT_SUCCESS)
+    rc = MQTTClient_connectURIVersion(handle, options, serverURI, 3, start, millisecsTimeout);
+
+	FUNC_EXIT_RC(rc);
 	return rc;
 }
 
@@ -1055,7 +1073,7 @@ exit:
 }
 
 
-int MQTTClient_disconnect1(MQTTClient handle, int timeout, int internal)
+int MQTTClient_disconnect1(MQTTClient handle, int timeout, int internal, int stop)
 {
 	MQTTClients* m = handle;
 	START_TIME_TYPE start;
@@ -1095,14 +1113,15 @@ int MQTTClient_disconnect1(MQTTClient handle, int timeout, int internal)
 	if (Thread_check_sem(m->connect_sem))
 		Thread_post_sem(m->connect_sem);
 	if (Thread_check_sem(m->connack_sem))
-		Thread_post_sem(m->connack_sem);
+    Thread_post_sem(m->connect_sem);
 	if (Thread_check_sem(m->suback_sem))
 		Thread_post_sem(m->suback_sem);
 	if (Thread_check_sem(m->unsuback_sem))
 		Thread_post_sem(m->unsuback_sem);
 
 exit:
-	MQTTClient_stop();
+  if (stop)
+  	MQTTClient_stop();
 	if (internal && m->cl && was_connected)
 	{
 		Log(TRACE_MIN, -1, "Calling connectionLost for client %s", m->c->clientID);
@@ -1116,7 +1135,7 @@ exit:
 
 int MQTTClient_disconnect_internal(MQTTClient handle, int timeout)
 {
-	return MQTTClient_disconnect1(handle, timeout, 1);
+	return MQTTClient_disconnect1(handle, timeout, 1, 1);
 }
 
 
@@ -1128,7 +1147,7 @@ void MQTTProtocol_closeSession(Clients* c, int sendwill)
 
 int MQTTClient_disconnect(MQTTClient handle, int timeout)
 {
-	return MQTTClient_disconnect1(handle, timeout, 0);
+	return MQTTClient_disconnect1(handle, timeout, 0, 1);
 }
 
 
@@ -1550,7 +1569,7 @@ MQTTPacket* MQTTClient_waitfor(MQTTClient handle, int packet_type, int* rc, long
 		else if (packet_type == UNSUBACK)
 			*rc = Thread_wait_sem(m->unsuback_sem, timeout);
 		if (*rc == 0 && packet_type != CONNECT && m->pack == NULL)
-			Log(TRACE_MIN, -1, "waitfor unexpectedly is NULL for client %s, packet_type %d", m->c->clientID, packet_type);
+			Log(TRACE_MIN, -1, "waitfor unexpectedly is NULL for client %s, packet_type %d, timeout %ld", m->c->clientID, packet_type, timeout);
 		pack = m->pack;
 	}
 	else

src/MQTTPacket.c

@@ -518,7 +518,7 @@ void MQTTPacket_freePublish(Publish* pack)
  * @param type the MQTT packet type e.g. SUBACK
  * @param msgid the MQTT message id to use
  * @param dup boolean - whether to set the MQTT DUP flag
- * @param socket the open socket to send the data to
+ * @param net the network handle to send the data to
  * @return the completion code (e.g. TCPSOCKET_COMPLETE)
  */
 int MQTTPacket_send_ack(int type, int msgid, int dup, networkHandles *net)
@@ -532,6 +532,8 @@ int MQTTPacket_send_ack(int type, int msgid, int dup, networkHandles *net)
 	header.byte = 0;
 	header.bits.type = type;
 	header.bits.dup = dup;
+  if (type == PUBREL)
+    header.bits.qos = 1;
 	writeInt(&ptr, msgid);
 	if ((rc = MQTTPacket_send(net, header, buf, 2)) != TCPSOCKET_INTERRUPTED)
 		free(buf);

src/MQTTPacketOut.c

@@ -36,20 +36,20 @@
 /**
  * Send an MQTT CONNECT packet down a socket.
  * @param client a structure from which to get all the required values
+ * @param MQTTVersion the MQTT version to connect with
  * @return the completion code (e.g. TCPSOCKET_COMPLETE)
  */
-int MQTTPacket_send_connect(Clients* client)
+int MQTTPacket_send_connect(Clients* client, int MQTTVersion)
 {
 	char *buf, *ptr;
 	Connect packet;
-	int rc, len;
+	int rc = -1, len;
 
 	FUNC_ENTRY;
 	packet.header.byte = 0;
 	packet.header.bits.type = CONNECT;
-	packet.header.bits.qos = 1;
 
-	len = 12 + strlen(client->clientID)+2;
+	len = ((MQTTVersion == 3) ? 12 : 10) + strlen(client->clientID)+2;
 	if (client->will)
 		len += strlen(client->will->topic)+2 + strlen(client->will->msg)+2;
 	if (client->username)
@@ -58,8 +58,18 @@ int MQTTPacket_send_connect(Clients* client)
 		len += strlen(client->password)+2;
 
 	ptr = buf = malloc(len);
-	writeUTF(&ptr, "MQIsdp");
-	writeChar(&ptr, (char)3);
+  if (MQTTVersion == 3)
+  {
+  	writeUTF(&ptr, "MQIsdp");
+	  writeChar(&ptr, (char)3);
+  }
+  else if (MQTTVersion == 4)
+  {
+  	writeUTF(&ptr, "MQTT");
+	  writeChar(&ptr, (char)4);
+  }
+  else
+    goto exit;
 
 	packet.flags.all = 0;
 	packet.flags.bits.cleanstart = client->cleansession;
@@ -90,6 +100,7 @@ int MQTTPacket_send_connect(Clients* client)
 
 	rc = MQTTPacket_send(&client->net, packet.header, buf, len);
 	Log(LOG_PROTOCOL, 0, NULL, client->net.socket, client->clientID, client->cleansession, rc);
+exit:
 	free(buf);
 	FUNC_EXIT_RC(rc);
 	return rc;

src/MQTTPacketOut.h

@@ -20,7 +20,7 @@
 
 #include "MQTTPacket.h"
 
-int MQTTPacket_send_connect(Clients* client);
+int MQTTPacket_send_connect(Clients* client, int MQTTVersion);
 void* MQTTPacket_connack(unsigned char aHeader, char* data, int datalen);
 
 int MQTTPacket_send_pingreq(networkHandles* net, char* clientID);

src/MQTTProtocolOut.c

@@ -77,18 +77,15 @@ char* MQTTProtocol_addressPort(char* ip_address, int* port)
 /**
  * MQTT outgoing connect processing for a client
  * @param ip_address the TCP address:port to connect to
- * @param clientID the MQTT client id to use
- * @param cleansession MQTT cleansession flag
- * @param keepalive MQTT keepalive timeout in seconds
- * @param willMessage pointer to the will message to be used, if any
- * @param username MQTT 3.1 username, or NULL
- * @param password MQTT 3.1 password, or NULL
- * @return the new client structure
+ * @param aClient a structure with all MQTT data needed
+ * @param int ssl
+ * @param int MQTTVersion the MQTT version to connect with (3 or 4)
+ * @return return code
  */
 #if defined(OPENSSL)
-int MQTTProtocol_connect(char* ip_address, Clients* aClient, int ssl)
+int MQTTProtocol_connect(char* ip_address, Clients* aClient, int ssl, int MQTTVersion)
 #else
-int MQTTProtocol_connect(char* ip_address, Clients* aClient)
+  int MQTTProtocol_connect(char* ip_address, Clients* aClient, int MQTTVersion)
 #endif
 {
 	int rc, port;
@@ -120,7 +117,7 @@ int MQTTProtocol_connect(char* ip_address, Clients* aClient)
 		if (rc == 0)
 		{
 			/* Now send the MQTT connect packet */
-			if ((rc = MQTTPacket_send_connect(aClient)) == 0)
+		  if ((rc = MQTTPacket_send_connect(aClient, MQTTVersion)) == 0)
 				aClient->connect_state = 3; /* MQTT Connect sent - wait for CONNACK */ 
 			else
 				aClient->connect_state = 0;

src/MQTTProtocolOut.h

@@ -30,9 +30,9 @@
 
 void MQTTProtocol_reconnect(char* ip_address, Clients* client);
 #if defined(OPENSSL)
-int MQTTProtocol_connect(char* ip_address, Clients* acClients, int ssl);
+int MQTTProtocol_connect(char* ip_address, Clients* acClients, int ssl, int MQTTVersion);
 #else
-int MQTTProtocol_connect(char* ip_address, Clients* acClients);
+int MQTTProtocol_connect(char* ip_address, Clients* acClients, int MQTTVersion);
 #endif
 int MQTTProtocol_handlePingresps(void* pack, int sock);
 int MQTTProtocol_subscribe(Clients* client, List* topics, List* qoss);

test/test1.c

@@ -948,6 +948,17 @@ int test6_socket_close(int socket)
 	return rc;
 }
 
+typedef struct
+{
+	int socket;
+	time_t lastContact;
+#if defined(OPENSSL)
+	SSL* ssl;
+	SSL_CTX* ctx;
+#endif
+} networkHandles;
+
+
 typedef struct
 {
 	char* clientID;					/**< the string id of the client */
@@ -957,22 +968,12 @@ typedef struct
 	unsigned int connected : 1;		/**< whether it is currently connected */
 	unsigned int good : 1; 			/**< if we have an error on the socket we turn this off */
 	unsigned int ping_outstanding : 1;
-	unsigned int connect_state : 2;
-	int socket;
-	int msgID;
-	int keepAliveInterval;
-	int retryInterval;
-	int maxInflightMessages;
-	time_t lastContact;
-	void* will;
-	void* inboundMsgs;
-	void* outboundMsgs;				/**< in flight */
-	void* messageQueue;
-	void* phandle;  /* the persistence handle */
-	MQTTClient_persistence* persistence; /* a persistence implementation */
-	int connectOptionsVersion;
+	int connect_state : 4;
+	networkHandles net;
+/* ... */
 } Clients;
 
+
 typedef struct
 {
 	char* serverURI;
@@ -1056,7 +1057,7 @@ int test6(struct Options options)
 	/* now send the command which will break the connection and cause the will message to be sent */
 	/*rc = MQTTClient_publish(test6_c1, mqttsas_topic, strlen("TERMINATE"), "TERMINATE", 0, 0, NULL);
 	assert("Good rc from publish", rc == MQTTCLIENT_SUCCESS, "rc was %d\n", rc);*/
-  	test6_socket_close(((MQTTClients*)test6_c1)->c->socket); 
+  test6_socket_close(((MQTTClients*)test6_c1)->c->net.socket); 
 
 	MyLog(LOGA_INFO, "Waiting to receive the will message");
 	count = 0;

test/test4.c

@@ -1094,7 +1094,6 @@ int test6(struct Options options)
 		#else
 			usleep(10000L);
 		#endif		
-	
 
 	MQTTAsync_destroy(&cinfo.c);