Add function_by_name to Rust API for applying platform types

Author: zznop

rust/src/platform.rs

@@ -20,8 +20,8 @@ use crate::{
     rc::*,
     string::*,
     types::{
-        QualifiedNameAndType, TypeContainer, TypeLibrary, TypeParserError, TypeParserErrorSeverity,
-        TypeParserResult,
+        QualifiedName, QualifiedNameAndType, Type, TypeContainer, TypeLibrary, TypeParserError,
+        TypeParserErrorSeverity, TypeParserResult,
     },
 };
 use binaryninjacore_sys::*;
@@ -279,6 +279,22 @@ impl Platform {
         }
     }
 
+    pub fn function_by_name<T: Into<QualifiedName>>(
+        &self,
+        name: T,
+        exact_match: bool,
+    ) -> Option<Ref<Type>> {
+        let mut raw_name = QualifiedName::into_raw(name.into());
+        unsafe {
+            let type_handle = BNGetPlatformFunctionByName(self.handle, &mut raw_name, exact_match);
+            QualifiedName::free_raw(raw_name);
+            if type_handle.is_null() {
+                return None;
+            }
+            Some(Type::ref_from_raw(type_handle))
+        }
+    }
+
     // TODO: system_calls
     // TODO: add a helper function to define a system call (platform function with a specific type)
 

rust/tests/platform.rs

@@ -21,6 +21,17 @@ fn test_platform_types() {
     assert_ne!(platform_types.len(), 0);
 }
 
+#[test]
+fn test_platform_function_by_name() {
+    let _session = Session::new().expect("Failed to initialize session");
+    let platform = Platform::by_name("windows-x86_64").expect("windows-x86_64 exists");
+    let platform_functions = platform.functions();
+    assert_ne!(platform_functions.len(), 0);
+
+    let function = platform_functions.get(0);
+    assert!(platform.function_by_name(function.name, true).is_some());
+}
+
 #[test]
 fn test_platform_calling_conventions() {
     let _session = Session::new().expect("Failed to initialize session");