tests/test-nbd-extents.sh

   1 #!/usr/bin/env bash
   2 # nbdkit
   3 # Copyright (C) 2019 Red Hat Inc.
   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 # Unfortunately the output of this test depends on the PAGE_SIZE
  34 # defined in common/sparse/sparse.c and would change (breaking the
  35 # test) if we ever changed that definition.
  36
  37 source ./functions.sh
  38 set -e
  39 set -x
  40
  41 requires jq --version
  42 requires qemu-img --version
  43 requires qemu-img map --help
  44
  45 out="test-nbd-extents.out"
  46 expected="test-nbd-extents.expected"
  47 sock=`mktemp -u`
  48 pid1="test-nbd-extents.pid1"
  49 pid2="test-nbd-extents.pid2"
  50 pid3="test-nbd-extents.pid3"
  51 pid4="test-nbd-extents.pid4"
  52 pid5="test-nbd-extents.pid5"
  53 files="$out $expected $sock $pid1 $pid2 $pid3 $pid4 $pid5"
  54 rm -f $files
  55 cleanup_fn rm -f $files
  56
  57 do_test ()
  58 {
  59     start_nbdkit -P "$4" -U "$sock" \
  60            --filter=truncate \
  61            data data="$1" size="$2" \
  62            truncate="$3"
  63     # We use jq to normalize the output and convert it to plain text.
  64     nbdkit -U - nbd socket="$sock" \
  65            --run 'qemu-img map -f raw --output=json $nbd' |
  66         jq -c '.[] | {start:.start, length:.length, data:.data, zero:.zero}' \
  67            > $out
  68     rm -f "$sock"
  69     if ! cmp $out $expected; then
  70         echo "$0: output did not match expected data"
  71         echo "expected:"
  72         cat $expected
  73         echo "output:"
  74         cat $out
  75         exit 1
  76     fi
  77 }
  78
  79 # Completely sparse disk.
  80 cat > $expected <<'EOF'
  81 {"start":0,"length":65536,"data":false,"zero":true}
  82 EOF
  83 do_test "" 1M 65536 "$pid1"
  84
  85 # Completely allocated disk.
  86 cat > $expected <<'EOF'
  87 {"start":0,"length":32768,"data":true,"zero":false}
  88 EOF
  89 do_test "1 @32768 1 @65536 1 @98304 1" 128K 32768 "$pid2"
  90
  91 #----------------------------------------------------------------------
  92 # The above are the easy cases.  Now let's truncate to a larger
  93 # size which should create a hole at the end.
  94
  95 # Completely sparse disk.
  96 cat > $expected <<'EOF'
  97 {"start":0,"length":1048576,"data":false,"zero":true}
  98 EOF
  99 do_test "" 65536 1M "$pid3"
 100
 101 # Completely allocated disk.
 102 cat > $expected <<'EOF'
 103 {"start":0,"length":512,"data":true,"zero":false}
 104 {"start":512,"length":1048064,"data":false,"zero":true}
 105 EOF
 106 do_test "1" 512 1M "$pid4"
 107
 108 # Zero-length plugin.  Unlike nbdkit-zero-plugin, the data plugin
 109 # advertises extents and so will behave differently.
 110 cat > $expected <<'EOF'
 111 {"start":0,"length":0,"data":false,"zero":false}
 112 EOF
 113 do_test "" 0 0 "$pid5"