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