Bender Robotics / Mbed 2 deprecated PLAUCI_full

posilani dat

Dependencies:   FatFileSystemCpp mbed PowerControl USBHostLite

WiflyInterface/Socket/UDPSocket.cpp@26:5674b8978551, 2017-05-23 (annotated)

Committer:
PavelKumpan
Date:
Tue May 23 18:42:14 2017 +0000
Revision:
26:5674b8978551
Parent:
7:805c16a071d0 
Recreated communication protocol.

Who changed what in which revision?

UserRevisionLine numberNew contents of line
jkaderka 7:805c16a071d0 1 /* Copyright (C) 2012 mbed.org, MIT License
jkaderka 7:805c16a071d0 2 *
jkaderka 7:805c16a071d0 3 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software
jkaderka 7:805c16a071d0 4 * and associated documentation files (the "Software"), to deal in the Software without restriction,
jkaderka 7:805c16a071d0 5 * including without limitation the rights to use, copy, modify, merge, publish, distribute,
jkaderka 7:805c16a071d0 6 * sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
jkaderka 7:805c16a071d0 7 * furnished to do so, subject to the following conditions:
jkaderka 7:805c16a071d0 8 *
jkaderka 7:805c16a071d0 9 * The above copyright notice and this permission notice shall be included in all copies or
jkaderka 7:805c16a071d0 10 * substantial portions of the Software.
jkaderka 7:805c16a071d0 11 *
jkaderka 7:805c16a071d0 12 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
jkaderka 7:805c16a071d0 13 * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
jkaderka 7:805c16a071d0 14 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
jkaderka 7:805c16a071d0 15 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
jkaderka 7:805c16a071d0 16 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
jkaderka 7:805c16a071d0 17 */
jkaderka 7:805c16a071d0 18
jkaderka 7:805c16a071d0 19 #include "UDPSocket.h"
jkaderka 7:805c16a071d0 20
jkaderka 7:805c16a071d0 21 #include <string>
jkaderka 7:805c16a071d0 22 #include <algorithm>
jkaderka 7:805c16a071d0 23
jkaderka 7:805c16a071d0 24 UDPSocket::UDPSocket()
jkaderka 7:805c16a071d0 25 {
jkaderka 7:805c16a071d0 26 endpoint_configured = false;
jkaderka 7:805c16a071d0 27 endpoint_read = false;
jkaderka 7:805c16a071d0 28 }
jkaderka 7:805c16a071d0 29
jkaderka 7:805c16a071d0 30 int UDPSocket::init(void)
jkaderka 7:805c16a071d0 31 {
jkaderka 7:805c16a071d0 32 wifi->setProtocol(UDP);
jkaderka 7:805c16a071d0 33 wifi->exit();
jkaderka 7:805c16a071d0 34 return 0;
jkaderka 7:805c16a071d0 35 }
jkaderka 7:805c16a071d0 36
jkaderka 7:805c16a071d0 37 // Server initialization
jkaderka 7:805c16a071d0 38 int UDPSocket::bind(int port)
jkaderka 7:805c16a071d0 39 {
jkaderka 7:805c16a071d0 40 char cmd[17];
jkaderka 7:805c16a071d0 41
jkaderka 7:805c16a071d0 42 // set local port
jkaderka 7:805c16a071d0 43 sprintf(cmd, "set i l %d\r", port);
jkaderka 7:805c16a071d0 44 if (!wifi->sendCommand(cmd, "AOK"))
jkaderka 7:805c16a071d0 45 return -1;
jkaderka 7:805c16a071d0 46
jkaderka 7:805c16a071d0 47 // save
jkaderka 7:805c16a071d0 48 if (!wifi->sendCommand("save\r", "Stor"))
jkaderka 7:805c16a071d0 49 return -1;
jkaderka 7:805c16a071d0 50
jkaderka 7:805c16a071d0 51 // reboot
jkaderka 7:805c16a071d0 52 wifi->reboot();
jkaderka 7:805c16a071d0 53
jkaderka 7:805c16a071d0 54 // set udp protocol
jkaderka 7:805c16a071d0 55 wifi->setProtocol(UDP);
jkaderka 7:805c16a071d0 56
jkaderka 7:805c16a071d0 57 // connect the network
jkaderka 7:805c16a071d0 58 if (wifi->isDHCP()) {
jkaderka 7:805c16a071d0 59 if (!wifi->sendCommand("join\r", "DHCP=ON", NULL, 10000))
jkaderka 7:805c16a071d0 60 return -1;
jkaderka 7:805c16a071d0 61 } else {
jkaderka 7:805c16a071d0 62 if (!wifi->sendCommand("join\r", "Associated", NULL, 10000))
jkaderka 7:805c16a071d0 63 return -1;
jkaderka 7:805c16a071d0 64 }
jkaderka 7:805c16a071d0 65
jkaderka 7:805c16a071d0 66 // exit
jkaderka 7:805c16a071d0 67 wifi->exit();
jkaderka 7:805c16a071d0 68 wifi->flush();
jkaderka 7:805c16a071d0 69 return 0;
jkaderka 7:805c16a071d0 70 }
jkaderka 7:805c16a071d0 71
jkaderka 7:805c16a071d0 72 // -1 if unsuccessful, else number of bytes written
jkaderka 7:805c16a071d0 73 int UDPSocket::sendTo(Endpoint &remote, char *packet, int length)
jkaderka 7:805c16a071d0 74 {
jkaderka 7:805c16a071d0 75 Timer tmr;
jkaderka 7:805c16a071d0 76 int idx = 0;
jkaderka 7:805c16a071d0 77
jkaderka 7:805c16a071d0 78 confEndpoint(remote);
jkaderka 7:805c16a071d0 79
jkaderka 7:805c16a071d0 80 tmr.start();
jkaderka 7:805c16a071d0 81
jkaderka 7:805c16a071d0 82 while ((tmr.read_ms() < _timeout) || _blocking) {
jkaderka 7:805c16a071d0 83
jkaderka 7:805c16a071d0 84 idx += wifi->send(packet, length);
jkaderka 7:805c16a071d0 85
jkaderka 7:805c16a071d0 86 if (idx == length)
jkaderka 7:805c16a071d0 87 return idx;
jkaderka 7:805c16a071d0 88 }
jkaderka 7:805c16a071d0 89 return (idx == 0) ? -1 : idx;
jkaderka 7:805c16a071d0 90 }
jkaderka 7:805c16a071d0 91
jkaderka 7:805c16a071d0 92 // -1 if unsuccessful, else number of bytes received
jkaderka 7:805c16a071d0 93 int UDPSocket::receiveFrom(Endpoint &remote, char *buffer, int length)
jkaderka 7:805c16a071d0 94 {
jkaderka 7:805c16a071d0 95 Timer tmr;
jkaderka 7:805c16a071d0 96 int idx = 0;
jkaderka 7:805c16a071d0 97 int nb_available = 0;
jkaderka 7:805c16a071d0 98 int time = -1;
jkaderka 7:805c16a071d0 99
jkaderka 7:805c16a071d0 100 if (_blocking) {
jkaderka 7:805c16a071d0 101 while (1) {
jkaderka 7:805c16a071d0 102 nb_available = wifi->readable();
jkaderka 7:805c16a071d0 103 if (nb_available != 0) {
jkaderka 7:805c16a071d0 104 break;
jkaderka 7:805c16a071d0 105 }
jkaderka 7:805c16a071d0 106 }
jkaderka 7:805c16a071d0 107 }
jkaderka 7:805c16a071d0 108
jkaderka 7:805c16a071d0 109 tmr.start();
jkaderka 7:805c16a071d0 110
jkaderka 7:805c16a071d0 111 while (time < _timeout) {
jkaderka 7:805c16a071d0 112
jkaderka 7:805c16a071d0 113 nb_available = wifi->readable();
jkaderka 7:805c16a071d0 114 for (int i = 0; i < min(nb_available, length); i++) {
jkaderka 7:805c16a071d0 115 buffer[idx] = wifi->getc();
jkaderka 7:805c16a071d0 116 idx++;
jkaderka 7:805c16a071d0 117 }
jkaderka 7:805c16a071d0 118
jkaderka 7:805c16a071d0 119 if (idx == length) {
jkaderka 7:805c16a071d0 120 break;
jkaderka 7:805c16a071d0 121 }
jkaderka 7:805c16a071d0 122
jkaderka 7:805c16a071d0 123 time = tmr.read_ms();
jkaderka 7:805c16a071d0 124 }
jkaderka 7:805c16a071d0 125
jkaderka 7:805c16a071d0 126 readEndpoint(remote);
jkaderka 7:805c16a071d0 127 return (idx == 0) ? -1 : idx;
jkaderka 7:805c16a071d0 128 }
jkaderka 7:805c16a071d0 129
jkaderka 7:805c16a071d0 130 bool UDPSocket::confEndpoint(Endpoint & ep)
jkaderka 7:805c16a071d0 131 {
jkaderka 7:805c16a071d0 132 char * host;
jkaderka 7:805c16a071d0 133 char cmd[30];
jkaderka 7:805c16a071d0 134 if (!endpoint_configured) {
jkaderka 7:805c16a071d0 135 host = ep.get_address();
jkaderka 7:805c16a071d0 136 if (host[0] != '\0') {
jkaderka 7:805c16a071d0 137 // set host
jkaderka 7:805c16a071d0 138 sprintf(cmd, "set i h %s\r", host);
jkaderka 7:805c16a071d0 139 if (!wifi->sendCommand(cmd, "AOK"))
jkaderka 7:805c16a071d0 140 return false;
jkaderka 7:805c16a071d0 141
jkaderka 7:805c16a071d0 142 // set remote port
jkaderka 7:805c16a071d0 143 sprintf(cmd, "set i r %d\r", ep.get_port());
jkaderka 7:805c16a071d0 144 if (!wifi->sendCommand(cmd, "AOK"))
jkaderka 7:805c16a071d0 145 return false;
jkaderka 7:805c16a071d0 146
jkaderka 7:805c16a071d0 147 wifi->exit();
jkaderka 7:805c16a071d0 148 endpoint_configured = true;
jkaderka 7:805c16a071d0 149 return true;
jkaderka 7:805c16a071d0 150 }
jkaderka 7:805c16a071d0 151 }
jkaderka 7:805c16a071d0 152 return true;
jkaderka 7:805c16a071d0 153 }
jkaderka 7:805c16a071d0 154
jkaderka 7:805c16a071d0 155 bool UDPSocket::readEndpoint(Endpoint & ep)
jkaderka 7:805c16a071d0 156 {
jkaderka 7:805c16a071d0 157 char recv[256];
jkaderka 7:805c16a071d0 158 int begin = 0;
jkaderka 7:805c16a071d0 159 int end = 0;
jkaderka 7:805c16a071d0 160 string str;
jkaderka 7:805c16a071d0 161 string addr;
jkaderka 7:805c16a071d0 162 int port;
jkaderka 7:805c16a071d0 163 if (!endpoint_read) {
jkaderka 7:805c16a071d0 164 if (!wifi->sendCommand("get ip\r", NULL, recv))
jkaderka 7:805c16a071d0 165 return false;
jkaderka 7:805c16a071d0 166 wifi->exit();
jkaderka 7:805c16a071d0 167 str = recv;
jkaderka 7:805c16a071d0 168 begin = str.find("HOST=");
jkaderka 7:805c16a071d0 169 end = str.find("PROTO=");
jkaderka 7:805c16a071d0 170 if (begin != string::npos && end != string::npos) {
jkaderka 7:805c16a071d0 171 str = str.substr(begin + 5, end - begin - 5);
jkaderka 7:805c16a071d0 172 int pos = str.find(":");
jkaderka 7:805c16a071d0 173 if (pos != string::npos) {
jkaderka 7:805c16a071d0 174 addr = str.substr(0, pos);
jkaderka 7:805c16a071d0 175 port = atoi(str.substr(pos + 1).c_str());
jkaderka 7:805c16a071d0 176 ep.set_address(addr.c_str(), port);
jkaderka 7:805c16a071d0 177 endpoint_read = true;
jkaderka 7:805c16a071d0 178 wifi->flush();
jkaderka 7:805c16a071d0 179 return true;
jkaderka 7:805c16a071d0 180 }
jkaderka 7:805c16a071d0 181 }
jkaderka 7:805c16a071d0 182 wifi->flush();
jkaderka 7:805c16a071d0 183 }
jkaderka 7:805c16a071d0 184 return false;
jkaderka 7:805c16a071d0 185 }