wip

KiraCoding · Sep 13, 2024 · 36c4338 · 36c4338
1 parent cf5f650
commit 36c4338
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/src/header.rs b/src/header.rs
diff --git a/src/program.rs b/src/program.rs
@@ -144,9 +144,7 @@ impl Program {
 
     fn init() -> Self {
         let base = Base::program();
-
         let pe = unsafe { PeView::module(base.as_nonnull().as_ptr()) };
-
         let len = pe.nt_headers().OptionalHeader.SizeOfImage as usize;
 
         let sections = pe

diff --git a/src/section.rs b/src/section.rs
@@ -38,17 +38,23 @@ impl Section {
     }
 
     pub fn contains(&self, pattern: &[u8]) -> bool {
+        assert!(pattern.len() >= 1);
+
         self.find(pattern).is_some()
     }
 
     pub fn find(&self, pattern: &[u8]) -> Option<NonNull<u8>> {
+        assert!(pattern.len() >= 1);
+
         self.as_slice()
             .par_windows(pattern.len())
             .position_first(|window| window == pattern)
             .map(|offset| unsafe { self.base.add(offset) })
     }
 
     pub fn rfind(&self, pattern: &[u8]) -> Option<NonNull<u8>> {
+        assert!(pattern.len() >= 1);
+
         self.as_slice()
             .par_windows(pattern.len())
             .position_last(|window| window == pattern)