Initial commit

Author: PolovinaD

.gitignore

@@ -0,0 +1,4 @@
+.idea
+domains.txt
+wordlist.txt
+out.txt

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2021 Nenad Popovic
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -0,0 +1,50 @@
+mkpath
+------
+Make paths using a wordlist
+
+Read a wordlist file and generate paths for given domain or list of domains.
+Input from wordlist file is lowercased and unique words are processed. Additionally, wordlist can be
+filtered using regex. 
+
+```
+Usage of mksub:
+  -d string
+    	Input domain
+  -df string
+    	Input domain file, one domain per line
+  -l int
+    	Path level to generate (default 1) (default 1)
+  -o string
+    	Output file (optional)
+  -r string
+    	Regex to filter words from wordlist file
+  -w string
+    	Wordlist file
+```
+
+### Example
+
+##### wordlist.txt
+```
+dev
+DEV
+*
+foo/bar
+prod
+```
+```shell script
+> go run mksub.go -d example.com -l 2 -w input.txt -r "^[a-zA-Z0-9\.-_]+$"
+example.com/dev/
+example.com/foo/bar/
+example.com/prod/
+example.com/foo/bar/dev/
+example.com/prod/dev/
+example.com/dev/dev/
+example.com/dev/foo/bar/
+example.com/foo/bar/foo/bar/
+example.com/prod/foo/bar/
+example.com/dev/prod/
+example.com/foo/bar/prod/
+example.com/prod/prod/
+
+```

mkpath.go

@@ -0,0 +1,106 @@
+package main
+
+import (
+	"bufio"
+	"flag"
+	"fmt"
+	"os"
+	"regexp"
+	"strings"
+)
+
+func main() {
+	domain := flag.String("d", "", "Input domain")
+	domainFile := flag.String("df", "", "Input domain file, one domain per line")
+	wordlist := flag.String("w", "", "Wordlist file")
+	r := flag.String("r", "", "Regex to filter words from wordlist file")
+	level := flag.Int("l", 1, "Subdomain level to generate (default 1)")
+	output := flag.String("o", "", "Output file (optional)")
+	flag.Parse()
+
+	inputDomains := make([]string, 0)
+	if *domain != "" {
+		inputDomains = append(inputDomains, *domain)
+	}
+	if *domainFile != "" {
+		inputFile, err := os.Open(*domainFile)
+		if err != nil {
+			fmt.Println(err.Error())
+			os.Exit(1)
+		}
+		defer inputFile.Close()
+		scanner := bufio.NewScanner(inputFile)
+		for scanner.Scan() {
+			inputDomains = append(inputDomains, scanner.Text())
+		}
+	}
+	if len(inputDomains) == 0 {
+		fmt.Println("No input provided")
+		os.Exit(1)
+	}
+
+	wordlistFile, err := os.Open(*wordlist)
+	if err != nil {
+		fmt.Println(err.Error())
+		os.Exit(1)
+	}
+	defer wordlistFile.Close()
+
+	var reg *regexp.Regexp
+	if *r != "" {
+		reg, err = regexp.Compile(*r)
+		if err != nil {
+			fmt.Println(err.Error())
+			os.Exit(1)
+		}
+	}
+
+	var outputFile *os.File
+	if *output != "" {
+		outputFile, err = os.Create(*output)
+		if err != nil {
+			fmt.Println(err.Error())
+			os.Exit(1)
+		}
+		defer outputFile.Close()
+	}
+
+	wordSet := make(map[string]bool)
+	scanner := bufio.NewScanner(wordlistFile)
+
+	for scanner.Scan() {
+		word := strings.ToLower(scanner.Text())
+		if reg != nil {
+			if !reg.Match([]byte(word)) {
+				continue
+			}
+		}
+		if _, isOld := wordSet[word]; word != "" && !isOld {
+			wordSet[word] = true
+		}
+	}
+
+	results := make([]string, 0)
+	for i := 0; i < *level; i += 1 {
+		toMerge := results[0:]
+		if len(toMerge) == 0 {
+			for word := range wordSet {
+				results = append(results, word)
+			}
+		} else {
+			for _, sd := range toMerge {
+				for word := range wordSet {
+					results = append(results, fmt.Sprintf("%s.%s", word, sd))
+				}
+			}
+		}
+	}
+	for _, domain := range inputDomains {
+		for _, subpath := range results {
+			fmt.Println(domain + "/" + subpath + "/")
+			if outputFile != nil {
+				_, _ = outputFile.WriteString(domain + "/" + subpath + "/" + "\n")
+			}
+		}
+	}
+}