conntrack.go

package conntrack

import (
	"strconv"
	"time"

	"inet.af/netaddr"
)

type Entry struct {
	Src       netaddr.IPPort
	Dst       netaddr.IPPort
	TxBytes   uint64
	TxPackets uint64
	RxBytes   uint64
	RxPackets uint64
	Lifetime  time.Time
	Proto     uint8
}

type EntriesFilter func(e *Entry) bool

func All() EntriesFilter {
	return func(e *Entry) bool {
		return true
	}
}

func FilterBySrcIP(ips map[netaddr.IP]struct{}) EntriesFilter {
	return func(e *Entry) bool {
		_, found := ips[e.Src.IP()]
		return found
	}
}

type Client interface {
	ListEntries(filter EntriesFilter) ([]*Entry, error)
	Close() error
}

var protoNames = map[uint8]string{
	0:  "ANY",
	1:  "ICMP",
	6:  "TCP",
	17: "UDP",
	58: "ICMPv6",
}

func ProtoString(p uint8) string {
	if _, ok := protoNames[p]; ok {
		return protoNames[p]
	}
	return strconv.Itoa(int(p))
}