Fix signal handling

awnumar · Mar 11, 2020 · 2fc8c81 · 2fc8c81
1 parent a477fe1
commit 2fc8c81
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/examples/socketkey/socketkey.go b/examples/socketkey/socketkey.go
@@ -45,7 +45,7 @@ func SocketKey(size int) {
 	memguard.CatchSignal(func(s os.Signal) {
 		fmt.Println("Received signal:", s.String())
 		listener.Close()
-	})
+	}, os.Interrupt, os.Kill)
 
 	// Purge the session before returning.
 	defer memguard.Purge()

diff --git a/examples/socketkey/socketkey_test.go b/examples/socketkey/socketkey_test.go
@@ -5,3 +5,10 @@ import "testing"
 func TestSocketKey(t *testing.T) {
 	SocketKey(4096)
 }
+
+func BenchmarkSocketKey32(b *testing.B) {
+	b.ReportAllocs()
+	for i := 0; i < b.N; i++ {
+		SocketKey(32)
+	}
+}