Kenji Arai / mbed-os_TYBLE16

mbed-os5 only for TYBLE16

Dependents:   TYBLE16_simple_data_logger TYBLE16_MP3_Air

features/netsocket/TCPSocket.cpp@1:9db0e321a9f4, 2019-12-31 (annotated)

Committer:
kenjiArai
Date:
Tue Dec 31 06:02:27 2019 +0000
Revision:
1:9db0e321a9f4
Parent:
0:5b88d5760320 
updated based on mbed-os5.15.0

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 "TCPSocket.h"
kenjiArai 0:5b88d5760320 18 #include "Timer.h"
kenjiArai 0:5b88d5760320 19 #include "mbed_assert.h"
kenjiArai 0:5b88d5760320 20
kenjiArai 0:5b88d5760320 21 TCPSocket::TCPSocket()
kenjiArai 0:5b88d5760320 22 {
kenjiArai 0:5b88d5760320 23 _socket_stats.stats_update_proto(this, NSAPI_TCP);
kenjiArai 0:5b88d5760320 24 }
kenjiArai 0:5b88d5760320 25
kenjiArai 0:5b88d5760320 26 TCPSocket::TCPSocket(TCPSocket *parent, nsapi_socket_t socket, SocketAddress address)
kenjiArai 0:5b88d5760320 27 {
kenjiArai 0:5b88d5760320 28 _socket = socket;
kenjiArai 0:5b88d5760320 29 _stack = parent->_stack;
kenjiArai 0:5b88d5760320 30 _factory_allocated = true;
kenjiArai 0:5b88d5760320 31 _remote_peer = address;
kenjiArai 0:5b88d5760320 32 _socket_stats.stats_new_socket_entry(this);
kenjiArai 0:5b88d5760320 33 _event = mbed::Callback<void()>(this, &TCPSocket::event);
kenjiArai 0:5b88d5760320 34 _stack->socket_attach(socket, &mbed::Callback<void()>::thunk, &_event);
kenjiArai 0:5b88d5760320 35 }
kenjiArai 0:5b88d5760320 36
kenjiArai 0:5b88d5760320 37 TCPSocket::~TCPSocket()
kenjiArai 0:5b88d5760320 38 {
kenjiArai 0:5b88d5760320 39 }
kenjiArai 0:5b88d5760320 40
kenjiArai 0:5b88d5760320 41 nsapi_protocol_t TCPSocket::get_proto()
kenjiArai 0:5b88d5760320 42 {
kenjiArai 0:5b88d5760320 43 return NSAPI_TCP;
kenjiArai 0:5b88d5760320 44 }
kenjiArai 0:5b88d5760320 45
kenjiArai 0:5b88d5760320 46 nsapi_error_t TCPSocket::connect(const SocketAddress &address)
kenjiArai 0:5b88d5760320 47 {
kenjiArai 0:5b88d5760320 48 _lock.lock();
kenjiArai 0:5b88d5760320 49 nsapi_error_t ret;
kenjiArai 0:5b88d5760320 50
kenjiArai 0:5b88d5760320 51 // If this assert is hit then there are two threads
kenjiArai 0:5b88d5760320 52 // performing a send at the same time which is undefined
kenjiArai 0:5b88d5760320 53 // behavior
kenjiArai 0:5b88d5760320 54 MBED_ASSERT(_writers == 0);
kenjiArai 0:5b88d5760320 55 _writers++;
kenjiArai 0:5b88d5760320 56
kenjiArai 0:5b88d5760320 57 bool blocking_connect_in_progress = false;
kenjiArai 0:5b88d5760320 58
kenjiArai 0:5b88d5760320 59 while (true) {
kenjiArai 0:5b88d5760320 60 if (!_socket) {
kenjiArai 0:5b88d5760320 61 ret = NSAPI_ERROR_NO_SOCKET;
kenjiArai 0:5b88d5760320 62 break;
kenjiArai 0:5b88d5760320 63 }
kenjiArai 0:5b88d5760320 64
kenjiArai 0:5b88d5760320 65 core_util_atomic_flag_clear(&_pending);
kenjiArai 0:5b88d5760320 66 ret = _stack->socket_connect(_socket, address);
kenjiArai 0:5b88d5760320 67 if ((_timeout == 0) || !(ret == NSAPI_ERROR_IN_PROGRESS || ret == NSAPI_ERROR_ALREADY)) {
kenjiArai 0:5b88d5760320 68 _socket_stats.stats_update_socket_state(this, SOCK_CONNECTED);
kenjiArai 0:5b88d5760320 69 break;
kenjiArai 0:5b88d5760320 70 } else {
kenjiArai 0:5b88d5760320 71 blocking_connect_in_progress = true;
kenjiArai 0:5b88d5760320 72
kenjiArai 0:5b88d5760320 73 uint32_t flag;
kenjiArai 0:5b88d5760320 74
kenjiArai 0:5b88d5760320 75 // Release lock before blocking so other threads
kenjiArai 0:5b88d5760320 76 // accessing this object aren't blocked
kenjiArai 0:5b88d5760320 77 _lock.unlock();
kenjiArai 0:5b88d5760320 78 flag = _event_flag.wait_any(WRITE_FLAG, _timeout);
kenjiArai 0:5b88d5760320 79 _lock.lock();
kenjiArai 0:5b88d5760320 80 if (flag & osFlagsError) {
kenjiArai 0:5b88d5760320 81 // Timeout break
kenjiArai 0:5b88d5760320 82 break;
kenjiArai 0:5b88d5760320 83 }
kenjiArai 0:5b88d5760320 84 }
kenjiArai 0:5b88d5760320 85 }
kenjiArai 0:5b88d5760320 86
kenjiArai 0:5b88d5760320 87 _writers--;
kenjiArai 0:5b88d5760320 88 if (!_socket) {
kenjiArai 0:5b88d5760320 89 _event_flag.set(FINISHED_FLAG);
kenjiArai 0:5b88d5760320 90 }
kenjiArai 0:5b88d5760320 91
kenjiArai 0:5b88d5760320 92 /* Non-blocking connect gives "EISCONN" once done - convert to OK for blocking mode if we became connected during this call */
kenjiArai 0:5b88d5760320 93 if (ret == NSAPI_ERROR_IS_CONNECTED && blocking_connect_in_progress) {
kenjiArai 0:5b88d5760320 94 _socket_stats.stats_update_socket_state(this, SOCK_CONNECTED);
kenjiArai 0:5b88d5760320 95 ret = NSAPI_ERROR_OK;
kenjiArai 0:5b88d5760320 96 }
kenjiArai 0:5b88d5760320 97
kenjiArai 0:5b88d5760320 98 if (ret == NSAPI_ERROR_OK || ret == NSAPI_ERROR_IN_PROGRESS) {
kenjiArai 0:5b88d5760320 99 _remote_peer = address;
kenjiArai 0:5b88d5760320 100 _socket_stats.stats_update_peer(this, _remote_peer);
kenjiArai 0:5b88d5760320 101 }
kenjiArai 0:5b88d5760320 102
kenjiArai 0:5b88d5760320 103 _lock.unlock();
kenjiArai 0:5b88d5760320 104 return ret;
kenjiArai 0:5b88d5760320 105 }
kenjiArai 0:5b88d5760320 106
kenjiArai 0:5b88d5760320 107 nsapi_error_t TCPSocket::connect(const char *host, uint16_t port)
kenjiArai 0:5b88d5760320 108 {
kenjiArai 0:5b88d5760320 109 SocketAddress address;
kenjiArai 0:5b88d5760320 110 if (!_socket) {
kenjiArai 0:5b88d5760320 111 return NSAPI_ERROR_NO_SOCKET;
kenjiArai 0:5b88d5760320 112 }
kenjiArai 0:5b88d5760320 113 nsapi_error_t err;
kenjiArai 0:5b88d5760320 114 if (!strcmp(_interface_name, "")) {
kenjiArai 0:5b88d5760320 115 err = _stack->gethostbyname(host, &address);
kenjiArai 0:5b88d5760320 116 } else {
kenjiArai 0:5b88d5760320 117 err = _stack->gethostbyname(host, &address, NSAPI_UNSPEC, _interface_name);
kenjiArai 0:5b88d5760320 118 }
kenjiArai 0:5b88d5760320 119 if (err) {
kenjiArai 0:5b88d5760320 120 return NSAPI_ERROR_DNS_FAILURE;
kenjiArai 0:5b88d5760320 121 }
kenjiArai 0:5b88d5760320 122
kenjiArai 0:5b88d5760320 123 address.set_port(port);
kenjiArai 0:5b88d5760320 124
kenjiArai 0:5b88d5760320 125 // connect is thread safe
kenjiArai 0:5b88d5760320 126 return connect(address);
kenjiArai 0:5b88d5760320 127 }
kenjiArai 0:5b88d5760320 128
kenjiArai 0:5b88d5760320 129 nsapi_size_or_error_t TCPSocket::send(const void *data, nsapi_size_t size)
kenjiArai 0:5b88d5760320 130 {
kenjiArai 0:5b88d5760320 131 _lock.lock();
kenjiArai 0:5b88d5760320 132 const uint8_t *data_ptr = static_cast<const uint8_t *>(data);
kenjiArai 0:5b88d5760320 133 nsapi_size_or_error_t ret;
kenjiArai 0:5b88d5760320 134 nsapi_size_t written = 0;
kenjiArai 0:5b88d5760320 135
kenjiArai 0:5b88d5760320 136 // If this assert is hit then there are two threads
kenjiArai 0:5b88d5760320 137 // performing a send at the same time which is undefined
kenjiArai 0:5b88d5760320 138 // behavior
kenjiArai 0:5b88d5760320 139 MBED_ASSERT(_writers == 0);
kenjiArai 0:5b88d5760320 140 _writers++;
kenjiArai 0:5b88d5760320 141
kenjiArai 0:5b88d5760320 142 // Unlike recv, we should write the whole thing if blocking. POSIX only
kenjiArai 0:5b88d5760320 143 // allows partial as a side-effect of signal handling; it normally tries to
kenjiArai 0:5b88d5760320 144 // write everything if blocking. Without signals we can always write all.
kenjiArai 0:5b88d5760320 145 while (true) {
kenjiArai 0:5b88d5760320 146 if (!_socket) {
kenjiArai 0:5b88d5760320 147 ret = NSAPI_ERROR_NO_SOCKET;
kenjiArai 0:5b88d5760320 148 break;
kenjiArai 0:5b88d5760320 149 }
kenjiArai 0:5b88d5760320 150
kenjiArai 0:5b88d5760320 151 core_util_atomic_flag_clear(&_pending);
kenjiArai 0:5b88d5760320 152 ret = _stack->socket_send(_socket, data_ptr + written, size - written);
kenjiArai 0:5b88d5760320 153 if (ret >= 0) {
kenjiArai 0:5b88d5760320 154 written += ret;
kenjiArai 0:5b88d5760320 155 if (written >= size) {
kenjiArai 0:5b88d5760320 156 break;
kenjiArai 0:5b88d5760320 157 }
kenjiArai 0:5b88d5760320 158 }
kenjiArai 0:5b88d5760320 159 if (_timeout == 0) {
kenjiArai 0:5b88d5760320 160 break;
kenjiArai 0:5b88d5760320 161 } else if (ret == NSAPI_ERROR_WOULD_BLOCK) {
kenjiArai 0:5b88d5760320 162 uint32_t flag;
kenjiArai 0:5b88d5760320 163
kenjiArai 0:5b88d5760320 164 // Release lock before blocking so other threads
kenjiArai 0:5b88d5760320 165 // accessing this object aren't blocked
kenjiArai 0:5b88d5760320 166 _lock.unlock();
kenjiArai 0:5b88d5760320 167 flag = _event_flag.wait_any(WRITE_FLAG, _timeout);
kenjiArai 0:5b88d5760320 168 _lock.lock();
kenjiArai 0:5b88d5760320 169
kenjiArai 0:5b88d5760320 170 if (flag & osFlagsError) {
kenjiArai 0:5b88d5760320 171 // Timeout break
kenjiArai 0:5b88d5760320 172 break;
kenjiArai 0:5b88d5760320 173 }
kenjiArai 0:5b88d5760320 174 } else if (ret < 0) {
kenjiArai 0:5b88d5760320 175 break;
kenjiArai 0:5b88d5760320 176 }
kenjiArai 0:5b88d5760320 177 }
kenjiArai 0:5b88d5760320 178
kenjiArai 0:5b88d5760320 179 _writers--;
kenjiArai 0:5b88d5760320 180 if (!_socket) {
kenjiArai 0:5b88d5760320 181 _event_flag.set(FINISHED_FLAG);
kenjiArai 0:5b88d5760320 182 }
kenjiArai 0:5b88d5760320 183
kenjiArai 0:5b88d5760320 184 _lock.unlock();
kenjiArai 0:5b88d5760320 185 if (ret <= 0 && ret != NSAPI_ERROR_WOULD_BLOCK) {
kenjiArai 0:5b88d5760320 186 return ret;
kenjiArai 0:5b88d5760320 187 } else if (written == 0) {
kenjiArai 0:5b88d5760320 188 return NSAPI_ERROR_WOULD_BLOCK;
kenjiArai 0:5b88d5760320 189 } else {
kenjiArai 0:5b88d5760320 190 _socket_stats.stats_update_sent_bytes(this, written);
kenjiArai 0:5b88d5760320 191 return written;
kenjiArai 0:5b88d5760320 192 }
kenjiArai 0:5b88d5760320 193 }
kenjiArai 0:5b88d5760320 194
kenjiArai 0:5b88d5760320 195 nsapi_size_or_error_t TCPSocket::sendto(const SocketAddress &address, const void *data, nsapi_size_t size)
kenjiArai 0:5b88d5760320 196 {
kenjiArai 0:5b88d5760320 197 (void)address;
kenjiArai 0:5b88d5760320 198 return send(data, size);
kenjiArai 0:5b88d5760320 199 }
kenjiArai 0:5b88d5760320 200
kenjiArai 0:5b88d5760320 201 nsapi_size_or_error_t TCPSocket::recv(void *data, nsapi_size_t size)
kenjiArai 0:5b88d5760320 202 {
kenjiArai 0:5b88d5760320 203 _lock.lock();
kenjiArai 0:5b88d5760320 204 nsapi_size_or_error_t ret;
kenjiArai 0:5b88d5760320 205
kenjiArai 0:5b88d5760320 206 // If this assert is hit then there are two threads
kenjiArai 0:5b88d5760320 207 // performing a recv at the same time which is undefined
kenjiArai 0:5b88d5760320 208 // behavior
kenjiArai 0:5b88d5760320 209 MBED_ASSERT(_readers == 0);
kenjiArai 0:5b88d5760320 210 _readers++;
kenjiArai 0:5b88d5760320 211
kenjiArai 0:5b88d5760320 212 while (true) {
kenjiArai 0:5b88d5760320 213 if (!_socket) {
kenjiArai 0:5b88d5760320 214 ret = NSAPI_ERROR_NO_SOCKET;
kenjiArai 0:5b88d5760320 215 break;
kenjiArai 0:5b88d5760320 216 }
kenjiArai 0:5b88d5760320 217
kenjiArai 0:5b88d5760320 218 core_util_atomic_flag_clear(&_pending);
kenjiArai 0:5b88d5760320 219 ret = _stack->socket_recv(_socket, data, size);
kenjiArai 0:5b88d5760320 220 if ((_timeout == 0) || (ret != NSAPI_ERROR_WOULD_BLOCK)) {
kenjiArai 0:5b88d5760320 221 _socket_stats.stats_update_recv_bytes(this, ret);
kenjiArai 0:5b88d5760320 222 break;
kenjiArai 0:5b88d5760320 223 } else {
kenjiArai 0:5b88d5760320 224 uint32_t flag;
kenjiArai 0:5b88d5760320 225
kenjiArai 0:5b88d5760320 226 // Release lock before blocking so other threads
kenjiArai 0:5b88d5760320 227 // accessing this object aren't blocked
kenjiArai 0:5b88d5760320 228 _lock.unlock();
kenjiArai 0:5b88d5760320 229 flag = _event_flag.wait_any(READ_FLAG, _timeout);
kenjiArai 0:5b88d5760320 230 _lock.lock();
kenjiArai 0:5b88d5760320 231
kenjiArai 0:5b88d5760320 232 if (flag & osFlagsError) {
kenjiArai 0:5b88d5760320 233 // Timeout break
kenjiArai 0:5b88d5760320 234 ret = NSAPI_ERROR_WOULD_BLOCK;
kenjiArai 0:5b88d5760320 235 break;
kenjiArai 0:5b88d5760320 236 }
kenjiArai 0:5b88d5760320 237 }
kenjiArai 0:5b88d5760320 238 }
kenjiArai 0:5b88d5760320 239
kenjiArai 0:5b88d5760320 240 _readers--;
kenjiArai 0:5b88d5760320 241 if (!_socket) {
kenjiArai 0:5b88d5760320 242 _event_flag.set(FINISHED_FLAG);
kenjiArai 0:5b88d5760320 243 }
kenjiArai 0:5b88d5760320 244
kenjiArai 0:5b88d5760320 245 _lock.unlock();
kenjiArai 0:5b88d5760320 246 return ret;
kenjiArai 0:5b88d5760320 247 }
kenjiArai 0:5b88d5760320 248
kenjiArai 0:5b88d5760320 249 nsapi_size_or_error_t TCPSocket::recvfrom(SocketAddress *address, void *data, nsapi_size_t size)
kenjiArai 0:5b88d5760320 250 {
kenjiArai 0:5b88d5760320 251 if (address) {
kenjiArai 0:5b88d5760320 252 *address = _remote_peer;
kenjiArai 0:5b88d5760320 253 }
kenjiArai 0:5b88d5760320 254 return recv(data, size);
kenjiArai 0:5b88d5760320 255 }
kenjiArai 0:5b88d5760320 256
kenjiArai 0:5b88d5760320 257 nsapi_error_t TCPSocket::listen(int backlog)
kenjiArai 0:5b88d5760320 258 {
kenjiArai 0:5b88d5760320 259 _lock.lock();
kenjiArai 0:5b88d5760320 260 nsapi_error_t ret;
kenjiArai 0:5b88d5760320 261
kenjiArai 0:5b88d5760320 262 if (!_socket) {
kenjiArai 0:5b88d5760320 263 ret = NSAPI_ERROR_NO_SOCKET;
kenjiArai 0:5b88d5760320 264 } else {
kenjiArai 0:5b88d5760320 265 ret = _stack->socket_listen(_socket, backlog);
kenjiArai 0:5b88d5760320 266 if (NSAPI_ERROR_OK == ret) {
kenjiArai 0:5b88d5760320 267 _socket_stats.stats_update_socket_state(this, SOCK_LISTEN);
kenjiArai 0:5b88d5760320 268 }
kenjiArai 0:5b88d5760320 269 }
kenjiArai 0:5b88d5760320 270
kenjiArai 0:5b88d5760320 271 _lock.unlock();
kenjiArai 0:5b88d5760320 272 return ret;
kenjiArai 0:5b88d5760320 273 }
kenjiArai 0:5b88d5760320 274
kenjiArai 0:5b88d5760320 275 TCPSocket *TCPSocket::accept(nsapi_error_t *error)
kenjiArai 0:5b88d5760320 276 {
kenjiArai 0:5b88d5760320 277 _lock.lock();
kenjiArai 0:5b88d5760320 278 TCPSocket *connection = NULL;
kenjiArai 0:5b88d5760320 279 nsapi_error_t ret;
kenjiArai 0:5b88d5760320 280
kenjiArai 0:5b88d5760320 281 _readers++;
kenjiArai 0:5b88d5760320 282
kenjiArai 0:5b88d5760320 283 while (true) {
kenjiArai 0:5b88d5760320 284 if (!_socket) {
kenjiArai 0:5b88d5760320 285 ret = NSAPI_ERROR_NO_SOCKET;
kenjiArai 0:5b88d5760320 286 break;
kenjiArai 0:5b88d5760320 287 }
kenjiArai 0:5b88d5760320 288
kenjiArai 0:5b88d5760320 289 core_util_atomic_flag_clear(&_pending);
kenjiArai 0:5b88d5760320 290 void *socket;
kenjiArai 0:5b88d5760320 291 SocketAddress address;
kenjiArai 0:5b88d5760320 292 ret = _stack->socket_accept(_socket, &socket, &address);
kenjiArai 0:5b88d5760320 293
kenjiArai 0:5b88d5760320 294 if (0 == ret) {
kenjiArai 0:5b88d5760320 295 connection = new TCPSocket(this, socket, address);
kenjiArai 0:5b88d5760320 296 _socket_stats.stats_update_peer(connection, address);
kenjiArai 0:5b88d5760320 297 _socket_stats.stats_update_socket_state(connection, SOCK_CONNECTED);
kenjiArai 0:5b88d5760320 298 break;
kenjiArai 0:5b88d5760320 299 } else if ((_timeout == 0) || (ret != NSAPI_ERROR_WOULD_BLOCK)) {
kenjiArai 0:5b88d5760320 300 break;
kenjiArai 0:5b88d5760320 301 } else {
kenjiArai 0:5b88d5760320 302 uint32_t flag;
kenjiArai 0:5b88d5760320 303
kenjiArai 0:5b88d5760320 304 // Release lock before blocking so other threads
kenjiArai 0:5b88d5760320 305 // accessing this object aren't blocked
kenjiArai 0:5b88d5760320 306 _lock.unlock();
kenjiArai 0:5b88d5760320 307 flag = _event_flag.wait_any(READ_FLAG, _timeout);
kenjiArai 0:5b88d5760320 308 _lock.lock();
kenjiArai 0:5b88d5760320 309
kenjiArai 0:5b88d5760320 310 if (flag & osFlagsError) {
kenjiArai 0:5b88d5760320 311 // Timeout break
kenjiArai 0:5b88d5760320 312 ret = NSAPI_ERROR_WOULD_BLOCK;
kenjiArai 0:5b88d5760320 313 break;
kenjiArai 0:5b88d5760320 314 }
kenjiArai 0:5b88d5760320 315 }
kenjiArai 0:5b88d5760320 316 }
kenjiArai 0:5b88d5760320 317
kenjiArai 0:5b88d5760320 318 _readers--;
kenjiArai 0:5b88d5760320 319 if (!_socket) {
kenjiArai 0:5b88d5760320 320 _event_flag.set(FINISHED_FLAG);
kenjiArai 0:5b88d5760320 321 }
kenjiArai 0:5b88d5760320 322 _lock.unlock();
kenjiArai 0:5b88d5760320 323 if (error) {
kenjiArai 0:5b88d5760320 324 *error = ret;
kenjiArai 0:5b88d5760320 325 }
kenjiArai 0:5b88d5760320 326 return connection;
kenjiArai 0:5b88d5760320 327 }