-
Notifications
You must be signed in to change notification settings - Fork 0
/
tracker.go
126 lines (108 loc) · 2.49 KB
/
tracker.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
package main
import (
"fmt"
"io/ioutil"
"net"
"net/http"
"net/url"
"time"
)
type Tracker struct {
Url string
Metainfo *Metainfo
TrackerResponse *TrackerResponse
}
func NewTracker(url string, metainfo *Metainfo) *Tracker {
return &Tracker{
Url: url,
Metainfo: metainfo,
TrackerResponse: &TrackerResponse{
State: TrackerResponseStateNotSent,
},
}
}
func (t *Tracker) SendRequest() (err error) {
responseBody, err := t.httpGetPeerRequest()
if err != nil {
return
}
t.TrackerResponse, err = NewTrackerResponse(responseBody)
return
}
func (t *Tracker) urlWithParams() string {
params := t.generateParams()
return t.Url + params
}
func (t *Tracker) httpGetPeerRequest() (responseBody []byte, err error) {
//fmt.Printf("GET %v\n", t.urlWithParams())
transport := http.Transport{
Dial: dialTimeout,
}
client := http.Client{
Transport: &transport,
}
resp, err := client.Get(t.urlWithParams())
if err != nil {
return
}
defer resp.Body.Close()
responseBody, err = ioutil.ReadAll(resp.Body)
if err != nil {
return
}
return
}
func (t *Tracker) generateParams() (urlParams string) {
paramDict := map[string]string{}
paramDict["info_hash"] = t.infoHashParam()
paramDict["peer_id"] = t.peerIdParam()
paramDict["port"] = t.portParam()
paramDict["uploaded"] = t.uploadedParam()
paramDict["downloaded"] = t.downloadedParam()
paramDict["left"] = t.leftParam()
paramDict["compact"] = t.compactParam()
paramDict["no_peer_id"] = t.noPeerIdParam()
paramDict["event"] = t.eventParam()
for key, value := range paramDict {
if urlParams == "" {
urlParams += "?"
} else {
urlParams += "&"
}
urlParams += key + "=" + value
}
return
}
func (t *Tracker) infoHashParam() string {
hash := string(t.Metainfo.InfoDictionary.Hash)
return url.QueryEscape(hash)
}
func (t *Tracker) peerIdParam() string {
return "15620985492012023883"
}
func (t *Tracker) leftParam() string {
// TODO make this dynamic
return fmt.Sprintf("%v", t.Metainfo.InfoDictionary.Length())
}
func (t *Tracker) portParam() string {
return "6881"
}
func (t *Tracker) uploadedParam() string {
return "0"
}
func (t *Tracker) downloadedParam() string {
return "0"
}
func (t *Tracker) compactParam() string {
return "1"
}
func (t *Tracker) noPeerIdParam() string {
return "0"
}
func (t *Tracker) eventParam() string {
return "started"
}
func dialTimeout(network, addr string) (net.Conn, error) {
var timeout = time.Duration(2 * time.Second)
return net.DialTimeout(network, addr, timeout)
}