Fix memory leak in base address detection

1 files changed, 16 insertions, 0 deletions

diff --git a/examples/triage/baseaddress.cpp b/examples/triage/baseaddress.cpp
index cb8f5176..436066eb 100644
--- a/examples/triage/baseaddress.cpp
+++ b/examples/triage/baseaddress.cpp
@@ -29,6 +29,22 @@ string BaseAddressDetectionConfidenceToString(BNBaseAddressDetectionConfidence l
 }
 
 
+BaseAddressDetectionThread::BaseAddressDetectionThread(BaseAddressDetectionQtInputs* widgetInputs,
+	BinaryNinja::Ref<BinaryNinja::BinaryView> bv)
+{
+	m_inputs = widgetInputs;
+	m_view = bv;
+	m_baseDetection = new BinaryNinja::BaseAddressDetection(m_view);
+}
+
+
+BaseAddressDetectionThread::~BaseAddressDetectionThread()
+{
+	if (m_baseDetection)
+		delete m_baseDetection;
+}
+
+
 void BaseAddressDetectionThread::run()
 {
 	BaseAddressDetectionQtResults results;