bug fix !!!

SeeFlowerX · Dec 10, 2022 · e77d9d2 · e77d9d2
1 parent 7da18cc
commit e77d9d2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/raw_syscalls.c b/src/raw_syscalls.c
@@ -298,6 +298,11 @@ int raw_syscalls_sys_enter(struct bpf_raw_tracepoint_args* ctx) {
             }
         }
     } else {
+        // 可能是展开循环或者处于else分支的原因 这里必须得重新获取一次 arg_mask
+        struct arg_mask_t* arg_mask = bpf_map_lookup_elem(&arg_mask_map, &data->syscall_id);
+        if (arg_mask == NULL) {
+            return 0;
+        }
         // 展开循环
         #pragma unroll
         for (int i = 0; i < 6; i++) {