From df67ba2b415d9157e4b56723c0b92695a4124ea0 Mon Sep 17 00:00:00 2001 From: iximeow Date: Sun, 23 Jun 2024 10:48:30 -0700 Subject: actually use new can_rep in 32b and 16b modes --- src/protected_mode/display.rs | 2 +- src/real_mode/display.rs | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) (limited to 'src') diff --git a/src/protected_mode/display.rs b/src/protected_mode/display.rs index ed5d902..9cc949d 100644 --- a/src/protected_mode/display.rs +++ b/src/protected_mode/display.rs @@ -3376,7 +3376,7 @@ fn contextualize_intel(instr: &Instruction, colors: } if instr.prefixes.rep_any() { - if [Opcode::MOVS, Opcode::CMPS, Opcode::LODS, Opcode::STOS, Opcode::INS, Opcode::OUTS].contains(&instr.opcode) { + if instr.opcode.can_rep() { if instr.prefixes.rep() { write!(out, "rep ")?; } else if instr.prefixes.repnz() { diff --git a/src/real_mode/display.rs b/src/real_mode/display.rs index 9332011..b06e3d2 100644 --- a/src/real_mode/display.rs +++ b/src/real_mode/display.rs @@ -3376,7 +3376,7 @@ fn contextualize_intel(instr: &Instruction, colors: } if instr.prefixes.rep_any() { - if [Opcode::MOVS, Opcode::CMPS, Opcode::LODS, Opcode::STOS, Opcode::INS, Opcode::OUTS].contains(&instr.opcode) { + if instr.opcode.can_rep() { if instr.prefixes.rep() { write!(out, "rep ")?; } else if instr.prefixes.repnz() { -- cgit v1.1