添加 --no-tids 支持对tid进行过滤

Author: Your Name

Makefile

@@ -2,6 +2,11 @@ CMD_CLANG ?= clang
 CMD_GO ?= go
 CMD_RM ?= rm
 
+DEBUG_PRINT ?=
+ifeq ($(DEBUG),1)
+DEBUG_PRINT := -DDEBUG_PRINT
+endif
+
 .PHONY: all
 all: ebpf assets build
 	@echo $(shell date)
@@ -22,6 +27,7 @@ ebpf:
 	-nostdlibinc \
 	-no-canonical-prefixes \
 	-O2 \
+	$(DEBUG_PRINT)	\
 	-isystem external/bionic/libc/include \
 	-isystem external/bionic/libc/kernel/uapi \
 	-isystem external/bionic/libc/kernel/uapi/asm-arm64 \
@@ -38,6 +44,7 @@ ebpf:
 	-nostdlibinc \
 	-no-canonical-prefixes \
 	-O2 \
+	$(DEBUG_PRINT)	\
 	-isystem external/bionic/libc/include \
 	-isystem external/bionic/libc/kernel/uapi \
 	-isystem external/bionic/libc/kernel/uapi/asm-arm64 \

cli/cmd/root.go

@@ -78,6 +78,19 @@ func persistentPreRunEFunc(command *cobra.Command, args []string) error {
         target_config.Pid = global_config.Pid
     }
 
