fix(sync): ban peer if sending invalid prev_header (#3955)

Description --- - bans peer for ChainLinkBroken error Motivation and Context --- Node should not continue syncing from a peer that does not properly link headers How Has This Been Tested? --- Manually, my node banned peers that sent an incorrect prev header
tari-project · Mar 24, 2022 · 384ab0c · 384ab0c
1 parent 48eb86e
commit 384ab0c
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 2 deletions.
diff --git a/base_layer/core/src/base_node/sync/header_sync/error.rs b/base_layer/core/src/base_node/sync/header_sync/error.rs
@@ -67,8 +67,12 @@ pub enum BlockHeaderSyncError {
     NetworkSilence,
     #[error("Invalid protocol response: {0}")]
     InvalidProtocolResponse(String),
-    #[error("Headers did not form a chain. Expected {actual} to equal the previous hash {expected}")]
-    ChainLinkBroken { actual: String, expected: String },
+    #[error("Header at height {height} did not form a chain. Expected {actual} to equal the previous hash {expected}")]
+    ChainLinkBroken {
+        height: u64,
+        actual: String,
+        expected: String,
+    },
     #[error("Block error: {0}")]
     BlockError(#[from] BlockError),
     #[error(

diff --git a/base_layer/core/src/base_node/sync/header_sync/synchronizer.rs b/base_layer/core/src/base_node/sync/header_sync/synchronizer.rs
@@ -180,6 +180,19 @@ impl<'a, B: BlockchainBackend + 'static> HeaderSynchronizer<'a, B> {
                     })
                     .await?;
                 },
+                Err(BlockHeaderSyncError::ChainLinkBroken {
+                    height,
+                    actual,
+                    expected,
+                }) => {
+                    let reason = BanReason::ChainLinkBroken {
+                        height,
+                        actual: actual.to_string(),
+                        expected: expected.to_string(),
+                    };
+                    warn!(target: LOG_TARGET, "Chain link broken: {}", reason);
+                    self.ban_peer_long(node_id, reason).await?;
+                },
                 Err(ref err @ BlockHeaderSyncError::WeakerChain { claimed, actual, local }) => {
                     warn!(target: LOG_TARGET, "{}", err);
                     self.ban_peer_long(node_id, BanReason::PeerCouldNotProvideStrongerChain {
@@ -794,6 +807,12 @@ enum BanReason {
         "Peer claimed an accumulated difficulty of {claimed} but validated difficulty was {actual} <= local: {local}"
     )]
     PeerCouldNotProvideStrongerChain { claimed: u128, actual: u128, local: u128 },
+    #[error("Header at height {height} did not form a chain. Expected {actual} to equal the previous hash {expected}")]
+    ChainLinkBroken {
+        height: u64,
+        actual: String,
+        expected: String,
+    },
 }
 
 struct ChainSplitInfo {

diff --git a/base_layer/core/src/base_node/sync/header_sync/validator.rs b/base_layer/core/src/base_node/sync/header_sync/validator.rs
@@ -129,6 +129,7 @@ impl<B: BlockchainBackend + 'static> BlockHeaderSyncValidator<B> {
         }
         if header.prev_hash != state.previous_accum.hash {
             return Err(BlockHeaderSyncError::ChainLinkBroken {
+                height: header.height,
                 actual: header.prev_hash.to_hex(),
                 expected: state.previous_accum.hash.to_hex(),
             });