xnu-7195.101.1.tar.gz

[apple/xnu.git] / tests / telemetry.c

diff --git a/tests/telemetry.c b/tests/telemetry.c
index abf66285b60955b0331f7963f484ff5f75bce9fb..68d99ad9444a55c41915bc234d0d9deee96b7b42 100644 (file)
--- a/tests/telemetry.c
+++ b/tests/telemetry.c
@@ -4,12 +4,15 @@
 #include <darwintest.h>
 #include <dispatch/dispatch.h>
 #include <ktrace/ktrace.h>
+#include <kperf/kperf.h>
 #include <kern/debug.h>
 #include <notify.h>
 #include <sys/kdebug.h>
 #include <sys/sysctl.h>
 #include <TargetConditionals.h>
 
+#include "ktrace_helpers.h"
+
 enum telemetry_pmi {
        TELEMETRY_PMI_NONE,
        TELEMETRY_PMI_INSTRS,
@@ -132,6 +135,7 @@ thread_spin(__unused void *arg)
 T_DECL(microstackshot_pmi, "attempt to configure microstackshots on PMI")
 {
        skip_if_pmi_unsupported();
+       start_controlling_ktrace();
 
        T_SETUPBEGIN;
        ktrace_session_t s = ktrace_session_create();