overflow_discovery/discovery/port.go

package discovery

import (
	"fmt"
	"net"

	"git.loafle.net/overflow/overflow_discovery/api/module/discovery/model"
	"git.loafle.net/overflow/overflow_discovery/discovery/ipv4"
	"git.loafle.net/overflow/overflow_discovery/discovery/ipv6"
)

func scanPort(host *model.Host, dp *model.DiscoveryPort, resultChan chan interface{}, errChan chan error, doneChan chan<- struct{}, stopChan chan struct{}) {
	defer func() {
		doneChan <- struct{}{}
	}()

	_, ipNet, err := net.ParseCIDR(host.Zone.Network)
	if nil != err {
		errChan <- err
		return
	}
	switch len(ipNet.IP) {
	case net.IPv4len:
		ipv4.ScanPort(host, dp, resultChan, errChan, stopChan)
	case net.IPv6len:
		ipv6.ScanPort(host, dp, resultChan, errChan, stopChan)

	default:
		errChan <- fmt.Errorf("Discovery: Not supported ip length")
		return
	}
}