123
UdpSocket.cpp@15:53715cc81c63, 2019-09-07 (annotated)
- Committer:
- hudakz
- Date:
- Sat Sep 07 17:42:42 2019 +0000
- Revision:
- 15:53715cc81c63
- Parent:
- 14:7648334eb41b
Timeout parameter added for the 'connect' function, SPI speed reduced from 20 to 10 Mb/s, debug messages fixed ...
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
hudakz | 9:a156d3de5647 | 1 | /* |
hudakz | 9:a156d3de5647 | 2 | UIPUdp.cpp - Arduino implementation of a UIP wrapper class. |
hudakz | 9:a156d3de5647 | 3 | Copyright (c) 2013 Norbert Truchsess <norbert.truchsess@t-online.de> |
hudakz | 9:a156d3de5647 | 4 | All rights reserved. |
hudakz | 9:a156d3de5647 | 5 | |
hudakz | 9:a156d3de5647 | 6 | This program is free software: you can redistribute it and/or modify |
hudakz | 9:a156d3de5647 | 7 | it under the terms of the GNU General Public License as published by |
hudakz | 9:a156d3de5647 | 8 | the Free Software Foundation, either version 3 of the License, or |
hudakz | 9:a156d3de5647 | 9 | (at your option) any later version. |
hudakz | 9:a156d3de5647 | 10 | |
hudakz | 9:a156d3de5647 | 11 | This program is distributed in the hope that it will be useful, |
hudakz | 9:a156d3de5647 | 12 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
hudakz | 9:a156d3de5647 | 13 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
hudakz | 9:a156d3de5647 | 14 | GNU General Public License for more details. |
hudakz | 9:a156d3de5647 | 15 | |
hudakz | 9:a156d3de5647 | 16 | You should have received a copy of the GNU General Public License |
hudakz | 9:a156d3de5647 | 17 | along with this program. If not, see <http://www.gnu.org/licenses/>. |
hudakz | 9:a156d3de5647 | 18 | */ |
hudakz | 9:a156d3de5647 | 19 | #include "UipEthernet.h" |
hudakz | 9:a156d3de5647 | 20 | #include "UdpSocket.h" |
hudakz | 9:a156d3de5647 | 21 | #include "DnsClient.h" |
hudakz | 9:a156d3de5647 | 22 | |
hudakz | 9:a156d3de5647 | 23 | extern "C" |
hudakz | 9:a156d3de5647 | 24 | { |
hudakz | 9:a156d3de5647 | 25 | #include "utility/uip-conf.h" |
hudakz | 9:a156d3de5647 | 26 | #include "utility/uip.h" |
hudakz | 9:a156d3de5647 | 27 | #include "utility/uip_arp.h" |
hudakz | 9:a156d3de5647 | 28 | } |
hudakz | 9:a156d3de5647 | 29 | #if UIP_UDP |
hudakz | 9:a156d3de5647 | 30 | #define UIP_ARPHDRSIZE 42 |
hudakz | 9:a156d3de5647 | 31 | #define UDPBUF ((struct uip_udpip_hdr*) &uip_buf[UIP_LLH_LEN]) |
hudakz | 9:a156d3de5647 | 32 | |
hudakz | 14:7648334eb41b | 33 | /** |
hudakz | 14:7648334eb41b | 34 | * @brief |
hudakz | 14:7648334eb41b | 35 | * @note |
hudakz | 14:7648334eb41b | 36 | * @param |
hudakz | 14:7648334eb41b | 37 | * @retval |
hudakz | 14:7648334eb41b | 38 | */ |
hudakz | 9:a156d3de5647 | 39 | UdpSocket::UdpSocket() : |
hudakz | 14:7648334eb41b | 40 | _uip_udp_conn(NULL), |
hudakz | 14:7648334eb41b | 41 | _timeout_ms(1000) |
hudakz | 9:a156d3de5647 | 42 | { |
hudakz | 9:a156d3de5647 | 43 | memset(&appdata, 0, sizeof(appdata)); |
hudakz | 9:a156d3de5647 | 44 | } |
hudakz | 9:a156d3de5647 | 45 | |
hudakz | 14:7648334eb41b | 46 | /** |
hudakz | 14:7648334eb41b | 47 | * @brief |
hudakz | 14:7648334eb41b | 48 | * @note |
hudakz | 14:7648334eb41b | 49 | * @param |
hudakz | 14:7648334eb41b | 50 | * @retval |
hudakz | 14:7648334eb41b | 51 | */ |
hudakz | 14:7648334eb41b | 52 | UdpSocket::UdpSocket(int timeout_ms) : |
hudakz | 14:7648334eb41b | 53 | _uip_udp_conn(NULL), |
hudakz | 14:7648334eb41b | 54 | _timeout_ms(timeout_ms) |
hudakz | 14:7648334eb41b | 55 | { |
hudakz | 14:7648334eb41b | 56 | memset(&appdata, 0, sizeof(appdata)); |
hudakz | 14:7648334eb41b | 57 | } |
hudakz | 14:7648334eb41b | 58 | |
hudakz | 14:7648334eb41b | 59 | /** |
hudakz | 14:7648334eb41b | 60 | * @brief |
hudakz | 14:7648334eb41b | 61 | * @note |
hudakz | 14:7648334eb41b | 62 | * @param |
hudakz | 14:7648334eb41b | 63 | * @retval |
hudakz | 14:7648334eb41b | 64 | */ |
hudakz | 14:7648334eb41b | 65 | UdpSocket::UdpSocket(UipEthernet* net, int timeout_ms /*= 1000*/ ) : |
hudakz | 14:7648334eb41b | 66 | _uip_udp_conn(NULL), |
hudakz | 14:7648334eb41b | 67 | _timeout_ms(timeout_ms) |
hudakz | 14:7648334eb41b | 68 | { |
hudakz | 14:7648334eb41b | 69 | memset(&appdata, 0, sizeof(appdata)); |
hudakz | 14:7648334eb41b | 70 | if (UipEthernet::ethernet != net) |
hudakz | 14:7648334eb41b | 71 | UipEthernet::ethernet = net; |
hudakz | 14:7648334eb41b | 72 | } |
hudakz | 14:7648334eb41b | 73 | |
hudakz | 9:a156d3de5647 | 74 | // initialize, start listening on specified port. Returns 1 if successful, 0 if there are no sockets available to use |
hudakz | 9:a156d3de5647 | 75 | uint8_t UdpSocket::begin(uint16_t port) |
hudakz | 9:a156d3de5647 | 76 | { |
hudakz | 9:a156d3de5647 | 77 | if (!_uip_udp_conn) { |
hudakz | 9:a156d3de5647 | 78 | _uip_udp_conn = uip_udp_new(NULL, 0); |
hudakz | 9:a156d3de5647 | 79 | } |
hudakz | 9:a156d3de5647 | 80 | |
hudakz | 9:a156d3de5647 | 81 | if (_uip_udp_conn) { |
hudakz | 9:a156d3de5647 | 82 | uip_udp_bind(_uip_udp_conn, htons(port)); |
hudakz | 9:a156d3de5647 | 83 | _uip_udp_conn->appstate = &appdata; |
hudakz | 9:a156d3de5647 | 84 | return 1; |
hudakz | 9:a156d3de5647 | 85 | } |
hudakz | 9:a156d3de5647 | 86 | |
hudakz | 9:a156d3de5647 | 87 | return 0; |
hudakz | 9:a156d3de5647 | 88 | } |
hudakz | 9:a156d3de5647 | 89 | |
hudakz | 9:a156d3de5647 | 90 | // Finish with the UDP socket |
hudakz | 9:a156d3de5647 | 91 | void UdpSocket::stop() |
hudakz | 9:a156d3de5647 | 92 | { |
hudakz | 9:a156d3de5647 | 93 | if (_uip_udp_conn) { |
hudakz | 9:a156d3de5647 | 94 | uip_udp_remove(_uip_udp_conn); |
hudakz | 9:a156d3de5647 | 95 | _uip_udp_conn->appstate = NULL; |
hudakz | 9:a156d3de5647 | 96 | _uip_udp_conn = NULL; |
hudakz | 11:647d53d146f1 | 97 | UipEthernet::ethernet->enc28j60Eth.freeBlock(appdata.packet_in); |
hudakz | 11:647d53d146f1 | 98 | UipEthernet::ethernet->enc28j60Eth.freeBlock(appdata.packet_next); |
hudakz | 11:647d53d146f1 | 99 | UipEthernet::ethernet->enc28j60Eth.freeBlock(appdata.packet_out); |
hudakz | 9:a156d3de5647 | 100 | memset(&appdata, 0, sizeof(appdata)); |
hudakz | 9:a156d3de5647 | 101 | } |
hudakz | 9:a156d3de5647 | 102 | } |
hudakz | 9:a156d3de5647 | 103 | |
hudakz | 14:7648334eb41b | 104 | /** |
hudakz | 14:7648334eb41b | 105 | * @brief |
hudakz | 14:7648334eb41b | 106 | * @note |
hudakz | 14:7648334eb41b | 107 | * @param |
hudakz | 14:7648334eb41b | 108 | * @retval |
hudakz | 14:7648334eb41b | 109 | */ |
hudakz | 14:7648334eb41b | 110 | void UdpSocket::close() |
hudakz | 14:7648334eb41b | 111 | { |
hudakz | 14:7648334eb41b | 112 | stop(); |
hudakz | 14:7648334eb41b | 113 | } |
hudakz | 14:7648334eb41b | 114 | |
hudakz | 9:a156d3de5647 | 115 | // Sending UDP packets |
hudakz | 9:a156d3de5647 | 116 | // Start building up a packet to send to the remote host specific in ip and port |
hudakz | 9:a156d3de5647 | 117 | |
hudakz | 9:a156d3de5647 | 118 | // Returns 1 if successful, 0 if there was a problem with the supplied IP address or port |
hudakz | 9:a156d3de5647 | 119 | int UdpSocket::beginPacket(IpAddress ip, uint16_t port) |
hudakz | 9:a156d3de5647 | 120 | { |
hudakz | 9:a156d3de5647 | 121 | UipEthernet::ethernet->tick(); |
hudakz | 9:a156d3de5647 | 122 | if (ip && port) { |
hudakz | 9:a156d3de5647 | 123 | uip_ipaddr_t ripaddr; |
hudakz | 9:a156d3de5647 | 124 | uip_ip_addr(&ripaddr, ip); |
hudakz | 9:a156d3de5647 | 125 | #ifdef UIPETHERNET_DEBUG_UDP |
hudakz | 9:a156d3de5647 | 126 | printf("udp beginPacket, "); |
hudakz | 9:a156d3de5647 | 127 | #endif |
hudakz | 9:a156d3de5647 | 128 | if (_uip_udp_conn) { |
hudakz | 9:a156d3de5647 | 129 | _uip_udp_conn->rport = htons(port); |
hudakz | 9:a156d3de5647 | 130 | uip_ipaddr_copy(_uip_udp_conn->ripaddr, &ripaddr); |
hudakz | 9:a156d3de5647 | 131 | } |
hudakz | 9:a156d3de5647 | 132 | else { |
hudakz | 9:a156d3de5647 | 133 | _uip_udp_conn = uip_udp_new(&ripaddr, htons(port)); |
hudakz | 9:a156d3de5647 | 134 | if (_uip_udp_conn) |
hudakz | 9:a156d3de5647 | 135 | { |
hudakz | 9:a156d3de5647 | 136 | #ifdef UIPETHERNET_DEBUG_UDP |
hudakz | 9:a156d3de5647 | 137 | printf("new connection, "); |
hudakz | 9:a156d3de5647 | 138 | #endif |
hudakz | 9:a156d3de5647 | 139 | _uip_udp_conn->appstate = &appdata; |
hudakz | 9:a156d3de5647 | 140 | } |
hudakz | 9:a156d3de5647 | 141 | else |
hudakz | 9:a156d3de5647 | 142 | { |
hudakz | 9:a156d3de5647 | 143 | #ifdef UIPETHERNET_DEBUG_UDP |
hudakz | 9:a156d3de5647 | 144 | printf("failed to allocate new connection\r\n"); |
hudakz | 9:a156d3de5647 | 145 | #endif |
hudakz | 9:a156d3de5647 | 146 | return 0; |
hudakz | 9:a156d3de5647 | 147 | } |
hudakz | 9:a156d3de5647 | 148 | } |
hudakz | 9:a156d3de5647 | 149 | |
hudakz | 9:a156d3de5647 | 150 | #ifdef UIPETHERNET_DEBUG_UDP |
hudakz | 15:53715cc81c63 | 151 | char buf[16]; |
hudakz | 15:53715cc81c63 | 152 | printf("rip: %s, port: %d\r\n", ip.toString(buf), port); |
hudakz | 9:a156d3de5647 | 153 | #endif |
hudakz | 9:a156d3de5647 | 154 | } |
hudakz | 9:a156d3de5647 | 155 | |
hudakz | 9:a156d3de5647 | 156 | if (_uip_udp_conn) { |
hudakz | 9:a156d3de5647 | 157 | if (appdata.packet_out == NOBLOCK) { |
hudakz | 11:647d53d146f1 | 158 | appdata.packet_out = UipEthernet::ethernet->enc28j60Eth.allocBlock(UIP_UDP_MAXPACKETSIZE); |
hudakz | 9:a156d3de5647 | 159 | appdata.out_pos = UIP_UDP_PHYH_LEN; |
hudakz | 15:53715cc81c63 | 160 | if (appdata.packet_out != NOBLOCK) { |
hudakz | 9:a156d3de5647 | 161 | return 1; |
hudakz | 15:53715cc81c63 | 162 | } |
hudakz | 9:a156d3de5647 | 163 | #ifdef UIPETHERNET_DEBUG_UDP |
hudakz | 15:53715cc81c63 | 164 | else { |
hudakz | 9:a156d3de5647 | 165 | printf("failed to allocate memory for packet\r\n"); |
hudakz | 15:53715cc81c63 | 166 | } |
hudakz | 9:a156d3de5647 | 167 | #endif |
hudakz | 9:a156d3de5647 | 168 | } |
hudakz | 9:a156d3de5647 | 169 | |
hudakz | 9:a156d3de5647 | 170 | #ifdef UIPETHERNET_DEBUG_UDP |
hudakz | 15:53715cc81c63 | 171 | else { |
hudakz | 9:a156d3de5647 | 172 | printf("previous packet on that connection not sent yet\r\n"); |
hudakz | 15:53715cc81c63 | 173 | } |
hudakz | 9:a156d3de5647 | 174 | #endif |
hudakz | 9:a156d3de5647 | 175 | } |
hudakz | 9:a156d3de5647 | 176 | |
hudakz | 9:a156d3de5647 | 177 | return 0; |
hudakz | 9:a156d3de5647 | 178 | } |
hudakz | 9:a156d3de5647 | 179 | |
hudakz | 9:a156d3de5647 | 180 | // Start building up a packet to send to the remote host specific in host and port |
hudakz | 9:a156d3de5647 | 181 | |
hudakz | 9:a156d3de5647 | 182 | // Returns 1 if successful, 0 if there was a problem resolving the hostname or port |
hudakz | 9:a156d3de5647 | 183 | int UdpSocket::beginPacket(const char* host, uint16_t port) |
hudakz | 9:a156d3de5647 | 184 | { |
hudakz | 9:a156d3de5647 | 185 | // Look up the host first |
hudakz | 9:a156d3de5647 | 186 | int ret = 0; |
hudakz | 9:a156d3de5647 | 187 | DnsClient dns; |
hudakz | 9:a156d3de5647 | 188 | IpAddress remote_addr; |
hudakz | 9:a156d3de5647 | 189 | |
hudakz | 9:a156d3de5647 | 190 | dns.begin(UipEthernet::ethernet->dnsServerIP()); |
hudakz | 9:a156d3de5647 | 191 | ret = dns.getHostByName(host, remote_addr); |
hudakz | 9:a156d3de5647 | 192 | if (ret == 1) { |
hudakz | 9:a156d3de5647 | 193 | return beginPacket(remote_addr, port); |
hudakz | 9:a156d3de5647 | 194 | } |
hudakz | 9:a156d3de5647 | 195 | else { |
hudakz | 9:a156d3de5647 | 196 | return ret; |
hudakz | 9:a156d3de5647 | 197 | } |
hudakz | 9:a156d3de5647 | 198 | } |
hudakz | 9:a156d3de5647 | 199 | |
hudakz | 9:a156d3de5647 | 200 | // Finish off this packet and send it |
hudakz | 9:a156d3de5647 | 201 | |
hudakz | 9:a156d3de5647 | 202 | // Returns 1 if the packet was sent successfully, 0 if there was an error |
hudakz | 9:a156d3de5647 | 203 | int UdpSocket::endPacket() |
hudakz | 9:a156d3de5647 | 204 | { |
hudakz | 9:a156d3de5647 | 205 | if (_uip_udp_conn && appdata.packet_out != NOBLOCK) { |
hudakz | 9:a156d3de5647 | 206 | appdata.send = true; |
hudakz | 11:647d53d146f1 | 207 | UipEthernet::ethernet->enc28j60Eth.resizeBlock(appdata.packet_out, 0, appdata.out_pos); |
hudakz | 9:a156d3de5647 | 208 | uip_udp_periodic_conn(_uip_udp_conn); |
hudakz | 9:a156d3de5647 | 209 | if (uip_len > 0) { |
hudakz | 9:a156d3de5647 | 210 | _send(&appdata); |
hudakz | 9:a156d3de5647 | 211 | return 1; |
hudakz | 9:a156d3de5647 | 212 | } |
hudakz | 9:a156d3de5647 | 213 | } |
hudakz | 9:a156d3de5647 | 214 | |
hudakz | 9:a156d3de5647 | 215 | return 0; |
hudakz | 9:a156d3de5647 | 216 | } |
hudakz | 9:a156d3de5647 | 217 | |
hudakz | 9:a156d3de5647 | 218 | // Write a single byte into the packet |
hudakz | 9:a156d3de5647 | 219 | size_t UdpSocket::write(uint8_t c) |
hudakz | 9:a156d3de5647 | 220 | { |
hudakz | 9:a156d3de5647 | 221 | return write(&c, 1); |
hudakz | 9:a156d3de5647 | 222 | } |
hudakz | 9:a156d3de5647 | 223 | |
hudakz | 9:a156d3de5647 | 224 | // Write size bytes from buffer into the packet |
hudakz | 9:a156d3de5647 | 225 | size_t UdpSocket::write(const uint8_t* buffer, size_t size) |
hudakz | 9:a156d3de5647 | 226 | { |
hudakz | 9:a156d3de5647 | 227 | if (appdata.packet_out != NOBLOCK) { |
hudakz | 11:647d53d146f1 | 228 | size_t ret = UipEthernet::ethernet->enc28j60Eth.writePacket |
hudakz | 9:a156d3de5647 | 229 | ( |
hudakz | 9:a156d3de5647 | 230 | appdata.packet_out, |
hudakz | 9:a156d3de5647 | 231 | appdata.out_pos, |
hudakz | 9:a156d3de5647 | 232 | (uint8_t*)buffer, |
hudakz | 9:a156d3de5647 | 233 | size |
hudakz | 9:a156d3de5647 | 234 | ); |
hudakz | 9:a156d3de5647 | 235 | appdata.out_pos += ret; |
hudakz | 9:a156d3de5647 | 236 | return ret; |
hudakz | 9:a156d3de5647 | 237 | } |
hudakz | 9:a156d3de5647 | 238 | |
hudakz | 9:a156d3de5647 | 239 | return 0; |
hudakz | 9:a156d3de5647 | 240 | } |
hudakz | 9:a156d3de5647 | 241 | |
hudakz | 9:a156d3de5647 | 242 | // Start processing the next available incoming packet |
hudakz | 9:a156d3de5647 | 243 | |
hudakz | 9:a156d3de5647 | 244 | // Returns the size of the packet in bytes, or 0 if no packets are available |
hudakz | 9:a156d3de5647 | 245 | int UdpSocket::parsePacket() |
hudakz | 9:a156d3de5647 | 246 | { |
hudakz | 9:a156d3de5647 | 247 | UipEthernet::ethernet->tick(); |
hudakz | 9:a156d3de5647 | 248 | #ifdef UIPETHERNET_DEBUG_UDP |
hudakz | 9:a156d3de5647 | 249 | if (appdata.packet_in != NOBLOCK) { |
hudakz | 9:a156d3de5647 | 250 | printf("udp parsePacket freeing previous packet: %d\r\n", appdata.packet_in); |
hudakz | 9:a156d3de5647 | 251 | } |
hudakz | 9:a156d3de5647 | 252 | #endif |
hudakz | 11:647d53d146f1 | 253 | UipEthernet::ethernet->enc28j60Eth.freeBlock(appdata.packet_in); |
hudakz | 9:a156d3de5647 | 254 | |
hudakz | 9:a156d3de5647 | 255 | appdata.packet_in = appdata.packet_next; |
hudakz | 9:a156d3de5647 | 256 | appdata.packet_next = NOBLOCK; |
hudakz | 9:a156d3de5647 | 257 | |
hudakz | 9:a156d3de5647 | 258 | #ifdef UIPETHERNET_DEBUG_UDP |
hudakz | 9:a156d3de5647 | 259 | if (appdata.packet_in != NOBLOCK) { |
hudakz | 9:a156d3de5647 | 260 | printf("udp parsePacket received packet: %d", appdata.packet_in); |
hudakz | 9:a156d3de5647 | 261 | } |
hudakz | 9:a156d3de5647 | 262 | #endif |
hudakz | 9:a156d3de5647 | 263 | |
hudakz | 11:647d53d146f1 | 264 | int size = UipEthernet::ethernet->enc28j60Eth.blockSize(appdata.packet_in); |
hudakz | 9:a156d3de5647 | 265 | #ifdef UIPETHERNET_DEBUG_UDP |
hudakz | 9:a156d3de5647 | 266 | if (appdata.packet_in != NOBLOCK) { |
hudakz | 9:a156d3de5647 | 267 | printf(", size: %d\r\n", size); |
hudakz | 9:a156d3de5647 | 268 | } |
hudakz | 9:a156d3de5647 | 269 | #endif |
hudakz | 9:a156d3de5647 | 270 | return size; |
hudakz | 9:a156d3de5647 | 271 | } |
hudakz | 9:a156d3de5647 | 272 | |
hudakz | 9:a156d3de5647 | 273 | // Number of bytes remaining in the current packet |
hudakz | 14:7648334eb41b | 274 | size_t UdpSocket::available() |
hudakz | 9:a156d3de5647 | 275 | { |
hudakz | 9:a156d3de5647 | 276 | UipEthernet::ethernet->tick(); |
hudakz | 11:647d53d146f1 | 277 | return UipEthernet::ethernet->enc28j60Eth.blockSize(appdata.packet_in); |
hudakz | 9:a156d3de5647 | 278 | } |
hudakz | 9:a156d3de5647 | 279 | |
hudakz | 14:7648334eb41b | 280 | // Read a single byte from the current packet. Returns -1 if no byte is available. |
hudakz | 9:a156d3de5647 | 281 | int UdpSocket::read() |
hudakz | 9:a156d3de5647 | 282 | { |
hudakz | 9:a156d3de5647 | 283 | static unsigned char c; |
hudakz | 9:a156d3de5647 | 284 | if (read(&c, 1) > 0) { |
hudakz | 9:a156d3de5647 | 285 | return c; |
hudakz | 9:a156d3de5647 | 286 | } |
hudakz | 9:a156d3de5647 | 287 | |
hudakz | 9:a156d3de5647 | 288 | return -1; |
hudakz | 9:a156d3de5647 | 289 | } |
hudakz | 9:a156d3de5647 | 290 | |
hudakz | 9:a156d3de5647 | 291 | // Read up to len bytes from the current packet and place them into buffer |
hudakz | 9:a156d3de5647 | 292 | |
hudakz | 9:a156d3de5647 | 293 | // Returns the number of bytes read, or 0 if none are available |
hudakz | 14:7648334eb41b | 294 | size_t UdpSocket::read(unsigned char* buffer, size_t len) |
hudakz | 9:a156d3de5647 | 295 | { |
hudakz | 9:a156d3de5647 | 296 | UipEthernet::ethernet->tick(); |
hudakz | 9:a156d3de5647 | 297 | if (appdata.packet_in != NOBLOCK) { |
hudakz | 11:647d53d146f1 | 298 | memaddress read = UipEthernet::ethernet->enc28j60Eth.readPacket(appdata.packet_in, 0, buffer, len); |
hudakz | 11:647d53d146f1 | 299 | if (read == UipEthernet::ethernet->enc28j60Eth.blockSize(appdata.packet_in)) { |
hudakz | 11:647d53d146f1 | 300 | UipEthernet::ethernet->enc28j60Eth.freeBlock(appdata.packet_in); |
hudakz | 9:a156d3de5647 | 301 | appdata.packet_in = NOBLOCK; |
hudakz | 9:a156d3de5647 | 302 | } |
hudakz | 9:a156d3de5647 | 303 | else |
hudakz | 11:647d53d146f1 | 304 | UipEthernet::ethernet->enc28j60Eth.resizeBlock(appdata.packet_in, read); |
hudakz | 9:a156d3de5647 | 305 | return read; |
hudakz | 9:a156d3de5647 | 306 | } |
hudakz | 9:a156d3de5647 | 307 | |
hudakz | 9:a156d3de5647 | 308 | return 0; |
hudakz | 9:a156d3de5647 | 309 | } |
hudakz | 9:a156d3de5647 | 310 | |
hudakz | 9:a156d3de5647 | 311 | // Return the next byte from the current packet without moving on to the next byte |
hudakz | 9:a156d3de5647 | 312 | int UdpSocket::peek() |
hudakz | 9:a156d3de5647 | 313 | { |
hudakz | 9:a156d3de5647 | 314 | UipEthernet::ethernet->tick(); |
hudakz | 9:a156d3de5647 | 315 | if (appdata.packet_in != NOBLOCK) { |
hudakz | 9:a156d3de5647 | 316 | unsigned char c; |
hudakz | 11:647d53d146f1 | 317 | if (UipEthernet::ethernet->enc28j60Eth.readPacket(appdata.packet_in, 0, &c, 1) == 1) |
hudakz | 9:a156d3de5647 | 318 | return c; |
hudakz | 9:a156d3de5647 | 319 | } |
hudakz | 9:a156d3de5647 | 320 | |
hudakz | 9:a156d3de5647 | 321 | return -1; |
hudakz | 9:a156d3de5647 | 322 | } |
hudakz | 9:a156d3de5647 | 323 | |
hudakz | 9:a156d3de5647 | 324 | // Finish reading the current packet |
hudakz | 9:a156d3de5647 | 325 | void UdpSocket::flush() |
hudakz | 9:a156d3de5647 | 326 | { |
hudakz | 9:a156d3de5647 | 327 | UipEthernet::ethernet->tick(); |
hudakz | 11:647d53d146f1 | 328 | UipEthernet::ethernet->enc28j60Eth.freeBlock(appdata.packet_in); |
hudakz | 9:a156d3de5647 | 329 | appdata.packet_in = NOBLOCK; |
hudakz | 9:a156d3de5647 | 330 | } |
hudakz | 9:a156d3de5647 | 331 | |
hudakz | 9:a156d3de5647 | 332 | // Return the IP address of the host who sent the current incoming packet |
hudakz | 9:a156d3de5647 | 333 | IpAddress UdpSocket::remoteIP() |
hudakz | 9:a156d3de5647 | 334 | { |
hudakz | 9:a156d3de5647 | 335 | return _uip_udp_conn ? ip_addr_uip(_uip_udp_conn->ripaddr) : IpAddress(); |
hudakz | 9:a156d3de5647 | 336 | } |
hudakz | 9:a156d3de5647 | 337 | |
hudakz | 9:a156d3de5647 | 338 | // Return the port of the host who sent the current incoming packet |
hudakz | 9:a156d3de5647 | 339 | uint16_t UdpSocket::remotePort() |
hudakz | 9:a156d3de5647 | 340 | { |
hudakz | 9:a156d3de5647 | 341 | return _uip_udp_conn ? ntohs(_uip_udp_conn->rport) : 0; |
hudakz | 9:a156d3de5647 | 342 | } |
hudakz | 9:a156d3de5647 | 343 | |
hudakz | 9:a156d3de5647 | 344 | // UIP callback function |
hudakz | 9:a156d3de5647 | 345 | void uipudp_appcall() |
hudakz | 9:a156d3de5647 | 346 | { |
hudakz | 9:a156d3de5647 | 347 | if (uip_udp_userdata_t * data = (uip_udp_userdata_t *) (uip_udp_conn->appstate)) { |
hudakz | 9:a156d3de5647 | 348 | if (uip_newdata()) { |
hudakz | 9:a156d3de5647 | 349 | if (data->packet_next == NOBLOCK) { |
hudakz | 9:a156d3de5647 | 350 | uip_udp_conn->rport = UDPBUF->srcport; |
hudakz | 9:a156d3de5647 | 351 | uip_ipaddr_copy(uip_udp_conn->ripaddr, UDPBUF->srcipaddr); |
hudakz | 11:647d53d146f1 | 352 | data->packet_next = UipEthernet::ethernet->enc28j60Eth.allocBlock(ntohs(UDPBUF->udplen) - UIP_UDPH_LEN); |
hudakz | 9:a156d3de5647 | 353 | |
hudakz | 9:a156d3de5647 | 354 | //if we are unable to allocate memory the packet is dropped. udp doesn't guarantee packet delivery |
hudakz | 9:a156d3de5647 | 355 | if (data->packet_next != NOBLOCK) { |
hudakz | 9:a156d3de5647 | 356 | //discard Linklevel and IP and udp-header and any trailing bytes: |
hudakz | 11:647d53d146f1 | 357 | UipEthernet::ethernet->enc28j60Eth.copyPacket |
hudakz | 9:a156d3de5647 | 358 | ( |
hudakz | 9:a156d3de5647 | 359 | data->packet_next, |
hudakz | 9:a156d3de5647 | 360 | 0, |
hudakz | 9:a156d3de5647 | 361 | UipEthernet::inPacket, |
hudakz | 9:a156d3de5647 | 362 | UIP_UDP_PHYH_LEN, |
hudakz | 11:647d53d146f1 | 363 | UipEthernet::ethernet->enc28j60Eth.blockSize(data->packet_next) |
hudakz | 9:a156d3de5647 | 364 | ); |
hudakz | 9:a156d3de5647 | 365 | #ifdef UIPETHERNET_DEBUG_UDP |
hudakz | 9:a156d3de5647 | 366 | printf |
hudakz | 9:a156d3de5647 | 367 | ( |
hudakz | 9:a156d3de5647 | 368 | "udp, uip_newdata received packet: %d, size: %d\r\n", |
hudakz | 9:a156d3de5647 | 369 | data->packet_next, |
hudakz | 15:53715cc81c63 | 370 | UipEthernet::ethernet->enc28j60Eth.blockSize(data->packet_next) |
hudakz | 9:a156d3de5647 | 371 | ); |
hudakz | 9:a156d3de5647 | 372 | #endif |
hudakz | 9:a156d3de5647 | 373 | } |
hudakz | 9:a156d3de5647 | 374 | } |
hudakz | 9:a156d3de5647 | 375 | } |
hudakz | 9:a156d3de5647 | 376 | |
hudakz | 9:a156d3de5647 | 377 | if (uip_poll() && data->send) |
hudakz | 9:a156d3de5647 | 378 | { |
hudakz | 9:a156d3de5647 | 379 | //set uip_slen (uip private) by calling uip_udp_send |
hudakz | 9:a156d3de5647 | 380 | #ifdef UIPETHERNET_DEBUG_UDP |
hudakz | 9:a156d3de5647 | 381 | printf |
hudakz | 9:a156d3de5647 | 382 | ( |
hudakz | 9:a156d3de5647 | 383 | "udp, uip_poll preparing packet to send: %d, size: %d\r\n", |
hudakz | 9:a156d3de5647 | 384 | data->packet_out, |
hudakz | 15:53715cc81c63 | 385 | UipEthernet::ethernet->enc28j60Eth.blockSize(data->packet_out) |
hudakz | 9:a156d3de5647 | 386 | ); |
hudakz | 9:a156d3de5647 | 387 | #endif |
hudakz | 9:a156d3de5647 | 388 | UipEthernet::uipPacket = data->packet_out; |
hudakz | 9:a156d3de5647 | 389 | UipEthernet::uipHeaderLen = UIP_UDP_PHYH_LEN; |
hudakz | 9:a156d3de5647 | 390 | uip_udp_send(data->out_pos - (UIP_UDP_PHYH_LEN)); |
hudakz | 9:a156d3de5647 | 391 | } |
hudakz | 9:a156d3de5647 | 392 | } |
hudakz | 9:a156d3de5647 | 393 | } |
hudakz | 9:a156d3de5647 | 394 | |
hudakz | 9:a156d3de5647 | 395 | /** |
hudakz | 9:a156d3de5647 | 396 | * @brief |
hudakz | 9:a156d3de5647 | 397 | * @note |
hudakz | 9:a156d3de5647 | 398 | * @param |
hudakz | 9:a156d3de5647 | 399 | * @retval |
hudakz | 9:a156d3de5647 | 400 | */ |
hudakz | 9:a156d3de5647 | 401 | void UdpSocket::_send(uip_udp_userdata_t* data) |
hudakz | 9:a156d3de5647 | 402 | { |
hudakz | 9:a156d3de5647 | 403 | uip_arp_out(); //add arp |
hudakz | 9:a156d3de5647 | 404 | if (uip_len == UIP_ARPHDRSIZE) { |
hudakz | 9:a156d3de5647 | 405 | UipEthernet::uipPacket = NOBLOCK; |
hudakz | 9:a156d3de5647 | 406 | UipEthernet::packetState &= ~UIPETHERNET_SENDPACKET; |
hudakz | 9:a156d3de5647 | 407 | #ifdef UIPETHERNET_DEBUG_UDP |
hudakz | 9:a156d3de5647 | 408 | printf("udp, uip_poll results in ARP-packet\r\n"); |
hudakz | 9:a156d3de5647 | 409 | #endif |
hudakz | 9:a156d3de5647 | 410 | } |
hudakz | 9:a156d3de5647 | 411 | else { |
hudakz | 9:a156d3de5647 | 412 | //arp found ethaddr for ip (otherwise packet is replaced by arp-request) |
hudakz | 9:a156d3de5647 | 413 | data->send = false; |
hudakz | 9:a156d3de5647 | 414 | data->packet_out = NOBLOCK; |
hudakz | 9:a156d3de5647 | 415 | UipEthernet::packetState |= UIPETHERNET_SENDPACKET; |
hudakz | 9:a156d3de5647 | 416 | #ifdef UIPETHERNET_DEBUG_UDP |
hudakz | 15:53715cc81c63 | 417 | printf("udp, uip_packet to send: %d\r\n", UipEthernet::uipPacket); |
hudakz | 9:a156d3de5647 | 418 | #endif |
hudakz | 9:a156d3de5647 | 419 | } |
hudakz | 9:a156d3de5647 | 420 | |
hudakz | 9:a156d3de5647 | 421 | UipEthernet::ethernet->network_send(); |
hudakz | 9:a156d3de5647 | 422 | } |
hudakz | 9:a156d3de5647 | 423 | #endif |
hudakz | 14:7648334eb41b | 424 | |
hudakz | 14:7648334eb41b | 425 | /** |
hudakz | 14:7648334eb41b | 426 | * @brief |
hudakz | 14:7648334eb41b | 427 | * @note |
hudakz | 14:7648334eb41b | 428 | * @param |
hudakz | 14:7648334eb41b | 429 | * @retval |
hudakz | 14:7648334eb41b | 430 | */ |
hudakz | 14:7648334eb41b | 431 | nsapi_size_or_error_t UdpSocket::sendto(const char* host, uint16_t port, const void* data, size_t size) |
hudakz | 14:7648334eb41b | 432 | { |
hudakz | 14:7648334eb41b | 433 | DnsClient dns; |
hudakz | 14:7648334eb41b | 434 | IpAddress address; |
hudakz | 14:7648334eb41b | 435 | uint32_t address_bytes; |
hudakz | 14:7648334eb41b | 436 | |
hudakz | 14:7648334eb41b | 437 | dns.begin(UipEthernet::ethernet->dnsServerIP()); |
hudakz | 14:7648334eb41b | 438 | if (dns.getHostByName(host, address) == 1) { |
hudakz | 14:7648334eb41b | 439 | address_bytes = address; |
hudakz | 14:7648334eb41b | 440 | _remote_addr = SocketAddress(&address_bytes, NSAPI_IPv4, port); |
hudakz | 14:7648334eb41b | 441 | } |
hudakz | 14:7648334eb41b | 442 | else { |
hudakz | 14:7648334eb41b | 443 | _remote_addr = SocketAddress(host, port); |
hudakz | 14:7648334eb41b | 444 | } |
hudakz | 14:7648334eb41b | 445 | |
hudakz | 14:7648334eb41b | 446 | if (beginPacket(host, port) == 0) { |
hudakz | 14:7648334eb41b | 447 | stop(); |
hudakz | 14:7648334eb41b | 448 | return NSAPI_ERROR_NO_ADDRESS; |
hudakz | 14:7648334eb41b | 449 | } |
hudakz | 14:7648334eb41b | 450 | |
hudakz | 14:7648334eb41b | 451 | if (write((uint8_t*)data, size) == 0) { |
hudakz | 14:7648334eb41b | 452 | stop(); |
hudakz | 14:7648334eb41b | 453 | return NSAPI_ERROR_WOULD_BLOCK; |
hudakz | 14:7648334eb41b | 454 | }; |
hudakz | 14:7648334eb41b | 455 | |
hudakz | 14:7648334eb41b | 456 | if (endPacket() <= 0) { |
hudakz | 14:7648334eb41b | 457 | stop(); |
hudakz | 14:7648334eb41b | 458 | return NSAPI_ERROR_WOULD_BLOCK; |
hudakz | 14:7648334eb41b | 459 | } |
hudakz | 14:7648334eb41b | 460 | |
hudakz | 14:7648334eb41b | 461 | return NSAPI_ERROR_OK; |
hudakz | 14:7648334eb41b | 462 | } |
hudakz | 14:7648334eb41b | 463 | |
hudakz | 14:7648334eb41b | 464 | /** |
hudakz | 14:7648334eb41b | 465 | * @brief |
hudakz | 14:7648334eb41b | 466 | * @note |
hudakz | 14:7648334eb41b | 467 | * @param |
hudakz | 14:7648334eb41b | 468 | * @retval |
hudakz | 14:7648334eb41b | 469 | */ |
hudakz | 14:7648334eb41b | 470 | nsapi_size_or_error_t UdpSocket::sendto(const SocketAddress& address, const void* data, size_t size) |
hudakz | 14:7648334eb41b | 471 | { |
hudakz | 14:7648334eb41b | 472 | IpAddress ip_addr(address.get_addr().bytes); |
hudakz | 14:7648334eb41b | 473 | uint16_t port = address.get_port(); |
hudakz | 14:7648334eb41b | 474 | |
hudakz | 14:7648334eb41b | 475 | _remote_addr = address; |
hudakz | 14:7648334eb41b | 476 | |
hudakz | 14:7648334eb41b | 477 | if (beginPacket(ip_addr, port) == 0) { |
hudakz | 14:7648334eb41b | 478 | stop(); |
hudakz | 14:7648334eb41b | 479 | return NSAPI_ERROR_NO_ADDRESS; |
hudakz | 14:7648334eb41b | 480 | } |
hudakz | 14:7648334eb41b | 481 | |
hudakz | 14:7648334eb41b | 482 | if (write((uint8_t*)data, size) == 0) { |
hudakz | 14:7648334eb41b | 483 | stop(); |
hudakz | 14:7648334eb41b | 484 | return NSAPI_ERROR_WOULD_BLOCK; |
hudakz | 14:7648334eb41b | 485 | }; |
hudakz | 14:7648334eb41b | 486 | |
hudakz | 14:7648334eb41b | 487 | if (endPacket() <= 0) { |
hudakz | 14:7648334eb41b | 488 | stop(); |
hudakz | 14:7648334eb41b | 489 | return NSAPI_ERROR_WOULD_BLOCK; |
hudakz | 14:7648334eb41b | 490 | } |
hudakz | 14:7648334eb41b | 491 | |
hudakz | 14:7648334eb41b | 492 | return NSAPI_ERROR_OK; |
hudakz | 14:7648334eb41b | 493 | } |
hudakz | 14:7648334eb41b | 494 | |
hudakz | 14:7648334eb41b | 495 | /** |
hudakz | 14:7648334eb41b | 496 | * @brief |
hudakz | 14:7648334eb41b | 497 | * @note |
hudakz | 14:7648334eb41b | 498 | * @param |
hudakz | 14:7648334eb41b | 499 | * @retval |
hudakz | 14:7648334eb41b | 500 | */ |
hudakz | 14:7648334eb41b | 501 | nsapi_size_or_error_t UdpSocket::recvfrom(SocketAddress* address, void* data, size_t size) |
hudakz | 14:7648334eb41b | 502 | { |
hudakz | 14:7648334eb41b | 503 | *address = _remote_addr; |
hudakz | 14:7648334eb41b | 504 | |
hudakz | 14:7648334eb41b | 505 | Timer timer; |
hudakz | 14:7648334eb41b | 506 | int success; |
hudakz | 14:7648334eb41b | 507 | |
hudakz | 14:7648334eb41b | 508 | timer.start(); |
hudakz | 14:7648334eb41b | 509 | do { |
hudakz | 14:7648334eb41b | 510 | success = parsePacket(); |
hudakz | 14:7648334eb41b | 511 | } while (!success && (timer.read_ms() < _timeout_ms)); |
hudakz | 14:7648334eb41b | 512 | |
hudakz | 14:7648334eb41b | 513 | if (!success) { |
hudakz | 14:7648334eb41b | 514 | stop(); |
hudakz | 14:7648334eb41b | 515 | return NSAPI_ERROR_WOULD_BLOCK; |
hudakz | 14:7648334eb41b | 516 | } |
hudakz | 14:7648334eb41b | 517 | |
hudakz | 14:7648334eb41b | 518 | size_t n; |
hudakz | 14:7648334eb41b | 519 | size_t recv_count = 0; |
hudakz | 14:7648334eb41b | 520 | uint8_t* pos = (uint8_t*)data; |
hudakz | 14:7648334eb41b | 521 | |
hudakz | 14:7648334eb41b | 522 | do { |
hudakz | 14:7648334eb41b | 523 | if (recv_count + available() <= size) { |
hudakz | 14:7648334eb41b | 524 | n = read(pos, available()); |
hudakz | 14:7648334eb41b | 525 | pos += n; |
hudakz | 14:7648334eb41b | 526 | recv_count += n; |
hudakz | 14:7648334eb41b | 527 | } |
hudakz | 14:7648334eb41b | 528 | else { |
hudakz | 14:7648334eb41b | 529 | return NSAPI_ERROR_NO_MEMORY; |
hudakz | 14:7648334eb41b | 530 | } |
hudakz | 14:7648334eb41b | 531 | } while ((available() > 0) && (recv_count < size)); |
hudakz | 14:7648334eb41b | 532 | |
hudakz | 14:7648334eb41b | 533 | flush(); |
hudakz | 14:7648334eb41b | 534 | |
hudakz | 14:7648334eb41b | 535 | return NSAPI_ERROR_OK; |
hudakz | 14:7648334eb41b | 536 | } |