add regex support for -eh (#4437)

projectdiscovery · Nov 28, 2023 · 7cd340b · 7cd340b
1 parent 578e301
commit 7cd340b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/core/inputs/hybrid/hmap.go b/pkg/core/inputs/hybrid/hmap.go
@@ -8,6 +8,7 @@ import (
 	"fmt"
 	"io"
 	"os"
+	"regexp"
 	"strings"
 	"sync"
 	"time"
@@ -437,7 +438,8 @@ func (i *Input) delItem(metaInput *contextargs.MetaInput) {
 			return err
 		}
 
-		if tmpUrl.Host == targetUrl.Host {
+		matched, _ := regexp.MatchString(metaInput.Input, tmpUrl.Host)
+		if tmpUrl.Host == targetUrl.Host || matched {
 			_ = i.hostMap.Del(tmpKey)
 			i.excludedHosts[tmpKey] = struct{}{}
 			i.excludedCount++