From 4a1d5c29b5baec20e363a976fb9bc1fc7c1b0ca8 Mon Sep 17 00:00:00 2001
From: Peter LaFosse <peter@vector35.com>
Date: Sun, 18 Nov 2018 14:05:34 -0500
Subject: Add Find APIs with progress

---
 binaryview.cpp | 40 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

(limited to 'binaryview.cpp')

diff --git a/binaryview.cpp b/binaryview.cpp
index 984edda8..afcb4560 100644
--- a/binaryview.cpp
+++ b/binaryview.cpp
@@ -2001,6 +2001,46 @@ bool BinaryView::FindNextConstant(uint64_t start, uint64_t constant, uint64_t& r
 }
 
 
+struct FindProgressCallbackContext
+{
+	std::function<bool(size_t, size_t)> func;
+};
+
+
+static bool FindProgressCallback(void* ctxt, size_t progress, size_t total)
+{
+	FindProgressCallbackContext* cb = (FindProgressCallbackContext*)ctxt;
+	return cb->func(progress, total);
+}
+
+
+bool BinaryView::FindNextData(uint64_t start, uint64_t end, const DataBuffer& data, uint64_t& addr, BNFindFlag flags,
+	const std::function<bool(size_t current, size_t total)>& progress)
+{
+	FindProgressCallbackContext fp;
+	fp.func = progress;
+	return BNFindNextDataWithProgress(m_object, start, end, data.GetBufferObject(), &addr, flags, &fp, FindProgressCallback);
+}
+
+
+bool BinaryView::FindNextText(uint64_t start, uint64_t end, const std::string& data, uint64_t& addr, Ref<DisassemblySettings> settings,
+	BNFindFlag flags, const std::function<bool(size_t current, size_t total)>& progress)
+{
+	FindProgressCallbackContext fp;
+	fp.func = progress;
+	return BNFindNextTextWithProgress(m_object, start, end, data.c_str(), &addr, settings->GetObject(), flags, &fp, FindProgressCallback);
+}
+
+
+bool BinaryView::FindNextConstant(uint64_t start, uint64_t end, uint64_t constant, uint64_t& addr, Ref<DisassemblySettings> settings,
+	const std::function<bool(size_t current, size_t total)>& progress)
+{
+	FindProgressCallbackContext fp;
+	fp.func = progress;
+	return BNFindNextConstantWithProgress(m_object, start, end, constant, &addr, settings->GetObject(), &fp, FindProgressCallback);
+}
+
+
 void BinaryView::Reanalyze()
 {
 	BNReanalyzeAllFunctions(m_object);
-- 
cgit v1.3.1