From 206ecd79f43a50f487ef6addded374366ff4a702 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Do=C4=9Fan=20Can=20Bak=C4=B1r?= Date: Wed, 21 Feb 2024 14:32:24 +0300 Subject: [PATCH 1/2] remove scheme --- v2/pkg/runner/enumerate.go | 1 + 1 file changed, 1 insertion(+) diff --git a/v2/pkg/runner/enumerate.go b/v2/pkg/runner/enumerate.go index ac3c70dd0..6c7cac7b9 100644 --- a/v2/pkg/runner/enumerate.go +++ b/v2/pkg/runner/enumerate.go @@ -63,6 +63,7 @@ func (r *Runner) EnumerateSingleDomainWithCtx(ctx context.Context, domain string continue } subdomain := strings.ReplaceAll(strings.ToLower(result.Value), "*.", "") + subdomain = strings.TrimPrefix(strings.TrimPrefix(subdomain, "http://"), "https://") if matchSubdomain := r.filterAndMatchSubdomain(subdomain); matchSubdomain { if _, ok := uniqueMap[subdomain]; !ok { From 4e5adaf5fbef2ee706f0fcfd6c807cba6b8a7360 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Do=C4=9Fan=20Can=20Bak=C4=B1r?= Date: Fri, 15 Mar 2024 14:51:05 +0300 Subject: [PATCH 2/2] use replacer --- v2/pkg/runner/enumerate.go | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/v2/pkg/runner/enumerate.go b/v2/pkg/runner/enumerate.go index 6c7cac7b9..59c5d5041 100644 --- a/v2/pkg/runner/enumerate.go +++ b/v2/pkg/runner/enumerate.go @@ -18,6 +18,12 @@ import ( const maxNumCount = 2 +var replacer = strings.NewReplacer( + "*.", "", + "http://", "", + "https://", "", +) + // EnumerateSingleDomain wraps EnumerateSingleDomainWithCtx with an empty context func (r *Runner) EnumerateSingleDomain(domain string, writers []io.Writer) error { return r.EnumerateSingleDomainWithCtx(context.Background(), domain, writers) @@ -62,8 +68,7 @@ func (r *Runner) EnumerateSingleDomainWithCtx(ctx context.Context, domain string skippedCounts[result.Source]++ continue } - subdomain := strings.ReplaceAll(strings.ToLower(result.Value), "*.", "") - subdomain = strings.TrimPrefix(strings.TrimPrefix(subdomain, "http://"), "https://") + subdomain := replacer.Replace(result.Value) if matchSubdomain := r.filterAndMatchSubdomain(subdomain); matchSubdomain { if _, ok := uniqueMap[subdomain]; !ok {