userfaultfd: Add ioctl tests
* tests/ioctl_uffdio.c: New file. * tests/ioctl_uffdio.test: New test. * tests/.gitignore: Add ioctl_uffdio. * tests/Makefile.am (check_PROGRAMS): Likewise. (DECODER_TESTS): Add ioctl_uffdio.test.
This commit is contained in:
Dr. David Alan Gilbert
parent
5d1216a56b
commit
0efd888e22
1
tests/.gitignore
vendored
1
tests/.gitignore
vendored
@ -88,6 +88,7 @@ getuid32
|
||||
getxxid
|
||||
inet-cmsg
|
||||
ioctl
|
||||
ioctl_uffdio
|
||||
ioctl_v4l2
|
||||
ioperm
|
||||
iopl
|
||||
|
||||
@ -143,6 +143,7 @@ check_PROGRAMS = \
|
||||
getxxid \
|
||||
inet-cmsg \
|
||||
ioctl \
|
||||
ioctl_uffdio \
|
||||
ioctl_v4l2 \
|
||||
ioperm \
|
||||
iopl \
|
||||
@ -424,6 +425,7 @@ DECODER_TESTS = \
|
||||
getxxid.test \
|
||||
inet-cmsg.test \
|
||||
ioctl.test \
|
||||
ioctl_uffdio.test \
|
||||
ioctl_v4l2.test \
|
||||
ioperm.test \
|
||||
iopl.test \
|
||||
|
||||
218
tests/ioctl_uffdio.c
Normal file
218
tests/ioctl_uffdio.c
Normal file
@ -0,0 +1,218 @@
|
||||
/*
|
||||
* Copyright (c) 2015-2016 Dmitry V. Levin <ldv@altlinux.org>
|
||||
* Copyright (c) 2016 Red Hat, Inc.
|
||||
* All rights reserved.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions
|
||||
* are met:
|
||||
* 1. Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
* 2. Redistributions in binary form must reproduce the above copyright
|
||||
* notice, this list of conditions and the following disclaimer in the
|
||||
* documentation and/or other materials provided with the distribution.
|
||||
* 3. The name of the author may not be used to endorse or promote products
|
||||
* derived from this software without specific prior written permission.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
|
||||
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
||||
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
|
||||
* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
|
||||
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
|
||||
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
|
||||
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#include "tests.h"
|
||||
#include <sys/syscall.h>
|
||||
|
||||
#if defined __NR_userfaultfd && defined HAVE_LINUX_USERFAULTFD_H
|
||||
|
||||
# include <fcntl.h>
|
||||
# include <inttypes.h>
|
||||
# include <stdint.h>
|
||||
# include <stdio.h>
|
||||
# include <string.h>
|
||||
# include <unistd.h>
|
||||
|
||||
# include <sys/ioctl.h>
|
||||
# include <sys/mman.h>
|
||||
# include <linux/ioctl.h>
|
||||
# include <linux/userfaultfd.h>
|
||||
|
||||
int
|
||||
main(void)
|
||||
{
|
||||
int rc;
|
||||
int fd = syscall(__NR_userfaultfd, O_NONBLOCK);
|
||||
size_t pagesize = getpagesize();
|
||||
|
||||
if (fd < 0)
|
||||
perror_msg_and_skip("userfaultfd");
|
||||
|
||||
/* ---- API ---- */
|
||||
(void) tail_alloc(1);
|
||||
struct uffdio_api *api_struct = tail_alloc(sizeof(*api_struct));
|
||||
|
||||
/* With a bad fd */
|
||||
memset(api_struct, 0, sizeof(*api_struct));
|
||||
rc = ioctl(-1, UFFDIO_API, api_struct);
|
||||
printf("ioctl(-1, UFFDIO_API, {api=0, features=0}) = %d %s (%m)\n",
|
||||
rc, errno2name());
|
||||
/* With a bad pointer */
|
||||
rc = ioctl(fd, UFFDIO_API, NULL);
|
||||
printf("ioctl(%d, UFFDIO_API, NULL) = %d %s (%m)\n",
|
||||
fd, rc, errno2name());
|
||||
/* Normal call */
|
||||
api_struct->api = UFFD_API;
|
||||
api_struct->features = 0;
|
||||
rc = ioctl(fd, UFFDIO_API, api_struct);
|
||||
printf("ioctl(%d, UFFDIO_API, {api=0xaa, features=0, "
|
||||
"features.out=%#" PRIx64 ", " "ioctls=1<<_UFFDIO_REGISTER|"
|
||||
"1<<_UFFDIO_UNREGISTER|1<<_UFFDIO_API",
|
||||
fd, (uint64_t)api_struct->features);
|
||||
api_struct->ioctls &= ~(1ull<<_UFFDIO_REGISTER|
|
||||
1ull<<_UFFDIO_UNREGISTER|
|
||||
1ull<<_UFFDIO_API);
|
||||
if (api_struct->ioctls)
|
||||
printf("|%#" PRIx64, (uint64_t)api_struct->ioctls);
|
||||
printf("}) = %d\n", rc);
|
||||
|
||||
/* For the rest of the tests we need some anonymous memory */
|
||||
void *area1 = mmap(NULL, pagesize, PROT_READ|PROT_WRITE,
|
||||
MAP_PRIVATE|MAP_ANONYMOUS,
|
||||
-1, 0);
|
||||
if (area1 == MAP_FAILED)
|
||||
perror_msg_and_fail("mmap area1");
|
||||
void *area2 = mmap(NULL, pagesize, PROT_READ|PROT_WRITE,
|
||||
MAP_PRIVATE|MAP_ANONYMOUS,
|
||||
-1, 0);
|
||||
if (area2 == MAP_FAILED)
|
||||
perror_msg_and_fail("mmap area2");
|
||||
madvise(area2, pagesize, MADV_DONTNEED);
|
||||
*(char *)area1 = 42;
|
||||
|
||||
/* ---- REGISTER ---- */
|
||||
(void) tail_alloc(1);
|
||||
struct uffdio_register *register_struct =
|
||||
tail_alloc(sizeof(*register_struct));
|
||||
memset(register_struct, 0, sizeof(*register_struct));
|
||||
|
||||
rc = ioctl(-1, UFFDIO_REGISTER, register_struct);
|
||||
printf("ioctl(-1, UFFDIO_REGISTER, {range={start=0, len=0}, "
|
||||
"mode=0}) = %d %s (%m)\n", rc, errno2name());
|
||||
|
||||
rc = ioctl(fd, UFFDIO_REGISTER, NULL);
|
||||
printf("ioctl(%d, UFFDIO_REGISTER, NULL) = %d %s (%m)\n",
|
||||
fd, rc, errno2name());
|
||||
|
||||
register_struct->range.start = (uint64_t)(uintptr_t)area2;
|
||||
register_struct->range.len = pagesize;
|
||||
register_struct->mode = UFFDIO_REGISTER_MODE_MISSING;
|
||||
rc = ioctl(fd, UFFDIO_REGISTER, register_struct);
|
||||
printf("ioctl(%d, UFFDIO_REGISTER, {range={start=%p, len=%#zx}, "
|
||||
"mode=UFFDIO_REGISTER_MODE_MISSING, ioctls="
|
||||
"1<<_UFFDIO_WAKE|1<<_UFFDIO_COPY|1<<_UFFDIO_ZEROPAGE",
|
||||
fd, area2, pagesize);
|
||||
register_struct->ioctls &= ~(1ull<<_UFFDIO_WAKE|
|
||||
1ull<<_UFFDIO_COPY|
|
||||
1ull<<_UFFDIO_ZEROPAGE);
|
||||
if (register_struct->ioctls)
|
||||
printf("|%#" PRIx64, (uint64_t)register_struct->ioctls);
|
||||
printf("}) = %d\n", rc);
|
||||
|
||||
/* With area2 registered we can now do the atomic copies onto it
|
||||
* but be careful not to access it in any other way otherwise
|
||||
* userfaultfd will cause us to stall.
|
||||
*/
|
||||
/* ---- COPY ---- */
|
||||
(void) tail_alloc(1);
|
||||
struct uffdio_copy *copy_struct = tail_alloc(sizeof(*copy_struct));
|
||||
|
||||
memset(copy_struct, 0, sizeof(*copy_struct));
|
||||
rc = ioctl(-1, UFFDIO_COPY, copy_struct);
|
||||
printf("ioctl(-1, UFFDIO_COPY, {dst=0, src=0, len=0, mode=0"
|
||||
"}) = %d %s (%m)\n", rc, errno2name());
|
||||
|
||||
rc = ioctl(fd, UFFDIO_COPY, NULL);
|
||||
printf("ioctl(%d, UFFDIO_COPY, NULL) = %d %s (%m)\n",
|
||||
fd, rc, errno2name());
|
||||
|
||||
copy_struct->dst = (uint64_t)(uintptr_t)area2;
|
||||
copy_struct->src = (uint64_t)(uintptr_t)area1;
|
||||
copy_struct->len = pagesize;
|
||||
copy_struct->mode = UFFDIO_COPY_MODE_DONTWAKE;
|
||||
rc = ioctl(fd, UFFDIO_COPY, copy_struct);
|
||||
printf("ioctl(%d, UFFDIO_COPY, {dst=%p, src=%p, len=%#zx,"
|
||||
" mode=UFFDIO_COPY_MODE_DONTWAKE, copy=%#zx}) = %d\n",
|
||||
fd, area2, area1, pagesize, pagesize, rc);
|
||||
|
||||
/* ---- ZEROPAGE ---- */
|
||||
(void) tail_alloc(1);
|
||||
struct uffdio_zeropage *zero_struct = tail_alloc(sizeof(*zero_struct));
|
||||
madvise(area2, pagesize, MADV_DONTNEED);
|
||||
|
||||
memset(zero_struct, 0, sizeof(*zero_struct));
|
||||
rc = ioctl(-1, UFFDIO_ZEROPAGE, zero_struct);
|
||||
printf("ioctl(-1, UFFDIO_ZEROPAGE, {range={start=0, len=0}, mode=0"
|
||||
"}) = %d %s (%m)\n", rc, errno2name());
|
||||
|
||||
rc = ioctl(fd, UFFDIO_ZEROPAGE, NULL);
|
||||
printf("ioctl(%d, UFFDIO_ZEROPAGE, NULL) = %d %s (%m)\n",
|
||||
fd, rc, errno2name());
|
||||
|
||||
zero_struct->range.start = (uint64_t)(uintptr_t)area2;
|
||||
zero_struct->range.len = pagesize;
|
||||
zero_struct->mode = UFFDIO_ZEROPAGE_MODE_DONTWAKE;
|
||||
rc = ioctl(fd, UFFDIO_ZEROPAGE, zero_struct);
|
||||
printf("ioctl(%d, UFFDIO_ZEROPAGE, {range={start=%p, len=%#zx},"
|
||||
" mode=UFFDIO_ZEROPAGE_MODE_DONTWAKE, zeropage=%#zx}) = %d\n",
|
||||
fd, area2, pagesize, pagesize, rc);
|
||||
|
||||
/* ---- WAKE ---- */
|
||||
(void) tail_alloc(1);
|
||||
struct uffdio_range *range_struct = tail_alloc(sizeof(*range_struct));
|
||||
memset(range_struct, 0, sizeof(*range_struct));
|
||||
|
||||
rc = ioctl(-1, UFFDIO_WAKE, range_struct);
|
||||
printf("ioctl(-1, UFFDIO_WAKE, {start=0, len=0}) = %d %s (%m)\n",
|
||||
rc, errno2name());
|
||||
|
||||
rc = ioctl(fd, UFFDIO_WAKE, NULL);
|
||||
printf("ioctl(%d, UFFDIO_WAKE, NULL) = %d %s (%m)\n",
|
||||
fd, rc, errno2name());
|
||||
|
||||
range_struct->start = (uint64_t)(uintptr_t)area2;
|
||||
range_struct->len = pagesize;
|
||||
rc = ioctl(fd, UFFDIO_WAKE, range_struct);
|
||||
printf("ioctl(%d, UFFDIO_WAKE, {start=%p, len=%#zx}) = %d\n",
|
||||
fd, area2, pagesize, rc);
|
||||
|
||||
/* ---- UNREGISTER ---- */
|
||||
memset(range_struct, 0, sizeof(*range_struct));
|
||||
|
||||
rc = ioctl(-1, UFFDIO_UNREGISTER, range_struct);
|
||||
printf("ioctl(-1, UFFDIO_UNREGISTER, {start=0, len=0}) = %d %s (%m)\n",
|
||||
rc, errno2name());
|
||||
|
||||
rc = ioctl(fd, UFFDIO_UNREGISTER, NULL);
|
||||
printf("ioctl(%d, UFFDIO_UNREGISTER, NULL) = %d %s (%m)\n",
|
||||
fd, rc, errno2name());
|
||||
|
||||
range_struct->start = (uint64_t)(uintptr_t)area2;
|
||||
range_struct->len = pagesize;
|
||||
rc = ioctl(fd, UFFDIO_UNREGISTER, range_struct);
|
||||
printf("ioctl(%d, UFFDIO_UNREGISTER, {start=%p, len=%#zx}) = %d\n",
|
||||
fd, area2, pagesize, rc);
|
||||
puts("+++ exited with 0 +++");
|
||||
return 0;
|
||||
}
|
||||
|
||||
#else
|
||||
|
||||
SKIP_MAIN_UNDEFINED("__NR_userfaultfd && HAVE_LINUX_USERFAULTFD_H")
|
||||
|
||||
#endif
|
||||
5
tests/ioctl_uffdio.test
Executable file
5
tests/ioctl_uffdio.test
Executable file
@ -0,0 +1,5 @@
|
||||
#!/bin/sh
|
||||
|
||||
# Check userfaultfd ioctl decoding.
|
||||
|
||||
. "${srcdir=.}/ioctl.test"
|
||||
Loading…
x
Reference in New Issue
Block a user