Add ZYDIS_ATTRIB_HAS_EEVEX

zyantific · Oct 29, 2024 · c105172 · c105172
1 parent 7e0ec6f
commit c105172
Show file tree

Hide file tree

Showing 5 changed files with 6,114 additions and 6,095 deletions.
diff --git a/include/Zydis/Internal/SharedData.h b/include/Zydis/Internal/SharedData.h
@@ -859,6 +859,9 @@ typedef struct ZydisInstructionDefinitionEVEX_
     ZyanU8 mask_override                   ZYAN_BITFIELD(ZYDIS_MASK_OVERRIDE_REQUIRED_BITS);
     ZyanU8 broadcast                       ZYAN_BITFIELD(ZYDIS_EVEX_STATIC_BROADCAST_REQUIRED_BITS);
 #endif
+    ZyanU8 is_eevex                        ZYAN_BITFIELD( 1);
+    /*ZyanU8 accepts_zu                      ZYAN_BITFIELD( 1);
+    ZyanU8 accepts_nf                      ZYAN_BITFIELD( 1);*/
 } ZydisInstructionDefinitionEVEX;
 #endif
 

diff --git a/include/Zydis/SharedTypes.h b/include/Zydis/SharedTypes.h
@@ -731,6 +731,10 @@ typedef ZyanU64 ZydisInstructionAttributes;
  * The instruction uses the APX source condition code.
  */
 #define ZYDIS_ATTRIB_HAS_SCC                    (1ULL << 47)
+/**
+* The instruction has the `EEVEX` (extended `EVEX`) prefix.
+*/
+#define ZYDIS_ATTRIB_HAS_EEVEX                  (1ULL << 48) // TODO: rebase
 
 // TODO: PPX Hint
 

diff --git a/src/Decoder.c b/src/Decoder.c
@@ -5188,6 +5188,17 @@ static ZyanStatus ZydisDecodeInstruction(ZydisDecoderState* state,
                     ZydisGetInstructionDefinition(instruction->encoding, node->value, &definition);
                 }
 
+                if (instruction->encoding == ZYDIS_INSTRUCTION_ENCODING_EVEX)
+                {
+                    const ZydisInstructionDefinitionEVEX* evex_definition = 
+                        (const ZydisInstructionDefinitionEVEX*)definition;
+
+                    if (evex_definition->is_eevex)
+                    {
+                        instruction->attributes |= ZYDIS_ATTRIB_HAS_EEVEX;
+                    }
+                }
+
                 instruction->mnemonic = definition->mnemonic;
 
 #ifndef ZYDIS_MINIMAL_MODE