Type library C++ api & import recording

1 files changed, 32 insertions, 0 deletions

diff --git a/platform.cpp b/platform.cpp
index b3ec68e8..06d65539 100644
--- a/platform.cpp
+++ b/platform.cpp
@@ -342,6 +342,38 @@ map<uint32_t, QualifiedNameAndType> Platform::GetSystemCalls()
 }
 
 
+vector<Ref<TypeLibrary>> Platform::GetTypeLibraries()
+{
+	size_t count;
+	BNTypeLibrary** libs = BNGetPlatformTypeLibraries(m_object, &count);
+
+	vector<Ref<TypeLibrary>> result;
+	for (size_t i = 0; i < count; ++i)
+	{
+		result.push_back(new TypeLibrary(BNNewTypeLibraryReference(libs[i])));
+	}
+
+	BNFreeTypeLibraryList(libs, count);
+	return result;
+}
+
+
+vector<Ref<TypeLibrary>> Platform::GetTypeLibrariesByName(const std::string& name)
+{
+	size_t count;
+	BNTypeLibrary** libs = BNGetPlatformTypeLibrariesByName(m_object, name.c_str(), &count);
+
+	vector<Ref<TypeLibrary>> result;
+	for (size_t i = 0; i < count; ++i)
+	{
+		result.push_back(new TypeLibrary(BNNewTypeLibraryReference(libs[i])));
+	}
+
+	BNFreeTypeLibraryList(libs, count);
+	return result;
+}
+
+
 Ref<Type> Platform::GetTypeByName(const QualifiedName& name)
 {
 	BNQualifiedName nameObj = name.GetAPIObject();