X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=lightning%2Fsrc%2Frouting%2Fscoring.rs;h=95ad581105af19a0ef7a927936d5b84d14b2bc62;hb=a02982fbba53d2b630cc024d1fed6ed37ed0b4e0;hp=524f0ed3158b40dabae37d9516009a2588b6cd08;hpb=87a6e013f704ad905cb32452f53b264e9b322918;p=rust-lightning

diff --git a/lightning/src/routing/scoring.rs b/lightning/src/routing/scoring.rs
index 524f0ed3..95ad5811 100644
--- a/lightning/src/routing/scoring.rs
+++ b/lightning/src/routing/scoring.rs
@@ -102,6 +102,12 @@ pub trait Score $(: $supertrait)* {
 
 	/// Handles updating channel penalties after successfully routing along a path.
 	fn payment_path_successful(&mut self, path: &[&RouteHop]);
+
+	/// Handles updating channel penalties after a probe over the given path failed.
+	fn probe_failed(&mut self, path: &[&RouteHop], short_channel_id: u64);
+
+	/// Handles updating channel penalties after a probe over the given path succeeded.
+	fn probe_successful(&mut self, path: &[&RouteHop]);
 }
 
 impl<S: Score, T: DerefMut<Target=S> $(+ $supertrait)*> Score for T {
@@ -118,6 +124,14 @@ impl<S: Score, T: DerefMut<Target=S> $(+ $supertrait)*> Score for T {
 	fn payment_path_successful(&mut self, path: &[&RouteHop]) {
 		self.deref_mut().payment_path_successful(path)
 	}
+
+	fn probe_failed(&mut self, path: &[&RouteHop], short_channel_id: u64) {
+		self.deref_mut().probe_failed(path, short_channel_id)
+	}
+
+	fn probe_successful(&mut self, path: &[&RouteHop]) {
+		self.deref_mut().probe_successful(path)
+	}
 }
 } }
 
@@ -241,6 +255,10 @@ impl Score for FixedPenaltyScorer {
 	fn payment_path_failed(&mut self, _path: &[&RouteHop], _short_channel_id: u64) {}
 
 	fn payment_path_successful(&mut self, _path: &[&RouteHop]) {}
+
+	fn probe_failed(&mut self, _path: &[&RouteHop], _short_channel_id: u64) {}
+
+	fn probe_successful(&mut self, _path: &[&RouteHop]) {}
 }
 
 impl Writeable for FixedPenaltyScorer {
@@ -811,6 +829,14 @@ impl<G: Deref<Target = NetworkGraph<L>>, L: Deref, T: Time> Score for Probabilis
 			}
 		}
 	}
+
+	fn probe_failed(&mut self, path: &[&RouteHop], short_channel_id: u64) {
+		self.payment_path_failed(path, short_channel_id)
+	}
+
+	fn probe_successful(&mut self, path: &[&RouteHop]) {
+		self.payment_path_failed(path, u64::max_value())
+	}
 }
 
 mod approx {