3 Copyright (C) 2010 celeron55, Perttu Ahola <celeron55@gmail.com>
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation; either version 2 of the License, or
8 (at your option) any later version.
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
15 You should have received a copy of the GNU General Public License along
16 with this program; if not, write to the Free Software Foundation, Inc.,
17 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
28 // Debug printing options
29 // Set to 1 for debug output
31 // This is prepended to everything printed here
34 bool g_sockets_initialized = false;
40 if(WSAStartup( MAKEWORD(2,2), &WsaData ) != NO_ERROR)
41 throw SocketException("WSAStartup failed");
44 g_sockets_initialized = true;
47 void sockets_cleanup()
60 Address::Address(unsigned int address, unsigned short port)
66 Address::Address(unsigned int a, unsigned int b,
67 unsigned int c, unsigned int d,
70 m_address = (a<<24) | (b<<16) | ( c<<8) | d;
74 bool Address::operator==(Address &address)
76 return (m_address == address.m_address
77 && m_port == address.m_port);
80 bool Address::operator!=(Address &address)
82 return !(*this == address);
85 void Address::Resolve(const char *name)
87 struct addrinfo *resolved;
88 int e = getaddrinfo(name, NULL, NULL, &resolved);
90 throw ResolveError("");
92 FIXME: This is an ugly hack; change the whole class
93 to store the address as sockaddr
95 struct sockaddr_in *t = (struct sockaddr_in*)resolved->ai_addr;
96 m_address = ntohl(t->sin_addr.s_addr);
97 freeaddrinfo(resolved);
100 unsigned int Address::getAddress() const
105 unsigned short Address::getPort() const
110 void Address::setAddress(unsigned int address)
115 void Address::setAddress(unsigned int a, unsigned int b,
116 unsigned int c, unsigned int d)
118 m_address = (a<<24) | (b<<16) | ( c<<8) | d;
121 void Address::setPort(unsigned short port)
126 void Address::print(std::ostream *s) const
128 (*s)<<((m_address>>24)&0xff)<<"."
129 <<((m_address>>16)&0xff)<<"."
130 <<((m_address>>8)&0xff)<<"."
131 <<((m_address>>0)&0xff)<<":"
135 void Address::print() const
140 UDPSocket::UDPSocket()
142 if(g_sockets_initialized == false)
143 throw SocketException("Sockets not initialized");
145 m_handle = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
148 dstream<<DPS<<"UDPSocket("<<(int)m_handle<<")::UDPSocket()"<<std::endl;
152 throw SocketException("Failed to create socket");
156 DWORD nonblocking = 0;
157 if(ioctlsocket(m_handle, FIONBIO, &nonblocking) != 0)
159 throw SocketException("Failed set non-blocking mode");
163 if(fcntl(m_handle, F_SETFL, O_NONBLOCK, nonblocking) == -1)
165 throw SocketException("Failed set non-blocking mode");
172 UDPSocket::~UDPSocket()
175 dstream<<DPS<<"UDPSocket("<<(int)m_handle<<")::~UDPSocket()"<<std::endl;
178 closesocket(m_handle);
184 void UDPSocket::Bind(unsigned short port)
187 dstream<<DPS<<"UDPSocket("<<(int)m_handle
188 <<")::Bind(): port="<<port<<std::endl;
191 address.sin_family = AF_INET;
192 address.sin_addr.s_addr = INADDR_ANY;
193 address.sin_port = htons(port);
195 if(bind(m_handle, (const sockaddr*)&address, sizeof(sockaddr_in)) < 0)
197 #ifndef DISABLE_ERRNO
198 dstream<<(int)m_handle<<": Bind failed: "<<strerror(errno)<<std::endl;
200 throw SocketException("Failed to bind socket");
204 void UDPSocket::Send(const Address & destination, const void * data, int size)
206 bool dumping_packet = false;
207 if(INTERNET_SIMULATOR)
208 dumping_packet = (myrand()%10==0); //easy
209 //dumping_packet = (myrand()%4==0); // hard
212 /*dstream<<DPS<<"UDPSocket("<<(int)m_handle
213 <<")::Send(): destination=";*/
215 dstream<<(int)m_handle<<" -> ";
217 dstream<<", size="<<size<<", data=";
218 for(int i=0; i<size && i<20; i++){
219 if(i%2==0) DEBUGPRINT(" ");
220 DEBUGPRINT("%.2X", ((int)((const char*)data)[i])&0xff);
225 dstream<<" (DUMPED BY INTERNET_SIMULATOR)";
228 else if(dumping_packet)
230 // Lol let's forget it
231 dstream<<"UDPSocket::Send(): "
232 "INTERNET_SIMULATOR: dumping packet."
240 address.sin_family = AF_INET;
241 address.sin_addr.s_addr = htonl(destination.getAddress());
242 address.sin_port = htons(destination.getPort());
244 int sent = sendto(m_handle, (const char*)data, size,
245 0, (sockaddr*)&address, sizeof(sockaddr_in));
249 throw SendFailedException("Failed to send packet");
253 int UDPSocket::Receive(Address & sender, void * data, int size)
255 if(WaitData(m_timeout_ms) == false)
261 socklen_t address_len = sizeof(address);
263 int received = recvfrom(m_handle, (char*)data,
264 size, 0, (sockaddr*)&address, &address_len);
269 unsigned int address_ip = ntohl(address.sin_addr.s_addr);
270 unsigned int address_port = ntohs(address.sin_port);
272 sender = Address(address_ip, address_port);
275 //dstream<<DPS<<"UDPSocket("<<(int)m_handle<<")::Receive(): sender=";
276 dstream<<DPS<<(int)m_handle<<" <- ";
278 //dstream<<", received="<<received<<std::endl;
279 dstream<<", size="<<received<<", data=";
280 for(int i=0; i<received && i<20; i++){
281 if(i%2==0) DEBUGPRINT(" ");
282 DEBUGPRINT("%.2X", ((int)((const char*)data)[i])&0xff);
292 int UDPSocket::GetHandle()
297 void UDPSocket::setTimeoutMs(int timeout_ms)
299 m_timeout_ms = timeout_ms;
302 bool UDPSocket::WaitData(int timeout_ms)
307 // Initialize the set
309 FD_SET(m_handle, &readset);
311 // Initialize time out struct
314 tv.tv_usec = timeout_ms * 1000;
316 result = select(m_handle+1, &readset, NULL, NULL, &tv);
320 /*dstream<<"Select timed out (timeout_ms="
321 <<timeout_ms<<")"<<std::endl;*/
326 #ifndef DISABLE_ERRNO
327 dstream<<(int)m_handle<<": Select failed: "<<strerror(errno)<<std::endl;
330 int e = WSAGetLastError();
331 dstream<<(int)m_handle<<": WSAGetLastError()="<<e<<std::endl;
332 if(e == 10004 /*=WSAEINTR*/)
334 dstream<<"WARNING: Ignoring WSAEINTR."<<std::endl;
338 throw SocketException("Select failed");
340 else if(FD_ISSET(m_handle, &readset) == false){
342 //dstream<<"Select reported no data in m_handle"<<std::endl;
347 //dstream<<"Select reported data in m_handle"<<std::endl;