From e878199be4a74c880acd5d38c33965b47d7630d0 Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Thu, 25 Aug 2016 20:42:33 -0400
Subject: Adding APIs for database save/load progress and constant references

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

(limited to 'binaryview.cpp')

diff --git a/binaryview.cpp b/binaryview.cpp
index cff3ed10..14393930 100644
--- a/binaryview.cpp
+++ b/binaryview.cpp
@@ -518,12 +518,25 @@ bool BinaryView::CreateDatabase(const string& path)
 }
 
 
+bool BinaryView::CreateDatabase(const string& path,
+	const function<void(size_t progress, size_t total)>& progressCallback)
+{
+	return m_file->CreateDatabase(path, this, progressCallback);
+}
+
+
 bool BinaryView::SaveAutoSnapshot()
 {
 	return m_file->SaveAutoSnapshot(this);
 }
 
 
+bool BinaryView::SaveAutoSnapshot(const function<void(size_t progress, size_t total)>& progressCallback)
+{
+	return m_file->SaveAutoSnapshot(this, progressCallback);
+}
+
+
 void BinaryView::BeginUndoActions()
 {
 	m_file->BeginUndoActions();
-- 
cgit v1.3.1


From 3978e18b09ca745fd08defb8f00fbece267beaf2 Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Tue, 30 Aug 2016 21:59:40 -0400
Subject: Add API to reanalyze

---
 binaryninjaapi.h   |  4 ++++
 binaryninjacore.h  |  3 +++
 binaryview.cpp     |  6 ++++++
 function.cpp       |  6 ++++++
 python/__init__.py | 16 ++++++++++++++++
 5 files changed, 35 insertions(+)

(limited to 'binaryview.cpp')

diff --git a/binaryninjaapi.h b/binaryninjaapi.h
index f60ee159..7c417623 100644
--- a/binaryninjaapi.h
+++ b/binaryninjaapi.h
@@ -919,6 +919,8 @@ namespace BinaryNinja
 		void UndefineUserType(const std::string& name);
 
 		bool FindNextData(uint64_t start, const DataBuffer& data, uint64_t& result, BNFindFlag flags = NoFindFlags);
+
+		void Reanalyze();
 	};
 
 	class BinaryData: public BinaryView
@@ -1672,6 +1674,8 @@ namespace BinaryNinja
 			size_t operand);
 		void SetIntegerConstantDisplayType(Architecture* arch, uint64_t instrAddr, uint64_t value, size_t operand,
 			BNIntegerDisplayType type);
+
+		void Reanalyze();
 	};
 
 	struct FunctionGraphEdge
diff --git a/binaryninjacore.h b/binaryninjacore.h
index c293f940..3aa0a1e6 100644
--- a/binaryninjacore.h
+++ b/binaryninjacore.h
@@ -1565,6 +1565,9 @@ extern "C"
 	BINARYNINJACOREAPI void BNUndefineAnalysisType(BNBinaryView* view, const char* name);
 	BINARYNINJACOREAPI void BNUndefineUserAnalysisType(BNBinaryView* view, const char* name);
 
+	BINARYNINJACOREAPI void BNReanalyzeAllFunctions(BNBinaryView* view);
+	BINARYNINJACOREAPI void BNReanalyzeFunction(BNFunction* func);
+
 	// Disassembly settings
 	BINARYNINJACOREAPI BNDisassemblySettings* BNCreateDisassemblySettings(void);
 	BINARYNINJACOREAPI BNDisassemblySettings* BNNewDisassemblySettingsReference(BNDisassemblySettings* settings);
diff --git a/binaryview.cpp b/binaryview.cpp
index 14393930..6631f3d6 100644
--- a/binaryview.cpp
+++ b/binaryview.cpp
@@ -1440,6 +1440,12 @@ bool BinaryView::FindNextData(uint64_t start, const DataBuffer& data, uint64_t&
 }
 
 
+void BinaryView::Reanalyze()
+{
+	BNReanalyzeAllFunctions(m_object);
+}
+
+
 BinaryData::BinaryData(FileMetadata* file): BinaryView(BNCreateBinaryDataView(file->GetObject()))
 {
 }
diff --git a/function.cpp b/function.cpp
index 78a33bda..cbd9ec40 100644
--- a/function.cpp
+++ b/function.cpp
@@ -572,3 +572,9 @@ void Function::SetIntegerConstantDisplayType(Architecture* arch, uint64_t instrA
 {
 	BNSetIntegerConstantDisplayType(m_object, arch->GetObject(), instrAddr, value, operand, type);
 }
+
+
+void Function::Reanalyze()
+{
+	BNReanalyzeFunction(m_object);
+}
diff --git a/python/__init__.py b/python/__init__.py
index e9f036ae..82aaba32 100644
--- a/python/__init__.py
+++ b/python/__init__.py
@@ -3231,6 +3231,14 @@ class BinaryView(object):
 			return None
 		return result.value
 
+	def reanalyze(self):
+		"""
+		``reanalyze`` causes all functions to be reanalyzed. This function does not wait for the analysis to finish.
+
+		:rtype: None
+		"""
+		core.BNReanalyzeAllFunctions(self.handle)
+
 	def __setattr__(self, name, value):
 		try:
 			object.__setattr__(self,name,value)
@@ -4677,6 +4685,14 @@ class Function(object):
 			display_type = core.BNIntegerDisplayType_by_name[display_type]
 		core.BNSetIntegerConstantDisplayType(self.handle, arch.handle, instr_addr, value, operand, display_type)
 
+	def reanalyze(self):
+		"""
+		``reanalyze`` causes this functions to be reanalyzed. This function does not wait for the analysis to finish.
+
+		:rtype: None
+		"""
+		core.BNReanalyzeFunction(self.handle)
+
 class BasicBlockEdge:
 	def __init__(self, branch_type, target, arch):
 		self.type = core.BNBranchType_names[branch_type]
-- 
cgit v1.3.1