Nicolas Borla / Mbed OS ROME2_Robot_Firmware
Committer:
boro
Date:
Tue Mar 09 13:10:40 2021 +0000
Revision:
3:6fe17b8a6d62
Parent:
0:4beb2ea291ec
SDBlockDevice added

Who changed what in which revision?

UserRevisionLine numberNew contents of line
boro 0:4beb2ea291ec 1 /* Socket
boro 0:4beb2ea291ec 2 * Copyright (c) 2015 ARM Limited
boro 0:4beb2ea291ec 3 *
boro 0:4beb2ea291ec 4 * Licensed under the Apache License, Version 2.0 (the "License");
boro 0:4beb2ea291ec 5 * you may not use this file except in compliance with the License.
boro 0:4beb2ea291ec 6 * You may obtain a copy of the License at
boro 0:4beb2ea291ec 7 *
boro 0:4beb2ea291ec 8 * http://www.apache.org/licenses/LICENSE-2.0
boro 0:4beb2ea291ec 9 *
boro 0:4beb2ea291ec 10 * Unless required by applicable law or agreed to in writing, software
boro 0:4beb2ea291ec 11 * distributed under the License is distributed on an "AS IS" BASIS,
boro 0:4beb2ea291ec 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
boro 0:4beb2ea291ec 13 * See the License for the specific language governing permissions and
boro 0:4beb2ea291ec 14 * limitations under the License.
boro 0:4beb2ea291ec 15 */
boro 0:4beb2ea291ec 16
boro 0:4beb2ea291ec 17 #include "TCPServer.h"
boro 0:4beb2ea291ec 18 #include "mbed.h"
boro 0:4beb2ea291ec 19
boro 0:4beb2ea291ec 20 TCPServer::TCPServer()
boro 0:4beb2ea291ec 21 : _pending(0), _accept_sem(0)
boro 0:4beb2ea291ec 22 {
boro 0:4beb2ea291ec 23 }
boro 0:4beb2ea291ec 24
boro 0:4beb2ea291ec 25 TCPServer::~TCPServer()
boro 0:4beb2ea291ec 26 {
boro 0:4beb2ea291ec 27 close();
boro 0:4beb2ea291ec 28 }
boro 0:4beb2ea291ec 29
boro 0:4beb2ea291ec 30 nsapi_protocol_t TCPServer::get_proto()
boro 0:4beb2ea291ec 31 {
boro 0:4beb2ea291ec 32 return NSAPI_TCP;
boro 0:4beb2ea291ec 33 }
boro 0:4beb2ea291ec 34
boro 0:4beb2ea291ec 35 nsapi_error_t TCPServer::listen(int backlog)
boro 0:4beb2ea291ec 36 {
boro 0:4beb2ea291ec 37 _lock.lock();
boro 0:4beb2ea291ec 38 nsapi_error_t ret;
boro 0:4beb2ea291ec 39
boro 0:4beb2ea291ec 40 if (!_socket) {
boro 0:4beb2ea291ec 41 ret = NSAPI_ERROR_NO_SOCKET;
boro 0:4beb2ea291ec 42 } else {
boro 0:4beb2ea291ec 43 ret = _stack->socket_listen(_socket, backlog);
boro 0:4beb2ea291ec 44 }
boro 0:4beb2ea291ec 45
boro 0:4beb2ea291ec 46 _lock.unlock();
boro 0:4beb2ea291ec 47 return ret;
boro 0:4beb2ea291ec 48 }
boro 0:4beb2ea291ec 49
boro 0:4beb2ea291ec 50 nsapi_error_t TCPServer::accept(TCPSocket *connection, SocketAddress *address)
boro 0:4beb2ea291ec 51 {
boro 0:4beb2ea291ec 52 _lock.lock();
boro 0:4beb2ea291ec 53 nsapi_error_t ret;
boro 0:4beb2ea291ec 54
boro 0:4beb2ea291ec 55 while (true) {
boro 0:4beb2ea291ec 56 if (!_socket) {
boro 0:4beb2ea291ec 57 ret = NSAPI_ERROR_NO_SOCKET;
boro 0:4beb2ea291ec 58 break;
boro 0:4beb2ea291ec 59 }
boro 0:4beb2ea291ec 60
boro 0:4beb2ea291ec 61 _pending = 0;
boro 0:4beb2ea291ec 62 void *socket;
boro 0:4beb2ea291ec 63 ret = _stack->socket_accept(_socket, &socket, address);
boro 0:4beb2ea291ec 64
boro 0:4beb2ea291ec 65 if (0 == ret) {
boro 0:4beb2ea291ec 66 connection->_lock.lock();
boro 0:4beb2ea291ec 67
boro 0:4beb2ea291ec 68 if (connection->_socket) {
boro 0:4beb2ea291ec 69 connection->close();
boro 0:4beb2ea291ec 70 }
boro 0:4beb2ea291ec 71
boro 0:4beb2ea291ec 72 connection->_stack = _stack;
boro 0:4beb2ea291ec 73 connection->_socket = socket;
boro 0:4beb2ea291ec 74 connection->_event = Callback<void()>(connection, &TCPSocket::event);
boro 0:4beb2ea291ec 75 _stack->socket_attach(socket, &Callback<void()>::thunk, &connection->_event);
boro 0:4beb2ea291ec 76
boro 0:4beb2ea291ec 77 connection->_lock.unlock();
boro 0:4beb2ea291ec 78 break;
boro 0:4beb2ea291ec 79 } else if (NSAPI_ERROR_WOULD_BLOCK != ret) {
boro 0:4beb2ea291ec 80 break;
boro 0:4beb2ea291ec 81 } else {
boro 0:4beb2ea291ec 82 int32_t count;
boro 0:4beb2ea291ec 83
boro 0:4beb2ea291ec 84 // Release lock before blocking so other threads
boro 0:4beb2ea291ec 85 // accessing this object aren't blocked
boro 0:4beb2ea291ec 86 _lock.unlock();
boro 0:4beb2ea291ec 87 count = _accept_sem.wait(_timeout);
boro 0:4beb2ea291ec 88 _lock.lock();
boro 0:4beb2ea291ec 89
boro 0:4beb2ea291ec 90 if (count < 1) {
boro 0:4beb2ea291ec 91 // Semaphore wait timed out so break out and return
boro 0:4beb2ea291ec 92 ret = NSAPI_ERROR_WOULD_BLOCK;
boro 0:4beb2ea291ec 93 break;
boro 0:4beb2ea291ec 94 }
boro 0:4beb2ea291ec 95 }
boro 0:4beb2ea291ec 96 }
boro 0:4beb2ea291ec 97
boro 0:4beb2ea291ec 98 _lock.unlock();
boro 0:4beb2ea291ec 99 return ret;
boro 0:4beb2ea291ec 100 }
boro 0:4beb2ea291ec 101
boro 0:4beb2ea291ec 102 void TCPServer::event()
boro 0:4beb2ea291ec 103 {
boro 0:4beb2ea291ec 104 int32_t acount = _accept_sem.wait(0);
boro 0:4beb2ea291ec 105 if (acount <= 1) {
boro 0:4beb2ea291ec 106 _accept_sem.release();
boro 0:4beb2ea291ec 107 }
boro 0:4beb2ea291ec 108
boro 0:4beb2ea291ec 109 _pending += 1;
boro 0:4beb2ea291ec 110 if (_callback && _pending == 1) {
boro 0:4beb2ea291ec 111 _callback();
boro 0:4beb2ea291ec 112 }
boro 0:4beb2ea291ec 113 }