Juan Carlos Jimenez / niMQTT

niMQTT

Dependencies:   EthernetInterface mbed-rtos

Fork of niMQTT by Guilhem Saurel

niMQTT.cpp@9:3be69efa4402, 2014-03-03 (annotated)

Committer:
Palantir
Date:
Mon Mar 03 14:53:08 2014 +0000
Revision:
9:3be69efa4402
Parent:
8:438958bb9df3 
Add second thread, so we don't block

Who changed what in which revision?

UserRevisionLine numberNew contents of line
Nim65s 0:d4dfed20c6ea 1 #include "niMQTT.h"
Nim65s 0:d4dfed20c6ea 2
Palantir 9:3be69efa4402 3 niMQTT::niMQTT(char *server, void (*callback)(const char*, const char*),
Palantir 9:3be69efa4402 4 char *id, int port, char *username, char *password, bool debug):
Palantir 9:3be69efa4402 5 server(server), port(port), id(id), callback(callback), username(username),
Palantir 9:3be69efa4402 6 password(password),
Palantir 9:3be69efa4402 7 debug(debug), connected(true), message_id(0),
Palantir 9:3be69efa4402 8 ping_thread(&niMQTT::ping_thread_starter, this),
Palantir 9:3be69efa4402 9 recv_thread(&niMQTT::recv_thread_starter, this),
Palantir 9:3be69efa4402 10 waiting_new_packet(true), packet_sent(false), waiting_connack(0),
Palantir 9:3be69efa4402 11 waiting_suback(0), waiting_pingresp(0)
Palantir 9:3be69efa4402 12 {
Nim65s 0:d4dfed20c6ea 13 init();
Nim65s 0:d4dfed20c6ea 14 }
Nim65s 0:d4dfed20c6ea 15
Nim65s 0:d4dfed20c6ea 16 int niMQTT::init() {
Nim65s 0:d4dfed20c6ea 17 if (debug) printf("*init\r\n");
Nim65s 0:d4dfed20c6ea 18 socket = new TCPSocketConnection;
Nim65s 0:d4dfed20c6ea 19 do printf("Socket connection...\r\n"); while (socket->connect(server, port) < 0);
Nim65s 0:d4dfed20c6ea 20 socket->set_blocking(true, KEEP_ALIVE*500); // KEEP_ALIVE / 2 in seconds
Nim65s 0:d4dfed20c6ea 21
Nim65s 0:d4dfed20c6ea 22 printf("Socket connected.\r\n");
Nim65s 0:d4dfed20c6ea 23
Palantir 9:3be69efa4402 24 ping_thread.signal_set(START_THREAD);
Palantir 9:3be69efa4402 25 recv_thread.signal_set(START_THREAD);
Nim65s 0:d4dfed20c6ea 26
Nim65s 0:d4dfed20c6ea 27 return connect();
Nim65s 0:d4dfed20c6ea 28 }
Nim65s 0:d4dfed20c6ea 29
Nim65s 0:d4dfed20c6ea 30 int niMQTT::send(char *packet, int size) {
Nim65s 0:d4dfed20c6ea 31 //if (debug) {
Nim65s 0:d4dfed20c6ea 32 printf("*send: ");
Nim65s 0:d4dfed20c6ea 33 for(int i=0; i<size; i++) printf("0x%x ", packet[i]);
Nim65s 0:d4dfed20c6ea 34 printf("\r\n");
Nim65s 0:d4dfed20c6ea 35 //}
Nim65s 0:d4dfed20c6ea 36
Nim65s 0:d4dfed20c6ea 37 int j = -1;
Nim65s 0:d4dfed20c6ea 38 do j = socket->send_all(packet, size); while (j < 0);
Nim65s 0:d4dfed20c6ea 39
Nim65s 0:d4dfed20c6ea 40 if (j != size) printf ("%d bytes sent (%d expected)...\r\n", j, size);
Nim65s 0:d4dfed20c6ea 41 else if (debug) printf("packet sent\r\n");
Nim65s 0:d4dfed20c6ea 42 packet_sent = true;
Nim65s 0:d4dfed20c6ea 43
Nim65s 0:d4dfed20c6ea 44 return (j == size);
Nim65s 0:d4dfed20c6ea 45 }
Nim65s 0:d4dfed20c6ea 46
Nim65s 0:d4dfed20c6ea 47 int niMQTT::recv() {
Nim65s 0:d4dfed20c6ea 48 if (debug) printf("*recv\r\n");
Nim65s 0:d4dfed20c6ea 49
Nim65s 0:d4dfed20c6ea 50 int timeout = 0;
Nim65s 0:d4dfed20c6ea 51 while (!waiting_new_packet && timeout++ != TIMEOUT/100) wait(0.1);
Nim65s 0:d4dfed20c6ea 52 if (timeout >= TIMEOUT/100) {
Nim65s 0:d4dfed20c6ea 53 printf("RECV TIMEOUT\r\n");
Nim65s 0:d4dfed20c6ea 54 if (waiting_connack > 0) printf("CONNACK not received !\r\n");
Nim65s 7:d01d8f0bac58 55 if (waiting_suback > 0) printf("SUBACK not received !\r\n");
Nim65s 0:d4dfed20c6ea 56 if (waiting_pingresp > 0) printf("PINGRESP not received !\r\n");
Nim65s 4:afbc7b066cff 57 reconnect();
Nim65s 0:d4dfed20c6ea 58 return -1;
Nim65s 0:d4dfed20c6ea 59 }
Nim65s 0:d4dfed20c6ea 60
Nim65s 0:d4dfed20c6ea 61 if (debug) printf("Receiving new packet...\r\n");
Nim65s 0:d4dfed20c6ea 62
Nim65s 0:d4dfed20c6ea 63 char header_received;
Nim65s 0:d4dfed20c6ea 64 socket->receive(&header_received, 1);
Nim65s 0:d4dfed20c6ea 65 if (debug) printf("Received 0x%x\r\n", header_received);
Nim65s 0:d4dfed20c6ea 66
Nim65s 0:d4dfed20c6ea 67 waiting_new_packet = false;
Nim65s 5:3d21020a2826 68 //bool DUP = ((header_received & 4) == 4);
Nim65s 5:3d21020a2826 69 //int QoS = (header_received & 6);
Nim65s 5:3d21020a2826 70 //bool RETAIN = ((header_received & 1) == 1);
Nim65s 0:d4dfed20c6ea 71
Nim65s 0:d4dfed20c6ea 72 switch (header_received & 0xf0) {
Nim65s 0:d4dfed20c6ea 73 case CONNACK: connack(); break;
Nim65s 0:d4dfed20c6ea 74 case PUBLISH: publish_received(); break;
Nim65s 0:d4dfed20c6ea 75 case PUBACK: puback_received(); break;
Nim65s 0:d4dfed20c6ea 76 case SUBACK: suback(); break;
Nim65s 0:d4dfed20c6ea 77 case UNSUBACK: suback(true); break;
Nim65s 0:d4dfed20c6ea 78 case PINGRESP: pingresp(); break;
Nim65s 4:afbc7b066cff 79 default: waiting_new_packet = true; reconnect(); printf("BAD HEADER: 0x%x\r\n", header_received); return -1;
Nim65s 0:d4dfed20c6ea 80 }
Nim65s 0:d4dfed20c6ea 81
Nim65s 0:d4dfed20c6ea 82 return 0;
Nim65s 0:d4dfed20c6ea 83 }
Nim65s 0:d4dfed20c6ea 84
Nim65s 0:d4dfed20c6ea 85 int niMQTT::connect() {
Nim65s 0:d4dfed20c6ea 86 if (debug) printf("*connect\r\n");
Nim65s 0:d4dfed20c6ea 87 int username_length = strlen(username);
Nim65s 0:d4dfed20c6ea 88 int password_length = strlen(password);
Nim65s 0:d4dfed20c6ea 89 int id_length = strlen(id);
Nim65s 0:d4dfed20c6ea 90
Nim65s 0:d4dfed20c6ea 91 int use_username = (username_length != 0);
Nim65s 0:d4dfed20c6ea 92 int use_password = (password_length != 0);
Nim65s 0:d4dfed20c6ea 93
Nim65s 0:d4dfed20c6ea 94 char variable_header[] = {0,6,77,81,73,115,100,112,3,
Nim65s 0:d4dfed20c6ea 95 use_username << 7 | use_password << 6,
Nim65s 0:d4dfed20c6ea 96 KEEP_ALIVE / 256, KEEP_ALIVE % 256 };
Nim65s 0:d4dfed20c6ea 97
Nim65s 0:d4dfed20c6ea 98 int remaining_length = 14 + id_length + username_length + password_length + 2*(use_username + use_password);
Nim65s 0:d4dfed20c6ea 99 int packet_length = 2 + remaining_length;
Nim65s 0:d4dfed20c6ea 100
Nim65s 0:d4dfed20c6ea 101 char fixed_header[] = { CONNECT, remaining_length };
Nim65s 0:d4dfed20c6ea 102
Nim65s 0:d4dfed20c6ea 103 char packet[packet_length];
Nim65s 0:d4dfed20c6ea 104 memcpy(packet, fixed_header, 2);
Nim65s 0:d4dfed20c6ea 105 memcpy(packet + 2, variable_header, 12);
Nim65s 0:d4dfed20c6ea 106
Nim65s 0:d4dfed20c6ea 107 // Adds the payload: id
Nim65s 0:d4dfed20c6ea 108 char id_size[2] = { id_length / 256, id_length % 256 };
Nim65s 0:d4dfed20c6ea 109 memcpy(packet + 14, id_size, 2);
Nim65s 0:d4dfed20c6ea 110 memcpy(packet + 16, id, id_length);
Nim65s 0:d4dfed20c6ea 111
Nim65s 0:d4dfed20c6ea 112 // Adds username & Password to the payload
Nim65s 0:d4dfed20c6ea 113 if (use_username) {
Nim65s 0:d4dfed20c6ea 114 char username_size[2] = { username_length / 256, username_length % 256 };
Nim65s 0:d4dfed20c6ea 115 memcpy(packet + 16 + id_length, username_size, 2);
Nim65s 0:d4dfed20c6ea 116 memcpy(packet + 18 + id_length, username, username_length);
Nim65s 0:d4dfed20c6ea 117 }
Nim65s 0:d4dfed20c6ea 118 if (use_password) {
Nim65s 0:d4dfed20c6ea 119 char password_size[2] = { password_length / 256, password_length % 256 };
Nim65s 0:d4dfed20c6ea 120 memcpy(packet + 18 + id_length + username_length, password_size, 2);
Nim65s 0:d4dfed20c6ea 121 memcpy(packet + 20 + id_length + username_length, password, password_length);
Nim65s 0:d4dfed20c6ea 122 }
Nim65s 0:d4dfed20c6ea 123
Nim65s 0:d4dfed20c6ea 124 waiting_connack++;
Nim65s 0:d4dfed20c6ea 125
Nim65s 0:d4dfed20c6ea 126 return send(packet, packet_length);
Nim65s 0:d4dfed20c6ea 127 }
Nim65s 0:d4dfed20c6ea 128
Nim65s 0:d4dfed20c6ea 129 int niMQTT::connack() {
Nim65s 0:d4dfed20c6ea 130 if (debug) printf("CONNACK Received\r\n");
Nim65s 0:d4dfed20c6ea 131 if (waiting_connack > 0) waiting_connack--;
Nim65s 4:afbc7b066cff 132 else {
Nim65s 4:afbc7b066cff 133 printf("CONNACK UNEXPECTED !\r\n");
Nim65s 4:afbc7b066cff 134 reconnect();
Nim65s 4:afbc7b066cff 135 return -2;
Nim65s 4:afbc7b066cff 136 }
Nim65s 0:d4dfed20c6ea 137
Nim65s 0:d4dfed20c6ea 138 char resp[3];
Nim65s 0:d4dfed20c6ea 139 socket->receive(resp, 3);
Nim65s 0:d4dfed20c6ea 140 waiting_new_packet = true;
Nim65s 0:d4dfed20c6ea 141
Nim65s 4:afbc7b066cff 142 if (resp[0] != 0x2) {
Nim65s 4:afbc7b066cff 143 printf("Wrong second byte of CONNACK, get 0x%x instead of 0x2\r\n", resp[1]);
Nim65s 4:afbc7b066cff 144 reconnect();
Nim65s 4:afbc7b066cff 145 return -2;
Nim65s 4:afbc7b066cff 146 }
Nim65s 0:d4dfed20c6ea 147 switch (resp[2]) {
Nim65s 0:d4dfed20c6ea 148 case 0: printf("Connection Accepted\r\n"); break;
Nim65s 0:d4dfed20c6ea 149 case 1: printf("Connection Refused: unacceptable protocol version\r\n"); break;
Nim65s 0:d4dfed20c6ea 150 case 2: printf("Connection Refused: identifier rejected\r\n"); break;
Nim65s 0:d4dfed20c6ea 151 case 3: printf("Connection Refused: server unavailable\r\n"); break;
Nim65s 0:d4dfed20c6ea 152 case 4: printf("Connection Refused: bad user name or password\r\n"); break;
Nim65s 0:d4dfed20c6ea 153 case 5: printf("Connection Refused: not authorized\r\n"); break;
Nim65s 0:d4dfed20c6ea 154 default: printf("I have no idea what I am doing\r\n");
Nim65s 0:d4dfed20c6ea 155 }
Nim65s 0:d4dfed20c6ea 156
Nim65s 0:d4dfed20c6ea 157 return (resp[2] == 0);
Nim65s 0:d4dfed20c6ea 158 }
Nim65s 0:d4dfed20c6ea 159
Nim65s 0:d4dfed20c6ea 160 int niMQTT::pub(char *topic, char *message) {
Nim65s 0:d4dfed20c6ea 161 if (debug) printf("*pub\r\n");
Nim65s 0:d4dfed20c6ea 162 int topic_length = strlen(topic);
Nim65s 0:d4dfed20c6ea 163 int message_length = strlen(message);
Nim65s 0:d4dfed20c6ea 164
Nim65s 0:d4dfed20c6ea 165 int remaining_length = topic_length + message_length + 2;
Nim65s 0:d4dfed20c6ea 166 int remaining_length_2 = remaining_length_length(remaining_length);
Nim65s 0:d4dfed20c6ea 167 int packet_length = 1 + remaining_length + remaining_length_2;
Nim65s 0:d4dfed20c6ea 168
Nim65s 0:d4dfed20c6ea 169 char header = PUBLISH;
Nim65s 0:d4dfed20c6ea 170 char packet[packet_length];
Nim65s 0:d4dfed20c6ea 171 // header
Nim65s 0:d4dfed20c6ea 172 memcpy(packet, &header, 1);
Nim65s 0:d4dfed20c6ea 173 get_remaining_length(remaining_length, packet);
Nim65s 0:d4dfed20c6ea 174
Nim65s 0:d4dfed20c6ea 175 // variable header: topic name
Nim65s 0:d4dfed20c6ea 176 char topic_size[2] = { topic_length / 256, topic_length % 256 };
Nim65s 0:d4dfed20c6ea 177 memcpy(packet + 1 + remaining_length_2, topic_size, 2);
Nim65s 0:d4dfed20c6ea 178 memcpy(packet + 3 + remaining_length_2, topic, topic_length);
Nim65s 0:d4dfed20c6ea 179
Nim65s 0:d4dfed20c6ea 180 // payload: message
Nim65s 0:d4dfed20c6ea 181 memcpy(packet + 3 + remaining_length_2 + topic_length, message, message_length);
Nim65s 0:d4dfed20c6ea 182
Nim65s 0:d4dfed20c6ea 183 return send(packet, packet_length);
Nim65s 0:d4dfed20c6ea 184 }
Nim65s 0:d4dfed20c6ea 185
Nim65s 0:d4dfed20c6ea 186 void niMQTT::publish_received() {
Nim65s 0:d4dfed20c6ea 187 //remaining length
Nim65s 0:d4dfed20c6ea 188 int remaining_length = decode_remaining_length();
Nim65s 0:d4dfed20c6ea 189
Nim65s 0:d4dfed20c6ea 190 // topic
Nim65s 0:d4dfed20c6ea 191 char mqtt_utf8_length[2];
Nim65s 0:d4dfed20c6ea 192 socket->receive(mqtt_utf8_length, 2);
Nim65s 0:d4dfed20c6ea 193 int utf8_length = mqtt_utf8_length[0] * 256 + mqtt_utf8_length[1];
Nim65s 0:d4dfed20c6ea 194
Nim65s 0:d4dfed20c6ea 195 if (debug) printf("PUBLISH Received: %i, %i\r\n", remaining_length, utf8_length);
Nim65s 0:d4dfed20c6ea 196
Nim65s 0:d4dfed20c6ea 197 char topic[utf8_length + 1];
Nim65s 0:d4dfed20c6ea 198 socket->receive(topic, utf8_length);
Nim65s 0:d4dfed20c6ea 199 topic[utf8_length] = 0;
Nim65s 0:d4dfed20c6ea 200
Nim65s 0:d4dfed20c6ea 201 // payload
Nim65s 0:d4dfed20c6ea 202 int message_length = remaining_length - utf8_length - 2;
Nim65s 0:d4dfed20c6ea 203 char message[message_length + 1];
Nim65s 0:d4dfed20c6ea 204 socket->receive(message, message_length);
Nim65s 0:d4dfed20c6ea 205 message[message_length] = 0;
Nim65s 0:d4dfed20c6ea 206
Nim65s 0:d4dfed20c6ea 207 waiting_new_packet = true;
Nim65s 0:d4dfed20c6ea 208
Nim65s 8:438958bb9df3 209 call_callback(topic, message);
Nim65s 0:d4dfed20c6ea 210 }
Nim65s 0:d4dfed20c6ea 211
Nim65s 0:d4dfed20c6ea 212 int niMQTT::puback() {
Nim65s 0:d4dfed20c6ea 213 if (debug) printf("*puback\r\n");
Nim65s 0:d4dfed20c6ea 214 char fixed_header[] = { PUBACK, 2, message_id / 256, message_id % 256 };
Nim65s 0:d4dfed20c6ea 215 return send(fixed_header, 4);
Nim65s 0:d4dfed20c6ea 216 }
Nim65s 0:d4dfed20c6ea 217
Nim65s 0:d4dfed20c6ea 218 int niMQTT::puback_received() {
Nim65s 0:d4dfed20c6ea 219 waiting_new_packet = true;
Nim65s 0:d4dfed20c6ea 220 return 0; // TODO
Nim65s 0:d4dfed20c6ea 221 }
Nim65s 0:d4dfed20c6ea 222
Nim65s 0:d4dfed20c6ea 223 int niMQTT::sub(char *topic, bool unsub) {
Nim65s 0:d4dfed20c6ea 224 if (debug) printf("*sub\r\n");
Nim65s 0:d4dfed20c6ea 225 char command = (unsub) ? UNSUBSCRIBE : SUBSCRIBE;
Nim65s 0:d4dfed20c6ea 226 int topic_length = strlen(topic);
Nim65s 0:d4dfed20c6ea 227
Nim65s 0:d4dfed20c6ea 228 int remaining_length = topic_length + 5;
Nim65s 0:d4dfed20c6ea 229 int remaining_length_2 = remaining_length_length(remaining_length);
Nim65s 0:d4dfed20c6ea 230 int packet_length = 1 + remaining_length + remaining_length_2;
Nim65s 0:d4dfed20c6ea 231
Nim65s 0:d4dfed20c6ea 232 char header = command | LEAST_ONCE;
Nim65s 0:d4dfed20c6ea 233 char packet[packet_length];
Nim65s 0:d4dfed20c6ea 234 // header
Nim65s 0:d4dfed20c6ea 235 memcpy(packet, &header, 1);
Nim65s 0:d4dfed20c6ea 236 get_remaining_length(remaining_length, packet);
Nim65s 0:d4dfed20c6ea 237
Nim65s 0:d4dfed20c6ea 238 // variable header: message identifier
Nim65s 0:d4dfed20c6ea 239 message_id++;
Nim65s 0:d4dfed20c6ea 240 char variable_header[] = { message_id / 256, message_id % 256 };
Nim65s 0:d4dfed20c6ea 241 memcpy(packet + 1 + remaining_length_2, variable_header, 2);
Nim65s 0:d4dfed20c6ea 242
Nim65s 0:d4dfed20c6ea 243 // payload: topic name & requested QoS
Nim65s 0:d4dfed20c6ea 244 char topic_size[2] = { topic_length / 256, topic_length % 256 };
Nim65s 0:d4dfed20c6ea 245 char requested_qos = MOST_ONCE;
Nim65s 0:d4dfed20c6ea 246 memcpy(packet + 3 + remaining_length_2, topic_size, 2);
Nim65s 0:d4dfed20c6ea 247 memcpy(packet + 5 + remaining_length_2, topic, topic_length);
Nim65s 0:d4dfed20c6ea 248 memcpy(packet + 5 + remaining_length_2 + topic_length, &requested_qos, 1);
Nim65s 0:d4dfed20c6ea 249
Nim65s 0:d4dfed20c6ea 250 waiting_suback++;
Nim65s 0:d4dfed20c6ea 251
Nim65s 0:d4dfed20c6ea 252 return send(packet, packet_length);
Nim65s 0:d4dfed20c6ea 253 }
Nim65s 0:d4dfed20c6ea 254
Nim65s 0:d4dfed20c6ea 255 int niMQTT::suback(bool unsub) {
Nim65s 0:d4dfed20c6ea 256 if (debug) printf("SUBACK received\r\n");
Nim65s 0:d4dfed20c6ea 257 if (waiting_suback > 0) waiting_suback--;
Nim65s 4:afbc7b066cff 258 else {
Nim65s 4:afbc7b066cff 259 printf("SUBACK UNEXPECTED !\r\n");
Nim65s 4:afbc7b066cff 260 reconnect();
Nim65s 4:afbc7b066cff 261 return -2;
Nim65s 4:afbc7b066cff 262 }
Nim65s 0:d4dfed20c6ea 263
Nim65s 5:3d21020a2826 264 //char command = (unsub) ? UNSUBACK : SUBACK;
Nim65s 0:d4dfed20c6ea 265
Nim65s 0:d4dfed20c6ea 266 int remaining_length = decode_remaining_length();
Nim65s 0:d4dfed20c6ea 267
Nim65s 0:d4dfed20c6ea 268 // Variable Header
Nim65s 0:d4dfed20c6ea 269 char var_resp[remaining_length];
Nim65s 0:d4dfed20c6ea 270 socket->receive(var_resp, remaining_length);
Nim65s 0:d4dfed20c6ea 271 waiting_new_packet = true;
Nim65s 0:d4dfed20c6ea 272 if (debug) {
Nim65s 0:d4dfed20c6ea 273 printf("suback: ");
Nim65s 0:d4dfed20c6ea 274 for (int j=0; j<remaining_length; j++) printf("0x%x ", var_resp[j]);
Nim65s 0:d4dfed20c6ea 275 printf("\r\n");
Nim65s 0:d4dfed20c6ea 276 }
Nim65s 0:d4dfed20c6ea 277
Nim65s 0:d4dfed20c6ea 278 if (var_resp[0] * 256 + var_resp[1] != message_id) {
Nim65s 0:d4dfed20c6ea 279 printf("wrong message identifer in (UN)SUBACK, get %i instead of %i...\r\n", var_resp[0] * 256 + var_resp[1], message_id);
Nim65s 0:d4dfed20c6ea 280 }
Nim65s 0:d4dfed20c6ea 281
Nim65s 0:d4dfed20c6ea 282 // here we should do things about the QoS if /unsuback, but let's say it's 0.
Nim65s 0:d4dfed20c6ea 283
Nim65s 0:d4dfed20c6ea 284 return (var_resp[0] * 256 + var_resp[1] == message_id);
Nim65s 0:d4dfed20c6ea 285 }
Nim65s 0:d4dfed20c6ea 286
Nim65s 0:d4dfed20c6ea 287 int niMQTT::pingreq () {
Nim65s 0:d4dfed20c6ea 288 if (debug) printf("*pingreq\r\n");
Nim65s 0:d4dfed20c6ea 289 char fixed_header[] = { PINGREQ, 0 };
Nim65s 0:d4dfed20c6ea 290 waiting_pingresp++;
Nim65s 0:d4dfed20c6ea 291 return send(fixed_header, 2);
Nim65s 0:d4dfed20c6ea 292 }
Nim65s 0:d4dfed20c6ea 293
Nim65s 0:d4dfed20c6ea 294 int niMQTT::pingresp() {
Nim65s 0:d4dfed20c6ea 295 if (debug) printf("PINGRESP Received\r\n");
Nim65s 0:d4dfed20c6ea 296 if (waiting_pingresp > 0) waiting_pingresp--;
Nim65s 4:afbc7b066cff 297 else {
Nim65s 4:afbc7b066cff 298 printf("PINGRESP Unexpected !\r\n");
Nim65s 4:afbc7b066cff 299 reconnect();
Nim65s 4:afbc7b066cff 300 return -2;
Nim65s 4:afbc7b066cff 301 }
Nim65s 0:d4dfed20c6ea 302
Nim65s 0:d4dfed20c6ea 303 char resp;
Nim65s 0:d4dfed20c6ea 304 socket->receive(&resp, 1);
Nim65s 0:d4dfed20c6ea 305 waiting_new_packet = true;
Nim65s 0:d4dfed20c6ea 306
Nim65s 4:afbc7b066cff 307 if (resp != 0) {
Nim65s 4:afbc7b066cff 308 printf("Wrong second byte of PINGRESP, get 0x%x instead of 0x0\r\n", resp);
Nim65s 4:afbc7b066cff 309 reconnect();
Nim65s 4:afbc7b066cff 310 return -2;
Nim65s 4:afbc7b066cff 311 }
Nim65s 0:d4dfed20c6ea 312
Nim65s 0:d4dfed20c6ea 313 return (resp == 0);
Nim65s 0:d4dfed20c6ea 314 }
Nim65s 0:d4dfed20c6ea 315
Nim65s 0:d4dfed20c6ea 316 int niMQTT::disconnect() {
Nim65s 0:d4dfed20c6ea 317 if (debug) printf("*disconnect\r\n");
Nim65s 0:d4dfed20c6ea 318 char fixed_header[] = { DISCONNECT, 0 };
Nim65s 0:d4dfed20c6ea 319 return send(fixed_header, 2);
Nim65s 0:d4dfed20c6ea 320 }
Nim65s 0:d4dfed20c6ea 321
Nim65s 0:d4dfed20c6ea 322 niMQTT::~niMQTT() {
Nim65s 0:d4dfed20c6ea 323 if (debug) printf("*~niMQTT()\r\n");
Nim65s 0:d4dfed20c6ea 324 connected = false;
Nim65s 0:d4dfed20c6ea 325 disconnect();
Nim65s 0:d4dfed20c6ea 326 socket->close();
Nim65s 0:d4dfed20c6ea 327 delete socket;
Nim65s 0:d4dfed20c6ea 328 }
Nim65s 0:d4dfed20c6ea 329
Nim65s 4:afbc7b066cff 330 void niMQTT::reconnect() {
Nim65s 4:afbc7b066cff 331 if (debug) printf("Reconnecting...\r\n");
Nim65s 4:afbc7b066cff 332 disconnect();
Nim65s 4:afbc7b066cff 333 socket->close();
Nim65s 4:afbc7b066cff 334
Nim65s 4:afbc7b066cff 335 do printf("Socket connection...\r\n"); while (socket->connect(server, port) < 0);
Nim65s 4:afbc7b066cff 336 socket->set_blocking(true, KEEP_ALIVE*500); // KEEP_ALIVE / 2 in seconds
Nim65s 4:afbc7b066cff 337
Nim65s 4:afbc7b066cff 338 connect();
Nim65s 4:afbc7b066cff 339 }
Nim65s 4:afbc7b066cff 340
Palantir 9:3be69efa4402 341 void niMQTT::ping_thread_starter(void const *p) {
Nim65s 0:d4dfed20c6ea 342 niMQTT *instance = (niMQTT*)p;
Palantir 9:3be69efa4402 343 instance->ping_thread_worker();
Palantir 9:3be69efa4402 344 }
Palantir 9:3be69efa4402 345
Palantir 9:3be69efa4402 346 void niMQTT::recv_thread_starter(void const *p) {
Palantir 9:3be69efa4402 347 niMQTT *instance = (niMQTT*)p;
Palantir 9:3be69efa4402 348 instance->recv_thread_worker();
Nim65s 0:d4dfed20c6ea 349 }
Nim65s 0:d4dfed20c6ea 350
Palantir 9:3be69efa4402 351 void niMQTT::ping_thread_worker() {
Palantir 9:3be69efa4402 352 if (debug) printf("*ping_thread_worker\r\n");
Palantir 9:3be69efa4402 353 ping_thread.signal_wait(START_THREAD);
Nim65s 0:d4dfed20c6ea 354 while (connected) {
Palantir 9:3be69efa4402 355 if (debug) printf("New loop in ping thread worker\r\n");
Nim65s 0:d4dfed20c6ea 356 Thread::wait(KEEP_ALIVE*100); // KEEP_ALIVE / 10 in seconds
Nim65s 0:d4dfed20c6ea 357 if (!packet_sent) pingreq();
Nim65s 0:d4dfed20c6ea 358 packet_sent = false;
Nim65s 0:d4dfed20c6ea 359 }
Nim65s 0:d4dfed20c6ea 360 }
Nim65s 0:d4dfed20c6ea 361
Palantir 9:3be69efa4402 362 void niMQTT::recv_thread_worker() {
Palantir 9:3be69efa4402 363 if (debug) printf("*recv_thread_worker\r\n");
Palantir 9:3be69efa4402 364 recv_thread.signal_wait(START_THREAD);
Palantir 9:3be69efa4402 365 while (connected) {
Palantir 9:3be69efa4402 366 if (debug) printf("New loop in recv thread worker\r\n");
Palantir 9:3be69efa4402 367 recv();
Palantir 9:3be69efa4402 368 packet_sent = false;
Palantir 9:3be69efa4402 369 }
Palantir 9:3be69efa4402 370 }
Palantir 9:3be69efa4402 371
Nim65s 0:d4dfed20c6ea 372 void niMQTT::get_remaining_length(int remaining_length, char *packet) {
Nim65s 0:d4dfed20c6ea 373 int X = remaining_length;
Nim65s 0:d4dfed20c6ea 374 int n = 1;
Nim65s 0:d4dfed20c6ea 375 char digit;
Nim65s 0:d4dfed20c6ea 376 do {
Nim65s 0:d4dfed20c6ea 377 digit = X % 0x80;
Nim65s 0:d4dfed20c6ea 378 X /= 0x80;
Nim65s 0:d4dfed20c6ea 379 if (X > 0) digit |= 0x80;
Nim65s 0:d4dfed20c6ea 380 memcpy(packet + n, &digit, 1);
Nim65s 0:d4dfed20c6ea 381 n++;
Nim65s 0:d4dfed20c6ea 382 } while (X > 0);
Nim65s 0:d4dfed20c6ea 383 }
Nim65s 0:d4dfed20c6ea 384
Nim65s 0:d4dfed20c6ea 385 int niMQTT::decode_remaining_length() {
Nim65s 0:d4dfed20c6ea 386 int multiplier = 1;
Nim65s 0:d4dfed20c6ea 387 int value = 0;
Nim65s 0:d4dfed20c6ea 388 char digit = 0;
Nim65s 0:d4dfed20c6ea 389 do {
Nim65s 0:d4dfed20c6ea 390 while (socket->receive(&digit, 1) < 0) wait(0.1);
Nim65s 0:d4dfed20c6ea 391 value += (digit & 127) * multiplier;
Nim65s 0:d4dfed20c6ea 392 multiplier *= 128;
Nim65s 0:d4dfed20c6ea 393 } while ((digit & 0x80) != 0);
Nim65s 0:d4dfed20c6ea 394 return value;
Nim65s 0:d4dfed20c6ea 395 }
Nim65s 0:d4dfed20c6ea 396
Nim65s 0:d4dfed20c6ea 397 int niMQTT::remaining_length_length(int remaining_length) {
Nim65s 0:d4dfed20c6ea 398 int X = remaining_length;
Nim65s 0:d4dfed20c6ea 399 int rll = 0;
Nim65s 0:d4dfed20c6ea 400 do {
Nim65s 0:d4dfed20c6ea 401 rll++;
Nim65s 0:d4dfed20c6ea 402 X /= 0x80;
Nim65s 0:d4dfed20c6ea 403 } while (X > 0);
Nim65s 0:d4dfed20c6ea 404 return rll;
Nim65s 0:d4dfed20c6ea 405 }
Nim65s 8:438958bb9df3 406
Nim65s 8:438958bb9df3 407 void niMQTT::call_callback(const char *topic, const char *message) {
Nim65s 8:438958bb9df3 408 callback(topic, message);
Nim65s 8:438958bb9df3 409 }