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

fixed bug in llmp test

Browse Source
This commit is contained in:
Dominik Maier 2020-11-22 04:30:12 +01:00
parent 9ffc34b973
commit 2760cb5099
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
llmp_test/src/main.rs
View File

@ -40,20 +40,27 @@ unsafe fn test_adder_clientloop(client: *mut llmp_client, _data: *mut c_void) ->
let mut last_result: u32 = 0; let mut last_result: u32 = 0;
let mut current_result: u32 = 0; let mut current_result: u32 = 0;
loop { loop {
let mut msg_counter = 0;
loop { loop {
let last_msg = llmp_client_recv(client); let last_msg = llmp_client_recv(client);
if last_msg == 0 as *mut llmp_message { if last_msg == 0 as *mut llmp_message {
break; break;
} }
msg_counter += 1;
match (*last_msg).tag { match (*last_msg).tag {
TAG_SIMPLE_U32_V1 => { TAG_SIMPLE_U32_V1 => {
current_result = last_result.wrapping_add(u32_from_msg(last_msg)); current_result = current_result.wrapping_add(u32_from_msg(last_msg));
} }
_ => println!("Adder Client ignored unknown message {}", (*last_msg).tag), _ => println!("Adder Client ignored unknown message {}", (*last_msg).tag),
}; };
} }
if current_result != last_result { if current_result != last_result {
println!(
"Adder handled {} messages, reporting {} to broker",
msg_counter, current_result
);
let llmp_message = llmp_client_alloc_next(client, size_of::<u32>()); let llmp_message = llmp_client_alloc_next(client, size_of::<u32>());
std::ptr::copy( std::ptr::copy(
current_result.to_be_bytes().as_ptr(), current_result.to_be_bytes().as_ptr(),