Make 'tools/wakeuptime' use an offcpu filter

This modifies 'tools/wakeuptime' to include a filter in offcpu() for capturing data only for a given pid or for user threads only. Signed-off-by: Sandipan Das <sandipan@linux.vnet.ibm.com>
sjtuhjh · Nov 17, 2017 · 3edb453 · 3edb453
1 parent 2c2d46f
commit 3edb453
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tools/wakeuptime.py b/tools/wakeuptime.py
@@ -101,8 +101,13 @@ def signal_ignore(signal, frame):
 
 int offcpu(struct pt_regs *ctx) {
     u32 pid = bpf_get_current_pid_tgid();
-    u64 ts = bpf_ktime_get_ns();
-    // XXX: should filter here too, but need task_struct
+    struct task_struct *p = (struct task_struct *) bpf_get_current_task();
+    u64 ts;
+
+    if (FILTER)
+        return 0;
+
+    ts = bpf_ktime_get_ns();
     start.update(&pid, &ts);
     return 0;
 }