Remove MonitorUpdateId from persist trait

Author: G8XSU

fuzz/src/utils/test_persister.rs

@@ -1,6 +1,5 @@
 use lightning::chain;
 use lightning::chain::{chainmonitor, channelmonitor};
-use lightning::chain::chainmonitor::MonitorUpdateId;
 use lightning::chain::transaction::OutPoint;
 use lightning::util::test_channel_signer::TestChannelSigner;
 
@@ -10,11 +9,11 @@ pub struct TestPersister {
 	pub update_ret: Mutex<chain::ChannelMonitorUpdateStatus>,
 }
 impl chainmonitor::Persist<TestChannelSigner> for TestPersister {
-	fn persist_new_channel(&self, _funding_txo: OutPoint, _data: &channelmonitor::ChannelMonitor<TestChannelSigner>, _update_id: MonitorUpdateId) -> chain::ChannelMonitorUpdateStatus {
+	fn persist_new_channel(&self, _funding_txo: OutPoint, _data: &channelmonitor::ChannelMonitor<TestChannelSigner>) -> chain::ChannelMonitorUpdateStatus {
 		self.update_ret.lock().unwrap().clone()
 	}
 
-	fn update_persisted_channel(&self, _funding_txo: OutPoint, _update: Option<&channelmonitor::ChannelMonitorUpdate>, _data: &channelmonitor::ChannelMonitor<TestChannelSigner>, _update_id: MonitorUpdateId) -> chain::ChannelMonitorUpdateStatus {
+	fn update_persisted_channel(&self, _funding_txo: OutPoint, _update: Option<&channelmonitor::ChannelMonitorUpdate>, _data: &channelmonitor::ChannelMonitor<TestChannelSigner>) -> chain::ChannelMonitorUpdateStatus {
 		self.update_ret.lock().unwrap().clone()
 	}
 

lightning-persister/src/fs_store.rs

@@ -448,8 +448,6 @@ mod tests {
 		nodes[1].node.force_close_broadcasting_latest_txn(&chan.2, &nodes[0].node.get_our_node_id()).unwrap();
 		check_closed_event!(nodes[1], 1, ClosureReason::HolderForceClosed, [nodes[0].node.get_our_node_id()], 100000);
 		let mut added_monitors = nodes[1].chain_monitor.added_monitors.lock().unwrap();
-		let update_map = nodes[1].chain_monitor.latest_monitor_update_id.lock().unwrap();
-		let update_id = update_map.get(&added_monitors[0].1.channel_id()).unwrap();
 
 		// Set the store's directory to read-only, which should result in
 		// returning an unrecoverable failure when we then attempt to persist a
@@ -463,7 +461,7 @@ mod tests {
 			txid: Txid::from_str("8984484a580b825b9972d7adb15050b3ab624ccd731946b3eeddb92f4e7ef6be").unwrap(),
 			index: 0
 		};
-		match store.persist_new_channel(test_txo, &added_monitors[0].1, update_id.2) {
+		match store.persist_new_channel(test_txo, &added_monitors[0].1) {
 			ChannelMonitorUpdateStatus::UnrecoverableError => {},
 			_ => panic!("unexpected result from persisting new channel")
 		}

lightning/src/chain/chainmonitor.rs

@@ -47,46 +47,6 @@ use core::ops::Deref;
 use core::sync::atomic::{AtomicUsize, Ordering};
 use bitcoin::secp256k1::PublicKey;
 
-mod update_origin {
-	#[derive(Debug, Clone, Copy, Hash, PartialEq, Eq)]
-	/// A specific update's ID stored in a `MonitorUpdateId`, separated out to make the contents
-	/// entirely opaque.
-	pub(crate) enum UpdateOrigin {
-		/// An update that was generated by the `ChannelManager` (via our [`crate::chain::Watch`]
-		/// implementation). This corresponds to an actual [ChannelMonitorUpdate::update_id] field
-		/// and [ChannelMonitor::get_latest_update_id].
-		///
-		/// [ChannelMonitor::get_latest_update_id]: crate::chain::channelmonitor::ChannelMonitor::get_latest_update_id
-		/// [ChannelMonitorUpdate::update_id]: crate::chain::channelmonitor::ChannelMonitorUpdate::update_id
-		OffChain(u64),
-		/// An update that was generated during blockchain processing. The ID here is specific to the
-		/// generating [ChannelMonitor] and does *not* correspond to any on-disk IDs.
-		///
-		/// [ChannelMonitor]: crate::chain::channelmonitor::ChannelMonitor
-		ChainSync(u64),
-	}
-}
-
-#[cfg(any(feature = "_test_utils", test))]
-pub(crate) use update_origin::UpdateOrigin;
-#[cfg(not(any(feature = "_test_utils", test)))]
-use update_origin::UpdateOrigin;
-
-/// An opaque identifier describing a specific [`Persist`] method call.
-#[derive(Debug, Clone, Copy, Hash, PartialEq, Eq)]
-pub struct MonitorUpdateId {
-	pub(crate) contents: UpdateOrigin,
-}
-
-impl MonitorUpdateId {
-	pub(crate) fn from_monitor_update(update: &ChannelMonitorUpdate) -> Self {
-		Self { contents: UpdateOrigin::OffChain(update.update_id) }
-	}
-	pub(crate) fn from_new_monitor<ChannelSigner: WriteableEcdsaChannelSigner>(monitor: &ChannelMonitor<ChannelSigner>) -> Self {
-		Self { contents: UpdateOrigin::OffChain(monitor.get_latest_update_id()) }
-	}
-}
-
 /// `Persist` defines behavior for persisting channel monitors: this could mean
 /// writing once to disk, and/or uploading to one or more backup services.
 ///
@@ -105,7 +65,9 @@ impl MonitorUpdateId {
 ///    [`ChainMonitor::get_monitor`].
 ///
 ///    Once a full [`ChannelMonitor`] has been persisted, all pending updates for that channel can
-///    be marked as complete via [`ChainMonitor::channel_monitor_updated`].
+///    be marked as complete via [`ChainMonitor::channel_monitor_updated`]. It is only necessary to
+///    call [`ChainMonitor::channel_monitor_updated`] when an `update_id` is provided and when you
+///    return [`ChannelMonitorUpdateStatus::InProgress`].
 ///
 ///    If at some point no further progress can be made towards persisting the pending updates, the
 ///    node should simply shut down.
@@ -117,9 +79,9 @@ impl MonitorUpdateId {
 /// # For those implementing asynchronous persistence
 ///
 ///  All calls should generally spawn a background task and immediately return
-///  [`ChannelMonitorUpdateStatus::InProgress`]. Once the update completes,
-///  [`ChainMonitor::channel_monitor_updated`] should be called with the corresponding
-///  [`MonitorUpdateId`].
+///  [`ChannelMonitorUpdateStatus::InProgress`]. Once the update completes, it is only necessary to
+///  call [`ChainMonitor::channel_monitor_updated`] when an `update_id` is provided and when you
+///  return [`ChannelMonitorUpdateStatus::InProgress`].
 ///
 ///  Note that unlike the direct [`chain::Watch`] interface,
 ///  [`ChainMonitor::channel_monitor_updated`] must be called once for *each* update which occurs.
@@ -150,15 +112,16 @@ pub trait Persist<ChannelSigner: WriteableEcdsaChannelSigner> {
 	/// channel's outpoint (and it is up to you to maintain a correct mapping between the outpoint
 	/// and the stored channel data). Note that you **must** persist every new monitor to disk.
 	///
-	/// The `update_id` is used to identify this call to [`ChainMonitor::channel_monitor_updated`],
-	/// if you return [`ChannelMonitorUpdateStatus::InProgress`].
+	/// The `update_id` uniquely identifies a call to [`ChainMonitor::channel_monitor_updated`].
+	/// It is only necessary to call [`ChainMonitor::channel_monitor_updated`] when an `update_id` is
+	/// provided and when you return [`ChannelMonitorUpdateStatus::InProgress`].
 	///
 	/// See [`Writeable::write`] on [`ChannelMonitor`] for writing out a `ChannelMonitor`
 	/// and [`ChannelMonitorUpdateStatus`] for requirements when returning errors.
 	///
 	/// [`ChannelManager`]: crate::ln::channelmanager::ChannelManager
 	/// [`Writeable::write`]: crate::util::ser::Writeable::write
-	fn persist_new_channel(&self, channel_funding_outpoint: OutPoint, data: &ChannelMonitor<ChannelSigner>, update_id: MonitorUpdateId) -> ChannelMonitorUpdateStatus;
+	fn persist_new_channel(&self, channel_funding_outpoint: OutPoint, data: &ChannelMonitor<ChannelSigner>) -> ChannelMonitorUpdateStatus;
 
 	/// Update one channel's data. The provided [`ChannelMonitor`] has already applied the given
 	/// update.
@@ -185,15 +148,16 @@ pub trait Persist<ChannelSigner: WriteableEcdsaChannelSigner> {
 	/// them in batches. The size of each monitor grows `O(number of state updates)`
 	/// whereas updates are small and `O(1)`.
 	///
-	/// The `update_id` is used to identify this call to [`ChainMonitor::channel_monitor_updated`],
-	/// if you return [`ChannelMonitorUpdateStatus::InProgress`].
+	/// The `update_id` uniquely identifies a call to [`ChainMonitor::channel_monitor_updated`].
+	/// It is only necessary to call [`ChainMonitor::channel_monitor_updated`] when an `update_id` is
+	/// provided and when you return [`ChannelMonitorUpdateStatus::InProgress`].
 	///
 	/// See [`Writeable::write`] on [`ChannelMonitor`] for writing out a `ChannelMonitor`,
 	/// [`Writeable::write`] on [`ChannelMonitorUpdate`] for writing out an update, and
 	/// [`ChannelMonitorUpdateStatus`] for requirements when returning errors.
 	///
 	/// [`Writeable::write`]: crate::util::ser::Writeable::write
-	fn update_persisted_channel(&self, channel_funding_outpoint: OutPoint, update: Option<&ChannelMonitorUpdate>, data: &ChannelMonitor<ChannelSigner>, update_id: MonitorUpdateId) -> ChannelMonitorUpdateStatus;
+	fn update_persisted_channel(&self, channel_funding_outpoint: OutPoint, update: Option<&ChannelMonitorUpdate>, data: &ChannelMonitor<ChannelSigner>) -> ChannelMonitorUpdateStatus;
 	/// Prevents the channel monitor from being loaded on startup.
 	///
 	/// Archiving the data in a backup location (rather than deleting it fully) is useful for
@@ -209,13 +173,12 @@ struct MonitorHolder<ChannelSigner: WriteableEcdsaChannelSigner> {
 	/// update_persisted_channel, the user returns a
 	/// [`ChannelMonitorUpdateStatus::InProgress`], and then calls channel_monitor_updated
 	/// immediately, racing our insertion of the pending update into the contained Vec.
-	pending_monitor_updates: Mutex<Vec<MonitorUpdateId>>,
+	pending_monitor_updates: Mutex<Vec<u64>>,
 }
 
 impl<ChannelSigner: WriteableEcdsaChannelSigner> MonitorHolder<ChannelSigner> {
-	fn has_pending_offchain_updates(&self, pending_monitor_updates_lock: &MutexGuard<Vec<MonitorUpdateId>>) -> bool {
-		pending_monitor_updates_lock.iter().any(|update_id|
-			if let UpdateOrigin::OffChain(_) = update_id.contents { true } else { false })
+	fn has_pending_updates(&self, pending_monitor_updates_lock: &MutexGuard<Vec<u64>>) -> bool {
+		!pending_monitor_updates_lock.is_empty()
 	}
 }
 
@@ -346,20 +309,11 @@ where C::Target: chain::Filter,
 		let mut txn_outputs;
 		{
 			txn_outputs = process(monitor, txdata);
-			let chain_sync_update_id = self.sync_persistence_id.get_increment();
-			let update_id = MonitorUpdateId {
-				contents: UpdateOrigin::ChainSync(chain_sync_update_id),
-			};
-
-			log_trace!(logger, "Syncing Channel Monitor for channel {} for block-data update_id {}",
-				log_funding_info!(monitor),
-				chain_sync_update_id
-			);
-			match self.persister.update_persisted_channel(*funding_outpoint, None, monitor, update_id) {
+			log_trace!(logger, "Syncing Channel Monitor for channel {}", log_funding_info!(monitor));
+			match self.persister.update_persisted_channel(*funding_outpoint, None, monitor) {
 				ChannelMonitorUpdateStatus::Completed =>
-					log_trace!(logger, "Finished syncing Channel Monitor for channel {} for block-data update_id {}",
-						log_funding_info!(monitor),
-						chain_sync_update_id
+					log_trace!(logger, "Finished syncing Channel Monitor for channel {} for block-data",
+						log_funding_info!(monitor)
 					),
 				ChannelMonitorUpdateStatus::InProgress => {
 					log_debug!(logger, "Channel Monitor sync for channel {} in progress.", log_funding_info!(monitor));
@@ -464,15 +418,15 @@ where C::Target: chain::Filter,
 
 	#[cfg(not(c_bindings))]
 	/// Lists the pending updates for each [`ChannelMonitor`] (by `OutPoint` being monitored).
-	pub fn list_pending_monitor_updates(&self) -> HashMap<OutPoint, Vec<MonitorUpdateId>> {
+	pub fn list_pending_monitor_updates(&self) -> HashMap<OutPoint, Vec<u64>> {
 		hash_map_from_iter(self.monitors.read().unwrap().iter().map(|(outpoint, holder)| {
 			(*outpoint, holder.pending_monitor_updates.lock().unwrap().clone())
 		}))
 	}
 
 	#[cfg(c_bindings)]
 	/// Lists the pending updates for each [`ChannelMonitor`] (by `OutPoint` being monitored).
-	pub fn list_pending_monitor_updates(&self) -> Vec<(OutPoint, Vec<MonitorUpdateId>)> {
+	pub fn list_pending_monitor_updates(&self) -> Vec<(OutPoint, Vec<u64>)> {
 		self.monitors.read().unwrap().iter().map(|(outpoint, holder)| {
 			(*outpoint, holder.pending_monitor_updates.lock().unwrap().clone())
 		}).collect()
@@ -500,47 +454,43 @@ where C::Target: chain::Filter,
 	///
 	/// Returns an [`APIError::APIMisuseError`] if `funding_txo` does not match any currently
 	/// registered [`ChannelMonitor`]s.
-	pub fn channel_monitor_updated(&self, funding_txo: OutPoint, completed_update_id: MonitorUpdateId) -> Result<(), APIError> {
+	pub fn channel_monitor_updated(&self, funding_txo: OutPoint, completed_update_id: u64) -> Result<(), APIError> {
 		let monitors = self.monitors.read().unwrap();
 		let monitor_data = if let Some(mon) = monitors.get(&funding_txo) { mon } else {
 			return Err(APIError::APIMisuseError { err: format!("No ChannelMonitor matching funding outpoint {:?} found", funding_txo) });
 		};
 		let mut pending_monitor_updates = monitor_data.pending_monitor_updates.lock().unwrap();
 		pending_monitor_updates.retain(|update_id| *update_id != completed_update_id);
 
-		match completed_update_id {
-			MonitorUpdateId { contents: UpdateOrigin::OffChain(completed_update_id) } => {
-				// Note that we only check for `UpdateOrigin::OffChain` failures here - if
-				// we're being told that a `UpdateOrigin::OffChain` monitor update completed,
-				// we only care about ensuring we don't tell the `ChannelManager` to restore
-				// the channel to normal operation until all `UpdateOrigin::OffChain` updates
-				// complete.
-				// If there's some `UpdateOrigin::ChainSync` update still pending that's okay
-				// - we can still update our channel state, just as long as we don't return
-				// `MonitorEvent`s from the monitor back to the `ChannelManager` until they
-				// complete.
-				let monitor_is_pending_updates = monitor_data.has_pending_offchain_updates(&pending_monitor_updates);
-				log_debug!(self.logger, "Completed off-chain monitor update {} for channel with funding outpoint {:?}, {}",
-					completed_update_id,
-					funding_txo,
-					if monitor_is_pending_updates {
-						"still have pending off-chain updates"
-					} else {
-						"all off-chain updates complete, returning a MonitorEvent"
-					});
-				if monitor_is_pending_updates {
-					// If there are still monitor updates pending, we cannot yet construct a
-					// Completed event.
-					return Ok(());
-				}
-				let channel_id = monitor_data.monitor.channel_id();
-				self.pending_monitor_events.lock().unwrap().push((funding_txo, channel_id, vec![MonitorEvent::Completed {
-					funding_txo, channel_id,
-					monitor_update_id: monitor_data.monitor.get_latest_update_id(),
-				}], monitor_data.monitor.get_counterparty_node_id()));
-			},
-			MonitorUpdateId { contents: UpdateOrigin::ChainSync(_) } => {},
+		// Note that we only check for `UpdateOrigin::OffChain` failures here - if
+		// we're being told that a `UpdateOrigin::OffChain` monitor update completed,
+		// we only care about ensuring we don't tell the `ChannelManager` to restore
+		// the channel to normal operation until all `UpdateOrigin::OffChain` updates
+		// complete.
+		// If there's some `UpdateOrigin::ChainSync` update still pending that's okay
+		// - we can still update our channel state, just as long as we don't return
+		// `MonitorEvent`s from the monitor back to the `ChannelManager` until they
+		// complete.
+		let monitor_is_pending_updates = monitor_data.has_pending_updates(&pending_monitor_updates);
+		log_debug!(self.logger, "Completed off-chain monitor update {} for channel with funding outpoint {:?}, {}",
+			completed_update_id,
+			funding_txo,
+			if monitor_is_pending_updates {
+				"still have pending off-chain updates"
+			} else {
+				"all off-chain updates complete, returning a MonitorEvent"
+			});
+		if monitor_is_pending_updates {
+			// If there are still monitor updates pending, we cannot yet construct a
+			// Completed event.
+			return Ok(());
 		}
+		let channel_id = monitor_data.monitor.channel_id();
+		self.pending_monitor_events.lock().unwrap().push((funding_txo, channel_id, vec![MonitorEvent::Completed {
+			funding_txo, channel_id,
+			monitor_update_id: monitor_data.monitor.get_latest_update_id(),
+		}], monitor_data.monitor.get_counterparty_node_id()));
+
 		self.event_notifier.notify();
 		Ok(())
 	}
@@ -771,9 +721,9 @@ where C::Target: chain::Filter,
 			hash_map::Entry::Vacant(e) => e,
 		};
 		log_trace!(logger, "Got new ChannelMonitor for channel {}", log_funding_info!(monitor));
-		let update_id = MonitorUpdateId::from_new_monitor(&monitor);
+		let update_id = monitor.get_latest_update_id();
 		let mut pending_monitor_updates = Vec::new();
-		let persist_res = self.persister.persist_new_channel(funding_outpoint, &monitor, update_id);
+		let persist_res = self.persister.persist_new_channel(funding_outpoint, &monitor);
 		match persist_res {
 			ChannelMonitorUpdateStatus::InProgress => {
 				log_info!(logger, "Persistence of new ChannelMonitor for channel {} in progress", log_funding_info!(monitor));
@@ -823,7 +773,7 @@ where C::Target: chain::Filter,
 				log_trace!(logger, "Updating ChannelMonitor to id {} for channel {}", update.update_id, log_funding_info!(monitor));
 				let update_res = monitor.update_monitor(update, &self.broadcaster, &self.fee_estimator, &self.logger);
 
-				let update_id = MonitorUpdateId::from_monitor_update(update);
+				let update_id = update.update_id;
 				let mut pending_monitor_updates = monitor_state.pending_monitor_updates.lock().unwrap();
 				let persist_res = if update_res.is_err() {
 					// Even if updating the monitor returns an error, the monitor's state will
@@ -832,9 +782,9 @@ where C::Target: chain::Filter,
 					// while reading `channel_monitor` with updates from storage. Instead, we should persist
 					// the entire `channel_monitor` here.
 					log_warn!(logger, "Failed to update ChannelMonitor for channel {}. Going ahead and persisting the entire ChannelMonitor", log_funding_info!(monitor));
-					self.persister.update_persisted_channel(funding_txo, None, monitor, update_id)
+					self.persister.update_persisted_channel(funding_txo, None, monitor)
 				} else {
-					self.persister.update_persisted_channel(funding_txo, Some(update), monitor, update_id)
+					self.persister.update_persisted_channel(funding_txo, Some(update), monitor)
 				};
 				match persist_res {
 					ChannelMonitorUpdateStatus::InProgress => {