From ca968c4e5040f1ff4b5b5e12d9312360d807dcc1 Mon Sep 17 00:00:00 2001 From: littleblack111 Date: Mon, 3 Jun 2024 19:56:39 +0800 Subject: added submodules, change . -> dot (see man stow --dotfiles) --- dot-oh-my-zsh/plugins/shell-proxy/ssh-proxy.py | 36 ++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100755 dot-oh-my-zsh/plugins/shell-proxy/ssh-proxy.py (limited to 'dot-oh-my-zsh/plugins/shell-proxy/ssh-proxy.py') diff --git a/dot-oh-my-zsh/plugins/shell-proxy/ssh-proxy.py b/dot-oh-my-zsh/plugins/shell-proxy/ssh-proxy.py new file mode 100755 index 00000000..a498c84b --- /dev/null +++ b/dot-oh-my-zsh/plugins/shell-proxy/ssh-proxy.py @@ -0,0 +1,36 @@ +#!/usr/bin/env python3 +import os +import subprocess +import sys +from urllib.parse import urlparse + +proxy = next(os.environ[_] for _ in ("HTTP_PROXY", "HTTPS_PROXY") if _ in os.environ) + +parsed = urlparse(proxy) + +proxy_protocols = { + "http": "connect", + "https": "connect", + "socks": "5", + "socks5": "5", + "socks4": "4", + "socks4a": "4", +} + +if parsed.scheme not in proxy_protocols: + raise TypeError('unsupported proxy protocol: "{}"'.format(parsed.scheme)) + +def make_argv(): + yield "nc" + if sys.platform == 'linux': + # caveats: macOS built-in netcat command not supported proxy-type + yield "-X" # --proxy-type + # Supported protocols are 4 (SOCKS v4), 5 (SOCKS v5) and connect (HTTP proxy). + # Default SOCKS v5 is used. + yield proxy_protocols[parsed.scheme] + yield "-x" # --proxy + yield parsed.netloc # proxy-host:proxy-port + yield sys.argv[1] # host + yield sys.argv[2] # port + +subprocess.call(make_argv()) -- cgit v1.2.3