ClearML/clearml-session
1
0
Fork 0
mirror of https://github.com/clearml/clearml-session synced 2025-02-07 13:22:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #26 from mukhery/bugfix-key-login

Browse Source 
bugfix: handle ssh key login
This commit is contained in:
Jake Henning 2023-07-18 23:35:03 +03:00 committed by GitHub
commit 44827aec0a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
clearml_session/__main__.py
View File

@ -696,7 +696,7 @@ def wait_for_machine(state, task):
def start_ssh_tunnel(username, remote_address, ssh_port, ssh_password, local_remote_pair_list, debug=False): def start_ssh_tunnel(username, remote_address, ssh_port, ssh_password, local_remote_pair_list, debug=False):
print('Starting SSH tunnel to {}@{}, port {}'.format(username, remote_address, ssh_port)) print('Starting SSH tunnel to {}@{}, port {}'.format(username, remote_address, ssh_port))
child = None child = None
args = ['-N', '-C', args = ['-C',
'{}@{}'.format(username, remote_address), '-p', '{}'.format(ssh_port), '{}@{}'.format(username, remote_address), '-p', '{}'.format(ssh_port),
'-o', 'UserKnownHostsFile=/dev/null', '-o', 'UserKnownHostsFile=/dev/null',
'-o', 'Compression=yes', '-o', 'Compression=yes',
@ -717,7 +717,8 @@ def start_ssh_tunnel(username, remote_address, ssh_port, ssh_password, local_rem
args=args, args=args,
logfile=fd, timeout=20, encoding='utf-8') logfile=fd, timeout=20, encoding='utf-8')
i = child.expect([r'(?i)password:', r'\(yes\/no\)', r'.*[$#] ', pexpect.EOF]) # Match only "(yes/no" in order to handle both (yes/no) and (yes/no/[fingerprint])
i = child.expect([r'(?i)password:', r'\(yes\/no', r'.*[$#] ', pexpect.EOF])
if i == 0: if i == 0:
child.sendline(ssh_password) child.sendline(ssh_password)
try: try: