diff --git a/internal/check.go b/internal/check.go new file mode 100644 index 0000000..5bf0569 --- /dev/null +++ b/internal/check.go @@ -0,0 +1 @@ +package internal diff --git a/internal/option.go b/internal/option.go index 4ee5192..6a79024 100644 --- a/internal/option.go +++ b/internal/option.go @@ -52,6 +52,7 @@ type RequestOptions struct { Cookie string `long:"cookie"` Force bool `long:"force"` SimhashDistance int `long:"distance"` + CheckOnly bool `long:"--check-only"` } type MiscOptions struct { @@ -72,19 +73,20 @@ func (opt *Option) PrepareRunner() (*Runner, error) { } var err error r := &Runner{ - Progress: uiprogress.New(), - Threads: opt.Threads, - PoolSize: opt.PoolSize, - Mod: opt.Mod, - Timeout: opt.Timeout, - Deadline: opt.Deadline, - Offset: opt.Offset, - Limit: opt.Limit, - URLList: make(chan string), - OutputCh: make(chan *pkg.Baseline, 100), - FuzzyCh: make(chan *pkg.Baseline, 100), - Fuzzy: opt.Fuzzy, - Force: opt.Force, + Progress: uiprogress.New(), + Threads: opt.Threads, + PoolSize: opt.PoolSize, + Mod: opt.Mod, + Timeout: opt.Timeout, + Deadline: opt.Deadline, + Offset: opt.Offset, + Limit: opt.Limit, + URLList: make(chan string), + OutputCh: make(chan *pkg.Baseline, 100), + FuzzyCh: make(chan *pkg.Baseline, 100), + Fuzzy: opt.Fuzzy, + Force: opt.Force, + CheckOnly: opt.CheckOnly, } err = pkg.LoadTemplates()