diff options
| -rw-r--r-- | talpid_openvpn_plugin/src/ffi/consts.rs | 75 |
1 files changed, 66 insertions, 9 deletions
diff --git a/talpid_openvpn_plugin/src/ffi/consts.rs b/talpid_openvpn_plugin/src/ffi/consts.rs index ca82606751..6b18b77466 100644 --- a/talpid_openvpn_plugin/src/ffi/consts.rs +++ b/talpid_openvpn_plugin/src/ffi/consts.rs @@ -20,6 +20,42 @@ pub const OPENVPN_PLUGIN_TLS_FINAL: c_int = 10; pub const OPENVPN_PLUGIN_ENABLE_PF: c_int = 11; pub const OPENVPN_PLUGIN_ROUTE_PREDOWN: c_int = 12; pub const OPENVPN_PLUGIN_N: c_int = 13; +error_chain!{ + errors { + InvalidEnumVariant { + description("Integer does not match any enum variant") + } + } +} + +#[derive(Debug, Copy, Clone, Eq, PartialEq, Hash)] +pub enum OpenVpnPluginEvent { + Up = 0, + Down = 1, + RouteUp = 2, + IpChange = 3, + TlsVerify = 4, + AuthUserPassVerify = 5, + ClientConnect = 6, + ClientDisconnect = 7, + LearnAddress = 8, + ClientConnectV2 = 9, + TlsFinal = 10, + EnablePf = 11, + RoutePredown = 12, + N = 13, +} + +impl OpenVpnPluginEvent { + pub fn from_int(i: c_int) -> Result<OpenVpnPluginEvent> { + if i >= OpenVpnPluginEvent::Up as c_int && i <= OpenVpnPluginEvent::N as c_int { + Ok(unsafe { ::std::mem::transmute_copy::<c_int, OpenVpnPluginEvent>(&i) }) + } else { + Err(ErrorKind::InvalidEnumVariant.into()) + } + } +} + lazy_static! { pub static ref PLUGIN_EVENT_NAMES: HashMap<c_int, &'static str> = { @@ -60,20 +96,41 @@ mod tests { use super::*; #[test] - fn plugin_event_name_up() { - let name = plugin_event_name(0); - assert_eq!("PLUGIN_UP", name); + fn from_int_first() { + let result = OpenVpnPluginEvent::from_int(0); + assert_matches!(result, Ok(OpenVpnPluginEvent::Up)); + } + + #[test] + fn from_int_last() { + let result = OpenVpnPluginEvent::from_int(13); + assert_matches!(result, Ok(OpenVpnPluginEvent::N)); + } + + #[test] + fn from_int_all_valid() { + for i in 0..13 { + if OpenVpnPluginEvent::from_int(0).is_err() { + panic!("{} not covered", i); + } + } + } + + #[test] + fn from_int_negative() { + let result = OpenVpnPluginEvent::from_int(-5); + assert_matches!(result, Err(Error(ErrorKind::InvalidEnumVariant, _))); } #[test] - fn plugin_event_name_n() { - let name = plugin_event_name(13); - assert_eq!("PLUGIN_N", name); + fn from_int_invalid() { + let result = OpenVpnPluginEvent::from_int(14); + assert_matches!(result, Err(Error(ErrorKind::InvalidEnumVariant, _))); } #[test] - fn plugin_event_name_not_existing() { - let name = plugin_event_name(-15); - assert_eq!("UNKNOWN", name); + fn event_enum_to_str() { + let result = format!("{:?}", OpenVpnPluginEvent::Up); + assert_eq!("Up", result); } } |