Add API/UI to control dead store elimination of specific variables

1 files changed, 20 insertions, 0 deletions

diff --git a/function.cpp b/function.cpp
index 2daf5522..340f1e53 100644
--- a/function.cpp
+++ b/function.cpp
@@ -1900,6 +1900,26 @@ void Function::SetGotoLabelName(uint64_t labelId, const std::string& name)
 }
 
 
+BNDeadStoreElimination Function::GetVariableDeadStoreElimination(const Variable& var)
+{
+	BNVariable varData;
+	varData.type = var.type;
+	varData.index = var.index;
+	varData.storage = var.storage;
+	return BNGetFunctionVariableDeadStoreElimination(m_object, &varData);
+}
+
+
+void Function::SetVariableDeadStoreElimination(const Variable& var, BNDeadStoreElimination mode)
+{
+	BNVariable varData;
+	varData.type = var.type;
+	varData.index = var.index;
+	varData.storage = var.storage;
+	BNSetFunctionVariableDeadStoreElimination(m_object, &varData, mode);
+}
+
+
 AdvancedFunctionAnalysisDataRequestor::AdvancedFunctionAnalysisDataRequestor(Function* func): m_func(func)
 {
 	if (m_func)