gaspersic/solana
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Zk instructions check length (#24103)

Browse Source 
* zk-token-sdk: add a length check before decoding proof instruction

* zk-token-sdk: fix minor spelling

* zk-token-sdk: one-liner for length check

* zk-token-sdk: one-liner fix
This commit is contained in:
samkim-crypto
2022-04-05 09:40:45 -03:00
committed by GitHub
parent 2b718d00b0
commit ba92ba0e06
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
programs/zk-token-proof/src/lib.rs
View File

@ -56,7 +56,7 @@ pub fn process_instruction(
verify::<WithdrawData>(invoke_context) verify::<WithdrawData>(invoke_context)
} }
ProofInstruction::VerifyWithdrawWithheldTokens => { ProofInstruction::VerifyWithdrawWithheldTokens => {
ic_msg!(invoke_context, "VerifyWithdraw"); ic_msg!(invoke_context, "VerifyWithdrawWithheldTokens");
verify::<WithdrawData>(invoke_context) verify::<WithdrawData>(invoke_context)
} }
ProofInstruction::VerifyTransfer => { ProofInstruction::VerifyTransfer => {

2
zk-token-sdk/src/zk_token_proof_instruction.rs
View File

@ -73,7 +73,7 @@ impl ProofInstruction {
} }
pub fn decode_type(input: &[u8]) -> Option<Self> { pub fn decode_type(input: &[u8]) -> Option<Self> {
FromPrimitive::from_u8(input[0]) input.get(0).and_then(|x| FromPrimitive::from_u8(*x))
} }
pub fn decode_data<T: Pod>(input: &[u8]) -> Option<&T> { pub fn decode_data<T: Pod>(input: &[u8]) -> Option<&T> {