63 lines
1.7 KiB
Go
63 lines
1.7 KiB
Go
// Copyright 2015 The Chihaya Authors. All rights reserved.
|
|
// Use of this source code is governed by the BSD 2-Clause license,
|
|
// which can be found in the LICENSE file.
|
|
|
|
package udp
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"time"
|
|
|
|
"github.com/chihaya/chihaya/tracker/models"
|
|
)
|
|
|
|
// Writer implements the tracker.Writer interface for the UDP protocol.
|
|
type Writer struct {
|
|
buf *bytes.Buffer
|
|
|
|
connectionID []byte
|
|
transactionID []byte
|
|
}
|
|
|
|
// WriteError writes the failure reason as a null-terminated string.
|
|
func (w *Writer) WriteError(err error) error {
|
|
w.writeHeader(3)
|
|
w.buf.WriteString(err.Error())
|
|
w.buf.WriteRune('\000')
|
|
return nil
|
|
}
|
|
|
|
// WriteAnnounce encodes an announce response according to the UDP spec.
|
|
func (w *Writer) WriteAnnounce(res *models.AnnounceResponse) error {
|
|
w.writeHeader(1)
|
|
binary.Write(w.buf, binary.BigEndian, uint32(res.Interval/time.Second))
|
|
binary.Write(w.buf, binary.BigEndian, uint32(res.Incomplete))
|
|
binary.Write(w.buf, binary.BigEndian, uint32(res.Complete))
|
|
|
|
for _, peer := range res.IPv4Peers {
|
|
w.buf.Write(peer.IP)
|
|
binary.Write(w.buf, binary.BigEndian, peer.Port)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// WriteAnnounce encodes a scrape response according to the UDP spec.
|
|
func (w *Writer) WriteScrape(res *models.ScrapeResponse) error {
|
|
w.writeHeader(2)
|
|
|
|
for _, torrent := range res.Files {
|
|
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.Leechers.Len()))
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// writeHeader writes the action and transaction ID to the response.
|
|
func (w *Writer) writeHeader(action uint32) {
|
|
binary.Write(w.buf, binary.BigEndian, action)
|
|
w.buf.Write(w.transactionID)
|
|
}
|