X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=fuzz%2Fsrc%2Frouter.rs;h=b0f052dbc646f613a1d8c038c88c96c3604ef88e;hb=df1c4ee1506061adf285c09638d401ec6e8397c3;hp=e80e080f31786bc76b40b10bc9b6fb96d27c8e2a;hpb=f40e47c1ef6957ab548345162ffbd540bc7fe0a2;p=rust-lightning diff --git a/fuzz/src/router.rs b/fuzz/src/router.rs index e80e080f..b0f052db 100644 --- a/fuzz/src/router.rs +++ b/fuzz/src/router.rs @@ -12,14 +12,17 @@ use bitcoin::blockdata::transaction::TxOut; use bitcoin::hash_types::BlockHash; use lightning::chain; -use lightning::ln::channelmanager::ChannelDetails; +use lightning::chain::transaction::OutPoint; +use lightning::ln::channelmanager::{ChannelDetails, ChannelCounterparty}; use lightning::ln::features::InitFeatures; use lightning::ln::msgs; -use lightning::routing::router::{get_route, RouteHintHop}; +use lightning::routing::router::{find_route, PaymentParameters, RouteHint, RouteHintHop, RouteParameters}; +use lightning::routing::scoring::FixedPenaltyScorer; use lightning::util::logger::Logger; use lightning::util::ser::Readable; use lightning::routing::network_graph::{NetworkGraph, RoutingFees}; +use bitcoin::hashes::Hash; use bitcoin::secp256k1::key::PublicKey; use bitcoin::network::constants::Network; use bitcoin::blockdata::constants::genesis_block; @@ -131,6 +134,7 @@ pub fn do_test(data: &[u8], out: Out) { msgs::DecodeError::BadLengthDescriptor => return, msgs::DecodeError::ShortRead => panic!("We picked the length..."), msgs::DecodeError::Io(e) => panic!("{:?}", e), + msgs::DecodeError::UnsupportedCompression => return, } } }} @@ -157,7 +161,7 @@ pub fn do_test(data: &[u8], out: Out) { let logger: Arc = Arc::new(test_logger::TestLogger::new("".to_owned(), out)); let our_pubkey = get_pubkey!(); - let mut net_graph = NetworkGraph::new(genesis_block(Network::Bitcoin).header.block_hash()); + let net_graph = NetworkGraph::new(genesis_block(Network::Bitcoin).header.block_hash()); let mut node_pks = HashSet::new(); let mut scid = 42; @@ -204,27 +208,41 @@ pub fn do_test(data: &[u8], out: Out) { let rnid = node_pks.iter().skip(slice_to_be16(get_slice!(2))as usize % node_pks.len()).next().unwrap(); first_hops_vec.push(ChannelDetails { channel_id: [0; 32], + counterparty: ChannelCounterparty { + node_id: *rnid, + features: InitFeatures::known(), + unspendable_punishment_reserve: 0, + forwarding_info: None, + outbound_htlc_minimum_msat: None, + outbound_htlc_maximum_msat: None, + }, + funding_txo: Some(OutPoint { txid: bitcoin::Txid::from_slice(&[0; 32]).unwrap(), index: 0 }), + channel_type: None, short_channel_id: Some(scid), - remote_network_id: *rnid, - counterparty_features: InitFeatures::known(), + inbound_scid_alias: None, channel_value_satoshis: slice_to_be64(get_slice!(8)), - user_id: 0, - inbound_capacity_msat: 0, - is_live: true, + user_channel_id: 0, inbound_capacity_msat: 0, + unspendable_punishment_reserve: None, + confirmations_required: None, + force_close_spend_delay: None, + is_outbound: true, is_funding_locked: true, + is_usable: true, is_public: true, + balance_msat: 0, outbound_capacity_msat: 0, - counterparty_forwarding_info: None, + inbound_htlc_minimum_msat: None, + inbound_htlc_maximum_msat: None, }); } Some(&first_hops_vec[..]) }, }; - let mut last_hops_vec = Vec::new(); + let mut last_hops = Vec::new(); { let count = get_slice!(1)[0]; for _ in 0..count { scid += 1; let rnid = node_pks.iter().skip(slice_to_be16(get_slice!(2))as usize % node_pks.len()).next().unwrap(); - last_hops_vec.push(RouteHintHop { + last_hops.push(RouteHint(vec![RouteHintHop { src_node_id: *rnid, short_channel_id: scid, fees: RoutingFees { @@ -234,15 +252,20 @@ pub fn do_test(data: &[u8], out: Out) { cltv_expiry_delta: slice_to_be16(get_slice!(2)), htlc_minimum_msat: Some(slice_to_be64(get_slice!(8))), htlc_maximum_msat: None, - }); + }])); } } - let last_hops = &last_hops_vec[..]; + let scorer = FixedPenaltyScorer::with_penalty(0); + let random_seed_bytes: [u8; 32] = [get_slice!(1)[0]; 32]; for target in node_pks.iter() { - let _ = get_route(&our_pubkey, &net_graph, target, None, + let route_params = RouteParameters { + payment_params: PaymentParameters::from_node_id(*target).with_route_hints(last_hops.clone()), + final_value_msat: slice_to_be64(get_slice!(8)), + final_cltv_expiry_delta: slice_to_be32(get_slice!(4)), + }; + let _ = find_route(&our_pubkey, &route_params, &net_graph, first_hops.map(|c| c.iter().collect::>()).as_ref().map(|a| a.as_slice()), - &last_hops.iter().collect::>(), - slice_to_be64(get_slice!(8)), slice_to_be32(get_slice!(4)), Arc::clone(&logger)); + Arc::clone(&logger), &scorer, &random_seed_bytes); } }, }