wakwak koba / MQTTClient

result of success : 1 to 0, max msg size: 127B to 16383B, fixed existing bugs

Fork of MQTTClient by wakwak koba

MQTTClient.cpp@10:6fa55ad359f0, 2016-01-04 (annotated)

Committer:
wakwak_koba
Date:
Mon Jan 04 07:24:12 2016 +0000
Revision:
10:6fa55ad359f0
Parent:
9:a0e39cea763a
Child:
11:2a8c86bfdeca
20151231?

Who changed what in which revision?

UserRevisionLine numberNew contents of line
SomeRandomBloke 4:9e25b78e0352 1 /*
SomeRandomBloke 0:260fb10c0755 2 MQTTClient.cpp
SomeRandomBloke 0:260fb10c0755 3 Based on MQTTClient from http://ceit.uq.edu.au/content/mqttclient-mbed-version-20
SomeRandomBloke 0:260fb10c0755 4 A simple MQTT client for mbed, version 2.0
SomeRandomBloke 0:260fb10c0755 5 By Yilun FAN, @CEIT, @JAN 2011
SomeRandomBloke 0:260fb10c0755 6
SomeRandomBloke 0:260fb10c0755 7 Bug fixes and additions by Andrew Lindsay (andrew [at] thiseldo [dot] co [dot] uk)
SomeRandomBloke 0:260fb10c0755 8
SomeRandomBloke 0:260fb10c0755 9 Permission is hereby granted, free of charge, to any person obtaining a copy
SomeRandomBloke 0:260fb10c0755 10 of this software and associated documentation files (the "Software"), to deal
SomeRandomBloke 0:260fb10c0755 11 in the Software without restriction, including without limitation the rights
SomeRandomBloke 0:260fb10c0755 12 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
SomeRandomBloke 0:260fb10c0755 13 copies of the Software, and to permit persons to whom the Software is
SomeRandomBloke 0:260fb10c0755 14 furnished to do so, subject to the following conditions:
SomeRandomBloke 0:260fb10c0755 15
SomeRandomBloke 0:260fb10c0755 16 The above copyright notice and this permission notice shall be included in
SomeRandomBloke 0:260fb10c0755 17 all copies or substantial portions of the Software.
SomeRandomBloke 0:260fb10c0755 18
SomeRandomBloke 0:260fb10c0755 19 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
SomeRandomBloke 0:260fb10c0755 20 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
SomeRandomBloke 0:260fb10c0755 21 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
SomeRandomBloke 0:260fb10c0755 22 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
SomeRandomBloke 0:260fb10c0755 23 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
SomeRandomBloke 0:260fb10c0755 24 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
SomeRandomBloke 0:260fb10c0755 25 THE SOFTWARE.
SomeRandomBloke 0:260fb10c0755 26 */
SomeRandomBloke 0:260fb10c0755 27
SomeRandomBloke 0:260fb10c0755 28 #include "MQTTClient.h"
SomeRandomBloke 0:260fb10c0755 29
SomeRandomBloke 1:a50b6c866a3b 30 /** Default Constructor
SomeRandomBloke 1:a50b6c866a3b 31 */
SomeRandomBloke 1:a50b6c866a3b 32 MQTTClient::MQTTClient() {}
SomeRandomBloke 1:a50b6c866a3b 33
SomeRandomBloke 1:a50b6c866a3b 34 /** Default Destructor
SomeRandomBloke 1:a50b6c866a3b 35 */
SomeRandomBloke 1:a50b6c866a3b 36 MQTTClient::~MQTTClient() {}
SomeRandomBloke 1:a50b6c866a3b 37
SomeRandomBloke 3:ac326c5b065c 38 /** Alternative Constructor with parameters
SomeRandomBloke 9:a0e39cea763a 39 *
SomeRandomBloke 9:a0e39cea763a 40 * Allow object to be constructed with minimum parameters.
SomeRandomBloke 1:a50b6c866a3b 41 *
SomeRandomBloke 1:a50b6c866a3b 42 * @param server The IP address of the server to connect to
SomeRandomBloke 1:a50b6c866a3b 43 * @param port The TCP/IP port on the server to connect to
SomeRandomBloke 1:a50b6c866a3b 44 * @param callback Callback function to handle subscription to topics
SomeRandomBloke 1:a50b6c866a3b 45 */
SomeRandomBloke 0:260fb10c0755 46 MQTTClient::MQTTClient(IpAddr server, int port, void (*callback)(char*, char*)) {
SomeRandomBloke 0:260fb10c0755 47 this->port = port;
SomeRandomBloke 0:260fb10c0755 48 callback_server = callback;
wakwak_koba 10:6fa55ad359f0 49
wakwak_koba 10:6fa55ad359f0 50 host.setPort(port);
wakwak_koba 10:6fa55ad359f0 51 host.setIp(server);
wakwak_koba 10:6fa55ad359f0 52 host.setName("localhost");
wakwak_koba 10:6fa55ad359f0 53
SomeRandomBloke 9:a0e39cea763a 54 this->userName = NULL;
SomeRandomBloke 9:a0e39cea763a 55 this->password = NULL;
SomeRandomBloke 0:260fb10c0755 56 connected = false;
SomeRandomBloke 0:260fb10c0755 57 sessionOpened = false;
wakwak_koba 10:6fa55ad359f0 58
wakwak_koba 10:6fa55ad359f0 59 pTCPSocket = NULL;
SomeRandomBloke 0:260fb10c0755 60 timer.start();
SomeRandomBloke 0:260fb10c0755 61 }
SomeRandomBloke 0:260fb10c0755 62
wakwak_koba 10:6fa55ad359f0 63 MQTTClient::MQTTClient(IpAddr server, int port, char *id, char *userName, char *password, void (*callback)(char*, char*)) {
SomeRandomBloke 0:260fb10c0755 64 this->port = port;
SomeRandomBloke 0:260fb10c0755 65 callback_server = callback;
wakwak_koba 10:6fa55ad359f0 66
wakwak_koba 10:6fa55ad359f0 67 host.setPort(port);
wakwak_koba 10:6fa55ad359f0 68 host.setIp(server);
wakwak_koba 10:6fa55ad359f0 69 host.setName("localhost");
wakwak_koba 10:6fa55ad359f0 70
wakwak_koba 10:6fa55ad359f0 71 this->userName = userName;
wakwak_koba 10:6fa55ad359f0 72 this->password = password;
wakwak_koba 10:6fa55ad359f0 73 clientId = id;
SomeRandomBloke 0:260fb10c0755 74 connected = false;
SomeRandomBloke 0:260fb10c0755 75 sessionOpened = false;
wakwak_koba 10:6fa55ad359f0 76
wakwak_koba 10:6fa55ad359f0 77 pTCPSocket = NULL;
SomeRandomBloke 0:260fb10c0755 78 timer.start();
SomeRandomBloke 0:260fb10c0755 79 }
SomeRandomBloke 0:260fb10c0755 80
SomeRandomBloke 1:a50b6c866a3b 81 /** Send a message of specified size
SomeRandomBloke 9:a0e39cea763a 82 *
SomeRandomBloke 1:a50b6c866a3b 83 * @param msg message string
SomeRandomBloke 1:a50b6c866a3b 84 * @param size Size of the message to send
wakwak_koba 10:6fa55ad359f0 85 * @returns value to indicate message sent successfully or not, -1 for error, 0 for success.
SomeRandomBloke 1:a50b6c866a3b 86 */
SomeRandomBloke 0:260fb10c0755 87 int MQTTClient::send_data(const char* msg, int size) {
wakwak_koba 10:6fa55ad359f0 88 writtable = false;
wakwak_koba 10:6fa55ad359f0 89
wakwak_koba 10:6fa55ad359f0 90 /* printf("send_data length %d\n", size);
wakwak_koba 10:6fa55ad359f0 91 for (int i = 0; i < size; i++) {
wakwak_koba 10:6fa55ad359f0 92 printf("%x, ", msg[i]);
wakwak_koba 10:6fa55ad359f0 93 }
wakwak_koba 10:6fa55ad359f0 94 printf("\r\n"); */
wakwak_koba 10:6fa55ad359f0 95
SomeRandomBloke 0:260fb10c0755 96 int transLen = pTCPSocket->send(msg, size);
wakwak_koba 10:6fa55ad359f0 97 pool();
wakwak_koba 10:6fa55ad359f0 98
SomeRandomBloke 0:260fb10c0755 99 /*Check send length matches the message length*/
SomeRandomBloke 0:260fb10c0755 100 if (transLen != size) {
SomeRandomBloke 0:260fb10c0755 101 for (int i = 0; i < size; i++) {
SomeRandomBloke 0:260fb10c0755 102 printf("%x, ", msg[i]);
SomeRandomBloke 0:260fb10c0755 103 }
SomeRandomBloke 0:260fb10c0755 104 printf("Error on send.\r\n");
SomeRandomBloke 0:260fb10c0755 105 return -1;
SomeRandomBloke 0:260fb10c0755 106 }
wakwak_koba 10:6fa55ad359f0 107
wakwak_koba 10:6fa55ad359f0 108 return 0;
SomeRandomBloke 0:260fb10c0755 109 }
SomeRandomBloke 0:260fb10c0755 110
SomeRandomBloke 1:a50b6c866a3b 111 /** Start a MQTT session, build CONNECT packet
SomeRandomBloke 9:a0e39cea763a 112 *
SomeRandomBloke 1:a50b6c866a3b 113 * @param id The client name shown on MQTT server.
wakwak_koba 10:6fa55ad359f0 114 * @returns -1 for error, 0 for success
SomeRandomBloke 1:a50b6c866a3b 115 */
SomeRandomBloke 0:260fb10c0755 116 int MQTTClient::open_session(char* id) {
SomeRandomBloke 0:260fb10c0755 117 /*variable header*/
wakwak_koba 10:6fa55ad359f0 118 char var_header[] = {0x00,0x06,0x4d,0x51,0x49,0x73,0x64,0x70,0x03,0x02,0x00,keepalive/500,0x00,strlen(id)};
SomeRandomBloke 0:260fb10c0755 119
SomeRandomBloke 0:260fb10c0755 120 /*fixed header: 2 bytes, big endian*/
SomeRandomBloke 0:260fb10c0755 121 char fixed_header[] = {MQTTCONNECT,12+strlen(id)+2};
SomeRandomBloke 0:260fb10c0755 122 short usernameLen = strlen( userName );
SomeRandomBloke 0:260fb10c0755 123 short passwordLen = strlen( password );
SomeRandomBloke 0:260fb10c0755 124 var_header[9] |= (usernameLen > 0 ? 0x80 : 0x00 );
SomeRandomBloke 0:260fb10c0755 125 var_header[9] |= (passwordLen > 0 ? 0x40 : 0x00 );
SomeRandomBloke 0:260fb10c0755 126
wakwak_koba 10:6fa55ad359f0 127 printf("fixed %d, var %d id %d, username %d, password %d\n",sizeof(fixed_header), sizeof(var_header), strlen(id), usernameLen, passwordLen );
wakwak_koba 10:6fa55ad359f0 128 char packet[sizeof(fixed_header) + sizeof(var_header) + strlen(id) + usernameLen + (usernameLen > 0 ? 2 : 0) + passwordLen + (passwordLen > 0 ? 2 : 0) ];
SomeRandomBloke 0:260fb10c0755 129
SomeRandomBloke 0:260fb10c0755 130 memset(packet,0,sizeof(packet));
SomeRandomBloke 0:260fb10c0755 131 memcpy(packet,fixed_header,sizeof(fixed_header));
SomeRandomBloke 0:260fb10c0755 132 memcpy(packet+sizeof(fixed_header),var_header,sizeof(var_header));
SomeRandomBloke 0:260fb10c0755 133 memcpy(packet+sizeof(fixed_header)+sizeof(var_header), id, strlen(id));
SomeRandomBloke 0:260fb10c0755 134 if ( usernameLen > 0 ) {
SomeRandomBloke 0:260fb10c0755 135 packet[sizeof(fixed_header)+sizeof(var_header)+strlen(id)] = 0x00;
SomeRandomBloke 0:260fb10c0755 136 packet[sizeof(fixed_header)+sizeof(var_header)+strlen(id)+1] = usernameLen;
SomeRandomBloke 0:260fb10c0755 137 memcpy(packet+sizeof(fixed_header)+sizeof(var_header)+strlen(id)+2, userName, usernameLen);
SomeRandomBloke 0:260fb10c0755 138 packet[1] += (usernameLen + 2);
SomeRandomBloke 0:260fb10c0755 139 }
SomeRandomBloke 0:260fb10c0755 140 if ( passwordLen > 0 ) {
SomeRandomBloke 0:260fb10c0755 141 packet[sizeof(fixed_header)+sizeof(var_header)+strlen(id) + usernameLen + (usernameLen > 0 ? 2 : 0) ] = 0x00;
SomeRandomBloke 0:260fb10c0755 142 packet[sizeof(fixed_header)+sizeof(var_header)+strlen(id) + usernameLen + (usernameLen > 0 ? 2 : 0)+1] = passwordLen;
SomeRandomBloke 0:260fb10c0755 143 memcpy(packet+sizeof(fixed_header)+sizeof(var_header)+strlen(id)+ usernameLen + (usernameLen > 0 ? 2 : 0)+2, password, passwordLen);
SomeRandomBloke 0:260fb10c0755 144 packet[1] += (passwordLen + 2);
SomeRandomBloke 0:260fb10c0755 145 }
SomeRandomBloke 0:260fb10c0755 146
wakwak_koba 10:6fa55ad359f0 147 /* printf("Packet length %d\n", sizeof(packet));
SomeRandomBloke 0:260fb10c0755 148 for (int i = 0; i < sizeof(packet); i++) {
SomeRandomBloke 0:260fb10c0755 149 printf("%x, ", packet[i]);
SomeRandomBloke 0:260fb10c0755 150 }
wakwak_koba 10:6fa55ad359f0 151 printf("\r\n");*/
wakwak_koba 10:6fa55ad359f0 152
wakwak_koba 10:6fa55ad359f0 153 if (send_data(packet, sizeof(packet))) {
SomeRandomBloke 0:260fb10c0755 154 return -1;
SomeRandomBloke 0:260fb10c0755 155 }
wakwak_koba 10:6fa55ad359f0 156 printf("Sent\n");
wakwak_koba 10:6fa55ad359f0 157 return 0;
SomeRandomBloke 0:260fb10c0755 158 }
SomeRandomBloke 0:260fb10c0755 159
SomeRandomBloke 1:a50b6c866a3b 160 /** Open TCP port, connect to server on given IP address.
SomeRandomBloke 9:a0e39cea763a 161 *
wakwak_koba 10:6fa55ad359f0 162 * @returns -1: If connect to server failed. -2: Failed to open session on server. 0: Connection accessed.
wakwak_koba 10:6fa55ad359f0 163 */
wakwak_koba 10:6fa55ad359f0 164 int MQTTClient::connect() {
wakwak_koba 10:6fa55ad359f0 165 return connect(clientId);
wakwak_koba 10:6fa55ad359f0 166 }
wakwak_koba 10:6fa55ad359f0 167
wakwak_koba 10:6fa55ad359f0 168 /** Open TCP port, connect to server on given IP address.
wakwak_koba 10:6fa55ad359f0 169 *
SomeRandomBloke 1:a50b6c866a3b 170 * @param id The client name shown on MQTT server.
wakwak_koba 10:6fa55ad359f0 171 * @returns -1: If connect to server failed. -2: Failed to open session on server. 0: Connection accessed.
SomeRandomBloke 1:a50b6c866a3b 172 */
SomeRandomBloke 0:260fb10c0755 173 int MQTTClient::connect(char* id) {
SomeRandomBloke 0:260fb10c0755 174 /*Initial TCP socket*/
wakwak_koba 10:6fa55ad359f0 175 if(!pTCPSocket) {
wakwak_koba 10:6fa55ad359f0 176 pTCPSocket = new TCPSocket;
wakwak_koba 10:6fa55ad359f0 177 pTCPSocket->setOnEvent(this, &MQTTClient::onTCPSocketEvent);
wakwak_koba 10:6fa55ad359f0 178 }
SomeRandomBloke 0:260fb10c0755 179
SomeRandomBloke 0:260fb10c0755 180 /*Trying to connect to host*/
SomeRandomBloke 0:260fb10c0755 181 printf("Trying to connect to host..\r\n\r\n");
SomeRandomBloke 0:260fb10c0755 182 TCPSocketErr err = pTCPSocket->connect(host);
SomeRandomBloke 0:260fb10c0755 183
SomeRandomBloke 0:260fb10c0755 184 Net::poll();
SomeRandomBloke 0:260fb10c0755 185 if (err) {
SomeRandomBloke 0:260fb10c0755 186 printf("Error connecting to host [%d]\r\n", (int) err);
SomeRandomBloke 0:260fb10c0755 187 return -1;
SomeRandomBloke 0:260fb10c0755 188 }
SomeRandomBloke 9:a0e39cea763a 189 printf("Connect to host..\r\n\r\n");
SomeRandomBloke 0:260fb10c0755 190
SomeRandomBloke 0:260fb10c0755 191 /*Wait TCP connection with server to be established*/
SomeRandomBloke 0:260fb10c0755 192 int i = 0;
SomeRandomBloke 0:260fb10c0755 193 while (!connected) {
SomeRandomBloke 0:260fb10c0755 194 Net::poll();
SomeRandomBloke 0:260fb10c0755 195 wait(1);
SomeRandomBloke 0:260fb10c0755 196 i++;
SomeRandomBloke 0:260fb10c0755 197 printf("Wait for connections %d..\r\n", i);
SomeRandomBloke 0:260fb10c0755 198 if (i == 35) {//If wait too long, give up.
SomeRandomBloke 0:260fb10c0755 199 return -1;
SomeRandomBloke 0:260fb10c0755 200 }
SomeRandomBloke 0:260fb10c0755 201 }
SomeRandomBloke 0:260fb10c0755 202
SomeRandomBloke 0:260fb10c0755 203 /*Send open session message to server*/
wakwak_koba 10:6fa55ad359f0 204 if(open_session(id)) {
wakwak_koba 10:6fa55ad359f0 205 printf("Error open session [%s]\r\n", id);
wakwak_koba 10:6fa55ad359f0 206 return -1;
wakwak_koba 10:6fa55ad359f0 207 }
SomeRandomBloke 0:260fb10c0755 208
SomeRandomBloke 0:260fb10c0755 209 /*Wait server notice of open sesion*/
SomeRandomBloke 7:c12d2bfe40e2 210 i = 0;
SomeRandomBloke 0:260fb10c0755 211 while (!sessionOpened) {
SomeRandomBloke 0:260fb10c0755 212 Net::poll();
SomeRandomBloke 0:260fb10c0755 213 wait(1);
SomeRandomBloke 9:a0e39cea763a 214 if (!connected || ++i >40) {
SomeRandomBloke 0:260fb10c0755 215 break;
SomeRandomBloke 0:260fb10c0755 216 }
SomeRandomBloke 9:a0e39cea763a 217 printf("Wait for session %d..\r\n", i);
SomeRandomBloke 0:260fb10c0755 218 }
wakwak_koba 10:6fa55ad359f0 219 // if (!connected) { wakwak_koba
wakwak_koba 10:6fa55ad359f0 220 if (!connected || !sessionOpened) {
SomeRandomBloke 0:260fb10c0755 221 return -2;
SomeRandomBloke 0:260fb10c0755 222 }
SomeRandomBloke 0:260fb10c0755 223 lastActivity = timer.read_ms();
wakwak_koba 10:6fa55ad359f0 224 return 0;
SomeRandomBloke 0:260fb10c0755 225 }
SomeRandomBloke 0:260fb10c0755 226
SomeRandomBloke 1:a50b6c866a3b 227 /** Publish a message on a topic.
SomeRandomBloke 1:a50b6c866a3b 228 *
SomeRandomBloke 1:a50b6c866a3b 229 * @param pub_topic The topic name the massage will be publish on.
SomeRandomBloke 1:a50b6c866a3b 230 * @param msg The massage to be published.
wakwak_koba 10:6fa55ad359f0 231 * @returns -1: Failed to publish message. 0: Publish sucessed.
SomeRandomBloke 1:a50b6c866a3b 232 */
wakwak_koba 10:6fa55ad359f0 233 int MQTTClient::publish(char* pub_topic, char* msg, bool retain) {
wakwak_koba 10:6fa55ad359f0 234 uint8_t var_header_pub[strlen(pub_topic)+2];
SomeRandomBloke 0:260fb10c0755 235 strcpy((char *)&var_header_pub[2], pub_topic);
SomeRandomBloke 0:260fb10c0755 236 var_header_pub[0] = 0;
SomeRandomBloke 0:260fb10c0755 237 var_header_pub[1] = strlen(pub_topic);
wakwak_koba 10:6fa55ad359f0 238
wakwak_koba 10:6fa55ad359f0 239 // uint8_t fixed_header_pub[] = {MQTTPUBLISH,sizeof(var_header_pub)+strlen(msg)};
wakwak_koba 10:6fa55ad359f0 240 int var_len = sizeof(var_header_pub)+strlen(msg);
wakwak_koba 10:6fa55ad359f0 241 uint8_t fixed_header_pub[1 + (var_len < 128 ? 1 : 2)];
wakwak_koba 10:6fa55ad359f0 242 fixed_header_pub[0] = MQTTPUBLISH | (retain ? 0x01 : 0x00);
SomeRandomBloke 0:260fb10c0755 243
wakwak_koba 10:6fa55ad359f0 244 if(var_len < 128)
wakwak_koba 10:6fa55ad359f0 245 fixed_header_pub[1] = var_len;
wakwak_koba 10:6fa55ad359f0 246 else {
wakwak_koba 10:6fa55ad359f0 247 fixed_header_pub[1] = (var_len % 0x80) | 0x80;
wakwak_koba 10:6fa55ad359f0 248 fixed_header_pub[2] = (var_len / 0x80);
wakwak_koba 10:6fa55ad359f0 249 }
wakwak_koba 10:6fa55ad359f0 250
SomeRandomBloke 0:260fb10c0755 251 uint8_t packet_pub[sizeof(fixed_header_pub)+sizeof(var_header_pub)+strlen(msg)];
SomeRandomBloke 0:260fb10c0755 252 memset(packet_pub,0,sizeof(packet_pub));
SomeRandomBloke 0:260fb10c0755 253 memcpy(packet_pub,fixed_header_pub,sizeof(fixed_header_pub));
SomeRandomBloke 0:260fb10c0755 254 memcpy(packet_pub+sizeof(fixed_header_pub),var_header_pub,sizeof(var_header_pub));
SomeRandomBloke 0:260fb10c0755 255 memcpy(packet_pub+sizeof(fixed_header_pub)+sizeof(var_header_pub),msg,strlen(msg));
SomeRandomBloke 0:260fb10c0755 256
wakwak_koba 10:6fa55ad359f0 257 if (send_data((char*)packet_pub, sizeof(packet_pub))) {
SomeRandomBloke 0:260fb10c0755 258 return -1;
SomeRandomBloke 0:260fb10c0755 259 }
wakwak_koba 10:6fa55ad359f0 260
wakwak_koba 10:6fa55ad359f0 261 return 0;
SomeRandomBloke 0:260fb10c0755 262 }
SomeRandomBloke 0:260fb10c0755 263
SomeRandomBloke 1:a50b6c866a3b 264 /** Disconnect from server
SomeRandomBloke 1:a50b6c866a3b 265 */
SomeRandomBloke 0:260fb10c0755 266 void MQTTClient::disconnect() {
SomeRandomBloke 0:260fb10c0755 267 char packet_224[] = {0xe0, 0x00};
SomeRandomBloke 0:260fb10c0755 268 send_data((char*)packet_224, 2);
SomeRandomBloke 0:260fb10c0755 269
wakwak_koba 10:6fa55ad359f0 270 pTCPSocket->close();
wakwak_koba 10:6fa55ad359f0 271 delete pTCPSocket;
wakwak_koba 10:6fa55ad359f0 272 pTCPSocket = NULL;
SomeRandomBloke 0:260fb10c0755 273 connected = false;
SomeRandomBloke 0:260fb10c0755 274 }
SomeRandomBloke 0:260fb10c0755 275
SomeRandomBloke 1:a50b6c866a3b 276 /** Read data from receive packet
SomeRandomBloke 9:a0e39cea763a 277 * Determine what needs to be done with packet.
SomeRandomBloke 1:a50b6c866a3b 278 */
SomeRandomBloke 0:260fb10c0755 279 void MQTTClient::read_data() {
SomeRandomBloke 0:260fb10c0755 280 char buffer[1024];
SomeRandomBloke 0:260fb10c0755 281 int len = 0, readLen;
SomeRandomBloke 0:260fb10c0755 282
SomeRandomBloke 0:260fb10c0755 283 while ((readLen = pTCPSocket->recv(buffer, 1024)) != 0) {
SomeRandomBloke 0:260fb10c0755 284 len += readLen;
SomeRandomBloke 0:260fb10c0755 285 }
SomeRandomBloke 0:260fb10c0755 286
SomeRandomBloke 0:260fb10c0755 287 buffer[len] = '\0';
SomeRandomBloke 0:260fb10c0755 288
SomeRandomBloke 0:260fb10c0755 289
wakwak_koba 10:6fa55ad359f0 290 /* printf("Read length: %d %d\r\n", len, readLen);
SomeRandomBloke 0:260fb10c0755 291
SomeRandomBloke 0:260fb10c0755 292 for (int i = 0; i < len; i++) {
SomeRandomBloke 0:260fb10c0755 293 printf("%2X ", buffer[i]);
SomeRandomBloke 0:260fb10c0755 294 }
wakwak_koba 10:6fa55ad359f0 295 printf("\r\n"); */
SomeRandomBloke 0:260fb10c0755 296
SomeRandomBloke 0:260fb10c0755 297 char type = buffer[0]>>4;
SomeRandomBloke 0:260fb10c0755 298 if ( type == 2 ) { // CONNACK
SomeRandomBloke 0:260fb10c0755 299 printf("CONNACK\n");
SomeRandomBloke 0:260fb10c0755 300
SomeRandomBloke 0:260fb10c0755 301 } else if (type == 3) { // PUBLISH
SomeRandomBloke 0:260fb10c0755 302 if (callback_server) {
SomeRandomBloke 0:260fb10c0755 303 short index = 1;
SomeRandomBloke 0:260fb10c0755 304 short multiplier = 1;
SomeRandomBloke 9:a0e39cea763a 305 short value = 0;
SomeRandomBloke 0:260fb10c0755 306 uint8_t digit;
SomeRandomBloke 0:260fb10c0755 307 do {
SomeRandomBloke 9:a0e39cea763a 308 digit = buffer[index++];
SomeRandomBloke 9:a0e39cea763a 309 value += (digit & 127) * multiplier;
SomeRandomBloke 0:260fb10c0755 310 multiplier *= 128;
SomeRandomBloke 0:260fb10c0755 311 } while ((digit & 128) != 0);
SomeRandomBloke 0:260fb10c0755 312 printf( "variable length %d, index %d\n", value, index );
SomeRandomBloke 9:a0e39cea763a 313
SomeRandomBloke 0:260fb10c0755 314 // uint8_t tl = (buffer[2]<<3)+buffer[3];
SomeRandomBloke 0:260fb10c0755 315 uint8_t tl = (buffer[index]<<3)+buffer[index+1];
SomeRandomBloke 0:260fb10c0755 316 printf("Topic len %d\n",tl);
SomeRandomBloke 0:260fb10c0755 317 char topic[tl+1];
SomeRandomBloke 0:260fb10c0755 318 for (int i=0; i<tl; i++) {
SomeRandomBloke 0:260fb10c0755 319 topic[i] = buffer[index+2+i];
SomeRandomBloke 0:260fb10c0755 320 }
SomeRandomBloke 0:260fb10c0755 321 topic[tl] = 0;
SomeRandomBloke 0:260fb10c0755 322 // ignore msgID - only support QoS 0 subs
SomeRandomBloke 0:260fb10c0755 323 char *payload = buffer+index+2+tl;
SomeRandomBloke 0:260fb10c0755 324 callback_server(topic,(char*)payload);
SomeRandomBloke 0:260fb10c0755 325 }
SomeRandomBloke 0:260fb10c0755 326 } else if (type == 12) { // PINGREG -- Ask for alive
SomeRandomBloke 0:260fb10c0755 327 char packet_208[] = {0xd0, 0x00};
SomeRandomBloke 0:260fb10c0755 328 send_data((char*)packet_208, 2);
SomeRandomBloke 0:260fb10c0755 329 lastActivity = timer.read_ms();
SomeRandomBloke 0:260fb10c0755 330 }
SomeRandomBloke 0:260fb10c0755 331 }
SomeRandomBloke 0:260fb10c0755 332
SomeRandomBloke 1:a50b6c866a3b 333 /** Check for session opened
SomeRandomBloke 1:a50b6c866a3b 334 */
SomeRandomBloke 0:260fb10c0755 335 void MQTTClient::read_open_session() {
SomeRandomBloke 0:260fb10c0755 336 char buffer[32];
SomeRandomBloke 0:260fb10c0755 337 int len = 0, readLen;
SomeRandomBloke 0:260fb10c0755 338
SomeRandomBloke 0:260fb10c0755 339 while ((readLen = pTCPSocket->recv(buffer, 32)) != 0) {
SomeRandomBloke 0:260fb10c0755 340 len += readLen;
SomeRandomBloke 0:260fb10c0755 341 }
SomeRandomBloke 0:260fb10c0755 342
SomeRandomBloke 0:260fb10c0755 343 if (len == 4 && buffer[3] == 0) {
SomeRandomBloke 0:260fb10c0755 344 printf("Session opened\r\n");
SomeRandomBloke 0:260fb10c0755 345 sessionOpened = true;
SomeRandomBloke 0:260fb10c0755 346 }
SomeRandomBloke 0:260fb10c0755 347 }
SomeRandomBloke 0:260fb10c0755 348
SomeRandomBloke 1:a50b6c866a3b 349 /** Subscribe to a topic
SomeRandomBloke 9:a0e39cea763a 350 *
SomeRandomBloke 1:a50b6c866a3b 351 * @param topic The topic name to be subscribed.
wakwak_koba 10:6fa55ad359f0 352 * @returns -1: Failed to subscribe to topic. 0: Subscribe sucessed.
SomeRandomBloke 1:a50b6c866a3b 353 */
SomeRandomBloke 0:260fb10c0755 354 int MQTTClient::subscribe(char* topic) {
SomeRandomBloke 0:260fb10c0755 355
SomeRandomBloke 0:260fb10c0755 356 if (connected) {
SomeRandomBloke 0:260fb10c0755 357 uint8_t var_header_topic[] = {0,10};
SomeRandomBloke 0:260fb10c0755 358 uint8_t fixed_header_topic[] = {MQTTSUBSCRIBE,sizeof(var_header_topic)+strlen(topic)+3};
SomeRandomBloke 0:260fb10c0755 359
SomeRandomBloke 0:260fb10c0755 360 //utf topic
SomeRandomBloke 0:260fb10c0755 361 uint8_t utf_topic[strlen(topic)+3];
SomeRandomBloke 0:260fb10c0755 362 strcpy((char *)&utf_topic[2], topic);
SomeRandomBloke 0:260fb10c0755 363
SomeRandomBloke 0:260fb10c0755 364 utf_topic[0] = 0;
SomeRandomBloke 0:260fb10c0755 365 utf_topic[1] = strlen(topic);
SomeRandomBloke 0:260fb10c0755 366 utf_topic[sizeof(utf_topic)-1] = 0;
SomeRandomBloke 0:260fb10c0755 367
SomeRandomBloke 0:260fb10c0755 368 char packet_topic[sizeof(var_header_topic)+sizeof(fixed_header_topic)+strlen(topic)+3];
SomeRandomBloke 0:260fb10c0755 369 memset(packet_topic,0,sizeof(packet_topic));
SomeRandomBloke 0:260fb10c0755 370 memcpy(packet_topic,fixed_header_topic,sizeof(fixed_header_topic));
SomeRandomBloke 0:260fb10c0755 371 memcpy(packet_topic+sizeof(fixed_header_topic),var_header_topic,sizeof(var_header_topic));
SomeRandomBloke 0:260fb10c0755 372 memcpy(packet_topic+sizeof(fixed_header_topic)+sizeof(var_header_topic),utf_topic,sizeof(utf_topic));
SomeRandomBloke 0:260fb10c0755 373
wakwak_koba 10:6fa55ad359f0 374 if (send_data(packet_topic, sizeof(packet_topic))) {
SomeRandomBloke 0:260fb10c0755 375 return -1;
SomeRandomBloke 0:260fb10c0755 376 }
wakwak_koba 10:6fa55ad359f0 377 return 0;
SomeRandomBloke 0:260fb10c0755 378 }
SomeRandomBloke 0:260fb10c0755 379 return -1;
SomeRandomBloke 0:260fb10c0755 380 }
SomeRandomBloke 0:260fb10c0755 381
SomeRandomBloke 1:a50b6c866a3b 382 /** Send heartbeat/keep-alive message
SomeRandomBloke 9:a0e39cea763a 383 *
SomeRandomBloke 1:a50b6c866a3b 384 */
SomeRandomBloke 0:260fb10c0755 385 void MQTTClient::live() {
SomeRandomBloke 0:260fb10c0755 386 if (connected) {
SomeRandomBloke 0:260fb10c0755 387 int t = timer.read_ms();
wakwak_koba 10:6fa55ad359f0 388 if (t - lastActivity > keepalive) {
SomeRandomBloke 0:260fb10c0755 389 //Send 192 0 to broker
wakwak_koba 10:6fa55ad359f0 390 //printf("Send 192\r\n");
SomeRandomBloke 0:260fb10c0755 391 char packet_192[] = {0xc0, 0x00};
SomeRandomBloke 0:260fb10c0755 392 send_data((char*)packet_192, 2);
SomeRandomBloke 0:260fb10c0755 393 lastActivity = t;
SomeRandomBloke 0:260fb10c0755 394 }
SomeRandomBloke 0:260fb10c0755 395 }
SomeRandomBloke 0:260fb10c0755 396 }
SomeRandomBloke 0:260fb10c0755 397
wakwak_koba 10:6fa55ad359f0 398 void MQTTClient::pool() {
wakwak_koba 10:6fa55ad359f0 399 Net::poll();
wakwak_koba 10:6fa55ad359f0 400 }
wakwak_koba 10:6fa55ad359f0 401
SomeRandomBloke 1:a50b6c866a3b 402 /** TCP Socket event handling
SomeRandomBloke 1:a50b6c866a3b 403 *
SomeRandomBloke 1:a50b6c866a3b 404 * @param e Event object
SomeRandomBloke 1:a50b6c866a3b 405 */
SomeRandomBloke 0:260fb10c0755 406 void MQTTClient::onTCPSocketEvent(TCPSocketEvent e) {
SomeRandomBloke 0:260fb10c0755 407 switch (e) {
SomeRandomBloke 0:260fb10c0755 408 case TCPSOCKET_ACCEPT:
SomeRandomBloke 0:260fb10c0755 409 printf("New TCPSocketEvent: TCPSOCKET_ACCEPT\r\n");
SomeRandomBloke 0:260fb10c0755 410 break;
SomeRandomBloke 0:260fb10c0755 411 case TCPSOCKET_CONNECTED:
SomeRandomBloke 0:260fb10c0755 412 printf("New TCPSocketEvent: TCPSOCKET_CONNECTED\r\n");
SomeRandomBloke 0:260fb10c0755 413 connected = true;
SomeRandomBloke 0:260fb10c0755 414 break;
SomeRandomBloke 0:260fb10c0755 415 case TCPSOCKET_WRITEABLE:
wakwak_koba 10:6fa55ad359f0 416 // printf("New TCPSocketEvent: TCPSOCKET_WRITEABLE\r\n");
wakwak_koba 10:6fa55ad359f0 417 writtable = true;
SomeRandomBloke 0:260fb10c0755 418 break;
SomeRandomBloke 0:260fb10c0755 419 case TCPSOCKET_READABLE:
SomeRandomBloke 0:260fb10c0755 420 printf("New TCPSocketEvent: TCPSOCKET_READABLE\r\n");
SomeRandomBloke 0:260fb10c0755 421 if (!sessionOpened) {
SomeRandomBloke 0:260fb10c0755 422 read_open_session();
SomeRandomBloke 0:260fb10c0755 423 }
SomeRandomBloke 0:260fb10c0755 424 read_data();
SomeRandomBloke 0:260fb10c0755 425 break;
SomeRandomBloke 0:260fb10c0755 426 case TCPSOCKET_CONTIMEOUT:
SomeRandomBloke 0:260fb10c0755 427 printf("New TCPSocketEvent: TCPSOCKET_CONTIMEOUT\r\n");
SomeRandomBloke 0:260fb10c0755 428 break;
SomeRandomBloke 0:260fb10c0755 429 case TCPSOCKET_CONRST:
SomeRandomBloke 0:260fb10c0755 430 printf("New TCPSocketEvent: TCPSOCKET_CONRST\r\n");
SomeRandomBloke 0:260fb10c0755 431 break;
SomeRandomBloke 0:260fb10c0755 432 case TCPSOCKET_CONABRT:
SomeRandomBloke 0:260fb10c0755 433 printf("New TCPSocketEvent: TCPSOCKET_CONABRT\r\n");
SomeRandomBloke 0:260fb10c0755 434 break;
SomeRandomBloke 0:260fb10c0755 435 case TCPSOCKET_ERROR:
SomeRandomBloke 0:260fb10c0755 436 printf("New TCPSocketEvent: TCPSOCKET_ERROR\r\n");
SomeRandomBloke 0:260fb10c0755 437 break;
SomeRandomBloke 0:260fb10c0755 438 case TCPSOCKET_DISCONNECTED:
SomeRandomBloke 0:260fb10c0755 439 printf("New TCPSocketEvent: TCPSOCKET_DISCONNECTED\r\n");
SomeRandomBloke 0:260fb10c0755 440 pTCPSocket->close();
SomeRandomBloke 0:260fb10c0755 441 connected = false;
SomeRandomBloke 0:260fb10c0755 442 break;
SomeRandomBloke 0:260fb10c0755 443 }
SomeRandomBloke 0:260fb10c0755 444 }