146 lines
3.3 KiB
Go
146 lines
3.3 KiB
Go
package torrent
|
|
|
|
import (
|
|
"io"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/anacrolix/missinggo/pubsub"
|
|
"github.com/bradfitz/iter"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/anacrolix/torrent/metainfo"
|
|
pp "github.com/anacrolix/torrent/peer_protocol"
|
|
"github.com/anacrolix/torrent/storage"
|
|
)
|
|
|
|
// Ensure that no race exists between sending a bitfield, and a subsequent
|
|
// Have that would potentially alter it.
|
|
func TestSendBitfieldThenHave(t *testing.T) {
|
|
r, w := io.Pipe()
|
|
c := &connection{
|
|
t: &Torrent{
|
|
cl: &Client{},
|
|
},
|
|
r: r,
|
|
w: w,
|
|
}
|
|
c.writerCond.L = &c.t.cl.mu
|
|
go c.writer(time.Minute)
|
|
c.mu().Lock()
|
|
c.Bitfield([]bool{false, true, false})
|
|
c.mu().Unlock()
|
|
c.mu().Lock()
|
|
c.Have(2)
|
|
c.mu().Unlock()
|
|
b := make([]byte, 15)
|
|
n, err := io.ReadFull(r, b)
|
|
c.mu().Lock()
|
|
// This will cause connection.writer to terminate.
|
|
c.closed.Set()
|
|
c.mu().Unlock()
|
|
require.NoError(t, err)
|
|
require.EqualValues(t, 15, n)
|
|
// Here we see that the bitfield doesn't have piece 2 set, as that should
|
|
// arrive in the following Have message.
|
|
require.EqualValues(t, "\x00\x00\x00\x02\x05@\x00\x00\x00\x05\x04\x00\x00\x00\x02", string(b))
|
|
}
|
|
|
|
type torrentStorage struct {
|
|
writeSem sync.Mutex
|
|
}
|
|
|
|
func (me *torrentStorage) Close() error { return nil }
|
|
|
|
func (me *torrentStorage) Piece(mp metainfo.Piece) storage.PieceImpl {
|
|
return me
|
|
}
|
|
|
|
func (me *torrentStorage) Completion() storage.Completion {
|
|
return storage.Completion{}
|
|
}
|
|
|
|
func (me *torrentStorage) MarkComplete() error {
|
|
return nil
|
|
}
|
|
|
|
func (me *torrentStorage) MarkNotComplete() error {
|
|
return nil
|
|
}
|
|
|
|
func (me *torrentStorage) ReadAt([]byte, int64) (int, error) {
|
|
panic("shouldn't be called")
|
|
}
|
|
|
|
func (me *torrentStorage) WriteAt(b []byte, _ int64) (int, error) {
|
|
if len(b) != defaultChunkSize {
|
|
panic(len(b))
|
|
}
|
|
me.writeSem.Unlock()
|
|
return len(b), nil
|
|
}
|
|
|
|
func BenchmarkConnectionMainReadLoop(b *testing.B) {
|
|
cl := &Client{}
|
|
ts := &torrentStorage{}
|
|
t := &Torrent{
|
|
cl: cl,
|
|
storage: &storage.Torrent{ts},
|
|
pieceStateChanges: pubsub.NewPubSub(),
|
|
}
|
|
require.NoError(b, t.setInfo(&metainfo.Info{
|
|
Pieces: make([]byte, 20),
|
|
Length: 1 << 20,
|
|
PieceLength: 1 << 20,
|
|
}))
|
|
t.setChunkSize(defaultChunkSize)
|
|
t.pendingPieces.Set(0, PiecePriorityNormal.BitmapPriority())
|
|
r, w := io.Pipe()
|
|
cn := &connection{
|
|
t: t,
|
|
r: r,
|
|
}
|
|
mrlErr := make(chan error)
|
|
cl.mu.Lock()
|
|
go func() {
|
|
err := cn.mainReadLoop()
|
|
if err != nil {
|
|
mrlErr <- err
|
|
}
|
|
close(mrlErr)
|
|
}()
|
|
msg := pp.Message{
|
|
Type: pp.Piece,
|
|
Piece: make([]byte, defaultChunkSize),
|
|
}
|
|
wb, err := msg.MarshalBinary()
|
|
require.NoError(b, err)
|
|
b.SetBytes(int64(len(msg.Piece)))
|
|
ts.writeSem.Lock()
|
|
for range iter.N(b.N) {
|
|
cl.mu.Lock()
|
|
t.pieces[0].dirtyChunks.Clear()
|
|
cl.mu.Unlock()
|
|
n, err := w.Write(wb)
|
|
require.NoError(b, err)
|
|
require.EqualValues(b, len(wb), n)
|
|
ts.writeSem.Lock()
|
|
}
|
|
w.Close()
|
|
require.NoError(b, <-mrlErr)
|
|
require.EqualValues(b, b.N, cn.UsefulChunksReceived)
|
|
}
|
|
|
|
func TestConnectionReceiveBadChunkIndex(t *testing.T) {
|
|
cn := connection{
|
|
t: &Torrent{},
|
|
}
|
|
require.False(t, cn.t.haveInfo())
|
|
assert.NotPanics(t, func() { cn.receiveChunk(&pp.Message{}) })
|
|
cn.t.info = &metainfo.Info{}
|
|
require.True(t, cn.t.haveInfo())
|
|
assert.NotPanics(t, func() { cn.receiveChunk(&pp.Message{}) })
|
|
}
|