tests/test-delay-shutdown.sh

   1 #!/usr/bin/env bash
   2 # nbdkit
   3 # Copyright Red Hat
   4 #
   5 # Redistribution and use in source and binary forms, with or without
   6 # modification, are permitted provided that the following conditions are
   7 # met:
   8 #
   9 # * Redistributions of source code must retain the above copyright
  10 # notice, this list of conditions and the following disclaimer.
  11 #
  12 # * Redistributions in binary form must reproduce the above copyright
  13 # notice, this list of conditions and the following disclaimer in the
  14 # documentation and/or other materials provided with the distribution.
  15 #
  16 # * Neither the name of Red Hat nor the names of its contributors may be
  17 # used to endorse or promote products derived from this software without
  18 # specific prior written permission.
  19 #
  20 # THIS SOFTWARE IS PROVIDED BY RED HAT AND CONTRIBUTORS ''AS IS'' AND
  21 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  22 # THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
  23 # PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL RED HAT OR
  24 # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  25 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  26 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  27 # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  28 # ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  29 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  30 # OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31 # SUCH DAMAGE.
  32
  33 source ./functions.sh
  34 set -x
  35
  36 # Cannot use kill pidfile below on Windows, but must use taskkill instead.
  37 if is_windows; then
  38     echo "$0: this test needs to be revised to work on Windows"
  39     exit 77
  40 fi
  41
  42 requires qemu-io --version
  43 requires timeout 60s true
  44
  45 pidfile=delay-shutdown.pid
  46 sock=$(mktemp -u /tmp/nbdkit-test-sock.XXXXXX)
  47 files="$pidfile $sock"
  48 cleanup_fn rm -f $files
  49 fail=0
  50
  51 # Create a server that delays reads and forces only one connection at a time.
  52 # This tests that the delay filter's use of nbdkit_nanosleep is able to
  53 # react to both connection death and server shutdown without finishing
  54 # the entire delay duration.
  55 start_nbdkit -P $pidfile -U $sock --filter=noparallel --filter=delay \
  56     null 1M serialize=connections rdelay=60
  57
  58 # Early client death should not stall connection of second client.
  59 trap '' ERR
  60
  61 # Find out the exit status for timed out processes
  62 timeout 1s sleep 5 >/dev/null
  63 timeout_status=$?
  64 if test $timeout_status = 0; then
  65     echo "Failed to get timeout exit status"
  66     exit 1
  67 fi
  68
  69 timeout 1s qemu-io -f raw "nbd+unix:///?socket=$sock" -c 'r 0 512' </dev/null
  70 test $? = $timeout_status || {
  71     echo "Unexpected status; qemu-io should have been killed for timing out"
  72     fail=1
  73 }
  74 timeout 5s qemu-io -f raw "nbd+unix:///?socket=$sock" -c 'quit' </dev/null
  75 test $? = 0 || {
  76     echo "Unexpected status; nbdkit was not responsive to allow second qemu-io"
  77     fail=1
  78 }
  79
  80 # The server's response to shutdown signals should not stall on delays
  81 qemu-io -f raw "nbd+unix:///?socket=$sock" -c 'r 0 512' </dev/null &
  82 pid=$!
  83 sleep 1
  84 kill -s INT "$(cat "$pidfile")"
  85 for (( i=0; i < 5; i++ )); do
  86     kill -s 0 "$(cat "$pidfile")" || break
  87     sleep 1
  88 done
  89 if [ $i = 5 ]; then
  90     echo "Unexpected status; nbdkit didn't react fast enough to signal"
  91     fail=1
  92 fi
  93 wait $pid
  94
  95 exit $fail