Clément BENOIT / Socket

fort Socket

Fork of Socket by mbed official

UDPSocket.cpp@2:b227d242f3c7, 2012-07-23 (annotated)

Committer:
emilmont
Date:
Mon Jul 23 11:52:50 2012 +0000
Revision:
2:b227d242f3c7
Parent:
1:8080965f5d76
Child:
3:e6474399e057
tidyup

Who changed what in which revision?

UserRevisionLine numberNew contents of line
donatien 0:1f77255a22f5 1 /* Copyright (C) 2012 mbed.org, MIT License
donatien 0:1f77255a22f5 2 *
donatien 0:1f77255a22f5 3 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software
donatien 0:1f77255a22f5 4 * and associated documentation files (the "Software"), to deal in the Software without restriction,
donatien 0:1f77255a22f5 5 * including without limitation the rights to use, copy, modify, merge, publish, distribute,
donatien 0:1f77255a22f5 6 * sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
donatien 0:1f77255a22f5 7 * furnished to do so, subject to the following conditions:
donatien 0:1f77255a22f5 8 *
donatien 0:1f77255a22f5 9 * The above copyright notice and this permission notice shall be included in all copies or
donatien 0:1f77255a22f5 10 * substantial portions of the Software.
donatien 0:1f77255a22f5 11 *
donatien 0:1f77255a22f5 12 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
donatien 0:1f77255a22f5 13 * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
donatien 0:1f77255a22f5 14 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
donatien 0:1f77255a22f5 15 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
donatien 0:1f77255a22f5 16 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
donatien 0:1f77255a22f5 17 */
donatien 0:1f77255a22f5 18
donatien 0:1f77255a22f5 19 #include "UDPSocket.h"
donatien 0:1f77255a22f5 20
donatien 0:1f77255a22f5 21 #include <cstring>
donatien 0:1f77255a22f5 22
donatien 0:1f77255a22f5 23 using std::memset;
donatien 0:1f77255a22f5 24
donatien 0:1f77255a22f5 25 UDPSocket::UDPSocket() : m_sock(-1)
donatien 0:1f77255a22f5 26 {
donatien 0:1f77255a22f5 27 }
donatien 0:1f77255a22f5 28
donatien 0:1f77255a22f5 29 UDPSocket::~UDPSocket()
donatien 0:1f77255a22f5 30 {
donatien 0:1f77255a22f5 31 close(); //Don't want to leak
donatien 0:1f77255a22f5 32 }
donatien 0:1f77255a22f5 33
donatien 0:1f77255a22f5 34 int UDPSocket::bind(int port)
donatien 0:1f77255a22f5 35 {
donatien 0:1f77255a22f5 36 int ret = init();
donatien 0:1f77255a22f5 37 if( ret < 0 )
donatien 0:1f77255a22f5 38 {
donatien 0:1f77255a22f5 39 return -1;
donatien 0:1f77255a22f5 40 }
donatien 0:1f77255a22f5 41
donatien 0:1f77255a22f5 42 struct sockaddr_in localHost;
donatien 0:1f77255a22f5 43 std::memset(&localHost, 0, sizeof(localHost));
donatien 0:1f77255a22f5 44
donatien 0:1f77255a22f5 45 localHost.sin_family = AF_INET;
donatien 0:1f77255a22f5 46 localHost.sin_port = htons(port);
donatien 0:1f77255a22f5 47 localHost.sin_addr.s_addr = INADDR_ANY;
donatien 0:1f77255a22f5 48
donatien 0:1f77255a22f5 49 ret = ::bind(m_sock, (const struct sockaddr *)&localHost, sizeof(localHost));
donatien 0:1f77255a22f5 50 if (ret < 0)
donatien 0:1f77255a22f5 51 {
donatien 0:1f77255a22f5 52 close();
donatien 0:1f77255a22f5 53 return -1;
donatien 0:1f77255a22f5 54 }
donatien 0:1f77255a22f5 55
donatien 0:1f77255a22f5 56 return 0;
donatien 0:1f77255a22f5 57 }
donatien 0:1f77255a22f5 58
donatien 0:1f77255a22f5 59 // -1 if unsuccessful, else number of bytes written
donatien 1:8080965f5d76 60 int UDPSocket::sendTo(char* data, int length, char* host, int port, int timeout)
donatien 0:1f77255a22f5 61 {
donatien 0:1f77255a22f5 62 if( m_sock < 0 )
donatien 0:1f77255a22f5 63 {
donatien 0:1f77255a22f5 64 return -1;
donatien 0:1f77255a22f5 65 }
donatien 0:1f77255a22f5 66
donatien 0:1f77255a22f5 67 struct sockaddr_in remoteHost;
donatien 0:1f77255a22f5 68
donatien 0:1f77255a22f5 69 //Populate m_remoteHost
donatien 0:1f77255a22f5 70 std::memset(&remoteHost, 0, sizeof(struct sockaddr_in));
donatien 0:1f77255a22f5 71
donatien 0:1f77255a22f5 72 //Resolve DNS address or populate hard-coded IP address
donatien 0:1f77255a22f5 73 struct hostent *server = ::gethostbyname(host);
donatien 0:1f77255a22f5 74 if(server == NULL)
donatien 0:1f77255a22f5 75 {
donatien 0:1f77255a22f5 76 return -1; //Could not resolve address
donatien 0:1f77255a22f5 77 }
donatien 0:1f77255a22f5 78 std::memcpy((char*)&remoteHost.sin_addr.s_addr, (char*)server->h_addr_list[0], server->h_length);
donatien 0:1f77255a22f5 79
donatien 0:1f77255a22f5 80 remoteHost.sin_family = AF_INET;
donatien 0:1f77255a22f5 81 remoteHost.sin_port = htons(port);
donatien 0:1f77255a22f5 82
donatien 0:1f77255a22f5 83 size_t writtenLen = 0;
donatien 0:1f77255a22f5 84 struct timeval t_val; //t_val will be decremented on each call to select()
donatien 0:1f77255a22f5 85 t_val.tv_sec = timeout / 1000;
donatien 0:1f77255a22f5 86 t_val.tv_usec = (timeout - (t_val.tv_sec * 1000)) * 1000;
donatien 0:1f77255a22f5 87 while(writtenLen < length)
donatien 0:1f77255a22f5 88 {
donatien 0:1f77255a22f5 89 //Wait for socket to be writeable
donatien 0:1f77255a22f5 90 //Creating FS set
donatien 0:1f77255a22f5 91 fd_set socksSet;
donatien 0:1f77255a22f5 92 FD_ZERO(&socksSet);
donatien 0:1f77255a22f5 93 FD_SET(m_sock, &socksSet);
donatien 0:1f77255a22f5 94
donatien 0:1f77255a22f5 95 int ret = ::select(FD_SETSIZE, NULL, &socksSet, NULL, &t_val);
donatien 0:1f77255a22f5 96 if(ret <= 0 || !FD_ISSET(m_sock, &socksSet))
donatien 0:1f77255a22f5 97 {
donatien 0:1f77255a22f5 98 return writtenLen; //Timeout -- FIXME should we return -1 or writtenLength ?
donatien 0:1f77255a22f5 99 }
donatien 0:1f77255a22f5 100
donatien 0:1f77255a22f5 101 ret = ::sendto(m_sock, data + writtenLen, length - writtenLen, 0, (const struct sockaddr *)&remoteHost, sizeof(remoteHost));
donatien 0:1f77255a22f5 102 if( ret > 0)
donatien 0:1f77255a22f5 103 {
donatien 0:1f77255a22f5 104 writtenLen += ret;
donatien 0:1f77255a22f5 105 continue;
donatien 0:1f77255a22f5 106 }
donatien 0:1f77255a22f5 107 else if( ret == 0 )
donatien 0:1f77255a22f5 108 {
donatien 0:1f77255a22f5 109 return writtenLen; //Connection was closed by server -- FIXME how do we signal that the connection was closed ?
donatien 0:1f77255a22f5 110 }
donatien 0:1f77255a22f5 111 else
donatien 0:1f77255a22f5 112 {
donatien 0:1f77255a22f5 113 return -1; //Connnection error
donatien 0:1f77255a22f5 114 }
donatien 0:1f77255a22f5 115 }
donatien 0:1f77255a22f5 116
donatien 0:1f77255a22f5 117 return writtenLen;
donatien 0:1f77255a22f5 118 }
donatien 0:1f77255a22f5 119
donatien 0:1f77255a22f5 120 // -1 if unsuccessful, else number of bytes received
donatien 1:8080965f5d76 121 int UDPSocket::receiveFrom(char* data, int length, char** host, int* port, int timeout)
donatien 0:1f77255a22f5 122 {
donatien 0:1f77255a22f5 123 if( m_sock < 0 )
donatien 0:1f77255a22f5 124 {
donatien 0:1f77255a22f5 125 return -1;
donatien 0:1f77255a22f5 126 }
donatien 0:1f77255a22f5 127
donatien 0:1f77255a22f5 128 struct sockaddr_in remoteHost;
donatien 0:1f77255a22f5 129
donatien 0:1f77255a22f5 130 //Populate m_remoteHost
donatien 0:1f77255a22f5 131 std::memset(&remoteHost, 0, sizeof(struct sockaddr_in));
donatien 0:1f77255a22f5 132
donatien 0:1f77255a22f5 133 socklen_t remoteHostLen = sizeof(remoteHost);
donatien 0:1f77255a22f5 134
donatien 0:1f77255a22f5 135 size_t readLen = 0;
donatien 0:1f77255a22f5 136 struct timeval t_val; //t_val will be decremented on each call to select()
donatien 0:1f77255a22f5 137 t_val.tv_sec = timeout / 1000;
donatien 0:1f77255a22f5 138 t_val.tv_usec = (timeout - (t_val.tv_sec * 1000)) * 1000;
donatien 0:1f77255a22f5 139
donatien 0:1f77255a22f5 140 //No loop here as we don't want to mix packets from different sources
donatien 0:1f77255a22f5 141 do
donatien 0:1f77255a22f5 142 {
donatien 0:1f77255a22f5 143 //Wait for socket to be readable
donatien 0:1f77255a22f5 144 //Creating FS set
donatien 0:1f77255a22f5 145 fd_set socksSet;
donatien 0:1f77255a22f5 146 FD_ZERO(&socksSet);
donatien 0:1f77255a22f5 147 FD_SET(m_sock, &socksSet);
donatien 0:1f77255a22f5 148 int ret = ::select(FD_SETSIZE, &socksSet, NULL, NULL, &t_val);
donatien 0:1f77255a22f5 149 if(ret <= 0 || !FD_ISSET(m_sock, &socksSet))
donatien 0:1f77255a22f5 150 {
donatien 0:1f77255a22f5 151 break; //Timeout -- FIXME should we return -1 or writtenLength ?
donatien 0:1f77255a22f5 152 }
donatien 0:1f77255a22f5 153
donatien 0:1f77255a22f5 154 ret = ::recvfrom(m_sock, data + readLen, length - readLen, 0, (struct sockaddr*)&remoteHost, &remoteHostLen);
donatien 0:1f77255a22f5 155 if( ret > 0)
donatien 0:1f77255a22f5 156 {
donatien 0:1f77255a22f5 157 readLen += ret;
donatien 0:1f77255a22f5 158 }
donatien 0:1f77255a22f5 159 else if( ret == 0 )
donatien 0:1f77255a22f5 160 {
donatien 0:1f77255a22f5 161 //Connection was closed by server -- FIXME how do we signal that the connection was closed ?
donatien 0:1f77255a22f5 162 //Continue and populate address
donatien 0:1f77255a22f5 163 }
donatien 0:1f77255a22f5 164 else
donatien 0:1f77255a22f5 165 {
donatien 0:1f77255a22f5 166 return -1; //Connection error
donatien 0:1f77255a22f5 167 }
donatien 0:1f77255a22f5 168 } while(0);
donatien 0:1f77255a22f5 169
donatien 0:1f77255a22f5 170 static char hostBuf[16];
donatien 0:1f77255a22f5 171 inet_ntoa_r(remoteHost.sin_addr, hostBuf, sizeof(hostBuf));
donatien 0:1f77255a22f5 172
donatien 0:1f77255a22f5 173 *host = hostBuf;
donatien 0:1f77255a22f5 174 *port = ntohs(remoteHost.sin_port);
donatien 0:1f77255a22f5 175
donatien 0:1f77255a22f5 176 return readLen;
donatien 0:1f77255a22f5 177 }
donatien 0:1f77255a22f5 178
donatien 0:1f77255a22f5 179 int UDPSocket::close()
donatien 0:1f77255a22f5 180 {
donatien 0:1f77255a22f5 181 if( m_sock < 0 )
donatien 0:1f77255a22f5 182 {
donatien 0:1f77255a22f5 183 return -1;
donatien 0:1f77255a22f5 184 }
donatien 0:1f77255a22f5 185
donatien 0:1f77255a22f5 186 ::close(m_sock);
donatien 0:1f77255a22f5 187 m_sock = -1;
donatien 0:1f77255a22f5 188
donatien 0:1f77255a22f5 189 return 0;
donatien 0:1f77255a22f5 190 }
donatien 0:1f77255a22f5 191
donatien 0:1f77255a22f5 192 int UDPSocket::init()
donatien 0:1f77255a22f5 193 {
donatien 0:1f77255a22f5 194 if( m_sock != -1 )
donatien 0:1f77255a22f5 195 {
donatien 0:1f77255a22f5 196 return -1;
donatien 0:1f77255a22f5 197 }
donatien 0:1f77255a22f5 198 m_sock = ::socket(AF_INET, SOCK_DGRAM, 0); //UDP socket
donatien 0:1f77255a22f5 199 if (m_sock < 0)
donatien 0:1f77255a22f5 200 {
donatien 0:1f77255a22f5 201 return -1; //Could not create socket (Out of memory / available descriptors)
donatien 0:1f77255a22f5 202 }
donatien 0:1f77255a22f5 203 return 0;
donatien 0:1f77255a22f5 204 }