Add UI action and API to control early returns in HLIL

1 files changed, 12 insertions, 0 deletions

diff --git a/function.cpp b/function.cpp
index fec9e108..5ce3a843 100644
--- a/function.cpp
+++ b/function.cpp
@@ -2837,6 +2837,18 @@ void Function::SetConditionInverted(uint64_t addr, bool invert)
 }
 
 
+BNEarlyReturn Function::GetEarlyReturn(uint64_t addr)
+{
+	return BNGetEarlyReturn(m_object, addr);
+}
+
+
+void Function::SetEarlyReturn(uint64_t addr, BNEarlyReturn mode)
+{
+	BNSetEarlyReturn(m_object, addr, mode);
+}
+
+
 std::map<Variable, std::set<Variable>> Function::GetMergedVariables()
 {
 	size_t count;