luckyimnotanigger/tun2proxy
1
0
Fork 0
mirror of https://github.com/tun2proxy/tun2proxy.git synced 2025-06-08 07:37:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

refine code

Browse source
This commit is contained in:
ssrlive 2024-09-14 21:38:15 +08:00
parent c583e884b5
commit e817257866
1 changed files with 10 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

19
src/win_svc.rs
View file

@ -63,14 +63,15 @@ fn run_service(_arguments: Vec<std::ffi::OsString>) -> Result<(), crate::BoxErro
// Tell the system that the service is running now // Tell the system that the service is running now
status_handle.set_service_status(next_status.clone())?; status_handle.set_service_status(next_status.clone())?;
let args = crate::Args::parse_args(); // main logic here
{
let args = crate::Args::parse_args();
let default = format!("{:?},trust_dns_proto=warn", args.verbosity); let default = format!("{:?},trust_dns_proto=warn", args.verbosity);
env_logger::Builder::from_env(env_logger::Env::default().default_filter_or(default)).init(); env_logger::Builder::from_env(env_logger::Env::default().default_filter_or(default)).init();
let rt = tokio::runtime::Builder::new_multi_thread().enable_all().build()?; let rt = tokio::runtime::Builder::new_multi_thread().enable_all().build()?;
rt.block_on(async { rt.block_on(async {
{
unsafe extern "C" fn traffic_cb(status: *const crate::TrafficStatus, _: *mut std::ffi::c_void) { unsafe extern "C" fn traffic_cb(status: *const crate::TrafficStatus, _: *mut std::ffi::c_void) {
let status = &*status; let status = &*status;
log::debug!("Traffic: ▲ {} : ▼ {}", status.tx, status.rx); log::debug!("Traffic: ▲ {} : ▼ {}", status.tx, status.rx);
@ -80,9 +81,9 @@ fn run_service(_arguments: Vec<std::ffi::OsString>) -> Result<(), crate::BoxErro
if let Err(err) = crate::desktop_run_async(args, shutdown_token).await { if let Err(err) = crate::desktop_run_async(args, shutdown_token).await {
log::error!("main loop error: {}", err); log::error!("main loop error: {}", err);
} }
} Ok::<(), crate::Error>(())
Ok::<(), crate::Error>(()) })?;
})?; }
// Tell the system that the service is stopped now // Tell the system that the service is stopped now
next_status.current_state = windows_service::service::ServiceState::Stopped; next_status.current_state = windows_service::service::ServiceState::Stopped;