Add check for duplicate transactional inputs and outputs in validators

tari-project · May 14, 2020 · b951f43 · b951f43
1 parent 7d48672
commit b951f43
Showing 1 changed file with 34 additions and 2 deletions.
diff --git a/base_layer/core/src/validation/block_validators.rs b/base_layer/core/src/validation/block_validators.rs
@@ -67,6 +67,12 @@ impl StatelessValidation<Block> for StatelessBlockValidator {
         trace!(target: LOG_TARGET, "SV - Coinbase output is ok for {} ", &block_id);
         check_block_weight(block, &self.consensus_constants)?;
         trace!(target: LOG_TARGET, "SV - Block weight is ok for {} ", &block_id);
+        check_duplicate_transactions_inputs(block)?;
+        trace!(
+            target: LOG_TARGET,
+            "SV - No duplicate inputs or outputs for {} ",
+            &block_id
+        );
         // Check that the inputs are are allowed to be spent
         block.check_stxo_rules().map_err(BlockValidationError::from)?;
         trace!(target: LOG_TARGET, "SV - Output constraints are ok for {} ", &block_id);
@@ -108,12 +114,18 @@ impl<B: BlockchainBackend> Validation<Block, B> for FullConsensusValidator {
         trace!(target: LOG_TARGET, "FCV - Coinbase output ok for {}", &block_id);
         check_block_weight(block, &self.rules.consensus_constants())?;
         trace!(target: LOG_TARGET, "FCV - Block weight ok for {}", &block_id);
+        check_duplicate_transactions_inputs(block)?;
+        trace!(
+            target: LOG_TARGET,
+            "FCV - No duplicate inputs or outputs for {} ",
+            &block_id
+        );
         check_cut_through(block)?;
         trace!(target: LOG_TARGET, "FCV - Cut-though correct for {}", &block_id);
         block.check_stxo_rules().map_err(BlockValidationError::from)?;
-        trace!(target: LOG_TARGET, "FCV - Cut-though correct for {}", &block_id);
+        trace!(target: LOG_TARGET, "FCV - STxO rules correct for {}", &block_id);
         check_accounting_balance(block, self.rules.clone(), &self.factories)?;
-        trace!(target: LOG_TARGET, "FCV - Cut-though correct for {}", &block_id);
+        trace!(target: LOG_TARGET, "FCV - accounting balance correct for {}", &block_id);
         check_inputs_are_utxos(block, db)?;
         trace!(target: LOG_TARGET, "FCV - All inputs are valid for {}", &block_id);
         check_mmr_roots(block, db)?;
@@ -179,6 +191,26 @@ fn check_coinbase_output(block: &Block, consensus_constants: &ConsensusConstants
         .map_err(ValidationError::from)
 }
 
+// This function checks for duplicate inputs and outputs. There should be no duplicate inputs or outputs in a block
+fn check_duplicate_transactions_inputs(block: &Block) -> Result<(), ValidationError> {
+    trace!(
+        target: LOG_TARGET,
+        "Checking duplicate inputs and outputs on block with hash {}",
+        block.hash().to_hex()
+    );
+    for i in 1..block.body.inputs().len() {
+        if block.body.inputs()[i..].contains(&block.body.inputs()[i - 1]) {
+            return Err(ValidationError::CustomError("Duplicate Input".to_string()));
+        }
+    }
+    for i in 1..block.body.outputs().len() {
+        if block.body.outputs()[i..].contains(&block.body.outputs()[i - 1]) {
+            return Err(ValidationError::CustomError("Duplicate Output".to_string()));
+        }
+    }
+    Ok(())
+}
+
 /// This function checks that all inputs in the blocks are valid UTXO's to be spend
 fn check_inputs_are_utxos<B: BlockchainBackend>(block: &Block, db: &B) -> Result<(), ValidationError> {
     for utxo in block.body.inputs() {