scan.py

#!/usr/bin/env python3
import socket
import random
import sys
import traceback
import logging
import os
import datetime
import requests
import hashlib
import re
from optparse import OptionParser
from struct import pack
from time import time, strftime, localtime
from utils import tobuf, toutf, tostr, dict_to_info, info_to_dict
from alarm import send_alarm
import config

MASTERSERVER_ADR = {
#	'ms.xash.su:27010': None,
	'mentality.rip:27010': None,
#	'ms.tyabus.co.uk:27010': None,
}
MS_SCAN_REQUEST = '1\xFF0.0.0.0:0\0'
COLORSTRING_RE = re.compile(r'(\^[0-9]|[|])')
PACKET_HEADER = b'\xff\xff\xff\xff'

def get_socket(options, timeout=1):
	sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM | socket.SOCK_NONBLOCK)
	sock.settimeout(timeout)
	if options.netdev:
		sock.setsockopt(socket.SOL_SOCKET, socket.SO_BINDTODEVICE, tobuf(options.netdev))
	sock.bind(('0.0.0.0', 0))
	return sock

def outofbound_print(sock, addr, message):
	p = tobuf('\xff\xff\xff\xff' + message)
	sock.sendto(p, addr)

def get_all_masters(family = socket.AF_INET):
	for hostport in MASTERSERVER_ADR:
		(host, port) = hostport.split(':')
		res = socket.getaddrinfo(host, port, family = family, proto = socket.IPPROTO_UDP)
		MASTERSERVER_ADR[hostport] = res[0][-1]

def master_query(sock, hostport, family, gamedir, protocol):
	key = hex(random.randint(0, 2**32))[2:]
	query = MS_SCAN_REQUEST + dict_to_info({
		'clver': '0.20',
		'nat': '0',
		'protocol': protocol,
		'gamedir': gamedir,
		'key': key,
	})
	print(tobuf(query))
	sock.sendto(tobuf(query) + b'\x00', MASTERSERVER_ADR[hostport])
	return key

def parse_ip_list(data):
	servers = []
	i = 0
	while data[i:i+6] != b'\0\0\0\0\0\0':
		ip = '%d.%d.%d.%d' % (data[i], data[i+1], data[i+2], data[i+3])
		port = data[i+4] * 256 + data[i+5]
		server = (ip, port)
		servers += [server]
		i += 6
	return servers

def is_master(addr):
	for i in MASTERSERVER_ADR:
		if MASTERSERVER_ADR[i] == addr:
			return True
	return False

def collect_servers(sock):
	while True:
		try:
			(data, addr) = sock.recvfrom(2048)
		except BlockingIOError:
			continue
		except (socket.timeout, TimeoutError):
			return list()

		if not is_master(addr):
			print('Expected response from master but got from %s:%i' % (addr[0], addr[1]))
			continue

		if data[0:4] != PACKET_HEADER:
			print('Unhandled packet %s from %s:%i' % (data, addr[0], addr[1]))
			continue

		data = data.strip(PACKET_HEADER)
		if data.startswith(b'f\n'):
			data = data.strip(b'f\n')
			return parse_ip_list(data)
	return list()

def validate_server_info(sock):
	while True:
		try:
			(data, addr) = sock.recvfrom(2048)
		except BlockingIOError:
			continue
		except (socket.timeout, TimeoutError):
			continue

		if data[0:4] != PACKET_HEADER:
			print('Unhandled packet %s from %s:%i' % (data, addr[0], addr[1]))
			continue
		data = data.strip(PACKET_HEADER)

		if data.startswith(b'info\n\\'): # infostring begin
			data = data.strip(b'info\n\\')
			info = info_to_dict(tostr(data))

			print(addr, ' ', data)
		else:
			print('broken ', addr, ' ', data)

def probe_servers_from_multiple_sockets(options, server_list):
	bad_servers = []
	ok_servers = []
	sock = get_socket(options)

#	random.shuffle(server_list)

	for server in server_list:
		outofbound_print(sock, server, 'info %d' % options.protocol)

#	random.shuffle(server_list)

	validate_server_info(sock)

def main(options):
	sock = get_socket(options, 5)

	if options.unban and not options.probe:
		print('--unban must be used with --probe-server')
		return

	get_all_masters()

	server_list = list()

	# step 1: get all servers
	for addr in MASTERSERVER_ADR:
		master_query(sock, addr, socket.AF_INET, options.gamedir, options.protocol)
		server_list += collect_servers(sock)

	if options.list:
		random.shuffle(server_list)
		for i in server_list:
			print("%s:%d" % (i[0], i[1]))
		return

	print('Found servers: %d' % (len(server_list)))

	# step 2: send to each server from new socket info request and validate response
	bad_servers = probe_servers_from_multiple_sockets(options, server_list)

if __name__ == "__main__":
	parser = OptionParser()
	parser.add_option('--probe-server', action='store', dest='probe', default=None)
	parser.add_option('-g', '--gamedir', action='store', dest='gamedir', default='valve')
	parser.add_option('-p', '--protocol', action='store', dest='protocol', default=48, type='int')
	parser.add_option('-d', '--netdev', action='store', dest='netdev', default=None)
	parser.add_option('-l', '--list-ips', action='store_true', dest='list', default=False)
	parser.add_option('--dry-run', action='store_true', dest='dry', default=False)
	parser.add_option('--unban', action='store_true', dest='unban', default=False)

	(options, args) = parser.parse_args()
	if not options.list:
		print('=============================================')
	main(options)
	if not options.list:
		print('ENDENDENDENDENDENDENDENDENDENDENDENDENDENDEND')