Method Invoke

core/pool/checkpool.go:148–212

(v interface{})

}

func (pool *CheckPool) Invoke(v interface{}) {
	defer func() {
		pool.reqCount.Add(1)
		pool.wg.Done()
	}()

	unit := v.(*Unit)

	// 为Check请求创建RequestConfig，使用配置的参数
	checkReqConfig := &ihttp.RequestConfig{
		Method:          pool.Request.Method, // 使用配置的 Method
		Headers:         pool.Request.Headers,
		Host:            pool.Request.Host, // 使用配置的 Host
		Path:            pool.Request.Path, // 使用配置的 Path
		Body:            pool.Request.Body, // 使用配置的 Body
		RandomUserAgent: pool.Request.RandomUserAgent,
	}

	req, err := checkReqConfig.Build(pool.ctx, pool.ClientType, unit.path, "", "", "")
	if err != nil {
		logs.Log.Debug(err.Error())
		bl := &baseline.Baseline{
			SprayResult: &parsers.SprayResult{
				UrlString: unit.path,
				IsValid:   false,
				ErrString: err.Error(),
				Reason:    pkg.ErrUrlError.Error(),
				ReqDepth:  unit.depth,
			},
		}
		pool.sendProcess(bl)
		return
	}
	start := time.Now()
	var bl *baseline.Baseline
	resp, reqerr := pool.client.Do(req)
	if reqerr != nil {
		pool.failedCount.Add(1)
		bl = &baseline.Baseline{
			SprayResult: &parsers.SprayResult{
				UrlString: unit.path,
				IsValid:   false,
				ErrString: reqerr.Error(),
				Reason:    pkg.ErrRequestFailed.Error(),
				ReqDepth:  unit.depth,
			},
		}
		logs.Log.Debugf("%s, %s", unit.path, reqerr.Error())
		pool.doUpgrade(bl)
	} else {
		bl = baseline.NewBaseline(req.URI(), req.Host(), resp)
		bl.ReqDepth = unit.depth
		bl.Collect()
		if bl.Status == 400 {
			pool.doUpgrade(bl)
		}
	}
	bl.ReqDepth = unit.depth