Skip to content

Commit c302763

Browse files
committed
Add read_channel_monitors utility
This replaces the `FilesystemPersister::read_channelmonitors` method, as we can now implement a single utility for all `KVStore`s.
1 parent d40ab06 commit c302763

File tree

1 file changed

+49
-3
lines changed

1 file changed

+49
-3
lines changed

lightning/src/util/persist.rs

Lines changed: 49 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,9 @@
99
//! and [`ChannelMonitor`] all in one place.
1010
1111
use core::ops::Deref;
12-
use bitcoin::hashes::hex::ToHex;
12+
use bitcoin::hashes::hex::{FromHex, ToHex};
13+
use bitcoin::{BlockHash, Txid};
14+
1315
use crate::io;
1416
use crate::prelude::{Vec, String};
1517
use crate::routing::scoring::WriteableScore;
@@ -24,7 +26,7 @@ use crate::ln::channelmanager::ChannelManager;
2426
use crate::routing::router::Router;
2527
use crate::routing::gossip::NetworkGraph;
2628
use crate::util::logger::Logger;
27-
use crate::util::ser::Writeable;
29+
use crate::util::ser::{ReadableArgs, Writeable};
2830

2931
/// The namespace under which the [`ChannelManager`] will be persisted.
3032
pub const CHANNEL_MANAGER_PERSISTENCE_NAMESPACE: &str = "";
@@ -156,4 +158,48 @@ impl<ChannelSigner: WriteableEcdsaChannelSigner, K: KVStore> Persist<ChannelSign
156158
}
157159
}
158160

159-
161+
/// Read previously persisted [`ChannelMonitor`]s from the store.
162+
pub fn read_channel_monitors<K: Deref, ES: Deref, SP: Deref>(
163+
kv_store: K, entropy_source: ES, signer_provider: SP,
164+
) -> crate::io::Result<Vec<(BlockHash, ChannelMonitor<<SP::Target as SignerProvider>::Signer>)>>
165+
where
166+
K::Target: KVStore,
167+
ES::Target: EntropySource + Sized,
168+
SP::Target: SignerProvider + Sized,
169+
{
170+
let mut res = Vec::new();
171+
172+
for stored_key in kv_store.list(CHANNEL_MONITOR_PERSISTENCE_NAMESPACE)? {
173+
let txid = Txid::from_hex(stored_key.split_at(64).0).map_err(|_| {
174+
crate::io::Error::new(crate::io::ErrorKind::InvalidData, "Invalid tx ID in stored key")
175+
})?;
176+
177+
let index: u16 = stored_key.split_at(65).1.parse().map_err(|_| {
178+
crate::io::Error::new(crate::io::ErrorKind::InvalidData, "Invalid tx index in stored key")
179+
})?;
180+
181+
match <(BlockHash, ChannelMonitor<<SP::Target as SignerProvider>::Signer>)>::read(
182+
&mut kv_store.read(CHANNEL_MONITOR_PERSISTENCE_NAMESPACE, &stored_key)?,
183+
(&*entropy_source, &*signer_provider),
184+
) {
185+
Ok((block_hash, channel_monitor)) => {
186+
if channel_monitor.get_funding_txo().0.txid != txid
187+
|| channel_monitor.get_funding_txo().0.index != index
188+
{
189+
return Err(crate::io::Error::new(
190+
crate::io::ErrorKind::InvalidData,
191+
"ChannelMonitor was stored under the wrong key",
192+
));
193+
}
194+
res.push((block_hash, channel_monitor));
195+
}
196+
Err(_) => {
197+
return Err(crate::io::Error::new(
198+
crate::io::ErrorKind::InvalidData,
199+
"Failed to deserialize ChannelMonitor"
200+
))
201+
}
202+
}
203+
}
204+
Ok(res)
205+
}

0 commit comments

Comments
 (0)