model/discovery/DiscoverHost.go

package discovery

import (
	"fmt"
	"net"

	"git.loafle.net/commons/util-go/net/cidr"
	"git.loafle.net/overflow/model/meta"
)

type DiscoverHost struct {
	MetaIPType *meta.MetaIPType `json:"metaIPType,omitempty"`

	FirstScanRange string   `json:"firstScanRange,omitempty"`
	LastScanRange  string   `json:"lastScanRange,omitempty"`
	ExcludeHosts   []string `json:"excludeHosts,omitempty"`
	IncludeHosts   []string `json:"includeHosts,omitempty"`

	DiscoverPort *DiscoverPort `json:"discoverPort,omitempty"`
}

func (dh *DiscoverHost) TargetHosts(network string) ([]net.IP, error) {
	cr, err := cidr.NewCIDRRanger(network)
	if nil != err {
		return nil, err
	}

	var firstIP net.IP
	if "" != dh.FirstScanRange {
		firstIP = net.ParseIP(dh.FirstScanRange)
		if nil == firstIP {
			return nil, fmt.Errorf("IP(%v) of FirstScanRange host is not valid", firstIP)
		}
	}
	var lastIP net.IP
	if "" != dh.LastScanRange {
		lastIP = net.ParseIP(dh.LastScanRange)
		if nil == lastIP {
			return nil, fmt.Errorf("IP(%v) of LastScanRange host is not valid", lastIP)
		}
	}

	includeIPs := make([]net.IP, 0)
	for _, iHost := range dh.IncludeHosts {
		iIP := net.ParseIP(iHost)
		if nil == iIP {
			return nil, fmt.Errorf("IP(%v) of include host is not valid", iHost)
		}
		includeIPs = append(includeIPs, iIP)
	}

	excludeIPs := make([]net.IP, 0)
	for _, eHost := range dh.ExcludeHosts {
		eIP := net.ParseIP(eHost)
		if nil == eIP {
			return nil, fmt.Errorf("IP(%v) of exclude host is not valid", eHost)
		}
		excludeIPs = append(excludeIPs, eIP)
	}

	ranges, err := cr.Ranges(firstIP, lastIP, includeIPs, excludeIPs)
	if nil != err {
		return nil, err
	}

	return ranges, nil
}