Réécriture en cours de ssh-fwd
This commit is contained in:
parent
c637cd7dd7
commit
9ff68dda52
1 changed files with 44 additions and 53 deletions
97
bin/ssh-fwd
97
bin/ssh-fwd
|
@ -1,68 +1,59 @@
|
|||
#!/bin/bash
|
||||
|
||||
# Alias si le terminal est kitty
|
||||
if [ "$TERM" = xterm-kitty ]; then
|
||||
ssh=(kitty +kitten ssh)
|
||||
else
|
||||
ssh=(ssh)
|
||||
fi
|
||||
|
||||
# Variables
|
||||
declare -a args
|
||||
|
||||
# Foncitons
|
||||
error() {
|
||||
>&2 printf '\033[1;31m%s\033[0m \033[1m%s\033[0m\n' ERROR: "$*"
|
||||
>&2 printf '\033[1;31mERROR:\033[0m \033[1m%s\033[0m\n' "$error"
|
||||
return "$1"
|
||||
}
|
||||
|
||||
help() {
|
||||
local name
|
||||
name="$(basename "$0")"
|
||||
cat << EOF
|
||||
${name}: utilisation:
|
||||
${name} SERVEUR PORT_LOCAL PORT_DISTANT [CIBLE] [-- ARGUMENTS_SSH...]
|
||||
EOF
|
||||
error_null_arg() {
|
||||
local error
|
||||
error="Variable \"${1}\" can't be null"
|
||||
error 1
|
||||
}
|
||||
|
||||
error_help() {
|
||||
error "Invalid usage"
|
||||
>&2 help
|
||||
exit 1
|
||||
parse_variable() {
|
||||
! :
|
||||
}
|
||||
|
||||
requested_port_bridge_pid() {
|
||||
lsof -nP -i TCP -s TCP:LISTEN | awk '($1 == "ssh" && $9 ~ /.*:'"${1}"'/) { print $2 }' | uniq | head -n 1
|
||||
}
|
||||
# Parser les variables
|
||||
trap error_null_arg EXIT
|
||||
set -e
|
||||
|
||||
# Analyser les arguments
|
||||
while [ -n "$1" ]; do
|
||||
case "$1" in
|
||||
--)
|
||||
ssh_host="$1"
|
||||
shift
|
||||
break
|
||||
;;
|
||||
*)
|
||||
args+=("$1")
|
||||
shift
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
# Il doit y avoir au moins 3 arguments dans le tableau
|
||||
if [ "${#args[@]}" -lt 3 ]; then
|
||||
error_help
|
||||
fi
|
||||
parse_variable ssh_host "$1" 'SSH host'
|
||||
parse_variable local_port 'local port'
|
||||
parse_variable distant_port 'distant port'
|
||||
|
||||
pid="$(requested_port_bridge_pid "${args[1]}")"
|
||||
if [ -z "$pid" ]; then
|
||||
exec "${ssh[@]}" \
|
||||
-f \
|
||||
-N \
|
||||
-L "${args[1]}:${args[3]:-localhost}:${args[2]}" \
|
||||
"${args[0]}" \
|
||||
"$@"
|
||||
trap - EXIT
|
||||
set +e
|
||||
|
||||
# Lancer la redirection SSH
|
||||
start_forwarding() {
|
||||
ssh \
|
||||
-f -N \
|
||||
-L "${local_port}:${target}:${distant_port}" \
|
||||
"$@" \
|
||||
"$ssh_host"
|
||||
}
|
||||
|
||||
# Supprimer la redirection SSH
|
||||
stop_forwarding() {
|
||||
start_forwarding -O cancel
|
||||
}
|
||||
|
||||
# Retourner 0 si le port de la redirection est déjà utilisé par SSH
|
||||
is_forwarding_active() {
|
||||
lsof -i -P -n \
|
||||
| awk '($1 == "ssh" && $10 == "(LISTEN)") { print $9 }' \
|
||||
| cut -d : -f 2 \
|
||||
| grep -q "$local_port"
|
||||
}
|
||||
|
||||
# Branchement principal
|
||||
if ! is_forwarding_active; then
|
||||
start_forwarding
|
||||
else
|
||||
echo "Suppression de la redirection..."
|
||||
kill "$pid"
|
||||
stop_forwarding
|
||||
fi
|
||||
|
||||
|
|
Loading…
Reference in a new issue