Add feature-wrapping for new ConfidentialTransferInstruction::Transfe…

…rWithFee (#3116)
solana-labs · Apr 26, 2022 · 51ece68 · 51ece68
1 parent b3fe5b6
commit 51ece68
Showing 1 changed file with 14 additions and 7 deletions.
diff --git a/token/program-2022/src/extension/confidential_transfer/processor.rs b/token/program-2022/src/extension/confidential_transfer/processor.rs
@@ -1164,13 +1164,20 @@ pub(crate) fn process_instruction(
         }
         ConfidentialTransferInstruction::TransferWithFee => {
             msg!("ConfidentialTransferInstruction::TransferWithFee");
-            let data = decode_instruction_data::<TransferWithFeeInstructionData>(input)?;
-            process_transfer(
-                program_id,
-                accounts,
-                data.new_source_decryptable_available_balance,
-                data.proof_instruction_offset as i64,
-            )
+            #[cfg(feature = "zk-ops")]
+            {
+                let data = decode_instruction_data::<TransferWithFeeInstructionData>(input)?;
+                process_transfer(
+                    program_id,
+                    accounts,
+                    data.new_source_decryptable_available_balance,
+                    data.proof_instruction_offset as i64,
+                )
+            }
+            #[cfg(not(feature = "zk-ops"))]
+            {
+                Err(ProgramError::InvalidInstructionData)
+            }
         }
         ConfidentialTransferInstruction::ApplyPendingBalance => {
             msg!("ConfidentialTransferInstruction::ApplyPendingBalance");