From c5c93fc82b8929d04f62d241ca50228de60fa5f4 Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Fri, 13 Jul 2018 18:43:34 -0400
Subject: Add ability to update custom flow graphs

---
 binaryninjaapi.h | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

(limited to 'binaryninjaapi.h')

diff --git a/binaryninjaapi.h b/binaryninjaapi.h
index 1133625e..dd685d2b 100644
--- a/binaryninjaapi.h
+++ b/binaryninjaapi.h
@@ -2585,8 +2585,11 @@ namespace BinaryNinja
 		static void PrepareForLayoutCallback(void* ctxt);
 		static void PopulateNodesCallback(void* ctxt);
 		static void CompleteLayoutCallback(void* ctxt);
+		static BNFlowGraph* UpdateCallback(void* ctxt);
 
 	protected:
+		FlowGraph(BNFlowGraph* graph);
+
 		void FinishPrepareForLayout();
 		virtual void PrepareForLayout();
 		virtual void PopulateNodes();
@@ -2594,7 +2597,6 @@ namespace BinaryNinja
 
 	public:
 		FlowGraph();
-		FlowGraph(BNFlowGraph* graph);
 		~FlowGraph();
 
 		BNFlowGraph* GetGraphObject() const { return m_graph; }
@@ -2629,6 +2631,15 @@ namespace BinaryNinja
 		void SetMediumLevelILFunction(MediumLevelILFunction* func);
 
 		void Show(const std::string& title);
+
+		virtual Ref<FlowGraph> Update();
+	};
+
+	class CoreFlowGraph: public FlowGraph
+	{
+	public:
+		CoreFlowGraph(BNFlowGraph* graph);
+		virtual Ref<FlowGraph> Update() override;
 	};
 
 	struct LowLevelILLabel: public BNLowLevelILLabel
-- 
cgit v1.3.1