#pragma once
#include <transform.hpp>

namespace vm
{
	namespace handler
	{
		using instr_callback_t = bool(*)(const zydis_decoded_instr_t& instr);

		enum mnemonic_t
		{
			INVALID,
			PUSHVSP,
			SHRQ,
			MULQ,
			DIVQ,
			JMP,
			VMEXIT,

			SREGQ,
			SREGDW,
			SREGW,

			LREGQ,
			LREGDW,

			LCONSTQ,
			LCONSTBZXW,
			LCONSTBSXDW,
			LCONSTDWSXQ,
			LCONSTWSXQ,
			LCONSTDW,

			READQ,
			READDW,
			READW,

			WRITEQ,
			WRITEDW,
			WRITEW,

			ADDQ,
			ADDDW,

			SHLQ,
			SHLDW,

			NANDQ,
			NANDDW
		};

        enum extention_t
        {
            none,
            sign_extend,
            zero_extend
        };

        struct profile_t
        {
            const char* name;
			mnemonic_t mnemonic;
            u8 imm_size;
			std::vector<instr_callback_t> signature;
            extention_t extention;
        };

		namespace profile
		{
			extern vm::handler::profile_t sregq;
			extern vm::handler::profile_t sregdw;
			extern vm::handler::profile_t sregw;

			extern vm::handler::profile_t lregq;
			extern vm::handler::profile_t lregdw;

			extern vm::handler::profile_t lconstq;
			extern vm::handler::profile_t lconstbzxw;
			extern vm::handler::profile_t lconstbsxdw;
			extern vm::handler::profile_t lconstdwsxq;
			extern vm::handler::profile_t lconstwsxq;
			extern vm::handler::profile_t lconstdw;

			extern vm::handler::profile_t addq;
			extern vm::handler::profile_t adddw;

			extern vm::handler::profile_t shlq;
			extern vm::handler::profile_t shldw;

			extern vm::handler::profile_t nandq;
			extern vm::handler::profile_t nanddw;

			extern vm::handler::profile_t writeq;
			extern vm::handler::profile_t writedw;

			extern vm::handler::profile_t shrq;
			extern vm::handler::profile_t pushvsp;
			extern vm::handler::profile_t mulq;
			extern vm::handler::profile_t divq;
			extern vm::handler::profile_t jmp;
			extern vm::handler::profile_t readq;
			extern vm::handler::profile_t vmexit;

			inline std::vector<vm::handler::profile_t*> all =
			{
				&sregq, &sregdw, &sregw,
				&lregq, &lregdw,
				&lconstq, &lconstbzxw, &lconstbsxdw, &lconstdwsxq, &lconstwsxq, &lconstdw,
				&addq, &adddw,
				&shlq, &shldw,
				&writeq, &writedw,
				&nandq, &nanddw,

				&shrq,
				&readq,
				&mulq,
				&pushvsp,
				&divq,
				&jmp,
				&vmexit
			};
		}
	}
}