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 <[email protected]>
yingtaojuzi · 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;
 }