01a1d30192ad383a4ae913c01de71ab94b5a2a38
[xonotic/darkplaces.git] / net_udp.c
1 /*
2 Copyright (C) 1996-1997 Id Software, Inc.
3
4 This program is free software; you can redistribute it and/or
5 modify it under the terms of the GNU General Public License
6 as published by the Free Software Foundation; either version 2
7 of the License, or (at your option) any later version.
8
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
12
13 See the GNU General Public License for more details.
14
15 You should have received a copy of the GNU General Public License
16 along with this program; if not, write to the Free Software
17 Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
18
19 */
20 #include "quakedef.h"
21 #include "net_udp.h"
22 #ifdef WIN32
23 #include "winquake.h"
24 #define MAXHOSTNAMELEN          256
25 #else
26 #include <unistd.h>
27 #include <sys/types.h>
28 #include <sys/socket.h>
29 #include <netinet/in.h>
30 #include <arpa/inet.h>
31 #include <netdb.h>
32 #include <sys/param.h>
33 #include <sys/ioctl.h>
34 #include <errno.h>
35
36 #ifdef __sun__
37 #include <sys/filio.h>
38 #endif
39
40 #ifdef NeXT
41 #include <libc.h>
42 #endif
43 #endif
44
45 static int net_acceptsocket = -1;               // socket for fielding new connections
46 static int net_controlsocket;
47 static int net_broadcastsocket = 0;
48 static struct qsockaddr broadcastaddr;
49
50 static union {int i;unsigned char d[4];} myAddr;
51
52 //=============================================================================
53
54 #ifdef WIN32
55 WSADATA         winsockdata;
56 #endif
57
58 int UDP_Init (void)
59 {
60         int i;
61         struct hostent *local = NULL;
62         char buff[MAXHOSTNAMELEN];
63
64         if (COM_CheckParm ("-noudp"))
65                 return -1;
66
67 #ifdef WIN32
68         if (WSAStartup (MAKEWORD(1, 1), &winsockdata))
69         {
70                 Con_SafePrintf ("Winsock initialization failed.\n");
71                 return -1;
72         }
73 #endif
74
75         // loopback as a worst case fallback
76         myAddr.d[0] = 127;myAddr.d[1] = 0;myAddr.d[2] = 0;myAddr.d[3] = 1;
77
78         if ((i = COM_CheckParm("-ip")) != 0 && i < com_argc)
79         {
80                 myAddr.i = inet_addr(com_argv[i+1]);
81                 Con_Printf("Binding to IP Interface Address of %i.%i.%i.%i\n", myAddr.d[0], myAddr.d[1], myAddr.d[2], myAddr.d[3]);
82         }
83         else if (gethostname(buff, MAXHOSTNAMELEN) != -1)
84         {
85                 buff[MAXHOSTNAMELEN - 1] = 0;
86                 local = gethostbyname(buff);
87                 if (local != NULL)
88                         myAddr.i = *((int *)local->h_addr_list[0]);
89         }
90
91         sprintf(my_tcpip_address, "%d.%d.%d.%d", myAddr.d[0], myAddr.d[1], myAddr.d[2], myAddr.d[3]);
92
93         if ((net_controlsocket = UDP_OpenSocket (0)) == -1)
94         {
95                 Con_Printf("UDP_Init: Unable to open control socket\n");
96 #ifdef WIN32
97                 WSACleanup ();
98 #endif
99                 return -1;
100         }
101
102         ((struct sockaddr_in *)&broadcastaddr)->sin_family = AF_INET;
103         ((struct sockaddr_in *)&broadcastaddr)->sin_addr.s_addr = htonl(INADDR_BROADCAST);
104         ((struct sockaddr_in *)&broadcastaddr)->sin_port = htons((unsigned short)net_hostport);
105
106         Con_Printf("UDP Initialized\n");
107         tcpipAvailable = true;
108
109         return net_controlsocket;
110 }
111
112 //=============================================================================
113
114 void UDP_Shutdown (void)
115 {
116         UDP_Listen (false);
117         UDP_CloseSocket (net_controlsocket);
118 #ifdef WIN32
119         WSACleanup ();
120 #endif
121 }
122
123 //=============================================================================
124
125 void UDP_Listen (qboolean state)
126 {
127         // enable listening
128         if (state)
129         {
130                 if (net_acceptsocket != -1)
131                         return;
132                 if ((net_acceptsocket = UDP_OpenSocket (net_hostport)) == -1)
133                         Sys_Error ("UDP_Listen: Unable to open accept socket\n");
134                 return;
135         }
136
137         // disable listening
138         if (net_acceptsocket == -1)
139                 return;
140         UDP_CloseSocket (net_acceptsocket);
141         net_acceptsocket = -1;
142 }
143
144 //=============================================================================
145
146 int UDP_OpenSocket (int port)
147 {
148         int newsocket;
149         struct sockaddr_in address;
150
151         if ((newsocket = socket (PF_INET, SOCK_DGRAM, IPPROTO_UDP)) == -1)
152                 return -1;
153
154         {
155 #ifdef WIN32
156                 u_long _true = 1;
157                 if (ioctlsocket (newsocket, FIONBIO, &_true) == -1)
158                 {
159                         closesocket (newsocket);
160 #else
161                 char _true = 1;
162                 if (ioctl (newsocket, FIONBIO, &_true) == -1)
163                 {
164                         close (newsocket);
165 #endif
166                         Sys_Error("UDP_OpenSocket: unable to do a ioctl FIONBIO on the socket\n");
167                 }
168         }
169
170         address.sin_family = AF_INET;
171         address.sin_addr.s_addr = myAddr.i;
172         address.sin_port = htons((unsigned short)port);
173         if (bind(newsocket, (void *)&address, sizeof(address)) == -1)
174         {
175 #ifdef WIN32
176                 closesocket(newsocket);
177 #else
178                 close(newsocket);
179 #endif
180                 Sys_Error ("UDP_OpenSocket: Unable to bind to %s", UDP_AddrToString((struct qsockaddr *)&address));
181         }
182
183         return newsocket;
184 }
185
186 //=============================================================================
187
188 int UDP_CloseSocket (int socket)
189 {
190         if (net_broadcastsocket == socket)
191                 net_broadcastsocket = 0;
192 #ifdef WIN32
193         return closesocket (socket);
194 #else
195         return close (socket);
196 #endif
197 }
198
199 //=============================================================================
200
201 int UDP_Connect (int socket, struct qsockaddr *addr)
202 {
203         return 0;
204 }
205
206 //=============================================================================
207
208 int UDP_CheckNewConnections (void)
209 {
210         char buf[4096];
211 #ifndef WIN32
212         unsigned long   available;
213         struct sockaddr_in      from;
214         socklen_t                       fromlen;
215 #endif
216
217         if (net_acceptsocket == -1)
218                 return -1;
219
220 #ifdef WIN32
221         if (recvfrom (net_acceptsocket, buf, sizeof(buf), MSG_PEEK, NULL, NULL) >= 0)
222                 return net_acceptsocket;
223 #else
224         if (ioctl (net_acceptsocket, FIONREAD, &available) == -1)
225                 Sys_Error ("UDP: ioctlsocket (FIONREAD) failed\n");
226         if (available)
227                 return net_acceptsocket;
228         recvfrom (net_acceptsocket, buf, 0, 0, (struct sockaddr *) &from, &fromlen);
229 #endif
230         return -1;
231 }
232
233 //=============================================================================
234
235 int UDP_Recv (qbyte *buf, int len, struct qsockaddr *addr)
236 {
237         return UDP_Read (net_acceptsocket, buf, len, addr);
238 }
239
240 //=============================================================================
241
242 int UDP_Send (qbyte *buf, int len, struct qsockaddr *addr)
243 {
244         return UDP_Write (net_acceptsocket, buf, len, addr);
245 }
246
247 //=============================================================================
248
249 int UDP_Read (int socket, qbyte *buf, int len, struct qsockaddr *addr)
250 {
251         int addrlen = sizeof (struct qsockaddr);
252         int ret;
253
254         ret = recvfrom (socket, buf, len, 0, (struct sockaddr *)addr, &addrlen);
255         if (ret == -1)
256         {
257 #ifdef WIN32
258                 int e = WSAGetLastError();
259                 if (e == WSAEWOULDBLOCK || e == WSAECONNREFUSED)
260                         return 0;
261                 Con_Printf("UDP_Read: WASGetLastError == %i\n", e);
262 #else
263                 if (errno == EWOULDBLOCK || errno == ECONNREFUSED)
264                         return 0;
265                 Con_Printf("UDP_Read: errno == %i (%s)\n", errno, strerror(errno));
266 #endif
267         }
268         return ret;
269 }
270
271 //=============================================================================
272
273 int UDP_MakeSocketBroadcastCapable (int socket)
274 {
275         int i = 1;
276
277         // make this socket broadcast capable
278         if (setsockopt(socket, SOL_SOCKET, SO_BROADCAST, (char *)&i, sizeof(i)) < 0)
279                 return -1;
280         net_broadcastsocket = socket;
281
282         return 0;
283 }
284
285 //=============================================================================
286
287 int UDP_Broadcast (int socket, qbyte *buf, int len)
288 {
289         int ret;
290
291         if (socket != net_broadcastsocket)
292         {
293                 if (net_broadcastsocket != 0)
294                         Sys_Error("Attempted to use multiple broadcasts sockets\n");
295                 ret = UDP_MakeSocketBroadcastCapable (socket);
296                 if (ret == -1)
297                 {
298                         Con_Printf("Unable to make socket broadcast capable\n");
299                         return ret;
300                 }
301         }
302
303         return UDP_Write (socket, buf, len, &broadcastaddr);
304 }
305
306 //=============================================================================
307
308 int UDP_Write (int socket, qbyte *buf, int len, struct qsockaddr *addr)
309 {
310         int ret;
311
312         ret = sendto (socket, buf, len, 0, (struct sockaddr *)addr, sizeof(struct qsockaddr));
313         if (ret == -1)
314         {
315 #ifdef WIN32
316                 int e = WSAGetLastError();
317                 if (e == WSAEWOULDBLOCK)
318                         return 0;
319                 Con_Printf("UDP_Write: WASGetLastError == %i\n", e);
320 #else
321                 if (errno == EWOULDBLOCK)
322                         return 0;
323                 Con_Printf("UDP_Write: errno == %i (%s)\n", errno, strerror(errno));
324 #endif
325         }
326         return ret;
327 }
328
329 //=============================================================================
330
331 char *UDP_AddrToString (const struct qsockaddr *addr)
332 {
333         static char buffer[22]; // only 22 needed (3 + 1 + 3 + 1 + 3 + 1 + 3 + 1 + 5 + null)
334         unsigned char *ip = (char *)(&((struct sockaddr_in *)addr)->sin_addr.s_addr);
335         sprintf(buffer, "%d.%d.%d.%d:%d", ip[0], ip[1], ip[2], ip[3], ntohs(((struct sockaddr_in *)addr)->sin_port));
336         return buffer;
337 }
338
339 //=============================================================================
340
341 int UDP_StringToAddr (const char *string, struct qsockaddr *addr)
342 {
343         int ha1, ha2, ha3, ha4, hp;
344         int ipaddr;
345
346         sscanf(string, "%d.%d.%d.%d:%d", &ha1, &ha2, &ha3, &ha4, &hp);
347         ipaddr = (ha1 << 24) | (ha2 << 16) | (ha3 << 8) | ha4;
348
349         addr->sa_family = AF_INET;
350         ((struct sockaddr_in *)addr)->sin_addr.s_addr = htonl(ipaddr);
351         ((struct sockaddr_in *)addr)->sin_port = htons((unsigned short)hp);
352         return 0;
353 }
354
355 //=============================================================================
356
357 int UDP_GetSocketAddr (int socket, struct qsockaddr *addr)
358 {
359         int ret;
360         int addrlen = sizeof(struct qsockaddr);
361         memset(addr, 0, sizeof(struct qsockaddr));
362         ret = getsockname(socket, (struct sockaddr *)addr, &addrlen);
363         if (ret == -1)
364         {
365 #ifdef WIN32
366                 int e = WSAGetLastError();
367                 Con_Printf("UDP_GetSocketAddr: WASGetLastError == %i\n", e);
368 #else
369                 Con_Printf("UDP_GetSocketAddr: errno == %i (%s)\n", errno, strerror(errno));
370 #endif
371         }
372         return ret;
373 }
374
375 //=============================================================================
376
377 int UDP_GetNameFromAddr (const struct qsockaddr *addr, char *name)
378 {
379         struct hostent *hostentry;
380
381         hostentry = gethostbyaddr ((char *)&((struct sockaddr_in *)addr)->sin_addr, sizeof(struct in_addr), AF_INET);
382         if (hostentry)
383         {
384                 strncpy (name, (char *)hostentry->h_name, NET_NAMELEN - 1);
385                 return 0;
386         }
387
388         strcpy (name, UDP_AddrToString (addr));
389         return 0;
390 }
391
392 //=============================================================================
393
394 int UDP_GetAddrFromName(const char *name, struct qsockaddr *addr)
395 {
396         struct hostent *hostentry;
397
398         if (name[0] >= '0' && name[0] <= '9')
399                 return UDP_StringToAddr (name, addr);
400
401         hostentry = gethostbyname (name);
402         if (!hostentry)
403                 return -1;
404
405         addr->sa_family = AF_INET;
406         ((struct sockaddr_in *)addr)->sin_port = htons((unsigned short)net_hostport);
407         ((struct sockaddr_in *)addr)->sin_addr.s_addr = *(int *)hostentry->h_addr_list[0];
408
409         return 0;
410 }
411
412 //=============================================================================
413
414 int UDP_AddrCompare (const struct qsockaddr *addr1, const struct qsockaddr *addr2)
415 {
416         if (addr1->sa_family != addr2->sa_family)
417                 return -1;
418
419         if (((struct sockaddr_in *)addr1)->sin_addr.s_addr != ((struct sockaddr_in *)addr2)->sin_addr.s_addr)
420                 return -1;
421
422         if (((struct sockaddr_in *)addr1)->sin_port != ((struct sockaddr_in *)addr2)->sin_port)
423                 return 1;
424
425         return 0;
426 }
427
428 //=============================================================================
429
430 int UDP_GetSocketPort (struct qsockaddr *addr)
431 {
432         return ntohs(((struct sockaddr_in *)addr)->sin_port);
433 }
434
435
436 int UDP_SetSocketPort (struct qsockaddr *addr, int port)
437 {
438         ((struct sockaddr_in *)addr)->sin_port = htons((unsigned short)port);
439         return 0;
440 }
441