Mirrored_Repos/houdini
1
0
Fork 0
mirror of https://github.com/solero/houdini.git synced 2025-04-26 16:46:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove duplicate code in world auth handlers

Browse Source
This commit is contained in:
Ben 2019-10-12 00:14:09 +01:00
parent 5bf149236a
commit 33fe814273
1 changed files with 34 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

84
houdini/handlers/login/world.py
View File

@ -14,17 +14,40 @@ handle_version_check = login.handle_version_check
handle_random_key = login.handle_random_key handle_random_key = login.handle_random_key
async def world_login(p, data):
if len(p.server.penguins_by_id) >= p.server.server_config['Capacity']:
return await p.send_error_and_disconnect(103)
if data is None:
return await p.send_error_and_disconnect(100)
if not data.active:
return await p.close()
if data.permaban:
return await p.close()
active_ban = await Ban.query.where((Ban.penguin_id == data.id) & (Ban.expires >= datetime.now())).gino.scalar()
if active_ban is not None:
return await p.close()
if data.id in p.server.penguins_by_id:
await p.server.penguins_by_id[data.id].close()
p.logger.info(f'{data.username} logged in successfully')
p.data = data
await p.send_xt('l')
@handlers.handler(XMLPacket('login'), client=ClientType.Vanilla) @handlers.handler(XMLPacket('login'), client=ClientType.Vanilla)
@handlers.allow_once @handlers.allow_once
@handlers.depends_on_packet(XMLPacket('verChk'), XMLPacket('rndK')) @handlers.depends_on_packet(XMLPacket('verChk'), XMLPacket('rndK'))
async def handle_login(p, credentials: WorldCredentials): async def handle_login(p, credentials: WorldCredentials):
if len(p.server.penguins_by_id) >= p.server.server_config['Capacity']:
return await p.send_error_and_disconnect(103)
tr = p.server.redis.multi_exec() tr = p.server.redis.multi_exec()
tr.get('{}.lkey'.format(credentials.username)) tr.get(f'{credentials.username}.lkey')
tr.get('{}.ckey'.format(credentials.username)) tr.get(f'{credentials.username}.ckey')
tr.delete('{}.lkey'.format(credentials.username), '{}.ckey'.format(credentials.username)) tr.delete(f'{credentials.username}.lkey', f'{credentials.username}.ckey')
login_key, confirmation_hash, _ = await tr.execute() login_key, confirmation_hash, _ = await tr.execute()
if login_key is None or confirmation_hash is None: if login_key is None or confirmation_hash is None:
@ -41,39 +64,17 @@ async def handle_login(p, credentials: WorldCredentials):
data = await Penguin.get(credentials.id) data = await Penguin.get(credentials.id)
if data is None:
return await p.send_error_and_disconnect(100)
if not data.active:
return await p.close()
if data.permaban:
return await p.close()
active_ban = await Ban.query.where((Ban.penguin_id == data.id) & (Ban.expires >= datetime.now())).gino.scalar()
if active_ban is not None:
return await p.close()
if data.id in p.server.penguins_by_id:
await p.server.penguins_by_id[data.id].close()
p.logger.info('{} logged in successfully'.format(data.username))
p.data = data
p.login_key = login_key p.login_key = login_key
await p.send_xt('l') await world_login(p, data)
@handlers.handler(XMLPacket('login'), client=ClientType.Legacy) @handlers.handler(XMLPacket('login'), client=ClientType.Legacy)
@handlers.allow_once @handlers.allow_once
@handlers.depends_on_packet(XMLPacket('verChk'), XMLPacket('rndK')) @handlers.depends_on_packet(XMLPacket('verChk'), XMLPacket('rndK'))
async def handle_legacy_login(p, credentials: Credentials): async def handle_legacy_login(p, credentials: Credentials):
if len(p.server.penguins_by_id) >= p.server.server_config['Capacity']:
return await p.send_error_and_disconnect(103)
tr = p.server.redis.multi_exec() tr = p.server.redis.multi_exec()
tr.get('{}.lkey'.format(credentials.username)) tr.get(f'{credentials.username}.lkey')
tr.delete('{}.lkey'.format(credentials.username), '{}.ckey'.format(credentials.username)) tr.delete(f'{credentials.username}.lkey', '{credentials.username}.ckey')
login_key, _ = await tr.execute() login_key, _ = await tr.execute()
login_key = login_key.decode() login_key = login_key.decode()
@ -84,24 +85,5 @@ async def handle_legacy_login(p, credentials: Credentials):
data = await Penguin.query.where(Penguin.username == credentials.username).gino.first() data = await Penguin.query.where(Penguin.username == credentials.username).gino.first()
if data is None:
return await p.send_error_and_disconnect(100)
if not data.active:
return await p.close()
if data.permaban:
return await p.close()
active_ban = await Ban.query.where((Ban.penguin_id == data.id) & (Ban.expires >= datetime.now())).gino.scalar()
if active_ban is not None:
return await p.close()
if data.id in p.server.penguins_by_id:
await p.server.penguins_by_id[data.id].close()
p.logger.info('{} logged in successfully'.format(data.username))
p.data = data
p.login_key = login_key p.login_key = login_key
await p.send_xt('l') await world_login(p, data)