Method PrepareConfig

core/runner.go:104–175

()

}

func (r *Runner) PrepareConfig() *pool.Config {
	// 准备HTTP请求配置
	headers := make(http.Header)
	for k, v := range r.Headers {
		headers.Set(k, v)
	}

	if headers.Get("User-Agent") == "" {
		headers.Set("User-Agent", pkg.DefaultUserAgent)
	}

	if headers.Get("Accept") == "" {
		headers.Set("Accept", "*/*")
	}

	requestConfig := &ihttp.RequestConfig{
		Method:          r.Method,
		Headers:         headers,
		Host:            r.Host,
		Path:            r.Path,
		Body:            r.Body,
		RandomUserAgent: r.RandomUserAgent,
	}

	config := &pool.Config{
		Thread:         r.Threads,
		Timeout:        time.Duration(r.Timeout) * time.Second,
		RateLimit:      r.RateLimit,
		Mod:            pool.ModMap[r.Mod],
		Request:        requestConfig,
		OutputCh:       r.OutputCh,
		FuzzyCh:        r.FuzzyCh,
		Outwg:          r.OutWg,
		Fuzzy:          r.Fuzzy,
		CheckPeriod:    r.CheckPeriod,
		ErrPeriod:      int32(r.ErrPeriod),
		BreakThreshold: int32(r.BreakThreshold),
		MatchExpr:      r.MatchExpr,
		FilterExpr:     r.FilterExpr,
		RecuExpr:       r.RecursiveExpr,
		AppendRule:     r.AppendRules, // 对有效目录追加规则, 根据rule生成
		AppendWords:    r.AppendWords, // 对有效目录追加字典
		Fns:            r.Fns,
		//IgnoreWaf:       r.IgnoreWaf,
		Crawl:             r.CrawlPlugin,
		Scope:             r.Scope,
		Active:            r.Finger,
		Bak:               r.BakPlugin,
		Fuzzuli:           r.FuzzuliPlugin,
		Common:            r.CommonPlugin,
		Poc:               r.PocPlugin,
		RetryLimit:        r.RetryCount,
		ClientType:        r.ClientType,
		Random:            r.Random,
		Index:             r.Index,
		MaxRecursionDepth: r.Depth,
		MaxRedirect:       3,
		MaxAppendDepth:    r.AppendDepth,