fix: make RawTarget conform FQDN

0x2E · Jul 17, 2023 · aa2b7a9 · aa2b7a9
1 parent ff4ec43
commit aa2b7a9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/internal/conf/conf.go b/internal/conf/conf.go
@@ -41,11 +41,11 @@ func (c *Config) Verify() error {
 		dn = dn[dot+1:]
 	}
 	c.Target = dn
-
 	if _, ok := dns.IsDomainName(c.Target); !ok {
 		return errors.New("invalid domain name: " + c.Target)
 	}
 	c.Target = dns.Fqdn(c.Target)
+	c.RawTarget = dns.Fqdn(c.RawTarget)
 
 	if c.Wordlist != "" {
 		f, err := os.Open(c.Wordlist)

diff --git a/internal/module/wordlist.go b/internal/module/wordlist.go
@@ -44,7 +44,11 @@ func RunWordlist(ctx context.Context, toNext chan<- *Task) error {
 		dSet[word] = struct{}{}
 
 		dn := fn(word)
+		// logrus.Debug(dn)
 		putTask(toNext, dn)
 	}
+
+	// try main domain too
+	putTask(toNext, conf.C.Target)
 	return nil
 }