mirror of
https://github.com/netbirdio/netbird.git
synced 2025-07-20 16:03:23 +02:00
c02e236196
adds NetFlow functionality to track and log network traffic information between peers, with features including: - Flow logging for TCP, UDP, and ICMP traffic - Integration with connection tracking system - Resource ID tracking in NetFlow events - DNS and exit node collection configuration - Flow API and Redis cache in management - Memory-based flow storage implementation - Kernel conntrack counters and userspace counters - TCP state machine improvements for more accurate tracking - Migration from net.IP to netip.Addr in the userspace firewall
82 lines
1.8 KiB
Go
82 lines
1.8 KiB
Go
package peer
|
|
|
|
import (
|
|
"net/netip"
|
|
"sync"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type routeIDLookup struct {
|
|
localMap sync.Map
|
|
remoteMap sync.Map
|
|
resolvedIPs sync.Map
|
|
}
|
|
|
|
func (r *routeIDLookup) AddLocalRouteID(resourceID string, route netip.Prefix) {
|
|
_, exists := r.localMap.LoadOrStore(route, resourceID)
|
|
if exists {
|
|
log.Tracef("resourceID %s already exists in local map", resourceID)
|
|
}
|
|
}
|
|
|
|
func (r *routeIDLookup) RemoveLocalRouteID(route netip.Prefix) {
|
|
r.localMap.Delete(route)
|
|
}
|
|
|
|
func (r *routeIDLookup) AddRemoteRouteID(resourceID string, route netip.Prefix) {
|
|
_, exists := r.remoteMap.LoadOrStore(route, resourceID)
|
|
if exists {
|
|
log.Tracef("resourceID %s already exists in remote map", resourceID)
|
|
}
|
|
}
|
|
|
|
func (r *routeIDLookup) RemoveRemoteRouteID(route netip.Prefix) {
|
|
r.remoteMap.Delete(route)
|
|
}
|
|
|
|
func (r *routeIDLookup) AddResolvedIP(resourceID string, route netip.Prefix) {
|
|
r.resolvedIPs.Store(route.Addr(), resourceID)
|
|
}
|
|
|
|
func (r *routeIDLookup) RemoveResolvedIP(route netip.Prefix) {
|
|
r.resolvedIPs.Delete(route.Addr())
|
|
}
|
|
|
|
// Lookup returns the resource ID for the given IP address
|
|
// and a bool indicating if the IP is an exit node
|
|
func (r *routeIDLookup) Lookup(ip netip.Addr) (string, bool) {
|
|
var isExitNode bool
|
|
|
|
resId, ok := r.resolvedIPs.Load(ip)
|
|
if ok {
|
|
return resId.(string), false
|
|
}
|
|
|
|
var resourceID string
|
|
r.localMap.Range(func(key, value interface{}) bool {
|
|
pref := key.(netip.Prefix)
|
|
if pref.Contains(ip) {
|
|
resourceID = value.(string)
|
|
isExitNode = pref.Bits() == 0
|
|
return false
|
|
|
|
}
|
|
return true
|
|
})
|
|
|
|
if resourceID == "" {
|
|
r.remoteMap.Range(func(key, value interface{}) bool {
|
|
pref := key.(netip.Prefix)
|
|
if pref.Contains(ip) {
|
|
resourceID = value.(string)
|
|
isExitNode = pref.Bits() == 0
|
|
return false
|
|
}
|
|
return true
|
|
})
|
|
}
|
|
|
|
return resourceID, isExitNode
|
|
}
|