Skip to content

Commit 0597864

Browse files
committed
fixup: Add OurPeerStorage for serialized Peer Storage backups
1 parent d80b377 commit 0597864

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

lightning/src/ln/our_peer_storage.rs

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
//! It supports encryption and decryption to maintain data integrity and security during
1212
//! transmission.
1313
//!
14-
use crate::chain::chainmonitor::PeerStorageKey;
14+
use crate::sign::PeerStorageKey;
1515

1616
use crate::crypto::chacha20poly1305rfc::ChaCha20Poly1305RFC;
1717
use crate::prelude::*;
@@ -35,11 +35,11 @@ use crate::prelude::*;
3535
/// ## Example
3636
/// ```
3737
/// use lightning::ln::our_peer_storage::OurPeerStorage;
38-
/// use lightning::chain::chainmonitor::PeerStorageKey;
39-
/// let key = PeerStorageKey::new([0u8; 32]);
40-
/// let our_peer_storage = OurPeerStorage::create_from_data(key.clone(), vec![1,2,3]);
38+
/// use lightning::sign::PeerStorageKey;
39+
/// let key = PeerStorageKey{inner: [0u8; 32]};
40+
/// let our_peer_storage = OurPeerStorage::create_from_data(key.clone(), vec![1, 2, 3]);
4141
/// let decrypted_data = our_peer_storage.decrypt_our_peer_storage(key).unwrap();
42-
/// assert_eq!(decrypted_data, vec![1 , 2, 3]);
42+
/// assert_eq!(decrypted_data, vec![1, 2, 3]);
4343
/// ```
4444
#[derive(PartialEq)]
4545
pub struct OurPeerStorage {
@@ -71,7 +71,7 @@ impl OurPeerStorage {
7171
let mut nonce = [0; 12];
7272
nonce[4..].copy_from_slice(&n.to_le_bytes()[..]);
7373

74-
let mut chacha = ChaCha20Poly1305RFC::new(key.as_bytes(), &nonce, b"");
74+
let mut chacha = ChaCha20Poly1305RFC::new(&key.inner, &nonce, b"");
7575
let mut tag = [0; 16];
7676
chacha.encrypt_full_message_in_place(&mut ser_channels[0..plaintext_len], &mut tag);
7777

@@ -98,7 +98,7 @@ impl OurPeerStorage {
9898
let mut nonce = [0; 12];
9999
nonce[4..].copy_from_slice(&n.to_le_bytes()[..]);
100100

101-
let mut chacha = ChaCha20Poly1305RFC::new(key.as_bytes(), &nonce, b"");
101+
let mut chacha = ChaCha20Poly1305RFC::new(&key.inner, &nonce, b"");
102102

103103
if chacha.check_decrypt_in_place(encrypted_data, tag).is_err() {
104104
return Err(());

0 commit comments

Comments
 (0)