48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
package bittorrent
|
|
|
|
import (
|
|
"net"
|
|
|
|
"github.com/chihaya/chihaya/pkg/log"
|
|
)
|
|
|
|
// ErrInvalidIP indicates an invalid IP for an Announce.
|
|
var ErrInvalidIP = ClientError("invalid IP")
|
|
|
|
// SanitizeAnnounce enforces a max and default NumWant and coerces the peer's
|
|
// IP address into the proper format.
|
|
func SanitizeAnnounce(r *AnnounceRequest, maxNumWant, defaultNumWant uint32) error {
|
|
if !r.NumWantProvided {
|
|
r.NumWant = defaultNumWant
|
|
} else if r.NumWant > maxNumWant {
|
|
r.NumWant = maxNumWant
|
|
}
|
|
|
|
if ip := r.Peer.IP.To4(); ip != nil {
|
|
r.Peer.IP.IP = ip
|
|
r.Peer.IP.AddressFamily = IPv4
|
|
} else if len(r.Peer.IP.IP) == net.IPv6len { // implies r.Peer.IP.To4() == nil
|
|
r.Peer.IP.AddressFamily = IPv6
|
|
} else {
|
|
return ErrInvalidIP
|
|
}
|
|
|
|
log.Debug("sanitized announce", r, log.Fields{
|
|
"maxNumWant": maxNumWant,
|
|
"defaultNumWant": defaultNumWant,
|
|
})
|
|
return nil
|
|
}
|
|
|
|
// SanitizeScrape enforces a max number of infohashes for a single scrape
|
|
// request.
|
|
func SanitizeScrape(r *ScrapeRequest, maxScrapeInfoHashes uint32) error {
|
|
if len(r.InfoHashes) > int(maxScrapeInfoHashes) {
|
|
r.InfoHashes = r.InfoHashes[:maxScrapeInfoHashes]
|
|
}
|
|
|
|
log.Debug("sanitized scrape", r, log.Fields{
|
|
"maxScrapeInfoHashes": maxScrapeInfoHashes,
|
|
})
|
|
return nil
|
|
}
|