1*e7776783SApple OSS Distributions #ifndef KTRACE_HELPERS_H
2*e7776783SApple OSS Distributions #define KTRACE_HELPERS_H
3*e7776783SApple OSS Distributions
4*e7776783SApple OSS Distributions #include <darwintest.h>
5*e7776783SApple OSS Distributions #include <ktrace/ktrace.h>
6*e7776783SApple OSS Distributions #include <libproc.h>
7*e7776783SApple OSS Distributions #include <sys/sysctl.h>
8*e7776783SApple OSS Distributions #include <sys/kdebug.h>
9*e7776783SApple OSS Distributions
10*e7776783SApple OSS Distributions static inline void
reset_ktrace(void)11*e7776783SApple OSS Distributions reset_ktrace(void)
12*e7776783SApple OSS Distributions {
13*e7776783SApple OSS Distributions (void)sysctl((int[]){ CTL_KERN, KERN_KDEBUG, KERN_KDREMOVE }, 3,
14*e7776783SApple OSS Distributions NULL, 0, NULL, 0);
15*e7776783SApple OSS Distributions kperf_reset();
16*e7776783SApple OSS Distributions }
17*e7776783SApple OSS Distributions
18*e7776783SApple OSS Distributions static inline void
start_controlling_ktrace(void)19*e7776783SApple OSS Distributions start_controlling_ktrace(void)
20*e7776783SApple OSS Distributions {
21*e7776783SApple OSS Distributions T_SETUPBEGIN;
22*e7776783SApple OSS Distributions
23*e7776783SApple OSS Distributions int state = 0;
24*e7776783SApple OSS Distributions size_t statesz = sizeof(state);
25*e7776783SApple OSS Distributions int ret = sysctlbyname("ktrace.state", &state, &statesz, NULL, 0);
26*e7776783SApple OSS Distributions T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "getting ktrace state");
27*e7776783SApple OSS Distributions
28*e7776783SApple OSS Distributions if (state == 1) {
29*e7776783SApple OSS Distributions int ownerpid = 0;
30*e7776783SApple OSS Distributions size_t pidsz = sizeof(ownerpid);
31*e7776783SApple OSS Distributions ret = sysctlbyname("ktrace.owning_pid", &ownerpid, &pidsz, NULL, 0);
32*e7776783SApple OSS Distributions T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "getting owning pid");
33*e7776783SApple OSS Distributions
34*e7776783SApple OSS Distributions if (ownerpid <= 0) {
35*e7776783SApple OSS Distributions T_LOG("ktrace is in foreground, but no owner");
36*e7776783SApple OSS Distributions goto out;
37*e7776783SApple OSS Distributions }
38*e7776783SApple OSS Distributions
39*e7776783SApple OSS Distributions char ownername[1024];
40*e7776783SApple OSS Distributions ret = proc_name(ownerpid, ownername, sizeof(ownername));
41*e7776783SApple OSS Distributions if (ret == 0) {
42*e7776783SApple OSS Distributions T_LOG("ktrace is in foreground, but owner (%d) has no name", ownerpid);
43*e7776783SApple OSS Distributions goto out;
44*e7776783SApple OSS Distributions }
45*e7776783SApple OSS Distributions
46*e7776783SApple OSS Distributions T_LOG("ktrace is in foreground, owned by %s, sending SIGKILL", ownername);
47*e7776783SApple OSS Distributions kill(ownerpid, SIGKILL);
48*e7776783SApple OSS Distributions usleep(500000);
49*e7776783SApple OSS Distributions
50*e7776783SApple OSS Distributions ret = proc_name(ownerpid, ownername, sizeof(ownername));
51*e7776783SApple OSS Distributions T_QUIET; T_ASSERT_EQ(ret, 0, "should have killed ktrace owner");
52*e7776783SApple OSS Distributions }
53*e7776783SApple OSS Distributions
54*e7776783SApple OSS Distributions out:
55*e7776783SApple OSS Distributions reset_ktrace();
56*e7776783SApple OSS Distributions T_ATEND(reset_ktrace);
57*e7776783SApple OSS Distributions T_SETUPEND;
58*e7776783SApple OSS Distributions }
59*e7776783SApple OSS Distributions
60*e7776783SApple OSS Distributions static inline void
assert_kdebug_test(enum kdebug_test flavor,const char * msg)61*e7776783SApple OSS Distributions assert_kdebug_test(enum kdebug_test flavor, const char *msg)
62*e7776783SApple OSS Distributions {
63*e7776783SApple OSS Distributions size_t size = flavor;
64*e7776783SApple OSS Distributions int mib[] = { CTL_KERN, KERN_KDEBUG, KERN_KDTEST };
65*e7776783SApple OSS Distributions T_ASSERT_POSIX_SUCCESS(sysctl(mib, sizeof(mib) / sizeof(mib[0]), NULL,
66*e7776783SApple OSS Distributions &size, NULL, 0), "KERN_KDTEST %d: %s", flavor, msg);
67*e7776783SApple OSS Distributions }
68*e7776783SApple OSS Distributions
69*e7776783SApple OSS Distributions static inline uint64_t
ns_from_abs(ktrace_session_t s,uint64_t abstime)70*e7776783SApple OSS Distributions ns_from_abs(ktrace_session_t s, uint64_t abstime)
71*e7776783SApple OSS Distributions {
72*e7776783SApple OSS Distributions uint64_t ns = 0;
73*e7776783SApple OSS Distributions int error = ktrace_convert_timestamp_to_nanoseconds(s, abstime, &ns);
74*e7776783SApple OSS Distributions T_QUIET; T_ASSERT_POSIX_ZERO(error, "convert abstime to nanoseconds");
75*e7776783SApple OSS Distributions return ns;
76*e7776783SApple OSS Distributions }
77*e7776783SApple OSS Distributions
78*e7776783SApple OSS Distributions static inline uint64_t
relns_from_abs(ktrace_session_t s,uint64_t abstime)79*e7776783SApple OSS Distributions relns_from_abs(ktrace_session_t s, uint64_t abstime)
80*e7776783SApple OSS Distributions {
81*e7776783SApple OSS Distributions return ns_from_abs(s, abstime - ktrace_get_earliest_timestamp(s));
82*e7776783SApple OSS Distributions }
83*e7776783SApple OSS Distributions
84*e7776783SApple OSS Distributions #endif /* !defined(KTRACE_HELPERS_H) */
85