Implement m_throw and m_rethrow matchers

Author: AmrDeveloper

docs/matchers/exceptions.md

@@ -1,7 +1,9 @@
 ### Exceptions Instructions Matchers functions
 
-|   Function   | Parameters |   Return    |                     Description                      |
-| :----------: | :--------: | :---------: | :--------------------------------------------------: |
-|   m_invoke   |            | InstMatcher |   Build Inst Matcher that match invoke Instruction   |
-| m_landingpad |            | InstMatcher | Build Inst Matcher that match landingpad Instruction |
-|   m_resume   |            | InstMatcher |   Build Inst Matcher that match resume Instruction   |
+|   Function   | Parameters |   Return    |                      Description                      |
+| :----------: | :--------: | :---------: | :---------------------------------------------------: |
+|   m_invoke   |            | InstMatcher |   Build Inst Matcher that match invoke Instruction    |
+| m_landingpad |            | InstMatcher | Build Inst Matcher that match landingpad Instruction  |
+|   m_resume   |            | InstMatcher |   Build Inst Matcher that match resume Instruction    |
+|   m_throw    |            | InstMatcher |  Build Inst Matcher that match call to `__cxa_throw`  |
+|  m_rethrow   |            | InstMatcher | Build Inst Matcher that match call to `__cxa_rethrow` |

src/functions/matchers/exception.rs

@@ -5,6 +5,8 @@ use crate::ir::values::InstMatcherValue;
 use crate::matchers::exception::InvokeInstMatcher;
 use crate::matchers::exception::LandingPadInstMatcher;
 use crate::matchers::exception::ResumeInstMatcher;
+use crate::matchers::exception::RethrowInstMatcher;
+use crate::matchers::exception::ThrowInstMatcher;
 
 use gitql_core::signature::Signature;
 use gitql_core::signature::StandardFunction;
@@ -17,6 +19,8 @@ pub fn register_exception_inst_matchers_functions(
     map.insert("m_invoke", match_invoke_inst);
     map.insert("m_landingpad", match_landingpad_inst);
     map.insert("m_resume", match_resume_inst);
+    map.insert("m_throw", match_throw_inst);
+    map.insert("m_rethrow", match_rethrow_inst);
 }
 
 #[inline(always)]
@@ -46,6 +50,22 @@ pub fn register_exception_inst_matchers_function_signatures(
             return_type: Box::new(InstMatcherType),
         },
     );
+
+    map.insert(
+        "m_throw",
+        Signature {
+            parameters: vec![],
+            return_type: Box::new(InstMatcherType),
+        },
+    );
+
+    map.insert(
+        "m_rethrow",
+        Signature {
+            parameters: vec![],
+            return_type: Box::new(InstMatcherType),
+        },
+    );
 }
 
 fn match_invoke_inst(_values: &[Box<dyn Value>]) -> Box<dyn Value> {
@@ -62,3 +82,13 @@ fn match_resume_inst(_values: &[Box<dyn Value>]) -> Box<dyn Value> {
     let matcher = Box::new(ResumeInstMatcher);
     Box::new(InstMatcherValue { matcher })
 }
+
+fn match_throw_inst(_values: &[Box<dyn Value>]) -> Box<dyn Value> {
+    let matcher = Box::new(ThrowInstMatcher);
+    Box::new(InstMatcherValue { matcher })
+}
+
+fn match_rethrow_inst(_values: &[Box<dyn Value>]) -> Box<dyn Value> {
+    let matcher = Box::new(RethrowInstMatcher);
+    Box::new(InstMatcherValue { matcher })
+}

src/matchers/exception.rs

@@ -1,7 +1,10 @@
-use inkwell::llvm_sys::core::{LLVMGetInstructionOpcode, LLVMIsAResumeInst};
+use inkwell::llvm_sys::core::LLVMGetInstructionOpcode;
+use inkwell::llvm_sys::core::LLVMIsAResumeInst;
 use inkwell::llvm_sys::prelude::LLVMValueRef;
 use inkwell::llvm_sys::LLVMOpcode;
 
+use crate::matchers::matchers_helper::is_call_or_invoke_inst_with_specific_name;
+
 use super::Matcher;
 
 #[derive(Clone)]
@@ -33,3 +36,23 @@ impl Matcher<LLVMValueRef> for ResumeInstMatcher {
         unsafe { !LLVMIsAResumeInst(*instruction).is_null() }
     }
 }
+
+#[derive(Clone)]
+pub struct ThrowInstMatcher;
+
+impl Matcher<LLVMValueRef> for ThrowInstMatcher {
+    #[allow(clippy::not_unsafe_ptr_arg_deref)]
+    fn is_match(&self, instruction: &LLVMValueRef) -> bool {
+        is_call_or_invoke_inst_with_specific_name(instruction, "__cxa_throw")
+    }
+}
+
+#[derive(Clone)]
+pub struct RethrowInstMatcher;
+
+impl Matcher<LLVMValueRef> for RethrowInstMatcher {
+    #[allow(clippy::not_unsafe_ptr_arg_deref)]
+    fn is_match(&self, instruction: &LLVMValueRef) -> bool {
+        is_call_or_invoke_inst_with_specific_name(instruction, "__cxa_rethrow")
+    }
+}

src/matchers/matchers_helper.rs

@@ -0,0 +1,25 @@
+use inkwell::llvm_sys::core::LLVMGetCalledValue;
+use inkwell::llvm_sys::core::LLVMGetInstructionOpcode;
+use inkwell::llvm_sys::core::LLVMGetValueName2;
+use inkwell::llvm_sys::prelude::LLVMValueRef;
+use inkwell::llvm_sys::LLVMOpcode;
+
+pub(crate) fn is_call_or_invoke_inst_with_specific_name(
+    instruction: &LLVMValueRef,
+    name: &'static str,
+) -> bool {
+    unsafe {
+        if matches!(
+            LLVMGetInstructionOpcode(*instruction),
+            LLVMOpcode::LLVMCall | LLVMOpcode::LLVMInvoke
+        ) {
+            let mut len: usize = 0;
+            let called_value = LLVMGetCalledValue(*instruction);
+            let name_ptr = LLVMGetValueName2(called_value, &mut len);
+            let name_slice = std::slice::from_raw_parts(name_ptr as *const u8, len);
+            let called_name = std::str::from_utf8_unchecked(name_slice).to_string();
+            return called_name.eq(name);
+        }
+        false
+    }
+}

src/matchers/mod.rs

@@ -15,6 +15,8 @@ pub mod other;
 pub mod types;
 pub mod usage;
 
+mod matchers_helper;
+
 dyn_clone::clone_trait_object!(<T> Matcher<T>);
 
 pub trait Matcher<T: Clone>: DynClone {