1 /* 2 * Copyright (c) 2014-2016 Dmitry V. Levin <ldv (at) altlinux.org> 3 * Copyright (c) 2016-2017 The strace developers. 4 * All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 3. The name of the author may not be used to endorse or promote products 15 * derived from this software without specific prior written permission. 16 * 17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 20 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 */ 28 29 #include "tests.h" 30 31 #ifdef HAVE_PREADV 32 33 # include <fcntl.h> 34 # include <stdio.h> 35 # include <sys/uio.h> 36 # include <unistd.h> 37 38 # define LEN 8 39 40 static void 41 print_iov(const struct iovec *iov) 42 { 43 unsigned int i; 44 unsigned char *buf = iov->iov_base; 45 46 fputs("{iov_base=\"", stdout); 47 for (i = 0; i < iov->iov_len; ++i) 48 printf("\\%d", (int) buf[i]); 49 printf("\", iov_len=%u}", (unsigned) iov->iov_len); 50 } 51 52 static void 53 print_iovec(const struct iovec *iov, unsigned int cnt) 54 { 55 unsigned int i; 56 putchar('['); 57 for (i = 0; i < cnt; ++i) { 58 if (i) 59 fputs(", ", stdout); 60 print_iov(&iov[i]); 61 } 62 putchar(']'); 63 } 64 65 int 66 main(void) 67 { 68 const off_t offset = 0xdefaceddeadbeefLL; 69 char *buf = tail_alloc(LEN); 70 TAIL_ALLOC_OBJECT_CONST_PTR(struct iovec, iov); 71 iov->iov_base = buf; 72 iov->iov_len = LEN; 73 74 (void) close(0); 75 if (open("/dev/zero", O_RDONLY)) 76 perror_msg_and_fail("open"); 77 78 if (preadv(0, iov, 1, offset) != LEN) 79 perror_msg_and_fail("preadv"); 80 printf("preadv(0, "); 81 print_iovec(iov, 1); 82 printf(", 1, %lld) = %u\n", (long long) offset, LEN); 83 84 if (preadv(0, iov, 1, -1) != -1) 85 perror_msg_and_fail("preadv"); 86 printf("preadv(0, [{iov_base=%p, iov_len=%zu}], 1, -1) = " 87 "-1 EINVAL (%m)\n", iov->iov_base, iov->iov_len); 88 89 if (preadv(0, NULL, 1, -2) != -1) 90 perror_msg_and_fail("preadv"); 91 printf("preadv(0, NULL, 1, -2) = -1 EINVAL (%m)\n"); 92 93 if (preadv(0, iov, 0, -3) != -1) 94 perror_msg_and_fail("preadv"); 95 printf("preadv(0, [], 0, -3) = -1 EINVAL (%m)\n"); 96 97 static const char tmp[] = "preadv-tmpfile"; 98 int fd = open(tmp, O_RDWR | O_CREAT | O_TRUNC, 0600); 99 if (fd < 0) 100 perror_msg_and_fail("open"); 101 if (unlink(tmp)) 102 perror_msg_and_fail("unlink"); 103 104 static const char w[] = "0123456789abcde"; 105 if (write(fd, w, LENGTH_OF(w)) != LENGTH_OF(w)) 106 perror_msg_and_fail("write"); 107 108 static const char r0_c[] = "01234567"; 109 static const char r1_c[] = "89abcde"; 110 111 const unsigned int r_len = (LENGTH_OF(w) + 1) / 2; 112 void *r0 = tail_alloc(r_len); 113 const struct iovec r0_iov_[] = { 114 { 115 .iov_base = r0, 116 .iov_len = r_len 117 } 118 }; 119 const struct iovec *r_iov = tail_memdup(r0_iov_, sizeof(r0_iov_)); 120 121 long rc; 122 123 rc = preadv(fd, r_iov, ARRAY_SIZE(r0_iov_), 0); 124 if (rc != (int) r_len) 125 perror_msg_and_fail("preadv: expected %u, returned %ld", 126 r_len, rc); 127 printf("preadv(%d, [{iov_base=\"%s\", iov_len=%u}], %u, 0) = %u\n", 128 fd, r0_c, r_len, (unsigned int) ARRAY_SIZE(r0_iov_), r_len); 129 130 void *r1 = tail_alloc(r_len); 131 void *r2 = tail_alloc(LENGTH_OF(w)); 132 const struct iovec r1_iov_[] = { 133 { 134 .iov_base = r1, 135 .iov_len = r_len 136 }, 137 { 138 .iov_base = r2, 139 .iov_len = LENGTH_OF(w) 140 } 141 }; 142 r_iov = tail_memdup(r1_iov_, sizeof(r1_iov_)); 143 144 rc = preadv(fd, r_iov, ARRAY_SIZE(r1_iov_), r_len); 145 if (rc != (int) LENGTH_OF(w) - (int) r_len) 146 perror_msg_and_fail("preadv: expected %d, returned %ld", 147 (int) LENGTH_OF(w) - r_len, rc); 148 printf("preadv(%d, [{iov_base=\"%s\", iov_len=%u}" 149 ", {iov_base=\"\", iov_len=%u}], %u, %u) = %u\n", 150 fd, r1_c, r_len, LENGTH_OF(w), 151 (unsigned int) ARRAY_SIZE(r1_iov_), 152 r_len, LENGTH_OF(w) - r_len); 153 154 puts("+++ exited with 0 +++"); 155 return 0; 156 } 157 158 #else 159 160 SKIP_MAIN_UNDEFINED("HAVE_PREADV") 161 162 #endif 163
