Zk instructions check length (#24103)

* 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
2022-04-05 09:40:45 -03:00
parent 2b718d00b0
commit ba92ba0e06
2 changed files with 2 additions and 2 deletions
--- a/zk-token-sdk/src/zk_token_proof_instruction.rs
+++ b/zk-token-sdk/src/zk_token_proof_instruction.rs
@ -73,7 +73,7 @@ impl ProofInstruction {
    }

    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> {