Set Client.utpSock

diff --git a/client.go b/client.go
index 19458318e6cce188058224380496586cb716bd0a..9809fd6699a16746ae985fbe09e644a67a51aae5 100644 (file)
--- a/client.go
+++ b/client.go
@@ -467,14 +467,13 @@ func NewClient(cfg *Config) (cl *Client, err error) {
                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 @@ func NewClient(cfg *Config) (cl *Client, err error) {
                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 {