Replace ChainWatchInterface in NetGraphMsgHandler
[rust-lightning] / lightning / src / routing / router.rs
1 //! The top-level routing/network map tracking logic lives here.
2 //!
3 //! You probably want to create a NetGraphMsgHandler and use that as your RoutingMessageHandler and then
4 //! interrogate it to get routes for your own payments.
5
6 use bitcoin::secp256k1::key::PublicKey;
7
8 use ln::channelmanager;
9 use ln::features::{ChannelFeatures, NodeFeatures};
10 use ln::msgs::{DecodeError, ErrorAction, LightningError, MAX_VALUE_MSAT};
11 use routing::network_graph::{NetworkGraph, RoutingFees};
12 use util::ser::{Writeable, Readable};
13 use util::logger::Logger;
14
15 use std::cmp;
16 use std::collections::{HashMap,BinaryHeap};
17 use std::ops::Deref;
18
19 /// A hop in a route
20 #[derive(Clone, PartialEq)]
21 pub struct RouteHop {
22         /// The node_id of the node at this hop.
23         pub pubkey: PublicKey,
24         /// The node_announcement features of the node at this hop. For the last hop, these may be
25         /// amended to match the features present in the invoice this node generated.
26         pub node_features: NodeFeatures,
27         /// The channel that should be used from the previous hop to reach this node.
28         pub short_channel_id: u64,
29         /// The channel_announcement features of the channel that should be used from the previous hop
30         /// to reach this node.
31         pub channel_features: ChannelFeatures,
32         /// The fee taken on this hop. For the last hop, this should be the full value of the payment.
33         pub fee_msat: u64,
34         /// The CLTV delta added for this hop. For the last hop, this should be the full CLTV value
35         /// expected at the destination, in excess of the current block height.
36         pub cltv_expiry_delta: u32,
37 }
38
39 impl Writeable for Vec<RouteHop> {
40         fn write<W: ::util::ser::Writer>(&self, writer: &mut W) -> Result<(), ::std::io::Error> {
41                 (self.len() as u8).write(writer)?;
42                 for hop in self.iter() {
43                         hop.pubkey.write(writer)?;
44                         hop.node_features.write(writer)?;
45                         hop.short_channel_id.write(writer)?;
46                         hop.channel_features.write(writer)?;
47                         hop.fee_msat.write(writer)?;
48                         hop.cltv_expiry_delta.write(writer)?;
49                 }
50                 Ok(())
51         }
52 }
53
54 impl Readable for Vec<RouteHop> {
55         fn read<R: ::std::io::Read>(reader: &mut R) -> Result<Vec<RouteHop>, DecodeError> {
56                 let hops_count: u8 = Readable::read(reader)?;
57                 let mut hops = Vec::with_capacity(hops_count as usize);
58                 for _ in 0..hops_count {
59                         hops.push(RouteHop {
60                                 pubkey: Readable::read(reader)?,
61                                 node_features: Readable::read(reader)?,
62                                 short_channel_id: Readable::read(reader)?,
63                                 channel_features: Readable::read(reader)?,
64                                 fee_msat: Readable::read(reader)?,
65                                 cltv_expiry_delta: Readable::read(reader)?,
66                         });
67                 }
68                 Ok(hops)
69         }
70 }
71
72 /// A route directs a payment from the sender (us) to the recipient. If the recipient supports MPP,
73 /// it can take multiple paths. Each path is composed of one or more hops through the network.
74 #[derive(Clone, PartialEq)]
75 pub struct Route {
76         /// The list of routes taken for a single (potentially-)multi-part payment. The pubkey of the
77         /// last RouteHop in each path must be the same.
78         /// Each entry represents a list of hops, NOT INCLUDING our own, where the last hop is the
79         /// destination. Thus, this must always be at least length one. While the maximum length of any
80         /// given path is variable, keeping the length of any path to less than 20 should currently
81         /// ensure it is viable.
82         pub paths: Vec<Vec<RouteHop>>,
83 }
84
85 impl Writeable for Route {
86         fn write<W: ::util::ser::Writer>(&self, writer: &mut W) -> Result<(), ::std::io::Error> {
87                 (self.paths.len() as u64).write(writer)?;
88                 for hops in self.paths.iter() {
89                         hops.write(writer)?;
90                 }
91                 Ok(())
92         }
93 }
94
95 impl Readable for Route {
96         fn read<R: ::std::io::Read>(reader: &mut R) -> Result<Route, DecodeError> {
97                 let path_count: u64 = Readable::read(reader)?;
98                 let mut paths = Vec::with_capacity(cmp::min(path_count, 128) as usize);
99                 for _ in 0..path_count {
100                         paths.push(Readable::read(reader)?);
101                 }
102                 Ok(Route { paths })
103         }
104 }
105
106 /// A channel descriptor which provides a last-hop route to get_route
107 pub struct RouteHint {
108         /// The node_id of the non-target end of the route
109         pub src_node_id: PublicKey,
110         /// The short_channel_id of this channel
111         pub short_channel_id: u64,
112         /// The fees which must be paid to use this channel
113         pub fees: RoutingFees,
114         /// The difference in CLTV values between this node and the next node.
115         pub cltv_expiry_delta: u16,
116         /// The minimum value, in msat, which must be relayed to the next hop.
117         pub htlc_minimum_msat: u64,
118 }
119
120 #[derive(Eq, PartialEq)]
121 struct RouteGraphNode {
122         pubkey: PublicKey,
123         lowest_fee_to_peer_through_node: u64,
124         lowest_fee_to_node: u64,
125 }
126
127 impl cmp::Ord for RouteGraphNode {
128         fn cmp(&self, other: &RouteGraphNode) -> cmp::Ordering {
129                 other.lowest_fee_to_peer_through_node.cmp(&self.lowest_fee_to_peer_through_node)
130                         .then_with(|| other.pubkey.serialize().cmp(&self.pubkey.serialize()))
131         }
132 }
133
134 impl cmp::PartialOrd for RouteGraphNode {
135         fn partial_cmp(&self, other: &RouteGraphNode) -> Option<cmp::Ordering> {
136                 Some(self.cmp(other))
137         }
138 }
139
140 struct DummyDirectionalChannelInfo {
141         cltv_expiry_delta: u32,
142         htlc_minimum_msat: u64,
143         fees: RoutingFees,
144 }
145
146
147 /// Gets a route from us to the given target node.
148 ///
149 /// Extra routing hops between known nodes and the target will be used if they are included in
150 /// last_hops.
151 ///
152 /// If some channels aren't announced, it may be useful to fill in a first_hops with the
153 /// results from a local ChannelManager::list_usable_channels() call. If it is filled in, our
154 /// view of our local channels (from net_graph_msg_handler) will be ignored, and only those in first_hops
155 /// will be used.
156 ///
157 /// Panics if first_hops contains channels without short_channel_ids
158 /// (ChannelManager::list_usable_channels will never include such channels).
159 ///
160 /// The fees on channels from us to next-hops are ignored (as they are assumed to all be
161 /// equal), however the enabled/disabled bit on such channels as well as the htlc_minimum_msat
162 /// *is* checked as they may change based on the receiving node.
163 pub fn get_route<L: Deref>(our_node_id: &PublicKey, network: &NetworkGraph, target: &PublicKey, first_hops: Option<&[channelmanager::ChannelDetails]>,
164         last_hops: &[RouteHint], final_value_msat: u64, final_cltv: u32, logger: L) -> Result<Route, LightningError> where L::Target: Logger {
165         // TODO: Obviously *only* using total fee cost sucks. We should consider weighting by
166         // uptime/success in using a node in the past.
167         if *target == *our_node_id {
168                 return Err(LightningError{err: "Cannot generate a route to ourselves".to_owned(), action: ErrorAction::IgnoreError});
169         }
170
171         if final_value_msat > MAX_VALUE_MSAT {
172                 return Err(LightningError{err: "Cannot generate a route of more value than all existing satoshis".to_owned(), action: ErrorAction::IgnoreError});
173         }
174
175         // We do a dest-to-source Dijkstra's sorting by each node's distance from the destination
176         // plus the minimum per-HTLC fee to get from it to another node (aka "shitty A*").
177         // TODO: There are a few tweaks we could do, including possibly pre-calculating more stuff
178         // to use as the A* heuristic beyond just the cost to get one node further than the current
179         // one.
180
181         let dummy_directional_info = DummyDirectionalChannelInfo { // used for first_hops routes
182                 cltv_expiry_delta: 0,
183                 htlc_minimum_msat: 0,
184                 fees: RoutingFees {
185                         base_msat: 0,
186                         proportional_millionths: 0,
187                 }
188         };
189
190         let mut targets = BinaryHeap::new(); //TODO: Do we care about switching to eg Fibbonaci heap?
191         let mut dist = HashMap::with_capacity(network.get_nodes().len());
192
193         let mut first_hop_targets = HashMap::with_capacity(if first_hops.is_some() { first_hops.as_ref().unwrap().len() } else { 0 });
194         if let Some(hops) = first_hops {
195                 for chan in hops {
196                         let short_channel_id = chan.short_channel_id.expect("first_hops should be filled in with usable channels, not pending ones");
197                         if chan.remote_network_id == *target {
198                                 return Ok(Route {
199                                         paths: vec![vec![RouteHop {
200                                                 pubkey: chan.remote_network_id,
201                                                 node_features: chan.counterparty_features.to_context(),
202                                                 short_channel_id,
203                                                 channel_features: chan.counterparty_features.to_context(),
204                                                 fee_msat: final_value_msat,
205                                                 cltv_expiry_delta: final_cltv,
206                                         }]],
207                                 });
208                         }
209                         first_hop_targets.insert(chan.remote_network_id, (short_channel_id, chan.counterparty_features.clone()));
210                 }
211                 if first_hop_targets.is_empty() {
212                         return Err(LightningError{err: "Cannot route when there are no outbound routes away from us".to_owned(), action: ErrorAction::IgnoreError});
213                 }
214         }
215
216         macro_rules! add_entry {
217                 // Adds entry which goes from $src_node_id to $dest_node_id
218                 // over the channel with id $chan_id with fees described in
219                 // $directional_info.
220                 ( $chan_id: expr, $src_node_id: expr, $dest_node_id: expr, $directional_info: expr, $chan_features: expr, $starting_fee_msat: expr ) => {
221                         //TODO: Explore simply adding fee to hit htlc_minimum_msat
222                         if $starting_fee_msat as u64 + final_value_msat >= $directional_info.htlc_minimum_msat {
223                                 let proportional_fee_millions = ($starting_fee_msat + final_value_msat).checked_mul($directional_info.fees.proportional_millionths as u64);
224                                 if let Some(new_fee) = proportional_fee_millions.and_then(|part| {
225                                                 ($directional_info.fees.base_msat as u64).checked_add(part / 1000000) })
226                                 {
227                                         let mut total_fee = $starting_fee_msat as u64;
228                                         let hm_entry = dist.entry(&$src_node_id);
229                                         let old_entry = hm_entry.or_insert_with(|| {
230                                                 let node = network.get_nodes().get(&$src_node_id).unwrap();
231                                                 let mut fee_base_msat = u32::max_value();
232                                                 let mut fee_proportional_millionths = u32::max_value();
233                                                 if let Some(fees) = node.lowest_inbound_channel_fees {
234                                                         fee_base_msat = fees.base_msat;
235                                                         fee_proportional_millionths = fees.proportional_millionths;
236                                                 };
237                                                 (u64::max_value(),
238                                                         fee_base_msat,
239                                                         fee_proportional_millionths,
240                                                         RouteHop {
241                                                                 pubkey: $dest_node_id.clone(),
242                                                                 node_features: NodeFeatures::empty(),
243                                                                 short_channel_id: 0,
244                                                                 channel_features: $chan_features.clone(),
245                                                                 fee_msat: 0,
246                                                                 cltv_expiry_delta: 0,
247                                                         },
248                                                 )
249                                         });
250                                         if $src_node_id != *our_node_id {
251                                                 // Ignore new_fee for channel-from-us as we assume all channels-from-us
252                                                 // will have the same effective-fee
253                                                 total_fee += new_fee;
254                                                 if let Some(fee_inc) = final_value_msat.checked_add(total_fee).and_then(|inc| { (old_entry.2 as u64).checked_mul(inc) }) {
255                                                         total_fee += fee_inc / 1000000 + (old_entry.1 as u64);
256                                                 } else {
257                                                         // max_value means we'll always fail the old_entry.0 > total_fee check
258                                                         total_fee = u64::max_value();
259                                                 }
260                                         }
261                                         let new_graph_node = RouteGraphNode {
262                                                 pubkey: $src_node_id,
263                                                 lowest_fee_to_peer_through_node: total_fee,
264                                                 lowest_fee_to_node: $starting_fee_msat as u64 + new_fee,
265                                         };
266                                         if old_entry.0 > total_fee {
267                                                 targets.push(new_graph_node);
268                                                 old_entry.0 = total_fee;
269                                                 old_entry.3 = RouteHop {
270                                                         pubkey: $dest_node_id.clone(),
271                                                         node_features: NodeFeatures::empty(),
272                                                         short_channel_id: $chan_id.clone(),
273                                                         channel_features: $chan_features.clone(),
274                                                         fee_msat: new_fee, // This field is ignored on the last-hop anyway
275                                                         cltv_expiry_delta: $directional_info.cltv_expiry_delta as u32,
276                                                 }
277                                         }
278                                 }
279                         }
280                 };
281         }
282
283         macro_rules! add_entries_to_cheapest_to_target_node {
284                 ( $node: expr, $node_id: expr, $fee_to_target_msat: expr ) => {
285                         if first_hops.is_some() {
286                                 if let Some(&(ref first_hop, ref features)) = first_hop_targets.get(&$node_id) {
287                                         add_entry!(first_hop, *our_node_id, $node_id, dummy_directional_info, features.to_context(), $fee_to_target_msat);
288                                 }
289                         }
290
291                         let features;
292                         if let Some(node_info) = $node.announcement_info.as_ref() {
293                                 features = node_info.features.clone();
294                         } else {
295                                 features = NodeFeatures::empty();
296                         }
297
298                         if !features.requires_unknown_bits() {
299                                 for chan_id in $node.channels.iter() {
300                                         let chan = network.get_channels().get(chan_id).unwrap();
301                                         if !chan.features.requires_unknown_bits() {
302                                                 if chan.node_one == *$node_id {
303                                                         // ie $node is one, ie next hop in A* is two, via the two_to_one channel
304                                                         if first_hops.is_none() || chan.node_two != *our_node_id {
305                                                                 if let Some(two_to_one) = chan.two_to_one.as_ref() {
306                                                                         if two_to_one.enabled {
307                                                                                 add_entry!(chan_id, chan.node_two, chan.node_one, two_to_one, chan.features, $fee_to_target_msat);
308                                                                         }
309                                                                 }
310                                                         }
311                                                 } else {
312                                                         if first_hops.is_none() || chan.node_one != *our_node_id {
313                                                                 if let Some(one_to_two) = chan.one_to_two.as_ref() {
314                                                                         if one_to_two.enabled {
315                                                                                 add_entry!(chan_id, chan.node_one, chan.node_two, one_to_two, chan.features, $fee_to_target_msat);
316                                                                         }
317                                                                 }
318
319                                                         }
320                                                 }
321                                         }
322                                 }
323                         }
324                 };
325         }
326
327         match network.get_nodes().get(target) {
328                 None => {},
329                 Some(node) => {
330                         add_entries_to_cheapest_to_target_node!(node, target, 0);
331                 },
332         }
333
334         for hop in last_hops.iter() {
335                 if first_hops.is_none() || hop.src_node_id != *our_node_id { // first_hop overrules last_hops
336                         if network.get_nodes().get(&hop.src_node_id).is_some() {
337                                 if first_hops.is_some() {
338                                         if let Some(&(ref first_hop, ref features)) = first_hop_targets.get(&hop.src_node_id) {
339                                                 // Currently there are no channel-context features defined, so we are a
340                                                 // bit lazy here. In the future, we should pull them out via our
341                                                 // ChannelManager, but there's no reason to waste the space until we
342                                                 // need them.
343                                                 add_entry!(first_hop, *our_node_id , hop.src_node_id, dummy_directional_info, features.to_context(), 0);
344                                         }
345                                 }
346                                 // BOLT 11 doesn't allow inclusion of features for the last hop hints, which
347                                 // really sucks, cause we're gonna need that eventually.
348                                 add_entry!(hop.short_channel_id, hop.src_node_id, target, hop, ChannelFeatures::empty(), 0);
349                         }
350                 }
351         }
352
353         while let Some(RouteGraphNode { pubkey, lowest_fee_to_node, .. }) = targets.pop() {
354                 if pubkey == *our_node_id {
355                         let mut res = vec!(dist.remove(&our_node_id).unwrap().3);
356                         loop {
357                                 if let Some(&(_, ref features)) = first_hop_targets.get(&res.last().unwrap().pubkey) {
358                                         res.last_mut().unwrap().node_features = features.to_context();
359                                 } else if let Some(node) = network.get_nodes().get(&res.last().unwrap().pubkey) {
360                                         if let Some(node_info) = node.announcement_info.as_ref() {
361                                                 res.last_mut().unwrap().node_features = node_info.features.clone();
362                                         } else {
363                                                 res.last_mut().unwrap().node_features = NodeFeatures::empty();
364                                         }
365                                 } else {
366                                         // We should be able to fill in features for everything except the last
367                                         // hop, if the last hop was provided via a BOLT 11 invoice (though we
368                                         // should be able to extend it further as BOLT 11 does have feature
369                                         // flags for the last hop node itself).
370                                         assert!(res.last().unwrap().pubkey == *target);
371                                 }
372                                 if res.last().unwrap().pubkey == *target {
373                                         break;
374                                 }
375
376                                 let new_entry = match dist.remove(&res.last().unwrap().pubkey) {
377                                         Some(hop) => hop.3,
378                                         None => return Err(LightningError{err: "Failed to find a non-fee-overflowing path to the given destination".to_owned(), action: ErrorAction::IgnoreError}),
379                                 };
380                                 res.last_mut().unwrap().fee_msat = new_entry.fee_msat;
381                                 res.last_mut().unwrap().cltv_expiry_delta = new_entry.cltv_expiry_delta;
382                                 res.push(new_entry);
383                         }
384                         res.last_mut().unwrap().fee_msat = final_value_msat;
385                         res.last_mut().unwrap().cltv_expiry_delta = final_cltv;
386                         let route = Route { paths: vec![res] };
387                         log_trace!(logger, "Got route: {}", log_route!(route));
388                         return Ok(route);
389                 }
390
391                 match network.get_nodes().get(&pubkey) {
392                         None => {},
393                         Some(node) => {
394                                 add_entries_to_cheapest_to_target_node!(node, &pubkey, lowest_fee_to_node);
395                         },
396                 }
397         }
398
399         Err(LightningError{err: "Failed to find a path to the given destination".to_owned(), action: ErrorAction::IgnoreError})
400 }
401
402 #[cfg(test)]
403 mod tests {
404         use routing::router::{get_route, RouteHint, RoutingFees};
405         use routing::network_graph::NetGraphMsgHandler;
406         use ln::features::{ChannelFeatures, InitFeatures, NodeFeatures};
407         use ln::msgs::{ErrorAction, LightningError, OptionalField, UnsignedChannelAnnouncement, ChannelAnnouncement, RoutingMessageHandler,
408            NodeAnnouncement, UnsignedNodeAnnouncement, ChannelUpdate, UnsignedChannelUpdate};
409         use ln::channelmanager;
410         use util::test_utils;
411         use util::ser::Writeable;
412
413         use bitcoin::hashes::sha256d::Hash as Sha256dHash;
414         use bitcoin::hashes::Hash;
415         use bitcoin::network::constants::Network;
416         use bitcoin::blockdata::constants::genesis_block;
417         use bitcoin::util::hash::BitcoinHash;
418
419         use hex;
420
421         use bitcoin::secp256k1::key::{PublicKey,SecretKey};
422         use bitcoin::secp256k1::{Secp256k1, All};
423
424         use std::sync::Arc;
425
426         // Using the same keys for LN and BTC ids
427         fn add_channel(net_graph_msg_handler: &NetGraphMsgHandler<Arc<test_utils::TestChainSource>, Arc<test_utils::TestLogger>>, secp_ctx: &Secp256k1<All>, node_1_privkey: &SecretKey,
428            node_2_privkey: &SecretKey, features: ChannelFeatures, short_channel_id: u64) {
429                 let node_id_1 = PublicKey::from_secret_key(&secp_ctx, node_1_privkey);
430                 let node_id_2 = PublicKey::from_secret_key(&secp_ctx, node_2_privkey);
431
432                 let unsigned_announcement = UnsignedChannelAnnouncement {
433                         features,
434                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
435                         short_channel_id,
436                         node_id_1,
437                         node_id_2,
438                         bitcoin_key_1: node_id_1,
439                         bitcoin_key_2: node_id_2,
440                         excess_data: Vec::new(),
441                 };
442
443                 let msghash = hash_to_message!(&Sha256dHash::hash(&unsigned_announcement.encode()[..])[..]);
444                 let valid_announcement = ChannelAnnouncement {
445                         node_signature_1: secp_ctx.sign(&msghash, node_1_privkey),
446                         node_signature_2: secp_ctx.sign(&msghash, node_2_privkey),
447                         bitcoin_signature_1: secp_ctx.sign(&msghash, node_1_privkey),
448                         bitcoin_signature_2: secp_ctx.sign(&msghash, node_2_privkey),
449                         contents: unsigned_announcement.clone(),
450                 };
451                 match net_graph_msg_handler.handle_channel_announcement(&valid_announcement) {
452                         Ok(res) => assert!(res),
453                         _ => panic!()
454                 };
455         }
456
457         fn update_channel(net_graph_msg_handler: &NetGraphMsgHandler<Arc<test_utils::TestChainSource>, Arc<test_utils::TestLogger>>, secp_ctx: &Secp256k1<All>, node_privkey: &SecretKey, update: UnsignedChannelUpdate) {
458                 let msghash = hash_to_message!(&Sha256dHash::hash(&update.encode()[..])[..]);
459                 let valid_channel_update = ChannelUpdate {
460                         signature: secp_ctx.sign(&msghash, node_privkey),
461                         contents: update.clone()
462                 };
463
464                 match net_graph_msg_handler.handle_channel_update(&valid_channel_update) {
465                         Ok(res) => assert!(res),
466                         // Err(_) => panic!()
467                         Err(e) => println!("{:?}", e.err)
468                 };
469         }
470
471
472         fn add_or_update_node(net_graph_msg_handler: &NetGraphMsgHandler<Arc<test_utils::TestChainSource>, Arc<test_utils::TestLogger>>, secp_ctx: &Secp256k1<All>, node_privkey: &SecretKey,
473            features: NodeFeatures, timestamp: u32) {
474                 let node_id = PublicKey::from_secret_key(&secp_ctx, node_privkey);
475                 let unsigned_announcement = UnsignedNodeAnnouncement {
476                         features,
477                         timestamp,
478                         node_id,
479                         rgb: [0; 3],
480                         alias: [0; 32],
481                         addresses: Vec::new(),
482                         excess_address_data: Vec::new(),
483                         excess_data: Vec::new(),
484                 };
485                 let msghash = hash_to_message!(&Sha256dHash::hash(&unsigned_announcement.encode()[..])[..]);
486                 let valid_announcement = NodeAnnouncement {
487                         signature: secp_ctx.sign(&msghash, node_privkey),
488                         contents: unsigned_announcement.clone()
489                 };
490
491                 match net_graph_msg_handler.handle_node_announcement(&valid_announcement) {
492                         Ok(_) => (),
493                         Err(_) => panic!()
494                 };
495         }
496
497         #[test]
498         fn route_test() {
499                 let secp_ctx = Secp256k1::new();
500                 let our_privkey = &SecretKey::from_slice(&hex::decode("0101010101010101010101010101010101010101010101010101010101010101").unwrap()[..]).unwrap();
501                 let our_id = PublicKey::from_secret_key(&secp_ctx, our_privkey);
502                 let logger = Arc::new(test_utils::TestLogger::new());
503                 let net_graph_msg_handler = NetGraphMsgHandler::new(None, Arc::clone(&logger));
504                 // Build network from our_id to node8:
505                 //
506                 //        -1(1)2-  node1  -1(3)2-
507                 //       /                       \
508                 // our_id -1(12)2- node8 -1(13)2--- node3
509                 //       \                       /
510                 //        -1(2)2-  node2  -1(4)2-
511                 //
512                 //
513                 // chan1  1-to-2: disabled
514                 // chan1  2-to-1: enabled, 0 fee
515                 //
516                 // chan2  1-to-2: enabled, ignored fee
517                 // chan2  2-to-1: enabled, 0 fee
518                 //
519                 // chan3  1-to-2: enabled, 0 fee
520                 // chan3  2-to-1: enabled, 100 msat fee
521                 //
522                 // chan4  1-to-2: enabled, 100% fee
523                 // chan4  2-to-1: enabled, 0 fee
524                 //
525                 // chan12 1-to-2: enabled, ignored fee
526                 // chan12 2-to-1: enabled, 0 fee
527                 //
528                 // chan13 1-to-2: enabled, 200% fee
529                 // chan13 2-to-1: enabled, 0 fee
530                 //
531                 //
532                 //       -1(5)2- node4 -1(8)2--
533                 //       |         2          |
534                 //       |       (11)         |
535                 //      /          1           \
536                 // node3--1(6)2- node5 -1(9)2--- node7 (not in global route map)
537                 //      \                      /
538                 //       -1(7)2- node6 -1(10)2-
539                 //
540                 // chan5  1-to-2: enabled, 100 msat fee
541                 // chan5  2-to-1: enabled, 0 fee
542                 //
543                 // chan6  1-to-2: enabled, 0 fee
544                 // chan6  2-to-1: enabled, 0 fee
545                 //
546                 // chan7  1-to-2: enabled, 100% fee
547                 // chan7  2-to-1: enabled, 0 fee
548                 //
549                 // chan8  1-to-2: enabled, variable fee (0 then 1000 msat)
550                 // chan8  2-to-1: enabled, 0 fee
551                 //
552                 // chan9  1-to-2: enabled, 1001 msat fee
553                 // chan9  2-to-1: enabled, 0 fee
554                 //
555                 // chan10 1-to-2: enabled, 0 fee
556                 // chan10 2-to-1: enabled, 0 fee
557                 //
558                 // chan11 1-to-2: enabled, 0 fee
559                 // chan11 2-to-1: enabled, 0 fee
560
561                 let node1_privkey = &SecretKey::from_slice(&hex::decode("0202020202020202020202020202020202020202020202020202020202020202").unwrap()[..]).unwrap();
562                 let node2_privkey = &SecretKey::from_slice(&hex::decode("0303030303030303030303030303030303030303030303030303030303030303").unwrap()[..]).unwrap();
563                 let node3_privkey = &SecretKey::from_slice(&hex::decode("0404040404040404040404040404040404040404040404040404040404040404").unwrap()[..]).unwrap();
564                 let node4_privkey = &SecretKey::from_slice(&hex::decode("0505050505050505050505050505050505050505050505050505050505050505").unwrap()[..]).unwrap();
565                 let node5_privkey = &SecretKey::from_slice(&hex::decode("0606060606060606060606060606060606060606060606060606060606060606").unwrap()[..]).unwrap();
566                 let node6_privkey = &SecretKey::from_slice(&hex::decode("0707070707070707070707070707070707070707070707070707070707070707").unwrap()[..]).unwrap();
567                 let node7_privkey = &SecretKey::from_slice(&hex::decode("0808080808080808080808080808080808080808080808080808080808080808").unwrap()[..]).unwrap();
568                 let node8_privkey = &SecretKey::from_slice(&hex::decode("0909090909090909090909090909090909090909090909090909090909090909").unwrap()[..]).unwrap();
569
570
571                 let node1 = PublicKey::from_secret_key(&secp_ctx, node1_privkey);
572                 let node2 = PublicKey::from_secret_key(&secp_ctx, node2_privkey);
573                 let node3 = PublicKey::from_secret_key(&secp_ctx, node3_privkey);
574                 let node4 = PublicKey::from_secret_key(&secp_ctx, node4_privkey);
575                 let node5 = PublicKey::from_secret_key(&secp_ctx, node5_privkey);
576                 let node6 = PublicKey::from_secret_key(&secp_ctx, node6_privkey);
577                 let node7 = PublicKey::from_secret_key(&secp_ctx, node7_privkey);
578                 let node8 = PublicKey::from_secret_key(&secp_ctx, node8_privkey);
579
580                 macro_rules! id_to_feature_flags {
581                         // Set the feature flags to the id'th odd (ie non-required) feature bit so that we can
582                         // test for it later.
583                         ($id: expr) => { {
584                                 let idx = ($id - 1) * 2 + 1;
585                                 if idx > 8*3 {
586                                         vec![1 << (idx - 8*3), 0, 0, 0]
587                                 } else if idx > 8*2 {
588                                         vec![1 << (idx - 8*2), 0, 0]
589                                 } else if idx > 8*1 {
590                                         vec![1 << (idx - 8*1), 0]
591                                 } else {
592                                         vec![1 << idx]
593                                 }
594                         } }
595                 }
596
597                 add_channel(&net_graph_msg_handler, &secp_ctx, our_privkey, node1_privkey, ChannelFeatures::from_le_bytes(id_to_feature_flags!(1)), 1);
598                 update_channel(&net_graph_msg_handler, &secp_ctx, node1_privkey, UnsignedChannelUpdate {
599                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
600                         short_channel_id: 1,
601                         timestamp: 1,
602                         flags: 1,
603                         cltv_expiry_delta: 0,
604                         htlc_minimum_msat: 0,
605                         htlc_maximum_msat: OptionalField::Absent,
606                         fee_base_msat: 0,
607                         fee_proportional_millionths: 0,
608                         excess_data: Vec::new()
609                 });
610                 add_or_update_node(&net_graph_msg_handler, &secp_ctx, node1_privkey, NodeFeatures::from_le_bytes(id_to_feature_flags!(1)), 0);
611
612                 add_channel(&net_graph_msg_handler, &secp_ctx, our_privkey, node2_privkey, ChannelFeatures::from_le_bytes(id_to_feature_flags!(2)), 2);
613                 update_channel(&net_graph_msg_handler, &secp_ctx, our_privkey, UnsignedChannelUpdate {
614                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
615                         short_channel_id: 2,
616                         timestamp: 1,
617                         flags: 0,
618                         cltv_expiry_delta: u16::max_value(),
619                         htlc_minimum_msat: 0,
620                         htlc_maximum_msat: OptionalField::Absent,
621                         fee_base_msat: u32::max_value(),
622                         fee_proportional_millionths: u32::max_value(),
623                         excess_data: Vec::new()
624                 });
625                 update_channel(&net_graph_msg_handler, &secp_ctx, node2_privkey, UnsignedChannelUpdate {
626                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
627                         short_channel_id: 2,
628                         timestamp: 1,
629                         flags: 1,
630                         cltv_expiry_delta: 0,
631                         htlc_minimum_msat: 0,
632                         htlc_maximum_msat: OptionalField::Absent,
633                         fee_base_msat: 0,
634                         fee_proportional_millionths: 0,
635                         excess_data: Vec::new()
636                 });
637
638                 add_or_update_node(&net_graph_msg_handler, &secp_ctx, node2_privkey, NodeFeatures::from_le_bytes(id_to_feature_flags!(2)), 0);
639
640                 add_channel(&net_graph_msg_handler, &secp_ctx, our_privkey, node8_privkey, ChannelFeatures::from_le_bytes(id_to_feature_flags!(12)), 12);
641                 update_channel(&net_graph_msg_handler, &secp_ctx, our_privkey, UnsignedChannelUpdate {
642                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
643                         short_channel_id: 12,
644                         timestamp: 1,
645                         flags: 0,
646                         cltv_expiry_delta: u16::max_value(),
647                         htlc_minimum_msat: 0,
648                         htlc_maximum_msat: OptionalField::Absent,
649                         fee_base_msat: u32::max_value(),
650                         fee_proportional_millionths: u32::max_value(),
651                         excess_data: Vec::new()
652                 });
653                 update_channel(&net_graph_msg_handler, &secp_ctx, node8_privkey, UnsignedChannelUpdate {
654                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
655                         short_channel_id: 12,
656                         timestamp: 1,
657                         flags: 1,
658                         cltv_expiry_delta: 0,
659                         htlc_minimum_msat: 0,
660                         htlc_maximum_msat: OptionalField::Absent,
661                         fee_base_msat: 0,
662                         fee_proportional_millionths: 0,
663                         excess_data: Vec::new()
664                 });
665
666
667                 add_or_update_node(&net_graph_msg_handler, &secp_ctx, node8_privkey, NodeFeatures::from_le_bytes(id_to_feature_flags!(8)), 0);
668
669                 add_channel(&net_graph_msg_handler, &secp_ctx, node1_privkey, node3_privkey, ChannelFeatures::from_le_bytes(id_to_feature_flags!(3)), 3);
670                 update_channel(&net_graph_msg_handler, &secp_ctx, node1_privkey, UnsignedChannelUpdate {
671                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
672                         short_channel_id: 3,
673                         timestamp: 1,
674                         flags: 0,
675                         cltv_expiry_delta: (3 << 8) | 1,
676                         htlc_minimum_msat: 0,
677                         htlc_maximum_msat: OptionalField::Absent,
678                         fee_base_msat: 0,
679                         fee_proportional_millionths: 0,
680                         excess_data: Vec::new()
681                 });
682                 update_channel(&net_graph_msg_handler, &secp_ctx, node3_privkey, UnsignedChannelUpdate {
683                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
684                         short_channel_id: 3,
685                         timestamp: 1,
686                         flags: 1,
687                         cltv_expiry_delta: (3 << 8) | 2,
688                         htlc_minimum_msat: 0,
689                         htlc_maximum_msat: OptionalField::Absent,
690                         fee_base_msat: 100,
691                         fee_proportional_millionths: 0,
692                         excess_data: Vec::new()
693                 });
694
695
696                 add_channel(&net_graph_msg_handler, &secp_ctx, node2_privkey, node3_privkey, ChannelFeatures::from_le_bytes(id_to_feature_flags!(4)), 4);
697                 update_channel(&net_graph_msg_handler, &secp_ctx, node2_privkey, UnsignedChannelUpdate {
698                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
699                         short_channel_id: 4,
700                         timestamp: 1,
701                         flags: 0,
702                         cltv_expiry_delta: (4 << 8) | 1,
703                         htlc_minimum_msat: 0,
704                         htlc_maximum_msat: OptionalField::Absent,
705                         fee_base_msat: 0,
706                         fee_proportional_millionths: 1000000,
707                         excess_data: Vec::new()
708                 });
709                 update_channel(&net_graph_msg_handler, &secp_ctx, node3_privkey, UnsignedChannelUpdate {
710                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
711                         short_channel_id: 4,
712                         timestamp: 1,
713                         flags: 1,
714                         cltv_expiry_delta: (4 << 8) | 2,
715                         htlc_minimum_msat: 0,
716                         htlc_maximum_msat: OptionalField::Absent,
717                         fee_base_msat: 0,
718                         fee_proportional_millionths: 0,
719                         excess_data: Vec::new()
720                 });
721
722                 add_channel(&net_graph_msg_handler, &secp_ctx, node8_privkey, node3_privkey, ChannelFeatures::from_le_bytes(id_to_feature_flags!(13)), 13);
723                 update_channel(&net_graph_msg_handler, &secp_ctx, node8_privkey, UnsignedChannelUpdate {
724                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
725                         short_channel_id: 13,
726                         timestamp: 1,
727                         flags: 0,
728                         cltv_expiry_delta: (13 << 8) | 1,
729                         htlc_minimum_msat: 0,
730                         htlc_maximum_msat: OptionalField::Absent,
731                         fee_base_msat: 0,
732                         fee_proportional_millionths: 2000000,
733                         excess_data: Vec::new()
734                 });
735                 update_channel(&net_graph_msg_handler, &secp_ctx, node3_privkey, UnsignedChannelUpdate {
736                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
737                         short_channel_id: 13,
738                         timestamp: 1,
739                         flags: 1,
740                         cltv_expiry_delta: (13 << 8) | 2,
741                         htlc_minimum_msat: 0,
742                         htlc_maximum_msat: OptionalField::Absent,
743                         fee_base_msat: 0,
744                         fee_proportional_millionths: 0,
745                         excess_data: Vec::new()
746                 });
747                 add_or_update_node(&net_graph_msg_handler, &secp_ctx, node3_privkey, NodeFeatures::from_le_bytes(id_to_feature_flags!(3)), 0);
748                 add_channel(&net_graph_msg_handler, &secp_ctx, node3_privkey, node5_privkey, ChannelFeatures::from_le_bytes(id_to_feature_flags!(6)), 6);
749
750                 update_channel(&net_graph_msg_handler, &secp_ctx, node3_privkey, UnsignedChannelUpdate {
751                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
752                         short_channel_id: 6,
753                         timestamp: 1,
754                         flags: 0,
755                         cltv_expiry_delta: (6 << 8) | 1,
756                         htlc_minimum_msat: 0,
757                         htlc_maximum_msat: OptionalField::Absent,
758                         fee_base_msat: 0,
759                         fee_proportional_millionths: 0,
760                         excess_data: Vec::new()
761                 });
762                 update_channel(&net_graph_msg_handler, &secp_ctx, node5_privkey, UnsignedChannelUpdate {
763                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
764                         short_channel_id: 6,
765                         timestamp: 1,
766                         flags: 1,
767                         cltv_expiry_delta: (6 << 8) | 2,
768                         htlc_minimum_msat: 0,
769                         htlc_maximum_msat: OptionalField::Absent,
770                         fee_base_msat: 0,
771                         fee_proportional_millionths: 0,
772                         excess_data: Vec::new()
773                 });
774
775                 add_channel(&net_graph_msg_handler, &secp_ctx, node5_privkey, node4_privkey, ChannelFeatures::from_le_bytes(id_to_feature_flags!(11)), 11);
776                 update_channel(&net_graph_msg_handler, &secp_ctx, node5_privkey, UnsignedChannelUpdate {
777                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
778                         short_channel_id: 11,
779                         timestamp: 1,
780                         flags: 0,
781                         cltv_expiry_delta: (11 << 8) | 1,
782                         htlc_minimum_msat: 0,
783                         htlc_maximum_msat: OptionalField::Absent,
784                         fee_base_msat: 0,
785                         fee_proportional_millionths: 0,
786                         excess_data: Vec::new()
787                 });
788                 update_channel(&net_graph_msg_handler, &secp_ctx, node4_privkey, UnsignedChannelUpdate {
789                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
790                         short_channel_id: 11,
791                         timestamp: 1,
792                         flags: 1,
793                         cltv_expiry_delta: (11 << 8) | 2,
794                         htlc_minimum_msat: 0,
795                         htlc_maximum_msat: OptionalField::Absent,
796                         fee_base_msat: 0,
797                         fee_proportional_millionths: 0,
798                         excess_data: Vec::new()
799                 });
800                 add_or_update_node(&net_graph_msg_handler, &secp_ctx, node5_privkey, NodeFeatures::from_le_bytes(id_to_feature_flags!(5)), 0);
801                 add_or_update_node(&net_graph_msg_handler, &secp_ctx, node4_privkey, NodeFeatures::from_le_bytes(id_to_feature_flags!(4)), 0);
802
803                 add_channel(&net_graph_msg_handler, &secp_ctx, node3_privkey, node6_privkey, ChannelFeatures::from_le_bytes(id_to_feature_flags!(7)), 7);
804
805                 update_channel(&net_graph_msg_handler, &secp_ctx, node3_privkey, UnsignedChannelUpdate {
806                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
807                         short_channel_id: 7,
808                         timestamp: 1,
809                         flags: 0,
810                         cltv_expiry_delta: (7 << 8) | 1,
811                         htlc_minimum_msat: 0,
812                         htlc_maximum_msat: OptionalField::Absent,
813                         fee_base_msat: 0,
814                         fee_proportional_millionths: 1000000,
815                         excess_data: Vec::new()
816                 });
817                 update_channel(&net_graph_msg_handler, &secp_ctx, node6_privkey, UnsignedChannelUpdate {
818                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
819                         short_channel_id: 7,
820                         timestamp: 1,
821                         flags: 1,
822                         cltv_expiry_delta: (7 << 8) | 2,
823                         htlc_minimum_msat: 0,
824                         htlc_maximum_msat: OptionalField::Absent,
825                         fee_base_msat: 0,
826                         fee_proportional_millionths: 0,
827                         excess_data: Vec::new()
828                 });
829
830                 add_or_update_node(&net_graph_msg_handler, &secp_ctx, node6_privkey, NodeFeatures::from_le_bytes(id_to_feature_flags!(6)), 0);
831
832                 // Simple route to 3 via 2
833                 let route = get_route(&our_id, &net_graph_msg_handler.network_graph.read().unwrap(), &node3, None, &Vec::new(), 100, 42, Arc::clone(&logger)).unwrap();
834                 assert_eq!(route.paths[0].len(), 2);
835
836                 assert_eq!(route.paths[0][0].pubkey, node2);
837                 assert_eq!(route.paths[0][0].short_channel_id, 2);
838                 assert_eq!(route.paths[0][0].fee_msat, 100);
839                 assert_eq!(route.paths[0][0].cltv_expiry_delta, (4 << 8) | 1);
840                 assert_eq!(route.paths[0][0].node_features.le_flags(), &id_to_feature_flags!(2));
841                 assert_eq!(route.paths[0][0].channel_features.le_flags(), &id_to_feature_flags!(2));
842
843                 assert_eq!(route.paths[0][1].pubkey, node3);
844                 assert_eq!(route.paths[0][1].short_channel_id, 4);
845                 assert_eq!(route.paths[0][1].fee_msat, 100);
846                 assert_eq!(route.paths[0][1].cltv_expiry_delta, 42);
847                 assert_eq!(route.paths[0][1].node_features.le_flags(), &id_to_feature_flags!(3));
848                 assert_eq!(route.paths[0][1].channel_features.le_flags(), &id_to_feature_flags!(4));
849
850
851                 // // Disable channels 4 and 12 by flags=2
852                 update_channel(&net_graph_msg_handler, &secp_ctx, node2_privkey, UnsignedChannelUpdate {
853                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
854                         short_channel_id: 4,
855                         timestamp: 2,
856                         flags: 2, // to disable
857                         cltv_expiry_delta: 0,
858                         htlc_minimum_msat: 0,
859                         htlc_maximum_msat: OptionalField::Absent,
860                         fee_base_msat: 0,
861                         fee_proportional_millionths: 0,
862                         excess_data: Vec::new()
863                 });
864                 update_channel(&net_graph_msg_handler, &secp_ctx, our_privkey, UnsignedChannelUpdate {
865                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
866                         short_channel_id: 12,
867                         timestamp: 2,
868                         flags: 2, // to disable
869                         cltv_expiry_delta: 0,
870                         htlc_minimum_msat: 0,
871                         htlc_maximum_msat: OptionalField::Absent,
872                         fee_base_msat: 0,
873                         fee_proportional_millionths: 0,
874                         excess_data: Vec::new()
875                 });
876
877                 // If all the channels require some features we don't understand, route should fail
878                 if let Err(LightningError{err, action: ErrorAction::IgnoreError}) = get_route(&our_id, &net_graph_msg_handler.network_graph.read().unwrap(), &node3, None, &Vec::new(), 100, 42, Arc::clone(&logger)) {
879                         assert_eq!(err, "Failed to find a path to the given destination");
880                 } else { panic!(); }
881
882                 // If we specify a channel to node8, that overrides our local channel view and that gets used
883                 let our_chans = vec![channelmanager::ChannelDetails {
884                         channel_id: [0; 32],
885                         short_channel_id: Some(42),
886                         remote_network_id: node8.clone(),
887                         counterparty_features: InitFeatures::from_le_bytes(vec![0b11]),
888                         channel_value_satoshis: 0,
889                         user_id: 0,
890                         outbound_capacity_msat: 0,
891                         inbound_capacity_msat: 0,
892                         is_live: true,
893                 }];
894                 let route = get_route(&our_id, &net_graph_msg_handler.network_graph.read().unwrap(), &node3, Some(&our_chans),  &Vec::new(), 100, 42, Arc::clone(&logger)).unwrap();
895                 assert_eq!(route.paths[0].len(), 2);
896
897                 assert_eq!(route.paths[0][0].pubkey, node8);
898                 assert_eq!(route.paths[0][0].short_channel_id, 42);
899                 assert_eq!(route.paths[0][0].fee_msat, 200);
900                 assert_eq!(route.paths[0][0].cltv_expiry_delta, (13 << 8) | 1);
901                 assert_eq!(route.paths[0][0].node_features.le_flags(), &vec![0b11]); // it should also override our view of their features
902                 assert_eq!(route.paths[0][0].channel_features.le_flags(), &Vec::<u8>::new()); // No feature flags will meet the relevant-to-channel conversion
903
904                 assert_eq!(route.paths[0][1].pubkey, node3);
905                 assert_eq!(route.paths[0][1].short_channel_id, 13);
906                 assert_eq!(route.paths[0][1].fee_msat, 100);
907                 assert_eq!(route.paths[0][1].cltv_expiry_delta, 42);
908                 assert_eq!(route.paths[0][1].node_features.le_flags(), &id_to_feature_flags!(3));
909                 assert_eq!(route.paths[0][1].channel_features.le_flags(), &id_to_feature_flags!(13));
910
911                 // Re-enable channels 4 and 12
912                 update_channel(&net_graph_msg_handler, &secp_ctx, node2_privkey, UnsignedChannelUpdate {
913                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
914                         short_channel_id: 4,
915                         timestamp: 3,
916                         flags: 0, // to enable
917                         cltv_expiry_delta: (4 << 8) | 1,
918                         htlc_minimum_msat: 0,
919                         htlc_maximum_msat: OptionalField::Absent,
920                         fee_base_msat: 0,
921                         fee_proportional_millionths: 1000000,
922                         excess_data: Vec::new()
923                 });
924                 update_channel(&net_graph_msg_handler, &secp_ctx, our_privkey, UnsignedChannelUpdate {
925                         chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
926                         short_channel_id: 12,
927                         timestamp: 3,
928                         flags: 0, // to enable
929                         cltv_expiry_delta: u16::max_value(),
930                         htlc_minimum_msat: 0,
931                         htlc_maximum_msat: OptionalField::Absent,
932                         fee_base_msat: u32::max_value(),
933                         fee_proportional_millionths: u32::max_value(),
934                         excess_data: Vec::new()
935                 });
936                 // Disable nodes 1, 2, and 8 by requiring unknown feature bits
937                 let mut unknown_features = NodeFeatures::known();
938                 unknown_features.set_required_unknown_bits();
939                 add_or_update_node(&net_graph_msg_handler, &secp_ctx, node1_privkey, unknown_features.clone(), 1);
940                 add_or_update_node(&net_graph_msg_handler, &secp_ctx, node2_privkey, unknown_features.clone(), 1);
941                 add_or_update_node(&net_graph_msg_handler, &secp_ctx, node8_privkey, unknown_features.clone(), 1);
942
943                 // // If all nodes require some features we don't understand, route should fail
944                 // if let Err(LightningError{err, action: ErrorAction::IgnoreError}) = get_route(&our_id, &net_graph_msg_handler, &node3, None, &Vec::new(), 100, 42, Arc::clone(&logger)) {
945                 //      assert_eq!(err, "Failed to find a path to the given destination");
946                 // } else { panic!(); }
947
948                 // If we specify a channel to node8, that overrides our local channel view and that gets used
949                 let our_chans = vec![channelmanager::ChannelDetails {
950                         channel_id: [0; 32],
951                         short_channel_id: Some(42),
952                         remote_network_id: node8.clone(),
953                         counterparty_features: InitFeatures::from_le_bytes(vec![0b11]),
954                         channel_value_satoshis: 0,
955                         user_id: 0,
956                         outbound_capacity_msat: 0,
957                         inbound_capacity_msat: 0,
958                         is_live: true,
959                 }];
960                 let route = get_route(&our_id, &net_graph_msg_handler.network_graph.read().unwrap(), &node3, Some(&our_chans), &Vec::new(), 100, 42, Arc::clone(&logger)).unwrap();
961                 assert_eq!(route.paths[0].len(), 2);
962
963                 assert_eq!(route.paths[0][0].pubkey, node8);
964                 assert_eq!(route.paths[0][0].short_channel_id, 42);
965                 assert_eq!(route.paths[0][0].fee_msat, 200);
966                 assert_eq!(route.paths[0][0].cltv_expiry_delta, (13 << 8) | 1);
967                 assert_eq!(route.paths[0][0].node_features.le_flags(), &vec![0b11]); // it should also override our view of their features
968                 assert_eq!(route.paths[0][0].channel_features.le_flags(), &Vec::<u8>::new()); // No feature flags will meet the relevant-to-channel conversion
969
970                 assert_eq!(route.paths[0][1].pubkey, node3);
971                 assert_eq!(route.paths[0][1].short_channel_id, 13);
972                 assert_eq!(route.paths[0][1].fee_msat, 100);
973                 assert_eq!(route.paths[0][1].cltv_expiry_delta, 42);
974                 assert_eq!(route.paths[0][1].node_features.le_flags(), &id_to_feature_flags!(3));
975                 assert_eq!(route.paths[0][1].channel_features.le_flags(), &id_to_feature_flags!(13));
976
977                 // Re-enable nodes 1, 2, and 8
978                 add_or_update_node(&net_graph_msg_handler, &secp_ctx, node1_privkey, NodeFeatures::from_le_bytes(id_to_feature_flags!(1)), 2);
979                 add_or_update_node(&net_graph_msg_handler, &secp_ctx, node2_privkey, NodeFeatures::from_le_bytes(id_to_feature_flags!(2)), 2);
980                 add_or_update_node(&net_graph_msg_handler, &secp_ctx, node8_privkey, NodeFeatures::from_le_bytes(id_to_feature_flags!(8)), 2);
981
982                 // Note that we don't test disabling node 3 and failing to route to it, as we (somewhat
983                 // naively) assume that the user checked the feature bits on the invoice, which override
984                 // the node_announcement.
985
986                 // Route to 1 via 2 and 3 because our channel to 1 is disabled
987                 let route = get_route(&our_id, &net_graph_msg_handler.network_graph.read().unwrap(), &node1, None, &Vec::new(), 100, 42, Arc::clone(&logger)).unwrap();
988                 assert_eq!(route.paths[0].len(), 3);
989
990                 assert_eq!(route.paths[0][0].pubkey, node2);
991                 assert_eq!(route.paths[0][0].short_channel_id, 2);
992                 assert_eq!(route.paths[0][0].fee_msat, 200);
993                 assert_eq!(route.paths[0][0].cltv_expiry_delta, (4 << 8) | 1);
994                 assert_eq!(route.paths[0][0].node_features.le_flags(), &id_to_feature_flags!(2));
995                 assert_eq!(route.paths[0][0].channel_features.le_flags(), &id_to_feature_flags!(2));
996
997                 assert_eq!(route.paths[0][1].pubkey, node3);
998                 assert_eq!(route.paths[0][1].short_channel_id, 4);
999                 assert_eq!(route.paths[0][1].fee_msat, 100);
1000                 assert_eq!(route.paths[0][1].cltv_expiry_delta, (3 << 8) | 2);
1001                 assert_eq!(route.paths[0][1].node_features.le_flags(), &id_to_feature_flags!(3));
1002                 assert_eq!(route.paths[0][1].channel_features.le_flags(), &id_to_feature_flags!(4));
1003
1004                 assert_eq!(route.paths[0][2].pubkey, node1);
1005                 assert_eq!(route.paths[0][2].short_channel_id, 3);
1006                 assert_eq!(route.paths[0][2].fee_msat, 100);
1007                 assert_eq!(route.paths[0][2].cltv_expiry_delta, 42);
1008                 assert_eq!(route.paths[0][2].node_features.le_flags(), &id_to_feature_flags!(1));
1009                 assert_eq!(route.paths[0][2].channel_features.le_flags(), &id_to_feature_flags!(3));
1010
1011                 // If we specify a channel to node8, that overrides our local channel view and that gets used
1012                 let our_chans = vec![channelmanager::ChannelDetails {
1013                         channel_id: [0; 32],
1014                         short_channel_id: Some(42),
1015                         remote_network_id: node8.clone(),
1016                         counterparty_features: InitFeatures::from_le_bytes(vec![0b11]),
1017                         channel_value_satoshis: 0,
1018                         user_id: 0,
1019                         outbound_capacity_msat: 0,
1020                         inbound_capacity_msat: 0,
1021                         is_live: true,
1022                 }];
1023                 let route = get_route(&our_id, &net_graph_msg_handler.network_graph.read().unwrap(), &node3, Some(&our_chans), &Vec::new(), 100, 42, Arc::clone(&logger)).unwrap();
1024                 assert_eq!(route.paths[0].len(), 2);
1025
1026                 assert_eq!(route.paths[0][0].pubkey, node8);
1027                 assert_eq!(route.paths[0][0].short_channel_id, 42);
1028                 assert_eq!(route.paths[0][0].fee_msat, 200);
1029                 assert_eq!(route.paths[0][0].cltv_expiry_delta, (13 << 8) | 1);
1030                 assert_eq!(route.paths[0][0].node_features.le_flags(), &vec![0b11]);
1031                 assert_eq!(route.paths[0][0].channel_features.le_flags(), &Vec::<u8>::new()); // No feature flags will meet the relevant-to-channel conversion
1032
1033                 assert_eq!(route.paths[0][1].pubkey, node3);
1034                 assert_eq!(route.paths[0][1].short_channel_id, 13);
1035                 assert_eq!(route.paths[0][1].fee_msat, 100);
1036                 assert_eq!(route.paths[0][1].cltv_expiry_delta, 42);
1037                 assert_eq!(route.paths[0][1].node_features.le_flags(), &id_to_feature_flags!(3));
1038                 assert_eq!(route.paths[0][1].channel_features.le_flags(), &id_to_feature_flags!(13));
1039
1040                 let zero_fees = RoutingFees {
1041                         base_msat: 0,
1042                         proportional_millionths: 0,
1043                 };
1044                 let mut last_hops = vec!(RouteHint {
1045                                 src_node_id: node4.clone(),
1046                                 short_channel_id: 8,
1047                                 fees: zero_fees,
1048                                 cltv_expiry_delta: (8 << 8) | 1,
1049                                 htlc_minimum_msat: 0,
1050                         }, RouteHint {
1051                                 src_node_id: node5.clone(),
1052                                 short_channel_id: 9,
1053                                 fees: RoutingFees {
1054                                         base_msat: 1001,
1055                                         proportional_millionths: 0,
1056                                 },
1057                                 cltv_expiry_delta: (9 << 8) | 1,
1058                                 htlc_minimum_msat: 0,
1059                         }, RouteHint {
1060                                 src_node_id: node6.clone(),
1061                                 short_channel_id: 10,
1062                                 fees: zero_fees,
1063                                 cltv_expiry_delta: (10 << 8) | 1,
1064                                 htlc_minimum_msat: 0,
1065                         });
1066
1067                 // Simple test across 2, 3, 5, and 4 via a last_hop channel
1068                 let route = get_route(&our_id, &net_graph_msg_handler.network_graph.read().unwrap(), &node7, None, &last_hops, 100, 42, Arc::clone(&logger)).unwrap();
1069                 assert_eq!(route.paths[0].len(), 5);
1070
1071                 assert_eq!(route.paths[0][0].pubkey, node2);
1072                 assert_eq!(route.paths[0][0].short_channel_id, 2);
1073                 assert_eq!(route.paths[0][0].fee_msat, 100);
1074                 assert_eq!(route.paths[0][0].cltv_expiry_delta, (4 << 8) | 1);
1075                 assert_eq!(route.paths[0][0].node_features.le_flags(), &id_to_feature_flags!(2));
1076                 assert_eq!(route.paths[0][0].channel_features.le_flags(), &id_to_feature_flags!(2));
1077
1078                 assert_eq!(route.paths[0][1].pubkey, node3);
1079                 assert_eq!(route.paths[0][1].short_channel_id, 4);
1080                 assert_eq!(route.paths[0][1].fee_msat, 0);
1081                 assert_eq!(route.paths[0][1].cltv_expiry_delta, (6 << 8) | 1);
1082                 assert_eq!(route.paths[0][1].node_features.le_flags(), &id_to_feature_flags!(3));
1083                 assert_eq!(route.paths[0][1].channel_features.le_flags(), &id_to_feature_flags!(4));
1084
1085                 assert_eq!(route.paths[0][2].pubkey, node5);
1086                 assert_eq!(route.paths[0][2].short_channel_id, 6);
1087                 assert_eq!(route.paths[0][2].fee_msat, 0);
1088                 assert_eq!(route.paths[0][2].cltv_expiry_delta, (11 << 8) | 1);
1089                 assert_eq!(route.paths[0][2].node_features.le_flags(), &id_to_feature_flags!(5));
1090                 assert_eq!(route.paths[0][2].channel_features.le_flags(), &id_to_feature_flags!(6));
1091
1092                 assert_eq!(route.paths[0][3].pubkey, node4);
1093                 assert_eq!(route.paths[0][3].short_channel_id, 11);
1094                 assert_eq!(route.paths[0][3].fee_msat, 0);
1095                 assert_eq!(route.paths[0][3].cltv_expiry_delta, (8 << 8) | 1);
1096                 // If we have a peer in the node map, we'll use their features here since we don't have
1097                 // a way of figuring out their features from the invoice:
1098                 assert_eq!(route.paths[0][3].node_features.le_flags(), &id_to_feature_flags!(4));
1099                 assert_eq!(route.paths[0][3].channel_features.le_flags(), &id_to_feature_flags!(11));
1100
1101                 assert_eq!(route.paths[0][4].pubkey, node7);
1102                 assert_eq!(route.paths[0][4].short_channel_id, 8);
1103                 assert_eq!(route.paths[0][4].fee_msat, 100);
1104                 assert_eq!(route.paths[0][4].cltv_expiry_delta, 42);
1105                 assert_eq!(route.paths[0][4].node_features.le_flags(), &Vec::<u8>::new()); // We dont pass flags in from invoices yet
1106                 assert_eq!(route.paths[0][4].channel_features.le_flags(), &Vec::<u8>::new()); // We can't learn any flags from invoices, sadly
1107
1108                 // Simple test with outbound channel to 4 to test that last_hops and first_hops connect
1109                 let our_chans = vec![channelmanager::ChannelDetails {
1110                         channel_id: [0; 32],
1111                         short_channel_id: Some(42),
1112                         remote_network_id: node4.clone(),
1113                         counterparty_features: InitFeatures::from_le_bytes(vec![0b11]),
1114                         channel_value_satoshis: 0,
1115                         user_id: 0,
1116                         outbound_capacity_msat: 0,
1117                         inbound_capacity_msat: 0,
1118                         is_live: true,
1119                 }];
1120                 let route = get_route(&our_id, &net_graph_msg_handler.network_graph.read().unwrap(), &node7, Some(&our_chans), &last_hops, 100, 42, Arc::clone(&logger)).unwrap();
1121                 assert_eq!(route.paths[0].len(), 2);
1122
1123                 assert_eq!(route.paths[0][0].pubkey, node4);
1124                 assert_eq!(route.paths[0][0].short_channel_id, 42);
1125                 assert_eq!(route.paths[0][0].fee_msat, 0);
1126                 assert_eq!(route.paths[0][0].cltv_expiry_delta, (8 << 8) | 1);
1127                 assert_eq!(route.paths[0][0].node_features.le_flags(), &vec![0b11]);
1128                 assert_eq!(route.paths[0][0].channel_features.le_flags(), &Vec::<u8>::new()); // No feature flags will meet the relevant-to-channel conversion
1129
1130                 assert_eq!(route.paths[0][1].pubkey, node7);
1131                 assert_eq!(route.paths[0][1].short_channel_id, 8);
1132                 assert_eq!(route.paths[0][1].fee_msat, 100);
1133                 assert_eq!(route.paths[0][1].cltv_expiry_delta, 42);
1134                 assert_eq!(route.paths[0][1].node_features.le_flags(), &Vec::<u8>::new()); // We dont pass flags in from invoices yet
1135                 assert_eq!(route.paths[0][1].channel_features.le_flags(), &Vec::<u8>::new()); // We can't learn any flags from invoices, sadly
1136
1137                 last_hops[0].fees.base_msat = 1000;
1138
1139                 // Revert to via 6 as the fee on 8 goes up
1140                 let route = get_route(&our_id, &net_graph_msg_handler.network_graph.read().unwrap(), &node7, None, &last_hops, 100, 42, Arc::clone(&logger)).unwrap();
1141                 assert_eq!(route.paths[0].len(), 4);
1142
1143                 assert_eq!(route.paths[0][0].pubkey, node2);
1144                 assert_eq!(route.paths[0][0].short_channel_id, 2);
1145                 assert_eq!(route.paths[0][0].fee_msat, 200); // fee increased as its % of value transferred across node
1146                 assert_eq!(route.paths[0][0].cltv_expiry_delta, (4 << 8) | 1);
1147                 assert_eq!(route.paths[0][0].node_features.le_flags(), &id_to_feature_flags!(2));
1148                 assert_eq!(route.paths[0][0].channel_features.le_flags(), &id_to_feature_flags!(2));
1149
1150                 assert_eq!(route.paths[0][1].pubkey, node3);
1151                 assert_eq!(route.paths[0][1].short_channel_id, 4);
1152                 assert_eq!(route.paths[0][1].fee_msat, 100);
1153                 assert_eq!(route.paths[0][1].cltv_expiry_delta, (7 << 8) | 1);
1154                 assert_eq!(route.paths[0][1].node_features.le_flags(), &id_to_feature_flags!(3));
1155                 assert_eq!(route.paths[0][1].channel_features.le_flags(), &id_to_feature_flags!(4));
1156
1157                 assert_eq!(route.paths[0][2].pubkey, node6);
1158                 assert_eq!(route.paths[0][2].short_channel_id, 7);
1159                 assert_eq!(route.paths[0][2].fee_msat, 0);
1160                 assert_eq!(route.paths[0][2].cltv_expiry_delta, (10 << 8) | 1);
1161                 // If we have a peer in the node map, we'll use their features here since we don't have
1162                 // a way of figuring out their features from the invoice:
1163                 assert_eq!(route.paths[0][2].node_features.le_flags(), &id_to_feature_flags!(6));
1164                 assert_eq!(route.paths[0][2].channel_features.le_flags(), &id_to_feature_flags!(7));
1165
1166                 assert_eq!(route.paths[0][3].pubkey, node7);
1167                 assert_eq!(route.paths[0][3].short_channel_id, 10);
1168                 assert_eq!(route.paths[0][3].fee_msat, 100);
1169                 assert_eq!(route.paths[0][3].cltv_expiry_delta, 42);
1170                 assert_eq!(route.paths[0][3].node_features.le_flags(), &Vec::<u8>::new()); // We dont pass flags in from invoices yet
1171                 assert_eq!(route.paths[0][3].channel_features.le_flags(), &Vec::<u8>::new()); // We can't learn any flags from invoices, sadly
1172
1173                 // ...but still use 8 for larger payments as 6 has a variable feerate
1174                 let route = get_route(&our_id, &net_graph_msg_handler.network_graph.read().unwrap(), &node7, None, &last_hops, 2000, 42, Arc::clone(&logger)).unwrap();
1175                 assert_eq!(route.paths[0].len(), 5);
1176
1177                 assert_eq!(route.paths[0][0].pubkey, node2);
1178                 assert_eq!(route.paths[0][0].short_channel_id, 2);
1179                 assert_eq!(route.paths[0][0].fee_msat, 3000);
1180                 assert_eq!(route.paths[0][0].cltv_expiry_delta, (4 << 8) | 1);
1181                 assert_eq!(route.paths[0][0].node_features.le_flags(), &id_to_feature_flags!(2));
1182                 assert_eq!(route.paths[0][0].channel_features.le_flags(), &id_to_feature_flags!(2));
1183
1184                 assert_eq!(route.paths[0][1].pubkey, node3);
1185                 assert_eq!(route.paths[0][1].short_channel_id, 4);
1186                 assert_eq!(route.paths[0][1].fee_msat, 0);
1187                 assert_eq!(route.paths[0][1].cltv_expiry_delta, (6 << 8) | 1);
1188                 assert_eq!(route.paths[0][1].node_features.le_flags(), &id_to_feature_flags!(3));
1189                 assert_eq!(route.paths[0][1].channel_features.le_flags(), &id_to_feature_flags!(4));
1190
1191                 assert_eq!(route.paths[0][2].pubkey, node5);
1192                 assert_eq!(route.paths[0][2].short_channel_id, 6);
1193                 assert_eq!(route.paths[0][2].fee_msat, 0);
1194                 assert_eq!(route.paths[0][2].cltv_expiry_delta, (11 << 8) | 1);
1195                 assert_eq!(route.paths[0][2].node_features.le_flags(), &id_to_feature_flags!(5));
1196                 assert_eq!(route.paths[0][2].channel_features.le_flags(), &id_to_feature_flags!(6));
1197
1198                 assert_eq!(route.paths[0][3].pubkey, node4);
1199                 assert_eq!(route.paths[0][3].short_channel_id, 11);
1200                 assert_eq!(route.paths[0][3].fee_msat, 1000);
1201                 assert_eq!(route.paths[0][3].cltv_expiry_delta, (8 << 8) | 1);
1202                 // If we have a peer in the node map, we'll use their features here since we don't have
1203                 // a way of figuring out their features from the invoice:
1204                 assert_eq!(route.paths[0][3].node_features.le_flags(), &id_to_feature_flags!(4));
1205                 assert_eq!(route.paths[0][3].channel_features.le_flags(), &id_to_feature_flags!(11));
1206
1207                 assert_eq!(route.paths[0][4].pubkey, node7);
1208                 assert_eq!(route.paths[0][4].short_channel_id, 8);
1209                 assert_eq!(route.paths[0][4].fee_msat, 2000);
1210                 assert_eq!(route.paths[0][4].cltv_expiry_delta, 42);
1211                 assert_eq!(route.paths[0][4].node_features.le_flags(), &Vec::<u8>::new()); // We dont pass flags in from invoices yet
1212                 assert_eq!(route.paths[0][4].channel_features.le_flags(), &Vec::<u8>::new()); // We can't learn any flags from invoices, sadly
1213         }
1214 }