Sort resolvers, move warnings & changed short flag names
This commit is contained in:
parent
4028682b9d
commit
14cd55baea
@ -9,6 +9,7 @@ import (
|
|||||||
|
|
||||||
"git.supernets.org/perp/blink/v1/internal/bar"
|
"git.supernets.org/perp/blink/v1/internal/bar"
|
||||||
"git.supernets.org/perp/blink/v1/internal/file"
|
"git.supernets.org/perp/blink/v1/internal/file"
|
||||||
|
"git.supernets.org/perp/blink/v1/internal/util"
|
||||||
"git.supernets.org/perp/blink/v1/pkg/dns"
|
"git.supernets.org/perp/blink/v1/pkg/dns"
|
||||||
"git.supernets.org/perp/blink/v1/pkg/runner"
|
"git.supernets.org/perp/blink/v1/pkg/runner"
|
||||||
"github.com/fatih/color"
|
"github.com/fatih/color"
|
||||||
@ -80,26 +81,15 @@ var rootCmd = &cobra.Command{
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
|
resolversLines = util.Port(resolversLines)
|
||||||
slices.Sort(resolversLines)
|
slices.Sort(resolversLines)
|
||||||
opts.Resolvers = slices.Compact(resolversLines)
|
opts.Resolvers = slices.Compact(resolversLines)
|
||||||
|
|
||||||
// Warnings
|
|
||||||
if opts.UDP {
|
|
||||||
log.Warn().Msg("UDP is enabled")
|
|
||||||
}
|
|
||||||
|
|
||||||
if opts.IPv6 {
|
|
||||||
log.Warn().Msg("IPv6 is enabled")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Store domains
|
// Store domains
|
||||||
domains := make(map[string]int)
|
domains := make(map[string]int)
|
||||||
|
|
||||||
// Store task count
|
|
||||||
tasks := len(opts.Wordlist) * len(opts.Domains)
|
|
||||||
|
|
||||||
// Create bar
|
// Create bar
|
||||||
pb := bar.New(tasks)
|
pb := bar.New(len(opts.Wordlist) * len(opts.Domains))
|
||||||
|
|
||||||
// Handle results
|
// Handle results
|
||||||
opts.OnResult = func(result *dns.Result) {
|
opts.OnResult = func(result *dns.Result) {
|
||||||
@ -148,6 +138,15 @@ var rootCmd = &cobra.Command{
|
|||||||
// Create runner
|
// Create runner
|
||||||
run := runner.New(&opts)
|
run := runner.New(&opts)
|
||||||
|
|
||||||
|
// Print warnings
|
||||||
|
if opts.UDP {
|
||||||
|
log.Warn().Msg("UDP is enabled")
|
||||||
|
}
|
||||||
|
|
||||||
|
if opts.IPv6 {
|
||||||
|
log.Warn().Msg("IPv6 is enabled")
|
||||||
|
}
|
||||||
|
|
||||||
// Wildcard enabled
|
// Wildcard enabled
|
||||||
if wildcard {
|
if wildcard {
|
||||||
// Go through domains
|
// Go through domains
|
||||||
@ -178,9 +177,9 @@ func init() {
|
|||||||
rootCmd.Flags().StringVarP(&resolvers, "resolvers", "r", "", "Path to resolvers")
|
rootCmd.Flags().StringVarP(&resolvers, "resolvers", "r", "", "Path to resolvers")
|
||||||
rootCmd.Flags().BoolVarP(&opts.UDP, "udp", "u", false, "Query using UDP")
|
rootCmd.Flags().BoolVarP(&opts.UDP, "udp", "u", false, "Query using UDP")
|
||||||
rootCmd.Flags().BoolVarP(&opts.IPv6, "ipv6", "i", false, "Query for IPv6")
|
rootCmd.Flags().BoolVarP(&opts.IPv6, "ipv6", "i", false, "Query for IPv6")
|
||||||
rootCmd.Flags().BoolVarP(&wildcard, "wildcard", "c", false, "Query for wildcard")
|
rootCmd.Flags().BoolVarP(&wildcard, "wildcard", "d", false, "Query for wildcard")
|
||||||
rootCmd.Flags().IntVarP(&opts.Timeout, "timeout", "d", 5, "Query timeout")
|
rootCmd.Flags().IntVarP(&opts.Timeout, "timeout", "t", 5, "Query timeout")
|
||||||
rootCmd.Flags().IntVarP(&opts.Threads, "threads", "t", 1, "Concurrent threads")
|
rootCmd.Flags().IntVarP(&opts.Threads, "threads", "c", 1, "Concurrent threads")
|
||||||
rootCmd.Flags().BoolVarP(&verbose, "verbose", "v", false, "Verbose logging")
|
rootCmd.Flags().BoolVarP(&verbose, "verbose", "v", false, "Verbose logging")
|
||||||
|
|
||||||
// Set flag options
|
// Set flag options
|
||||||
|
Loading…
Reference in New Issue
Block a user