[src/trunk]: src/tests/kernel/arch/amd64 Add new tests dbregs_dr[0123]_trap_c...

[kamil <kamil%NetBSD.org@localhost>]

details:   https://anonhg.NetBSD.org/src/rev/083ec7e3c0dc
branches:  trunk
changeset: 821862:083ec7e3c0dc
user:      kamil <kamil%NetBSD.org@localhost>
date:      Mon Feb 20 02:56:03 2017 +0000

description:
Add new tests dbregs_dr[0123]_trap_code in arch/amd64

Add new tests:
 - dbregs_dr0_trap_code
 - dbregs_dr1_trap_code
 - dbregs_dr2_trap_code
 - dbregs_dr3_trap_code

This is the final set of tests for trap types that are supposed to be
supported by all amd64 CPUs.

Traps for code (instruction) execution must be set to 1 byte, otherwise
they are undefined. x86 code traps must point to the first byte of an
instruction.

Sponsored by <The NetBSD Foundation>

diffstat:

 tests/kernel/arch/amd64/t_ptrace_wait.c |  561 +++++++++++++++++++++++++++++++-
 1 files changed, 559 insertions(+), 2 deletions(-)

diffs (truncated from 588 to 300 lines):

diff -r 7f1170bd2cea -r 083ec7e3c0dc tests/kernel/arch/amd64/t_ptrace_wait.c
--- a/tests/kernel/arch/amd64/t_ptrace_wait.c   Mon Feb 20 01:38:28 2017 +0000
+++ b/tests/kernel/arch/amd64/t_ptrace_wait.c   Mon Feb 20 02:56:03 2017 +0000
@@ -1,4 +1,4 @@
-/*     $NetBSD: t_ptrace_wait.c,v 1.18 2017/02/20 01:34:53 kamil Exp $ */
+/*     $NetBSD: t_ptrace_wait.c,v 1.19 2017/02/20 02:56:03 kamil Exp $ */
 
 /*-
  * Copyright (c) 2016 The NetBSD Foundation, Inc.
@@ -27,7 +27,7 @@
  */
 
 #include <sys/cdefs.h>
-__RCSID("$NetBSD: t_ptrace_wait.c,v 1.18 2017/02/20 01:34:53 kamil Exp $");
+__RCSID("$NetBSD: t_ptrace_wait.c,v 1.19 2017/02/20 02:56:03 kamil Exp $");
 
 #include <sys/param.h>
 #include <sys/types.h>
@@ -5986,6 +5986,558 @@
 }
 #endif
 
