[zk-token-proof] include VerifyBatchRangeProofU256 in the `enable_z…

…k_transfer_with_fee` feature gate (#34747) include `VerifyBatchRangeProofU256` in the `enable_zk_transfer_with_fee` feature
solana-labs · Jan 12, 2024 · bc13642 · bc13642
1 parent 4385ed1
commit bc13642
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/programs/zk-token-proof/src/lib.rs b/programs/zk-token-proof/src/lib.rs
@@ -261,6 +261,11 @@ declare_process_instruction!(Entrypoint, 0, |invoke_context| {
             )
         }
         ProofInstruction::VerifyBatchedRangeProofU256 => {
+            // transfer with fee related proofs are not enabled
+            if !enable_zk_transfer_with_fee {
+                return Err(InstructionError::InvalidInstructionData);
+            }
+
             if native_programs_consume_cu {
                 invoke_context
                     .consume_checked(VERIFY_BATCHED_RANGE_PROOF_U256_COMPUTE_UNITS)