commit b20f5e741f086bf37b19365566b01b5bf6fa251e
author lorddoskias <lorddoskias@users.noreply.github.com> Sat May 30 19:17:33 2020 +0300
committer GitHub <noreply@github.com> Sat May 30 09:17:33 2020 -0700
tree b6a6f8ccdd1f6c9c38a9c7ceab1f787e5f40e17f
parent 5558e36bd97ace7bc3efe3a70d0c9d4fc0d34e2a

offwaketime: Add support for --state (#2940)

Since offwaketime is really an amalgamation of offcputime and wakeuptime
there is no reason why it shouldn't support the --state argument of the
former.

Co-authored-by: Nikolay Borisov <nborisov@suse.com>

tools/offwaketime.py

diff --git a/tools/offwaketime.py b/tools/offwaketime.py
index 665b666..88dc68c 100755
--- a/tools/offwaketime.py
+++ b/tools/offwaketime.py
@@ -93,6 +93,9 @@
     type=positive_nonzero_int,
     help="the amount of time in microseconds under which we " +
          "store traces (default U64_MAX)")
+parser.add_argument("--state", type=positive_int,
+    help="filter on this thread state bitmask (eg, 2 == TASK_UNINTERRUPTIBLE" +
+         ") see include/linux/sched.h")
 parser.add_argument("--ebpf", action="store_true",
     help=argparse.SUPPRESS)
 args = parser.parse_args()
@@ -147,7 +150,7 @@
     u32 pid = p->pid;
     u32 tgid = p->tgid;
 
-    if (!(THREAD_FILTER)) {
+    if (!((THREAD_FILTER) && (STATE_FILTER))) {
         return 0;
     }
 
@@ -171,7 +174,7 @@
     u64 ts = bpf_ktime_get_ns();
 
     // Record timestamp for the previous Process (Process going into waiting)
-    if (THREAD_FILTER) {
+    if ((THREAD_FILTER) && (STATE_FILTER)) {
         start.update(&pid, &ts);
     }
 
@@ -234,7 +237,15 @@
 else:
     thread_context = "all threads"
     thread_filter = '1'
+if args.state == 0:
+    state_filter = 'p->state == 0'
+elif args.state:
+    # these states are sometimes bitmask checked
+    state_filter = 'p->state & %d' % args.state
+else:
+    state_filter = '1'
 bpf_text = bpf_text.replace('THREAD_FILTER', thread_filter)
+bpf_text = bpf_text.replace('STATE_FILTER', state_filter)
 
 # set stack storage size
 bpf_text = bpf_text.replace('STACK_STORAGE_SIZE', str(args.stack_storage_size))

tools/offwaketime_example.txt

diff --git a/tools/offwaketime_example.txt b/tools/offwaketime_example.txt
index 8291e2f..02c5a04 100644
--- a/tools/offwaketime_example.txt
+++ b/tools/offwaketime_example.txt
@@ -307,7 +307,7 @@
 # ./offwaketime -h
 usage: offwaketime [-h] [-p PID | -t TID | -u | -k] [-U | -K] [-d] [-f]
                    [--stack-storage-size STACK_STORAGE_SIZE]
-                   [-m MIN_BLOCK_TIME] [-M MAX_BLOCK_TIME]
+                   [-m MIN_BLOCK_TIME] [-M MAX_BLOCK_TIME] [--state STATE]
                    [duration]
 
 Summarize blocked time by kernel stack trace + waker stack
@@ -340,6 +340,8 @@
   -M MAX_BLOCK_TIME, --max-block-time MAX_BLOCK_TIME
                         the amount of time in microseconds under which we
                         store traces (default U64_MAX)
+  --state STATE         filter on this thread state bitmask (eg, 2 ==
+                        TASK_UNINTERRUPTIBLE) see include/linux/sched.h
 
 examples:
     ./offwaketime             # trace off-CPU + waker stack time until Ctrl-C