BA / Mbed OS BaBoRo_test2

mbed-os/features/netsocket/TCPServer.cpp@0:02dd72d1d465, 2018-04-24 (annotated)

Committer:
borlanic
Date:
Tue Apr 24 11:45:18 2018 +0000
Revision:
0:02dd72d1d465
BaBoRo_test2 - backup 1

Who changed what in which revision?

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