zk-token-sdk: add verify transfer with fee instruction (#22924)

programs/zk-token-proof/src/lib.rs

@@ -53,5 +53,9 @@ pub fn process_instruction(
             ic_msg!(invoke_context, "VerifyTransfer");
             verify::<TransferData>(input, invoke_context)
         }
+        ProofInstruction::VerifyTransferWithFee => {
+            ic_msg!(invoke_context, "VerifyTransferWithFee");
+            verify::<TransferWithFeeData>(input, invoke_context)
+        }
     }
 }

zk-token-sdk/src/instruction/mod.rs

@@ -14,7 +14,10 @@ use {
     },
     curve25519_dalek::scalar::Scalar,
 };
-pub use {close_account::CloseAccountData, transfer::TransferData, withdraw::WithdrawData};
+pub use {
+    close_account::CloseAccountData, transfer::TransferData,
+    transfer_with_fee::TransferWithFeeData, withdraw::WithdrawData,
+};
 
 /// Constant for 2^32
 #[cfg(not(target_arch = "bpf"))]

zk-token-sdk/src/zk_token_proof_instruction.rs

@@ -39,6 +39,16 @@ pub enum ProofInstruction {
     ///   `TransferData`
     ///
     VerifyTransfer,
+
+    /// Verify a `TransferWithFeeData` struct
+    ///
+    /// Accounts expected by this instruction:
+    ///   None
+    ///
+    /// Data expected by this instruction:
+    ///   `TransferWithFeeData`
+    ///
+    VerifyTransferWithFee,
 }
 
 impl ProofInstruction {
@@ -76,3 +86,7 @@ pub fn verify_withdraw(proof_data: &WithdrawData) -> Instruction {
 pub fn verify_transfer(proof_data: &TransferData) -> Instruction {
     ProofInstruction::VerifyTransfer.encode(proof_data)
 }
+
+pub fn verify_transfer_with_fee(proof_data: &TransferWithFeeData) -> Instruction {
+    ProofInstruction::VerifyTransferWithFee.encode(proof_data)
+}