overflow_discovery/discovery/service.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 scanService(port *model.Port, ds *model.DiscoveryService, resultChan chan interface{}, errChan chan error, doneChan chan<- struct{}, stopChan chan struct{}) {
	defer func() {
		doneChan <- struct{}{}
	}()

	_, ipNet, err := net.ParseCIDR(port.Host.Zone.Network)
	if nil != err {
		errChan <- err
		return
	}

	switch len(ipNet.IP) {
	case net.IPv4len:
		ipv4.ScanService(port, ds, resultChan, errChan, stopChan)
	case net.IPv6len:
		ipv6.ScanService(port, ds, resultChan, errChan, stopChan)

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