use std::cell::Cell;

use libafl_bolts::os::{fork, ForkResult};
use libafl_bolts::shmem::{ShMem, ShMemProvider as _};

fn do_stuff(shmem: &impl ShMem) {
    let p = shmem.as_ptr_of::<Cell<usize>>().unwrap();
    for i in 0..128 {
        let val = unsafe { (*p).get() };
        println!("val = {val}");
        unsafe { (*p).set(val + 1) };
    }
}

fn main() {
    let mut shmem_provider = libafl_bolts::shmem::MmapShMemProvider::default();
    let shmem = shmem_provider.new_on_shmem(Cell::new(0usize)).unwrap();
    shmem_provider.pre_fork().unwrap();
    match unsafe { fork() }.unwrap() {
        ForkResult::Parent(handle) => {
            shmem_provider.post_fork(false).unwrap();
            do_stuff(&shmem);
            handle.status();
        }
        ForkResult::Child => {
            shmem_provider.post_fork(true).unwrap();
            do_stuff(&shmem);
            std::process::abort();
        }
    }
    let p = shmem.as_ptr_of::<Cell<usize>>().unwrap();
    assert_eq!(256, unsafe { (*p).get() });
}