Kenji Arai / mbed-os_TYBLE16

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?

UserRevisionLine numberNew 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 }