Преглед изворни кода

incorporate CID option into DTLS configuration

Vladislav Yarmak пре 8 месеци
родитељ
комит
d271d25e50
2 измењених фајлова са 7 додато и 0 уклоњено
  1. 4 0
      client/client.go
  2. 3 0
      server/server.go

+ 4 - 0
client/client.go

@@ -65,6 +65,10 @@ func New(cfg *Config) (*Client, error) {
 		CipherSuites:         cfg.CipherSuites,
 		EllipticCurves:       cfg.EllipticCurves,
 	}
+	if cfg.EnableCID {
+		client.dtlsConfig.ConnectionIDGenerator = dtls.OnlySendCIDGenerator()
+	}
+
 	lc := udp.ListenConfig{
 		Backlog: Backlog,
 	}

+ 3 - 0
server/server.go

@@ -71,6 +71,9 @@ func New(cfg *Config) (*Server, error) {
 			return nil
 		},
 	}
+	if cfg.EnableCID {
+		srv.dtlsConfig.ConnectionIDGenerator = dtls.RandomCIDGenerator(8)
+	}
 	srv.listener, err = dtls.Listen("udp", net.UDPAddrFromAddrPort(lAddrPort), srv.dtlsConfig)
 	if err != nil {
 		cancelCtx()