crypto/openssh/regress/multiplex.sh

   1 #       $OpenBSD: multiplex.sh,v 1.27 2014/12/22 06:14:29 djm Exp $
   2 #       Placed in the Public Domain.
   3
   4 CTL=/tmp/openssh.regress.ctl-sock.$$
   5
   6 tid="connection multiplexing"
   7
   8 NC=$OBJ/netcat
   9
  10 trace "will use ProxyCommand $proxycmd"
  11 if config_defined DISABLE_FD_PASSING ; then
  12         echo "skipped (not supported on this platform)"
  13         exit 0
  14 fi
  15
  16 P=3301  # test port
  17
  18 wait_for_mux_master_ready()
  19 {
  20         for i in 1 2 3 4 5; do
  21                 ${SSH} -F $OBJ/ssh_config -S $CTL -Ocheck otherhost \
  22                     >/dev/null 2>&1 && return 0
  23                 sleep $i
  24         done
  25         fatal "mux master never becomes ready"
  26 }
  27
  28 start_sshd
  29
  30 start_mux_master()
  31 {
  32         trace "start master, fork to background"
  33         ${SSH} -Nn2 -MS$CTL -F $OBJ/ssh_config -oSendEnv="_XXX_TEST" somehost \
  34             -E $TEST_REGRESS_LOGFILE 2>&1 &
  35         # NB. $SSH_PID will be killed by test-exec.sh:cleanup on fatal errors.
  36         SSH_PID=$!
  37         wait_for_mux_master_ready
  38 }
  39
  40 start_mux_master
  41
  42 verbose "test $tid: envpass"
  43 trace "env passing over multiplexed connection"
  44 _XXX_TEST=blah ${SSH} -F $OBJ/ssh_config -oSendEnv="_XXX_TEST" -S$CTL otherhost sh << 'EOF'
  45         test X"$_XXX_TEST" = X"blah"
  46 EOF
  47 if [ $? -ne 0 ]; then
  48         fail "environment not found"
  49 fi
  50
  51 verbose "test $tid: transfer"
  52 rm -f ${COPY}
  53 trace "ssh transfer over multiplexed connection and check result"
  54 ${SSH} -F $OBJ/ssh_config -S$CTL otherhost cat ${DATA} > ${COPY}
  55 test -f ${COPY}                 || fail "ssh -Sctl: failed copy ${DATA}"
  56 cmp ${DATA} ${COPY}             || fail "ssh -Sctl: corrupted copy of ${DATA}"
  57
  58 rm -f ${COPY}
  59 trace "ssh transfer over multiplexed connection and check result"
  60 ${SSH} -F $OBJ/ssh_config -S $CTL otherhost cat ${DATA} > ${COPY}
  61 test -f ${COPY}                 || fail "ssh -S ctl: failed copy ${DATA}"
  62 cmp ${DATA} ${COPY}             || fail "ssh -S ctl: corrupted copy of ${DATA}"
  63
  64 rm -f ${COPY}
  65 trace "sftp transfer over multiplexed connection and check result"
  66 echo "get ${DATA} ${COPY}" | \
  67         ${SFTP} -S ${SSH} -F $OBJ/ssh_config -oControlPath=$CTL otherhost >>$TEST_REGRESS_LOGFILE 2>&1
  68 test -f ${COPY}                 || fail "sftp: failed copy ${DATA}"
  69 cmp ${DATA} ${COPY}             || fail "sftp: corrupted copy of ${DATA}"
  70
  71 rm -f ${COPY}
  72 trace "scp transfer over multiplexed connection and check result"
  73 ${SCP} -S ${SSH} -F $OBJ/ssh_config -oControlPath=$CTL otherhost:${DATA} ${COPY} >>$TEST_REGRESS_LOGFILE 2>&1
  74 test -f ${COPY}                 || fail "scp: failed copy ${DATA}"
  75 cmp ${DATA} ${COPY}             || fail "scp: corrupted copy of ${DATA}"
  76
  77 rm -f ${COPY}
  78 verbose "test $tid: forward"
  79 trace "forward over TCP/IP and check result"
  80 $NC -N -l 127.0.0.1 $((${PORT} + 1)) < ${DATA} > /dev/null &
  81 netcat_pid=$!
  82 ${SSH} -F $OBJ/ssh_config -S $CTL -Oforward -L127.0.0.1:$((${PORT} + 2)):127.0.0.1:$((${PORT} + 1)) otherhost >>$TEST_SSH_LOGFILE 2>&1
  83 $NC 127.0.0.1 $((${PORT} + 2)) < /dev/null > ${COPY}
  84 cmp ${DATA} ${COPY}             || fail "ssh: corrupted copy of ${DATA}"
  85 kill $netcat_pid 2>/dev/null
  86 rm -f ${COPY} $OBJ/unix-[123].fwd
  87
  88 trace "forward over UNIX and check result"
  89 $NC -N -Ul $OBJ/unix-1.fwd < ${DATA} > /dev/null &
  90 netcat_pid=$!
  91 ${SSH} -F $OBJ/ssh_config -S $CTL -Oforward -L$OBJ/unix-2.fwd:$OBJ/unix-1.fwd otherhost >>$TEST_SSH_LOGFILE 2>&1
  92 ${SSH} -F $OBJ/ssh_config -S $CTL -Oforward -R$OBJ/unix-3.fwd:$OBJ/unix-2.fwd otherhost >>$TEST_SSH_LOGFILE 2>&1
  93 $NC -U $OBJ/unix-3.fwd < /dev/null > ${COPY} 2>/dev/null
  94 cmp ${DATA} ${COPY}             || fail "ssh: corrupted copy of ${DATA}"
  95 kill $netcat_pid 2>/dev/null
  96 rm -f ${COPY} $OBJ/unix-[123].fwd
  97
  98 for s in 0 1 4 5 44; do
  99         trace "exit status $s over multiplexed connection"
 100         verbose "test $tid: status $s"
 101         ${SSH} -F $OBJ/ssh_config -S $CTL otherhost exit $s
 102         r=$?
 103         if [ $r -ne $s ]; then
 104                 fail "exit code mismatch for protocol $p: $r != $s"
 105         fi
 106
 107         # same with early close of stdout/err
 108         trace "exit status $s with early close over multiplexed connection"
 109         ${SSH} -F $OBJ/ssh_config -S $CTL -n otherhost \
 110                 exec sh -c \'"sleep 2; exec > /dev/null 2>&1; sleep 3; exit $s"\'
 111         r=$?
 112         if [ $r -ne $s ]; then
 113                 fail "exit code (with sleep) mismatch for protocol $p: $r != $s"
 114         fi
 115 done
 116
 117 verbose "test $tid: cmd check"
 118 ${SSH} -F $OBJ/ssh_config -S $CTL -Ocheck otherhost >>$TEST_REGRESS_LOGFILE 2>&1 \
 119     || fail "check command failed"
 120
 121 verbose "test $tid: cmd forward local (TCP)"
 122 ${SSH} -F $OBJ/ssh_config -S $CTL -Oforward -L $P:localhost:$PORT otherhost \
 123      || fail "request local forward failed"
 124 ${SSH} -F $OBJ/ssh_config -p$P otherhost true \
 125      || fail "connect to local forward port failed"
 126 ${SSH} -F $OBJ/ssh_config -S $CTL -Ocancel -L $P:localhost:$PORT otherhost \
 127      || fail "cancel local forward failed"
 128 ${SSH} -F $OBJ/ssh_config -p$P otherhost true \
 129      && fail "local forward port still listening"
 130
 131 verbose "test $tid: cmd forward remote (TCP)"
 132 ${SSH} -F $OBJ/ssh_config -S $CTL -Oforward -R $P:localhost:$PORT otherhost \
 133      || fail "request remote forward failed"
 134 ${SSH} -F $OBJ/ssh_config -p$P otherhost true \
 135      || fail "connect to remote forwarded port failed"
 136 ${SSH} -F $OBJ/ssh_config -S $CTL -Ocancel -R $P:localhost:$PORT otherhost \
 137      || fail "cancel remote forward failed"
 138 ${SSH} -F $OBJ/ssh_config -p$P otherhost true \
 139      && fail "remote forward port still listening"
 140
 141 verbose "test $tid: cmd forward local (UNIX)"
 142 ${SSH} -F $OBJ/ssh_config -S $CTL -Oforward -L $OBJ/unix-1.fwd:localhost:$PORT otherhost \
 143      || fail "request local forward failed"
 144 echo "" | $NC -U $OBJ/unix-1.fwd | grep "Protocol mismatch" >/dev/null 2>&1 \
 145      || fail "connect to local forward path failed"
 146 ${SSH} -F $OBJ/ssh_config -S $CTL -Ocancel -L $OBJ/unix-1.fwd:localhost:$PORT otherhost \
 147      || fail "cancel local forward failed"
 148 N=$(echo "xyzzy" | $NC -U $OBJ/unix-1.fwd 2>&1 | grep "xyzzy" | wc -l)
 149 test ${N} -eq 0 || fail "local forward path still listening"
 150 rm -f $OBJ/unix-1.fwd
 151
 152 verbose "test $tid: cmd forward remote (UNIX)"
 153 ${SSH} -F $OBJ/ssh_config -S $CTL -Oforward -R $OBJ/unix-1.fwd:localhost:$PORT otherhost \
 154      || fail "request remote forward failed"
 155 echo "" | $NC -U $OBJ/unix-1.fwd | grep "Protocol mismatch" >/dev/null 2>&1 \
 156      || fail "connect to remote forwarded path failed"
 157 ${SSH} -F $OBJ/ssh_config -S $CTL -Ocancel -R $OBJ/unix-1.fwd:localhost:$PORT otherhost \
 158      || fail "cancel remote forward failed"
 159 N=$(echo "xyzzy" | $NC -U $OBJ/unix-1.fwd 2>&1 | grep "xyzzy" | wc -l)
 160 test ${N} -eq 0 || fail "remote forward path still listening"
 161 rm -f $OBJ/unix-1.fwd
 162
 163 verbose "test $tid: cmd exit"
 164 ${SSH} -F $OBJ/ssh_config -S $CTL -Oexit otherhost >>$TEST_REGRESS_LOGFILE 2>&1 \
 165     || fail "send exit command failed"
 166
 167 # Wait for master to exit
 168 wait $SSH_PID
 169 kill -0 $SSH_PID >/dev/null 2>&1 && fail "exit command failed"
 170
 171 # Restart master and test -O stop command with master using -N
 172 verbose "test $tid: cmd stop"
 173 trace "restart master, fork to background"
 174 start_mux_master
 175
 176 # start a long-running command then immediately request a stop
 177 ${SSH} -F $OBJ/ssh_config -S $CTL otherhost "sleep 10; exit 0" \
 178      >>$TEST_REGRESS_LOGFILE 2>&1 &
 179 SLEEP_PID=$!
 180 ${SSH} -F $OBJ/ssh_config -S $CTL -Ostop otherhost >>$TEST_REGRESS_LOGFILE 2>&1 \
 181     || fail "send stop command failed"
 182
 183 # wait until both long-running command and master have exited.
 184 wait $SLEEP_PID
 185 [ $! != 0 ] || fail "waiting for concurrent command"
 186 wait $SSH_PID
 187 [ $! != 0 ] || fail "waiting for master stop"
 188 kill -0 $SSH_PID >/dev/null 2>&1 && fatal "stop command failed"
 189 SSH_PID="" # Already gone, so don't kill in cleanup
 190