commit 54044d51f2298844bddb211c7e984012726ad7a4
author detailyang <detailyang@gmail.com> Thu Jan 10 00:43:39 2019 +0800
committer yonghong-song <ys114321@gmail.com> Wed Jan 09 08:43:39 2019 -0800
tree 71044307dcb6ed085d9f33641579352f85e606ef
parent 9f851e2a13700b414bea361d727e7bd3ce6c1ecc

tools: allow tcpaccept filter via PID (#2117)

add missing FILTER in the bpf program.

tools/tcpaccept.py

diff --git a/tools/tcpaccept.py b/tools/tcpaccept.py
index e66d334..6c36051 100755
--- a/tools/tcpaccept.py
+++ b/tools/tcpaccept.py
@@ -84,6 +84,8 @@
     struct sock *newsk = (struct sock *)PT_REGS_RC(ctx);
     u32 pid = bpf_get_current_pid_tgid();
 
+    ##FILTER_PID##
+
     if (newsk == NULL)
         return 0;
 
@@ -161,6 +163,9 @@
     if (args->protocol != IPPROTO_TCP)
         return 0;
     u32 pid = bpf_get_current_pid_tgid();
+
+    ##FILTER_PID##
+
     // pull in details
     u16 family = 0, lport = 0;
     family = args->family;
@@ -197,10 +202,10 @@
 
 # code substitutions
 if args.pid:
-    bpf_text = bpf_text.replace('FILTER',
+    bpf_text = bpf_text.replace('##FILTER_PID##',
         'if (pid != %s) { return 0; }' % args.pid)
 else:
-    bpf_text = bpf_text.replace('FILTER', '')
+    bpf_text = bpf_text.replace('##FILTER_PID##', '')
 if debug or args.ebpf:
     print(bpf_text)
     if args.ebpf: