A fork of the original interface for OS/2. Features a correctly-implemented recv (but retains the old behavior via recv2).
Dependents: weather_clock weather_clock
Socket/UDPSocket.cpp@4:0bcec6272784, 2012-12-20 (annotated)
- Committer:
- samux
- Date:
- Thu Dec 20 15:08:58 2012 +0000
- Revision:
- 4:0bcec6272784
- Parent:
- 1:fb4494783863
- Child:
- 12:c5f0eac67a8a
proper join after reboot - increase size flush
Who changed what in which revision?
User | Revision | Line number | New 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 "UDPSocket.h" |
samux | 1:fb4494783863 | 20 | |
samux | 1:fb4494783863 | 21 | #include <string> |
samux | 1:fb4494783863 | 22 | #include <algorithm> |
samux | 1:fb4494783863 | 23 | |
samux | 1:fb4494783863 | 24 | UDPSocket::UDPSocket() |
samux | 1:fb4494783863 | 25 | { |
samux | 1:fb4494783863 | 26 | endpoint_configured = false; |
samux | 1:fb4494783863 | 27 | endpoint_read = false; |
samux | 1:fb4494783863 | 28 | } |
samux | 1:fb4494783863 | 29 | |
samux | 1:fb4494783863 | 30 | int UDPSocket::init(void) |
samux | 1:fb4494783863 | 31 | { |
samux | 1:fb4494783863 | 32 | wifi->setProtocol(UDP); |
samux | 1:fb4494783863 | 33 | wifi->exit(); |
samux | 1:fb4494783863 | 34 | return 0; |
samux | 1:fb4494783863 | 35 | } |
samux | 1:fb4494783863 | 36 | |
samux | 1:fb4494783863 | 37 | // Server initialization |
samux | 1:fb4494783863 | 38 | int UDPSocket::bind(int port) |
samux | 1:fb4494783863 | 39 | { |
samux | 1:fb4494783863 | 40 | char cmd[17]; |
samux | 1:fb4494783863 | 41 | |
samux | 1:fb4494783863 | 42 | // set local port |
samux | 1:fb4494783863 | 43 | sprintf(cmd, "set i l %d\r", port); |
samux | 1:fb4494783863 | 44 | if (!wifi->sendCommand(cmd, "AOK")) |
samux | 1:fb4494783863 | 45 | return -1; |
samux | 4:0bcec6272784 | 46 | |
samux | 4:0bcec6272784 | 47 | // save |
samux | 4:0bcec6272784 | 48 | if (!wifi->sendCommand("save\r", "Stor")) |
samux | 4:0bcec6272784 | 49 | return -1; |
samux | 4:0bcec6272784 | 50 | |
samux | 4:0bcec6272784 | 51 | // reboot |
samux | 4:0bcec6272784 | 52 | wifi->reboot(); |
samux | 4:0bcec6272784 | 53 | |
samux | 4:0bcec6272784 | 54 | // set udp protocol |
samux | 4:0bcec6272784 | 55 | wifi->setProtocol(UDP); |
samux | 4:0bcec6272784 | 56 | |
samux | 4:0bcec6272784 | 57 | // connect the network |
samux | 4:0bcec6272784 | 58 | if (wifi->isDHCP()) { |
samux | 4:0bcec6272784 | 59 | if (!wifi->sendCommand("join\r", "DHCP=ON", NULL, 10000)) |
samux | 4:0bcec6272784 | 60 | return -1; |
samux | 4:0bcec6272784 | 61 | } else { |
samux | 4:0bcec6272784 | 62 | if (!wifi->sendCommand("join\r", "Associated", NULL, 10000)) |
samux | 4:0bcec6272784 | 63 | return -1; |
samux | 4:0bcec6272784 | 64 | } |
samux | 4:0bcec6272784 | 65 | |
samux | 4:0bcec6272784 | 66 | // exit |
samux | 1:fb4494783863 | 67 | wifi->exit(); |
samux | 4:0bcec6272784 | 68 | wifi->flush(); |
samux | 1:fb4494783863 | 69 | return 0; |
samux | 1:fb4494783863 | 70 | } |
samux | 1:fb4494783863 | 71 | |
samux | 1:fb4494783863 | 72 | // -1 if unsuccessful, else number of bytes written |
samux | 1:fb4494783863 | 73 | int UDPSocket::sendTo(Endpoint &remote, char *packet, int length) |
samux | 1:fb4494783863 | 74 | { |
samux | 1:fb4494783863 | 75 | Timer tmr; |
samux | 1:fb4494783863 | 76 | int idx = 0; |
samux | 1:fb4494783863 | 77 | |
samux | 1:fb4494783863 | 78 | confEndpoint(remote); |
samux | 1:fb4494783863 | 79 | |
samux | 1:fb4494783863 | 80 | tmr.start(); |
samux | 1:fb4494783863 | 81 | |
samux | 1:fb4494783863 | 82 | while ((tmr.read_ms() < _timeout) || _blocking) { |
samux | 1:fb4494783863 | 83 | |
samux | 1:fb4494783863 | 84 | idx += wifi->send(packet, length); |
samux | 1:fb4494783863 | 85 | |
samux | 1:fb4494783863 | 86 | if (idx == length) |
samux | 1:fb4494783863 | 87 | return idx; |
samux | 1:fb4494783863 | 88 | } |
samux | 1:fb4494783863 | 89 | return (idx == 0) ? -1 : idx; |
samux | 1:fb4494783863 | 90 | } |
samux | 1:fb4494783863 | 91 | |
samux | 1:fb4494783863 | 92 | // -1 if unsuccessful, else number of bytes received |
samux | 1:fb4494783863 | 93 | int UDPSocket::receiveFrom(Endpoint &remote, char *buffer, int length) |
samux | 1:fb4494783863 | 94 | { |
samux | 1:fb4494783863 | 95 | Timer tmr; |
samux | 1:fb4494783863 | 96 | int idx = 0; |
samux | 1:fb4494783863 | 97 | int nb_available = 0; |
samux | 1:fb4494783863 | 98 | int time = -1; |
samux | 1:fb4494783863 | 99 | |
samux | 1:fb4494783863 | 100 | if (_blocking) { |
samux | 1:fb4494783863 | 101 | while (1) { |
samux | 1:fb4494783863 | 102 | nb_available = wifi->readable(); |
samux | 1:fb4494783863 | 103 | if (nb_available != 0) { |
samux | 1:fb4494783863 | 104 | break; |
samux | 1:fb4494783863 | 105 | } |
samux | 1:fb4494783863 | 106 | } |
samux | 1:fb4494783863 | 107 | } |
samux | 1:fb4494783863 | 108 | |
samux | 1:fb4494783863 | 109 | tmr.start(); |
samux | 1:fb4494783863 | 110 | |
samux | 1:fb4494783863 | 111 | while (time < _timeout) { |
samux | 1:fb4494783863 | 112 | |
samux | 1:fb4494783863 | 113 | nb_available = wifi->readable(); |
samux | 1:fb4494783863 | 114 | for (int i = 0; i < min(nb_available, length); i++) { |
samux | 1:fb4494783863 | 115 | buffer[idx] = wifi->getc(); |
samux | 1:fb4494783863 | 116 | idx++; |
samux | 1:fb4494783863 | 117 | } |
samux | 1:fb4494783863 | 118 | |
samux | 1:fb4494783863 | 119 | if (idx == length) { |
samux | 1:fb4494783863 | 120 | break; |
samux | 1:fb4494783863 | 121 | } |
samux | 1:fb4494783863 | 122 | |
samux | 1:fb4494783863 | 123 | time = tmr.read_ms(); |
samux | 1:fb4494783863 | 124 | } |
samux | 1:fb4494783863 | 125 | |
samux | 1:fb4494783863 | 126 | readEndpoint(remote); |
samux | 1:fb4494783863 | 127 | return (idx == 0) ? -1 : idx; |
samux | 1:fb4494783863 | 128 | } |
samux | 1:fb4494783863 | 129 | |
samux | 1:fb4494783863 | 130 | bool UDPSocket::confEndpoint(Endpoint & ep) |
samux | 1:fb4494783863 | 131 | { |
samux | 1:fb4494783863 | 132 | char * host; |
samux | 1:fb4494783863 | 133 | char cmd[30]; |
samux | 1:fb4494783863 | 134 | if (!endpoint_configured) { |
samux | 1:fb4494783863 | 135 | host = ep.get_address(); |
samux | 1:fb4494783863 | 136 | if (host[0] != '\0') { |
samux | 1:fb4494783863 | 137 | // set host |
samux | 1:fb4494783863 | 138 | sprintf(cmd, "set i h %s\r", host); |
samux | 1:fb4494783863 | 139 | if (!wifi->sendCommand(cmd, "AOK")) |
samux | 1:fb4494783863 | 140 | return false; |
samux | 1:fb4494783863 | 141 | |
samux | 1:fb4494783863 | 142 | // set remote port |
samux | 1:fb4494783863 | 143 | sprintf(cmd, "set i r %d\r", ep.get_port()); |
samux | 1:fb4494783863 | 144 | if (!wifi->sendCommand(cmd, "AOK")) |
samux | 1:fb4494783863 | 145 | return false; |
samux | 1:fb4494783863 | 146 | |
samux | 1:fb4494783863 | 147 | wifi->exit(); |
samux | 1:fb4494783863 | 148 | endpoint_configured = true; |
samux | 1:fb4494783863 | 149 | return true; |
samux | 1:fb4494783863 | 150 | } |
samux | 1:fb4494783863 | 151 | } |
samux | 1:fb4494783863 | 152 | return true; |
samux | 1:fb4494783863 | 153 | } |
samux | 1:fb4494783863 | 154 | |
samux | 1:fb4494783863 | 155 | bool UDPSocket::readEndpoint(Endpoint & ep) |
samux | 1:fb4494783863 | 156 | { |
samux | 1:fb4494783863 | 157 | char recv[256]; |
samux | 1:fb4494783863 | 158 | int begin = 0; |
samux | 1:fb4494783863 | 159 | int end = 0; |
samux | 1:fb4494783863 | 160 | string str; |
samux | 1:fb4494783863 | 161 | string addr; |
samux | 1:fb4494783863 | 162 | int port; |
samux | 1:fb4494783863 | 163 | if (!endpoint_read) { |
samux | 1:fb4494783863 | 164 | if (!wifi->sendCommand("get ip\r", NULL, recv)) |
samux | 1:fb4494783863 | 165 | return false; |
samux | 1:fb4494783863 | 166 | wifi->exit(); |
samux | 1:fb4494783863 | 167 | str = recv; |
samux | 1:fb4494783863 | 168 | begin = str.find("HOST="); |
samux | 1:fb4494783863 | 169 | end = str.find("PROTO="); |
samux | 1:fb4494783863 | 170 | if (begin != string::npos && end != string::npos) { |
samux | 1:fb4494783863 | 171 | str = str.substr(begin + 5, end - begin - 5); |
samux | 1:fb4494783863 | 172 | int pos = str.find(":"); |
samux | 1:fb4494783863 | 173 | if (pos != string::npos) { |
samux | 1:fb4494783863 | 174 | addr = str.substr(0, pos); |
samux | 1:fb4494783863 | 175 | port = atoi(str.substr(pos + 1).c_str()); |
samux | 1:fb4494783863 | 176 | ep.set_address(addr.c_str(), port); |
samux | 1:fb4494783863 | 177 | endpoint_read = true; |
samux | 1:fb4494783863 | 178 | wifi->flush(); |
samux | 1:fb4494783863 | 179 | return true; |
samux | 1:fb4494783863 | 180 | } |
samux | 1:fb4494783863 | 181 | } |
samux | 1:fb4494783863 | 182 | wifi->flush(); |
samux | 1:fb4494783863 | 183 | } |
samux | 1:fb4494783863 | 184 | return false; |
samux | 1:fb4494783863 | 185 | } |