From 9c8d4eac3dadf6273b34ba193c219de9283285b3 Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty.wagner@gmail.com>
Date: Tue, 3 Jun 2025 13:42:04 -0600
Subject: Add UI action and API to control early returns in HLIL

---
 ui/flowgraphwidget.h | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'ui/flowgraphwidget.h')

diff --git a/ui/flowgraphwidget.h b/ui/flowgraphwidget.h
index b69ec38b..d988da12 100644
--- a/ui/flowgraphwidget.h
+++ b/ui/flowgraphwidget.h
@@ -170,6 +170,8 @@ class BINARYNINJAUIAPI FlowGraphWidget :
 	BNExprFolding getCurrentExprFolding();
 	std::optional<uint64_t> getCurrentInvertableConditionAddress();
 	bool getCurrentConditionInverted();
+	std::optional<uint64_t> getCurrentEarlyReturnAddress();
+	BNEarlyReturn getCurrentEarlyReturn();
 	std::optional<std::pair<BinaryNinja::Variable, BinaryNinja::Variable>> getMergeVariablesAtCurrentLocation();
 
   protected:
@@ -401,6 +403,7 @@ class BINARYNINJAUIAPI FlowGraphWidget :
 	void setCurrentVariableDeadStoreElimination(BNDeadStoreElimination elimination);
 	void setCurrentExprFolding(BNExprFolding folding);
 	void toggleConditionInverted();
+	void setCurrentEarlyReturn(BNEarlyReturn earlyReturn);
 	void splitToNewTabAndNavigateFromCursorPosition();
 	void splitToNewWindowAndNavigateFromCursorPosition();
 	void splitToNewPaneAndNavigateFromCursorPosition();
-- 
cgit v1.3.1