remove verify_partial_witness_signature

near · Nov 8, 2024 · c8eab17 · c8eab17
1 parent 69d0922
commit c8eab17
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 25 deletions.
diff --git a/chain/chain/src/test_utils/kv_runtime.rs b/chain/chain/src/test_utils/kv_runtime.rs
@@ -1022,13 +1022,6 @@ impl EpochManagerAdapter for MockEpochManager {
         Ok(true)
     }
 
-    fn verify_partial_witness_signature(
-        &self,
-        _partial_witness: &PartialEncodedStateWitness,
-    ) -> Result<bool, Error> {
-        Ok(true)
-    }
-
     fn verify_witness_contract_accesses_signature(
         &self,
         _accesses: &ChunkContractAccesses,

diff --git a/chain/client/src/stateless_validation/validate.rs b/chain/client/src/stateless_validation/validate.rs
@@ -62,7 +62,9 @@ pub fn validate_partial_encoded_state_witness(
         return Ok(false);
     }
 
-    if !epoch_manager.verify_partial_witness_signature(&partial_witness)? {
+    let chunk_producer =
+        epoch_manager.get_chunk_producer_info(&partial_witness.chunk_production_key())?;
+    if !partial_witness.verify(chunk_producer.public_key()) {
         return Err(Error::InvalidPartialChunkStateWitness("Invalid signature".to_string()));
     }
 

diff --git a/chain/epoch-manager/src/adapter.rs b/chain/epoch-manager/src/adapter.rs
@@ -491,11 +491,6 @@ pub trait EpochManagerAdapter: Send + Sync {
         endorsement: &ChunkEndorsement,
     ) -> Result<bool, Error>;
 
-    fn verify_partial_witness_signature(
-        &self,
-        partial_witness: &PartialEncodedStateWitness,
-    ) -> Result<bool, Error>;
-
     fn verify_witness_contract_accesses_signature(
         &self,
         accesses: &ChunkContractAccesses,
@@ -1177,15 +1172,6 @@ impl EpochManagerAdapter for EpochManagerHandle {
         Ok(endorsement.verify(validator.public_key()))
     }
 
-    fn verify_partial_witness_signature(
-        &self,
-        partial_witness: &PartialEncodedStateWitness,
-    ) -> Result<bool, Error> {
-        let chunk_producer =
-            self.read().get_chunk_producer_info(&partial_witness.chunk_production_key())?;
-        Ok(partial_witness.verify(chunk_producer.public_key()))
-    }
-
     fn verify_witness_contract_accesses_signature(
         &self,
         accesses: &ChunkContractAccesses,

diff --git a/chain/epoch-manager/src/tests/mod.rs b/chain/epoch-manager/src/tests/mod.rs
@@ -3360,11 +3360,13 @@ fn test_verify_partial_witness_signature() {
         7,
         signer.as_ref(),
     );
-    assert!(epoch_manager.verify_partial_witness_signature(&partial_witness).unwrap());
+    let chunk_producer =
+        epoch_manager.get_chunk_producer_info(&partial_witness.chunk_production_key()).unwrap();
+    assert!(partial_witness.verify(chunk_producer.public_key()));
 
     // Check invalid chunk state witness signature.
     partial_witness.signature = Signature::default();
-    assert!(!epoch_manager.verify_partial_witness_signature(&partial_witness).unwrap());
+    assert!(!partial_witness.verify(chunk_producer.public_key()));
 
     // Check chunk state witness invalidity when signer is not a chunk validator.
     let bad_signer = Arc::new(create_test_signer("test2"));
@@ -3376,7 +3378,7 @@ fn test_verify_partial_witness_signature() {
         7,
         bad_signer.as_ref(),
     );
-    assert!(!epoch_manager.verify_partial_witness_signature(&bad_partial_witness).unwrap());
+    assert!(!bad_partial_witness.verify(chunk_producer.public_key()));
 }
 
 /// Simulate the blockchain over a few epochs and verify that possible_epochs_of_height_around_tip()