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
|
package main
import (
"encoding/json"
"fmt"
"os"
"strings"
bencode "github.com/jackpal/bencode-go" // Available if you need it!
)
func createClient() *Client {
fn := os.Args[2]
c := NewClient("00112233445566778899", 6881)
_, err := c.AddTorrentFile(fn)
if err != nil {
panic(err)
}
return c
}
func main() {
command := os.Args[1]
switch command {
case "decode":
bencodedValue := os.Args[2]
decoded, err := bencode.Decode(strings.NewReader(bencodedValue))
if err != nil {
fmt.Println(err)
return
}
jsonOutput, _ := json.Marshal(decoded)
fmt.Println(string(jsonOutput))
case "info":
fn := os.Args[2]
c := createClient()
meta := c.Torrents[fn].Meta
fmt.Printf("Tracker URL: %s\n", meta.Announce)
fmt.Printf("Length: %d\n", meta.Info.Length)
infoHash, err := meta.InfoHash()
if err != nil {
panic(err)
}
fmt.Printf("Info Hash: %x", infoHash)
fmt.Printf("Piece Length: %d\n", meta.Info.PieceLength)
fmt.Println("Piece Hashes:")
for _, h := range meta.PieceHashes() {
fmt.Printf("%x\n", h)
}
case "peers":
fn := os.Args[2]
c := createClient()
pr, err := c.GetPeers(fn)
if err != nil {
panic(err)
}
for _, peer := range pr.Peers {
fmt.Println(peer)
}
case "handshake":
fn := os.Args[2]
c := createClient()
peerAddr := os.Args[3]
hs, err := c.Handshake(fn, peerAddr)
if err != nil {
panic(err)
}
fmt.Printf("Peer ID: %s\n", hs.PeerIdHex())
default:
fmt.Println("Unknown command: " + command)
os.Exit(1)
}
}
|