1 /*
2 * Copyright (c) 2015-2016 Dmitry V. Levin <ldv@altlinux.org>
3 * Copyright (c) 2016 Red Hat, Inc.
4 * Copyright (c) 2016-2017 The strace developers.
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
33 #if defined __NR_userfaultfd && defined HAVE_LINUX_USERFAULTFD_H
34
35 # include <fcntl.h>
36 # include <inttypes.h>
37 # include <stdint.h>
38 # include <stdio.h>
39 # include <string.h>
40 # include <unistd.h>
41
42 # include <sys/ioctl.h>
43 # include <sys/mman.h>
44 # include <linux/ioctl.h>
45 # include <linux/userfaultfd.h>
46
47 #include "xlat.h"
48 #include "xlat/uffd_api_features.h"
49
50 int
main(void)51 main(void)
52 {
53 int rc;
54 int fd = syscall(__NR_userfaultfd, O_NONBLOCK);
55 size_t pagesize = getpagesize();
56
57 if (fd < 0)
58 perror_msg_and_skip("userfaultfd");
59
60 /* ---- API ---- */
61 TAIL_ALLOC_OBJECT_CONST_PTR(struct uffdio_api, api_struct);
62
63 /* With a bad fd */
64 memset(api_struct, 0, sizeof(*api_struct));
65 rc = ioctl(-1, UFFDIO_API, api_struct);
66 printf("ioctl(-1, UFFDIO_API, {api=0, features=0}) = %d %s (%m)\n",
67 rc, errno2name());
68 /* With a bad pointer */
69 rc = ioctl(fd, UFFDIO_API, NULL);
70 printf("ioctl(%d, UFFDIO_API, NULL) = %d %s (%m)\n",
71 fd, rc, errno2name());
72 /* Normal call */
73 api_struct->api = UFFD_API;
74 api_struct->features = 0;
75 rc = ioctl(fd, UFFDIO_API, api_struct);
76 printf("ioctl(%d, UFFDIO_API, {api=0xaa, features=0", fd);
77 if (api_struct->features) {
78 printf(" => features=");
79 printflags(uffd_api_features, api_struct->features,
80 "UFFD_FEATURE_???");
81 }
82 printf(", ioctls=1<<_UFFDIO_REGISTER|"
83 "1<<_UFFDIO_UNREGISTER|1<<_UFFDIO_API");
84 api_struct->ioctls &= ~(1ull<<_UFFDIO_REGISTER|
85 1ull<<_UFFDIO_UNREGISTER|
86 1ull<<_UFFDIO_API);
87 if (api_struct->ioctls)
88 printf("|%#" PRIx64, (uint64_t)api_struct->ioctls);
89 printf("}) = %d\n", rc);
90
91 /* For the rest of the tests we need some anonymous memory */
92 void *area1 = mmap(NULL, pagesize, PROT_READ|PROT_WRITE,
93 MAP_PRIVATE|MAP_ANONYMOUS,
94 -1, 0);
95 if (area1 == MAP_FAILED)
96 perror_msg_and_fail("mmap area1");
97 void *area2 = mmap(NULL, pagesize, PROT_READ|PROT_WRITE,
98 MAP_PRIVATE|MAP_ANONYMOUS,
99 -1, 0);
100 if (area2 == MAP_FAILED)
101 perror_msg_and_fail("mmap area2");
102 madvise(area2, pagesize, MADV_DONTNEED);
103 *(char *)area1 = 42;
104
105 /* ---- REGISTER ---- */
106 struct uffdio_register *register_struct =
107 tail_alloc(sizeof(*register_struct));
108 memset(register_struct, 0, sizeof(*register_struct));
109
110 rc = ioctl(-1, UFFDIO_REGISTER, register_struct);
111 printf("ioctl(-1, UFFDIO_REGISTER, {range={start=0, len=0}, "
112 "mode=0}) = %d %s (%m)\n", rc, errno2name());
113
114 rc = ioctl(fd, UFFDIO_REGISTER, NULL);
115 printf("ioctl(%d, UFFDIO_REGISTER, NULL) = %d %s (%m)\n",
116 fd, rc, errno2name());
117
118 register_struct->range.start = (uint64_t)(uintptr_t)area2;
119 register_struct->range.len = pagesize;
120 register_struct->mode = UFFDIO_REGISTER_MODE_MISSING;
121 rc = ioctl(fd, UFFDIO_REGISTER, register_struct);
122 printf("ioctl(%d, UFFDIO_REGISTER, {range={start=%p, len=%#zx}, "
123 "mode=UFFDIO_REGISTER_MODE_MISSING, ioctls="
124 "1<<_UFFDIO_WAKE|1<<_UFFDIO_COPY|1<<_UFFDIO_ZEROPAGE",
125 fd, area2, pagesize);
126 register_struct->ioctls &= ~(1ull<<_UFFDIO_WAKE|
127 1ull<<_UFFDIO_COPY|
128 1ull<<_UFFDIO_ZEROPAGE);
129 if (register_struct->ioctls)
130 printf("|%#" PRIx64, (uint64_t)register_struct->ioctls);
131 printf("}) = %d\n", rc);
132
133 /* With area2 registered we can now do the atomic copies onto it
134 * but be careful not to access it in any other way otherwise
135 * userfaultfd will cause us to stall.
136 */
137 /* ---- COPY ---- */
138 TAIL_ALLOC_OBJECT_CONST_PTR(struct uffdio_copy, copy_struct);
139
140 memset(copy_struct, 0, sizeof(*copy_struct));
141 rc = ioctl(-1, UFFDIO_COPY, copy_struct);
142 printf("ioctl(-1, UFFDIO_COPY, {dst=0, src=0, len=0, mode=0"
143 "}) = %d %s (%m)\n", rc, errno2name());
144
145 rc = ioctl(fd, UFFDIO_COPY, NULL);
146 printf("ioctl(%d, UFFDIO_COPY, NULL) = %d %s (%m)\n",
147 fd, rc, errno2name());
148
149 copy_struct->dst = (uint64_t)(uintptr_t)area2;
150 copy_struct->src = (uint64_t)(uintptr_t)area1;
151 copy_struct->len = pagesize;
152 copy_struct->mode = UFFDIO_COPY_MODE_DONTWAKE;
153 rc = ioctl(fd, UFFDIO_COPY, copy_struct);
154 printf("ioctl(%d, UFFDIO_COPY, {dst=%p, src=%p, len=%#zx,"
155 " mode=UFFDIO_COPY_MODE_DONTWAKE, copy=%#zx}) = %d\n",
156 fd, area2, area1, pagesize, pagesize, rc);
157
158 /* ---- ZEROPAGE ---- */
159 TAIL_ALLOC_OBJECT_CONST_PTR(struct uffdio_zeropage, zero_struct);
160 madvise(area2, pagesize, MADV_DONTNEED);
161
162 memset(zero_struct, 0, sizeof(*zero_struct));
163 rc = ioctl(-1, UFFDIO_ZEROPAGE, zero_struct);
164 printf("ioctl(-1, UFFDIO_ZEROPAGE, {range={start=0, len=0}, mode=0"
165 "}) = %d %s (%m)\n", rc, errno2name());
166
167 rc = ioctl(fd, UFFDIO_ZEROPAGE, NULL);
168 printf("ioctl(%d, UFFDIO_ZEROPAGE, NULL) = %d %s (%m)\n",
169 fd, rc, errno2name());
170
171 zero_struct->range.start = (uint64_t)(uintptr_t)area2;
172 zero_struct->range.len = pagesize;
173 zero_struct->mode = UFFDIO_ZEROPAGE_MODE_DONTWAKE;
174 rc = ioctl(fd, UFFDIO_ZEROPAGE, zero_struct);
175 printf("ioctl(%d, UFFDIO_ZEROPAGE, {range={start=%p, len=%#zx},"
176 " mode=UFFDIO_ZEROPAGE_MODE_DONTWAKE, zeropage=%#zx}) = %d\n",
177 fd, area2, pagesize, pagesize, rc);
178
179 /* ---- WAKE ---- */
180 TAIL_ALLOC_OBJECT_CONST_PTR(struct uffdio_range, range_struct);
181 memset(range_struct, 0, sizeof(*range_struct));
182
183 rc = ioctl(-1, UFFDIO_WAKE, range_struct);
184 printf("ioctl(-1, UFFDIO_WAKE, {start=0, len=0}) = %d %s (%m)\n",
185 rc, errno2name());
186
187 rc = ioctl(fd, UFFDIO_WAKE, NULL);
188 printf("ioctl(%d, UFFDIO_WAKE, NULL) = %d %s (%m)\n",
189 fd, rc, errno2name());
190
191 range_struct->start = (uint64_t)(uintptr_t)area2;
192 range_struct->len = pagesize;
193 rc = ioctl(fd, UFFDIO_WAKE, range_struct);
194 printf("ioctl(%d, UFFDIO_WAKE, {start=%p, len=%#zx}) = %d\n",
195 fd, area2, pagesize, rc);
196
197 /* ---- UNREGISTER ---- */
198 memset(range_struct, 0, sizeof(*range_struct));
199
200 rc = ioctl(-1, UFFDIO_UNREGISTER, range_struct);
201 printf("ioctl(-1, UFFDIO_UNREGISTER, {start=0, len=0}) = %d %s (%m)\n",
202 rc, errno2name());
203
204 rc = ioctl(fd, UFFDIO_UNREGISTER, NULL);
205 printf("ioctl(%d, UFFDIO_UNREGISTER, NULL) = %d %s (%m)\n",
206 fd, rc, errno2name());
207
208 range_struct->start = (uint64_t)(uintptr_t)area2;
209 range_struct->len = pagesize;
210 rc = ioctl(fd, UFFDIO_UNREGISTER, range_struct);
211 printf("ioctl(%d, UFFDIO_UNREGISTER, {start=%p, len=%#zx}) = %d\n",
212 fd, area2, pagesize, rc);
213 puts("+++ exited with 0 +++");
214 return 0;
215 }
216
217 #else
218
219 SKIP_MAIN_UNDEFINED("__NR_userfaultfd && HAVE_LINUX_USERFAULTFD_H")
220
221 #endif
222