Marco Zecchini / Mbed OS Example_RTOS

Rtos API example

mbed-os/features/netsocket/TCPServer.cpp@0:9fca2b23d0ba, 2019-02-23 (annotated)

Committer:
marcozecchini
Date:
Sat Feb 23 12:13:36 2019 +0000
Revision:
0:9fca2b23d0ba
final commit

Who changed what in which revision?

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