Adapt to changes to native program entrypoint

This commit is contained in:
Michael Vines 2021-10-21 10:49:29 -07:00
parent 9c88f8205b
commit f2e7f0c32b
2 changed files with 7 additions and 9 deletions

View File

@ -14,4 +14,5 @@ getrandom = { version = "0.1", features = ["dummy"] }
num-derive = "0.3"
num-traits = "0.2"
solana-sdk = "=1.9.0"
solana-program-runtime = "=1.9.0"
spl-zk-token-sdk = { path = "../sdk" }

View File

@ -2,16 +2,15 @@
use {
bytemuck::Pod,
solana_sdk::{
ic_msg, instruction::InstructionError, process_instruction::InvokeContext, pubkey::Pubkey,
},
solana_program_runtime::{ic_msg, invoke_context::InvokeContext},
solana_sdk::instruction::InstructionError,
spl_zk_token_sdk::zk_token_proof_instruction::*,
std::result::Result,
};
fn verify<T: Pod + Verifiable>(
input: &[u8],
invoke_context: &mut dyn InvokeContext,
invoke_context: &mut InvokeContext,
) -> Result<(), InstructionError> {
let proof = ProofInstruction::decode_data::<T>(input).ok_or_else(|| {
ic_msg!(invoke_context, "invalid proof data");
@ -25,9 +24,9 @@ fn verify<T: Pod + Verifiable>(
}
pub fn process_instruction(
program_id: &Pubkey,
_first_instruction_account: usize,
input: &[u8],
invoke_context: &mut dyn InvokeContext,
invoke_context: &mut InvokeContext,
) -> Result<(), InstructionError> {
if invoke_context.invoke_depth() != 1 {
// Not supported as an inner instruction
@ -40,9 +39,7 @@ pub fn process_instruction(
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)?
{
match ProofInstruction::decode_type(input).ok_or(InstructionError::InvalidInstructionData)? {
ProofInstruction::VerifyCloseAccount => {
ic_msg!(invoke_context, "VerifyCloseAccount");
verify::<CloseAccountData>(input, invoke_context)