From: Dmitry V. Levin Date: Tue, 1 Aug 2017 20:59:48 +0000 (+0000) Subject: tests: add test_process_vm_readv and test_ptrace_peekdata to libtests X-Git-Tag: v4.19~189 X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=4f03dc3fc667ea221b96e7324235504190f22a12;p=strace tests: add test_process_vm_readv and test_ptrace_peekdata to libtests * tests/test_ucopy.c: New file. * tests/test_ucopy.h: Likewise. * tests/Makefile.am (libtests_a_SOURCES): Add them. --- diff --git a/tests/Makefile.am b/tests/Makefile.am index 5687a8d8..dd716519 100644 --- a/tests/Makefile.am +++ b/tests/Makefile.am @@ -65,6 +65,8 @@ libtests_a_SOURCES = \ tail_alloc.c \ test_netlink.h \ test_nlattr.h \ + test_ucopy.c \ + test_ucopy.h \ tests.h \ tprintf.c \ # end of libtests_a_SOURCES diff --git a/tests/test_ucopy.c b/tests/test_ucopy.c new file mode 100644 index 00000000..9ddffbca --- /dev/null +++ b/tests/test_ucopy.c @@ -0,0 +1,162 @@ +/* + * Test whether process_vm_readv and PTRACE_PEEKDATA work. + * + * Copyright (c) 2016-2017 Dmitry V. Levin + * 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 +#include +#include +#include +#include +#include +#include + +#include "test_ucopy.h" + +#ifndef HAVE_PROCESS_VM_READV + +# include +# include "scno.h" +static ssize_t +strace_process_vm_readv(pid_t pid, + const struct iovec *lvec, + unsigned long liovcnt, + const struct iovec *rvec, + unsigned long riovcnt, + unsigned long flags) +{ + return syscall(__NR_process_vm_readv, + (long) pid, lvec, liovcnt, rvec, riovcnt, flags); +} +# define process_vm_readv strace_process_vm_readv + +#endif /* !HAVE_PROCESS_VM_READV */ + +static bool +call_process_vm_readv(const int pid, long *const addr) +{ + long data = 0; + + const struct iovec local = { + .iov_base = &data, + .iov_len = sizeof(data) + }; + const struct iovec remote = { + .iov_base = addr, + .iov_len = sizeof(*addr) + }; + + return process_vm_readv(pid, &local, 1, &remote, 1, 0) == sizeof(data) + && data == 1; +} + +static bool +call_ptrace_peekdata(const int pid, long *const addr) +{ + return ptrace(PTRACE_PEEKDATA, pid, addr, 0) == 1; +} + +static bool +test_ucopy(bool (*fn)(int pid, long *addr)) +{ + static long data; + + data = 0; + bool rc = false; + int saved = 0; + + pid_t pid = fork(); + if (pid < 0) + perror_msg_and_fail("fork"); + + if (!pid) { + data = 1; + if (ptrace(PTRACE_TRACEME, 0, 0, 0)) + perror_msg_and_fail("PTRACE_TRACEME"); + raise(SIGSTOP); + _exit(0); + } + + for (;;) { + int status, tracee; + + errno = 0; + tracee = wait(&status); + if (tracee != pid) { + if (errno == EINTR) + continue; + saved = errno; + kill(pid, SIGKILL); + errno = saved; + perror_msg_and_fail("wait"); + } + if (WIFEXITED(status)) { + if (WEXITSTATUS(status) == 0) + break; + error_msg_and_fail("unexpected exit status %u", + WEXITSTATUS(status)); + } + if (WIFSIGNALED(status)) + error_msg_and_fail("unexpected signal %u", + WTERMSIG(status)); + if (!WIFSTOPPED(status) || WSTOPSIG(status) != SIGSTOP) { + kill(pid, SIGKILL); + error_msg_and_fail("unexpected wait status %x", + status); + } + + errno = 0; + rc = fn(pid, &data); + if (!rc) + saved = errno; + + if (ptrace(PTRACE_CONT, pid, 0, 0)) { + saved = errno; + kill(pid, SIGKILL); + errno = saved; + perror_msg_and_fail("PTRACE_CONT"); + } + } + + if (!rc) + errno = saved; + return rc; +} + +bool +test_process_vm_readv(void) +{ + return test_ucopy(call_process_vm_readv); +} + +bool +test_ptrace_peekdata(void) +{ + return test_ucopy(call_ptrace_peekdata); +} diff --git a/tests/test_ucopy.h b/tests/test_ucopy.h new file mode 100644 index 00000000..c2a70200 --- /dev/null +++ b/tests/test_ucopy.h @@ -0,0 +1,7 @@ +#include + +extern bool +test_process_vm_readv(void); + +extern bool +test_ptrace_peekdata(void);