File content as of revision 3:9b00db719afa:

#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", str, '\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 str[10];
    
    wait(0.25);
    if(!wifi->CmdMode())
    {
        printf("Websocket::connected: cannot enter in cmd mode\r\n");
        return false;
    }
    wait(0.25);
    
    wifi->Send("show c\r\n", "NO");
    wifi->read(str);
    
    if(str[3] == '1')
    {
        if(!wifi->exit())
        {
            printf("Websocket::connected: cannot exit\r\n");
            return false;
        }
        return true;
    }
    return false;
        
}