adds feature-gate code to enforce retransmitter signature verification

anza-xyz · Jun 24, 2024 · 8abd1d5 · 8abd1d5
1 parent fae0a50
commit 8abd1d5
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/sdk/src/feature_set.rs b/sdk/src/feature_set.rs
@@ -817,6 +817,10 @@ pub mod migrate_config_program_to_core_bpf {
     solana_sdk::declare_id!("2Fr57nzzkLYXW695UdDxDeR5fhnZWSttZeZYemrnpGFV");
 }
 
+pub mod verify_retransmitter_signature {
+    solana_sdk::declare_id!("BZ5g4hRbu5hLQQBdPyo2z9icGyJ8Khiyj3QS6dhWijTb");
+}
+
 lazy_static! {
     /// Map of feature identifiers to user-visible description
     pub static ref FEATURE_NAMES: HashMap<Pubkey, &'static str> = [
@@ -1016,6 +1020,7 @@ lazy_static! {
         (migrate_feature_gate_program_to_core_bpf::id(), "Migrate Feature Gate program to Core BPF (programify) #1003"),
         (vote_only_full_fec_sets::id(), "vote only full fec sets"),
         (migrate_config_program_to_core_bpf::id(), "Migrate Config program to Core BPF #1378"),
+        (verify_retransmitter_signature::id(), "Verify retransmitter signature #1840"),
         /*************** ADD NEW FEATURES HERE ***************/
     ]
     .iter()

diff --git a/turbine/src/sigverify_shreds.rs b/turbine/src/sigverify_shreds.rs
@@ -1,6 +1,6 @@
 use {
     crate::{
-        cluster_nodes::{self, ClusterNodesCache},
+        cluster_nodes::{self, check_feature_activation, ClusterNodesCache},
         retransmit_stage::RetransmitStage,
     },
     crossbeam_channel::{Receiver, RecvTimeoutError, SendError, Sender},
@@ -19,6 +19,7 @@ use {
     },
     solana_sdk::{
         clock::Slot,
+        feature_set,
         pubkey::Pubkey,
         signature::{Keypair, Signer},
     },
@@ -201,6 +202,19 @@ fn run_shred_sigverify<const K: usize>(
                     stats
                         .num_invalid_retransmitter
                         .fetch_add(1, Ordering::Relaxed);
+                    if shred::layout::get_slot(shred)
+                        .map(|slot| {
+                            check_feature_activation(
+                                &feature_set::verify_retransmitter_signature::id(),
+                                slot,
+                                &root_bank,
+                            )
+                        })
+                        .unwrap_or_default()
+                    {
+                        packet.meta_mut().set_discard(true);
+                        return;
+                    }
                 }
                 // We can ignore Error::InvalidShredVariant because that
                 // basically means that the shred is of a variant which