101 lines
2.6 KiB
Go
101 lines
2.6 KiB
Go
package torrent
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/anacrolix/torrent/common"
|
|
"github.com/anacrolix/torrent/metainfo"
|
|
pp "github.com/anacrolix/torrent/peer_protocol"
|
|
"github.com/anacrolix/torrent/segments"
|
|
"github.com/anacrolix/torrent/webseed"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type webseedPeer struct {
|
|
client webseed.Client
|
|
requests map[Request]webseed.Request
|
|
peer Peer
|
|
}
|
|
|
|
var _ peerImpl = (*webseedPeer)(nil)
|
|
|
|
func (me *webseedPeer) connStatusString() string {
|
|
return me.client.Url
|
|
}
|
|
|
|
func (ws *webseedPeer) String() string {
|
|
return fmt.Sprintf("webseed peer for %q", ws.client.Url)
|
|
}
|
|
|
|
func (ws *webseedPeer) onGotInfo(info *metainfo.Info) {
|
|
ws.client.FileIndex = segments.NewIndex(common.LengthIterFromUpvertedFiles(info.UpvertedFiles()))
|
|
ws.client.Info = info
|
|
}
|
|
|
|
func (ws *webseedPeer) _postCancel(r Request) {
|
|
ws.cancel(r)
|
|
}
|
|
|
|
func (ws *webseedPeer) writeInterested(interested bool) bool {
|
|
return true
|
|
}
|
|
|
|
func (ws *webseedPeer) cancel(r Request) bool {
|
|
ws.requests[r].Cancel()
|
|
return true
|
|
}
|
|
|
|
func (ws *webseedPeer) intoSpec(r Request) webseed.RequestSpec {
|
|
return webseed.RequestSpec{ws.peer.t.requestOffset(r), int64(r.Length)}
|
|
}
|
|
|
|
func (ws *webseedPeer) request(r Request) bool {
|
|
webseedRequest := ws.client.NewRequest(ws.intoSpec(r))
|
|
ws.requests[r] = webseedRequest
|
|
go ws.requestResultHandler(r, webseedRequest)
|
|
return true
|
|
}
|
|
|
|
func (ws *webseedPeer) connectionFlags() string {
|
|
return "WS"
|
|
}
|
|
|
|
// TODO: This is called when banning peers. Perhaps we want to be able to ban webseeds too. We could
|
|
// return bool if this is even possible, and if it isn't, skip to the next drop candidate.
|
|
func (ws *webseedPeer) drop() {}
|
|
|
|
func (ws *webseedPeer) updateRequests() {
|
|
ws.peer.doRequestState()
|
|
}
|
|
|
|
func (ws *webseedPeer) onClose() {}
|
|
|
|
func (ws *webseedPeer) requestResultHandler(r Request, webseedRequest webseed.Request) {
|
|
result := <-webseedRequest.Result
|
|
ws.peer.t.cl.lock()
|
|
defer ws.peer.t.cl.unlock()
|
|
if result.Err != nil {
|
|
ws.peer.logger.Printf("Request %v rejected: %v", r, result.Err)
|
|
// Always close for now. We need to filter out temporary errors, but this is a nightmare in
|
|
// Go. Currently a bad webseed URL can starve out the good ones due to the chunk selection
|
|
// algorithm.
|
|
const closeOnAllErrors = false
|
|
if closeOnAllErrors || strings.Contains(errors.Cause(result.Err).Error(), "unsupported protocol scheme") {
|
|
ws.peer.close()
|
|
} else {
|
|
ws.peer.remoteRejectedRequest(r)
|
|
}
|
|
} else {
|
|
err := ws.peer.receiveChunk(&pp.Message{
|
|
Type: pp.Piece,
|
|
Index: r.Index,
|
|
Begin: r.Begin,
|
|
Piece: result.Bytes,
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|