mbed-os5 only for TYBLE16
Dependents: TYBLE16_simple_data_logger TYBLE16_MP3_Air
features/netsocket/UDPSocket.cpp@0:5b88d5760320, 2019-12-17 (annotated)
- Committer:
- kenjiArai
- Date:
- Tue Dec 17 23:23:45 2019 +0000
- Revision:
- 0:5b88d5760320
- Child:
- 1:9db0e321a9f4
mbed-os5 only for TYBLE16
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
kenjiArai | 0:5b88d5760320 | 1 | /* Socket |
kenjiArai | 0:5b88d5760320 | 2 | * Copyright (c) 2015 ARM Limited |
kenjiArai | 0:5b88d5760320 | 3 | * |
kenjiArai | 0:5b88d5760320 | 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
kenjiArai | 0:5b88d5760320 | 5 | * you may not use this file except in compliance with the License. |
kenjiArai | 0:5b88d5760320 | 6 | * You may obtain a copy of the License at |
kenjiArai | 0:5b88d5760320 | 7 | * |
kenjiArai | 0:5b88d5760320 | 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
kenjiArai | 0:5b88d5760320 | 9 | * |
kenjiArai | 0:5b88d5760320 | 10 | * Unless required by applicable law or agreed to in writing, software |
kenjiArai | 0:5b88d5760320 | 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
kenjiArai | 0:5b88d5760320 | 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
kenjiArai | 0:5b88d5760320 | 13 | * See the License for the specific language governing permissions and |
kenjiArai | 0:5b88d5760320 | 14 | * limitations under the License. |
kenjiArai | 0:5b88d5760320 | 15 | */ |
kenjiArai | 0:5b88d5760320 | 16 | |
kenjiArai | 0:5b88d5760320 | 17 | #include "UDPSocket.h" |
kenjiArai | 0:5b88d5760320 | 18 | #include "Timer.h" |
kenjiArai | 0:5b88d5760320 | 19 | #include "mbed_assert.h" |
kenjiArai | 0:5b88d5760320 | 20 | |
kenjiArai | 0:5b88d5760320 | 21 | UDPSocket::UDPSocket() |
kenjiArai | 0:5b88d5760320 | 22 | { |
kenjiArai | 0:5b88d5760320 | 23 | _socket_stats.stats_update_proto(this, NSAPI_UDP); |
kenjiArai | 0:5b88d5760320 | 24 | } |
kenjiArai | 0:5b88d5760320 | 25 | |
kenjiArai | 0:5b88d5760320 | 26 | UDPSocket::~UDPSocket() |
kenjiArai | 0:5b88d5760320 | 27 | { |
kenjiArai | 0:5b88d5760320 | 28 | } |
kenjiArai | 0:5b88d5760320 | 29 | |
kenjiArai | 0:5b88d5760320 | 30 | nsapi_protocol_t UDPSocket::get_proto() |
kenjiArai | 0:5b88d5760320 | 31 | { |
kenjiArai | 0:5b88d5760320 | 32 | return NSAPI_UDP; |
kenjiArai | 0:5b88d5760320 | 33 | } |
kenjiArai | 0:5b88d5760320 | 34 | |
kenjiArai | 0:5b88d5760320 | 35 | nsapi_error_t UDPSocket::connect(const SocketAddress &address) |
kenjiArai | 0:5b88d5760320 | 36 | { |
kenjiArai | 0:5b88d5760320 | 37 | _remote_peer = address; |
kenjiArai | 0:5b88d5760320 | 38 | _socket_stats.stats_update_peer(this, _remote_peer); |
kenjiArai | 0:5b88d5760320 | 39 | _socket_stats.stats_update_socket_state(this, SOCK_CONNECTED); |
kenjiArai | 0:5b88d5760320 | 40 | return NSAPI_ERROR_OK; |
kenjiArai | 0:5b88d5760320 | 41 | } |
kenjiArai | 0:5b88d5760320 | 42 | |
kenjiArai | 0:5b88d5760320 | 43 | nsapi_size_or_error_t UDPSocket::sendto(const char *host, uint16_t port, const void *data, nsapi_size_t size) |
kenjiArai | 0:5b88d5760320 | 44 | { |
kenjiArai | 0:5b88d5760320 | 45 | SocketAddress address; |
kenjiArai | 0:5b88d5760320 | 46 | nsapi_size_or_error_t err; |
kenjiArai | 0:5b88d5760320 | 47 | |
kenjiArai | 0:5b88d5760320 | 48 | if (!strcmp(_interface_name, "")) { |
kenjiArai | 0:5b88d5760320 | 49 | err = _stack->gethostbyname(host, &address); |
kenjiArai | 0:5b88d5760320 | 50 | } else { |
kenjiArai | 0:5b88d5760320 | 51 | err = _stack->gethostbyname(host, &address, NSAPI_UNSPEC, _interface_name); |
kenjiArai | 0:5b88d5760320 | 52 | } |
kenjiArai | 0:5b88d5760320 | 53 | |
kenjiArai | 0:5b88d5760320 | 54 | if (err) { |
kenjiArai | 0:5b88d5760320 | 55 | return NSAPI_ERROR_DNS_FAILURE; |
kenjiArai | 0:5b88d5760320 | 56 | } |
kenjiArai | 0:5b88d5760320 | 57 | |
kenjiArai | 0:5b88d5760320 | 58 | address.set_port(port); |
kenjiArai | 0:5b88d5760320 | 59 | |
kenjiArai | 0:5b88d5760320 | 60 | // sendto is thread safe |
kenjiArai | 0:5b88d5760320 | 61 | return sendto(address, data, size); |
kenjiArai | 0:5b88d5760320 | 62 | } |
kenjiArai | 0:5b88d5760320 | 63 | |
kenjiArai | 0:5b88d5760320 | 64 | nsapi_size_or_error_t UDPSocket::sendto(const SocketAddress &address, const void *data, nsapi_size_t size) |
kenjiArai | 0:5b88d5760320 | 65 | { |
kenjiArai | 0:5b88d5760320 | 66 | _lock.lock(); |
kenjiArai | 0:5b88d5760320 | 67 | nsapi_size_or_error_t ret; |
kenjiArai | 0:5b88d5760320 | 68 | |
kenjiArai | 0:5b88d5760320 | 69 | _writers++; |
kenjiArai | 0:5b88d5760320 | 70 | if (_socket) { |
kenjiArai | 0:5b88d5760320 | 71 | _socket_stats.stats_update_socket_state(this, SOCK_OPEN); |
kenjiArai | 0:5b88d5760320 | 72 | _socket_stats.stats_update_peer(this, address); |
kenjiArai | 0:5b88d5760320 | 73 | } |
kenjiArai | 0:5b88d5760320 | 74 | while (true) { |
kenjiArai | 0:5b88d5760320 | 75 | if (!_socket) { |
kenjiArai | 0:5b88d5760320 | 76 | ret = NSAPI_ERROR_NO_SOCKET; |
kenjiArai | 0:5b88d5760320 | 77 | break; |
kenjiArai | 0:5b88d5760320 | 78 | } |
kenjiArai | 0:5b88d5760320 | 79 | |
kenjiArai | 0:5b88d5760320 | 80 | core_util_atomic_flag_clear(&_pending); |
kenjiArai | 0:5b88d5760320 | 81 | nsapi_size_or_error_t sent = _stack->socket_sendto(_socket, address, data, size); |
kenjiArai | 0:5b88d5760320 | 82 | if ((0 == _timeout) || (NSAPI_ERROR_WOULD_BLOCK != sent)) { |
kenjiArai | 0:5b88d5760320 | 83 | _socket_stats.stats_update_sent_bytes(this, sent); |
kenjiArai | 0:5b88d5760320 | 84 | ret = sent; |
kenjiArai | 0:5b88d5760320 | 85 | break; |
kenjiArai | 0:5b88d5760320 | 86 | } else { |
kenjiArai | 0:5b88d5760320 | 87 | uint32_t flag; |
kenjiArai | 0:5b88d5760320 | 88 | |
kenjiArai | 0:5b88d5760320 | 89 | // Release lock before blocking so other threads |
kenjiArai | 0:5b88d5760320 | 90 | // accessing this object aren't blocked |
kenjiArai | 0:5b88d5760320 | 91 | _lock.unlock(); |
kenjiArai | 0:5b88d5760320 | 92 | flag = _event_flag.wait_any(WRITE_FLAG, _timeout); |
kenjiArai | 0:5b88d5760320 | 93 | _lock.lock(); |
kenjiArai | 0:5b88d5760320 | 94 | |
kenjiArai | 0:5b88d5760320 | 95 | if (flag & osFlagsError) { |
kenjiArai | 0:5b88d5760320 | 96 | // Timeout break |
kenjiArai | 0:5b88d5760320 | 97 | ret = NSAPI_ERROR_WOULD_BLOCK; |
kenjiArai | 0:5b88d5760320 | 98 | break; |
kenjiArai | 0:5b88d5760320 | 99 | } |
kenjiArai | 0:5b88d5760320 | 100 | } |
kenjiArai | 0:5b88d5760320 | 101 | } |
kenjiArai | 0:5b88d5760320 | 102 | |
kenjiArai | 0:5b88d5760320 | 103 | _writers--; |
kenjiArai | 0:5b88d5760320 | 104 | if (!_socket || !_writers) { |
kenjiArai | 0:5b88d5760320 | 105 | _event_flag.set(FINISHED_FLAG); |
kenjiArai | 0:5b88d5760320 | 106 | } |
kenjiArai | 0:5b88d5760320 | 107 | _lock.unlock(); |
kenjiArai | 0:5b88d5760320 | 108 | return ret; |
kenjiArai | 0:5b88d5760320 | 109 | } |
kenjiArai | 0:5b88d5760320 | 110 | |
kenjiArai | 0:5b88d5760320 | 111 | nsapi_size_or_error_t UDPSocket::send(const void *data, nsapi_size_t size) |
kenjiArai | 0:5b88d5760320 | 112 | { |
kenjiArai | 0:5b88d5760320 | 113 | if (!_remote_peer) { |
kenjiArai | 0:5b88d5760320 | 114 | return NSAPI_ERROR_NO_ADDRESS; |
kenjiArai | 0:5b88d5760320 | 115 | } |
kenjiArai | 0:5b88d5760320 | 116 | return sendto(_remote_peer, data, size); |
kenjiArai | 0:5b88d5760320 | 117 | } |
kenjiArai | 0:5b88d5760320 | 118 | |
kenjiArai | 0:5b88d5760320 | 119 | nsapi_size_or_error_t UDPSocket::recvfrom(SocketAddress *address, void *buffer, nsapi_size_t size) |
kenjiArai | 0:5b88d5760320 | 120 | { |
kenjiArai | 0:5b88d5760320 | 121 | _lock.lock(); |
kenjiArai | 0:5b88d5760320 | 122 | nsapi_size_or_error_t ret; |
kenjiArai | 0:5b88d5760320 | 123 | SocketAddress ignored; |
kenjiArai | 0:5b88d5760320 | 124 | |
kenjiArai | 0:5b88d5760320 | 125 | if (!address) { |
kenjiArai | 0:5b88d5760320 | 126 | address = &ignored; |
kenjiArai | 0:5b88d5760320 | 127 | } |
kenjiArai | 0:5b88d5760320 | 128 | |
kenjiArai | 0:5b88d5760320 | 129 | _readers++; |
kenjiArai | 0:5b88d5760320 | 130 | |
kenjiArai | 0:5b88d5760320 | 131 | if (_socket) { |
kenjiArai | 0:5b88d5760320 | 132 | _socket_stats.stats_update_socket_state(this, SOCK_OPEN); |
kenjiArai | 0:5b88d5760320 | 133 | } |
kenjiArai | 0:5b88d5760320 | 134 | while (true) { |
kenjiArai | 0:5b88d5760320 | 135 | if (!_socket) { |
kenjiArai | 0:5b88d5760320 | 136 | ret = NSAPI_ERROR_NO_SOCKET; |
kenjiArai | 0:5b88d5760320 | 137 | break; |
kenjiArai | 0:5b88d5760320 | 138 | } |
kenjiArai | 0:5b88d5760320 | 139 | |
kenjiArai | 0:5b88d5760320 | 140 | core_util_atomic_flag_clear(&_pending); |
kenjiArai | 0:5b88d5760320 | 141 | nsapi_size_or_error_t recv = _stack->socket_recvfrom(_socket, address, buffer, size); |
kenjiArai | 0:5b88d5760320 | 142 | |
kenjiArai | 0:5b88d5760320 | 143 | // Filter incomming packets using connected peer address |
kenjiArai | 0:5b88d5760320 | 144 | if (recv >= 0 && _remote_peer && _remote_peer != *address) { |
kenjiArai | 0:5b88d5760320 | 145 | continue; |
kenjiArai | 0:5b88d5760320 | 146 | } |
kenjiArai | 0:5b88d5760320 | 147 | |
kenjiArai | 0:5b88d5760320 | 148 | _socket_stats.stats_update_peer(this, _remote_peer); |
kenjiArai | 0:5b88d5760320 | 149 | // Non-blocking sockets always return. Blocking only returns when success or errors other than WOULD_BLOCK |
kenjiArai | 0:5b88d5760320 | 150 | if ((0 == _timeout) || (NSAPI_ERROR_WOULD_BLOCK != recv)) { |
kenjiArai | 0:5b88d5760320 | 151 | ret = recv; |
kenjiArai | 0:5b88d5760320 | 152 | _socket_stats.stats_update_recv_bytes(this, recv); |
kenjiArai | 0:5b88d5760320 | 153 | break; |
kenjiArai | 0:5b88d5760320 | 154 | } else { |
kenjiArai | 0:5b88d5760320 | 155 | uint32_t flag; |
kenjiArai | 0:5b88d5760320 | 156 | |
kenjiArai | 0:5b88d5760320 | 157 | // Release lock before blocking so other threads |
kenjiArai | 0:5b88d5760320 | 158 | // accessing this object aren't blocked |
kenjiArai | 0:5b88d5760320 | 159 | _lock.unlock(); |
kenjiArai | 0:5b88d5760320 | 160 | flag = _event_flag.wait_any(READ_FLAG, _timeout); |
kenjiArai | 0:5b88d5760320 | 161 | _lock.lock(); |
kenjiArai | 0:5b88d5760320 | 162 | |
kenjiArai | 0:5b88d5760320 | 163 | if (flag & osFlagsError) { |
kenjiArai | 0:5b88d5760320 | 164 | // Timeout break |
kenjiArai | 0:5b88d5760320 | 165 | ret = NSAPI_ERROR_WOULD_BLOCK; |
kenjiArai | 0:5b88d5760320 | 166 | break; |
kenjiArai | 0:5b88d5760320 | 167 | } |
kenjiArai | 0:5b88d5760320 | 168 | } |
kenjiArai | 0:5b88d5760320 | 169 | } |
kenjiArai | 0:5b88d5760320 | 170 | |
kenjiArai | 0:5b88d5760320 | 171 | _readers--; |
kenjiArai | 0:5b88d5760320 | 172 | if (!_socket || !_readers) { |
kenjiArai | 0:5b88d5760320 | 173 | _event_flag.set(FINISHED_FLAG); |
kenjiArai | 0:5b88d5760320 | 174 | } |
kenjiArai | 0:5b88d5760320 | 175 | |
kenjiArai | 0:5b88d5760320 | 176 | _lock.unlock(); |
kenjiArai | 0:5b88d5760320 | 177 | return ret; |
kenjiArai | 0:5b88d5760320 | 178 | } |
kenjiArai | 0:5b88d5760320 | 179 | |
kenjiArai | 0:5b88d5760320 | 180 | nsapi_size_or_error_t UDPSocket::recv(void *buffer, nsapi_size_t size) |
kenjiArai | 0:5b88d5760320 | 181 | { |
kenjiArai | 0:5b88d5760320 | 182 | return recvfrom(NULL, buffer, size); |
kenjiArai | 0:5b88d5760320 | 183 | } |
kenjiArai | 0:5b88d5760320 | 184 | |
kenjiArai | 0:5b88d5760320 | 185 | Socket *UDPSocket::accept(nsapi_error_t *error) |
kenjiArai | 0:5b88d5760320 | 186 | { |
kenjiArai | 0:5b88d5760320 | 187 | if (error) { |
kenjiArai | 0:5b88d5760320 | 188 | *error = NSAPI_ERROR_UNSUPPORTED; |
kenjiArai | 0:5b88d5760320 | 189 | } |
kenjiArai | 0:5b88d5760320 | 190 | return NULL; |
kenjiArai | 0:5b88d5760320 | 191 | } |
kenjiArai | 0:5b88d5760320 | 192 | |
kenjiArai | 0:5b88d5760320 | 193 | nsapi_error_t UDPSocket::listen(int) |
kenjiArai | 0:5b88d5760320 | 194 | { |
kenjiArai | 0:5b88d5760320 | 195 | return NSAPI_ERROR_UNSUPPORTED; |
kenjiArai | 0:5b88d5760320 | 196 | } |