Client of WebSocket protocol
Fork of WebSocketClient by
Diff: Websocket.cpp
- Revision:
- 8:c9da00db9d33
- Parent:
- 7:4567996414a5
--- a/Websocket.cpp Fri Feb 08 12:33:04 2013 +0000 +++ b/Websocket.cpp Fri Feb 24 20:06:49 2017 +0000 @@ -1,7 +1,11 @@ #include "Websocket.h" +#include "Pmed_reset.h" +#include "Log.h" #define MAX_TRY_WRITE 20 #define MAX_TRY_READ 10 +#define MAX_TRY_CONNECT 50 +#define WAIT_CONNECT_TIME 10000 //Debug is disabled by default #if 0 @@ -14,14 +18,166 @@ #define ERR(x, ...) #endif -#define INFO(x, ...) printf("[WebSocket : INFO]"x"\r\n", ##__VA_ARGS__); +#define INFO(x, ...) printf("[WebSocket : INFO]"x"\r\n", ##__VA_ARGS__); + +char * Websocket::captureMessage; +bool Websocket::haveMessage; +bool Websocket::isConect; +bool disconnectedWS = false; -Websocket::Websocket(char * url) { +void Websocket::Websocket_Thread(void const *arg) +{ + //Log::writeEntry("WebSocket Thread Start"); + printf("Thread WebSocket Iniciada\r\n"); + int i = 0, f = 0; + char *comando = NULL; + char *tomada = NULL; + char *canal = NULL; + char *limite = NULL; + char server[30]; + char buffer[1024] = {}; + //bool external = false; + strcpy(server, "ws://"); + strcat(server, Settings::get_networkServer()); + strcat(server, ":8080"); + + //printf("Servidor WS: %s", server); + + Websocket ws(server); + + while(1){ + if(ws.is_connected()){ + if (f == 0){ + //Log::writeEntry("WebSocket Conectado."); + //printf("\nMensagem do MBED para servidor WebSocket em PHP\n\n"); + //ws.send("#*MBEDStart*#Mensagem do MBED para servidor WebSocket em PHP\n"); //mensagem para registrar login do mbed no banco de dados (ainda por implementar) + f = 1; + } + if(disconnectedWS != false){ + disconnectedWS = false; + } + if(haveMessage){ + ws.send(captureMessage); + haveMessage = false; + captureMessage = NULL; + } + if (ws.read(buffer)) { + //ws.send("#*MBEDTest*#"); + //printf("Recebido Mensagem WebSocket: %s\r\n", buffer); + comando = strtok(buffer, ":"); + if(strcmp(comando, "test") == 0){ + //printf("\r\nRecebido por WebSocket o Comando de Teste."); + ws.send("#*MBEDTest*#"); + //Log::writeEntry("Recebido por WebSocket o Comando de Teste."); + } + if(strcmp(comando, "capture") == 0){ + tomada = strtok(NULL, ":"); + canal = strtok(NULL, " "); + /* + printf("Comando: %s\n", comando); + printf("Tomada: %s\n", tomada); + printf("Canal: %s\n", canal); + printf("Tamanho da string: %d\n", strlen(buffer)); + */ + + EventDetector::externalCapture(atoi(tomada), canal[0]); + } + if(strcmp(comando, "reset") == 0){ + Pmed_reset("Reiniciando o MBED por WebSocket."); + } + + if(strcmp(comando, "setLimit") == 0 || strcmp(comando, "setStandByLimit") == 0){ + tomada = strtok(NULL, ":"); + canal = strtok(NULL, ":"); + limite = strtok(NULL, " "); + for(i=0; (unsigned)i < _PRTGMD_SETTINGS_MAX_NUMBER_OF_OUTLETS_; i++){ + if(atoi(tomada) == Settings::get_outlet(i)){ + if(strcmp(comando, "setLimit") == 0){ + if(strcmp(canal, "p") == 0){ + printf("Entrou canal fase e setou limite %s para tomada %s\r\n", limite, tomada); + Settings::set_limit(i*2,strtod(limite, NULL)); + } + if(strcmp(canal, "d") == 0){ + printf("Entrou canal fuga e setou limite %s para tomada %s\r\n", limite, tomada); + Settings::set_limit(i*2+1,strtod(limite, NULL)); + } + } + if(strcmp(comando, "setStandByLimit") == 0){ + if(strcmp(canal, "p") == 0){ + printf("Entrou canal StandBy de fase e setou limite %s para tomada %s\r\n", limite, tomada); + Settings::set_standbyLimit(i*2,strtod(limite, NULL)); + } + if(strcmp(canal, "d") == 0){ + printf("Entrou canal StandBy de fuga e setou limite %s para tomada %s\r\n", limite, tomada); + Settings::set_standbyLimit(i*2+1,strtod(limite, NULL)); + } + } + break; + }else{ + if(i == _PRTGMD_SETTINGS_MAX_NUMBER_OF_OUTLETS_ - 1){ + printf("Tomada %d nao encontrada para atualizar limite!\r\n", atoi(tomada)); + } + } + } + //printf("Comando: %s\n", comando); + //printf("Tomada: %d\n", atoi(tomada)); + //printf("Canal: %s\n", canal); + //printf("Limite: %f\n", strtod(limite, NULL)); + } + /* + if(strcmp(comando, "iniciar") == 0){ + printf("\n\nChegou o Iniciar!!!\n\n"); + external = true; + } + if(strcmp(comando, "parar") == 0){ + printf("\n\nChegou o Parar!!!\n\n"); + external = false; + } + if(external){ + + }*/ + } + /* + if (!ws.is_connected()){ + printf("Caiu a conexao!!!\n"); + Thread::wait(Settings::get_delayTry()); + for(int i=0; i < MAX_TRY_CONNECT; i++){ + Thread::wait(Settings::get_delayTry()); + //if(ws.connect()){ + // break; + //} + } + //Websocket ws("ws://192.168.103.101:8080"); + ws.connect(); + }*/ + }else{ + if(!disconnectedWS){ + if(ws.close()){ + disconnectedWS = true; + Log::writeEntry("WebSocket Desconectado."); + } + } + Thread::wait(WAIT_CONNECT_TIME); + if(ws.connect()){ + //Log::writeEntry("WebSocket Conectado."); + printf("Thread WebSocket Conectou no Servidor %s\r\n", Settings::get_networkServer()); + } + //Thread::wait(10000); + } + } + +} + +Websocket::Websocket(char const * url) { fillFields(url); socket.set_blocking(false, 400); } -void Websocket::fillFields(char * url) { +bool Websocket::wsIsConnected(){ + return isConect; +} + +void Websocket::fillFields(char const * url) { int ret = parseURL(url, scheme, sizeof(scheme), host, sizeof(host), &port, path, sizeof(path)); if(ret) { @@ -119,7 +275,7 @@ while (socket.connect(host, port) < 0) { ERR("Unable to connect to (%s) on port (%d)", host, port); - wait(0.2); + Thread::wait(200); return false; } @@ -141,19 +297,18 @@ sprintf(cmd, "Sec-WebSocket-Version: 13\r\n\r\n"); int ret = write(cmd, strlen(cmd)); - if (ret != strlen(cmd)) { + if ((unsigned)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); @@ -171,8 +326,8 @@ close(); return false; } - - INFO("\r\nhost: %s\r\npath: %s\r\nport: %d\r\n\r\n", host, path, port); + + //INFO("\r\nhost: %s\r\npath: %s\r\nport: %d\r\n\r\n", host, path, port); return true; } @@ -290,7 +445,7 @@ if (nb != len_msg) return false; - for (i = 0; i < len_msg; i++) { + for (i = 0; (unsigned)i < len_msg; i++) { message[i] = message[i] ^ mask[i % 4]; } @@ -300,10 +455,10 @@ } bool Websocket::close() { - if (!is_connected()) - return false; + //if (!is_connected()) + //return false; - int ret = socket.close(); + int ret = socket.close(true); if (ret < 0) { ERR("Could not disconnect"); return false; @@ -312,6 +467,7 @@ } bool Websocket::is_connected() { + isConect = socket.is_connected(); return socket.is_connected(); }