mbed official / WiflyInterface

WiflyInterface

Dependents:   test_over_wifly

Fork of WiflyInterface by mbed official

Socket/TCPSocketServer.cpp@9:c77799a03294, 2014-01-28 (annotated)

Committer:
Bogdan Marinescu
Date:
Tue Jan 28 11:00:48 2014 +0200
Revision:
9:c77799a03294
Parent:
4:0bcec6272784 
Address various potential buffer overflow issues



https://mbed.org/comments/cr/93/4779/#c8499

Who changed what in which revision?

UserRevisionLine numberNew contents of line
samux 1:fb4494783863 1 /* Copyright (C) 2012 mbed.org, MIT License
samux 1:fb4494783863 2 *
samux 1:fb4494783863 3 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software
samux 1:fb4494783863 4 * and associated documentation files (the "Software"), to deal in the Software without restriction,
samux 1:fb4494783863 5 * including without limitation the rights to use, copy, modify, merge, publish, distribute,
samux 1:fb4494783863 6 * sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
samux 1:fb4494783863 7 * furnished to do so, subject to the following conditions:
samux 1:fb4494783863 8 *
samux 1:fb4494783863 9 * The above copyright notice and this permission notice shall be included in all copies or
samux 1:fb4494783863 10 * substantial portions of the Software.
samux 1:fb4494783863 11 *
samux 1:fb4494783863 12 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
samux 1:fb4494783863 13 * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
samux 1:fb4494783863 14 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
samux 1:fb4494783863 15 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
samux 1:fb4494783863 16 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
samux 1:fb4494783863 17 */
samux 1:fb4494783863 18
samux 1:fb4494783863 19 #include "TCPSocketServer.h"
samux 1:fb4494783863 20 #include <string>
samux 1:fb4494783863 21
samux 1:fb4494783863 22 TCPSocketServer::TCPSocketServer() {}
samux 1:fb4494783863 23
samux 1:fb4494783863 24 // Server initialization
samux 1:fb4494783863 25 int TCPSocketServer::bind(int port) {
samux 1:fb4494783863 26 char cmd[20];
samux 1:fb4494783863 27
samux 1:fb4494783863 28 // set TCP protocol
samux 1:fb4494783863 29 wifi->setProtocol(TCP);
samux 1:fb4494783863 30
samux 1:fb4494783863 31 // set local port
samux 1:fb4494783863 32 sprintf(cmd, "set i l %d\r", port);
samux 1:fb4494783863 33 if (!wifi->sendCommand(cmd, "AOK"))
samux 1:fb4494783863 34 return -1;
samux 3:9aa05e19c62e 35
samux 3:9aa05e19c62e 36 // save
samux 3:9aa05e19c62e 37 if (!wifi->sendCommand("save\r", "Stor"))
samux 3:9aa05e19c62e 38 return -1;
samux 3:9aa05e19c62e 39
samux 3:9aa05e19c62e 40 // reboot
samux 3:9aa05e19c62e 41 wifi->reboot();
samux 3:9aa05e19c62e 42
samux 3:9aa05e19c62e 43 // connect the network
samux 4:0bcec6272784 44 if (wifi->isDHCP()) {
samux 4:0bcec6272784 45 if (!wifi->sendCommand("join\r", "DHCP=ON", NULL, 10000))
samux 4:0bcec6272784 46 return -1;
samux 4:0bcec6272784 47 } else {
samux 4:0bcec6272784 48 if (!wifi->sendCommand("join\r", "Associated", NULL, 10000))
samux 4:0bcec6272784 49 return -1;
samux 4:0bcec6272784 50 }
samux 1:fb4494783863 51
samux 3:9aa05e19c62e 52 // exit
samux 1:fb4494783863 53 wifi->exit();
samux 3:9aa05e19c62e 54
samux 3:9aa05e19c62e 55 wait(0.2);
samux 3:9aa05e19c62e 56 wifi->flush();
samux 1:fb4494783863 57 return 0;
samux 1:fb4494783863 58 }
samux 1:fb4494783863 59
samux 1:fb4494783863 60 int TCPSocketServer::listen(int backlog) {
samux 1:fb4494783863 61 if (backlog != 1)
samux 1:fb4494783863 62 return -1;
samux 1:fb4494783863 63 return 0;
samux 1:fb4494783863 64 }
samux 1:fb4494783863 65
samux 1:fb4494783863 66
samux 1:fb4494783863 67 int TCPSocketServer::accept(TCPSocketConnection& connection) {
samux 1:fb4494783863 68 int nb_available = 0, pos = 0;
samux 1:fb4494783863 69 char c;
samux 1:fb4494783863 70 string str;
samux 1:fb4494783863 71 bool o_find = false;
samux 1:fb4494783863 72 while (1) {
samux 1:fb4494783863 73 while(!wifi->readable());
samux 1:fb4494783863 74 nb_available = wifi->readable();
samux 1:fb4494783863 75 for (int i = 0; i < nb_available; i++) {
samux 1:fb4494783863 76 c = wifi->getc();
samux 1:fb4494783863 77 if (c == '*') {
samux 1:fb4494783863 78 o_find = true;
samux 1:fb4494783863 79 }
samux 1:fb4494783863 80 if (o_find && c != '\r' && c != '\n') {
samux 1:fb4494783863 81 str += c;
samux 1:fb4494783863 82 pos = str.find("*OPEN*");
samux 1:fb4494783863 83 if (pos != string::npos) {
samux 1:fb4494783863 84 wifi->flush();
samux 1:fb4494783863 85 return 0;
samux 1:fb4494783863 86 }
samux 1:fb4494783863 87 }
samux 1:fb4494783863 88 }
samux 1:fb4494783863 89 }
samux 1:fb4494783863 90 }