aboutsummaryrefslogtreecommitdiff
path: root/cmd/btcheck/main.go
blob: 24e454075d4a4a515b4bfe4bef1850014b09b2fc (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
package main

import (
	"fmt"
	"io/ioutil"
	"log"
	"os"

	"dim13.org/btget/bencode"
	"dim13.org/btget/meta"
)

func Open(fname string) (meta.Torrent, error) {
	var tor meta.Torrent
	body, err := ioutil.ReadFile(fname)
	if err != nil {
		return meta.Torrent{}, err
	}
	err = bencode.Unmarshal(body, &tor)
	if err != nil {
		return meta.Torrent{}, err
	}
	return tor, nil
}

func main() {
	if len(os.Args) != 2 {
		log.Fatal("usage: ", os.Args[0], " <torrent>")
	}
	tor, err := Open(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.Printf("%6.2f%%\n", pieces.Good())
}