aboutsummaryrefslogtreecommitdiff
path: root/cmd/btcheck/main.go
blob: 6d79e8be9e8617c8e070155239c2bdaa074e9bc7 (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
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)
	for i, p := range tor.Info.Pieces {
		off := tor.Info.Offset(i)
		n, err := tor.Info.ReadAt(buf, off)
		if err != nil {
			log.Fatal(err)
		}
		fmt.Printf("%d %d %d %x %v\n", i, off, n, p, p.Check(buf[:n]))
	}

	id, _ := peer.NewID()
	req := tracker.Request{
		InfoHash: tor.Info.Hash(),
		PeerID:   []byte(id),
		Compact:  true,
		Port:     6881,
	}
	pool := make(map[string]tracker.Peer)
	for p := range req.Poll(tor.Announce) {
		addr := p.Addr()
		if _, ok := pool[addr]; !ok {
			log.Println("new peer", p)
			pool[addr] = p
		}
	}
}