diff --git a/lib/sshyp.py b/lib/sshyp.py index fe3e482..00df5dc 100755 --- a/lib/sshyp.py +++ b/lib/sshyp.py @@ -193,9 +193,9 @@ def whitelist_verify(_port, _username_ssh, _ip, _client_device_id, _identity): except CalledProcessError: _i, _full_password = 0, '' _server_whitelist = run(('ssh', '-i', _identity, '-p', _port, f"{_username_ssh}@{_ip}", - f'python3 -c \'from os import listdir; print(listdir("/home/{_username_ssh}' - f'/.config/sshyp/whitelist"))\''), - stdout=PIPE, text=True).stdout.rstrip()[2:-2].split("', '") + 'python3 -c \'from os import listdir; [print(_id, end="\x1f") for _id in ' + f'listdir("/home/{_username_ssh}/.config/sshyp/whitelist")]\''), + stdout=PIPE, text=True).stdout.rstrip().split('\x1f') for _device_id in _server_whitelist: if _device_id == _client_device_id: from getpass import getpass