diff --git a/pkg/runner/options.go b/pkg/runner/options.go index 0fb567e..463d0bc 100644 --- a/pkg/runner/options.go +++ b/pkg/runner/options.go @@ -39,9 +39,9 @@ func ParseOptions() *Options { flags.CreateGroup("options", "Options", flags.BoolVarP(&options.fast, "fast", "f", false, "Fast flag for dnsgen"), - flags.BoolVarP(&options.cleanup, "cleanup", "c", false, "Unessential files cleanup"), + flags.BoolVarP(&options.cleanup, "cleanup", "c", false, "Clean up all files except the final result"), flags.BoolVarP(&options.all, "all", "a", false, "All flag for subfinder"), - flags.BoolVarP(&options.silent, "silent", "s", false, "Show only resolved subdomains"), + flags.BoolVarP(&options.silent, "silent", "s", false, "Only show resolved subdomains"), flags.BoolVarP(&options.enrich, "enrich", "en", false, "Enrich flag for alterx"), flags.StringVarP(&options.permutationTool, "permutation-tool", "pt", "alterx", "Permutation tool (dnsgen or alterx)"), ) @@ -128,6 +128,11 @@ func (options *Options) validateOptions() error { return fmt.Errorf("a file under the name %v already exists", options.output) } + // checks if both silent and verbose flags are used + if options.verbose && options.silent { + return errors.New("can't use both silent and verbose mode") + } + return nil } diff --git a/pkg/runner/runner.go b/pkg/runner/runner.go index 333f908..988dcd6 100644 --- a/pkg/runner/runner.go +++ b/pkg/runner/runner.go @@ -164,10 +164,7 @@ func (runner *Runner) Start() error { } //print results to stdout - printResults( - runner.options.silent, - path.Join(runner.options.domain, runner.options.output), - ) + printResults(path.Join(runner.options.domain, runner.options.output)) //cleanup only if the flag is set if runner.options.cleanup { diff --git a/pkg/runner/utils.go b/pkg/runner/utils.go index 13bf969..ae126db 100644 --- a/pkg/runner/utils.go +++ b/pkg/runner/utils.go @@ -136,7 +136,7 @@ func cleanup(dir string) error { //list of files files := []string{ - "abuseipdb.subs", "subfinder.subs", "generated.subs", "shuffledns_phase1.in", + "abuseipdb.subs", "subfinder.subs", "generated.subs", "shuffledns_phase1.in", "shuffledns_phase1.out", "shuffledns_phase2.in", "shuffledns_phase2.out", "permutation.in", } @@ -182,7 +182,7 @@ func makeSubsFromWordlist(domain, wordlistFilename, generatedFilename string) er return nil } -func printResults(silentFlag bool, file string) error { +func printResults(file string) error { gologger.Debug().Msg("printing final results") f, err := os.Open(file)