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

Discard no tc mm (follow-up) (#2584)

Browse Source 
* forgot to set the flags
This commit is contained in:
Romain Malmain 2024-10-03 20:26:28 +02:00 committed by GitHub
parent 9a50868058
commit 15157e0b72
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -9,7 +9,7 @@ use std::{
#[cfg(feature = "llmp_compression")]
use libafl_bolts::llmp::LLMP_FLAG_COMPRESSED;
use libafl_bolts::{
llmp::{Flags, LlmpBrokerInner, LlmpHook, LlmpMsgHookResult, Tag},
llmp::{Flags, LlmpBrokerInner, LlmpHook, LlmpMsgHookResult, Tag, LLMP_FLAG_FROM_MM},
ownedref::OwnedRef,
shmem::ShMemProvider,
ClientId, Error,
@ -255,20 +255,22 @@ where
let msg = msg.into_owned().unwrap().into_vec();
#[cfg(feature = "llmp_compression")]
match state_wr_lock.compressor().maybe_compress(msg.as_ref()) {
Some(comp_buf) => {
Ok((_LLMP_TAG_TO_MAIN, LLMP_FLAG_COMPRESSED, comp_buf))
}
None => Ok((_LLMP_TAG_TO_MAIN, Flags(0), msg)),
Some(comp_buf) => Ok((
_LLMP_TAG_TO_MAIN,
LLMP_FLAG_COMPRESSED | LLMP_FLAG_FROM_MM,
comp_buf,
)),
None => Ok((_LLMP_TAG_TO_MAIN, LLMP_FLAG_FROM_MM, msg)),
}
#[cfg(not(feature = "llmp_compression"))]
Ok((_LLMP_TAG_TO_MAIN, Flags(0), msg))
Ok((_LLMP_TAG_TO_MAIN, LLMP_FLAG_FROM_MM, msg))
}
MultiMachineMsg::Event(evt) => {
let evt = evt.into_owned().unwrap();
let (inner_flags, buf) =
Self::try_compress(&mut state_wr_lock, evt.as_ref())?;
Ok((_LLMP_TAG_TO_MAIN, inner_flags, buf))
Ok((_LLMP_TAG_TO_MAIN, inner_flags | LLMP_FLAG_FROM_MM, buf))
}
})
.collect();