mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-15 03:11:02 +01:00
101 lines
2.5 KiB
Go
101 lines
2.5 KiB
Go
package peer
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
relayClient "github.com/netbirdio/netbird/relay/client"
|
|
)
|
|
|
|
type RelayConnInfo struct {
|
|
relayedConn net.Conn
|
|
rosenpassPubKey []byte
|
|
rosenpassAddr string
|
|
}
|
|
|
|
type WorkerRelayCallbacks struct {
|
|
OnConnReady func(RelayConnInfo)
|
|
OnDisconnected func()
|
|
}
|
|
|
|
type WorkerRelay struct {
|
|
ctx context.Context
|
|
log *log.Entry
|
|
config ConnConfig
|
|
relayManager *relayClient.Manager
|
|
conn WorkerRelayCallbacks
|
|
}
|
|
|
|
func NewWorkerRelay(ctx context.Context, log *log.Entry, config ConnConfig, relayManager *relayClient.Manager, callbacks WorkerRelayCallbacks) *WorkerRelay {
|
|
return &WorkerRelay{
|
|
ctx: ctx,
|
|
log: log,
|
|
config: config,
|
|
relayManager: relayManager,
|
|
conn: callbacks,
|
|
}
|
|
}
|
|
|
|
func (w *WorkerRelay) OnNewOffer(remoteOfferAnswer *OfferAnswer) {
|
|
if !w.isRelaySupported(remoteOfferAnswer) {
|
|
w.log.Infof("Relay is not supported by remote peer")
|
|
return
|
|
}
|
|
|
|
// the relayManager will return with error in case if the connection has lost with relay server
|
|
currentRelayAddress, err := w.relayManager.RelayAddress()
|
|
if err != nil {
|
|
w.log.Infof("local Relay connection is lost, skipping connection attempt")
|
|
return
|
|
}
|
|
|
|
srv := w.preferredRelayServer(currentRelayAddress.String(), remoteOfferAnswer.RelaySrvAddress)
|
|
|
|
relayedConn, err := w.relayManager.OpenConn(srv, w.config.Key, w.conn.OnDisconnected)
|
|
if err != nil {
|
|
// todo handle all type errors
|
|
if errors.Is(err, relayClient.ErrConnAlreadyExists) {
|
|
w.log.Infof("do not need to reopen relay connection")
|
|
return
|
|
}
|
|
w.log.Infof("do not need to reopen relay connection: %s", err)
|
|
return
|
|
}
|
|
|
|
w.log.Debugf("Relay connection established with %s", srv)
|
|
go w.conn.OnConnReady(RelayConnInfo{
|
|
relayedConn: relayedConn,
|
|
rosenpassPubKey: remoteOfferAnswer.RosenpassPubKey,
|
|
rosenpassAddr: remoteOfferAnswer.RosenpassAddr,
|
|
})
|
|
}
|
|
|
|
func (w *WorkerRelay) RelayAddress() (net.Addr, error) {
|
|
return w.relayManager.RelayAddress()
|
|
}
|
|
|
|
func (w *WorkerRelay) IsController() bool {
|
|
return w.config.LocalKey > w.config.Key
|
|
}
|
|
|
|
func (w *WorkerRelay) RelayIsSupportedLocally() bool {
|
|
return w.relayManager.HasRelayAddress()
|
|
}
|
|
|
|
func (w *WorkerRelay) isRelaySupported(answer *OfferAnswer) bool {
|
|
if !w.relayManager.HasRelayAddress() {
|
|
return false
|
|
}
|
|
return answer.RelaySrvAddress != ""
|
|
}
|
|
|
|
func (w *WorkerRelay) preferredRelayServer(myRelayAddress, remoteRelayAddress string) string {
|
|
if w.IsController() {
|
|
return myRelayAddress
|
|
}
|
|
return remoteRelayAddress
|
|
}
|