Leahnaya/dwc_network_server_emulator
1
0
Fork 0
mirror of https://github.com/barronwaffles/dwc_network_server_emulator.git synced 2026-07-12 22:21:55 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added natneg listing to backend server

Browse Source
This commit is contained in:
polaris 2014-05-20 11:17:52 -04:00
parent 3a4faf53d4
commit 25dd2a9d27
1 changed files with 34 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
gamespy_backend_server.py
View File

@ -46,6 +46,7 @@ class GameSpyServerDatabase(BaseManager):
class GameSpyBackendServer(object):
def __init__(self):
self.server_list = {}
self.natneg_list = {}
GameSpyServerDatabase.register("get_server_list", callable=lambda:self.server_list)
GameSpyServerDatabase.register("find_servers", callable=self.find_servers)
@ -53,6 +54,9 @@ class GameSpyBackendServer(object):
GameSpyServerDatabase.register("find_server_by_local_address", callable=self.find_server_by_local_address)
GameSpyServerDatabase.register("update_server_list", callable=self.update_server_list)
GameSpyServerDatabase.register("delete_server", callable=self.delete_server)
GameSpyServerDatabase.register("add_natneg_server", callable=self.add_natneg_server)
GameSpyServerDatabase.register("get_natneg_server", callable=self.get_natneg_server)
GameSpyServerDatabase.register("delete_natneg_server", callable=self.delete_natneg_server)
def start(self):
address = ("127.0.0.1", 27500)
@ -334,12 +338,12 @@ class GameSpyBackendServer(object):
requested = {}
for field in fields:
if not field in result:
if field in server:
requested[field] = server[field]
else:
# Return a dummy value. What's the normal behavior of the real server in this case?
requested[field] = ""
#if not field in result:
if field in server:
requested[field] = server[field]
else:
# Return a dummy value. What's the normal behavior of the real server in this case?
requested[field] = ""
result['requested'] = requested
@ -366,6 +370,8 @@ class GameSpyBackendServer(object):
self.server_list[gameid].append(value)
logger.log(logging.DEBUG, "%s servers: %d" % (gameid, len(self.server_list[gameid])))
return value
def delete_server(self, gameid, session):
if not gameid in self.server_list:
# Nothing to do if no servers for that game even exist.
@ -397,7 +403,7 @@ class GameSpyBackendServer(object):
for gameid in self.server_list:
for server in self.server_list[gameid]:
logger.log(logging.DEBUG, "publicip: %s == %s ? %d" % (server['publicip'], publicip, server['publicip'] == publicip))
if server['publicip'] == publicip and server['localport'] == str(localport):
if server['publicip'] == publicip and (server['localport'] == str(localport) or server['publicport'] == str(localport)):
for x in range(0, 10):
s = 'localip%d' % x
if s in server:
@ -406,7 +412,7 @@ class GameSpyBackendServer(object):
else:
for server in self.server_list[gameid]:
logger.log(logging.DEBUG, "publicip: %s == %s ? %d" % (server['publicip'], publicip, server['publicip'] == publicip))
if server['publicip'] == publicip and server['localport'] == str(localport):
if server['publicip'] == publicip and (server['localport'] == str(localport) or server['publicport'] == str(localport)):
for x in range(0, 10):
s = 'localip%d' % x
if s in server:
@ -414,6 +420,26 @@ class GameSpyBackendServer(object):
return server
return None
def add_natneg_server(self, cookie, server):
if cookie not in self.natneg_list:
self.natneg_list[cookie] = []
logger.log(logging.DEBUG, "Added natneg server %d" % (cookie))
self.natneg_list[cookie].append(server)
def get_natneg_server(self, cookie):
if cookie in self.natneg_list:
return self.natneg_list[cookie]
return None
def delete_natneg_server(self, cookie):
# TODO: Find a good time to prune the natneg server listing.
if cookie in self.natneg_list:
del self.natneg_list[cookie]
logger.log(logging.DEBUG, "Deleted natneg server %d" % (cookie))
if __name__ == '__main__':
freeze_support()