From 319e2382612c60a7e6ac128c15152166015225ca Mon Sep 17 00:00:00 2001
From: Callan Bryant <callan.bryant@gmail.com>
Date: Sat, 7 Mar 2020 18:45:50 +0000
Subject: [PATCH] bugfix: add/remove peers without restarting interface

---
 sync.go | 46 +++++++++++++++++++++++++++++++++++++++-------
 1 file changed, 39 insertions(+), 7 deletions(-)

diff --git a/sync.go b/sync.go
index 19c3b1d..896fccc 100644
--- a/sync.go
+++ b/sync.go
@@ -12,17 +12,49 @@ func Sync() {
 }
 
 func ConfigureDevice(conf *DsnetConfig) {
-	wgConfig := wgtypes.Config{
-		PrivateKey:   &conf.PrivateKey.Key,
-		ListenPort:   &conf.ListenPort,
-		ReplacePeers: true,
-		Peers:        conf.GetWgPeerConfigs(),
-	}
-
 	wg, err := wgctrl.New()
 	check(err)
 	defer wg.Close()
 
+	dev, err := wg.Device(conf.InterfaceName)
+
+	if err != nil {
+		ExitFail("Could not retrieve device '%s' (%v)", conf.InterfaceName, err)
+	}
+
+	peers := conf.GetWgPeerConfigs()
+
+	// compare peers to see if any exist on the device and not the config. If
+	// so, they should be removed by appending a dummy peer with Remove:true + pubkey.
+	knownKeys := make(map[wgtypes.Key]bool)
+
+	for _, peer := range peers {
+		knownKeys[peer.PublicKey] = true
+	}
+
+	// find deleted peers, and append dummy "remove" peers
+	for _, peer := range dev.Peers {
+		if !knownKeys[peer.PublicKey] {
+			peers = append(peers, wgtypes.PeerConfig{
+				PublicKey: peer.PublicKey,
+				Remove: true,
+			})
+		}
+	}
+
+	wgConfig := wgtypes.Config{
+		PrivateKey:   &conf.PrivateKey.Key,
+		ListenPort:   &conf.ListenPort,
+		// ReplacePeers with the same peers results in those peers losing
+		// connection, so it's not possible to do declarative configuration
+		// idempotently with ReplacePeers like I had assumed. Instead, peers
+		// must be removed imperatively with Remove:true. Peers can still be
+		// added/updated with ConfigureDevice declaratively.
+		ReplacePeers: false,
+		Peers:        peers,
+	}
+
+
 	err = wg.ConfigureDevice(conf.InterfaceName, wgConfig)
 
 	if err != nil {