added an overloaded vm::handler::get_profile function which takes a

mnemonic

include/vmprofiler.hpp

@@ -116,6 +116,7 @@ namespace vm
         // can be used on calc_jmp...
         bool get_operand_transforms( const zydis_routine_t &vm_handler, transform::map_t &transforms );
         vm::handler::profile_t *get_profile( handler_t &vm_handler );
+        vm::handler::profile_t *get_profile( vm::handler::mnemonic_t mnemonic );
 
         namespace table
         {

src/vmhandler.cpp

@@ -196,7 +196,7 @@ namespace vm
                 transform_instr =
                     std::find_if( ++transform_instr, vm_handler.end(), []( const zydis_instr_t &instr_data ) -> bool {
                         return vm::transform::valid( instr_data.instr.mnemonic ) &&
-                               instr_data.instr.operands[0].actions & ZYDIS_OPERAND_ACTION_WRITE &&
+                               instr_data.instr.operands[ 0 ].actions & ZYDIS_OPERAND_ACTION_WRITE &&
                                util::reg::compare( instr_data.instr.operands[ 0 ].reg.value, ZYDIS_REGISTER_RAX );
                     } );
 
@@ -219,8 +219,10 @@ namespace vm
 
                 for ( auto &instr : vprofile->signature )
                 {
-                    contains = std::find_if(contains, vm_handler->instrs.end(),
+                    contains =
-                        [ & ]( zydis_instr_t &instr_data ) -> bool { return instr( instr_data.instr ); } );
+                        std::find_if( contains, vm_handler->instrs.end(), [ & ]( zydis_instr_t &instr_data ) -> bool {
+                            return instr( instr_data.instr );
+                        } );
 
                     if ( contains == vm_handler->instrs.end() )
                         return false;
@@ -236,6 +238,15 @@ namespace vm
             return nullptr;
         }
 
+        vm::handler::profile_t *get_profile( vm::handler::mnemonic_t mnemonic )
+        {
+            auto result = std::find_if(
+                vm::handler::profile::all.begin(), vm::handler::profile::all.end(),
+                [ & ]( vm::handler::profile_t *profile ) -> bool { return profile->mnemonic == mnemonic; } );
+
+            return result != vm::handler::profile::all.end() ? *result : nullptr;
+        }
+
         namespace table
         {
             std::uintptr_t *get( const zydis_routine_t &vm_entry )