client.go | 11 +++++------ diff --git a/client.go b/client.go index 19458318e6cce188058224380496586cb716bd0a..9809fd6699a16746ae985fbe09e644a67a51aae5 100644 --- a/client.go +++ b/client.go @@ -467,14 +467,13 @@ } cl.listeners = append(cl.listeners, l) go cl.acceptConnections(l, false) } - var utpSock *utp.Socket if !cl.disableUTP { - utpSock, err = utp.NewSocket(listenAddr()) + cl.utpSock, err = utp.NewSocket(listenAddr()) if err != nil { return } - cl.listeners = append(cl.listeners, utpSock) - go cl.acceptConnections(utpSock, true) + cl.listeners = append(cl.listeners, cl.utpSock) + go cl.acceptConnections(cl.utpSock, true) } if !cfg.NoDHT { dhtCfg := cfg.DHTConfig @@ -484,8 +483,8 @@ } if dhtCfg.Addr == "" { dhtCfg.Addr = listenAddr() } - if dhtCfg.Conn == nil && utpSock != nil { - dhtCfg.Conn = utpSock + if dhtCfg.Conn == nil && cl.utpSock != nil { + dhtCfg.Conn = cl.utpSock } cl.dHT, err = dht.NewServer(dhtCfg) if cl.ipBlockList != nil {