aboutsummaryrefslogtreecommitdiff
path: root/cmd/btcheck/main.go
blob: 0bbd91dbcee429cbf5187910996d085157700634 (plain)
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
package main

import (
	"fmt"
	"log"
	"os"

	"dim13.org/btget/meta"
	"dim13.org/btget/peer"
	"dim13.org/btget/tracker"
)

func main() {
	if len(os.Args) != 2 {
		log.Fatal("usage: ", os.Args[0], " <torrent>")
	}
	tor, err := meta.New(os.Args[1])
	if err != nil {
		log.Fatal(err)
	}
	buf := make([]byte, tor.Info.PieceLength)
	pieces := tor.Info.GetPieces()
	for i, p := range pieces {
		n, err := tor.Info.ReadAt(buf, p.Offset)
		if err != nil {
			log.Fatal(err)
		}
		fmt.Println(i, p.Offset, n, p.Check(buf[:n]))
		pieces[i] = p
	}
	fmt.Println("good pieces:", pieces.Good())
	fmt.Println("bit field:", pieces.BitField())

	id, _ := peer.NewID()
	req := tracker.Request{
		InfoHash: tor.Info.Hash(),
		PeerID:   []byte(id),
		Compact:  true,
		Port:     6881,
	}
	resp, err := req.Get(tor.Announce)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println(resp.Interval)
	fmt.Println(resp.Peers)
}