From: Matt Joiner <anacrolix@gmail.com>
Date: Mon, 9 Feb 2015 13:17:04 +0000 (+1100)
Subject: uTP raw Socket use is now via .PacketConn
X-Git-Tag: v1.0.0~1343
X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=3c9fdba3457077c360fa79fe92ed2bc285e54514;p=btrtrc.git

uTP raw Socket use is now via .PacketConn
---

diff --git a/client.go b/client.go
index 4cc60251..394cea23 100644
--- 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 5afcf2d3..f7668238 100644
--- 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)
 			}