tools/test/stress2/misc/rdwr.sh

   1 #!/bin/sh
   2
   3 #
   4 # Copyright (c) 2012 Peter Holm <pho@FreeBSD.org>
   5 # All rights reserved.
   6 #
   7 # Redistribution and use in source and binary forms, with or without
   8 # modification, are permitted provided that the following conditions
   9 # are met:
  10 # 1. Redistributions of source code must retain the above copyright
  11 #    notice, this list of conditions and the following disclaimer.
  12 # 2. 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 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  17 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19 # ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  20 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  22 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  23 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  24 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  25 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26 # SUCH DAMAGE.
  27 #
  28
  29 # Test with read/write length of INT_MAX (i386) or INT_MAX+1 (amd64)
  30
  31 . ../default.cfg
  32
  33 [ `id -u ` -ne 0 ] && echo "Must be root!" && exit 1
  34
  35 odir=`pwd`
  36 cd /tmp
  37 sed '1,/^EOF/d' < $odir/$0 > rdwr.c
  38 mycc -o rdwr -Wall rdwr.c || exit
  39 rm -f rdwr.c
  40
  41 oldclamp=`sysctl debug.devfs_iosize_max_clamp 2>/dev/null |
  42     awk '{print $NF}'`
  43 if [ `uname -m` = amd64 ]; then
  44         [ "$oldclamp" = "1" ] && sysctl debug.devfs_iosize_max_clamp=0
  45 fi
  46 for j in `jot 10`; do
  47         /tmp/rdwr || { echo FAIL; break; }
  48 done
  49 if [ `uname -m` = amd64 ]; then
  50         [ "$oldclamp" = "1" ] && sysctl debug.devfs_iosize_max_clamp=1
  51 fi
  52
  53 rm -f /tmp/rdwr
  54 exit
  55 EOF
  56 #include <sys/types.h>
  57 #include <sys/limits.h>
  58 #include <sys/mman.h>
  59 #include <sys/uio.h>
  60 #include <err.h>
  61 #include <fcntl.h>
  62 #include <unistd.h>
  63
  64 int
  65 main(int argc, char **argv)
  66 {
  67         int fd1, fd2;
  68         size_t len;
  69         void *p;
  70         struct iovec iov;
  71
  72         alarm(120);
  73         if ((fd1 = open("/dev/null", O_RDWR, 0)) == -1)
  74                 err(1, "open /dev/null");
  75
  76         if ((fd2 = open("/dev/zero", O_RDWR)) == -1)
  77                 err(1, "open /dev/zero");
  78
  79         if (sizeof(size_t) == sizeof(int32_t))
  80                 len = (size_t)INT_MAX;          /* i386  */
  81         else
  82                 len = (size_t)INT_MAX + 1;      /* amd64 */
  83
  84         if ((p = mmap(0, len, PROT_READ | PROT_WRITE, MAP_PRIVATE, fd2, 0)) ==
  85                         MAP_FAILED)
  86                 err(1, "mmap");
  87
  88         if (read(fd2, p, len) != len)
  89                 err(1, "read");
  90
  91         if (write(fd1, p, len) != len)
  92                 err(1, "write");
  93
  94         if (pread(fd2, p, len, 0) != len)
  95                 err(1, "pread");
  96
  97         if (pwrite(fd1, p, len, 0) != len)
  98                 err(1, "pwrite");
  99
 100         iov.iov_base = p;
 101         iov.iov_len = len;
 102         if (readv(fd2, &iov, 1) != len)
 103                 err(1, "readv");
 104
 105         if (writev(fd1, &iov, 1) != len)
 106                 err(1, "writev");
 107
 108         if (preadv(fd2, &iov, 1, 0) != len)
 109                 err(1, "preadv");
 110
 111         if (pwritev(fd1, &iov, 1, 0) != len)
 112                 err(1, "pwritev");
 113
 114         close(fd1);
 115         close(fd2);
 116
 117         return (0);
 118 }