commit 151fe198988ce3ab10964f4fca4401978caa18f1
author Hengqi Chen <chenhengqi@outlook.com> Sun May 16 17:18:27 2021 +0800
committer yonghong-song <ys114321@gmail.com> Mon May 17 12:01:31 2021 -0700
tree d62d880d628416612e525eb9a0c23e8949f8be31
parent 89c96a8cc45041e93cda6f6318170f56cb8e0184

tools: filter/display using PID intead of TID

Signed-off-by: Hengqi Chen <chenhengqi@outlook.com>

tools/gethostlatency.py

diff --git a/tools/gethostlatency.py b/tools/gethostlatency.py
index 0ba5a1e..353055d 100755
--- a/tools/gethostlatency.py
+++ b/tools/gethostlatency.py
@@ -61,14 +61,16 @@
         return 0;
 
     struct val_t val = {};
-    u32 pid = bpf_get_current_pid_tgid();
+    u64 pid_tgid = bpf_get_current_pid_tgid();
+    u32 pid = pid_tgid >> 32;
+    u32 tid = (u32)pid_tgid;
 
     if (bpf_get_current_comm(&val.comm, sizeof(val.comm)) == 0) {
         bpf_probe_read_user(&val.host, sizeof(val.host),
                        (void *)PT_REGS_PARM1(ctx));
-        val.pid = bpf_get_current_pid_tgid();
+        val.pid = pid;
         val.ts = bpf_ktime_get_ns();
-        start.update(&pid, &val);
+        start.update(&tid, &val);
     }
 
     return 0;
@@ -78,11 +80,12 @@
     struct val_t *valp;
     struct data_t data = {};
     u64 delta;
-    u32 pid = bpf_get_current_pid_tgid();
+    u64 pid_tgid = bpf_get_current_pid_tgid();
+    u32 tid = (u32)pid_tgid;
 
     u64 tsp = bpf_ktime_get_ns();
 
-    valp = start.lookup(&pid);
+    valp = start.lookup(&tid);
     if (valp == 0)
         return 0;       // missed start
 
@@ -91,7 +94,7 @@
     data.pid = valp->pid;
     data.delta = tsp - valp->ts;
     events.perf_submit(ctx, &data, sizeof(data));
-    start.delete(&pid);
+    start.delete(&tid);
     return 0;
 }
 """
@@ -113,11 +116,11 @@
                    pid=args.pid)
 
 # header
-print("%-9s %-6s %-16s %10s %s" % ("TIME", "PID", "COMM", "LATms", "HOST"))
+print("%-9s %-7s %-16s %10s %s" % ("TIME", "PID", "COMM", "LATms", "HOST"))
 
 def print_event(cpu, data, size):
     event = b["events"].event(data)
-    print("%-9s %-6d %-16s %10.2f %s" % (strftime("%H:%M:%S"), event.pid,
+    print("%-9s %-7d %-16s %10.2f %s" % (strftime("%H:%M:%S"), event.pid,
         event.comm.decode('utf-8', 'replace'), (float(event.delta) / 1000000),
         event.host.decode('utf-8', 'replace')))