@@ -56,22 +56,15 @@ use crate::ln::msgs::DecodeError;
5656#[ derive( PartialEq ) ]
5757pub struct OurPeerStorage {
5858 version : u8 ,
59- timestamp : u32 ,
59+ // If the block height is 0, OurPeerStorage doesn't have any channels backed up.
60+ block_height : u32 ,
6061 ser_channels : Vec < u8 > ,
6162}
6263
6364impl OurPeerStorage {
6465 /// Returns a [`OurPeerStorage`] with version 1 and current timestamp.
6566 pub fn new ( ) -> Self {
66- let duration_since_epoch = std:: time:: SystemTime :: now ( )
67- . duration_since ( std:: time:: SystemTime :: UNIX_EPOCH )
68- . expect ( "Time must be > 1970" ) ;
69-
70- Self {
71- version : 1 ,
72- timestamp : duration_since_epoch. as_secs ( ) as u32 ,
73- ser_channels : Vec :: new ( ) ,
74- }
67+ Self { version : 1 , block_height : 0u32 , ser_channels : Vec :: new ( ) }
7568 }
7669
7770 /// Stubs a channel inside [`OurPeerStorage`]
@@ -84,6 +77,10 @@ impl OurPeerStorage {
8477 self . ser_channels . clone ( )
8578 }
8679
80+ pub fn get_block_height ( & self ) -> u32 {
81+ self . block_height
82+ }
83+
8784 /// Encrypt [`OurPeerStorage`] using the `key` and return a `Vec<u8>` containing the result.
8885 pub fn encrypt_our_peer_storage ( & self , key : [ u8 ; 32 ] ) -> Vec < u8 > {
8986 let n = 0u64 ;
@@ -135,7 +132,7 @@ impl OurPeerStorage {
135132impl Writeable for OurPeerStorage {
136133 fn write < W : Writer > ( & self , writer : & mut W ) -> Result < ( ) , Error > {
137134 write_ver_prefix ! ( writer, self . version, 1 ) ;
138- self . timestamp . write ( writer) ?;
135+ self . block_height . write ( writer) ?;
139136 self . ser_channels . write ( writer) ?;
140137 Ok ( ( ) )
141138 }
@@ -144,10 +141,10 @@ impl Writeable for OurPeerStorage {
144141impl Readable for OurPeerStorage {
145142 fn read < R : io:: Read > ( reader : & mut R ) -> Result < Self , DecodeError > {
146143 let ver = read_ver_prefix ! ( reader, 1u8 ) ;
147- let timestamp : u32 = Readable :: read ( reader) ?;
144+ let block_height : u32 = Readable :: read ( reader) ?;
148145 let ser_channels = <Vec < u8 > as Readable >:: read ( reader) ?;
149146
150- let ps = OurPeerStorage { version : ver, timestamp , ser_channels } ;
147+ let ps = OurPeerStorage { version : ver, block_height , ser_channels } ;
151148 Ok ( ps)
152149 }
153150}
0 commit comments