raw_sync 0.1.5

Lightweight wrapper around OS synchronization primitives
Documentation
use std::ffi::CString;
use std::mem::size_of;
use std::ptr::null_mut;

use winapi::{
    shared::ntdef::{FALSE, NULL, TRUE},
    um::{
        handleapi::CloseHandle,
        synchapi::{CreateEventA, OpenEventA, ResetEvent, SetEvent, WaitForSingleObject},
        winbase::{INFINITE, WAIT_OBJECT_0},
        winnt::{EVENT_MODIFY_STATE, HANDLE, SYNCHRONIZE},
    },
};

use super::{EventImpl, EventInit, EventState};
use crate::{Result, Timeout};

pub struct Event {
    handle: HANDLE,
}
impl Drop for Event {
    fn drop(&mut self) {
        //trace!("CloseHandle(0x{:X})", self.handle as usize);
        unsafe { CloseHandle(self.handle) };
    }
}
impl EventInit for Event {
    fn size_of(_addr: Option<*mut u8>) -> usize {
        size_of::<u32>()
    }

    #[allow(clippy::new_ret_no_self)]
    unsafe fn new(mem: *mut u8, auto_reset: bool) -> Result<(Box<dyn EventImpl>, usize)> {
        let mut handle: HANDLE = NULL;
        let mut id: u32 = 0;
        while handle == NULL {
            id = rand::random::<u32>();
            let path = CString::new(format!("event_{}", id)).unwrap();
            //trace!("CreateEventA(NULL, '{:?}', '{}')",!auto_reset,path.to_string_lossy());
            handle = CreateEventA(
                null_mut(),
                if auto_reset { FALSE } else { TRUE } as _,
                FALSE as _,
                path.as_ptr() as *mut _,
            );
        }

        let obj: Box<dyn EventImpl> = Box::new(Event { handle });
        *(mem as *mut u32) = id;
        Ok((obj, Self::size_of(None)))
    }

    unsafe fn from_existing(mem: *mut u8) -> Result<(Box<dyn EventImpl>, usize)> {
        let id: u32 = *(mem as *mut u32);
        let path = CString::new(format!("event_{}", id)).unwrap();
        //trace!("OpenEventA('{}')", path.to_string_lossy());
        let handle = OpenEventA(
            EVENT_MODIFY_STATE | SYNCHRONIZE, // request full access
            FALSE as _,                       // handle not inheritable
            path.as_ptr() as *mut _,
        );

        if handle == NULL {
            return Err(From::from(format!(
                "Failed to open event {}",
                path.to_string_lossy()
            )));
        }

        Ok((Box::new(Event { handle }), Self::size_of(None)))
    }
}
impl EventImpl for Event {
    fn wait(&self, timeout: Timeout) -> Result<()> {
        //trace!("WaitForSingleObject(0x{:X})", self.handle as usize);
        let wait_res = unsafe {
            WaitForSingleObject(
                self.handle,
                match timeout {
                    Timeout::Infinite => INFINITE,
                    Timeout::Val(dur) => dur.as_millis() as _,
                },
            )
        };

        if wait_res == WAIT_OBJECT_0 {
            Ok(())
        } else {
            Err(From::from(format!(
                "Failed waiting for event : 0x{:X}",
                wait_res
            )))
        }
    }

    fn set(&self, state: EventState) -> Result<()> {
        let res = match state {
            EventState::Clear => {
                //trace!("ResetEvent(0x{:X})", self.handle as usize);
                unsafe { ResetEvent(self.handle) }
            }
            EventState::Signaled => {
                //trace!("SetEvent(0x{:X})", self.handle as usize);
                unsafe { SetEvent(self.handle) }
            }
        };

        if res != 0 {
            Ok(())
        } else {
            Err(From::from(format!(
                "Failed setting event state : 0x{:X}",
                res
            )))
        }
    }
}