+#if defined(HAVE_DBREGS)
+ATF_TC(dbregs_dr0_trap_code);
+ATF_TC_HEAD(dbregs_dr0_trap_code, tc)
+{
+       atf_tc_set_md_var(tc, "descr",
+           "Verify that setting trap with DR0 triggers SIGTRAP "
+           "(break on code execution trap)");
+}
+
+ATF_TC_BODY(dbregs_dr0_trap_code, tc)
+{
+       const int exitval = 5;
+       const int sigval = SIGSTOP;
+       pid_t child, wpid;
+#if defined(TWAIT_HAVE_STATUS)
+       int status;
+#endif
+       struct dbreg r1;
+       struct dbreg r2;
+       /* Number of available CPU Debug Registers on AMD64 */
+       const size_t len = 16;
+       size_t i;
+       volatile int watchme = 1;
+       union u dr7;
+
+       struct ptrace_siginfo info;
+       memset(&info, 0, sizeof(info));
+
+       dr7.raw = 0;
+       dr7.bits.global_dr0_breakpoint = 1;
+       dr7.bits.condition_dr0 = 0;     /* 0b00 -- break on code execution */
+       dr7.bits.len_dr0 = 0;           /* 0b00 -- 1 byte */
+
+       printf("Assert that known number of Debug Registers (%zu) is valid\n",
+           len);
+       ATF_REQUIRE_EQ(__arraycount(r1.dr), len);
+       ATF_REQUIRE_EQ(__arraycount(r2.dr), len);
+
+       printf("Before forking process PID=%d\n", getpid());
+       ATF_REQUIRE((child = fork()) != -1);
+       if (child == 0) {
+               printf("Before calling PT_TRACE_ME from child %d\n", getpid());
+               FORKEE_ASSERT(ptrace(PT_TRACE_ME, 0, NULL, 0) != -1);
+
+               printf("Before raising %s from child\n", strsignal(sigval));
+               FORKEE_ASSERT(raise(sigval) == 0);
+
+               printf("check_happy(%d)=%d\n", watchme, check_happy(watchme));
+
+               printf("Before raising %s from child\n", strsignal(sigval));
+               FORKEE_ASSERT(raise(sigval) == 0);
+
+               printf("Before exiting of the child process\n");
+               _exit(exitval);
+       }
+       printf("Parent process PID=%d, child's PID=%d\n", getpid(), child);
+
+       printf("Before calling %s() for the child\n", TWAIT_FNAME);
+       TWAIT_REQUIRE_SUCCESS(wpid = TWAIT_GENERIC(child, &status, 0), child);
+
+       validate_status_stopped(status, sigval);
+
+       printf("Call GETDBREGS for the child process (r1)\n");
+       ATF_REQUIRE(ptrace(PT_GETDBREGS, child, &r1, 0) != -1);
+
+       printf("State of the debug registers (r1):\n");
+       for (i = 0; i < __arraycount(r1.dr); i++)
+               printf("r1[%zu]=%#lx\n", i, r1.dr[i]);
+
+       r1.dr[0] = (long)(intptr_t)check_happy;
+       printf("Set DR0 (r1.dr[0]) to new value %#lx\n", r1.dr[0]);
+
+       r1.dr[7] = dr7.raw;
+       printf("Set DR7 (r1.dr[7]) to new value %#lx\n", r1.dr[7]);
+
+       printf("New state of the debug registers (r1):\n");
+       for (i = 0; i < __arraycount(r1.dr); i++)
+               printf("r1[%zu]=%#lx\n", i, r1.dr[i]);
+
+       printf("Call SETDBREGS for the child process (r1)\n");
+       ATF_REQUIRE(ptrace(PT_SETDBREGS, child, &r1, 0) != -1);
+
+       printf("Call CONTINUE for the child process\n");
+       ATF_REQUIRE(ptrace(PT_CONTINUE, child, (void *)1, 0) != -1);
+
+       printf("Before calling %s() for the child\n", TWAIT_FNAME);
+       TWAIT_REQUIRE_SUCCESS(wpid = TWAIT_GENERIC(child, &status, 0), child);
+
+       validate_status_stopped(status, SIGTRAP);
+
+       printf("Before calling ptrace(2) with PT_GET_SIGINFO for child\n");
+       ATF_REQUIRE(ptrace(PT_GET_SIGINFO, child, &info, sizeof(info)) != -1);
+
+       printf("Signal traced to lwpid=%d\n", info.psi_lwpid);
+       printf("Signal properties: si_signo=%#x si_code=%#x si_errno=%#x\n",
+           info.psi_siginfo.si_signo, info.psi_siginfo.si_code,
+           info.psi_siginfo.si_errno);
+
+       printf("Before checking siginfo_t\n");
+       ATF_REQUIRE_EQ(info.psi_siginfo.si_signo, SIGTRAP);
+       ATF_REQUIRE_EQ(info.psi_siginfo.si_code, TRAP_DBREG);
+
+       printf("Remove code trap from check_happy=%p\n", check_happy);
+       dr7.bits.global_dr0_breakpoint = 0;
+       r1.dr[7] = dr7.raw;
+       printf("Set DR7 (r1.dr[7]) to new value %#lx\n", r1.dr[7]);
+
+       printf("Call SETDBREGS for the child process (r1)\n");
+       ATF_REQUIRE(ptrace(PT_SETDBREGS, child, &r1, 0) != -1);
+
+       printf("Call CONTINUE for the child process\n");
+       ATF_REQUIRE(ptrace(PT_CONTINUE, child, (void *)1, 0) != -1);
+
+       printf("Before calling %s() for the child\n", TWAIT_FNAME);
+       TWAIT_REQUIRE_SUCCESS(wpid = TWAIT_GENERIC(child, &status, 0), child);
+
+       validate_status_stopped(status, sigval);
+
+       printf("Call GETDBREGS for the child process (r2)\n");
+       ATF_REQUIRE(ptrace(PT_GETDBREGS, child, &r2, 0) != -1);
+
+       printf("Assert that (r1) and (r2) are the same\n");
+       ATF_REQUIRE(memcmp(&r1, &r2, len) == 0);
+
+       printf("Before resuming the child process where it left off and "
+           "without signal to be sent\n");
+       ATF_REQUIRE(ptrace(PT_CONTINUE, child, (void *)1, 0) != -1);
+
+       printf("Before calling %s() for the child\n", TWAIT_FNAME);
+       TWAIT_REQUIRE_SUCCESS(wpid = TWAIT_GENERIC(child, &status, 0), child);
+
+       validate_status_exited(status, exitval);
+
+       printf("Before calling %s() for the child\n", TWAIT_FNAME);
+       TWAIT_REQUIRE_FAILURE(ECHILD, wpid = TWAIT_GENERIC(child, &status, 0));
+}
+#endif
+
+#if defined(HAVE_DBREGS)
+ATF_TC(dbregs_dr1_trap_code);
+ATF_TC_HEAD(dbregs_dr1_trap_code, tc)
+{
+       atf_tc_set_md_var(tc, "descr",
+           "Verify that setting trap with DR1 triggers SIGTRAP "
+           "(break on code execution trap)");
+}
+
+ATF_TC_BODY(dbregs_dr1_trap_code, tc)
+{
+       const int exitval = 5;
+       const int sigval = SIGSTOP;
+       pid_t child, wpid;
+#if defined(TWAIT_HAVE_STATUS)
+       int status;
+#endif
+       struct dbreg r1;
+       struct dbreg r2;
+       /* Number of available CPU Debug Registers on AMD64 */
+       const size_t len = 16;
+       size_t i;
+       volatile int watchme = 1;
+       union u dr7;
+
+       struct ptrace_siginfo info;
+       memset(&info, 0, sizeof(info));
+
+       dr7.raw = 0;
+       dr7.bits.global_dr1_breakpoint = 1;
+       dr7.bits.condition_dr1 = 0;     /* 0b00 -- break on code execution */
+       dr7.bits.len_dr1 = 0;           /* 0b00 -- 1 byte */
+
+       printf("Assert that known number of Debug Registers (%zu) is valid\n",
+           len);
+       ATF_REQUIRE_EQ(__arraycount(r1.dr), len);
+       ATF_REQUIRE_EQ(__arraycount(r2.dr), len);
+
+       printf("Before forking process PID=%d\n", getpid());
+       ATF_REQUIRE((child = fork()) != -1);
+       if (child == 0) {
+               printf("Before calling PT_TRACE_ME from child %d\n", getpid());
+               FORKEE_ASSERT(ptrace(PT_TRACE_ME, 0, NULL, 0) != -1);
+
+               printf("Before raising %s from child\n", strsignal(sigval));
+               FORKEE_ASSERT(raise(sigval) == 0);
+
+               printf("check_happy(%d)=%d\n", watchme, check_happy(watchme));
+
+               printf("Before raising %s from child\n", strsignal(sigval));
+               FORKEE_ASSERT(raise(sigval) == 0);
+
+               printf("Before exiting of the child process\n");
+               _exit(exitval);
+       }
+       printf("Parent process PID=%d, child's PID=%d\n", getpid(), child);
+
+       printf("Before calling %s() for the child\n", TWAIT_FNAME);
+       TWAIT_REQUIRE_SUCCESS(wpid = TWAIT_GENERIC(child, &status, 0), child);
+
+       validate_status_stopped(status, sigval);
+
+       printf("Call GETDBREGS for the child process (r1)\n");
+       ATF_REQUIRE(ptrace(PT_GETDBREGS, child, &r1, 0) != -1);
+
+       printf("State of the debug registers (r1):\n");
+       for (i = 0; i < __arraycount(r1.dr); i++)
+               printf("r1[%zu]=%#lx\n", i, r1.dr[i]);
+
+       r1.dr[1] = (long)(intptr_t)check_happy;
+       printf("Set DR1 (r1.dr[1]) to new value %#lx\n", r1.dr[1]);
+
+       r1.dr[7] = dr7.raw;
+       printf("Set DR7 (r1.dr[7]) to new value %#lx\n", r1.dr[7]);
+
+       printf("New state of the debug registers (r1):\n");
+       for (i = 0; i < __arraycount(r1.dr); i++)
+               printf("r1[%zu]=%#lx\n", i, r1.dr[i]);
+
+       printf("Call SETDBREGS for the child process (r1)\n");
+       ATF_REQUIRE(ptrace(PT_SETDBREGS, child, &r1, 0) != -1);
+
+       printf("Call CONTINUE for the child process\n");
+       ATF_REQUIRE(ptrace(PT_CONTINUE, child, (void *)1, 0) != -1);
+
+       printf("Before calling %s() for the child\n", TWAIT_FNAME);
+       TWAIT_REQUIRE_SUCCESS(wpid = TWAIT_GENERIC(child, &status, 0), child);
+
+       validate_status_stopped(status, SIGTRAP);
+
+       printf("Before calling ptrace(2) with PT_GET_SIGINFO for child\n");
+       ATF_REQUIRE(ptrace(PT_GET_SIGINFO, child, &info, sizeof(info)) != -1);
+
+       printf("Signal traced to lwpid=%d\n", info.psi_lwpid);
+       printf("Signal properties: si_signo=%#x si_code=%#x si_errno=%#x\n",
+           info.psi_siginfo.si_signo, info.psi_siginfo.si_code,
+           info.psi_siginfo.si_errno);
+
+       printf("Before checking siginfo_t\n");
+       ATF_REQUIRE_EQ(info.psi_siginfo.si_signo, SIGTRAP);
+       ATF_REQUIRE_EQ(info.psi_siginfo.si_code, TRAP_DBREG);
+
+       printf("Remove code trap from check_happy=%p\n", check_happy);
+       dr7.bits.global_dr1_breakpoint = 0;
+       r1.dr[7] = dr7.raw;
+       printf("Set DR7 (r1.dr[7]) to new value %#lx\n", r1.dr[7]);
+
+       printf("Call SETDBREGS for the child process (r1)\n");
+       ATF_REQUIRE(ptrace(PT_SETDBREGS, child, &r1, 0) != -1);
+
+       printf("Call CONTINUE for the child process\n");
+       ATF_REQUIRE(ptrace(PT_CONTINUE, child, (void *)1, 0) != -1);
+
+       printf("Before calling %s() for the child\n", TWAIT_FNAME);
+       TWAIT_REQUIRE_SUCCESS(wpid = TWAIT_GENERIC(child, &status, 0), child);
+
+       validate_status_stopped(status, sigval);
+
+       printf("Call GETDBREGS for the child process (r2)\n");
+       ATF_REQUIRE(ptrace(PT_GETDBREGS, child, &r2, 0) != -1);
+
+       printf("Assert that (r1) and (r2) are the same\n");
+       ATF_REQUIRE(memcmp(&r1, &r2, len) == 0);
+
+       printf("Before resuming the child process where it left off and "
+           "without signal to be sent\n");
+       ATF_REQUIRE(ptrace(PT_CONTINUE, child, (void *)1, 0) != -1);
+
+       printf("Before calling %s() for the child\n", TWAIT_FNAME);
+       TWAIT_REQUIRE_SUCCESS(wpid = TWAIT_GENERIC(child, &status, 0), child);
+
+       validate_status_exited(status, exitval);
+
+       printf("Before calling %s() for the child\n", TWAIT_FNAME);
+       TWAIT_REQUIRE_FAILURE(ECHILD, wpid = TWAIT_GENERIC(child, &status, 0));
+}
+#endif
+
+#if defined(HAVE_DBREGS)
+ATF_TC(dbregs_dr2_trap_code);