SYS-OSS/FRET-LibAFL
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

libafl multimachine: disable ratelimiting (#2558)

Browse Source 
* disable rate limiting for now

* fix

* clippy

---------

Co-authored-by: Dongjia "toka" Zhang <tokazerkje@outlook.com>
This commit is contained in:
Romain Malmain 2024-09-30 15:57:10 +02:00 committed by GitHub
parent 17def0390d
commit 173aeddbcc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 29 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libafl/src/events/broker_hooks/centralized_multi_machine.rs
View File

@ -210,6 +210,8 @@ where
.send_interesting_event_to_nodes(&mm_msg)
.await?;
log::debug!("msg sent.");
Ok(())
});

45
libafl/src/events/multi_machine.rs
View File

@ -32,7 +32,7 @@ use crate::{
inputs::{Input, NopInput},
};
const MAX_NB_RECEIVED_AT_ONCE: usize = 10;
// const MAX_NB_RECEIVED_AT_ONCE: usize = 100;
#[bitflags(default = SendToParent | SendToChildren)]
#[repr(u8)]
@ -394,7 +394,9 @@ where
// 1. Read msg size
let mut node_msg_len: [u8; 4] = [0; 4];
log::debug!("Receiving msg len...");
stream.read_exact(&mut node_msg_len).await?;
log::debug!("msg len received.");
let node_msg_len = u32::from_le_bytes(node_msg_len) as usize;
// 2. Read msg
@ -404,7 +406,9 @@ where
unsafe {
node_msg.set_len(node_msg_len);
}
log::debug!("Receiving msg...");
stream.read_exact(node_msg.as_mut_slice()).await?;
log::debug!("msg received.");
let node_msg = node_msg.into_boxed_slice();
Ok(Some(MultiMachineMsg::from_llmp_msg(node_msg)))
@ -420,16 +424,19 @@ where
let msg_len = u32::to_le_bytes(serialized_msg.len() as u32);
// 0. Write the dummy byte
log::debug!("Sending dummy byte");
log::debug!("Sending dummy byte...");
stream.write_all(&[DUMMY_BYTE]).await?;
log::debug!("dummy byte sent.");
// 1. Write msg size
log::debug!("Sending msg len");
log::debug!("Sending msg len...");
stream.write_all(&msg_len).await?;
log::debug!("msg len sent.");
// 2. Write msg
log::debug!("Sending msg");
log::debug!("Sending msg...");
stream.write_all(serialized_msg).await?;
log::debug!("msg sent.");
Ok(())
}
@ -483,11 +490,11 @@ where
{
let mut ids_to_remove: Vec<NodeId> = Vec::new();
for (child_id, child_stream) in &mut self.children {
log::debug!("Sending to child...");
if (Self::write_msg(child_stream, msg).await).is_err() {
log::debug!("Sending to child {child_id:?}...");
if let Err(err) = Self::write_msg(child_stream, msg).await {
// most likely the child disconnected. drop the connection later on and continue.
log::debug!(
"The child disconnected. We won't try to communicate with it again."
"The child disconnected. We won't try to communicate with it again. Error: {err:?}"
);
ids_to_remove.push(*child_id);
}
@ -510,15 +517,17 @@ where
msgs: &mut Vec<MultiMachineMsg<'a, I>>,
) -> Result<(), Error> {
log::debug!("Checking for new events from other nodes...");
let mut nb_received = 0usize;
// let mut nb_received = 0usize;
// Our (potential) parent could have something for us
if let Some(parent) = &mut self.parent {
loop {
// Exit if received a lot of inputs at once.
if nb_received > MAX_NB_RECEIVED_AT_ONCE {
return Ok(());
}
// TODO: this causes problems in some cases, it could freeze all fuzzer instances.
// if nb_received > MAX_NB_RECEIVED_AT_ONCE {
// log::debug!("hitting MAX_NB_RECEIVED_AT_ONCE limit...");
// return Ok(());
// }
log::debug!("Receiving from parent...");
match Self::read_msg(parent).await {
@ -526,7 +535,7 @@ where
log::debug!("Received event from parent");
// The parent has something for us, we store it
msgs.push(msg);
nb_received += 1;
// nb_received += 1;
}
Ok(None) => {
@ -562,17 +571,17 @@ where
for (child_id, child_stream) in &mut self.children {
loop {
// Exit if received a lot of inputs at once.
if nb_received > MAX_NB_RECEIVED_AT_ONCE {
return Ok(());
}
// if nb_received > MAX_NB_RECEIVED_AT_ONCE {
// return Ok(());
//}
log::debug!("Receiving from child {:?}...", child_id);
log::debug!("Receiving from child {child_id:?}...");
match Self::read_msg(child_stream).await {
Ok(Some(msg)) => {
// The parent has something for us, we store it
log::debug!("Received event from child!");
msgs.push(msg);
nb_received += 1;
// nb_received += 1;
}
Ok(None) => {
@ -584,7 +593,7 @@ where
Err(Error::OsError(e, _, _)) => {
// most likely the parent disconnected. drop the connection
log::error!(
"The parent disconnected. We won't try to communicate with it again."
"The child disconnected. We won't try to communicate with it again."
);
log::error!("Error: {e:?}");
ids_to_remove.push(*child_id);