From 40bdfac7b71dd1cbe9223b82518053f80268e2a8 Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty.wagner@gmail.com>
Date: Tue, 16 Aug 2022 20:49:02 -0400
Subject: Splitting of variables

---
 function.cpp | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

(limited to 'function.cpp')

diff --git a/function.cpp b/function.cpp
index 06815c39..96795ef0 100644
--- a/function.cpp
+++ b/function.cpp
@@ -2611,6 +2611,38 @@ void Function::UnmergeVariables(const Variable& target, const std::set<Variable>
 }
 
 
+std::set<Variable> Function::GetSplitVariables()
+{
+	size_t count = 0;
+	BNVariable* vars = BNGetSplitVariables(m_object, &count);
+
+	std::set<Variable> result;
+	for (size_t i = 0; i < count; i++)
+	{
+		Variable var;
+		var.type = vars[i].type;
+		var.index = vars[i].index;
+		var.storage = vars[i].storage;
+		result.insert(var);
+	}
+
+	BNFreeVariableList(vars);
+	return result;
+}
+
+
+void Function::SplitVariable(const Variable& var)
+{
+	BNSplitVariable(m_object, &var);
+}
+
+
+void Function::UnsplitVariable(const Variable& var)
+{
+	BNUnsplitVariable(m_object, &var);
+}
+
+
 vector<ILReferenceSource> Function::GetMediumLevelILVariableReferences(const Variable& var)
 {
 	size_t count;
-- 
cgit v1.3.1