ClearML/clearml-session
1
0
Fork 0
mirror of https://github.com/clearml/clearml-session synced 2025-06-26 18:16:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Enhance SSH tunnel error handling to include permission denied case

Browse Source
This commit is contained in:
Philipp Kolberg 2025-01-23 10:08:25 +01:00
parent 01c7e9756f
commit 951398bdb3
No known key found for this signature in database
GPG Key ID: 1804EB0943274F8B
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clearml_session/__main__.py
View File

@ -937,7 +937,7 @@ def start_ssh_tunnel(username, remote_address, ssh_port, ssh_password, local_rem
logfile=fd, timeout=20, encoding='utf-8')
# 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])
i = child.expect([r'(?i)password:', r'\(yes\/no', r'.*[$#] ', r'Permission denied \(publickey,password\)', pexpect.EOF])
if i == 0:
child.sendline(ssh_password)
try:
@ -974,6 +974,8 @@ def start_ssh_tunnel(username, remote_address, ssh_port, ssh_password, local_rem
raise ValueError('Incorrect password')
except pexpect.TIMEOUT:
pass
elif i == 3:
raise Exception("SSH tunneling failed: please enable password authentication in your SSH client config.")
except Exception as ex:
if debug:
print("ERROR: running local SSH client [{}] failed connecting to {}: {}".format(command, args, ex))