fork of WebSocketClient with some fixes
Fork of WebSocketClient by
Websocket.cpp
- Committer:
- samux
- Date:
- 2012-08-20
- Revision:
- 4:466f90b7849a
- Parent:
- 3:9589afa4712e
- Child:
- 5:bb09d7a6c92f
File content as of revision 4:466f90b7849a:
#include "Websocket.h" #include <string> #define MAX_TRY_WRITE 50 #define MAX_TRY_READ 30 //Debug is disabled by default #if 0 #define DBG(x, ...) std::printf("[WebSocket : DBG]"x"\r\n", ##__VA_ARGS__); #define WARN(x, ...) std::printf("[WebSocket : WARN]"x"\r\n", ##__VA_ARGS__); #define ERR(x, ...) std::printf("[WebSocket : ERR]"x"\r\n", ##__VA_ARGS__); #else #define DBG(x, ...) #define WARN(x, ...) #define ERR(x, ...) #endif #define INFO(x, ...) printf("[WebSocket : INFO]"x"\r\n", ##__VA_ARGS__); Websocket::Websocket(char * url) { fillFields(url); socket.set_blocking(false, 400); } void Websocket::fillFields(char * url) { char *res = NULL; char *res1 = NULL; char buf[50]; strcpy(buf, url); res = strtok(buf, ":"); if (strcmp(res, "ws")) { DBG("\r\nFormat printfor: please use: \"ws://ip-or-domain[:port]/path\"\r\n\r\n"); } else { //ip_domain and port res = strtok(NULL, "/"); //path res1 = strtok(NULL, " "); if (res1 != NULL) { path = res1; } //ip_domain res = strtok(res, ":"); //port res1 = strtok(NULL, " "); if (res1 != NULL) { port = res1; } else { port = "80"; } if (res != NULL) { ip_domain = res; } } } bool Websocket::connect() { char cmd[200]; while (socket.connect(ip_domain.c_str(), atoi(port.c_str())) < 0) { ERR("Unable to connect to (%s) on port (%d)\r\n", ip_domain.c_str(), atoi(port.c_str())); wait(0.2); } // sent http header to upgrade to the ws protocol sprintf(cmd, "GET /%s HTTP/1.1\r\n", path.c_str()); write(cmd, strlen(cmd)); sprintf(cmd, "Host: %s:%s\r\n", ip_domain.c_str(), port.c_str()); write(cmd, strlen(cmd)); sprintf(cmd, "Upgrade: WebSocket\r\n"); write(cmd, strlen(cmd)); sprintf(cmd, "Connection: Upgrade\r\n"); write(cmd, strlen(cmd)); sprintf(cmd, "Sec-WebSocket-Key: L159VM0TWUzyDxwJEIEzjw==\r\n"); write(cmd, strlen(cmd)); sprintf(cmd, "Sec-WebSocket-Version: 13\r\n\r\n"); int ret = write(cmd, strlen(cmd)); if (ret != strlen(cmd)) { close(); ERR("Could not send request"); return false; } ret = read(cmd, 200, 100); if (ret < 0) { close(); ERR("Could not receive answer\r\n"); return false; } cmd[ret] = '\0'; DBG("recv: %s\r\n", cmd); if ( strstr(cmd, "DdLWT/1JcX+nQFHebYP+rqEx5xI=") == NULL ) { ERR("Wrong answer from server, got \"%s\" instead\r\n", cmd); do { ret = read(cmd, 200, 100); if (ret < 0) { ERR("Could not receive answer\r\n"); return false; } cmd[ret] = '\0'; printf("%s",cmd); } while (ret > 0); close(); return false; } 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()); return true; } int Websocket::sendLength(uint32_t len, char * msg) { if (len < 126) { msg[0] = len | (1<<7); return 1; } else if (len < 65535) { msg[0] = 126 | (1<<7); msg[1] = (len >> 8) & 0xff; msg[2] = len & 0xff; return 3; } else { msg[0] = 127 | (1<<7); for (int i = 0; i < 8; i++) { msg[i+1] = (len >> i*8) & 0xff; } return 9; } } int Websocket::readChar(char * pC, bool block) { return read(pC, 1, 1); } int Websocket::sendOpcode(uint8_t opcode, char * msg) { msg[0] = 0x80 | (opcode & 0x0f); return 1; } int Websocket::sendMask(char * msg) { for (int i = 0; i < 4; i++) { msg[i] = 0; } return 4; } int Websocket::send(char * str) { char msg[strlen(str) + 15]; int idx = 0; idx = sendOpcode(0x01, msg); idx += sendLength(strlen(str), msg + idx); idx += sendMask(msg + idx); memcpy(msg+idx, str, strlen(str)); int res = write(msg, idx + strlen(str)); return res; } bool Websocket::read(char * message) { int i = 0; uint32_t len_msg; char opcode = 0; char c; char mask[4] = {0, 0, 0, 0}; bool is_masked = false; Timer tmr; // read the opcode tmr.start(); while (true) { if (tmr.read() > 3) { DBG("timeout ws\r\n"); return false; } socket.set_blocking(false, 1); if (socket.receive(&opcode, 1) != 1) { socket.set_blocking(false, 2000); return false; } socket.set_blocking(false, 2000); if (opcode == 0x81) break; } DBG("opcode: 0x%X\r\n", opcode); readChar(&c); len_msg = c & 0x7f; is_masked = c & 0x80; if (len_msg == 126) { readChar(&c); len_msg = c << 8; readChar(&c); len_msg += c; } else if (len_msg == 127) { len_msg = 0; for (int i = 0; i < 8; i++) { readChar(&c); len_msg += (c << (7-i)*8); } } if (len_msg == 0) { return false; } DBG("length: %d\r\n", len_msg); if (is_masked) { for (i = 0; i < 4; i++) readChar(&c); mask[i] = c; } int nb = read(message, len_msg, len_msg); if (nb != len_msg) return false; for (i = 0; i < len_msg; i++) { message[i] = message[i] ^ mask[i % 4]; } message[len_msg] = '\0'; return true; } bool Websocket::close() { if (!is_connected()) return false; int ret = socket.close(); if (ret < 0) { ERR("Could not disconnect"); return false; } return true; } bool Websocket::is_connected() { return socket.is_connected(); } std::string Websocket::getPath() { return path; } int Websocket::write(char * str, int len) { int res = 0, idx = 0; for (int j = 0; j < MAX_TRY_WRITE; j++) { if ((res = socket.send_all(str + idx, len - idx)) == -1) continue; idx += res; if (idx == len) return len; } return (idx == 0) ? -1 : idx; } int Websocket::read(char * str, int len, int min_len) { int res = 0, idx = 0; for (int j = 0; j < MAX_TRY_WRITE; j++) { if ((res = socket.receive_all(str + idx, len - idx)) == -1) continue; idx += res; if (idx == len || (min_len != -1 && idx > min_len)) return idx; } return (idx == 0) ? -1 : idx; }