tests/functions.sh.in

   1 # nbdkit
   2 # Common functions used by the tests.
   3 # @configure_input@
   4 # Copyright (C) 2017-2019 Red Hat Inc.
   5 #
   6 # Redistribution and use in source and binary forms, with or without
   7 # modification, are permitted provided that the following conditions are
   8 # met:
   9 #
  10 # * Redistributions of source code must retain the above copyright
  11 # notice, this list of conditions and the following disclaimer.
  12 #
  13 # * Redistributions in binary form must reproduce the above copyright
  14 # notice, this list of conditions and the following disclaimer in the
  15 # documentation and/or other materials provided with the distribution.
  16 #
  17 # * Neither the name of Red Hat nor the names of its contributors may be
  18 # used to endorse or promote products derived from this software without
  19 # specific prior written permission.
  20 #
  21 # THIS SOFTWARE IS PROVIDED BY RED HAT AND CONTRIBUTORS ''AS IS'' AND
  22 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  23 # THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
  24 # PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL RED HAT OR
  25 # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  28 # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  29 # ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  30 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  31 # OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32 # SUCH DAMAGE.
  33
  34 # cleanup_fn f [args]
  35 #
  36 # A generic trap handling function.  This runs the function or command
  37 # f plus optional args when the script exits for any reason.
  38 declare -a _cleanup_hook
  39 cleanup_fn ()
  40 {
  41     _cleanup_hook[${#_cleanup_hook[@]}]="$@"
  42 }
  43
  44 _run_cleanup_hooks ()
  45 {
  46     status=$?
  47     set +e
  48     trap '' INT QUIT TERM EXIT ERR
  49     echo $0: run cleanup hooks: exit code $status
  50
  51     for (( i = 0; i < ${#_cleanup_hook[@]}; ++i )); do
  52         ${_cleanup_hook[i]}
  53     done
  54
  55     exit $status
  56 }
  57 trap _run_cleanup_hooks INT QUIT TERM EXIT ERR
  58
  59 # requires program [args]
  60 #
  61 # Check that ‘program [args]’ works.  If not, skip the test.
  62 # For example to check that qemu-img is available, do:
  63 #
  64 # requires qemu-img --version
  65 requires ()
  66 {
  67     ( "$@" ) </dev/null >/dev/null 2>&1 || {
  68         echo "$0: ‘$*’ failed with error code $?"
  69         echo "$0: test prerequisite is missing or not working"
  70         exit 77
  71     }
  72 }
  73
  74 # qemu cannot connect to ::1 if IPv6 is disabled because of
  75 # the way it uses getaddrinfo.  See:
  76 # https://bugzilla.redhat.com/show_bug.cgi?id=808147
  77 # https://lists.fedoraproject.org/archives/list/devel@lists.fedoraproject.org/thread/SXDLSZ3GKXL6NDAKP4MPJ25IMHKN67X3/
  78 requires_ipv6_loopback ()
  79 {
  80     requires qemu-img --version
  81
  82     # This should fail with "Connection refused".  If IPv6 is broken
  83     # then it fails with "Address family for hostname not supported"
  84     # instead.  It's very unlikely that port 1 is open.
  85     if LANG=C qemu-img info "nbd:[::1]:1" |& \
  86        grep -sq "Address family for hostname not supported"; then
  87         echo "$0: IPv6 loopback is not available, skipping this test"
  88         exit 77
  89     fi
  90 }
  91
  92 # start_nbdkit -P pidfile args...
  93 #
  94 # Run nbdkit with args and wait for it to start up.  If it fails to
  95 # start up, exit with an error message.  Also a cleanup handler is
  96 # installed automatically which kills nbdkit on exit.
  97 start_nbdkit ()
  98 {
  99     # -P <pidfile> must be the first two parameters.
 100     if [ "$1" != "-P" ]; then
 101        echo "$0: start_nbdkit: -P <pidfile> option must be first"
 102        exit 1
 103     fi
 104     pidfile="$2"
 105
 106     # Run nbdkit.
 107     nbdkit -v "$@"
 108
 109     # Wait for the pidfile to appear.
 110     for i in {1..60}; do
 111         if test -s "$pidfile"; then
 112             break
 113         fi
 114         sleep 1
 115     done
 116     if ! test -s "$pidfile"; then
 117         echo "$0: start_nbdkit: PID file $pidfile was not created"
 118         exit 1
 119     fi
 120
 121     # Kill nbdkit on exit.
 122     cleanup_fn kill "$(cat "$pidfile")"
 123 }
 124
 125 # foreach_plugin f [args]
 126 #
 127 # For each plugin that was built, run the function or command f with
 128 # the plugin name as the first argument, optionally followed by the
 129 # remaining args.
 130 foreach_plugin ()
 131 {
 132     f="$1"
 133     shift
 134
 135     for p in @plugins@; do
 136         # Was the plugin built?
 137         d="@top_builddir@/plugins/$p"
 138         if [ -f "$d/.libs/nbdkit-$p-plugin.so" ] ||
 139            [ -f "$d/nbdkit-$p-plugin" ]; then
 140             # Yes so run the test.
 141             "$f" "$p" "$@"
 142         fi
 143     done
 144 }
 145
 146 # pick_unused_port
 147 #
 148 # Picks and returns an "unused" port, setting the global variable
 149 # $port.
 150 #
 151 # This is inherently racy so we only use it where it's absolutely
 152 # necessary (eg. testing TLS because qemu cannot do TLS over a Unix
 153 # domain socket).
 154 pick_unused_port ()
 155 {
 156     requires ss --version
 157
 158     # Start at a random port to make it less likely that two parallel
 159     # tests will conflict.
 160     port=$(( 50000 + (RANDOM%15000) ))
 161     while ss -ltn | grep -sqE ":$port\b"; do
 162         ((port++))
 163         if [ $port -eq 65000 ]; then port=50000; fi
 164     done
 165     echo picked unused port $port
 166 }