Base address detection widget in Triage view

Initial implementation of base address detection UI widget in triage
summary

1 files changed, 13 insertions, 0 deletions

diff --git a/examples/triage/view.cpp b/examples/triage/view.cpp
index 87cdb61a..579f8917 100644
--- a/examples/triage/view.cpp
+++ b/examples/triage/view.cpp
@@ -10,6 +10,7 @@
 #include "librariesinfo.h"
 #include "headers.h"
 #include "strings.h"
+#include "baseaddress.h"
 #include "fontsettings.h"
 #include <binaryninjacore.h>
 
@@ -52,6 +53,18 @@ TriageView::TriageView(QWidget* parent, BinaryViewRef data) : QScrollArea(parent
 		delete hdr;
 	}
 
+	auto fileMetadata = m_data->GetFile();
+	auto existingViews = fileMetadata->GetExistingViews();
+	if ((existingViews.size() == 2 && fileMetadata->GetViewOfType("Mapped")) || existingViews.size() == 1)
+	{
+		// Binary either only has raw view (Open for triage mode) or raw and mapped view
+		QGroupBox* baseDetectionGroup = new QGroupBox("Base Address Detection", container);
+		QVBoxLayout* baseDetectionLayout = new QVBoxLayout();
+		baseDetectionLayout->addWidget(new BaseAddressDetectionWidget(this, data));
+		baseDetectionGroup->setLayout(baseDetectionLayout);
+		layout->addWidget(baseDetectionGroup);
+	}
+
 	QGroupBox* librariesGroup = new QGroupBox("Libraries", container);
 	QVBoxLayout* librariesLayout = new QVBoxLayout();
 	librariesLayout->addWidget(new LibrariesWidget(this, data));