Diff: Websocket.cpp

Revision:

0:21fe3b05249f

Child:

2:4a841609a3a3

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Websocket.cpp	Fri Aug 12 11:21:45 2011 +0000
@@ -0,0 +1,156 @@
+#include "Websocket.h"
+
+Websocket::Websocket(char * url, int port, Wifly * wifi)
+{
+    char *res = NULL;
+    char buf[30];
+    strcpy(buf, url);
+     
+    res = strtok(buf, ":");
+    if (strcmp(res, "ws"))
+    {
+        this->ip_domain = NULL;
+        this->path = NULL;
+        printf("\r\nFormat error: please use: \"ws://ip-or-domain/path\"\r\n\r\n");
+    }
+    else
+    {
+        res = strtok(NULL, ":");
+        res = strtok(res, "/");
+        if(res != NULL)
+        {
+            ip_domain = (char *) malloc (sizeof(char) * strlen(res));
+            strcpy(this->ip_domain, res);
+        }
+        res = strtok(NULL, " ");
+        if(res != NULL)
+        {
+            path = (char *) malloc (sizeof(char) * strlen(res));
+            strcpy(this->path, res);
+        }
+        
+        this->port = port;
+        this->wifi = wifi;
+        
+    }
+}
+
+bool Websocket::connect()
+{
+    char cmd[50];
+    wifi->Send("exit\r", "NO");
+    //enter in cmd mode
+    while(!wifi->Send("$$$", "CMD"))
+    {
+        printf("cannot enter in CMD mode\r\n");
+        wifi->exit();
+    }
+    
+
+    //open the connection
+    sprintf(cmd, "open %s %d\r\n", ip_domain, port);
+    if(!wifi->Send(cmd, "OPEN"))
+    {   
+        printf("Websocket::connect cannot open\r\n");
+        return false;
+    }
+    
+    
+    //send websocket HTTP header
+    sprintf(cmd, "GET /%s HTTP/1.1\r\n", path);
+    wifi->Send(cmd, "NO");
+    
+    sprintf(cmd, "Host: %s:%d\r\n", ip_domain, port);
+    wifi->Send(cmd, "NO");
+    
+    wifi->Send("Upgrade: WebSocket\r\n", "NO");
+    
+    sprintf(cmd, "Origin: http:%s:%d\r\n", ip_domain, port);
+    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;
+        
+    printf("\r\nip_domain: %s\r\npath: /%s\r\nport: %d\r\n\r\n",this->ip_domain, this->path, this->port);
+    return true;
+}
+
+void Websocket::Send(char * str)
+{
+    char cmd[100];
+    wifi->putc('\x00');
+    sprintf(cmd, "%s%c", "test", '\xff');
+    wifi->Send(cmd, "NO");
+}
+
+bool Websocket::read(char * message)
+{
+    int i = 0;
+    char char_read;
+    
+    if (!wifi->readable())
+    {
+        message = NULL;
+        return false;
+    }
+        
+    if (wifi->getc() != 0x00)
+    {
+        message = NULL;
+        return false;
+    }
+        
+    while ( (char_read = wifi->getc()) != 0xff)
+        message[i++] = char_read;
+        
+    return true;
+}
+
+bool Websocket::close()
+{
+    if(!wifi->CmdMode())
+    {
+        printf("Websocket::close: cannot enter in cmd mode\r\n");
+        return false;
+    }
+    
+    wifi->Send("close\r", "NO");
+    
+    if(!wifi->exit())
+        return false;
+        
+    return true;
+}
+
+
+
+bool Websocket::connected()
+{
+    char cmd[30];
+    
+    if(!wifi->CmdMode())
+    {
+        printf("Websocket::connected: cannot enter in cmd mode\r\n");
+        return false;
+    }
+    
+    //try to (re)open the connection
+    sprintf(cmd, "open %s %d\r\n", ip_domain, port);
+    if(wifi->Send(cmd, "Connected"))
+    {   
+        if(!wifi->exit())
+            return false;
+        return true;
+    }
+    else
+    {
+        wifi->exit();
+        return false;
+    }
+        
+    
+}
\ No newline at end of file