Merge pull request #28 from devilcove/feature/virt-subnet

bugfixes when adding/deleting peers

Author: mattkasun

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),

wireguard.go

@@ -185,9 +185,9 @@ func (wg *wireguard) ReplacePeer(newPeer wgtypes.PeerConfig) {
 }
 
 func (wg *wireguard) DeletePeer(key string) {
-	for _, peer := range wg.Config.Peers {
+	for i, peer := range wg.Config.Peers {
 		if peer.PublicKey.String() == key {
-			peer.Remove = true
+			wg.Config.Peers[i].Remove = true
 			break
 		}
 	}