Marco Mayer / Mbed OS Queue

Important changes to repositories hosted on mbed.com

Mbed hosted mercurial repositories are deprecated and are due to be permanently deleted in July 2026.

To keep a copy of this software download the repository Zip archive or clone locally using Mercurial.

It is also possible to export all your personal repositories from the account settings page.

mbed-os/features/netsocket/TCPSocket.cpp@2:c7897a3f5f11, 2020-04-22 (annotated)

Committer:
demayer
Date:
Wed Apr 22 11:50:00 2020 +0000
Revision:
2:c7897a3f5f11
Parent:
0:6bf0743ece18 
test

Who changed what in which revision?

UserRevisionLine numberNew contents of line
demayer 0:6bf0743ece18 1 /* Socket
demayer 0:6bf0743ece18 2 * Copyright (c) 2015 ARM Limited
demayer 0:6bf0743ece18 3 *
demayer 0:6bf0743ece18 4 * Licensed under the Apache License, Version 2.0 (the "License");
demayer 0:6bf0743ece18 5 * you may not use this file except in compliance with the License.
demayer 0:6bf0743ece18 6 * You may obtain a copy of the License at
demayer 0:6bf0743ece18 7 *
demayer 0:6bf0743ece18 8 * http://www.apache.org/licenses/LICENSE-2.0
demayer 0:6bf0743ece18 9 *
demayer 0:6bf0743ece18 10 * Unless required by applicable law or agreed to in writing, software
demayer 0:6bf0743ece18 11 * distributed under the License is distributed on an "AS IS" BASIS,
demayer 0:6bf0743ece18 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
demayer 0:6bf0743ece18 13 * See the License for the specific language governing permissions and
demayer 0:6bf0743ece18 14 * limitations under the License.
demayer 0:6bf0743ece18 15 */
demayer 0:6bf0743ece18 16
demayer 0:6bf0743ece18 17 #include "TCPSocket.h"
demayer 0:6bf0743ece18 18 #include "Timer.h"
demayer 0:6bf0743ece18 19 #include "mbed_assert.h"
demayer 0:6bf0743ece18 20
demayer 0:6bf0743ece18 21 #define READ_FLAG 0x1u
demayer 0:6bf0743ece18 22 #define WRITE_FLAG 0x2u
demayer 0:6bf0743ece18 23
demayer 0:6bf0743ece18 24 TCPSocket::TCPSocket()
demayer 0:6bf0743ece18 25 : _pending(0), _event_flag(),
demayer 0:6bf0743ece18 26 _read_in_progress(false), _write_in_progress(false)
demayer 0:6bf0743ece18 27 {
demayer 0:6bf0743ece18 28 }
demayer 0:6bf0743ece18 29
demayer 0:6bf0743ece18 30 TCPSocket::~TCPSocket()
demayer 0:6bf0743ece18 31 {
demayer 0:6bf0743ece18 32 close();
demayer 0:6bf0743ece18 33 }
demayer 0:6bf0743ece18 34
demayer 0:6bf0743ece18 35 nsapi_protocol_t TCPSocket::get_proto()
demayer 0:6bf0743ece18 36 {
demayer 0:6bf0743ece18 37 return NSAPI_TCP;
demayer 0:6bf0743ece18 38 }
demayer 0:6bf0743ece18 39
demayer 0:6bf0743ece18 40 nsapi_error_t TCPSocket::connect(const SocketAddress &address)
demayer 0:6bf0743ece18 41 {
demayer 0:6bf0743ece18 42 _lock.lock();
demayer 0:6bf0743ece18 43 nsapi_error_t ret;
demayer 0:6bf0743ece18 44
demayer 0:6bf0743ece18 45 // If this assert is hit then there are two threads
demayer 0:6bf0743ece18 46 // performing a send at the same time which is undefined
demayer 0:6bf0743ece18 47 // behavior
demayer 0:6bf0743ece18 48 MBED_ASSERT(!_write_in_progress);
demayer 0:6bf0743ece18 49 _write_in_progress = true;
demayer 0:6bf0743ece18 50
demayer 0:6bf0743ece18 51 bool blocking_connect_in_progress = false;
demayer 0:6bf0743ece18 52
demayer 0:6bf0743ece18 53 while (true) {
demayer 0:6bf0743ece18 54 if (!_socket) {
demayer 0:6bf0743ece18 55 ret = NSAPI_ERROR_NO_SOCKET;
demayer 0:6bf0743ece18 56 break;
demayer 0:6bf0743ece18 57 }
demayer 0:6bf0743ece18 58
demayer 0:6bf0743ece18 59 _pending = 0;
demayer 0:6bf0743ece18 60 ret = _stack->socket_connect(_socket, address);
demayer 0:6bf0743ece18 61 if ((_timeout == 0) || !(ret == NSAPI_ERROR_IN_PROGRESS || ret == NSAPI_ERROR_ALREADY)) {
demayer 0:6bf0743ece18 62 break;
demayer 0:6bf0743ece18 63 } else {
demayer 0:6bf0743ece18 64 blocking_connect_in_progress = true;
demayer 0:6bf0743ece18 65
demayer 0:6bf0743ece18 66 uint32_t flag;
demayer 0:6bf0743ece18 67
demayer 0:6bf0743ece18 68 // Release lock before blocking so other threads
demayer 0:6bf0743ece18 69 // accessing this object aren't blocked
demayer 0:6bf0743ece18 70 _lock.unlock();
demayer 0:6bf0743ece18 71 flag = _event_flag.wait_any(WRITE_FLAG, _timeout);
demayer 0:6bf0743ece18 72 _lock.lock();
demayer 0:6bf0743ece18 73 if (flag & osFlagsError) {
demayer 0:6bf0743ece18 74 // Timeout break
demayer 0:6bf0743ece18 75 break;
demayer 0:6bf0743ece18 76 }
demayer 0:6bf0743ece18 77 }
demayer 0:6bf0743ece18 78 }
demayer 0:6bf0743ece18 79
demayer 0:6bf0743ece18 80 _write_in_progress = false;
demayer 0:6bf0743ece18 81
demayer 0:6bf0743ece18 82 /* Non-blocking connect gives "EISCONN" once done - convert to OK for blocking mode if we became connected during this call */
demayer 0:6bf0743ece18 83 if (ret == NSAPI_ERROR_IS_CONNECTED && blocking_connect_in_progress) {
demayer 0:6bf0743ece18 84 ret = NSAPI_ERROR_OK;
demayer 0:6bf0743ece18 85 }
demayer 0:6bf0743ece18 86
demayer 0:6bf0743ece18 87 _lock.unlock();
demayer 0:6bf0743ece18 88 return ret;
demayer 0:6bf0743ece18 89 }
demayer 0:6bf0743ece18 90
demayer 0:6bf0743ece18 91 nsapi_error_t TCPSocket::connect(const char *host, uint16_t port)
demayer 0:6bf0743ece18 92 {
demayer 0:6bf0743ece18 93 SocketAddress address;
demayer 0:6bf0743ece18 94 nsapi_error_t err = _stack->gethostbyname(host, &address);
demayer 0:6bf0743ece18 95 if (err) {
demayer 0:6bf0743ece18 96 return NSAPI_ERROR_DNS_FAILURE;
demayer 0:6bf0743ece18 97 }
demayer 0:6bf0743ece18 98
demayer 0:6bf0743ece18 99 address.set_port(port);
demayer 0:6bf0743ece18 100
demayer 0:6bf0743ece18 101 // connect is thread safe
demayer 0:6bf0743ece18 102 return connect(address);
demayer 0:6bf0743ece18 103 }
demayer 0:6bf0743ece18 104
demayer 0:6bf0743ece18 105 nsapi_size_or_error_t TCPSocket::send(const void *data, nsapi_size_t size)
demayer 0:6bf0743ece18 106 {
demayer 0:6bf0743ece18 107 _lock.lock();
demayer 0:6bf0743ece18 108 nsapi_size_or_error_t ret;
demayer 0:6bf0743ece18 109
demayer 0:6bf0743ece18 110 // If this assert is hit then there are two threads
demayer 0:6bf0743ece18 111 // performing a send at the same time which is undefined
demayer 0:6bf0743ece18 112 // behavior
demayer 0:6bf0743ece18 113 MBED_ASSERT(!_write_in_progress);
demayer 0:6bf0743ece18 114 _write_in_progress = true;
demayer 0:6bf0743ece18 115
demayer 0:6bf0743ece18 116 while (true) {
demayer 0:6bf0743ece18 117 if (!_socket) {
demayer 0:6bf0743ece18 118 ret = NSAPI_ERROR_NO_SOCKET;
demayer 0:6bf0743ece18 119 break;
demayer 0:6bf0743ece18 120 }
demayer 0:6bf0743ece18 121
demayer 0:6bf0743ece18 122 _pending = 0;
demayer 0:6bf0743ece18 123 ret = _stack->socket_send(_socket, data, size);
demayer 0:6bf0743ece18 124 if ((_timeout == 0) || (ret != NSAPI_ERROR_WOULD_BLOCK)) {
demayer 0:6bf0743ece18 125 break;
demayer 0:6bf0743ece18 126 } else {
demayer 0:6bf0743ece18 127 uint32_t flag;
demayer 0:6bf0743ece18 128
demayer 0:6bf0743ece18 129 // Release lock before blocking so other threads
demayer 0:6bf0743ece18 130 // accessing this object aren't blocked
demayer 0:6bf0743ece18 131 _lock.unlock();
demayer 0:6bf0743ece18 132 flag = _event_flag.wait_any(WRITE_FLAG, _timeout);
demayer 0:6bf0743ece18 133 _lock.lock();
demayer 0:6bf0743ece18 134
demayer 0:6bf0743ece18 135 if (flag & osFlagsError) {
demayer 0:6bf0743ece18 136 // Timeout break
demayer 0:6bf0743ece18 137 ret = NSAPI_ERROR_WOULD_BLOCK;
demayer 0:6bf0743ece18 138 break;
demayer 0:6bf0743ece18 139 }
demayer 0:6bf0743ece18 140 }
demayer 0:6bf0743ece18 141 }
demayer 0:6bf0743ece18 142
demayer 0:6bf0743ece18 143 _write_in_progress = false;
demayer 0:6bf0743ece18 144 _lock.unlock();
demayer 0:6bf0743ece18 145 return ret;
demayer 0:6bf0743ece18 146 }
demayer 0:6bf0743ece18 147
demayer 0:6bf0743ece18 148 nsapi_size_or_error_t TCPSocket::recv(void *data, nsapi_size_t size)
demayer 0:6bf0743ece18 149 {
demayer 0:6bf0743ece18 150 _lock.lock();
demayer 0:6bf0743ece18 151 nsapi_size_or_error_t ret;
demayer 0:6bf0743ece18 152
demayer 0:6bf0743ece18 153 // If this assert is hit then there are two threads
demayer 0:6bf0743ece18 154 // performing a recv at the same time which is undefined
demayer 0:6bf0743ece18 155 // behavior
demayer 0:6bf0743ece18 156 MBED_ASSERT(!_read_in_progress);
demayer 0:6bf0743ece18 157 _read_in_progress = true;
demayer 0:6bf0743ece18 158
demayer 0:6bf0743ece18 159 while (true) {
demayer 0:6bf0743ece18 160 if (!_socket) {
demayer 0:6bf0743ece18 161 ret = NSAPI_ERROR_NO_SOCKET;
demayer 0:6bf0743ece18 162 break;
demayer 0:6bf0743ece18 163 }
demayer 0:6bf0743ece18 164
demayer 0:6bf0743ece18 165 _pending = 0;
demayer 0:6bf0743ece18 166 ret = _stack->socket_recv(_socket, data, size);
demayer 0:6bf0743ece18 167 if ((_timeout == 0) || (ret != NSAPI_ERROR_WOULD_BLOCK)) {
demayer 0:6bf0743ece18 168 break;
demayer 0:6bf0743ece18 169 } else {
demayer 0:6bf0743ece18 170 uint32_t flag;
demayer 0:6bf0743ece18 171
demayer 0:6bf0743ece18 172 // Release lock before blocking so other threads
demayer 0:6bf0743ece18 173 // accessing this object aren't blocked
demayer 0:6bf0743ece18 174 _lock.unlock();
demayer 0:6bf0743ece18 175 flag = _event_flag.wait_any(READ_FLAG, _timeout);
demayer 0:6bf0743ece18 176 _lock.lock();
demayer 0:6bf0743ece18 177
demayer 0:6bf0743ece18 178 if (flag & osFlagsError) {
demayer 0:6bf0743ece18 179 // Timeout break
demayer 0:6bf0743ece18 180 ret = NSAPI_ERROR_WOULD_BLOCK;
demayer 0:6bf0743ece18 181 break;
demayer 0:6bf0743ece18 182 }
demayer 0:6bf0743ece18 183 }
demayer 0:6bf0743ece18 184 }
demayer 0:6bf0743ece18 185
demayer 0:6bf0743ece18 186 _read_in_progress = false;
demayer 0:6bf0743ece18 187 _lock.unlock();
demayer 0:6bf0743ece18 188 return ret;
demayer 0:6bf0743ece18 189 }
demayer 0:6bf0743ece18 190
demayer 0:6bf0743ece18 191 void TCPSocket::event()
demayer 0:6bf0743ece18 192 {
demayer 0:6bf0743ece18 193 _event_flag.set(READ_FLAG|WRITE_FLAG);
demayer 0:6bf0743ece18 194
demayer 0:6bf0743ece18 195 _pending += 1;
demayer 0:6bf0743ece18 196 if (_callback && _pending == 1) {
demayer 0:6bf0743ece18 197 _callback();
demayer 0:6bf0743ece18 198 }
demayer 0:6bf0743ece18 199 }