fixup! [trackerscraper] Add custom DNS lookup function
This commit is contained in:
parent
b33b45baf4
commit
487352fa5b
|
@ -92,7 +92,7 @@ type ClientConfig struct {
|
|||
HTTPProxy func(*http.Request) (*url.URL, error)
|
||||
// Takes a tracker's hostname and requests DNS A and AAAA records.
|
||||
// Used in case DNS lookups require a special setup (i.e., dns-over-https)
|
||||
TrackerIpFetcher func(*url.URL) ([]net.IP, error)
|
||||
LookupTrackerIp func(*url.URL) ([]net.IP, error)
|
||||
// HTTPUserAgent changes default UserAgent for HTTP requests
|
||||
HTTPUserAgent string
|
||||
// Updated occasionally to when there's been some changes to client
|
||||
|
|
|
@ -1577,9 +1577,9 @@ func (t *Torrent) startScrapingTracker(_url string) {
|
|||
}
|
||||
}
|
||||
newAnnouncer := &trackerScraper{
|
||||
u: *u,
|
||||
t: t,
|
||||
ipFetcher: t.cl.config.TrackerIpFetcher,
|
||||
u: *u,
|
||||
t: t,
|
||||
lookupTrackerIp: t.cl.config.LookupTrackerIp,
|
||||
}
|
||||
go newAnnouncer.Run()
|
||||
return newAnnouncer
|
||||
|
|
|
@ -18,10 +18,10 @@ import (
|
|||
// Announces a torrent to a tracker at regular intervals, when peers are
|
||||
// required.
|
||||
type trackerScraper struct {
|
||||
u url.URL
|
||||
t *Torrent
|
||||
lastAnnounce trackerAnnounceResult
|
||||
ipFetcher func(*url.URL) ([]net.IP, error)
|
||||
u url.URL
|
||||
t *Torrent
|
||||
lastAnnounce trackerAnnounceResult
|
||||
lookupTrackerIp func(*url.URL) ([]net.IP, error)
|
||||
}
|
||||
|
||||
type torrentTrackerAnnouncer interface {
|
||||
|
@ -68,8 +68,8 @@ type trackerAnnounceResult struct {
|
|||
|
||||
func (me *trackerScraper) getIp() (ip net.IP, err error) {
|
||||
var ips []net.IP
|
||||
if me.ipFetcher != nil {
|
||||
ips, err = me.ipFetcher(&me.u)
|
||||
if me.lookupTrackerIp != nil {
|
||||
ips, err = me.lookupTrackerIp(&me.u)
|
||||
} else {
|
||||
// Do a regular dns lookup
|
||||
ips, err = net.LookupIP(me.u.Hostname())
|
||||
|
|
Loading…
Reference in New Issue