From 5b7d7ed940d62b25a7056682184aa60c8d05b65e Mon Sep 17 00:00:00 2001
From: Antoine Riard <ariard@student.42.fr>
Date: Thu, 28 Jan 2021 17:45:36 -0500
Subject: [PATCH] Add PeerManager::disconnect_by_node_id()

This public method allows a client to easily disconnect peers while only
owning its node id. It will clean up peer state and disconnect properly
its descriptor.
---
 lightning/src/ln/peer_handler.rs | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/lightning/src/ln/peer_handler.rs b/lightning/src/ln/peer_handler.rs
index 890df356c..e083c0119 100644
--- a/lightning/src/ln/peer_handler.rs
+++ b/lightning/src/ln/peer_handler.rs
@@ -1179,6 +1179,24 @@ impl<Descriptor: SocketDescriptor, CM: Deref, RM: Deref, L: Deref> PeerManager<D
 		};
 	}
 
+	/// Disconnect a peer given its node id.
+	///
+	/// Set no_connection_possible to true to prevent any further connection with this peer,
+	/// force-closing any channels we have with it.
+	///
+	/// If a peer is connected, this will call `disconnect_socket` on the descriptor for the peer,
+	/// so be careful about reentrancy issues.
+	pub fn disconnect_by_node_id(&self, node_id: PublicKey, no_connection_possible: bool) {
+		let mut peers_lock = self.peers.lock().unwrap();
+		if let Some(mut descriptor) = peers_lock.node_id_to_descriptor.remove(&node_id) {
+			log_trace!(self.logger, "Disconnecting peer with id {} due to client request", node_id);
+			peers_lock.peers.remove(&descriptor);
+			peers_lock.peers_needing_send.remove(&descriptor);
+			self.message_handler.chan_handler.peer_disconnected(&node_id, no_connection_possible);
+			descriptor.disconnect_socket();
+		}
+	}
+
 	/// This function should be called roughly once every 30 seconds.
 	/// It will send pings to each peer and disconnect those which did not respond to the last round of pings.
 
-- 
2.39.5