Diff: aws-iot/src/aws_iot_mqtt_client.cpp

Revision:

0:cd5404401c2f

Child:

2:db59f45ad3d3

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/aws-iot/src/aws_iot_mqtt_client.cpp	Wed Apr 12 14:07:09 2017 +0200
@@ -0,0 +1,310 @@
+/*
+* Copyright 2015-2016 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+*
+* Licensed under the Apache License, Version 2.0 (the "License").
+* You may not use this file except in compliance with the License.
+* A copy of the License is located at
+*
+* http://aws.amazon.com/apache2.0
+*
+* or in the "license" file accompanying this file. This file is distributed
+* on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+* express or implied. See the License for the specific language governing
+* permissions and limitations under the License.
+*/
+
+// Based on Eclipse Paho.
+/*******************************************************************************
+ * Copyright (c) 2014 IBM Corp.
+ *
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * and Eclipse Distribution License v1.0 which accompany this distribution.
+ *
+ * The Eclipse Public License is available at
+ *    http://www.eclipse.org/legal/epl-v10.html
+ * and the Eclipse Distribution License is available at
+ *   http://www.eclipse.org/org/documents/edl-v10.php.
+ *
+ * Contributors:
+ *    Allan Stockdill-Mander/Ian Craggs - initial API and implementation and/or initial documentation
+ *******************************************************************************/
+
+/**
+ * @file aws_iot_mqtt_client.c
+ * @brief MQTT client API definitions
+ */
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#include "aws_iot_log.h"
+#include "aws_iot_mqtt_client_interface.h"
+
+#ifdef _ENABLE_THREAD_SUPPORT_
+#include "threads_interface.h"
+#endif
+
+const IoT_Client_Init_Params iotClientInitParamsDefault = IoT_Client_Init_Params_initializer;
+const IoT_MQTT_Will_Options iotMqttWillOptionsDefault = IoT_MQTT_Will_Options_Initializer;
+const IoT_Client_Connect_Params iotClientConnectParamsDefault = IoT_Client_Connect_Params_initializer;
+
+ClientState aws_iot_mqtt_get_client_state(AWS_IoT_Client *pClient) {
+	FUNC_ENTRY;
+	if(NULL == pClient) {
+		return CLIENT_STATE_INVALID;
+	}
+
+	FUNC_EXIT_RC(pClient->clientStatus.clientState);
+}
+
+#ifdef _ENABLE_THREAD_SUPPORT_
+IoT_Error_t aws_iot_mqtt_client_lock_mutex(AWS_IoT_Client *pClient, IoT_Mutex_t *pMutex) {
+	FUNC_ENTRY;
+	IoT_Error_t threadRc = FAILURE;
+
+	if(NULL == pClient || NULL == pMutex){
+		FUNC_EXIT_RC(NULL_VALUE_ERROR);
+	}
+
+	if(false == pClient->clientData.isBlockOnThreadLockEnabled) {
+		threadRc = aws_iot_thread_mutex_trylock(pMutex);
+	} else {
+		threadRc = aws_iot_thread_mutex_lock(pMutex);
+		/* Should never return Error because the above request blocks until lock is obtained */
+	}
+
+	if(SUCCESS != threadRc) {
+		FUNC_EXIT_RC(threadRc);
+	}
+
+	FUNC_EXIT_RC(SUCCESS);
+}
+
+IoT_Error_t aws_iot_mqtt_client_unlock_mutex(AWS_IoT_Client *pClient, IoT_Mutex_t *pMutex) {
+	if(NULL == pClient || NULL == pMutex) {
+		return NULL_VALUE_ERROR;
+	}
+	IOT_UNUSED(pClient);
+	return aws_iot_thread_mutex_unlock(pMutex);
+}
+#endif
+
+IoT_Error_t aws_iot_mqtt_set_client_state(AWS_IoT_Client *pClient, ClientState expectedCurrentState,
+										  ClientState newState) {
+	IoT_Error_t rc;
+#ifdef _ENABLE_THREAD_SUPPORT_
+	IoT_Error_t threadRc = FAILURE;
+#endif
+
+	FUNC_ENTRY;
+	if(NULL == pClient) {
+		FUNC_EXIT_RC(NULL_VALUE_ERROR);
+	}
+
+#ifdef _ENABLE_THREAD_SUPPORT_
+	rc = aws_iot_mqtt_client_lock_mutex(pClient, &(pClient->clientData.state_change_mutex));
+	if(SUCCESS != rc) {
+		return rc;
+	}
+#endif
+	if(expectedCurrentState == aws_iot_mqtt_get_client_state(pClient)) {
+		pClient->clientStatus.clientState = newState;
+		rc = SUCCESS;
+	} else {
+		rc = MQTT_UNEXPECTED_CLIENT_STATE_ERROR;
+	}
+
+#ifdef _ENABLE_THREAD_SUPPORT_
+	threadRc = aws_iot_mqtt_client_unlock_mutex(pClient, &(pClient->clientData.state_change_mutex));
+	if(SUCCESS == rc && SUCCESS != threadRc) {
+		rc = threadRc;
+	}
+#endif
+
+	FUNC_EXIT_RC(rc);
+}
+
+IoT_Error_t aws_iot_mqtt_set_connect_params(AWS_IoT_Client *pClient, IoT_Client_Connect_Params *pNewConnectParams) {
+	FUNC_ENTRY;
+	if(NULL == pClient || NULL == pNewConnectParams) {
+		FUNC_EXIT_RC(NULL_VALUE_ERROR);
+	}
+
+	pClient->clientData.options.isWillMsgPresent = pNewConnectParams->isWillMsgPresent;
+	pClient->clientData.options.MQTTVersion = pNewConnectParams->MQTTVersion;
+	pClient->clientData.options.pClientID = pNewConnectParams->pClientID;
+	pClient->clientData.options.clientIDLen = pNewConnectParams->clientIDLen;
+	pClient->clientData.options.pUsername = pNewConnectParams->pUsername;
+	pClient->clientData.options.usernameLen = pNewConnectParams->usernameLen;
+	pClient->clientData.options.pPassword = pNewConnectParams->pUsername;
+	pClient->clientData.options.passwordLen = pNewConnectParams->passwordLen;
+	pClient->clientData.options.will.pTopicName = pNewConnectParams->will.pTopicName;
+	pClient->clientData.options.will.topicNameLen = pNewConnectParams->will.topicNameLen;
+	pClient->clientData.options.will.pMessage = pNewConnectParams->will.pMessage;
+	pClient->clientData.options.will.msgLen = pNewConnectParams->will.msgLen;
+	pClient->clientData.options.will.qos = pNewConnectParams->will.qos;
+	pClient->clientData.options.will.isRetained = pNewConnectParams->will.isRetained;
+	pClient->clientData.options.keepAliveIntervalInSec = pNewConnectParams->keepAliveIntervalInSec;
+	pClient->clientData.options.isCleanSession = pNewConnectParams->isCleanSession;
+
+	FUNC_EXIT_RC(SUCCESS);
+}
+
+IoT_Error_t aws_iot_mqtt_init(AWS_IoT_Client *pClient, IoT_Client_Init_Params *pInitParams) {
+	uint32_t i;
+	IoT_Error_t rc;
+	IoT_Client_Connect_Params default_options = IoT_Client_Connect_Params_initializer;
+
+	FUNC_ENTRY;
+
+	if(NULL == pClient || NULL == pInitParams || NULL == pInitParams->pHostURL || 0 == pInitParams->port /*||
+	   NULL == pInitParams->pRootCALocation || NULL == pInitParams->pDevicePrivateKeyLocation ||
+	   NULL == pInitParams->pDeviceCertLocation*/) {
+		FUNC_EXIT_RC(NULL_VALUE_ERROR);
+	}
+
+	for(i = 0; i < AWS_IOT_MQTT_NUM_SUBSCRIBE_HANDLERS; ++i) {
+		pClient->clientData.messageHandlers[i].topicName = NULL;
+		pClient->clientData.messageHandlers[i].pApplicationHandler = NULL;
+		pClient->clientData.messageHandlers[i].pApplicationHandlerData = NULL;
+		pClient->clientData.messageHandlers[i].qos = QOS0;
+	}
+
+	pClient->clientData.packetTimeoutMs = pInitParams->mqttPacketTimeout_ms;
+	pClient->clientData.commandTimeoutMs = pInitParams->mqttCommandTimeout_ms;
+	pClient->clientData.writeBufSize = AWS_IOT_MQTT_TX_BUF_LEN;
+	pClient->clientData.readBufSize = AWS_IOT_MQTT_RX_BUF_LEN;
+	pClient->clientData.counterNetworkDisconnected = 0;
+	pClient->clientData.disconnectHandler = pInitParams->disconnectHandler;
+	pClient->clientData.disconnectHandlerData = pInitParams->disconnectHandlerData;
+	pClient->clientData.nextPacketId = 1;
+
+	/* Initialize default connection options */
+	rc = aws_iot_mqtt_set_connect_params(pClient, &default_options);
+	if(SUCCESS != rc) {
+		FUNC_EXIT_RC(rc);
+	}
+
+#ifdef _ENABLE_THREAD_SUPPORT_
+	pClient->clientData.isBlockOnThreadLockEnabled = pInitParams->isBlockOnThreadLockEnabled;
+	rc = aws_iot_thread_mutex_init(&(pClient->clientData.state_change_mutex));
+	if(SUCCESS != rc) {
+		FUNC_EXIT_RC(rc);
+	}
+	rc = aws_iot_thread_mutex_init(&(pClient->clientData.tls_read_mutex));
+	if(SUCCESS != rc) {
+		FUNC_EXIT_RC(rc);
+	}
+	rc = aws_iot_thread_mutex_init(&(pClient->clientData.tls_write_mutex));
+	if(SUCCESS != rc) {
+		FUNC_EXIT_RC(rc);
+	}
+#endif
+
+	pClient->clientStatus.isPingOutstanding = 0;
+	pClient->clientStatus.isAutoReconnectEnabled = pInitParams->enableAutoReconnect;
+
+	rc = iot_tls_init(&(pClient->networkStack), pInitParams->pRootCALocation, pInitParams->pDeviceCertLocation,
+					  pInitParams->pDevicePrivateKeyLocation, pInitParams->pHostURL, pInitParams->port,
+					  pInitParams->tlsHandshakeTimeout_ms, pInitParams->isSSLHostnameVerify);
+
+	if(SUCCESS != rc) {
+		pClient->clientStatus.clientState = CLIENT_STATE_INVALID;
+		FUNC_EXIT_RC(rc);
+	}
+
+	init_timer(&(pClient->pingTimer));
+	init_timer(&(pClient->reconnectDelayTimer));
+
+	pClient->clientStatus.clientState = CLIENT_STATE_INITIALIZED;
+
+	FUNC_EXIT_RC(SUCCESS);
+}
+
+uint16_t aws_iot_mqtt_get_next_packet_id(AWS_IoT_Client *pClient) {
+	return pClient->clientData.nextPacketId = (uint16_t) ((MAX_PACKET_ID == pClient->clientData.nextPacketId) ? 1 : (
+			pClient->clientData.nextPacketId + 1));
+}
+
+bool aws_iot_mqtt_is_client_connected(AWS_IoT_Client *pClient) {
+	bool isConnected;
+
+	FUNC_ENTRY;
+
+	if(NULL == pClient) {
+		IOT_WARN(" Client is null! ");
+		FUNC_EXIT_RC(false);
+	}
+
+	switch(pClient->clientStatus.clientState) {
+		case CLIENT_STATE_INVALID:
+		case CLIENT_STATE_INITIALIZED:
+		case CLIENT_STATE_CONNECTING:
+			isConnected = false;
+			break;
+		case CLIENT_STATE_CONNECTED_IDLE:
+		case CLIENT_STATE_CONNECTED_YIELD_IN_PROGRESS:
+		case CLIENT_STATE_CONNECTED_PUBLISH_IN_PROGRESS:
+		case CLIENT_STATE_CONNECTED_SUBSCRIBE_IN_PROGRESS:
+		case CLIENT_STATE_CONNECTED_UNSUBSCRIBE_IN_PROGRESS:
+		case CLIENT_STATE_CONNECTED_RESUBSCRIBE_IN_PROGRESS:
+		case CLIENT_STATE_CONNECTED_WAIT_FOR_CB_RETURN:
+			isConnected = true;
+			break;
+		case CLIENT_STATE_DISCONNECTING:
+		case CLIENT_STATE_DISCONNECTED_ERROR:
+		case CLIENT_STATE_DISCONNECTED_MANUALLY:
+		case CLIENT_STATE_PENDING_RECONNECT:
+		default:
+			isConnected = false;
+			break;
+	}
+
+	FUNC_EXIT_RC(isConnected);
+}
+
+bool aws_iot_is_autoreconnect_enabled(AWS_IoT_Client *pClient) {
+	FUNC_ENTRY;
+	if(NULL == pClient) {
+		IOT_WARN(" Client is null! ");
+		FUNC_EXIT_RC(false);
+	}
+
+	FUNC_EXIT_RC(pClient->clientStatus.isAutoReconnectEnabled);
+}
+
+IoT_Error_t aws_iot_mqtt_autoreconnect_set_status(AWS_IoT_Client *pClient, bool newStatus) {
+	FUNC_ENTRY;
+	if(NULL == pClient) {
+		FUNC_EXIT_RC(NULL_VALUE_ERROR);
+	}
+	pClient->clientStatus.isAutoReconnectEnabled = newStatus;
+	FUNC_EXIT_RC(SUCCESS);
+}
+
+IoT_Error_t aws_iot_mqtt_set_disconnect_handler(AWS_IoT_Client *pClient, iot_disconnect_handler pDisconnectHandler,
+												void *pDisconnectHandlerData) {
+	FUNC_ENTRY;
+	if(NULL == pClient || NULL == pDisconnectHandler) {
+		FUNC_EXIT_RC(NULL_VALUE_ERROR);
+	}
+
+	pClient->clientData.disconnectHandler = pDisconnectHandler;
+	pClient->clientData.disconnectHandlerData = pDisconnectHandlerData;
+	FUNC_EXIT_RC(SUCCESS);
+}
+
+uint32_t aws_iot_mqtt_get_network_disconnected_count(AWS_IoT_Client *pClient) {
+	return pClient->clientData.counterNetworkDisconnected;
+}
+
+void aws_iot_mqtt_reset_network_disconnected_count(AWS_IoT_Client *pClient) {
+	pClient->clientData.counterNetworkDisconnected = 0;
+}
+
+#ifdef __cplusplus
+}
+#endif
+