Skip to content

Commit eacef3a

Browse files
committed
Split up generic parameters that used to comprise KeysInterface.
1 parent 5824e22 commit eacef3a

27 files changed

+543
-434
lines changed

fuzz/src/chanmon_consistency.rs

+7-5
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ impl chain::Watch<EnforcingSigner> for TestChainMonitor {
159159
hash_map::Entry::Vacant(_) => panic!("Didn't have monitor on update call"),
160160
};
161161
let deserialized_monitor = <(BlockHash, channelmonitor::ChannelMonitor<EnforcingSigner>)>::
162-
read(&mut Cursor::new(&map_entry.get().1), &*self.keys).unwrap().1;
162+
read(&mut Cursor::new(&map_entry.get().1), (&*self.keys, &*self.keys)).unwrap().1;
163163
deserialized_monitor.update_monitor(&update, &&TestBroadcaster{}, &FuzzEstimator { ret_val: atomic::AtomicU32::new(253) }, &self.logger).unwrap();
164164
let mut ser = VecWriter(Vec::new());
165165
deserialized_monitor.write(&mut ser).unwrap();
@@ -326,7 +326,7 @@ fn check_payment_err(send_err: PaymentSendFailure) {
326326
}
327327
}
328328

329-
type ChanMan<'a> = ChannelManager<Arc<TestChainMonitor>, Arc<TestBroadcaster>, Arc<KeyProvider>, Arc<FuzzEstimator>, &'a FuzzRouter, Arc<dyn Logger>>;
329+
type ChanMan<'a> = ChannelManager<Arc<TestChainMonitor>, Arc<TestBroadcaster>, Arc<KeyProvider>, Arc<KeyProvider>, Arc<KeyProvider>, Arc<FuzzEstimator>, &'a FuzzRouter, Arc<dyn Logger>>;
330330

331331
#[inline]
332332
fn get_payment_secret_hash(dest: &ChanMan, payment_id: &mut u8) -> Option<(PaymentSecret, PaymentHash)> {
@@ -416,7 +416,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out) {
416416
network,
417417
best_block: BestBlock::from_genesis(network),
418418
};
419-
(ChannelManager::new($fee_estimator.clone(), monitor.clone(), broadcast.clone(), &router, Arc::clone(&logger), keys_manager.clone(), config, params),
419+
(ChannelManager::new($fee_estimator.clone(), monitor.clone(), broadcast.clone(), &router, Arc::clone(&logger), keys_manager.clone(), keys_manager.clone(), keys_manager.clone(), config, params),
420420
monitor, keys_manager)
421421
} }
422422
}
@@ -437,7 +437,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out) {
437437
let mut monitors = HashMap::new();
438438
let mut old_monitors = $old_monitors.latest_monitors.lock().unwrap();
439439
for (outpoint, (update_id, monitor_ser)) in old_monitors.drain() {
440-
monitors.insert(outpoint, <(BlockHash, ChannelMonitor<EnforcingSigner>)>::read(&mut Cursor::new(&monitor_ser), &*$keys_manager).expect("Failed to read monitor").1);
440+
monitors.insert(outpoint, <(BlockHash, ChannelMonitor<EnforcingSigner>)>::read(&mut Cursor::new(&monitor_ser), (&*$keys_manager, &*$keys_manager)).expect("Failed to read monitor").1);
441441
chain_monitor.latest_monitors.lock().unwrap().insert(outpoint, (update_id, monitor_ser));
442442
}
443443
let mut monitor_refs = HashMap::new();
@@ -446,7 +446,9 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out) {
446446
}
447447

