netbird/client/internal/peer/worker_relay.go

124 lines
3.1 KiB
Go
Raw Normal View History

package peer
import (
"context"
"errors"
"math/rand"
"net"
"time"
log "github.com/sirupsen/logrus"
relayClient "github.com/netbirdio/netbird/relay/client"
)
type OnRelayReadyCallback func(info RelayConnInfo)
type RelayConnInfo struct {
relayedConn net.Conn
rosenpassPubKey []byte
rosenpassAddr string
}
2024-06-18 11:22:40 +02:00
type WorkerRelay struct {
ctx context.Context
log *log.Entry
relayManager *relayClient.Manager
config ConnConfig
onRelayConnReadyFN OnRelayReadyCallback
onStatusChanged func(ConnStatus)
doHandshakeFn DoHandshake
}
func NewWorkerRelay(ctx context.Context, log *log.Entry, relayManager *relayClient.Manager, config ConnConfig, onRelayConnReadyFN OnRelayReadyCallback, onStatusChanged func(ConnStatus), doHandshakeFn DoHandshake) *WorkerRelay {
2024-06-18 11:22:40 +02:00
return &WorkerRelay{
ctx: ctx,
log: log,
relayManager: relayManager,
config: config,
onRelayConnReadyFN: onRelayConnReadyFN,
onStatusChanged: onStatusChanged,
doHandshakeFn: doHandshakeFn,
}
}
// SetupRelayConnection todo: this function is not completed. Make no sense to put it in a for loop because we are not waiting for any event
2024-06-18 11:22:40 +02:00
func (w *WorkerRelay) SetupRelayConnection() {
for {
2024-06-18 11:22:40 +02:00
if !w.waitForReconnectTry() {
return
}
2024-06-18 11:22:40 +02:00
remoteOfferAnswer, err := w.doHandshakeFn()
if err != nil {
if errors.Is(err, ErrSignalIsNotReady) {
2024-06-18 11:22:40 +02:00
w.log.Infof("signal client isn't ready, skipping connection attempt")
}
2024-06-18 11:22:40 +02:00
w.log.Errorf("failed to do handshake: %v", err)
continue
}
2024-06-18 11:22:40 +02:00
if !w.isRelaySupported(remoteOfferAnswer) {
// todo should we retry?
continue
}
// the relayManager will return with error in case if the connection has lost with relay server
2024-06-18 11:22:40 +02:00
currentRelayAddress, err := w.relayManager.RelayAddress()
if err != nil {
continue
}
2024-06-18 11:22:40 +02:00
srv := w.preferredRelayServer(currentRelayAddress.String(), remoteOfferAnswer.RelaySrvAddress)
relayedConn, err := w.relayManager.OpenConn(srv, w.config.Key)
if err != nil {
continue
}
2024-06-18 11:22:40 +02:00
go w.onRelayConnReadyFN(RelayConnInfo{
relayedConn: relayedConn,
rosenpassPubKey: remoteOfferAnswer.RosenpassPubKey,
rosenpassAddr: remoteOfferAnswer.RosenpassAddr,
})
2024-06-18 11:10:17 +02:00
// todo: waitForDisconnection()
}
}
2024-06-18 11:22:40 +02:00
func (w *WorkerRelay) RelayAddress() (net.Addr, error) {
return w.relayManager.RelayAddress()
}
// todo check my side too
2024-06-18 11:22:40 +02:00
func (w *WorkerRelay) isRelaySupported(answer *OfferAnswer) bool {
return answer.RelaySrvAddress != ""
}
2024-06-18 11:22:40 +02:00
func (w *WorkerRelay) preferredRelayServer(myRelayAddress, remoteRelayAddress string) string {
if w.config.LocalKey > w.config.Key {
return myRelayAddress
}
return remoteRelayAddress
}
2024-06-18 11:22:40 +02:00
func (w *WorkerRelay) RelayIsSupportedLocally() bool {
return w.relayManager.HasRelayAddress()
}
// waitForReconnectTry waits for a random duration before trying to reconnect
2024-06-18 11:22:40 +02:00
func (w *WorkerRelay) waitForReconnectTry() bool {
minWait := 500
maxWait := 2000
duration := time.Duration(rand.Intn(maxWait-minWait)+minWait) * time.Millisecond
2024-06-18 11:20:01 +02:00
timeout := time.NewTimer(duration)
defer timeout.Stop()
select {
2024-06-18 11:22:40 +02:00
case <-w.ctx.Done():
return false
2024-06-18 11:20:01 +02:00
case <-timeout.C:
return true
}
}