Duy tran / MQTT-v-16

v-16

Dependencies:   FP MQTTPacket

MQTTClient.h@0:818366b014c9, 2017-12-12 (annotated)

Committer:
DuyLionTran
Date:
Tue Dec 12 15:58:09 2017 +0000
Revision:
0:818366b014c9
version 1.6;

Who changed what in which revision?

UserRevisionLine numberNew contents of line
DuyLionTran 0:818366b014c9 1 /*******************************************************************************
DuyLionTran 0:818366b014c9 2 * Copyright (c) 2014, 2017 IBM Corp.
DuyLionTran 0:818366b014c9 3 *
DuyLionTran 0:818366b014c9 4 * All rights reserved. This program and the accompanying materials
DuyLionTran 0:818366b014c9 5 * are made available under the terms of the Eclipse Public License v1.0
DuyLionTran 0:818366b014c9 6 * and Eclipse Distribution License v1.0 which accompany this distribution.
DuyLionTran 0:818366b014c9 7 *
DuyLionTran 0:818366b014c9 8 * The Eclipse Public License is available at
DuyLionTran 0:818366b014c9 9 * http://www.eclipse.org/legal/epl-v10.html
DuyLionTran 0:818366b014c9 10 * and the Eclipse Distribution License is available at
DuyLionTran 0:818366b014c9 11 * http://www.eclipse.org/org/documents/edl-v10.php.
DuyLionTran 0:818366b014c9 12 *
DuyLionTran 0:818366b014c9 13 * Contributors:
DuyLionTran 0:818366b014c9 14 * Ian Craggs - initial API and implementation and/or initial documentation
DuyLionTran 0:818366b014c9 15 * Ian Craggs - fix for bug 458512 - QoS 2 messages
DuyLionTran 0:818366b014c9 16 * Ian Craggs - fix for bug 460389 - send loop uses wrong length
DuyLionTran 0:818366b014c9 17 * Ian Craggs - fix for bug 464169 - clearing subscriptions
DuyLionTran 0:818366b014c9 18 * Ian Craggs - fix for bug 464551 - enums and ints can be different size
DuyLionTran 0:818366b014c9 19 * Mark Sonnentag - fix for bug 475204 - inefficient instantiation of Timer
DuyLionTran 0:818366b014c9 20 * Ian Craggs - fix for bug 475749 - packetid modified twice
DuyLionTran 0:818366b014c9 21 * Ian Craggs - add ability to set message handler separately #6
DuyLionTran 0:818366b014c9 22 *******************************************************************************/
DuyLionTran 0:818366b014c9 23
DuyLionTran 0:818366b014c9 24 #if !defined(MQTTCLIENT_H)
DuyLionTran 0:818366b014c9 25 #define MQTTCLIENT_H
DuyLionTran 0:818366b014c9 26
DuyLionTran 0:818366b014c9 27 #include "FP.h"
DuyLionTran 0:818366b014c9 28 #include "MQTTPacket.h"
DuyLionTran 0:818366b014c9 29 #include <stdio.h>
DuyLionTran 0:818366b014c9 30 #include "MQTTLogging.h"
DuyLionTran 0:818366b014c9 31
DuyLionTran 0:818366b014c9 32 #if !defined(MQTTCLIENT_QOS1)
DuyLionTran 0:818366b014c9 33 #define MQTTCLIENT_QOS1 1
DuyLionTran 0:818366b014c9 34 #endif
DuyLionTran 0:818366b014c9 35 #if !defined(MQTTCLIENT_QOS2)
DuyLionTran 0:818366b014c9 36 #define MQTTCLIENT_QOS2 0
DuyLionTran 0:818366b014c9 37 #endif
DuyLionTran 0:818366b014c9 38
DuyLionTran 0:818366b014c9 39 namespace MQTT
DuyLionTran 0:818366b014c9 40 {
DuyLionTran 0:818366b014c9 41
DuyLionTran 0:818366b014c9 42
DuyLionTran 0:818366b014c9 43 enum QoS { QOS0, QOS1, QOS2 };
DuyLionTran 0:818366b014c9 44
DuyLionTran 0:818366b014c9 45 // all failure return codes must be negative
DuyLionTran 0:818366b014c9 46 enum returnCode { BUFFER_OVERFLOW = -2, FAILURE = -1, SUCCESS = 0 };
DuyLionTran 0:818366b014c9 47
DuyLionTran 0:818366b014c9 48
DuyLionTran 0:818366b014c9 49 struct Message
DuyLionTran 0:818366b014c9 50 {
DuyLionTran 0:818366b014c9 51 enum QoS qos;
DuyLionTran 0:818366b014c9 52 bool retained;
DuyLionTran 0:818366b014c9 53 bool dup;
DuyLionTran 0:818366b014c9 54 unsigned short id;
DuyLionTran 0:818366b014c9 55 void *payload;
DuyLionTran 0:818366b014c9 56 size_t payloadlen;
DuyLionTran 0:818366b014c9 57 };
DuyLionTran 0:818366b014c9 58
DuyLionTran 0:818366b014c9 59
DuyLionTran 0:818366b014c9 60 struct MessageData
DuyLionTran 0:818366b014c9 61 {
DuyLionTran 0:818366b014c9 62 MessageData(MQTTString &aTopicName, struct Message &aMessage) : message(aMessage), topicName(aTopicName)
DuyLionTran 0:818366b014c9 63 { }
DuyLionTran 0:818366b014c9 64
DuyLionTran 0:818366b014c9 65 struct Message &message;
DuyLionTran 0:818366b014c9 66 MQTTString &topicName;
DuyLionTran 0:818366b014c9 67 };
DuyLionTran 0:818366b014c9 68
DuyLionTran 0:818366b014c9 69
DuyLionTran 0:818366b014c9 70 struct connackData
DuyLionTran 0:818366b014c9 71 {
DuyLionTran 0:818366b014c9 72 int rc;
DuyLionTran 0:818366b014c9 73 bool sessionPresent;
DuyLionTran 0:818366b014c9 74 };
DuyLionTran 0:818366b014c9 75
DuyLionTran 0:818366b014c9 76
DuyLionTran 0:818366b014c9 77 struct subackData
DuyLionTran 0:818366b014c9 78 {
DuyLionTran 0:818366b014c9 79 int grantedQoS;
DuyLionTran 0:818366b014c9 80 };
DuyLionTran 0:818366b014c9 81
DuyLionTran 0:818366b014c9 82
DuyLionTran 0:818366b014c9 83 class PacketId
DuyLionTran 0:818366b014c9 84 {
DuyLionTran 0:818366b014c9 85 public:
DuyLionTran 0:818366b014c9 86 PacketId()
DuyLionTran 0:818366b014c9 87 {
DuyLionTran 0:818366b014c9 88 next = 0;
DuyLionTran 0:818366b014c9 89 }
DuyLionTran 0:818366b014c9 90
DuyLionTran 0:818366b014c9 91 int getNext()
DuyLionTran 0:818366b014c9 92 {
DuyLionTran 0:818366b014c9 93 return next = (next == MAX_PACKET_ID) ? 1 : next + 1;
DuyLionTran 0:818366b014c9 94 }
DuyLionTran 0:818366b014c9 95
DuyLionTran 0:818366b014c9 96 private:
DuyLionTran 0:818366b014c9 97 static const int MAX_PACKET_ID = 65535;
DuyLionTran 0:818366b014c9 98 int next;
DuyLionTran 0:818366b014c9 99 };
DuyLionTran 0:818366b014c9 100
DuyLionTran 0:818366b014c9 101
DuyLionTran 0:818366b014c9 102 /**
DuyLionTran 0:818366b014c9 103 * @class Client
DuyLionTran 0:818366b014c9 104 * @brief blocking, non-threaded MQTT client API
DuyLionTran 0:818366b014c9 105 *
DuyLionTran 0:818366b014c9 106 * This version of the API blocks on all method calls, until they are complete. This means that only one
DuyLionTran 0:818366b014c9 107 * MQTT request can be in process at any one time.
DuyLionTran 0:818366b014c9 108 * @param Network a network class which supports send, receive
DuyLionTran 0:818366b014c9 109 * @param Timer a timer class with the methods:
DuyLionTran 0:818366b014c9 110 */
DuyLionTran 0:818366b014c9 111 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE = 100, int MAX_MESSAGE_HANDLERS = 5>
DuyLionTran 0:818366b014c9 112 class Client
DuyLionTran 0:818366b014c9 113 {
DuyLionTran 0:818366b014c9 114
DuyLionTran 0:818366b014c9 115 public:
DuyLionTran 0:818366b014c9 116
DuyLionTran 0:818366b014c9 117 typedef void (*messageHandler)(MessageData&);
DuyLionTran 0:818366b014c9 118
DuyLionTran 0:818366b014c9 119 /** Construct the client
DuyLionTran 0:818366b014c9 120 * @param network - pointer to an instance of the Network class - must be connected to the endpoint
DuyLionTran 0:818366b014c9 121 * before calling MQTT connect
DuyLionTran 0:818366b014c9 122 * @param limits an instance of the Limit class - to alter limits as required
DuyLionTran 0:818366b014c9 123 */
DuyLionTran 0:818366b014c9 124 Client(Network& network, unsigned int command_timeout_ms = 30000);
DuyLionTran 0:818366b014c9 125
DuyLionTran 0:818366b014c9 126 /** Set the default message handling callback - used for any message which does not match a subscription message handler
DuyLionTran 0:818366b014c9 127 * @param mh - pointer to the callback function. Set to 0 to remove.
DuyLionTran 0:818366b014c9 128 */
DuyLionTran 0:818366b014c9 129 void setDefaultMessageHandler(messageHandler mh)
DuyLionTran 0:818366b014c9 130 {
DuyLionTran 0:818366b014c9 131 if (mh != 0)
DuyLionTran 0:818366b014c9 132 defaultMessageHandler.attach(mh);
DuyLionTran 0:818366b014c9 133 else
DuyLionTran 0:818366b014c9 134 defaultMessageHandler.detach();
DuyLionTran 0:818366b014c9 135 }
DuyLionTran 0:818366b014c9 136
DuyLionTran 0:818366b014c9 137 /** Set a message handling callback. This can be used outside of the the subscribe method.
DuyLionTran 0:818366b014c9 138 * @param topicFilter - a topic pattern which can include wildcards
DuyLionTran 0:818366b014c9 139 * @param mh - pointer to the callback function. If 0, removes the callback if any
DuyLionTran 0:818366b014c9 140 */
DuyLionTran 0:818366b014c9 141 int setMessageHandler(const char* topicFilter, messageHandler mh);
DuyLionTran 0:818366b014c9 142
DuyLionTran 0:818366b014c9 143 /** MQTT Connect - send an MQTT connect packet down the network and wait for a Connack
DuyLionTran 0:818366b014c9 144 * The nework object must be connected to the network endpoint before calling this
DuyLionTran 0:818366b014c9 145 * Default connect options are used
DuyLionTran 0:818366b014c9 146 * @return success code -
DuyLionTran 0:818366b014c9 147 */
DuyLionTran 0:818366b014c9 148 int connect();
DuyLionTran 0:818366b014c9 149
DuyLionTran 0:818366b014c9 150 /** MQTT Connect - send an MQTT connect packet down the network and wait for a Connack
DuyLionTran 0:818366b014c9 151 * The nework object must be connected to the network endpoint before calling this
DuyLionTran 0:818366b014c9 152 * @param options - connect options
DuyLionTran 0:818366b014c9 153 * @return success code -
DuyLionTran 0:818366b014c9 154 */
DuyLionTran 0:818366b014c9 155 int connect(MQTTPacket_connectData& options);
DuyLionTran 0:818366b014c9 156
DuyLionTran 0:818366b014c9 157 /** MQTT Connect - send an MQTT connect packet down the network and wait for a Connack
DuyLionTran 0:818366b014c9 158 * The nework object must be connected to the network endpoint before calling this
DuyLionTran 0:818366b014c9 159 * @param options - connect options
DuyLionTran 0:818366b014c9 160 * @param connackData - connack data to be returned
DuyLionTran 0:818366b014c9 161 * @return success code -
DuyLionTran 0:818366b014c9 162 */
DuyLionTran 0:818366b014c9 163 int connect(MQTTPacket_connectData& options, connackData& data);
DuyLionTran 0:818366b014c9 164
DuyLionTran 0:818366b014c9 165 /** MQTT Publish - send an MQTT publish packet and wait for all acks to complete for all QoSs
DuyLionTran 0:818366b014c9 166 * @param topic - the topic to publish to
DuyLionTran 0:818366b014c9 167 * @param message - the message to send
DuyLionTran 0:818366b014c9 168 * @return success code -
DuyLionTran 0:818366b014c9 169 */
DuyLionTran 0:818366b014c9 170 int publish(const char* topicName, Message& message);
DuyLionTran 0:818366b014c9 171
DuyLionTran 0:818366b014c9 172 /** MQTT Publish - send an MQTT publish packet and wait for all acks to complete for all QoSs
DuyLionTran 0:818366b014c9 173 * @param topic - the topic to publish to
DuyLionTran 0:818366b014c9 174 * @param payload - the data to send
DuyLionTran 0:818366b014c9 175 * @param payloadlen - the length of the data
DuyLionTran 0:818366b014c9 176 * @param qos - the QoS to send the publish at
DuyLionTran 0:818366b014c9 177 * @param retained - whether the message should be retained
DuyLionTran 0:818366b014c9 178 * @return success code -
DuyLionTran 0:818366b014c9 179 */
DuyLionTran 0:818366b014c9 180 int publish(const char* topicName, void* payload, size_t payloadlen, enum QoS qos = QOS0, bool retained = false);
DuyLionTran 0:818366b014c9 181
DuyLionTran 0:818366b014c9 182 /** MQTT Publish - send an MQTT publish packet and wait for all acks to complete for all QoSs
DuyLionTran 0:818366b014c9 183 * @param topic - the topic to publish to
DuyLionTran 0:818366b014c9 184 * @param payload - the data to send
DuyLionTran 0:818366b014c9 185 * @param payloadlen - the length of the data
DuyLionTran 0:818366b014c9 186 * @param id - the packet id used - returned
DuyLionTran 0:818366b014c9 187 * @param qos - the QoS to send the publish at
DuyLionTran 0:818366b014c9 188 * @param retained - whether the message should be retained
DuyLionTran 0:818366b014c9 189 * @return success code -
DuyLionTran 0:818366b014c9 190 */
DuyLionTran 0:818366b014c9 191 int publish(const char* topicName, void* payload, size_t payloadlen, unsigned short& id, enum QoS qos = QOS1, bool retained = false);
DuyLionTran 0:818366b014c9 192
DuyLionTran 0:818366b014c9 193 /** MQTT Subscribe - send an MQTT subscribe packet and wait for the suback
DuyLionTran 0:818366b014c9 194 * @param topicFilter - a topic pattern which can include wildcards
DuyLionTran 0:818366b014c9 195 * @param qos - the MQTT QoS to subscribe at
DuyLionTran 0:818366b014c9 196 * @param mh - the callback function to be invoked when a message is received for this subscription
DuyLionTran 0:818366b014c9 197 * @return success code -
DuyLionTran 0:818366b014c9 198 */
DuyLionTran 0:818366b014c9 199 int subscribe(const char* topicFilter, enum QoS qos, messageHandler mh);
DuyLionTran 0:818366b014c9 200
DuyLionTran 0:818366b014c9 201 /** MQTT Subscribe - send an MQTT subscribe packet and wait for the suback
DuyLionTran 0:818366b014c9 202 * @param topicFilter - a topic pattern which can include wildcards
DuyLionTran 0:818366b014c9 203 * @param qos - the MQTT QoS to subscribe at©
DuyLionTran 0:818366b014c9 204 * @param mh - the callback function to be invoked when a message is received for this subscription
DuyLionTran 0:818366b014c9 205 * @param
DuyLionTran 0:818366b014c9 206 * @return success code -
DuyLionTran 0:818366b014c9 207 */
DuyLionTran 0:818366b014c9 208 int subscribe(const char* topicFilter, enum QoS qos, messageHandler mh, subackData &data);
DuyLionTran 0:818366b014c9 209
DuyLionTran 0:818366b014c9 210 /** MQTT Unsubscribe - send an MQTT unsubscribe packet and wait for the unsuback
DuyLionTran 0:818366b014c9 211 * @param topicFilter - a topic pattern which can include wildcards
DuyLionTran 0:818366b014c9 212 * @return success code -
DuyLionTran 0:818366b014c9 213 */
DuyLionTran 0:818366b014c9 214 int unsubscribe(const char* topicFilter);
DuyLionTran 0:818366b014c9 215
DuyLionTran 0:818366b014c9 216 /** MQTT Disconnect - send an MQTT disconnect packet, and clean up any state
DuyLionTran 0:818366b014c9 217 * @return success code -
DuyLionTran 0:818366b014c9 218 */
DuyLionTran 0:818366b014c9 219 int disconnect();
DuyLionTran 0:818366b014c9 220
DuyLionTran 0:818366b014c9 221 /** A call to this API must be made within the keepAlive interval to keep the MQTT connection alive
DuyLionTran 0:818366b014c9 222 * yield can be called if no other MQTT operation is needed. This will also allow messages to be
DuyLionTran 0:818366b014c9 223 * received.
DuyLionTran 0:818366b014c9 224 * @param timeout_ms the time to wait, in milliseconds
DuyLionTran 0:818366b014c9 225 * @return success code - on failure, this means the client has disconnected
DuyLionTran 0:818366b014c9 226 */
DuyLionTran 0:818366b014c9 227 int yield(unsigned long timeout_ms = 1000L);
DuyLionTran 0:818366b014c9 228
DuyLionTran 0:818366b014c9 229 /** Is the client connected?
DuyLionTran 0:818366b014c9 230 * @return flag - is the client connected or not?
DuyLionTran 0:818366b014c9 231 */
DuyLionTran 0:818366b014c9 232 bool isConnected()
DuyLionTran 0:818366b014c9 233 {
DuyLionTran 0:818366b014c9 234 return isconnected;
DuyLionTran 0:818366b014c9 235 }
DuyLionTran 0:818366b014c9 236
DuyLionTran 0:818366b014c9 237 private:
DuyLionTran 0:818366b014c9 238
DuyLionTran 0:818366b014c9 239 void closeSession();
DuyLionTran 0:818366b014c9 240 void cleanSession();
DuyLionTran 0:818366b014c9 241 int cycle(Timer& timer);
DuyLionTran 0:818366b014c9 242 int waitfor(int packet_type, Timer& timer);
DuyLionTran 0:818366b014c9 243 int keepalive();
DuyLionTran 0:818366b014c9 244 int publish(int len, Timer& timer, enum QoS qos);
DuyLionTran 0:818366b014c9 245
DuyLionTran 0:818366b014c9 246 int decodePacket(int* value, int timeout);
DuyLionTran 0:818366b014c9 247 int readPacket(Timer& timer);
DuyLionTran 0:818366b014c9 248 int sendPacket(int length, Timer& timer);
DuyLionTran 0:818366b014c9 249 int deliverMessage(MQTTString& topicName, Message& message);
DuyLionTran 0:818366b014c9 250 bool isTopicMatched(char* topicFilter, MQTTString& topicName);
DuyLionTran 0:818366b014c9 251
DuyLionTran 0:818366b014c9 252 Network& ipstack;
DuyLionTran 0:818366b014c9 253 unsigned long command_timeout_ms;
DuyLionTran 0:818366b014c9 254
DuyLionTran 0:818366b014c9 255 unsigned char sendbuf[MAX_MQTT_PACKET_SIZE];
DuyLionTran 0:818366b014c9 256 unsigned char readbuf[MAX_MQTT_PACKET_SIZE];
DuyLionTran 0:818366b014c9 257
DuyLionTran 0:818366b014c9 258 Timer last_sent, last_received;
DuyLionTran 0:818366b014c9 259 unsigned int keepAliveInterval;
DuyLionTran 0:818366b014c9 260 bool ping_outstanding;
DuyLionTran 0:818366b014c9 261 bool cleansession;
DuyLionTran 0:818366b014c9 262
DuyLionTran 0:818366b014c9 263 PacketId packetid;
DuyLionTran 0:818366b014c9 264
DuyLionTran 0:818366b014c9 265 struct MessageHandlers
DuyLionTran 0:818366b014c9 266 {
DuyLionTran 0:818366b014c9 267 const char* topicFilter;
DuyLionTran 0:818366b014c9 268 FP<void, MessageData&> fp;
DuyLionTran 0:818366b014c9 269 } messageHandlers[MAX_MESSAGE_HANDLERS]; // Message handlers are indexed by subscription topic
DuyLionTran 0:818366b014c9 270
DuyLionTran 0:818366b014c9 271 FP<void, MessageData&> defaultMessageHandler;
DuyLionTran 0:818366b014c9 272
DuyLionTran 0:818366b014c9 273 bool isconnected;
DuyLionTran 0:818366b014c9 274
DuyLionTran 0:818366b014c9 275 #if MQTTCLIENT_QOS1 || MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 276 unsigned char pubbuf[MAX_MQTT_PACKET_SIZE]; // store the last publish for sending on reconnect
DuyLionTran 0:818366b014c9 277 int inflightLen;
DuyLionTran 0:818366b014c9 278 unsigned short inflightMsgid;
DuyLionTran 0:818366b014c9 279 enum QoS inflightQoS;
DuyLionTran 0:818366b014c9 280 #endif
DuyLionTran 0:818366b014c9 281
DuyLionTran 0:818366b014c9 282 #if MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 283 bool pubrel;
DuyLionTran 0:818366b014c9 284 #if !defined(MAX_INCOMING_QOS2_MESSAGES)
DuyLionTran 0:818366b014c9 285 #define MAX_INCOMING_QOS2_MESSAGES 10
DuyLionTran 0:818366b014c9 286 #endif
DuyLionTran 0:818366b014c9 287 unsigned short incomingQoS2messages[MAX_INCOMING_QOS2_MESSAGES];
DuyLionTran 0:818366b014c9 288 bool isQoS2msgidFree(unsigned short id);
DuyLionTran 0:818366b014c9 289 bool useQoS2msgid(unsigned short id);
DuyLionTran 0:818366b014c9 290 void freeQoS2msgid(unsigned short id);
DuyLionTran 0:818366b014c9 291 #endif
DuyLionTran 0:818366b014c9 292
DuyLionTran 0:818366b014c9 293 };
DuyLionTran 0:818366b014c9 294
DuyLionTran 0:818366b014c9 295 }
DuyLionTran 0:818366b014c9 296
DuyLionTran 0:818366b014c9 297
DuyLionTran 0:818366b014c9 298 template<class Network, class Timer, int a, int MAX_MESSAGE_HANDLERS>
DuyLionTran 0:818366b014c9 299 void MQTT::Client<Network, Timer, a, MAX_MESSAGE_HANDLERS>::cleanSession()
DuyLionTran 0:818366b014c9 300 {
DuyLionTran 0:818366b014c9 301 for (int i = 0; i < MAX_MESSAGE_HANDLERS; ++i)
DuyLionTran 0:818366b014c9 302 messageHandlers[i].topicFilter = 0;
DuyLionTran 0:818366b014c9 303
DuyLionTran 0:818366b014c9 304 #if MQTTCLIENT_QOS1 || MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 305 inflightMsgid = 0;
DuyLionTran 0:818366b014c9 306 inflightQoS = QOS0;
DuyLionTran 0:818366b014c9 307 #endif
DuyLionTran 0:818366b014c9 308
DuyLionTran 0:818366b014c9 309 #if MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 310 pubrel = false;
DuyLionTran 0:818366b014c9 311 for (int i = 0; i < MAX_INCOMING_QOS2_MESSAGES; ++i)
DuyLionTran 0:818366b014c9 312 incomingQoS2messages[i] = 0;
DuyLionTran 0:818366b014c9 313 #endif
DuyLionTran 0:818366b014c9 314 }
DuyLionTran 0:818366b014c9 315
DuyLionTran 0:818366b014c9 316
DuyLionTran 0:818366b014c9 317 template<class Network, class Timer, int a, int MAX_MESSAGE_HANDLERS>
DuyLionTran 0:818366b014c9 318 void MQTT::Client<Network, Timer, a, MAX_MESSAGE_HANDLERS>::closeSession()
DuyLionTran 0:818366b014c9 319 {
DuyLionTran 0:818366b014c9 320 ping_outstanding = false;
DuyLionTran 0:818366b014c9 321 isconnected = false;
DuyLionTran 0:818366b014c9 322 if (cleansession)
DuyLionTran 0:818366b014c9 323 cleanSession();
DuyLionTran 0:818366b014c9 324 }
DuyLionTran 0:818366b014c9 325
DuyLionTran 0:818366b014c9 326
DuyLionTran 0:818366b014c9 327 template<class Network, class Timer, int a, int MAX_MESSAGE_HANDLERS>
DuyLionTran 0:818366b014c9 328 MQTT::Client<Network, Timer, a, MAX_MESSAGE_HANDLERS>::Client(Network& network, unsigned int command_timeout_ms) : ipstack(network), packetid()
DuyLionTran 0:818366b014c9 329 {
DuyLionTran 0:818366b014c9 330 this->command_timeout_ms = command_timeout_ms;
DuyLionTran 0:818366b014c9 331 cleansession = true;
DuyLionTran 0:818366b014c9 332 closeSession();
DuyLionTran 0:818366b014c9 333 }
DuyLionTran 0:818366b014c9 334
DuyLionTran 0:818366b014c9 335
DuyLionTran 0:818366b014c9 336 #if MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 337 template<class Network, class Timer, int a, int b>
DuyLionTran 0:818366b014c9 338 bool MQTT::Client<Network, Timer, a, b>::isQoS2msgidFree(unsigned short id)
DuyLionTran 0:818366b014c9 339 {
DuyLionTran 0:818366b014c9 340 for (int i = 0; i < MAX_INCOMING_QOS2_MESSAGES; ++i)
DuyLionTran 0:818366b014c9 341 {
DuyLionTran 0:818366b014c9 342 if (incomingQoS2messages[i] == id)
DuyLionTran 0:818366b014c9 343 return false;
DuyLionTran 0:818366b014c9 344 }
DuyLionTran 0:818366b014c9 345 return true;
DuyLionTran 0:818366b014c9 346 }
DuyLionTran 0:818366b014c9 347
DuyLionTran 0:818366b014c9 348
DuyLionTran 0:818366b014c9 349 template<class Network, class Timer, int a, int b>
DuyLionTran 0:818366b014c9 350 bool MQTT::Client<Network, Timer, a, b>::useQoS2msgid(unsigned short id)
DuyLionTran 0:818366b014c9 351 {
DuyLionTran 0:818366b014c9 352 for (int i = 0; i < MAX_INCOMING_QOS2_MESSAGES; ++i)
DuyLionTran 0:818366b014c9 353 {
DuyLionTran 0:818366b014c9 354 if (incomingQoS2messages[i] == 0)
DuyLionTran 0:818366b014c9 355 {
DuyLionTran 0:818366b014c9 356 incomingQoS2messages[i] = id;
DuyLionTran 0:818366b014c9 357 return true;
DuyLionTran 0:818366b014c9 358 }
DuyLionTran 0:818366b014c9 359 }
DuyLionTran 0:818366b014c9 360 return false;
DuyLionTran 0:818366b014c9 361 }
DuyLionTran 0:818366b014c9 362
DuyLionTran 0:818366b014c9 363
DuyLionTran 0:818366b014c9 364 template<class Network, class Timer, int a, int b>
DuyLionTran 0:818366b014c9 365 void MQTT::Client<Network, Timer, a, b>::freeQoS2msgid(unsigned short id)
DuyLionTran 0:818366b014c9 366 {
DuyLionTran 0:818366b014c9 367 for (int i = 0; i < MAX_INCOMING_QOS2_MESSAGES; ++i)
DuyLionTran 0:818366b014c9 368 {
DuyLionTran 0:818366b014c9 369 if (incomingQoS2messages[i] == id)
DuyLionTran 0:818366b014c9 370 {
DuyLionTran 0:818366b014c9 371 incomingQoS2messages[i] = 0;
DuyLionTran 0:818366b014c9 372 return;
DuyLionTran 0:818366b014c9 373 }
DuyLionTran 0:818366b014c9 374 }
DuyLionTran 0:818366b014c9 375 }
DuyLionTran 0:818366b014c9 376 #endif
DuyLionTran 0:818366b014c9 377
DuyLionTran 0:818366b014c9 378
DuyLionTran 0:818366b014c9 379 template<class Network, class Timer, int a, int b>
DuyLionTran 0:818366b014c9 380 int MQTT::Client<Network, Timer, a, b>::sendPacket(int length, Timer& timer)
DuyLionTran 0:818366b014c9 381 {
DuyLionTran 0:818366b014c9 382 int rc = FAILURE,
DuyLionTran 0:818366b014c9 383 sent = 0;
DuyLionTran 0:818366b014c9 384
DuyLionTran 0:818366b014c9 385 while (sent < length)
DuyLionTran 0:818366b014c9 386 {
DuyLionTran 0:818366b014c9 387 rc = ipstack.write(&sendbuf[sent], length - sent, timer.left_ms());
DuyLionTran 0:818366b014c9 388 if (rc < 0) // there was an error writing the data
DuyLionTran 0:818366b014c9 389 break;
DuyLionTran 0:818366b014c9 390 sent += rc;
DuyLionTran 0:818366b014c9 391 if (timer.expired()) // only check expiry after at least one attempt to write
DuyLionTran 0:818366b014c9 392 break;
DuyLionTran 0:818366b014c9 393 }
DuyLionTran 0:818366b014c9 394 if (sent == length)
DuyLionTran 0:818366b014c9 395 {
DuyLionTran 0:818366b014c9 396 if (this->keepAliveInterval > 0)
DuyLionTran 0:818366b014c9 397 last_sent.countdown(this->keepAliveInterval); // record the fact that we have successfully sent the packet
DuyLionTran 0:818366b014c9 398 rc = SUCCESS;
DuyLionTran 0:818366b014c9 399 }
DuyLionTran 0:818366b014c9 400 else
DuyLionTran 0:818366b014c9 401 rc = FAILURE;
DuyLionTran 0:818366b014c9 402
DuyLionTran 0:818366b014c9 403 #if defined(MQTT_DEBUG)
DuyLionTran 0:818366b014c9 404 char printbuf[150];
DuyLionTran 0:818366b014c9 405 DEBUG("Rc %d from sending packet %s\r\n", rc,
DuyLionTran 0:818366b014c9 406 MQTTFormat_toServerString(printbuf, sizeof(printbuf), sendbuf, length));
DuyLionTran 0:818366b014c9 407 #endif
DuyLionTran 0:818366b014c9 408 return rc;
DuyLionTran 0:818366b014c9 409 }
DuyLionTran 0:818366b014c9 410
DuyLionTran 0:818366b014c9 411
DuyLionTran 0:818366b014c9 412 template<class Network, class Timer, int a, int b>
DuyLionTran 0:818366b014c9 413 int MQTT::Client<Network, Timer, a, b>::decodePacket(int* value, int timeout)
DuyLionTran 0:818366b014c9 414 {
DuyLionTran 0:818366b014c9 415 unsigned char c;
DuyLionTran 0:818366b014c9 416 int multiplier = 1;
DuyLionTran 0:818366b014c9 417 int len = 0;
DuyLionTran 0:818366b014c9 418 const int MAX_NO_OF_REMAINING_LENGTH_BYTES = 4;
DuyLionTran 0:818366b014c9 419
DuyLionTran 0:818366b014c9 420 *value = 0;
DuyLionTran 0:818366b014c9 421 do
DuyLionTran 0:818366b014c9 422 {
DuyLionTran 0:818366b014c9 423 int rc = MQTTPACKET_READ_ERROR;
DuyLionTran 0:818366b014c9 424
DuyLionTran 0:818366b014c9 425 if (++len > MAX_NO_OF_REMAINING_LENGTH_BYTES)
DuyLionTran 0:818366b014c9 426 {
DuyLionTran 0:818366b014c9 427 rc = MQTTPACKET_READ_ERROR; /* bad data */
DuyLionTran 0:818366b014c9 428 goto exit;
DuyLionTran 0:818366b014c9 429 }
DuyLionTran 0:818366b014c9 430 rc = ipstack.read(&c, 1, timeout);
DuyLionTran 0:818366b014c9 431 if (rc != 1)
DuyLionTran 0:818366b014c9 432 goto exit;
DuyLionTran 0:818366b014c9 433 *value += (c & 127) * multiplier;
DuyLionTran 0:818366b014c9 434 multiplier *= 128;
DuyLionTran 0:818366b014c9 435 } while ((c & 128) != 0);
DuyLionTran 0:818366b014c9 436 exit:
DuyLionTran 0:818366b014c9 437 return len;
DuyLionTran 0:818366b014c9 438 }
DuyLionTran 0:818366b014c9 439
DuyLionTran 0:818366b014c9 440
DuyLionTran 0:818366b014c9 441 /**
DuyLionTran 0:818366b014c9 442 * If any read fails in this method, then we should disconnect from the network, as on reconnect
DuyLionTran 0:818366b014c9 443 * the packets can be retried.
DuyLionTran 0:818366b014c9 444 * @param timeout the max time to wait for the packet read to complete, in milliseconds
DuyLionTran 0:818366b014c9 445 * @return the MQTT packet type, 0 if none, -1 if error
DuyLionTran 0:818366b014c9 446 */
DuyLionTran 0:818366b014c9 447 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int b>
DuyLionTran 0:818366b014c9 448 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, b>::readPacket(Timer& timer)
DuyLionTran 0:818366b014c9 449 {
DuyLionTran 0:818366b014c9 450 int rc = FAILURE;
DuyLionTran 0:818366b014c9 451 MQTTHeader header = {0};
DuyLionTran 0:818366b014c9 452 int len = 0;
DuyLionTran 0:818366b014c9 453 int rem_len = 0;
DuyLionTran 0:818366b014c9 454
DuyLionTran 0:818366b014c9 455 /* 1. read the header byte. This has the packet type in it */
DuyLionTran 0:818366b014c9 456 rc = ipstack.read(readbuf, 1, timer.left_ms());
DuyLionTran 0:818366b014c9 457 if (rc != 1)
DuyLionTran 0:818366b014c9 458 goto exit;
DuyLionTran 0:818366b014c9 459
DuyLionTran 0:818366b014c9 460 len = 1;
DuyLionTran 0:818366b014c9 461 /* 2. read the remaining length. This is variable in itself */
DuyLionTran 0:818366b014c9 462 decodePacket(&rem_len, timer.left_ms());
DuyLionTran 0:818366b014c9 463 len += MQTTPacket_encode(readbuf + 1, rem_len); /* put the original remaining length into the buffer */
DuyLionTran 0:818366b014c9 464
DuyLionTran 0:818366b014c9 465 if (rem_len > (MAX_MQTT_PACKET_SIZE - len))
DuyLionTran 0:818366b014c9 466 {
DuyLionTran 0:818366b014c9 467 rc = BUFFER_OVERFLOW;
DuyLionTran 0:818366b014c9 468 goto exit;
DuyLionTran 0:818366b014c9 469 }
DuyLionTran 0:818366b014c9 470
DuyLionTran 0:818366b014c9 471 /* 3. read the rest of the buffer using a callback to supply the rest of the data */
DuyLionTran 0:818366b014c9 472 if (rem_len > 0 && (ipstack.read(readbuf + len, rem_len, timer.left_ms()) != rem_len))
DuyLionTran 0:818366b014c9 473 goto exit;
DuyLionTran 0:818366b014c9 474
DuyLionTran 0:818366b014c9 475 header.byte = readbuf[0];
DuyLionTran 0:818366b014c9 476 rc = header.bits.type;
DuyLionTran 0:818366b014c9 477 if (this->keepAliveInterval > 0)
DuyLionTran 0:818366b014c9 478 last_received.countdown(this->keepAliveInterval); // record the fact that we have successfully received a packet
DuyLionTran 0:818366b014c9 479 exit:
DuyLionTran 0:818366b014c9 480
DuyLionTran 0:818366b014c9 481 #if defined(MQTT_DEBUG)
DuyLionTran 0:818366b014c9 482 if (rc >= 0)
DuyLionTran 0:818366b014c9 483 {
DuyLionTran 0:818366b014c9 484 char printbuf[50];
DuyLionTran 0:818366b014c9 485 DEBUG("Rc %d receiving packet %s\r\n", rc,
DuyLionTran 0:818366b014c9 486 MQTTFormat_toClientString(printbuf, sizeof(printbuf), readbuf, len));
DuyLionTran 0:818366b014c9 487 }
DuyLionTran 0:818366b014c9 488 #endif
DuyLionTran 0:818366b014c9 489 return rc;
DuyLionTran 0:818366b014c9 490 }
DuyLionTran 0:818366b014c9 491
DuyLionTran 0:818366b014c9 492
DuyLionTran 0:818366b014c9 493 // assume topic filter and name is in correct format
DuyLionTran 0:818366b014c9 494 // # can only be at end
DuyLionTran 0:818366b014c9 495 // + and # can only be next to separator
DuyLionTran 0:818366b014c9 496 template<class Network, class Timer, int a, int b>
DuyLionTran 0:818366b014c9 497 bool MQTT::Client<Network, Timer, a, b>::isTopicMatched(char* topicFilter, MQTTString& topicName)
DuyLionTran 0:818366b014c9 498 {
DuyLionTran 0:818366b014c9 499 char* curf = topicFilter;
DuyLionTran 0:818366b014c9 500 char* curn = topicName.lenstring.data;
DuyLionTran 0:818366b014c9 501 char* curn_end = curn + topicName.lenstring.len;
DuyLionTran 0:818366b014c9 502
DuyLionTran 0:818366b014c9 503 while (*curf && curn < curn_end)
DuyLionTran 0:818366b014c9 504 {
DuyLionTran 0:818366b014c9 505 if (*curn == '/' && *curf != '/')
DuyLionTran 0:818366b014c9 506 break;
DuyLionTran 0:818366b014c9 507 if (*curf != '+' && *curf != '#' && *curf != *curn)
DuyLionTran 0:818366b014c9 508 break;
DuyLionTran 0:818366b014c9 509 if (*curf == '+')
DuyLionTran 0:818366b014c9 510 { // skip until we meet the next separator, or end of string
DuyLionTran 0:818366b014c9 511 char* nextpos = curn + 1;
DuyLionTran 0:818366b014c9 512 while (nextpos < curn_end && *nextpos != '/')
DuyLionTran 0:818366b014c9 513 nextpos = ++curn + 1;
DuyLionTran 0:818366b014c9 514 }
DuyLionTran 0:818366b014c9 515 else if (*curf == '#')
DuyLionTran 0:818366b014c9 516 curn = curn_end - 1; // skip until end of string
DuyLionTran 0:818366b014c9 517 curf++;
DuyLionTran 0:818366b014c9 518 curn++;
DuyLionTran 0:818366b014c9 519 };
DuyLionTran 0:818366b014c9 520
DuyLionTran 0:818366b014c9 521 return (curn == curn_end) && (*curf == '\0');
DuyLionTran 0:818366b014c9 522 }
DuyLionTran 0:818366b014c9 523
DuyLionTran 0:818366b014c9 524
DuyLionTran 0:818366b014c9 525
DuyLionTran 0:818366b014c9 526 template<class Network, class Timer, int a, int MAX_MESSAGE_HANDLERS>
DuyLionTran 0:818366b014c9 527 int MQTT::Client<Network, Timer, a, MAX_MESSAGE_HANDLERS>::deliverMessage(MQTTString& topicName, Message& message)
DuyLionTran 0:818366b014c9 528 {
DuyLionTran 0:818366b014c9 529 int rc = FAILURE;
DuyLionTran 0:818366b014c9 530
DuyLionTran 0:818366b014c9 531 // we have to find the right message handler - indexed by topic
DuyLionTran 0:818366b014c9 532 for (int i = 0; i < MAX_MESSAGE_HANDLERS; ++i)
DuyLionTran 0:818366b014c9 533 {
DuyLionTran 0:818366b014c9 534 if (messageHandlers[i].topicFilter != 0 && (MQTTPacket_equals(&topicName, (char*)messageHandlers[i].topicFilter) ||
DuyLionTran 0:818366b014c9 535 isTopicMatched((char*)messageHandlers[i].topicFilter, topicName)))
DuyLionTran 0:818366b014c9 536 {
DuyLionTran 0:818366b014c9 537 if (messageHandlers[i].fp.attached())
DuyLionTran 0:818366b014c9 538 {
DuyLionTran 0:818366b014c9 539 MessageData md(topicName, message);
DuyLionTran 0:818366b014c9 540 messageHandlers[i].fp(md);
DuyLionTran 0:818366b014c9 541 rc = SUCCESS;
DuyLionTran 0:818366b014c9 542 }
DuyLionTran 0:818366b014c9 543 }
DuyLionTran 0:818366b014c9 544 }
DuyLionTran 0:818366b014c9 545
DuyLionTran 0:818366b014c9 546 if (rc == FAILURE && defaultMessageHandler.attached())
DuyLionTran 0:818366b014c9 547 {
DuyLionTran 0:818366b014c9 548 MessageData md(topicName, message);
DuyLionTran 0:818366b014c9 549 defaultMessageHandler(md);
DuyLionTran 0:818366b014c9 550 rc = SUCCESS;
DuyLionTran 0:818366b014c9 551 }
DuyLionTran 0:818366b014c9 552
DuyLionTran 0:818366b014c9 553 return rc;
DuyLionTran 0:818366b014c9 554 }
DuyLionTran 0:818366b014c9 555
DuyLionTran 0:818366b014c9 556
DuyLionTran 0:818366b014c9 557
DuyLionTran 0:818366b014c9 558 template<class Network, class Timer, int a, int b>
DuyLionTran 0:818366b014c9 559 int MQTT::Client<Network, Timer, a, b>::yield(unsigned long timeout_ms)
DuyLionTran 0:818366b014c9 560 {
DuyLionTran 0:818366b014c9 561 int rc = SUCCESS;
DuyLionTran 0:818366b014c9 562 Timer timer;
DuyLionTran 0:818366b014c9 563
DuyLionTran 0:818366b014c9 564 timer.countdown_ms(timeout_ms);
DuyLionTran 0:818366b014c9 565 while (!timer.expired())
DuyLionTran 0:818366b014c9 566 {
DuyLionTran 0:818366b014c9 567 if (cycle(timer) < 0)
DuyLionTran 0:818366b014c9 568 {
DuyLionTran 0:818366b014c9 569 rc = FAILURE;
DuyLionTran 0:818366b014c9 570 break;
DuyLionTran 0:818366b014c9 571 }
DuyLionTran 0:818366b014c9 572 }
DuyLionTran 0:818366b014c9 573
DuyLionTran 0:818366b014c9 574 return rc;
DuyLionTran 0:818366b014c9 575 }
DuyLionTran 0:818366b014c9 576
DuyLionTran 0:818366b014c9 577
DuyLionTran 0:818366b014c9 578 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int b>
DuyLionTran 0:818366b014c9 579 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, b>::cycle(Timer& timer)
DuyLionTran 0:818366b014c9 580 {
DuyLionTran 0:818366b014c9 581 // get one piece of work off the wire and one pass through
DuyLionTran 0:818366b014c9 582 int len = 0,
DuyLionTran 0:818366b014c9 583 rc = SUCCESS;
DuyLionTran 0:818366b014c9 584
DuyLionTran 0:818366b014c9 585 int packet_type = readPacket(timer); // read the socket, see what work is due
DuyLionTran 0:818366b014c9 586
DuyLionTran 0:818366b014c9 587 switch (packet_type)
DuyLionTran 0:818366b014c9 588 {
DuyLionTran 0:818366b014c9 589 default:
DuyLionTran 0:818366b014c9 590 // no more data to read, unrecoverable. Or read packet fails due to unexpected network error
DuyLionTran 0:818366b014c9 591 rc = packet_type;
DuyLionTran 0:818366b014c9 592 goto exit;
DuyLionTran 0:818366b014c9 593 case NSAPI_ERROR_WOULD_BLOCK:
DuyLionTran 0:818366b014c9 594 case NSAPI_ERROR_OK: // timed out reading packet
DuyLionTran 0:818366b014c9 595 break;
DuyLionTran 0:818366b014c9 596 case CONNACK:
DuyLionTran 0:818366b014c9 597 case PUBACK:
DuyLionTran 0:818366b014c9 598 case SUBACK:
DuyLionTran 0:818366b014c9 599 break;
DuyLionTran 0:818366b014c9 600 case PUBLISH:
DuyLionTran 0:818366b014c9 601 {
DuyLionTran 0:818366b014c9 602 MQTTString topicName = MQTTString_initializer;
DuyLionTran 0:818366b014c9 603 Message msg;
DuyLionTran 0:818366b014c9 604 int intQoS;
DuyLionTran 0:818366b014c9 605 msg.payloadlen = 0; /* this is a size_t, but deserialize publish sets this as int */
DuyLionTran 0:818366b014c9 606 if (MQTTDeserialize_publish((unsigned char*)&msg.dup, &intQoS, (unsigned char*)&msg.retained, (unsigned short*)&msg.id, &topicName,
DuyLionTran 0:818366b014c9 607 (unsigned char**)&msg.payload, (int*)&msg.payloadlen, readbuf, MAX_MQTT_PACKET_SIZE) != 1)
DuyLionTran 0:818366b014c9 608 goto exit;
DuyLionTran 0:818366b014c9 609 msg.qos = (enum QoS)intQoS;
DuyLionTran 0:818366b014c9 610 #if MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 611 if (msg.qos != QOS2)
DuyLionTran 0:818366b014c9 612 #endif
DuyLionTran 0:818366b014c9 613 deliverMessage(topicName, msg);
DuyLionTran 0:818366b014c9 614 #if MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 615 else if (isQoS2msgidFree(msg.id))
DuyLionTran 0:818366b014c9 616 {
DuyLionTran 0:818366b014c9 617 if (useQoS2msgid(msg.id))
DuyLionTran 0:818366b014c9 618 deliverMessage(topicName, msg);
DuyLionTran 0:818366b014c9 619 else
DuyLionTran 0:818366b014c9 620 WARN("Maximum number of incoming QoS2 messages exceeded");
DuyLionTran 0:818366b014c9 621 }
DuyLionTran 0:818366b014c9 622 #endif
DuyLionTran 0:818366b014c9 623 #if MQTTCLIENT_QOS1 || MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 624 if (msg.qos != QOS0)
DuyLionTran 0:818366b014c9 625 {
DuyLionTran 0:818366b014c9 626 if (msg.qos == QOS1)
DuyLionTran 0:818366b014c9 627 len = MQTTSerialize_ack(sendbuf, MAX_MQTT_PACKET_SIZE, PUBACK, 0, msg.id);
DuyLionTran 0:818366b014c9 628 else if (msg.qos == QOS2)
DuyLionTran 0:818366b014c9 629 len = MQTTSerialize_ack(sendbuf, MAX_MQTT_PACKET_SIZE, PUBREC, 0, msg.id);
DuyLionTran 0:818366b014c9 630 if (len <= 0)
DuyLionTran 0:818366b014c9 631 rc = FAILURE;
DuyLionTran 0:818366b014c9 632 else
DuyLionTran 0:818366b014c9 633 rc = sendPacket(len, timer);
DuyLionTran 0:818366b014c9 634 if (rc == FAILURE)
DuyLionTran 0:818366b014c9 635 goto exit; // there was a problem
DuyLionTran 0:818366b014c9 636 }
DuyLionTran 0:818366b014c9 637 break;
DuyLionTran 0:818366b014c9 638 #endif
DuyLionTran 0:818366b014c9 639 }
DuyLionTran 0:818366b014c9 640 #if MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 641 case PUBREC:
DuyLionTran 0:818366b014c9 642 case PUBREL:
DuyLionTran 0:818366b014c9 643 unsigned short mypacketid;
DuyLionTran 0:818366b014c9 644 unsigned char dup, type;
DuyLionTran 0:818366b014c9 645 if (MQTTDeserialize_ack(&type, &dup, &mypacketid, readbuf, MAX_MQTT_PACKET_SIZE) != 1)
DuyLionTran 0:818366b014c9 646 rc = FAILURE;
DuyLionTran 0:818366b014c9 647 else if ((len = MQTTSerialize_ack(sendbuf, MAX_MQTT_PACKET_SIZE,
DuyLionTran 0:818366b014c9 648 (packet_type == PUBREC) ? PUBREL : PUBCOMP, 0, mypacketid)) <= 0)
DuyLionTran 0:818366b014c9 649 rc = FAILURE;
DuyLionTran 0:818366b014c9 650 else if ((rc = sendPacket(len, timer)) != SUCCESS) // send the PUBREL packet
DuyLionTran 0:818366b014c9 651 rc = FAILURE; // there was a problem
DuyLionTran 0:818366b014c9 652 if (rc == FAILURE)
DuyLionTran 0:818366b014c9 653 goto exit; // there was a problem
DuyLionTran 0:818366b014c9 654 if (packet_type == PUBREL)
DuyLionTran 0:818366b014c9 655 freeQoS2msgid(mypacketid);
DuyLionTran 0:818366b014c9 656 break;
DuyLionTran 0:818366b014c9 657
DuyLionTran 0:818366b014c9 658 case PUBCOMP:
DuyLionTran 0:818366b014c9 659 break;
DuyLionTran 0:818366b014c9 660 #endif
DuyLionTran 0:818366b014c9 661 case PINGRESP:
DuyLionTran 0:818366b014c9 662 ping_outstanding = false;
DuyLionTran 0:818366b014c9 663 break;
DuyLionTran 0:818366b014c9 664 }
DuyLionTran 0:818366b014c9 665
DuyLionTran 0:818366b014c9 666 if (keepalive() != SUCCESS)
DuyLionTran 0:818366b014c9 667 //check only keepalive FAILURE status so that previous FAILURE status can be considered as FAULT
DuyLionTran 0:818366b014c9 668 rc = FAILURE;
DuyLionTran 0:818366b014c9 669
DuyLionTran 0:818366b014c9 670 exit:
DuyLionTran 0:818366b014c9 671 if (rc == SUCCESS)
DuyLionTran 0:818366b014c9 672 rc = packet_type;
DuyLionTran 0:818366b014c9 673 else if (isconnected)
DuyLionTran 0:818366b014c9 674 closeSession();
DuyLionTran 0:818366b014c9 675 return rc;
DuyLionTran 0:818366b014c9 676 }
DuyLionTran 0:818366b014c9 677
DuyLionTran 0:818366b014c9 678
DuyLionTran 0:818366b014c9 679 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int b>
DuyLionTran 0:818366b014c9 680 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, b>::keepalive()
DuyLionTran 0:818366b014c9 681 {
DuyLionTran 0:818366b014c9 682 int rc = SUCCESS;
DuyLionTran 0:818366b014c9 683 static Timer ping_sent;
DuyLionTran 0:818366b014c9 684
DuyLionTran 0:818366b014c9 685 if (keepAliveInterval == 0)
DuyLionTran 0:818366b014c9 686 goto exit;
DuyLionTran 0:818366b014c9 687
DuyLionTran 0:818366b014c9 688 if (ping_outstanding)
DuyLionTran 0:818366b014c9 689 {
DuyLionTran 0:818366b014c9 690 if (ping_sent.expired())
DuyLionTran 0:818366b014c9 691 {
DuyLionTran 0:818366b014c9 692 rc = FAILURE; // session failure
DuyLionTran 0:818366b014c9 693 #if defined(MQTT_DEBUG)
DuyLionTran 0:818366b014c9 694 DEBUG("PINGRESP not received in keepalive interval\r\n");
DuyLionTran 0:818366b014c9 695 #endif
DuyLionTran 0:818366b014c9 696 }
DuyLionTran 0:818366b014c9 697 }
DuyLionTran 0:818366b014c9 698 else if (last_sent.expired() || last_received.expired())
DuyLionTran 0:818366b014c9 699 {
DuyLionTran 0:818366b014c9 700 Timer timer(1000);
DuyLionTran 0:818366b014c9 701 int len = MQTTSerialize_pingreq(sendbuf, MAX_MQTT_PACKET_SIZE);
DuyLionTran 0:818366b014c9 702 if (len > 0 && (rc = sendPacket(len, timer)) == SUCCESS) // send the ping packet
DuyLionTran 0:818366b014c9 703 {
DuyLionTran 0:818366b014c9 704 ping_outstanding = true;
DuyLionTran 0:818366b014c9 705 ping_sent.countdown(this->keepAliveInterval);
DuyLionTran 0:818366b014c9 706 }
DuyLionTran 0:818366b014c9 707 }
DuyLionTran 0:818366b014c9 708 exit:
DuyLionTran 0:818366b014c9 709 return rc;
DuyLionTran 0:818366b014c9 710 }
DuyLionTran 0:818366b014c9 711
DuyLionTran 0:818366b014c9 712
DuyLionTran 0:818366b014c9 713 // only used in single-threaded mode where one command at a time is in process
DuyLionTran 0:818366b014c9 714 template<class Network, class Timer, int a, int b>
DuyLionTran 0:818366b014c9 715 int MQTT::Client<Network, Timer, a, b>::waitfor(int packet_type, Timer& timer)
DuyLionTran 0:818366b014c9 716 {
DuyLionTran 0:818366b014c9 717 int rc = FAILURE;
DuyLionTran 0:818366b014c9 718
DuyLionTran 0:818366b014c9 719 do
DuyLionTran 0:818366b014c9 720 {
DuyLionTran 0:818366b014c9 721 if (timer.expired())
DuyLionTran 0:818366b014c9 722 break; // we timed out
DuyLionTran 0:818366b014c9 723 rc = cycle(timer);
DuyLionTran 0:818366b014c9 724 }
DuyLionTran 0:818366b014c9 725 while (rc != packet_type && rc >= 0);
DuyLionTran 0:818366b014c9 726
DuyLionTran 0:818366b014c9 727 return rc;
DuyLionTran 0:818366b014c9 728 }
DuyLionTran 0:818366b014c9 729
DuyLionTran 0:818366b014c9 730
DuyLionTran 0:818366b014c9 731 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int b>
DuyLionTran 0:818366b014c9 732 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, b>::connect(MQTTPacket_connectData& options, connackData& data)
DuyLionTran 0:818366b014c9 733 {
DuyLionTran 0:818366b014c9 734 Timer connect_timer(command_timeout_ms);
DuyLionTran 0:818366b014c9 735 int rc = FAILURE;
DuyLionTran 0:818366b014c9 736 int len = 0;
DuyLionTran 0:818366b014c9 737
DuyLionTran 0:818366b014c9 738 if (isconnected) // don't send connect packet again if we are already connected
DuyLionTran 0:818366b014c9 739 goto exit;
DuyLionTran 0:818366b014c9 740
DuyLionTran 0:818366b014c9 741 this->keepAliveInterval = options.keepAliveInterval;
DuyLionTran 0:818366b014c9 742 this->cleansession = options.cleansession;
DuyLionTran 0:818366b014c9 743 if ((len = MQTTSerialize_connect(sendbuf, MAX_MQTT_PACKET_SIZE, &options)) <= 0)
DuyLionTran 0:818366b014c9 744 goto exit;
DuyLionTran 0:818366b014c9 745 if ((rc = sendPacket(len, connect_timer)) != SUCCESS) // send the connect packet
DuyLionTran 0:818366b014c9 746 goto exit; // there was a problem
DuyLionTran 0:818366b014c9 747
DuyLionTran 0:818366b014c9 748 if (this->keepAliveInterval > 0)
DuyLionTran 0:818366b014c9 749 last_received.countdown(this->keepAliveInterval);
DuyLionTran 0:818366b014c9 750 // this will be a blocking call, wait for the connack
DuyLionTran 0:818366b014c9 751 if (waitfor(CONNACK, connect_timer) == CONNACK)
DuyLionTran 0:818366b014c9 752 {
DuyLionTran 0:818366b014c9 753 data.rc = 0;
DuyLionTran 0:818366b014c9 754 data.sessionPresent = false;
DuyLionTran 0:818366b014c9 755 if (MQTTDeserialize_connack((unsigned char*)&data.sessionPresent,
DuyLionTran 0:818366b014c9 756 (unsigned char*)&data.rc, readbuf, MAX_MQTT_PACKET_SIZE) == 1)
DuyLionTran 0:818366b014c9 757 rc = data.rc;
DuyLionTran 0:818366b014c9 758 else
DuyLionTran 0:818366b014c9 759 rc = FAILURE;
DuyLionTran 0:818366b014c9 760 }
DuyLionTran 0:818366b014c9 761 else
DuyLionTran 0:818366b014c9 762 rc = FAILURE;
DuyLionTran 0:818366b014c9 763
DuyLionTran 0:818366b014c9 764 #if MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 765 // resend any inflight publish
DuyLionTran 0:818366b014c9 766 if (inflightMsgid > 0 && inflightQoS == QOS2 && pubrel)
DuyLionTran 0:818366b014c9 767 {
DuyLionTran 0:818366b014c9 768 if ((len = MQTTSerialize_ack(sendbuf, MAX_MQTT_PACKET_SIZE, PUBREL, 0, inflightMsgid)) <= 0)
DuyLionTran 0:818366b014c9 769 rc = FAILURE;
DuyLionTran 0:818366b014c9 770 else
DuyLionTran 0:818366b014c9 771 rc = publish(len, connect_timer, inflightQoS);
DuyLionTran 0:818366b014c9 772 }
DuyLionTran 0:818366b014c9 773 else
DuyLionTran 0:818366b014c9 774 #endif
DuyLionTran 0:818366b014c9 775 #if MQTTCLIENT_QOS1 || MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 776 if (inflightMsgid > 0)
DuyLionTran 0:818366b014c9 777 {
DuyLionTran 0:818366b014c9 778 memcpy(sendbuf, pubbuf, MAX_MQTT_PACKET_SIZE);
DuyLionTran 0:818366b014c9 779 rc = publish(inflightLen, connect_timer, inflightQoS);
DuyLionTran 0:818366b014c9 780 }
DuyLionTran 0:818366b014c9 781 #endif
DuyLionTran 0:818366b014c9 782
DuyLionTran 0:818366b014c9 783 exit:
DuyLionTran 0:818366b014c9 784 if (rc == SUCCESS)
DuyLionTran 0:818366b014c9 785 {
DuyLionTran 0:818366b014c9 786 isconnected = true;
DuyLionTran 0:818366b014c9 787 ping_outstanding = false;
DuyLionTran 0:818366b014c9 788 }
DuyLionTran 0:818366b014c9 789 return rc;
DuyLionTran 0:818366b014c9 790 }
DuyLionTran 0:818366b014c9 791
DuyLionTran 0:818366b014c9 792
DuyLionTran 0:818366b014c9 793 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int b>
DuyLionTran 0:818366b014c9 794 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, b>::connect(MQTTPacket_connectData& options)
DuyLionTran 0:818366b014c9 795 {
DuyLionTran 0:818366b014c9 796 connackData data;
DuyLionTran 0:818366b014c9 797 return connect(options, data);
DuyLionTran 0:818366b014c9 798 }
DuyLionTran 0:818366b014c9 799
DuyLionTran 0:818366b014c9 800
DuyLionTran 0:818366b014c9 801 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int b>
DuyLionTran 0:818366b014c9 802 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, b>::connect()
DuyLionTran 0:818366b014c9 803 {
DuyLionTran 0:818366b014c9 804 MQTTPacket_connectData default_options = MQTTPacket_connectData_initializer;
DuyLionTran 0:818366b014c9 805 return connect(default_options);
DuyLionTran 0:818366b014c9 806 }
DuyLionTran 0:818366b014c9 807
DuyLionTran 0:818366b014c9 808
DuyLionTran 0:818366b014c9 809 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int MAX_MESSAGE_HANDLERS>
DuyLionTran 0:818366b014c9 810 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, MAX_MESSAGE_HANDLERS>::setMessageHandler(const char* topicFilter, messageHandler messageHandler)
DuyLionTran 0:818366b014c9 811 {
DuyLionTran 0:818366b014c9 812 int rc = FAILURE;
DuyLionTran 0:818366b014c9 813 int i = -1;
DuyLionTran 0:818366b014c9 814
DuyLionTran 0:818366b014c9 815 // first check for an existing matching slot
DuyLionTran 0:818366b014c9 816 for (i = 0; i < MAX_MESSAGE_HANDLERS; ++i)
DuyLionTran 0:818366b014c9 817 {
DuyLionTran 0:818366b014c9 818 if (messageHandlers[i].topicFilter != 0 && strcmp(messageHandlers[i].topicFilter, topicFilter) == 0)
DuyLionTran 0:818366b014c9 819 {
DuyLionTran 0:818366b014c9 820 if (messageHandler == 0) // remove existing
DuyLionTran 0:818366b014c9 821 {
DuyLionTran 0:818366b014c9 822 messageHandlers[i].topicFilter = 0;
DuyLionTran 0:818366b014c9 823 messageHandlers[i].fp.detach();
DuyLionTran 0:818366b014c9 824 }
DuyLionTran 0:818366b014c9 825 rc = SUCCESS; // return i when adding new subscription
DuyLionTran 0:818366b014c9 826 break;
DuyLionTran 0:818366b014c9 827 }
DuyLionTran 0:818366b014c9 828 }
DuyLionTran 0:818366b014c9 829 // if no existing, look for empty slot (unless we are removing)
DuyLionTran 0:818366b014c9 830 if (messageHandler != 0) {
DuyLionTran 0:818366b014c9 831 if (rc == FAILURE)
DuyLionTran 0:818366b014c9 832 {
DuyLionTran 0:818366b014c9 833 for (i = 0; i < MAX_MESSAGE_HANDLERS; ++i)
DuyLionTran 0:818366b014c9 834 {
DuyLionTran 0:818366b014c9 835 if (messageHandlers[i].topicFilter == 0)
DuyLionTran 0:818366b014c9 836 {
DuyLionTran 0:818366b014c9 837 rc = SUCCESS;
DuyLionTran 0:818366b014c9 838 break;
DuyLionTran 0:818366b014c9 839 }
DuyLionTran 0:818366b014c9 840 }
DuyLionTran 0:818366b014c9 841 }
DuyLionTran 0:818366b014c9 842 if (i < MAX_MESSAGE_HANDLERS)
DuyLionTran 0:818366b014c9 843 {
DuyLionTran 0:818366b014c9 844 messageHandlers[i].topicFilter = topicFilter;
DuyLionTran 0:818366b014c9 845 messageHandlers[i].fp.attach(messageHandler);
DuyLionTran 0:818366b014c9 846 }
DuyLionTran 0:818366b014c9 847 }
DuyLionTran 0:818366b014c9 848 return rc;
DuyLionTran 0:818366b014c9 849 }
DuyLionTran 0:818366b014c9 850
DuyLionTran 0:818366b014c9 851
DuyLionTran 0:818366b014c9 852 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int MAX_MESSAGE_HANDLERS>
DuyLionTran 0:818366b014c9 853 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, MAX_MESSAGE_HANDLERS>::subscribe(const char* topicFilter,
DuyLionTran 0:818366b014c9 854 enum QoS qos, messageHandler messageHandler, subackData& data)
DuyLionTran 0:818366b014c9 855 {
DuyLionTran 0:818366b014c9 856 int rc = FAILURE;
DuyLionTran 0:818366b014c9 857 Timer timer(command_timeout_ms);
DuyLionTran 0:818366b014c9 858 int len = 0;
DuyLionTran 0:818366b014c9 859 MQTTString topic = {(char*)topicFilter, {0, 0}};
DuyLionTran 0:818366b014c9 860
DuyLionTran 0:818366b014c9 861 if (!isconnected)
DuyLionTran 0:818366b014c9 862 goto exit;
DuyLionTran 0:818366b014c9 863
DuyLionTran 0:818366b014c9 864 len = MQTTSerialize_subscribe(sendbuf, MAX_MQTT_PACKET_SIZE, 0, packetid.getNext(), 1, &topic, (int*)&qos);
DuyLionTran 0:818366b014c9 865 if (len <= 0)
DuyLionTran 0:818366b014c9 866 goto exit;
DuyLionTran 0:818366b014c9 867 if ((rc = sendPacket(len, timer)) != SUCCESS) // send the subscribe packet
DuyLionTran 0:818366b014c9 868 goto exit; // there was a problem
DuyLionTran 0:818366b014c9 869
DuyLionTran 0:818366b014c9 870 if (waitfor(SUBACK, timer) == SUBACK) // wait for suback
DuyLionTran 0:818366b014c9 871 {
DuyLionTran 0:818366b014c9 872 int count = 0;
DuyLionTran 0:818366b014c9 873 unsigned short mypacketid;
DuyLionTran 0:818366b014c9 874 data.grantedQoS = 0;
DuyLionTran 0:818366b014c9 875 if (MQTTDeserialize_suback(&mypacketid, 1, &count, &data.grantedQoS, readbuf, MAX_MQTT_PACKET_SIZE) == 1)
DuyLionTran 0:818366b014c9 876 {
DuyLionTran 0:818366b014c9 877 if (data.grantedQoS != 0x80)
DuyLionTran 0:818366b014c9 878 rc = setMessageHandler(topicFilter, messageHandler);
DuyLionTran 0:818366b014c9 879 }
DuyLionTran 0:818366b014c9 880 }
DuyLionTran 0:818366b014c9 881 else
DuyLionTran 0:818366b014c9 882 rc = FAILURE;
DuyLionTran 0:818366b014c9 883
DuyLionTran 0:818366b014c9 884 exit:
DuyLionTran 0:818366b014c9 885 if (rc == FAILURE)
DuyLionTran 0:818366b014c9 886 closeSession();
DuyLionTran 0:818366b014c9 887 return rc;
DuyLionTran 0:818366b014c9 888 }
DuyLionTran 0:818366b014c9 889
DuyLionTran 0:818366b014c9 890
DuyLionTran 0:818366b014c9 891 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int MAX_MESSAGE_HANDLERS>
DuyLionTran 0:818366b014c9 892 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, MAX_MESSAGE_HANDLERS>::subscribe(const char* topicFilter, enum QoS qos, messageHandler messageHandler)
DuyLionTran 0:818366b014c9 893 {
DuyLionTran 0:818366b014c9 894 subackData data;
DuyLionTran 0:818366b014c9 895 return subscribe(topicFilter, qos, messageHandler, data);
DuyLionTran 0:818366b014c9 896 }
DuyLionTran 0:818366b014c9 897
DuyLionTran 0:818366b014c9 898
DuyLionTran 0:818366b014c9 899 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int MAX_MESSAGE_HANDLERS>
DuyLionTran 0:818366b014c9 900 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, MAX_MESSAGE_HANDLERS>::unsubscribe(const char* topicFilter)
DuyLionTran 0:818366b014c9 901 {
DuyLionTran 0:818366b014c9 902 int rc = FAILURE;
DuyLionTran 0:818366b014c9 903 Timer timer(command_timeout_ms);
DuyLionTran 0:818366b014c9 904 MQTTString topic = {(char*)topicFilter, {0, 0}};
DuyLionTran 0:818366b014c9 905 int len = 0;
DuyLionTran 0:818366b014c9 906
DuyLionTran 0:818366b014c9 907 if (!isconnected)
DuyLionTran 0:818366b014c9 908 goto exit;
DuyLionTran 0:818366b014c9 909
DuyLionTran 0:818366b014c9 910 if ((len = MQTTSerialize_unsubscribe(sendbuf, MAX_MQTT_PACKET_SIZE, 0, packetid.getNext(), 1, &topic)) <= 0)
DuyLionTran 0:818366b014c9 911 goto exit;
DuyLionTran 0:818366b014c9 912 if ((rc = sendPacket(len, timer)) != SUCCESS) // send the unsubscribe packet
DuyLionTran 0:818366b014c9 913 goto exit; // there was a problem
DuyLionTran 0:818366b014c9 914
DuyLionTran 0:818366b014c9 915 if (waitfor(UNSUBACK, timer) == UNSUBACK)
DuyLionTran 0:818366b014c9 916 {
DuyLionTran 0:818366b014c9 917 unsigned short mypacketid; // should be the same as the packetid above
DuyLionTran 0:818366b014c9 918 if (MQTTDeserialize_unsuback(&mypacketid, readbuf, MAX_MQTT_PACKET_SIZE) == 1)
DuyLionTran 0:818366b014c9 919 {
DuyLionTran 0:818366b014c9 920 // remove the subscription message handler associated with this topic, if there is one
DuyLionTran 0:818366b014c9 921 setMessageHandler(topicFilter, 0);
DuyLionTran 0:818366b014c9 922 }
DuyLionTran 0:818366b014c9 923 }
DuyLionTran 0:818366b014c9 924 else
DuyLionTran 0:818366b014c9 925 rc = FAILURE;
DuyLionTran 0:818366b014c9 926
DuyLionTran 0:818366b014c9 927 exit:
DuyLionTran 0:818366b014c9 928 if (rc != SUCCESS)
DuyLionTran 0:818366b014c9 929 closeSession();
DuyLionTran 0:818366b014c9 930 return rc;
DuyLionTran 0:818366b014c9 931 }
DuyLionTran 0:818366b014c9 932
DuyLionTran 0:818366b014c9 933
DuyLionTran 0:818366b014c9 934 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int b>
DuyLionTran 0:818366b014c9 935 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, b>::publish(int len, Timer& timer, enum QoS qos)
DuyLionTran 0:818366b014c9 936 {
DuyLionTran 0:818366b014c9 937 int rc;
DuyLionTran 0:818366b014c9 938
DuyLionTran 0:818366b014c9 939 if ((rc = sendPacket(len, timer)) != SUCCESS) // send the publish packet
DuyLionTran 0:818366b014c9 940 goto exit; // there was a problem
DuyLionTran 0:818366b014c9 941
DuyLionTran 0:818366b014c9 942 #if MQTTCLIENT_QOS1
DuyLionTran 0:818366b014c9 943 if (qos == QOS1)
DuyLionTran 0:818366b014c9 944 {
DuyLionTran 0:818366b014c9 945 if (waitfor(PUBACK, timer) == PUBACK)
DuyLionTran 0:818366b014c9 946 {
DuyLionTran 0:818366b014c9 947 unsigned short mypacketid;
DuyLionTran 0:818366b014c9 948 unsigned char dup, type;
DuyLionTran 0:818366b014c9 949 if (MQTTDeserialize_ack(&type, &dup, &mypacketid, readbuf, MAX_MQTT_PACKET_SIZE) != 1)
DuyLionTran 0:818366b014c9 950 rc = FAILURE;
DuyLionTran 0:818366b014c9 951 else if (inflightMsgid == mypacketid)
DuyLionTran 0:818366b014c9 952 inflightMsgid = 0;
DuyLionTran 0:818366b014c9 953 }
DuyLionTran 0:818366b014c9 954 else
DuyLionTran 0:818366b014c9 955 rc = FAILURE;
DuyLionTran 0:818366b014c9 956 }
DuyLionTran 0:818366b014c9 957 #endif
DuyLionTran 0:818366b014c9 958 #if MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 959 else if (qos == QOS2)
DuyLionTran 0:818366b014c9 960 {
DuyLionTran 0:818366b014c9 961 if (waitfor(PUBCOMP, timer) == PUBCOMP)
DuyLionTran 0:818366b014c9 962 {
DuyLionTran 0:818366b014c9 963 unsigned short mypacketid;
DuyLionTran 0:818366b014c9 964 unsigned char dup, type;
DuyLionTran 0:818366b014c9 965 if (MQTTDeserialize_ack(&type, &dup, &mypacketid, readbuf, MAX_MQTT_PACKET_SIZE) != 1)
DuyLionTran 0:818366b014c9 966 rc = FAILURE;
DuyLionTran 0:818366b014c9 967 else if (inflightMsgid == mypacketid)
DuyLionTran 0:818366b014c9 968 inflightMsgid = 0;
DuyLionTran 0:818366b014c9 969 }
DuyLionTran 0:818366b014c9 970 else
DuyLionTran 0:818366b014c9 971 rc = FAILURE;
DuyLionTran 0:818366b014c9 972 }
DuyLionTran 0:818366b014c9 973 #endif
DuyLionTran 0:818366b014c9 974
DuyLionTran 0:818366b014c9 975 exit:
DuyLionTran 0:818366b014c9 976 if (rc != SUCCESS)
DuyLionTran 0:818366b014c9 977 closeSession();
DuyLionTran 0:818366b014c9 978 return rc;
DuyLionTran 0:818366b014c9 979 }
DuyLionTran 0:818366b014c9 980
DuyLionTran 0:818366b014c9 981
DuyLionTran 0:818366b014c9 982
DuyLionTran 0:818366b014c9 983 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int b>
DuyLionTran 0:818366b014c9 984 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, b>::publish(const char* topicName, void* payload, size_t payloadlen, unsigned short& id, enum QoS qos, bool retained)
DuyLionTran 0:818366b014c9 985 {
DuyLionTran 0:818366b014c9 986 int rc = FAILURE;
DuyLionTran 0:818366b014c9 987 Timer timer(command_timeout_ms);
DuyLionTran 0:818366b014c9 988 MQTTString topicString = MQTTString_initializer;
DuyLionTran 0:818366b014c9 989 int len = 0;
DuyLionTran 0:818366b014c9 990
DuyLionTran 0:818366b014c9 991 if (!isconnected)
DuyLionTran 0:818366b014c9 992 goto exit;
DuyLionTran 0:818366b014c9 993
DuyLionTran 0:818366b014c9 994 topicString.cstring = (char*)topicName;
DuyLionTran 0:818366b014c9 995
DuyLionTran 0:818366b014c9 996 #if MQTTCLIENT_QOS1 || MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 997 if (qos == QOS1 || qos == QOS2)
DuyLionTran 0:818366b014c9 998 id = packetid.getNext();
DuyLionTran 0:818366b014c9 999 #endif
DuyLionTran 0:818366b014c9 1000
DuyLionTran 0:818366b014c9 1001 len = MQTTSerialize_publish(sendbuf, MAX_MQTT_PACKET_SIZE, 0, qos, retained, id,
DuyLionTran 0:818366b014c9 1002 topicString, (unsigned char*)payload, payloadlen);
DuyLionTran 0:818366b014c9 1003 if (len <= 0)
DuyLionTran 0:818366b014c9 1004 goto exit;
DuyLionTran 0:818366b014c9 1005
DuyLionTran 0:818366b014c9 1006 #if MQTTCLIENT_QOS1 || MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 1007 if (!cleansession)
DuyLionTran 0:818366b014c9 1008 {
DuyLionTran 0:818366b014c9 1009 memcpy(pubbuf, sendbuf, len);
DuyLionTran 0:818366b014c9 1010 inflightMsgid = id;
DuyLionTran 0:818366b014c9 1011 inflightLen = len;
DuyLionTran 0:818366b014c9 1012 inflightQoS = qos;
DuyLionTran 0:818366b014c9 1013 #if MQTTCLIENT_QOS2
DuyLionTran 0:818366b014c9 1014 pubrel = false;
DuyLionTran 0:818366b014c9 1015 #endif
DuyLionTran 0:818366b014c9 1016 }
DuyLionTran 0:818366b014c9 1017 #endif
DuyLionTran 0:818366b014c9 1018
DuyLionTran 0:818366b014c9 1019 rc = publish(len, timer, qos);
DuyLionTran 0:818366b014c9 1020 exit:
DuyLionTran 0:818366b014c9 1021 return rc;
DuyLionTran 0:818366b014c9 1022 }
DuyLionTran 0:818366b014c9 1023
DuyLionTran 0:818366b014c9 1024
DuyLionTran 0:818366b014c9 1025 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int b>
DuyLionTran 0:818366b014c9 1026 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, b>::publish(const char* topicName, void* payload, size_t payloadlen, enum QoS qos, bool retained)
DuyLionTran 0:818366b014c9 1027 {
DuyLionTran 0:818366b014c9 1028 unsigned short id = 0; // dummy - not used for anything
DuyLionTran 0:818366b014c9 1029 return publish(topicName, payload, payloadlen, id, qos, retained);
DuyLionTran 0:818366b014c9 1030 }
DuyLionTran 0:818366b014c9 1031
DuyLionTran 0:818366b014c9 1032
DuyLionTran 0:818366b014c9 1033 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int b>
DuyLionTran 0:818366b014c9 1034 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, b>::publish(const char* topicName, Message& message)
DuyLionTran 0:818366b014c9 1035 {
DuyLionTran 0:818366b014c9 1036 return publish(topicName, message.payload, message.payloadlen, message.qos, message.retained);
DuyLionTran 0:818366b014c9 1037 }
DuyLionTran 0:818366b014c9 1038
DuyLionTran 0:818366b014c9 1039
DuyLionTran 0:818366b014c9 1040 template<class Network, class Timer, int MAX_MQTT_PACKET_SIZE, int b>
DuyLionTran 0:818366b014c9 1041 int MQTT::Client<Network, Timer, MAX_MQTT_PACKET_SIZE, b>::disconnect()
DuyLionTran 0:818366b014c9 1042 {
DuyLionTran 0:818366b014c9 1043 int rc = FAILURE;
DuyLionTran 0:818366b014c9 1044 Timer timer(command_timeout_ms); // we might wait for incomplete incoming publishes to complete
DuyLionTran 0:818366b014c9 1045 int len = MQTTSerialize_disconnect(sendbuf, MAX_MQTT_PACKET_SIZE);
DuyLionTran 0:818366b014c9 1046 if (len > 0)
DuyLionTran 0:818366b014c9 1047 rc = sendPacket(len, timer); // send the disconnect packet
DuyLionTran 0:818366b014c9 1048 closeSession();
DuyLionTran 0:818366b014c9 1049 return rc;
DuyLionTran 0:818366b014c9 1050 }
DuyLionTran 0:818366b014c9 1051
DuyLionTran 0:818366b014c9 1052 #endif