Merge pull request #743 from projectdiscovery/feat-376-dns-proxy

Adding proxy
projectdiscovery · Nov 11, 2024 · aaf7d90 · aaf7d90
2 parents 89f443f + 3698edc
commit aaf7d90
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/internal/runner/options.go b/internal/runner/options.go
@@ -72,6 +72,7 @@ type Options struct {
 	HealthCheck        bool
 	DisableUpdateCheck bool
 	PdcpAuth           string
+	Proxy              string
 }
 
 // ShouldLoadResume resume file
@@ -180,6 +181,7 @@ func ParseOptions() *Options {
 		flagSet.StringVarP(&options.Resolvers, "resolver", "r", "", "list of resolvers to use (file or comma separated)"),
 		flagSet.IntVarP(&options.WildcardThreshold, "wildcard-threshold", "wt", 5, "wildcard filter threshold"),
 		flagSet.StringVarP(&options.WildcardDomain, "wildcard-domain", "wd", "", "domain name for wildcard filtering (other flags will be ignored - only json output is supported)"),
+		flagSet.StringVar(&options.Proxy, "proxy", "", "proxy to use (eg socks5://127.0.0.1:8080)"),
 	)
 
 	_ = flagSet.Parse()

diff --git a/internal/runner/runner.go b/internal/runner/runner.go
@@ -57,6 +57,7 @@ func New(options *Options) (*Runner, error) {
 	dnsxOptions.TraceMaxRecursion = options.TraceMaxRecursion
 	dnsxOptions.Hostsfile = options.HostsFile
 	dnsxOptions.OutputCDN = options.OutputCDN
+	dnsxOptions.Proxy = options.Proxy
 	if options.Resolvers != "" {
 		dnsxOptions.BaseResolvers = []string{}
 		// If it's a file load resolvers from it

diff --git a/libs/dnsx/dnsx.go b/libs/dnsx/dnsx.go
@@ -4,7 +4,6 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
-	"math"
 
 	miekgdns "github.com/miekg/dns"
 	"github.com/projectdiscovery/cdncheck"
@@ -30,6 +29,7 @@ type Options struct {
 	Hostsfile         bool
 	OutputCDN         bool
 	QueryAll          bool
+	Proxy             string
 }
 
 // ResponseData to show output result
@@ -72,7 +72,7 @@ var DefaultOptions = Options{
 	BaseResolvers:     DefaultResolvers,
 	MaxRetries:        5,
 	QuestionTypes:     []uint16{miekgdns.TypeA},
-	TraceMaxRecursion: math.MaxUint16,
+	TraceMaxRecursion: 255,
 	Hostsfile:         true,
 }
 
@@ -94,6 +94,7 @@ func New(options Options) (*DNSX, error) {
 		BaseResolvers: options.BaseResolvers,
 		MaxRetries:    options.MaxRetries,
 		Hostsfile:     options.Hostsfile,
+		Proxy:         options.Proxy,
 	}
 
 	dnsClient, err := retryabledns.NewWithOptions(retryablednsOptions)