uTP raw Socket use is now via .PacketConn

diff --git a/client.go b/client.go
index 4cc6025173441cfcd581816bb78575345dd3fea2..394cea2349ac0d191d3790290381840933d1a8f3 100644 (file)
--- a/client.go
+++ b/client.go
@@ -490,7 +490,7 @@ func NewClient(cfg *Config) (cl *Client, err error) {
                        dhtCfg.Addr = listenAddr()
                }
                if dhtCfg.Conn == nil && cl.utpSock != nil {
-                       dhtCfg.Conn = cl.utpSock
+                       dhtCfg.Conn = cl.utpSock.PacketConn()
                }
                cl.dHT, err = dht.NewServer(dhtCfg)
                if cl.ipBlockList != nil {

diff --git a/client_test.go b/client_test.go
index 5afcf2d38ce3019a815713c84a310b7655decda3..f7668238f0d0e8b6768156e2312a9753bd241967 100644 (file)
--- a/client_test.go
+++ b/client_test.go
@@ -162,7 +162,7 @@ func TestUTPRawConn(t *testing.T) {
                defer close(readerStopped)
                b := make([]byte, 500)
                for i := 0; i < N; i++ {
-                       n, _, err := l.ReadFrom(b)
+                       n, _, err := l.PacketConn().ReadFrom(b)
                        if err != nil {
                                t.Fatalf("error reading from raw conn: %s", err)
                        }