add logic to disallow blocks on consecutive relay slots

Signed-off-by: Charles Ferrell <charlie@manta.network>
Manta-Network · Apr 9, 2024 · 0a73df3 · 0a73df3
1 parent 63d95c7
commit 0a73df3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/runtime/calamari/src/lib.rs b/runtime/calamari/src/lib.rs
@@ -1340,6 +1340,11 @@ impl_runtime_apis! {
         fn can_author(author: NimbusId, relay_parent: u32, parent_header: &<Block as BlockT>::Header) -> bool {
             let next_block_number = parent_header.number + 1;
             let slot = relay_parent;
+            if cumulus_pallet_parachain_system::Pallet::<Self>::last_relay_block_number() + 1 == relay_parent {
+                log::debug!("Cannot Author blocks on consecutive slots");
+                return false
+            }
+
             // Because the staking solution calculates the next staking set at the beginning
             // of the first block in the new round, the only way to accurately predict the
             // authors is to compute the selection during prediction.

diff --git a/runtime/manta/src/lib.rs b/runtime/manta/src/lib.rs
@@ -1261,6 +1261,11 @@ impl_runtime_apis! {
         fn can_author(author: NimbusId, relay_parent: u32, parent_header: &<Block as BlockT>::Header) -> bool {
             let next_block_number = parent_header.number + 1;
             let slot = relay_parent;
+            if cumulus_pallet_parachain_system::Pallet::<Self>::last_relay_block_number() + 1 == relay_parent {
+                log::debug!("Cannot Author blocks on consecutive slots");
+                return false
+            }
+
             // Because the staking solution calculates the next staking set at the beginning
             // of the first block in the new round, the only way to accurately predict the
             // authors is to compute the selection during prediction.