sevan
diff --git a/‎tools/funccount.py‎
Lines changed: 5 additions & 4 deletions b/‎tools/funccount.py‎
Lines changed: 5 additions & 4 deletions
@@ -19,6 +19,7 @@
 from time import sleep, strftime
 import argparse
 import signal
+import os
 
 # arguments
 examples = """examples:
@@ -76,12 +77,12 @@ def signal_ignore(signal, frame):
     return 0;
 }
 """
+filter_text = "u32 pid; pid = bpf_get_current_pid_tgid(); "
 if args.pid:
-    bpf_text = bpf_text.replace('FILTER',
-        ('u32 pid; pid = bpf_get_current_pid_tgid(); ' +
-        'if (pid != %s) { return 0; }') % (args.pid))
+    filter_text += "if (pid != %s) { return 0; }" % args.pid
 else:
-    bpf_text = bpf_text.replace('FILTER', '')
+    filter_text += "if (pid == %s) { return 0; }" % os.getpid()
+bpf_text = bpf_text.replace('FILTER', filter_text)
 if debug:
     print(bpf_text)
 b = BPF(text=bpf_text)