Add wordlist generator

jrozner · jrozner · commit 045ed69f3878 · 2016-02-21T22:38:23.000-08:00
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,4 @@
 .DS_Store
 .*swp
 cmd/sonar/sonar
+words.go
diff --git a/cmd/sonar/main.go b/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 {
diff --git a/cmd/wordlist_generator/main.go b/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)
+	}
+}
diff --git a/wordlist.go b/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
