ddnet/scripts/tw_api.py

# coding: utf-8
from socket import *
import struct
import sys

def get_server_info(address, port):
	try:
		sock = socket(AF_INET, SOCK_DGRAM) 
		sock.settimeout(1.5); 
		sock.sendto("\xff\xff\xff\xff\xff\xff\xff\xff\xff\xffgief", (address, port)) 
		data, addr = sock.recvfrom(1024) 
		sock.close() 
	 
		data = data[14:] # skip header
	 	
		slots = data.split("\x00")
		server_info = {}
		server_info["version"] = slots[0]
		server_info["name"] = slots[1]
		server_info["map"] = slots[2]
		server_info["gametype_id"] = int(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+1]
			player["score"] = slots[8+i*2]
			server_info["players"] += [player]
			
		gametypes = ["dm", "tdm", "ctf"]
		try: server_info["gametype_name"] = gametypes[server_info["gametype_id"]]
		except: server_info["gametype_name"] = "unknown"
		
		return server_info
	except:
		return None
		
def get_server_count(address, port):
	try:
		sock = socket(AF_INET, SOCK_DGRAM) 
		sock.settimeout(1.5); 
		sock.sendto("\xff\xff\xff\xff\xff\xff\xff\xff\xff\xffcoun", (address, port)) 
		data, addr = sock.recvfrom(1024) 
		sock.close() 
	 
		data = data[14:] # skip header
		return struct.unpack(">H", data)[0]
	except:
		return -1

def get_servers(address):
	counter = 0
	master_port = 8300
	servers = []
 
	try:
		sock = socket(AF_INET, SOCK_DGRAM) 
		sock.settimeout(1.5) 
		sock.sendto("\x20\x00\x00\x00\x00\x00\xff\xff\xff\xffreqt", (address, master_port)) 
		
		while 1:
			data, addr = sock.recvfrom(1024)
			
			data = data[14:] 
			num_servers = len(data) / 6 

			for n in range(0, num_servers): 
				ip = ".".join(map(str, map(ord, data[n*6:n*6+4]))) 
				port = ord(data[n*6+5]) * 256 + ord(data[n*6+4]) 
				servers += [[ip, port]]
			
			# and we are done
			if num_servers < 128:
				break
			
		sock.close()

		return servers
	except:
		return None		


def get_all_servers():
	servers = []
	for i in range(1, 16):
		addr = "master%d.teeworlds.com"%i
		list = get_servers(addr)
		if list:
			print addr, "had", len(list), "servers"
			servers += list
	return servers

servers = get_all_servers()
total_players = 0
if 1:
	for server in servers:
		print "checking server", server[0], server[1]
		info = get_server_info(server[0], server[1])
		if info:
			total_players += len(info["players"])

print total_players, "on", len(servers), 'servers'