BA / SerialCom

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.

Fork of OmniWheels by Gustav Atmel

mbed-os/features/netsocket/TCPSocket.cpp@1:9c5af431a1f1, 2018-05-01 (annotated)

Committer:
gustavatmel
Date:
Tue May 01 15:47:08 2018 +0000
Revision:
1:9c5af431a1f1
sdf

Who changed what in which revision?

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