Add API to execute code on the main thread

1 files changed, 22 insertions, 0 deletions

diff --git a/binaryninjaapi.h b/binaryninjaapi.h
index c8d54c18..d3234dcc 100644
--- a/binaryninjaapi.h
+++ b/binaryninjaapi.h
@@ -273,6 +273,8 @@ namespace BinaryNinja
 	class Platform;
 	class Type;
 	class DataBuffer;
+	class MainThreadAction;
+	class MainThreadActionHandler;
 
 	/*! Logs to the error console with the given BNLogLevel.
 
@@ -365,6 +367,10 @@ namespace BinaryNinja
 	                Type** outType,
 	                std::vector<std::string>& outVarName);
 
+	void RegisterMainThread(MainThreadActionHandler* handler);
+	Ref<MainThreadAction> ExecuteOnMainThread(const std::function<void()>& action);
+	void ExecuteOnMainThreadAndWait(const std::function<void()>& action);
+
 	class DataBuffer
 	{
 		BNDataBuffer* m_buffer;
@@ -2153,4 +2159,20 @@ namespace BinaryNinja
 		CoreScriptingProvider(BNScriptingProvider* provider);
 		virtual Ref<ScriptingInstance> CreateNewInstance() override;
 	};
+
+	class MainThreadAction: public CoreRefCountObject<BNMainThreadAction,
+		BNNewMainThreadActionReference, BNFreeMainThreadAction>
+	{
+	public:
+		MainThreadAction(BNMainThreadAction* action);
+		void Execute();
+		bool IsDone() const;
+		void Wait();
+	};
+
+	class MainThreadActionHandler
+	{
+	public:
+		virtual void AddMainThreadAction(MainThreadAction* action) = 0;
+	};
 }