Add endian argument to registering archs with views

1 files changed, 6 insertions, 6 deletions

diff --git a/binaryviewtype.cpp b/binaryviewtype.cpp
index 12883bd6..e7feb68c 100644
--- a/binaryviewtype.cpp
+++ b/binaryviewtype.cpp
@@ -106,24 +106,24 @@ vector<Ref<BinaryViewType>> BinaryViewType::GetViewTypesForData(BinaryView* data
 }
 
 
-void BinaryViewType::RegisterArchitecture(const string& name, uint32_t id, Architecture* arch)
+void BinaryViewType::RegisterArchitecture(const string& name, uint32_t id, BNEndianness endian, Architecture* arch)
 {
 	Ref<BinaryViewType> type = BinaryViewType::GetByName(name);
 	if (!type)
 		return;
-	type->RegisterArchitecture(id, arch);
+	type->RegisterArchitecture(id, endian, arch);
 }
 
 
-void BinaryViewType::RegisterArchitecture(uint32_t id, Architecture* arch)
+void BinaryViewType::RegisterArchitecture(uint32_t id, BNEndianness endian, Architecture* arch)
 {
-	BNRegisterArchitectureForViewType(m_object, id, arch->GetObject());
+	BNRegisterArchitectureForViewType(m_object, id, endian, arch->GetObject());
 }
 
 
-Ref<Architecture> BinaryViewType::GetArchitecture(uint32_t id)
+Ref<Architecture> BinaryViewType::GetArchitecture(uint32_t id, BNEndianness endian)
 {
-	BNArchitecture* arch = BNGetArchitectureForViewType(m_object, id);
+	BNArchitecture* arch = BNGetArchitectureForViewType(m_object, id, endian);
 	if (!arch)
 		return nullptr;
 	return new CoreArchitecture(arch);