From 6436615567547349434351468012512505109552 Mon Sep 17 00:00:00 2001
From: Glenn Smith <glenn@vector35.com>
Date: Mon, 6 Nov 2023 15:28:59 -0500
Subject: Type Browser

---
 binaryview.cpp | 102 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 102 insertions(+)

(limited to 'binaryview.cpp')

diff --git a/binaryview.cpp b/binaryview.cpp
index 4bbd3aa3..077f2b4a 100644
--- a/binaryview.cpp
+++ b/binaryview.cpp
@@ -2491,6 +2491,108 @@ std::vector<Confidence<Ref<Type>>> BinaryView::GetTypesReferenced(const Qualifie
 }
 
 
+unordered_set<QualifiedName> BinaryView::GetOutgoingDirectTypeReferences(const QualifiedName& type)
+{
+	size_t count;
+	BNQualifiedName apiType = type.GetAPIObject();
+	BNQualifiedName* apiResult = BNGetOutgoingDirectTypeReferences(m_object, &apiType, &count);
+	QualifiedName::FreeAPIObject(&apiType);
+	if (!apiResult)
+		return {};
+
+	unordered_set<QualifiedName> result;
+	for (size_t i = 0; i < count; i ++)
+	{
+		result.insert(QualifiedName::FromAPIObject(&apiResult[i]));
+	}
+	BNFreeTypeNameList(apiResult, count);
+	return result;
+}
+
+
+unordered_set<QualifiedName> BinaryView::GetOutgoingRecursiveTypeReferences(const QualifiedName& type)
+{
+	return GetOutgoingRecursiveTypeReferences(unordered_set<QualifiedName>{type});
+}
+
+
+unordered_set<QualifiedName> BinaryView::GetOutgoingRecursiveTypeReferences(const unordered_set<QualifiedName>& types)
+{
+	size_t count;
+	vector<BNQualifiedName> apiTypes;
+	for (auto& type: types)
+	{
+		apiTypes.push_back(type.GetAPIObject());
+	}
+	BNQualifiedName* apiResult = BNGetOutgoingRecursiveTypeReferences(m_object, apiTypes.data(), apiTypes.size(), &count);
+	for (auto& type: apiTypes)
+	{
+		QualifiedName::FreeAPIObject(&type);
+	}
+	if (!apiResult)
+		return {};
+
+	unordered_set<QualifiedName> result;
+	for (size_t i = 0; i < count; i ++)
+	{
+		result.insert(QualifiedName::FromAPIObject(&apiResult[i]));
+	}
+	BNFreeTypeNameList(apiResult, count);
+	return result;
+}
+
+
+unordered_set<QualifiedName> BinaryView::GetIncomingDirectTypeReferences(const QualifiedName& type)
+{
+	size_t count;
+	BNQualifiedName apiType = type.GetAPIObject();
+	BNQualifiedName* apiResult = BNGetIncomingDirectTypeReferences(m_object, &apiType, &count);
+	QualifiedName::FreeAPIObject(&apiType);
+	if (!apiResult)
+		return {};
+
+	unordered_set<QualifiedName> result;
+	for (size_t i = 0; i < count; i ++)
+	{
+		result.insert(QualifiedName::FromAPIObject(&apiResult[i]));
+	}
+	BNFreeTypeNameList(apiResult, count);
+	return result;
+}
+
+
+unordered_set<QualifiedName> BinaryView::GetIncomingRecursiveTypeReferences(const QualifiedName& type)
+{
+	return GetIncomingRecursiveTypeReferences(unordered_set<QualifiedName>{type});
+}
+
+
+unordered_set<QualifiedName> BinaryView::GetIncomingRecursiveTypeReferences(const unordered_set<QualifiedName>& types)
+{
+	size_t count;
+	vector<BNQualifiedName> apiTypes;
+	for (auto& type: types)
+	{
+		apiTypes.push_back(type.GetAPIObject());
+	}
+	BNQualifiedName* apiResult = BNGetIncomingRecursiveTypeReferences(m_object, apiTypes.data(), apiTypes.size(), &count);
+	for (auto& type: apiTypes)
+	{
+		QualifiedName::FreeAPIObject(&type);
+	}
+	if (!apiResult)
+		return {};
+
+	unordered_set<QualifiedName> result;
+	for (size_t i = 0; i < count; i ++)
+	{
+		result.insert(QualifiedName::FromAPIObject(&apiResult[i]));
+	}
+	BNFreeTypeNameList(apiResult, count);
+	return result;
+}
+
+
 vector<uint64_t> BinaryView::GetCallees(ReferenceSource callSite)
 {
 	size_t count;
-- 
cgit v1.3.1