From 6789e3e3deb1e2da83acb06aeff2a83ad3a91c97 Mon Sep 17 00:00:00 2001 From: Randall Winkhart Date: Thu, 4 Jan 2024 14:08:02 -0500 Subject: [PATCH] Improve method of serving whitelist contents to clients --- lib/sshyp.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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