Implement m_zext and m_sext Matchers

Author: AmrDeveloper

docs/matchers/cast.md

@@ -2,6 +2,8 @@
 
 |     Function      | Parameters |   Return    |                        Description                         |
 | :---------------: | :--------: | :---------: | :--------------------------------------------------------: |
+|      m_zext       |            | InstMatcher |     Build Inst Matcher that matchers zext instruction      |
+|      m_sext       |            | InstMatcher |     Build Inst Matcher that matchers sext instruction      |
 |   m_int_to_ptr    |            | InstMatcher |   Build Inst Matcher that matchers IntToPtr instruction    |
 |   m_ptr_to_int    |            | InstMatcher |   Build Inst Matcher that matchers PtrToInt instruction    |
 |    m_bit_cast     |            | InstMatcher |   Build Inst Matcher that matchers Bit cast instruction    |

src/functions/matchers/cast.rs

@@ -10,6 +10,8 @@ use crate::matchers::cast::CastInstMatcher;
 
 #[inline(always)]
 pub fn register_cast_matchers_function(map: &mut HashMap<&'static str, StandardFunction>) {
+    map.insert("m_zext", match_zext);
+    map.insert("m_sext", match_sext);
     map.insert("m_int_to_ptr", match_int_to_ptr_inst);
     map.insert("m_ptr_to_int", match_ptr_to_int_inst);
     map.insert("m_bit_cast", match_bit_cast_inst);
@@ -18,6 +20,10 @@ pub fn register_cast_matchers_function(map: &mut HashMap<&'static str, StandardF
 
 #[inline(always)]
 pub fn register_cast_matchers_function_signatures(map: &mut HashMap<&'static str, Signature>) {
+    map.insert("m_zext", Signature::with_return(Box::new(InstMatcherType)));
+
+    map.insert("m_sext", Signature::with_return(Box::new(InstMatcherType)));
+
     map.insert(
         "m_int_to_ptr",
         Signature::with_return(Box::new(InstMatcherType)),
@@ -39,6 +45,16 @@ pub fn register_cast_matchers_function_signatures(map: &mut HashMap<&'static str
     );
 }
 
+fn match_zext(_values: &[Box<dyn Value>]) -> Box<dyn Value> {
+    let matcher = Box::new(CastInstMatcher::create_zext());
+    Box::new(InstMatcherValue { matcher })
+}
+
+fn match_sext(_values: &[Box<dyn Value>]) -> Box<dyn Value> {
+    let matcher = Box::new(CastInstMatcher::create_sext());
+    Box::new(InstMatcherValue { matcher })
+}
+
 fn match_int_to_ptr_inst(_values: &[Box<dyn Value>]) -> Box<dyn Value> {
     let matcher = Box::new(CastInstMatcher::create_int_to_ptr());
     Box::new(InstMatcherValue { matcher })

src/matchers/cast.rs

@@ -8,13 +8,17 @@ use super::InstMatcher;
 enum CastMatcherKind {
     IntToPtr,
     PtrToInt,
+    ZExt,
+    SExt,
     BitCast,
     AddrSpaceCast,
 }
 
 impl CastMatcherKind {
     pub fn llvm_opcode(&self) -> LLVMOpcode {
         match self {
+            CastMatcherKind::ZExt => LLVMOpcode::LLVMZExt,
+            CastMatcherKind::SExt => LLVMOpcode::LLVMSExt,
             CastMatcherKind::IntToPtr => LLVMOpcode::LLVMIntToPtr,
             CastMatcherKind::PtrToInt => LLVMOpcode::LLVMPtrToInt,
             CastMatcherKind::BitCast => LLVMOpcode::LLVMBitCast,
@@ -29,6 +33,18 @@ pub struct CastInstMatcher {
 }
 
 impl CastInstMatcher {
+    pub fn create_zext() -> CastInstMatcher {
+        CastInstMatcher {
+            matcher_kind: CastMatcherKind::ZExt,
+        }
+    }
+
+    pub fn create_sext() -> CastInstMatcher {
+        CastInstMatcher {
+            matcher_kind: CastMatcherKind::SExt,
+        }
+    }
+
     pub fn create_int_to_ptr() -> CastInstMatcher {
         CastInstMatcher {
             matcher_kind: CastMatcherKind::IntToPtr,