LBRYCommunity/tracker
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity

udp: implement BEP45 response

Browse source
This commit is contained in:
Jimmy Zelinskie 2015-04-04 15:41:58 -04:00
parent 938542ac74
commit d317dfc069
2 changed files with 45 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
udp/protocol.go
View file

@ -18,6 +18,7 @@ const (
announceActionID announceActionID
scrapeActionID scrapeActionID
errorActionID errorActionID
announceDualStackActionID
) )
var ( var (

54
udp/writer.go
View file

@ -28,14 +28,33 @@ func (w *Writer) WriteError(err error) error {
return nil return nil
} }
// WriteAnnounce encodes an announce response according to the UDP spec. // WriteAnnounce encodes an announce response by selecting the proper announce
func (w *Writer) WriteAnnounce(res *models.AnnounceResponse) error { // format based on the BitTorrent spec.
w.writeHeader(announceActionID) func (w *Writer) WriteAnnounce(resp *models.AnnounceResponse) (err error) {
binary.Write(w.buf, binary.BigEndian, uint32(res.Interval/time.Second)) if resp.Announce.HasIPv6() {
binary.Write(w.buf, binary.BigEndian, uint32(res.Incomplete)) err = w.WriteAnnounceIPv6(resp)
binary.Write(w.buf, binary.BigEndian, uint32(res.Complete)) } else {
err = w.WriteAnnounceIPv4(resp)
}
for _, peer := range res.IPv4Peers { return
}
// WriteAnnounceIPv6 encodes an announce response according to BEP45.
func (w *Writer) WriteAnnounceIPv6(resp *models.AnnounceResponse) error {
w.writeHeader(announceDualStackActionID)
binary.Write(w.buf, binary.BigEndian, uint32(resp.Interval/time.Second))
binary.Write(w.buf, binary.BigEndian, uint32(resp.Incomplete))
binary.Write(w.buf, binary.BigEndian, uint32(resp.Complete))
binary.Write(w.buf, binary.BigEndian, uint32(len(resp.IPv4Peers)))
binary.Write(w.buf, binary.BigEndian, uint32(len(resp.IPv6Peers)))
for _, peer := range resp.IPv4Peers {
w.buf.Write(peer.IP)
binary.Write(w.buf, binary.BigEndian, peer.Port)
}
for _, peer := range resp.IPv6Peers {
w.buf.Write(peer.IP) w.buf.Write(peer.IP)
binary.Write(w.buf, binary.BigEndian, peer.Port) binary.Write(w.buf, binary.BigEndian, peer.Port)
} }
@ -43,11 +62,26 @@ func (w *Writer) WriteAnnounce(res *models.AnnounceResponse) error {
return nil return nil
} }
// WriteScrape encodes a scrape response according to the UDP spec. // WriteAnnounceIPv4 encodes an announce response according to BEP15.
func (w *Writer) WriteScrape(res *models.ScrapeResponse) error { func (w *Writer) WriteAnnounceIPv4(resp *models.AnnounceResponse) error {
w.writeHeader(announceActionID)
binary.Write(w.buf, binary.BigEndian, uint32(resp.Interval/time.Second))
binary.Write(w.buf, binary.BigEndian, uint32(resp.Incomplete))
binary.Write(w.buf, binary.BigEndian, uint32(resp.Complete))
for _, peer := range resp.IPv4Peers {
w.buf.Write(peer.IP)
binary.Write(w.buf, binary.BigEndian, peer.Port)
}
return nil
}
// WriteScrape encodes a scrape response according to BEP15.
func (w *Writer) WriteScrape(resp *models.ScrapeResponse) error {
w.writeHeader(scrapeActionID) w.writeHeader(scrapeActionID)
for _, torrent := range res.Files { for _, torrent := range resp.Files {
binary.Write(w.buf, binary.BigEndian, uint32(torrent.Seeders.Len())) binary.Write(w.buf, binary.BigEndian, uint32(torrent.Seeders.Len()))
binary.Write(w.buf, binary.BigEndian, uint32(torrent.Snatches)) binary.Write(w.buf, binary.BigEndian, uint32(torrent.Snatches))
binary.Write(w.buf, binary.BigEndian, uint32(torrent.Leechers.Len())) binary.Write(w.buf, binary.BigEndian, uint32(torrent.Leechers.Len()))