misc/tst-preadvwritev-common.c

   1 /* Common definitions for preadv and pwritev.
   2    Copyright (C) 2016-2017 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4
   5    The GNU C Library is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU Lesser General Public
   7    License as published by the Free Software Foundation; either
   8    version 2.1 of the License, or (at your option) any later version.
   9
  10    The GNU C Library is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13    Lesser General Public License for more details.
  14
  15    You should have received a copy of the GNU Lesser General Public
  16    License along with the GNU C Library; if not, see
  17    <http://www.gnu.org/licenses/>.  */
  18
  19 #include <stdio.h>
  20 #include <stdint.h>
  21 #include <errno.h>
  22 #include <string.h>
  23 #include <sys/uio.h>
  24 #include <sys/stat.h>
  25
  26 #include <support/check.h>
  27 #include <support/temp_file.h>
  28
  29 static char *temp_filename;
  30 static int temp_fd;
  31
  32 static int do_test (void);
  33
  34 static void
  35 do_prepare (int argc, char **argv)
  36 {
  37   temp_fd = create_temp_file ("tst-preadvwritev.", &temp_filename);
  38   if (temp_fd == -1)
  39     FAIL_EXIT1 ("cannot create temporary file");
  40 }
  41 #define PREPARE do_prepare
  42
  43 #ifndef PREADV
  44 # define PREADV(__fd, __iov, __iovcnt, __offset) \
  45   preadv (__fd, __iov, __iovcnt, __offset)
  46 #endif
  47
  48 #ifndef PWRITEV
  49 # define PWRITEV(__fd, __iov, __iovcnt, __offset) \
  50   pwritev (__fd, __iov, __iovcnt, __offset)
  51 #endif
  52
  53 static int
  54 do_test_with_offset (off_t offset)
  55 {
  56   struct iovec iov[2];
  57   ssize_t ret;
  58
  59   char buf1[32];
  60   char buf2[64];
  61
  62   memset (buf1, 0xf0, sizeof buf1);
  63   memset (buf2, 0x0f, sizeof buf2);
  64
  65   /* Write two buffer with 32 and 64 bytes respectively.  */
  66   memset (iov, 0, sizeof iov);
  67   iov[0].iov_base = buf1;
  68   iov[0].iov_len = sizeof buf1;
  69   iov[1].iov_base = buf2;
  70   iov[1].iov_len = sizeof buf2;
  71
  72   ret = PWRITEV (temp_fd, iov, 2, offset);
  73   if (ret == -1)
  74     FAIL_RET ("first pwritev returned -1");
  75   if (ret != (sizeof buf1 + sizeof buf2))
  76     FAIL_RET ("first pwritev returned an unexpected value");
  77
  78   ret = PWRITEV (temp_fd, iov, 2, sizeof buf1 + sizeof buf2 + offset);
  79   if (ret == -1)
  80     FAIL_RET ("second pwritev returned -1");
  81   if (ret != (sizeof buf1 + sizeof buf2))
  82     FAIL_RET ("second pwritev returned an unexpected value");
  83
  84   char buf3[32];
  85   char buf4[64];
  86
  87   memset (buf3, 0x0f, sizeof buf3);
  88   memset (buf4, 0xf0, sizeof buf4);
  89
  90   iov[0].iov_base = buf3;
  91   iov[0].iov_len = sizeof buf3;
  92   iov[1].iov_base = buf4;
  93   iov[1].iov_len = sizeof buf4;
  94
  95   /* Now read two buffer with 32 and 64 bytes respectively.  */
  96   ret = PREADV (temp_fd, iov, 2, offset);
  97   if (ret == -1)
  98     FAIL_RET ("first preadv returned -1");
  99   if (ret != (sizeof buf3 + sizeof buf4))
 100     FAIL_RET ("first preadv returned an unexpected value");
 101
 102   if (memcmp (buf1, buf3, sizeof buf1) != 0)
 103     FAIL_RET ("first buffer from first preadv different than expected");
 104   if (memcmp (buf2, buf4, sizeof buf2) != 0)
 105     FAIL_RET ("second buffer from first preadv different than expected");
 106
 107   ret = PREADV (temp_fd, iov, 2, sizeof buf3 + sizeof buf4 + offset);
 108   if (ret == -1)
 109     FAIL_RET ("second preadv returned -1");
 110   if (ret != (sizeof buf3 + sizeof buf4))
 111     FAIL_RET ("second preadv returned an unexpected value");
 112
 113   /* And compare the buffers read and written to check if there are equal.  */
 114   if (memcmp (buf1, buf3, sizeof buf1) != 0)
 115     FAIL_RET ("first buffer from second preadv different than expected");
 116   if (memcmp (buf2, buf4, sizeof buf2) != 0)
 117     FAIL_RET ("second buffer from second preadv different than expected");
 118
 119   return 0;
 120 }
 121
 122 #include <support/test-driver.c>