169 lines
5.4 KiB
Bash
Executable File
169 lines
5.4 KiB
Bash
Executable File
# $OpenBSD: forward-control.sh,v 1.2 2013/11/18 05:09:32 naddy Exp $
|
|
# Placed in the Public Domain.
|
|
|
|
tid="sshd control of local and remote forwarding"
|
|
|
|
LFWD_PORT=3320
|
|
RFWD_PORT=3321
|
|
CTL=$OBJ/ctl-sock
|
|
READY=$OBJ/ready
|
|
|
|
wait_for_file_to_appear() {
|
|
_path=$1
|
|
_n=0
|
|
while test ! -f $_path ; do
|
|
test $_n -eq 1 && trace "waiting for $_path to appear"
|
|
_n=`expr $_n + 1`
|
|
test $_n -ge 20 && return 1
|
|
sleep 1
|
|
done
|
|
return 0
|
|
}
|
|
|
|
wait_for_process_to_exit() {
|
|
_pid=$1
|
|
_n=0
|
|
while kill -0 $_pid 2>/dev/null ; do
|
|
test $_n -eq 1 && trace "waiting for $_pid to exit"
|
|
_n=`expr $_n + 1`
|
|
test $_n -ge 20 && return 1
|
|
sleep 1
|
|
done
|
|
return 0
|
|
}
|
|
|
|
# usage: check_lfwd protocol Y|N message
|
|
check_lfwd() {
|
|
_proto=$1
|
|
_expected=$2
|
|
_message=$3
|
|
rm -f $READY
|
|
${SSH} -oProtocol=$_proto -F $OBJ/ssh_proxy \
|
|
-L$LFWD_PORT:127.0.0.1:$PORT \
|
|
-o ExitOnForwardFailure=yes \
|
|
-n host exec sh -c \'"sleep 60 & echo \$! > $READY ; wait "\' \
|
|
>/dev/null 2>&1 &
|
|
_sshpid=$!
|
|
wait_for_file_to_appear $READY || \
|
|
fatal "check_lfwd ssh fail: $_message"
|
|
${SSH} -F $OBJ/ssh_config -p $LFWD_PORT \
|
|
-oConnectionAttempts=4 host true >/dev/null 2>&1
|
|
_result=$?
|
|
kill $_sshpid `cat $READY` 2>/dev/null
|
|
wait_for_process_to_exit $_sshpid
|
|
if test "x$_expected" = "xY" -a $_result -ne 0 ; then
|
|
fail "check_lfwd failed (expecting success): $_message"
|
|
elif test "x$_expected" = "xN" -a $_result -eq 0 ; then
|
|
fail "check_lfwd succeeded (expecting failure): $_message"
|
|
elif test "x$_expected" != "xY" -a "x$_expected" != "xN" ; then
|
|
fatal "check_lfwd invalid argument \"$_expected\""
|
|
else
|
|
verbose "check_lfwd done (expecting $_expected): $_message"
|
|
fi
|
|
}
|
|
|
|
# usage: check_rfwd protocol Y|N message
|
|
check_rfwd() {
|
|
_proto=$1
|
|
_expected=$2
|
|
_message=$3
|
|
rm -f $READY
|
|
${SSH} -oProtocol=$_proto -F $OBJ/ssh_proxy \
|
|
-R$RFWD_PORT:127.0.0.1:$PORT \
|
|
-o ExitOnForwardFailure=yes \
|
|
-n host exec sh -c \'"sleep 60 & echo \$! > $READY ; wait "\' \
|
|
>/dev/null 2>&1 &
|
|
_sshpid=$!
|
|
wait_for_file_to_appear $READY
|
|
_result=$?
|
|
if test $_result -eq 0 ; then
|
|
${SSH} -F $OBJ/ssh_config -p $RFWD_PORT \
|
|
-oConnectionAttempts=4 host true >/dev/null 2>&1
|
|
_result=$?
|
|
kill $_sshpid `cat $READY` 2>/dev/null
|
|
wait_for_process_to_exit $_sshpid
|
|
fi
|
|
if test "x$_expected" = "xY" -a $_result -ne 0 ; then
|
|
fail "check_rfwd failed (expecting success): $_message"
|
|
elif test "x$_expected" = "xN" -a $_result -eq 0 ; then
|
|
fail "check_rfwd succeeded (expecting failure): $_message"
|
|
elif test "x$_expected" != "xY" -a "x$_expected" != "xN" ; then
|
|
fatal "check_rfwd invalid argument \"$_expected\""
|
|
else
|
|
verbose "check_rfwd done (expecting $_expected): $_message"
|
|
fi
|
|
}
|
|
|
|
start_sshd
|
|
cp ${OBJ}/sshd_proxy ${OBJ}/sshd_proxy.bak
|
|
cp ${OBJ}/authorized_keys_${USER} ${OBJ}/authorized_keys_${USER}.bak
|
|
|
|
# Sanity check: ensure the default config allows forwarding
|
|
for p in 1 2 ; do
|
|
check_lfwd $p Y "proto $p, default configuration"
|
|
check_rfwd $p Y "proto $p, default configuration"
|
|
done
|
|
|
|
# Usage: all_tests yes|local|remote|no Y|N Y|N Y|N Y|N Y|N Y|N
|
|
all_tests() {
|
|
_tcpfwd=$1
|
|
_plain_lfwd=$2
|
|
_plain_rfwd=$3
|
|
_nopermit_lfwd=$4
|
|
_nopermit_rfwd=$5
|
|
_permit_lfwd=$6
|
|
_permit_rfwd=$7
|
|
_badfwd=127.0.0.1:22
|
|
_goodfwd=127.0.0.1:${PORT}
|
|
for _proto in 1 2 ; do
|
|
cp ${OBJ}/authorized_keys_${USER}.bak \
|
|
${OBJ}/authorized_keys_${USER}
|
|
_prefix="proto $_proto, AllowTcpForwarding=$_tcpfwd"
|
|
# No PermitOpen
|
|
( cat ${OBJ}/sshd_proxy.bak ;
|
|
echo "AllowTcpForwarding $_tcpfwd" ) \
|
|
> ${OBJ}/sshd_proxy
|
|
check_lfwd $_proto $_plain_lfwd "$_prefix"
|
|
check_rfwd $_proto $_plain_rfwd "$_prefix"
|
|
# PermitOpen via sshd_config that doesn't match
|
|
( cat ${OBJ}/sshd_proxy.bak ;
|
|
echo "AllowTcpForwarding $_tcpfwd" ;
|
|
echo "PermitOpen $_badfwd" ) \
|
|
> ${OBJ}/sshd_proxy
|
|
check_lfwd $_proto $_nopermit_lfwd "$_prefix, !PermitOpen"
|
|
check_rfwd $_proto $_nopermit_rfwd "$_prefix, !PermitOpen"
|
|
# PermitOpen via sshd_config that does match
|
|
( cat ${OBJ}/sshd_proxy.bak ;
|
|
echo "AllowTcpForwarding $_tcpfwd" ;
|
|
echo "PermitOpen $_badfwd $_goodfwd" ) \
|
|
> ${OBJ}/sshd_proxy
|
|
# NB. permitopen via authorized_keys should have same
|
|
# success/fail as via sshd_config
|
|
# permitopen via authorized_keys that doesn't match
|
|
sed "s/^/permitopen=\"$_badfwd\" /" \
|
|
< ${OBJ}/authorized_keys_${USER}.bak \
|
|
> ${OBJ}/authorized_keys_${USER} || fatal "sed 1 fail"
|
|
( cat ${OBJ}/sshd_proxy.bak ;
|
|
echo "AllowTcpForwarding $_tcpfwd" ) \
|
|
> ${OBJ}/sshd_proxy
|
|
check_lfwd $_proto $_nopermit_lfwd "$_prefix, !permitopen"
|
|
check_rfwd $_proto $_nopermit_rfwd "$_prefix, !permitopen"
|
|
# permitopen via authorized_keys that does match
|
|
sed "s/^/permitopen=\"$_badfwd\",permitopen=\"$_goodfwd\" /" \
|
|
< ${OBJ}/authorized_keys_${USER}.bak \
|
|
> ${OBJ}/authorized_keys_${USER} || fatal "sed 2 fail"
|
|
( cat ${OBJ}/sshd_proxy.bak ;
|
|
echo "AllowTcpForwarding $_tcpfwd" ) \
|
|
> ${OBJ}/sshd_proxy
|
|
check_lfwd $_proto $_permit_lfwd "$_prefix, permitopen"
|
|
check_rfwd $_proto $_permit_rfwd "$_prefix, permitopen"
|
|
done
|
|
}
|
|
|
|
# no-permitopen mismatch-permitopen match-permitopen
|
|
# AllowTcpForwarding local remote local remote local remote
|
|
all_tests yes Y Y N Y Y Y
|
|
all_tests local Y N N N Y N
|
|
all_tests remote N Y N Y N Y
|
|
all_tests no N N N N N N
|