From 3a1a81c3ddd8340432ecd2687d3a25195cc97d40 Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty.wagner@gmail.com>
Date: Fri, 6 Jun 2025 16:22:48 -0600
Subject: Add UI action and API to control switch recovery 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 d988da12..90b86000 100644
--- a/ui/flowgraphwidget.h
+++ b/ui/flowgraphwidget.h
@@ -172,6 +172,8 @@ class BINARYNINJAUIAPI FlowGraphWidget :
 	bool getCurrentConditionInverted();
 	std::optional<uint64_t> getCurrentEarlyReturnAddress();
 	BNEarlyReturn getCurrentEarlyReturn();
+	std::optional<uint64_t> getCurrentSwitchRecoveryAddress();
+	BNSwitchRecovery getCurrentSwitchRecovery();
 	std::optional<std::pair<BinaryNinja::Variable, BinaryNinja::Variable>> getMergeVariablesAtCurrentLocation();
 
   protected:
@@ -404,6 +406,7 @@ class BINARYNINJAUIAPI FlowGraphWidget :
 	void setCurrentExprFolding(BNExprFolding folding);
 	void toggleConditionInverted();
 	void setCurrentEarlyReturn(BNEarlyReturn earlyReturn);
+	void setCurrentSwitchRecovery(BNSwitchRecovery recovery);
 	void splitToNewTabAndNavigateFromCursorPosition();
 	void splitToNewWindowAndNavigateFromCursorPosition();
 	void splitToNewPaneAndNavigateFromCursorPosition();
-- 
cgit v1.3.1