kevin/ohmyzsh
1
Fork 0
mirror of https://github.com/ohmyzsh/ohmyzsh.git synced 2024-11-15 01:48:34 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix(shell-proxy): make ssh-proxy compatible with macOS (#10640)

Browse Source
This commit is contained in:
septs 2022-04-12 19:26:22 +08:00 committed by GitHub
parent 846f417eb8
commit 9fa3f46122
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
plugins/shell-proxy/.editorconfig Normal file
View File

@ -0,0 +1,3 @@
[*.py]
indent_size = 4
indent_style = space

23
plugins/shell-proxy/ssh-proxy.py
View File

@ -20,14 +20,17 @@ proxy_protocols = {
if parsed.scheme not in proxy_protocols:
raise TypeError('unsupported proxy protocol: "{}"'.format(parsed.scheme))
argv = [
"nc",
"-X",
proxy_protocols[parsed.scheme], # Supported protocols are 4 (SOCKS v4), 5 (SOCKS v5) and connect (HTTP proxy). Default SOCKS v5 is used.
"-x",
parsed.netloc, # proxy-host:proxy-port
sys.argv[1], # host
sys.argv[2], # port
]
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(argv)
subprocess.call(make_argv())