Change OTX to use the domain API

lc · Oct 15, 2020 · 5caa956 · 5caa956
1 parent 3c0eb59
commit 5caa956
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/providers/otx.go b/providers/otx.go
@@ -3,6 +3,7 @@ package providers
 import (
 	"encoding/json"
 	"fmt"
+	"strings"
 )
 
 type OTXProvider struct {
@@ -51,7 +52,13 @@ func (o *OTXProvider) Fetch(domain string, results chan<- string) error {
 		_ = resp.Body.Close()
 
 		for _, entry := range result.URLList {
-			results <- entry.URL
+			if o.IncludeSubdomains {
+				results <- entry.URL
+			} else {
+				if strings.EqualFold(domain, entry.Hostname) {
+					results <- entry.URL
+				}
+			}
 		}
 
 		if !result.HasNext {

diff --git a/providers/provider.go b/providers/provider.go
@@ -8,7 +8,7 @@ import (
 
 const (
 	// Version of gau
-	Version = `1.0.6`
+	Version = `1.0.7`
 	// UserAgent for the HTTP Client
 	userAgent = "Mozilla/5.0 (compatible; gau/" + Version + "; https://github.com/lc/gau)"
 )