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/TCPServer.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 "TCPServer.h"
be_bryan 0:b74591d5ab33 18 #include "mbed.h"
be_bryan 0:b74591d5ab33 19
be_bryan 0:b74591d5ab33 20 TCPServer::TCPServer()
be_bryan 0:b74591d5ab33 21 : _pending(0), _accept_sem(0)
be_bryan 0:b74591d5ab33 22 {
be_bryan 0:b74591d5ab33 23 }
be_bryan 0:b74591d5ab33 24
be_bryan 0:b74591d5ab33 25 TCPServer::~TCPServer()
be_bryan 0:b74591d5ab33 26 {
be_bryan 0:b74591d5ab33 27 close();
be_bryan 0:b74591d5ab33 28 }
be_bryan 0:b74591d5ab33 29
be_bryan 0:b74591d5ab33 30 nsapi_protocol_t TCPServer::get_proto()
be_bryan 0:b74591d5ab33 31 {
be_bryan 0:b74591d5ab33 32 return NSAPI_TCP;
be_bryan 0:b74591d5ab33 33 }
be_bryan 0:b74591d5ab33 34
be_bryan 0:b74591d5ab33 35 nsapi_error_t TCPServer::listen(int backlog)
be_bryan 0:b74591d5ab33 36 {
be_bryan 0:b74591d5ab33 37 _lock.lock();
be_bryan 0:b74591d5ab33 38 nsapi_error_t ret;
be_bryan 0:b74591d5ab33 39
be_bryan 0:b74591d5ab33 40 if (!_socket) {
be_bryan 0:b74591d5ab33 41 ret = NSAPI_ERROR_NO_SOCKET;
be_bryan 0:b74591d5ab33 42 } else {
be_bryan 0:b74591d5ab33 43 ret = _stack->socket_listen(_socket, backlog);
be_bryan 0:b74591d5ab33 44 }
be_bryan 0:b74591d5ab33 45
be_bryan 0:b74591d5ab33 46 _lock.unlock();
be_bryan 0:b74591d5ab33 47 return ret;
be_bryan 0:b74591d5ab33 48 }
be_bryan 0:b74591d5ab33 49
be_bryan 0:b74591d5ab33 50 nsapi_error_t TCPServer::accept(TCPSocket *connection, SocketAddress *address)
be_bryan 0:b74591d5ab33 51 {
be_bryan 0:b74591d5ab33 52 _lock.lock();
be_bryan 0:b74591d5ab33 53 nsapi_error_t ret;
be_bryan 0:b74591d5ab33 54
be_bryan 0:b74591d5ab33 55 while (true) {
be_bryan 0:b74591d5ab33 56 if (!_socket) {
be_bryan 0:b74591d5ab33 57 ret = NSAPI_ERROR_NO_SOCKET;
be_bryan 0:b74591d5ab33 58 break;
be_bryan 0:b74591d5ab33 59 }
be_bryan 0:b74591d5ab33 60
be_bryan 0:b74591d5ab33 61 _pending = 0;
be_bryan 0:b74591d5ab33 62 void *socket;
be_bryan 0:b74591d5ab33 63 ret = _stack->socket_accept(_socket, &socket, address);
be_bryan 0:b74591d5ab33 64
be_bryan 0:b74591d5ab33 65 if (0 == ret) {
be_bryan 0:b74591d5ab33 66 connection->_lock.lock();
be_bryan 0:b74591d5ab33 67
be_bryan 0:b74591d5ab33 68 if (connection->_socket) {
be_bryan 0:b74591d5ab33 69 connection->close();
be_bryan 0:b74591d5ab33 70 }
be_bryan 0:b74591d5ab33 71
be_bryan 0:b74591d5ab33 72 connection->_stack = _stack;
be_bryan 0:b74591d5ab33 73 connection->_socket = socket;
be_bryan 0:b74591d5ab33 74 connection->_event = Callback<void()>(connection, &TCPSocket::event);
be_bryan 0:b74591d5ab33 75 _stack->socket_attach(socket, &Callback<void()>::thunk, &connection->_event);
be_bryan 0:b74591d5ab33 76
be_bryan 0:b74591d5ab33 77 connection->_lock.unlock();
be_bryan 0:b74591d5ab33 78 break;
be_bryan 0:b74591d5ab33 79 } else if (NSAPI_ERROR_WOULD_BLOCK != ret) {
be_bryan 0:b74591d5ab33 80 break;
be_bryan 0:b74591d5ab33 81 } else {
be_bryan 0:b74591d5ab33 82 int32_t count;
be_bryan 0:b74591d5ab33 83
be_bryan 0:b74591d5ab33 84 // Release lock before blocking so other threads
be_bryan 0:b74591d5ab33 85 // accessing this object aren't blocked
be_bryan 0:b74591d5ab33 86 _lock.unlock();
be_bryan 0:b74591d5ab33 87 count = _accept_sem.wait(_timeout);
be_bryan 0:b74591d5ab33 88 _lock.lock();
be_bryan 0:b74591d5ab33 89
be_bryan 0:b74591d5ab33 90 if (count < 1) {
be_bryan 0:b74591d5ab33 91 // Semaphore wait timed out so break out and return
be_bryan 0:b74591d5ab33 92 ret = NSAPI_ERROR_WOULD_BLOCK;
be_bryan 0:b74591d5ab33 93 break;
be_bryan 0:b74591d5ab33 94 }
be_bryan 0:b74591d5ab33 95 }
be_bryan 0:b74591d5ab33 96 }
be_bryan 0:b74591d5ab33 97
be_bryan 0:b74591d5ab33 98 _lock.unlock();
be_bryan 0:b74591d5ab33 99 return ret;
be_bryan 0:b74591d5ab33 100 }
be_bryan 0:b74591d5ab33 101
be_bryan 0:b74591d5ab33 102 void TCPServer::event()
be_bryan 0:b74591d5ab33 103 {
be_bryan 0:b74591d5ab33 104 int32_t acount = _accept_sem.wait(0);
be_bryan 0:b74591d5ab33 105 if (acount <= 1) {
be_bryan 0:b74591d5ab33 106 _accept_sem.release();
be_bryan 0:b74591d5ab33 107 }
be_bryan 0:b74591d5ab33 108
be_bryan 0:b74591d5ab33 109 _pending += 1;
be_bryan 0:b74591d5ab33 110 if (_callback && _pending == 1) {
be_bryan 0:b74591d5ab33 111 _callback();
be_bryan 0:b74591d5ab33 112 }
be_bryan 0:b74591d5ab33 113 }