mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-12 09:50:47 +01:00
71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
|
package peer
|
||
|
|
||
|
import (
|
||
|
"github.com/pion/ice/v3"
|
||
|
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||
|
|
||
|
signal "github.com/netbirdio/netbird/signal/client"
|
||
|
sProto "github.com/netbirdio/netbird/signal/proto"
|
||
|
)
|
||
|
|
||
|
type Signaler struct {
|
||
|
signal signal.Client
|
||
|
wgPrivateKey wgtypes.Key
|
||
|
}
|
||
|
|
||
|
func NewSignaler(signal signal.Client, wgPrivateKey wgtypes.Key) *Signaler {
|
||
|
return &Signaler{
|
||
|
signal: signal,
|
||
|
wgPrivateKey: wgPrivateKey,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (s *Signaler) SignalOffer(offer OfferAnswer, remoteKey string) error {
|
||
|
return s.signalOfferAnswer(offer, remoteKey, sProto.Body_OFFER)
|
||
|
}
|
||
|
|
||
|
func (s *Signaler) SignalAnswer(offer OfferAnswer, remoteKey string) error {
|
||
|
return s.signalOfferAnswer(offer, remoteKey, sProto.Body_ANSWER)
|
||
|
}
|
||
|
|
||
|
func (s *Signaler) SignalICECandidate(candidate ice.Candidate, remoteKey string) error {
|
||
|
return s.signal.Send(&sProto.Message{
|
||
|
Key: s.wgPrivateKey.PublicKey().String(),
|
||
|
RemoteKey: remoteKey,
|
||
|
Body: &sProto.Body{
|
||
|
Type: sProto.Body_CANDIDATE,
|
||
|
Payload: candidate.Marshal(),
|
||
|
},
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (s *Signaler) Ready() bool {
|
||
|
return s.signal.Ready()
|
||
|
}
|
||
|
|
||
|
// SignalOfferAnswer signals either an offer or an answer to remote peer
|
||
|
func (s *Signaler) signalOfferAnswer(offerAnswer OfferAnswer, remoteKey string, bodyType sProto.Body_Type) error {
|
||
|
msg, err := signal.MarshalCredential(
|
||
|
s.wgPrivateKey,
|
||
|
offerAnswer.WgListenPort,
|
||
|
remoteKey,
|
||
|
&signal.Credential{
|
||
|
UFrag: offerAnswer.IceCredentials.UFrag,
|
||
|
Pwd: offerAnswer.IceCredentials.Pwd,
|
||
|
},
|
||
|
bodyType,
|
||
|
offerAnswer.RosenpassPubKey,
|
||
|
offerAnswer.RosenpassAddr,
|
||
|
offerAnswer.RelaySrvAddress)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
err = s.signal.Send(msg)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|