mirror of
https://github.com/de2tla2f/pymaster.git
synced 2024-11-27 05:11:29 +00:00
Compare commits
4 commits
4ad8a8a59c
...
76079d0fdb
Author | SHA1 | Date | |
---|---|---|---|
76079d0fdb | |||
e9b16fd990 | |||
f50347eac6 | |||
45f1378679 |
|
@ -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)
|
||||||
|
|
|
@ -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
|
||||||
|
|
Loading…
Reference in a new issue