Ssh: add proxyjump solution

This commit is contained in:
Akemi Izuko 2024-08-25 16:02:04 -06:00
parent e8e455ee4d
commit 1e38b26cad
Signed by: akemi
GPG key ID: 8DE0764E1809E9FC

30
ssh_daemon/proxyjump.py Normal file
View file

@ -0,0 +1,30 @@
#!/usr/bin/env python3
import os
import sys
import subprocess
def get_vm_port(username):
user_map = {
'user0': 9000,
'user1': 9001,
'user2': 9002
}
return user_map.get(username)
def main():
username = os.environ.get('USER')
if not username:
print("Error: Unable to determine username", file=sys.stderr)
sys.exit(1)
port = 9002 #get_vm_port(username)
if not port:
print(f"Error: No VM associated with user {username}", file=sys.stderr)
sys.exit(1)
# Use os.execvp to replace the current process with ssh
os.execvp('ssh', ['ssh', '-p', str(port), 'root@localhost'])
if __name__ == "__main__":
main()