BA / SerialCom

Fork of OmniWheels by Gustav Atmel

Committer:
gustavatmel
Date:
Tue May 01 15:47:08 2018 +0000
Revision:
1:9c5af431a1f1
sdf

Who changed what in which revision?

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