From d8a15adb1d3645e73095144bf0d0115cb73500a9 Mon Sep 17 00:00:00 2001
From: Alwin Berger <alwin.berger@tu-dortmund.de>
Date: Tue, 22 Oct 2024 14:25:51 +0200
Subject: [PATCH] move codepieces

---
 fuzzers/FRET/src/systemstate/mod.rs | 101 +++++++++++++++-------------
 1 file changed, 55 insertions(+), 46 deletions(-)

diff --git a/fuzzers/FRET/src/systemstate/mod.rs b/fuzzers/FRET/src/systemstate/mod.rs
index e4883b99b6..ce308f361a 100644
--- a/fuzzers/FRET/src/systemstate/mod.rs
+++ b/fuzzers/FRET/src/systemstate/mod.rs
@@ -36,6 +36,7 @@ pub enum CaptureEvent {
     Undefined,
 }
 
+// ============================= State info
 
 /// Raw info Dump from Qemu
 #[derive(Debug, Default)]
@@ -179,6 +180,8 @@ impl fmt::Display for ReducedFreeRTOSSystemState {
     }
 }
 
+// ============================= Interval info
+
 // #[derive(Debug, Default, Serialize, Deserialize, Clone, PartialEq, Eq, Hash)]
 // pub enum ExecLevel {
 //     APP = 0,
@@ -244,51 +247,9 @@ impl ExecInterval {
     }
 }
 
-// Wrapper around Vec<RefinedFreeRTOSSystemState> to attach as Metadata
-#[derive(Debug, Default, Serialize, Deserialize, Clone)]
-pub struct FreeRTOSSystemStateMetadata {
-    pub inner: Vec<ReducedFreeRTOSSystemState>,
-    // TODO: Add abbs and memory reads
-    trace_length: usize,
-    indices: Vec<usize>,    // Hashed enumeration of States
-    tcref: isize,
-}
-impl FreeRTOSSystemStateMetadata {
-    pub fn new(inner: Vec<ReducedFreeRTOSSystemState>) -> Self{
-        let tmp = inner.iter().enumerate().map(|x| compute_hash(x) as usize).collect();
-        Self {trace_length: inner.len(), inner: inner, indices: tmp, tcref: 0}
-    }
-}
-pub fn compute_hash<T>(obj: T) -> u64 
-where
-    T: Hash
-{
-    let mut s = DefaultHasher::new();
-    obj.hash(&mut s);
-    s.finish()
-}
-
-// impl AsSlice for FreeRTOSSystemStateMetadata {
-//     /// Convert the slice of system-states to a slice of hashes over enumerated states
-//     fn as_slice(&self) -> &[usize] {
-//         self.indices.as_slice()
-//     }
-
-//     type Entry = usize;
-// }
-
-impl HasRefCnt for FreeRTOSSystemStateMetadata {
-    fn refcnt(&self) -> isize {
-        self.tcref
-    }
-
-    fn refcnt_mut(&mut self) -> &mut isize {
-        &mut self.tcref
-    }
-}
-
-libafl_bolts::impl_serdeany!(FreeRTOSSystemStateMetadata);
+// ============================= Atomic Basic Block
 
+/// A single-entry multiple-exit region between api calls. May be used referenced in multiple intervals.
 #[derive(Default, Serialize, Deserialize, Clone)]
 pub struct AtomicBasicBlock {
     start: GuestAddr,
@@ -300,7 +261,7 @@ pub struct AtomicBasicBlock {
 
 impl PartialEq for AtomicBasicBlock {
     fn eq(&self, other: &Self) -> bool {
-        self.start == other.start && self.ends == other.ends && self.level == other.level
+        self.start == other.start && self.ends == other.ends && self.level == other.level && self.instance_name == other.instance_name
     }
 }
 
@@ -313,6 +274,7 @@ impl Hash for AtomicBasicBlock {
         let mut keys : Vec<_> = self.ends.iter().collect();
         keys.sort();
         self.level.hash(state);
+        self.instance_name.hash(state);
         keys.hash(state);
     }
 }
@@ -392,4 +354,51 @@ fn get_task_names(trace: &Vec<ReducedFreeRTOSSystemState>) -> HashSet<String> {
     ret
 }
 
-libafl_bolts::impl_serdeany!(AtomicBasicBlock);
\ No newline at end of file
+libafl_bolts::impl_serdeany!(AtomicBasicBlock);
+
+// ============================= Per testcase metadata
+
+// Wrapper around Vec<RefinedFreeRTOSSystemState> to attach as Metadata
+#[derive(Debug, Default, Serialize, Deserialize, Clone)]
+pub struct FreeRTOSSystemStateMetadata {
+    pub inner: Vec<ReducedFreeRTOSSystemState>,
+    // TODO: Add abbs and memory reads
+    trace_length: usize,
+    indices: Vec<usize>,    // Hashed enumeration of States
+    tcref: isize,
+}
+impl FreeRTOSSystemStateMetadata {
+    pub fn new(inner: Vec<ReducedFreeRTOSSystemState>) -> Self{
+        let tmp = inner.iter().enumerate().map(|x| compute_hash(x) as usize).collect();
+        Self {trace_length: inner.len(), inner: inner, indices: tmp, tcref: 0}
+    }
+}
+pub fn compute_hash<T>(obj: T) -> u64 
+where
+    T: Hash
+{
+    let mut s = DefaultHasher::new();
+    obj.hash(&mut s);
+    s.finish()
+}
+
+// impl AsSlice for FreeRTOSSystemStateMetadata {
+//     /// Convert the slice of system-states to a slice of hashes over enumerated states
+//     fn as_slice(&self) -> &[usize] {
+//         self.indices.as_slice()
+//     }
+
+//     type Entry = usize;
+// }
+
+impl HasRefCnt for FreeRTOSSystemStateMetadata {
+    fn refcnt(&self) -> isize {
+        self.tcref
+    }
+
+    fn refcnt_mut(&mut self) -> &mut isize {
+        &mut self.tcref
+    }
+}
+
+libafl_bolts::impl_serdeany!(FreeRTOSSystemStateMetadata);