Method queryIPCIDR

sources/agent/driftnet/driftnet.go:153–215

(ctx context.Context, session *sources.Session, driftnetRequest *DriftnetRequest, results chan sources.Result)

}

func (agent *Agent) queryIPCIDR(ctx context.Context, session *sources.Session, driftnetRequest *DriftnetRequest, results chan sources.Result) {
	var targetCIDR = driftnetRequest.Query

	if iputil.IsIP(targetCIDR) {
		if iputil.IsIPv4(targetCIDR) {
			targetCIDR = iputil.AsIPV4CIDR(targetCIDR)
		} else if iputil.IsIPv6(targetCIDR) {
			targetCIDR = iputil.AsIPV6CIDR(targetCIDR)
		}
	}

	requestCIDRs, splitError := mapcidr.SplitByNumber(targetCIDR, 1024)
	if splitError != nil {
		sources.SendResult(ctx, results, sources.Result{Source: agent.Name(), Error: splitError})
		return
	}

	totalReportedResults := 0
	for _, requestCIDR := range requestCIDRs {
		if ctx.Err() != nil {
			return
		}
		driftnetRequest.Query = requestCIDR.String()

		resp, queryError := agent.queryURL(ctx, session, OpenPortIPPortsURL, driftnetRequest)

		if queryError != nil {
			if resp != nil && resp.StatusCode == http.StatusNoContent {
				continue
			}

			sources.SendResult(ctx, results, sources.Result{Source: agent.Name(), Error: queryError})
			return
		}
		defer func() {
			_ = resp.Body.Close()
		}()

		driftnetResponse := &DriftnetAPIOpenIPPortResponse{}
		if err := json.NewDecoder(resp.Body).Decode(driftnetResponse); err != nil {
			sources.SendResult(ctx, results, sources.Result{Source: agent.Name(), Error: err})
			return
		}

		for ip, portsResponse := range driftnetResponse.Values {
			result := sources.Result{Source: agent.Name(), IP: ip}
			result.Raw, _ = json.Marshal(driftnetResponse)

			for port := range portsResponse.Values {
				result.Port = port
				if !sources.SendResult(ctx, results, result) {
					return
				}

				totalReportedResults += 1

				if driftnetRequest.ResultLimit > 0 && totalReportedResults >= driftnetRequest.ResultLimit {
					return