Adds --subs-only to assetfinder

Author: tomnomnom

assetfinder/main.go

@@ -12,13 +12,16 @@ import (
 )
 
 func main() {
+	var subsOnly bool
+	flag.BoolVar(&subsOnly, "subs-only", false, "Only incluse subdomains of search domain")
 	flag.Parse()
 
 	domain := flag.Arg(0)
 	if domain == "" {
 		fmt.Println("no domain specified")
 		return
 	}
+	domain = strings.ToLower(domain)
 
 	sources := []fetchFn{
 		fetchCertSpotter,
@@ -68,6 +71,9 @@ func main() {
 		if _, ok := printed[n]; ok {
 			continue
 		}
+		if subsOnly && !strings.HasSuffix(n, domain) {
+			continue
+		}
 		fmt.Println(n)
 		printed[n] = true
 	}