cmd/btrtrc/main.go

   1 package main
   2
   3 import (
   4         "flag"
   5         "log"
   6         "net"
   7         "os"
   8         "os/signal"
   9         "strings"
  10         "syscall"
  11
  12         "github.com/anacrolix/dht/v2"
  13         analog "github.com/anacrolix/log"
  14
  15         "github.com/anacrolix/torrent"
  16         "github.com/anacrolix/torrent/storage"
  17 )
  18
  19 const TorrentExt = ".torrent"
  20
  21 func main() {
  22         log.SetFlags(log.Ldate | log.Ltime)
  23         dhtBoot := flag.String("dht", "dht.cypherpunks.ru:8991", "Comma-separated list of DHT bootstrap nodes")
  24         addr := flag.String("bind", "[::]:6881", "Address to bind to")
  25         pub4 := flag.String("4", "", "External IPv4 address")
  26         pub6 := flag.String("6", "", "External IPv6 address")
  27         debug := flag.Bool("debug", false, "Enable debug messages")
  28         noDHT := flag.Bool("nodht", false, "Disable DHT")
  29         verify := flag.Bool("verify", false, "Force verification of provided torrents")
  30         flag.Parse()
  31
  32         dht.DefaultGlobalBootstrapHostPorts = strings.Split(*dhtBoot, ",")
  33         cc := torrent.NewDefaultClientConfig()
  34         cc.Debug = *debug
  35         cc.DisableAcceptRateLimiting = true
  36         cc.NoDefaultPortForwarding = true
  37         cc.DisableWebtorrent = true
  38         cc.Logger = analog.Default.WithNames("main", "client")
  39         cc.DefaultStorage = storage.NewFileWithCompletion(".", NewBFPieceCompletion())
  40         if *verify {
  41                 doVerify(cc, flag.Args())
  42                 return
  43         }
  44         cc.Seed = true
  45         if *pub4 == "" {
  46                 cc.DisableIPv4 = true
  47         } else {
  48                 cc.PublicIp4 = net.ParseIP(*pub4).To4()
  49         }
  50         if *pub6 == "" {
  51                 cc.DisableIPv6 = true
  52         } else {
  53                 cc.PublicIp6 = net.ParseIP(*pub6).To16()
  54         }
  55         cc.NoDHT = *noDHT
  56         cc.SetListenAddr(*addr)
  57         client, err := torrent.NewClient(cc)
  58         if err != nil {
  59                 log.Fatalln("torrent.NewClient:", err)
  60         }
  61         defer client.Close()
  62
  63         needsShutdown := make(chan os.Signal)
  64         overallStatusCancel := make(chan struct{})
  65         txStatsDumperCancel := make(chan struct{})
  66         signal.Notify(needsShutdown, syscall.SIGTERM, syscall.SIGINT)
  67         go func() {
  68                 <-needsShutdown
  69                 overallStatusCancel <- struct{}{}
  70                 txStatsDumperCancel <- struct{}{}
  71                 client.Close()
  72         }()
  73
  74         fifosPrepare()
  75         log.Println("started", client.PublicIPs())
  76         go overallStatus(client, overallStatusCancel)
  77         go fifoList(client)
  78         go fifoDHTList(client)
  79         go fifoAdd(client)
  80         go fifoDel(client)
  81         go txStatsDumper(client, txStatsDumperCancel)
  82         <-client.Closed()
  83         <-overallStatusCancel
  84         <-txStatsDumperCancel
  85         fifosCleanup()
  86         log.Println("finished")
  87 }