aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/protected_mode/display.rs2
-rw-r--r--src/real_mode/display.rs2
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() {