forked from anacrolix/torrent
/
tracker_scraper.go
77 lines (68 loc) · 1.79 KB
/
tracker_scraper.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
package torrent
import (
"log"
"time"
"github.com/anacrolix/missinggo"
"github.com/anacrolix/torrent/tracker"
)
// Announces a torrent to a tracker at regular intervals, when peers are
// required.
type trackerScraper struct {
url string
// Causes the trackerScraper to stop running.
stop missinggo.Event
t *Torrent
}
func trackerToTorrentPeers(ps []tracker.Peer) (ret []Peer) {
ret = make([]Peer, 0, len(ps))
for _, p := range ps {
ret = append(ret, Peer{
IP: p.IP,
Port: p.Port,
Source: peerSourceTracker,
})
}
return
}
// Return how long to wait before trying again. For most errors, we return 5
// minutes, a relatively quick turn around for DNS changes.
func (me *trackerScraper) announce() time.Duration {
blocked, urlToUse, host, err := me.t.cl.prepareTrackerAnnounceUnlocked(me.url)
if err != nil {
log.Printf("error preparing announce to %q: %s", me.url, err)
return 5 * time.Minute
}
if blocked {
log.Printf("announce to tracker %q blocked by IP", me.url)
return 5 * time.Minute
}
me.t.cl.mu.Lock()
req := me.t.announceRequest()
me.t.cl.mu.Unlock()
res, err := tracker.AnnounceHost(urlToUse, &req, host)
if err != nil {
log.Printf("error announcing %s %q to %q: %s", me.t.InfoHash().HexString(), me.t.Name(), me.url, err)
return 5 * time.Minute
}
me.t.AddPeers(trackerToTorrentPeers(res.Peers))
return time.Duration(res.Interval) * time.Second
}
func (me *trackerScraper) Run() {
for {
select {
case <-me.t.closed.LockedChan(&me.t.cl.mu):
return
case <-me.stop.LockedChan(&me.t.cl.mu):
return
case <-me.t.wantPeersEvent.LockedChan(&me.t.cl.mu):
}
intervalChan := time.After(me.announce())
select {
case <-me.t.closed.LockedChan(&me.t.cl.mu):
return
case <-me.stop.LockedChan(&me.t.cl.mu):
return
case <-intervalChan:
}
}
}