fix: check pool net set headers

Author: M09Ic

core/pool/brutepool.go

@@ -107,14 +107,6 @@ type BrutePool struct {
 }
 
 func (pool *BrutePool) Init() error {
-	if pool.Headers.Get("User-Agent") == "" {
-		pool.Headers.Set("User-Agent", pkg.DefaultUserAgent)
-	}
-
-	if pool.Headers.Get("Accept") == "" {
-		pool.Headers.Set("Accept", "*/*")
-	}
-
 	pool.initwg.Add(2)
 	if pool.Index != "/" {
 		logs.Log.Logf(pkg.LogVerbose, "custom index url: %s", pkg.BaseURL(pool.url)+pkg.FormatURL(pkg.BaseURL(pool.url), pool.Index))

core/pool/checkpool.go

@@ -36,7 +36,7 @@ func NewCheckPool(ctx context.Context, config *Config) (*CheckPool, error) {
 			processCh:  make(chan *baseline.Baseline, config.Thread),
 		},
 	}
-	pool.Headers = map[string]string{"Connection": "close"}
+	pool.Headers.Set("Connection", "close")
 	p, _ := ants.NewPoolWithFunc(config.Thread, pool.Invoke)
 
 	pool.Pool = p

core/runner.go

@@ -114,6 +114,14 @@ func (r *Runner) PrepareConfig() *pool.Config {
 		config.Headers.Set(k, v)
 	}
 
+	if config.Headers.Get("User-Agent") == "" {
+		config.Headers.Set("User-Agent", pkg.DefaultUserAgent)
+	}
+
+	if config.Headers.Get("Accept") == "" {
+		config.Headers.Set("Accept", "*/*")
+	}
+
 	return config
 }