1.0

src/Assembler.h

@@ -0,0 +1,60 @@
+#pragma once
+#ifndef __ASSEMBLER_H__
+#define __ASSEMBLER_H__
+#include <vector>
+#include "asmjit/asmjit.h"
+
+using namespace asmjit;
+using namespace x86;
+class CAssembler
+{
+public:
+	CAssembler(Arch arch = Arch::kX86) :
+		m_Environment(arch)
+	{
+		Prepare();
+	}
+
+	~CAssembler() {}
+public:
+	void Prepare()
+	{
+		m_CodeHolder.reset();
+		m_CodeHolder.init(m_Environment, 0);
+		m_CodeHolder.attach(&m_Assembler);
+		//m_Assembler.addDiagnosticOptions(asmjit::DiagnosticOptions::kValidateAssembler);
+	}
+
+	Assembler* GetAssembler()
+	{
+		return &m_Assembler;
+	}
+
+	Section* GetEmitCode()
+	{
+		return m_CodeHolder.textSection();
+	}
+
+	std::vector <unsigned char> GetBytes(bool clear = true)
+	{
+		Section* pSection = this->GetEmitCode();
+		std::vector <unsigned char>  AsmCode;
+		AsmCode.resize(pSection->bufferSize());
+		memcpy(AsmCode.data(), pSection->data(), pSection->bufferSize());
+		if (clear) this->Prepare();
+		return AsmCode;
+	}
+
+
+protected:
+
+
+private:
+	CodeHolder m_CodeHolder;
+	Environment m_Environment;
+	Assembler m_Assembler;
+};
+
+
+#endif // __ASSEMBLER_H__
+