linux/ohmyzsh
1
0
Fork 0
mirror of https://github.com/ohmyzsh/ohmyzsh.git synced 2026-03-27 03:14:56 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

add port-forward and proxy func in ssh cli

Browse source
This commit is contained in:
Erfan Ghavam 2026-02-25 17:05:58 +03:30
commit 42962b7ed2
1 changed files with 59 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

59
plugins/ssh/ssh.plugin.zsh
View file

@ -51,3 +51,62 @@ function ssh_unload_key {
ssh-add -d "$keyfile"
fi
}
############################################################
# Port forwarding
function ssh_port_forward {
if [[ $# -lt 2 ]]; then
echo "Usage: ssh port-forward <local_port>:<remote_port> user@host [ssh options]"
return 1
fi
local ports="$1"
shift
local local_port="${ports%%:*}"
local remote_port="${ports##*:}"
if [[ -z "$local_port" || -z "$remote_port" ]]; then
echo "Invalid port format. Use local_port:remote_port"
return 1
fi
command ssh -N -L "${local_port}:127.0.0.1:${remote_port}" "$@"
}
############################################################
# SSH SOCKS proxy
function ssh_proxy {
local local_port="$1"
if [[ -z "$local_port" ]]; then
echo "Usage: ssh proxy <local_port> user@host [ssh options]"
return 1
fi
shift
if [[ -z "$local_port" || "$local_port" != <-> ]]; then
echo "Invalid port. Use a numeric local port."
return 1
fi
if [[ $# -lt 1 ]]; then
echo "Usage: ssh proxy local_port user@host [ssh options]"
return 1
fi
command ssh -D "${local_port}" -N "$@"
}
############################################################
# Configure ssh command to support custom subcommands
function ssh {
case "$1" in
port-forward|pf)
shift
ssh_port_forward "$@"
return $?
;;
proxy|px)
shift
ssh_proxy "$@"
return $?
;;
*)
command ssh "$@"
;;
esac
}