| Alexey Ivanov | cc01a9c | 2019-01-16 09:50:46 -0800 | [diff] [blame] | 1 | #!/usr/bin/python |
| Alexei Starovoitov | b6e43a2 | 2015-06-18 14:41:17 -0700 | [diff] [blame] | 2 | # Copyright (c) PLUMgrid, Inc. |
| 3 | # Licensed under the Apache License, Version 2.0 (the "License") |
| 4 | |
| Brenden Blanco | c35989d | 2015-09-02 18:04:07 -0700 | [diff] [blame] | 5 | from bcc import BPF |
| Alexei Starovoitov | b6e43a2 | 2015-06-18 14:41:17 -0700 | [diff] [blame] | 6 | from time import sleep |
| 7 | |
| Adam Drescher | 84ce85b | 2019-05-28 17:29:46 -0400 | [diff] [blame] | 8 | b = BPF(src_file="task_switch.c") |
| Guodong Xu | 00b72fd | 2021-03-13 02:23:47 +0000 | [diff] [blame] | 9 | b.attach_kprobe(event_re="^finish_task_switch$|^finish_task_switch\.isra\.\d$", |
| 10 | fn_name="count_sched") |
| Alexei Starovoitov | b6e43a2 | 2015-06-18 14:41:17 -0700 | [diff] [blame] | 11 | |
| 12 | # generate many schedule events |
| 13 | for i in range(0, 100): sleep(0.01) |
| 14 | |
| Brenden Blanco | c8b6698 | 2015-08-28 23:15:19 -0700 | [diff] [blame] | 15 | for k, v in b["stats"].items(): |
| Alexei Starovoitov | b6e43a2 | 2015-06-18 14:41:17 -0700 | [diff] [blame] | 16 | print("task_switch[%5d->%5d]=%u" % (k.prev_pid, k.curr_pid, v.value)) |