Nukesor/pueue
1
1
Fork 0
mirror of https://github.com/Nukesor/pueue.git synced 2024-10-04 03:37:33 +03:00
Code Issues Projects Releases Wiki Activity

Separate logon/logoff into separate arms

Browse Source
This commit is contained in:
Cherry 2024-09-02 06:46:06 -07:00
parent 0efaaf88e8
commit 26dc46d917
No known key found for this signature in database
GPG Key ID: 8B5DD74E59C6C5AB
1 changed files with 20 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
pueue/src/daemon/service.rs
View File

@ -41,7 +41,7 @@ use windows_service::{
service::{
ServiceAccess, ServiceControl, ServiceControlAccept, ServiceErrorControl, ServiceExitCode,
ServiceInfo, ServiceStartType, ServiceState, ServiceStatus, ServiceType,
SessionChangeReason,
SessionChangeParam, SessionChangeReason, SessionNotification,
},
service_control_handler::{self, ServiceControlHandlerResult},
service_dispatcher,
@ -150,26 +150,30 @@ fn run_service() -> Result<()> {
ServiceControlHandlerResult::NoError
}
ServiceControl::SessionChange(param) => {
match param.reason {
SessionChangeReason::SessionLogon => {
if !spawner.running() {
if let Err(e) = spawner.start(Some(param.notification.session_id)) {
error!("failed to spawn: {e}");
}
}
// Logon
ServiceControl::SessionChange(SessionChangeParam {
reason: SessionChangeReason::SessionLogon,
notification: SessionNotification { session_id, .. },
}) => {
if !spawner.running() {
if let Err(e) = spawner.start(Some(session_id)) {
error!("failed to spawn: {e}");
}
SessionChangeReason::SessionLogoff => {
spawner.stop();
}
_ => (),
}
ServiceControlHandlerResult::NoError
}
// Logoff
ServiceControl::SessionChange(SessionChangeParam {
reason: SessionChangeReason::SessionLogoff,
..
}) => {
spawner.stop();
ServiceControlHandlerResult::NoError
}
// All services must accept Interrogate even if it's a no-op.
ServiceControl::Interrogate => ServiceControlHandlerResult::NoError,
@ -412,7 +416,7 @@ impl Spawner {
let mut handle = self.child.lock().unwrap().0 .0;
if handle.is_invalid() {
while !self.running.load(Ordering::Relaxed) {
while !self.running() {
thread::park();
}