[BNTL] Allow decompressing standalone TypeLibrary objects

Previously you must have written the type library to disk

Author: emesare

binaryninjaapi.h

@@ -19915,21 +19915,6 @@ namespace BinaryNinja {
 		*/
 		TypeLibrary(Ref<Architecture> arch, const std::string& name);
 
-		/*! Decompresses a type library from a file
-
-			\param path
-			\return The string contents of the decompressed type library
-		*/
-		std::string Decompress(const std::string& path);
-
-		/*! Decompresses a type library from a file
-
-			\param path
-			\param output
-			\return True if the type library was successfully decompressed
-		*/
-		static bool DecompressToFile(const std::string& path, const std::string& output);
-
 		/*! Loads a finalized type library instance from file
 
 			\param path
@@ -19957,9 +19942,17 @@ namespace BinaryNinja {
 		/*! Saves a finalized type library instance to file
 
 			\param path
+			\return True if the type library was successfully written to the file
 		*/
 		bool WriteToFile(const std::string& path);
 
+		/*! Decompresses the type library to a JSON file
+
+			\param path
+			\return True if the type library was successfully decompressed
+		*/
+		bool DecompressToFile(const std::string& path);
+		
 		/*! The Architecture this type library is associated with
 
 			\return

binaryninjacore.h

@@ -6814,7 +6814,7 @@ extern "C"
 	BINARYNINJACOREAPI BNTypeLibrary* BNNewTypeLibraryReference(BNTypeLibrary* lib);
 	BINARYNINJACOREAPI BNTypeLibrary* BNDuplicateTypeLibrary(BNTypeLibrary* lib);
 	BINARYNINJACOREAPI BNTypeLibrary* BNLoadTypeLibraryFromFile(const char* path);
-	BINARYNINJACOREAPI bool BNTypeLibraryDecompressToFile(const char* file, const char* output);
+	BINARYNINJACOREAPI bool BNTypeLibraryDecompressToFile(BNTypeLibrary* lib, const char* output);
 	BINARYNINJACOREAPI void BNFreeTypeLibrary(BNTypeLibrary* lib);
 
 	BINARYNINJACOREAPI BNTypeLibrary* BNLookupTypeLibraryByName(BNArchitecture* arch, const char* name);

python/typelibrary.py

@@ -56,17 +56,6 @@ def new(arch: 'architecture.Architecture', name:str) -> 'TypeLibrary':
 		"""
 		return TypeLibrary(core.BNNewTypeLibrary(arch.handle, name))
 
-	@staticmethod
-	def decompress_to_file(path: str, output: str) -> bool:
-		"""
-		Decompresses a type library file to a file on disk.
-
-		:param str path:
-		:param str output:
-		:rtype: bool
-		"""
-		return core.BNTypeLibraryDecompressToFile(path, output)
-
 	@staticmethod
 	def load_from_file(path: str) -> Optional['TypeLibrary']:
 		"""
@@ -92,6 +81,17 @@ def write_to_file(self, path: str) -> None:
 		if not core.BNWriteTypeLibraryToFile(self.handle, path):
 			raise OSError(f"Failed to write type library to '{path}'")
 
+	def decompress_to_file(self, path: str) -> None:
+		"""
+		Decompresses the type library file to a file on disk.
+
+		:param str path:
+		:rtype: bool
+		:raises: OSError if saving the file fails
+		"""
+		if not core.BNTypeLibraryDecompressToFile(self.handle, path):
+			raise OSError(f"Failed to decompress type library to '{path}'")
+
 	@staticmethod
 	def from_name(arch: architecture.Architecture, name: str):
 		"""

rust/src/types/library.rs

@@ -70,13 +70,6 @@ impl TypeLibrary {
         unsafe { Array::new(result, count, ()) }
     }
 
-    /// Decompresses a type library file to a JSON file at the given `output_path`.
-    pub fn decompress_to_file(path: &Path, output_path: &Path) -> bool {
-        let path = path.to_cstr();
-        let output = output_path.to_cstr();
-        unsafe { BNTypeLibraryDecompressToFile(path.as_ptr(), output.as_ptr()) }
-    }
-
     /// Loads a finalized type library instance from the given `path`.
     ///
     /// The returned type library cannot be modified.
@@ -94,6 +87,12 @@ impl TypeLibrary {
         unsafe { BNWriteTypeLibraryToFile(self.as_raw(), path.as_ptr()) }
     }
 
+    /// Decompresses the type library file to a JSON file at the given `output_path`.
+    pub fn decompress_to_file(&self, output_path: &Path) -> bool {
+        let path = output_path.to_cstr();
+        unsafe { BNTypeLibraryDecompressToFile(self.handle.as_ptr(), path.as_ptr()) }
+    }
+
     /// Looks up the first type library found with a matching name. Keep in mind that names are not
     /// necessarily unique.
     ///

typelibrary.cpp

@@ -14,9 +14,9 @@ TypeLibrary::TypeLibrary(Ref<Architecture> arch, const std::string& name)
 }
 
 
-bool TypeLibrary::DecompressToFile(const std::string& path, const std::string& output)
+bool TypeLibrary::DecompressToFile(const std::string& path)
 {
-	return BNTypeLibraryDecompressToFile(path.c_str(), output.c_str());
+	return BNTypeLibraryDecompressToFile(m_object, path.c_str());
 }