xmr: sign step 03 - permutation length check added

ph4r05 · Oct 3, 2018 · 8fa4066 · 8fa4066
1 parent 55a593a
commit 8fa4066
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/apps/monero/protocol/signing/step_03_inputs_permutation.py b/src/apps/monero/protocol/signing/step_03_inputs_permutation.py
@@ -26,8 +26,11 @@ async def tsx_inputs_permutation(state: State, permutation: list):
     """
     Set permutation on the inputs - sorted by key image on host.
     """
-    state.source_permutation = permutation
+    if len(permutation) != state.input_count:
+        raise ValueError('Invalid permutation size')
     _check_permutation(permutation)
+
+    state.source_permutation = permutation
     state.current_input_index = -1
 
     return MoneroTransactionInputsPermutationAck()