From: Matt Joiner <anacrolix@gmail.com>
Date: Sat, 20 Feb 2021 11:16:54 +0000 (+1100)
Subject: Add ClientConfig.ConfigureAnacrolixDhtServer
X-Git-Tag: v1.26.0-alpha~14
X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=cff461c89844fc439c69fcc22f03f7fc6ef20e56;p=btrtrc.git

Add ClientConfig.ConfigureAnacrolixDhtServer
---

diff --git a/client.go b/client.go
index cc59dd34..fa17ccb5 100644
--- a/client.go
+++ b/client.go
@@ -361,6 +361,9 @@ func (cl *Client) newAnacrolixDhtServer(conn net.PacketConn) (s *dht.Server, err
 		OnQuery:            cl.config.DHTOnQuery,
 		Logger:             cl.logger.WithContextText(fmt.Sprintf("dht server on %v", conn.LocalAddr().String())),
 	}
+	if f := cl.config.ConfigureAnacrolixDhtServer; f != nil {
+		f(&cfg)
+	}
 	s, err = dht.NewServer(&cfg)
 	if err == nil {
 		go func() {
diff --git a/config.go b/config.go
index 0cfde9d0..c7b08965 100644
--- a/config.go
+++ b/config.go
@@ -37,6 +37,9 @@ type ClientConfig struct {
 	// Don't create a DHT.
 	NoDHT            bool `long:"disable-dht"`
 	DhtStartingNodes func(network string) dht.StartingNodesGetter
+	// Called for each anacrolix/dht Server created for the Client.
+	ConfigureAnacrolixDhtServer func(*dht.ServerConfig)
+
 	// Never send chunks to peers.
 	NoUpload bool `long:"no-upload"`
 	// Disable uploading even when it isn't fair.
diff --git a/go.sum b/go.sum
index 85e1ff99..153b9937 100644
--- a/go.sum
+++ b/go.sum
@@ -58,6 +58,7 @@ github.com/alexflint/go-scalar v1.0.0 h1:NGupf1XV/Xb04wXskDFzS0KWOLH632W/EO4fAFi
 github.com/alexflint/go-scalar v1.0.0/go.mod h1:GpHzbCOZXEKMEcygYQ5n/aa4Aq84zbxjy3MxYW0gjYw=
 github.com/anacrolix/dht v0.0.0-20180412060941-24cbf25b72a4 h1:0yHJvFiGQhJ1gSHJOR8xzmnx45orEt7uiIB6guf0+zc=
 github.com/anacrolix/dht v0.0.0-20180412060941-24cbf25b72a4/go.mod h1:hQfX2BrtuQsLQMYQwsypFAab/GvHg8qxwVi4OJdR1WI=
+github.com/anacrolix/dht v1.0.1 h1:a7zVMiZWfPiToAUbjMZYeI3UvmsDP3j8vH5EDIAjM9c=
 github.com/anacrolix/dht/v2 v2.0.1 h1:gOHJ+OKqJ4Eb48OYStZm4AlWr1/nSA2TWlzb/+t36SA=
 github.com/anacrolix/dht/v2 v2.0.1/go.mod h1:GbTT8BaEtfqab/LPd5tY41f3GvYeii3mmDUK300Ycyo=
 github.com/anacrolix/dht/v2 v2.2.1-0.20191103020011-1dba080fb358 h1:Fji8X0K+FTn9UUupodVCCnQG7wNXETUARGT+8P5s+N0=