413 lines
12 KiB
Go
413 lines
12 KiB
Go
package origin
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"math/rand"
|
|
"net"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/cloudflare/cloudflared/connection"
|
|
"github.com/cloudflare/cloudflared/h2mux"
|
|
"github.com/cloudflare/cloudflared/signal"
|
|
tunnelpogs "github.com/cloudflare/cloudflared/tunnelrpc/pogs"
|
|
)
|
|
|
|
const (
|
|
// Waiting time before retrying a failed tunnel connection
|
|
tunnelRetryDuration = time.Second * 10
|
|
// SRV record resolution TTL
|
|
resolveTTL = time.Hour
|
|
// Interval between registering new tunnels
|
|
registrationInterval = time.Second
|
|
|
|
subsystemRefreshAuth = "refresh_auth"
|
|
// Maximum exponent for 'Authenticate' exponential backoff
|
|
refreshAuthMaxBackoff = 10
|
|
// Waiting time before retrying a failed 'Authenticate' connection
|
|
refreshAuthRetryDuration = time.Second * 10
|
|
// Maximum time to make an Authenticate RPC
|
|
authTokenTimeout = time.Second * 30
|
|
)
|
|
|
|
var (
|
|
errJWTUnset = errors.New("JWT unset")
|
|
errEventDigestUnset = errors.New("event digest unset")
|
|
)
|
|
|
|
type Supervisor struct {
|
|
cloudflaredUUID uuid.UUID
|
|
config *TunnelConfig
|
|
edgeIPs []*net.TCPAddr
|
|
// nextUnusedEdgeIP is the index of the next addr k edgeIPs to try
|
|
nextUnusedEdgeIP int
|
|
lastResolve time.Time
|
|
resolverC chan resolveResult
|
|
tunnelErrors chan tunnelError
|
|
tunnelsConnecting map[int]chan struct{}
|
|
// nextConnectedIndex and nextConnectedSignal are used to wait for all
|
|
// currently-connecting tunnels to finish connecting so we can reset backoff timer
|
|
nextConnectedIndex int
|
|
nextConnectedSignal chan struct{}
|
|
|
|
logger *logrus.Entry
|
|
|
|
jwtLock *sync.RWMutex
|
|
jwt []byte
|
|
|
|
eventDigestLock *sync.RWMutex
|
|
eventDigest []byte
|
|
}
|
|
|
|
type resolveResult struct {
|
|
edgeIPs []*net.TCPAddr
|
|
err error
|
|
}
|
|
|
|
type tunnelError struct {
|
|
index int
|
|
err error
|
|
}
|
|
|
|
func NewSupervisor(config *TunnelConfig, u uuid.UUID) *Supervisor {
|
|
return &Supervisor{
|
|
cloudflaredUUID: u,
|
|
config: config,
|
|
tunnelErrors: make(chan tunnelError),
|
|
tunnelsConnecting: map[int]chan struct{}{},
|
|
logger: config.Logger.WithField("subsystem", "supervisor"),
|
|
jwtLock: &sync.RWMutex{},
|
|
eventDigestLock: &sync.RWMutex{},
|
|
}
|
|
}
|
|
|
|
func (s *Supervisor) Run(ctx context.Context, connectedSignal *signal.Signal) error {
|
|
logger := s.config.Logger
|
|
if err := s.initialize(ctx, connectedSignal); err != nil {
|
|
return err
|
|
}
|
|
var tunnelsWaiting []int
|
|
tunnelsActive := s.config.HAConnections
|
|
|
|
backoff := BackoffHandler{MaxRetries: s.config.Retries, BaseTime: tunnelRetryDuration, RetryForever: true}
|
|
var backoffTimer <-chan time.Time
|
|
|
|
refreshAuthBackoff := &BackoffHandler{MaxRetries: refreshAuthMaxBackoff, BaseTime: refreshAuthRetryDuration, RetryForever: true}
|
|
var refreshAuthBackoffTimer <-chan time.Time
|
|
|
|
if s.config.UseReconnectToken {
|
|
if timer, err := s.refreshAuth(ctx, refreshAuthBackoff, s.authenticate); err == nil {
|
|
refreshAuthBackoffTimer = timer
|
|
} else {
|
|
logger.WithError(err).Errorf("initial refreshAuth failed, retrying in %v", refreshAuthRetryDuration)
|
|
refreshAuthBackoffTimer = time.After(refreshAuthRetryDuration)
|
|
}
|
|
}
|
|
|
|
for {
|
|
select {
|
|
// Context cancelled
|
|
case <-ctx.Done():
|
|
for tunnelsActive > 0 {
|
|
<-s.tunnelErrors
|
|
tunnelsActive--
|
|
}
|
|
return nil
|
|
// startTunnel returned with error
|
|
// (note that this may also be caused by context cancellation)
|
|
case tunnelError := <-s.tunnelErrors:
|
|
tunnelsActive--
|
|
if tunnelError.err != nil {
|
|
logger.WithError(tunnelError.err).Warn("Tunnel disconnected due to error")
|
|
tunnelsWaiting = append(tunnelsWaiting, tunnelError.index)
|
|
s.waitForNextTunnel(tunnelError.index)
|
|
|
|
if backoffTimer == nil {
|
|
backoffTimer = backoff.BackoffTimer()
|
|
}
|
|
|
|
// If the error is a dial error, the problem is likely to be network related
|
|
// try another addr before refreshing since we are likely to get back the
|
|
// same IPs in the same order. Same problem with duplicate connection error.
|
|
if s.unusedIPs() {
|
|
s.replaceEdgeIP(tunnelError.index)
|
|
} else {
|
|
s.refreshEdgeIPs()
|
|
}
|
|
}
|
|
// Backoff was set and its timer expired
|
|
case <-backoffTimer:
|
|
backoffTimer = nil
|
|
for _, index := range tunnelsWaiting {
|
|
go s.startTunnel(ctx, index, s.newConnectedTunnelSignal(index))
|
|
}
|
|
tunnelsActive += len(tunnelsWaiting)
|
|
tunnelsWaiting = nil
|
|
// Time to call Authenticate
|
|
case <-refreshAuthBackoffTimer:
|
|
newTimer, err := s.refreshAuth(ctx, refreshAuthBackoff, s.authenticate)
|
|
if err != nil {
|
|
logger.WithError(err).Error("Authentication failed")
|
|
// Permanent failure. Leave the `select` without setting the
|
|
// channel to be non-null, so we'll never hit this case of the `select` again.
|
|
continue
|
|
}
|
|
refreshAuthBackoffTimer = newTimer
|
|
// Tunnel successfully connected
|
|
case <-s.nextConnectedSignal:
|
|
if !s.waitForNextTunnel(s.nextConnectedIndex) && len(tunnelsWaiting) == 0 {
|
|
// No more tunnels outstanding, clear backoff timer
|
|
backoff.SetGracePeriod()
|
|
}
|
|
// DNS resolution returned
|
|
case result := <-s.resolverC:
|
|
s.lastResolve = time.Now()
|
|
s.resolverC = nil
|
|
if result.err == nil {
|
|
logger.Debug("Service discovery refresh complete")
|
|
s.edgeIPs = result.edgeIPs
|
|
} else {
|
|
logger.WithError(result.err).Error("Service discovery error")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Returns nil if initialization succeeded, else the initialization error.
|
|
func (s *Supervisor) initialize(ctx context.Context, connectedSignal *signal.Signal) error {
|
|
logger := s.logger
|
|
|
|
edgeIPs, err := s.resolveEdgeIPs()
|
|
if err != nil {
|
|
logger.Infof("ResolveEdgeIPs err")
|
|
return err
|
|
}
|
|
s.edgeIPs = edgeIPs
|
|
if s.config.HAConnections > len(edgeIPs) {
|
|
logger.Warnf("You requested %d HA connections but I can give you at most %d.", s.config.HAConnections, len(edgeIPs))
|
|
s.config.HAConnections = len(edgeIPs)
|
|
}
|
|
s.lastResolve = time.Now()
|
|
// check entitlement and version too old error before attempting to register more tunnels
|
|
s.nextUnusedEdgeIP = s.config.HAConnections
|
|
go s.startFirstTunnel(ctx, connectedSignal)
|
|
select {
|
|
case <-ctx.Done():
|
|
<-s.tunnelErrors
|
|
return ctx.Err()
|
|
case tunnelError := <-s.tunnelErrors:
|
|
return tunnelError.err
|
|
case <-connectedSignal.Wait():
|
|
}
|
|
// At least one successful connection, so start the rest
|
|
for i := 1; i < s.config.HAConnections; i++ {
|
|
ch := signal.New(make(chan struct{}))
|
|
go s.startTunnel(ctx, i, ch)
|
|
time.Sleep(registrationInterval)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// startTunnel starts the first tunnel connection. The resulting error will be sent on
|
|
// s.tunnelErrors. It will send a signal via connectedSignal if registration succeed
|
|
func (s *Supervisor) startFirstTunnel(ctx context.Context, connectedSignal *signal.Signal) {
|
|
err := ServeTunnelLoop(ctx, s, s.config, s.getEdgeIP(0), 0, connectedSignal, s.cloudflaredUUID)
|
|
defer func() {
|
|
s.tunnelErrors <- tunnelError{index: 0, err: err}
|
|
}()
|
|
|
|
for s.unusedIPs() {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
default:
|
|
}
|
|
switch err.(type) {
|
|
case nil:
|
|
return
|
|
// try the next address if it was a dialError(network problem) or
|
|
// dupConnRegisterTunnelError
|
|
case connection.DialError, dupConnRegisterTunnelError:
|
|
s.replaceEdgeIP(0)
|
|
default:
|
|
return
|
|
}
|
|
err = ServeTunnelLoop(ctx, s, s.config, s.getEdgeIP(0), 0, connectedSignal, s.cloudflaredUUID)
|
|
}
|
|
}
|
|
|
|
// startTunnel starts a new tunnel connection. The resulting error will be sent on
|
|
// s.tunnelErrors.
|
|
func (s *Supervisor) startTunnel(ctx context.Context, index int, connectedSignal *signal.Signal) {
|
|
err := ServeTunnelLoop(ctx, s, s.config, s.getEdgeIP(index), uint8(index), connectedSignal, s.cloudflaredUUID)
|
|
s.tunnelErrors <- tunnelError{index: index, err: err}
|
|
}
|
|
|
|
func (s *Supervisor) newConnectedTunnelSignal(index int) *signal.Signal {
|
|
sig := make(chan struct{})
|
|
s.tunnelsConnecting[index] = sig
|
|
s.nextConnectedSignal = sig
|
|
s.nextConnectedIndex = index
|
|
return signal.New(sig)
|
|
}
|
|
|
|
func (s *Supervisor) waitForNextTunnel(index int) bool {
|
|
delete(s.tunnelsConnecting, index)
|
|
s.nextConnectedSignal = nil
|
|
for k, v := range s.tunnelsConnecting {
|
|
s.nextConnectedIndex = k
|
|
s.nextConnectedSignal = v
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (s *Supervisor) getEdgeIP(index int) *net.TCPAddr {
|
|
return s.edgeIPs[index%len(s.edgeIPs)]
|
|
}
|
|
|
|
func (s *Supervisor) resolveEdgeIPs() ([]*net.TCPAddr, error) {
|
|
// If --edge is specfied, resolve edge server addresses
|
|
if len(s.config.EdgeAddrs) > 0 {
|
|
return connection.ResolveAddrs(s.config.EdgeAddrs)
|
|
}
|
|
// Otherwise lookup edge server addresses through service discovery
|
|
return connection.EdgeDiscovery(s.logger)
|
|
}
|
|
|
|
func (s *Supervisor) refreshEdgeIPs() {
|
|
if s.resolverC != nil {
|
|
return
|
|
}
|
|
if time.Since(s.lastResolve) < resolveTTL {
|
|
return
|
|
}
|
|
s.resolverC = make(chan resolveResult)
|
|
go func() {
|
|
edgeIPs, err := s.resolveEdgeIPs()
|
|
s.resolverC <- resolveResult{edgeIPs: edgeIPs, err: err}
|
|
}()
|
|
}
|
|
|
|
func (s *Supervisor) unusedIPs() bool {
|
|
return s.nextUnusedEdgeIP < len(s.edgeIPs)
|
|
}
|
|
|
|
func (s *Supervisor) replaceEdgeIP(badIPIndex int) {
|
|
s.edgeIPs[badIPIndex] = s.edgeIPs[s.nextUnusedEdgeIP]
|
|
s.nextUnusedEdgeIP++
|
|
}
|
|
|
|
func (s *Supervisor) ReconnectToken() ([]byte, error) {
|
|
s.jwtLock.RLock()
|
|
defer s.jwtLock.RUnlock()
|
|
if s.jwt == nil {
|
|
return nil, errJWTUnset
|
|
}
|
|
return s.jwt, nil
|
|
}
|
|
|
|
func (s *Supervisor) SetReconnectToken(jwt []byte) {
|
|
s.jwtLock.Lock()
|
|
defer s.jwtLock.Unlock()
|
|
s.jwt = jwt
|
|
}
|
|
|
|
func (s *Supervisor) EventDigest() ([]byte, error) {
|
|
s.eventDigestLock.RLock()
|
|
defer s.eventDigestLock.RUnlock()
|
|
if s.eventDigest == nil {
|
|
return nil, errEventDigestUnset
|
|
}
|
|
return s.eventDigest, nil
|
|
}
|
|
|
|
func (s *Supervisor) SetEventDigest(eventDigest []byte) {
|
|
s.eventDigestLock.Lock()
|
|
defer s.eventDigestLock.Unlock()
|
|
s.eventDigest = eventDigest
|
|
}
|
|
|
|
func (s *Supervisor) refreshAuth(
|
|
ctx context.Context,
|
|
backoff *BackoffHandler,
|
|
authenticate func(ctx context.Context, numPreviousAttempts int) (tunnelpogs.AuthOutcome, error),
|
|
) (retryTimer <-chan time.Time, err error) {
|
|
logger := s.config.Logger.WithField("subsystem", subsystemRefreshAuth)
|
|
authOutcome, err := authenticate(ctx, backoff.Retries())
|
|
if err != nil {
|
|
if duration, ok := backoff.GetBackoffDuration(ctx); ok {
|
|
logger.WithError(err).Warnf("Retrying in %v", duration)
|
|
return backoff.BackoffTimer(), nil
|
|
}
|
|
return nil, err
|
|
}
|
|
// clear backoff timer
|
|
backoff.SetGracePeriod()
|
|
|
|
switch outcome := authOutcome.(type) {
|
|
case tunnelpogs.AuthSuccess:
|
|
s.SetReconnectToken(outcome.JWT())
|
|
return timeAfter(outcome.RefreshAfter()), nil
|
|
case tunnelpogs.AuthUnknown:
|
|
duration := outcome.RefreshAfter()
|
|
logger.WithError(outcome).Warnf("Retrying in %v", duration)
|
|
return timeAfter(duration), nil
|
|
case tunnelpogs.AuthFail:
|
|
return nil, outcome
|
|
default:
|
|
return nil, fmt.Errorf("Unexpected outcome type %T", authOutcome)
|
|
}
|
|
}
|
|
|
|
func (s *Supervisor) authenticate(ctx context.Context, numPreviousAttempts int) (tunnelpogs.AuthOutcome, error) {
|
|
arbitraryEdgeIP := s.getEdgeIP(rand.Int())
|
|
edgeConn, err := connection.DialEdge(ctx, dialTimeout, s.config.TlsConfig, arbitraryEdgeIP)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer edgeConn.Close()
|
|
|
|
handler := h2mux.MuxedStreamFunc(func(*h2mux.MuxedStream) error {
|
|
// This callback is invoked by h2mux when the edge initiates a stream.
|
|
return nil // noop
|
|
})
|
|
muxerConfig := s.config.muxerConfig(handler)
|
|
muxerConfig.Logger = muxerConfig.Logger.WithField("subsystem", subsystemRefreshAuth)
|
|
muxer, err := h2mux.Handshake(edgeConn, edgeConn, muxerConfig, s.config.Metrics.activeStreams)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
go muxer.Serve(ctx)
|
|
defer func() {
|
|
// If we don't wait for the muxer shutdown here, edgeConn.Close() runs before the muxer connections are done,
|
|
// and the user sees log noise: "error writing data", "connection closed unexpectedly"
|
|
<-muxer.Shutdown()
|
|
}()
|
|
|
|
tunnelServer, err := connection.NewRPCClient(ctx, muxer, s.logger.WithField("subsystem", subsystemRefreshAuth), openStreamTimeout)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer tunnelServer.Close()
|
|
|
|
const arbitraryConnectionID = uint8(0)
|
|
registrationOptions := s.config.RegistrationOptions(arbitraryConnectionID, edgeConn.LocalAddr().String(), s.cloudflaredUUID)
|
|
registrationOptions.NumPreviousAttempts = uint8(numPreviousAttempts)
|
|
authResponse, err := tunnelServer.Authenticate(
|
|
ctx,
|
|
s.config.OriginCert,
|
|
s.config.Hostname,
|
|
registrationOptions,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return authResponse.Outcome(), nil
|
|
}
|