Simple websocket client

Dependents:   Websocket_Ethernet_HelloWorld Websocket_Wifly_HelloWorld RPC_Wifly_HelloWorld RPC_Ethernet_HelloWorld ... more

Committer:
samux
Date:
Mon Aug 20 08:50:51 2012 +0000
Revision:
4:466f90b7849a
Parent:
3:9589afa4712e
Child:
5:bb09d7a6c92f
avoid calling send() multiple times for one ws message

Who changed what in which revision?

UserRevisionLine numberNew contents of line
samux 3:9589afa4712e 1 #include "Websocket.h"
samux 3:9589afa4712e 2 #include <string>
samux 3:9589afa4712e 3
samux 3:9589afa4712e 4 #define MAX_TRY_WRITE 50
samux 3:9589afa4712e 5 #define MAX_TRY_READ 30
samux 3:9589afa4712e 6
samux 3:9589afa4712e 7 //Debug is disabled by default
samux 3:9589afa4712e 8 #if 0
samux 3:9589afa4712e 9 #define DBG(x, ...) std::printf("[WebSocket : DBG]"x"\r\n", ##__VA_ARGS__);
samux 3:9589afa4712e 10 #define WARN(x, ...) std::printf("[WebSocket : WARN]"x"\r\n", ##__VA_ARGS__);
samux 3:9589afa4712e 11 #define ERR(x, ...) std::printf("[WebSocket : ERR]"x"\r\n", ##__VA_ARGS__);
samux 3:9589afa4712e 12 #else
samux 3:9589afa4712e 13 #define DBG(x, ...)
samux 3:9589afa4712e 14 #define WARN(x, ...)
samux 3:9589afa4712e 15 #define ERR(x, ...)
samux 3:9589afa4712e 16 #endif
samux 3:9589afa4712e 17
samux 3:9589afa4712e 18 #define INFO(x, ...) printf("[WebSocket : INFO]"x"\r\n", ##__VA_ARGS__);
samux 3:9589afa4712e 19
samux 3:9589afa4712e 20 Websocket::Websocket(char * url) {
samux 3:9589afa4712e 21 fillFields(url);
samux 3:9589afa4712e 22 socket.set_blocking(false, 400);
samux 3:9589afa4712e 23 }
samux 3:9589afa4712e 24
samux 3:9589afa4712e 25 void Websocket::fillFields(char * url) {
samux 3:9589afa4712e 26 char *res = NULL;
samux 3:9589afa4712e 27 char *res1 = NULL;
samux 3:9589afa4712e 28
samux 3:9589afa4712e 29 char buf[50];
samux 3:9589afa4712e 30 strcpy(buf, url);
samux 3:9589afa4712e 31
samux 3:9589afa4712e 32 res = strtok(buf, ":");
samux 3:9589afa4712e 33 if (strcmp(res, "ws")) {
samux 3:9589afa4712e 34 DBG("\r\nFormat printfor: please use: \"ws://ip-or-domain[:port]/path\"\r\n\r\n");
samux 3:9589afa4712e 35 } else {
samux 3:9589afa4712e 36 //ip_domain and port
samux 3:9589afa4712e 37 res = strtok(NULL, "/");
samux 3:9589afa4712e 38
samux 3:9589afa4712e 39 //path
samux 3:9589afa4712e 40 res1 = strtok(NULL, " ");
samux 3:9589afa4712e 41 if (res1 != NULL) {
samux 3:9589afa4712e 42 path = res1;
samux 3:9589afa4712e 43 }
samux 3:9589afa4712e 44
samux 3:9589afa4712e 45 //ip_domain
samux 3:9589afa4712e 46 res = strtok(res, ":");
samux 3:9589afa4712e 47
samux 3:9589afa4712e 48 //port
samux 3:9589afa4712e 49 res1 = strtok(NULL, " ");
samux 3:9589afa4712e 50 if (res1 != NULL) {
samux 3:9589afa4712e 51 port = res1;
samux 3:9589afa4712e 52 } else {
samux 3:9589afa4712e 53 port = "80";
samux 3:9589afa4712e 54 }
samux 3:9589afa4712e 55
samux 3:9589afa4712e 56 if (res != NULL) {
samux 3:9589afa4712e 57 ip_domain = res;
samux 3:9589afa4712e 58 }
samux 3:9589afa4712e 59 }
samux 3:9589afa4712e 60 }
samux 3:9589afa4712e 61
samux 3:9589afa4712e 62
samux 3:9589afa4712e 63 bool Websocket::connect() {
samux 3:9589afa4712e 64 char cmd[200];
samux 3:9589afa4712e 65
samux 3:9589afa4712e 66 while (socket.connect(ip_domain.c_str(), atoi(port.c_str())) < 0) {
samux 3:9589afa4712e 67 ERR("Unable to connect to (%s) on port (%d)\r\n", ip_domain.c_str(), atoi(port.c_str()));
samux 3:9589afa4712e 68 wait(0.2);
samux 3:9589afa4712e 69 }
samux 3:9589afa4712e 70
samux 3:9589afa4712e 71 // sent http header to upgrade to the ws protocol
samux 3:9589afa4712e 72 sprintf(cmd, "GET /%s HTTP/1.1\r\n", path.c_str());
samux 3:9589afa4712e 73 write(cmd, strlen(cmd));
samux 3:9589afa4712e 74
samux 3:9589afa4712e 75 sprintf(cmd, "Host: %s:%s\r\n", ip_domain.c_str(), port.c_str());
samux 3:9589afa4712e 76 write(cmd, strlen(cmd));
samux 3:9589afa4712e 77
samux 3:9589afa4712e 78 sprintf(cmd, "Upgrade: WebSocket\r\n");
samux 3:9589afa4712e 79 write(cmd, strlen(cmd));
samux 3:9589afa4712e 80
samux 3:9589afa4712e 81 sprintf(cmd, "Connection: Upgrade\r\n");
samux 3:9589afa4712e 82 write(cmd, strlen(cmd));
samux 3:9589afa4712e 83
samux 3:9589afa4712e 84 sprintf(cmd, "Sec-WebSocket-Key: L159VM0TWUzyDxwJEIEzjw==\r\n");
samux 3:9589afa4712e 85 write(cmd, strlen(cmd));
samux 3:9589afa4712e 86
samux 3:9589afa4712e 87 sprintf(cmd, "Sec-WebSocket-Version: 13\r\n\r\n");
samux 3:9589afa4712e 88 int ret = write(cmd, strlen(cmd));
samux 3:9589afa4712e 89 if (ret != strlen(cmd)) {
samux 3:9589afa4712e 90 close();
samux 3:9589afa4712e 91 ERR("Could not send request");
samux 3:9589afa4712e 92 return false;
samux 3:9589afa4712e 93 }
samux 3:9589afa4712e 94
samux 3:9589afa4712e 95 ret = read(cmd, 200, 100);
samux 3:9589afa4712e 96 if (ret < 0) {
samux 3:9589afa4712e 97 close();
samux 3:9589afa4712e 98 ERR("Could not receive answer\r\n");
samux 3:9589afa4712e 99 return false;
samux 3:9589afa4712e 100 }
samux 3:9589afa4712e 101
samux 3:9589afa4712e 102 cmd[ret] = '\0';
samux 3:9589afa4712e 103 DBG("recv: %s\r\n", cmd);
samux 3:9589afa4712e 104
samux 3:9589afa4712e 105 if ( strstr(cmd, "DdLWT/1JcX+nQFHebYP+rqEx5xI=") == NULL ) {
samux 3:9589afa4712e 106 ERR("Wrong answer from server, got \"%s\" instead\r\n", cmd);
samux 3:9589afa4712e 107 do {
samux 3:9589afa4712e 108 ret = read(cmd, 200, 100);
samux 3:9589afa4712e 109 if (ret < 0) {
samux 3:9589afa4712e 110 ERR("Could not receive answer\r\n");
samux 3:9589afa4712e 111 return false;
samux 3:9589afa4712e 112 }
samux 3:9589afa4712e 113 cmd[ret] = '\0';
samux 3:9589afa4712e 114 printf("%s",cmd);
samux 3:9589afa4712e 115 } while (ret > 0);
samux 3:9589afa4712e 116 close();
samux 3:9589afa4712e 117 return false;
samux 3:9589afa4712e 118 }
samux 3:9589afa4712e 119
samux 3:9589afa4712e 120 INFO("\r\nip_domain: %s\r\npath: /%s\r\nport: %s\r\n\r\n", ip_domain.c_str(), path.c_str(), port.c_str());
samux 3:9589afa4712e 121 return true;
samux 3:9589afa4712e 122 }
samux 3:9589afa4712e 123
samux 4:466f90b7849a 124 int Websocket::sendLength(uint32_t len, char * msg) {
samux 3:9589afa4712e 125
samux 3:9589afa4712e 126 if (len < 126) {
samux 4:466f90b7849a 127 msg[0] = len | (1<<7);
samux 3:9589afa4712e 128 return 1;
samux 3:9589afa4712e 129 } else if (len < 65535) {
samux 4:466f90b7849a 130 msg[0] = 126 | (1<<7);
samux 4:466f90b7849a 131 msg[1] = (len >> 8) & 0xff;
samux 4:466f90b7849a 132 msg[2] = len & 0xff;
samux 3:9589afa4712e 133 return 3;
samux 3:9589afa4712e 134 } else {
samux 4:466f90b7849a 135 msg[0] = 127 | (1<<7);
samux 3:9589afa4712e 136 for (int i = 0; i < 8; i++) {
samux 4:466f90b7849a 137 msg[i+1] = (len >> i*8) & 0xff;
samux 3:9589afa4712e 138 }
samux 3:9589afa4712e 139 return 9;
samux 3:9589afa4712e 140 }
samux 3:9589afa4712e 141 }
samux 3:9589afa4712e 142
samux 3:9589afa4712e 143 int Websocket::readChar(char * pC, bool block) {
samux 3:9589afa4712e 144 return read(pC, 1, 1);
samux 3:9589afa4712e 145 }
samux 3:9589afa4712e 146
samux 4:466f90b7849a 147 int Websocket::sendOpcode(uint8_t opcode, char * msg) {
samux 4:466f90b7849a 148 msg[0] = 0x80 | (opcode & 0x0f);
samux 4:466f90b7849a 149 return 1;
samux 3:9589afa4712e 150 }
samux 3:9589afa4712e 151
samux 4:466f90b7849a 152 int Websocket::sendMask(char * msg) {
samux 3:9589afa4712e 153 for (int i = 0; i < 4; i++) {
samux 4:466f90b7849a 154 msg[i] = 0;
samux 3:9589afa4712e 155 }
samux 3:9589afa4712e 156 return 4;
samux 3:9589afa4712e 157 }
samux 3:9589afa4712e 158
samux 3:9589afa4712e 159 int Websocket::send(char * str) {
samux 4:466f90b7849a 160 char msg[strlen(str) + 15];
samux 4:466f90b7849a 161 int idx = 0;
samux 4:466f90b7849a 162 idx = sendOpcode(0x01, msg);
samux 4:466f90b7849a 163 idx += sendLength(strlen(str), msg + idx);
samux 4:466f90b7849a 164 idx += sendMask(msg + idx);
samux 4:466f90b7849a 165 memcpy(msg+idx, str, strlen(str));
samux 4:466f90b7849a 166 int res = write(msg, idx + strlen(str));
samux 3:9589afa4712e 167 return res;
samux 3:9589afa4712e 168 }
samux 3:9589afa4712e 169
samux 3:9589afa4712e 170
samux 3:9589afa4712e 171 bool Websocket::read(char * message) {
samux 3:9589afa4712e 172 int i = 0;
samux 3:9589afa4712e 173 uint32_t len_msg;
samux 3:9589afa4712e 174 char opcode = 0;
samux 3:9589afa4712e 175 char c;
samux 3:9589afa4712e 176 char mask[4] = {0, 0, 0, 0};
samux 3:9589afa4712e 177 bool is_masked = false;
samux 3:9589afa4712e 178 Timer tmr;
samux 3:9589afa4712e 179
samux 3:9589afa4712e 180 // read the opcode
samux 3:9589afa4712e 181 tmr.start();
samux 3:9589afa4712e 182 while (true) {
samux 3:9589afa4712e 183 if (tmr.read() > 3) {
samux 3:9589afa4712e 184 DBG("timeout ws\r\n");
samux 3:9589afa4712e 185 return false;
samux 3:9589afa4712e 186 }
samux 3:9589afa4712e 187
samux 3:9589afa4712e 188 socket.set_blocking(false, 1);
samux 3:9589afa4712e 189 if (socket.receive(&opcode, 1) != 1) {
samux 3:9589afa4712e 190 socket.set_blocking(false, 2000);
samux 3:9589afa4712e 191 return false;
samux 3:9589afa4712e 192 }
samux 3:9589afa4712e 193
samux 3:9589afa4712e 194 socket.set_blocking(false, 2000);
samux 3:9589afa4712e 195
samux 3:9589afa4712e 196 if (opcode == 0x81)
samux 3:9589afa4712e 197 break;
samux 3:9589afa4712e 198 }
samux 3:9589afa4712e 199 DBG("opcode: 0x%X\r\n", opcode);
samux 3:9589afa4712e 200
samux 3:9589afa4712e 201 readChar(&c);
samux 3:9589afa4712e 202 len_msg = c & 0x7f;
samux 3:9589afa4712e 203 is_masked = c & 0x80;
samux 3:9589afa4712e 204 if (len_msg == 126) {
samux 3:9589afa4712e 205 readChar(&c);
samux 3:9589afa4712e 206 len_msg = c << 8;
samux 3:9589afa4712e 207 readChar(&c);
samux 3:9589afa4712e 208 len_msg += c;
samux 3:9589afa4712e 209 } else if (len_msg == 127) {
samux 3:9589afa4712e 210 len_msg = 0;
samux 3:9589afa4712e 211 for (int i = 0; i < 8; i++) {
samux 3:9589afa4712e 212 readChar(&c);
samux 3:9589afa4712e 213 len_msg += (c << (7-i)*8);
samux 3:9589afa4712e 214 }
samux 3:9589afa4712e 215 }
samux 3:9589afa4712e 216
samux 3:9589afa4712e 217 if (len_msg == 0) {
samux 3:9589afa4712e 218 return false;
samux 3:9589afa4712e 219 }
samux 3:9589afa4712e 220 DBG("length: %d\r\n", len_msg);
samux 3:9589afa4712e 221
samux 3:9589afa4712e 222 if (is_masked) {
samux 3:9589afa4712e 223 for (i = 0; i < 4; i++)
samux 3:9589afa4712e 224 readChar(&c);
samux 3:9589afa4712e 225 mask[i] = c;
samux 3:9589afa4712e 226 }
samux 3:9589afa4712e 227
samux 3:9589afa4712e 228 int nb = read(message, len_msg, len_msg);
samux 3:9589afa4712e 229 if (nb != len_msg)
samux 3:9589afa4712e 230 return false;
samux 3:9589afa4712e 231
samux 3:9589afa4712e 232 for (i = 0; i < len_msg; i++) {
samux 3:9589afa4712e 233 message[i] = message[i] ^ mask[i % 4];
samux 3:9589afa4712e 234 }
samux 3:9589afa4712e 235
samux 3:9589afa4712e 236 message[len_msg] = '\0';
samux 3:9589afa4712e 237
samux 3:9589afa4712e 238 return true;
samux 3:9589afa4712e 239 }
samux 3:9589afa4712e 240
samux 3:9589afa4712e 241 bool Websocket::close() {
samux 3:9589afa4712e 242 if (!is_connected())
samux 3:9589afa4712e 243 return false;
samux 3:9589afa4712e 244
samux 3:9589afa4712e 245 int ret = socket.close();
samux 3:9589afa4712e 246 if (ret < 0) {
samux 3:9589afa4712e 247 ERR("Could not disconnect");
samux 3:9589afa4712e 248 return false;
samux 3:9589afa4712e 249 }
samux 3:9589afa4712e 250 return true;
samux 3:9589afa4712e 251 }
samux 3:9589afa4712e 252
samux 3:9589afa4712e 253 bool Websocket::is_connected() {
samux 3:9589afa4712e 254 return socket.is_connected();
samux 3:9589afa4712e 255 }
samux 3:9589afa4712e 256
samux 3:9589afa4712e 257 std::string Websocket::getPath() {
samux 3:9589afa4712e 258 return path;
samux 3:9589afa4712e 259 }
samux 3:9589afa4712e 260
samux 3:9589afa4712e 261 int Websocket::write(char * str, int len) {
samux 3:9589afa4712e 262 int res = 0, idx = 0;
samux 3:9589afa4712e 263
samux 3:9589afa4712e 264 for (int j = 0; j < MAX_TRY_WRITE; j++) {
samux 3:9589afa4712e 265
samux 3:9589afa4712e 266 if ((res = socket.send_all(str + idx, len - idx)) == -1)
samux 3:9589afa4712e 267 continue;
samux 3:9589afa4712e 268
samux 3:9589afa4712e 269 idx += res;
samux 3:9589afa4712e 270
samux 3:9589afa4712e 271 if (idx == len)
samux 3:9589afa4712e 272 return len;
samux 3:9589afa4712e 273 }
samux 3:9589afa4712e 274
samux 3:9589afa4712e 275 return (idx == 0) ? -1 : idx;
samux 3:9589afa4712e 276 }
samux 3:9589afa4712e 277
samux 3:9589afa4712e 278 int Websocket::read(char * str, int len, int min_len) {
samux 3:9589afa4712e 279 int res = 0, idx = 0;
samux 3:9589afa4712e 280
samux 3:9589afa4712e 281 for (int j = 0; j < MAX_TRY_WRITE; j++) {
samux 3:9589afa4712e 282
samux 3:9589afa4712e 283 if ((res = socket.receive_all(str + idx, len - idx)) == -1)
samux 3:9589afa4712e 284 continue;
samux 3:9589afa4712e 285
samux 3:9589afa4712e 286 idx += res;
samux 3:9589afa4712e 287
samux 3:9589afa4712e 288 if (idx == len || (min_len != -1 && idx > min_len))
samux 3:9589afa4712e 289 return idx;
samux 3:9589afa4712e 290 }
samux 3:9589afa4712e 291
samux 3:9589afa4712e 292 return (idx == 0) ? -1 : idx;
samux 3:9589afa4712e 293 }