Alibek Omarov
7 years ago
13 changed files with 321 additions and 25 deletions
@ -0,0 +1,256 @@ |
|||||||
|
/*
|
||||||
|
masterlist.c - multi-master list |
||||||
|
Copyright (C) 2018 mittorn |
||||||
|
|
||||||
|
This program is free software: you can redistribute it and/or modify |
||||||
|
it under the terms of the GNU General Public License as published by |
||||||
|
the Free Software Foundation, either version 3 of the License, or |
||||||
|
(at your option) any later version. |
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful, |
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
GNU General Public License for more details. |
||||||
|
*/ |
||||||
|
#include "common.h" |
||||||
|
#include "netchan.h" |
||||||
|
|
||||||
|
typedef struct master_s |
||||||
|
{ |
||||||
|
struct master_s *next; |
||||||
|
qboolean sent; |
||||||
|
qboolean save; |
||||||
|
string address; |
||||||
|
} master_t; |
||||||
|
|
||||||
|
struct masterlist_s |
||||||
|
{ |
||||||
|
master_t *list; |
||||||
|
qboolean modified; |
||||||
|
} ml; |
||||||
|
|
||||||
|
/*
|
||||||
|
======================== |
||||||
|
NET_SendToMasters |
||||||
|
|
||||||
|
Send request to all masterservers list |
||||||
|
return true if would block |
||||||
|
======================== |
||||||
|
*/ |
||||||
|
qboolean NET_SendToMasters( netsrc_t sock, size_t len, const void *data ) |
||||||
|
{ |
||||||
|
master_t *list; |
||||||
|
qboolean wait = false; |
||||||
|
|
||||||
|
for( list = ml.list; list; list = list->next ) |
||||||
|
{ |
||||||
|
netadr_t adr; |
||||||
|
int res; |
||||||
|
|
||||||
|
if( list->sent ) |
||||||
|
continue; |
||||||
|
|
||||||
|
res = NET_StringToAdrNB( list->address, &adr ); |
||||||
|
|
||||||
|
if( !res ) |
||||||
|
{ |
||||||
|
MsgDev( D_INFO, "Can't resolve adr: %s\n", list->address ); |
||||||
|
list->sent = true; |
||||||
|
continue; |
||||||
|
} |
||||||
|
|
||||||
|
if( res == 2 ) |
||||||
|
{ |
||||||
|
list->sent = false; |
||||||
|
wait = true; |
||||||
|
continue; |
||||||
|
} |
||||||
|
|
||||||
|
list->sent = true; |
||||||
|
|
||||||
|
NET_SendPacket( sock, len, data, adr ); |
||||||
|
} |
||||||
|
|
||||||
|
if( !wait ) |
||||||
|
{ |
||||||
|
list = ml.list; |
||||||
|
|
||||||
|
while( list ) |
||||||
|
{ |
||||||
|
list->sent = false; |
||||||
|
list = list->next; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return wait; |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
======================== |
||||||
|
NET_AddMaster |
||||||
|
|
||||||
|
Add master to the list |
||||||
|
======================== |
||||||
|
*/ |
||||||
|
static void NET_AddMaster( char *addr, qboolean save ) |
||||||
|
{ |
||||||
|
master_t *master, *last; |
||||||
|
|
||||||
|
for( last = ml.list; last && last->next; last = last->next ) |
||||||
|
{ |
||||||
|
if( !Q_strcmp( last->address, addr ) ) // already exists
|
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
master = Mem_Alloc( host.mempool, sizeof( master_t ) ); |
||||||
|
Q_strncpy( master->address, addr, MAX_STRING ); |
||||||
|
master->sent = false; |
||||||
|
master->save = save; |
||||||
|
master->next = NULL; |
||||||
|
|
||||||
|
// link in
|
||||||
|
if( last ) |
||||||
|
last->next = master; |
||||||
|
else |
||||||
|
ml.list = master; |
||||||
|
} |
||||||
|
|
||||||
|
static void NET_AddMaster_f( void ) |
||||||
|
{ |
||||||
|
if( Cmd_Argc() != 2 ) |
||||||
|
{ |
||||||
|
Msg( S_USAGE "addmaster <address>\n"); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
NET_AddMaster( Cmd_Argv( 1 ), true ); // save them into config
|
||||||
|
ml.modified = true; // save config
|
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
======================== |
||||||
|
NET_ClearMasters |
||||||
|
|
||||||
|
Clear master list |
||||||
|
======================== |
||||||
|
*/ |
||||||
|
static void NET_ClearMasters_f( void ) |
||||||
|
{ |
||||||
|
while( ml.list ) |
||||||
|
{ |
||||||
|
master_t *prev = ml.list; |
||||||
|
ml.list = ml.list->next; |
||||||
|
Mem_Free( prev ); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
======================== |
||||||
|
NET_ListMasters_f |
||||||
|
|
||||||
|
Display current master linked list |
||||||
|
======================== |
||||||
|
*/ |
||||||
|
static void NET_ListMasters_f( void ) |
||||||
|
{ |
||||||
|
master_t *list; |
||||||
|
int i; |
||||||
|
|
||||||
|
Msg( "Master servers\n=============\n" ); |
||||||
|
|
||||||
|
|
||||||
|
for( i = 1, list = ml.list; list; i++, list = list->next ) |
||||||
|
{ |
||||||
|
Msg( "%d\t%s\n", i, list->address ); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
======================== |
||||||
|
NET_LoadMasters |
||||||
|
|
||||||
|
Load master server list from xashcomm.lst |
||||||
|
======================== |
||||||
|
*/ |
||||||
|
static void NET_LoadMasters( void ) |
||||||
|
{ |
||||||
|
byte *afile, *pfile; |
||||||
|
char token[MAX_TOKEN]; |
||||||
|
|
||||||
|
pfile = afile = FS_LoadFile( "xashcomm.lst", NULL, true ); |
||||||
|
|
||||||
|
if( !afile ) // file doesn't exist yet
|
||||||
|
{ |
||||||
|
MsgDev( D_INFO, "Cannot load xashcomm.lst\n" ); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
// format: master <addr>\n
|
||||||
|
while( ( pfile = COM_ParseFile( pfile, token ) ) ) |
||||||
|
{ |
||||||
|
if( !Q_strcmp( token, "master" ) ) // load addr
|
||||||
|
{ |
||||||
|
pfile = COM_ParseFile( pfile, token ); |
||||||
|
|
||||||
|
NET_AddMaster( token, true ); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
Mem_Free( afile ); |
||||||
|
|
||||||
|
ml.modified = false; |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
======================== |
||||||
|
NET_SaveMasters |
||||||
|
|
||||||
|
Save master server list to xashcomm.lst, except for default |
||||||
|
======================== |
||||||
|
*/ |
||||||
|
void NET_SaveMasters( void ) |
||||||
|
{ |
||||||
|
file_t *f; |
||||||
|
master_t *m; |
||||||
|
|
||||||
|
if( !ml.modified ) |
||||||
|
{ |
||||||
|
MsgDev( D_NOTE, "Master server list not changed\n" ); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
f = FS_Open( "xashcomm.lst", "w", true ); |
||||||
|
|
||||||
|
if( !f ) |
||||||
|
{ |
||||||
|
MsgDev( D_ERROR, "Couldn't write xashcomm.lst\n" ); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
for( m = ml.list; m; m = m->next ) |
||||||
|
{ |
||||||
|
if( m->save ) |
||||||
|
FS_Printf( f, "master %s\n", m->address ); |
||||||
|
} |
||||||
|
|
||||||
|
FS_Close( f ); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
======================== |
||||||
|
NET_InitMasters |
||||||
|
|
||||||
|
Initialize master server list |
||||||
|
======================== |
||||||
|
*/ |
||||||
|
void NET_InitMasters( void ) |
||||||
|
{ |
||||||
|
Cmd_AddRestrictedCommand( "addmaster", NET_AddMaster_f, "add address to masterserver list" ); |
||||||
|
Cmd_AddRestrictedCommand( "clearmasters", NET_ClearMasters_f, "clear masterserver list" ); |
||||||
|
Cmd_AddCommand( "listmasters", NET_ListMasters_f, "list masterservers" ); |
||||||
|
|
||||||
|
// keep main master always there
|
||||||
|
NET_AddMaster( MASTERSERVER_ADR, false ); |
||||||
|
NET_AddMaster( MASTERSERVER_ADR2, false ); |
||||||
|
NET_LoadMasters( ); |
||||||
|
} |
Loading…
Reference in new issue