From 11cac460aafd90ff9c3ce66a4acb5a983de597b1 Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Mon, 4 Oct 2021 18:14:26 -0700
Subject: [PATCH] Consume compute units since proof verification is an
 expensive operation

---
 programs/zk-token-proof/src/lib.rs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/programs/zk-token-proof/src/lib.rs b/programs/zk-token-proof/src/lib.rs
index 11a8667cfe..866baf7d9d 100644
--- 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)?
     {