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?
User | Revision | Line number | New 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 | } |