out of order when oneshot

jschwinger233 · Nov 26, 2023 · c283160 · c283160
1 parent 1bd586c
commit c283160
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -35,7 +35,9 @@ func main() {
 		}
 	}()
 
-	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+	ctx, stop := signal.NotifyContext(ctx, os.Interrupt, syscall.SIGTERM)
 	defer stop()
 
 	if err = bpfObjs.Load(bpf.LoadOptions{
@@ -118,7 +120,10 @@ func main() {
 
 		if config.Oneshot {
 			if ksym, _ := utils.Addr2ksym(skb.Meta.At); ksym == "kfree_skbmem" {
-				break
+				go func() {
+					time.Sleep(1 * time.Second)
+					cancel()
+				}()
 			}
 		}
 	}