Skip to content

Commit 9ad6525

Browse files
committed
Clean up allocations in internal_peer_storage_retrieval
Clarifies some log messages and removes some unecessary allocations in both happy and error paths.
1 parent cd31737 commit 9ad6525

File tree

1 file changed

+11
-16
lines changed

1 file changed

+11
-16
lines changed

lightning/src/ln/channelmanager.rs

Lines changed: 11 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -8547,36 +8547,31 @@ This indicates a bug inside LDK. Please report this error at https://github.com/
85478547
}
85488548

85498549
#[rustfmt::skip]
8550-
fn internal_peer_storage_retrieval(&self, counterparty_node_id: PublicKey, msg: msgs::PeerStorageRetrieval) -> Result<(), MsgHandleErrInternal> {
8550+
fn internal_peer_storage_retrieval(&self, peer_node_id: PublicKey, msg: msgs::PeerStorageRetrieval) -> Result<(), MsgHandleErrInternal> {
85518551
// TODO: Check if have any stale or missing ChannelMonitor.
8552-
let logger = WithContext::from(&self.logger, Some(counterparty_node_id), None, None);
8553-
let err = MsgHandleErrInternal::from_chan_no_close(
8552+
let logger = WithContext::from(&self.logger, Some(peer_node_id), None, None);
8553+
let err = || MsgHandleErrInternal::from_chan_no_close(
85548554
ChannelError::Ignore("Invalid PeerStorageRetrieval message received.".into()),
85558555
ChannelId([0; 32]),
85568556
);
8557-
let err_str = || {
8558-
format!("Invalid PeerStorage received from {}", counterparty_node_id)
8559-
};
85608557

85618558
let encrypted_ops = match EncryptedOurPeerStorage::new(msg.data) {
85628559
Ok(encrypted_ops) => encrypted_ops,
8563-
Err(_) => {
8564-
log_debug!(logger, "{}", err_str());
8565-
return Err(err);
8560+
Err(()) => {
8561+
log_debug!(logger, "Received a peer backup which wasn't long enough to be valid");
8562+
return Err(err());
85668563
}
85678564
};
85688565

8569-
let decrypted_data = match encrypted_ops.decrypt(&self.node_signer.get_peer_storage_key()) {
8566+
let decrypted = match encrypted_ops.decrypt(&self.node_signer.get_peer_storage_key()) {
85708567
Ok(decrypted_ops) => decrypted_ops.into_vec(),
8571-
Err(_) => {
8572-
log_debug!(logger, "{}", err_str());
8573-
return Err(err);
8568+
Err(()) => {
8569+
log_debug!(logger, "Received a peer backup which was corrupted");
8570+
return Err(err());
85748571
}
85758572
};
85768573

8577-
if decrypted_data.is_empty() {
8578-
log_debug!(logger, "Received a peer storage from peer {} with 0 channels.", log_pubkey!(counterparty_node_id));
8579-
}
8574+
log_trace!(logger, "Got valid {}-byte peer backup from {}", decrypted.len(), peer_node_id);
85808575

85818576
Ok(())
85828577
}

0 commit comments

Comments
 (0)