-
Timothee Besset authoredTimothee Besset authored
approve-ssh-key 2.51 KiB
#!/usr/bin/env python3
import sys
import os
import logging
import tempfile
from urllib.parse import quote_plus as urllib_quote_plus
import subprocess
import json
ENABLE_SSHD = '/usr/bin/steamos-polkit-helpers/steamos-enable-sshd'
logger = logging.getLogger(os.path.realpath(__file__))
import devkit_utils
if __name__ == '__main__':
logger.setLevel(logging.INFO)
ch = logging.StreamHandler()
ch.setLevel(logging.INFO)
ch.setFormatter(logging.Formatter('%(name)s:%(message)s'))
logger.addHandler(ch)
request = open(sys.argv[1], 'rt').read()
requestIP = 'Unknown'
try:
requestIP = sys.argv[2]
except NameError:
logger.warning('request IP not given as argument to hook')
pass
key_identifier = request.split(' ')[2]
request = f'Development host at IP {requestIP} key: {key_identifier!r}'
ret = {}
try:
devkit_utils.validate_steam_client()
except devkit_utils.SteamClientNotRunningException as e:
msg = 'Steam is not running'
logger.warning(msg)
ret['error'] = msg
else:
with tempfile.TemporaryDirectory(prefix='approve-ssh-key') as tempdir:
logger.info('Sending pairing request to Steam')
response = os.path.join(tempdir, 'response')
cmd = 'approve-ssh-key?response={}&request={}'.format(
urllib_quote_plus(response),
urllib_quote_plus(request),
)
devkit_utils.execute_steam_client_command(cmd)
try:
with devkit_utils.wait_on_file_response(response, timeout=30) as f:
logger.debug('Got response from Steam client')
# Make sure sshd is enabled to support development features
if os.path.exists(ENABLE_SSHD):
subprocess.check_call(ENABLE_SSHD)
else:
subprocess.check_call('sudo systemctl enable --now sshd', shell=True)
except devkit_utils.SteamResponse_Timeout:
ret['error'] = 'timeout - Steam did not respond to the pairing request'
except devkit_utils.SteamResponse_Error as e:
ret['error'] = e.error_response
# json dictionary response gets written to stdout
# NOTE: the devkit service unfortunately smashes stdout/stderr all together in a text response, but we can still work with that
sys.stdout.flush()
sys.stderr.flush()
print(json.dumps(ret))
retcode = 1 if 'error' in ret else 0
sys.exit(retcode)