TCPSocketServer.cpp

00001 /* Copyright (C) 2012 mbed.org, MIT License
00002  *
00003  * Permission is hereby granted, free of charge, to any person obtaining a copy of this software
00004  * and associated documentation files (the "Software"), to deal in the Software without restriction,
00005  * including without limitation the rights to use, copy, modify, merge, publish, distribute,
00006  * sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
00007  * furnished to do so, subject to the following conditions:
00008  *
00009  * The above copyright notice and this permission notice shall be included in all copies or
00010  * substantial portions of the Software.
00011  *
00012  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
00013  * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
00014  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
00015  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
00016  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
00017  */
00018 
00019 #include "TCPSocketServer.h"
00020 #include <string>
00021 
00022 TCPSocketServer::TCPSocketServer() {}
00023 
00024 // Server initialization
00025 int TCPSocketServer::bind(int port) {
00026     char cmd[20];
00027     
00028     // set TCP protocol
00029     wifi->setProtocol(TCP);
00030     
00031     // set local port
00032     sprintf(cmd, "set i l %d\r", port);
00033     if (!wifi->sendCommand(cmd, "AOK"))
00034         return -1;
00035     
00036     // save
00037     if (!wifi->sendCommand("save\r", "Stor"))
00038         return -1;
00039     
00040     // reboot
00041     wifi->reboot();
00042     
00043     // connect the network
00044     if (wifi->isDHCP()) {
00045         if (!wifi->sendCommand("join\r", "DHCP=ON", NULL, 10000))
00046             return -1;
00047     } else {
00048         if (!wifi->sendCommand("join\r", "Associated", NULL, 10000))
00049             return -1;
00050     }
00051         
00052     // exit
00053     wifi->exit();
00054     
00055     wait(0.2);
00056     wifi->flush();
00057     return 0;
00058 }
00059 
00060 int TCPSocketServer::listen(int backlog) {
00061     if (backlog != 1)
00062         return -1;
00063     return 0;
00064 }
00065 
00066 
00067 int TCPSocketServer::accept(TCPSocketConnection& connection) {
00068     int nb_available = 0, pos = 0;
00069     char c;
00070     string str;
00071     bool o_find = false;
00072     while (1) {
00073         while(!wifi->readable());
00074         nb_available = wifi->readable();
00075         for (int i = 0; i < nb_available; i++) {
00076             c = wifi->getc();
00077             if (c == '*') {
00078                 o_find = true;
00079             }
00080             if (o_find && c != '\r' && c != '\n') {
00081                 str += c;
00082                 pos = str.find("*OPEN*");
00083                 if (pos != string::npos) {
00084                     wifi->flush();
00085                     return 0;
00086                 }
00087             }
00088         }
00089     }
00090 }
00091 
00092 int TCPSocketServer::acceptWithTimeout(TCPSocketConnection& connection, int timeout) {
00093     int nb_available = 0, pos = 0;
00094     char c;
00095     string str;
00096     bool o_find = false;
00097     time_t currTime = time(NULL);
00098     while (1) {
00099         while(!wifi->readable() && time(NULL)-currTime<=timeout);
00100         if (time(NULL)-currTime>timeout) {
00101             wifi->flush();
00102             return -1;
00103         }
00104         nb_available = wifi->readable();
00105         for (int i = 0; i < nb_available; i++) {
00106             c = wifi->getc();
00107             if (c == '*') {
00108                 o_find = true;
00109             }
00110             if (o_find && c != '\r' && c != '\n') {
00111                 str += c;
00112                 pos = str.find("*OPEN*");
00113                 if (pos != string::npos) {
00114                     wifi->flush();
00115                     return 0;
00116                 }
00117             }
00118         }
00119     }
00120 }