214fe6b93c
WebRTC conns are providing the correct IP for peer priority calculations, so use that instead of trying to guess (which doesn't work if there are no regular conn listeners attached to the Client. (cherry picked from commit e86bb5fee3958dc90a3d012469b1352005d6c9ad)
62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
package torrent
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/anacrolix/dht/v2/krpc"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/anacrolix/torrent/metainfo"
|
|
pp "github.com/anacrolix/torrent/peer_protocol"
|
|
)
|
|
|
|
func TestPexConnState(t *testing.T) {
|
|
var cl Client
|
|
cl.init(TestingConfig(t))
|
|
cl.initLogger()
|
|
torrent := cl.newTorrent(metainfo.Hash{}, nil)
|
|
addr := &net.TCPAddr{IP: net.IPv6loopback, Port: 4747}
|
|
c := cl.newConnection(nil, newConnectionOpts{
|
|
remoteAddr: addr,
|
|
network: addr.Network(),
|
|
})
|
|
c.PeerExtensionIDs = make(map[pp.ExtensionName]pp.ExtensionNumber)
|
|
c.PeerExtensionIDs[pp.ExtensionNamePex] = pexExtendedId
|
|
c.messageWriter.mu.Lock()
|
|
c.setTorrent(torrent)
|
|
if err := torrent.addPeerConn(c); err != nil {
|
|
t.Log(err)
|
|
}
|
|
|
|
connWriteCond := c.messageWriter.writeCond.Signaled()
|
|
c.pex.Init(c)
|
|
require.True(t, c.pex.IsEnabled(), "should get enabled")
|
|
defer c.pex.Close()
|
|
|
|
var out pp.Message
|
|
writerCalled := false
|
|
testWriter := func(m pp.Message) bool {
|
|
writerCalled = true
|
|
out = m
|
|
return true
|
|
}
|
|
<-connWriteCond
|
|
c.pex.Share(testWriter)
|
|
require.True(t, writerCalled)
|
|
require.EqualValues(t, pp.Extended, out.Type)
|
|
require.EqualValues(t, pexExtendedId, out.ExtendedID)
|
|
|
|
x, err := pp.LoadPexMsg(out.ExtendedPayload)
|
|
require.NoError(t, err)
|
|
targx := pp.PexMsg{
|
|
Added: krpc.CompactIPv4NodeAddrs(nil),
|
|
AddedFlags: []pp.PexPeerFlags{},
|
|
Added6: krpc.CompactIPv6NodeAddrs{
|
|
mustNodeAddr(addr),
|
|
},
|
|
Added6Flags: []pp.PexPeerFlags{0},
|
|
}
|
|
require.EqualValues(t, targx, x)
|
|
}
|