Type Browser

1 files changed, 102 insertions, 0 deletions

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;