fork of WebSocketClient with some fixes
Fork of WebSocketClient by
Websocket.cpp@8:ccedee13be8d, 2013-10-23 (annotated)
- Committer:
- mbed_official
- Date:
- Wed Oct 23 09:58:39 2013 +0000
- Revision:
- 8:ccedee13be8d
- Parent:
- 7:4567996414a5
- Child:
- 9:efa2c147bee1
Apache V2 License
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
mbed_official | 8:ccedee13be8d | 1 | /* mbed Microcontroller Library |
mbed_official | 8:ccedee13be8d | 2 | * Copyright (c) 2006-2013 ARM Limited |
mbed_official | 8:ccedee13be8d | 3 | * |
mbed_official | 8:ccedee13be8d | 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
mbed_official | 8:ccedee13be8d | 5 | * you may not use this file except in compliance with the License. |
mbed_official | 8:ccedee13be8d | 6 | * You may obtain a copy of the License at |
mbed_official | 8:ccedee13be8d | 7 | * |
mbed_official | 8:ccedee13be8d | 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
mbed_official | 8:ccedee13be8d | 9 | * |
mbed_official | 8:ccedee13be8d | 10 | * Unless required by applicable law or agreed to in writing, software |
mbed_official | 8:ccedee13be8d | 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
mbed_official | 8:ccedee13be8d | 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
mbed_official | 8:ccedee13be8d | 13 | * See the License for the specific language governing permissions and |
mbed_official | 8:ccedee13be8d | 14 | * limitations under the License. |
mbed_official | 8:ccedee13be8d | 15 | */ |
samux | 3:9589afa4712e | 16 | #include "Websocket.h" |
samux | 3:9589afa4712e | 17 | |
samux | 7:4567996414a5 | 18 | #define MAX_TRY_WRITE 20 |
samux | 7:4567996414a5 | 19 | #define MAX_TRY_READ 10 |
samux | 3:9589afa4712e | 20 | |
samux | 3:9589afa4712e | 21 | //Debug is disabled by default |
samux | 3:9589afa4712e | 22 | #if 0 |
samux | 3:9589afa4712e | 23 | #define DBG(x, ...) std::printf("[WebSocket : DBG]"x"\r\n", ##__VA_ARGS__); |
samux | 3:9589afa4712e | 24 | #define WARN(x, ...) std::printf("[WebSocket : WARN]"x"\r\n", ##__VA_ARGS__); |
samux | 3:9589afa4712e | 25 | #define ERR(x, ...) std::printf("[WebSocket : ERR]"x"\r\n", ##__VA_ARGS__); |
samux | 3:9589afa4712e | 26 | #else |
samux | 3:9589afa4712e | 27 | #define DBG(x, ...) |
samux | 3:9589afa4712e | 28 | #define WARN(x, ...) |
samux | 3:9589afa4712e | 29 | #define ERR(x, ...) |
samux | 3:9589afa4712e | 30 | #endif |
samux | 3:9589afa4712e | 31 | |
samux | 3:9589afa4712e | 32 | #define INFO(x, ...) printf("[WebSocket : INFO]"x"\r\n", ##__VA_ARGS__); |
samux | 3:9589afa4712e | 33 | |
samux | 3:9589afa4712e | 34 | Websocket::Websocket(char * url) { |
samux | 3:9589afa4712e | 35 | fillFields(url); |
samux | 3:9589afa4712e | 36 | socket.set_blocking(false, 400); |
samux | 3:9589afa4712e | 37 | } |
samux | 3:9589afa4712e | 38 | |
samux | 3:9589afa4712e | 39 | void Websocket::fillFields(char * url) { |
donatien | 6:86e89a0369b9 | 40 | int ret = parseURL(url, scheme, sizeof(scheme), host, sizeof(host), &port, path, sizeof(path)); |
donatien | 6:86e89a0369b9 | 41 | if(ret) |
donatien | 6:86e89a0369b9 | 42 | { |
donatien | 6:86e89a0369b9 | 43 | ERR("URL parsing failed; please use: \"ws://ip-or-domain[:port]/path\""); |
donatien | 6:86e89a0369b9 | 44 | return; |
donatien | 6:86e89a0369b9 | 45 | } |
samux | 3:9589afa4712e | 46 | |
donatien | 6:86e89a0369b9 | 47 | if(port == 0) //TODO do handle WSS->443 |
donatien | 6:86e89a0369b9 | 48 | { |
donatien | 6:86e89a0369b9 | 49 | port = 80; |
donatien | 6:86e89a0369b9 | 50 | } |
donatien | 6:86e89a0369b9 | 51 | |
donatien | 6:86e89a0369b9 | 52 | if(strcmp(scheme, "ws")) |
donatien | 6:86e89a0369b9 | 53 | { |
donatien | 6:86e89a0369b9 | 54 | ERR("Wrong scheme, please use \"ws\" instead"); |
donatien | 6:86e89a0369b9 | 55 | } |
donatien | 6:86e89a0369b9 | 56 | } |
samux | 3:9589afa4712e | 57 | |
donatien | 6:86e89a0369b9 | 58 | int Websocket::parseURL(const char* url, char* scheme, size_t maxSchemeLen, char* host, size_t maxHostLen, uint16_t* port, char* path, size_t maxPathLen) //Parse URL |
donatien | 6:86e89a0369b9 | 59 | { |
donatien | 6:86e89a0369b9 | 60 | char* schemePtr = (char*) url; |
donatien | 6:86e89a0369b9 | 61 | char* hostPtr = (char*) strstr(url, "://"); |
donatien | 6:86e89a0369b9 | 62 | if(hostPtr == NULL) |
donatien | 6:86e89a0369b9 | 63 | { |
donatien | 6:86e89a0369b9 | 64 | WARN("Could not find host"); |
donatien | 6:86e89a0369b9 | 65 | return -1; //URL is invalid |
donatien | 6:86e89a0369b9 | 66 | } |
donatien | 6:86e89a0369b9 | 67 | |
donatien | 6:86e89a0369b9 | 68 | if( maxSchemeLen < hostPtr - schemePtr + 1 ) //including NULL-terminating char |
donatien | 6:86e89a0369b9 | 69 | { |
donatien | 6:86e89a0369b9 | 70 | WARN("Scheme str is too small (%d >= %d)", maxSchemeLen, hostPtr - schemePtr + 1); |
donatien | 6:86e89a0369b9 | 71 | return -1; |
donatien | 6:86e89a0369b9 | 72 | } |
donatien | 6:86e89a0369b9 | 73 | memcpy(scheme, schemePtr, hostPtr - schemePtr); |
donatien | 6:86e89a0369b9 | 74 | scheme[hostPtr - schemePtr] = '\0'; |
donatien | 6:86e89a0369b9 | 75 | |
donatien | 6:86e89a0369b9 | 76 | hostPtr+=3; |
donatien | 6:86e89a0369b9 | 77 | |
donatien | 6:86e89a0369b9 | 78 | size_t hostLen = 0; |
samux | 3:9589afa4712e | 79 | |
donatien | 6:86e89a0369b9 | 80 | char* portPtr = strchr(hostPtr, ':'); |
donatien | 6:86e89a0369b9 | 81 | if( portPtr != NULL ) |
donatien | 6:86e89a0369b9 | 82 | { |
donatien | 6:86e89a0369b9 | 83 | hostLen = portPtr - hostPtr; |
donatien | 6:86e89a0369b9 | 84 | portPtr++; |
donatien | 6:86e89a0369b9 | 85 | if( sscanf(portPtr, "%hu", port) != 1) |
donatien | 6:86e89a0369b9 | 86 | { |
donatien | 6:86e89a0369b9 | 87 | WARN("Could not find port"); |
donatien | 6:86e89a0369b9 | 88 | return -1; |
donatien | 6:86e89a0369b9 | 89 | } |
donatien | 6:86e89a0369b9 | 90 | } |
donatien | 6:86e89a0369b9 | 91 | else |
donatien | 6:86e89a0369b9 | 92 | { |
donatien | 6:86e89a0369b9 | 93 | *port=0; |
donatien | 6:86e89a0369b9 | 94 | } |
donatien | 6:86e89a0369b9 | 95 | char* pathPtr = strchr(hostPtr, '/'); |
donatien | 6:86e89a0369b9 | 96 | if( hostLen == 0 ) |
donatien | 6:86e89a0369b9 | 97 | { |
donatien | 6:86e89a0369b9 | 98 | hostLen = pathPtr - hostPtr; |
donatien | 6:86e89a0369b9 | 99 | } |
samux | 3:9589afa4712e | 100 | |
donatien | 6:86e89a0369b9 | 101 | if( maxHostLen < hostLen + 1 ) //including NULL-terminating char |
donatien | 6:86e89a0369b9 | 102 | { |
donatien | 6:86e89a0369b9 | 103 | WARN("Host str is too small (%d >= %d)", maxHostLen, hostLen + 1); |
donatien | 6:86e89a0369b9 | 104 | return -1; |
donatien | 6:86e89a0369b9 | 105 | } |
donatien | 6:86e89a0369b9 | 106 | memcpy(host, hostPtr, hostLen); |
donatien | 6:86e89a0369b9 | 107 | host[hostLen] = '\0'; |
samux | 3:9589afa4712e | 108 | |
donatien | 6:86e89a0369b9 | 109 | size_t pathLen; |
donatien | 6:86e89a0369b9 | 110 | char* fragmentPtr = strchr(hostPtr, '#'); |
donatien | 6:86e89a0369b9 | 111 | if(fragmentPtr != NULL) |
donatien | 6:86e89a0369b9 | 112 | { |
donatien | 6:86e89a0369b9 | 113 | pathLen = fragmentPtr - pathPtr; |
donatien | 6:86e89a0369b9 | 114 | } |
donatien | 6:86e89a0369b9 | 115 | else |
donatien | 6:86e89a0369b9 | 116 | { |
donatien | 6:86e89a0369b9 | 117 | pathLen = strlen(pathPtr); |
donatien | 6:86e89a0369b9 | 118 | } |
donatien | 6:86e89a0369b9 | 119 | |
donatien | 6:86e89a0369b9 | 120 | if( maxPathLen < pathLen + 1 ) //including NULL-terminating char |
donatien | 6:86e89a0369b9 | 121 | { |
donatien | 6:86e89a0369b9 | 122 | WARN("Path str is too small (%d >= %d)", maxPathLen, pathLen + 1); |
donatien | 6:86e89a0369b9 | 123 | return -1; |
donatien | 6:86e89a0369b9 | 124 | } |
donatien | 6:86e89a0369b9 | 125 | memcpy(path, pathPtr, pathLen); |
donatien | 6:86e89a0369b9 | 126 | path[pathLen] = '\0'; |
donatien | 6:86e89a0369b9 | 127 | |
donatien | 6:86e89a0369b9 | 128 | return 0; |
samux | 3:9589afa4712e | 129 | } |
samux | 3:9589afa4712e | 130 | |
samux | 3:9589afa4712e | 131 | |
samux | 3:9589afa4712e | 132 | bool Websocket::connect() { |
samux | 3:9589afa4712e | 133 | char cmd[200]; |
samux | 3:9589afa4712e | 134 | |
donatien | 6:86e89a0369b9 | 135 | while (socket.connect(host, port) < 0) { |
donatien | 6:86e89a0369b9 | 136 | ERR("Unable to connect to (%s) on port (%d)", host, port); |
samux | 3:9589afa4712e | 137 | wait(0.2); |
donatien | 6:86e89a0369b9 | 138 | return false; |
samux | 3:9589afa4712e | 139 | } |
samux | 3:9589afa4712e | 140 | |
samux | 3:9589afa4712e | 141 | // sent http header to upgrade to the ws protocol |
donatien | 6:86e89a0369b9 | 142 | sprintf(cmd, "GET %s HTTP/1.1\r\n", path); |
samux | 3:9589afa4712e | 143 | write(cmd, strlen(cmd)); |
donatien | 6:86e89a0369b9 | 144 | |
donatien | 6:86e89a0369b9 | 145 | sprintf(cmd, "Host: %s:%d\r\n", host, port); |
samux | 3:9589afa4712e | 146 | write(cmd, strlen(cmd)); |
samux | 3:9589afa4712e | 147 | |
samux | 3:9589afa4712e | 148 | sprintf(cmd, "Upgrade: WebSocket\r\n"); |
samux | 3:9589afa4712e | 149 | write(cmd, strlen(cmd)); |
samux | 3:9589afa4712e | 150 | |
samux | 3:9589afa4712e | 151 | sprintf(cmd, "Connection: Upgrade\r\n"); |
samux | 3:9589afa4712e | 152 | write(cmd, strlen(cmd)); |
samux | 3:9589afa4712e | 153 | |
samux | 3:9589afa4712e | 154 | sprintf(cmd, "Sec-WebSocket-Key: L159VM0TWUzyDxwJEIEzjw==\r\n"); |
samux | 3:9589afa4712e | 155 | write(cmd, strlen(cmd)); |
samux | 3:9589afa4712e | 156 | |
samux | 3:9589afa4712e | 157 | sprintf(cmd, "Sec-WebSocket-Version: 13\r\n\r\n"); |
samux | 3:9589afa4712e | 158 | int ret = write(cmd, strlen(cmd)); |
samux | 3:9589afa4712e | 159 | if (ret != strlen(cmd)) { |
samux | 3:9589afa4712e | 160 | close(); |
samux | 3:9589afa4712e | 161 | ERR("Could not send request"); |
samux | 3:9589afa4712e | 162 | return false; |
samux | 3:9589afa4712e | 163 | } |
samux | 3:9589afa4712e | 164 | |
samux | 3:9589afa4712e | 165 | ret = read(cmd, 200, 100); |
samux | 3:9589afa4712e | 166 | if (ret < 0) { |
samux | 3:9589afa4712e | 167 | close(); |
samux | 3:9589afa4712e | 168 | ERR("Could not receive answer\r\n"); |
samux | 3:9589afa4712e | 169 | return false; |
samux | 3:9589afa4712e | 170 | } |
samux | 3:9589afa4712e | 171 | |
samux | 3:9589afa4712e | 172 | cmd[ret] = '\0'; |
samux | 3:9589afa4712e | 173 | DBG("recv: %s\r\n", cmd); |
samux | 3:9589afa4712e | 174 | |
samux | 3:9589afa4712e | 175 | if ( strstr(cmd, "DdLWT/1JcX+nQFHebYP+rqEx5xI=") == NULL ) { |
samux | 3:9589afa4712e | 176 | ERR("Wrong answer from server, got \"%s\" instead\r\n", cmd); |
samux | 3:9589afa4712e | 177 | do { |
samux | 3:9589afa4712e | 178 | ret = read(cmd, 200, 100); |
samux | 3:9589afa4712e | 179 | if (ret < 0) { |
samux | 3:9589afa4712e | 180 | ERR("Could not receive answer\r\n"); |
samux | 3:9589afa4712e | 181 | return false; |
samux | 3:9589afa4712e | 182 | } |
samux | 3:9589afa4712e | 183 | cmd[ret] = '\0'; |
samux | 3:9589afa4712e | 184 | printf("%s",cmd); |
samux | 3:9589afa4712e | 185 | } while (ret > 0); |
samux | 3:9589afa4712e | 186 | close(); |
samux | 3:9589afa4712e | 187 | return false; |
samux | 3:9589afa4712e | 188 | } |
samux | 3:9589afa4712e | 189 | |
donatien | 6:86e89a0369b9 | 190 | INFO("\r\nhost: %s\r\npath: %s\r\nport: %d\r\n\r\n", host, path, port); |
samux | 3:9589afa4712e | 191 | return true; |
samux | 3:9589afa4712e | 192 | } |
samux | 3:9589afa4712e | 193 | |
samux | 4:466f90b7849a | 194 | int Websocket::sendLength(uint32_t len, char * msg) { |
samux | 3:9589afa4712e | 195 | |
samux | 3:9589afa4712e | 196 | if (len < 126) { |
samux | 4:466f90b7849a | 197 | msg[0] = len | (1<<7); |
samux | 3:9589afa4712e | 198 | return 1; |
samux | 3:9589afa4712e | 199 | } else if (len < 65535) { |
samux | 4:466f90b7849a | 200 | msg[0] = 126 | (1<<7); |
samux | 4:466f90b7849a | 201 | msg[1] = (len >> 8) & 0xff; |
samux | 4:466f90b7849a | 202 | msg[2] = len & 0xff; |
samux | 3:9589afa4712e | 203 | return 3; |
samux | 3:9589afa4712e | 204 | } else { |
samux | 4:466f90b7849a | 205 | msg[0] = 127 | (1<<7); |
samux | 3:9589afa4712e | 206 | for (int i = 0; i < 8; i++) { |
samux | 4:466f90b7849a | 207 | msg[i+1] = (len >> i*8) & 0xff; |
samux | 3:9589afa4712e | 208 | } |
samux | 3:9589afa4712e | 209 | return 9; |
samux | 3:9589afa4712e | 210 | } |
samux | 3:9589afa4712e | 211 | } |
samux | 3:9589afa4712e | 212 | |
samux | 3:9589afa4712e | 213 | int Websocket::readChar(char * pC, bool block) { |
samux | 3:9589afa4712e | 214 | return read(pC, 1, 1); |
samux | 3:9589afa4712e | 215 | } |
samux | 3:9589afa4712e | 216 | |
samux | 4:466f90b7849a | 217 | int Websocket::sendOpcode(uint8_t opcode, char * msg) { |
samux | 4:466f90b7849a | 218 | msg[0] = 0x80 | (opcode & 0x0f); |
samux | 4:466f90b7849a | 219 | return 1; |
samux | 3:9589afa4712e | 220 | } |
samux | 3:9589afa4712e | 221 | |
samux | 4:466f90b7849a | 222 | int Websocket::sendMask(char * msg) { |
samux | 3:9589afa4712e | 223 | for (int i = 0; i < 4; i++) { |
samux | 4:466f90b7849a | 224 | msg[i] = 0; |
samux | 3:9589afa4712e | 225 | } |
samux | 3:9589afa4712e | 226 | return 4; |
samux | 3:9589afa4712e | 227 | } |
samux | 3:9589afa4712e | 228 | |
samux | 3:9589afa4712e | 229 | int Websocket::send(char * str) { |
samux | 4:466f90b7849a | 230 | char msg[strlen(str) + 15]; |
samux | 4:466f90b7849a | 231 | int idx = 0; |
samux | 4:466f90b7849a | 232 | idx = sendOpcode(0x01, msg); |
samux | 4:466f90b7849a | 233 | idx += sendLength(strlen(str), msg + idx); |
samux | 4:466f90b7849a | 234 | idx += sendMask(msg + idx); |
samux | 4:466f90b7849a | 235 | memcpy(msg+idx, str, strlen(str)); |
samux | 4:466f90b7849a | 236 | int res = write(msg, idx + strlen(str)); |
samux | 3:9589afa4712e | 237 | return res; |
samux | 3:9589afa4712e | 238 | } |
samux | 3:9589afa4712e | 239 | |
samux | 3:9589afa4712e | 240 | |
samux | 3:9589afa4712e | 241 | bool Websocket::read(char * message) { |
samux | 3:9589afa4712e | 242 | int i = 0; |
samux | 3:9589afa4712e | 243 | uint32_t len_msg; |
samux | 3:9589afa4712e | 244 | char opcode = 0; |
samux | 3:9589afa4712e | 245 | char c; |
samux | 3:9589afa4712e | 246 | char mask[4] = {0, 0, 0, 0}; |
samux | 3:9589afa4712e | 247 | bool is_masked = false; |
samux | 3:9589afa4712e | 248 | Timer tmr; |
samux | 3:9589afa4712e | 249 | |
samux | 3:9589afa4712e | 250 | // read the opcode |
samux | 3:9589afa4712e | 251 | tmr.start(); |
samux | 3:9589afa4712e | 252 | while (true) { |
samux | 3:9589afa4712e | 253 | if (tmr.read() > 3) { |
samux | 3:9589afa4712e | 254 | DBG("timeout ws\r\n"); |
samux | 3:9589afa4712e | 255 | return false; |
samux | 3:9589afa4712e | 256 | } |
donatien | 5:bb09d7a6c92f | 257 | |
donatien | 5:bb09d7a6c92f | 258 | if(!socket.is_connected()) |
donatien | 5:bb09d7a6c92f | 259 | { |
donatien | 5:bb09d7a6c92f | 260 | WARN("Connection was closed by server"); |
donatien | 5:bb09d7a6c92f | 261 | return false; |
donatien | 5:bb09d7a6c92f | 262 | } |
samux | 3:9589afa4712e | 263 | |
samux | 3:9589afa4712e | 264 | socket.set_blocking(false, 1); |
samux | 3:9589afa4712e | 265 | if (socket.receive(&opcode, 1) != 1) { |
samux | 3:9589afa4712e | 266 | socket.set_blocking(false, 2000); |
samux | 3:9589afa4712e | 267 | return false; |
samux | 3:9589afa4712e | 268 | } |
samux | 3:9589afa4712e | 269 | |
samux | 3:9589afa4712e | 270 | socket.set_blocking(false, 2000); |
samux | 3:9589afa4712e | 271 | |
samux | 3:9589afa4712e | 272 | if (opcode == 0x81) |
samux | 3:9589afa4712e | 273 | break; |
samux | 3:9589afa4712e | 274 | } |
samux | 3:9589afa4712e | 275 | DBG("opcode: 0x%X\r\n", opcode); |
samux | 3:9589afa4712e | 276 | |
samux | 3:9589afa4712e | 277 | readChar(&c); |
samux | 3:9589afa4712e | 278 | len_msg = c & 0x7f; |
samux | 3:9589afa4712e | 279 | is_masked = c & 0x80; |
samux | 3:9589afa4712e | 280 | if (len_msg == 126) { |
samux | 3:9589afa4712e | 281 | readChar(&c); |
samux | 3:9589afa4712e | 282 | len_msg = c << 8; |
samux | 3:9589afa4712e | 283 | readChar(&c); |
samux | 3:9589afa4712e | 284 | len_msg += c; |
samux | 3:9589afa4712e | 285 | } else if (len_msg == 127) { |
samux | 3:9589afa4712e | 286 | len_msg = 0; |
samux | 3:9589afa4712e | 287 | for (int i = 0; i < 8; i++) { |
samux | 3:9589afa4712e | 288 | readChar(&c); |
samux | 3:9589afa4712e | 289 | len_msg += (c << (7-i)*8); |
samux | 3:9589afa4712e | 290 | } |
samux | 3:9589afa4712e | 291 | } |
samux | 3:9589afa4712e | 292 | |
samux | 3:9589afa4712e | 293 | if (len_msg == 0) { |
samux | 3:9589afa4712e | 294 | return false; |
samux | 3:9589afa4712e | 295 | } |
samux | 3:9589afa4712e | 296 | DBG("length: %d\r\n", len_msg); |
samux | 3:9589afa4712e | 297 | |
samux | 3:9589afa4712e | 298 | if (is_masked) { |
samux | 3:9589afa4712e | 299 | for (i = 0; i < 4; i++) |
samux | 3:9589afa4712e | 300 | readChar(&c); |
samux | 3:9589afa4712e | 301 | mask[i] = c; |
samux | 3:9589afa4712e | 302 | } |
samux | 3:9589afa4712e | 303 | |
samux | 3:9589afa4712e | 304 | int nb = read(message, len_msg, len_msg); |
samux | 3:9589afa4712e | 305 | if (nb != len_msg) |
samux | 3:9589afa4712e | 306 | return false; |
samux | 3:9589afa4712e | 307 | |
samux | 3:9589afa4712e | 308 | for (i = 0; i < len_msg; i++) { |
samux | 3:9589afa4712e | 309 | message[i] = message[i] ^ mask[i % 4]; |
samux | 3:9589afa4712e | 310 | } |
samux | 3:9589afa4712e | 311 | |
samux | 3:9589afa4712e | 312 | message[len_msg] = '\0'; |
samux | 3:9589afa4712e | 313 | |
samux | 3:9589afa4712e | 314 | return true; |
samux | 3:9589afa4712e | 315 | } |
samux | 3:9589afa4712e | 316 | |
samux | 3:9589afa4712e | 317 | bool Websocket::close() { |
samux | 3:9589afa4712e | 318 | if (!is_connected()) |
samux | 3:9589afa4712e | 319 | return false; |
samux | 3:9589afa4712e | 320 | |
samux | 3:9589afa4712e | 321 | int ret = socket.close(); |
samux | 3:9589afa4712e | 322 | if (ret < 0) { |
samux | 3:9589afa4712e | 323 | ERR("Could not disconnect"); |
samux | 3:9589afa4712e | 324 | return false; |
samux | 3:9589afa4712e | 325 | } |
samux | 3:9589afa4712e | 326 | return true; |
samux | 3:9589afa4712e | 327 | } |
samux | 3:9589afa4712e | 328 | |
samux | 3:9589afa4712e | 329 | bool Websocket::is_connected() { |
samux | 3:9589afa4712e | 330 | return socket.is_connected(); |
samux | 3:9589afa4712e | 331 | } |
samux | 3:9589afa4712e | 332 | |
donatien | 6:86e89a0369b9 | 333 | char* Websocket::getPath() { |
samux | 3:9589afa4712e | 334 | return path; |
samux | 3:9589afa4712e | 335 | } |
samux | 3:9589afa4712e | 336 | |
samux | 3:9589afa4712e | 337 | int Websocket::write(char * str, int len) { |
samux | 3:9589afa4712e | 338 | int res = 0, idx = 0; |
samux | 3:9589afa4712e | 339 | |
samux | 3:9589afa4712e | 340 | for (int j = 0; j < MAX_TRY_WRITE; j++) { |
donatien | 5:bb09d7a6c92f | 341 | |
donatien | 5:bb09d7a6c92f | 342 | if(!socket.is_connected()) |
donatien | 5:bb09d7a6c92f | 343 | { |
donatien | 5:bb09d7a6c92f | 344 | WARN("Connection was closed by server"); |
donatien | 5:bb09d7a6c92f | 345 | break; |
donatien | 5:bb09d7a6c92f | 346 | } |
samux | 3:9589afa4712e | 347 | |
samux | 3:9589afa4712e | 348 | if ((res = socket.send_all(str + idx, len - idx)) == -1) |
samux | 3:9589afa4712e | 349 | continue; |
samux | 3:9589afa4712e | 350 | |
samux | 3:9589afa4712e | 351 | idx += res; |
samux | 3:9589afa4712e | 352 | |
samux | 3:9589afa4712e | 353 | if (idx == len) |
samux | 3:9589afa4712e | 354 | return len; |
samux | 3:9589afa4712e | 355 | } |
samux | 3:9589afa4712e | 356 | |
samux | 3:9589afa4712e | 357 | return (idx == 0) ? -1 : idx; |
samux | 3:9589afa4712e | 358 | } |
samux | 3:9589afa4712e | 359 | |
samux | 3:9589afa4712e | 360 | int Websocket::read(char * str, int len, int min_len) { |
samux | 3:9589afa4712e | 361 | int res = 0, idx = 0; |
samux | 3:9589afa4712e | 362 | |
samux | 3:9589afa4712e | 363 | for (int j = 0; j < MAX_TRY_WRITE; j++) { |
samux | 3:9589afa4712e | 364 | |
samux | 3:9589afa4712e | 365 | if ((res = socket.receive_all(str + idx, len - idx)) == -1) |
samux | 3:9589afa4712e | 366 | continue; |
samux | 3:9589afa4712e | 367 | |
samux | 3:9589afa4712e | 368 | idx += res; |
samux | 3:9589afa4712e | 369 | |
samux | 3:9589afa4712e | 370 | if (idx == len || (min_len != -1 && idx > min_len)) |
samux | 3:9589afa4712e | 371 | return idx; |
samux | 3:9589afa4712e | 372 | } |
samux | 3:9589afa4712e | 373 | |
samux | 3:9589afa4712e | 374 | return (idx == 0) ? -1 : idx; |
samux | 3:9589afa4712e | 375 | } |