set endpoint when adding/updating peer

devilcove · Apr 22, 2024 · 8be36fa · 8be36fa
1 parent 9f9e80a
commit 8be36fa
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/internal/agent/interface.go b/internal/agent/interface.go
@@ -7,6 +7,7 @@ import (
 	"log/slog"
 	"net"
 	"slices"
+	"strconv"
 	"time"
 
 	"github.com/devilcove/boltdb"
@@ -372,8 +373,13 @@ func convertPeerToWG(netPeer plexus.NetworkPeer, peers []plexus.NetworkPeer) (wg
 	if err != nil {
 		return wgtypes.PeerConfig{}, err
 	}
+	addr, err := net.ResolveUDPAddr("udp", netPeer.Endpoint.String()+":"+strconv.Itoa(netPeer.PublicListenPort))
+	if err != nil {
+		return wgtypes.PeerConfig{}, err
+	}
 	return wgtypes.PeerConfig{
 		PublicKey:                   key,
+		Endpoint:                    addr,
 		PersistentKeepaliveInterval: &keepalive,
 		ReplaceAllowedIPs:           true,
 		AllowedIPs:                  getAllowedIPs(netPeer, peers),