Implement LanguageRepresentation and LineFormatter in Rust API

Author: rbran

rust/src/function.rs

@@ -40,6 +40,7 @@ pub use binaryninjacore_sys::BNHighlightStandardColor as HighlightStandardColor;
 use crate::architecture::RegisterId;
 use crate::confidence::Conf;
 use crate::high_level_il::HighLevelILFunction;
+use crate::language_representation::CoreLanguageRepresentationFunction;
 use crate::low_level_il::{LiftedILFunction, RegularLowLevelILFunction};
 use crate::medium_level_il::MediumLevelILFunction;
 use crate::variable::{
@@ -476,6 +477,42 @@ impl Function {
         }
     }
 
+    /// Get the language representation of the function.
+    ///
+    ///  * `language` - The language representation, ex. "Pseudo C".
+    pub fn language_representation<S: BnStrCompatible>(
+        &self,
+        language: S,
+    ) -> Option<Ref<CoreLanguageRepresentationFunction>> {
+        let lang_name = language.into_bytes_with_nul();
+        let repr = unsafe {
+            BNGetFunctionLanguageRepresentation(
+                self.handle,
+                lang_name.as_ref().as_ptr() as *const c_char,
+            )
+        };
+        NonNull::new(repr)
+            .map(|handle| unsafe { CoreLanguageRepresentationFunction::ref_from_raw(handle) })
+    }
+
+    /// Get the language representation of the function, if available.
+    ///
+    ///  * `language` - The language representation, ex. "Pseudo C".
+    pub fn language_representation_if_available<S: BnStrCompatible>(
+        &self,
+        language: S,
+    ) -> Option<Ref<CoreLanguageRepresentationFunction>> {
+        let lang_name = language.into_bytes_with_nul();
+        let repr = unsafe {
+            BNGetFunctionLanguageRepresentationIfAvailable(
+                self.handle,
+                lang_name.as_ref().as_ptr() as *const c_char,
+            )
+        };
+        NonNull::new(repr)
+            .map(|handle| unsafe { CoreLanguageRepresentationFunction::ref_from_raw(handle) })
+    }
+
     pub fn high_level_il(&self, full_ast: bool) -> Result<Ref<HighLevelILFunction>, ()> {
         unsafe {
             let hlil_ptr = BNGetFunctionHighLevelIL(self.handle);

rust/src/high_level_il.rs

@@ -4,6 +4,7 @@ mod function;
 mod instruction;
 mod lift;
 pub mod operation;
+pub mod token_emitter;
 
 pub use self::block::*;
 pub use self::function::*;

rust/src/high_level_il/function.rs

@@ -15,12 +15,17 @@ pub struct HighLevelILFunction {
 }
 
 impl HighLevelILFunction {
+    pub(crate) unsafe fn from_raw(handle: *mut BNHighLevelILFunction, full_ast: bool) -> Self {
+        debug_assert!(!handle.is_null());
+        Self { handle, full_ast }
+    }
+
     pub(crate) unsafe fn ref_from_raw(
         handle: *mut BNHighLevelILFunction,
         full_ast: bool,
     ) -> Ref<Self> {
         debug_assert!(!handle.is_null());
-        Self { handle, full_ast }.to_owned()
+        Ref::new(Self { handle, full_ast })
     }
 
     pub fn instruction_from_index(