netbird/client/internal/acl/id/id.go

package id

import (
	"crypto/sha256"
	"encoding/hex"
	"fmt"
	"net/netip"
	"strconv"

	"github.com/netbirdio/netbird/client/firewall/manager"
)

type RuleID string

func (r RuleID) GetRuleID() string {
	return string(r)
}

func GenerateRouteRuleKey(
	sources []netip.Prefix,
	destination netip.Prefix,
	proto manager.Protocol,
	sPort *manager.Port,
	dPort *manager.Port,
	action manager.Action,
) RuleID {
	manager.SortPrefixes(sources)

	h := sha256.New()

	// Write all fields to the hasher, with delimiters
	h.Write([]byte("sources:"))
	for _, src := range sources {
		h.Write([]byte(src.String()))
		h.Write([]byte(","))
	}

	h.Write([]byte("destination:"))
	h.Write([]byte(destination.String()))

	h.Write([]byte("proto:"))
	h.Write([]byte(proto))

	h.Write([]byte("sPort:"))
	if sPort != nil {
		h.Write([]byte(sPort.String()))
	} else {
		h.Write([]byte("<nil>"))
	}

	h.Write([]byte("dPort:"))
	if dPort != nil {
		h.Write([]byte(dPort.String()))
	} else {
		h.Write([]byte("<nil>"))
	}

	h.Write([]byte("action:"))
	h.Write([]byte(strconv.Itoa(int(action))))
	hash := hex.EncodeToString(h.Sum(nil))

	// prepend destination prefix to be able to identify the rule
	return RuleID(fmt.Sprintf("%s-%s", destination.String(), hash[:16]))
}