From dd6200b7111adfa883aa191c4e74ce437a7ad8b1 Mon Sep 17 00:00:00 2001
From: Andrea Fioraldi <andreafioraldi@gmail.com>
Date: Wed, 16 Dec 2020 13:55:35 +0100
Subject: [PATCH] corpus and state serde test

---
 afl/src/engines/mod.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/afl/src/engines/mod.rs b/afl/src/engines/mod.rs
index 2a88d09407..5475701d00 100644
--- a/afl/src/engines/mod.rs
+++ b/afl/src/engines/mod.rs
@@ -420,7 +420,7 @@ mod tests {
     use crate::inputs::bytes::BytesInput;
     use crate::mutators::{mutation_bitflip, ComposedByMutations, StdScheduledMutator};
     use crate::stages::mutational::StdMutationalStage;
-    use crate::tuples::tuple_list;
+    use crate::tuples::{tuple_list_type, tuple_list};
     use crate::utils::StdRand;
 
     fn harness<I>(_executor: &dyn Executor<I>, _buf: &[u8]) -> ExitKind {
@@ -458,5 +458,13 @@ mod tests {
                 )
                 .expect(&format!("Error in iter {}", i));
         }
+
+        let state_serialized = postcard::to_allocvec(&state).unwrap();
+        let state_deserialized: State<BytesInput, StdRand, tuple_list_type!(), tuple_list_type!()> = postcard::from_bytes(state_serialized.as_slice()).unwrap();
+        assert_eq!(state.executions, state_deserialized.executions);
+
+        let corpus_serialized = postcard::to_allocvec(&corpus).unwrap();
+        let corpus_deserialized: InMemoryCorpus<BytesInput, StdRand> = postcard::from_bytes(corpus_serialized.as_slice()).unwrap();
+        assert_eq!(corpus.count(), corpus_deserialized.count());
     }
 }