@@ -9,7 +9,7 @@ use std::{
9
9
#[ cfg( feature = "llmp_compression" ) ]
10
10
use libafl_bolts:: llmp:: LLMP_FLAG_COMPRESSED ;
11
11
use libafl_bolts:: {
12
- llmp:: { Flags , LlmpBrokerInner , LlmpHook , LlmpMsgHookResult , Tag } ,
12
+ llmp:: { Flags , LlmpBrokerInner , LlmpHook , LlmpMsgHookResult , Tag , LLMP_FLAG_FROM_MM } ,
13
13
ownedref:: OwnedRef ,
14
14
shmem:: ShMemProvider ,
15
15
ClientId , Error ,
@@ -255,20 +255,22 @@ where
255
255
let msg = msg. into_owned ( ) . unwrap ( ) . into_vec ( ) ;
256
256
#[ cfg( feature = "llmp_compression" ) ]
257
257
match state_wr_lock. compressor ( ) . maybe_compress ( msg. as_ref ( ) ) {
258
- Some ( comp_buf) => {
259
- Ok ( ( _LLMP_TAG_TO_MAIN, LLMP_FLAG_COMPRESSED , comp_buf) )
260
- }
261
- None => Ok ( ( _LLMP_TAG_TO_MAIN, Flags ( 0 ) , msg) ) ,
258
+ Some ( comp_buf) => Ok ( (
259
+ _LLMP_TAG_TO_MAIN,
260
+ LLMP_FLAG_COMPRESSED | LLMP_FLAG_FROM_MM ,
261
+ comp_buf,
262
+ ) ) ,
263
+ None => Ok ( ( _LLMP_TAG_TO_MAIN, LLMP_FLAG_FROM_MM , msg) ) ,
262
264
}
263
265
#[ cfg( not( feature = "llmp_compression" ) ) ]
264
- Ok ( ( _LLMP_TAG_TO_MAIN, Flags ( 0 ) , msg) )
266
+ Ok ( ( _LLMP_TAG_TO_MAIN, LLMP_FLAG_FROM_MM , msg) )
265
267
}
266
268
MultiMachineMsg :: Event ( evt) => {
267
269
let evt = evt. into_owned ( ) . unwrap ( ) ;
268
270
let ( inner_flags, buf) =
269
271
Self :: try_compress ( & mut state_wr_lock, evt. as_ref ( ) ) ?;
270
272
271
- Ok ( ( _LLMP_TAG_TO_MAIN, inner_flags, buf) )
273
+ Ok ( ( _LLMP_TAG_TO_MAIN, inner_flags | LLMP_FLAG_FROM_MM , buf) )
272
274
}
273
275
} )
274
276
. collect ( ) ;
0 commit comments