get given domain on resultcallback fx (#800)

Co-authored-by: Sandeep Singh <sandeep@projectdiscovery.io>
projectdiscovery · Mar 31, 2023 · 5075ae3 · 5075ae3
1 parent 2ced65c
commit 5075ae3
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/v2/pkg/resolve/resolve.go b/v2/pkg/resolve/resolve.go
@@ -25,6 +25,7 @@ type ResolutionPool struct {
 
 // HostEntry defines a host with the source
 type HostEntry struct {
+	Domain string
 	Host   string
 	Source string
 }

diff --git a/v2/pkg/runner/enumerate.go b/v2/pkg/runner/enumerate.go
@@ -79,7 +79,7 @@ func (r *Runner) EnumerateSingleDomainWithCtx(ctx context.Context, domain string
 						continue
 					}
 
-					hostEntry := resolve.HostEntry{Host: subdomain, Source: result.Source}
+					hostEntry := resolve.HostEntry{Domain: domain, Host: subdomain, Source: result.Source}
 
 					uniqueMap[subdomain] = hostEntry
 					// If the user asked to remove wildcard then send on the resolve

diff --git a/v2/pkg/runner/outputter.go b/v2/pkg/runner/outputter.go
@@ -128,7 +128,7 @@ func writeJSONHostIP(input string, results map[string]resolve.Result, writer io.
 func (o *OutputWriter) WriteHostNoWildcard(input string, results map[string]resolve.Result, writer io.Writer) error {
 	hosts := make(map[string]resolve.HostEntry)
 	for host, result := range results {
-		hosts[host] = resolve.HostEntry{Host: result.Host, Source: result.Source}
+		hosts[host] = resolve.HostEntry{Domain: host, Host: result.Host, Source: result.Source}
 	}
 
 	return o.WriteHost(input, hosts, writer)