448448
let read_args = ChannelManagerReadArgs {
449-
keys_manager,
449+
entropy_source: keys_manager.clone(),
450+
node_signer: keys_manager.clone(),
451+
signer_provider: keys_manager.clone(),
450452
fee_estimator: $fee_estimator.clone(),
451453
chain_monitor: chain_monitor.clone(),
452454
tx_broadcaster: broadcast.clone(),

fuzz/src/chanmon_deser.rs

+2-2
Original file line numberDiff line numberDiff line change
@@ -22,10 +22,10 @@ impl Writer for VecWriter {
2222

2323
#[inline]
2424
pub fn do_test<Out: test_logger::Output>(data: &[u8], _out: Out) {
25-
if let Ok((latest_block_hash, monitor)) = <(BlockHash, channelmonitor::ChannelMonitor<EnforcingSigner>)>::read(&mut Cursor::new(data), &OnlyReadsKeysInterface {}) {
25+
if let Ok((latest_block_hash, monitor)) = <(BlockHash, channelmonitor::ChannelMonitor<EnforcingSigner>)>::read(&mut Cursor::new(data), (&OnlyReadsKeysInterface {}, &OnlyReadsKeysInterface {})) {
2626
let mut w = VecWriter(Vec::new());
2727
monitor.write(&mut w).unwrap();
28-
let deserialized_copy = <(BlockHash, channelmonitor::ChannelMonitor<EnforcingSigner>)>::read(&mut Cursor::new(&w.0), &OnlyReadsKeysInterface {}).unwrap();
28+
let deserialized_copy = <(BlockHash, channelmonitor::ChannelMonitor<EnforcingSigner>)>::read(&mut Cursor::new(&w.0), (&OnlyReadsKeysInterface {}, &OnlyReadsKeysInterface {})).unwrap();
2929
assert!(latest_block_hash == deserialized_copy.0);
3030
assert!(monitor == deserialized_copy.1);
3131
}

fuzz/src/full_stack.rs

+2-2
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ impl<'a> std::hash::Hash for Peer<'a> {
182182

183183
type ChannelMan<'a> = ChannelManager<
184184
Arc<chainmonitor::ChainMonitor<EnforcingSigner, Arc<dyn chain::Filter>, Arc<TestBroadcaster>, Arc<FuzzEstimator>, Arc<dyn Logger>, Arc<TestPersister>>>,
185-
Arc<TestBroadcaster>, Arc<KeyProvider>, Arc<FuzzEstimator>, &'a FuzzRouter, Arc<dyn Logger>>;
185+
Arc<TestBroadcaster>, Arc<KeyProvider>, Arc<KeyProvider>, Arc<KeyProvider>, Arc<FuzzEstimator>, &'a FuzzRouter, Arc<dyn Logger>>;
186186
type PeerMan<'a> = PeerManager<Peer<'a>, Arc<ChannelMan<'a>>, Arc<P2PGossipSync<Arc<NetworkGraph<Arc<dyn Logger>>>, Arc<dyn chain::Access>, Arc<dyn Logger>>>, IgnoringMessageHandler, Arc<dyn Logger>, IgnoringMessageHandler>;
187187

188188
struct MoneyLossDetector<'a> {
@@ -441,7 +441,7 @@ pub fn do_test(data: &[u8], logger: &Arc<dyn Logger>) {
441441
network,
442442
best_block: BestBlock::from_genesis(network),
443443
};
444-
let channelmanager = Arc::new(ChannelManager::new(fee_est.clone(), monitor.clone(), broadcast.clone(), &router, Arc::clone(&logger), keys_manager.clone(), config, params));
444+
let channelmanager = Arc::new(ChannelManager::new(fee_est.clone(), monitor.clone(), broadcast.clone(), &router, Arc::clone(&logger), keys_manager.clone(), keys_manager.clone(), keys_manager.clone(), config, params));
445445
// Adding new calls to `EntropySource::get_secure_random_bytes` during startup can change all the
446446
// keys subsequently generated in this test. Rather than regenerating all the messages manually,
447447
// it's easier to just increment the counter here so the keys don't change.

fuzz/src/onion_message.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ pub fn do_test<L: Logger>(data: &[u8], logger: &L) {
3030
counter: AtomicU64::new(0),
3131
};
3232
let custom_msg_handler = TestCustomMessageHandler {};
33-
let onion_messenger = OnionMessenger::new(&keys_manager, logger, &custom_msg_handler);
33+
let onion_messenger = OnionMessenger::new(&keys_manager, &keys_manager, logger, &custom_msg_handler);
3434
let mut pk = [2; 33]; pk[1] = 0xff;
3535
let peer_node_id_not_used = PublicKey::from_slice(&pk).unwrap();
3636
onion_messenger.handle_onion_message(&peer_node_id_not_used, &msg);

lightning-background-processor/src/lib.rs

+23-15
Original file line numberDiff line numberDiff line change
@@ -341,7 +341,9 @@ pub async fn process_events_async<
341341
CF: 'static + Deref + Send + Sync,
342342
CW: 'static + Deref + Send + Sync,
343343
T: 'static + Deref + Send + Sync,
344-
K: 'static + Deref + Send + Sync,
344+
ES: 'static + Deref + Send + Sync,
345+
NS: 'static + Deref + Send + Sync,
346+
SP: 'static + Deref + Send + Sync,
345347
F: 'static + Deref + Send + Sync,
346348
R: 'static + Deref + Send + Sync,
347349
G: 'static + Deref<Target = NetworkGraph<L>> + Send + Sync,
@@ -354,8 +356,8 @@ pub async fn process_events_async<
354356
EventHandlerFuture: core::future::Future<Output = ()>,
355357
EventHandler: Fn(Event) -> EventHandlerFuture,
356358
PS: 'static + Deref + Send,
357-
M: 'static + Deref<Target = ChainMonitor<<K::Target as SignerProvider>::Signer, CF, T, F, L, P>> + Send + Sync,
358-
CM: 'static + Deref<Target = ChannelManager<CW, T, K, F, R, L>> + Send + Sync,
359+
M: 'static + Deref<Target = ChainMonitor<<SP::Target as SignerProvider>::Signer, CF, T, F, L, P>> + Send + Sync,
360+
CM: 'static + Deref<Target = ChannelManager<CW, T, ES, NS, SP, F, R, L>> + Send + Sync,
359361
PGS: 'static + Deref<Target = P2PGossipSync<G, CA, L>> + Send + Sync,
360362
RGS: 'static + Deref<Target = RapidGossipSync<G, L>> + Send,
361363
UMH: 'static + Deref + Send + Sync,
@@ -372,18 +374,20 @@ pub async fn process_events_async<
372374
where
373375
CA::Target: 'static + chain::Access,
374376
CF::Target: 'static + chain::Filter,
375-
CW::Target: 'static + chain::Watch<<K::Target as SignerProvider>::Signer>,
377+
CW::Target: 'static + chain::Watch<<SP::Target as SignerProvider>::Signer>,
376378
T::Target: 'static + BroadcasterInterface,
377-
K::Target: 'static + EntropySource + NodeSigner + SignerProvider,
379+
ES::Target: 'static + EntropySource,
380+
NS::Target: 'static + NodeSigner,
381+
SP::Target: 'static + SignerProvider,
378382
F::Target: 'static + FeeEstimator,
379383
R::Target: 'static + Router,
380384
L::Target: 'static + Logger,
381-
P::Target: 'static + Persist<<K::Target as SignerProvider>::Signer>,
385+
P::Target: 'static + Persist<<SP::Target as SignerProvider>::Signer>,
382386
CMH::Target: 'static + ChannelMessageHandler,
383387
OMH::Target: 'static + OnionMessageHandler,
384388
RMH::Target: 'static + RoutingMessageHandler,
385389
UMH::Target: 'static + CustomMessageHandler,
386-
PS::Target: 'static + Persister<'a, CW, T, K, F, R, L, SC>,
390+
PS::Target: 'static + Persister<'a, CW, T, ES, NS, SP, F, R, L, SC>,
387391
{
388392
let mut should_break = true;
389393
let async_event_handler = |event| {
@@ -461,7 +465,9 @@ impl BackgroundProcessor {
461465
CF: 'static + Deref + Send + Sync,
462466
CW: 'static + Deref + Send + Sync,
463467
T: 'static + Deref + Send + Sync,
464-
K: 'static + Deref + Send + Sync,
468+
ES: 'static + Deref + Send + Sync,
469+
NS: 'static + Deref + Send + Sync,
470+
SP: 'static + Deref + Send + Sync,
465471
F: 'static + Deref + Send + Sync,
466472
R: 'static + Deref + Send + Sync,
467473
G: 'static + Deref<Target = NetworkGraph<L>> + Send + Sync,
@@ -473,8 +479,8 @@ impl BackgroundProcessor {
473479
RMH: 'static + Deref + Send + Sync,
474480
EH: 'static + EventHandler + Send,
475481
PS: 'static + Deref + Send,
476-
M: 'static + Deref<Target = ChainMonitor<<K::Target as SignerProvider>::Signer, CF, T, F, L, P>> + Send + Sync,
477-
CM: 'static + Deref<Target = ChannelManager<CW, T, K, F, R, L>> + Send + Sync,
482+
M: 'static + Deref<Target = ChainMonitor<<SP::Target as SignerProvider>::Signer, CF, T, F, L, P>> + Send + Sync,
483+
CM: 'static + Deref<Target = ChannelManager<CW, T, ES, NS, SP, F, R, L>> + Send + Sync,
478484
PGS: 'static + Deref<Target = P2PGossipSync<G, CA, L>> + Send + Sync,
479485
RGS: 'static + Deref<Target = RapidGossipSync<G, L>> + Send,
480486
UMH: 'static + Deref + Send + Sync,
@@ -488,18 +494,20 @@ impl BackgroundProcessor {
488494
where
489495
CA::Target: 'static + chain::Access,
490496
CF::Target: 'static + chain::Filter,
491-
CW::Target: 'static + chain::Watch<<K::Target as SignerProvider>::Signer>,
497+
CW::Target: 'static + chain::Watch<<SP::Target as SignerProvider>::Signer>,
492498
T::Target: 'static + BroadcasterInterface,
493-
K::Target: 'static + EntropySource + NodeSigner + SignerProvider,
499+
ES::Target: 'static + EntropySource,
500+
NS::Target: 'static + NodeSigner,
501+
SP::Target: 'static + SignerProvider,
494502
F::Target: 'static + FeeEstimator,
495503
R::Target: 'static + Router,
496504
L::Target: 'static + Logger,
497-
P::Target: 'static + Persist<<K::Target as SignerProvider>::Signer>,
505+
P::Target: 'static + Persist<<SP::Target as SignerProvider>::Signer>,
498506
CMH::Target: 'static + ChannelMessageHandler,
499507
OMH::Target: 'static + OnionMessageHandler,
500508
RMH::Target: 'static + RoutingMessageHandler,
501509
UMH::Target: 'static + CustomMessageHandler,
502-
PS::Target: 'static + Persister<'a, CW, T, K, F, R, L, SC>,
510+
PS::Target: 'static + Persister<'a, CW, T, ES, NS, SP, F, R, L, SC>,
503511
{
504512
let stop_thread = Arc::new(AtomicBool::new(false));
505513
let stop_thread_clone = stop_thread.clone();
@@ -742,7 +750,7 @@ mod tests {
742750
let chain_monitor = Arc::new(chainmonitor::ChainMonitor::new(Some(chain_source.clone()), tx_broadcaster.clone(), logger.clone(), fee_estimator.clone(), persister.clone()));
743751
let best_block = BestBlock::from_genesis(network);
744752
let params = ChainParameters { network, best_block };
745-
let manager = Arc::new(ChannelManager::new(fee_estimator.clone(), chain_monitor.clone(), tx_broadcaster.clone(), router.clone(), logger.clone(), keys_manager.clone(), UserConfig::default(), params));
753+
let manager = Arc::new(ChannelManager::new(fee_estimator.clone(), chain_monitor.clone(), tx_broadcaster.clone(), router.clone(), logger.clone(), keys_manager.clone(), keys_manager.clone(), keys_manager.clone(), UserConfig::default(), params));
746754
let p2p_gossip_sync = Arc::new(P2PGossipSync::new(network_graph.clone(), Some(chain_source.clone()), logger.clone()));
747755
let rapid_gossip_sync = Arc::new(RapidGossipSync::new(network_graph.clone()));
748756
let msg_handler = MessageHandler { chan_handler: Arc::new(test_utils::TestChannelMessageHandler::new()), route_handler: Arc::new(test_utils::TestRoutingMessageHandler::new()), onion_message_handler: IgnoringMessageHandler{}};

lightning-block-sync/src/init.rs

+14-8
Original file line numberDiff line numberDiff line change
@@ -61,18 +61,22 @@ BlockSourceResult<ValidatedBlockHeader> where B::Target: BlockSource {
6161
///
6262
/// async fn init_sync<
6363
/// B: BlockSource,
64-
/// K: EntropySource + NodeSigner + SignerProvider,
64+
/// ES: EntropySource,
65+
/// NS: NodeSigner,
66+
/// SP: SignerProvider,
6567
/// T: BroadcasterInterface,
6668
/// F: FeeEstimator,
6769
/// R: Router,
6870
/// L: Logger,
6971
/// C: chain::Filter,
70-
/// P: chainmonitor::Persist<K::Signer>,
72+
/// P: chainmonitor::Persist<SP::Signer>,
7173
/// >(
7274
/// block_source: &B,
73-
/// chain_monitor: &ChainMonitor<K::Signer, &C, &T, &F, &L, &P>,
75+
/// chain_monitor: &ChainMonitor<SP::Signer, &C, &T, &F, &L, &P>,
7476
/// config: UserConfig,
75-
/// keys_manager: &K,
77+
/// entropy_source: &ES,
78+
/// node_signer: &NS,
79+
/// signer_provider: &SP,
7680
/// tx_broadcaster: &T,
7781
/// fee_estimator: &F,
7882
/// router: &R,
@@ -81,14 +85,16 @@ BlockSourceResult<ValidatedBlockHeader> where B::Target: BlockSource {
8185
/// ) {
8286
/// // Read a serialized channel monitor paired with the block hash when it was persisted.
8387
/// let serialized_monitor = "...";
84-
/// let (monitor_block_hash, mut monitor) = <(BlockHash, ChannelMonitor<K::Signer>)>::read(
85-
/// &mut Cursor::new(&serialized_monitor), keys_manager).unwrap();
88+
/// let (monitor_block_hash, mut monitor) = <(BlockHash, ChannelMonitor<SP::Signer>)>::read(
89+
/// &mut Cursor::new(&serialized_monitor), (entropy_source, signer_provider)).unwrap();
8690
///
8791
/// // Read the channel manager paired with the block hash when it was persisted.
8892
/// let serialized_manager = "...";
8993
/// let (manager_block_hash, mut manager) = {
9094
/// let read_args = ChannelManagerReadArgs::new(
91-
/// keys_manager,
95+
/// entropy_source,
96+
/// node_signer,
97+
/// signer_provider,
9298
/// fee_estimator,
9399
/// chain_monitor,
94100
/// tx_broadcaster,
@@ -97,7 +103,7 @@ BlockSourceResult<ValidatedBlockHeader> where B::Target: BlockSource {
97103
/// config,
98104
/// vec![&mut monitor],
99105
/// );
100-
/// <(BlockHash, ChannelManager<&ChainMonitor<K::Signer, &C, &T, &F, &L, &P>, &T, &K, &F, &R, &L>)>::read(
106+
/// <(BlockHash, ChannelManager<&ChainMonitor<SP::Signer, &C, &T, &F, &L, &P>, &T, &ES, &NS, &SP, &F, &R, &L>)>::read(
101107
/// &mut Cursor::new(&serialized_manager), read_args).unwrap()
102108
/// };
103109
///

0 commit comments

Comments
 (0)