add a function to initialize session ticket keys in a Config

quic-go · Jun 24, 2023 · 8ae6b8e · 8ae6b8e
1 parent fea6306
commit 8ae6b8e
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/unsafe.go b/unsafe.go
@@ -94,3 +94,8 @@ func compareStruct(a, b reflect.Type) bool {
 	}
 	return true
 }
+
+// InitSessionTicketKeys triggers the initialization of session ticket keys.
+func InitSessionTicketKeys(conf *Config) {
+	fromConfig(conf).ticketKeys(nil)
+}
diff --git a/unsafe_test.go b/unsafe_test.go
@@ -170,3 +170,17 @@ func TestClientSessionStateReinterpretCast(t *testing.T) {
 // 		t.Fatal("failed")
 // 	}
 // }
+
+func TestInitSessionTicketKeys(t *testing.T) {
+	c1 := testConfig.Clone()
+	InitSessionTicketKeys(c1)
+	c2 := c1.Clone()
+
+	stk := fromConfig(c1).autoSessionTicketKeys
+	if len(stk) == 0 {
+		t.Fatal("no session ticket keys")
+	}
+	if !reflect.DeepEqual(stk, fromConfig(c2).autoSessionTicketKeys) {
+		t.Fatal("session ticket keys don't match")
+	}
+}