From dad28e674e2b11e4091618876a4575b6c87013bd Mon Sep 17 00:00:00 2001
From: Yaroslav Kolomiiets <yarikos@gmail.com>
Date: Thu, 16 Apr 2020 13:17:10 +0100
Subject: [PATCH] cmd/torrent: add -ipv4/-ipv6 options

---
 cmd/torrent/main.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/cmd/torrent/main.go b/cmd/torrent/main.go
index 4ec5a849..a346cdfb 100644
--- 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