X-Git-Url: https://code.octet-stream.net/netwatcher/blobdiff_plain/9ea683eddb05362f02bca60e4337955dfc5871dc..fd5b83a034dd0cc1751ddbfe1df813fdf798b8b1:/src/watch_win.rs?ds=sidebyside diff --git a/src/watch_win.rs b/src/watch_win.rs index 41612b8..3a4a301 100644 --- a/src/watch_win.rs +++ b/src/watch_win.rs @@ -23,7 +23,7 @@ struct WatchState { /// The last result that we captured, for diffing prev_list: List, /// User's callback - cb: Box, + cb: Box, } pub(crate) struct WatchHandle { @@ -39,7 +39,7 @@ impl Drop for WatchHandle { } } -pub(crate) fn watch_interfaces( +pub(crate) fn watch_interfaces( callback: F, ) -> Result { let state = Box::pin(Mutex::new(WatchState { @@ -60,10 +60,8 @@ pub(crate) fn watch_interfaces( }; match res { NO_ERROR => { - // Trigger an initial update. - // This is allowed to race with true updates because it - // will always calculate a diff and discard no-ops. - handle_notif(&mut state.lock().unwrap()); + // Trigger an initial update + handle_notif(&mut state.lock().unwrap(), crate::list::list_interfaces()?); // Then return the handle Ok(WatchHandle { hnd, _state: state }) } @@ -81,15 +79,19 @@ unsafe extern "system" fn notif( ) { let state_ptr = ctx as *const Mutex; unsafe { - let state_guard = &mut *state_ptr.as_ref().unwrap().lock().unwrap(); - handle_notif(state_guard); + let state_guard = &mut *state_ptr + .as_ref() + .expect("callback ctx should never be null") + .lock() + .unwrap(); + let Ok(new_list) = crate::list::list_interfaces() else { + return; + }; + handle_notif(state_guard, new_list); } } -fn handle_notif(state: &mut WatchState) { - let Ok(new_list) = crate::list::list_interfaces() else { - return; - }; +fn handle_notif(state: &mut WatchState, new_list: List) { if new_list == state.prev_list { return; }