god-eye/internal/modules/asn/asn.go

// Package asn is a Fase 0.6 adapter around v1 network.ASNScanner. Expands
// discovery by enumerating IPs within the target's ASN/CIDR blocks.
package asn

import (
	"time"

	"god-eye/internal/eventbus"
	"god-eye/internal/module"
	"god-eye/internal/network"
	"god-eye/internal/store"
)

// CtxPassthrough is used to thread module.Context.Ctx into network helpers.

const ModuleName = "discovery.asn"

type asnModule struct{}

func Register() { module.Register(&asnModule{}) }

func (*asnModule) Name() string                   { return ModuleName }
func (*asnModule) Phase() module.Phase            { return module.PhaseResolution }
func (*asnModule) Consumes() []eventbus.EventType { return []eventbus.EventType{eventbus.EventDNSResolved} }
func (*asnModule) Produces() []eventbus.EventType { return nil }
func (*asnModule) DefaultEnabled() bool           { return false } // opt-in

func (*asnModule) Run(mctx module.Context) error {
	if !mctx.Config.Bool("asn_scan", false) {
		return nil
	}
	timeout := mctx.Config.Int("timeout", 10)

	hosts := mctx.Store.All(mctx.Ctx)
	seenIP := make(map[string]struct{})
	for _, h := range hosts {
		for _, ip := range h.IPs {
			seenIP[ip] = struct{}{}
		}
	}

	scanner := network.NewASNScanner(timeout)
	for ip := range seenIP {
		if mctx.Ctx.Err() != nil {
			break
		}
		info, err := scanner.GetASNInfo(mctx.Ctx, ip)
		if err != nil || info == nil {
			continue
		}
		_ = mctx.Store.Upsert(mctx.Ctx, ipToFirstHost(mctx, ip), func(h *store.Host) {
			if h.ASN == "" {
				h.ASN = info.ASN
			}
			if h.Org == "" {
				h.Org = info.Name
			}
			if h.Country == "" {
				h.Country = info.Country
			}
		})
	}
	return nil
}

// ipToFirstHost returns the first subdomain mapped to ip in the store.
func ipToFirstHost(mctx module.Context, ip string) string {
	for _, h := range mctx.Store.All(mctx.Ctx) {
		for _, rip := range h.IPs {
			if rip == ip {
				return h.Subdomain
			}
		}
	}
	return ""
}

var _ = time.Now