extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2025-01-31 10:31:58 +01:00
Code Issues Packages Projects Releases Wiki Activity

Discover NAT side ip with STUN

Browse Source
This commit is contained in:
Zoltán Papp 2024-04-24 18:59:55 +02:00
parent 318d379658
commit adb787954e
1 changed files with 53 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
client/internal/relay/turn.go
View File

@ -52,7 +52,11 @@ func (r *PermanentTurn) Open() error {
return err
}
r.turnClient = client
r.listen()
err = r.turnClient.Listen()
if err != nil {
log.Errorf("failed to listen turn client: %v", err)
return err
}
relayConn, err := client.Allocate()
if err != nil {
@ -61,7 +65,7 @@ func (r *PermanentTurn) Open() error {
}
r.relayConn = relayConn
srvReflexiveAddress, err := r.discoverPublicIP()
srvReflexiveAddress, err := r.discoverPublicIPByStun()
if err != nil {
log.Errorf("failed to discover public IP: %v", err)
return err
@ -79,6 +83,26 @@ func (r *PermanentTurn) SrvRefAddr() net.Addr {
}
func (r *PermanentTurn) PunchHole(mappedAddr net.Addr) error {
/*
err := r.turnClient.CreatePermission(mappedAddr)
if err != nil {
log.Errorf("---- failed to create permission: %v", err)
return err
}
msg, err := stun.Build(stun.BindingRequest, stun.TransactionID,
stun.Fingerprint,
)
if err != nil {
log.Errorf("--- failed to build stun message: %v", err)
return nil
}
_, err = r.relayConn.WriteTo(msg.Raw, mappedAddr)
if err != nil {
log.Errorf("failed to write to relay conn: %v", err)
return err
}
*/
_, err := r.relayConn.WriteTo([]byte("Hello"), mappedAddr)
return err
}
@ -117,6 +141,33 @@ func (r *PermanentTurn) discoverPublicIP() (*net.UDPAddr, error) {
return udpAddr, nil
}
func (r *PermanentTurn) discoverPublicIPByStun() (*net.UDPAddr, error) {
c, err := stun.DialURI(r.stunURI, &stun.DialConfig{})
if err != nil {
panic(err)
}
message := stun.MustBuild(stun.TransactionID, stun.BindingRequest)
var addr *net.UDPAddr
err = c.Do(message, func(res stun.Event) {
if res.Error != nil {
panic(res.Error)
}
var xorAddr stun.XORMappedAddress
if err := xorAddr.GetFrom(res.Message); err != nil {
log.Errorf("failed to get xor address: %v", err)
return
}
addr = &net.UDPAddr{
IP: xorAddr.IP,
Port: xorAddr.Port,
}
})
if err != nil {
return nil, err
}
return addr, nil
}
func (r *PermanentTurn) listen() {
if !r.turnClientListenLock.TryLock() {
return