diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/long_mode/mod.rs | 32 |
1 files changed, 14 insertions, 18 deletions
diff --git a/src/long_mode/mod.rs b/src/long_mode/mod.rs index 4d5cf59..533efa4 100644 --- a/src/long_mode/mod.rs +++ b/src/long_mode/mod.rs @@ -6466,9 +6466,7 @@ impl DecodeCtx { fn read_opc_hotpath< T: Reader<<Arch as yaxpeax_arch::Arch>::Address, <Arch as yaxpeax_arch::Arch>::Word>, S: DescriptionSink<FieldDescription>, ->(&mut self, words: &mut T, instruction: &mut Instruction, sink: &mut S) -> Result<Result<OperandCode, (u8, OpcodeRecord)>, DecodeError> { - let mut b = words.next().ok().ok_or(DecodeError::ExhaustedInput)?; - +>(&mut self, mut b: u8, nextb: &mut u8, record: &mut OpcodeRecord, words: &mut T, instruction: &mut Instruction, sink: &mut S) -> Result<bool, DecodeError> { if b >= 0x40 && b < 0x50 { sink.record((words.offset() - 1) as u32 * 8, (words.offset() - 1) as u32 * 8 + 7, FieldDescription { desc: InnerDescription::RexPrefix(b), @@ -6480,7 +6478,7 @@ fn read_opc_hotpath< } self.rb_size = RegisterBank::rB; b = words.next().ok().ok_or(DecodeError::ExhaustedInput)?; - record = unsafe { + *record = unsafe { core::ptr::read_volatile(&OPCODES[b as usize]) }; } else if b == 0x66 { @@ -6489,24 +6487,20 @@ fn read_opc_hotpath< id: words.offset() as u32 * 8 - 8, }); b = words.next().ok().ok_or(DecodeError::ExhaustedInput)?; - record = unsafe { + *record = unsafe { core::ptr::read_volatile(&OPCODES[b as usize]) }; instruction.prefixes.set_operand_size(); self.vqp_size = RegisterBank::W; - } else { - record = unsafe { - core::ptr::read_volatile(&OPCODES[b as usize]) - }; } if let Interpretation::Instruction(opc) = record.0 { record_opcode_record_found(words, sink, opc, record.1, 1); instruction.opcode = opc; - return Ok(Ok(record.1)); + return Ok(true); } else if b == 0x0f { let b = words.next().ok().ok_or(DecodeError::ExhaustedInput)?; - let (record, len) = if b == 0x38 { + let (r, len) = if b == 0x38 { let b = words.next().ok().ok_or(DecodeError::ExhaustedInput)?; (self.read_0f38_opcode(b, &mut instruction.prefixes), 3) } else if b == 0x3a { @@ -6515,15 +6509,17 @@ fn read_opc_hotpath< } else { (self.read_0f_opcode(b, &mut instruction.prefixes), 2) }; + *record = r; if let Interpretation::Instruction(opc) = record.0 { record_opcode_record_found(words, sink, opc, record.1, len); instruction.opcode = opc; } else { unsafe { unreachable_unchecked(); } } - return Ok(Ok(record.1)); + return Ok(true); } else { - return Ok(Err((b, record))); + *nextb = b; + return Ok(false); } } @@ -6533,6 +6529,8 @@ fn read_with_annotations< S: DescriptionSink<FieldDescription>, >(mut self, decoder: &InstDecoder, words: &mut T, instruction: &mut Instruction, sink: &mut S) -> Result<(), DecodeError> { words.mark(); + let mut nextb = words.next().ok().ok_or(DecodeError::ExhaustedInput)?; + let mut next_rec = OPCODES[nextb as usize]; instruction.prefixes = Prefixes::new(0); // const RAXRAXRAXRAX: [RegSpec; 4] = [RegSpec::rax(); 4]; @@ -6541,10 +6539,9 @@ fn read_with_annotations< instruction.regs[1] = RegSpec::rax(); instruction.regs[2] = RegSpec::rax(); - let hotpath_res = self.read_opc_hotpath(words, instruction, sink)?; - let record: OperandCode = match hotpath_res { - Ok(code) => code, - Err((mut nextb, mut next_rec)) => { + let record: OperandCode = if self.read_opc_hotpath(nextb, &mut nextb, &mut next_rec, words, instruction, sink)? { + next_rec.1 + } else { let prefixes = &mut instruction.prefixes; let record = loop { let mut record = next_rec; @@ -6687,7 +6684,6 @@ fn read_with_annotations< } record.1 - } }; self.read_operands(decoder, words, instruction, record, sink)?; |