Make fuzzer aware of UD0_COMPAT mode

zyantific · Jan 25, 2024 · c882bc6 · c882bc6
1 parent a8c754f
commit c882bc6
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tools/ZydisFuzzEncoder.c b/tools/ZydisFuzzEncoder.c
@@ -287,6 +287,16 @@ int ZydisFuzzTarget(ZydisStreamRead read_fn, void *stream_ctx)
         abort();
     }
 
+    if (request.mnemonic == ZYDIS_MNEMONIC_UD0 && request.operand_count == 0)
+    {
+        status = ZydisDecoderEnableMode(&decoder, ZYDIS_DECODER_MODE_UD0_COMPAT, ZYAN_TRUE);
+        if (!ZYAN_SUCCESS(status))
+        {
+            fputs("Failed to enable UD0_COMPAT mode\n", ZYAN_STDERR);
+            abort();
+        }
+    }
+
     ZydisDecodedInstruction insn1;
     ZydisDecodedOperand operands1[ZYDIS_MAX_OPERAND_COUNT];
     status = ZydisDecoderDecodeFull(&decoder, encoded_instruction, encoded_length, &insn1,