cmd/torrent-verify/main.go

   1 package main
   2
   3 import (
   4         "bytes"
   5         "crypto/sha1"
   6         "fmt"
   7         "io"
   8         "log"
   9         "os"
  10         "path/filepath"
  11
  12         "github.com/anacrolix/tagflag"
  13         "github.com/edsrzf/mmap-go"
  14
  15         "github.com/anacrolix/torrent/metainfo"
  16         "github.com/anacrolix/torrent/mmap_span"
  17 )
  18
  19 func mmapFile(name string) (mm mmap.MMap, err error) {
  20         f, err := os.Open(name)
  21         if err != nil {
  22                 return
  23         }
  24         defer f.Close()
  25         fi, err := f.Stat()
  26         if err != nil {
  27                 return
  28         }
  29         if fi.Size() == 0 {
  30                 return
  31         }
  32         return mmap.MapRegion(f, -1, mmap.RDONLY, mmap.COPY, 0)
  33 }
  34
  35 func verifyTorrent(info *metainfo.Info, root string) error {
  36         span := new(mmap_span.MMapSpan)
  37         for _, file := range info.UpvertedFiles() {
  38                 filename := filepath.Join(append([]string{root, info.Name}, file.Path...)...)
  39                 mm, err := mmapFile(filename)
  40                 if err != nil {
  41                         return err
  42                 }
  43                 if int64(len(mm)) != file.Length {
  44                         return fmt.Errorf("file %q has wrong length", filename)
  45                 }
  46                 span.Append(mm)
  47         }
  48         span.InitIndex()
  49         for i, numPieces := 0, info.NumPieces(); i < numPieces; i += 1 {
  50                 p := info.Piece(i)
  51                 hash := sha1.New()
  52                 _, err := io.Copy(hash, io.NewSectionReader(span, p.Offset(), p.Length()))
  53                 if err != nil {
  54                         return err
  55                 }
  56                 good := bytes.Equal(hash.Sum(nil), p.Hash().Bytes())
  57                 if !good {
  58                         return fmt.Errorf("hash mismatch at piece %d", i)
  59                 }
  60                 fmt.Printf("%d: %v: %v\n", i, p.Hash(), good)
  61         }
  62         return nil
  63 }
  64
  65 func main() {
  66         log.SetFlags(log.Flags() | log.Lshortfile)
  67         flags := struct {
  68                 DataDir string
  69                 tagflag.StartPos
  70                 TorrentFile string
  71         }{}
  72         tagflag.Parse(&flags)
  73         metaInfo, err := metainfo.LoadFromFile(flags.TorrentFile)
  74         if err != nil {
  75                 log.Fatal(err)
  76         }
  77         info, err := metaInfo.UnmarshalInfo()
  78         if err != nil {
  79                 log.Fatalf("error unmarshalling info: %s", err)
  80         }
  81         err = verifyTorrent(&info, flags.DataDir)
  82         if err != nil {
  83                 log.Fatalf("torrent failed verification: %s", err)
  84         }
  85 }