ing
This commit is contained in:
crusader
parent
8239f564f7
commit
c59c3ed31b
|
|
@ -1,82 +0,0 @@
|
||||||
package discovery
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"git.loafle.net/overflow/overflow_discovery/discovery/types"
|
|
||||||
)
|
|
||||||
|
|
||||||
type Discoverer interface {
|
|
||||||
Discover(endded chan<- error) error
|
|
||||||
DiscoverZone(endded chan<- error) error
|
|
||||||
DiscoverHost(endded chan<- error) error
|
|
||||||
DiscoverPort(endded chan<- error) error
|
|
||||||
DiscoverService(endded chan<- error) error
|
|
||||||
DiscoverUDPService(endded chan<- error) error
|
|
||||||
}
|
|
||||||
|
|
||||||
type discovery struct {
|
|
||||||
zd zoneDiscoverer
|
|
||||||
hd hostDiscoverer
|
|
||||||
pd portDiscoverer
|
|
||||||
|
|
||||||
endded chan<- error
|
|
||||||
zoneEndded chan error
|
|
||||||
hostEndded chan error
|
|
||||||
portEndded chan error
|
|
||||||
|
|
||||||
zoneFound chan *types.DiscoveryZone
|
|
||||||
}
|
|
||||||
|
|
||||||
func New() Discoverer {
|
|
||||||
d := &discovery{}
|
|
||||||
d.zd = newZoneDiscoverer()
|
|
||||||
d.hd = newHostDiscoverer()
|
|
||||||
d.pd = newPortDiscoverer()
|
|
||||||
|
|
||||||
return d
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *discovery) Discover(endded chan<- error) error {
|
|
||||||
d.endded = endded
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *discovery) DiscoverZone(endded chan<- error) error {
|
|
||||||
d.endded = endded
|
|
||||||
|
|
||||||
// 1. When discoverd zone
|
|
||||||
// 2. When occurred error
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *discovery) DiscoverHost(endded chan<- error) error {
|
|
||||||
d.endded = endded
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *discovery) DiscoverPort(endded chan<- error) error {
|
|
||||||
d.endded = endded
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *discovery) DiscoverService(endded chan<- error) error {
|
|
||||||
d.endded = endded
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *discovery) DiscoverUDPService(endded chan<- error) error {
|
|
||||||
d.endded = endded
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *discovery) Shutdown(ctx context.Context) error {
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
@ -1,31 +0,0 @@
|
||||||
package discovery
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"git.loafle.net/overflow/overflow_discovery/commons"
|
|
||||||
)
|
|
||||||
|
|
||||||
type hostDiscoverer interface {
|
|
||||||
commons.EndableStarter
|
|
||||||
commons.Shutdowner
|
|
||||||
}
|
|
||||||
|
|
||||||
type hostDiscovery struct {
|
|
||||||
}
|
|
||||||
|
|
||||||
func newHostDiscoverer() hostDiscoverer {
|
|
||||||
hd := &hostDiscovery{}
|
|
||||||
|
|
||||||
return hd
|
|
||||||
}
|
|
||||||
|
|
||||||
func (hd *hostDiscovery) EndableStart(endded chan<- error) error {
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (hd *hostDiscovery) Shutdown(ctx context.Context) error {
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
@ -1,31 +0,0 @@
|
||||||
package discovery
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"git.loafle.net/overflow/overflow_discovery/commons"
|
|
||||||
)
|
|
||||||
|
|
||||||
type portDiscoverer interface {
|
|
||||||
commons.EndableStarter
|
|
||||||
commons.Shutdowner
|
|
||||||
}
|
|
||||||
|
|
||||||
type portDiscovery struct {
|
|
||||||
}
|
|
||||||
|
|
||||||
func newPortDiscoverer() portDiscoverer {
|
|
||||||
pd := &portDiscovery{}
|
|
||||||
|
|
||||||
return pd
|
|
||||||
}
|
|
||||||
|
|
||||||
func (pd *portDiscovery) EndableStart(endded chan<- error) error {
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (pd *portDiscovery) Shutdown(ctx context.Context) error {
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
@ -1,31 +0,0 @@
|
||||||
package discovery
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"git.loafle.net/overflow/overflow_discovery/commons"
|
|
||||||
)
|
|
||||||
|
|
||||||
type serviceDiscoverer interface {
|
|
||||||
commons.EndableStarter
|
|
||||||
commons.Shutdowner
|
|
||||||
}
|
|
||||||
|
|
||||||
type serviceDiscovery struct {
|
|
||||||
}
|
|
||||||
|
|
||||||
func newServiceDiscoverer() portDiscoverer {
|
|
||||||
pd := &serviceDiscovery{}
|
|
||||||
|
|
||||||
return pd
|
|
||||||
}
|
|
||||||
|
|
||||||
func (sd *serviceDiscovery) EndableStart(endded chan<- error) error {
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (sd *serviceDiscovery) Shutdown(ctx context.Context) error {
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
@ -1,21 +0,0 @@
|
||||||
package types
|
|
||||||
|
|
||||||
const (
|
|
||||||
TYPE_TCP = "TCP"
|
|
||||||
TYPE_UDP = "UDP"
|
|
||||||
TYPE_TLS = "TLS"
|
|
||||||
|
|
||||||
SEND = "Send"
|
|
||||||
RECV = "Recv"
|
|
||||||
TIMEOUT = "Timeout"
|
|
||||||
RECV_RST = "Closed"
|
|
||||||
|
|
||||||
SUCCESS = "Success"
|
|
||||||
|
|
||||||
TYPE_DISCOVERY = "DISCOVERY"
|
|
||||||
TYPE_DETECTNETWORK = "DETECTNETWORK"
|
|
||||||
TYPE_SENSOR = "SENSOR"
|
|
||||||
|
|
||||||
RANGE_TYPE_RANGE = "RANGE"
|
|
||||||
RANGE_TYPE_TARGET = "TARGET"
|
|
||||||
)
|
|
||||||
|
|
@ -1,28 +0,0 @@
|
||||||
package types
|
|
||||||
|
|
||||||
import (
|
|
||||||
"sync"
|
|
||||||
|
|
||||||
"git.loafle.net/overflow/overflow_discovery/discovery/types/timestamp"
|
|
||||||
)
|
|
||||||
|
|
||||||
type DiscoveryHistory struct {
|
|
||||||
ID int `json:"id,omitempty"`
|
|
||||||
StartDate timestamp.Timestamp `json:"startDate"`
|
|
||||||
EndDate timestamp.Timestamp `json:"endDate"`
|
|
||||||
Result bool `json:"result"`
|
|
||||||
Zone *DiscoveryZone `json:"zone"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewDiscoveryHistory(zone *DiscoveryZone) DiscoveryHistory {
|
|
||||||
return DiscoveryHistory{
|
|
||||||
Zone: zone,
|
|
||||||
StartDate: timestamp.Now(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type DiscoveryMgr struct {
|
|
||||||
Stop chan bool
|
|
||||||
DiscoveryWg *sync.WaitGroup
|
|
||||||
Zone *DiscoveryZone
|
|
||||||
}
|
|
||||||
|
|
@ -1,92 +0,0 @@
|
||||||
package types
|
|
||||||
|
|
||||||
import (
|
|
||||||
"net"
|
|
||||||
"sync"
|
|
||||||
|
|
||||||
"git.loafle.net/overflow/overflow_discovery/discovery/types/timestamp"
|
|
||||||
)
|
|
||||||
|
|
||||||
type DiscoveryHost struct {
|
|
||||||
M *sync.RWMutex `json:"-"`
|
|
||||||
Ports_ map[string]*DiscoveryPort `json:"-"`
|
|
||||||
HostName string `json:"-"`
|
|
||||||
PortDiscoveryTime timestamp.Timestamp `json:"-"`
|
|
||||||
Zone *DiscoveryZone `json:"-"`
|
|
||||||
Histories []*PortScanHistory `json:"-"`
|
|
||||||
|
|
||||||
FirstScanRange uint16 `json:"firstScanRange"`
|
|
||||||
LastScanRange uint16 `json:"lastScanRange"`
|
|
||||||
Name string `json:"name"`
|
|
||||||
|
|
||||||
ID int `json:"id,omitempty"`
|
|
||||||
IP string `json:"ip"`
|
|
||||||
Mac string `json:"mac"`
|
|
||||||
Ports []*DiscoveryPort `json:"ports"`
|
|
||||||
CreateDate timestamp.Timestamp `json:"createDate,omitempty"`
|
|
||||||
UpdateDate timestamp.Timestamp `json:"updateDate,omitempty"`
|
|
||||||
|
|
||||||
Os string `json:"os,omitempty"`
|
|
||||||
Target bool `json:"target,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *DiscoveryHost) SetPort(key string, value *DiscoveryPort) {
|
|
||||||
h.M.Lock()
|
|
||||||
defer h.M.Unlock()
|
|
||||||
h.Ports_[key] = value
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *DiscoveryHost) GetPort(key string) *DiscoveryPort {
|
|
||||||
h.M.RLock()
|
|
||||||
defer h.M.RUnlock()
|
|
||||||
return h.Ports_[key]
|
|
||||||
}
|
|
||||||
|
|
||||||
type HostScanHistory struct {
|
|
||||||
ID int `json:"id,omitempty"`
|
|
||||||
IP string `json:"ip"`
|
|
||||||
CreateDate timestamp.Timestamp `json:"createDate,omitempty"`
|
|
||||||
SendDate timestamp.Timestamp `json:"sendDate"`
|
|
||||||
ResultDate timestamp.Timestamp `json:"resultDate"`
|
|
||||||
ResultType string `json:"resultType"`
|
|
||||||
Description string `json:"description"`
|
|
||||||
Zone *DiscoveryZone `json:"-"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type HostMgr struct {
|
|
||||||
DiscoveryMgr
|
|
||||||
ToPort chan *DiscoveryHost
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewHost(ip net.IP, macAddr net.HardwareAddr) *DiscoveryHost {
|
|
||||||
host := &DiscoveryHost{
|
|
||||||
M: new(sync.RWMutex),
|
|
||||||
Ports_: make(map[string]*DiscoveryPort, 100),
|
|
||||||
IP: ip.String(),
|
|
||||||
Mac: macAddr.String(),
|
|
||||||
CreateDate: timestamp.Now(),
|
|
||||||
UpdateDate: timestamp.Now(),
|
|
||||||
}
|
|
||||||
|
|
||||||
return host
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewHostHistory(ip net.IP, resultType string, des string) *HostScanHistory {
|
|
||||||
hh := &HostScanHistory{
|
|
||||||
IP: ip.String(),
|
|
||||||
SendDate: timestamp.Now(),
|
|
||||||
ResultType: resultType,
|
|
||||||
Description: des,
|
|
||||||
}
|
|
||||||
|
|
||||||
return hh
|
|
||||||
}
|
|
||||||
func (p *DiscoveryHost) AddHistory(h *PortScanHistory) {
|
|
||||||
p.Histories = append(p.Histories, h)
|
|
||||||
}
|
|
||||||
|
|
||||||
type PortRange struct {
|
|
||||||
Types string
|
|
||||||
First int
|
|
||||||
Second int
|
|
||||||
}
|
|
||||||
|
|
@ -1,65 +0,0 @@
|
||||||
package types
|
|
||||||
|
|
||||||
import (
|
|
||||||
"strconv"
|
|
||||||
|
|
||||||
"git.loafle.net/overflow/overflow_discovery/discovery/types/timestamp"
|
|
||||||
)
|
|
||||||
|
|
||||||
type DiscoveryPort struct {
|
|
||||||
Port_ string `json:"-"`
|
|
||||||
|
|
||||||
ID int `json:"id,omitempty"`
|
|
||||||
CreateDate timestamp.Timestamp `json:"createDate,omitempty"`
|
|
||||||
UpdateDate timestamp.Timestamp `json:"updateDate,omitempty"`
|
|
||||||
Host *DiscoveryHost `json:"-"`
|
|
||||||
Services []*DiscoveryService `json:"services"`
|
|
||||||
Histories []*ServiceScanHistory `json:"-"`
|
|
||||||
|
|
||||||
PortType string `json:"portType"`
|
|
||||||
Number uint16 `json:"portNumber"`
|
|
||||||
|
|
||||||
Target bool `json:"target,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type PortScanHistory struct {
|
|
||||||
ID int `json:"id,omitempty"`
|
|
||||||
CreateDate timestamp.Timestamp `json:"createDate,omitempty"`
|
|
||||||
Host *DiscoveryHost `json:"-"`
|
|
||||||
PortNumber uint16 `json:"portNumber"`
|
|
||||||
PortType string `json:"portType"`
|
|
||||||
DirectionType string `json:"directionType"`
|
|
||||||
Description string `json:"description"`
|
|
||||||
StartDate timestamp.Timestamp `json:"startDate"`
|
|
||||||
EndDate timestamp.Timestamp `json:"endDate"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type PortMgr struct {
|
|
||||||
DiscoveryMgr
|
|
||||||
FromHost chan *DiscoveryHost
|
|
||||||
ToService chan *DiscoveryPort
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewPort(port string, host *DiscoveryHost, portType string) *DiscoveryPort {
|
|
||||||
rPort := &DiscoveryPort{Port_: port, Host: host, PortType: portType}
|
|
||||||
i, _ := strconv.ParseUint(port, 10, 16)
|
|
||||||
rPort.Number = uint16(i)
|
|
||||||
return rPort
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewPortScanHistory(host *DiscoveryHost, port uint16, portType string, directionType string, description string) *PortScanHistory {
|
|
||||||
rPortSH := &PortScanHistory{
|
|
||||||
CreateDate: timestamp.Now(),
|
|
||||||
Host: host,
|
|
||||||
PortNumber: port,
|
|
||||||
PortType: portType,
|
|
||||||
DirectionType: directionType,
|
|
||||||
Description: description,
|
|
||||||
}
|
|
||||||
host.AddHistory(rPortSH)
|
|
||||||
return rPortSH
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *DiscoveryPort) SetHistory(h []*ServiceScanHistory) {
|
|
||||||
p.Histories = h
|
|
||||||
}
|
|
||||||
|
|
@ -1,45 +0,0 @@
|
||||||
package types
|
|
||||||
|
|
||||||
import (
|
|
||||||
"git.loafle.net/overflow/overflow_discovery/discovery/types/timestamp"
|
|
||||||
)
|
|
||||||
|
|
||||||
type DiscoveryService struct {
|
|
||||||
ID int `json:"id,omitempty"`
|
|
||||||
CreateDate timestamp.Timestamp `json:"createDate,omitempty"`
|
|
||||||
UpdateDate timestamp.Timestamp `json:"updateDate,omitempty"`
|
|
||||||
Port *DiscoveryPort `json:"-"`
|
|
||||||
PortType string `json:"portType"` /*tls or normal*/
|
|
||||||
ServiceName string `json:"serviceName"`
|
|
||||||
|
|
||||||
Target bool `json:"target,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewService(ptype string, serviceName string, port *DiscoveryPort) *DiscoveryService {
|
|
||||||
service := &DiscoveryService{PortType: ptype, ServiceName: serviceName, Port: port}
|
|
||||||
return service
|
|
||||||
}
|
|
||||||
|
|
||||||
const (
|
|
||||||
DIRECTION_SEND = "Send"
|
|
||||||
DIRECTION_RECV = "Recv"
|
|
||||||
)
|
|
||||||
|
|
||||||
type ServiceScanHistory struct {
|
|
||||||
ID int `json:"id,omitempty"`
|
|
||||||
CreateDate timestamp.Timestamp `json:"createDate,omitempty"`
|
|
||||||
Port *DiscoveryPort `json:"-"`
|
|
||||||
ServiceName string `json:"serviceName"`
|
|
||||||
Direction string `json:"direction"`
|
|
||||||
Packet []byte `json:"packet"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewServiceScanHistory(port *DiscoveryPort, serviceName string, direction string, packet []byte) *ServiceScanHistory {
|
|
||||||
return &ServiceScanHistory{
|
|
||||||
CreateDate: timestamp.Now(),
|
|
||||||
Port: port,
|
|
||||||
ServiceName: serviceName,
|
|
||||||
Direction: direction,
|
|
||||||
Packet: packet,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,53 +0,0 @@
|
||||||
package types
|
|
||||||
|
|
||||||
import (
|
|
||||||
"net"
|
|
||||||
"sync"
|
|
||||||
)
|
|
||||||
|
|
||||||
type discoverTestZoneCallback func(host *DiscoveryZone)
|
|
||||||
|
|
||||||
type DiscoveryZone struct {
|
|
||||||
ID int `json:"id,omitempty"`
|
|
||||||
IP string `json:"ip"`
|
|
||||||
Netmask string `json:"mask"`
|
|
||||||
Iface string `json:"iface"`
|
|
||||||
Mac string `json:"mac"`
|
|
||||||
FirstScanRange int64 `json:"firstScanRange"`
|
|
||||||
LastScanRange int64 `json:"lastScanRange"`
|
|
||||||
Histories []*HostScanHistory `json:"-"`
|
|
||||||
Hosts []*DiscoveryHost `json:"hosts"`
|
|
||||||
CurrentTypes string `json:"-"`
|
|
||||||
|
|
||||||
Hosts_ map[string]*DiscoveryHost `json:"-"`
|
|
||||||
M *sync.RWMutex `json:"-"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (z *DiscoveryZone) SetHost(ip string, value *DiscoveryHost) {
|
|
||||||
z.M.Lock()
|
|
||||||
defer z.M.Unlock()
|
|
||||||
z.Hosts_[ip] = value
|
|
||||||
}
|
|
||||||
|
|
||||||
func (z *DiscoveryZone) GetHost(ip string) (*DiscoveryHost, bool) {
|
|
||||||
z.M.RLock()
|
|
||||||
defer z.M.RUnlock()
|
|
||||||
h, ok := z.Hosts_[ip]
|
|
||||||
return h, ok
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *DiscoveryZone) AddHistory(h *HostScanHistory) {
|
|
||||||
p.Histories = append(p.Histories, h)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (z *DiscoveryZone) CidrString() string {
|
|
||||||
netmask := net.IPMask(net.ParseIP(z.Netmask))
|
|
||||||
str := z.IP + "/" + netmask.String()
|
|
||||||
|
|
||||||
_, ipnet, err := net.ParseCIDR(str)
|
|
||||||
if nil != err {
|
|
||||||
//logging.Logger.Error(fmt.Sprintf("ParseCIDR: %v", err))
|
|
||||||
}
|
|
||||||
|
|
||||||
return ipnet.String()
|
|
||||||
}
|
|
||||||
|
|
@ -1,272 +0,0 @@
|
||||||
package types
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
|
||||||
"net"
|
|
||||||
//"strconv"
|
|
||||||
//"strings"
|
|
||||||
"testing"
|
|
||||||
//"bytes"
|
|
||||||
//"encoding/binary"
|
|
||||||
//"bytes"
|
|
||||||
//"io/ioutil"
|
|
||||||
"git.loafle.net/overflow/overflow_discovery/discovery/types/timestamp"
|
|
||||||
//"net/http"
|
|
||||||
)
|
|
||||||
|
|
||||||
type DiscoveryZone222 struct {
|
|
||||||
MinIp string `json:"-"`
|
|
||||||
MaxIp string `json:"-"`
|
|
||||||
LocalIp string `json:"ip"`
|
|
||||||
Cidr string `json:"cidr"`
|
|
||||||
Inter net.Interface `json:"-"`
|
|
||||||
//Hosts map[string]*Host `json:"-"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestZone222(t *testing.T) {
|
|
||||||
|
|
||||||
zz := DiscoveryZone222{}
|
|
||||||
|
|
||||||
zz.MaxIp = "192.168.1.254"
|
|
||||||
zz.MinIp = "192.168.1.1"
|
|
||||||
zz.LocalIp = "192.168.1.207"
|
|
||||||
zz.Cidr = "192.168.1.0/24"
|
|
||||||
|
|
||||||
jb, _ := json.Marshal(zz)
|
|
||||||
|
|
||||||
fmt.Println(string(jb))
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
type DiscoveryZone1 struct {
|
|
||||||
MinIp string `json:"-"`
|
|
||||||
MaxIp string `json:"-"`
|
|
||||||
Ip_ string `json:"-"`
|
|
||||||
Cidr_ string `json:"-"`
|
|
||||||
Inter_ net.Interface `json:"-"`
|
|
||||||
//Hosts map[string]*Host `json:"-"`
|
|
||||||
|
|
||||||
Ip int64 `json:"ip"`
|
|
||||||
Cidr int64 `json:"cidr"`
|
|
||||||
|
|
||||||
HostScanHistories []*DiscoveryHostHistory `json:"hostScanHistories"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type DiscoveryHostHistory struct {
|
|
||||||
ID int `json:"id"`
|
|
||||||
Ip int64 `json:"ip"`
|
|
||||||
|
|
||||||
CreateDate timestamp.Timestamp `json:"createDate"`
|
|
||||||
SendDate timestamp.Timestamp `json:"sendDate"`
|
|
||||||
ResultDate timestamp.Timestamp `json:"resultDate"`
|
|
||||||
ResultType string `json:"resultType"`
|
|
||||||
Description string `json:"description"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDiscoveryZone(t *testing.T) {
|
|
||||||
|
|
||||||
zones := NewZones()
|
|
||||||
|
|
||||||
fmt.Println(len(zones))
|
|
||||||
fmt.Println(zones[0].LocalIp_)
|
|
||||||
fmt.Println(zones[0].Cidr)
|
|
||||||
|
|
||||||
var zone DiscoveryZone = DiscoveryZone{
|
|
||||||
MinIp_: "192.168.0.1",
|
|
||||||
MaxIp_: "192.168.0.254",
|
|
||||||
Cidr: "192.168.0.0/24",
|
|
||||||
LocalIp_: "192.168.0.203",
|
|
||||||
Inter_: net.Interface{},
|
|
||||||
Hosts_: make(map[string]*DiscoveryHost, 255),
|
|
||||||
Ip: 233312321,
|
|
||||||
Cidr_: 123123321,
|
|
||||||
}
|
|
||||||
|
|
||||||
var zones1 []*DiscoveryZone
|
|
||||||
zones1 = append(zones1, &zone)
|
|
||||||
z := NewZonesByDiscoveryZone(zones1)
|
|
||||||
|
|
||||||
fmt.Println(len(z))
|
|
||||||
//for _, zone := range zones {
|
|
||||||
// //a := inet_strton(zone.LocalIp)
|
|
||||||
// //fmt.Println(a)
|
|
||||||
// //
|
|
||||||
// //str := inet_ntoa(a)
|
|
||||||
// //
|
|
||||||
// //fmt.Println(str)
|
|
||||||
//
|
|
||||||
// //aa := CidrToInt64(zone.Cidr)
|
|
||||||
// //
|
|
||||||
// //fmt.Println(aa)
|
|
||||||
// //
|
|
||||||
// //str := Int64ToCidr(aa)
|
|
||||||
// //
|
|
||||||
// //fmt.Println(str)
|
|
||||||
//
|
|
||||||
//}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
//func CreateDHH(ip string) *DiscoveryHostHistory {
|
|
||||||
// dhh := &DiscoveryHostHistory{}
|
|
||||||
//
|
|
||||||
// dhh.CreateDate = timestamp.Now()
|
|
||||||
// dhh.SendDate = timestamp.Now()
|
|
||||||
// dhh.ResultDate = timestamp.Now()
|
|
||||||
// dhh.Description = "testest"
|
|
||||||
//
|
|
||||||
// strIp := "192.168.1."
|
|
||||||
// strIp += ip
|
|
||||||
// dhh.Ip = inet_strton(strIp)
|
|
||||||
//
|
|
||||||
// return dhh
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func TestInsertDZone(t *testing.T) {
|
|
||||||
//
|
|
||||||
// dz := DiscoveryZone{}
|
|
||||||
//
|
|
||||||
// dz.Cidr_ = "192.168.1.0/24"
|
|
||||||
// dz.Ip_ = "192.168.1.207"
|
|
||||||
//
|
|
||||||
// dz.Cidr = CidrToInt64(dz.Cidr_)
|
|
||||||
// dz.Ip = inet_strton(dz.Ip_)
|
|
||||||
//
|
|
||||||
// dz.HostScanHistories = append(dz.HostScanHistories, CreateDHH("103"), CreateDHH("104"), CreateDHH("105"), CreateDHH("106"))
|
|
||||||
//
|
|
||||||
// fmt.Println(inet_ntoa(dz.Ip))
|
|
||||||
// fmt.Println(Int64ToCidr(dz.Cidr))
|
|
||||||
//
|
|
||||||
// jb, _ := json.Marshal(dz)
|
|
||||||
//
|
|
||||||
// fmt.Println(string(jb))
|
|
||||||
//
|
|
||||||
// resp, _ := http.Post("http://192.168.1.207:8080/discoveryZone", "application/json", bytes.NewBuffer(jb))
|
|
||||||
// bb, _ := ioutil.ReadAll(resp.Body)
|
|
||||||
//
|
|
||||||
// ddzz := DiscoveryZone{}
|
|
||||||
//
|
|
||||||
// json.Unmarshal(bb, &ddzz)
|
|
||||||
//
|
|
||||||
// fmt.Println(ddzz)
|
|
||||||
//
|
|
||||||
// resp.Body.Close()
|
|
||||||
//
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func TestSelectDZone(t *testing.T) {
|
|
||||||
//
|
|
||||||
// resp, _ := http.Get("http://192.168.1.207:8080/discoveryZone/1")
|
|
||||||
// bb, _ := ioutil.ReadAll(resp.Body)
|
|
||||||
// ddzz := DiscoveryZone{}
|
|
||||||
//
|
|
||||||
// fmt.Println(string(bb))
|
|
||||||
// json.Unmarshal(bb, &ddzz)
|
|
||||||
//
|
|
||||||
// fmt.Println(ddzz)
|
|
||||||
//
|
|
||||||
// fmt.Println(Int64ToCidr(ddzz.Cidr))
|
|
||||||
// fmt.Println(inet_ntoa(ddzz.Ip))
|
|
||||||
//
|
|
||||||
// resp.Body.Close()
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func inet_ntoa(ipnr int64) net.IP {
|
|
||||||
// var bytes [4]byte
|
|
||||||
// bytes[0] = byte(ipnr & 0xFF)
|
|
||||||
// bytes[1] = byte((ipnr >> 8) & 0xFF)
|
|
||||||
// bytes[2] = byte((ipnr >> 16) & 0xFF)
|
|
||||||
// bytes[3] = byte((ipnr >> 24) & 0xFF)
|
|
||||||
//
|
|
||||||
// return net.IPv4(bytes[3], bytes[2], bytes[1], bytes[0])
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//// Convert net.IP to int64 , http://www.outofmemory.cn
|
|
||||||
//func inet_aton(ipnr net.IP) int64 {
|
|
||||||
// bits := strings.Split(ipnr.String(), ".")
|
|
||||||
//
|
|
||||||
// b0, _ := strconv.Atoi(bits[0])
|
|
||||||
// b1, _ := strconv.Atoi(bits[1])
|
|
||||||
// b2, _ := strconv.Atoi(bits[2])
|
|
||||||
// b3, _ := strconv.Atoi(bits[3])
|
|
||||||
//
|
|
||||||
// var sum int64
|
|
||||||
//
|
|
||||||
// sum += int64(b0) << 24
|
|
||||||
// sum += int64(b1) << 16
|
|
||||||
// sum += int64(b2) << 8
|
|
||||||
// sum += int64(b3)
|
|
||||||
//
|
|
||||||
// return sum
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func inet_strton(ipnr string) int64 {
|
|
||||||
// bits := strings.Split(ipnr, ".")
|
|
||||||
//
|
|
||||||
// b0, _ := strconv.Atoi(bits[0])
|
|
||||||
// b1, _ := strconv.Atoi(bits[1])
|
|
||||||
// b2, _ := strconv.Atoi(bits[2])
|
|
||||||
// b3, _ := strconv.Atoi(bits[3])
|
|
||||||
//
|
|
||||||
// var sum int64
|
|
||||||
//
|
|
||||||
// sum += int64(b0) << 24
|
|
||||||
// sum += int64(b1) << 16
|
|
||||||
// sum += int64(b2) << 8
|
|
||||||
// sum += int64(b3)
|
|
||||||
//
|
|
||||||
// return sum
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func CidrToInt64(cidr string) int64 {
|
|
||||||
//
|
|
||||||
// bits := strings.Split(cidr, ".")
|
|
||||||
//
|
|
||||||
// b0, _ := strconv.Atoi(bits[0])
|
|
||||||
// b1, _ := strconv.Atoi(bits[1])
|
|
||||||
// b2, _ := strconv.Atoi(bits[2])
|
|
||||||
// //b3, _ := strconv.Atoi(bits[3])
|
|
||||||
//
|
|
||||||
// masks := strings.Split(bits[3], "/")
|
|
||||||
//
|
|
||||||
// bb3 := masks[0]
|
|
||||||
// bb4 := masks[1]
|
|
||||||
//
|
|
||||||
// b3, _ := strconv.Atoi(bb3)
|
|
||||||
// b4, _ := strconv.Atoi(bb4)
|
|
||||||
//
|
|
||||||
// var sum int64
|
|
||||||
//
|
|
||||||
// sum += int64(b0) << 32
|
|
||||||
// sum += int64(b1) << 24
|
|
||||||
// sum += int64(b2) << 16
|
|
||||||
// sum += int64(b3) << 8
|
|
||||||
// sum += int64(b4)
|
|
||||||
//
|
|
||||||
// return sum
|
|
||||||
//
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func Int64ToCidr(cidrInt int64) string {
|
|
||||||
//
|
|
||||||
// c1 := cidrInt & 0xFF
|
|
||||||
// c2 := (cidrInt >> 8) & 0xFF
|
|
||||||
// c3 := (cidrInt >> 16) & 0xFF
|
|
||||||
// c4 := (cidrInt >> 24) & 0xFF
|
|
||||||
// c5 := (cidrInt >> 32) & 0xFF
|
|
||||||
//
|
|
||||||
// var cidr string
|
|
||||||
//
|
|
||||||
// cidr += strconv.FormatInt(c5, 10)
|
|
||||||
// cidr += "."
|
|
||||||
// cidr += strconv.FormatInt(c4, 10)
|
|
||||||
// cidr += "."
|
|
||||||
// cidr += strconv.FormatInt(c3, 10)
|
|
||||||
// cidr += "."
|
|
||||||
// cidr += strconv.FormatInt(c2, 10)
|
|
||||||
// cidr += "/"
|
|
||||||
// cidr += strconv.FormatInt(c1, 10)
|
|
||||||
//
|
|
||||||
// return cidr
|
|
||||||
//}
|
|
||||||
|
|
@ -1,12 +0,0 @@
|
||||||
package types
|
|
||||||
|
|
||||||
type ServiceScanInfo struct {
|
|
||||||
History []*ServiceScanHistory
|
|
||||||
Port *DiscoveryPort
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewServiceScanInfo(port *DiscoveryPort) *ServiceScanInfo {
|
|
||||||
s := &ServiceScanInfo{}
|
|
||||||
s.Port = port
|
|
||||||
return s
|
|
||||||
}
|
|
||||||
|
|
@ -1,37 +0,0 @@
|
||||||
package timestamp
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"strconv"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
type Timestamp time.Time
|
|
||||||
|
|
||||||
func (t Timestamp) MarshalJSON() ([]byte, error) {
|
|
||||||
ts := time.Time(t).Unix()
|
|
||||||
stamp := fmt.Sprint(ts * 1000)
|
|
||||||
return []byte(stamp), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *Timestamp) UnmarshalJSON(b []byte) error {
|
|
||||||
ts, err := strconv.Atoi(string(b))
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
*t = Timestamp(time.Unix(int64(ts)/1000, 0))
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t Timestamp) String() string {
|
|
||||||
return time.Time(t).String()
|
|
||||||
}
|
|
||||||
|
|
||||||
func Now() Timestamp {
|
|
||||||
return Timestamp(time.Now())
|
|
||||||
}
|
|
||||||
|
|
||||||
func Date(year int, month time.Month, day int) Timestamp {
|
|
||||||
return Timestamp(time.Date(year, month, day, 0, 0, 0, 0, time.UTC))
|
|
||||||
}
|
|
||||||
|
|
@ -1,31 +0,0 @@
|
||||||
package discovery
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"git.loafle.net/overflow/overflow_discovery/commons"
|
|
||||||
)
|
|
||||||
|
|
||||||
type zoneDiscoverer interface {
|
|
||||||
commons.EndableStarter
|
|
||||||
commons.Shutdowner
|
|
||||||
}
|
|
||||||
|
|
||||||
type zoneDiscovery struct {
|
|
||||||
}
|
|
||||||
|
|
||||||
func newZoneDiscoverer() zoneDiscoverer {
|
|
||||||
zd := &zoneDiscovery{}
|
|
||||||
|
|
||||||
return zd
|
|
||||||
}
|
|
||||||
|
|
||||||
func (zd *zoneDiscovery) EndableStart(endded chan<- error) error {
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (zd *zoneDiscovery) Shutdown(ctx context.Context) error {
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
5
build.sh
Executable file
5
build.sh
Executable file
|
|
@ -0,0 +1,5 @@
|
||||||
|
rm ./dist/*
|
||||||
|
CGO_ENABLED=1 go build -a --installsuffix cgo --ldflags="-s" -o ./dist/overflow_probe_container_discovery
|
||||||
|
|
||||||
|
chmod u+s ./dist/overflow_probe_container_discovery
|
||||||
|
cp ./dist/overflow_probe_container_discovery /project/overFlow/
|
||||||
|
|
@ -39,8 +39,8 @@ type PCapScanner interface {
|
||||||
}
|
}
|
||||||
|
|
||||||
type pCapScan struct {
|
type pCapScan struct {
|
||||||
zone *discoveryM.Zone
|
|
||||||
pCapHandle *pcap.Handle
|
pCapHandle *pcap.Handle
|
||||||
|
zone *discoveryM.Zone
|
||||||
|
|
||||||
arpListenerChanMtx sync.RWMutex
|
arpListenerChanMtx sync.RWMutex
|
||||||
arpListenerChans []chan *layers.ARP
|
arpListenerChans []chan *layers.ARP
|
||||||
|
|
|
||||||
1
main.go
1
main.go
|
|
@ -19,7 +19,6 @@ var (
|
||||||
func init() {
|
func init() {
|
||||||
pidPath = flag.String("pid-path", "/tmp/discovery-container.pid", "The path of pid file")
|
pidPath = flag.String("pid-path", "/tmp/discovery-container.pid", "The path of pid file")
|
||||||
flag.Parse()
|
flag.Parse()
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user