123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410 |
- use crate::{
- config::{Config, IVerge},
- log_err,
- };
- use anyhow::{anyhow, Result};
- use auto_launch::{AutoLaunch, AutoLaunchBuilder};
- use once_cell::sync::OnceCell;
- use parking_lot::Mutex;
- use std::env::current_exe;
- use std::sync::Arc;
- use sysproxy::{Autoproxy, Sysproxy};
- use tauri::async_runtime::Mutex as TokioMutex;
- pub struct Sysopt {
-
- cur_sysproxy: Arc<Mutex<Option<Sysproxy>>>,
-
-
- old_sysproxy: Arc<Mutex<Option<Sysproxy>>>,
-
- cur_autoproxy: Arc<Mutex<Option<Autoproxy>>>,
-
-
- old_autoproxy: Arc<Mutex<Option<Autoproxy>>>,
-
- auto_launch: Arc<Mutex<Option<AutoLaunch>>>,
-
- guard_state: Arc<TokioMutex<bool>>,
- }
- #[cfg(target_os = "windows")]
- static DEFAULT_BYPASS: &str = "localhost;127.*;192.168.*;10.*;172.16.*;172.17.*;172.18.*;172.19.*;172.20.*;172.21.*;172.22.*;172.23.*;172.24.*;172.25.*;172.26.*;172.27.*;172.28.*;172.29.*;172.30.*;172.31.*;<local>";
- #[cfg(target_os = "linux")]
- static DEFAULT_BYPASS: &str = "localhost,127.0.0.1,192.168.0.0/16,10.0.0.0/8,172.16.0.0/12,::1";
- #[cfg(target_os = "macos")]
- static DEFAULT_BYPASS: &str =
- "127.0.0.1,192.168.0.0/16,10.0.0.0/8,172.16.0.0/12,localhost,*.local,*.crashlytics.com,<local>";
- fn get_bypass() -> String {
- let bypass = DEFAULT_BYPASS.to_string();
- let res = {
- let verge = Config::verge();
- let verge = verge.latest();
- verge.system_proxy_bypass.clone()
- };
- let custom_bypass = match res {
- Some(bypass) => bypass,
- None => "".to_string(),
- };
- #[cfg(target_os = "windows")]
- let bypass = if custom_bypass.is_empty() {
- bypass
- } else {
- format!("{};{}", bypass, custom_bypass)
- };
- #[cfg(not(target_os = "windows"))]
- let bypass = if custom_bypass.is_empty() {
- bypass
- } else {
- format!("{},{}", bypass, custom_bypass)
- };
- bypass
- }
- impl Sysopt {
- pub fn global() -> &'static Sysopt {
- static SYSOPT: OnceCell<Sysopt> = OnceCell::new();
- SYSOPT.get_or_init(|| Sysopt {
- cur_sysproxy: Arc::new(Mutex::new(None)),
- old_sysproxy: Arc::new(Mutex::new(None)),
- cur_autoproxy: Arc::new(Mutex::new(None)),
- old_autoproxy: Arc::new(Mutex::new(None)),
- auto_launch: Arc::new(Mutex::new(None)),
- guard_state: Arc::new(TokioMutex::new(false)),
- })
- }
-
- pub fn init_sysproxy(&self) -> Result<()> {
- let port = Config::verge()
- .latest()
- .verge_mixed_port
- .unwrap_or(Config::clash().data().get_mixed_port());
- let pac_port = IVerge::get_singleton_port();
- let (enable, pac) = {
- let verge = Config::verge();
- let verge = verge.latest();
- (
- verge.enable_system_proxy.unwrap_or(false),
- verge.proxy_auto_config.unwrap_or(false),
- )
- };
- let mut sys = Sysproxy {
- enable,
- host: String::from("127.0.0.1"),
- port,
- bypass: get_bypass(),
- };
- let mut auto = Autoproxy {
- enable,
- url: format!("http://127.0.0.1:{pac_port}/commands/pac"),
- };
- if pac {
- sys.enable = false;
- let old = Sysproxy::get_system_proxy().ok();
- sys.set_system_proxy()?;
- *self.old_sysproxy.lock() = old;
- *self.cur_sysproxy.lock() = Some(sys);
- let old = Autoproxy::get_auto_proxy().ok();
- auto.set_auto_proxy()?;
- *self.old_autoproxy.lock() = old;
- *self.cur_autoproxy.lock() = Some(auto);
- } else {
- auto.enable = false;
- let old = Autoproxy::get_auto_proxy().ok();
- auto.set_auto_proxy()?;
- *self.old_autoproxy.lock() = old;
- *self.cur_autoproxy.lock() = Some(auto);
- let old = Sysproxy::get_system_proxy().ok();
- sys.set_system_proxy()?;
- *self.old_sysproxy.lock() = old;
- *self.cur_sysproxy.lock() = Some(sys);
- }
-
- self.guard_proxy();
- Ok(())
- }
-
- pub fn update_sysproxy(&self) -> Result<()> {
- let mut cur_sysproxy = self.cur_sysproxy.lock();
- let old_sysproxy = self.old_sysproxy.lock();
- let mut cur_autoproxy = self.cur_autoproxy.lock();
- let old_autoproxy = self.old_autoproxy.lock();
- let (enable, pac) = {
- let verge = Config::verge();
- let verge = verge.latest();
- (
- verge.enable_system_proxy.unwrap_or(false),
- verge.proxy_auto_config.unwrap_or(false),
- )
- };
- if pac && (cur_autoproxy.is_none() || old_autoproxy.is_none()) {
- drop(cur_autoproxy);
- drop(old_autoproxy);
- return self.init_sysproxy();
- }
- if !pac && (cur_sysproxy.is_none() || old_sysproxy.is_none()) {
- drop(cur_sysproxy);
- drop(old_sysproxy);
- return self.init_sysproxy();
- }
- let port = Config::verge()
- .latest()
- .verge_mixed_port
- .unwrap_or(Config::clash().data().get_mixed_port());
- let pac_port = IVerge::get_singleton_port();
- let mut sysproxy = cur_sysproxy.take().unwrap();
- sysproxy.bypass = get_bypass();
- sysproxy.port = port;
- let mut autoproxy = cur_autoproxy.take().unwrap();
- autoproxy.url = format!("http://127.0.0.1:{pac_port}/commands/pac");
- if pac {
- sysproxy.enable = false;
- sysproxy.set_system_proxy()?;
- *cur_sysproxy = Some(sysproxy);
- autoproxy.enable = enable;
- autoproxy.set_auto_proxy()?;
- *cur_autoproxy = Some(autoproxy);
- } else {
- autoproxy.enable = false;
- autoproxy.set_auto_proxy()?;
- *cur_autoproxy = Some(autoproxy);
- sysproxy.enable = enable;
- sysproxy.set_system_proxy()?;
- *cur_sysproxy = Some(sysproxy);
- }
- Ok(())
- }
-
- pub fn reset_sysproxy(&self) -> Result<()> {
- let mut cur_sysproxy = self.cur_sysproxy.lock();
- let mut old_sysproxy = self.old_sysproxy.lock();
- let mut cur_autoproxy = self.cur_autoproxy.lock();
- let mut old_autoproxy = self.old_autoproxy.lock();
- let cur_sysproxy = cur_sysproxy.take();
- let cur_autoproxy = cur_autoproxy.take();
- if let Some(mut old) = old_sysproxy.take() {
-
-
- let port_same = cur_sysproxy.map_or(true, |cur| old.port == cur.port);
- if old.enable && port_same {
- old.enable = false;
- log::info!(target: "app", "reset proxy by disabling the original proxy");
- } else {
- log::info!(target: "app", "reset proxy to the original proxy");
- }
- old.set_system_proxy()?;
- } else if let Some(mut cur @ Sysproxy { enable: true, .. }) = cur_sysproxy {
-
- log::info!(target: "app", "reset proxy by disabling the current proxy");
- cur.enable = false;
- cur.set_system_proxy()?;
- } else {
- log::info!(target: "app", "reset proxy with no action");
- }
- if let Some(mut old) = old_autoproxy.take() {
-
-
- let url_same = cur_autoproxy.map_or(true, |cur| old.url == cur.url);
- if old.enable && url_same {
- old.enable = false;
- log::info!(target: "app", "reset proxy by disabling the original proxy");
- } else {
- log::info!(target: "app", "reset proxy to the original proxy");
- }
- old.set_auto_proxy()?;
- } else if let Some(mut cur @ Autoproxy { enable: true, .. }) = cur_autoproxy {
-
- log::info!(target: "app", "reset proxy by disabling the current proxy");
- cur.enable = false;
- cur.set_auto_proxy()?;
- } else {
- log::info!(target: "app", "reset proxy with no action");
- }
- Ok(())
- }
-
- pub fn init_launch(&self) -> Result<()> {
- let app_exe = current_exe()?;
-
- let app_name = app_exe
- .file_stem()
- .and_then(|f| f.to_str())
- .ok_or(anyhow!("failed to get file stem"))?;
- let app_path = app_exe
- .as_os_str()
- .to_str()
- .ok_or(anyhow!("failed to get app_path"))?
- .to_string();
-
- #[cfg(target_os = "windows")]
- let app_path = format!("\"{app_path}\"");
-
- #[cfg(target_os = "macos")]
- let app_path = (|| -> Option<String> {
- let path = std::path::PathBuf::from(&app_path);
- let path = path.parent()?.parent()?.parent()?;
- let extension = path.extension()?.to_str()?;
- match extension == "app" {
- true => Some(path.as_os_str().to_str()?.to_string()),
- false => None,
- }
- })()
- .unwrap_or(app_path);
-
- #[cfg(target_os = "linux")]
- let app_path = {
- use crate::core::handle::Handle;
- use tauri::Manager;
- let handle = Handle::global();
- match handle.app_handle.lock().as_ref() {
- Some(app_handle) => {
- let appimage = app_handle.env().appimage;
- appimage
- .and_then(|p| p.to_str().map(|s| s.to_string()))
- .unwrap_or(app_path)
- }
- None => app_path,
- }
- };
- let auto = AutoLaunchBuilder::new()
- .set_app_name(app_name)
- .set_app_path(&app_path)
- .build()?;
- *self.auto_launch.lock() = Some(auto);
- Ok(())
- }
-
- pub fn update_launch(&self) -> Result<()> {
- let auto_launch = self.auto_launch.lock();
- if auto_launch.is_none() {
- drop(auto_launch);
- return self.init_launch();
- }
- let enable = { Config::verge().latest().enable_auto_launch };
- let enable = enable.unwrap_or(false);
- let auto_launch = auto_launch.as_ref().unwrap();
- match enable {
- true => auto_launch.enable()?,
- false => log_err!(auto_launch.disable()),
- };
- Ok(())
- }
-
-
- pub fn guard_proxy(&self) {
- use tokio::time::{sleep, Duration};
- let guard_state = self.guard_state.clone();
- tauri::async_runtime::spawn(async move {
-
- let mut state = guard_state.lock().await;
- if *state {
- return;
- }
- *state = true;
- drop(state);
-
- let mut wait_secs = 10u64;
- loop {
- sleep(Duration::from_secs(wait_secs)).await;
- let (enable, guard, guard_duration, pac) = {
- let verge = Config::verge();
- let verge = verge.latest();
- (
- verge.enable_system_proxy.unwrap_or(false),
- verge.enable_proxy_guard.unwrap_or(false),
- verge.proxy_guard_duration.unwrap_or(10),
- verge.proxy_auto_config.unwrap_or(false),
- )
- };
-
- if !enable || !guard {
- break;
- }
-
- wait_secs = guard_duration;
- log::debug!(target: "app", "try to guard the system proxy");
- let port = {
- Config::verge()
- .latest()
- .verge_mixed_port
- .unwrap_or(Config::clash().data().get_mixed_port())
- };
- let pac_port = IVerge::get_singleton_port();
- if pac {
- let autoproxy = Autoproxy {
- enable: true,
- url: format!("http://127.0.0.1:{pac_port}/commands/pac"),
- };
- log_err!(autoproxy.set_auto_proxy());
- } else {
- let sysproxy = Sysproxy {
- enable: true,
- host: "127.0.0.1".into(),
- port,
- bypass: get_bypass(),
- };
- log_err!(sysproxy.set_system_proxy());
- }
- }
- let mut state = guard_state.lock().await;
- *state = false;
- drop(state);
- });
- }
- }
|