fix wakeuptime's raw_tp: get awakened task from ctx not current

Signed-off-by: Wenbo Zhang <ethercflow@gmail.com>

diff --git a/tools/wakeuptime.py b/tools/wakeuptime.py
index 9fefca6fa02d0bd369a8dcbfe49849f4fe618703..0723f8af5904eb07cf7350885762468ad82c2411 100755 (executable)
--- a/tools/wakeuptime.py
+++ b/tools/wakeuptime.py
@@ -164,7 +164,7 @@ RAW_TRACEPOINT_PROBE(sched_switch)
 RAW_TRACEPOINT_PROBE(sched_wakeup)
 {
     // TP_PROTO(struct task_struct *p)
-    struct task_struct *p = (struct task_struct *) bpf_get_current_task();
+    struct task_struct *p = (struct task_struct *)ctx->args[0];
     return wakeup(ctx, p);
 }
 """