-
Notifications
You must be signed in to change notification settings - Fork 43
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
6 changed files
with
153 additions
and
43 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,99 @@ | ||
| #![cfg(windows)] | ||
|
|
||
| const SERVICE_NAME: &str = "overtls"; | ||
|
|
||
| windows_service::define_windows_service!(ffi_service_main, my_service_main); | ||
|
|
||
| pub fn start_service() -> Result<(), windows_service::Error> { | ||
| // Register generated `ffi_service_main` with the system and start the service, | ||
| // blocking this thread until the service is stopped. | ||
| windows_service::service_dispatcher::start(SERVICE_NAME, ffi_service_main)?; | ||
| Ok(()) | ||
| } | ||
|
|
||
| fn my_service_main(arguments: Vec<std::ffi::OsString>) { | ||
| // The entry point where execution will start on a background thread after a call to | ||
| // `service_dispatcher::start` from `main`. | ||
|
|
||
| if let Err(_e) = run_service(arguments) { | ||
| // Handle errors in some way. | ||
| } | ||
| } | ||
|
|
||
| fn run_service(_arguments: Vec<std::ffi::OsString>) -> Result<(), crate::BoxError> { | ||
| use windows_service::service::ServiceControl; | ||
| use windows_service::service_control_handler::{self, ServiceControlHandlerResult}; | ||
|
|
||
| let shutdown_token = crate::CancellationToken::new(); | ||
| let shutdown_token_clone = shutdown_token.clone(); | ||
|
|
||
| let event_handler = move |control_event| -> ServiceControlHandlerResult { | ||
| match control_event { | ||
| ServiceControl::Stop => { | ||
| // Handle stop event and return control back to the system. | ||
| shutdown_token.cancel(); | ||
| ServiceControlHandlerResult::NoError | ||
| } | ||
| // All services must accept Interrogate even if it's a no-op. | ||
| ServiceControl::Interrogate => ServiceControlHandlerResult::NoError, | ||
| _ => ServiceControlHandlerResult::NotImplemented, | ||
| } | ||
| }; | ||
|
|
||
| // Register system service event handler | ||
| let status_handle = service_control_handler::register(SERVICE_NAME, event_handler)?; | ||
|
|
||
| let mut next_status = windows_service::service::ServiceStatus { | ||
| // Should match the one from system service registry | ||
| service_type: windows_service::service::ServiceType::OWN_PROCESS, | ||
| // The new state | ||
| current_state: windows_service::service::ServiceState::Running, | ||
| // Accept stop events when running | ||
| controls_accepted: windows_service::service::ServiceControlAccept::STOP, | ||
| // Used to report an error when starting or stopping only, otherwise must be zero | ||
| exit_code: windows_service::service::ServiceExitCode::Win32(0), | ||
| // Only used for pending states, otherwise must be zero | ||
| checkpoint: 0, | ||
| // Only used for pending states, otherwise must be zero | ||
| wait_hint: std::time::Duration::default(), | ||
| // Unused for setting status | ||
| process_id: None, | ||
| }; | ||
|
|
||
| // Tell the system that the service is running now | ||
| status_handle.set_service_status(next_status.clone())?; | ||
|
|
||
| let opt = crate::CmdOpt::parse_cmd(); | ||
| let is_server = opt.is_server(); | ||
|
|
||
| let mut config = if let Some(file) = opt.config { | ||
| crate::Config::from_config_file(file)? | ||
| } else if let Some(ref url_of_node) = opt.url_of_node { | ||
| let mut cfg = crate::Config::from_ssr_url(url_of_node)?; | ||
| cfg.set_listen_addr(opt.listen_addr.unwrap_or(std::net::SocketAddr::from(([127, 0, 0, 1], 1080)))); | ||
| cfg | ||
| } else { | ||
| return Err("Config file or node URL is required".into()); | ||
| }; | ||
| config.set_cache_dns(opt.cache_dns); | ||
|
|
||
| if opt.generate_url { | ||
| return Err("Generate URL is not supported in Windows service".into()); | ||
| } | ||
|
|
||
| config.check_correctness(is_server)?; | ||
|
|
||
| let rt = tokio::runtime::Builder::new_multi_thread().enable_all().build()?; | ||
| rt.block_on(async { | ||
| if let Err(err) = crate::async_main(config, false, shutdown_token_clone).await { | ||
| log::error!("Error: {}", err); | ||
| } | ||
| Ok::<(), crate::Error>(()) | ||
| })?; | ||
|
|
||
| // Tell the system that the service is stopped now | ||
| next_status.current_state = windows_service::service::ServiceState::Stopped; | ||
| status_handle.set_service_status(next_status)?; | ||
|
|
||
| Ok(()) | ||
| } |