diff --git a/src/gui/app.rs b/src/gui/app.rs index d93460b..890ca27 100644 --- a/src/gui/app.rs +++ b/src/gui/app.rs @@ -7,6 +7,7 @@ use crate::{ }, util::{ consts::{colours, game::WINDOW_TITLE, path}, + firewall::Firewall, persistent_state::PersistentState, system_info::SystemInfo, win, @@ -49,6 +50,7 @@ pub struct App { stage: Stage, pub flags: Flags, pub system_info: SystemInfo, + pub firewall: Firewall, pub anti_afk: features::anti_afk::AntiAfk, empty_session: features::empty_session::EmptySession, force_close: features::force_close::ForceClose, diff --git a/src/util/firewall.rs b/src/util/firewall.rs index 1ece93b..cbba3e5 100644 --- a/src/util/firewall.rs +++ b/src/util/firewall.rs @@ -11,17 +11,20 @@ use windows::{ core::BSTR, }; +#[derive(Debug)] pub struct Firewall { policy: INetFwPolicy2, } -impl Firewall { - pub fn new() -> Result> { - Ok(Self { - policy: unsafe { CoCreateInstance(&NetFwPolicy2, None, CLSCTX_INPROC_SERVER) }?, - }) +impl Default for Firewall { + fn default() -> Self { + Self { + policy: unsafe { CoCreateInstance(&NetFwPolicy2, None, CLSCTX_INPROC_SERVER) }.unwrap(), + } } +} +impl Firewall { pub fn add( &self, name: &str,