Diff: Websocket.cpp

Revision:

0:08afdb5fb5ca

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Websocket.cpp	Thu Oct 20 09:30:01 2011 +0000
@@ -0,0 +1,195 @@
+#include "Websocket.h"
+#include <string>
+
+Websocket::Websocket(char * url, Wifly * wifi) {
+    this->wifi = wifi;
+    netif = WIF;
+    fillFields(url);
+}
+
+
+void Websocket::fillFields(char * url) {
+    char *res = NULL;
+    char *res1 = NULL;
+
+    char buf[50];
+    strcpy(buf, url);
+
+    res = strtok(buf, ":");
+    if (strcmp(res, "ws")) {
+#ifdef DEBUG
+        printf("\r\nFormat error: please use: \"ws://ip-or-domain[:port]/path\"\r\n\r\n");
+#endif
+    } 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, " ");
+        //port
+        port = (res1 != NULL) ? res1 : "80";
+
+        if (res != NULL)
+            ip_domain = res;
+    }
+}
+
+
+bool Websocket::connect() {
+    char cmd[50];
+    if (netif == WIF) {
+        wifi->send("exit\r", "NO");
+        //enter in cmd mode
+        while (!wifi->send("$$$", "CMD")) {
+#ifdef DEBUG
+            printf("cannot enter in CMD mode\r\n");
+#endif
+            wifi->exit();
+        }
+
+
+        //open the connection
+        sprintf(cmd, "open %s %s\r\n", ip_domain.c_str(), port.c_str());
+        if (!wifi->send(cmd, "OPEN*")) {
+#ifdef DEBUG
+            printf("Websocket::connect cannot open\r\n");
+#endif
+            return false;
+        }
+
+
+        //send websocket HTTP header
+        sprintf(cmd, "GET /%s HTTP/1.1\r\n", path.c_str());
+        wifi->send(cmd, "NO");
+
+        sprintf(cmd, "Host: %s:%s\r\n", ip_domain.c_str(), port.c_str());
+        wifi->send(cmd, "NO");
+
+        wifi->send("Upgrade: WebSocket\r\n", "NO");
+
+        sprintf(cmd, "Origin: http:%s:%s\r\n", ip_domain.c_str(), port.c_str());
+        wifi->send(cmd, "NO");
+
+
+        wifi->send("Connection: Upgrade\r\n", "NO");
+        wifi->send("Sec-WebSocket-Key1: 4 @1  46546xW%0l 1 5\r\n", "NO");
+        wifi->send("Sec-WebSocket-key2: 12998 5 Y3 1  .P00\r\n\r\n", "NO");
+        if (!wifi->send("^n:ds[4U", "8jKS'y:G*Co,Wxa-"))
+            return false;
+#ifdef DEBUG
+        printf("\r\nip_domain: %s\r\npath: /%s\r\nport: %s\r\n\r\n",this->ip_domain.c_str(), this->path.c_str(), this->port.c_str());
+#endif
+        return true;
+    }
+    //the program shouldn't be here
+    return false;
+}
+
+void Websocket::send(char * str) {
+    if (netif == WIF) {
+        wifi->putc('\x00');
+        wifi->send(str, "NO");
+        wifi->putc('\xff');
+    }
+}
+
+bool Websocket::read(char * message) {
+    int i = 0;
+
+    if (netif == WIF) {
+        if (!wifi->read(message))
+            return false;
+
+        //we check if the first byte is 0x00
+        if (message == NULL || message[0] != 0x00) {
+            message = NULL;
+            return false;
+        }
+
+        while (message[i + 1] != 0xff && i < strlen(message + 1))
+            i++;
+
+        if (message[i+1] == 0xff) {
+            message[i+1] = 0;
+            memcpy(message, message + 1, strlen(message + 1) + 1);
+            return true;
+        } else {
+            message = NULL;
+            return false;
+        }
+    }
+    //the program shouldn't be here
+    return false;
+}
+
+bool Websocket::close() {
+    if (netif == WIF) {
+        if (!wifi->cmdMode()) {
+#ifdef DEBUG
+            printf("Websocket::close: cannot enter in cmd mode\r\n");
+#endif
+            return false;
+        }
+
+        wifi->send("close\r", "NO");
+
+        if (!wifi->exit())
+            return false;
+    }
+    //the program shouldn't be here
+    return false;
+}
+
+
+
+bool Websocket::connected() {
+    if (netif == WIF) {
+        char str[10];
+
+        wait(0.25);
+        if (!wifi->cmdMode()) {
+#ifdef DEBUG
+            printf("Websocket::connected: cannot enter in cmd mode\r\n");
+#endif
+            return false;
+        }
+        wait(0.25);
+
+        wifi->send("show c\r\n", "NO", str);
+
+        if (str[3] == '1') {
+            if (!wifi->exit()) {
+#ifdef DEBUG
+                printf("Websocket::connected: cannot exit\r\n");
+#endif
+                return false;
+            }
+            return true;
+        }
+        if (!wifi->exit()) {
+#ifdef DEBUG
+            printf("Websocket::connected: cannot exit\r\n");
+#endif
+        }
+        return false;
+    }
+    //the program shouldn't be here
+    return false;
+}
+
+std::string Websocket::getPath()
+{
+    return path;
+}
+
+
+
+