Add UI action and API to control early returns in HLIL

1 files changed, 3 insertions, 0 deletions

diff --git a/ui/linearview.h b/ui/linearview.h
index 2ea8e37f..eea40d7f 100644
--- a/ui/linearview.h
+++ b/ui/linearview.h
@@ -320,6 +320,8 @@ class BINARYNINJAUIAPI LinearView : public QAbstractScrollArea, public View, pub
 	BNExprFolding getCurrentExprFolding();
 	std::optional<uint64_t> getCurrentInvertableConditionAddress();
 	bool getCurrentConditionInverted();
+	std::optional<uint64_t> getCurrentEarlyReturnAddress();
+	BNEarlyReturn getCurrentEarlyReturn();
 
 	void setDataButtonVisible(bool visible);
 	std::optional<std::pair<BinaryNinja::Variable, BinaryNinja::Variable>> getMergeVariablesAtCurrentLocation();
@@ -412,6 +414,7 @@ private Q_SLOTS:
 	void setCurrentVariableDeadStoreElimination(BNDeadStoreElimination elimination);
 	void setCurrentExprFolding(BNExprFolding folding);
 	void toggleConditionInverted();
+	void setCurrentEarlyReturn(BNEarlyReturn earlyReturn);
 
 Q_SIGNALS:
 	void notifyResizeEvent(int width, int height);