mirror of
https://github.com/sshuttle/sshuttle.git
synced 2024-11-25 01:13:37 +01:00
124 lines
2.8 KiB
Bash
Executable File
124 lines
2.8 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -e
|
|
|
|
function with_set_x() {
|
|
set -x
|
|
"$@"
|
|
{
|
|
ec=$?
|
|
set +x
|
|
return $ec
|
|
} 2>/dev/null
|
|
}
|
|
|
|
function log() {
|
|
echo "$*" >&2
|
|
}
|
|
|
|
ssh_cmd='ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null'
|
|
ssh_copy_id=false
|
|
args=()
|
|
while [[ $# -gt 0 ]]; do
|
|
arg=$1
|
|
shift
|
|
case "$arg" in
|
|
-v)
|
|
ssh_cmd+='-v'
|
|
;;
|
|
-r)
|
|
args+=("$arg" "$1")
|
|
shift
|
|
continue
|
|
;;
|
|
--copy-id)
|
|
ssh_copy_id=true
|
|
continue
|
|
;;
|
|
--server-py=*)
|
|
server_pyenv_ver="${arg#*=}"
|
|
continue
|
|
;;
|
|
--client-py=*)
|
|
client_pyenv_ver="${arg#*=}"
|
|
continue
|
|
;;
|
|
-6)
|
|
ipv6_only=true
|
|
continue
|
|
;;
|
|
--sshuttle-bin=*)
|
|
sshuttle_bin="${arg#*=}"
|
|
continue
|
|
;;
|
|
-*) ;;
|
|
*)
|
|
if [[ -z "$node" ]]; then
|
|
node=$arg
|
|
continue
|
|
fi
|
|
;;
|
|
esac
|
|
args+=("$arg")
|
|
done
|
|
|
|
port="2222"
|
|
user="test:test"
|
|
|
|
if [[ $node == node-* ]]; then
|
|
pycmd="/pyenv/shims/python"
|
|
ssh_cmd+=" -o SetEnv=PYENV_VERSION=${server_pyenv_ver:-'3'}"
|
|
args=("--python=$pycmd" "${args[@]}")
|
|
host=$("$(dirname "$0")/test-bed" get-ip "$node")
|
|
index=${node#node-}
|
|
if [[ $ipv6_only == true ]]; then
|
|
args+=("2001:0DB8::/112")
|
|
else
|
|
args+=("10.55.$index.0/24")
|
|
fi
|
|
else
|
|
host=$node
|
|
fi
|
|
|
|
if ! command -v sshpass >/dev/null; then
|
|
log "sshpass is not found. You might have to manually enter ssh password: 'test'"
|
|
user=${user%:*}
|
|
fi
|
|
|
|
if [[ $ssh_copy_id == true ]]; then
|
|
log "Trying to make it passwordless"
|
|
with_set_x ssh-copy-id -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -p "$port" "$user@$host"
|
|
fi
|
|
|
|
if [[ -z $sshuttle_bin || "$sshuttle_bin" == dev ]]; then
|
|
cd "$(dirname "$0")/.."
|
|
export PYTHONPATH="."
|
|
if [[ -n $client_pyenv_ver ]]; then
|
|
log "Using pyenv version: $client_pyenv_ver"
|
|
command -v pyenv &>/dev/null || log "You have to install pyenv to use --client-py" && exit 1
|
|
sshuttle_cmd=(/usr/bin/env PYENV_VERSION="$client_pyenv_ver" pyenv exec python -m sshuttle)
|
|
else
|
|
log "Using best python version availble"
|
|
if [ -x "$(command -v python3)" ] &&
|
|
python3 -c "import sys; sys.exit(not sys.version_info > (3, 5))"; then
|
|
sshuttle_cmd=(python3 -m sshuttle)
|
|
else
|
|
sshuttle_cmd=(python -m sshuttle)
|
|
fi
|
|
fi
|
|
else
|
|
[[ -n $client_pyenv_ver ]] && log "Can't specify --client-py when --sshuttle-bin is specified" && exit 1
|
|
sshuttle_cmd=("$sshuttle_bin")
|
|
fi
|
|
|
|
if [[ " ${args[*]} " != *" --ssh-cmd "* ]]; then
|
|
args=("--ssh-cmd" "$ssh_cmd" "${args[@]}")
|
|
fi
|
|
|
|
if [[ " ${args[*]} " != *" -r "* ]]; then
|
|
args=("-r" "$user@$host:$port" "${args[@]}")
|
|
fi
|
|
|
|
set -x
|
|
"${sshuttle_cmd[@]}" --version
|
|
exec "${sshuttle_cmd[@]}" "${args[@]}"
|