arch/win32/ipx_mcast4.c

   1 /* $Id: ipx_mcast4.c,v 1.3 2003-11-26 12:26:25 btb Exp $ */
   2
   3 /*
   4  *
   5  * "ipx driver" for IPv4 multicasting
   6  *
   7  */
   8
   9
  10 #ifdef HAVE_CONFIG_H
  11 #include <conf.h>
  12 #endif
  13
  14 #include <stdio.h>
  15 #include <string.h>
  16 #include <stdarg.h>
  17 #ifndef _MSC_VER
  18 #include <unistd.h>
  19 #endif
  20 #include <stdlib.h>
  21 #include <ctype.h>
  22
  23 #include <winsock2.h>
  24 #include <ws2tcpip.h>
  25
  26 #include "ipx_mcast4.h"
  27 #include "args.h"
  28 #include "error.h"
  29 #include "../../main/multi.h"
  30 #include "../../main/newmenu.h"
  31
  32 //#define IPX_MCAST4_DEBUG
  33
  34 extern unsigned char ipx_MyAddress[10];
  35
  36 #define UDP_BASEPORT 28342
  37 #define PORTSHIFT_TOLERANCE 0x100
  38 #define MAX_PACKETSIZE 8192
  39
  40 /* OUR port. Can be changed by "@X[+=]..." argument (X is the shift value)
  41  */
  42 static int baseport=UDP_BASEPORT;
  43
  44 static struct in_addr game_addr;    // The game's multicast address
  45
  46 #define MSGHDR "IPX_mcast4: "
  47
  48 #ifdef __GNUC__
  49 #ifdef IPX_MCAST4_DEBUG
  50 static void msg(const char *fmt, ...)
  51 {
  52         va_list ap;
  53
  54         fputs(MSGHDR, stdout);
  55         va_start(ap, fmt);
  56         vprintf(fmt, ap);
  57         va_end(ap);
  58         putchar('\n');
  59 }
  60 #else
  61 #define msg(m...)
  62 #endif
  63 #else
  64 # ifdef IPX_MCAST4_DEBUG
  65 #  define msg printf
  66 # else
  67 #  define msg
  68 # endif
  69 #endif
  70
  71 #ifdef __GNUC__
  72 #define FAIL(m...) do{ nm_messagebox("Error", 1, "Ok", ##m); return -1; } while (0)
  73 #else
  74 # define FAIL() do{ nm_messagebox("Error", 1, "Ok", "IPX_mcast4 failure"); return -1; } while (0)
  75 #endif
  76
  77 #ifdef IPX_MCAST4_DEBUG
  78 /* Dump raw form of IP address/port by fancy output to user
  79  */
  80 static void dumpraddr(unsigned char *a)
  81 {
  82         short port;
  83
  84         printf("[%u.%u.%u.%u]", a[0], a[1], a[2], a[3]);
  85         port=(signed short)ntohs(*(unsigned short *)(a+4));
  86         if (port) printf(":%+d",port);
  87 }
  88
  89 /* Like dumpraddr() but for structure "sockaddr_in"
  90  */
  91 static void dumpaddr(struct sockaddr_in *sin)
  92 {
  93         unsigned short ports;
  94         unsigned char qhbuf[8];
  95
  96         memcpy(qhbuf + 0, &sin->sin_addr, 4);
  97         ports = htons(((short)ntohs(sin->sin_port)) - UDP_BASEPORT);
  98         memcpy(qhbuf + 4, &ports, 2);
  99         dumpraddr(qhbuf);
 100 }
 101 #endif
 102
 103 // The multicast address for Descent 2 game announcements.
 104 // TODO: Pick a better address for this
 105 #define DESCENT2_ANNOUNCE_ADDR inet_addr("239.255.1.2")
 106
 107 /* Open the socket and subscribe to the multicast session */
 108 static int ipx_mcast4_OpenSocket(ipx_socket_t *sk, int port)
 109 {
 110         u_char loop;
 111         struct ip_mreq mreq;
 112         struct sockaddr_in sin;
 113         int ttl = 128;
 114
 115         if((sk->fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0)
 116         {
 117                 sk->fd = -1;
 118                 FAIL("socket() creation failed on port %d: %m", port);
 119         }
 120
 121         // Bind to the port
 122         sin.sin_family = AF_INET;
 123         sin.sin_addr.s_addr = htonl(INADDR_ANY);
 124         sin.sin_port = htons(baseport);
 125         if (bind(sk->fd, (struct sockaddr *)&sin, sizeof(sin)))
 126         {
 127                 if (closesocket(sk->fd))
 128                         msg("close() failed during error recovery: %m");
 129                 sk->fd = -1;
 130                 FAIL("bind() to UDP port %d failed: %m", baseport);
 131         }
 132
 133         // Set the TTL so the packets can get out of the local network.
 134         if(setsockopt(sk->fd, IPPROTO_IP, IP_MULTICAST_TTL, (const void*)&ttl, sizeof(ttl)) < 0)
 135                 FAIL("setsockopt() failed to set TTL to 128");
 136
 137         // Disable multicast loopback
 138         loop = 0;
 139         if(setsockopt(sk->fd, IPPROTO_IP, IP_MULTICAST_LOOP, (const void*)&loop, sizeof(loop)) < 0)
 140                 FAIL("setsockopt() failed to disable multicast loopback: %m");
 141
 142         // Subscribe to the game announcement address
 143         memset(&mreq, 0, sizeof(mreq));
 144         mreq.imr_multiaddr.s_addr = DESCENT2_ANNOUNCE_ADDR;
 145         mreq.imr_interface.s_addr = INADDR_ANY;
 146         if(setsockopt(sk->fd, IPPROTO_IP, IP_ADD_MEMBERSHIP, (const void*)&mreq, sizeof(mreq)) < 0)
 147                 FAIL("setsockopt() failed to subscribe to the game announcement multicast group");
 148
 149         // We're not subscribed to a game address yet
 150         game_addr.s_addr = 0;
 151
 152         sk->socket = port;
 153         return 0;
 154 }
 155
 156 static void ipx_mcast4_CloseSocket(ipx_socket_t *sk)
 157 {
 158         if(closesocket(sk->fd) < 0)
 159                 msg("Close failed");
 160         sk->fd = -1;
 161 }
 162
 163 static int ipx_mcast4_SendPacket(ipx_socket_t *sk, IPXPacket_t *IPXHeader, u_char *data, int dataLen)
 164 {
 165         struct sockaddr_in toaddr;
 166         int i;
 167
 168         msg("SendPacket enter, dataLen=%d", dataLen);
 169
 170         if(dataLen < 0 || dataLen > MAX_PACKETSIZE)
 171                 return -1;
 172
 173         toaddr.sin_family = AF_INET;
 174         memcpy(&toaddr.sin_addr, IPXHeader->Destination.Node + 0, 4);
 175         //toaddr.sin_port = htons(((short)ntohs(*(unsigned short *)(IPXHeader->Destination.Node + 4))) + UDP_BASEPORT);
 176         // For now, just use the same port for everything
 177         toaddr.sin_port = htons(UDP_BASEPORT);
 178
 179         // If it's the broadcast address, then we want to send it to the
 180         // GAME ANNOUNCEMENT address.
 181         // Data to be sent to the GAME has the destination already set by
 182         // ipx_mcast4_SendGamePacket
 183         if(toaddr.sin_addr.s_addr == INADDR_BROADCAST)
 184                 toaddr.sin_addr.s_addr = DESCENT2_ANNOUNCE_ADDR;
 185
 186 #ifdef IPX_MCAST4_DEBUG
 187         printf(MSGHDR "sendto((%d),Node=[4] %02X %02X,Socket=%02X %02X,s_port=%u,",
 188                dataLen,
 189                IPXHeader->Destination.Node[4], IPXHeader->Destination.Node[5],
 190                IPXHeader->Destination.Socket[0], IPXHeader->Destination.Socket[1],
 191                ntohs(toaddr.sin_port));
 192         dumpaddr(&toaddr);
 193         puts(").");
 194 #endif
 195
 196         i = sendto(sk->fd, data, dataLen, 0, (struct sockaddr *)&toaddr, sizeof(toaddr));
 197         return i;
 198 }
 199
 200 static int ipx_mcast4_ReceivePacket(ipx_socket_t *sk, char *outbuf, int outbufsize, struct ipx_recv_data *rd)
 201 {
 202         int size;
 203         struct sockaddr_in fromaddr;
 204         int fromaddrsize = sizeof(fromaddr);
 205
 206         if((size = recvfrom(sk->fd, outbuf, outbufsize, 0, (struct sockaddr*)&fromaddr, &fromaddrsize)) < 0)
 207                 return -1;
 208
 209 #ifdef IPX_MCAST4_DEBUG
 210         printf(MSGHDR "Got packet from ");
 211         dumpaddr(&fromaddr);
 212         puts("");
 213 #endif
 214
 215         // We have the packet, now fill out the receive data.
 216         memset(rd, 0, sizeof(*rd));
 217         memcpy(rd->src_node, &fromaddr.sin_addr, 4);
 218         // TODO: Include the port like in ipx_udp.c
 219         rd->pkt_type = 0;
 220
 221         return size;
 222 }
 223
 224 /* Handle the netgame aux data
 225  * Byte 0 is the protocol version number.
 226  * Bytes 1-4 are the IPv4 multicast session to join, in network byte order.
 227  */
 228 static int ipx_mcast4_HandleNetgameAuxData(ipx_socket_t *sk, const u_char buf[NETGAME_AUX_SIZE])
 229 {
 230         // Extract the multicast session and subscribe to it.  We should
 231         // now be getting packets intended for the players of this game.
 232
 233         // Note that we stay subscribed to the game announcement session,
 234         // so we can reply to game info requests
 235         struct ip_mreq mreq;
 236         int ttl = 128;
 237
 238         // Check the protocol version
 239         if(buf[0] != IPX_MCAST4_VERSION)
 240         {
 241                 FAIL("mcast4 protocol\nversion mismatch!\nGame version is %02x,\nour version is %02x", buf[0], IPX_MCAST4_VERSION);
 242         }
 243
 244         // Get the multicast session
 245         memcpy(&game_addr, buf + 1, sizeof(game_addr));
 246
 247 #ifdef IPX_MCAST4_DEBUG
 248         {
 249                 struct sockaddr_in tmpaddr;
 250                 tmpaddr.sin_addr = game_addr;
 251                 tmpaddr.sin_port = 0;
 252
 253                 printf("Handling netgame aux data: Subscribing to ");
 254                 dumpaddr(&tmpaddr);
 255                 puts("");
 256         }
 257 #endif
 258
 259         // Set the TTL so the packets can get out of the local network.
 260         if(setsockopt(sk->fd, IPPROTO_IP, IP_MULTICAST_TTL, (const void*)&ttl, sizeof(ttl)) < 0)
 261                 FAIL("setsockopt() failed to set TTL to 128");
 262
 263         memset(&mreq, 0, sizeof(mreq));
 264         mreq.imr_multiaddr = game_addr;
 265         mreq.imr_interface.s_addr = INADDR_ANY;
 266
 267         if(setsockopt(sk->fd, IPPROTO_IP, IP_ADD_MEMBERSHIP, (const void*)&mreq, sizeof(mreq)) < 0)
 268                 FAIL("setsockopt() failed to subscribe to the game group");
 269
 270         return 0;
 271 }
 272
 273 /* Create the netgame aux data.
 274  * Byte 0 is the protcol version number.
 275  * Bytes 1-4 hold the IPv4 multicast session for the game.
 276  */
 277 static void ipx_mcast4_InitNetgameAuxData(ipx_socket_t *sk, u_char buf[NETGAME_AUX_SIZE])
 278 {
 279         char addr[16];
 280         Assert(game_addr.s_addr == 0);
 281
 282         // The first byte is the version number
 283         buf[0] = IPX_MCAST4_VERSION;
 284
 285         // Generate a random session
 286         // game_addr = inet_makeaddr(239*256 + 255, d_rand() % 0xFFFF);
 287         sprintf(addr, "%i.%i.%i.%i", 239, 255, d_rand() % 0xFF, d_rand() % 0xFF);
 288         game_addr.s_addr = inet_addr(addr);
 289         memcpy(buf + 1, &game_addr, sizeof(game_addr));
 290
 291         // Since we're obviously the hosting machine, subscribe to this address
 292         ipx_mcast4_HandleNetgameAuxData(sk, buf);
 293 }
 294
 295 static void ipx_mcast4_HandleLeaveGame(ipx_socket_t *sk)
 296 {
 297         // We left the game, so unsubscribe from its multicast session
 298         struct ip_mreq mreq;
 299
 300         Assert(game_addr.s_addr != 0);
 301
 302 #ifdef IPX_MCAST4_DEBUG
 303         printf("Unsubscribing from game's multicast group: ");
 304         dumpraddr(&game_addr.s_addr);
 305         printf("\n");
 306 #endif
 307
 308         mreq.imr_multiaddr = game_addr;
 309         mreq.imr_interface.s_addr = INADDR_ANY;
 310         if(setsockopt(sk->fd, IPPROTO_IP, IP_DROP_MEMBERSHIP, (const void*)&mreq, sizeof(mreq)) < 0)
 311                 msg("setsockopt() failed unsubscribing from previous group!");
 312         game_addr.s_addr = 0;
 313 }
 314
 315 // Send a packet to every member of the game.  We can just multicast it here.
 316 static int ipx_mcast4_SendGamePacket(ipx_socket_t *sk, ubyte *data, int dataLen)
 317 {
 318         struct sockaddr_in toaddr;
 319         int i;
 320
 321         memset(&toaddr, 0, sizeof(toaddr));
 322         toaddr.sin_family = AF_INET;
 323         toaddr.sin_addr = game_addr;
 324         toaddr.sin_port = htons(UDP_BASEPORT);
 325
 326         msg("ipx_mcast4_SendGamePacket");
 327
 328         i = sendto(sk->fd, data, dataLen, 0, (struct sockaddr *)&toaddr, sizeof(toaddr));
 329
 330         return i;
 331 }
 332
 333 // Pull this in from ipx_udp.c since it's the same for us.
 334 extern int ipx_udp_GetMyAddress();
 335
 336 struct ipx_driver ipx_mcast4 = {
 337         ipx_udp_GetMyAddress,
 338         ipx_mcast4_OpenSocket,
 339         ipx_mcast4_CloseSocket,
 340         ipx_mcast4_SendPacket,
 341         ipx_mcast4_ReceivePacket,
 342         ipx_general_PacketReady,
 343         ipx_mcast4_InitNetgameAuxData,
 344         ipx_mcast4_HandleNetgameAuxData,
 345         ipx_mcast4_HandleLeaveGame,
 346         ipx_mcast4_SendGamePacket
 347 };