Add option to follow redirects (#13)

This commit is contained in:
Sebastian Lawniczak 2019-04-03 11:54:32 +02:00 committed by Joona Hoikkala
parent 9934cfdfc3
commit 53361352aa
3 changed files with 26 additions and 19 deletions

View file

@ -70,6 +70,7 @@ func main() {
flag.StringVar(&opts.outputFormat, "of", "json", "Output file format. Available formats: json, csv, ecsv")
flag.BoolVar(&conf.Quiet, "s", false, "Do not print additional information (silent mode)")
flag.BoolVar(&conf.StopOn403, "sf", false, "Stop when > 90% of responses return 403 Forbidden")
flag.BoolVar(&conf.FollowRedirects, "r", false, "Follow redirects")
flag.IntVar(&conf.Threads, "t", 40, "Number of concurrent threads.")
flag.BoolVar(&opts.showVersion, "V", false, "Show version information.")
flag.Parse()

View file

@ -28,6 +28,7 @@ type Config struct {
OutputFile string
OutputFormat string
StopOn403 bool
FollowRedirects bool
Delay optRange
Filters []FilterProvider
Matchers []FilterProvider
@ -48,6 +49,7 @@ func NewConfig(ctx context.Context) Config {
conf.Data = ""
conf.Quiet = false
conf.StopOn403 = false
conf.FollowRedirects = false
conf.ProxyURL = http.ProxyFromEnvironment
conf.Filters = make([]FilterProvider, 0)
conf.Delay = optRange{0, 0, false, false}

View file

@ -38,6 +38,10 @@ func NewSimpleRunner(conf *ffuf.Config) ffuf.RunnerProvider {
InsecureSkipVerify: conf.TLSSkipVerify,
},
}}
if conf.FollowRedirects {
simplerunner.client.CheckRedirect = nil
}
return &simplerunner
}