regress/test-exec.sh

   1 #       $OpenBSD: test-exec.sh,v 1.28 2005/05/20 23:14:15 djm Exp $
   2 #       Placed in the Public Domain.
   3
   4 #SUDO=sudo
   5
   6 # Unbreak GNU head(1)
   7 _POSIX2_VERSION=199209
   8 export _POSIX2_VERSION
   9
  10 case `uname -s 2>/dev/null` in
  11 OSF1*)
  12         BIN_SH=xpg4
  13         export BIN_SH
  14         ;;
  15 esac
  16
  17 if [ ! -z "$TEST_SSH_PORT" ]; then
  18         PORT="$TEST_SSH_PORT"
  19 else
  20         PORT=4242
  21 fi
  22
  23 if [ -x /usr/ucb/whoami ]; then
  24         USER=`/usr/ucb/whoami`
  25 elif whoami >/dev/null 2>&1; then
  26         USER=`whoami`
  27 else
  28         USER=`id -un`
  29 fi
  30
  31 OBJ=$1
  32 if [ "x$OBJ" = "x" ]; then
  33         echo '$OBJ not defined'
  34         exit 2
  35 fi
  36 if [ ! -d $OBJ ]; then
  37         echo "not a directory: $OBJ"
  38         exit 2
  39 fi
  40 SCRIPT=$2
  41 if [ "x$SCRIPT" = "x" ]; then
  42         echo '$SCRIPT not defined'
  43         exit 2
  44 fi
  45 if [ ! -f $SCRIPT ]; then
  46         echo "not a file: $SCRIPT"
  47         exit 2
  48 fi
  49 if $TEST_SHELL -n $SCRIPT; then
  50         true
  51 else
  52         echo "syntax error in $SCRIPT"
  53         exit 2
  54 fi
  55 unset SSH_AUTH_SOCK
  56
  57 SRC=`dirname ${SCRIPT}`
  58
  59 # defaults
  60 SSH=ssh
  61 SSHD=sshd
  62 SSHAGENT=ssh-agent
  63 SSHADD=ssh-add
  64 SSHKEYGEN=ssh-keygen
  65 SSHKEYSCAN=ssh-keyscan
  66 SFTP=sftp
  67 SFTPSERVER=/usr/libexec/openssh/sftp-server
  68 SCP=scp
  69
  70 if [ "x$TEST_SSH_SSH" != "x" ]; then
  71         SSH="${TEST_SSH_SSH}"
  72 fi
  73 if [ "x$TEST_SSH_SSHD" != "x" ]; then
  74         SSHD="${TEST_SSH_SSHD}"
  75 fi
  76 if [ "x$TEST_SSH_SSHAGENT" != "x" ]; then
  77         SSHAGENT="${TEST_SSH_SSHAGENT}"
  78 fi
  79 if [ "x$TEST_SSH_SSHADD" != "x" ]; then
  80         SSHADD="${TEST_SSH_SSHADD}"
  81 fi
  82 if [ "x$TEST_SSH_SSHKEYGEN" != "x" ]; then
  83         SSHKEYGEN="${TEST_SSH_SSHKEYGEN}"
  84 fi
  85 if [ "x$TEST_SSH_SSHKEYSCAN" != "x" ]; then
  86         SSHKEYSCAN="${TEST_SSH_SSHKEYSCAN}"
  87 fi
  88 if [ "x$TEST_SSH_SFTP" != "x" ]; then
  89         SFTP="${TEST_SSH_SFTP}"
  90 fi
  91 if [ "x$TEST_SSH_SFTPSERVER" != "x" ]; then
  92         SFTPSERVER="${TEST_SSH_SFTPSERVER}"
  93 fi
  94 if [ "x$TEST_SSH_SCP" != "x" ]; then
  95         SCP="${TEST_SSH_SCP}"
  96 fi
  97
  98 # Path to sshd must be absolute for rexec
  99 case "$SSHD" in
 100 /*) ;;
 101 *) SSHD=`which sshd` ;;
 102 esac
 103
 104 if [ "x$TEST_SSH_LOGFILE" = "x" ]; then
 105         TEST_SSH_LOGFILE=/dev/null
 106 fi
 107
 108 # these should be used in tests
 109 export SSH SSHD SSHAGENT SSHADD SSHKEYGEN SSHKEYSCAN SFTP SFTPSERVER SCP
 110 #echo $SSH $SSHD $SSHAGENT $SSHADD $SSHKEYGEN $SSHKEYSCAN $SFTP $SFTPSERVER $SCP
 111
 112 # helper
 113 echon()
 114 {
 115        if [ "x`echo -n`" = "x" ]; then
 116                echo -n "$@"
 117        elif [ "x`echo '\c'`" = "x" ]; then
 118                echo "$@\c"
 119        else
 120                fatal "Don't know how to echo without newline."
 121        fi
 122 }
 123
 124 have_prog()
 125 {
 126         saved_IFS="$IFS"
 127         IFS=":"
 128         for i in $PATH
 129         do
 130                 if [ -x $i/$1 ]; then
 131                         IFS="$saved_IFS"
 132                         return 0
 133                 fi
 134         done
 135         IFS="$saved_IFS"
 136         return 1
 137 }
 138
 139 cleanup ()
 140 {
 141         if [ -f $PIDFILE ]; then
 142                 pid=`cat $PIDFILE`
 143                 if [ "X$pid" = "X" ]; then
 144                         echo no sshd running
 145                 else
 146                         if [ $pid -lt 2 ]; then
 147                                 echo bad pid for ssd: $pid
 148                         else
 149                                 $SUDO kill $pid
 150                         fi
 151                 fi
 152         fi
 153 }
 154
 155 trace ()
 156 {
 157         echo "trace: $@" >>$TEST_SSH_LOGFILE
 158         if [ "X$TEST_SSH_TRACE" = "Xyes" ]; then
 159                 echo "$@"
 160         fi
 161 }
 162
 163 verbose ()
 164 {
 165         echo "verbose: $@" >>$TEST_SSH_LOGFILE
 166         if [ "X$TEST_SSH_QUIET" != "Xyes" ]; then
 167                 echo "$@"
 168         fi
 169 }
 170
 171
 172 fail ()
 173 {
 174         echo "FAIL: $@" >>$TEST_SSH_LOGFILE
 175         RESULT=1
 176         echo "$@"
 177 }
 178
 179 fatal ()
 180 {
 181         echo "FATAL: $@" >>$TEST_SSH_LOGFILE
 182         echon "FATAL: "
 183         fail "$@"
 184         cleanup
 185         exit $RESULT
 186 }
 187
 188 RESULT=0
 189 PIDFILE=$OBJ/pidfile
 190
 191 trap fatal 3 2
 192
 193 # create server config
 194 cat << EOF > $OBJ/sshd_config
 195         StrictModes             no
 196         Port                    $PORT
 197         AddressFamily           inet
 198         ListenAddress           127.0.0.1
 199         #ListenAddress          ::1
 200         PidFile                 $PIDFILE
 201         AuthorizedKeysFile      $OBJ/authorized_keys_%u
 202         LogLevel                VERBOSE
 203         AcceptEnv               _XXX_TEST_*
 204         AcceptEnv               _XXX_TEST
 205         Subsystem       sftp    $SFTPSERVER
 206 EOF
 207
 208 if [ ! -z "$TEST_SSH_SSHD_CONFOPTS" ]; then
 209         trace "adding sshd_config option $TEST_SSH_SSHD_CONFOPTS"
 210         echo "$TEST_SSH_SSHD_CONFOPTS" >> $OBJ/sshd_config
 211 fi
 212
 213 # server config for proxy connects
 214 cp $OBJ/sshd_config $OBJ/sshd_proxy
 215
 216 # allow group-writable directories in proxy-mode
 217 echo 'StrictModes no' >> $OBJ/sshd_proxy
 218
 219 # create client config
 220 cat << EOF > $OBJ/ssh_config
 221 Host *
 222         Hostname                127.0.0.1
 223         HostKeyAlias            localhost-with-alias
 224         Port                    $PORT
 225         User                    $USER
 226         GlobalKnownHostsFile    $OBJ/known_hosts
 227         UserKnownHostsFile      $OBJ/known_hosts
 228         RSAAuthentication       yes
 229         PubkeyAuthentication    yes
 230         ChallengeResponseAuthentication no
 231         HostbasedAuthentication no
 232         PasswordAuthentication  no
 233         BatchMode               yes
 234         StrictHostKeyChecking   yes
 235 EOF
 236
 237 if [ ! -z "$TEST_SSH_SSH_CONFOPTS" ]; then
 238         trace "adding ssh_config option $TEST_SSH_SSHD_CONFOPTS"
 239         echo "$TEST_SSH_SSH_CONFOPTS" >> $OBJ/ssh_config
 240 fi
 241
 242 rm -f $OBJ/known_hosts $OBJ/authorized_keys_$USER
 243
 244 trace "generate keys"
 245 for t in rsa rsa1; do
 246         # generate user key
 247         rm -f $OBJ/$t
 248         ${SSHKEYGEN} -b 1024 -q -N '' -t $t  -f $OBJ/$t ||\
 249                 fail "ssh-keygen for $t failed"
 250
 251         # known hosts file for client
 252         (
 253                 echon 'localhost-with-alias,127.0.0.1,::1 '
 254                 cat $OBJ/$t.pub
 255         ) >> $OBJ/known_hosts
 256
 257         # setup authorized keys
 258         cat $OBJ/$t.pub >> $OBJ/authorized_keys_$USER
 259         echo IdentityFile $OBJ/$t >> $OBJ/ssh_config
 260
 261         # use key as host key, too
 262         $SUDO cp $OBJ/$t $OBJ/host.$t
 263         echo HostKey $OBJ/host.$t >> $OBJ/sshd_config
 264
 265         # don't use SUDO for proxy connect
 266         echo HostKey $OBJ/$t >> $OBJ/sshd_proxy
 267 done
 268 chmod 644 $OBJ/authorized_keys_$USER
 269
 270 # create a proxy version of the client config
 271 (
 272         cat $OBJ/ssh_config
 273         echo proxycommand ${SUDO} sh ${SRC}/sshd-log-wrapper.sh ${SSHD} ${TEST_SSH_LOGFILE} -i -f $OBJ/sshd_proxy
 274 ) > $OBJ/ssh_proxy
 275
 276 # check proxy config
 277 ${SSHD} -t -f $OBJ/sshd_proxy   || fatal "sshd_proxy broken"
 278
 279 start_sshd ()
 280 {
 281         # start sshd
 282         $SUDO ${SSHD} -f $OBJ/sshd_config -t    || fatal "sshd_config broken"
 283         $SUDO ${SSHD} -f $OBJ/sshd_config -e >>$TEST_SSH_LOGFILE 2>&1
 284
 285         trace "wait for sshd"
 286         i=0;
 287         while [ ! -f $PIDFILE -a $i -lt 10 ]; do
 288                 i=`expr $i + 1`
 289                 sleep $i
 290         done
 291
 292         test -f $PIDFILE || fatal "no sshd running on port $PORT"
 293 }
 294
 295 # source test body
 296 . $SCRIPT
 297
 298 # kill sshd
 299 cleanup
 300 if [ $RESULT -eq 0 ]; then
 301         verbose ok $tid
 302 else
 303         echo failed $tid
 304 fi
 305 exit $RESULT