mirror of
https://github.com/netbirdio/netbird.git
synced 2026-09-12 17:59:06 +02:00
Zoltán Papp
73cffdb702
Files move directly between peers over the overlay, with no server in the path. The receiver listens on the WireGuard address only, so the port is unreachable from outside the tunnel, and every offer is matched to a known peer before anything is read. Consent is the default: an offer carries metadata alone, and no payload moves until the receiver accepts. Policy is per profile and device-local — off, ask, or auto-accept, with per-sender exceptions on top. Policy and history live in the profile's preferences, so removing a profile takes its file drop state with it. Transfers interrupted by a restart are settled on load; nothing survives to finish them, and left alone they would sit in the log as permanently pending. The Android bindings pull payload bytes through a chunk-returning stream: gomobile copies a []byte argument into a fresh Java array and never copies it back, so a fill-my-buffer method would hand back the right length with no data.
100 lines
2.7 KiB
Go
100 lines
2.7 KiB
Go
package peer
|
|
|
|
import (
|
|
"sync/atomic"
|
|
|
|
"github.com/pion/ice/v4"
|
|
log "github.com/sirupsen/logrus"
|
|
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
|
|
|
signal "github.com/netbirdio/netbird/shared/signal/client"
|
|
sProto "github.com/netbirdio/netbird/shared/signal/proto"
|
|
)
|
|
|
|
type Signaler struct {
|
|
signal signal.Client
|
|
wgPrivateKey wgtypes.Key
|
|
filedropPort atomic.Uint32
|
|
}
|
|
|
|
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()
|
|
}
|
|
|
|
// SetFiledropPort sets the file drop listen port advertised in offers and answers;
|
|
// 0 means the well-known default and is not put on the wire.
|
|
func (s *Signaler) SetFiledropPort(port uint16) {
|
|
s.filedropPort.Store(uint32(port))
|
|
}
|
|
|
|
// SignalOfferAnswer signals either an offer or an answer to remote peer
|
|
func (s *Signaler) signalOfferAnswer(offerAnswer OfferAnswer, remoteKey string, bodyType sProto.Body_Type) error {
|
|
var sessionIDBytes []byte
|
|
if offerAnswer.SessionID != nil {
|
|
var err error
|
|
sessionIDBytes, err = offerAnswer.SessionID.Bytes()
|
|
if err != nil {
|
|
log.Warnf("failed to get session ID bytes: %v", err)
|
|
}
|
|
}
|
|
msg, err := signal.MarshalCredential(s.wgPrivateKey, remoteKey, signal.CredentialPayload{
|
|
Type: bodyType,
|
|
WgListenPort: offerAnswer.WgListenPort,
|
|
FiledropPort: uint16(s.filedropPort.Load()),
|
|
Credential: &signal.Credential{
|
|
UFrag: offerAnswer.IceCredentials.UFrag,
|
|
Pwd: offerAnswer.IceCredentials.Pwd,
|
|
},
|
|
RosenpassPubKey: offerAnswer.RosenpassPubKey,
|
|
RosenpassAddr: offerAnswer.RosenpassAddr,
|
|
RelaySrvAddress: offerAnswer.RelaySrvAddress,
|
|
RelaySrvIP: offerAnswer.RelaySrvIP,
|
|
SessionID: sessionIDBytes,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err = s.signal.Send(msg); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Signaler) SignalIdle(remoteKey string) error {
|
|
return s.signal.Send(&sProto.Message{
|
|
Key: s.wgPrivateKey.PublicKey().String(),
|
|
RemoteKey: remoteKey,
|
|
Body: &sProto.Body{
|
|
Type: sProto.Body_GO_IDLE,
|
|
},
|
|
})
|
|
}
|