From a6e4cac56beab7e894e3b4c5b07425dcf9f67121 Mon Sep 17 00:00:00 2001
From: Andrea Fioraldi <andreafioraldi@gmail.com>
Date: Mon, 25 Jul 2022 10:35:53 +0200
Subject: [PATCH] prelude module (#709)

---
 libafl/src/bolts/mod.rs | 32 ++++++++++++++++++++++++++++++++
 libafl/src/lib.rs       | 20 ++++++++++++++++++++
 2 files changed, 52 insertions(+)

diff --git a/libafl/src/bolts/mod.rs b/libafl/src/bolts/mod.rs
index feb915d3c9..3e6497a58f 100644
--- a/libafl/src/bolts/mod.rs
+++ b/libafl/src/bolts/mod.rs
@@ -151,3 +151,35 @@ pub fn format_duration_hms(duration: &time::Duration) -> String {
     let secs = duration.as_secs();
     format!("{}h-{}m-{}s", (secs / 60) / 60, (secs / 60) % 60, secs % 60)
 }
+
+/// The purpose of this module is to alleviate imports of the bolts by adding a glob import.
+pub mod bolts_prelude {
+    pub use super::anymap::*;
+    #[cfg(feature = "std")]
+    pub use super::build_id::*;
+    #[cfg(all(
+        any(feature = "cli", feature = "frida_cli", feature = "qemu_cli"),
+        feature = "std"
+    ))]
+    pub use super::cli::*;
+    #[cfg(feature = "llmp_compression")]
+    pub use super::compress::*;
+    #[cfg(feature = "std")]
+    pub use super::core_affinity::*;
+    pub use super::cpu::*;
+    #[cfg(feature = "std")]
+    pub use super::fs::*;
+    #[cfg(feature = "std")]
+    pub use super::launcher::*;
+    pub use super::llmp::*;
+    #[cfg(all(feature = "std", unix))]
+    pub use super::minibsod::*;
+    pub use super::os::*;
+    pub use super::ownedref::*;
+    pub use super::rands::*;
+    pub use super::serdeany::*;
+    pub use super::shmem::*;
+    #[cfg(feature = "std")]
+    pub use super::staterestore::*;
+    pub use super::tuples::*;
+}
diff --git a/libafl/src/lib.rs b/libafl/src/lib.rs
index c1afb5cc64..8dfaeccc50 100644
--- a/libafl/src/lib.rs
+++ b/libafl/src/lib.rs
@@ -423,6 +423,26 @@ impl From<pyo3::PyErr> for Error {
 #[cfg(feature = "std")]
 impl std::error::Error for Error {}
 
+/// The purpose of this module is to alleviate imports of many components by adding a glob import.
+pub mod prelude {
+    pub use super::bolts::bolts_prelude::*;
+    pub use super::bolts::*;
+    pub use super::corpus::*;
+    pub use super::events::*;
+    pub use super::executors::*;
+    pub use super::feedbacks::*;
+    pub use super::fuzzer::*;
+    pub use super::generators::*;
+    pub use super::inputs::*;
+    pub use super::monitors::*;
+    pub use super::mutators::*;
+    pub use super::observers::*;
+    pub use super::schedulers::*;
+    pub use super::stages::*;
+    pub use super::state::*;
+    pub use super::*;
+}
+
 // TODO: no_std test
 #[cfg(feature = "std")]
 #[cfg(test)]