7041c014bb
* Added a MQTT pub/sub example of using AWS IoT (via ECC based TLS1.2 connection). * Fixed a buffer overflow issue when receiving large MQTT packet. * Reset TLS connection on read/write errors.
564 lines
16 KiB
C
564 lines
16 KiB
C
/*******************************************************************************
|
|
* 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
|
|
*******************************************************************************/
|
|
#include <espressif/esp_common.h>
|
|
#include <lwip/arch.h>
|
|
#include "MQTTClient.h"
|
|
|
|
void NewMessageData(MessageData* md, MQTTString* aTopicName, MQTTMessage* aMessgage) {
|
|
md->topic = aTopicName;
|
|
md->message = aMessgage;
|
|
}
|
|
|
|
|
|
int getNextPacketId(MQTTClient *c) {
|
|
return c->next_packetid = (c->next_packetid == MAX_PACKET_ID) ? 1 : c->next_packetid + 1;
|
|
}
|
|
|
|
|
|
int sendPacket(MQTTClient* c, int length, Timer* timer)
|
|
{
|
|
int rc = FAILURE,
|
|
sent = 0;
|
|
|
|
while (sent < length && !expired(timer))
|
|
{
|
|
rc = c->ipstack->mqttwrite(c->ipstack, &c->buf[sent], length - sent, left_ms(timer));
|
|
if (rc < 0) // there was an error writing the data
|
|
break;
|
|
sent += rc;
|
|
}
|
|
if (sent == length)
|
|
{
|
|
countdown(&(c->ping_timer), c->keepAliveInterval); // record the fact that we have successfully sent the packet
|
|
rc = SUCCESS;
|
|
}
|
|
else
|
|
rc = FAILURE;
|
|
return rc;
|
|
}
|
|
|
|
|
|
int decodePacket(MQTTClient* c, int* value, int timeout)
|
|
{
|
|
unsigned char i;
|
|
int multiplier = 1;
|
|
int len = 0;
|
|
const int MAX_NO_OF_REMAINING_LENGTH_BYTES = 4;
|
|
|
|
*value = 0;
|
|
do
|
|
{
|
|
int rc = MQTTPACKET_READ_ERROR;
|
|
|
|
if (++len > MAX_NO_OF_REMAINING_LENGTH_BYTES)
|
|
{
|
|
rc = MQTTPACKET_READ_ERROR; /* bad data */
|
|
goto exit;
|
|
}
|
|
rc = c->ipstack->mqttread(c->ipstack, &i, 1, timeout);
|
|
if (rc != 1)
|
|
{
|
|
goto exit;
|
|
}
|
|
*value += (i & 127) * multiplier;
|
|
multiplier *= 128;
|
|
} while ((i & 128) != 0);
|
|
exit:
|
|
return len;
|
|
}
|
|
|
|
|
|
// Return packet type. If no packet avilable, return FAILURE, or READ_ERROR if timeout
|
|
int readPacket(MQTTClient* c, Timer* timer)
|
|
{
|
|
int rc = FAILURE;
|
|
MQTTHeader header = {0};
|
|
int len = 0;
|
|
int rem_len = 0;
|
|
|
|
/* 1. read the header byte. This has the packet type in it */
|
|
if (c->ipstack->mqttread(c->ipstack, c->readbuf, 1, left_ms(timer)) != 1)
|
|
goto exit;
|
|
len = 1;
|
|
/* 2. read the remaining length. This is variable in itself */
|
|
len += decodePacket(c, &rem_len, left_ms(timer));
|
|
if (len <= 1 || len + rem_len > c->readbuf_size) /* if packet is too big to fit in our readbuf, abort */
|
|
{
|
|
rc = READ_ERROR;
|
|
goto exit;
|
|
}
|
|
MQTTPacket_encode(c->readbuf + 1, rem_len); /* put the original remaining length back into the buffer */
|
|
/* 3. read the rest of the buffer using a callback to supply the rest of the data */
|
|
if (rem_len > 0 && (c->ipstack->mqttread(c->ipstack, c->readbuf + len, rem_len, left_ms(timer)) != rem_len))
|
|
{
|
|
rc = READ_ERROR;
|
|
goto exit;
|
|
}
|
|
header.byte = c->readbuf[0];
|
|
rc = header.bits.type;
|
|
exit:
|
|
return rc;
|
|
}
|
|
|
|
|
|
// assume topic filter and name is in correct format
|
|
// # can only be at end
|
|
// + and # can only be next to separator
|
|
char isTopicMatched(char* topicFilter, MQTTString* topicName)
|
|
{
|
|
char* curf = topicFilter;
|
|
char* curn = topicName->lenstring.data;
|
|
char* curn_end = curn + topicName->lenstring.len;
|
|
|
|
while (*curf && curn < curn_end)
|
|
{
|
|
if (*curn == '/' && *curf != '/')
|
|
break;
|
|
if (*curf != '+' && *curf != '#' && *curf != *curn)
|
|
break;
|
|
if (*curf == '+')
|
|
{ // skip until we meet the next separator, or end of string
|
|
char* nextpos = curn + 1;
|
|
while (nextpos < curn_end && *nextpos != '/')
|
|
nextpos = ++curn + 1;
|
|
}
|
|
else if (*curf == '#')
|
|
curn = curn_end - 1; // skip until end of string
|
|
curf++;
|
|
curn++;
|
|
};
|
|
|
|
return (curn == curn_end) && (*curf == '\0');
|
|
}
|
|
|
|
|
|
int deliverMessage(MQTTClient* c, MQTTString* topicName, MQTTMessage* message)
|
|
{
|
|
int i;
|
|
int rc = FAILURE;
|
|
|
|
// we have to find the right message handler - indexed by topic
|
|
for (i = 0; i < MAX_MESSAGE_HANDLERS; ++i)
|
|
{
|
|
if (c->messageHandlers[i].topicFilter != 0 && (MQTTPacket_equals(topicName, (char*)c->messageHandlers[i].topicFilter) ||
|
|
isTopicMatched((char*)c->messageHandlers[i].topicFilter, topicName)))
|
|
{
|
|
if (c->messageHandlers[i].fp != NULL)
|
|
{
|
|
MessageData md;
|
|
NewMessageData(&md, topicName, message);
|
|
c->messageHandlers[i].fp(&md);
|
|
rc = SUCCESS;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (rc == FAILURE && c->defaultMessageHandler != NULL)
|
|
{
|
|
MessageData md;
|
|
NewMessageData(&md, topicName, message);
|
|
c->defaultMessageHandler(&md);
|
|
rc = SUCCESS;
|
|
}
|
|
|
|
return rc;
|
|
}
|
|
|
|
|
|
int keepalive(MQTTClient* c)
|
|
{
|
|
int rc = SUCCESS;
|
|
|
|
if (c->keepAliveInterval == 0)
|
|
{
|
|
rc = SUCCESS;
|
|
goto exit;
|
|
}
|
|
|
|
if (expired(&(c->ping_timer)))
|
|
{
|
|
if (c->ping_outstanding)
|
|
{
|
|
// if ping failure accumulated above MAX_FAIL_ALLOWED, the connection is broken
|
|
++(c->fail_count);
|
|
if (c->fail_count >= MAX_FAIL_ALLOWED)
|
|
{
|
|
rc = DISCONNECTED;
|
|
goto exit;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Timer timer;
|
|
InitTimer(&timer);
|
|
countdown_ms(&timer, 1000);
|
|
c->ping_outstanding = 1;
|
|
int len = MQTTSerialize_pingreq(c->buf, c->buf_size);
|
|
if (len > 0)
|
|
sendPacket(c, len, &timer);
|
|
}
|
|
// re-arm ping counter
|
|
countdown(&(c->ping_timer), c->keepAliveInterval);
|
|
}
|
|
|
|
exit:
|
|
return rc;
|
|
}
|
|
|
|
|
|
int cycle(MQTTClient* c, Timer* timer)
|
|
{
|
|
// read the socket, see what work is due
|
|
int packet_type = readPacket(c, timer);
|
|
|
|
int len = 0,
|
|
rc = SUCCESS;
|
|
|
|
switch (packet_type)
|
|
{
|
|
case CONNACK:
|
|
case PUBACK:
|
|
case SUBACK:
|
|
break;
|
|
case PUBLISH:
|
|
{
|
|
MQTTString topicName;
|
|
MQTTMessage msg;
|
|
if (MQTTDeserialize_publish((unsigned char*)&msg.dup, (int*)&msg.qos, (unsigned char*)&msg.retained, (unsigned short*)&msg.id, &topicName,
|
|
(unsigned char**)&msg.payload, (int*)&msg.payloadlen, c->readbuf, c->readbuf_size) != 1)
|
|
goto exit;
|
|
deliverMessage(c, &topicName, &msg);
|
|
if (msg.qos != QOS0)
|
|
{
|
|
if (msg.qos == QOS1)
|
|
len = MQTTSerialize_ack(c->buf, c->buf_size, PUBACK, 0, msg.id);
|
|
else if (msg.qos == QOS2)
|
|
len = MQTTSerialize_ack(c->buf, c->buf_size, PUBREC, 0, msg.id);
|
|
if (len <= 0)
|
|
rc = FAILURE;
|
|
else
|
|
rc = sendPacket(c, len, timer);
|
|
if (rc == FAILURE)
|
|
goto exit; // there was a problem
|
|
}
|
|
break;
|
|
}
|
|
case PUBREC:
|
|
{
|
|
unsigned short mypacketid;
|
|
unsigned char dup, type;
|
|
if (MQTTDeserialize_ack(&type, &dup, &mypacketid, c->readbuf, c->readbuf_size) != 1)
|
|
rc = FAILURE;
|
|
else if ((len = MQTTSerialize_ack(c->buf, c->buf_size, PUBREL, 0, mypacketid)) <= 0)
|
|
rc = FAILURE;
|
|
else if ((rc = sendPacket(c, len, timer)) != SUCCESS) // send the PUBREL packet
|
|
rc = FAILURE; // there was a problem
|
|
if (rc == FAILURE)
|
|
goto exit; // there was a problem
|
|
break;
|
|
}
|
|
case PUBCOMP:
|
|
break;
|
|
case PINGRESP:
|
|
{
|
|
c->ping_outstanding = 0;
|
|
c->fail_count = 0;
|
|
break;
|
|
}
|
|
case READ_ERROR:
|
|
{
|
|
c->isconnected = 0; // we simulate a disconnect if reading error
|
|
rc = DISCONNECTED; // so that the outer layer will reconnect and recover
|
|
break;
|
|
}
|
|
}
|
|
if (c->isconnected)
|
|
rc = keepalive(c);
|
|
exit:
|
|
if (rc == SUCCESS)
|
|
rc = packet_type;
|
|
return rc;
|
|
}
|
|
|
|
|
|
void NewMQTTClient(MQTTClient* c, Network* network, unsigned int command_timeout_ms, unsigned char* buf, size_t buf_size, unsigned char* readbuf, size_t readbuf_size)
|
|
{
|
|
int i;
|
|
c->ipstack = network;
|
|
|
|
for (i = 0; i < MAX_MESSAGE_HANDLERS; ++i)
|
|
c->messageHandlers[i].topicFilter = 0;
|
|
c->command_timeout_ms = command_timeout_ms;
|
|
c->buf = buf;
|
|
c->buf_size = buf_size;
|
|
c->readbuf = readbuf;
|
|
c->readbuf_size = readbuf_size;
|
|
c->isconnected = 0;
|
|
c->ping_outstanding = 0;
|
|
c->fail_count = 0;
|
|
c->defaultMessageHandler = NULL;
|
|
InitTimer(&(c->ping_timer));
|
|
}
|
|
|
|
|
|
int MQTTYield(MQTTClient* c, int timeout_ms)
|
|
{
|
|
int rc = SUCCESS;
|
|
Timer timer;
|
|
|
|
InitTimer(&timer);
|
|
countdown_ms(&timer, timeout_ms);
|
|
while (!expired(&timer))
|
|
{
|
|
rc = cycle(c, &timer);
|
|
// cycle could return 0 or packet_type or 65535 if nothing is read
|
|
// cycle returns DISCONNECTED only if keepalive() fails.
|
|
if (rc == DISCONNECTED)
|
|
break;
|
|
rc = SUCCESS;
|
|
}
|
|
return rc;
|
|
}
|
|
|
|
|
|
// only used in single-threaded mode where one command at a time is in process
|
|
int waitfor(MQTTClient* c, int packet_type, Timer* timer)
|
|
{
|
|
int rc = FAILURE;
|
|
|
|
do
|
|
{
|
|
if (expired(timer))
|
|
break; // we timed out
|
|
}
|
|
while ((rc = cycle(c, timer)) != packet_type);
|
|
|
|
return rc;
|
|
}
|
|
|
|
|
|
int MQTTConnect(MQTTClient* c, MQTTPacket_connectData* options)
|
|
{
|
|
Timer connect_timer;
|
|
int rc = FAILURE;
|
|
MQTTPacket_connectData default_options = MQTTPacket_connectData_initializer;
|
|
int len = 0;
|
|
|
|
InitTimer(&connect_timer);
|
|
countdown_ms(&connect_timer, c->command_timeout_ms);
|
|
|
|
if (c->isconnected) // don't send connect packet again if we are already connected
|
|
goto exit;
|
|
|
|
if (options == 0)
|
|
options = &default_options; // set default options if none were supplied
|
|
|
|
c->keepAliveInterval = options->keepAliveInterval;
|
|
countdown(&(c->ping_timer), c->keepAliveInterval);
|
|
|
|
if ((len = MQTTSerialize_connect(c->buf, c->buf_size, options)) <= 0)
|
|
goto exit;
|
|
if ((rc = sendPacket(c, len, &connect_timer)) != SUCCESS) // send the connect packet
|
|
goto exit; // there was a problem
|
|
|
|
// this will be a blocking call, wait for the connack
|
|
if (waitfor(c, CONNACK, &connect_timer) == CONNACK)
|
|
{
|
|
unsigned char connack_rc = 255;
|
|
char sessionPresent = 0;
|
|
if (MQTTDeserialize_connack((unsigned char*)&sessionPresent, &connack_rc, c->readbuf, c->readbuf_size) == 1)
|
|
rc = connack_rc;
|
|
else
|
|
rc = FAILURE;
|
|
}
|
|
else
|
|
rc = FAILURE;
|
|
|
|
exit:
|
|
if (rc == SUCCESS)
|
|
c->isconnected = 1;
|
|
return rc;
|
|
}
|
|
|
|
|
|
int MQTTSubscribe(MQTTClient* c, const char* topic, enum QoS qos, messageHandler handler)
|
|
{
|
|
int rc = FAILURE;
|
|
Timer timer;
|
|
int len = 0;
|
|
MQTTString topicStr = MQTTString_initializer;
|
|
topicStr.cstring = (char *)topic;
|
|
|
|
InitTimer(&timer);
|
|
countdown_ms(&timer, c->command_timeout_ms);
|
|
if (!c->isconnected)
|
|
goto exit;
|
|
|
|
len = MQTTSerialize_subscribe(c->buf, c->buf_size, 0, getNextPacketId(c), 1, &topicStr, (int*)&qos);
|
|
if (len <= 0)
|
|
goto exit;
|
|
if ((rc = sendPacket(c, len, &timer)) != SUCCESS) // send the subscribe packet
|
|
{
|
|
goto exit; // there was a problem
|
|
}
|
|
|
|
if (waitfor(c, SUBACK, &timer) == SUBACK) // wait for suback
|
|
{
|
|
int count = 0, grantedQoS = -1;
|
|
unsigned short mypacketid;
|
|
if (MQTTDeserialize_suback(&mypacketid, 1, &count, &grantedQoS, c->readbuf, c->readbuf_size) == 1)
|
|
rc = grantedQoS; // 0, 1, 2 or 0x80
|
|
if (rc != 0x80)
|
|
{
|
|
int i;
|
|
for (i = 0; i < MAX_MESSAGE_HANDLERS; ++i)
|
|
{
|
|
if (c->messageHandlers[i].topicFilter == 0)
|
|
{
|
|
c->messageHandlers[i].topicFilter = topic;
|
|
c->messageHandlers[i].fp = handler;
|
|
rc = 0;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else
|
|
rc = FAILURE;
|
|
|
|
exit:
|
|
return rc;
|
|
}
|
|
|
|
|
|
int MQTTUnsubscribe(MQTTClient* c, const char* topicFilter)
|
|
{
|
|
int rc = FAILURE;
|
|
Timer timer;
|
|
MQTTString topic = MQTTString_initializer;
|
|
topic.cstring = (char *)topicFilter;
|
|
int len = 0;
|
|
|
|
InitTimer(&timer);
|
|
countdown_ms(&timer, c->command_timeout_ms);
|
|
|
|
if (!c->isconnected)
|
|
goto exit;
|
|
|
|
if ((len = MQTTSerialize_unsubscribe(c->buf, c->buf_size, 0, getNextPacketId(c), 1, &topic)) <= 0)
|
|
goto exit;
|
|
if ((rc = sendPacket(c, len, &timer)) != SUCCESS) // send the subscribe packet
|
|
goto exit; // there was a problem
|
|
|
|
if (waitfor(c, UNSUBACK, &timer) == UNSUBACK)
|
|
{
|
|
unsigned short mypacketid; // should be the same as the packetid above
|
|
if (MQTTDeserialize_unsuback(&mypacketid, c->readbuf, c->readbuf_size) == 1)
|
|
rc = 0;
|
|
}
|
|
else
|
|
rc = FAILURE;
|
|
|
|
exit:
|
|
return rc;
|
|
}
|
|
|
|
|
|
int MQTTPublish(MQTTClient* c, const char* topic, MQTTMessage* message)
|
|
{
|
|
int rc = FAILURE;
|
|
Timer timer;
|
|
MQTTString topicStr = MQTTString_initializer;
|
|
topicStr.cstring = (char *)topic;
|
|
int len = 0;
|
|
|
|
InitTimer(&timer);
|
|
countdown_ms(&timer, c->command_timeout_ms);
|
|
|
|
if (!c->isconnected)
|
|
goto exit;
|
|
|
|
if (message->qos == QOS1 || message->qos == QOS2)
|
|
message->id = getNextPacketId(c);
|
|
|
|
len = MQTTSerialize_publish(c->buf, c->buf_size, 0, message->qos, message->retained, message->id,
|
|
topicStr, (unsigned char*)message->payload, message->payloadlen);
|
|
if (len <= 0)
|
|
goto exit;
|
|
if ((rc = sendPacket(c, len, &timer)) != SUCCESS) // send the subscribe packet
|
|
{
|
|
goto exit; // there was a problem
|
|
}
|
|
|
|
if (message->qos == QOS1)
|
|
{
|
|
if (waitfor(c, PUBACK, &timer) == PUBACK)
|
|
{
|
|
// We still can receive from broker, treat as recoverable
|
|
c->fail_count = 0;
|
|
unsigned short mypacketid;
|
|
unsigned char dup, type;
|
|
if (MQTTDeserialize_ack(&type, &dup, &mypacketid, c->readbuf, c->readbuf_size) != 1)
|
|
rc = FAILURE;
|
|
else
|
|
rc = SUCCESS;
|
|
}
|
|
else
|
|
{
|
|
rc = FAILURE;
|
|
}
|
|
|
|
}
|
|
else if (message->qos == QOS2)
|
|
{
|
|
if (waitfor(c, PUBCOMP, &timer) == PUBCOMP)
|
|
{
|
|
// We still can receive from broker, treat as recoverable
|
|
c->fail_count = 0;
|
|
unsigned short mypacketid;
|
|
unsigned char dup, type;
|
|
if (MQTTDeserialize_ack(&type, &dup, &mypacketid, c->readbuf, c->readbuf_size) != 1)
|
|
rc = FAILURE;
|
|
else
|
|
rc = SUCCESS;
|
|
}
|
|
else
|
|
{
|
|
rc = FAILURE;
|
|
}
|
|
}
|
|
|
|
exit:
|
|
return rc;
|
|
}
|
|
|
|
|
|
int MQTTDisconnect(MQTTClient* c)
|
|
{
|
|
int rc = FAILURE;
|
|
Timer timer; // we might wait for incomplete incoming publishes to complete
|
|
int len = MQTTSerialize_disconnect(c->buf, c->buf_size);
|
|
|
|
InitTimer(&timer);
|
|
countdown_ms(&timer, c->command_timeout_ms);
|
|
|
|
if (len > 0)
|
|
rc = sendPacket(c, len, &timer); // send the disconnect packet
|
|
|
|
c->isconnected = 0;
|
|
return rc;
|
|
}
|
|
|