cmd/torrent: add -ipv4/-ipv6 options

diff --git a/cmd/torrent/main.go b/cmd/torrent/main.go
index 4ec5a8493f4fc130afbc7d662ed72e746982b12c..a346cdfbc8ffe91bdaa8580a0e1f7bba0e248479 100644 (file)
--- a/cmd/torrent/main.go
+++ b/cmd/torrent/main.go
@@ -141,6 +141,8 @@ var flags = struct {
        Dht             bool
        TcpPeers        bool
        UtpPeers        bool
+       Ipv4            bool
+       Ipv6            bool
        tagflag.StartPos
        Torrent []string `arity:"+" help:"torrent file path or magnet uri"`
 }{
@@ -150,6 +152,8 @@ var flags = struct {
        Dht:          true,
        TcpPeers:     true,
        UtpPeers:     true,
+       Ipv4:         true,
+       Ipv6:         true,
 }
 
 func stdoutAndStderrAreSameFile() bool {
@@ -204,6 +208,8 @@ func downloadErr(args []string, parent *tagflag.Parser) error {
        clientConfig := torrent.NewDefaultClientConfig()
        clientConfig.DisableTCP = !flags.TcpPeers
        clientConfig.DisableUTP = !flags.UtpPeers
+       clientConfig.DisableIPv4 = !flags.Ipv4
+       clientConfig.DisableIPv6 = !flags.Ipv6
        clientConfig.DisableAcceptRateLimiting = true
        clientConfig.NoDHT = !flags.Dht
        clientConfig.Debug = flags.Debug