ddnet/src/mastersrv/mastersrv.cpp

355 lines
8.8 KiB
C++
Raw Normal View History

2010-05-29 07:25:38 +00:00
// copyright (c) 2007 magnus auvinen, see licence.txt for more info
#include <base/system.h>
2010-05-29 07:25:38 +00:00
#include <engine/shared/network.h>
#include "mastersrv.h"
enum {
MTU = 1400,
MAX_SERVERS_PER_PACKET=128,
MAX_PACKETS=16,
MAX_SERVERS=MAX_SERVERS_PER_PACKET*MAX_PACKETS,
EXPIRE_TIME = 90
};
2010-05-29 07:25:38 +00:00
struct CCheckServer
2007-07-21 21:28:56 +00:00
{
2010-05-29 07:25:38 +00:00
NETADDR m_Address;
NETADDR m_AltAddress;
int m_TryCount;
int64 m_TryTime;
};
2007-07-21 21:28:56 +00:00
2010-05-29 07:25:38 +00:00
static CCheckServer m_aCheckServers[MAX_SERVERS];
static int m_NumCheckServers = 0;
2010-05-29 07:25:38 +00:00
struct NETADDR_IPv4
{
2010-05-29 07:25:38 +00:00
unsigned char m_aIp[4];
unsigned short m_Port;
};
2010-05-29 07:25:38 +00:00
struct CServerEntry
{
2010-05-29 07:25:38 +00:00
NETADDR m_Address;
int64 m_Expire;
};
static CServerEntry m_aServers[MAX_SERVERS];
static int m_NumServers = 0;
2010-05-29 07:25:38 +00:00
struct CPacketData
{
2010-05-29 07:25:38 +00:00
int m_Size;
struct {
2010-05-29 07:25:38 +00:00
unsigned char m_aHeader[sizeof(SERVERBROWSE_LIST)];
NETADDR_IPv4 m_aServers[MAX_SERVERS_PER_PACKET];
} m_Data;
};
2008-02-10 21:54:52 +00:00
2010-05-29 07:25:38 +00:00
CPacketData m_aPackets[MAX_PACKETS];
static int m_NumPackets = 0;
struct CCountPacketData
2008-02-10 21:54:52 +00:00
{
2010-05-29 07:25:38 +00:00
unsigned char m_Header[sizeof(SERVERBROWSE_COUNT)];
unsigned char m_High;
unsigned char m_Low;
};
static CCountPacketData m_CountData;
2008-02-10 21:54:52 +00:00
//static int64 server_expire[MAX_SERVERS];
2010-05-29 07:25:38 +00:00
static CNetClient m_NetChecker; // NAT/FW checker
static CNetClient m_NetOp; // main
2007-07-21 21:28:56 +00:00
2010-05-29 07:25:38 +00:00
void BuildPackets()
{
2010-05-29 07:25:38 +00:00
CServerEntry *pCurrent = &m_aServers[0];
int ServersLeft = m_NumServers;
m_NumPackets = 0;
while(ServersLeft && m_NumPackets < MAX_PACKETS)
{
2010-05-29 07:25:38 +00:00
int Chunk = ServersLeft;
if(Chunk > MAX_SERVERS_PER_PACKET)
Chunk = MAX_SERVERS_PER_PACKET;
ServersLeft -= Chunk;
// copy header
2010-05-29 07:25:38 +00:00
mem_copy(m_aPackets[m_NumPackets].m_Data.m_aHeader, SERVERBROWSE_LIST, sizeof(SERVERBROWSE_LIST));
// copy server addresses
2010-05-29 07:25:38 +00:00
for(int i = 0; i < Chunk; i++)
{
// TODO: ipv6 support
2010-05-29 07:25:38 +00:00
m_aPackets[m_NumPackets].m_Data.m_aServers[i].m_aIp[0] = pCurrent->m_Address.ip[0];
m_aPackets[m_NumPackets].m_Data.m_aServers[i].m_aIp[1] = pCurrent->m_Address.ip[1];
m_aPackets[m_NumPackets].m_Data.m_aServers[i].m_aIp[2] = pCurrent->m_Address.ip[2];
m_aPackets[m_NumPackets].m_Data.m_aServers[i].m_aIp[3] = pCurrent->m_Address.ip[3];
m_aPackets[m_NumPackets].m_Data.m_aServers[i].m_Port = pCurrent->m_Address.port;
pCurrent++;
}
2010-05-29 07:25:38 +00:00
m_aPackets[m_NumPackets].m_Size = sizeof(SERVERBROWSE_LIST) + sizeof(NETADDR_IPv4)*Chunk;
2010-05-29 07:25:38 +00:00
m_NumPackets++;
}
}
2010-05-29 07:25:38 +00:00
void SendOk(NETADDR *pAddr)
2007-07-21 21:28:56 +00:00
{
2010-05-29 07:25:38 +00:00
CNetChunk p;
p.m_ClientID = -1;
p.m_Address = *pAddr;
p.m_Flags = NETSENDFLAG_CONNLESS;
p.m_DataSize = sizeof(SERVERBROWSE_FWOK);
p.m_pData = SERVERBROWSE_FWOK;
2008-02-10 21:54:52 +00:00
// send on both to be sure
2010-05-29 07:25:38 +00:00
m_NetChecker.Send(&p);
m_NetOp.Send(&p);
2007-07-21 21:28:56 +00:00
}
2010-05-29 07:25:38 +00:00
void SendError(NETADDR *pAddr)
2007-07-21 21:28:56 +00:00
{
2010-05-29 07:25:38 +00:00
CNetChunk p;
p.m_ClientID = -1;
p.m_Address = *pAddr;
p.m_Flags = NETSENDFLAG_CONNLESS;
p.m_DataSize = sizeof(SERVERBROWSE_FWERROR);
p.m_pData = SERVERBROWSE_FWERROR;
m_NetOp.Send(&p);
2007-07-21 21:28:56 +00:00
}
2010-05-29 07:25:38 +00:00
void SendCheck(NETADDR *pAddr)
2007-07-21 21:28:56 +00:00
{
2010-05-29 07:25:38 +00:00
CNetChunk p;
p.m_ClientID = -1;
p.m_Address = *pAddr;
p.m_Flags = NETSENDFLAG_CONNLESS;
p.m_DataSize = sizeof(SERVERBROWSE_FWCHECK);
p.m_pData = SERVERBROWSE_FWCHECK;
m_NetChecker.Send(&p);
2007-07-21 21:28:56 +00:00
}
2010-05-29 07:25:38 +00:00
void AddCheckserver(NETADDR *pInfo, NETADDR *pAlt)
2007-07-21 21:28:56 +00:00
{
// add server
2010-05-29 07:25:38 +00:00
if(m_NumCheckServers == MAX_SERVERS)
2007-07-21 21:28:56 +00:00
{
dbg_msg("mastersrv", "error: mastersrv is full");
return;
}
dbg_msg("mastersrv", "checking: %d.%d.%d.%d:%d (%d.%d.%d.%d:%d)",
2010-05-29 07:25:38 +00:00
pInfo->ip[0], pInfo->ip[1], pInfo->ip[2], pInfo->ip[3], pInfo->port,
pAlt->ip[0], pAlt->ip[1], pAlt->ip[2], pAlt->ip[3], pAlt->port);
m_aCheckServers[m_NumCheckServers].m_Address = *pInfo;
m_aCheckServers[m_NumCheckServers].m_AltAddress = *pAlt;
m_aCheckServers[m_NumCheckServers].m_TryCount = 0;
m_aCheckServers[m_NumCheckServers].m_TryTime = 0;
m_NumCheckServers++;
2007-07-21 21:28:56 +00:00
}
2010-05-29 07:25:38 +00:00
void AddServer(NETADDR *pInfo)
{
// see if server already exists in list
int i;
2010-05-29 07:25:38 +00:00
for(i = 0; i < m_NumServers; i++)
{
2010-05-29 07:25:38 +00:00
if(net_addr_comp(&m_aServers[i].m_Address, pInfo) == 0)
{
dbg_msg("mastersrv", "updated: %d.%d.%d.%d:%d",
2010-05-29 07:25:38 +00:00
pInfo->ip[0], pInfo->ip[1], pInfo->ip[2], pInfo->ip[3], pInfo->port);
m_aServers[i].m_Expire = time_get()+time_freq()*EXPIRE_TIME;
return;
}
}
// add server
2010-05-29 07:25:38 +00:00
if(m_NumServers == MAX_SERVERS)
{
dbg_msg("mastersrv", "error: mastersrv is full");
return;
}
dbg_msg("mastersrv", "added: %d.%d.%d.%d:%d",
2010-05-29 07:25:38 +00:00
pInfo->ip[0], pInfo->ip[1], pInfo->ip[2], pInfo->ip[3], pInfo->port);
m_aServers[m_NumServers].m_Address = *pInfo;
m_aServers[m_NumServers].m_Expire = time_get()+time_freq()*EXPIRE_TIME;
m_NumServers++;
}
2010-05-29 07:25:38 +00:00
void UpdateServers()
2007-07-21 21:28:56 +00:00
{
2010-05-29 07:25:38 +00:00
int64 Now = time_get();
int64 Freq = time_freq();
for(int i = 0; i < m_NumCheckServers; i++)
2007-07-21 21:28:56 +00:00
{
2010-05-29 07:25:38 +00:00
if(Now > m_aCheckServers[i].m_TryTime+Freq)
2007-07-21 21:28:56 +00:00
{
2010-05-29 07:25:38 +00:00
if(m_aCheckServers[i].m_TryCount == 10)
2007-07-21 21:28:56 +00:00
{
dbg_msg("mastersrv", "check failed: %d.%d.%d.%d:%d",
2010-05-29 07:25:38 +00:00
m_aCheckServers[i].m_Address.ip[0], m_aCheckServers[i].m_Address.ip[1],
m_aCheckServers[i].m_Address.ip[2], m_aCheckServers[i].m_Address.ip[3],m_aCheckServers[i].m_Address.port,
m_aCheckServers[i].m_AltAddress.ip[0], m_aCheckServers[i].m_AltAddress.ip[1],
m_aCheckServers[i].m_AltAddress.ip[2], m_aCheckServers[i].m_AltAddress.ip[3],m_aCheckServers[i].m_AltAddress.port);
2007-07-21 21:28:56 +00:00
// FAIL!!
2010-05-29 07:25:38 +00:00
SendError(&m_aCheckServers[i].m_Address);
m_aCheckServers[i] = m_aCheckServers[m_NumCheckServers-1];
m_NumCheckServers--;
2007-07-21 21:28:56 +00:00
i--;
}
else
{
2010-05-29 07:25:38 +00:00
m_aCheckServers[i].m_TryCount++;
m_aCheckServers[i].m_TryTime = Now;
if(m_aCheckServers[i].m_TryCount&1)
SendCheck(&m_aCheckServers[i].m_Address);
else
2010-05-29 07:25:38 +00:00
SendCheck(&m_aCheckServers[i].m_AltAddress);
2007-07-21 21:28:56 +00:00
}
}
}
}
2010-05-29 07:25:38 +00:00
void PurgeServers()
{
2010-05-29 07:25:38 +00:00
int64 Now = time_get();
int i = 0;
2010-05-29 07:25:38 +00:00
while(i < m_NumServers)
{
2010-05-29 07:25:38 +00:00
if(m_aServers[i].m_Expire < Now)
{
// remove server
dbg_msg("mastersrv", "expired: %d.%d.%d.%d:%d",
2010-05-29 07:25:38 +00:00
m_aServers[i].m_Address.ip[0], m_aServers[i].m_Address.ip[1],
m_aServers[i].m_Address.ip[2], m_aServers[i].m_Address.ip[3], m_aServers[i].m_Address.port);
m_aServers[i] = m_aServers[m_NumServers-1];
m_NumServers--;
}
else
i++;
}
}
2010-05-29 07:25:38 +00:00
int main(int argc, char **argv) // ignore_convention
{
2010-05-29 07:25:38 +00:00
int64 LastBuild = 0;
NETADDR BindAddr;
2008-10-01 18:34:43 +00:00
dbg_logger_stdout();
net_init();
2010-05-29 07:25:38 +00:00
mem_zero(&BindAddr, sizeof(BindAddr));
BindAddr.port = MASTERSERVER_PORT;
m_NetOp.Open(BindAddr, 0);
2010-05-29 07:25:38 +00:00
BindAddr.port = MASTERSERVER_PORT+1;
m_NetChecker.Open(BindAddr, 0);
// TODO: check socket for errors
//mem_copy(data.header, SERVERBROWSE_LIST, sizeof(SERVERBROWSE_LIST));
2010-05-29 07:25:38 +00:00
mem_copy(m_CountData.m_Header, SERVERBROWSE_COUNT, sizeof(SERVERBROWSE_COUNT));
2008-02-10 21:54:52 +00:00
dbg_msg("mastersrv", "started");
while(1)
{
2010-05-29 07:25:38 +00:00
m_NetOp.Update();
m_NetChecker.Update();
2010-05-29 07:25:38 +00:00
// process m_aPackets
CNetChunk Packet;
while(m_NetOp.Recv(&Packet))
{
2010-05-29 07:25:38 +00:00
if(Packet.m_DataSize == sizeof(SERVERBROWSE_HEARTBEAT)+2 &&
mem_comp(Packet.m_pData, SERVERBROWSE_HEARTBEAT, sizeof(SERVERBROWSE_HEARTBEAT)) == 0)
{
2010-05-29 07:25:38 +00:00
NETADDR Alt;
unsigned char *d = (unsigned char *)Packet.m_pData;
Alt = Packet.m_Address;
Alt.port =
(d[sizeof(SERVERBROWSE_HEARTBEAT)]<<8) |
d[sizeof(SERVERBROWSE_HEARTBEAT)+1];
// add it
2010-05-29 07:25:38 +00:00
AddCheckserver(&Packet.m_Address, &Alt);
}
2010-05-29 07:25:38 +00:00
else if(Packet.m_DataSize == sizeof(SERVERBROWSE_GETCOUNT) &&
mem_comp(Packet.m_pData, SERVERBROWSE_GETCOUNT, sizeof(SERVERBROWSE_GETCOUNT)) == 0)
2008-02-10 21:54:52 +00:00
{
2010-05-29 07:25:38 +00:00
dbg_msg("mastersrv", "count requested, responding with %d", m_NumServers);
2008-02-10 21:54:52 +00:00
2010-05-29 07:25:38 +00:00
CNetChunk p;
p.m_ClientID = -1;
p.m_Address = Packet.m_Address;
p.m_Flags = NETSENDFLAG_CONNLESS;
p.m_DataSize = sizeof(m_CountData);
p.m_pData = &m_CountData;
m_CountData.m_High = (m_NumServers>>8)&0xff;
m_CountData.m_Low = m_NumServers&0xff;
m_NetOp.Send(&p);
2008-02-10 21:54:52 +00:00
}
2010-05-29 07:25:38 +00:00
else if(Packet.m_DataSize == sizeof(SERVERBROWSE_GETLIST) &&
mem_comp(Packet.m_pData, SERVERBROWSE_GETLIST, sizeof(SERVERBROWSE_GETLIST)) == 0)
{
// someone requested the list
2010-05-29 07:25:38 +00:00
dbg_msg("mastersrv", "requested, responding with %d m_aServers", m_NumServers);
CNetChunk p;
p.m_ClientID = -1;
p.m_Address = Packet.m_Address;
p.m_Flags = NETSENDFLAG_CONNLESS;
2010-05-29 07:25:38 +00:00
for(int i = 0; i < m_NumPackets; i++)
{
2010-05-29 07:25:38 +00:00
p.m_DataSize = m_aPackets[i].m_Size;
p.m_pData = &m_aPackets[i].m_Data;
m_NetOp.Send(&p);
}
}
2007-07-21 21:28:56 +00:00
}
2010-05-29 07:25:38 +00:00
// process m_aPackets
while(m_NetChecker.Recv(&Packet))
2007-07-21 21:28:56 +00:00
{
2010-05-29 07:25:38 +00:00
if(Packet.m_DataSize == sizeof(SERVERBROWSE_FWRESPONSE) &&
mem_comp(Packet.m_pData, SERVERBROWSE_FWRESPONSE, sizeof(SERVERBROWSE_FWRESPONSE)) == 0)
2007-07-21 21:28:56 +00:00
{
// remove it from checking
2010-05-29 07:25:38 +00:00
for(int i = 0; i < m_NumCheckServers; i++)
2007-07-21 21:28:56 +00:00
{
2010-05-29 07:25:38 +00:00
if(net_addr_comp(&m_aCheckServers[i].m_Address, &Packet.m_Address) == 0 ||
net_addr_comp(&m_aCheckServers[i].m_AltAddress, &Packet.m_Address) == 0)
2007-07-21 21:28:56 +00:00
{
2010-05-29 07:25:38 +00:00
m_NumCheckServers--;
m_aCheckServers[i] = m_aCheckServers[m_NumCheckServers];
2007-07-21 21:28:56 +00:00
break;
}
}
2010-05-29 07:25:38 +00:00
AddServer(&Packet.m_Address);
SendOk(&Packet.m_Address);
2007-07-21 21:28:56 +00:00
}
}
2010-05-29 07:25:38 +00:00
if(time_get()-LastBuild > time_freq()*5)
{
2010-05-29 07:25:38 +00:00
LastBuild = time_get();
2010-05-29 07:25:38 +00:00
PurgeServers();
UpdateServers();
BuildPackets();
}
// be nice to the CPU
thread_sleep(1);
}
return 0;
}