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