+    target_config.TidBlacklistMask = 0
+    if global_config.TidsBlacklist != "" {
+        tids := strings.Split(global_config.TidsBlacklist, ",")
+        if len(tids) > 5 {
+            return fmt.Errorf("max tid blacklist count is 5, provided count:%d", len(tids))
+        }
+        for i, v := range tids {
+            value, _ := strconv.ParseUint(v, 10, 32)
+            target_config.TidBlacklist[i] = uint32(value)
+            target_config.TidBlacklistMask |= (1 << i)
+        }
+    }
+
     // 第二步 通过包名获取uid和库路径 先通过pm命令获取安装位置
     if global_config.Name != "" {
         return parseByPackage(global_config.Name)
@@ -210,6 +223,7 @@ func init() {
     rootCmd.PersistentFlags().StringVarP(&global_config.Name, "name", "n", "", "must set uid or package name")
     rootCmd.PersistentFlags().Uint64VarP(&global_config.Uid, "uid", "u", 0, "must set uid or package name")
     rootCmd.PersistentFlags().Uint64VarP(&global_config.Pid, "pid", "p", 0, "add pid to filter")
+    rootCmd.PersistentFlags().StringVarP(&global_config.TidsBlacklist, "no-tids", "", "", "tid black list, max 5")
     rootCmd.PersistentFlags().BoolVarP(&global_config.Debug, "debug", "d", false, "enable debug logging")
     rootCmd.PersistentFlags().StringVarP(&global_config.LoggerFile, "out", "o", "", "-o save the packets to file")
     rootCmd.PersistentFlags().BoolVarP(&global_config.Quiet, "quiet", "", false, "use with --log-file, wont logging to terminal when used")

cli/cmd/stack.go

@@ -117,11 +117,13 @@ func stackCommandFunc(command *cobra.Command, args []string) {
             Symbol:  stack_config.Symbol,
             Offset:  stack_config.Offset,
             SConfig: config.SConfig{
-                UnwindStack: stack_config.UnwindStack,
-                ShowRegs:    stack_config.ShowRegs,
-                RegName:     stack_config.RegName,
-                Uid:         target_config.Uid,
-                Pid:         target_config.Pid,
+                UnwindStack:      stack_config.UnwindStack,
+                ShowRegs:         stack_config.ShowRegs,
+                RegName:          stack_config.RegName,
+                Uid:              target_config.Uid,
+                Pid:              target_config.Pid,
+                TidBlacklist:     target_config.TidBlacklist,
+                TidBlacklistMask: target_config.TidBlacklistMask,
             },
         }
         if err := pConfig.Check(); err == nil {

src/common.h

@@ -7,6 +7,8 @@
 
 #define TASK_COMM_LEN 16
 
+#define MAX_TID_BLACKLIST_COUNT 5
+
 typedef signed char __s8;
 
 typedef unsigned char __u8;

src/stack.c

@@ -22,6 +22,8 @@ struct {
 struct filter_t {
     u32 uid;
     u32 pid;
+    u32 tid_blacklist_mask;
+    u32 tid_blacklist[MAX_TID_BLACKLIST_COUNT];
 };
 
 struct {
@@ -52,6 +54,30 @@ int probe_stack(struct pt_regs* ctx) {
         return 0;
     }
 
+    #ifdef DEBUG_PRINT
+    char fmt0[] = "debug0, tid:%d mask:%d\n";
+    bpf_trace_printk(fmt0, sizeof(fmt0), tid, filter->tid_blacklist_mask);
+    #endif
+
+    #pragma unroll
+    for (int i = 0; i < MAX_TID_BLACKLIST_COUNT; i++) {
+        if ((filter->tid_blacklist_mask & (1 << i))) {
+
+            #ifdef DEBUG_PRINT
+            char fmt1[] = "debug1, tid:%d filter_tid:%d\n";
+            bpf_trace_printk(fmt1, sizeof(fmt1), tid, filter->tid_blacklist[i]);
+            #endif
+
+            if (filter->tid_blacklist[i] == tid) {
+                // 在tid黑名单直接跳过
+                return 0;
+            }
+        } else {
+            // 避免不必要的循环
+            break;
+        }
+    }
+
     u32 zero = 0;
     struct hook_data_event_t* event = bpf_map_lookup_elem(&data_buffer_heap, &zero);
     if (event == NULL) {

user/config/config_global.go

@@ -1,14 +1,15 @@
 package config
 
 type GlobalConfig struct {
-    Quiet      bool
-    Prepare    bool
-    Name       string
-    Debug      bool
-    Uid        uint64
-    Pid        uint64
-    LoggerFile string
-    ExecPath   string
+    Quiet         bool
+    Prepare       bool
+    Name          string
+    Debug         bool
+    Uid           uint64
+    Pid           uint64
+    TidsBlacklist string
+    LoggerFile    string
+    ExecPath      string
 }
 
 func NewGlobalConfig() *GlobalConfig {

user/config/config_hook.go

@@ -60,8 +60,10 @@ func (this *ProbeConfig) Check() error {
 
 func (this *ProbeConfig) GetFilter() StackFilter {
     filter := StackFilter{
-        uid: uint32(this.Uid),
-        pid: uint32(this.Pid),
+        uid:                uint32(this.Uid),
+        pid:                uint32(this.Pid),
+        tid_blacklist_mask: this.TidBlacklistMask,
+        tid_blacklist:      this.TidBlacklist,
     }
     return filter
 }

user/config/config_target.go

@@ -1,12 +1,14 @@
 package config
 
 type TargetConfig struct {
-    Name        string
-    Uid         uint64
-    Pid         uint64
-    LibraryDirs []string
-    DataDir     string
-    Abi         string
+    Name             string
+    Uid              uint64
+    Pid              uint64
+    TidBlacklist     [MAX_TID_BLACKLIST_COUNT]uint32
+    TidBlacklistMask uint32
+    LibraryDirs      []string
+    DataDir          string
+    Abi              string
 }
 
 func NewTargetConfig() *TargetConfig {

user/config/iconfig.go

@@ -1,8 +1,12 @@
 package config
 
+const MAX_TID_BLACKLIST_COUNT = 5
+
 type StackFilter struct {
-	uid uint32
-	pid uint32
+	uid                uint32
+	pid                uint32
+	tid_blacklist_mask uint32
+	tid_blacklist      [MAX_TID_BLACKLIST_COUNT]uint32
 }
 
 type SyscallFilter struct {
@@ -18,12 +22,14 @@ type IConfig interface {
 }
 
 type SConfig struct {
-	Uid         uint64
-	Pid         uint64
-	UnwindStack bool
-	ShowRegs    bool
-	RegName     string
-	Debug       bool
+	Uid              uint64
+	Pid              uint64
+	TidBlacklistMask uint32
+	TidBlacklist     [MAX_TID_BLACKLIST_COUNT]uint32
+	UnwindStack      bool
+	ShowRegs         bool
+	RegName          string
+	Debug            bool
 }
 
 func (this *SConfig) SetDebug(debug bool) {