Fix doc

client/main.go

@@ -1,50 +1,13 @@
 package main
 
 import (
-	"context"
-	"fmt"
-	"math/rand"
-	"net"
-	"time"
+	"os"
 
-	"github.com/netbirdio/netbird/iface/bind"
+	"github.com/netbirdio/netbird/client/cmd"
 )
 
 func main() {
-
-	udp, err := net.ListenUDP("udp4", nil)
-	if err != nil {
-		panic(err)
-	}
-	muxDefault := bind.NewUniversalUDPMuxDefault(bind.UniversalUDPMuxParams{UDPConn: udp, XORMappedAddrCacheTTL: 25 * time.Second})
-
-	go func() {
-		muxDefault.ReadFromConn(context.TODO())
-	}()
-
-	addr, err := net.ResolveUDPAddr("udp4", "18.198.11.240:5555")
-	if err != nil {
-		panic(err)
+	if err := cmd.Execute(); err != nil {
+		os.Exit(1)
 	}
-	for i := 0; i < 2; i++ {
-		go func() {
-			millis := rand.Intn(400) + 50
-			time.Sleep(time.Duration(millis) * time.Millisecond)
-
-			for {
-				a, err := muxDefault.GetXORMappedAddr(addr, 3*time.Second)
-				if err != nil {
-					//fmt.Println(time.Now())
-					fmt.Println(time.Now().String() + ": " + err.Error())
-					continue
-				}
-				//fmt.Println(time.Now())
-				fmt.Println(a)
-				/*millis = rand.Intn(1000-300) + 300
-				time.Sleep(time.Duration(millis) * time.Millisecond)*/
-			}
-		}()
-	}
-
-	select {}
 }

iface/bind/udp_mux_universal.go

@@ -209,7 +209,7 @@ func (m *UniversalUDPMuxDefault) GetXORMappedAddr(serverAddr net.Addr, deadline
 
 	// otherwise, make a STUN request to discover the address
 	// or wait for already sent request to complete
-	waitAddrReceived, err := m.sendStun(serverAddr)
+	waitAddrReceived, err := m.sendSTUN(serverAddr)
 	if err != nil {
 		return nil, fmt.Errorf("%s: %s", "failed to send STUN packet", err)
 	}
@@ -221,8 +221,10 @@ func (m *UniversalUDPMuxDefault) GetXORMappedAddr(serverAddr net.Addr, deadline
 		var addr *stun.XORMappedAddress
 		m.mu.Lock()
 		// A very odd case that mappedAddr is nil.
-		// But can happen when the deadline property is larger than params.XORMappedAddrCacheTTL.
-		// We protect the code from panic.
+		// Can happen when the deadline property is larger than params.XORMappedAddrCacheTTL.
+		// Or when we don't receive a response to our m.sendSTUN request (the response is handled asynchronously) and
+		// the XORMapped expires meanwhile triggering a closure of the waitAddrReceived channel.
+		// We protect the code from panic here.
 		if mappedAddr, ok := m.xorMappedMap[serverAddr.String()]; ok {
 			addr = mappedAddr.addr
 		}
@@ -236,11 +238,11 @@ func (m *UniversalUDPMuxDefault) GetXORMappedAddr(serverAddr net.Addr, deadline
 	}
 }
 
-// sendStun sends a STUN request via UDP conn.
+// sendSTUN sends a STUN request via UDP conn.
 //
 // The returned channel is closed when the STUN response has been received.
 // Method is safe for concurrent use.
-func (m *UniversalUDPMuxDefault) sendStun(serverAddr net.Addr) (chan struct{}, error) {
+func (m *UniversalUDPMuxDefault) sendSTUN(serverAddr net.Addr) (chan struct{}, error) {
 	m.mu.Lock()
 	defer m.mu.Unlock()