[src/trunk]: src/tests/kernel Add new tests read_d_write_d_handshake[12] in t...

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

details:   https://anonhg.NetBSD.org/src/rev/764dfc09fdd5
branches:  trunk
changeset: 819233:764dfc09fdd5
user:      kamil <kamil%NetBSD.org@localhost>
date:      Wed Nov 23 21:14:23 2016 +0000

description:
Add new tests read_d_write_d_handshake[12] in t_ptrace_wait{,3,4,6,id,pid}

read_d_write_d_handshake1:
    Verify PT_READ_D with PT_WRITE_D handshake

read_d_write_d_handshake2:
    Verify PT_WRITE_D with PT_READ_D handshake

Sponsored by <The NetBSD Foundation>

diffstat:

 tests/kernel/t_ptrace_wait.c |  153 ++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 151 insertions(+), 2 deletions(-)

diffs (180 lines):

diff -r f3b52a721a2e -r 764dfc09fdd5 tests/kernel/t_ptrace_wait.c
--- a/tests/kernel/t_ptrace_wait.c      Wed Nov 23 20:56:47 2016 +0000
+++ b/tests/kernel/t_ptrace_wait.c      Wed Nov 23 21:14:23 2016 +0000
@@ -1,4 +1,4 @@
-/*     $NetBSD: t_ptrace_wait.c,v 1.25 2016/11/23 20:56:47 kamil Exp $ */
+/*     $NetBSD: t_ptrace_wait.c,v 1.26 2016/11/23 21:14:23 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.25 2016/11/23 20:56:47 kamil Exp $");
+__RCSID("$NetBSD: t_ptrace_wait.c,v 1.26 2016/11/23 21:14:23 kamil Exp $");
 
 #include <sys/param.h>
 #include <sys/types.h>
@@ -3279,6 +3279,152 @@
        TWAIT_REQUIRE_FAILURE(ECHILD, wpid = TWAIT_GENERIC(child, &status, 0));
 }
 
+ATF_TC(read_d_write_d_handshake1);
+ATF_TC_HEAD(read_d_write_d_handshake1, tc)
+{
+       atf_tc_set_md_var(tc, "descr",
+           "Verify PT_READ_D with PT_WRITE_D handshake");
+}
+
+ATF_TC_BODY(read_d_write_d_handshake1, tc)
+{
+       const int exitval = 5;
+       const int sigval = SIGSTOP;
+       pid_t child, wpid;
+       int lookup_me_fromtracee = 0;
+       const int magic_fromtracee = (int)random();
+       int lookup_me_totracee = 0;
+       const int magic_totracee = (int)random();
+#if defined(TWAIT_HAVE_STATUS)
+       int status;
+#endif
+
+       printf("Before forking process PID=%d\n", getpid());
+       child = atf_utils_fork();
+       if (child == 0) {
+               printf("Before calling PT_TRACE_ME from child %d\n", getpid());
+               FORKEE_ASSERT(ptrace(PT_TRACE_ME, 0, NULL, 0) != -1);
+
+               lookup_me_fromtracee = magic_fromtracee;
+
+               printf("Before raising %s from child\n", strsignal(sigval));
+               FORKEE_ASSERT(raise(sigval) == 0);
+
+               FORKEE_ASSERT_EQ(lookup_me_totracee, magic_totracee);
+
+               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("Read new lookup_me_fromtracee PID=%d by tracer (PID=%d)\n",
+           child, getpid());
+       errno = 0;
+       lookup_me_fromtracee =
+           ptrace(PT_READ_D, child, &lookup_me_fromtracee, 0);
+       ATF_REQUIRE_EQ(errno, 0);
+
+       ATF_REQUIRE_EQ_MSG(lookup_me_fromtracee, magic_fromtracee,
+           "got value %" PRIx8 " != expected %" PRIx8, lookup_me_fromtracee,
+           magic_fromtracee);
+
+       printf("Write new lookup_me_totracee to PID=%d from tracer (PID=%d)\n",
+           child, getpid());
+       ATF_REQUIRE
+           (ptrace(PT_WRITE_D, child, &lookup_me_totracee, magic_totracee)
+           != -1);
+
+       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));
+}
+
+ATF_TC(read_d_write_d_handshake2);
+ATF_TC_HEAD(read_d_write_d_handshake2, tc)
+{
+       atf_tc_set_md_var(tc, "descr",
+           "Verify PT_WRITE_D with PT_READ_D handshake");
+}
+
+ATF_TC_BODY(read_d_write_d_handshake2, tc)
+{
+       const int exitval = 5;
+       const int sigval = SIGSTOP;
+       pid_t child, wpid;
+       int lookup_me_fromtracee = 0;
+       const int magic_fromtracee = (int)random();
+       int lookup_me_totracee = 0;
+       const int magic_totracee = (int)random();
+#if defined(TWAIT_HAVE_STATUS)
+       int status;
+#endif
+
+       printf("Before forking process PID=%d\n", getpid());
+       child = atf_utils_fork();
+       if (child == 0) {
+               printf("Before calling PT_TRACE_ME from child %d\n", getpid());
+               FORKEE_ASSERT(ptrace(PT_TRACE_ME, 0, NULL, 0) != -1);
+
+               lookup_me_fromtracee = magic_fromtracee;
+
+               printf("Before raising %s from child\n", strsignal(sigval));
+               FORKEE_ASSERT(raise(sigval) == 0);
+
+               FORKEE_ASSERT_EQ(lookup_me_totracee, magic_totracee);
+
+               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("Write new lookup_me_totracee to PID=%d from tracer (PID=%d)\n",
+           child, getpid());
+       ATF_REQUIRE
+           (ptrace(PT_WRITE_D, child, &lookup_me_totracee, magic_totracee)
+           != -1);
+
+       printf("Read new lookup_me_fromtracee PID=%d by tracer (PID=%d)\n",
+           child, getpid());
+       errno = 0;
+       lookup_me_fromtracee =
+           ptrace(PT_READ_D, child, &lookup_me_fromtracee, 0);
+       ATF_REQUIRE_EQ(errno, 0);
+
+       ATF_REQUIRE_EQ_MSG(lookup_me_fromtracee, magic_fromtracee,
+           "got value %" PRIx8 " != expected %" PRIx8, lookup_me_fromtracee,
+           magic_fromtracee);
+
+       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));
+}
+
 #if defined(TWAIT_HAVE_PID)
 #define ATF_TP_ADD_TC_HAVE_PID(a,b)    ATF_TP_ADD_TC(a,b)
 #else
@@ -3334,5 +3480,8 @@
        ATF_TP_ADD_TC(tp, io_read_d_write_d_handshake1);
        ATF_TP_ADD_TC(tp, io_read_d_write_d_handshake2);
 
+       ATF_TP_ADD_TC(tp, read_d_write_d_handshake1);
+       ATF_TP_ADD_TC(tp, read_d_write_d_handshake2);
+
        return atf_no_error();
 }