Compare commits

...

4 Commits

Author SHA1 Message Date
ghost 76079d0fdb remove extra debug 2024-01-08 21:58:44 +02:00
ghost e9b16fd990 fix var name 2024-01-08 21:53:44 +02:00
ghost f50347eac6 improve debug 2024-01-08 21:10:13 +02:00
ghost 45f1378679 define gamedir, add key support 2024-01-08 21:09:01 +02:00
2 changed files with 28 additions and 18 deletions

View File

@ -100,16 +100,20 @@ class PyMaster:
for i in self.serverList: for i in self.serverList:
if time() > i.die: if time() > i.die:
logging.debug("Server removed by timeout")
self.serverList.remove(i) self.serverList.remove(i)
continue continue
if not i.check: if not i.check:
logging.debug("Invalid request")
continue continue
if nat != i.nat: if nat != i.nat:
logging.debug("NAT {0} mismatch {1}".format(i.nat, nat))
continue continue
if gamedir is not None and gamedir != i.gamedir: if gamedir is not None and gamedir != i.gamedir:
logging.debug("Game dir {0} mismatch node settings: {1}".format(i.gamedir, gamedir))
continue continue
if nat: if nat:
@ -172,11 +176,13 @@ class PyMaster:
else: else:
count += 1 count += 1
if count > MAX_SERVERS_FOR_IP: if count > MAX_SERVERS_FOR_IP:
logging.debug("Reached MAX_SERVERS_FOR_IP: {0}".format(MAX_SERVERS_FOR_IP))
return return
challenge = random.randint(0, 2**32 - 1) challenge = random.randint(0, 2**32 - 1)
# Add server to list # Add server to list
logging.debug("Added new server {0}:{1} with challenge {2}".format(addr[0], addr[1], challenge))
self.serverList.append(ServerEntry(addr, challenge)) self.serverList.append(ServerEntry(addr, challenge))
# And send him a challenge # And send him a challenge
@ -193,6 +199,7 @@ class PyMaster:
# Find a server with same address # Find a server with same address
for serverEntry in self.serverList: for serverEntry in self.serverList:
if serverEntry.addr == addr: if serverEntry.addr == addr:
logging.debug("Skipped same server address: {0}:{1}".format(addr[0], addr[1]))
break break
serverEntry.setInfoString(serverInfo) serverEntry.setInfoString(serverInfo)

View File

@ -5,7 +5,7 @@ import ipaddress
class ServerEntry: class ServerEntry:
challenge2 = 0 challenge2 = 0
gamedir = '' gamedir = 'valve'
protocol = 0 protocol = 0
players = 0 players = 0
maxplayers = 0 maxplayers = 0
@ -20,45 +20,48 @@ class ServerEntry:
region = 255 region = 255
product = '' product = ''
nat = 0 nat = 0
key = None
def setInfoString(self, data): def setInfoString(self, data):
infostring = data.replace('\n', '').replace('\r', '').replace('\0', '') infostring = data.replace('\n', '').replace('\r', '').replace('\0', '')
split = infostring.split('\\') split = infostring.split('\\')
for i in range(0, len(split), 2): for i in range(0, len(split), 2):
try: try:
key = split[i + 1] value = split[i + 1]
if( split[i] == 'challenge' ): if( split[i] == 'challenge' ):
self.challenge2 = int(key) self.challenge2 = int(value)
elif( split[i] == 'gamedir' ): elif( split[i] == 'gamedir' ):
self.gamedir = key.lower() # keep gamedir lowercase self.gamedir = value.lower() # keep gamedir lowercase
elif( split[i] == 'protocol' ): elif( split[i] == 'protocol' ):
self.protocol = int(key) self.protocol = int(value)
elif( split[i] == 'players' ): elif( split[i] == 'players' ):
self.players = int(key) self.players = int(value)
elif( split[i] == 'max' ): elif( split[i] == 'max' ):
self.maxplayers = int(key.split('.')[0]) self.maxplayers = int(value.split('.')[0])
elif( split[i] == 'bots' ): elif( split[i] == 'bots' ):
self.bots = int(key) self.bots = int(value)
elif( split[i] == 'map' ): elif( split[i] == 'map' ):
self.gamemap = key self.gamemap = value
elif( split[i] == 'version' ): elif( split[i] == 'version' ):
self.version = key self.version = value
elif( split[i] == 'type' ): elif( split[i] == 'type' ):
self.servtype = key self.servtype = value
elif( split[i] == 'password' ): elif( split[i] == 'password' ):
self.password = key self.password = value
elif( split[i] == 'os' ): elif( split[i] == 'os' ):
self.os = key self.os = value
elif( split[i] == 'secure' ): elif( split[i] == 'secure' ):
self.secure = key self.secure = value
elif( split[i] == 'lan' ): elif( split[i] == 'lan' ):
self.lan = key self.lan = value
elif( split[i] == 'region' ): elif( split[i] == 'region' ):
self.region = key self.region = value
elif( split[i] == 'product' ): elif( split[i] == 'product' ):
self.product = key self.product = value
elif( split[i] == 'nat' ): elif( split[i] == 'nat' ):
self.nat = int(key) self.nat = int(value)
elif split[i] == 'key':
self.key = int(value, 16)
except IndexError: except IndexError:
pass pass
self.check = self.challenge == self.challenge2 self.check = self.challenge == self.challenge2