wip

abbiecnt · Aug 26, 2024 · d211921 · d211921
1 parent fc0bfdf
commit d211921
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/program/mod.rs b/src/program/mod.rs
@@ -1,6 +1,8 @@
 use crate::{Base, Section};
 use core::ops::Index;
 use core::slice::{from_raw_parts, SliceIndex};
+use rayon::iter::IndexedParallelIterator;
+use rayon::slice::ParallelSlice;
 use std::sync::LazyLock;
 
 static PROGRAM: LazyLock<Program> = LazyLock::new(Program::init);
@@ -40,6 +42,20 @@ impl Program {
         &self.sections
     }
 
+    pub fn find(&self, pattern: &[u8]) -> Option<*const u8> {
+        self.as_slice()
+            .par_windows(pattern.len())
+            .position_first(|window| window == pattern)
+            .map(|offset| unsafe { self.as_ptr().add(offset) })
+    }
+
+    pub fn rfind(&self, pattern: &[u8]) -> Option<*const u8> {
+        self.as_slice()
+            .par_windows(pattern.len())
+            .position_last(|window| window == pattern)
+            .map(|offset| unsafe { self.as_ptr().add(offset) })
+    }
+
     fn init() -> Self {
         #[cfg(target_os = "windows")]
         {