Clean up line sorting
This commit is contained in:
parent
14cd55baea
commit
febed39b84
@ -69,21 +69,23 @@ var rootCmd = &cobra.Command{
|
|||||||
color.Red(banner)
|
color.Red(banner)
|
||||||
|
|
||||||
// Read wordlist path
|
// Read wordlist path
|
||||||
wordlistLines, err := file.Read(wordlist)
|
wLines, err := file.Read(wordlist)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
slices.Sort(wordlistLines)
|
|
||||||
opts.Wordlist = slices.Compact(wordlistLines)
|
|
||||||
|
|
||||||
// Read resolvers file
|
// Read resolvers file
|
||||||
resolversLines, err := file.Read(resolvers)
|
rLines, err := file.Read(resolvers)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
resolversLines = util.Port(resolversLines)
|
|
||||||
slices.Sort(resolversLines)
|
// Sort & compact lines + append port
|
||||||
opts.Resolvers = slices.Compact(resolversLines)
|
slices.Sort(wLines)
|
||||||
|
opts.Wordlist = slices.Compact(wLines)
|
||||||
|
slices.Sort(rLines)
|
||||||
|
opts.Resolvers = slices.Compact(rLines)
|
||||||
|
rLines = util.Port(rLines)
|
||||||
|
|
||||||
// Store domains
|
// Store domains
|
||||||
domains := make(map[string]int)
|
domains := make(map[string]int)
|
||||||
|
29
v1/internal/util/port.go
Normal file
29
v1/internal/util/port.go
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
package util
|
||||||
|
|
||||||
|
import "strings"
|
||||||
|
|
||||||
|
// Check for port in resolvers
|
||||||
|
func Port(resolvers []string) []string {
|
||||||
|
// No resolvers found
|
||||||
|
if len(resolvers) == 0 {
|
||||||
|
resolvers = append(resolvers, "1.1.1.1:53")
|
||||||
|
resolvers = append(resolvers, "9.9.9.9:53")
|
||||||
|
return resolvers
|
||||||
|
}
|
||||||
|
|
||||||
|
// Go through resolvers
|
||||||
|
for index, resolver := range resolvers {
|
||||||
|
// Split
|
||||||
|
split := strings.Split(resolver, ":")
|
||||||
|
|
||||||
|
// No port found
|
||||||
|
if len(split) == 1 {
|
||||||
|
// Ignore empty & commented lines
|
||||||
|
if resolver != "" && !strings.HasPrefix(resolver, "#") {
|
||||||
|
resolvers[index] = resolver + ":53"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return resolvers
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user