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) (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);
329 // if ((rand() & 255) > 220)
337 Con_Printf("Read error\n");
341 if (sfunc.AddrCompare(&readaddr, &sock->addr) != 0)
344 Con_DPrintf("Forged packet received\n");
345 Con_DPrintf("Expected: %s\n", StrAddr (&sock->addr));
346 Con_DPrintf("Received: %s\n", StrAddr (&readaddr));
351 if (length < NET_HEADERSIZE)
357 length = BigLong(packetBuffer.length);
358 flags = length & (~NETFLAG_LENGTH_MASK);
359 length &= NETFLAG_LENGTH_MASK;
361 if (flags & NETFLAG_CTL)
364 sequence = BigLong(packetBuffer.sequence);
367 if (flags & NETFLAG_UNRELIABLE)
369 if (sequence < sock->unreliableReceiveSequence)
371 Con_DPrintf("Got a stale datagram\n");
375 if (sequence != sock->unreliableReceiveSequence)
377 count = sequence - sock->unreliableReceiveSequence;
378 droppedDatagrams += count;
379 Con_DPrintf("Dropped %u datagram(s)\n", count);
381 sock->unreliableReceiveSequence = sequence + 1;
383 length -= NET_HEADERSIZE;
385 SZ_Clear (&net_message);
386 SZ_Write (&net_message, packetBuffer.data, length);
392 if (flags & NETFLAG_ACK)
394 if (sequence != (sock->sendSequence - 1))
396 Con_DPrintf("Stale ACK received\n");
399 if (sequence == sock->ackSequence)
402 if (sock->ackSequence != sock->sendSequence)
403 Con_DPrintf("ack sequencing error\n");
407 Con_DPrintf("Duplicate ACK received\n");
410 sock->sendMessageLength -= MAX_DATAGRAM;
411 if (sock->sendMessageLength > 0)
413 memcpy(sock->sendMessage, sock->sendMessage+MAX_DATAGRAM, sock->sendMessageLength);
414 sock->sendNext = true;
418 sock->sendMessageLength = 0;
419 sock->canSend = true;
424 if (flags & NETFLAG_DATA)
426 packetBuffer.length = BigLong(NET_HEADERSIZE | NETFLAG_ACK);
427 packetBuffer.sequence = BigLong(sequence);
428 sfunc.Write (sock->socket, (qbyte *)&packetBuffer, NET_HEADERSIZE, &readaddr);
430 if (sequence != sock->receiveSequence)
432 receivedDuplicateCount++;
435 sock->receiveSequence++;
437 length -= NET_HEADERSIZE;
439 if (flags & NETFLAG_EOM)
441 SZ_Clear(&net_message);
442 SZ_Write(&net_message, sock->receiveMessage, sock->receiveMessageLength);
443 SZ_Write(&net_message, packetBuffer.data, length);
444 sock->receiveMessageLength = 0;
450 memcpy(sock->receiveMessage + sock->receiveMessageLength, packetBuffer.data, length);
451 sock->receiveMessageLength += length;
457 SendMessageNext (sock);
463 void PrintStats(qsocket_t *s)
465 Con_Printf("canSend = %4u \n", s->canSend);
466 Con_Printf("sendSeq = %4u ", s->sendSequence);
467 Con_Printf("recvSeq = %4u \n", s->receiveSequence);
471 void NET_Stats_f (void)
475 if (Cmd_Argc () == 1)
477 Con_Printf("unreliable messages sent = %i\n", unreliableMessagesSent);
478 Con_Printf("unreliable messages recv = %i\n", unreliableMessagesReceived);
479 Con_Printf("reliable messages sent = %i\n", messagesSent);
480 Con_Printf("reliable messages received = %i\n", messagesReceived);
481 Con_Printf("packetsSent = %i\n", packetsSent);
482 Con_Printf("packetsReSent = %i\n", packetsReSent);
483 Con_Printf("packetsReceived = %i\n", packetsReceived);
484 Con_Printf("receivedDuplicateCount = %i\n", receivedDuplicateCount);
485 Con_Printf("shortPacketCount = %i\n", shortPacketCount);
486 Con_Printf("droppedDatagrams = %i\n", droppedDatagrams);
488 else if (strcmp(Cmd_Argv(1), "*") == 0)
490 for (s = net_activeSockets; s; s = s->next)
492 // LordHavoc: sockets are dynamically allocated now
493 //for (s = net_freeSockets; s; s = s->next)
498 for (s = net_activeSockets; s; s = s->next)
499 if (Q_strcasecmp(Cmd_Argv(1), s->address) == 0)
501 // LordHavoc: sockets are dynamically allocated now
503 // for (s = net_freeSockets; s; s = s->next)
504 // if (Q_strcasecmp(Cmd_Argv(1), s->address) == 0)
513 static qboolean testInProgress = false;
514 static int testPollCount;
515 static int testDriver;
516 static int testSocket;
518 static void Test_Poll(void);
519 PollProcedure testPollProcedure = {NULL, 0.0, Test_Poll};
521 static void Test_Poll(void)
523 struct qsockaddr clientaddr;
533 net_landriverlevel = testDriver;
537 len = dfunc.Read (testSocket, net_message.data, net_message.maxsize, &clientaddr);
538 if (len < sizeof(int))
541 net_message.cursize = len;
544 control = BigLong(*((int *)net_message.data));
548 if ((control & (~NETFLAG_LENGTH_MASK)) != NETFLAG_CTL)
550 if ((control & NETFLAG_LENGTH_MASK) != len)
553 if (MSG_ReadByte() != CCREP_PLAYER_INFO)
554 Sys_Error("Unexpected repsonse to Player Info request\n");
556 playerNumber = MSG_ReadByte();
557 strcpy(name, MSG_ReadString());
558 colors = MSG_ReadLong();
559 frags = MSG_ReadLong();
560 connectTime = MSG_ReadLong();
561 strcpy(address, MSG_ReadString());
563 Con_Printf("%s\n frags:%3i colors:%u %u time:%u\n %s\n", name, frags, colors >> 4, colors & 0x0f, connectTime / 60, address);
569 SchedulePollProcedure(&testPollProcedure, 0.1);
573 dfunc.CloseSocket(testSocket);
574 testInProgress = false;
578 static void Test_f (void)
582 int max = MAX_SCOREBOARD;
583 struct qsockaddr sendaddr;
590 if (host && hostCacheCount)
592 for (n = 0; n < hostCacheCount; n++)
593 if (Q_strcasecmp (host, hostcache[n].name) == 0)
595 if (hostcache[n].driver != myDriverLevel)
597 net_landriverlevel = hostcache[n].ldriver;
598 max = hostcache[n].maxusers;
599 memcpy(&sendaddr, &hostcache[n].addr, sizeof(struct qsockaddr));
602 if (n < hostCacheCount)
606 for (net_landriverlevel = 0; net_landriverlevel < net_numlandrivers; net_landriverlevel++)
608 if (!net_landrivers[net_landriverlevel].initialized)
611 // see if we can resolve the host name
612 if (dfunc.GetAddrFromName(host, &sendaddr) != -1)
615 if (net_landriverlevel == net_numlandrivers)
619 testSocket = dfunc.OpenSocket(0);
620 if (testSocket == -1)
623 testInProgress = true;
625 testDriver = net_landriverlevel;
627 for (n = 0; n < max; n++)
629 SZ_Clear(&net_message);
630 // save space for the header, filled in later
631 MSG_WriteLong(&net_message, 0);
632 MSG_WriteByte(&net_message, CCREQ_PLAYER_INFO);
633 MSG_WriteByte(&net_message, n);
634 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
635 dfunc.Write (testSocket, net_message.data, net_message.cursize, &sendaddr);
637 SZ_Clear(&net_message);
638 SchedulePollProcedure(&testPollProcedure, 0.1);
642 static qboolean test2InProgress = false;
643 static int test2Driver;
644 static int test2Socket;
646 static void Test2_Poll(void);
647 PollProcedure test2PollProcedure = {NULL, 0.0, Test2_Poll};
649 static void Test2_Poll(void)
651 struct qsockaddr clientaddr;
657 net_landriverlevel = test2Driver;
660 len = dfunc.Read (test2Socket, net_message.data, net_message.maxsize, &clientaddr);
661 if (len < sizeof(int))
664 net_message.cursize = len;
667 control = BigLong(*((int *)net_message.data));
671 if ((control & (~NETFLAG_LENGTH_MASK)) != NETFLAG_CTL)
673 if ((control & NETFLAG_LENGTH_MASK) != len)
676 if (MSG_ReadByte() != CCREP_RULE_INFO)
679 strcpy(name, MSG_ReadString());
682 strcpy(value, MSG_ReadString());
684 Con_Printf("%-16.16s %-16.16s\n", name, value);
686 SZ_Clear(&net_message);
687 // save space for the header, filled in later
688 MSG_WriteLong(&net_message, 0);
689 MSG_WriteByte(&net_message, CCREQ_RULE_INFO);
690 MSG_WriteString(&net_message, name);
691 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
692 dfunc.Write (test2Socket, net_message.data, net_message.cursize, &clientaddr);
693 SZ_Clear(&net_message);
696 SchedulePollProcedure(&test2PollProcedure, 0.05);
700 Con_Printf("Unexpected repsonse to Rule Info request\n");
702 dfunc.CloseSocket(test2Socket);
703 test2InProgress = false;
707 static void Test2_f (void)
711 struct qsockaddr sendaddr;
718 if (host && hostCacheCount)
720 for (n = 0; n < hostCacheCount; n++)
721 if (Q_strcasecmp (host, hostcache[n].name) == 0)
723 if (hostcache[n].driver != myDriverLevel)
725 net_landriverlevel = hostcache[n].ldriver;
726 memcpy(&sendaddr, &hostcache[n].addr, sizeof(struct qsockaddr));
729 if (n < hostCacheCount)
733 for (net_landriverlevel = 0; net_landriverlevel < net_numlandrivers; net_landriverlevel++)
735 if (!net_landrivers[net_landriverlevel].initialized)
738 // see if we can resolve the host name
739 if (dfunc.GetAddrFromName(host, &sendaddr) != -1)
742 if (net_landriverlevel == net_numlandrivers)
746 test2Socket = dfunc.OpenSocket(0);
747 if (test2Socket == -1)
750 test2InProgress = true;
751 test2Driver = net_landriverlevel;
753 SZ_Clear(&net_message);
754 // save space for the header, filled in later
755 MSG_WriteLong(&net_message, 0);
756 MSG_WriteByte(&net_message, CCREQ_RULE_INFO);
757 MSG_WriteString(&net_message, "");
758 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
759 dfunc.Write (test2Socket, net_message.data, net_message.cursize, &sendaddr);
760 SZ_Clear(&net_message);
761 SchedulePollProcedure(&test2PollProcedure, 0.05);
765 int Datagram_Init (void)
770 myDriverLevel = net_driverlevel;
771 Cmd_AddCommand ("net_stats", NET_Stats_f);
772 Cvar_RegisterVariable (&cl_port);
774 if (COM_CheckParm("-nolan"))
777 for (i = 0; i < net_numlandrivers; i++)
779 csock = net_landrivers[i].Init ();
782 net_landrivers[i].initialized = true;
783 net_landrivers[i].controlSock = csock;
787 Cmd_AddCommand ("ban", NET_Ban_f);
789 Cmd_AddCommand ("test", Test_f);
790 Cmd_AddCommand ("test2", Test2_f);
796 void Datagram_Shutdown (void)
801 // shutdown the lan drivers
803 for (i = 0; i < net_numlandrivers; i++)
805 if (net_landrivers[i].initialized)
807 net_landrivers[i].Shutdown ();
808 net_landrivers[i].initialized = false;
814 void Datagram_Close (qsocket_t *sock)
816 sfunc.CloseSocket(sock->socket);
820 void Datagram_Listen (qboolean state)
824 for (i = 0; i < net_numlandrivers; i++)
825 if (net_landrivers[i].initialized)
826 net_landrivers[i].Listen (state);
830 static qsocket_t *_Datagram_CheckNewConnections (void)
832 struct qsockaddr clientaddr;
833 struct qsockaddr newaddr;
843 acceptsock = dfunc.CheckNewConnections();
844 if (acceptsock == -1)
847 SZ_Clear(&net_message);
849 len = dfunc.Read (acceptsock, net_message.data, net_message.maxsize, &clientaddr);
850 if (len < sizeof(int))
852 net_message.cursize = len;
855 control = BigLong(*((int *)net_message.data));
859 if ((control & (~NETFLAG_LENGTH_MASK)) != NETFLAG_CTL)
861 if ((control & NETFLAG_LENGTH_MASK) != len)
864 command = MSG_ReadByte();
865 if (command == CCREQ_SERVER_INFO)
867 if (strcmp(MSG_ReadString(), "QUAKE") != 0)
870 SZ_Clear(&net_message);
871 // save space for the header, filled in later
872 MSG_WriteLong(&net_message, 0);
873 MSG_WriteByte(&net_message, CCREP_SERVER_INFO);
874 dfunc.GetSocketAddr(acceptsock, &newaddr);
875 MSG_WriteString(&net_message, dfunc.AddrToString(&newaddr));
876 MSG_WriteString(&net_message, hostname.string);
877 MSG_WriteString(&net_message, sv.name);
878 MSG_WriteByte(&net_message, net_activeconnections);
879 MSG_WriteByte(&net_message, svs.maxclients);
880 MSG_WriteByte(&net_message, NET_PROTOCOL_VERSION);
881 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
882 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
883 SZ_Clear(&net_message);
887 if (command == CCREQ_PLAYER_INFO)
894 playerNumber = MSG_ReadByte();
896 for (clientNumber = 0, client = svs.clients; clientNumber < svs.maxclients; clientNumber++, client++)
901 if (activeNumber == playerNumber)
905 if (clientNumber == svs.maxclients)
908 SZ_Clear(&net_message);
909 // save space for the header, filled in later
910 MSG_WriteLong(&net_message, 0);
911 MSG_WriteByte(&net_message, CCREP_PLAYER_INFO);
912 MSG_WriteByte(&net_message, playerNumber);
913 MSG_WriteString(&net_message, client->name);
914 MSG_WriteLong(&net_message, client->colors);
915 MSG_WriteLong(&net_message, (int)client->edict->v.frags);
916 MSG_WriteLong(&net_message, (int)(net_time - client->netconnection->connecttime));
917 MSG_WriteString(&net_message, client->netconnection->address);
918 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
919 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
920 SZ_Clear(&net_message);
925 if (command == CCREQ_RULE_INFO)
930 // find the search start location
931 prevCvarName = MSG_ReadString();
934 var = Cvar_FindVar (prevCvarName);
942 // search for the next server cvar
945 if (var->flags & CVAR_NOTIFY)
952 SZ_Clear(&net_message);
953 // save space for the header, filled in later
954 MSG_WriteLong(&net_message, 0);
955 MSG_WriteByte(&net_message, CCREP_RULE_INFO);
958 MSG_WriteString(&net_message, var->name);
959 MSG_WriteString(&net_message, var->string);
961 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
962 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
963 SZ_Clear(&net_message);
968 if (command != CCREQ_CONNECT)
971 if (strcmp(MSG_ReadString(), "QUAKE") != 0)
974 if (MSG_ReadByte() != NET_PROTOCOL_VERSION)
976 SZ_Clear(&net_message);
977 // save space for the header, filled in later
978 MSG_WriteLong(&net_message, 0);
979 MSG_WriteByte(&net_message, CCREP_REJECT);
980 MSG_WriteString(&net_message, "Incompatible version.\n");
981 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
982 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
983 SZ_Clear(&net_message);
989 if (clientaddr.sa_family == AF_INET)
991 unsigned long testAddr;
992 testAddr = ((struct sockaddr_in *)&clientaddr)->sin_addr.s_addr;
993 if ((testAddr & banMask) == banAddr)
995 SZ_Clear(&net_message);
996 // save space for the header, filled in later
997 MSG_WriteLong(&net_message, 0);
998 MSG_WriteByte(&net_message, CCREP_REJECT);
999 MSG_WriteString(&net_message, "You have been banned.\n");
1000 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
1001 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
1002 SZ_Clear(&net_message);
1008 // see if this guy is already connected
1009 for (s = net_activeSockets; s; s = s->next)
1011 if (s->driver != net_driverlevel)
1013 ret = dfunc.AddrCompare(&clientaddr, &s->addr);
1016 // is this a duplicate connection reqeust?
1017 if (ret == 0 && net_time - s->connecttime < 2.0)
1019 // yes, so send a duplicate reply
1020 SZ_Clear(&net_message);
1021 // save space for the header, filled in later
1022 MSG_WriteLong(&net_message, 0);
1023 MSG_WriteByte(&net_message, CCREP_ACCEPT);
1024 dfunc.GetSocketAddr(s->socket, &newaddr);
1025 MSG_WriteLong(&net_message, dfunc.GetSocketPort(&newaddr));
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);
1031 // it's somebody coming back in from a crash/disconnect
1032 // so close the old qsocket and let their retry get them back in
1038 // allocate a QSocket
1039 sock = NET_NewQSocket ();
1042 // no room; try to let him know
1043 SZ_Clear(&net_message);
1044 // save space for the header, filled in later
1045 MSG_WriteLong(&net_message, 0);
1046 MSG_WriteByte(&net_message, CCREP_REJECT);
1047 MSG_WriteString(&net_message, "Server is full.\n");
1048 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
1049 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
1050 SZ_Clear(&net_message);
1054 // allocate a network socket
1055 newsock = dfunc.OpenSocket(0);
1058 NET_FreeQSocket(sock);
1062 // connect to the client
1063 if (dfunc.Connect (newsock, &clientaddr) == -1)
1065 dfunc.CloseSocket(newsock);
1066 NET_FreeQSocket(sock);
1070 // everything is allocated, just fill in the details
1071 sock->socket = newsock;
1072 sock->landriver = net_landriverlevel;
1073 sock->addr = clientaddr;
1074 strcpy(sock->address, dfunc.AddrToString(&clientaddr));
1076 // send him back the info about the server connection he has been allocated
1077 SZ_Clear(&net_message);
1078 // save space for the header, filled in later
1079 MSG_WriteLong(&net_message, 0);
1080 MSG_WriteByte(&net_message, CCREP_ACCEPT);
1081 dfunc.GetSocketAddr(newsock, &newaddr);
1082 MSG_WriteLong(&net_message, dfunc.GetSocketPort(&newaddr));
1083 // MSG_WriteString(&net_message, dfunc.AddrToString(&newaddr));
1084 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
1085 dfunc.Write (acceptsock, net_message.data, net_message.cursize, &clientaddr);
1086 SZ_Clear(&net_message);
1091 qsocket_t *Datagram_CheckNewConnections (void)
1093 qsocket_t *ret = NULL;
1095 for (net_landriverlevel = 0; net_landriverlevel < net_numlandrivers; net_landriverlevel++)
1096 if (net_landrivers[net_landriverlevel].initialized)
1097 if ((ret = _Datagram_CheckNewConnections ()) != NULL)
1103 static void _Datagram_SearchForHosts (qboolean xmit)
1108 struct qsockaddr readaddr;
1109 struct qsockaddr myaddr;
1112 dfunc.GetSocketAddr (dfunc.controlSock, &myaddr);
1115 SZ_Clear(&net_message);
1116 // save space for the header, filled in later
1117 MSG_WriteLong(&net_message, 0);
1118 MSG_WriteByte(&net_message, CCREQ_SERVER_INFO);
1119 MSG_WriteString(&net_message, "QUAKE");
1120 MSG_WriteByte(&net_message, NET_PROTOCOL_VERSION);
1121 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
1122 dfunc.Broadcast(dfunc.controlSock, net_message.data, net_message.cursize);
1123 SZ_Clear(&net_message);
1126 while ((ret = dfunc.Read (dfunc.controlSock, net_message.data, net_message.maxsize, &readaddr)) > 0)
1128 if (ret < sizeof(int))
1130 net_message.cursize = ret;
1132 // don't answer our own query
1133 if (dfunc.AddrCompare(&readaddr, &myaddr) >= 0)
1136 // is the cache full?
1137 if (hostCacheCount == HOSTCACHESIZE)
1140 MSG_BeginReading ();
1141 control = BigLong(*((int *)net_message.data));
1145 if ((control & (~NETFLAG_LENGTH_MASK)) != NETFLAG_CTL)
1147 if ((control & NETFLAG_LENGTH_MASK) != ret)
1150 if (MSG_ReadByte() != CCREP_SERVER_INFO)
1153 dfunc.GetAddrFromName(MSG_ReadString(), &readaddr);
1154 // search the cache for this server
1155 for (n = 0; n < hostCacheCount; n++)
1156 if (dfunc.AddrCompare(&readaddr, &hostcache[n].addr) == 0)
1159 // is it already there?
1160 if (n < hostCacheCount)
1165 strcpy(hostcache[n].name, MSG_ReadString());
1166 strcpy(hostcache[n].map, MSG_ReadString());
1167 hostcache[n].users = MSG_ReadByte();
1168 hostcache[n].maxusers = MSG_ReadByte();
1169 if (MSG_ReadByte() != NET_PROTOCOL_VERSION)
1171 strcpy(hostcache[n].cname, hostcache[n].name);
1172 hostcache[n].cname[14] = 0;
1173 strcpy(hostcache[n].name, "*");
1174 strcat(hostcache[n].name, hostcache[n].cname);
1176 memcpy(&hostcache[n].addr, &readaddr, sizeof(struct qsockaddr));
1177 hostcache[n].driver = net_driverlevel;
1178 hostcache[n].ldriver = net_landriverlevel;
1179 strcpy(hostcache[n].cname, dfunc.AddrToString(&readaddr));
1181 // check for a name conflict
1182 for (i = 0; i < hostCacheCount; i++)
1186 if (Q_strcasecmp (hostcache[n].name, hostcache[i].name) == 0)
1188 i = strlen(hostcache[n].name);
1189 if (i < 15 && hostcache[n].name[i-1] > '8')
1191 hostcache[n].name[i] = '0';
1192 hostcache[n].name[i+1] = 0;
1195 hostcache[n].name[i-1]++;
1202 void Datagram_SearchForHosts (qboolean xmit)
1204 for (net_landriverlevel = 0; net_landriverlevel < net_numlandrivers; net_landriverlevel++)
1206 if (hostCacheCount == HOSTCACHESIZE)
1208 if (net_landrivers[net_landriverlevel].initialized)
1209 _Datagram_SearchForHosts (xmit);
1214 static qsocket_t *_Datagram_Connect (char *host)
1216 struct qsockaddr sendaddr;
1217 struct qsockaddr readaddr;
1226 // see if we can resolve the host name
1227 if (dfunc.GetAddrFromName(host, &sendaddr) == -1)
1230 newsock = dfunc.OpenSocket (cl_port.integer);
1234 sock = NET_NewQSocket ();
1237 sock->socket = newsock;
1238 sock->landriver = net_landriverlevel;
1240 // connect to the host
1241 if (dfunc.Connect (newsock, &sendaddr) == -1)
1244 // send the connection request
1245 Con_Printf("trying...\n");CL_UpdateScreen();CL_UpdateScreen();
1246 start_time = net_time;
1248 for (reps = 0; reps < 3; reps++)
1250 SZ_Clear(&net_message);
1251 // save space for the header, filled in later
1252 MSG_WriteLong(&net_message, 0);
1253 MSG_WriteByte(&net_message, CCREQ_CONNECT);
1254 MSG_WriteString(&net_message, "QUAKE");
1255 MSG_WriteByte(&net_message, NET_PROTOCOL_VERSION);
1256 *((int *)net_message.data) = BigLong(NETFLAG_CTL | (net_message.cursize & NETFLAG_LENGTH_MASK));
1257 dfunc.Write (newsock, net_message.data, net_message.cursize, &sendaddr);
1258 SZ_Clear(&net_message);
1261 ret = dfunc.Read (newsock, net_message.data, net_message.maxsize, &readaddr);
1262 // if we got something, validate it
1265 // is it from the right place?
1266 if (sfunc.AddrCompare(&readaddr, &sendaddr) != 0)
1269 Con_DPrintf("wrong reply address\n");
1270 Con_DPrintf("Expected: %s\n", StrAddr (&sendaddr));
1271 Con_DPrintf("Received: %s\n", StrAddr (&readaddr));
1278 if (ret < sizeof(int))
1284 net_message.cursize = ret;
1285 MSG_BeginReading ();
1287 control = BigLong(*((int *)net_message.data));
1294 if ((control & (~NETFLAG_LENGTH_MASK)) != NETFLAG_CTL)
1299 if ((control & NETFLAG_LENGTH_MASK) != ret)
1306 while (ret == 0 && (SetNetTime() - start_time) < 2.5);
1309 Con_Printf("still trying...\n");CL_UpdateScreen();CL_UpdateScreen();
1310 start_time = SetNetTime();
1315 reason = "No Response";
1316 Con_Printf("%s\n", reason);
1317 strcpy(m_return_reason, reason);
1323 reason = "Network Error";
1324 Con_Printf("%s\n", reason);
1325 strcpy(m_return_reason, reason);
1329 ret = MSG_ReadByte();
1330 if (ret == CCREP_REJECT)
1332 reason = MSG_ReadString();
1333 Con_Printf("%s", reason);
1334 strncpy(m_return_reason, reason, 31);
1338 if (ret == CCREP_ACCEPT)
1340 memcpy(&sock->addr, &sendaddr, sizeof(struct qsockaddr));
1341 dfunc.SetSocketPort (&sock->addr, MSG_ReadLong());
1345 reason = "Bad Response";
1346 Con_Printf("%s\n", reason);
1347 strcpy(m_return_reason, reason);
1351 dfunc.GetNameFromAddr (&sendaddr, sock->address);
1353 Con_Printf ("Connection accepted to %s\n", sock->address);
1354 sock->lastMessageTime = SetNetTime();
1356 // switch the connection to the specified address
1357 if (dfunc.Connect (newsock, &sock->addr) == -1)
1359 reason = "Connect to Game failed";
1360 Con_Printf("%s\n", reason);
1361 strcpy(m_return_reason, reason);
1365 m_return_onerror = false;
1369 NET_FreeQSocket(sock);
1371 dfunc.CloseSocket(newsock);
1372 if (m_return_onerror)
1374 key_dest = key_menu;
1375 m_state = m_return_state;
1376 m_return_onerror = false;
1381 qsocket_t *Datagram_Connect (char *host)
1383 qsocket_t *ret = NULL;
1385 for (net_landriverlevel = 0; net_landriverlevel < net_numlandrivers; net_landriverlevel++)
1386 if (net_landrivers[net_landriverlevel].initialized)
1387 if ((ret = _Datagram_Connect (host)) != NULL)