Share objectives between nodes

libafl/Cargo.toml

@@ -142,6 +142,9 @@ unicode = ["libafl_bolts/alloc", "ahash/std", "serde/rc", "bitvec"]
 ## Enable multi-part input formats and mutators
 multipart_inputs = ["arrayvec", "rand_trait"]
 
+## Share objectives across nodes
+share_objectives = []
+
 #! ## LibAFL-Bolts Features
 
 ## Provide the `#[derive(SerdeAny)]` macro.

libafl/src/events/centralized.rs

@@ -287,22 +287,29 @@ where
     ) -> Result<(), Error> {
         if !self.is_main {
             // secondary node
-            let mut is_tc = false;
-            // Forward to main only if new tc or heartbeat
+            let mut is_tc_or_obj = false;
+            // Forward to main only if new tc, heartbeat, or optionally, a new objective
             let should_be_forwarded = match &mut event {
                 Event::NewTestcase { forward_id, .. } => {
                     *forward_id = Some(ClientId(self.inner.mgr_id().0 as u32));
-                    is_tc = true;
+                    is_tc_or_obj = true;
                     true
                 }
                 Event::UpdateExecStats { .. } => true, // send it but this guy won't be handled. the only purpose is to keep this client alive else the broker thinks it is dead and will dc it
+
+                #[cfg(feature = "share_objectives")]
+                Event::Objective { .. } => {
+                    is_tc_or_obj = true;
+                    true
+                }
+
                 Event::Stop => true,
                 _ => false,
             };
 
             if should_be_forwarded {
                 self.forward_to_main(&event)?;
-                if is_tc {
+                if is_tc_or_obj {
                     // early return here because we only send it to centralized not main broker.
                     return Ok(());
                 }
@@ -675,6 +682,15 @@ where
                     log::debug!("[{}] {} was discarded...)", process::id(), event_name);
                 }
             }
+
+            #[cfg(feature = "share_objectives")]
+            Event::Objective { .. } => {
+                log::debug!("Received new Objective");
+
+                self.hooks.on_fire_all(state, client_id, &event)?;
+                self.inner.fire(state, event)?;
+            }
+
             Event::Stop => {
                 state.request_stop();
             }

libafl/src/events/llmp/mgr.rs

@@ -461,6 +461,12 @@ where
                     }
                 }
             }
+
+            #[cfg(feature = "share_objectives")]
+            Event::Objective { .. } => {
+                log::debug!("Received new Objective");
+            }
+
             Event::CustomBuf { tag, buf } => {
                 for handler in &mut self.custom_buf_handlers {
                     if handler(state, &tag, &buf)? == CustomBufEventResult::Handled {

libafl/src/events/llmp/mod.rs

@@ -321,6 +321,13 @@ where
                 }
                 Ok(())
             }
+
+            #[cfg(feature = "share_objectives")]
+            Event::Objective { .. } => {
+                log::info!("Received new Objective");
+                Ok(())
+            }
+
             Event::CustomBuf { tag, buf } => {
                 for handler in &mut self.custom_buf_handlers {
                     if handler(state, &tag, &buf)? == CustomBufEventResult::Handled {

libafl/src/events/tcp.rs

@@ -644,6 +644,12 @@ where
                     log::info!("Added received Testcase as item #{item}");
                 }
             }
+
+            #[cfg(feature = "share_objectives")]
+            Event::Objective { .. } => {
+                log::info!("Received new Objective");
+            }
+
             Event::CustomBuf { tag, buf } => {
                 for handler in &mut self.custom_buf_handlers {
                     if handler(state, &tag, &buf)? == CustomBufEventResult::Handled {