fix: pocscan

Author: niudaii

pkg/pocscan/nuclei/poc.go

@@ -11,7 +11,6 @@ import (
 	"github.com/projectdiscovery/nuclei/v2/pkg/core"
 	"github.com/projectdiscovery/nuclei/v2/pkg/output"
 	"github.com/projectdiscovery/nuclei/v2/pkg/protocols"
-	"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/hosterrorscache"
 	"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/interactsh"
 	"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/protocolinit"
 	"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/protocolstate"
@@ -74,8 +73,8 @@ func ParsePoc(templateString string) {
 }
 
 func InitExecuterOptions(dir string) (err error) {
-	cache := hosterrorscache.New(30, hosterrorscache.DefaultMaxHostsCount)
-	defer cache.Close()
+	//cache := hosterrorscache.New(30, hosterrorscache.DefaultMaxHostsCount)
+	//defer cache.Close()
 
 	mockProgress := &testutils.MockProgressClient{}
 	reportingClient, _ := reporting.New(&reporting.Options{}, "")
@@ -110,16 +109,16 @@ func InitExecuterOptions(dir string) (err error) {
 	catalog := disk.NewCatalog(dir)
 
 	ExecuterOptions = protocols.ExecuterOptions{
-		Output:          outputWriter,
-		Options:         options,
-		Progress:        mockProgress,
-		Catalog:         catalog,
-		IssuesClient:    reportingClient,
-		Interactsh:      interactClient,
-		RateLimiter:     ratelimit.New(context.Background(), 150, time.Second),
-		HostErrorsCache: cache,
-		Colorizer:       aurora.NewAurora(true),
-		ResumeCfg:       types.NewResumeCfg(),
+		Output:       outputWriter,
+		Options:      options,
+		Progress:     mockProgress,
+		Catalog:      catalog,
+		IssuesClient: reportingClient,
+		Interactsh:   interactClient,
+		RateLimiter:  ratelimit.New(context.Background(), 150, time.Second),
+		//HostErrorsCache: cache,
+		Colorizer: aurora.NewAurora(true),
+		ResumeCfg: types.NewResumeCfg(),
 	}
 
 	return
@@ -128,10 +127,11 @@ func InitExecuterOptions(dir string) (err error) {
 func InitEngine(timeout int, proxy string) (engine *core.Engine) {
 	ExecuterOptions.Options.Timeout = timeout
 	ExecuterOptions.Options.RateLimit = 1
-	ExecuterOptions.Options.ProxyInternal = true
 	if proxy == "" {
+		ExecuterOptions.Options.ProxyInternal = false
 		ExecuterOptions.Options.Proxy = []string{}
 	} else {
+		ExecuterOptions.Options.ProxyInternal = true
 		ExecuterOptions.Options.Proxy = []string{proxy}
 	}
 

pkg/pocscan/nuclei/proxy.go

@@ -19,6 +19,8 @@ import (
 func loadProxyServers(options *types.Options) error {
 	var proxyURLList []url.URL // bug
 	if len(options.Proxy) == 0 {
+		types.ProxySocksURL = ""
+		types.ProxyURL = ""
 		return nil
 	}
 	for _, p := range options.Proxy {

pkg/pocscan/runner.go

@@ -27,6 +27,7 @@ type Runner struct {
 
 func NewRunner(options *Options, gobyPocs []*goby.Poc, xrayPocs []*xray.Poc, nucleiTemplates []*nuclei.Template) (runner *Runner, err error) {
 	nucleiEngine := nuclei.InitEngine(options.Timeout, options.Proxy)
+	//nucleiEngine = nuclei.InitEngine(options.Timeout, "")
 	reqClient := utils.NewReqClient(options.Proxy, options.Timeout, options.Headers)
 	runner = &Runner{
 		gobyPocs:        gobyPocs,