1 package httpTrackerServer
11 "github.com/anacrolix/dht/v2/krpc"
12 "github.com/anacrolix/generics"
13 "github.com/anacrolix/log"
15 "github.com/anacrolix/torrent/bencode"
16 "github.com/anacrolix/torrent/tracker"
17 httpTracker "github.com/anacrolix/torrent/tracker/http"
21 Announce tracker.AnnounceHandler
22 // Called to derive an announcer's IP if non-nil. If not specified, the Request.RemoteAddr is
23 // used. Necessary for instances running behind reverse proxies for example.
24 RequestHost func(r *http.Request) (netip.Addr, error)
27 func unmarshalQueryKeyToArray(w http.ResponseWriter, key string, query url.Values) (ret [20]byte, ok bool) {
29 if len(str) != len(ret) {
30 http.Error(w, fmt.Sprintf("%v has wrong length", key), http.StatusBadRequest)
38 // Returns false if there was an error and it was served.
39 func (me Handler) requestHostAddr(r *http.Request) (_ netip.Addr, err error) {
40 if me.RequestHost != nil {
41 return me.RequestHost(r)
43 host, _, err := net.SplitHostPort(r.RemoteAddr)
47 return netip.ParseAddr(host)
50 var requestHeadersLogger = log.Default.WithNames("request", "headers")
52 func (me Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
54 var event tracker.AnnounceEvent
55 err := event.UnmarshalText([]byte(vs.Get("event")))
57 http.Error(w, err.Error(), http.StatusBadRequest)
60 infoHash, ok := unmarshalQueryKeyToArray(w, "info_hash", vs)
64 peerId, ok := unmarshalQueryKeyToArray(w, "peer_id", vs)
68 requestHeadersLogger.Levelf(log.Debug, "request RemoteAddr=%q, header=%q", r.RemoteAddr, r.Header)
69 addr, err := me.requestHostAddr(r)
71 log.Printf("error getting requester IP: %v", err)
72 http.Error(w, "error determining your IP", http.StatusBadGateway)
75 portU64, err := strconv.ParseUint(vs.Get("port"), 0, 16)
76 addrPort := netip.AddrPortFrom(addr, uint16(portU64))
77 peers, err := me.Announce.Serve(r.Context(), tracker.AnnounceRequest{
81 Port: addrPort.Port(),
83 }, addrPort, tracker.GetPeersOpts{generics.Some[uint](200)})
85 log.Printf("error serving announce: %v", err)
86 http.Error(w, "error handling announce", http.StatusInternalServerError)
89 var resp httpTracker.HttpResponse
90 resp.Interval = 5 * 60
91 resp.Peers.Compact = true
92 for _, peer := range peers {
93 if peer.Addr().Is4() {
94 resp.Peers.List = append(resp.Peers.List, tracker.Peer{
95 IP: peer.Addr().AsSlice(),
96 Port: int(peer.Port()),
98 } else if peer.Addr().Is6() {
99 resp.Peers6 = append(resp.Peers6, krpc.NodeAddr{
100 IP: peer.Addr().AsSlice(),
101 Port: int(peer.Port()),
105 err = bencode.NewEncoder(w).Encode(resp)
107 log.Printf("error encoding and writing response body: %v", err)