Consume compute units since proof verification is an expensive operation

2021-10-04 18:14:26 -07:00
parent 62b6eafd7c
commit 11cac460aa
1 changed files with 6 additions and 0 deletions
--- a/programs/zk-token-proof/src/lib.rs
+++ b/programs/zk-token-proof/src/lib.rs
@ -34,6 +34,12 @@ pub fn process_instruction(
        return Err(InstructionError::UnsupportedProgramId);
    }

+    // Consume compute units since proof verification is an expensive operation
+    {
+        let compute_meter = invoke_context.get_compute_meter();
+        compute_meter.borrow_mut().consume(25_000)?; // TODO: Tune the number of units consumed?
+    }
+
    match ProofInstruction::decode_type(program_id, input)
        .ok_or(InstructionError::InvalidInstructionData)?
    {