Implement m_callbr inst matcher

Author: AmrDeveloper

docs/matchers/call.md

@@ -2,5 +2,6 @@
 
 |  Function   | Parameters  |   Return    |                                 Description                                 |
 | :---------: | :---------: | :---------: | :-------------------------------------------------------------------------: |
-|   m_call    | (n : Text?) | InstMatcher |     Build Inst Matcher that match call instruction   with optional name     |
+|   m_call    | (n : Text?) | InstMatcher |      Build Inst Matcher that match call instruction with optional name      |
+|  m_callbr   | (n : Text?) | InstMatcher |     Build Inst Matcher that match callbr instruction with optional name     |
 | m_intrinsic | (n : Text?) | InstMatcher | Build Inst Matcher that match intrinsic call instruction with optional name |

src/functions/matchers/call.rs

@@ -8,12 +8,14 @@ use gitql_core::values::Value;
 
 use crate::ir::types::InstMatcherType;
 use crate::ir::values::InstMatcherValue;
+use crate::matchers::call::CallBrInstMatcher;
 use crate::matchers::call::CallInstMatcher;
 use crate::matchers::call::IntrinsicInstMatcher;
 
 #[inline(always)]
 pub fn register_call_inst_matchers_functions(map: &mut HashMap<&'static str, StandardFunction>) {
     map.insert("m_call", match_call_inst);
+    map.insert("m_callbr", match_callbr_inst);
     map.insert("m_intrinsic", match_intrinsic_inst);
 }
 
@@ -25,6 +27,12 @@ pub fn register_call_inst_matchers_function_signatures(map: &mut HashMap<&'stati
             .add_parameter(Box::new(OptionType::new(Some(Box::new(TextType))))),
     );
 
+    map.insert(
+        "m_callbr",
+        Signature::with_return(Box::new(InstMatcherType))
+            .add_parameter(Box::new(OptionType::new(Some(Box::new(TextType))))),
+    );
+
     map.insert(
         "m_intrinsic",
         Signature::with_return(Box::new(InstMatcherType))
@@ -43,6 +51,17 @@ fn match_call_inst(values: &[Box<dyn Value>]) -> Box<dyn Value> {
     Box::new(InstMatcherValue { matcher })
 }
 
+fn match_callbr_inst(values: &[Box<dyn Value>]) -> Box<dyn Value> {
+    let function_name = if values.is_empty() {
+        None
+    } else {
+        values[0].as_text()
+    };
+
+    let matcher = Box::new(CallBrInstMatcher::create_call(function_name));
+    Box::new(InstMatcherValue { matcher })
+}
+
 fn match_intrinsic_inst(values: &[Box<dyn Value>]) -> Box<dyn Value> {
     let intrinsic_name = if values.is_empty() {
         None

src/matchers/call.rs

@@ -40,6 +40,37 @@ impl Matcher<LLVMValueRef> for CallInstMatcher {
     }
 }
 
+#[derive(Clone)]
+pub struct CallBrInstMatcher {
+    pub name: Option<String>,
+}
+
+impl CallBrInstMatcher {
+    pub fn create_call(name: Option<String>) -> Self {
+        CallBrInstMatcher { name }
+    }
+}
+
+impl Matcher<LLVMValueRef> for CallBrInstMatcher {
+    #[allow(clippy::not_unsafe_ptr_arg_deref)]
+    fn is_match(&self, instruction: &LLVMValueRef) -> bool {
+        unsafe {
+            if LLVMGetInstructionOpcode(*instruction) == LLVMOpcode::LLVMCallBr {
+                if let Some(expected_name) = &self.name {
+                    let called_value = LLVMGetCalledValue(*instruction);
+                    let mut len: usize = 0;
+                    let name_ptr = LLVMGetValueName2(called_value, &mut len);
+                    let name_slice = std::slice::from_raw_parts(name_ptr as *const u8, len);
+                    let name = std::str::from_utf8_unchecked(name_slice).to_string();
+                    return name.eq(expected_name);
+                }
+                return true;
+            }
+        }
+        false
+    }
+}
+
 #[derive(Clone)]
 pub struct IntrinsicInstMatcher {
     pub name: Option<String>,