Files
ziti/controller/network/network.go
T
Paul Lorenz 8d11af3d97 Optimize network run loop. Fixes #1897
* When routers change (connect/disconnect), we re-check the mesh. We are passing in the routers to the channel and they can build up, which is unnecessary. We only need to signal once for all the routers that change since the last time we checked, that the mesh needs to be checked.
* We pass link changes through the run method, which is unnecessary since we pass it to a new goroutine. This inefficiency should be fixed.
* We're evaluating all faulted links for rerouting, even if they were still pending. Only reroute connected links.
* We're using time.After in each for loop, which can accumulate timers. Use a single Ticker instead.
2024-04-04 18:46:50 -04:00

1484 lines
44 KiB
Go

/*
Copyright NetFoundry Inc.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
https://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package network
import (
"bytes"
"compress/gzip"
"encoding/json"
"fmt"
"github.com/openziti/foundation/v2/concurrenz"
"github.com/openziti/foundation/v2/goroutines"
"github.com/openziti/storage/objectz"
fabricMetrics "github.com/openziti/ziti/common/metrics"
"github.com/openziti/ziti/common/pb/mgmt_pb"
"github.com/openziti/ziti/controller/event"
"github.com/openziti/ziti/controller/raft"
"os"
"path/filepath"
"runtime/debug"
"sort"
"strings"
"sync"
"time"
"github.com/openziti/foundation/v2/versions"
"github.com/openziti/ziti/controller/command"
"github.com/michaelquigley/pfxlog"
"github.com/openziti/channel/v2/protobufs"
"github.com/openziti/foundation/v2/debugz"
"github.com/openziti/foundation/v2/errorz"
"github.com/openziti/foundation/v2/sequence"
"github.com/openziti/identity"
"github.com/openziti/metrics"
"github.com/openziti/metrics/metrics_pb"
"github.com/openziti/storage/boltz"
"github.com/openziti/ziti/common/ctrl_msg"
"github.com/openziti/ziti/common/logcontext"
"github.com/openziti/ziti/common/pb/ctrl_pb"
"github.com/openziti/ziti/common/trace"
"github.com/openziti/ziti/controller/db"
"github.com/openziti/ziti/controller/xt"
"github.com/pkg/errors"
"github.com/sirupsen/logrus"
"go.etcd.io/bbolt"
)
const SmartRerouteAttempt = 99969996
// Config provides the values needed to create a Network instance
type Config interface {
GetId() *identity.TokenId
GetMetricsRegistry() metrics.Registry
GetOptions() *Options
GetCommandDispatcher() command.Dispatcher
GetDb() boltz.Db
GetVersionProvider() versions.VersionProvider
GetEventDispatcher() event.Dispatcher
GetCloseNotify() <-chan struct{}
}
type Network struct {
*Managers
nodeId string
options *Options
assembleAndCleanC chan struct{}
linkController *linkController
forwardingFaults chan struct{}
circuitController *circuitController
routeSenderController *routeSenderController
sequence *sequence.Sequence
eventDispatcher event.Dispatcher
traceController trace.Controller
routerPresenceHandlers []RouterPresenceHandler
capabilities []string
closeNotify <-chan struct{}
watchdogCh chan struct{}
lock sync.Mutex
strategyRegistry xt.Registry
lastSnapshot time.Time
metricsRegistry metrics.Registry
VersionProvider versions.VersionProvider
serviceEventMetrics metrics.UsageRegistry
serviceDialSuccessCounter metrics.IntervalCounter
serviceDialFailCounter metrics.IntervalCounter
serviceDialTimeoutCounter metrics.IntervalCounter
serviceDialOtherErrorCounter metrics.IntervalCounter
serviceTerminatorTimeoutCounter metrics.IntervalCounter
serviceTerminatorConnectionRefusedCounter metrics.IntervalCounter
serviceInvalidTerminatorCounter metrics.IntervalCounter
serviceMisconfiguredTerminatorCounter metrics.IntervalCounter
config Config
}
func NewNetwork(config Config) (*Network, error) {
stores, err := db.InitStores(config.GetDb(), config.GetCommandDispatcher().GetRateLimiter())
if err != nil {
return nil, err
}
if config.GetOptions().IntervalAgeThreshold != 0 {
metrics.SetIntervalAgeThreshold(config.GetOptions().IntervalAgeThreshold)
logrus.Infof("set interval age threshold to '%v'", config.GetOptions().IntervalAgeThreshold)
}
serviceEventMetrics := metrics.NewUsageRegistry(config.GetId().Token, nil, config.GetCloseNotify())
network := &Network{
nodeId: config.GetId().Token,
options: config.GetOptions(),
assembleAndCleanC: make(chan struct{}, 1),
linkController: newLinkController(config.GetOptions()),
forwardingFaults: make(chan struct{}, 1),
circuitController: newCircuitController(),
routeSenderController: newRouteSenderController(),
sequence: sequence.NewSequence(),
eventDispatcher: config.GetEventDispatcher(),
traceController: trace.NewController(config.GetCloseNotify()),
closeNotify: config.GetCloseNotify(),
watchdogCh: make(chan struct{}, 1),
strategyRegistry: xt.GlobalRegistry(),
lastSnapshot: time.Now().Add(-time.Hour),
metricsRegistry: config.GetMetricsRegistry(),
VersionProvider: config.GetVersionProvider(),
serviceEventMetrics: serviceEventMetrics,
serviceDialSuccessCounter: serviceEventMetrics.IntervalCounter("service.dial.success", time.Minute),
serviceDialFailCounter: serviceEventMetrics.IntervalCounter("service.dial.fail", time.Minute),
serviceDialTimeoutCounter: serviceEventMetrics.IntervalCounter("service.dial.timeout", time.Minute),
serviceDialOtherErrorCounter: serviceEventMetrics.IntervalCounter("service.dial.error_other", time.Minute),
serviceTerminatorTimeoutCounter: serviceEventMetrics.IntervalCounter("service.dial.terminator.timeout", time.Minute),
serviceTerminatorConnectionRefusedCounter: serviceEventMetrics.IntervalCounter("service.dial.terminator.connection_refused", time.Minute),
serviceInvalidTerminatorCounter: serviceEventMetrics.IntervalCounter("service.dial.terminator.invalid", time.Minute),
serviceMisconfiguredTerminatorCounter: serviceEventMetrics.IntervalCounter("service.dial.terminator.misconfigured", time.Minute),
config: config,
}
routerCommPool, err := network.createRouterCommPool(config)
if err != nil {
return nil, err
}
network.Managers = NewManagers(network, config.GetCommandDispatcher(), config.GetDb(), stores, routerCommPool)
network.Managers.Inspections.network = network
network.AddCapability("ziti.fabric")
network.showOptions()
network.relayControllerMetrics()
network.AddRouterPresenceHandler(network.Managers.RouterMessaging)
go network.Managers.RouterMessaging.run()
return network, nil
}
func (network *Network) createRouterCommPool(config Config) (goroutines.Pool, error) {
poolConfig := goroutines.PoolConfig{
QueueSize: config.GetOptions().RouterComm.QueueSize,
MinWorkers: 0,
MaxWorkers: config.GetOptions().RouterComm.MaxWorkers,
IdleTime: 30 * time.Second,
CloseNotify: config.GetCloseNotify(),
PanicHandler: func(err interface{}) {
pfxlog.Logger().WithField(logrus.ErrorKey, err).WithField("backtrace", string(debug.Stack())).Error("panic during message send to router")
},
}
fabricMetrics.ConfigureGoroutinesPoolMetrics(&poolConfig, config.GetMetricsRegistry(), "pool.router.messaging")
pool, err := goroutines.NewPool(poolConfig)
if err != nil {
return nil, errors.Wrap(err, "error creating router messaging pool")
}
return pool, nil
}
func (network *Network) relayControllerMetrics() {
go func() {
timer := time.NewTicker(network.options.MetricsReportInterval)
defer timer.Stop()
for {
select {
case <-timer.C:
if msg := network.metricsRegistry.Poll(); msg != nil {
network.eventDispatcher.AcceptMetricsMsg(msg)
}
case <-network.closeNotify:
return
}
}
}()
}
func (network *Network) InitServiceCounterDispatch(handler metrics.Handler) {
network.serviceEventMetrics.StartReporting(handler, network.GetOptions().MetricsReportInterval, 10)
}
func (network *Network) GetAppId() string {
return network.nodeId
}
func (network *Network) GetOptions() *Options {
return network.options
}
func (network *Network) GetDb() boltz.Db {
return network.db
}
func (network *Network) GetStores() *db.Stores {
return network.stores
}
func (network *Network) GetManagers() *Managers {
return network.Managers
}
func (network *Network) GetConnectedRouter(routerId string) *Router {
return network.Routers.getConnected(routerId)
}
func (network *Network) GetReloadedRouter(routerId string) (*Router, error) {
network.Routers.RemoveFromCache(routerId)
return network.Routers.Read(routerId)
}
func (network *Network) GetRouter(routerId string) (*Router, error) {
return network.Routers.Read(routerId)
}
func (network *Network) AllConnectedRouters() []*Router {
return network.Routers.allConnected()
}
func (network *Network) GetLink(linkId string) (*Link, bool) {
return network.linkController.get(linkId)
}
func (network *Network) GetAllLinks() []*Link {
return network.linkController.all()
}
func (network *Network) GetAllLinksForRouter(routerId string) []*Link {
r := network.GetConnectedRouter(routerId)
if r == nil {
return nil
}
return r.routerLinks.GetLinks()
}
func (network *Network) GetCircuit(circuitId string) (*Circuit, bool) {
return network.circuitController.get(circuitId)
}
func (network *Network) GetAllCircuits() []*Circuit {
return network.circuitController.all()
}
func (network *Network) GetCircuitStore() *objectz.ObjectStore[*Circuit] {
return network.circuitController.store
}
func (network *Network) GetLinkStore() *objectz.ObjectStore[*Link] {
return network.linkController.store
}
func (network *Network) RouteResult(rs *RouteStatus) bool {
return network.routeSenderController.forwardRouteResult(rs)
}
func (network *Network) newRouteSender(circuitId string) *routeSender {
rs := newRouteSender(circuitId, network.options.RouteTimeout, network, network.Terminators)
network.routeSenderController.addRouteSender(rs)
return rs
}
func (network *Network) removeRouteSender(rs *routeSender) {
network.routeSenderController.removeRouteSender(rs)
}
func (network *Network) GetEventDispatcher() event.Dispatcher {
return network.eventDispatcher
}
func (network *Network) GetTraceController() trace.Controller {
return network.traceController
}
func (network *Network) GetMetricsRegistry() metrics.Registry {
return network.metricsRegistry
}
func (network *Network) GetServiceEventsMetricsRegistry() metrics.UsageRegistry {
return network.serviceEventMetrics
}
func (network *Network) GetCloseNotify() <-chan struct{} {
return network.closeNotify
}
func (network *Network) ConnectedRouter(id string) bool {
return network.Routers.IsConnected(id)
}
func (network *Network) ConnectRouter(r *Router) {
network.linkController.buildRouterLinks(r)
network.Routers.markConnected(r)
time.AfterFunc(250*time.Millisecond, network.notifyAssembleAndClean)
for _, h := range network.routerPresenceHandlers {
go h.RouterConnected(r)
}
go network.ValidateTerminators(r)
}
func (network *Network) ValidateTerminators(r *Router) {
logger := pfxlog.Logger().WithField("routerId", r.Id)
result, err := network.Terminators.Query(fmt.Sprintf(`router.id = "%v" limit none`, r.Id))
if err != nil {
logger.WithError(err).Error("failed to get terminators for router")
return
}
logger.Debugf("%v terminators to validate", len(result.Entities))
if len(result.Entities) == 0 {
return
}
network.Managers.RouterMessaging.ValidateRouterTerminators(result.Entities)
}
type LinkValidationCallback func(detail *mgmt_pb.RouterLinkDetails)
func (n *Network) ValidateLinks(filter string, cb LinkValidationCallback) (int64, func(), error) {
result, err := n.Routers.BaseList(filter)
if err != nil {
return 0, nil, err
}
sem := concurrenz.NewSemaphore(10)
evalF := func() {
for _, router := range result.Entities {
connectedRouter := n.GetConnectedRouter(router.Id)
if connectedRouter != nil {
sem.Acquire()
go func() {
defer sem.Release()
n.linkController.ValidateRouterLinks(n, connectedRouter, cb)
}()
} else {
n.linkController.reportRouterLinksError(router, errors.New("router not connected"), cb)
}
}
}
return int64(len(result.Entities)), evalF, nil
}
type SdkTerminatorValidationCallback func(detail *mgmt_pb.RouterSdkTerminatorsDetails)
func (n *Network) ValidateRouterSdkTerminators(filter string, cb SdkTerminatorValidationCallback) (int64, func(), error) {
result, err := n.Routers.BaseList(filter)
if err != nil {
return 0, nil, err
}
sem := concurrenz.NewSemaphore(10)
evalF := func() {
for _, router := range result.Entities {
connectedRouter := n.GetConnectedRouter(router.Id)
if connectedRouter != nil {
sem.Acquire()
go func() {
defer sem.Release()
n.Routers.ValidateRouterSdkTerminators(connectedRouter, cb)
}()
} else {
n.Routers.reportRouterSdkTerminatorsError(router, errors.New("router not connected"), cb)
}
}
}
return int64(len(result.Entities)), evalF, nil
}
func (network *Network) DisconnectRouter(r *Router) {
// 1: remove Links for Router
for _, l := range r.routerLinks.GetLinks() {
wasConnected := l.CurrentState().Mode == Connected
if l.Src.Id == r.Id {
network.linkController.remove(l)
}
if wasConnected {
network.RerouteLink(l)
}
}
// 2: remove Router
network.Routers.markDisconnected(r)
for _, h := range network.routerPresenceHandlers {
h.RouterDisconnected(r)
}
network.notifyAssembleAndClean()
}
func (network *Network) notifyAssembleAndClean() {
select {
case network.assembleAndCleanC <- struct{}{}:
default:
}
}
func (network *Network) NotifyExistingLink(id string, iteration uint32, linkProtocol, dialAddress string, srcRouter *Router, dstRouterId string) {
log := pfxlog.Logger().
WithField("routerId", srcRouter.Id).
WithField("linkId", id).
WithField("destRouterId", dstRouterId).
WithField("iteration", iteration)
src := network.Routers.getConnected(srcRouter.Id)
if src == nil {
log.Info("ignoring links message processed after router disconnected")
return
}
if src != srcRouter || !srcRouter.Connected.Load() {
log.Info("ignoring links message processed from old router connection")
return
}
dst := network.Routers.getConnected(dstRouterId)
if dst == nil {
network.NotifyLinkIdEvent(id, event.LinkFromRouterDisconnectedDest)
}
link, created := network.linkController.routerReportedLink(id, iteration, linkProtocol, dialAddress, srcRouter, dst, dstRouterId)
if created {
network.NotifyLinkEvent(link, event.LinkFromRouterNew)
log.Info("router reported link added")
} else {
network.NotifyLinkEvent(link, event.LinkFromRouterKnown)
log.Info("router reported link already known")
}
}
func (network *Network) LinkConnected(msg *ctrl_pb.LinkConnected) error {
if l, found := network.linkController.get(msg.Id); found {
if state := l.CurrentState(); state.Mode != Pending {
return errors.Errorf("link [l/%v] state is %v, not pending, cannot mark connected", msg.Id, state.Mode)
}
l.SetState(Connected)
network.NotifyLinkConnected(l, msg)
return nil
}
return errors.Errorf("no such link [l/%s]", msg.Id)
}
func (network *Network) LinkFaulted(l *Link, dupe bool) error {
l.SetState(Failed)
if dupe {
network.NotifyLinkEvent(l, event.LinkDuplicate)
} else {
network.NotifyLinkEvent(l, event.LinkFault)
}
pfxlog.Logger().WithField("linkId", l.Id).Info("removing failed link")
network.linkController.remove(l)
return nil
}
func (network *Network) VerifyRouter(routerId string, fingerprints []string) error {
router, err := network.GetRouter(routerId)
if err != nil {
return err
}
routerFingerprint := router.Fingerprint
if routerFingerprint == nil {
return errors.Errorf("invalid router %v, not yet enrolled", routerId)
}
for _, fp := range fingerprints {
if fp == *routerFingerprint {
return nil
}
}
return errors.Errorf("could not verify fingerprint for router %v", routerId)
}
func (network *Network) RerouteLink(l *Link) {
// This is called from Channel.rxer() and thus may not block
go func() {
network.handleRerouteLink(l)
}()
}
func (network *Network) CreateCircuit(params CreateCircuitParams) (*Circuit, error) {
srcR := params.GetSourceRouter()
clientId := params.GetClientId()
service := params.GetServiceId()
ctx := params.GetLogContext()
deadline := params.GetDeadline()
startTime := time.Now()
instanceId, serviceId := parseInstanceIdAndService(service)
// 1: Allocate Circuit Identifier
circuitId, err := network.circuitController.nextCircuitId()
if err != nil {
network.CircuitFailedEvent(circuitId, params, startTime, nil, nil, CircuitFailureInvalidService)
return nil, err
}
ctx.WithFields(map[string]interface{}{
"circuitId": circuitId,
"serviceId": service,
"attemptNumber": 1,
})
logger := pfxlog.ChannelLogger(logcontext.SelectPath).Wire(ctx).Entry
attempt := uint32(0)
allCleanups := make(map[string]struct{})
rs := network.newRouteSender(circuitId)
defer func() { network.removeRouteSender(rs) }()
for {
// 2: Find Service
svc, err := network.Services.Read(serviceId)
if err != nil {
network.CircuitFailedEvent(circuitId, params, startTime, nil, nil, CircuitFailureInvalidService)
network.ServiceDialOtherError(serviceId)
return nil, err
}
logger = logger.WithField("serviceName", svc.Name)
// 3: select terminator
strategy, terminator, pathNodes, circuitErr := network.selectPath(srcR, svc, instanceId, ctx)
if circuitErr != nil {
network.CircuitFailedEvent(circuitId, params, startTime, nil, nil, circuitErr.Cause())
network.ServiceDialOtherError(serviceId)
return nil, circuitErr
}
// 4: Create Path
path, pathErr := network.CreatePathWithNodes(pathNodes)
if pathErr != nil {
network.CircuitFailedEvent(circuitId, params, startTime, nil, terminator, pathErr.Cause())
network.ServiceDialOtherError(serviceId)
return nil, pathErr
}
// get circuit tags
tags := params.GetCircuitTags(terminator)
// 4a: Create Route Messages
rms := path.CreateRouteMessages(attempt, circuitId, terminator, deadline)
rms[len(rms)-1].Egress.PeerData = clientId.Data
for _, msg := range rms {
msg.Context = &ctrl_pb.Context{
Fields: ctx.GetStringFields(),
ChannelMask: ctx.GetChannelsMask(),
}
msg.Tags = tags
}
// 5: Routing
logger.Debug("route attempt for circuit")
peerData, cleanups, circuitErr := rs.route(attempt, path, rms, strategy, terminator, ctx.Clone())
for k, v := range cleanups {
allCleanups[k] = v
}
if circuitErr != nil {
logger.WithError(circuitErr).Warn("route attempt for circuit failed")
network.CircuitFailedEvent(circuitId, params, startTime, path, terminator, circuitErr.Cause())
attempt++
ctx.WithField("attemptNumber", attempt+1)
logger = logger.WithField("attemptNumber", attempt+1)
if attempt < network.options.CreateCircuitRetries {
continue
} else {
// revert successful routes
logger.Warnf("circuit creation failed after [%d] attempts, sending cleanup unroutes", network.options.CreateCircuitRetries)
for cleanupRId := range allCleanups {
if r := network.GetConnectedRouter(cleanupRId); r != nil {
if err := sendUnroute(r, circuitId, true); err == nil {
logger.WithField("routerId", cleanupRId).Debug("sent cleanup unroute for circuit")
} else {
logger.WithField("routerId", cleanupRId).Error("error sending cleanup unroute for circuit")
}
} else {
logger.WithField("routerId", cleanupRId).Error("router for circuit cleanup not connected")
}
}
return nil, errors.Wrapf(circuitErr, "exceeded maximum [%d] retries creating circuit [c/%s]", network.options.CreateCircuitRetries, circuitId)
}
}
// 5.a: Unroute Abandoned Routers (from Previous Attempts)
usedRouters := make(map[string]struct{})
for _, r := range path.Nodes {
usedRouters[r.Id] = struct{}{}
}
cleanupCount := 0
for cleanupRId := range allCleanups {
if _, found := usedRouters[cleanupRId]; !found {
cleanupCount++
if r := network.GetConnectedRouter(cleanupRId); r != nil {
if err := sendUnroute(r, circuitId, true); err == nil {
logger.WithField("routerId", cleanupRId).Debug("sent abandoned cleanup unroute for circuit to router")
} else {
logger.WithField("routerId", cleanupRId).WithError(err).Error("error sending abandoned cleanup unroute for circuit to router")
}
} else {
logger.WithField("routerId", cleanupRId).Error("router not connected for circuit, abandoned cleanup")
}
}
}
logger.Debugf("cleaned up [%d] abandoned routers for circuit", cleanupCount)
path.InitiatorLocalAddr = string(clientId.Data[uint32(ctrl_msg.InitiatorLocalAddressHeader)])
path.InitiatorRemoteAddr = string(clientId.Data[uint32(ctrl_msg.InitiatorRemoteAddressHeader)])
path.TerminatorLocalAddr = string(peerData[uint32(ctrl_msg.TerminatorLocalAddressHeader)])
path.TerminatorRemoteAddr = string(peerData[uint32(ctrl_msg.TerminatorRemoteAddressHeader)])
delete(peerData, uint32(ctrl_msg.InitiatorLocalAddressHeader))
delete(peerData, uint32(ctrl_msg.InitiatorRemoteAddressHeader))
delete(peerData, uint32(ctrl_msg.TerminatorLocalAddressHeader))
delete(peerData, uint32(ctrl_msg.TerminatorRemoteAddressHeader))
now := time.Now()
// 6: Create Circuit Object
circuit := &Circuit{
Id: circuitId,
ClientId: clientId.Token,
ServiceId: svc.Id,
Path: path,
Terminator: terminator,
PeerData: peerData,
CreatedAt: now,
UpdatedAt: now,
Tags: tags,
}
network.circuitController.add(circuit)
creationTimespan := time.Since(startTime)
network.CircuitEvent(event.CircuitCreated, circuit, &creationTimespan)
logger.WithField("path", circuit.Path).
WithField("terminator_local_address", circuit.Path.TerminatorLocalAddr).
WithField("terminator_remote_address", circuit.Path.TerminatorRemoteAddr).
Debug("created circuit")
return circuit, nil
}
}
func (network *Network) ReportForwardingFaults(ffr *ForwardingFaultReport) {
select {
case network.forwardingFaults <- struct{}{}:
default:
}
go network.handleForwardingFaults(ffr)
}
func parseInstanceIdAndService(service string) (string, string) {
atIndex := strings.IndexRune(service, '@')
if atIndex < 0 {
return "", service
}
identityId := service[0:atIndex]
serviceId := service[atIndex+1:]
return identityId, serviceId
}
func (network *Network) selectPath(srcR *Router, svc *Service, instanceId string, ctx logcontext.Context) (xt.Strategy, xt.CostedTerminator, []*Router, CircuitError) {
paths := map[string]*PathAndCost{}
var weightedTerminators []xt.CostedTerminator
var errList []error
log := pfxlog.ChannelLogger(logcontext.SelectPath).Wire(ctx)
hasOfflineRouters := false
pathError := false
for _, terminator := range svc.Terminators {
if terminator.InstanceId != instanceId {
continue
}
pathAndCost, found := paths[terminator.Router]
if !found {
dstR := network.Routers.getConnected(terminator.GetRouterId())
if dstR == nil {
err := errors.Errorf("router with id=%v on terminator with id=%v for service name=%v is not online",
terminator.GetRouterId(), terminator.GetId(), svc.Name)
log.Debugf("error while calculating path for service %v: %v", svc.Id, err)
errList = append(errList, err)
hasOfflineRouters = true
continue
}
path, cost, err := network.shortestPath(srcR, dstR)
if err != nil {
log.Debugf("error while calculating path for service %v: %v", svc.Id, err)
errList = append(errList, err)
pathError = true
continue
}
pathAndCost = newPathAndCost(path, cost)
paths[terminator.GetRouterId()] = pathAndCost
}
dynamicCost := xt.GlobalCosts().GetDynamicCost(terminator.Id)
unbiasedCost := uint32(terminator.Cost) + uint32(dynamicCost) + pathAndCost.cost
biasedCost := terminator.Precedence.GetBiasedCost(unbiasedCost)
costedTerminator := &RoutingTerminator{
Terminator: terminator,
RouteCost: biasedCost,
}
weightedTerminators = append(weightedTerminators, costedTerminator)
}
if len(svc.Terminators) == 0 {
return nil, nil, nil, newCircuitErrorf(CircuitFailureNoTerminators, "service %v has no terminators", svc.Id)
}
if len(weightedTerminators) == 0 {
if pathError {
return nil, nil, nil, newCircuitErrWrap(CircuitFailureNoPath, errorz.MultipleErrors(errList))
}
if hasOfflineRouters {
return nil, nil, nil, newCircuitErrorf(CircuitFailureNoOnlineTerminators, "service %v has no online terminators for instanceId %v", svc.Id, instanceId)
}
return nil, nil, nil, newCircuitErrorf(CircuitFailureNoTerminators, "service %v has no terminators for instanceId %v", svc.Id, instanceId)
}
strategy, err := network.strategyRegistry.GetStrategy(svc.TerminatorStrategy)
if err != nil {
return nil, nil, nil, newCircuitErrWrap(CircuitFailureInvalidStrategy, err)
}
sort.Slice(weightedTerminators, func(i, j int) bool {
return weightedTerminators[i].GetRouteCost() < weightedTerminators[j].GetRouteCost()
})
terminator, err := strategy.Select(weightedTerminators)
if err != nil {
return nil, nil, nil, newCircuitErrorf(CircuitFailureStrategyError, "strategy %v errored selecting terminator for service %v: %v", svc.TerminatorStrategy, svc.Id, err)
}
if terminator == nil {
return nil, nil, nil, newCircuitErrorf(CircuitFailureStrategyError, "strategy %v did not select terminator for service %v", svc.TerminatorStrategy, svc.Id)
}
path := paths[terminator.GetRouterId()].path
if log.Logger.IsLevelEnabled(logrus.DebugLevel) {
buf := strings.Builder{}
buf.WriteString("[")
if len(weightedTerminators) > 0 {
buf.WriteString(fmt.Sprintf("%v: %v", weightedTerminators[0].GetId(), weightedTerminators[0].GetRouteCost()))
for _, t := range weightedTerminators[1:] {
buf.WriteString(", ")
buf.WriteString(fmt.Sprintf("%v: %v", t.GetId(), t.GetRouteCost()))
}
}
buf.WriteString("]")
var routerIds []string
for _, r := range path {
routerIds = append(routerIds, fmt.Sprintf("r/%s", r.Id))
}
pathStr := strings.Join(routerIds, "->")
log.Debugf("selected terminator %v for path %v from %v", terminator.GetId(), pathStr, buf.String())
}
return strategy, terminator, path, nil
}
func (network *Network) RemoveCircuit(circuitId string, now bool) error {
log := pfxlog.Logger().WithField("circuitId", circuitId)
if circuit, found := network.circuitController.get(circuitId); found {
for _, r := range circuit.Path.Nodes {
err := sendUnroute(r, circuit.Id, now)
if err != nil {
log.Errorf("error sending unroute to [r/%s] (%s)", r.Id, err)
}
}
network.circuitController.remove(circuit)
network.CircuitEvent(event.CircuitDeleted, circuit, nil)
if svc, err := network.Services.Read(circuit.ServiceId); err == nil {
if strategy, err := network.strategyRegistry.GetStrategy(svc.TerminatorStrategy); strategy != nil {
strategy.NotifyEvent(xt.NewCircuitRemoved(circuit.Terminator))
} else if err != nil {
log.WithError(err).WithField("terminatorStrategy", svc.TerminatorStrategy).Warn("failed to notify strategy of circuit end, invalid strategy")
}
} else {
log.WithError(err).Error("unable to get service for circuit")
}
log.Debug("removed circuit")
return nil
}
return InvalidCircuitError{circuitId: circuitId}
}
func (network *Network) CreatePath(srcR, dstR *Router) (*Path, error) {
ingressId, err := network.sequence.NextHash()
if err != nil {
return nil, err
}
egressId, err := network.sequence.NextHash()
if err != nil {
return nil, err
}
path := &Path{
Links: make([]*Link, 0),
IngressId: ingressId,
EgressId: egressId,
Nodes: make([]*Router, 0),
}
path.Nodes = append(path.Nodes, srcR)
path.Nodes = append(path.Nodes, dstR)
return network.UpdatePath(path)
}
func (network *Network) CreatePathWithNodes(nodes []*Router) (*Path, CircuitError) {
ingressId, err := network.sequence.NextHash()
if err != nil {
return nil, newCircuitErrWrap(CircuitFailureIdGenerationError, err)
}
egressId, err := network.sequence.NextHash()
if err != nil {
return nil, newCircuitErrWrap(CircuitFailureIdGenerationError, err)
}
path := &Path{
Nodes: nodes,
IngressId: ingressId,
EgressId: egressId,
}
if err := network.setLinks(path); err != nil {
return nil, newCircuitErrWrap(CircuitFailurePathMissingLink, err)
}
return path, nil
}
func (network *Network) UpdatePath(path *Path) (*Path, error) {
srcR := path.Nodes[0]
dstR := path.Nodes[len(path.Nodes)-1]
nodes, _, err := network.shortestPath(srcR, dstR)
if err != nil {
return nil, err
}
path2 := &Path{
Nodes: nodes,
IngressId: path.IngressId,
EgressId: path.EgressId,
InitiatorLocalAddr: path.InitiatorLocalAddr,
InitiatorRemoteAddr: path.InitiatorRemoteAddr,
TerminatorLocalAddr: path.TerminatorLocalAddr,
TerminatorRemoteAddr: path.TerminatorRemoteAddr,
}
if err := network.setLinks(path2); err != nil {
return nil, err
}
return path2, nil
}
func (network *Network) setLinks(path *Path) error {
if len(path.Nodes) > 1 {
for i := 0; i < len(path.Nodes)-1; i++ {
if link, found := network.linkController.leastExpensiveLink(path.Nodes[i], path.Nodes[i+1]); found {
path.Links = append(path.Links, link)
} else {
return errors.Errorf("no link from r/%v to r/%v", path.Nodes[i].Id, path.Nodes[i+1].Id)
}
}
}
return nil
}
func (network *Network) AddRouterPresenceHandler(h RouterPresenceHandler) {
network.routerPresenceHandlers = append(network.routerPresenceHandlers, h)
}
func (network *Network) Run() {
defer logrus.Info("exited")
logrus.Info("started")
go network.watchdog()
ticker := time.NewTicker(time.Duration(network.options.CycleSeconds) * time.Second)
defer ticker.Stop()
for {
select {
case <-network.assembleAndCleanC:
network.assemble()
network.clean()
case <-network.forwardingFaults:
network.clean()
case <-ticker.C:
network.assemble()
network.clean()
network.smart()
network.linkController.scanForDeadLinks()
case <-network.closeNotify:
network.eventDispatcher.RemoveMetricsMessageHandler(network)
network.metricsRegistry.DisposeAll()
return
}
// notify the watchdog that we're processing
select {
case network.watchdogCh <- struct{}{}:
default:
}
}
}
func (network *Network) watchdog() {
watchdogInterval := 2 * time.Duration(network.options.CycleSeconds) * time.Second
consecutiveFails := 0
for {
// check every 2x cycle seconds
time.Sleep(watchdogInterval)
select {
case <-network.watchdogCh:
consecutiveFails = 0
continue
case <-network.closeNotify:
return
default:
consecutiveFails++
// network.Run didn't complete, something is stalling it
pfxlog.Logger().
WithField("watchdogInterval", watchdogInterval.String()).
WithField("consecutiveFails", consecutiveFails).
Warn("network.Run did not finish within watchdog interval")
if consecutiveFails == 3 {
debugz.DumpStack()
}
}
}
}
func (network *Network) handleRerouteLink(l *Link) {
log := logrus.WithField("linkId", l.Id)
log.Info("changed link")
if err := network.rerouteLink(l, time.Now().Add(DefaultOptionsRouteTimeout)); err != nil {
log.WithError(err).Error("unexpected error rerouting link")
}
}
func (network *Network) handleForwardingFaults(ffr *ForwardingFaultReport) {
network.fault(ffr)
}
func (network *Network) AddCapability(capability string) {
network.lock.Lock()
defer network.lock.Unlock()
network.capabilities = append(network.capabilities, capability)
}
func (network *Network) GetCapabilities() []string {
network.lock.Lock()
defer network.lock.Unlock()
return network.capabilities
}
func (network *Network) RemoveLink(linkId string) {
log := pfxlog.Logger().WithField("linkId", linkId)
link, _ := network.linkController.get(linkId)
var iteration uint32
var routerList []*Router
if link != nil {
iteration = link.Iteration
routerList = []*Router{link.Src}
if dst := link.GetDest(); dst != nil {
routerList = append(routerList, dst)
}
log = log.WithField("srcRouterId", link.Src.Id).
WithField("dstRouterId", link.DstId).
WithField("iteration", iteration)
log.Info("deleting known link")
} else {
routerList = network.AllConnectedRouters()
log.Info("deleting unknown link (sending link fault to all connected routers)")
}
for _, router := range routerList {
fault := &ctrl_pb.Fault{
Subject: ctrl_pb.FaultSubject_LinkFault,
Id: linkId,
Iteration: iteration,
}
if ctrl := router.Control; ctrl != nil {
if err := protobufs.MarshalTyped(fault).WithTimeout(15 * time.Second).Send(ctrl); err != nil {
log.WithField("faultDestRouterId", router.Id).WithError(err).
Error("failed to send link fault to router on link removal")
} else {
log.WithField("faultDestRouterId", router.Id).WithError(err).
Info("sent link fault to router on link removal")
}
}
}
if link != nil {
network.linkController.remove(link)
network.RerouteLink(link)
}
}
func (network *Network) rerouteLink(l *Link, deadline time.Time) error {
circuits := network.circuitController.all()
for _, circuit := range circuits {
if circuit.Path.usesLink(l) {
log := logrus.WithField("linkId", l.Id).
WithField("circuitId", circuit.Id)
log.Info("circuit uses link")
if err := network.rerouteCircuit(circuit, deadline); err != nil {
log.WithError(err).Error("error rerouting circuit, removing")
if err := network.RemoveCircuit(circuit.Id, true); err != nil {
log.WithError(err).Error("error removing circuit after reroute failure")
}
}
}
}
return nil
}
func (network *Network) rerouteCircuitWithTries(circuit *Circuit, retries int) bool {
log := pfxlog.Logger().WithField("circuitId", circuit.Id)
for i := 0; i < retries; i++ {
deadline := time.Now().Add(DefaultOptionsRouteTimeout)
err := network.rerouteCircuit(circuit, deadline)
if err == nil {
return true
}
log.WithError(err).WithField("attempt", i).Error("error re-routing circuit")
}
if err := network.RemoveCircuit(circuit.Id, true); err != nil {
log.WithError(err).Error("failure while removing circuit after failed re-route attempt")
}
return false
}
func (network *Network) rerouteCircuit(circuit *Circuit, deadline time.Time) error {
log := pfxlog.Logger().WithField("circuitId", circuit.Id)
if circuit.Rerouting.CompareAndSwap(false, true) {
defer circuit.Rerouting.Store(false)
log.Warn("rerouting circuit")
if cq, err := network.UpdatePath(circuit.Path); err == nil {
circuit.Path = cq
circuit.UpdatedAt = time.Now()
rms := cq.CreateRouteMessages(SmartRerouteAttempt, circuit.Id, circuit.Terminator, deadline)
for i := 0; i < len(cq.Nodes); i++ {
if _, err := sendRoute(cq.Nodes[i], rms[i], network.options.RouteTimeout); err != nil {
log.WithError(err).Errorf("error sending route to [r/%s]", cq.Nodes[i].Id)
}
}
log.Info("rerouted circuit")
network.CircuitEvent(event.CircuitUpdated, circuit, nil)
return nil
} else {
return err
}
} else {
log.Info("not rerouting circuit, already in progress")
return nil
}
}
func (network *Network) smartReroute(circuit *Circuit, cq *Path, deadline time.Time) bool {
retry := false
log := pfxlog.Logger().WithField("circuitId", circuit.Id)
if circuit.Rerouting.CompareAndSwap(false, true) {
defer circuit.Rerouting.Store(false)
circuit.Path = cq
circuit.UpdatedAt = time.Now()
rms := cq.CreateRouteMessages(SmartRerouteAttempt, circuit.Id, circuit.Terminator, deadline)
for i := 0; i < len(cq.Nodes); i++ {
if _, err := sendRoute(cq.Nodes[i], rms[i], network.options.RouteTimeout); err != nil {
retry = true
log.WithField("routerId", cq.Nodes[i].Id).WithError(err).Error("error sending smart route update to router")
break
}
}
if !retry {
logrus.Debug("rerouted circuit")
network.CircuitEvent(event.CircuitUpdated, circuit, nil)
}
}
return retry
}
func (network *Network) AcceptMetricsMsg(metrics *metrics_pb.MetricsMessage) {
if metrics.SourceId == network.nodeId {
return // ignore metrics coming from the controller itself
}
log := pfxlog.Logger()
router, err := network.Routers.Read(metrics.SourceId)
if err != nil {
log.Debugf("could not find router [r/%s] while processing metrics", metrics.SourceId)
return
}
for _, link := range network.GetAllLinksForRouter(router.Id) {
metricId := "link." + link.Id + ".latency"
var latencyCost int64
var found bool
if latency, ok := metrics.Histograms[metricId]; ok {
latencyCost = int64(latency.Mean)
found = true
metricId = "link." + link.Id + ".queue_time"
if queueTime, ok := metrics.Histograms[metricId]; ok {
latencyCost += int64(queueTime.Mean)
}
}
if found {
if link.Src.Id == router.Id {
link.SetSrcLatency(latencyCost) // latency is in nanoseconds
} else if link.DstId == router.Id {
link.SetDstLatency(latencyCost) // latency is in nanoseconds
} else {
log.Warnf("link not for router")
}
}
}
}
func sendRoute(r *Router, createMsg *ctrl_pb.Route, timeout time.Duration) (xt.PeerData, error) {
log := pfxlog.Logger().WithField("routerId", r.Id).
WithField("circuitId", createMsg.CircuitId)
log.Debug("sending create route message")
msg, err := protobufs.MarshalTyped(createMsg).WithTimeout(timeout).SendForReply(r.Control)
if err != nil {
log.WithError(err).WithField("timeout", timeout).Error("error sending route message")
return nil, err
}
if msg.ContentType == ctrl_msg.RouteResultType {
_, success := msg.Headers[ctrl_msg.RouteResultSuccessHeader]
if !success {
message := "route error, but no error message from router"
if errMsg, found := msg.Headers[ctrl_msg.RouteResultErrorHeader]; found {
message = string(errMsg)
}
return nil, errors.New(message)
}
peerData := xt.PeerData{}
for k, v := range msg.Headers {
if k > 0 {
peerData[uint32(k)] = v
}
}
return peerData, nil
}
return nil, fmt.Errorf("unexpected response type %v received in reply to route request", msg.ContentType)
}
func sendUnroute(r *Router, circuitId string, now bool) error {
unroute := &ctrl_pb.Unroute{
CircuitId: circuitId,
Now: now,
}
return protobufs.MarshalTyped(unroute).Send(r.Control)
}
func (network *Network) showOptions() {
if jsonOptions, err := json.MarshalIndent(network.options, "", " "); err == nil {
pfxlog.Logger().Infof("network = %s", string(jsonOptions))
} else {
panic(err)
}
}
type renderConfig interface {
RenderJsonConfig() (string, error)
}
func (network *Network) Inspect(name string) (*string, error) {
lc := strings.ToLower(name)
if lc == "stackdump" {
result := debugz.GenerateStack()
return &result, nil
} else if strings.HasPrefix(lc, "metrics") {
msg := network.metricsRegistry.Poll()
js, err := json.Marshal(msg)
if err != nil {
return nil, errors.Wrap(err, "failed to marshal metrics to json")
}
result := string(js)
return &result, nil
} else if lc == "config" {
if rc, ok := network.config.(renderConfig); ok {
val, err := rc.RenderJsonConfig()
if err != nil {
return nil, errors.Wrap(err, "failed to marshal config to json")
}
return &val, nil
}
} else if lc == "cluster-config" {
if src, ok := network.Dispatcher.(renderConfig); ok {
val, err := src.RenderJsonConfig()
return &val, err
}
} else if lc == "connected-routers" {
var result []map[string]any
for _, r := range network.Routers.allConnected() {
status := map[string]any{}
status["Id"] = r.Id
status["Name"] = r.Name
status["Version"] = r.VersionInfo.Version
status["ConnectTime"] = r.ConnectTime.Format(time.RFC3339)
result = append(result, status)
}
val, err := json.Marshal(result)
strVal := string(val)
return &strVal, err
} else if lc == "connected-peers" {
if raftController, ok := network.Dispatcher.(*raft.Controller); ok {
members, err := raftController.ListMembers()
if err != nil {
return nil, err
}
result, err := json.Marshal(members)
if err != nil {
return nil, fmt.Errorf("failed to marshall cluster member list to json (%w)", err)
}
resultStr := string(result)
return &resultStr, nil
}
} else if lc == "router-messaging" {
routerMessagingState, err := network.Managers.RouterMessaging.Inspect()
if err != nil {
return nil, err
}
result, err := json.Marshal(routerMessagingState)
if err != nil {
return nil, fmt.Errorf("failed to marshall router messaging state to json (%w)", err)
}
resultStr := string(result)
return &resultStr, nil
}
return nil, nil
}
func (network *Network) routerDeleted(routerId string) {
circuits := network.GetAllCircuits()
for _, circuit := range circuits {
if circuit.HasRouter(routerId) {
path := circuit.Path
// If we're either the initiator, terminator (or both), cleanup the circuit since
// we won't be able to re-establish it, and we'll never get a circuit fault
if path.Nodes[0].Id == routerId || path.Nodes[len(path.Nodes)-1].Id == routerId {
if err := network.RemoveCircuit(circuit.Id, true); err != nil {
pfxlog.Logger().WithField("routerId", routerId).
WithField("circuitId", circuit.Id).
WithError(err).Error("unable to remove circuit after router was deleted")
}
}
}
}
}
var DbSnapshotTooFrequentError = dbSnapshotTooFrequentError{}
type dbSnapshotTooFrequentError struct{}
func (d dbSnapshotTooFrequentError) Error() string {
return "may snapshot database at most once per minute"
}
func (network *Network) SnapshotDatabase() error {
network.lock.Lock()
defer network.lock.Unlock()
if network.lastSnapshot.Add(time.Minute).After(time.Now()) {
return DbSnapshotTooFrequentError
}
pfxlog.Logger().Info("snapshotting database")
err := network.GetDb().View(func(tx *bbolt.Tx) error {
return network.GetDb().Snapshot(tx)
})
if err == nil {
network.lastSnapshot = time.Now()
}
return err
}
func (network *Network) SnapshotDatabaseToFile(path string) (string, error) {
network.lock.Lock()
defer network.lock.Unlock()
if network.lastSnapshot.Add(time.Minute).After(time.Now()) {
return "", DbSnapshotTooFrequentError
}
actualPath := path
if actualPath == "" {
actualPath = "DB_DIR/DB_FILE-DATE-TIME"
}
err := network.GetDb().View(func(tx *bbolt.Tx) error {
now := time.Now()
dateStr := now.Format("20060102")
timeStr := now.Format("150405")
actualPath = strings.ReplaceAll(actualPath, "DATE", dateStr)
actualPath = strings.ReplaceAll(actualPath, "TIME", timeStr)
currentIndex := db.LoadCurrentRaftIndex(tx)
actualPath = strings.ReplaceAll(actualPath, "RAFT_INDEX", fmt.Sprintf("%v", currentIndex))
actualPath = strings.ReplaceAll(actualPath, "DB_DIR", filepath.Dir(tx.DB().Path()))
actualPath = strings.ReplaceAll(actualPath, "DB_FILE", filepath.Base(tx.DB().Path()))
pfxlog.Logger().WithField("path", actualPath).Info("snapshotting database to file")
_, err := os.Stat(actualPath)
if err != nil {
if !os.IsNotExist(err) {
return err
}
} else {
pfxlog.Logger().Infof("bolt db backup already exists: %v", actualPath)
return errors.Errorf("bolt db backup already exists [%s]", actualPath)
}
file, err := os.OpenFile(actualPath, os.O_WRONLY|os.O_CREATE, 0600)
if err != nil {
return err
}
defer func() {
if err = file.Close(); err != nil {
pfxlog.Logger().WithError(err).Errorf("failed to close backup database file %v", actualPath)
}
}()
_, err = tx.WriteTo(file)
return err
})
if err == nil {
network.lastSnapshot = time.Now()
}
return actualPath, err
}
func (network *Network) RestoreSnapshot(cmd *command.SyncSnapshotCommand) error {
log := pfxlog.Logger()
currentSnapshotId, err := network.getDb().GetSnapshotId()
if err != nil {
log.WithError(err).Error("unable to get current snapshot id")
}
if currentSnapshotId != nil && *currentSnapshotId == cmd.SnapshotId {
log.WithField("snapshotId", cmd.SnapshotId).Info("snapshot already current, skipping reload")
return nil
}
buf := bytes.NewBuffer(cmd.Snapshot)
reader, err := gzip.NewReader(buf)
if err != nil {
return errors.Wrapf(err, "unable to create gz reader for reading migration snapshot during restore")
}
network.getDb().RestoreFromReader(reader)
return nil
}
func (network *Network) SnapshotToRaft() error {
buf := &bytes.Buffer{}
gzWriter := gzip.NewWriter(buf)
snapshotId, err := network.db.SnapshotToWriter(gzWriter)
if err != nil {
return err
}
if err = gzWriter.Close(); err != nil {
return errors.Wrap(err, "error finishing gz compression of migration snapshot")
}
cmd := &command.SyncSnapshotCommand{
SnapshotId: snapshotId,
Snapshot: buf.Bytes(),
SnapshotSink: network.RestoreSnapshot,
}
return network.Dispatch(cmd)
}
type Cache interface {
RemoveFromCache(id string)
}
func newPathAndCost(path []*Router, cost int64) *PathAndCost {
if cost > (1 << 20) {
cost = 1 << 20
}
return &PathAndCost{
path: path,
cost: uint32(cost),
}
}
type PathAndCost struct {
path []*Router
cost uint32
}
type InvalidCircuitError struct {
circuitId string
}
func (err InvalidCircuitError) Error() string {
return fmt.Sprintf("invalid circuit (%s)", err.circuitId)
}