add cli flag to disable DHT

This partially addresses #3.

diff --git a/cmd/torrent/main.go b/cmd/torrent/main.go
index 411e1d2767420419180d0a5ba02c0f7d0fea8d30..39246d062cbf1f2c0074fac767c235a76c61279f 100644 (file)
--- a/cmd/torrent/main.go
+++ b/cmd/torrent/main.go
@@ -23,6 +23,7 @@ var (
        // TODO: Check the default torrent listen port.
        listenAddr      = flag.String("listenAddr", ":50007", "incoming connection address")
        disableTrackers = flag.Bool("disableTrackers", false, "disable trackers")
+       disableDHT      = flag.Bool("disableDHT", false, "disable DHT")
        seed            = flag.Bool("seed", false, "seed after downloading")
        upload          = flag.Bool("upload", true, "upload data to peers")
 )
@@ -43,6 +44,7 @@ func main() {
                DisableTrackers: *disableTrackers,
                ListenAddr:      *listenAddr,
                NoUpload:        !*upload,
+               NoDHT:           *disableDHT,
        })
        if err != nil {
                log.Fatalf("error creating client: %s", err)