13 "github.com/anacrolix/dht/v2"
14 analog "github.com/anacrolix/log"
16 "github.com/anacrolix/torrent"
17 "github.com/anacrolix/torrent/storage"
20 const TorrentExt = ".torrent"
22 func saveTorrent(t *torrent.Torrent) error {
23 pth := t.InfoHash().HexString() + TorrentExt
24 if _, err := os.Stat(pth); err == nil {
28 t.Metainfo().Write(&b)
29 if err := os.WriteFile(pth, b.Bytes(), 0666); err != nil {
32 return os.Symlink(pth, t.Name()+TorrentExt)
36 log.SetFlags(log.Ldate | log.Ltime)
37 dhtBoot := flag.String("dht", "dht.cypherpunks.ru:8991", "Comma-separated list of DHT bootstrap nodes")
38 addr := flag.String("bind", "[::]:6881", "Address to bind to")
39 pub4 := flag.String("4", "", "External IPv4 address")
40 pub6 := flag.String("6", "", "External IPv6 address")
41 debug := flag.Bool("debug", false, "Enable debug messages")
42 noDHT := flag.Bool("nodht", false, "Disable DHT")
43 verify := flag.Bool("verify", false, "Force verification of provided torrents")
46 dht.DefaultGlobalBootstrapHostPorts = strings.Split(*dhtBoot, ",")
47 cc := torrent.NewDefaultClientConfig()
49 cc.DisableAcceptRateLimiting = true
50 cc.NoDefaultPortForwarding = true
51 cc.DisableWebtorrent = true
52 cc.Logger = analog.Default.WithNames("main", "client")
53 cc.DefaultStorage = storage.NewFileWithCompletion(".", NewBFPieceCompletion())
55 doVerify(cc, flag.Args())
62 cc.PublicIp4 = net.ParseIP(*pub4).To4()
67 cc.PublicIp6 = net.ParseIP(*pub6).To16()
70 cc.SetListenAddr(*addr)
71 client, err := torrent.NewClient(cc)
73 log.Fatalln("torrent.NewClient:", err)
77 needsShutdown := make(chan os.Signal)
78 overallStatusCancel := make(chan struct{})
79 txStatsDumperCancel := make(chan struct{})
80 signal.Notify(needsShutdown, syscall.SIGTERM, syscall.SIGINT)
83 overallStatusCancel <- struct{}{}
84 txStatsDumperCancel <- struct{}{}
89 log.Println("started", client.PublicIPs())
90 go overallStatus(client, overallStatusCancel)
92 go fifoDHTList(client)
95 go txStatsDumper(client, txStatsDumperCancel)
100 log.Println("finished")