avoid marker

solana-labs · Jul 14, 2021 · 795b81f · 795b81f
1 parent e6f637b
commit 795b81f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sdk/src/compute_budget.rs b/sdk/src/compute_budget.rs
@@ -42,9 +42,9 @@ pub fn process_request(
     compute_budget: &mut BpfComputeBudget,
     tx: &Transaction,
 ) -> Result<(), TransactionError> {
-    if !tx.message().instructions.is_empty() {
-        let error = TransactionError::InstructionError(0, InstructionError::InvalidInstructionData);
-        let instruction = &tx.message().instructions[0];
+    let error = TransactionError::InstructionError(0, InstructionError::InvalidInstructionData);
+    // Compute budget instruction must be in 1st or 2nd instruction (avoid nonce marker)
+    for instruction in tx.message().instructions.iter().take(2) {
         if check_id(instruction.program_id(&tx.message().account_keys)) {
             let ComputeBudgetInstruction::RequestUnits(units) =
                 try_from_slice_unchecked::<ComputeBudgetInstruction>(&instruction.data)