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
1616use crate :: crypto:: chacha20poly1305rfc:: ChaCha20Poly1305RFC ;
1717use 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 ) ]
4545pub 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