1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
|
#include <ctime>
#include "masterserver.h"
void CMasterServer::Init(int port)
{
netaddr4 addr(0, 0, 0, 0, port);
addr.port = port;
net_init();
m_UDPSocket.open(port);
m_TCPSocket.open(&addr);
m_TCPSocket.set_non_blocking();
m_TCPSocket.listen();
}
void CMasterServer::Shutdown()
{
m_UDPSocket.close();
}
void CMasterServer::Tick()
{
m_CurrentTime = time(NULL);
ListenForHeartBeats();
ListenForServerListPolls();
CleanUpServerList();
}
void CMasterServer::ListenForHeartBeats()
{
netaddr4 from;
char data[1024];
int dataSize;
// read udp data while there is data to read :)
while ((dataSize = m_UDPSocket.recv(&from, (char *)data, sizeof(data))) > 0)
{
// compare the received data size to the expected size
if (dataSize == HEARTBEAT_SIZE)
{
char *d = data;
int32 signature;
d = ReadInt32(d, &signature);
// make sure the signature is correct
if (signature == HEARTBEAT_SIGNATURE)
{
CServerInfo info;
info.Deserialize(d);
from.port = 8303;
info.SetAddress(&from);
dbg_msg("got heartbeat", "IP: %i.%i.%i.%i:%i", (int)from.ip[0], (int)from.ip[1], (int)from.ip[2], (int)from.ip[3], from.port);
dbg_msg("refresh", "okay. server count: %i", m_ServerCount);
ProcessHeartBeat(info);
}
else
{} // unexpected signature
}
else
{} // unknown data received
}
}
void CMasterServer::ProcessHeartBeat(CServerInfo info)
{
// find the corresponding server info
CServerInfo *serverInfo = FindServerInfo(info.IP(), info.Port());
// if it isn't in the list already, try to get an unused slot
if (!serverInfo)
serverInfo = GetUnusedSlot();
// if we STILL don't have one, we're out of luck.
if (!serverInfo)
return;
*serverInfo = info;
serverInfo->Refresh(m_CurrentTime);
// mark the server list packet as old
m_ServerListPacketIsOld = true;
}
CServerInfo *CMasterServer::FindServerInfo(int32 ip, int32 port)
{
// for now, just loop over the array
for (int i = 0; i < m_ServerCount; i++)
{
CServerInfo *info = &m_Servers[i];
if (info->IP() == ip && info->Port() == port)
return info;
}
return 0x0;
}
CServerInfo *CMasterServer::GetUnusedSlot()
{
if (m_ServerCount == MAXSERVERS)
return 0x0;
else
return &m_Servers[m_ServerCount++];
}
void CMasterServer::CleanUpServerList()
{
for (int i = 0; i < m_ServerCount; i++)
{
CServerInfo *serverInfo = &m_Servers[i];
// check if it's time to remove it from the list
if (serverInfo->LastRefresh() + HEARTBEAT_LIFETIME < m_CurrentTime)
{
if (i + 1 == m_ServerCount)
{
// if we're at the last one, just decrease m_ServerCount
--m_ServerCount;
}
else
{
// otherwise, copy the last slot here and then decrease i and m_ServerCount
*serverInfo = m_Servers[m_ServerCount - 1];
--i;
--m_ServerCount;
}
// mark the server list packet as old and outdated
m_ServerListPacketIsOld = true;
}
}
}
void CMasterServer::ListenForServerListPolls()
{
socket_tcp4 client;
// accept clients while there are clients to be accepted
while (m_TCPSocket.accept(&client))
{
// maybe we've prepared the packet already... it'd be silly to do it twice
if (m_ServerListPacketIsOld)
{
BuildServerListPacket();
}
// send the server list and then close the socket
client.send(m_ServerListPacket, m_ServerListPacketSize);
client.close();
}
}
void CMasterServer::BuildServerListPacket()
{
char *d = m_ServerListPacket;
d = WriteInt32(d, 'TWSL');
d = WriteInt32(d, MASTERSERVER_VERSION);
d = WriteInt32(d, m_ServerCount);
for (int i = 0; i < m_ServerCount; i++)
{
CServerInfo *info = &m_Servers[i];
d = info->Serialize(d);
}
m_ServerListPacketSize = d - m_ServerListPacket;
m_ServerListPacketIsOld = false;
}
|