mirror of
https://github.com/barronwaffles/dwc_network_server_emulator.git
synced 2026-07-13 06:31:07 -05:00
c59645a6aa
- Rewrote the PlayerSpy server to easily handle more than 1 connection. It now requires Twisted to run. - The basics for the buddy list have been added. Friends can be added and authorized, as well as sending status messages to buddies. The feature is still incomplete, however. It's enough to show a friends only game in Tetris DS, though.
247 lines
9.3 KiB
Python
247 lines
9.3 KiB
Python
from twisted.internet.protocol import Factory
|
|
from twisted.internet.endpoints import serverFromString
|
|
from twisted.protocols.basic import LineReceiver
|
|
from twisted.internet import reactor
|
|
|
|
import gamespy.gs_database as gs_database
|
|
import gamespy.gs_query as gs_query
|
|
import gamespy.gs_utility as gs_utils
|
|
import other.utils as utils
|
|
|
|
class PlayerSession(LineReceiver):
|
|
def __init__(self, sessions, addr):
|
|
self.sessions = sessions
|
|
self.setRawMode() # We're dealing with binary data so set to raw mode
|
|
self.db = gs_database.GamespyDatabase()
|
|
self.profileId = 0
|
|
self.address = addr
|
|
|
|
def get_ip_as_int(self, address):
|
|
ipaddress = 0
|
|
|
|
if address != None:
|
|
for n in address.split('.'):
|
|
ipaddress = (ipaddress << 8) | int(n)
|
|
|
|
return ipaddress
|
|
|
|
def connectionMade(self):
|
|
# Create new session id
|
|
self.session = ""
|
|
self.challenge = utils.generate_random_str(8)
|
|
|
|
msg_d = []
|
|
msg_d.append(('__cmd__', "lc"))
|
|
msg_d.append(('__cmd_val__', "1"))
|
|
msg_d.append(('challenge', self.challenge))
|
|
msg_d.append(('id', "1"))
|
|
msg = gs_query.create_gamespy_message(msg_d)
|
|
|
|
utils.print_log("SENDING: '%s'..." % msg)
|
|
self.transport.write(bytes(msg))
|
|
|
|
def connectionLost(self, reason):
|
|
if self.session in self.sessions:
|
|
del self.sessions[self.session]
|
|
|
|
def rawDataReceived(self, data):
|
|
utils.print_log("RESPONSE: %s" % data)
|
|
|
|
commands = gs_query.parse_gamespy_message(data)
|
|
|
|
for data_parsed in commands:
|
|
print data_parsed
|
|
|
|
if data_parsed['__cmd__'] == "login":
|
|
self.perform_login(data_parsed)
|
|
elif data_parsed['__cmd__'] == "logout":
|
|
self.perform_logout(data_parsed)
|
|
elif data_parsed['__cmd__'] == "getprofile":
|
|
self.perform_getprofile(data_parsed)
|
|
elif data_parsed['__cmd__'] == "updatepro":
|
|
self.perform_updatepro(data_parsed)
|
|
elif data_parsed['__cmd__'] == "ka":
|
|
self.perform_ka(data_parsed)
|
|
elif data_parsed['__cmd__'] == "status":
|
|
self.perform_status(data_parsed)
|
|
elif data_parsed['__cmd__'] == "bm":
|
|
self.perform_bm(data_parsed)
|
|
elif data_parsed['__cmd__'] == "addbuddy":
|
|
self.perform_addbuddy(data_parsed)
|
|
elif data_parsed['__cmd__'] == "authadd":
|
|
self.perform_authadd(data_parsed)
|
|
|
|
def perform_login(self, data_parsed):
|
|
authtoken_parsed = gs_utils.parse_authtoken(data_parsed['authtoken'])
|
|
print authtoken_parsed
|
|
|
|
# get correct information
|
|
userid = authtoken_parsed['userid']
|
|
password = authtoken_parsed['passwd']
|
|
uniquenick = utils.base32_encode(int(userid)) + authtoken_parsed['gsbrcd']
|
|
email = uniquenick + "@nds"
|
|
|
|
# Verify the client's response
|
|
valid_response = gs_utils.generate_response(self.challenge, authtoken_parsed['challenge'], data_parsed['challenge'], data_parsed['authtoken'])
|
|
if data_parsed['response'] != valid_response:
|
|
utils.print_log("ERROR: Got invalid response. Got %s, expected %s" % (data_parsed['response'], valid_response))
|
|
|
|
proof = gs_utils.generate_proof(self.challenge, authtoken_parsed['challenge'], data_parsed['challenge'], data_parsed['authtoken'])
|
|
|
|
valid_user = self.db.check_user_exists(userid)
|
|
if valid_user == False:
|
|
profileid = self.db.create_user(userid, password, email, uniquenick)
|
|
else:
|
|
profileid = self.db.perform_login(userid, password)
|
|
|
|
if profileid == None:
|
|
# Handle case where the user is invalid
|
|
print "Invalid password"
|
|
|
|
sesskey = self.db.create_session(profileid)
|
|
|
|
self.sessions[profileid] = self
|
|
|
|
msg_d = []
|
|
msg_d.append(('__cmd__', "lc"))
|
|
msg_d.append(('__cmd_val__', "2"))
|
|
msg_d.append(('sesskey', sesskey))
|
|
msg_d.append(('proof', proof))
|
|
msg_d.append(('userid', userid))
|
|
msg_d.append(('profileid', profileid))
|
|
msg_d.append(('uniquenick', uniquenick))
|
|
msg_d.append(('lt', gs_utils.base64_encode(utils.generate_random_str(16)))) # Some kind of token... don't know it gets used or generated, but it doesn't seem to have any negative effects if it's not properly generated.
|
|
msg_d.append(('id', data_parsed['id']))
|
|
msg = gs_query.create_gamespy_message(msg_d)
|
|
|
|
self.profileid = profileid
|
|
|
|
utils.print_log("SENDING: %s" % msg)
|
|
self.transport.write(bytes(msg))
|
|
|
|
def perform_logout(self, data_parsed):
|
|
print "Session %s has logged off" % (data_parsed['sesskey'])
|
|
self.db.delete_session(data_parsed['sesskey'])
|
|
|
|
def perform_getprofile(self, data_parsed):
|
|
#profile = self.db.get_profile_from_session_key(data_parsed['sesskey'])
|
|
profile = self.db.get_profile_from_profileid(data_parsed['profileid'])
|
|
|
|
msg_d = []
|
|
msg_d.append(('__cmd__', "pi"))
|
|
msg_d.append(('__cmd_val__', ""))
|
|
msg_d.append(('profileid', profile['profileid']))
|
|
msg_d.append(('nick', profile['uniquenick']))
|
|
msg_d.append(('userid', profile['userid']))
|
|
msg_d.append(('email', profile['email']))
|
|
msg_d.append(('sig', utils.generate_random_hex_str(32)))
|
|
msg_d.append(('uniquenick', profile['uniquenick']))
|
|
msg_d.append(('pid', profile['pid']))
|
|
msg_d.append(('lastname', profile['lastname']))
|
|
msg_d.append(('lon', profile['lon']))
|
|
msg_d.append(('lat', profile['lat']))
|
|
msg_d.append(('loc', profile['loc']))
|
|
msg_d.append(('id', data_parsed['id']))
|
|
msg = gs_query.create_gamespy_message(msg_d)
|
|
|
|
utils.print_log("SENDING: %s" % msg)
|
|
self.transport.write(bytes(msg))
|
|
|
|
|
|
def perform_updatepro(self, data_parsed):
|
|
sesskey = data_parsed['sesskey']
|
|
|
|
# Remove any fields not related to what we should be updating.
|
|
data_parsed.pop('__cmd__')
|
|
data_parsed.pop('__cmd_val__')
|
|
data_parsed.pop('updatepro')
|
|
data_parsed.pop('sesskey')
|
|
|
|
# Create a list of fields to be updated.
|
|
fields = []
|
|
for f in data_parsed:
|
|
fields.append((f, data_parsed[f]))
|
|
|
|
self.db.update_profile(sesskey, fields)
|
|
|
|
def perform_ka(self, data_parsed):
|
|
# No op
|
|
return
|
|
|
|
def perform_status(self, data_parsed):
|
|
sesskey = data_parsed['sesskey']
|
|
|
|
#fields = []
|
|
#fields.append(("stat", data_parsed['statstring']))
|
|
#fields.append(("loc", data_parsed['locstring']))
|
|
|
|
#self.db.update_profile(sesskey, fields)
|
|
|
|
self.status = data_parsed['__cmd_val__']
|
|
self.statstring = data_parsed['statstring']
|
|
self.locstring = data_parsed['locstring']
|
|
|
|
self.send_status_to_friends()
|
|
|
|
def perform_bm(self, data_parsed):
|
|
dest_profileid = data_parsed['t']
|
|
dest_msg = data_parsed['msg']
|
|
|
|
def perform_addbuddy(self, data_parsed):
|
|
# Sample: \addbuddy\\sesskey\231601763\newprofileid\476756820\reason\\final\
|
|
self.db.add_buddy(self.profileid, data_parsed['newprofileid'])
|
|
|
|
# In the case that the user is already a buddy:
|
|
# \bm\2\f\217936895\msg\|signed|f259f26d3273f8bda23c7c5e4bd8c5aa\final\
|
|
# \error\\err\1539\errmsg\The profile requested is already a buddy.\final\
|
|
# Handle later?
|
|
|
|
def perform_authadd(self, data_parsed):
|
|
# Sample: \authadd\\sesskey\231587549\fromprofileid\217936895\sig\f259f26d3273f8bda23c7c5e4bd8c5aa\final\
|
|
self.db.auth_buddy(self.profileid, data_parsed['newprofileid'])
|
|
|
|
# After authorization, send the person who was authorized a message like so:
|
|
# \bm\1\f\476756820\msg\I have authorized your request to add me to your list\final\
|
|
|
|
def send_status_to_friends(self):
|
|
# TODO: Cache buddy list so we don't have to query the database every time
|
|
buddies = self.db.get_buddy_list(self.profileid)
|
|
|
|
status_msg = "|s|%s|ss|%s|ls|%s|ip|%d|p|0|qm|0" % (self.status, self.statstring, self.locstring, self.get_ip_as_int(self.address.host))
|
|
|
|
msg_d = []
|
|
msg_d.append(('__cmd__', "bm"))
|
|
msg_d.append(('__cmd_val__', "100"))
|
|
msg_d.append(('f', self.profileid))
|
|
msg_d.append(('msg', status_msg))
|
|
msg = gs_query.create_gamespy_message(msg_d)
|
|
|
|
for buddy in buddies:
|
|
if buddy['buddyProfileId'] in self.sessions:
|
|
self.sessions[buddy['buddyProfileId']].transport.write(bytes(msg))
|
|
|
|
def get_status_from_friends(self):
|
|
# This will be called when the player logs in. Grab the player's buddy list and check the current sessions to
|
|
# see if anyone is online. If they are online, make them send an update to the calling client.
|
|
return
|
|
|
|
|
|
|
|
|
|
class PlayerFactory(Factory):
|
|
def __init__(self):
|
|
# Instead of storing the sessions in the database, it might make more sense to store them in the PlayerFactory.
|
|
self.sessions = {}
|
|
print "Now listening for connections..."
|
|
|
|
def buildProtocol(self, addr):
|
|
return PlayerSession(self.sessions, addr)
|
|
|
|
|
|
#reactor.listenTCP(8007, ChatFactory())
|
|
#reactor.run()
|
|
endpoint = serverFromString(reactor, "tcp:29900")
|
|
conn = endpoint.listen(PlayerFactory())
|
|
reactor.run()
|
|
|