Track accepted connections

diff --git a/client.go b/client.go
index 601386e70e0f8232d4744140f979ac280160c343..68464da92abecd3a3eb6a7c27c77f1ee9b732350 100644 (file)
--- a/client.go
+++ b/client.go
@@ -25,6 +25,7 @@ import (
        "github.com/anacrolix/missinggo"
        . "github.com/anacrolix/missinggo"
        "github.com/anacrolix/missinggo/bitmap"
+       "github.com/anacrolix/missinggo/pproffd"
        "github.com/anacrolix/missinggo/pubsub"
        "github.com/anacrolix/sync"
        "github.com/anacrolix/utp"
@@ -530,6 +531,7 @@ func (cl *Client) acceptConnections(l net.Listener, utp bool) {
        for {
                cl.waitAccept()
                conn, err := l.Accept()
+               conn = pproffd.WrapNetConn(conn)
                if cl.closed.IsSet() {
                        if conn != nil {
                                conn.Close()