Ismp updates (#81)

pallet-ismp/src/host.rs

@@ -18,7 +18,7 @@ use crate::{
     dispatcher::Receipt, primitives::ConsensusClientProvider, AllowedProxies, ChallengePeriod,
     Config, ConsensusClientUpdateTime, ConsensusStateClient, ConsensusStates,
     FrozenConsensusClients, FrozenHeights, LatestStateMachineHeight, Nonce, RequestCommitments,
-    RequestReceipts, ResponseReceipts, StateCommitments, UnbondingPeriod,
+    RequestReceipts, ResponseReceipts, StateCommitments, StateMachineUpdateTime, UnbondingPeriod,
 };
 use alloc::{format, string::ToString};
 use core::time::Duration;
@@ -71,6 +71,20 @@ impl<T: Config> IsmpHost for Host<T> {
             })
     }
 
+    fn state_machine_update_time(
+        &self,
+        state_machine_height: StateMachineHeight,
+    ) -> Result<Duration, Error> {
+        StateMachineUpdateTime::<T>::get(state_machine_height)
+            .map(|timestamp| Duration::from_secs(timestamp))
+            .ok_or_else(|| {
+                Error::ImplementationSpecific(format!(
+                    "Update time not found for {:?}",
+                    state_machine_height
+                ))
+            })
+    }
+
     fn consensus_state(&self, id: ConsensusClientId) -> Result<Vec<u8>, Error> {
         ConsensusStates::<T>::get(id)
             .ok_or_else(|| Error::ConsensusStateNotFound { consensus_state_id: id })
@@ -116,6 +130,18 @@ impl<T: Config> IsmpHost for Host<T> {
         Ok(())
     }
 
+    fn store_state_machine_update_time(
+        &self,
+        state_machine_height: StateMachineHeight,
+        timestamp: Duration,
+    ) -> Result<(), Error> {
+        StateMachineUpdateTime::<T>::insert(
+            state_machine_height,
+            timestamp.as_secs().saturated_into::<u64>(),
+        );
+        Ok(())
+    }
+
     fn store_state_machine_commitment(
         &self,
         height: StateMachineHeight,

pallet-ismp/src/lib.rs

@@ -221,6 +221,13 @@ pub mod pallet {
     pub type ConsensusClientUpdateTime<T: Config> =
         StorageMap<_, Twox64Concat, ConsensusClientId, u64, OptionQuery>;
 
+    /// Holds the timestamp at which a state machine height was updated.
+    /// Used in ensuring that the configured challenge period elapses.
+    #[pallet::storage]
+    #[pallet::getter(fn state_machine_update_time)]
+    pub type StateMachineUpdateTime<T: Config> =
+        StorageMap<_, Twox64Concat, StateMachineHeight, u64, OptionQuery>;
+
     /// Commitments for outgoing requests
     /// The key is the request commitment
     #[pallet::storage]

pallet-ismp/src/mocks/ismp.rs

@@ -136,13 +136,16 @@ where
         },
     )
     .unwrap();
-
-    set_timestamp::<T>(1000_000_000);
-    StateMachineHeight {
+    let height = StateMachineHeight {
         id: StateMachineId {
             state_id: StateMachine::Ethereum(Ethereum::ExecutionLayer),
             consensus_state_id: MOCK_CONSENSUS_STATE_ID,
         },
         height: 3,
-    }
+    };
+    host.store_state_machine_update_time(height, core::time::Duration::from_millis(1000_000))
+        .unwrap();
+
+    set_timestamp::<T>(1000_000_000);
+    height
 }