mirror of
https://github.com/ptitSeb/Serious-Engine
synced 2024-12-27 07:54:51 +01:00
108 lines
2.8 KiB
Python
108 lines
2.8 KiB
Python
|
#!/usr/bin/python -u
|
||
|
|
|
||
|
|
# GameAgent
|
||
|
|
import socket
|
||
|
|
import random
|
||
|
|
import struct
|
||
|
|
import time
|
||
|
|
import re
|
||
|
|
|
||
|
|
# global user/server arrays
|
||
|
|
servers = {}
|
||
|
|
players = {}
|
||
|
|
|
||
|
|
# load user thread
|
||
|
|
execfile("user.py")
|
||
|
|
execfile("server.py")
|
||
|
|
|
||
|
|
# listen on socket
|
||
|
|
sockListen = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||
|
|
sockListen.bind(("0.0.0.0", 9005))
|
||
|
|
print("Listening on UDP port 9005.")
|
||
|
|
|
||
|
|
def sendto(data, addr):
|
||
|
|
sockListen.sendto(data, addr)
|
||
|
|
|
||
|
|
# format address
|
||
|
|
def fmta(addr):
|
||
|
|
return addr[0] + ":" + str(addr[1])
|
||
|
|
|
||
|
|
def server_join(data, addr):
|
||
|
|
if(servers.has_key(fmta(addr))):
|
||
|
|
print("Join request from server already joined!")
|
||
|
|
print("Server join from " + fmta(addr))
|
||
|
|
s = Server()
|
||
|
|
s.addr = addr
|
||
|
|
s.join()
|
||
|
|
servers[fmta(addr)] = s
|
||
|
|
|
||
|
|
def server_challengeresponse(data, addr):
|
||
|
|
if(servers.has_key(fmta(addr))):
|
||
|
|
servers[fmta(addr)].challengeresponse(data)
|
||
|
|
else:
|
||
|
|
# server didn't join yet, make server join
|
||
|
|
server_join(data, addr)
|
||
|
|
|
||
|
|
def server_statechanged(data, addr):
|
||
|
|
if(servers.has_key(fmta(addr))):
|
||
|
|
if(servers[fmta(addr)].visible):
|
||
|
|
servers[fmta(addr)].statechanged(data)
|
||
|
|
else:
|
||
|
|
if(servers[fmta(addr)].prejoinactions > 5):
|
||
|
|
servers.pop(fmta(addr))
|
||
|
|
print("Server sent too many pre join actions, removing!")
|
||
|
|
else:
|
||
|
|
servers[fmta(addr)].prejoinactions += 1
|
||
|
|
else:
|
||
|
|
print("State change from server who didn't join yet!")
|
||
|
|
# server didn't join yet, make server join
|
||
|
|
server_join(data, addr)
|
||
|
|
|
||
|
|
def enum_trigger(data, addr):
|
||
|
|
p = Player()
|
||
|
|
p.addr = addr
|
||
|
|
|
||
|
|
buffer = "s"
|
||
|
|
topop = ""
|
||
|
|
for key in servers:
|
||
|
|
server = servers[key]
|
||
|
|
if(time.time() - server.lastheartbeat > 300):
|
||
|
|
# 5m have passed since last heartbeat
|
||
|
|
# server sends a heartbeat every 2.5 minutes
|
||
|
|
# we'll kick the server out because he's not doing this
|
||
|
|
print("Server " + fmta(server.addr) + " hasn't given a heartbeat since " + time.strftime("%c", time.gmtime(server.lastheartbeat)) + ", kicking!")
|
||
|
|
topop += key + ","
|
||
|
|
continue
|
||
|
|
if(not server.visible):
|
||
|
|
continue
|
||
|
|
if(len(buffer) + 6 > 1024):
|
||
|
|
sendto(buffer, addr)
|
||
|
|
buffer = "s"
|
||
|
|
parse = server.addr[0].split(".")
|
||
|
|
buffer += struct.pack("BBBBH", int(parse[0]), int(parse[1]), int(parse[2]), int(parse[3]), server.port)
|
||
|
|
sendto(buffer, addr)
|
||
|
|
for key in topop.split(","):
|
||
|
|
if(servers.has_key(key)):
|
||
|
|
servers.pop(key)
|
||
|
|
|
||
|
|
# masterserver main loop
|
||
|
|
while True:
|
||
|
|
data, addr = sockListen.recvfrom(1024)
|
||
|
|
|
||
|
|
packets = {
|
||
|
|
'q': server_join,
|
||
|
|
'0': server_challengeresponse,
|
||
|
|
'u': server_statechanged,
|
||
|
|
|
||
|
|
'e': enum_trigger,
|
||
|
|
}
|
||
|
|
|
||
|
|
packetID = data[0]
|
||
|
|
if(packets.has_key(packetID)):
|
||
|
|
packets[packetID](data[1:], addr)
|
||
|
|
else:
|
||
|
|
print("Packet from " + addr[0] + " denied because of unknown packet " + str(ord(packetID)))
|
||
|
|
|
||
|
|
# server died
|
||
|
|
print("Server shutting down.")
|