Inital implementation of trie

1 files changed, 35 insertions, 6 deletions

diff --git a/src/device.go b/src/device.go
index cd0835c..d03057d 100644
--- a/src/device.go
+++ b/src/device.go
@@ -5,10 +5,39 @@ import (
 )
 
 type Device struct {
-	mutex      sync.RWMutex
-	peers      map[NoisePublicKey]*Peer
-	privateKey NoisePrivateKey
-	publicKey  NoisePublicKey
-	fwMark     uint32
-	listenPort uint16
+	mutex        sync.RWMutex
+	peers        map[NoisePublicKey]*Peer
+	privateKey   NoisePrivateKey
+	publicKey    NoisePublicKey
+	fwMark       uint32
+	listenPort   uint16
+	routingTable RoutingTable
+}
+
+func (dev *Device) RemovePeer(key NoisePublicKey) {
+	dev.mutex.Lock()
+	defer dev.mutex.Unlock()
+	peer, ok := dev.peers[key]
+	if !ok {
+		return
+	}
+	peer.mutex.Lock()
+	dev.routingTable.RemovePeer(peer)
+	delete(dev.peers, key)
+}
+
+func (dev *Device) RemoveAllAllowedIps(peer *Peer) {
+
+}
+
+func (dev *Device) RemoveAllPeers() {
+	dev.mutex.Lock()
+	defer dev.mutex.Unlock()
+
+	for key, peer := range dev.peers {
+		peer.mutex.Lock()
+		dev.routingTable.RemovePeer(peer)
+		delete(dev.peers, key)
+		peer.mutex.Unlock()
+	}
 }