diff --git a/src/ssh_manager.py b/src/ssh_manager.py index 241eb58..de718ff 100644 --- a/src/ssh_manager.py +++ b/src/ssh_manager.py @@ -61,7 +61,7 @@ def main(): exit_program() elif i == 'b' or i == 'B': sub_menu() - elif "C" == str.lower(i): + elif "c" == str.lower(i): socks_ssh_tunnel() else: for c in cmp: @@ -83,22 +83,22 @@ def socks_ssh_tunnel(): cmp.append(Computer(line, count)) count += 1 - print(print_magenta("*") + " " + print_green("Socks Tunnel") + " " + print_magenta("*")) + print(print_magenta("*") + " " + print_green("Socks Tunnel") + " " + print_magenta("*")) for c in cmp: print_menu_option(str(c.menu_id) + ") " + c.host) print_menu_option("A) Exit") print_menu_option("B) Main") print(print_magenta("*" * len(WELCOME_MESSAGE))) i = input("Enter number of computer to connect to or enter to exit:") - if i == '' or 'A' == str.lower(i): + if i == '' or 'c' == str.lower(i): exit_program() - elif 'B' == str.lower(i): + elif 'c' == str.lower(i): main() else: for c in cmp: if int(i) == c.menu_id: print_red("Starting socks proxy on " + c.host + ":8123") - subprocess.call(["ssh", "D", "8123", "f", "C", "q", "N", c.host]) + subprocess.call(["ssh", "-D", "8123", "-C", "-q", "-N", c.host]) exit_program()