tests/test-blocksize-default.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 -e
  35 set -x
  36
  37 requires_plugin eval
  38 requires nbdsh -c 'print(h.get_block_size)'
  39 requires dd iflag=count_bytes </dev/null
  40
  41 # Create an nbdkit eval plugin which presents per-export block size
  42 # constraints based on the export name.
  43 # Check that the blocksize filter advertises full access, that size is
  44 # truncated to minblock constraints, and underlying plugin access uses
  45 # read-modify-write at appropriate boundaries.
  46 export script='
  47 import os
  48
  49 sock = os.environ["unixsocket"]
  50
  51 ha = h;
  52 ha.set_export_name("a")
  53 ha.connect_unix(sock)
  54 hb = nbd.NBD()
  55 hb.set_export_name("b")
  56 hb.connect_unix(sock)
  57
  58 assert ha.get_size() == 2 * 1024 * 1024 - 1 * 1024
  59 assert ha.get_block_size(nbd.SIZE_MINIMUM) == 1
  60 assert ha.get_block_size(nbd.SIZE_PREFERRED) == 128 * 1024
  61 assert ha.get_block_size(nbd.SIZE_MAXIMUM) == 4 * 1024 * 1024 * 1024 - 1
  62 assert ha.pread(1024 * 1024 + 1, 1) == b"\0" * (1024 * 1024 + 1)
  63
  64 assert hb.get_size() == 2 * 1024 * 1024 - 2 * 1024
  65 assert hb.get_block_size(nbd.SIZE_MINIMUM) == 1
  66 assert hb.get_block_size(nbd.SIZE_PREFERRED) == 64 * 1024
  67 assert hb.get_block_size(nbd.SIZE_MAXIMUM) == 4 * 1024 * 1024 * 1024 - 1
  68 assert hb.pread(1024 * 1024 + 1, 1) == b"\0" * (1024 * 1024 + 1)
  69
  70 ha.shutdown()
  71 hb.shutdown()
  72 '
  73 nbdkit -v -U - --filter=blocksize eval \
  74     open='echo $3' \
  75     get_size="echo $((2 * 1024 * 1024 - 1))" \
  76     block_size='case $2 in
  77       a) echo 1K 128K 512K ;;
  78       b) echo 2K 64K 800K ;;
  79       *) echo unknown export name >&2; exit 1 ;;
  80     esac' \
  81     pread='case $2.$3.$4 in
  82       a.$((1*1024)).0 | a.$((512*1024)).$((1*1024)) | \
  83       a.$((511*1024)).$((513*1024)) | a.$((1*1024)).$((1024*1024)) | \
  84       b.$((2*1024)).0 | b.$((800*1024)).$((2*1024)) | \
  85       b.$((222*1024)).$((802*1024)) | b.$((2*1024)).$((1024*1024)) )
  86         dd if=/dev/zero count=$3 iflag=count_bytes ;;
  87       *) echo EIO >&2; exit 1 ;;
  88     esac' \
  89     --run 'export unixsocket; nbdsh -c "$script"'