xnu-4903.221.2.tar.gz

[apple/xnu.git] / osfmk / kperf / task_samplers.c

diff --git a/osfmk/kperf/task_samplers.c b/osfmk/kperf/task_samplers.c
index f976518d4ec50eb7cfc401366f5ab9f4cfec9947..3d521b782988fd30d499f5097cc7aa0371afc517 100644 (file)
--- a/osfmk/kperf/task_samplers.c
+++ b/osfmk/kperf/task_samplers.c
@@ -33,22 +33,15 @@
 
 #include <kern/task.h>
 
-extern boolean_t memorystatus_proc_is_dirty_unsafe(void *v);
+extern void memorystatus_proc_flags_unsafe(void * v, boolean_t *is_dirty,
+               boolean_t *is_dirty_tracked, boolean_t *allow_idle_exit);
 
 void
-kperf_task_snapshot_sample(struct kperf_task_snapshot *tksn,
-                           struct kperf_context *ctx)
+kperf_task_snapshot_sample(task_t task, struct kperf_task_snapshot *tksn)
 {
-       thread_t thread;
-       task_t task;
-
        BUF_INFO(PERF_TK_SNAP_SAMPLE | DBG_FUNC_START);
 
        assert(tksn != NULL);
-       assert(ctx != NULL);
-
-       thread = ctx->cur_thread;
-       task = get_threadtask(thread);
 
        tksn->kptksn_flags = 0;
        if (task->effective_policy.tep_darwinbg) {
@@ -61,9 +54,17 @@ kperf_task_snapshot_sample(struct kperf_task_snapshot *tksn,
                tksn->kptksn_flags |= KPERF_TASK_FLAG_BOOSTED;
        }
 #if CONFIG_MEMORYSTATUS
-       if (memorystatus_proc_is_dirty_unsafe(task->bsd_info)) {
+       boolean_t dirty = FALSE, dirty_tracked = FALSE, allow_idle_exit = FALSE;
+       memorystatus_proc_flags_unsafe(task->bsd_info, &dirty, &dirty_tracked, &allow_idle_exit);
+       if (dirty) {
                tksn->kptksn_flags |= KPERF_TASK_FLAG_DIRTY;
        }
+       if (dirty_tracked) {
+               tksn->kptksn_flags |= KPERF_TASK_FLAG_DIRTY_TRACKED;
+       }
+       if (allow_idle_exit) {
+               tksn->kptksn_flags |= KPERF_TASK_ALLOW_IDLE_EXIT;
+       }
 #endif
 
        tksn->kptksn_suspend_count = task->suspend_count;
@@ -96,3 +97,11 @@ kperf_task_snapshot_log(struct kperf_task_snapshot *tksn)
                                        LOWER_32(tksn->kptksn_system_time_in_terminated_threads));
 #endif /* defined(__LP64__) */
 }
+
+void
+kperf_task_info_log(struct kperf_context *ctx)
+{
+       assert(ctx != NULL);
+
+       BUF_DATA(PERF_TK_INFO_DATA, ctx->cur_pid);
+}