Merge pull request #2 from mozillazg/fix-cancel

fix the program cannot be terminated through ctrl-c
jschwinger233 · Mar 12, 2023 · 79533b0 · 79533b0
2 parents 105f01e + 84ffa45
commit 79533b0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -96,6 +96,12 @@ func main() {
 
 	println("start tracing")
 	for {
+		select {
+		case <-ctx.Done():
+			return
+		default:
+		}
+
 		meta := bpf.SkbdumpSkbMeta{}
 		if err := bpfObjs.MetaQueue.LookupAndDelete(nil, &meta); err != nil {
 			select {