support removing peers

1 files changed, 12 insertions, 2 deletions

diff --git a/src/interface/mod.rs b/src/interface/mod.rs
index 93c926c..d9f078f 100644
--- a/src/interface/mod.rs
+++ b/src/interface/mod.rs
@@ -188,8 +188,18 @@ impl Interface {
                         state.router.add_allowed_ips(&info.allowed_ips, &peer);
                         let _ = state.pubkey_map.insert(info.pub_key, peer);
                     },
-                    UpdateEvent::RemovePeer(_pub_key) => {
-                        warn!("RemovePeer event not yet handled");
+                    UpdateEvent::RemovePeer(pub_key) => {
+                        if let Some(peer_ref) = state.pubkey_map.remove(&pub_key) {
+                            let peer    = peer_ref.borrow();
+                            let indices = peer.get_mapped_indices();
+
+                            for index in indices {
+                                let _ = state.index_map.remove(&index);
+                            }
+                            state.router.remove_allowed_ips(&peer.info.allowed_ips);
+                        } else {
+                            info!("RemovePeer request for nonexistent peer.");
+                        }
                     },
                     _ => warn!("unhandled UpdateEvent received")
                 }