mirror of
https://github.com/barronwaffles/dwc_network_server_emulator.git
synced 2026-03-21 17:34:21 -05:00
f9c710af99
Contains enough code to allow Tetris DS to get to the post-login wifi menu. This is very rough, basic code. I expect it to change quite a bit from its current format once I start adding more features. It's nowhere near ready for public use yet.
130 lines
4.4 KiB
Python
130 lines
4.4 KiB
Python
# Server emulator for gpcm.gs.nintendowifi.net
|
|
import socket
|
|
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
|
|
|
|
db = gs_database.GamespyDatabase()
|
|
|
|
address = ('0.0.0.0', 29900)
|
|
backlog = 10
|
|
size = 2048
|
|
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
s.bind(address)
|
|
s.listen(backlog)
|
|
|
|
utils.print_log("Server is now listening on listening on %s:%s..." % (address[0], address[1]))
|
|
|
|
while 1:
|
|
client, address = s.accept()
|
|
|
|
# TODO: Redo this part of the server so it'll handle multiple connections
|
|
utils.print_log("Received connection from %s:%s" % (address[0], address[1]))
|
|
|
|
# Send request for login information
|
|
challenge = utils.generate_random_str(8)
|
|
|
|
msg_d = []
|
|
msg_d.append(('__cmd__', "lc"))
|
|
msg_d.append(('__cmd_val__', "1"))
|
|
msg_d.append(('challenge', challenge))
|
|
msg_d.append(('id', "1"))
|
|
msg = gs_query.create_gamespy_message(msg_d)
|
|
|
|
utils.print_log("SENDING: '%s'..." % msg)
|
|
client.send(msg)
|
|
|
|
# Receive any command
|
|
accept_connection = True
|
|
while accept_connection:
|
|
data = client.recv(size).rstrip()
|
|
utils.print_log("RESPONSE: %s" % data)
|
|
|
|
commands = gs_query.parse_gamespy_message(data)
|
|
|
|
for data_parsed in commands:
|
|
msg_d = []
|
|
|
|
print data_parsed
|
|
|
|
if data_parsed['__cmd__'] == "login":
|
|
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"
|
|
nick = uniquenick
|
|
|
|
# Verify the client's response
|
|
valid_response = gs_utils.generate_response(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(challenge, authtoken_parsed['challenge'], data_parsed['challenge'], data_parsed['authtoken'])
|
|
|
|
valid_user = db.check_user_exists(userid)
|
|
profileid = None
|
|
if valid_user == False:
|
|
profileid = db.create_user(userid, password, email, uniquenick)
|
|
else:
|
|
profileid = db.perform_login(userid, password)
|
|
if profileid == None:
|
|
# Handle case where the user is invalid
|
|
print "Invalid password"
|
|
|
|
sesskey = db.create_session(profileid)
|
|
|
|
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', db.get_profileid_from_session_key(sesskey)))
|
|
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)
|
|
|
|
elif data_parsed['__cmd__'] == "getprofile":
|
|
msg_d = []
|
|
|
|
msg_d.append(('__cmd__', "pi"))
|
|
msg_d.append(('__cmd_val__', ""))
|
|
msg_d.append(('profileid', db.get_profileid_from_session_key(data_parsed.append(('sesskey')))))
|
|
msg_d.append(('nick', nick))
|
|
msg_d.append(('userid', userid))
|
|
msg_d.append(('email', email))
|
|
msg_d.append(('sig', sig))
|
|
msg_d.append(('uniquenick', uniquenick))
|
|
msg_d.append(('pid', pid))
|
|
msg_d.append(('lastname', lastname))
|
|
msg_d.append(('lon', lon))
|
|
msg_d.append(('lat', lat))
|
|
msg_d.append(('loc', loc))
|
|
msg_d.append(('id', data_parsed['id']))
|
|
msg = gs_query.create_gamespy_message(msg_d)
|
|
|
|
elif data_parsed['__cmd__'] == "updatepro":
|
|
# Handle properly later
|
|
# Assume that there will be other parameters besides lastname, so make it a loop or something along those lines later
|
|
db.update_profile(data_parsed['sesskey'], [("lastname", data_parsed['lastname'])])
|
|
|
|
elif data_parsed['__cmd__'] == "status":
|
|
# Handle status update
|
|
msg = ""
|
|
|
|
elif data_parsed['__cmd__'] == "logout":
|
|
print "Session %s has logged off" % (data_parsed['sesskey'])
|
|
db.delete_session(data_parsed['sesskey'])
|
|
accept_connection = False
|
|
|
|
utils.print_log("SENDING: %s" % msg)
|
|
client.send(msg)
|
|
|
|
client.close()
|