edg-l/ddnet
1
0
Fork 0
mirror of https://github.com/ddnet/ddnet.git synced 2024-11-19 22:48:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use tokens in tw_api.py

Browse source
This commit is contained in:
nheir 2018-11-10 00:31:59 +01:00
parent 8e912f932e
commit 234121a415
1 changed files with 119 additions and 149 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

268
scripts/tw_api.py
View file

@ -1,149 +1,135 @@
#!/bin/env python3
# coding: utf-8 # coding: utf-8
from socket import * from socket import *
import struct
import sys import sys
import threading import threading
import time import time
import random
NUM_MASTERSERVERS = 4 NUM_MASTERSERVERS = 4
MASTERSERVER_PORT = 8300 MASTERSERVER_PORT = 8283
TIMEOUT = 2 TIMEOUT = 2
SERVERTYPE_NORMAL = 0 SERVERTYPE_NORMAL = 0
SERVERTYPE_LEGACY = 1
PACKET_GETLIST = "\x20\x00\x00\x00\x00\x00\xff\xff\xff\xffreqt" PACKET_GETLIST = b"\xff\xff\xff\xffreq2"
PACKET_GETLIST2 = "\x20\x00\x00\x00\x00\x00\xff\xff\xff\xffreq2" PACKET_LIST = b"\xff\xff\xff\xfflis2"
PACKET_GETINFO = "\xff\xff\xff\xff\xff\xff\xff\xff\xff\xffgief"
PACKET_GETINFO2 = "\xff\xff\xff\xff\xff\xff\xff\xff\xff\xffgie2" + "\x00"
PACKET_GETINFO3 = "\xff\xff\xff\xff\xff\xff\xff\xff\xff\xffgie3" + "\x00"
PACKET_GETINFO = b"\xff\xff\xff\xffgie3"
PACKET_INFO = b"\xff\xff\xff\xffinf3"
def pack_control_msg_with_token(token_srv,token_cl):
b = [0,0,1,0,0,5,0,0,0]
b[0] = (token_srv >> 12) & 0xff
b[1] = (token_srv >> 4) & 0xff
b[2] |= (token_srv << 4) & 0xff
b[6] |= (token_cl >> 16) & 0x0f
b[7] |= (token_cl >> 8) & 0xff
b[8] |= token_cl & 0xff
return bytes(b)
def unpack_control_msg_with_token(msg):
b = list(msg)
token_cl = (b[0] << 12) + (b[1] << 4) + (b[2] >> 4)
token_srv = (b[6] << 16) + (b[7] << 8) + b[8]
return token_cl,token_srv
def header_connless(token_srv, token_cl):
b = [0,0,8,16,0,0]
b[0] = (token_srv >> 12) & 0xff
b[1] = (token_srv >> 4) & 0xff
b[2] |= (token_srv << 4) & 0xff
b[3] |= (token_cl >> 16) & 0x0f
b[4] |= (token_cl >> 8) & 0xff
b[5] |= token_cl & 0xff
return bytes(b)
class Server_Info(threading.Thread): class Server_Info(threading.Thread):
def __init__(self, address, type): def __init__(self, address):
self.address = address self.address = address
self.type = type
self.finished = False self.finished = False
threading.Thread.__init__(self, target = self.run) threading.Thread.__init__(self, target = self.run)
def run(self): def run(self):
self.info = None self.info = None
if self.type == SERVERTYPE_NORMAL: self.info = get_server_info(self.address)
self.info = get_server_info3(self.address)
elif self.type == SERVERTYPE_LEGACY:
self.info = get_server_info(self.address)
if self.info:
self.info = get_server_info2(self.address)
self.finished = True self.finished = True
def unpack_int(b):
l = list(b[:5])
i = 0
Sign = (l[i]>>6)&1;
res = l[i] & 0x3F;
for _ in (0,):
if not (l[i]&0x80):
break
i+=1
res |= (l[i]&(0x7F))<<(6);
if not (l[i]&0x80):
break
i+=1
res |= (l[i]&(0x7F))<<(6+7);
if not (l[i]&0x80):
break
i+=1
res |= (l[i]&(0x7F))<<(6+7+7);
if not (l[i]&0x80):
break
i+=1
res |= (l[i]&(0x7F))<<(6+7+7+7);
i += 1;
res ^= -Sign
return res, b[i:]
def get_server_info(address): def get_server_info(address):
try: try:
sock = socket(AF_INET, SOCK_DGRAM) sock = socket(AF_INET, SOCK_DGRAM)
sock.settimeout(TIMEOUT); sock.settimeout(TIMEOUT)
sock.sendto(PACKET_GETINFO, address) token = random.randrange(0x100000)
sock.sendto(pack_control_msg_with_token(-1,token),address)
data, addr = sock.recvfrom(1024) data, addr = sock.recvfrom(1024)
sock.close() token_cl, token_srv = unpack_control_msg_with_token(data)
assert(token_cl == token)
data = data[14:] # skip header sock.sendto(header_connless(token_srv, token_cl) + PACKET_GETINFO + b'\x00', address)
slots = data.split("\x00")
server_info = {}
server_info["version"] = slots[0]
server_info["name"] = slots[1]
server_info["map"] = slots[2]
server_info["gametype"] = slots[3]
server_info["flags"] = int(slots[4])
server_info["progression"] = int(slots[5])
server_info["num_players"] = int(slots[6])
server_info["max_players"] = int(slots[7])
server_info["players"] = []
for i in xrange(0, server_info["num_players"]):
player = {}
player["name"] = slots[8+i*2]
player["score"] = int(slots[8+i*2+1])
server_info["players"].append(player)
return server_info
except:
sock.close()
return None
def get_server_info2(address):
try:
sock = socket(AF_INET, SOCK_DGRAM)
sock.settimeout(TIMEOUT);
sock.sendto(PACKET_GETINFO2, address)
data, addr = sock.recvfrom(1024) data, addr = sock.recvfrom(1024)
head = header_connless(token_cl, token_srv) + PACKET_INFO + b'\x00'
assert(data[:len(head)] == head)
sock.close() sock.close()
data = data[14:] # skip header data = data[len(head):] # skip header
slots = data.split("\x00")
slots = data.split(b"\x00", maxsplit=5)
server_info = {} server_info = {}
server_info["token"] = slots[0] server_info["version"] = slots[0].decode()
server_info["version"] = slots[1] server_info["name"] = slots[1].decode()
server_info["name"] = slots[2] server_info["hostname"] = slots[2].decode()
server_info["map"] = slots[3] server_info["map"] = slots[3].decode()
server_info["gametype"] = slots[4] server_info["gametype"] = slots[4].decode()
server_info["flags"] = int(slots[5]) data = slots[5]
server_info["progression"] = int(slots[6]) # these integers should fit in one byte each
server_info["num_players"] = int(slots[7]) server_info["flags"], server_info["skill"], server_info["num_players"], server_info["max_players"], server_info["num_clients"], server_info["max_clients"] = tuple(data[:6])
server_info["max_players"] = int(slots[8]) data = data[6:]
server_info["players"] = [] server_info["players"] = []
for i in xrange(0, server_info["num_players"]): for i in range(server_info["num_clients"]):
player = {} player = {}
player["name"] = slots[9+i*2] slots = data.split(b"\x00", maxsplit=2)
player["score"] = int(slots[9+i*2+1]) player["name"] = slots[0].decode()
server_info["players"].append(player) player["clan"] = slots[1].decode()
data = slots[2]
return server_info player["country"], data = unpack_int(data)
player["score"], data = unpack_int(data)
except: is_player, data = unpack_int(data)
sock.close() if is_player:
return None
def get_server_info3(address):
try:
sock = socket(AF_INET, SOCK_DGRAM)
sock.settimeout(TIMEOUT);
sock.sendto(PACKET_GETINFO3, address)
data, addr = sock.recvfrom(1400)
sock.close()
data = data[14:] # skip header
slots = data.split("\x00")
server_info = {}
server_info["token"] = slots[0]
server_info["version"] = slots[1]
server_info["name"] = slots[2]
server_info["map"] = slots[3]
server_info["gametype"] = slots[4]
server_info["flags"] = int(slots[5])
server_info["num_players"] = int(slots[6])
server_info["max_players"] = int(slots[7])
server_info["num_clients"] = int(slots[8])
server_info["max_clients"] = int(slots[9])
server_info["players"] = []
for i in xrange(0, server_info["num_clients"]):
player = {}
player["name"] = slots[10+i*5]
player["clan"] = slots[10+i*5+1]
player["country"] = int(slots[10+i*5+2])
player["score"] = int(slots[10+i*5+3])
if int(slots[10+i*5+4]):
player["player"] = True player["player"] = True
else: else:
player["player"] = False player["player"] = False
@ -152,11 +138,12 @@ def get_server_info3(address):
return server_info return server_info
except: except:
# import traceback
# traceback.print_exc()
sock.close() sock.close()
return None return None
class Master_Server_Info(threading.Thread): class Master_Server_Info(threading.Thread):
def __init__(self, address): def __init__(self, address):
@ -165,7 +152,7 @@ class Master_Server_Info(threading.Thread):
threading.Thread.__init__(self, target = self.run) threading.Thread.__init__(self, target = self.run)
def run(self): def run(self):
self.servers = get_list(self.address) + get_list2(self.address) self.servers = get_list(self.address)
self.finished = True self.finished = True
@ -175,48 +162,33 @@ def get_list(address):
try: try:
sock = socket(AF_INET, SOCK_DGRAM) sock = socket(AF_INET, SOCK_DGRAM)
sock.settimeout(TIMEOUT) sock.settimeout(TIMEOUT)
sock.sendto(PACKET_GETLIST, address)
token = random.randrange(0x100000)
sock.sendto(pack_control_msg_with_token(-1,token),address)
data, addr = sock.recvfrom(1024)
token_cl, token_srv = unpack_control_msg_with_token(data)
assert(token_cl == token)
sock.sendto(header_connless(token_srv, token_cl) + PACKET_GETLIST, addr)
head = header_connless(token_cl, token_srv) + PACKET_LIST
while 1: while 1:
data, addr = sock.recvfrom(1024) data, addr = sock.recvfrom(1024)
assert(data[:len(head)] == head)
data = data[14:] data = data[len(head):]
num_servers = len(data) / 6 num_servers = len(data) // 18
for n in range(0, num_servers): for n in range(0, num_servers):
ip = ".".join(map(str, map(ord, data[n*6:n*6+4]))) if data[n*18:n*18+12] == b"\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff":
port = ord(data[n*6+5]) * 256 + ord(data[n*6+4]) ip = ".".join(map(str, data[n*18+12:n*18+16]))
servers += [[(ip, port), SERVERTYPE_LEGACY]]
except:
sock.close()
return servers
def get_list2(address):
servers = []
try:
sock = socket(AF_INET, SOCK_DGRAM)
sock.settimeout(TIMEOUT)
sock.sendto(PACKET_GETLIST2, address)
while 1:
data, addr = sock.recvfrom(1400)
data = data[14:]
num_servers = len(data) / 18
for n in range(0, num_servers):
if data[n*18:n*18+12] == "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff":
ip = ".".join(map(str, map(ord, data[n*18+12:n*18+16])))
else: else:
ip = ":".join(map(str, map(ord, data[n*18:n*18+16]))) ip = ":".join(map(str, data[n*18:n*18+16]))
port = (ord(data[n*18+16])<<8) + ord(data[n*18+17]) port = ((data[n*18+16])<<8) + data[n*18+17]
servers += [[(ip, port), SERVERTYPE_NORMAL]] servers += [(ip, port)]
except: except:
# import traceback
# traceback.print_exc()
sock.close() sock.close()
return servers return servers
@ -242,10 +214,10 @@ while len(master_servers) != 0:
servers_info = [] servers_info = []
print str(len(servers)) + " servers" print(str(len(servers)) + " servers")
for server in servers: for server in servers:
s = Server_Info(server[0], server[1]) s = Server_Info(server)
servers_info.append(s) servers_info.append(s)
s.start() s.start()
time.sleep(0.001) # avoid issues time.sleep(0.001) # avoid issues
@ -253,18 +225,16 @@ for server in servers:
num_players = 0 num_players = 0
num_clients = 0 num_clients = 0
player_names = []
while len(servers_info) != 0: while len(servers_info) != 0:
if servers_info[0].finished == True: if servers_info[0].finished == True:
if servers_info[0].info: if servers_info[0].info:
num_players += servers_info[0].info["num_players"] num_players += servers_info[0].info["num_players"]
if servers_info[0].type == SERVERTYPE_NORMAL: num_clients += servers_info[0].info["num_clients"]
num_clients += servers_info[0].info["num_clients"]
else:
num_clients += servers_info[0].info["num_players"]
del servers_info[0] del servers_info[0]
time.sleep(0.001) # be nice time.sleep(0.001) # be nice
print str(num_players) + " players and " + str(num_clients-num_players) + " spectators" print(str(num_players) + " players and " + str(num_clients-num_players) + " spectators")