Expose peer addresses via get_peer_addresses

Author: tnull

lightning/src/ln/peer_handler.rs

@@ -744,6 +744,25 @@ impl<Descriptor: SocketDescriptor, CM: Deref, RM: Deref, OM: Deref, L: Deref, CM
 		}).collect()
 	}
 
+	/// Get a map from node id to node addresses for peers which have completed the initial handshake.
+	///
+	/// For outbound connections, the key will be the same as the `their_node_id` parameter passed in to
+	/// `new_outbound_connection`, however entries will only appear once the initial handshake has
+	/// completed and we are sure the remote peer has the private key for the given `node_id`.
+	pub fn get_peer_addresses(&self) -> HashMap<PublicKey, NetAddress> {
+		let mut peer_addrs = HashMap::new();
+
+		let peers = self.peers.read().unwrap();
+		for peer_mutex in peers.values() {
+			let p = peer_mutex.lock().unwrap();
+			if p.channel_encryptor.is_ready_for_encryption() && p.their_features.is_some() &&
+				p.their_node_id.is_some() && p.their_net_address.is_some() {
+				peer_addrs.insert(p.their_node_id.unwrap(), p.their_net_address.clone().unwrap());
+			}
+		}
+		peer_addrs
+	}
+
 	fn get_ephemeral_key(&self) -> SecretKey {
 		let mut ephemeral_hash = self.ephemeral_key_midstate.clone();
 		let counter = self.peer_counter.get_increment();