xmr: output index check added in step6

ph4r05 · Oct 17, 2018 · c27ae90 · c27ae90
1 parent 973c457
commit c27ae90
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/apps/monero/protocol/signing/step_06_set_output.py b/src/apps/monero/protocol/signing/step_06_set_output.py
@@ -103,6 +103,8 @@ async def set_output(state: State, dst_entr, dst_entr_hmac, rsig_data):
 async def _validate(state: State, dst_entr, dst_entr_hmac):
     if state.current_input_index + 1 != state.input_count:
         raise ValueError("Invalid number of inputs")
+    if state.current_output_index >= state.output_count:
+        raise ValueError("Invalid output index")
     if dst_entr.amount <= 0:
         raise ValueError("Destination with wrong amount: %s" % dst_entr.amount)