UDPSocket.cpp

00001 /* Copyright (C) 2012 mbed.org, MIT License
00002  *
00003  * Permission is hereby granted, free of charge, to any person obtaining a copy of this software
00004  * and associated documentation files (the "Software"), to deal in the Software without restriction,
00005  * including without limitation the rights to use, copy, modify, merge, publish, distribute,
00006  * sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
00007  * furnished to do so, subject to the following conditions:
00008  *
00009  * The above copyright notice and this permission notice shall be included in all copies or
00010  * substantial portions of the Software.
00011  *
00012  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
00013  * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
00014  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
00015  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
00016  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
00017  */
00018 #if LWIP_UDP
00019 
00020 #include "Socket/UDPSocket.h"
00021 
00022 #include <cstring>
00023 
00024 using std::memset;
00025 
00026 UDPSocket::UDPSocket() {
00027 }
00028 
00029 int UDPSocket::init(void) {
00030     return init_socket(SOCK_DGRAM);
00031 }
00032 
00033 // Server initialization
00034 int UDPSocket::bind(int port) {
00035     if (init_socket(SOCK_DGRAM) < 0)
00036         return -1;
00037     
00038     struct sockaddr_in localHost;
00039     std::memset(&localHost, 0, sizeof(localHost));
00040     
00041     localHost.sin_family = AF_INET;
00042     localHost.sin_port = htons(port);
00043     localHost.sin_addr.s_addr = INADDR_ANY;
00044     
00045     if (lwip_bind(_sock_fd, (const struct sockaddr *) &localHost, sizeof(localHost)) < 0) {
00046         close();
00047         return -1;
00048     }
00049     
00050     return 0;
00051 }
00052 
00053 int UDPSocket::join_multicast_group(const char* address) {
00054     struct ip_mreq mreq;
00055     
00056     // Set up group address 
00057     mreq.imr_multiaddr.s_addr = inet_addr(address);
00058     mreq.imr_interface.s_addr = htonl(INADDR_ANY);
00059     
00060     return set_option(IPPROTO_IP, IP_ADD_MEMBERSHIP, &mreq, sizeof(mreq));
00061 }
00062 
00063 int UDPSocket::set_broadcasting(bool broadcast) {
00064     int option = (broadcast) ? (1) : (0);
00065     return set_option(SOL_SOCKET, SO_BROADCAST, &option, sizeof(option));
00066 }
00067 
00068 // -1 if unsuccessful, else number of bytes written
00069 int UDPSocket::sendTo(Endpoint &remote, char *packet, int length) {
00070     if (_sock_fd < 0)
00071         return -1;
00072     
00073     if (!_blocking) {
00074         TimeInterval timeout(_timeout);
00075         if (wait_writable(timeout) != 0)
00076             return 0;
00077     }
00078     
00079     return lwip_sendto(_sock_fd, packet, length, 0, (const struct sockaddr *) &remote._remoteHost, sizeof(remote._remoteHost));
00080 }
00081 
00082 // -1 if unsuccessful, else number of bytes received
00083 int UDPSocket::receiveFrom(Endpoint &remote, char *buffer, int length) {
00084     if (_sock_fd < 0)
00085         return -1;
00086     
00087     if (!_blocking) {
00088         TimeInterval timeout(_timeout);
00089         if (wait_readable(timeout) != 0)
00090             return 0;
00091     }
00092     remote.reset_address();
00093     socklen_t remoteHostLen = sizeof(remote._remoteHost);
00094     return lwip_recvfrom(_sock_fd, buffer, length, 0, (struct sockaddr*) &remote._remoteHost, &remoteHostLen);
00095 }
00096 
00097 #endif