deep testing for stage resumeability

AFLplusplus · Jan 7, 2024 · 2683c9f · 2683c9f
1 parent c6af1ca
commit 2683c9f
Showing 1 changed file with 52 additions and 0 deletions.
diff --git a/libafl/src/stages/logics.rs b/libafl/src/stages/logics.rs
@@ -386,6 +386,28 @@ mod test {
         test_resume(&completed, &mut state, tuple_list!(ifstage, resetstage));
     }
 
+    #[test]
+    fn check_resumability_if_deep() {
+        let (completed, stages) = test_resume_stages();
+        let ifstage = IfStage::new(
+            |_, _, _, _| Ok(true),
+            tuple_list!(IfStage::new(
+                |_, _, _, _| Ok(true),
+                tuple_list!(IfStage::new(
+                    |_, _, _, _| Ok(true),
+                    tuple_list!(IfStage::new(
+                        |_, _, _, _| Ok(true),
+                        tuple_list!(IfStage::new(|_, _, _, _| Ok(true), stages),),
+                    ),),
+                ))
+            )),
+        );
+
+        let mut state = test_std_state::<NopInput>();
+
+        test_resume(&completed, &mut state, tuple_list!(ifstage));
+    }
+
     #[derive(Debug)]
     pub struct PanicStage<S> {
         phantom: PhantomData<S>,
@@ -462,4 +484,34 @@ mod test {
 
         test_resume(&completed, &mut state, tuple_list!(ifstage, resetstage));
     }
+
+    #[test]
+    fn check_resumability_if_else_else_deep() {
+        let (completed, stages) = test_resume_stages();
+        let ifstage = IfElseStage::new(
+            |_, _, _, _| Ok(false),
+            tuple_list!(PanicStage::new()),
+            tuple_list!(IfElseStage::new(
+                |_, _, _, _| Ok(false),
+                tuple_list!(PanicStage::new()),
+                tuple_list!(IfElseStage::new(
+                    |_, _, _, _| Ok(false),
+                    tuple_list!(PanicStage::new()),
+                    tuple_list!(IfElseStage::new(
+                        |_, _, _, _| Ok(false),
+                        tuple_list!(PanicStage::new()),
+                        tuple_list!(IfElseStage::new(
+                            |_, _, _, _| Ok(false),
+                            tuple_list!(PanicStage::new()),
+                            stages,
+                        )),
+                    )),
+                )),
+            )),
+        );
+
+        let mut state = test_std_state::<NopInput>();
+
+        test_resume(&completed, &mut state, tuple_list!(ifstage));
+    }
 }