1 /*
2 * Check decoding of preadv2 and pwritev2 syscalls.
3 *
4 * Copyright (c) 2016 Dmitry V. Levin <ldv@altlinux.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 * 3. The name of the author may not be used to endorse or promote products
16 * derived from this software without specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
19 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
20 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
21 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
22 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
23 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 */
29
30 #include "tests.h"
31 #include <asm/unistd.h>
32 #include "scno.h"
33
34 #if defined __NR_preadv2 && defined __NR_pwritev2
35
36 # include <errno.h>
37 # include <fcntl.h>
38 # include <stdio.h>
39 # include <sys/uio.h>
40 # include <unistd.h>
41
42 static int
pr(const int fd,const struct iovec * const vec,const unsigned long vlen,const unsigned long pos)43 pr(const int fd, const struct iovec *const vec,
44 const unsigned long vlen, const unsigned long pos)
45 {
46 return syscall(__NR_preadv2, fd, vec, vlen, pos, 0L, 0L);
47 }
48
49 static int
pw(const int fd,const struct iovec * const vec,const unsigned long vlen,const unsigned long pos)50 pw(const int fd, const struct iovec *const vec,
51 const unsigned long vlen, const unsigned long pos)
52 {
53 return syscall(__NR_pwritev2, fd, vec, vlen, pos, 0L, 0L);
54 }
55
56 static void
dumpio(void)57 dumpio(void)
58 {
59 static char tmp[] = "preadv2-pwritev2-tmpfile";
60 if (open(tmp, O_CREAT|O_RDONLY|O_TRUNC, 0600) != 0)
61 perror_msg_and_fail("creat: %s", tmp);
62 if (open(tmp, O_WRONLY) != 1)
63 perror_msg_and_fail("open: %s", tmp);
64 if (unlink(tmp))
65 perror_msg_and_fail("unlink: %s", tmp);
66
67 static const char w0_c[] = "012";
68 const char *w0_d = hexdump_strdup(w0_c);
69 void *w0 = tail_memdup(w0_c, LENGTH_OF(w0_c));
70
71 static const char w1_c[] = "34567";
72 const char *w1_d = hexdump_strdup(w1_c);
73 void *w1 = tail_memdup(w1_c, LENGTH_OF(w1_c));
74
75 static const char w2_c[] = "89abcde";
76 const char *w2_d = hexdump_strdup(w2_c);
77 void *w2 = tail_memdup(w2_c, LENGTH_OF(w2_c));
78
79 long rc;
80
81 static const char r0_c[] = "01234567";
82 const char *r0_d = hexdump_strdup(r0_c);
83 static const char r1_c[] = "89abcde";
84 const char *r1_d = hexdump_strdup(r1_c);
85
86 const struct iovec w_iov_[] = {
87 {
88 .iov_base = w0,
89 .iov_len = LENGTH_OF(w0_c)
90 }, {
91 .iov_base = w1,
92 .iov_len = LENGTH_OF(w1_c)
93 }, {
94 .iov_base = w2,
95 .iov_len = LENGTH_OF(w2_c)
96 }
97 };
98 const struct iovec *w_iov = tail_memdup(w_iov_, sizeof(w_iov_));
99
100 rc = pw(1, w_iov, 0, 0);
101 if (rc)
102 perror_msg_and_fail("pwritev2: expected 0, returned %ld", rc);
103 tprintf("pwritev2(1, [], 0, 0, 0) = 0\n");
104
105 rc = pw(1, w_iov + ARRAY_SIZE(w_iov_) - 1, 2, 0);
106 tprintf("pwritev2(1, [{iov_base=\"%s\", iov_len=%u}, %p], 2, 0, 0)"
107 " = %ld %s (%m)\n",
108 w2_c, LENGTH_OF(w2_c), w_iov + ARRAY_SIZE(w_iov_),
109 rc, errno2name());
110
111 const unsigned int w_len =
112 LENGTH_OF(w0_c) + LENGTH_OF(w1_c) + LENGTH_OF(w2_c);
113
114 rc = pw(1, w_iov, ARRAY_SIZE(w_iov_), 0);
115 if (rc != (int) w_len)
116 perror_msg_and_fail("pwritev2: expected %u, returned %ld",
117 w_len, rc);
118 close(1);
119 tprintf("pwritev2(1, [{iov_base=\"%s\", iov_len=%u}"
120 ", {iov_base=\"%s\", iov_len=%u}"
121 ", {iov_base=\"%s\", iov_len=%u}], %u, 0, 0) = %u\n"
122 " * %u bytes in buffer 0\n"
123 " | 00000 %-49s %-16s |\n"
124 " * %u bytes in buffer 1\n"
125 " | 00000 %-49s %-16s |\n"
126 " * %u bytes in buffer 2\n"
127 " | 00000 %-49s %-16s |\n",
128 w0_c, LENGTH_OF(w0_c), w1_c, LENGTH_OF(w1_c),
129 w2_c, LENGTH_OF(w2_c), ARRAY_SIZE(w_iov_), w_len,
130 LENGTH_OF(w0_c), w0_d, w0_c,
131 LENGTH_OF(w1_c), w1_d, w1_c, LENGTH_OF(w2_c), w2_d, w2_c);
132
133 const unsigned int r_len = (w_len + 1) / 2;
134 void *r0 = tail_alloc(r_len);
135 const struct iovec r0_iov_[] = {
136 {
137 .iov_base = r0,
138 .iov_len = r_len
139 }
140 };
141 const struct iovec *r_iov = tail_memdup(r0_iov_, sizeof(r0_iov_));
142
143 rc = pr(0, r_iov, ARRAY_SIZE(r0_iov_), 0);
144 if (rc != (int) r_len)
145 perror_msg_and_fail("preadv2: expected %u, returned %ld",
146 r_len, rc);
147 tprintf("preadv2(0, [{iov_base=\"%s\", iov_len=%u}], %u, 0, 0) = %u\n"
148 " * %u bytes in buffer 0\n"
149 " | 00000 %-49s %-16s |\n",
150 r0_c, r_len, ARRAY_SIZE(r0_iov_), r_len, r_len, r0_d, r0_c);
151
152 void *r1 = tail_alloc(r_len);
153 void *r2 = tail_alloc(w_len);
154 const struct iovec r1_iov_[] = {
155 {
156 .iov_base = r1,
157 .iov_len = r_len
158 },
159 {
160 .iov_base = r2,
161 .iov_len = w_len
162 }
163 };
164 r_iov = tail_memdup(r1_iov_, sizeof(r1_iov_));
165
166 rc = pr(0, r_iov, ARRAY_SIZE(r1_iov_), r_len);
167 if (rc != (int) w_len - (int) r_len)
168 perror_msg_and_fail("preadv2: expected %d, returned %ld",
169 (int) w_len - r_len, rc);
170 tprintf("preadv2(0, [{iov_base=\"%s\", iov_len=%u}"
171 ", {iov_base=\"\", iov_len=%u}], %u, %u, 0) = %u\n"
172 " * %u bytes in buffer 0\n"
173 " | 00000 %-49s %-16s |\n",
174 r1_c, r_len, w_len, ARRAY_SIZE(r1_iov_),
175 r_len, w_len - r_len,
176 w_len - r_len, r1_d, r1_c);
177 close(0);
178 }
179
180 int
main(void)181 main(void)
182 {
183 const kernel_ulong_t vlen = (kernel_ulong_t) 0xfac1fed2dad3bef4ULL;
184 const unsigned long long pos = 0xfac5fed6dad7bef8;
185 const kernel_ulong_t pos_l = (kernel_ulong_t) pos;
186 const kernel_ulong_t pos_h =
187 (sizeof(kernel_ulong_t) == sizeof(long long)) ?
188 (kernel_ulong_t) 0xbadc0deddeadbeefULL : 0xfac5fed6UL;
189 int test_dumpio = 1;
190
191 tprintf("%s", "");
192
193 syscall(__NR_preadv2, -1, NULL, vlen, pos_l, pos_h, 1);
194 if (ENOSYS == errno)
195 test_dumpio = 0;
196 tprintf("preadv2(-1, NULL, %lu, %lld, RWF_HIPRI) = -1 %s (%m)\n",
197 (unsigned long) vlen, pos, errno2name());
198
199 syscall(__NR_pwritev2, -1, NULL, vlen, pos_l, pos_h, 1);
200 if (ENOSYS == errno)
201 test_dumpio = 0;
202 tprintf("pwritev2(-1, NULL, %lu, %lld, RWF_HIPRI) = -1 %s (%m)\n",
203 (unsigned long) vlen, pos, errno2name());
204
205 if (test_dumpio)
206 dumpio();
207
208 tprintf("%s\n", "+++ exited with 0 +++");
209 return 0;
210 }
211
212 #else
213
214 SKIP_MAIN_UNDEFINED("__NR_preadv2 && __NR_pwritev2")
215
216 #endif
217