Implement m_global_var & m_global_val matchers

Author: AmrDeveloper

src/functions/matchers/globals.rs

@@ -0,0 +1,32 @@
+use std::collections::HashMap;
+
+use gitql_core::signature::Signature;
+use gitql_core::signature::StandardFunction;
+use gitql_core::values::Value;
+
+use crate::functions::matcher_signature_without_parameters;
+use crate::ir::values::InstMatcherValue;
+use crate::matchers::globals::GlobalValueExprMatcher;
+use crate::matchers::globals::GlobalVariableExprMatcher;
+
+#[inline(always)]
+pub fn register_globals_matchers_functions(map: &mut HashMap<&'static str, StandardFunction>) {
+    map.insert("m_global_var", match_global_variable_expr_inst);
+    map.insert("m_global_val", match_global_value_expr_inst);
+}
+
+#[inline(always)]
+pub fn register_globals_matchers_function_signatures(map: &mut HashMap<&'static str, Signature>) {
+    map.insert("m_global_var", matcher_signature_without_parameters());
+    map.insert("m_global_val", matcher_signature_without_parameters());
+}
+
+fn match_global_variable_expr_inst(_values: &[Box<dyn Value>]) -> Box<dyn Value> {
+    let matcher = Box::new(GlobalVariableExprMatcher);
+    Box::new(InstMatcherValue { matcher })
+}
+
+fn match_global_value_expr_inst(_values: &[Box<dyn Value>]) -> Box<dyn Value> {
+    let matcher = Box::new(GlobalValueExprMatcher);
+    Box::new(InstMatcherValue { matcher })
+}

src/functions/matchers/mod.rs

@@ -8,6 +8,7 @@ pub mod debug;
 pub mod exception;
 pub mod fcmp;
 pub mod get_element_ptr;
+pub mod globals;
 pub mod icmp;
 pub mod operand_bundle;
 pub mod other;

src/functions/mod.rs

@@ -42,6 +42,8 @@ use matchers::usage::register_usage_matchers_functions;
 
 use crate::functions::matchers::get_element_ptr::register_get_element_ptr_inst_matchers_function_signatures;
 use crate::functions::matchers::get_element_ptr::register_get_element_ptr_inst_matchers_functions;
+use crate::functions::matchers::globals::register_globals_matchers_function_signatures;
+use crate::functions::matchers::globals::register_globals_matchers_functions;
 use crate::ir::types::InstMatcherType;
 use crate::ir::values::InstMatcherValue;
 use crate::matchers::other::AnyInstMatcher;
@@ -70,6 +72,7 @@ pub fn llvm_ir_functions() -> &'static HashMap<&'static str, StandardFunction> {
         register_debug_inst_matchers_functions(&mut map);
         register_operand_bundle_inst_matchers_functions(&mut map);
         register_get_element_ptr_inst_matchers_functions(&mut map);
+        register_globals_matchers_functions(&mut map);
         map
     })
 }
@@ -93,7 +96,7 @@ pub fn llvm_ir_function_signatures() -> HashMap<&'static str, Signature> {
     register_debug_inst_matchers_function_signatures(&mut map);
     register_operand_bundle_inst_matchers_function_signatures(&mut map);
     register_get_element_ptr_inst_matchers_function_signatures(&mut map);
-
+    register_globals_matchers_function_signatures(&mut map);
     map
 }
 

src/matchers/globals.rs

@@ -0,0 +1,23 @@
+use inkwell::llvm_sys::core::LLVMIsAGlobalValue;
+use inkwell::llvm_sys::core::LLVMIsAGlobalVariable;
+use inkwell::llvm_sys::prelude::LLVMValueRef;
+
+use super::Matcher;
+
+#[derive(Clone)]
+pub struct GlobalValueExprMatcher;
+
+impl Matcher<LLVMValueRef> for GlobalValueExprMatcher {
+    fn is_match(&self, instruction: &LLVMValueRef) -> bool {
+        unsafe { !LLVMIsAGlobalValue(*instruction).is_null() }
+    }
+}
+
+#[derive(Clone)]
+pub struct GlobalVariableExprMatcher;
+
+impl Matcher<LLVMValueRef> for GlobalVariableExprMatcher {
+    fn is_match(&self, instruction: &LLVMValueRef) -> bool {
+        unsafe { !LLVMIsAGlobalVariable(*instruction).is_null() }
+    }
+}

src/matchers/mod.rs

@@ -9,6 +9,7 @@ pub mod debug;
 pub mod exception;
 pub mod fcmp;
 pub mod get_element_ptr;
+pub mod globals;
 pub mod icmp;
 pub mod operand_bundle;
 pub mod other;