Add wordlist generator

.gitignore

@@ -1,3 +1,4 @@
 .DS_Store
 .*swp
 cmd/sonar/sonar
+words.go

cmd/sonar/main.go

@@ -52,7 +52,7 @@ func main() {
 	case (brute == true):
 		var wl sonar.Wordlist
 		if wordlist == "" {
-			wl = sonar.NewInternal(sonar.InternalWords)
+			wl = sonar.NewInternal(sonar.InternalWordlist)
 		} else {
 			fp, err := os.Open(wordlist)
 			if err != nil {

cmd/wordlist_generator/main.go

@@ -0,0 +1,39 @@
+package main
+
+import (
+	"bytes"
+	"flag"
+	"io/ioutil"
+	"log"
+	"strings"
+)
+
+func main() {
+	var wordlist, pkg, output string
+
+	flag.StringVar(&wordlist, "wordlist", "", "word list to generate from")
+	flag.StringVar(&pkg, "package", "sonar", "package to generate for")
+	flag.StringVar(&output, "output", "words.go", "file output source to")
+	flag.Parse()
+
+	if wordlist == "" {
+		log.Fatal("no wordlist specified")
+	}
+
+	words, err := ioutil.ReadFile(wordlist)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	wordSlice := strings.Split(string(words), "\n")
+	out := bytes.NewBuffer([]byte{})
+	_, err = out.Write([]byte("package " + pkg + "\n\nvar InternalWordlist = []string{\"" + strings.Join(wordSlice, "\", \"") + "\"}\n"))
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	err = ioutil.WriteFile(output, out.Bytes(), 0600)
+	if err != nil {
+		log.Fatal(err)
+	}
+}

wordlist.go

@@ -1,3 +1,4 @@
+//go:generate wordlist_generator -wordlist wordlist.txt
 package sonar
 
 import (
@@ -6,8 +7,6 @@ import (
 	"strings"
 )
 
-var InternalWords = []string{"www", "beta", "mail", "demo", "test"}
-
 type Wordlist interface {
 	readWords()
 	GetChannel() <-chan string