X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=lightning%2Fsrc%2Frouting%2Fscoring.rs;h=66900f2ba43bce7553bd951754c7684f3b314910;hb=feabd61bfe74be3fb8878d9c4ae2a186e60e61a4;hp=a3d970715872433d40572d98301b0309419876ab;hpb=a922830ebbea1207d178b455237abe5474598026;p=rust-lightning

diff --git a/lightning/src/routing/scoring.rs b/lightning/src/routing/scoring.rs
index a3d97071..66900f2b 100644
--- a/lightning/src/routing/scoring.rs
+++ b/lightning/src/routing/scoring.rs
@@ -188,12 +188,39 @@ pub struct MultiThreadedLockableScore<S: Score> {
 	score: Mutex<S>,
 }
 #[cfg(c_bindings)]
-/// (C-not exported)
+/// A locked `MultiThreadedLockableScore`.
+pub struct MultiThreadedLockableScoreLock<'a, S: Score>(MutexGuard<'a, S>);
+#[cfg(c_bindings)]
+impl<'a, T: Score + 'a> Score for MultiThreadedLockableScoreLock<'a, T> {
+	fn channel_penalty_msat(&self, scid: u64, source: &NodeId, target: &NodeId, usage: ChannelUsage) -> u64 {
+		self.0.channel_penalty_msat(scid, source, target, usage)
+	}
+	fn payment_path_failed(&mut self, path: &[&RouteHop], short_channel_id: u64) {
+		self.0.payment_path_failed(path, short_channel_id)
+	}
+	fn payment_path_successful(&mut self, path: &[&RouteHop]) {
+		self.0.payment_path_successful(path)
+	}
+	fn probe_failed(&mut self, path: &[&RouteHop], short_channel_id: u64) {
+		self.0.probe_failed(path, short_channel_id)
+	}
+	fn probe_successful(&mut self, path: &[&RouteHop]) {
+		self.0.probe_successful(path)
+	}
+}
+#[cfg(c_bindings)]
+impl<'a, T: Score + 'a> Writeable for MultiThreadedLockableScoreLock<'a, T> {
+	fn write<W: Writer>(&self, writer: &mut W) -> Result<(), io::Error> {
+		self.0.write(writer)
+	}
+}
+
+#[cfg(c_bindings)]
 impl<'a, T: Score + 'a> LockableScore<'a> for MultiThreadedLockableScore<T> {
-	type Locked = MutexGuard<'a, T>;
+	type Locked = MultiThreadedLockableScoreLock<'a, T>;
 
-	fn lock(&'a self) -> MutexGuard<'a, T> {
-		Mutex::lock(&self.score).unwrap()
+	fn lock(&'a self) -> MultiThreadedLockableScoreLock<'a, T> {
+		MultiThreadedLockableScoreLock(Mutex::lock(&self.score).unwrap())
 	}
 }
 
@@ -1281,7 +1308,7 @@ mod tests {
 	use util::time::Time;
 	use util::time::tests::SinceEpoch;
 
-	use ln::features::{ChannelFeatures, NodeFeatures};
+	use ln::channelmanager;
 	use ln::msgs::{ChannelAnnouncement, ChannelUpdate, UnsignedChannelAnnouncement, UnsignedChannelUpdate};
 	use routing::gossip::{EffectiveCapacity, NetworkGraph, NodeId};
 	use routing::router::RouteHop;
@@ -1372,7 +1399,7 @@ mod tests {
 		let node_2_secret = &SecretKey::from_slice(&[40; 32]).unwrap();
 		let secp_ctx = Secp256k1::new();
 		let unsigned_announcement = UnsignedChannelAnnouncement {
-			features: ChannelFeatures::known(),
+			features: channelmanager::provided_channel_features(),
 			chain_hash: genesis_hash,
 			short_channel_id,
 			node_id_1: PublicKey::from_secret_key(&secp_ctx, &node_1_key),
@@ -1426,25 +1453,25 @@ mod tests {
 		vec![
 			RouteHop {
 				pubkey: source_pubkey(),
-				node_features: NodeFeatures::known(),
+				node_features: channelmanager::provided_node_features(),
 				short_channel_id: 41,
-				channel_features: ChannelFeatures::known(),
+				channel_features: channelmanager::provided_channel_features(),
 				fee_msat: 1,
 				cltv_expiry_delta: 18,
 			},
 			RouteHop {
 				pubkey: target_pubkey(),
-				node_features: NodeFeatures::known(),
+				node_features: channelmanager::provided_node_features(),
 				short_channel_id: 42,
-				channel_features: ChannelFeatures::known(),
+				channel_features: channelmanager::provided_channel_features(),
 				fee_msat: 2,
 				cltv_expiry_delta: 18,
 			},
 			RouteHop {
 				pubkey: recipient_pubkey(),
-				node_features: NodeFeatures::known(),
+				node_features: channelmanager::provided_node_features(),
 				short_channel_id: 43,
-				channel_features: ChannelFeatures::known(),
+				channel_features: channelmanager::provided_channel_features(),
 				fee_msat: amount_msat,
 				cltv_expiry_delta: 18,
 			},