mirror of
https://github.com/ohmyzsh/ohmyzsh.git
synced 2026-01-30 02:44:42 +01:00
c1c9416e3a
the ssh-proxy.py generates `nc` commands to connect the proxy, however, in cygwin, it does not set the protocol type, leading to nc using wrong default protocol (socks5) if we actually use an http proxy.
37 lines
1.1 KiB
Python
Executable file
37 lines
1.1 KiB
Python
Executable file
#!/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 in {'linux', 'cygwin'}:
|
|
# caveats: the built-in netcat of most linux distributions and cygwin support proxy type
|
|
# 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())
|