wifly/socket interface for wifly modules
Dependents: WiFi neurGAI_WIFI thingspeak thingspeak2
Diff: Socket/UDPSocket.cpp
- Revision:
- 0:6ffb0aeb3972
- Child:
- 4:74bfdd00362a
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Socket/UDPSocket.cpp Fri Aug 17 08:28:04 2012 +0000 @@ -0,0 +1,160 @@ +/* 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 "UDPSocket.h" + +#include <string> +#include <algorithm> + +UDPSocket::UDPSocket() { + endpoint_configured = false; + endpoint_read = false; +} + +int UDPSocket::init(void) { + if (!wifi->sendCommand("set ip proto 1\r", "AOK")) + return -1; + wifi->exit(); + return 0; +} + +// Server initialization +int UDPSocket::bind(int port) { + // use udp auto pairing + char cmd[20]; + if (!wifi->sendCommand("set ip proto 1\r", "AOK")) + return -1; + if (!wifi->sendCommand("set ip flags 0x40\r", "AOK")) + return -1; + if (!wifi->sendCommand("set ip host 0.0.0.0\r", "AOK")) + return -1; + if (!wifi->sendCommand("set ip remote 0\r", "AOK")) + return -1; + sprintf(cmd, "set ip local %d\r", port); + if (!wifi->sendCommand(cmd, "AOK")) + return -1; + if (!wifi->sendCommand("save\r", "Stor")) + return -1; + + wifi->reboot(); + return 0; +} + +// -1 if unsuccessful, else number of bytes written +int UDPSocket::sendTo(Endpoint &remote, char *packet, int length) { + Timer tmr; + int idx = 0; + + confEndpoint(remote); + + tmr.start(); + + while ((tmr.read_ms() < _timeout) || _blocking) { + + idx += wifi->send(packet, length); + + if (idx == length) + return idx; + } + return (idx == 0) ? -1 : idx; +} + +// -1 if unsuccessful, else number of bytes received +int UDPSocket::receiveFrom(Endpoint &remote, char *buffer, int length) { + Timer tmr; + int idx = 0; + int nb_available = 0; + + + if (_blocking) { + while (1) { + nb_available = wifi->readable(); + if (nb_available != 0) { + break; + } + } + } + + tmr.start(); + + while (tmr.read_ms() < _timeout) { + + nb_available = wifi->readable(); + for (int i = 0; i < min(nb_available, length); i++) { + buffer[idx++] = wifi->getc(); + } + + if (idx == length) + break; + } + + readEndpoint(remote); + + return (idx == 0) ? -1 : idx; +} + +bool UDPSocket::confEndpoint(Endpoint & ep) { + char * host; + char cmd[30]; + if (!endpoint_configured) { + host = ep.get_address(); + if (host[0] != '\0') { + sprintf(cmd, "set ip host %s\r", host); + if (!wifi->sendCommand(cmd, "AOK")) + return false; + sprintf(cmd, "set ip remote %d\r", ep.get_port()); + if (!wifi->sendCommand(cmd, "AOK")) + return false; + if (!wifi->sendCommand("save\r", "Stor")) + return false; + wifi->reboot(); + endpoint_configured = true; + return true; + } + } + return true; +} + +bool UDPSocket::readEndpoint(Endpoint & ep) { + char recv[256]; + int begin = 0; + int end = 0; + string str; + string addr; + int port; + if (!endpoint_read) { + if (!wifi->sendCommand("get ip\r", NULL, recv)) + return false; + wifi->exit(); + str = recv; + begin = str.find("HOST="); + end = str.find("PROTO="); + if (begin != string::npos && end != string::npos) { + str = str.substr(begin + 5, end - begin - 5); + int pos = str.find(":"); + if (pos != string::npos) { + addr = str.substr(0, pos); + port = atoi(str.substr(pos + 1).c_str()); + ep.set_address(addr.c_str(), port); + endpoint_read = true; + return true; + } + } + } + return false; +}