EL4121 Embedded System / mbed-os

mbed-os

Dependents:   cobaLCDJoyMotor_Thread odometry_omni_3roda_v3 odometry_omni_3roda_v1 odometry_omni_3roda_v2 ... more

features/netsocket/TCPSocket.cpp@0:b74591d5ab33, 2017-12-11 (annotated)

Committer:
be_bryan
Date:
Mon Dec 11 17:54:04 2017 +0000
Revision:
0:b74591d5ab33
motor ++

Who changed what in which revision?

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