Add udp tracker scrape support

[btrtrc.git] / tracker / udp / protocol.go

diff --git a/tracker/udp/protocol.go b/tracker/udp/protocol.go
index 365d3c5ca59a996b140d618b2ebca2e38f9c11f8..4a8dc6681512f14710023f5992647b001b124ec4 100644 (file)
--- a/tracker/udp/protocol.go
+++ b/tracker/udp/protocol.go
@@ -53,6 +53,8 @@ type AnnounceResponseHeader struct {
        Seeders  int32
 }
 
+type InfoHash = [20]byte
+
 func marshal(data interface{}) (b []byte, err error) {
        var buf bytes.Buffer
        err = binary.Write(&buf, binary.BigEndian, data)
@@ -60,6 +62,14 @@ func marshal(data interface{}) (b []byte, err error) {
        return
 }
 
+func mustMarshal(data interface{}) []byte {
+       b, err := marshal(data)
+       if err != nil {
+               panic(err)
+       }
+       return b
+}
+
 func Write(w io.Writer, data interface{}) error {
        return binary.Write(w, binary.BigEndian, data)
 }