1 files changed, 95 insertions, 0 deletions
diff --git a/architecture.cpp b/architecture.cpp
new file mode 100644
index 00000000..60fa4efa
--- /dev/null
+++ b/architecture.cpp
@@ -0,0 +1,95 @@
+#include "binaryninjaapi.h"
+
+using namespace BinaryNinja;
+using namespace std;
+
+
+InstructionInfo::InstructionInfo()
+{
+	length = 0;
+	branchCount = 0;
+}
+
+
+void InstructionInfo::AddBranch(BNBranchType type, uint64_t target)
+{
+	if (branchCount >= BN_MAX_INSTRUCTION_BRANCHES)
+		return;
+	branchType[branchCount] = type;
+	branchTarget[branchCount++] = target;
+}
+
+
+Architecture::Architecture(BNArchitecture* arch): m_arch(arch)
+{
+}
+
+
+Architecture::Architecture(const string& name): m_nameForRegister(name)
+{
+}
+
+
+bool Architecture::GetInstructionInfoCallback(void* ctxt, BNBinaryView* data, uint64_t addr, BNInstructionInfo* result)
+{
+	Architecture* arch = (Architecture*)ctxt;
+	Ref<BinaryView> view = new BinaryView(BNNewViewReference(data));
+
+	InstructionInfo info;
+	bool ok = arch->GetInstructionInfo(view, addr, info);
+	*result = info;
+	return ok;
+}
+
+
+void Architecture::Register(Architecture* arch)
+{
+	BNCustomArchitecture callbacks;
+	callbacks.context = arch;
+	callbacks.getInstructionInfo = GetInstructionInfoCallback;
+	arch->m_arch = BNRegisterArchitecture(arch->m_nameForRegister.c_str(), &callbacks);
+}
+
+
+Ref<Architecture> Architecture::GetByName(const string& name)
+{
+	BNArchitecture* arch = BNGetArchitectureByName(name.c_str());
+	if (!arch)
+		return nullptr;
+	return new CoreArchitecture(arch);
+}
+
+
+vector<Ref<Architecture>> Architecture::GetList()
+{
+	BNArchitecture** archs;
+	size_t count;
+	archs = BNGetArchitectureList(&count);
+
+	vector<Ref<Architecture>> result;
+	for (size_t i = 0; i < count; i++)
+		result.push_back(new CoreArchitecture(BNNewArchitectureReference(archs[i])));
+
+	BNFreeArchitectureList(archs, count);
+	return result;
+}
+
+
+string Architecture::GetName() const
+{
+	char* name = BNGetArchitectureName(m_arch);
+	string result = name;
+	BNFreeString(name);
+	return result;
+}
+
+
+CoreArchitecture::CoreArchitecture(BNArchitecture* arch): Architecture(arch)
+{
+}
+
+
+bool CoreArchitecture::GetInstructionInfo(BinaryView* view, uint64_t addr, InstructionInfo& result)
+{
+	return BNGetInstructionInfo(m_arch, view->GetViewObject(), addr, &result);
+}