diff options
| -rw-r--r-- | src/protected_mode/display.rs | 2 | ||||
| -rw-r--r-- | src/real_mode/display.rs | 2 |
2 files changed, 2 insertions, 2 deletions
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<T: fmt::Write, Y: YaxColors>(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<T: fmt::Write, Y: YaxColors>(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() { |