Forked mbed official WiflyInterface (interface for Roving Networks Wifly modules) which includes the possibility to use TCPSocketServer::accept as a non-blocking cal.

Dependents:   WiFlyHTTPServerSample MultiThreadingHTTPServer

Fork of WiflyInterface by mbed official

Socket/TCPSocketServer.cpp

Committer:
leihen
Date:
2013-06-05
Revision:
9:4f6f2f35a21a
Parent:
5:48d55083d2ff

File content as of revision 9:4f6f2f35a21a:

/* Copyright (C) 2012 mbed.org, MIT License
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software
 * and associated documentation files (the "Software"), to deal in the Software without restriction,
 * including without limitation the rights to use, copy, modify, merge, publish, distribute,
 * sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all copies or
 * substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
 * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

#include "TCPSocketServer.h"
#include <string>

#define _DEBUG 0

#if (_DEBUG && !defined(TARGET_LPC11U24))
#define INFO(x, ...) std::printf("[TCPSocketServer : INFO]"x"\r\n", ##__VA_ARGS__);
#define WARN(x, ...) std::printf("[TCPSocketServer : WARN]"x"\r\n", ##__VA_ARGS__);
#define ERR(x, ...) std::printf("[TCPSocketServer : ERR]"x"\r\n", ##__VA_ARGS__);
#else
#define INFO(x, ...)
#define WARN(x, ...)
#define ERR(x, ...)
#endif


TCPSocketServer::TCPSocketServer() {connected = false;}

// Server initialization
int TCPSocketServer::bind(int port) {
    char cmd[20];
    
    // set TCP protocol
    wifi->setProtocol(TCP);
    
    // set local port
    sprintf(cmd, "set i l %d\r", port);
    if (!wifi->sendCommand(cmd, "AOK"))
        return -1;
    
    // save
    if (!wifi->sendCommand("save\r", "Stor"))
        return -1;
    
    // reboot
    wifi->reboot();
    
    // connect the network
    if (wifi->isDHCP()) {
        if (!wifi->sendCommand("join\r", "DHCP=ON", NULL, 10000))
            return -1;
    } else {
        if (!wifi->sendCommand("join\r", "Associated", NULL, 10000))
            return -1;
    }
        
    // exit
    wifi->exit();
    
    wait(0.2);
    wifi->flush();
    return 0;
}

int TCPSocketServer::listen(int backlog) {
    if (backlog != 1)
        return -1;
    return 0;
}


int TCPSocketServer::accept(TCPSocketConnection& connection) {
    char c;
    string str;
    bool o_find = false;
    Timer tm;
    while (1) {
        INFO("Trying accept (%s)\n", (connected ? "Connected" : "Unconnected"));
        while(!wifi->readable())
        {
            if (!_blocking && (tm.read_ms() > _timeout))
                return 1;
        }
        INFO("Data (\'%c\')\n", wifi->peek());
        if (!connected || (wifi->peek() == '*')) {
            while (!o_find) {
                c = wifi->getc();
                INFO("%c", c);
                str += c;
                if (c == '*') {
                     if (str.find("*OPEN*") != string::npos) {
                        // connection found !
                        INFO("Connection received !\n");
                        connected = true;
                        return 0;
                     }
                     if (str.find("*CLOS*") != string::npos) {
                        //  Connection closed !
                        INFO("Connection closed !\n");
                        str = "";
                        connected = false;
                        break;
                     }
                }
            }
        }
        else {
            //  We are still connected and there is new data, so just behave as if the accept was done.
            INFO("Connection not yet closed, recycling because new data is available.\n");
            return 0;
        }
    }
}