Skip to content

Commit 9c6a377

Browse files
committed
Clean up TestKeysInterface random bytes override interface
Its very confusing to have multiple fields that do the same thing, one of which isn't even used for its stated purpose anymore after the previous few commits.
1 parent 46211c4 commit 9c6a377

File tree

3 files changed

+7
-16
lines changed

3 files changed

+7
-16
lines changed

lightning/src/ln/functional_tests.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6001,7 +6001,7 @@ fn bolt2_open_channel_sending_node_checks_part1() { //This test needs to be on i
60016001
let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
60026002
//Force duplicate channel ids
60036003
for node in nodes.iter() {
6004-
*node.keys_manager.override_channel_id_priv.lock().unwrap() = Some([0; 32]);
6004+
*node.keys_manager.override_random_bytes.lock().unwrap() = Some([0; 32]);
60056005
}
60066006

60076007
// BOLT #2 spec: Sending node must ensure temporary_channel_id is unique from any other channel ID with the same peer.

lightning/src/ln/onion_route_tests.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,7 @@ fn test_onion_failure() {
317317
let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs, &[Some(config), Some(config), Some(node_2_cfg)]);
318318
let mut nodes = create_network(3, &node_cfgs, &node_chanmgrs);
319319
for node in nodes.iter() {
320-
*node.keys_manager.override_session_priv.lock().unwrap() = Some([3; 32]);
320+
*node.keys_manager.override_random_bytes.lock().unwrap() = Some([3; 32]);
321321
}
322322
let channels = [create_announced_chan_between_nodes(&nodes, 0, 1, InitFeatures::known(), InitFeatures::known()), create_announced_chan_between_nodes(&nodes, 1, 2, InitFeatures::known(), InitFeatures::known())];
323323
let (route, payment_hash, _, payment_secret) = get_route_and_payment_hash!(nodes[0], nodes[2], 40000);

lightning/src/util/test_utils.rs

Lines changed: 5 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -471,8 +471,7 @@ impl Logger for TestLogger {
471471

472472
pub struct TestKeysInterface {
473473
pub backing: keysinterface::PhantomKeysManager,
474-
pub override_session_priv: Mutex<Option<[u8; 32]>>,
475-
pub override_channel_id_priv: Mutex<Option<[u8; 32]>>,
474+
pub override_random_bytes: Mutex<Option<[u8; 32]>>,
476475
pub disable_revocation_policy_check: bool,
477476
enforcement_states: Mutex<HashMap<[u8;32], Arc<Mutex<EnforcementState>>>>,
478477
expectations: Mutex<Option<VecDeque<OnGetShutdownScriptpubkey>>>,
@@ -506,16 +505,9 @@ impl keysinterface::KeysInterface for TestKeysInterface {
506505
}
507506

508507
fn get_secure_random_bytes(&self) -> [u8; 32] {
509-
let override_channel_id = self.override_channel_id_priv.lock().unwrap();
510-
let override_session_key = self.override_session_priv.lock().unwrap();
511-
if override_channel_id.is_some() && override_session_key.is_some() {
512-
panic!("We don't know which override key to use!");
513-
}
514-
if let Some(key) = &*override_channel_id {
515-
return *key;
516-
}
517-
if let Some(key) = &*override_session_key {
518-
return *key;
508+
let override_random_bytes = self.override_random_bytes.lock().unwrap();
509+
if let Some(bytes) = &*override_random_bytes {
510+
return *bytes;
519511
}
520512
self.backing.get_secure_random_bytes()
521513
}
@@ -543,8 +535,7 @@ impl TestKeysInterface {
543535
let now = Duration::from_secs(genesis_block(network).header.time as u64);
544536
Self {
545537
backing: keysinterface::PhantomKeysManager::new(seed, now.as_secs(), now.subsec_nanos(), seed),
546-
override_session_priv: Mutex::new(None),
547-
override_channel_id_priv: Mutex::new(None),
538+
override_random_bytes: Mutex::new(None),
548539
disable_revocation_policy_check: false,
549540
enforcement_states: Mutex::new(HashMap::new()),
550541
expectations: Mutex::new(None),

0 commit comments

Comments
 (0)