"bytes"
"context"
"crypto/rand"
- "encoding/binary"
"errors"
"fmt"
"io"
- "io/ioutil"
"net"
"net/url"
"sync"
"github.com/anacrolix/dht/v2/krpc"
_ "github.com/anacrolix/envpprof"
- "github.com/anacrolix/torrent/tracker/udp"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
+
+ "github.com/anacrolix/torrent/tracker/udp"
)
var trackers = []string{
- "udp://tracker.coppersurfer.tk:6969",
- "udp://tracker.leechers-paradise.org:6969",
-}
-
-// Ensure net.IPs are stored big-endian, to match the way they're read from
-// the wire.
-func TestNetIPv4Bytes(t *testing.T) {
- ip := net.IP([]byte{127, 0, 0, 1})
- if ip.String() != "127.0.0.1" {
- t.FailNow()
- }
- if string(ip) != "\x7f\x00\x00\x01" {
- t.Fatal([]byte(ip))
- }
-}
-
-func TestMarshalAnnounceResponse(t *testing.T) {
- peers := krpc.CompactIPv4NodeAddrs{
- {[]byte{127, 0, 0, 1}, 2},
- {[]byte{255, 0, 0, 3}, 4},
- }
- b, err := peers.MarshalBinary()
- require.NoError(t, err)
- require.EqualValues(t,
- "\x7f\x00\x00\x01\x00\x02\xff\x00\x00\x03\x00\x04",
- b)
- require.EqualValues(t, 12, binary.Size(udp.AnnounceResponseHeader{}))
-}
-
-// Failure to write an entire packet to UDP is expected to given an error.
-func TestLongWriteUDP(t *testing.T) {
- t.Parallel()
- l, err := net.ListenUDP("udp4", nil)
- require.NoError(t, err)
- defer l.Close()
- c, err := net.DialUDP("udp", nil, l.LocalAddr().(*net.UDPAddr))
- if err != nil {
- t.Fatal(err)
- }
- defer c.Close()
- for msgLen := 1; ; msgLen *= 2 {
- n, err := c.Write(make([]byte, msgLen))
- if err != nil {
- require.Contains(t, err.Error(), "message too long")
- return
- }
- if n < msgLen {
- t.FailNow()
- }
- }
-}
-
-func TestShortBinaryRead(t *testing.T) {
- var data udp.ResponseHeader
- err := binary.Read(bytes.NewBufferString("\x00\x00\x00\x01"), binary.BigEndian, &data)
- if err != io.ErrUnexpectedEOF {
- t.FailNow()
- }
-}
-
-func TestConvertInt16ToInt(t *testing.T) {
- i := 50000
- if int(uint16(int16(i))) != 50000 {
- t.FailNow()
- }
+ "udp://tracker.opentrackr.org:1337/announce",
+ "udp://tracker.openbittorrent.com:6969/announce",
+ "udp://localhost:42069",
}
func TestAnnounceLocalhost(t *testing.T) {
},
}
var err error
- srv.pc, err = net.ListenPacket("udp", ":0")
+ srv.pc, err = net.ListenPacket("udp", "localhost:0")
require.NoError(t, err)
defer srv.pc.Close()
go func() {
}
rand.Read(req.PeerId[:])
copy(req.InfoHash[:], []uint8{0xa3, 0x56, 0x41, 0x43, 0x74, 0x23, 0xe6, 0x26, 0xd9, 0x38, 0x25, 0x4a, 0x6b, 0x80, 0x49, 0x10, 0xa6, 0x67, 0xa, 0xc1})
- var ctx context.Context
+ ctx, cancel := context.WithTimeout(context.Background(), DefaultTrackerAnnounceTimeout)
+ defer cancel()
if dl, ok := t.Deadline(); ok {
var cancel func()
ctx, cancel = context.WithDeadline(context.Background(), dl.Add(-time.Second))
t.Skip(err)
}
require.NoError(t, err)
- t.Log(ar)
+ t.Logf("%+v", ar)
}
func TestAnnounceRandomInfoHashThirdParty(t *testing.T) {
rand.Read(req.PeerId[:])
rand.Read(req.InfoHash[:])
wg := sync.WaitGroup{}
- ctx, cancel := context.WithCancel(context.Background())
+ ctx, cancel := context.WithTimeout(context.Background(), DefaultTrackerAnnounceTimeout)
defer cancel()
if dl, ok := t.Deadline(); ok {
var cancel func()
// Check that URLPath option is done correctly.
func TestURLPathOption(t *testing.T) {
- conn, err := net.ListenUDP("udp", nil)
+ conn, err := net.ListenPacket("udp", "localhost:0")
if err != nil {
panic(err)
}
announceErr <- err
}()
var b [512]byte
+ // conn.SetReadDeadline(time.Now().Add(time.Second))
_, addr, _ := conn.ReadFrom(b[:])
r := bytes.NewReader(b[:])
var h udp.RequestHeader
r = bytes.NewReader(b[:n])
udp.Read(r, &h)
udp.Read(r, &AnnounceRequest{})
- all, _ := ioutil.ReadAll(r)
+ all, _ := io.ReadAll(r)
if string(all) != "\x02\x09/announce" {
t.FailNow()
}