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