2 Copyright (C) 1996-1997 Id Software, Inc.
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.
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.
13 See the GNU General Public License for more details.
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.
22 // This is enables a simple IP banning mechanism
29 #include <sys/socket.h>
30 #include <arpa/inet.h>
32 #define AF_INET 2 /* internet */
37 struct { unsigned char s_b1,s_b2,s_b3,s_b4; } S_un_b;
38 struct { unsigned short s_w1,s_w2; } S_un_w;
42 #define s_addr S_un.S_addr /* can be used for most tcp & ip code */
46 unsigned short sin_port;
47 struct in_addr sin_addr;
50 char *inet_ntoa(struct in_addr in);
51 unsigned long inet_addr(const char *cp);
58 cvar_t cl_port = {CVAR_SAVE, "cl_port", "0"};
60 // these two macros are to make the code more readable
61 #define sfunc net_landrivers[sock->landriver]
62 #define dfunc net_landrivers[net_landriverlevel]
64 static int net_landriverlevel;
66 /* statistic counters */
68 int packetsReSent = 0;
69 int packetsReceived = 0;
70 int receivedDuplicateCount = 0;
71 int shortPacketCount = 0;
74 static int myDriverLevel;
79 unsigned int sequence;
80 qbyte data[MAX_DATAGRAM];
85 char *StrAddr (struct qsockaddr *addr)
88 qbyte *p = (qbyte *)addr;
91 for (n = 0; n < 16; n++)
92 sprintf (buf + n * 2, "%02x", *p++);
99 unsigned long banAddr = 0x00000000;
100 unsigned long banMask = 0xffffffff;
102 void NET_Ban_f (void)
106 void (*print) (const char *fmt, ...);
108 if (cmd_source == src_command)
112 Cmd_ForwardToServer ();
119 if (pr_global_struct->deathmatch)
121 print = SV_ClientPrintf;
127 if (((struct in_addr *)&banAddr)->s_addr)
129 strcpy(addrStr, inet_ntoa(*(struct in_addr *)&banAddr));
130 strcpy(maskStr, inet_ntoa(*(struct in_addr *)&banMask));
131 print("Banning %s [%s]\n", addrStr, maskStr);
134 print("Banning not active\n");
138 if (Q_strcasecmp(Cmd_Argv(1), "off") == 0)
139 banAddr = 0x00000000;
141 banAddr = inet_addr(Cmd_Argv(1));
142 banMask = 0xffffffff;
146 banAddr = inet_addr(Cmd_Argv(1));
147 banMask = inet_addr(Cmd_Argv(2));
151 print("BAN ip_address [mask]\n");
158 int Datagram_SendMessage (qsocket_t *sock, sizebuf_t *data)
160 unsigned int packetLen;
161 unsigned int dataLen;
165 if (data->cursize == 0)
166 Sys_Error("Datagram_SendMessage: zero length message\n");
168 if (data->cursize > NET_MAXMESSAGE)
169 Sys_Error("Datagram_SendMessage: message too big %u\n", data->cursize);
171 if (sock->canSend == false)
172 Sys_Error("SendMessage: called with canSend == false\n");
175 memcpy(sock->sendMessage, data->data, data->cursize);
176 sock->sendMessageLength = data->cursize;
178 if (data->cursize <= MAX_DATAGRAM)
180 dataLen = data->cursize;
185 dataLen = MAX_DATAGRAM;
188 packetLen = NET_HEADERSIZE + dataLen;
190 packetBuffer.length = BigLong(packetLen | (NETFLAG_DATA | eom));
191 packetBuffer.sequence = BigLong(sock->sendSequence++);
192 memcpy (packetBuffer.data, sock->sendMessage, dataLen);
194 sock->canSend = false;
196 if (sfunc.Write (sock->socket, (qbyte *)&packetBuffer, packetLen, &sock->addr) == -1)
199 sock->lastSendTime = net_time;
205 int SendMessageNext (qsocket_t *sock)
207 unsigned int packetLen;
208 unsigned int dataLen;
211 if (sock->sendMessageLength <= MAX_DATAGRAM)
213 dataLen = sock->sendMessageLength;
218 dataLen = MAX_DATAGRAM;
221 packetLen = NET_HEADERSIZE + dataLen;
223 packetBuffer.length = BigLong(packetLen | (NETFLAG_DATA | eom));
224 packetBuffer.sequence = BigLong(sock->sendSequence++);
225 memcpy (packetBuffer.data, sock->sendMessage, dataLen);
227 sock->sendNext = false;
229 if (sfunc.Write (sock->socket, (qbyte *)&packetBuffer, packetLen, &sock->addr) == -1)
232 sock->lastSendTime = net_time;
238 int ReSendMessage (qsocket_t *sock)
240 unsigned int packetLen;
241 unsigned int dataLen;
244 if (sock->sendMessageLength <= MAX_DATAGRAM)
246 dataLen = sock->sendMessageLength;
251 dataLen = MAX_DATAGRAM;
254 packetLen = NET_HEADERSIZE + dataLen;
256 packetBuffer.length = BigLong(packetLen | (NETFLAG_DATA | eom));
257 packetBuffer.sequence = BigLong(sock->sendSequence - 1);
258 memcpy (packetBuffer.data, sock->sendMessage, dataLen);
260 sock->sendNext = false;
262 if (sfunc.Write (sock->socket, (qbyte *)&packetBuffer, packetLen, &sock->addr) == -1)
265 sock->lastSendTime = net_time;
271 qboolean Datagram_CanSendMessage (qsocket_t *sock)
274 SendMessageNext (sock);
276 return sock->canSend;
280 qboolean Datagram_CanSendUnreliableMessage (qsocket_t *sock)
286 int Datagram_SendUnreliableMessage (qsocket_t *sock, sizebuf_t *data)
291 if (data->cursize == 0)
292 Sys_Error("Datagram_SendUnreliableMessage: zero length message\n");
294 if (data->cursize > MAX_DATAGRAM)
295 Sys_Error("Datagram_SendUnreliableMessage: message too big %u\n", data->cursize);
298 packetLen = NET_HEADERSIZE + data->cursize;
300 packetBuffer.length = BigLong(packetLen | NETFLAG_UNRELIABLE);
301 packetBuffer.sequence = BigLong(sock->unreliableSendSequence++);
302 memcpy (packetBuffer.data, data->data, data->cursize);
304 if (sfunc.Write (sock->socket, (qbyte *)&packetBuffer, packetLen, &sock->addr) == -1)
312 int Datagram_GetMessage (qsocket_t *sock)
317 struct qsockaddr readaddr;
318 unsigned int sequence;
322 if ((net_time - sock->lastSendTime) > 1.0)
323 ReSendMessage (sock);
327 length = sfunc.Read (sock->socket, (qbyte *)&packetBuffer, NET_DATAGRAMSIZE, &readaddr);
334 Con_Printf("Read error\n");
338 if (sfunc.AddrCompare(&readaddr, &sock->addr) != 0)
341 Con_DPrintf("Forged packet received\n");
342 Con_DPrintf("Expected: %s\n", StrAddr (&sock->addr));
343 Con_DPrintf("Received: %s\n", StrAddr (&readaddr));
348 if (length < NET_HEADERSIZE)
354 length = BigLong(packetBuffer.length);
355 flags = length & (~NETFLAG_LENGTH_MASK);
356 length &= NETFLAG_LENGTH_MASK;
358 if (flags & NETFLAG_CTL)
361 sequence = BigLong(packetBuffer.sequence);
364 if (flags & NETFLAG_UNRELIABLE)
366 if (sequence < sock->unreliableReceiveSequence)
368 Con_DPrintf("Got a stale datagram\n");
372 if (sequence != sock->unreliableReceiveSequence)
374 count = sequence - sock->unreliableReceiveSequence;
375 droppedDatagrams += count;
376 Con_DPrintf("Dropped %u datagram(s)\n", count);
378 sock->unreliableReceiveSequence = sequence + 1;
380 length -= NET_HEADERSIZE;
382 SZ_Clear (&net_message);
383 SZ_Write (&net_message, packetBuffer.data, length);
389 if (flags & NETFLAG_ACK)
391 if (sequence != (sock->sendSequence - 1))
393 Con_DPrintf("Stale ACK received\n");
396 if (sequence == sock->ackSequence)
399 if (sock->ackSequence != sock->sendSequence)
400 Con_DPrintf("ack sequencing error\n");
404 Con_DPrintf("Duplicate ACK received\n");
407 sock->sendMessageLength -= MAX_DATAGRAM;
408 if (sock->sendMessageLength > 0)
410 memcpy(sock->sendMessage, sock->sendMessage+MAX_DATAGRAM, sock->sendMessageLength);
411 sock->sendNext = true;
415 sock->sendMessageLength = 0;
416 sock->canSend = true;
421 if (flags & NETFLAG_DATA)
423 packetBuffer.length = BigLong(NET_HEADERSIZE | NETFLAG_ACK);
424 packetBuffer.sequence = BigLong(sequence);
425 sfunc.Write (sock->socket, (qbyte *)&packetBuffer, NET_HEADERSIZE, &readaddr);
427 if (sequence != sock->receiveSequence)
429 receivedDuplicateCount++;
432 sock->receiveSequence++;
434 length -= NET_HEADERSIZE;
436 if (flags & NETFLAG_EOM)
438 SZ_Clear(&net_message);
439 SZ_Write(&net_message, sock->receiveMessage, sock->receiveMessageLength);
440 SZ_Write(&net_message, packetBuffer.data, length);
441 sock->receiveMessageLength = 0;
447 memcpy(sock->receiveMessage + sock->receiveMessageLength, packetBuffer.data, length);
448 sock->receiveMessageLength += length;
454 SendMessageNext (sock);
460 void PrintStats(qsocket_t *s)
462 Con_Printf("canSend = %4u \n", s->canSend);
463 Con_Printf("sendSeq = %4u ", s->sendSequence);
464 Con_Printf("recvSeq = %4u \n", s->receiveSequence);
468 void NET_Stats_f (void)
472 if (Cmd_Argc () == 1)
474 Con_Printf("unreliable messages sent = %i\n", unreliableMessagesSent);
475 Con_Printf("unreliable messages recv = %i\n", unreliableMessagesReceived);
476 Con_Printf("reliable messages sent = %i\n", messagesSent);
477 Con_Printf("reliable messages received = %i\n", messagesReceived);
478 Con_Printf("packetsSent = %i\n", packetsSent);
479 Con_Printf("packetsReSent = %i\n", packetsReSent);
480 Con_Printf("packetsReceived = %i\n", packetsReceived);
481 Con_Printf("receivedDuplicateCount = %i\n", receivedDuplicateCount);
482 Con_Printf("shortPacketCount = %i\n", shortPacketCount);
483 Con_Printf("droppedDatagrams = %i\n", droppedDatagrams);
485 else if (strcmp(Cmd_Argv(1), "*") == 0)
487 for (s = net_activeSockets; s; s = s->next)
492 for (s = net_activeSockets; s; s = s->next)
493 if (Q_strcasecmp(Cmd_Argv(1), s->address) == 0)
502 static qboolean testInProgress = false;
503 static int testPollCount;
504 static int testDriver;
505 static int testSocket;
507 static void Test_Poll(void);
508 PollProcedure testPollProcedure = {NULL, 0.0, Test_Poll};
510 static void Test_Poll(void)
512 struct qsockaddr clientaddr;
523 net_landriverlevel = testDriver;
527 len = dfunc.Read (testSocket, net_message.data, net_message.maxsize, &clientaddr);
528 if (len < sizeof(int))
531 net_message.cursize = len;
534 control = BigLong(*((int *)net_message.data));
538 if ((control & (~NETFLAG_LENGTH_MASK)) != NETFLAG_CTL)
540 if ((control & NETFLAG_LENGTH_MASK) != len)
544 if (c != CCREP_PLAYER_INFO)
545 Sys_Error("Unexpected repsonse to Player Info request\n");
547 playerNumber = MSG_ReadByte();
548 strcpy(name, MSG_ReadString());
549 colors = MSG_ReadLong();
550 frags = MSG_ReadLong();
551 connectTime = MSG_ReadLong();
552 strcpy(address, MSG_ReadString());
554 Con_Printf("%s\n frags:%3i colors:%u %u time:%u\n %s\n", name, frags, colors >> 4, colors & 0x0f, connectTime / 60, address);
560 SchedulePollProcedure(&testPollProcedure, 0.1);
564 dfunc.CloseSocket(testSocket);
565 testInProgress = false;
569 static void Test_f (void)
572 int n, max = MAX_SCOREBOARD;
573 struct qsockaddr sendaddr;
580 if (host && hostCacheCount)
582 for (n = 0; n < hostCacheCount; n++)
583 if (Q_strcasecmp (host, hostcache[n].name) == 0)
585 if (hostcache[n].driver != myDriverLevel)
587 net_landriverlevel = hostcache[n].ldriver;
588 max = hostcache[n].maxusers;
589 memcpy(&sendaddr, &hostcache[n].addr, sizeof(struct qsockaddr));
592 if (n < hostCacheCount)
596 for (net_landriverlevel = 0; net_landriverlevel < net_numlandrivers; net_landriverlevel++)
598 if (!net_landrivers[net_landriverlevel].initialized)
601 // see if we can resolve the host name
602 if (dfunc.GetAddrFromName(host, &sendaddr) != -1)
605 if (net_landriverlevel == net_numlandrivers)
609 testSocket = dfunc.OpenSocket(0);
610 if (testSocket == -1)
613 testInProgress = true;
615 testDriver = net_landriverlevel;
617 for (n = 0; n < max; n++)
619 SZ_Clear(&net_message);
620 // save space for the header, filled in later
621 MSG_WriteLong(&net_message, 0);
622 MSG_WriteByte(&net_message, CCREQ_PLAYER_INFO);
623 MSG_WriteByte(&net_message, n);
624 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
625 dfunc.Write (testSocket, net_message.data, net_message.cursize, &sendaddr);
627 SZ_Clear(&net_message);
628 SchedulePollProcedure(&testPollProcedure, 0.1);
632 static qboolean test2InProgress = false;
633 static int test2Driver;
634 static int test2Socket;
636 static void Test2_Poll(void);
637 PollProcedure test2PollProcedure = {NULL, 0.0, Test2_Poll};
639 static void Test2_Poll(void)
641 struct qsockaddr clientaddr;
648 net_landriverlevel = test2Driver;
651 len = dfunc.Read (test2Socket, net_message.data, net_message.maxsize, &clientaddr);
652 if (len < sizeof(int))
655 net_message.cursize = len;
658 control = BigLong(*((int *)net_message.data));
662 if ((control & (~NETFLAG_LENGTH_MASK)) != NETFLAG_CTL)
664 if ((control & NETFLAG_LENGTH_MASK) != len)
668 if (c != CCREP_RULE_INFO)
671 strcpy(name, MSG_ReadString());
674 strcpy(value, MSG_ReadString());
676 Con_Printf("%-16.16s %-16.16s\n", name, value);
678 SZ_Clear(&net_message);
679 // save space for the header, filled in later
680 MSG_WriteLong(&net_message, 0);
681 MSG_WriteByte(&net_message, CCREQ_RULE_INFO);
682 MSG_WriteString(&net_message, name);
683 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
684 dfunc.Write (test2Socket, net_message.data, net_message.cursize, &clientaddr);
685 SZ_Clear(&net_message);
688 SchedulePollProcedure(&test2PollProcedure, 0.05);
692 Con_Printf("Unexpected repsonse to Rule Info request\n");
694 dfunc.CloseSocket(test2Socket);
695 test2InProgress = false;
699 static void Test2_f (void)
703 struct qsockaddr sendaddr;
710 if (host && hostCacheCount)
712 for (n = 0; n < hostCacheCount; n++)
713 if (Q_strcasecmp (host, hostcache[n].name) == 0)
715 if (hostcache[n].driver != myDriverLevel)
717 net_landriverlevel = hostcache[n].ldriver;
718 memcpy(&sendaddr, &hostcache[n].addr, sizeof(struct qsockaddr));
721 if (n < hostCacheCount)
725 for (net_landriverlevel = 0; net_landriverlevel < net_numlandrivers; net_landriverlevel++)
727 if (!net_landrivers[net_landriverlevel].initialized)
730 // see if we can resolve the host name
731 if (dfunc.GetAddrFromName(host, &sendaddr) != -1)
734 if (net_landriverlevel == net_numlandrivers)
738 test2Socket = dfunc.OpenSocket(0);
739 if (test2Socket == -1)
742 test2InProgress = true;
743 test2Driver = net_landriverlevel;
745 SZ_Clear(&net_message);
746 // save space for the header, filled in later
747 MSG_WriteLong(&net_message, 0);
748 MSG_WriteByte(&net_message, CCREQ_RULE_INFO);
749 MSG_WriteString(&net_message, "");
750 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
751 dfunc.Write (test2Socket, net_message.data, net_message.cursize, &sendaddr);
752 SZ_Clear(&net_message);
753 SchedulePollProcedure(&test2PollProcedure, 0.05);
757 int Datagram_Init (void)
762 myDriverLevel = net_driverlevel;
763 Cmd_AddCommand ("net_stats", NET_Stats_f);
764 Cvar_RegisterVariable (&cl_port);
766 if (COM_CheckParm("-nolan"))
769 for (i = 0; i < net_numlandrivers; i++)
771 csock = net_landrivers[i].Init ();
774 net_landrivers[i].initialized = true;
775 net_landrivers[i].controlSock = csock;
779 Cmd_AddCommand ("ban", NET_Ban_f);
781 Cmd_AddCommand ("test", Test_f);
782 Cmd_AddCommand ("test2", Test2_f);
788 void Datagram_Shutdown (void)
793 // shutdown the lan drivers
795 for (i = 0; i < net_numlandrivers; i++)
797 if (net_landrivers[i].initialized)
799 net_landrivers[i].Shutdown ();
800 net_landrivers[i].initialized = false;
806 void Datagram_Close (qsocket_t *sock)
808 sfunc.CloseSocket(sock->socket);
812 void Datagram_Listen (qboolean state)
816 for (i = 0; i < net_numlandrivers; i++)
817 if (net_landrivers[i].initialized)
818 net_landrivers[i].Listen (state);
822 static qsocket_t *_Datagram_CheckNewConnections (void)
824 struct qsockaddr clientaddr;
825 struct qsockaddr newaddr;
836 acceptsock = dfunc.CheckNewConnections();
837 if (acceptsock == -1)
840 SZ_Clear(&net_message);
842 len = dfunc.Read (acceptsock, net_message.data, net_message.maxsize, &clientaddr);
843 if (len < sizeof(int))
845 net_message.cursize = len;
848 control = BigLong(*((int *)net_message.data));
852 if ((control & (~NETFLAG_LENGTH_MASK)) != NETFLAG_CTL)
854 if ((control & NETFLAG_LENGTH_MASK) != len)
857 command = MSG_ReadByte();
858 if (command == CCREQ_SERVER_INFO)
860 if (strcmp(MSG_ReadString(), "QUAKE") != 0)
863 SZ_Clear(&net_message);
864 // save space for the header, filled in later
865 MSG_WriteLong(&net_message, 0);
866 MSG_WriteByte(&net_message, CCREP_SERVER_INFO);
867 dfunc.GetSocketAddr(acceptsock, &newaddr);
868 MSG_WriteString(&net_message, dfunc.AddrToString(&newaddr));
869 MSG_WriteString(&net_message, hostname.string);
870 MSG_WriteString(&net_message, sv.name);
871 MSG_WriteByte(&net_message, net_activeconnections);
872 MSG_WriteByte(&net_message, svs.maxclients);
873 MSG_WriteByte(&net_message, NET_PROTOCOL_VERSION);
874 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
875 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
876 SZ_Clear(&net_message);
880 if (command == CCREQ_PLAYER_INFO)
887 playerNumber = MSG_ReadByte();
889 for (clientNumber = 0, client = svs.clients; clientNumber < svs.maxclients; clientNumber++, client++)
894 if (activeNumber == playerNumber)
898 if (clientNumber == svs.maxclients)
901 SZ_Clear(&net_message);
902 // save space for the header, filled in later
903 MSG_WriteLong(&net_message, 0);
904 MSG_WriteByte(&net_message, CCREP_PLAYER_INFO);
905 MSG_WriteByte(&net_message, playerNumber);
906 MSG_WriteString(&net_message, client->name);
907 MSG_WriteLong(&net_message, client->colors);
908 MSG_WriteLong(&net_message, (int)client->edict->v.frags);
909 MSG_WriteLong(&net_message, (int)(net_time - client->netconnection->connecttime));
910 MSG_WriteString(&net_message, client->netconnection->address);
911 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
912 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
913 SZ_Clear(&net_message);
918 if (command == CCREQ_RULE_INFO)
923 // find the search start location
924 prevCvarName = MSG_ReadString();
925 var = Cvar_FindVarAfter(prevCvarName, CVAR_NOTIFY);
929 SZ_Clear(&net_message);
930 // save space for the header, filled in later
931 MSG_WriteLong(&net_message, 0);
932 MSG_WriteByte(&net_message, CCREP_RULE_INFO);
935 MSG_WriteString(&net_message, var->name);
936 MSG_WriteString(&net_message, var->string);
938 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
939 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
940 SZ_Clear(&net_message);
945 if (command != CCREQ_CONNECT)
948 if (strcmp(MSG_ReadString(), "QUAKE") != 0)
952 if (c != NET_PROTOCOL_VERSION)
954 SZ_Clear(&net_message);
955 // save space for the header, filled in later
956 MSG_WriteLong(&net_message, 0);
957 MSG_WriteByte(&net_message, CCREP_REJECT);
958 MSG_WriteString(&net_message, "Incompatible version.\n");
959 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
960 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
961 SZ_Clear(&net_message);
967 if (clientaddr.sa_family == AF_INET)
969 unsigned long testAddr;
970 testAddr = ((struct sockaddr_in *)&clientaddr)->sin_addr.s_addr;
971 if ((testAddr & banMask) == banAddr)
973 SZ_Clear(&net_message);
974 // save space for the header, filled in later
975 MSG_WriteLong(&net_message, 0);
976 MSG_WriteByte(&net_message, CCREP_REJECT);
977 MSG_WriteString(&net_message, "You have been banned.\n");
978 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
979 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
980 SZ_Clear(&net_message);
986 // see if this guy is already connected
987 for (s = net_activeSockets; s; s = s->next)
989 if (s->driver != net_driverlevel)
991 ret = dfunc.AddrCompare(&clientaddr, &s->addr);
994 // is this a duplicate connection reqeust?
995 if (ret == 0 && net_time - s->connecttime < 2.0)
997 // yes, so send a duplicate reply
998 SZ_Clear(&net_message);
999 // save space for the header, filled in later
1000 MSG_WriteLong(&net_message, 0);
1001 MSG_WriteByte(&net_message, CCREP_ACCEPT);
1002 dfunc.GetSocketAddr(s->socket, &newaddr);
1003 MSG_WriteLong(&net_message, dfunc.GetSocketPort(&newaddr));
1004 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
1005 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
1006 SZ_Clear(&net_message);
1009 // it's somebody coming back in from a crash/disconnect
1010 // so close the old qsocket and let their retry get them back in
1016 // allocate a QSocket
1017 sock = NET_NewQSocket ();
1020 // no room; try to let him know
1021 SZ_Clear(&net_message);
1022 // save space for the header, filled in later
1023 MSG_WriteLong(&net_message, 0);
1024 MSG_WriteByte(&net_message, CCREP_REJECT);
1025 MSG_WriteString(&net_message, "Server is full.\n");
1026 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
1027 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
1028 SZ_Clear(&net_message);
1032 // allocate a network socket
1033 newsock = dfunc.OpenSocket(0);
1036 NET_FreeQSocket(sock);
1040 // connect to the client
1041 if (dfunc.Connect (newsock, &clientaddr) == -1)
1043 dfunc.CloseSocket(newsock);
1044 NET_FreeQSocket(sock);
1048 // everything is allocated, just fill in the details
1049 sock->socket = newsock;
1050 sock->landriver = net_landriverlevel;
1051 sock->addr = clientaddr;
1052 strcpy(sock->address, dfunc.AddrToString(&clientaddr));
1054 // send him back the info about the server connection he has been allocated
1055 SZ_Clear(&net_message);
1056 // save space for the header, filled in later
1057 MSG_WriteLong(&net_message, 0);
1058 MSG_WriteByte(&net_message, CCREP_ACCEPT);
1059 dfunc.GetSocketAddr(newsock, &newaddr);
1060 MSG_WriteLong(&net_message, dfunc.GetSocketPort(&newaddr));
1061 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
1062 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
1063 SZ_Clear(&net_message);
1068 qsocket_t *Datagram_CheckNewConnections (void)
1070 qsocket_t *ret = NULL;
1072 for (net_landriverlevel = 0; net_landriverlevel < net_numlandrivers; net_landriverlevel++)
1073 if (net_landrivers[net_landriverlevel].initialized)
1074 if ((ret = _Datagram_CheckNewConnections ()) != NULL)
1080 static void _Datagram_SearchForHosts (qboolean xmit)
1085 struct qsockaddr readaddr;
1086 struct qsockaddr myaddr;
1090 dfunc.GetSocketAddr (dfunc.controlSock, &myaddr);
1093 SZ_Clear(&net_message);
1094 // save space for the header, filled in later
1095 MSG_WriteLong(&net_message, 0);
1096 MSG_WriteByte(&net_message, CCREQ_SERVER_INFO);
1097 MSG_WriteString(&net_message, "QUAKE");
1098 MSG_WriteByte(&net_message, NET_PROTOCOL_VERSION);
1099 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
1100 dfunc.Broadcast(dfunc.controlSock, net_message.data, net_message.cursize);
1101 SZ_Clear(&net_message);
1104 while ((ret = dfunc.Read (dfunc.controlSock, net_message.data, net_message.maxsize, &readaddr)) > 0)
1106 if (ret < sizeof(int))
1108 net_message.cursize = ret;
1110 // don't answer our own query
1111 if (dfunc.AddrCompare(&readaddr, &myaddr) >= 0)
1114 // is the cache full?
1115 if (hostCacheCount == HOSTCACHESIZE)
1118 MSG_BeginReading ();
1119 control = BigLong(*((int *)net_message.data));
1123 if ((control & (~NETFLAG_LENGTH_MASK)) != NETFLAG_CTL)
1125 if ((control & NETFLAG_LENGTH_MASK) != ret)
1129 if (c != CCREP_SERVER_INFO)
1132 dfunc.GetAddrFromName(MSG_ReadString(), &readaddr);
1133 // search the cache for this server
1134 for (n = 0; n < hostCacheCount; n++)
1135 if (dfunc.AddrCompare(&readaddr, &hostcache[n].addr) == 0)
1138 // is it already there?
1139 if (n < hostCacheCount)
1144 strcpy(hostcache[n].name, MSG_ReadString());
1145 strcpy(hostcache[n].map, MSG_ReadString());
1146 hostcache[n].users = MSG_ReadByte();
1147 hostcache[n].maxusers = MSG_ReadByte();
1149 if (c != NET_PROTOCOL_VERSION)
1151 strcpy(hostcache[n].cname, hostcache[n].name);
1152 hostcache[n].cname[14] = 0;
1153 strcpy(hostcache[n].name, "*");
1154 strcat(hostcache[n].name, hostcache[n].cname);
1156 memcpy(&hostcache[n].addr, &readaddr, sizeof(struct qsockaddr));
1157 hostcache[n].driver = net_driverlevel;
1158 hostcache[n].ldriver = net_landriverlevel;
1159 strcpy(hostcache[n].cname, dfunc.AddrToString(&readaddr));
1161 // check for a name conflict
1162 for (i = 0; i < hostCacheCount; i++)
1166 if (Q_strcasecmp (hostcache[n].name, hostcache[i].name) == 0)
1168 i = strlen(hostcache[n].name);
1169 if (i < 15 && hostcache[n].name[i-1] > '8')
1171 hostcache[n].name[i] = '0';
1172 hostcache[n].name[i+1] = 0;
1175 hostcache[n].name[i-1]++;
1182 void Datagram_SearchForHosts (qboolean xmit)
1184 for (net_landriverlevel = 0; net_landriverlevel < net_numlandrivers; net_landriverlevel++)
1186 if (hostCacheCount == HOSTCACHESIZE)
1188 if (net_landrivers[net_landriverlevel].initialized)
1189 _Datagram_SearchForHosts (xmit);
1194 static qsocket_t *_Datagram_Connect (char *host)
1196 struct qsockaddr sendaddr;
1197 struct qsockaddr readaddr;
1206 // see if we can resolve the host name
1207 if (dfunc.GetAddrFromName(host, &sendaddr) == -1)
1210 newsock = dfunc.OpenSocket (cl_port.integer);
1214 sock = NET_NewQSocket ();
1217 sock->socket = newsock;
1218 sock->landriver = net_landriverlevel;
1220 // connect to the host
1221 if (dfunc.Connect (newsock, &sendaddr) == -1)
1224 // send the connection request
1225 Con_Printf("trying...\n");CL_UpdateScreen();CL_UpdateScreen();
1226 start_time = net_time;
1228 for (reps = 0; reps < 3; reps++)
1230 SZ_Clear(&net_message);
1231 // save space for the header, filled in later
1232 MSG_WriteLong(&net_message, 0);
1233 MSG_WriteByte(&net_message, CCREQ_CONNECT);
1234 MSG_WriteString(&net_message, "QUAKE");
1235 MSG_WriteByte(&net_message, NET_PROTOCOL_VERSION);
1236 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
1237 dfunc.Write (newsock, net_message.data, net_message.cursize, &sendaddr);
1238 SZ_Clear(&net_message);
1241 ret = dfunc.Read (newsock, net_message.data, net_message.maxsize, &readaddr);
1242 // if we got something, validate it
1245 // is it from the right place?
1246 if (sfunc.AddrCompare(&readaddr, &sendaddr) != 0)
1248 Con_DPrintf("wrong reply address\n");
1249 Con_DPrintf("Expected: %s\n", StrAddr (&sendaddr));
1250 Con_DPrintf("Received: %s\n", StrAddr (&readaddr));
1256 if (ret < sizeof(int))
1262 net_message.cursize = ret;
1263 MSG_BeginReading ();
1265 control = BigLong(*((int *)net_message.data));
1272 if ((control & (~NETFLAG_LENGTH_MASK)) != NETFLAG_CTL)
1277 if ((control & NETFLAG_LENGTH_MASK) != ret)
1284 while (ret == 0 && (SetNetTime() - start_time) < 2.5);
1287 Con_Printf("still trying...\n");CL_UpdateScreen();CL_UpdateScreen();
1288 start_time = SetNetTime();
1293 reason = "No Response";
1294 Con_Printf("%s\n", reason);
1295 strcpy(m_return_reason, reason);
1301 reason = "Network Error";
1302 Con_Printf("%s\n", reason);
1303 strcpy(m_return_reason, reason);
1307 ret = MSG_ReadByte();
1308 if (ret == CCREP_REJECT)
1310 reason = MSG_ReadString();
1311 Con_Printf("%s", reason);
1312 strncpy(m_return_reason, reason, 31);
1316 if (ret == CCREP_ACCEPT)
1318 memcpy(&sock->addr, &sendaddr, sizeof(struct qsockaddr));
1319 dfunc.SetSocketPort (&sock->addr, MSG_ReadLong());
1323 reason = "Bad Response";
1324 Con_Printf("%s\n", reason);
1325 strcpy(m_return_reason, reason);
1329 dfunc.GetNameFromAddr (&sendaddr, sock->address);
1331 Con_Printf ("Connection accepted to %s\n", sock->address);
1332 sock->lastMessageTime = SetNetTime();
1334 // switch the connection to the specified address
1335 if (dfunc.Connect (newsock, &sock->addr) == -1)
1337 reason = "Connect to Game failed";
1338 Con_Printf("%s\n", reason);
1339 strcpy(m_return_reason, reason);
1343 m_return_onerror = false;
1347 NET_FreeQSocket(sock);
1349 dfunc.CloseSocket(newsock);
1350 if (m_return_onerror)
1352 key_dest = key_menu;
1353 m_state = m_return_state;
1354 m_return_onerror = false;
1359 qsocket_t *Datagram_Connect (char *host)
1361 qsocket_t *ret = NULL;
1363 for (net_landriverlevel = 0; net_landriverlevel < net_numlandrivers; net_landriverlevel++)
1364 if (net_landrivers[net_landriverlevel].initialized)
1365 if ((ret = _Datagram_Connect (host)) != NULL)