pefect / NetworkSocketAPI

ap

Dependencies:   DnsQuery

Dependents:   WizFi310_TCP_Echo_Server_Example

Fork of NetworkSocketAPI by NetworkSocketAPI

UDPSocket.cpp@117:ff83eb6a1ab9, 2016-04-20 (annotated)

Committer:
Christopher Haster
Date:
Wed Apr 20 19:48:17 2016 -0500
Revision:
117:ff83eb6a1ab9
Parent:
116:bc043343d753
Child:
118:96627c4b83d5
Fix bug in timeout logic

Who changed what in which revision?

UserRevisionLine numberNew contents of line
Christopher Haster 79:43a7e8c0d6cc 1 /* Socket
Christopher Haster 79:43a7e8c0d6cc 2 * Copyright (c) 2015 ARM Limited
Christopher Haster 79:43a7e8c0d6cc 3 *
Christopher Haster 79:43a7e8c0d6cc 4 * Licensed under the Apache License, Version 2.0 (the "License");
Christopher Haster 79:43a7e8c0d6cc 5 * you may not use this file except in compliance with the License.
Christopher Haster 79:43a7e8c0d6cc 6 * You may obtain a copy of the License at
Christopher Haster 79:43a7e8c0d6cc 7 *
Christopher Haster 79:43a7e8c0d6cc 8 * http://www.apache.org/licenses/LICENSE-2.0
Christopher Haster 79:43a7e8c0d6cc 9 *
Christopher Haster 79:43a7e8c0d6cc 10 * Unless required by applicable law or agreed to in writing, software
Christopher Haster 79:43a7e8c0d6cc 11 * distributed under the License is distributed on an "AS IS" BASIS,
Christopher Haster 79:43a7e8c0d6cc 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
Christopher Haster 79:43a7e8c0d6cc 13 * See the License for the specific language governing permissions and
Christopher Haster 79:43a7e8c0d6cc 14 * limitations under the License.
Christopher Haster 79:43a7e8c0d6cc 15 */
Christopher Haster 79:43a7e8c0d6cc 16
Christopher Haster 79:43a7e8c0d6cc 17 #include "UDPSocket.h"
Christopher Haster 82:97d166c4a193 18 #include "Timer.h"
Christopher Haster 79:43a7e8c0d6cc 19
Christopher Haster 90:0a988e4abb72 20 UDPSocket::UDPSocket()
Christopher Haster 90:0a988e4abb72 21 {
Christopher Haster 90:0a988e4abb72 22 }
Christopher Haster 90:0a988e4abb72 23
Christopher Haster 105:2fd212f8da61 24 UDPSocket::UDPSocket(NetworkStack *iface)
Christopher Haster 79:43a7e8c0d6cc 25 {
Christopher Haster 90:0a988e4abb72 26 open(iface);
Christopher Haster 90:0a988e4abb72 27 }
Christopher Haster 90:0a988e4abb72 28
Christopher Haster 105:2fd212f8da61 29 int UDPSocket::open(NetworkStack *iface)
Christopher Haster 90:0a988e4abb72 30 {
Christopher Haster 90:0a988e4abb72 31 return Socket::open(iface, NSAPI_UDP);
Christopher Haster 79:43a7e8c0d6cc 32 }
Christopher Haster 79:43a7e8c0d6cc 33
Christopher Haster 96:656011e49d9f 34 int UDPSocket::sendto(const char *host, uint16_t port, const void *data, unsigned size)
Christopher Haster 96:656011e49d9f 35 {
Christopher Haster 96:656011e49d9f 36 SocketAddress addr(_iface, host, port);
Christopher Haster 96:656011e49d9f 37 if (!addr) {
Christopher Haster 96:656011e49d9f 38 return NSAPI_ERROR_DNS_FAILURE;
Christopher Haster 96:656011e49d9f 39 }
Christopher Haster 96:656011e49d9f 40
Christopher Haster 96:656011e49d9f 41 return sendto(addr, data, size);
Christopher Haster 79:43a7e8c0d6cc 42 }
Christopher Haster 79:43a7e8c0d6cc 43
Christopher Haster 80:9c6673c93082 44 int UDPSocket::sendto(const SocketAddress &address, const void *data, unsigned size)
Christopher Haster 79:43a7e8c0d6cc 45 {
Christopher Haster 82:97d166c4a193 46 mbed::Timer timer;
Christopher Haster 82:97d166c4a193 47 timer.start();
Christopher Haster 82:97d166c4a193 48
Christopher Haster 82:97d166c4a193 49 while (true) {
Christopher Haster 82:97d166c4a193 50 if (!_socket) {
Christopher Haster 82:97d166c4a193 51 return NSAPI_ERROR_NO_SOCKET;
Christopher Haster 82:97d166c4a193 52 }
Christopher Haster 82:97d166c4a193 53
Christopher Haster 82:97d166c4a193 54 int sent = _iface->socket_sendto(_socket, address, data, size);
Christopher Haster 115:950b19eb0f02 55 if (sent != NSAPI_ERROR_WOULD_BLOCK
Christopher Haster 117:ff83eb6a1ab9 56 || (_timeout >= 0 && timer.read_ms() > _timeout)) {
Christopher Haster 82:97d166c4a193 57 return sent;
Christopher Haster 82:97d166c4a193 58 }
Christopher Haster 116:bc043343d753 59
Christopher Haster 116:bc043343d753 60 __WFI();
Christopher Haster 79:43a7e8c0d6cc 61 }
Christopher Haster 79:43a7e8c0d6cc 62 }
Christopher Haster 79:43a7e8c0d6cc 63
Christopher Haster 79:43a7e8c0d6cc 64 int UDPSocket::recvfrom(SocketAddress *address, void *buffer, unsigned size)
Christopher Haster 79:43a7e8c0d6cc 65 {
Christopher Haster 82:97d166c4a193 66 mbed::Timer timer;
Christopher Haster 82:97d166c4a193 67 timer.start();
Christopher Haster 82:97d166c4a193 68
Christopher Haster 82:97d166c4a193 69 while (true) {
Christopher Haster 82:97d166c4a193 70 if (!_socket) {
Christopher Haster 82:97d166c4a193 71 return NSAPI_ERROR_NO_SOCKET;
Christopher Haster 82:97d166c4a193 72 }
Christopher Haster 82:97d166c4a193 73
Christopher Haster 82:97d166c4a193 74 int recv = _iface->socket_recvfrom(_socket, address, buffer, size);
Christopher Haster 115:950b19eb0f02 75 if (recv != NSAPI_ERROR_WOULD_BLOCK
Christopher Haster 117:ff83eb6a1ab9 76 || (_timeout >= 0 && timer.read_ms() > _timeout)) {
Christopher Haster 82:97d166c4a193 77 return recv;
Christopher Haster 82:97d166c4a193 78 }
Christopher Haster 116:bc043343d753 79
Christopher Haster 116:bc043343d753 80 __WFI();
Christopher Haster 79:43a7e8c0d6cc 81 }
Christopher Haster 79:43a7e8c0d6cc 82 }