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?
User | Revision | Line number | New 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 | } |