From e1b47b2a23c4edd48f8b267a6d76bf8ee61a7454 Mon Sep 17 00:00:00 2001
From: mappzor <34216645+mappzor@users.noreply.github.com>
Date: Wed, 6 Nov 2024 02:16:32 +0100
Subject: [PATCH] Workaround for `xsha1`/`xsha256`

---
 tools/ZydisFuzzReEncoding.c | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/tools/ZydisFuzzReEncoding.c b/tools/ZydisFuzzReEncoding.c
index e0c59247..1f891536 100644
--- a/tools/ZydisFuzzReEncoding.c
+++ b/tools/ZydisFuzzReEncoding.c
@@ -89,6 +89,14 @@ int ZydisFuzzTarget(ZydisStreamRead read_fn, void *stream_ctx)
     {
         return EXIT_FAILURE;
     }
+    // TODO: Temporary workaround for `xsha1` and `xsha256` with OSIZE prefix
+    if (insn1.mnemonic == ZYDIS_MNEMONIC_XSHA1 || insn1.mnemonic == ZYDIS_MNEMONIC_XSHA256)
+    {
+        if (insn1.attributes & ZYDIS_ATTRIB_HAS_OPERANDSIZE)
+        {
+            return EXIT_SUCCESS;
+        }
+    }
 
     ZydisReEncodeInstruction(&decoder, &insn1, operands1, insn1.operand_count_visible, buffer);