1use std::collections::BTreeMap;
91use std::ffi::OsString;
92use std::fmt;
93use std::path::{Path, PathBuf};
94use std::time::Duration;
95
96use chrono::{DateTime, Utc};
97use runner_manager_domain::model::StartMode;
98use runner_manager_domain::path::LocalAbsolutePath;
99use serde::{Deserialize, Serialize};
100
101use crate::lock::{HostLock, LockError, LockKind};
102use crate::paths::AppPaths;
103#[cfg(windows)]
104use crate::runner_root_access::RootAdmission;
105use crate::runner_root_access::{
106 Reversal, RootAccessChange, RootAccessError, RootAccessReport, RootAccessSummary,
107};
108
109pub const SERVICE_NAME: &str = "runner-manager";
115pub const WINDOWS_SCM_HOST_ARGUMENT: &str = "--windows-service-host";
121
122pub const DISPLAY_NAME: &str = "GitHub Actions Runner Manager";
125
126pub const DESCRIPTION: &str =
128 "Starts ephemeral GitHub Actions self-hosted runners on this machine on demand.";
129
130pub const DAEMON_ARGUMENTS: [&str; 2] = ["daemon", "run"];
136
137pub const RECORD_FILE: &str = "service.toml";
145
146pub const CONTACT_FILE: &str = "github-contact.toml";
152
153pub const ROOT_REFUSAL_FILE: &str = "runner-root-refusal.toml";
157
158pub const LOG_FILE_STEM: &str = crate::logging::SERVICE_LOG_STEM;
169
170#[derive(Debug, Clone, PartialEq, Eq)]
183pub struct ServiceIdentity {
184 name: String,
185 display_name: String,
186 description: String,
187}
188
189impl ServiceIdentity {
190 #[must_use]
192 pub fn product() -> Self {
193 Self {
194 name: SERVICE_NAME.to_string(),
195 display_name: DISPLAY_NAME.to_string(),
196 description: DESCRIPTION.to_string(),
197 }
198 }
199
200 #[must_use]
207 pub fn fixture(tag: &str) -> Self {
208 let tag: String = tag
209 .chars()
210 .map(|c| {
211 if c.is_ascii_alphanumeric() {
212 c.to_ascii_lowercase()
213 } else {
214 '-'
215 }
216 })
217 .collect();
218 let name = format!("{SERVICE_NAME}-selftest-{tag}");
219 Self {
220 display_name: format!("{DISPLAY_NAME} (self-test fixture {tag})"),
221 description: "Disposable fixture created by runner-manager's own installer tests. \
222 Safe to remove."
223 .to_string(),
224 name,
225 }
226 }
227
228 #[must_use]
233 pub fn is_fixture(&self) -> bool {
234 self.name.starts_with(&format!("{SERVICE_NAME}-selftest-"))
235 }
236
237 #[must_use]
240 pub fn name(&self) -> &str {
241 &self.name
242 }
243
244 #[must_use]
246 pub fn display_name(&self) -> &str {
247 &self.display_name
248 }
249
250 #[must_use]
252 pub fn description(&self) -> &str {
253 &self.description
254 }
255
256 #[must_use]
262 pub fn launchd_label(&self) -> String {
263 format!(
264 "{}.{}.{}",
265 crate::paths::QUALIFIER,
266 crate::paths::ORGANIZATION,
267 self.name
268 )
269 }
270
271 #[must_use]
273 pub fn systemd_unit(&self) -> String {
274 format!("{}.service", self.name)
275 }
276}
277
278impl fmt::Display for ServiceIdentity {
279 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
280 f.write_str(&self.name)
281 }
282}
283
284#[derive(Debug, Clone, Copy, PartialEq, Eq)]
298pub struct RestartPolicy {
299 delay: Duration,
300 reset_after: Duration,
301}
302
303impl RestartPolicy {
304 pub const MIN_DELAY: Duration = Duration::from_secs(1);
311
312 pub const MAX_DELAY: Duration = Duration::from_secs(300);
315
316 pub const DEFAULT_DELAY: Duration = Duration::from_secs(15);
318
319 pub const DEFAULT_RESET_AFTER: Duration = Duration::from_secs(600);
321
322 pub fn new(delay: Duration, reset_after: Duration) -> Result<Self, ServiceError> {
330 if delay < Self::MIN_DELAY || delay > Self::MAX_DELAY {
331 return Err(ServiceError::RestartDelay {
332 requested_secs: delay.as_secs(),
333 min_secs: Self::MIN_DELAY.as_secs(),
334 max_secs: Self::MAX_DELAY.as_secs(),
335 });
336 }
337 if reset_after <= delay {
338 return Err(ServiceError::RestartResetWindow {
339 reset_secs: reset_after.as_secs(),
340 delay_secs: delay.as_secs(),
341 });
342 }
343 Ok(Self { delay, reset_after })
344 }
345
346 #[must_use]
348 pub const fn delay(&self) -> Duration {
349 self.delay
350 }
351
352 #[must_use]
354 pub const fn reset_after(&self) -> Duration {
355 self.reset_after
356 }
357
358 #[must_use]
375 pub const fn effective_delay(&self, kind: DefinitionKind) -> Duration {
376 match kind {
377 DefinitionKind::WindowsScheduledTask => {
378 let seconds = self.delay.as_secs();
379 let minutes = seconds.div_ceil(60);
380 Duration::from_secs(if minutes == 0 { 60 } else { minutes * 60 })
381 }
382 _ => self.delay,
383 }
384 }
385}
386
387impl Default for RestartPolicy {
388 fn default() -> Self {
389 Self {
390 delay: Self::DEFAULT_DELAY,
391 reset_after: Self::DEFAULT_RESET_AFTER,
392 }
393 }
394}
395
396impl fmt::Display for RestartPolicy {
397 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
398 write!(
399 f,
400 "restart after {}s, failure count resets after {}s",
401 self.delay.as_secs(),
402 self.reset_after.as_secs()
403 )
404 }
405}
406
407#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
418#[serde(rename_all = "snake_case")]
419pub enum ServiceAccount {
420 LocalSystem,
423 Root,
426 InvokingUser,
428}
429
430impl ServiceAccount {
431 #[must_use]
440 pub const fn for_definition(kind: DefinitionKind, mode: StartMode) -> Self {
441 match (kind, mode) {
442 (DefinitionKind::WindowsService, _) => Self::LocalSystem,
443 (DefinitionKind::WindowsScheduledTask, _) | (_, StartMode::Login) => Self::InvokingUser,
444 (_, StartMode::Boot) => Self::Root,
445 }
446 }
447
448 #[must_use]
451 pub const fn for_start_mode(mode: StartMode) -> Self {
452 Self::for_definition(host_definition_kind(mode), mode)
453 }
454
455 #[must_use]
457 pub const fn as_str(&self) -> &'static str {
458 match self {
459 Self::LocalSystem => "NT AUTHORITY\\SYSTEM",
460 Self::Root => "root",
461 Self::InvokingUser => "the invoking user",
462 }
463 }
464
465 #[must_use]
472 pub const fn justification(&self) -> &'static str {
473 match self {
474 Self::LocalSystem => {
475 "the machine-scoped store's DACL names SY, BA and OW only; LocalService and \
476 NetworkService cannot read it, and widening the DACL to reach them would grant \
477 every service on this host read access to the one credential this product holds"
478 }
479 Self::Root => {
480 "a boot-time registration runs outside every login session: on macOS the System \
481 Keychain is unlocked by /var/db/SystemKey, which is root-only, and on Linux the \
482 machine-scoped store is a 0600 file under /var/lib that only root can open \
483 before a session exists"
484 }
485 Self::InvokingUser => {
486 "a login-mode registration reads the user-scoped store, which is deliberately \
487 readable by exactly one account and needs no elevation at all"
488 }
489 }
490 }
491
492 #[must_use]
494 pub const fn needs_elevation(&self) -> bool {
495 matches!(self, Self::LocalSystem | Self::Root)
496 }
497}
498
499impl fmt::Display for ServiceAccount {
500 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
501 f.write_str(self.as_str())
502 }
503}
504
505#[derive(Debug, thiserror::Error)]
516pub enum ServiceError {
517 #[error("cannot install the service while an agent is already running on this host: {source}")]
523 LockHeld {
524 #[source]
526 source: Box<LockError>,
527 },
528
529 #[error(
532 "cannot tell whether an agent is already running on this host, so the service was not \
533 installed: {source}. Fix the reported problem with the state directory and try again."
534 )]
535 LockUnreadable {
536 #[source]
538 source: Box<LockError>,
539 },
540
541 #[error(
549 "this registration would run jobs under the default runner root, and that root could \
550 not be prepared, so nothing was registered: {source}"
551 )]
552 RunnerRoot {
553 #[source]
555 source: Box<RootAccessError>,
556 },
557
558 #[error(
560 "cannot resolve the absolute path of this executable, so there is nothing to register: \
561 {detail}. Run the installer from the installed binary rather than through a shell \
562 function or a wrapper that replaces argv[0]."
563 )]
564 BinaryPath {
565 detail: String,
567 },
568
569 #[error(
571 "{} is not a file, so registering it would create a service that cannot start. Install \
572 the product first, then run `service install` from the installed binary.",
573 path.display()
574 )]
575 BinaryMissing {
576 path: PathBuf,
578 },
579
580 #[error(
582 "a restart delay of {requested_secs}s is outside the supported range \
583 {min_secs}s-{max_secs}s. Below the floor the platform's own throttle overrides the \
584 value, so `service status` would report a delay that is not the one in force."
585 )]
586 RestartDelay {
587 requested_secs: u64,
589 min_secs: u64,
591 max_secs: u64,
593 },
594
595 #[error(
597 "a failure-count reset window of {reset_secs}s is not longer than the {delay_secs}s \
598 restart delay, so the count would reset between every pair of restarts and no \
599 start limit could ever apply."
600 )]
601 RestartResetWindow {
602 reset_secs: u64,
604 delay_secs: u64,
606 },
607
608 #[error(
614 "{name} is already registered to start at {existing}, and this asks for {requested}. \
615 Changing the start mode moves the registration between two different service \
616 managers, so it is not something an install does by itself: run `service uninstall` \
617 first, or change the start mode in the terminal UI (`runner-manager tui`), which \
618 switches it in place and keeps the registration running throughout."
619 )]
620 AlreadyInstalled {
621 name: String,
623 existing: StartMode,
625 requested: StartMode,
627 },
628
629 #[error("{name} is not registered on this host, so there is nothing to {operation}.")]
631 NotInstalled {
632 name: String,
634 operation: &'static str,
636 },
637
638 #[error("cannot {operation} the service record {}: {detail}", path.display())]
640 Record {
641 operation: &'static str,
643 path: PathBuf,
645 detail: String,
647 },
648
649 #[error(
651 "the service record {} was not written by this product, or was written by a version \
652 this one cannot read: {detail}. Run `service uninstall` and `service install` again; \
653 neither touches configuration, secrets, or the cache.",
654 path.display()
655 )]
656 RecordUnreadable {
657 path: PathBuf,
659 detail: String,
661 },
662
663 #[error(
677 "the service record {} is there and this account may not read it: {detail}. It was \
678 written by whichever account installed the service -- on a boot-mode host, `sudo \
679 service install`. Running `service install` again rewrites it readable; until then \
680 only what the service manager itself reports is available.",
681 path.display()
682 )]
683 RecordNotPermitted {
684 path: PathBuf,
686 detail: String,
688 },
689
690 #[error("cannot prepare this host's application-data directories: {source}")]
692 Paths {
693 #[source]
695 source: Box<crate::paths::PathsError>,
696 },
697
698 #[error("cannot {operation} {name} through {manager}: {detail}")]
700 Control {
701 operation: &'static str,
703 name: String,
705 manager: &'static str,
707 detail: String,
709 },
710
711 #[error(
713 "cannot {operation} {name}: {cause}. The attempted rollback also failed: {rollback}. \
714 Inspect `service status` before retrying."
715 )]
716 Rollback {
717 operation: &'static str,
719 name: String,
721 cause: String,
723 rollback: String,
725 },
726
727 #[error("{operation} {name} needs administrative rights: {detail}. {remedy}")]
729 NeedsElevation {
730 operation: &'static str,
732 name: String,
734 detail: String,
736 remedy: &'static str,
738 },
739}
740
741#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
753pub struct ServiceDirectories {
754 pub config: PathBuf,
756 pub state: PathBuf,
758 pub runtime: PathBuf,
760 pub logs: PathBuf,
762}
763
764impl ServiceDirectories {
765 #[must_use]
767 pub fn of(paths: &AppPaths) -> Self {
768 Self {
769 config: paths.config_dir().to_path_buf(),
770 state: paths.state_dir().to_path_buf(),
771 runtime: paths.runtime_dir().to_path_buf(),
772 logs: paths.logs_dir().to_path_buf(),
773 }
774 }
775
776 #[must_use]
778 pub fn all(&self) -> [&Path; 4] {
779 [
780 self.config.as_path(),
781 self.state.as_path(),
782 self.runtime.as_path(),
783 self.logs.as_path(),
784 ]
785 }
786
787 #[must_use]
790 pub fn log_file(&self) -> PathBuf {
791 self.logs.join(LOG_FILE_STEM)
792 }
793}
794
795#[derive(Debug, Clone, PartialEq, Eq)]
807pub struct InstallRequest {
808 start_mode: StartMode,
809 binary: Option<PathBuf>,
810 source_binary: Option<PathBuf>,
811 arguments: Vec<OsString>,
812 restart: RestartPolicy,
813 on_demand: bool,
814}
815
816impl InstallRequest {
817 #[must_use]
819 pub fn new(start_mode: StartMode) -> Self {
820 Self {
821 start_mode,
822 binary: None,
823 source_binary: None,
824 arguments: DAEMON_ARGUMENTS.iter().map(OsString::from).collect(),
825 restart: RestartPolicy::default(),
826 on_demand: false,
827 }
828 }
829
830 #[must_use]
840 pub const fn started_on_demand(mut self) -> Self {
841 self.on_demand = true;
842 self
843 }
844
845 #[must_use]
852 pub fn for_binary(mut self, binary: impl Into<PathBuf>) -> Self {
853 self.binary = Some(binary.into());
854 self
855 }
856
857 #[must_use]
874 pub fn copied_from(mut self, source: impl Into<PathBuf>) -> Self {
875 self.source_binary = Some(source.into());
876 self
877 }
878
879 #[must_use]
881 pub fn with_arguments<I, S>(mut self, arguments: I) -> Self
882 where
883 I: IntoIterator<Item = S>,
884 S: Into<OsString>,
885 {
886 self.arguments = arguments.into_iter().map(Into::into).collect();
887 self
888 }
889
890 #[must_use]
893 pub const fn with_restart(mut self, restart: RestartPolicy) -> Self {
894 self.restart = restart;
895 self
896 }
897
898 #[must_use]
900 pub const fn start_mode(&self) -> StartMode {
901 self.start_mode
902 }
903}
904
905#[derive(Debug, Clone, PartialEq, Eq)]
912pub struct InstallPlan {
913 identity: ServiceIdentity,
914 start_mode: StartMode,
915 binary: PathBuf,
916 source_binary: Option<PathBuf>,
917 arguments: Vec<OsString>,
918 account: ServiceAccount,
919 restart: RestartPolicy,
920 directories: ServiceDirectories,
921 secret_guard: Option<PathBuf>,
922 on_demand: bool,
923}
924
925impl InstallPlan {
926 pub fn resolve(
934 identity: ServiceIdentity,
935 request: &InstallRequest,
936 directories: ServiceDirectories,
937 ) -> Result<Self, ServiceError> {
938 let binary = match &request.binary {
939 Some(named) => absolute(named)?,
940 None => running_executable()?,
941 };
942 if !binary.is_file() {
943 return Err(ServiceError::BinaryMissing { path: binary });
944 }
945 let secret_guard = crate::secrets::PlatformSecretStore::for_start_mode(request.start_mode)
950 .ok()
951 .map(|store| store.guard());
952 Ok(Self {
953 identity,
954 start_mode: request.start_mode,
955 binary,
956 source_binary: request.source_binary.clone(),
957 arguments: request.arguments.clone(),
958 account: ServiceAccount::for_start_mode(request.start_mode),
959 restart: request.restart,
960 directories,
961 secret_guard,
962 on_demand: request.on_demand,
963 })
964 }
965
966 #[must_use]
973 pub fn unchecked(
974 identity: ServiceIdentity,
975 start_mode: StartMode,
976 binary: impl Into<PathBuf>,
977 directories: ServiceDirectories,
978 ) -> Self {
979 Self {
980 identity,
981 start_mode,
982 binary: binary.into(),
983 source_binary: None,
984 arguments: DAEMON_ARGUMENTS.iter().map(OsString::from).collect(),
985 account: ServiceAccount::for_start_mode(start_mode),
986 restart: RestartPolicy::default(),
987 directories,
988 secret_guard: None,
989 on_demand: false,
990 }
991 }
992
993 #[must_use]
996 pub const fn started_on_demand(mut self) -> Self {
997 self.on_demand = true;
998 self
999 }
1000
1001 #[must_use]
1003 pub const fn is_on_demand(&self) -> bool {
1004 self.on_demand
1005 }
1006
1007 #[must_use]
1014 pub fn with_secret_guard(mut self, guard: impl Into<PathBuf>) -> Self {
1015 self.secret_guard = Some(guard.into());
1016 self
1017 }
1018
1019 #[must_use]
1021 pub fn secret_guard(&self) -> Option<&Path> {
1022 self.secret_guard.as_deref()
1023 }
1024
1025 #[must_use]
1027 pub const fn with_restart(mut self, restart: RestartPolicy) -> Self {
1028 self.restart = restart;
1029 self
1030 }
1031
1032 #[must_use]
1034 pub fn with_arguments<I, S>(mut self, arguments: I) -> Self
1035 where
1036 I: IntoIterator<Item = S>,
1037 S: Into<OsString>,
1038 {
1039 self.arguments = arguments.into_iter().map(Into::into).collect();
1040 self
1041 }
1042
1043 #[must_use]
1045 pub const fn identity(&self) -> &ServiceIdentity {
1046 &self.identity
1047 }
1048
1049 #[must_use]
1051 pub const fn start_mode(&self) -> StartMode {
1052 self.start_mode
1053 }
1054
1055 #[must_use]
1057 pub fn binary(&self) -> &Path {
1058 &self.binary
1059 }
1060
1061 #[must_use]
1067 pub fn source_binary(&self) -> Option<&Path> {
1068 self.source_binary.as_deref()
1069 }
1070
1071 #[must_use]
1073 pub fn arguments(&self) -> &[OsString] {
1074 &self.arguments
1075 }
1076
1077 #[must_use]
1079 pub const fn account(&self) -> &ServiceAccount {
1080 &self.account
1081 }
1082
1083 #[must_use]
1085 pub const fn restart(&self) -> RestartPolicy {
1086 self.restart
1087 }
1088
1089 #[must_use]
1091 pub const fn directories(&self) -> &ServiceDirectories {
1092 &self.directories
1093 }
1094
1095 #[must_use]
1097 pub fn command_line(&self) -> String {
1098 let mut out = quote_argument(&self.binary.to_string_lossy());
1099 for argument in &self.arguments {
1100 out.push(' ');
1101 out.push_str("e_argument(&argument.to_string_lossy()));
1102 }
1103 out
1104 }
1105}
1106
1107fn retained_runner_root(change: &RootAccessChange) -> Option<String> {
1115 let reversal = change.revert();
1116 matches!(reversal, Reversal::Retained { .. }).then(|| reversal.to_string())
1117}
1118
1119fn rolled_back<T>(
1132 retained: Option<String>,
1133 rollback: Result<T, ServiceError>,
1134 operation: &'static str,
1135 identity: &ServiceIdentity,
1136 cause: ServiceError,
1137) -> ServiceError {
1138 let left_behind = match (rollback.err(), retained) {
1139 (Some(rollback), Some(retained)) => Some(format!("{rollback}; {retained}")),
1140 (Some(rollback), None) => Some(rollback.to_string()),
1141 (None, retained) => retained,
1142 };
1143 match left_behind {
1144 Some(rollback) => ServiceError::Rollback {
1145 operation,
1146 name: identity.name().to_string(),
1147 cause: cause.to_string(),
1148 rollback,
1149 },
1150 None => cause,
1151 }
1152}
1153
1154fn undo_runner_root(
1157 change: &RootAccessChange,
1158 operation: &'static str,
1159 identity: &ServiceIdentity,
1160 cause: ServiceError,
1161) -> ServiceError {
1162 rolled_back(
1163 retained_runner_root(change),
1164 Ok(()),
1165 operation,
1166 identity,
1167 cause,
1168 )
1169}
1170
1171fn absolute(path: &Path) -> Result<PathBuf, ServiceError> {
1185 std::path::absolute(path).map_err(|error| ServiceError::BinaryPath {
1186 detail: format!("{} could not be made absolute: {error}", path.display()),
1187 })
1188}
1189
1190fn running_executable() -> Result<PathBuf, ServiceError> {
1192 let raw = std::env::current_exe().map_err(|error| ServiceError::BinaryPath {
1193 detail: error.to_string(),
1194 })?;
1195 absolute(&raw)
1196}
1197
1198pub(crate) fn quote_argument(argument: &str) -> String {
1207 if !argument.is_empty() && !argument.contains([' ', '"', '\t', '\n']) {
1208 return argument.to_string();
1209 }
1210 let mut out = String::with_capacity(argument.len() + 2);
1211 out.push('"');
1212 let mut backslashes = 0usize;
1213 for c in argument.chars() {
1214 match c {
1215 '\\' => {
1216 backslashes += 1;
1217 out.push('\\');
1218 }
1219 '"' => {
1220 for _ in 0..=backslashes {
1223 out.push('\\');
1224 }
1225 out.push('"');
1226 backslashes = 0;
1227 }
1228 other => {
1229 backslashes = 0;
1230 out.push(other);
1231 }
1232 }
1233 }
1234 for _ in 0..backslashes {
1236 out.push('\\');
1237 }
1238 out.push('"');
1239 out
1240}
1241
1242#[must_use]
1253pub fn executable_from_command_line(command_line: &str) -> Option<PathBuf> {
1254 let trimmed = command_line.trim_start();
1255 if trimmed.is_empty() {
1256 return None;
1257 }
1258 let mut out = String::new();
1259 let mut chars = trimmed.chars().peekable();
1260 let quoted = chars.peek() == Some(&'"');
1261 if quoted {
1262 chars.next();
1263 let mut backslashes = 0usize;
1264 for c in chars {
1265 match c {
1266 '\\' => {
1267 backslashes += 1;
1268 }
1269 '"' => {
1270 out.extend(std::iter::repeat_n('\\', backslashes / 2));
1273 if backslashes.is_multiple_of(2) {
1274 break;
1275 }
1276 backslashes = 0;
1277 out.push('"');
1278 }
1279 other => {
1280 out.extend(std::iter::repeat_n('\\', backslashes));
1281 backslashes = 0;
1282 out.push(other);
1283 }
1284 }
1285 }
1286 } else {
1287 for c in chars {
1288 if c == ' ' || c == '\t' {
1289 break;
1290 }
1291 out.push(c);
1292 }
1293 }
1294 if out.is_empty() {
1295 None
1296 } else {
1297 Some(PathBuf::from(out))
1298 }
1299}
1300
1301#[derive(Debug, Clone, Copy, PartialEq, Eq)]
1307pub enum DefinitionKind {
1308 WindowsService,
1312 WindowsScheduledTask,
1314 LaunchdPlist,
1316 SystemdUnit,
1318}
1319
1320impl DefinitionKind {
1321 #[must_use]
1323 pub const fn manager(self) -> &'static str {
1324 match self {
1325 Self::WindowsService => "the Windows Service Control Manager",
1326 Self::WindowsScheduledTask => "Windows Task Scheduler",
1327 Self::LaunchdPlist => "launchd",
1328 Self::SystemdUnit => "systemd",
1329 }
1330 }
1331}
1332
1333impl fmt::Display for DefinitionKind {
1334 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1335 f.write_str(self.manager())
1336 }
1337}
1338
1339#[derive(Debug, Clone, PartialEq, Eq)]
1346pub struct ServiceDefinition {
1347 kind: DefinitionKind,
1348 text: String,
1349 install_path: Option<PathBuf>,
1350}
1351
1352impl ServiceDefinition {
1353 #[must_use]
1355 pub const fn kind(&self) -> DefinitionKind {
1356 self.kind
1357 }
1358
1359 #[must_use]
1361 pub fn text(&self) -> &str {
1362 &self.text
1363 }
1364
1365 #[must_use]
1372 pub fn install_path(&self) -> Option<&Path> {
1373 self.install_path.as_deref()
1374 }
1375
1376 pub fn for_host(plan: &InstallPlan) -> Result<Self, ServiceError> {
1389 Ok(match host_definition_kind(plan.start_mode()) {
1390 DefinitionKind::WindowsService => Self::windows_service(plan),
1391 DefinitionKind::WindowsScheduledTask => {
1392 Self::windows_scheduled_task(plan, &TaskPrincipal::current()?)
1393 }
1394 DefinitionKind::LaunchdPlist => Self::launchd(plan, host_home().as_deref()),
1395 DefinitionKind::SystemdUnit => Self::systemd(plan, host_home().as_deref()),
1396 })
1397 }
1398
1399 #[must_use]
1401 pub fn windows_service(plan: &InstallPlan) -> Self {
1402 Self {
1403 kind: DefinitionKind::WindowsService,
1404 text: windows_service_descriptor(plan),
1405 install_path: None,
1406 }
1407 }
1408
1409 #[must_use]
1411 pub fn windows_scheduled_task(plan: &InstallPlan, principal: &TaskPrincipal) -> Self {
1412 Self {
1413 kind: DefinitionKind::WindowsScheduledTask,
1414 text: windows_scheduled_task_xml(plan, principal),
1415 install_path: None,
1416 }
1417 }
1418
1419 #[must_use]
1424 pub fn launchd(plan: &InstallPlan, home: Option<&Path>) -> Self {
1425 let file = format!("{}.plist", plan.identity().launchd_label());
1426 let install_path = match plan.start_mode() {
1427 StartMode::Boot => Some(PathBuf::from(LAUNCH_DAEMONS_DIR).join(file)),
1428 StartMode::Login => home.map(|home| home.join(LAUNCH_AGENTS_SUBDIR).join(file)),
1429 };
1430 Self {
1431 kind: DefinitionKind::LaunchdPlist,
1432 text: launchd_plist(plan),
1433 install_path,
1434 }
1435 }
1436
1437 #[must_use]
1445 pub fn from_text(kind: DefinitionKind, text: impl Into<String>) -> Self {
1446 Self {
1447 kind,
1448 text: text.into(),
1449 install_path: None,
1450 }
1451 }
1452
1453 #[must_use]
1458 pub fn systemd(plan: &InstallPlan, home: Option<&Path>) -> Self {
1459 let file = plan.identity().systemd_unit();
1460 let install_path = match plan.start_mode() {
1461 StartMode::Boot => Some(PathBuf::from(SYSTEMD_SYSTEM_DIR).join(file)),
1462 StartMode::Login => home.map(|home| home.join(SYSTEMD_USER_SUBDIR).join(file)),
1463 };
1464 Self {
1465 kind: DefinitionKind::SystemdUnit,
1466 text: systemd_unit(plan),
1467 install_path,
1468 }
1469 }
1470}
1471
1472pub const LAUNCH_DAEMONS_DIR: &str = "/Library/LaunchDaemons";
1474pub const LAUNCH_AGENTS_SUBDIR: &str = "Library/LaunchAgents";
1476pub const SYSTEMD_SYSTEM_DIR: &str = "/etc/systemd/system";
1478pub const SYSTEMD_USER_SUBDIR: &str = ".config/systemd/user";
1480
1481const DOCUMENTATION: &str = "https://github.com/IvanMurzak/GitHub-Runner-Scaler-UI";
1484
1485pub const START_LIMIT_BURST: u32 = 5;
1493
1494#[must_use]
1515pub fn systemd_unit(plan: &InstallPlan) -> String {
1516 let identity = plan.identity();
1517 let restart = plan.restart();
1518 let directories = plan.directories();
1519
1520 let mut out = String::new();
1521 out.push_str("[Unit]\n");
1522 out.push_str(&format!("Description={}\n", identity.display_name()));
1523 out.push_str(&format!("Documentation={DOCUMENTATION}\n"));
1524 out.push_str("After=network-online.target\n");
1525 out.push_str("Wants=network-online.target\n");
1526 out.push_str(&format!(
1529 "StartLimitIntervalSec={}\n",
1530 restart.reset_after().as_secs()
1531 ));
1532 out.push_str(&format!("StartLimitBurst={START_LIMIT_BURST}\n"));
1533
1534 out.push_str("\n[Service]\n");
1535 out.push_str("Type=simple\n");
1536 out.push_str("KillMode=process\n");
1540 out.push_str(&format!("ExecStart={}\n", plan.command_line()));
1541 out.push_str(&format!(
1542 "WorkingDirectory={}\n",
1543 directories.state.display()
1544 ));
1545 out.push_str(&format!("SyslogIdentifier={identity}\n"));
1546 out.push_str("Restart=on-failure\n");
1547 out.push_str(&format!("RestartSec={}\n", restart.delay().as_secs()));
1548
1549 out.push_str("\n# Least privilege. See docs/service-account.md.\n");
1550 for directive in SYSTEMD_HARDENING {
1551 out.push_str(directive);
1552 out.push('\n');
1553 }
1554 let mut writable = directories.all().to_vec();
1555 if let Some(secret_directory) = plan.secret_guard().and_then(Path::parent) {
1556 writable.push(secret_directory);
1557 }
1558 out.push_str(&format!(
1559 "ReadWritePaths={}\n",
1560 writable
1561 .iter()
1562 .map(|path| quote_argument(&path.to_string_lossy()))
1563 .collect::<Vec<_>>()
1564 .join(" ")
1565 ));
1566
1567 out.push_str("\n[Install]\n");
1568 out.push_str(match plan.start_mode() {
1569 StartMode::Boot => "WantedBy=multi-user.target\n",
1570 StartMode::Login => "WantedBy=default.target\n",
1571 });
1572 out
1573}
1574
1575pub const SYSTEMD_HARDENING: [&str; 13] = [
1582 "NoNewPrivileges=yes",
1583 "CapabilityBoundingSet=",
1584 "AmbientCapabilities=",
1585 "PrivateTmp=yes",
1586 "PrivateDevices=yes",
1587 "ProtectSystem=strict",
1588 "ProtectKernelTunables=yes",
1589 "ProtectKernelModules=yes",
1590 "ProtectControlGroups=yes",
1591 "RestrictNamespaces=yes",
1592 "RestrictRealtime=yes",
1593 "RestrictSUIDSGID=yes",
1594 "RestrictAddressFamilies=AF_INET AF_INET6 AF_UNIX",
1595];
1596
1597#[must_use]
1606pub fn launchd_plist(plan: &InstallPlan) -> String {
1607 let identity = plan.identity();
1608 let directories = plan.directories();
1609 let mut out = String::new();
1610 out.push_str("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n");
1611 out.push_str(
1612 "<!DOCTYPE plist PUBLIC \"-//Apple//DTD PLIST 1.0//EN\" \
1613 \"http://www.apple.com/DTDs/PropertyList-1.0.dtd\">\n",
1614 );
1615 out.push_str("<plist version=\"1.0\">\n<dict>\n");
1616 out.push_str(&plist_string("Label", &identity.launchd_label()));
1617
1618 out.push_str(" <key>ProgramArguments</key>\n <array>\n");
1619 out.push_str(&format!(
1620 " <string>{}</string>\n",
1621 xml_escape(&plan.binary().to_string_lossy())
1622 ));
1623 for argument in plan.arguments() {
1624 out.push_str(&format!(
1625 " <string>{}</string>\n",
1626 xml_escape(&argument.to_string_lossy())
1627 ));
1628 }
1629 out.push_str(" </array>\n");
1630
1631 out.push_str(" <key>RunAtLoad</key>\n <true/>\n");
1632 out.push_str(" <key>KeepAlive</key>\n <dict>\n");
1633 out.push_str(" <key>SuccessfulExit</key>\n <false/>\n");
1634 out.push_str(" </dict>\n");
1635 out.push_str(&format!(
1636 " <key>ThrottleInterval</key>\n <integer>{}</integer>\n",
1637 plan.restart().delay().as_secs()
1638 ));
1639 out.push_str(&plist_string("ProcessType", "Background"));
1642 out.push_str(&plist_string(
1643 "WorkingDirectory",
1644 &directories.state.to_string_lossy(),
1645 ));
1646 out.push_str(&plist_string(
1647 "StandardOutPath",
1648 &directories
1649 .logs
1650 .join("runner-manager.launchd.out.log")
1651 .to_string_lossy(),
1652 ));
1653 out.push_str(&plist_string(
1654 "StandardErrorPath",
1655 &directories
1656 .logs
1657 .join("runner-manager.launchd.err.log")
1658 .to_string_lossy(),
1659 ));
1660
1661 match plan.start_mode() {
1662 StartMode::Boot => {
1663 out.push_str(&plist_string(
1666 "UserName",
1667 ServiceAccount::for_definition(DefinitionKind::LaunchdPlist, StartMode::Boot)
1668 .as_str(),
1669 ));
1670 out.push_str(" <key>SessionCreate</key>\n <false/>\n");
1672 }
1673 StartMode::Login => {
1674 }
1678 }
1679
1680 out.push_str("</dict>\n</plist>\n");
1681 out
1682}
1683
1684fn plist_string(key: &str, value: &str) -> String {
1686 format!(
1687 " <key>{}</key>\n <string>{}</string>\n",
1688 xml_escape(key),
1689 xml_escape(value)
1690 )
1691}
1692
1693#[derive(Debug, Clone, PartialEq, Eq)]
1701pub struct TaskPrincipal {
1702 user_id: String,
1703}
1704
1705impl TaskPrincipal {
1706 pub fn current() -> Result<Self, ServiceError> {
1714 let user = std::env::var("USERNAME")
1715 .ok()
1716 .filter(|value| !value.trim().is_empty());
1717 let Some(user) = user else {
1718 return Err(ServiceError::Control {
1719 operation: "identify the account for",
1720 name: SERVICE_NAME.to_string(),
1721 manager: "Windows Task Scheduler",
1722 detail: "this session reports no %USERNAME%, so there is no principal to \
1723 register a logon-triggered task for"
1724 .to_string(),
1725 });
1726 };
1727 let domain = std::env::var("USERDOMAIN")
1728 .ok()
1729 .filter(|value| !value.trim().is_empty());
1730 Ok(Self {
1731 user_id: match domain {
1732 Some(domain) => format!("{domain}\\{user}"),
1733 None => user,
1734 },
1735 })
1736 }
1737
1738 #[must_use]
1740 pub fn named(user_id: impl Into<String>) -> Self {
1741 Self {
1742 user_id: user_id.into(),
1743 }
1744 }
1745
1746 #[must_use]
1748 pub fn user_id(&self) -> &str {
1749 &self.user_id
1750 }
1751}
1752
1753#[must_use]
1760pub fn windows_scheduled_task_xml(plan: &InstallPlan, principal: &TaskPrincipal) -> String {
1761 let identity = plan.identity();
1762 let user = xml_escape(principal.user_id());
1763 let arguments = plan
1764 .arguments()
1765 .iter()
1766 .map(|argument| quote_argument(&argument.to_string_lossy()))
1767 .collect::<Vec<_>>()
1768 .join(" ");
1769 let mut out = String::new();
1770 out.push_str("<?xml version=\"1.0\" encoding=\"UTF-16\"?>\n");
1771 out.push_str(
1772 "<Task version=\"1.4\" \
1773 xmlns=\"http://schemas.microsoft.com/windows/2004/02/mit/task\">\n",
1774 );
1775 out.push_str(" <RegistrationInfo>\n");
1776 out.push_str(&format!(
1777 " <Description>{}</Description>\n",
1778 xml_escape(identity.description())
1779 ));
1780 out.push_str(&format!(
1781 " <URI>\\{}</URI>\n",
1782 xml_escape(identity.name())
1783 ));
1784 out.push_str(" </RegistrationInfo>\n");
1785
1786 out.push_str(" <Triggers>\n <LogonTrigger>\n");
1787 out.push_str(" <Enabled>true</Enabled>\n");
1788 out.push_str(&format!(" <UserId>{user}</UserId>\n"));
1789 out.push_str(" </LogonTrigger>\n </Triggers>\n");
1790
1791 out.push_str(" <Principals>\n <Principal id=\"Author\">\n");
1792 out.push_str(&format!(" <UserId>{user}</UserId>\n"));
1793 out.push_str(" <LogonType>InteractiveToken</LogonType>\n");
1794 out.push_str(" <RunLevel>LeastPrivilege</RunLevel>\n");
1795 out.push_str(" </Principal>\n </Principals>\n");
1796
1797 out.push_str(" <Settings>\n");
1798 out.push_str(" <MultipleInstancesPolicy>IgnoreNew</MultipleInstancesPolicy>\n");
1802 out.push_str(" <DisallowStartIfOnBatteries>false</DisallowStartIfOnBatteries>\n");
1803 out.push_str(" <StopIfGoingOnBatteries>false</StopIfGoingOnBatteries>\n");
1804 out.push_str(" <AllowHardTerminate>true</AllowHardTerminate>\n");
1805 out.push_str(" <StartWhenAvailable>true</StartWhenAvailable>\n");
1806 out.push_str(" <RunOnlyIfNetworkAvailable>false</RunOnlyIfNetworkAvailable>\n");
1807 out.push_str(" <IdleSettings>\n");
1808 out.push_str(" <StopOnIdleEnd>false</StopOnIdleEnd>\n");
1809 out.push_str(" <RestartOnIdle>false</RestartOnIdle>\n");
1810 out.push_str(" </IdleSettings>\n");
1811 out.push_str(" <AllowStartOnDemand>true</AllowStartOnDemand>\n");
1812 out.push_str(" <Enabled>true</Enabled>\n");
1813 out.push_str(" <Hidden>false</Hidden>\n");
1814 out.push_str(" <RunOnlyIfIdle>false</RunOnlyIfIdle>\n");
1815 out.push_str(" <WakeToRun>false</WakeToRun>\n");
1816 out.push_str(" <ExecutionTimeLimit>PT0S</ExecutionTimeLimit>\n");
1818 out.push_str(" <Priority>7</Priority>\n");
1819 out.push_str(" <RestartOnFailure>\n");
1820 out.push_str(&format!(
1821 " <Interval>{}</Interval>\n",
1822 iso8601_minutes(
1823 plan.restart()
1824 .effective_delay(DefinitionKind::WindowsScheduledTask)
1825 )
1826 ));
1827 out.push_str(&format!(" <Count>{START_LIMIT_BURST}</Count>\n"));
1828 out.push_str(" </RestartOnFailure>\n");
1829 out.push_str(" </Settings>\n");
1830
1831 out.push_str(" <Actions Context=\"Author\">\n <Exec>\n");
1832 out.push_str(&format!(
1833 " <Command>{}</Command>\n",
1834 xml_escape(&plan.binary().to_string_lossy())
1835 ));
1836 if !arguments.is_empty() {
1837 out.push_str(&format!(
1838 " <Arguments>{}</Arguments>\n",
1839 xml_escape(&arguments)
1840 ));
1841 }
1842 out.push_str(&format!(
1843 " <WorkingDirectory>{}</WorkingDirectory>\n",
1844 xml_escape(&plan.directories().state.to_string_lossy())
1845 ));
1846 out.push_str(" </Exec>\n </Actions>\n");
1847 out.push_str("</Task>\n");
1848 out
1849}
1850
1851fn iso8601_minutes(duration: Duration) -> String {
1854 format!("PT{}M", duration.as_secs() / 60)
1855}
1856
1857pub(crate) fn xml_escape(value: &str) -> String {
1865 let mut out = String::with_capacity(value.len());
1866 for c in value.chars() {
1867 match c {
1868 '&' => out.push_str("&"),
1869 '<' => out.push_str("<"),
1870 '>' => out.push_str(">"),
1871 '"' => out.push_str("""),
1872 '\'' => out.push_str("'"),
1873 other => out.push(other),
1874 }
1875 }
1876 out
1877}
1878
1879fn xml_unescape(value: &str) -> String {
1885 value
1886 .replace("<", "<")
1887 .replace(">", ">")
1888 .replace(""", "\"")
1889 .replace("'", "'")
1890 .replace("&", "&")
1891}
1892
1893#[derive(Debug, Clone, PartialEq, Eq)]
1902pub struct WindowsServiceSpec {
1903 pub name: String,
1905 pub display_name: String,
1907 pub description: String,
1909 pub automatic_start: bool,
1911 pub account: Option<String>,
1914 pub command_line: String,
1916 pub restart: RestartPolicy,
1918}
1919
1920#[must_use]
1922pub fn windows_service_spec(plan: &InstallPlan) -> WindowsServiceSpec {
1923 WindowsServiceSpec {
1924 name: plan.identity().name().to_string(),
1925 display_name: plan.identity().display_name().to_string(),
1926 description: plan.identity().description().to_string(),
1927 automatic_start: plan.start_mode() == StartMode::Boot && !plan.is_on_demand(),
1932 account: match ServiceAccount::for_definition(
1933 DefinitionKind::WindowsService,
1934 plan.start_mode(),
1935 ) {
1936 ServiceAccount::LocalSystem => None,
1939 other => Some(other.as_str().to_string()),
1940 },
1941 command_line: plan.command_line(),
1942 restart: plan.restart(),
1943 }
1944}
1945
1946#[must_use]
1949fn windows_service_descriptor(plan: &InstallPlan) -> String {
1950 let spec = windows_service_spec(plan);
1951 let mut out = String::new();
1952 out.push_str("[windows-service]\n");
1953 out.push_str(&format!("Name={}\n", spec.name));
1954 out.push_str(&format!("DisplayName={}\n", spec.display_name));
1955 out.push_str(&format!("Description={}\n", spec.description));
1956 out.push_str("ServiceType=OWN_PROCESS\n");
1960 out.push_str(&format!(
1961 "StartType={}\n",
1962 if spec.automatic_start {
1963 "AutoStart"
1964 } else {
1965 "OnDemand"
1966 }
1967 ));
1968 out.push_str("ErrorControl=Normal\n");
1969 out.push_str(&format!(
1970 "Account={}\n",
1971 spec.account
1972 .as_deref()
1973 .unwrap_or(ServiceAccount::LocalSystem.as_str())
1974 ));
1975 out.push_str(&format!("CommandLine={}\n", spec.command_line));
1976 out.push_str(&format!(
1977 "FailureActionRestartDelaySecs={}\n",
1978 spec.restart.delay().as_secs()
1979 ));
1980 out.push_str(&format!(
1981 "FailureActionsResetPeriodSecs={}\n",
1982 spec.restart.reset_after().as_secs()
1983 ));
1984 out.push_str("FailureActionsOnNonCrashFailures=true\n");
1988 out.push_str(&format!(
1989 "ReadWritePaths={}\n",
1990 plan.directories()
1991 .all()
1992 .iter()
1993 .map(|path| quote_argument(&path.to_string_lossy()))
1994 .collect::<Vec<_>>()
1995 .join(" ")
1996 ));
1997 out
1998}
1999
2000#[derive(Debug, Clone, Copy, PartialEq, Eq)]
2006pub enum FindingKind {
2007 Excess,
2010 Shortfall,
2013}
2014
2015impl fmt::Display for FindingKind {
2016 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
2017 f.write_str(match self {
2018 Self::Excess => "excess",
2019 Self::Shortfall => "shortfall",
2020 })
2021 }
2022}
2023
2024#[derive(Debug, Clone, PartialEq, Eq)]
2026pub struct PrivilegeFinding {
2027 pub kind: FindingKind,
2029 pub subject: String,
2031 pub detail: String,
2033}
2034
2035impl fmt::Display for PrivilegeFinding {
2036 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
2037 write!(f, "{}: {} -- {}", self.kind, self.subject, self.detail)
2038 }
2039}
2040
2041#[derive(Debug, Clone, PartialEq, Eq)]
2055pub struct PrivilegeReview {
2056 kind: DefinitionKind,
2057 account: ServiceAccount,
2058 controls: Vec<String>,
2059 findings: Vec<PrivilegeFinding>,
2060}
2061
2062impl PrivilegeReview {
2063 #[must_use]
2065 pub fn is_least_privilege(&self) -> bool {
2066 !self
2067 .findings
2068 .iter()
2069 .any(|finding| finding.kind == FindingKind::Excess)
2070 }
2071
2072 #[must_use]
2074 pub fn findings(&self) -> &[PrivilegeFinding] {
2075 &self.findings
2076 }
2077
2078 #[must_use]
2081 pub fn excesses(&self) -> Vec<&PrivilegeFinding> {
2082 self.findings
2083 .iter()
2084 .filter(|finding| finding.kind == FindingKind::Excess)
2085 .collect()
2086 }
2087
2088 #[must_use]
2094 pub fn controls(&self) -> &[String] {
2095 &self.controls
2096 }
2097
2098 #[must_use]
2100 pub const fn account(&self) -> &ServiceAccount {
2101 &self.account
2102 }
2103
2104 #[must_use]
2106 pub const fn kind(&self) -> DefinitionKind {
2107 self.kind
2108 }
2109}
2110
2111impl fmt::Display for PrivilegeReview {
2112 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
2113 writeln!(
2114 f,
2115 "{} runs as {} ({})",
2116 self.kind,
2117 self.account,
2118 self.account.justification()
2119 )?;
2120 for control in &self.controls {
2121 writeln!(f, " confirmed {control}")?;
2122 }
2123 for finding in &self.findings {
2124 writeln!(f, " {finding}")?;
2125 }
2126 if self.is_least_privilege() {
2127 write!(f, " verdict least privilege")
2128 } else {
2129 write!(
2130 f,
2131 " verdict NOT least privilege: {} excess(es)",
2132 self.excesses().len()
2133 )
2134 }
2135 }
2136}
2137
2138#[must_use]
2144pub fn review_least_privilege(
2145 definition: &ServiceDefinition,
2146 plan: &InstallPlan,
2147) -> PrivilegeReview {
2148 let mut controls = Vec::new();
2149 let mut findings = Vec::new();
2150 match definition.kind() {
2151 DefinitionKind::SystemdUnit => {
2152 review_systemd(definition.text(), plan, &mut controls, &mut findings);
2153 }
2154 DefinitionKind::LaunchdPlist => {
2155 review_launchd(definition.text(), plan, &mut controls, &mut findings);
2156 }
2157 DefinitionKind::WindowsScheduledTask => {
2158 review_scheduled_task(definition.text(), &mut controls, &mut findings);
2159 }
2160 DefinitionKind::WindowsService => {
2161 review_windows_service(definition.text(), plan, &mut controls, &mut findings);
2162 }
2163 }
2164 PrivilegeReview {
2165 kind: definition.kind(),
2166 account: ServiceAccount::for_definition(definition.kind(), plan.start_mode()),
2171 controls,
2172 findings,
2173 }
2174}
2175
2176fn permitted_paths(kind: DefinitionKind, plan: &InstallPlan) -> Vec<String> {
2178 let mut permitted = plan
2179 .directories()
2180 .all()
2181 .iter()
2182 .map(|path| path.to_string_lossy().into_owned())
2183 .collect::<Vec<_>>();
2184 if kind == DefinitionKind::SystemdUnit
2185 && let Some(secret_directory) = plan.secret_guard().and_then(Path::parent)
2186 {
2187 permitted.push(secret_directory.to_string_lossy().into_owned());
2188 }
2189 permitted
2190}
2191
2192fn same_path_text(left: &str, right: &str) -> bool {
2199 if cfg!(windows) {
2200 left.eq_ignore_ascii_case(right)
2201 } else {
2202 left == right
2203 }
2204}
2205
2206fn same_path_for(kind: DefinitionKind, left: &str, right: &str) -> bool {
2209 match kind {
2210 DefinitionKind::WindowsService | DefinitionKind::WindowsScheduledTask => {
2211 left.eq_ignore_ascii_case(right)
2212 }
2213 DefinitionKind::LaunchdPlist | DefinitionKind::SystemdUnit => left == right,
2214 }
2215}
2216
2217fn review_writable_paths(
2219 kind: DefinitionKind,
2220 subject: &str,
2221 listed: &[String],
2222 plan: &InstallPlan,
2223 controls: &mut Vec<String>,
2224 findings: &mut Vec<PrivilegeFinding>,
2225) {
2226 let permitted = permitted_paths(kind, plan);
2227 for entry in listed {
2228 if !permitted
2229 .iter()
2230 .any(|allowed| same_path_for(kind, allowed, entry))
2231 {
2232 findings.push(PrivilegeFinding {
2233 kind: FindingKind::Excess,
2234 subject: subject.to_string(),
2235 detail: format!(
2236 "{entry} is writable but is not one of this registration's required paths"
2237 ),
2238 });
2239 }
2240 }
2241 for allowed in &permitted {
2242 if !listed
2243 .iter()
2244 .any(|entry| same_path_for(kind, allowed, entry))
2245 {
2246 findings.push(PrivilegeFinding {
2247 kind: FindingKind::Shortfall,
2248 subject: subject.to_string(),
2249 detail: format!(
2250 "{allowed} is one of this registration's directories but is not writable, \
2251 so the daemon cannot use it"
2252 ),
2253 });
2254 }
2255 }
2256 if listed.len() == permitted.len() && findings.iter().all(|f| f.subject != subject) {
2257 controls.push(format!(
2258 "{subject} names exactly the required application-data and credential paths"
2259 ));
2260 }
2261}
2262
2263fn review_inbound_surface(
2271 text: &str,
2272 markers: &[(&str, &str)],
2273 controls: &mut Vec<String>,
2274 findings: &mut Vec<PrivilegeFinding>,
2275) {
2276 let mut clean = true;
2277 for (marker, detail) in markers {
2278 if text.contains(marker) {
2279 clean = false;
2280 findings.push(PrivilegeFinding {
2281 kind: FindingKind::Excess,
2282 subject: (*marker).to_string(),
2283 detail: (*detail).to_string(),
2284 });
2285 }
2286 }
2287 if clean {
2288 controls.push(
2289 "no socket, listener, or Mach service is published on the daemon's behalf".to_string(),
2290 );
2291 }
2292}
2293
2294fn review_systemd(
2295 text: &str,
2296 plan: &InstallPlan,
2297 controls: &mut Vec<String>,
2298 findings: &mut Vec<PrivilegeFinding>,
2299) {
2300 let directives = ini_directives(text, "Service");
2301 for expected in SYSTEMD_HARDENING {
2302 let (key, value) = expected
2303 .split_once('=')
2304 .expect("every hardening directive is written as key=value");
2305 match directives.get(key) {
2306 Some(actual) if actual == value => controls.push((*expected).to_string()),
2307 Some(actual) => findings.push(PrivilegeFinding {
2308 kind: FindingKind::Excess,
2309 subject: key.to_string(),
2310 detail: format!(
2311 "is `{actual}`, not `{value}`, so the unit keeps authority the \
2312 requirement does not ask for"
2313 ),
2314 }),
2315 None => findings.push(PrivilegeFinding {
2316 kind: FindingKind::Excess,
2317 subject: key.to_string(),
2318 detail: format!(
2319 "is absent, so the unit inherits systemd's default rather than `{value}`"
2320 ),
2321 }),
2322 }
2323 }
2324
2325 match directives.get("ReadWritePaths") {
2326 Some(value) => {
2327 let listed = split_quoted(value);
2328 review_writable_paths(
2329 DefinitionKind::SystemdUnit,
2330 "ReadWritePaths",
2331 &listed,
2332 plan,
2333 controls,
2334 findings,
2335 );
2336 }
2337 None => findings.push(PrivilegeFinding {
2338 kind: FindingKind::Shortfall,
2339 subject: "ReadWritePaths".to_string(),
2340 detail: "is absent, so `ProtectSystem=strict` leaves the daemon nowhere to write"
2341 .to_string(),
2342 }),
2343 }
2344
2345 if directives.contains_key("PrivateUsers")
2348 && directives.get("PrivateUsers") == Some(&"no".to_string())
2349 {
2350 findings.push(PrivilegeFinding {
2351 kind: FindingKind::Excess,
2352 subject: "PrivateUsers".to_string(),
2353 detail: "is explicitly disabled, which is broader than leaving it at systemd's default"
2354 .to_string(),
2355 });
2356 }
2357
2358 review_inbound_surface(
2359 text,
2360 &[
2361 (
2362 "ListenStream=",
2363 "asks systemd to open a listening socket for this service, which \
2364 07-security.md rule 2 forbids the product to have",
2365 ),
2366 (
2367 "ListenDatagram=",
2368 "asks systemd to open a listening socket for this service, which \
2369 07-security.md rule 2 forbids the product to have",
2370 ),
2371 ],
2372 controls,
2373 findings,
2374 );
2375}
2376
2377fn review_launchd(
2378 text: &str,
2379 plan: &InstallPlan,
2380 controls: &mut Vec<String>,
2381 findings: &mut Vec<PrivilegeFinding>,
2382) {
2383 match plist_string_value(text, "ProcessType").as_deref() {
2384 Some("Background") => controls.push("ProcessType=Background".to_string()),
2385 Some(other) => findings.push(PrivilegeFinding {
2386 kind: FindingKind::Excess,
2387 subject: "ProcessType".to_string(),
2388 detail: format!(
2389 "is `{other}`, which asks the scheduler for more CPU and I/O than a background \
2390 daemon needs"
2391 ),
2392 }),
2393 None => findings.push(PrivilegeFinding {
2394 kind: FindingKind::Excess,
2395 subject: "ProcessType".to_string(),
2396 detail: "is absent, so launchd applies its `Standard` default rather than \
2397 `Background`"
2398 .to_string(),
2399 }),
2400 }
2401
2402 match plan.start_mode() {
2403 StartMode::Boot => {
2404 if plist_bool_value(text, "SessionCreate") == Some(true) {
2405 findings.push(PrivilegeFinding {
2406 kind: FindingKind::Excess,
2407 subject: "SessionCreate".to_string(),
2408 detail: "asks launchd to create a security session for a job that runs \
2409 outside every login session and has no use for one"
2410 .to_string(),
2411 });
2412 } else {
2413 controls.push("SessionCreate is not requested".to_string());
2414 }
2415 match plist_string_value(text, "UserName").as_deref() {
2416 Some("root") => {
2417 controls.push("UserName=root, stated rather than inherited".to_string())
2418 }
2419 Some(other) => findings.push(PrivilegeFinding {
2420 kind: FindingKind::Shortfall,
2421 subject: "UserName".to_string(),
2422 detail: format!(
2423 "is `{other}`, which cannot unlock the System Keychain: \
2424 /var/db/SystemKey is root-only, so the daemon would start and then \
2425 find no credential"
2426 ),
2427 }),
2428 None => findings.push(PrivilegeFinding {
2429 kind: FindingKind::Shortfall,
2430 subject: "UserName".to_string(),
2431 detail: "is absent, so the account is launchd's implicit default and this \
2432 review cannot confirm it"
2433 .to_string(),
2434 }),
2435 }
2436 }
2437 StartMode::Login => {
2438 if let Some(named) = plist_string_value(text, "UserName") {
2439 findings.push(PrivilegeFinding {
2440 kind: FindingKind::Excess,
2441 subject: "UserName".to_string(),
2442 detail: format!(
2443 "names `{named}` in a LaunchAgent, which already runs as the operator; \
2444 naming an account here asks launchd for a switch a login-mode \
2445 registration has no reason to want"
2446 ),
2447 });
2448 } else {
2449 controls
2450 .push("no UserName: the agent runs as the operator and no other".to_string());
2451 }
2452 }
2453 }
2454
2455 review_inbound_surface(
2456 text,
2457 &[
2458 (
2459 "<key>Sockets</key>",
2460 "asks launchd to open a socket for this job, which 07-security.md rule 2 \
2461 forbids the product to have",
2462 ),
2463 (
2464 "<key>MachServices</key>",
2465 "publishes a Mach service, which is the RPC surface 07-security.md rule 2 \
2466 forbids the product to have",
2467 ),
2468 ],
2469 controls,
2470 findings,
2471 );
2472}
2473
2474fn review_scheduled_task(
2475 text: &str,
2476 controls: &mut Vec<String>,
2477 findings: &mut Vec<PrivilegeFinding>,
2478) {
2479 match xml_value(text, "RunLevel").as_deref() {
2480 Some("LeastPrivilege") => controls.push("RunLevel=LeastPrivilege".to_string()),
2481 Some(other) => findings.push(PrivilegeFinding {
2482 kind: FindingKind::Excess,
2483 subject: "RunLevel".to_string(),
2484 detail: format!(
2485 "is `{other}`, so the task runs with an elevated token whenever the operator is \
2486 an administrator"
2487 ),
2488 }),
2489 None => findings.push(PrivilegeFinding {
2490 kind: FindingKind::Excess,
2491 subject: "RunLevel".to_string(),
2492 detail: "is absent, so Task Scheduler decides the token rather than the definition"
2493 .to_string(),
2494 }),
2495 }
2496
2497 match xml_value(text, "LogonType").as_deref() {
2498 Some("InteractiveToken") => controls.push("LogonType=InteractiveToken".to_string()),
2499 Some(other) => findings.push(PrivilegeFinding {
2500 kind: FindingKind::Excess,
2501 subject: "LogonType".to_string(),
2502 detail: format!(
2503 "is `{other}`, which asks Windows to store or synthesise a credential for this \
2504 task; an interactive token needs neither"
2505 ),
2506 }),
2507 None => findings.push(PrivilegeFinding {
2508 kind: FindingKind::Shortfall,
2509 subject: "LogonType".to_string(),
2510 detail: "is absent, so this review cannot confirm that no credential is stored"
2511 .to_string(),
2512 }),
2513 }
2514}
2515
2516fn review_windows_service(
2517 text: &str,
2518 plan: &InstallPlan,
2519 controls: &mut Vec<String>,
2520 findings: &mut Vec<PrivilegeFinding>,
2521) {
2522 let directives = ini_directives(text, "windows-service");
2523
2524 match directives.get("ServiceType").map(String::as_str) {
2525 Some("OWN_PROCESS") => controls.push("ServiceType=OWN_PROCESS".to_string()),
2526 Some(other) => findings.push(PrivilegeFinding {
2527 kind: FindingKind::Excess,
2528 subject: "ServiceType".to_string(),
2529 detail: format!(
2530 "is `{other}`; an interactive or shared-process service reaches further than a \
2531 daemon that only talks to GitHub over HTTPS"
2532 ),
2533 }),
2534 None => findings.push(PrivilegeFinding {
2535 kind: FindingKind::Shortfall,
2536 subject: "ServiceType".to_string(),
2537 detail: "is absent, so this review cannot confirm the service is not interactive"
2538 .to_string(),
2539 }),
2540 }
2541
2542 match directives.get("Account").map(String::as_str) {
2546 Some(account) if account == ServiceAccount::LocalSystem.as_str() => {
2547 controls.push(format!(
2548 "Account={account}: the only stock account the machine-scoped store's DACL \
2549 (SY, BA, OW) admits"
2550 ));
2551 }
2552 Some(other) => findings.push(PrivilegeFinding {
2553 kind: FindingKind::Shortfall,
2554 subject: "Account".to_string(),
2555 detail: format!(
2556 "is `{other}`, which the machine-scoped store's DACL does not name, so the \
2557 daemon would start and then find no credential. Widening that DACL is not this \
2558 registration's to do: an ACE reaching `{other}` would also reach every other \
2559 service running under it"
2560 ),
2561 }),
2562 None => findings.push(PrivilegeFinding {
2563 kind: FindingKind::Shortfall,
2564 subject: "Account".to_string(),
2565 detail: "is absent, so this review cannot confirm which account was registered"
2566 .to_string(),
2567 }),
2568 }
2569
2570 match directives.get("ReadWritePaths") {
2571 Some(value) => {
2572 let listed = split_quoted(value);
2573 review_writable_paths(
2574 DefinitionKind::WindowsService,
2575 "ReadWritePaths",
2576 &listed,
2577 plan,
2578 controls,
2579 findings,
2580 );
2581 }
2582 None => findings.push(PrivilegeFinding {
2583 kind: FindingKind::Shortfall,
2584 subject: "ReadWritePaths".to_string(),
2585 detail: "is absent, so the directories the service was installed against are not \
2586 recorded"
2587 .to_string(),
2588 }),
2589 }
2590}
2591
2592fn ini_directives(text: &str, section: &str) -> BTreeMap<String, String> {
2600 let mut out = BTreeMap::new();
2601 let mut inside = false;
2602 for line in text.lines() {
2603 let line = line.trim();
2604 if line.starts_with('[') && line.ends_with(']') {
2605 inside = &line[1..line.len() - 1] == section;
2606 continue;
2607 }
2608 if !inside || line.is_empty() || line.starts_with('#') || line.starts_with(';') {
2609 continue;
2610 }
2611 if let Some((key, value)) = line.split_once('=') {
2612 out.insert(key.trim().to_string(), value.trim().to_string());
2613 }
2614 }
2615 out
2616}
2617
2618fn split_quoted(value: &str) -> Vec<String> {
2621 let mut out = Vec::new();
2622 let mut rest = value.trim();
2623 while !rest.is_empty() {
2624 if rest.starts_with('"') {
2625 if let Some(parsed) = executable_from_command_line(rest) {
2628 out.push(parsed.to_string_lossy().into_owned());
2629 }
2630 let mut depth = 0usize;
2632 let mut end = rest.len();
2633 for (index, c) in rest.char_indices() {
2634 match c {
2635 '\\' => depth += 1,
2636 '"' => {
2637 if depth.is_multiple_of(2) && index > 0 {
2638 end = index + 1;
2639 break;
2640 }
2641 depth = 0;
2642 }
2643 _ => depth = 0,
2644 }
2645 }
2646 rest = rest[end..].trim_start();
2647 } else {
2648 let end = rest.find(char::is_whitespace).unwrap_or(rest.len());
2649 out.push(rest[..end].to_string());
2650 rest = rest[end..].trim_start();
2651 }
2652 }
2653 out
2654}
2655
2656pub(crate) fn xml_value(text: &str, tag: &str) -> Option<String> {
2668 let open = format!("<{tag}>");
2669 let close = format!("</{tag}>");
2670 let start = text.find(&open)? + open.len();
2671 let end = text[start..].find(&close)? + start;
2672 Some(xml_unescape(text[start..end].trim()))
2673}
2674
2675#[cfg(any(windows, test))]
2681fn windows_login_task_starts_automatically(document: &str) -> bool {
2682 document.contains("<LogonTrigger>")
2683 && xml_value(document, "Enabled").as_deref() != Some("false")
2684}
2685
2686fn plist_value_after_key<'a>(text: &'a str, key: &str) -> Option<&'a str> {
2688 let marker = format!("<key>{key}</key>");
2689 let start = text.find(&marker)? + marker.len();
2690 Some(text[start..].trim_start())
2691}
2692
2693fn plist_string_value(text: &str, key: &str) -> Option<String> {
2694 let rest = plist_value_after_key(text, key)?;
2695 if !rest.starts_with("<string>") {
2696 return None;
2697 }
2698 xml_value(rest, "string")
2699}
2700
2701fn plist_bool_value(text: &str, key: &str) -> Option<bool> {
2702 let rest = plist_value_after_key(text, key)?;
2703 if rest.starts_with("<true/>") {
2704 Some(true)
2705 } else if rest.starts_with("<false/>") {
2706 Some(false)
2707 } else {
2708 None
2709 }
2710}
2711
2712pub const RECORD_SCHEMA_VERSION: u32 = 1;
2723
2724#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
2736pub struct InstallRecord {
2737 pub schema_version: u32,
2739 pub service_name: String,
2741 pub manager: String,
2743 pub start_mode: StartMode,
2745 pub account: ServiceAccount,
2747 pub binary: PathBuf,
2749 #[serde(default)]
2757 pub source_binary: Option<PathBuf>,
2758 pub arguments: Vec<String>,
2760 pub restart_delay_secs: u64,
2762 pub restart_reset_secs: u64,
2764 pub log_file: PathBuf,
2766 #[serde(default)]
2774 pub starts_on_demand: bool,
2775 pub definition_path: Option<PathBuf>,
2777 pub installed_at: DateTime<Utc>,
2779 pub installed_by_version: String,
2781 pub directories: ServiceDirectories,
2786}
2787
2788impl InstallRecord {
2789 #[must_use]
2791 pub fn path(paths: &AppPaths) -> PathBuf {
2792 paths.config_dir().join(RECORD_FILE)
2793 }
2794
2795 #[must_use]
2797 pub fn of(plan: &InstallPlan, definition: &ServiceDefinition, at: DateTime<Utc>) -> Self {
2798 Self {
2799 schema_version: RECORD_SCHEMA_VERSION,
2800 service_name: plan.identity().name().to_string(),
2801 manager: definition.kind().manager().to_string(),
2802 start_mode: plan.start_mode(),
2803 account: plan.account().clone(),
2804 binary: plan.binary().to_path_buf(),
2805 arguments: plan
2806 .arguments()
2807 .iter()
2808 .map(|argument| argument.to_string_lossy().into_owned())
2809 .collect(),
2810 restart_delay_secs: plan.restart().delay().as_secs(),
2811 restart_reset_secs: plan.restart().reset_after().as_secs(),
2812 starts_on_demand: plan.is_on_demand(),
2813 source_binary: plan.source_binary().map(Path::to_path_buf),
2814 log_file: plan.directories().log_file(),
2815 definition_path: definition.install_path().map(Path::to_path_buf),
2816 installed_at: at,
2817 installed_by_version: env!("CARGO_PKG_VERSION").to_string(),
2818 directories: plan.directories().clone(),
2819 }
2820 }
2821
2822 #[must_use]
2826 pub fn restart(&self) -> RestartPolicy {
2827 RestartPolicy::new(
2828 Duration::from_secs(self.restart_delay_secs),
2829 Duration::from_secs(self.restart_reset_secs),
2830 )
2831 .unwrap_or_default()
2832 }
2833
2834 pub fn read(paths: &AppPaths) -> Result<Option<Self>, ServiceError> {
2845 let path = Self::path(paths);
2846 let text = match std::fs::read_to_string(&path) {
2847 Ok(text) => text,
2848 Err(error) if error.kind() == std::io::ErrorKind::NotFound => return Ok(None),
2849 Err(error) if error.kind() == std::io::ErrorKind::PermissionDenied => {
2852 return Err(ServiceError::RecordNotPermitted {
2853 path,
2854 detail: error.to_string(),
2855 });
2856 }
2857 Err(error) => {
2858 return Err(ServiceError::Record {
2859 operation: "read",
2860 path,
2861 detail: error.to_string(),
2862 });
2863 }
2864 };
2865 let record: Self =
2866 toml::from_str(&text).map_err(|error| ServiceError::RecordUnreadable {
2867 path: path.clone(),
2868 detail: error.to_string(),
2869 })?;
2870 if record.schema_version != RECORD_SCHEMA_VERSION {
2871 return Err(ServiceError::RecordUnreadable {
2872 path,
2873 detail: format!(
2874 "it declares schema version {} and this build reads version {}",
2875 record.schema_version, RECORD_SCHEMA_VERSION
2876 ),
2877 });
2878 }
2879 Ok(Some(record))
2880 }
2881
2882 pub fn write(&self, paths: &AppPaths) -> Result<(), ServiceError> {
2888 use std::io::Write as _;
2889
2890 let path = Self::path(paths);
2891 let text = toml::to_string_pretty(self).map_err(|error| ServiceError::Record {
2892 operation: "encode",
2893 path: path.clone(),
2894 detail: error.to_string(),
2895 })?;
2896 if let Some(parent) = path.parent() {
2897 std::fs::create_dir_all(parent).map_err(|error| ServiceError::Record {
2898 operation: "write",
2899 path: path.clone(),
2900 detail: error.to_string(),
2901 })?;
2902 }
2903 let parent = path.parent().ok_or_else(|| ServiceError::Record {
2904 operation: "write",
2905 path: path.clone(),
2906 detail: "the record path has no parent directory".to_string(),
2907 })?;
2908 let mut temporary =
2909 tempfile::NamedTempFile::new_in(parent).map_err(|error| ServiceError::Record {
2910 operation: "write",
2911 path: path.clone(),
2912 detail: error.to_string(),
2913 })?;
2914 temporary
2915 .write_all(text.as_bytes())
2916 .and_then(|()| temporary.as_file().sync_all())
2917 .map_err(|error| ServiceError::Record {
2918 operation: "write",
2919 path: path.clone(),
2920 detail: error.to_string(),
2921 })?;
2922 #[cfg(unix)]
2938 {
2939 use std::os::unix::fs::PermissionsExt as _;
2940
2941 temporary
2942 .as_file()
2943 .set_permissions(std::fs::Permissions::from_mode(0o644))
2944 .map_err(|error| ServiceError::Record {
2945 operation: "write",
2946 path: path.clone(),
2947 detail: error.to_string(),
2948 })?;
2949 }
2950 temporary
2951 .persist(&path)
2952 .map(|_| ())
2953 .map_err(|error| ServiceError::Record {
2954 operation: "write",
2955 path,
2956 detail: error.error.to_string(),
2957 })
2958 }
2959
2960 pub fn remove(paths: &AppPaths) -> Result<bool, ServiceError> {
2966 let path = Self::path(paths);
2967 match std::fs::remove_file(&path) {
2968 Ok(()) => Ok(true),
2969 Err(error) if error.kind() == std::io::ErrorKind::NotFound => Ok(false),
2970 Err(error) => Err(ServiceError::Record {
2971 operation: "remove",
2972 path,
2973 detail: error.to_string(),
2974 }),
2975 }
2976 }
2977}
2978
2979const CONTACT_SCHEMA_VERSION: u32 = 1;
2985
2986#[derive(Debug, Clone, Serialize, Deserialize)]
2987struct ContactRecord {
2988 schema_version: u32,
2989 last_success: DateTime<Utc>,
2990}
2991
2992pub fn record_github_contact(paths: &AppPaths, at: DateTime<Utc>) -> Result<(), ServiceError> {
3013 let path = contact_path(paths);
3014 let record = ContactRecord {
3015 schema_version: CONTACT_SCHEMA_VERSION,
3016 last_success: at,
3017 };
3018 let failed = |detail: String| ServiceError::Record {
3019 operation: "write",
3020 path: path.clone(),
3021 detail,
3022 };
3023 let text = toml::to_string_pretty(&record).map_err(|error| failed(error.to_string()))?;
3024 let directory = path.parent().unwrap_or_else(|| Path::new("."));
3025 std::fs::create_dir_all(directory).map_err(|error| failed(error.to_string()))?;
3026 let temporary = path.with_extension("toml.new");
3027 std::fs::write(&temporary, text).map_err(|error| failed(error.to_string()))?;
3028 std::fs::rename(&temporary, &path).map_err(|error| failed(error.to_string()))
3029}
3030
3031pub fn last_github_contact(paths: &AppPaths) -> Result<Option<DateTime<Utc>>, ServiceError> {
3043 let path = contact_path(paths);
3044 let text = match std::fs::read_to_string(&path) {
3045 Ok(text) => text,
3046 Err(error) if error.kind() == std::io::ErrorKind::NotFound => return Ok(None),
3047 Err(error) => {
3048 return Err(ServiceError::Record {
3049 operation: "read",
3050 path,
3051 detail: error.to_string(),
3052 });
3053 }
3054 };
3055 let record: ContactRecord = toml::from_str(&text).map_err(|error| ServiceError::Record {
3056 operation: "read",
3057 path,
3058 detail: error.to_string(),
3059 })?;
3060 Ok(Some(record.last_success))
3061}
3062
3063#[must_use]
3065pub fn contact_path(paths: &AppPaths) -> PathBuf {
3066 paths.state_dir().join(CONTACT_FILE)
3067}
3068
3069const ROOT_REFUSAL_SCHEMA_VERSION: u32 = 1;
3074
3075#[derive(Debug, Clone, Serialize, Deserialize)]
3076struct RootRefusalFile {
3077 schema_version: u32,
3078 #[serde(default)]
3080 refusals: BTreeMap<String, RootRefusalEntry>,
3081}
3082
3083#[derive(Debug, Clone, Serialize, Deserialize)]
3084struct RootRefusalEntry {
3085 at: DateTime<Utc>,
3086 kind: String,
3087 root: String,
3088 detail: String,
3089}
3090
3091#[derive(Debug, Clone, PartialEq, Eq)]
3093pub struct RunnerRootRefusal {
3094 pub policy: String,
3096 pub at: DateTime<Utc>,
3098 pub kind: String,
3100 pub root: String,
3102 pub detail: String,
3105}
3106
3107pub fn record_runner_root_refusal(
3141 paths: &AppPaths,
3142 policy: &str,
3143 at: DateTime<Utc>,
3144 kind: &str,
3145 root: &str,
3146 detail: &str,
3147) -> Result<(), ServiceError> {
3148 let mut file = read_refusal_file(paths)?.unwrap_or(RootRefusalFile {
3149 schema_version: ROOT_REFUSAL_SCHEMA_VERSION,
3150 refusals: BTreeMap::new(),
3151 });
3152 file.schema_version = ROOT_REFUSAL_SCHEMA_VERSION;
3153 file.refusals.insert(
3154 policy.to_owned(),
3155 RootRefusalEntry {
3156 at,
3157 kind: kind.to_owned(),
3158 root: root.to_owned(),
3159 detail: detail.to_owned(),
3160 },
3161 );
3162 write_refusal_file(paths, &file)
3163}
3164
3165pub fn clear_runner_root_refusal(paths: &AppPaths, policy: &str) -> Result<(), ServiceError> {
3176 let Some(mut file) = read_refusal_file(paths)? else {
3177 return Ok(());
3178 };
3179 if file.refusals.remove(policy).is_none() {
3180 return Ok(());
3181 }
3182 if file.refusals.is_empty() {
3183 let path = root_refusal_path(paths);
3184 return match std::fs::remove_file(&path) {
3185 Ok(()) => Ok(()),
3186 Err(error) if error.kind() == std::io::ErrorKind::NotFound => Ok(()),
3187 Err(error) => Err(ServiceError::Record {
3188 operation: "remove",
3189 path,
3190 detail: error.to_string(),
3191 }),
3192 };
3193 }
3194 write_refusal_file(paths, &file)
3195}
3196
3197pub fn runner_root_refusals(paths: &AppPaths) -> Result<Vec<RunnerRootRefusal>, ServiceError> {
3209 Ok(read_refusal_file(paths)?
3210 .map(|file| {
3211 file.refusals
3212 .into_iter()
3213 .map(|(policy, entry)| RunnerRootRefusal {
3214 policy,
3215 at: entry.at,
3216 kind: entry.kind,
3217 root: entry.root,
3218 detail: entry.detail,
3219 })
3220 .collect()
3221 })
3222 .unwrap_or_default())
3223}
3224
3225fn read_refusal_file(paths: &AppPaths) -> Result<Option<RootRefusalFile>, ServiceError> {
3226 let path = root_refusal_path(paths);
3227 let text = match std::fs::read_to_string(&path) {
3228 Ok(text) => text,
3229 Err(error) if error.kind() == std::io::ErrorKind::NotFound => return Ok(None),
3230 Err(error) => {
3231 return Err(ServiceError::Record {
3232 operation: "read",
3233 path,
3234 detail: error.to_string(),
3235 });
3236 }
3237 };
3238 toml::from_str(&text)
3239 .map(Some)
3240 .map_err(|error| ServiceError::Record {
3241 operation: "read",
3242 path,
3243 detail: error.to_string(),
3244 })
3245}
3246
3247fn write_refusal_file(paths: &AppPaths, file: &RootRefusalFile) -> Result<(), ServiceError> {
3248 let path = root_refusal_path(paths);
3249 let failed = |detail: String| ServiceError::Record {
3250 operation: "write",
3251 path: path.clone(),
3252 detail,
3253 };
3254 let text = toml::to_string_pretty(file).map_err(|error| failed(error.to_string()))?;
3255 let directory = path.parent().unwrap_or_else(|| Path::new("."));
3256 std::fs::create_dir_all(directory).map_err(|error| failed(error.to_string()))?;
3257 let temporary = path.with_extension("toml.new");
3258 std::fs::write(&temporary, text).map_err(|error| failed(error.to_string()))?;
3259 std::fs::rename(&temporary, &path).map_err(|error| failed(error.to_string()))
3260}
3261
3262#[must_use]
3264pub fn root_refusal_path(paths: &AppPaths) -> PathBuf {
3265 paths.state_dir().join(ROOT_REFUSAL_FILE)
3266}
3267
3268#[derive(Debug, Clone, PartialEq, Eq)]
3279pub enum BinaryPath {
3280 Current {
3282 path: PathBuf,
3284 },
3285 Missing {
3292 recorded: PathBuf,
3294 },
3295 NotExecutable {
3298 recorded: PathBuf,
3300 detail: String,
3302 },
3303 Diverged {
3305 recorded: PathBuf,
3307 registered: PathBuf,
3309 },
3310}
3311
3312impl BinaryPath {
3313 #[must_use]
3315 pub const fn is_error(&self) -> bool {
3316 !matches!(self, Self::Current { .. })
3317 }
3318
3319 #[must_use]
3321 pub fn recorded(&self) -> &Path {
3322 match self {
3323 Self::Current { path } => path,
3324 Self::Missing { recorded }
3325 | Self::NotExecutable { recorded, .. }
3326 | Self::Diverged { recorded, .. } => recorded,
3327 }
3328 }
3329}
3330
3331impl fmt::Display for BinaryPath {
3332 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
3333 match self {
3334 Self::Current { path } => write!(f, "{}", path.display()),
3335 Self::Missing { recorded } => write!(
3336 f,
3337 "{} -- STALE: nothing is at the recorded path, so the service cannot start. A \
3338 package manager that moved the binary is the usual cause; an `npm i -g` \
3339 installation moves with the active Node version. Run `service install` again \
3340 from the binary that is now installed.",
3341 recorded.display()
3342 ),
3343 Self::NotExecutable { recorded, detail } => write!(
3344 f,
3345 "{} -- STALE: {detail}, so the service cannot start. Run `service install` again \
3346 from the installed binary.",
3347 recorded.display()
3348 ),
3349 Self::Diverged {
3350 recorded,
3351 registered,
3352 } => write!(
3353 f,
3354 "{} -- STALE: the service manager is registered to start {} instead. Something \
3355 has edited the registration since it was installed. Run `service uninstall` and \
3356 `service install`; neither touches configuration, secrets, or the cache.",
3357 recorded.display(),
3358 registered.display()
3359 ),
3360 }
3361 }
3362}
3363
3364#[must_use]
3372pub fn inspect_binary(recorded: &Path, registered: Option<&Path>) -> BinaryPath {
3373 match std::fs::metadata(recorded) {
3374 Err(error) if error.kind() == std::io::ErrorKind::NotFound => {
3375 return BinaryPath::Missing {
3376 recorded: recorded.to_path_buf(),
3377 };
3378 }
3379 Err(error) => {
3380 return BinaryPath::NotExecutable {
3381 recorded: recorded.to_path_buf(),
3382 detail: format!("it cannot be inspected ({error})"),
3383 };
3384 }
3385 Ok(metadata) if !metadata.is_file() => {
3386 return BinaryPath::NotExecutable {
3387 recorded: recorded.to_path_buf(),
3388 detail: "what is there is not a file".to_string(),
3389 };
3390 }
3391 Ok(_) => {}
3392 }
3393 if let Some(registered) = registered
3394 && !same_path_text(&recorded.to_string_lossy(), ®istered.to_string_lossy())
3395 {
3396 return BinaryPath::Diverged {
3397 recorded: recorded.to_path_buf(),
3398 registered: registered.to_path_buf(),
3399 };
3400 }
3401 BinaryPath::Current {
3402 path: recorded.to_path_buf(),
3403 }
3404}
3405
3406#[derive(Debug, Clone, PartialEq, Eq)]
3412pub struct Registration {
3413 pub manager: DefinitionKind,
3418 pub start_mode: StartMode,
3420 pub command_line: String,
3422 pub account: Option<String>,
3424 pub running: bool,
3426 pub starts_automatically: bool,
3433 pub restart_delay: Option<Duration>,
3442}
3443
3444impl Registration {
3445 #[must_use]
3447 pub fn binary(&self) -> Option<PathBuf> {
3448 executable_from_command_line(&self.command_line)
3449 }
3450}
3451
3452pub trait ServiceControl: fmt::Debug {
3458 fn manager(&self) -> DefinitionKind;
3460
3461 fn install(&self, plan: &InstallPlan) -> Result<ServiceDefinition, ServiceError>;
3467
3468 fn uninstall(&self, identity: &ServiceIdentity) -> Result<bool, ServiceError>;
3479
3480 fn query(&self, identity: &ServiceIdentity) -> Result<Option<Registration>, ServiceError>;
3486
3487 fn start(&self, identity: &ServiceIdentity) -> Result<(), ServiceError>;
3494
3495 fn stop(&self, identity: &ServiceIdentity) -> Result<bool, ServiceError>;
3501}
3502
3503pub trait ControlFactory: fmt::Debug + Send + Sync {
3509 fn control(&self, mode: StartMode) -> Result<Box<dyn ServiceControl>, ServiceError>;
3515}
3516
3517#[derive(Debug, Clone, Copy, Default)]
3519pub struct HostControls;
3520
3521#[derive(Debug, Clone)]
3527pub struct Installed {
3528 pub plan: InstallPlan,
3530 pub definition: ServiceDefinition,
3532 pub record: InstallRecord,
3534 pub review: PrivilegeReview,
3536 pub runner_root: RootAccessSummary,
3541 pub replaced_existing: bool,
3549}
3550
3551#[derive(Debug, Clone, PartialEq, Eq)]
3553pub struct Uninstalled {
3554 pub removed_registration: bool,
3556 pub removed_record: bool,
3558 pub removed_definition: Option<PathBuf>,
3560 pub preserved: Vec<PathBuf>,
3567}
3568
3569impl fmt::Display for Uninstalled {
3570 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
3571 if self.removed_registration {
3572 writeln!(f, "The service registration was removed.")?;
3573 } else {
3574 writeln!(f, "There was no service registration to remove.")?;
3575 }
3576 writeln!(f, "Nothing else was deleted. These are untouched:")?;
3577 for path in &self.preserved {
3578 writeln!(f, " {}", path.display())?;
3579 }
3580 write!(
3581 f,
3582 "The stored GitHub token is untouched too; `auth logout` is what purges it."
3583 )
3584 }
3585}
3586
3587#[derive(Debug, Clone, PartialEq, Eq)]
3589pub struct StartModeChange {
3590 pub from: StartMode,
3592 pub to: StartMode,
3594 pub changed: bool,
3596 pub store_scope: crate::secrets::SecretScope,
3599 pub runner_root: RootAccessSummary,
3605}
3606
3607impl fmt::Display for StartModeChange {
3608 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
3609 if !self.changed {
3610 return write!(f, "The service already starts at {}.", self.to);
3611 }
3612 write!(
3613 f,
3614 "The service now starts at {} instead of {}. It reads the {}-scoped secret store; \
3615 if the token was stored under the other scope, run `auth login` again. {}",
3616 self.to, self.from, self.store_scope, self.runner_root
3617 )
3618 }
3619}
3620
3621#[derive(Debug, Clone)]
3629pub struct ServiceOperations {
3630 paths: AppPaths,
3631 identity: ServiceIdentity,
3632 controls: std::sync::Arc<dyn ControlFactory>,
3633 runner_root: Option<LocalAbsolutePath>,
3634}
3635
3636impl ServiceOperations {
3637 #[must_use]
3639 pub fn on_this_host(paths: AppPaths) -> Self {
3640 Self::with_controls(
3641 paths,
3642 ServiceIdentity::product(),
3643 std::sync::Arc::new(HostControls),
3644 )
3645 }
3646
3647 #[must_use]
3655 pub fn with_controls(
3656 paths: AppPaths,
3657 identity: ServiceIdentity,
3658 controls: std::sync::Arc<dyn ControlFactory>,
3659 ) -> Self {
3660 Self {
3661 paths,
3662 identity,
3663 controls,
3664 runner_root: None,
3665 }
3666 }
3667
3668 #[must_use]
3687 pub fn with_runner_root(mut self, root: LocalAbsolutePath) -> Self {
3688 self.runner_root = Some(root);
3689 self
3690 }
3691
3692 #[must_use]
3694 pub const fn paths(&self) -> &AppPaths {
3695 &self.paths
3696 }
3697
3698 #[must_use]
3700 pub const fn identity(&self) -> &ServiceIdentity {
3701 &self.identity
3702 }
3703
3704 pub fn install(&self, request: &InstallRequest) -> Result<Installed, ServiceError> {
3724 self.paths
3725 .create_all()
3726 .map_err(|source| ServiceError::Paths {
3727 source: Box::new(source),
3728 })?;
3729
3730 let _guard = self.refuse_while_an_agent_runs()?;
3733
3734 let (replacing, previous_was_running) = match self.find_registration()? {
3763 Some((existing, registration)) if existing == request.start_mode() => {
3764 (true, registration.running)
3765 }
3766 Some((existing, _)) => {
3767 return Err(ServiceError::AlreadyInstalled {
3768 name: self.identity.name().to_string(),
3769 existing,
3770 requested: request.start_mode(),
3771 });
3772 }
3773 None => (false, false),
3774 };
3775
3776 let plan = InstallPlan::resolve(
3777 self.identity.clone(),
3778 request,
3779 ServiceDirectories::of(&self.paths),
3780 )?;
3781
3782 let control = self.controls.control(plan.start_mode())?;
3787
3788 let root = self.prepare_runner_root(plan.start_mode())?;
3794
3795 let previous = if replacing {
3800 InstallRecord::read(&self.paths).ok().flatten()
3801 } else {
3802 None
3803 };
3804 if replacing && let Err(cause) = control.uninstall(&self.identity) {
3805 return Err(undo_runner_root(&root, "install", &self.identity, cause));
3806 }
3807
3808 let definition = match control.install(&plan) {
3809 Ok(definition) => definition,
3810 Err(cause) => {
3811 let restored =
3814 self.reinstate(control.as_ref(), previous.as_ref(), previous_was_running);
3815 return Err(rolled_back(
3816 retained_runner_root(&root),
3817 restored,
3818 "install",
3819 &self.identity,
3820 cause,
3821 ));
3822 }
3823 };
3824 let review = review_least_privilege(&definition, &plan);
3825 let record = InstallRecord::of(&plan, &definition, Utc::now());
3826 if let Err(cause) = record.write(&self.paths) {
3827 let rollback = control
3828 .uninstall(&self.identity)
3829 .map(|_| ())
3830 .and_then(|()| {
3831 self.reinstate(control.as_ref(), previous.as_ref(), previous_was_running)
3832 });
3833 return Err(rolled_back(
3834 retained_runner_root(&root),
3835 rollback,
3836 "install",
3837 &self.identity,
3838 cause,
3839 ));
3840 }
3841
3842 if let Err(cause) = control.start(&self.identity) {
3849 let remove_new = control.uninstall(&self.identity).map(|_| ());
3850 let restore_registration = remove_new.and_then(|()| {
3851 self.reinstate(control.as_ref(), previous.as_ref(), previous_was_running)
3852 });
3853 let restore_record = restore_registration.and_then(|()| match &previous {
3854 Some(previous) => previous.write(&self.paths),
3855 None => InstallRecord::remove(&self.paths).map(|_| ()),
3856 });
3857 return Err(rolled_back(
3858 retained_runner_root(&root),
3859 restore_record,
3860 "install and start",
3861 &self.identity,
3862 cause,
3863 ));
3864 }
3865 Ok(Installed {
3866 plan,
3867 definition,
3868 record,
3869 review,
3870 runner_root: root.summary().clone(),
3871 replaced_existing: replacing,
3872 })
3873 }
3874
3875 fn reinstate(
3886 &self,
3887 control: &dyn ServiceControl,
3888 previous: Option<&InstallRecord>,
3889 was_running: bool,
3890 ) -> Result<(), ServiceError> {
3891 let Some(record) = previous else {
3892 return Ok(());
3893 };
3894 let plan = InstallPlan::unchecked(
3895 self.identity.clone(),
3896 record.start_mode,
3897 record.binary.clone(),
3898 record.directories.clone(),
3899 )
3900 .with_arguments(record.arguments.clone())
3901 .with_restart(record.restart());
3902 let plan = if record.starts_on_demand {
3903 plan.started_on_demand()
3904 } else {
3905 plan
3906 };
3907 let plan = match crate::secrets::PlatformSecretStore::for_start_mode(record.start_mode) {
3908 Ok(store) => plan.with_secret_guard(store.guard()),
3909 Err(_) => plan,
3910 };
3911 control.install(&plan)?;
3912 if was_running {
3913 control.start(&self.identity)?;
3914 }
3915 Ok(())
3916 }
3917
3918 pub fn uninstall(&self) -> Result<Uninstalled, ServiceError> {
3927 let record = InstallRecord::read(&self.paths).ok().flatten();
3928 let removed_definition = record
3929 .as_ref()
3930 .and_then(|record| record.definition_path.clone());
3931
3932 let mut removed_registration = false;
3933 for mode in [StartMode::Boot, StartMode::Login] {
3937 let control = self.controls.control(mode)?;
3938 if control.uninstall(&self.identity)? {
3939 removed_registration = true;
3940 }
3941 }
3942 let removed_record = InstallRecord::remove(&self.paths)?;
3943 Ok(Uninstalled {
3944 removed_registration,
3945 removed_record,
3946 removed_definition: removed_definition.filter(|_| removed_registration),
3947 preserved: self
3948 .paths
3949 .all()
3950 .iter()
3951 .map(|(_, path)| (*path).to_path_buf())
3952 .collect(),
3953 })
3954 }
3955
3956 pub fn set_start_mode(&self, to: StartMode) -> Result<StartModeChange, ServiceError> {
3978 let Some(record) = InstallRecord::read(&self.paths)? else {
3979 return Err(ServiceError::NotInstalled {
3980 name: self.identity.name().to_string(),
3981 operation: "switch the start mode of",
3982 });
3983 };
3984 let from = record.start_mode;
3985 if from == to {
3986 return Ok(StartModeChange {
3990 from,
3991 to,
3992 changed: false,
3993 store_scope: crate::secrets::SecretScope::for_start_mode(to),
3994 runner_root: RootAccessSummary::NotApplicable,
3995 });
3996 }
3997
3998 #[cfg(windows)]
3999 let arguments = {
4000 let mut arguments = record.arguments.clone();
4001 arguments.retain(|argument| argument != WINDOWS_SCM_HOST_ARGUMENT);
4002 if to == StartMode::Boot {
4003 arguments.push(WINDOWS_SCM_HOST_ARGUMENT.to_string());
4004 }
4005 arguments
4006 };
4007 #[cfg(not(windows))]
4008 let arguments = record.arguments.clone();
4009 let plan = InstallPlan::unchecked(
4010 self.identity.clone(),
4011 to,
4012 record.binary.clone(),
4013 record.directories.clone(),
4014 )
4015 .with_arguments(arguments)
4016 .with_restart(record.restart());
4017 let plan = if record.starts_on_demand {
4018 plan.started_on_demand()
4019 } else {
4020 plan
4021 };
4022 let plan = match crate::secrets::PlatformSecretStore::for_start_mode(to) {
4023 Ok(store) => plan.with_secret_guard(store.guard()),
4024 Err(_) => plan,
4025 };
4026
4027 let target = self.controls.control(to)?;
4034 let previous = self.controls.control(from)?;
4039
4040 let root = self.prepare_runner_root(to)?;
4048
4049 let definition = match target.install(&plan) {
4050 Ok(definition) => definition,
4051 Err(cause) => {
4052 return Err(undo_runner_root(
4053 &root,
4054 "switch start mode",
4055 &self.identity,
4056 cause,
4057 ));
4058 }
4059 };
4060 let next_record = InstallRecord::of(&plan, &definition, record.installed_at);
4061 if let Err(cause) = next_record.write(&self.paths) {
4062 return Err(rolled_back(
4063 retained_runner_root(&root),
4064 target.uninstall(&self.identity),
4065 "switch start mode",
4066 &self.identity,
4067 cause,
4068 ));
4069 }
4070
4071 if let Err(cause) = previous.uninstall(&self.identity) {
4075 let target_rollback = target.uninstall(&self.identity);
4076 let record_rollback = record.write(&self.paths);
4077 return Err(rolled_back(
4078 retained_runner_root(&root),
4079 target_rollback.and(record_rollback),
4080 "switch start mode",
4081 &self.identity,
4082 cause,
4083 ));
4084 }
4085 Ok(StartModeChange {
4086 from,
4087 to,
4088 changed: true,
4089 store_scope: crate::secrets::SecretScope::for_start_mode(to),
4090 runner_root: root.summary().clone(),
4091 })
4092 }
4093
4094 pub fn start(&self) -> Result<(), ServiceError> {
4100 let Some((mode, _)) = self.find_registration()? else {
4101 return Err(ServiceError::NotInstalled {
4102 name: self.identity.name().to_string(),
4103 operation: "start",
4104 });
4105 };
4106 self.controls.control(mode)?.start(&self.identity)
4107 }
4108
4109 pub fn stop(&self) -> Result<bool, ServiceError> {
4115 let Some((mode, _)) = self.find_registration()? else {
4116 return Err(ServiceError::NotInstalled {
4117 name: self.identity.name().to_string(),
4118 operation: "stop",
4119 });
4120 };
4121 self.controls.control(mode)?.stop(&self.identity)
4122 }
4123
4124 pub fn status(&self) -> Result<ServiceStatus, ServiceError> {
4143 let (record, record_refused) = match InstallRecord::read(&self.paths) {
4144 Ok(record) => (record, None),
4145 Err(refusal @ ServiceError::RecordNotPermitted { .. }) => (None, Some(refusal)),
4146 Err(error) => return Err(error),
4147 };
4148 let found = self.find_registration()?;
4149 let last_github_contact = last_github_contact(&self.paths)?;
4150 Ok(ServiceStatus::compose(
4151 self.identity.clone(),
4152 record,
4153 record_refused.as_ref(),
4154 found.map(|(_, registration)| registration),
4155 last_github_contact,
4156 &self.paths,
4157 ))
4158 }
4159
4160 fn prepare_runner_root(&self, mode: StartMode) -> Result<RootAccessChange, ServiceError> {
4170 #[cfg(not(windows))]
4171 {
4172 let _ = (mode, &self.runner_root);
4175 Ok(RootAccessChange::not_applicable())
4176 }
4177 #[cfg(windows)]
4178 {
4179 let wrap = |source| ServiceError::RunnerRoot {
4180 source: Box::new(source),
4181 };
4182 if let Some(root) = self
4200 .runner_root
4201 .as_ref()
4202 .filter(|_| self.identity.is_fixture() || cfg!(test))
4203 {
4204 let admission = RootAdmission::of_this_account().map_err(wrap)?;
4205 return crate::runner_root_access::reconcile(&self.paths, root, &admission)
4206 .map_err(wrap);
4207 }
4208
4209 let admission = match ServiceAccount::for_start_mode(mode) {
4210 ServiceAccount::LocalSystem => RootAdmission::LocalSystem,
4213 ServiceAccount::InvokingUser | ServiceAccount::Root => {
4217 RootAdmission::of_this_account().map_err(wrap)?
4218 }
4219 };
4220 crate::runner_root_access::ensure_default_root(&self.paths, &admission).map_err(wrap)
4221 }
4222 }
4223
4224 fn refuse_while_an_agent_runs(&self) -> Result<HostLock, ServiceError> {
4226 HostLock::try_acquire(&self.paths, LockKind::SingleInstance).map_err(
4227 |source| match source {
4228 held @ LockError::Held { .. } => ServiceError::LockHeld {
4229 source: Box::new(held),
4230 },
4231 other => ServiceError::LockUnreadable {
4232 source: Box::new(other),
4233 },
4234 },
4235 )
4236 }
4237
4238 fn find_registration(&self) -> Result<Option<(StartMode, Registration)>, ServiceError> {
4240 for mode in [StartMode::Boot, StartMode::Login] {
4241 let control = self.controls.control(mode)?;
4242 if let Some(registration) = control.query(&self.identity)? {
4243 return Ok(Some((registration.start_mode, registration)));
4244 }
4245 }
4246 Ok(None)
4247 }
4248}
4249
4250#[derive(Debug, Clone, PartialEq, Eq)]
4256pub struct StatusProblem {
4257 pub subject: &'static str,
4259 pub detail: String,
4261}
4262
4263impl fmt::Display for StatusProblem {
4264 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
4265 write!(f, "{}: {}", self.subject, self.detail)
4266 }
4267}
4268
4269#[derive(Debug, Clone)]
4278pub struct ServiceStatus {
4279 identity: ServiceIdentity,
4280 record: Option<InstallRecord>,
4281 registration: Option<Registration>,
4282 binary: Option<BinaryPath>,
4283 log_file: PathBuf,
4284 store: Option<crate::secrets::ActiveStore>,
4285 last_github_contact: Option<DateTime<Utc>>,
4286 runner_root: Option<(PathBuf, RootAccessReport)>,
4287 problems: Vec<StatusProblem>,
4288 notes: Vec<String>,
4289}
4290
4291impl ServiceStatus {
4292 fn compose(
4297 identity: ServiceIdentity,
4298 record: Option<InstallRecord>,
4299 record_refused: Option<&ServiceError>,
4300 registration: Option<Registration>,
4301 last_github_contact: Option<DateTime<Utc>>,
4302 paths: &AppPaths,
4303 ) -> Self {
4304 let mut problems = Vec::new();
4305 let mut notes = Vec::new();
4306
4307 if let Some(refusal) = record_refused {
4308 problems.push(StatusProblem {
4313 subject: "install record",
4314 detail: refusal.to_string(),
4315 });
4316 }
4317
4318 let log_file = record.as_ref().map_or_else(
4319 || ServiceDirectories::of(paths).log_file(),
4320 |record| record.log_file.clone(),
4321 );
4322
4323 let binary = record.as_ref().map(|record| {
4324 inspect_binary(
4325 &record.binary,
4326 registration
4327 .as_ref()
4328 .and_then(Registration::binary)
4329 .as_deref(),
4330 )
4331 });
4332 if let Some(state) = &binary
4333 && state.is_error()
4334 {
4335 problems.push(StatusProblem {
4336 subject: "binary",
4337 detail: state.to_string(),
4338 });
4339 }
4340
4341 match (&record, ®istration) {
4342 (Some(_), None) => problems.push(StatusProblem {
4343 subject: "registration",
4344 detail: "this host has a service record but no service manager knows the \
4345 registration. Run `service install` again; it deletes nothing."
4346 .to_string(),
4347 }),
4348 (None, Some(found)) if record_refused.is_none() => problems.push(StatusProblem {
4353 subject: "record",
4354 detail: format!(
4355 "{} holds a registration for this service but there is no install record, so \
4356 the path it was installed from and the directories it was installed against \
4357 are unknown. Run `service uninstall` and `service install`.",
4358 found.manager
4359 ),
4360 }),
4361 (None, Some(_)) => {}
4362 (Some(record), Some(found)) => {
4363 if record.start_mode != found.start_mode {
4364 problems.push(StatusProblem {
4365 subject: "start mode",
4366 detail: format!(
4367 "the record says {} and {} holds a {} registration. Switch the start \
4368 mode again to make them agree.",
4369 record.start_mode, found.manager, found.start_mode
4370 ),
4371 });
4372 }
4373 if record.start_mode == StartMode::Boot && !found.starts_automatically {
4374 problems.push(StatusProblem {
4375 subject: "start mode",
4376 detail: format!(
4377 "{} holds the registration but will not start it by itself, so this \
4378 host does not resume work after a reboot.",
4379 found.manager
4380 ),
4381 });
4382 }
4383 if let Some(actual) = found.restart_delay {
4384 let expected = record.restart().effective_delay(found.manager);
4385 if actual != expected {
4386 problems.push(StatusProblem {
4387 subject: "restart policy",
4388 detail: format!(
4389 "the record says the service restarts after {}s and {} reports \
4390 {}s. Something has edited the registration since it was \
4391 installed.",
4392 expected.as_secs(),
4393 found.manager,
4394 actual.as_secs()
4395 ),
4396 });
4397 } else if expected != record.restart().delay() {
4398 notes.push(format!(
4403 "{} expresses the restart delay in whole minutes, so the {}s asked \
4404 for is enforced as {}s. The service therefore never restarts faster \
4405 than the configured bound.",
4406 found.manager,
4407 record.restart().delay().as_secs(),
4408 expected.as_secs()
4409 ));
4410 }
4411 }
4412 if found.starts_automatically && !found.running {
4413 problems.push(StatusProblem {
4414 subject: "runtime",
4415 detail: format!(
4416 "{} holds an automatic registration, but the daemon is stopped. Run \
4417 `runner-manager service install --start-at {}`; if the service \
4418 manager denies replacement, retry from an elevated terminal. If it \
4419 stops again, inspect {}.",
4420 found.manager,
4421 found.start_mode,
4422 log_file.display()
4423 ),
4424 });
4425 }
4426 }
4427 (None, None) => {}
4428 }
4429
4430 let store = record.as_ref().and_then(|record| {
4435 let registered_mode = registration
4436 .as_ref()
4437 .map_or(record.start_mode, |found| found.start_mode);
4438 crate::secrets::PlatformSecretStore::for_start_mode(record.start_mode)
4439 .ok()
4440 .map(|store| crate::secrets::ActiveStore::of(&store, registered_mode))
4441 });
4442 if let Some(store) = &store
4443 && !store.agrees_with_start_mode()
4444 {
4445 problems.push(StatusProblem {
4446 subject: "secret store",
4447 detail: format!(
4448 "{store}. Run `auth login` again so the token is stored where the registered \
4449 start mode can read it."
4450 ),
4451 });
4452 }
4453
4454 if record.as_ref().map(|record| record.start_mode) == Some(StartMode::Login) {
4455 notes.push(
4456 "This registration starts at login, so the agent does not run until the operator \
4457 signs in; this host does not resume work after an unattended reboot."
4458 .to_string(),
4459 );
4460 }
4461 if last_github_contact.is_none() {
4462 notes.push(
4463 "GitHub has not been reached successfully since this host's state directory was \
4464 created."
4465 .to_string(),
4466 );
4467 }
4468
4469 let runner_root = crate::runner_root::default_runner_root(paths)
4476 .ok()
4477 .map(|root| {
4478 let path = root.as_path().to_path_buf();
4479 let report = crate::runner_root_access::report(&path);
4480 (path, report)
4481 });
4482 if let Some((
4492 path,
4493 RootAccessReport::Present {
4494 broad_write: true, ..
4495 },
4496 )) = &runner_root
4497 {
4498 notes.push(format!(
4499 "the platform default runner root {} can be written by ordinary local users, so \
4500 it is not a safe place to run jobs. `service install` refuses it rather than \
4501 tightening it, because the contents of a directory anybody could write cannot \
4502 be trusted: remove or empty it, or choose another root with `runner-manager \
4503 host set-runtime-root --path <PATH>`.",
4504 path.display()
4505 ));
4506 }
4507
4508 match runner_root_refusals(paths) {
4531 Ok(refusals) => {
4532 for refusal in refusals {
4533 notes.push(format!(
4534 "policy {} started no runner: its runner root {} refused the launch \
4535 ({}), last at {}. {} This clears when that policy next places a \
4536 runner.",
4537 refusal.policy,
4538 refusal.root,
4539 refusal.kind,
4540 refusal.at.to_rfc3339(),
4541 refusal.detail,
4542 ));
4543 }
4544 }
4545 Err(error) => notes.push(format!(
4546 "whether the agent could use its runner roots could not be read: {error}"
4547 )),
4548 }
4549
4550 Self {
4551 identity,
4552 record,
4553 registration,
4554 binary,
4555 log_file,
4556 store,
4557 last_github_contact,
4558 runner_root,
4559 problems,
4560 notes,
4561 }
4562 }
4563
4564 #[must_use]
4572 pub fn runner_root(&self) -> Option<(&Path, &RootAccessReport)> {
4573 self.runner_root
4574 .as_ref()
4575 .map(|(path, report)| (path.as_path(), report))
4576 }
4577
4578 #[must_use]
4580 pub const fn is_installed(&self) -> bool {
4581 self.registration.is_some() || self.record.is_some()
4582 }
4583
4584 #[must_use]
4586 pub fn is_running(&self) -> bool {
4587 self.registration
4588 .as_ref()
4589 .is_some_and(|registration| registration.running)
4590 }
4591
4592 #[must_use]
4594 pub fn is_healthy(&self) -> bool {
4595 self.problems.is_empty()
4596 }
4597
4598 #[must_use]
4600 pub fn problems(&self) -> &[StatusProblem] {
4601 &self.problems
4602 }
4603
4604 #[must_use]
4607 pub fn notes(&self) -> &[String] {
4608 &self.notes
4609 }
4610
4611 #[must_use]
4613 pub fn start_mode(&self) -> Option<StartMode> {
4614 self.record.as_ref().map(|record| record.start_mode)
4615 }
4616
4617 #[must_use]
4620 pub const fn binary(&self) -> Option<&BinaryPath> {
4621 self.binary.as_ref()
4622 }
4623
4624 #[must_use]
4626 pub fn log_file(&self) -> &Path {
4627 &self.log_file
4628 }
4629
4630 #[must_use]
4632 pub const fn last_github_contact(&self) -> Option<DateTime<Utc>> {
4633 self.last_github_contact
4634 }
4635
4636 #[must_use]
4639 pub const fn secret_store(&self) -> Option<&crate::secrets::ActiveStore> {
4640 self.store.as_ref()
4641 }
4642
4643 #[must_use]
4645 pub const fn record(&self) -> Option<&InstallRecord> {
4646 self.record.as_ref()
4647 }
4648
4649 #[must_use]
4651 pub const fn registration(&self) -> Option<&Registration> {
4652 self.registration.as_ref()
4653 }
4654}
4655
4656impl fmt::Display for ServiceStatus {
4657 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
4658 writeln!(f, "Service: {}", self.identity)?;
4659 match (&self.record, &self.registration) {
4660 (None, None) => {
4661 writeln!(
4662 f,
4663 " installed no. `service install` registers `{} {}`.",
4664 SERVICE_NAME,
4665 DAEMON_ARGUMENTS.join(" ")
4666 )?;
4667 }
4668 _ => {
4669 let manager = self
4670 .registration
4671 .as_ref()
4672 .map(|registration| registration.manager.manager());
4673 writeln!(
4674 f,
4675 " installed {}",
4676 manager.unwrap_or("yes, but no service manager knows it")
4677 )?;
4678 writeln!(
4679 f,
4680 " state {}",
4681 if self.is_running() {
4682 "running"
4683 } else {
4684 "not running"
4685 }
4686 )?;
4687 }
4688 }
4689 if let Some(record) = &self.record {
4690 writeln!(f, " start mode {}", record.start_mode)?;
4691 writeln!(f, " account {}", record.account)?;
4692 writeln!(f, " restart on failure {}", record.restart())?;
4693 writeln!(
4694 f,
4695 " arguments {}",
4696 record.arguments.join(" ")
4697 )?;
4698 }
4699 if let Some(binary) = &self.binary {
4700 writeln!(f, " binary {binary}")?;
4701 }
4702 writeln!(f, " diagnostic log {}", self.log_file.display())?;
4703 if let Some((path, report)) = &self.runner_root {
4704 writeln!(f, " default runner root {}", path.display())?;
4711 if *report != RootAccessReport::NotApplicable {
4712 writeln!(f, " default root access {report}")?;
4713 }
4714 }
4715 if let Some(store) = &self.store {
4716 writeln!(f, " secret store {store}")?;
4717 }
4718 writeln!(
4719 f,
4720 " last GitHub contact {}",
4721 match self.last_github_contact {
4722 Some(at) => at.to_rfc3339(),
4723 None => "never".to_string(),
4724 }
4725 )?;
4726 for note in &self.notes {
4727 writeln!(f, " note {note}")?;
4728 }
4729 for problem in &self.problems {
4730 writeln!(f, " ERROR {problem}")?;
4731 }
4732 write!(
4733 f,
4734 " verdict {}",
4735 if self.is_healthy() {
4736 "healthy"
4737 } else {
4738 "NOT healthy"
4739 }
4740 )
4741 }
4742}
4743
4744#[derive(Debug, Clone, Default)]
4761pub struct RecordingControls {
4762 state: std::sync::Arc<std::sync::Mutex<RecordingState>>,
4763}
4764
4765#[derive(Debug, Default)]
4766struct RecordingState {
4767 registrations: BTreeMap<(StartMode, String), Registration>,
4768 definitions: BTreeMap<String, ServiceDefinition>,
4769 calls: Vec<String>,
4770 #[cfg(test)]
4771 install_failures: BTreeMap<StartMode, String>,
4772 #[cfg(test)]
4773 after_install: BTreeMap<StartMode, TestInstallSideEffect>,
4774}
4775
4776#[cfg(test)]
4777#[derive(Debug, Clone)]
4778enum TestInstallSideEffect {
4779 HideDirectory { directory: PathBuf, hidden: PathBuf },
4780}
4781
4782impl RecordingControls {
4783 #[must_use]
4785 pub fn new() -> Self {
4786 Self::default()
4787 }
4788
4789 #[must_use]
4792 pub fn calls(&self) -> Vec<String> {
4793 self.state.lock().expect("not poisoned").calls.clone()
4794 }
4795
4796 #[must_use]
4798 pub fn registrations(&self) -> Vec<(StartMode, String, Registration)> {
4799 self.state
4800 .lock()
4801 .expect("not poisoned")
4802 .registrations
4803 .iter()
4804 .map(|((mode, name), registration)| (*mode, name.clone(), registration.clone()))
4805 .collect()
4806 }
4807
4808 #[must_use]
4810 pub fn definition(&self, name: &str) -> Option<ServiceDefinition> {
4811 self.state
4812 .lock()
4813 .expect("not poisoned")
4814 .definitions
4815 .get(name)
4816 .cloned()
4817 }
4818
4819 pub fn edit(&self, name: &str, edit: impl FnOnce(&mut Registration)) {
4829 let mut state = self.state.lock().expect("not poisoned");
4830 if let Some((_, registration)) = state
4831 .registrations
4832 .iter_mut()
4833 .find(|((_, held), _)| held == name)
4834 {
4835 edit(registration);
4836 }
4837 }
4838
4839 #[cfg(test)]
4840 fn fail_next_install(&self, mode: StartMode, detail: &str) {
4841 self.state
4842 .lock()
4843 .expect("not poisoned")
4844 .install_failures
4845 .insert(mode, detail.to_string());
4846 }
4847
4848 #[cfg(test)]
4849 fn hide_directory_after_install(&self, mode: StartMode, directory: PathBuf, hidden: PathBuf) {
4850 self.state
4851 .lock()
4852 .expect("not poisoned")
4853 .after_install
4854 .insert(
4855 mode,
4856 TestInstallSideEffect::HideDirectory { directory, hidden },
4857 );
4858 }
4859}
4860
4861impl ControlFactory for RecordingControls {
4862 fn control(&self, mode: StartMode) -> Result<Box<dyn ServiceControl>, ServiceError> {
4863 Ok(Box::new(RecordingControl {
4864 mode,
4865 state: std::sync::Arc::clone(&self.state),
4866 }))
4867 }
4868}
4869
4870#[derive(Debug)]
4871struct RecordingControl {
4872 mode: StartMode,
4873 state: std::sync::Arc<std::sync::Mutex<RecordingState>>,
4874}
4875
4876impl RecordingControl {
4877 fn note(&self, operation: &str, name: &str) {
4878 self.state
4879 .lock()
4880 .expect("not poisoned")
4881 .calls
4882 .push(format!("{operation} {name} ({})", self.mode));
4883 }
4884}
4885
4886impl ServiceControl for RecordingControl {
4887 fn manager(&self) -> DefinitionKind {
4888 host_definition_kind(self.mode)
4892 }
4893
4894 fn install(&self, plan: &InstallPlan) -> Result<ServiceDefinition, ServiceError> {
4895 self.note("install", plan.identity().name());
4896 #[cfg(test)]
4897 if let Some(detail) = self
4898 .state
4899 .lock()
4900 .expect("not poisoned")
4901 .install_failures
4902 .remove(&self.mode)
4903 {
4904 return Err(ServiceError::Control {
4905 operation: "install",
4906 name: plan.identity().name().to_string(),
4907 manager: "recording control",
4908 detail,
4909 });
4910 }
4911 let definition = ServiceDefinition::for_host(plan)?;
4916 let mut state = self.state.lock().expect("not poisoned");
4917 state.registrations.insert(
4918 (self.mode, plan.identity().name().to_string()),
4919 Registration {
4920 manager: host_definition_kind(self.mode),
4921 start_mode: self.mode,
4922 command_line: plan.command_line(),
4923 account: Some(plan.account().as_str().to_string()),
4924 running: false,
4925 starts_automatically: true,
4926 restart_delay: Some(plan.restart().delay()),
4927 },
4928 );
4929 state
4930 .definitions
4931 .insert(plan.identity().name().to_string(), definition.clone());
4932 #[cfg(test)]
4933 let side_effect = state.after_install.remove(&self.mode);
4934 drop(state);
4935 #[cfg(test)]
4936 if let Some(TestInstallSideEffect::HideDirectory { directory, hidden }) = side_effect {
4937 std::fs::rename(&directory, &hidden).expect("test fault can hide the record directory");
4938 std::fs::write(&directory, b"blocks recreation")
4939 .expect("test fault can block record directory recreation");
4940 }
4941 Ok(definition)
4942 }
4943
4944 fn uninstall(&self, identity: &ServiceIdentity) -> Result<bool, ServiceError> {
4945 self.note("uninstall", identity.name());
4946 let mut state = self.state.lock().expect("not poisoned");
4947 state.definitions.remove(identity.name());
4948 Ok(state
4949 .registrations
4950 .remove(&(self.mode, identity.name().to_string()))
4951 .is_some())
4952 }
4953
4954 fn query(&self, identity: &ServiceIdentity) -> Result<Option<Registration>, ServiceError> {
4955 self.note("query", identity.name());
4956 Ok(self
4957 .state
4958 .lock()
4959 .expect("not poisoned")
4960 .registrations
4961 .get(&(self.mode, identity.name().to_string()))
4962 .cloned())
4963 }
4964
4965 fn start(&self, identity: &ServiceIdentity) -> Result<(), ServiceError> {
4966 self.note("start", identity.name());
4967 let mut state = self.state.lock().expect("not poisoned");
4968 match state
4969 .registrations
4970 .get_mut(&(self.mode, identity.name().to_string()))
4971 {
4972 Some(registration) => {
4973 registration.running = true;
4974 Ok(())
4975 }
4976 None => Err(ServiceError::NotInstalled {
4977 name: identity.name().to_string(),
4978 operation: "start",
4979 }),
4980 }
4981 }
4982
4983 fn stop(&self, identity: &ServiceIdentity) -> Result<bool, ServiceError> {
4984 self.note("stop", identity.name());
4985 let mut state = self.state.lock().expect("not poisoned");
4986 match state
4987 .registrations
4988 .get_mut(&(self.mode, identity.name().to_string()))
4989 {
4990 Some(registration) => Ok(std::mem::replace(&mut registration.running, false)),
4991 None => Err(ServiceError::NotInstalled {
4992 name: identity.name().to_string(),
4993 operation: "stop",
4994 }),
4995 }
4996 }
4997}
4998
4999impl ControlFactory for HostControls {
5000 fn control(&self, mode: StartMode) -> Result<Box<dyn ServiceControl>, ServiceError> {
5001 sys::control(mode)
5002 }
5003}
5004
5005fn host_home() -> Option<PathBuf> {
5012 directories::BaseDirs::new().map(|dirs| dirs.home_dir().to_path_buf())
5013}
5014
5015#[cfg(unix)]
5017fn home_directory() -> Option<PathBuf> {
5018 host_home()
5019}
5020
5021fn run(program: &str, arguments: &[&std::ffi::OsStr]) -> std::io::Result<(bool, String, String)> {
5030 let output = std::process::Command::new(program)
5031 .args(arguments)
5032 .output()?;
5033 Ok((
5034 output.status.success(),
5035 String::from_utf8_lossy(&output.stdout).into_owned(),
5036 String::from_utf8_lossy(&output.stderr).into_owned(),
5037 ))
5038}
5039
5040#[must_use]
5043pub const fn host_definition_kind(mode: StartMode) -> DefinitionKind {
5044 if cfg!(windows) {
5045 match mode {
5046 StartMode::Boot => DefinitionKind::WindowsService,
5047 StartMode::Login => DefinitionKind::WindowsScheduledTask,
5048 }
5049 } else if cfg!(target_os = "macos") {
5050 DefinitionKind::LaunchdPlist
5051 } else {
5052 DefinitionKind::SystemdUnit
5053 }
5054}
5055
5056#[cfg(any(target_os = "macos", test))]
5063fn enable_launchd_registration(
5064 mut launchctl: impl FnMut(&[&std::ffi::OsStr]) -> (bool, String),
5065 domain: &str,
5066 service_target: &str,
5067 plist: &Path,
5068 name: &str,
5069 elevation_remedy: &'static str,
5070) -> Result<(), ServiceError> {
5071 let (enabled, cause) = launchctl(&[
5072 std::ffi::OsStr::new("enable"),
5073 std::ffi::OsStr::new(service_target),
5074 ]);
5075 if enabled {
5076 return Ok(());
5077 }
5078
5079 let (booted_out, bootout_detail) = launchctl(&[
5080 std::ffi::OsStr::new("bootout"),
5081 std::ffi::OsStr::new(service_target),
5082 ]);
5083 let removed = std::fs::remove_file(plist);
5084 if !booted_out || removed.is_err() {
5085 return Err(ServiceError::Rollback {
5086 operation: "enable launchd registration",
5087 name: name.to_string(),
5088 cause,
5089 rollback: format!(
5090 "launchctl bootout {domain}: {}; remove {}: {}",
5091 if booted_out {
5092 "succeeded".to_string()
5093 } else {
5094 bootout_detail
5095 },
5096 plist.display(),
5097 removed
5098 .err()
5099 .map_or_else(|| "succeeded".to_string(), |error| error.to_string())
5100 ),
5101 });
5102 }
5103
5104 if cause.to_ascii_lowercase().contains("permission denied") {
5105 Err(ServiceError::NeedsElevation {
5106 operation: "enable",
5107 name: name.to_string(),
5108 detail: cause,
5109 remedy: elevation_remedy,
5110 })
5111 } else {
5112 Err(ServiceError::Control {
5113 operation: "enable",
5114 name: name.to_string(),
5115 manager: "launchd",
5116 detail: cause,
5117 })
5118 }
5119}
5120
5121#[derive(Debug, Clone)]
5130pub struct ServiceShutdown(tokio::sync::watch::Receiver<bool>);
5131
5132impl ServiceShutdown {
5133 pub async fn wait(mut self) {
5135 if !*self.0.borrow() {
5136 let _ = self.0.changed().await;
5137 }
5138 }
5139}
5140
5141#[cfg(windows)]
5148pub fn run_windows_service_host<F>(run: F) -> Result<u8, ServiceError>
5149where
5150 F: FnOnce(ServiceShutdown) -> u8 + Send + 'static,
5151{
5152 windows_host::run(Box::new(run))
5153}
5154
5155#[cfg(windows)]
5156mod windows_host {
5157 use std::ffi::OsString;
5158 use std::sync::{Arc, Mutex, OnceLock, mpsc};
5159 use std::time::Duration;
5160
5161 use windows_service::service::{
5162 ServiceControl, ServiceControlAccept, ServiceExitCode, ServiceState, ServiceStatus,
5163 ServiceType,
5164 };
5165 use windows_service::service_control_handler::{
5166 self, ServiceControlHandlerResult, ServiceStatusHandle,
5167 };
5168
5169 use super::{SERVICE_NAME, ServiceError, ServiceShutdown};
5170
5171 type Runner = Box<dyn FnOnce(ServiceShutdown) -> u8 + Send>;
5172
5173 struct Invocation {
5174 run: Runner,
5175 result: mpsc::SyncSender<Result<u8, String>>,
5176 }
5177
5178 static INVOCATION: OnceLock<Mutex<Option<Invocation>>> = OnceLock::new();
5179
5180 windows_service::define_windows_service!(ffi_service_main, service_main);
5181
5182 pub(super) fn run(run: Runner) -> Result<u8, ServiceError> {
5183 let (result_tx, result_rx) = mpsc::sync_channel(1);
5184 let slot = INVOCATION.get_or_init(|| Mutex::new(None));
5185 let mut invocation = slot
5186 .lock()
5187 .map_err(|_| host_error("prepare", "the service-host slot is poisoned"))?;
5188 if invocation.is_some() {
5189 return Err(host_error(
5190 "prepare",
5191 "the service-host slot was already used",
5192 ));
5193 }
5194 *invocation = Some(Invocation {
5195 run,
5196 result: result_tx,
5197 });
5198 drop(invocation);
5199
5200 windows_service::service_dispatcher::start("", ffi_service_main)
5201 .map_err(|error| host_error("connect", &error.to_string()))?;
5202 result_rx
5203 .recv()
5204 .map_err(|error| host_error("finish", &error.to_string()))?
5205 .map_err(|detail| host_error("run", &detail))
5206 }
5207
5208 fn service_main(_arguments: Vec<OsString>) {
5209 let Some(invocation) = INVOCATION.get().and_then(|slot| slot.lock().ok()?.take()) else {
5210 return;
5211 };
5212 let result = run_service(invocation.run);
5213 let _ = invocation.result.send(result);
5214 }
5215
5216 fn run_service(run: Runner) -> Result<u8, String> {
5217 let (stop_tx, stop_rx) = tokio::sync::watch::channel(false);
5218 let status: Arc<Mutex<Option<ServiceStatusHandle>>> = Arc::new(Mutex::new(None));
5219 let handler_status = Arc::clone(&status);
5220 let handler = move |control| match control {
5221 ServiceControl::Interrogate => ServiceControlHandlerResult::NoError,
5222 ServiceControl::Stop | ServiceControl::Shutdown => {
5223 if let Some(handle) = handler_status.lock().ok().and_then(|guard| *guard) {
5224 let _ = handle.set_service_status(service_status(
5225 ServiceState::StopPending,
5226 ServiceControlAccept::empty(),
5227 1,
5228 Duration::from_secs(300),
5229 0,
5230 ));
5231 }
5232 let _ = stop_tx.send(true);
5233 ServiceControlHandlerResult::NoError
5234 }
5235 _ => ServiceControlHandlerResult::NotImplemented,
5236 };
5237 let handle = service_control_handler::register("", handler)
5238 .map_err(|error| format!("cannot register the service control handler: {error}"))?;
5239 *status
5240 .lock()
5241 .map_err(|_| "the service status handle is poisoned".to_string())? = Some(handle);
5242 handle
5243 .set_service_status(service_status(
5244 ServiceState::Running,
5245 ServiceControlAccept::STOP | ServiceControlAccept::SHUTDOWN,
5246 0,
5247 Duration::default(),
5248 0,
5249 ))
5250 .map_err(|error| format!("cannot report SERVICE_RUNNING: {error}"))?;
5251
5252 let exit = run(ServiceShutdown(stop_rx));
5253 handle
5254 .set_service_status(service_status(
5255 ServiceState::Stopped,
5256 ServiceControlAccept::empty(),
5257 0,
5258 Duration::default(),
5259 u32::from(exit),
5260 ))
5261 .map_err(|error| format!("cannot report SERVICE_STOPPED: {error}"))?;
5262 Ok(exit)
5263 }
5264
5265 fn service_status(
5266 state: ServiceState,
5267 accepted: ServiceControlAccept,
5268 checkpoint: u32,
5269 wait_hint: Duration,
5270 exit: u32,
5271 ) -> ServiceStatus {
5272 ServiceStatus {
5273 service_type: ServiceType::OWN_PROCESS,
5274 current_state: state,
5275 controls_accepted: accepted,
5276 exit_code: if exit == 0 {
5277 ServiceExitCode::Win32(0)
5278 } else {
5279 ServiceExitCode::ServiceSpecific(exit)
5280 },
5281 checkpoint,
5282 wait_hint,
5283 process_id: None,
5284 }
5285 }
5286
5287 fn host_error(operation: &'static str, detail: &str) -> ServiceError {
5288 ServiceError::Control {
5289 operation,
5290 name: SERVICE_NAME.to_string(),
5291 manager: "the Windows Service Control Manager",
5292 detail: detail.to_string(),
5293 }
5294 }
5295}
5296
5297#[cfg(windows)]
5298mod sys {
5299 use std::ffi::{OsStr, OsString};
5309 use std::time::{Duration, Instant};
5310
5311 use runner_manager_domain::model::StartMode;
5312 use windows_service::service::{
5313 ServiceAccess, ServiceAction, ServiceActionType, ServiceErrorControl,
5314 ServiceFailureActions, ServiceFailureResetPeriod, ServiceInfo, ServiceStartType,
5315 ServiceState, ServiceType,
5316 };
5317 use windows_service::service_manager::{ServiceManager, ServiceManagerAccess};
5318
5319 use super::{
5320 DefinitionKind, InstallPlan, Registration, ServiceControl, ServiceDefinition, ServiceError,
5321 ServiceIdentity, TaskPrincipal, run, windows_service_spec, xml_value,
5322 };
5323
5324 const SERVICE_DOES_NOT_EXIST: i32 = 1060;
5327 const SERVICE_MARKED_FOR_DELETE: i32 = 1072;
5331 const ACCESS_DENIED: i32 = 5;
5333 const DELETE_TIMEOUT: Duration = Duration::from_secs(30);
5334 const DELETE_POLL_INTERVAL: Duration = Duration::from_millis(200);
5335
5336 const ELEVATION_REMEDY: &str = "Run the command from an elevated prompt: right-click Windows Terminal or PowerShell and \
5337 choose \"Run as administrator\".";
5338
5339 pub(super) fn control(mode: StartMode) -> Result<Box<dyn ServiceControl>, ServiceError> {
5340 Ok(match mode {
5341 StartMode::Boot => Box::new(ScmControl),
5342 StartMode::Login => Box::new(TaskControl),
5343 })
5344 }
5345
5346 #[derive(Debug)]
5349 struct ScmControl;
5350
5351 fn scm_error(
5355 operation: &'static str,
5356 name: &str,
5357 error: &windows_service::Error,
5358 ) -> ServiceError {
5359 let raw = match error {
5360 windows_service::Error::Winapi(io) => io.raw_os_error(),
5361 _ => None,
5362 };
5363 let detail = match error {
5364 windows_service::Error::Winapi(io) => io.to_string(),
5365 other => other.to_string(),
5366 };
5367 if raw == Some(ACCESS_DENIED) {
5368 return ServiceError::NeedsElevation {
5369 operation,
5370 name: name.to_string(),
5371 detail,
5372 remedy: ELEVATION_REMEDY,
5373 };
5374 }
5375 ServiceError::Control {
5376 operation,
5377 name: name.to_string(),
5378 manager: "the Windows Service Control Manager",
5379 detail,
5380 }
5381 }
5382
5383 fn open_manager(
5384 access: ServiceManagerAccess,
5385 operation: &'static str,
5386 name: &str,
5387 ) -> Result<ServiceManager, ServiceError> {
5388 ServiceManager::local_computer(None::<&OsStr>, access)
5389 .map_err(|error| scm_error(operation, name, &error))
5390 }
5391
5392 impl ServiceControl for ScmControl {
5393 fn manager(&self) -> DefinitionKind {
5394 DefinitionKind::WindowsService
5395 }
5396
5397 fn install(&self, plan: &InstallPlan) -> Result<ServiceDefinition, ServiceError> {
5398 let spec = windows_service_spec(plan);
5399 let manager = open_manager(
5400 ServiceManagerAccess::CONNECT | ServiceManagerAccess::CREATE_SERVICE,
5401 "install",
5402 &spec.name,
5403 )?;
5404 let info = ServiceInfo {
5405 name: OsString::from(&spec.name),
5406 display_name: OsString::from(&spec.display_name),
5407 service_type: ServiceType::OWN_PROCESS,
5409 start_type: if spec.automatic_start {
5410 ServiceStartType::AutoStart
5411 } else {
5412 ServiceStartType::OnDemand
5413 },
5414 error_control: ServiceErrorControl::Normal,
5415 executable_path: plan.binary().to_path_buf(),
5416 launch_arguments: plan.arguments().to_vec(),
5417 dependencies: Vec::new(),
5418 account_name: spec.account.as_ref().map(OsString::from),
5420 account_password: None,
5423 };
5424 let service = manager
5425 .create_service(
5426 &info,
5427 ServiceAccess::CHANGE_CONFIG
5428 | ServiceAccess::QUERY_CONFIG
5429 | ServiceAccess::QUERY_STATUS
5430 | ServiceAccess::START
5431 | ServiceAccess::STOP
5432 | ServiceAccess::DELETE,
5433 )
5434 .map_err(|error| scm_error("install", &spec.name, &error))?;
5435 service
5436 .set_description(&spec.description)
5437 .map_err(|error| scm_error("describe", &spec.name, &error))?;
5438 service
5439 .update_failure_actions(ServiceFailureActions {
5440 reset_period: ServiceFailureResetPeriod::After(spec.restart.reset_after()),
5441 reboot_msg: None,
5442 command: None,
5443 actions: Some(vec![
5450 ServiceAction {
5451 action_type: ServiceActionType::Restart,
5452 delay: spec.restart.delay(),
5453 };
5454 3
5455 ]),
5456 })
5457 .map_err(|error| scm_error("set the restart policy of", &spec.name, &error))?;
5458 service
5461 .set_failure_actions_on_non_crash_failures(true)
5462 .map_err(|error| scm_error("set the restart policy of", &spec.name, &error))?;
5463 Ok(ServiceDefinition::windows_service(plan))
5464 }
5465
5466 fn uninstall(&self, identity: &ServiceIdentity) -> Result<bool, ServiceError> {
5467 let manager =
5468 open_manager(ServiceManagerAccess::CONNECT, "uninstall", identity.name())?;
5469 let service = match manager.open_service(
5470 identity.name(),
5471 ServiceAccess::QUERY_STATUS | ServiceAccess::STOP | ServiceAccess::DELETE,
5472 ) {
5473 Ok(service) => service,
5474 Err(error) if is_missing(&error) => return Ok(false),
5475 Err(error) => return Err(scm_error("uninstall", identity.name(), &error)),
5476 };
5477 if let Ok(status) = service.query_status()
5481 && status.current_state != ServiceState::Stopped
5482 {
5483 let _ = service.stop();
5484 }
5485 service
5486 .delete()
5487 .map_err(|error| scm_error("uninstall", identity.name(), &error))?;
5488
5489 drop(service);
5496 let absent = wait_until_scm_absent(DELETE_TIMEOUT, DELETE_POLL_INTERVAL, || {
5497 match manager.open_service(identity.name(), ServiceAccess::QUERY_STATUS) {
5498 Ok(service) => {
5499 drop(service);
5500 Ok(false)
5501 }
5502 Err(error) if is_missing(&error) => Ok(true),
5503 Err(error) if is_marked_for_delete(&error) => Ok(false),
5504 Err(error) => Err(scm_error("verify uninstall of", identity.name(), &error)),
5505 }
5506 })?;
5507 if !absent {
5508 return Err(ServiceError::Control {
5509 operation: "verify uninstall of",
5510 name: identity.name().to_string(),
5511 manager: "the Windows Service Control Manager",
5512 detail: format!(
5513 "the registration was still visible {} seconds after DeleteService; \
5514 retry `service uninstall` from an elevated prompt",
5515 DELETE_TIMEOUT.as_secs()
5516 ),
5517 });
5518 }
5519 Ok(true)
5520 }
5521
5522 fn query(&self, identity: &ServiceIdentity) -> Result<Option<Registration>, ServiceError> {
5523 let manager = open_manager(ServiceManagerAccess::CONNECT, "inspect", identity.name())?;
5524 let service = match manager.open_service(
5525 identity.name(),
5526 ServiceAccess::QUERY_CONFIG | ServiceAccess::QUERY_STATUS,
5527 ) {
5528 Ok(service) => service,
5529 Err(error) if is_missing(&error) => return Ok(None),
5530 Err(error) => return Err(scm_error("inspect", identity.name(), &error)),
5531 };
5532 let config = service
5533 .query_config()
5534 .map_err(|error| scm_error("inspect", identity.name(), &error))?;
5535 let status = service
5536 .query_status()
5537 .map_err(|error| scm_error("inspect", identity.name(), &error))?;
5538 let restart_delay = service.get_failure_actions().ok().and_then(|actions| {
5539 actions
5540 .actions
5541 .and_then(|actions| actions.into_iter().next())
5542 .filter(|action| action.action_type == ServiceActionType::Restart)
5543 .map(|action| action.delay)
5544 });
5545 Ok(Some(Registration {
5546 manager: DefinitionKind::WindowsService,
5547 start_mode: StartMode::Boot,
5548 command_line: config.executable_path.to_string_lossy().into_owned(),
5552 account: config
5553 .account_name
5554 .map(|account| account.to_string_lossy().into_owned()),
5555 running: status.current_state == ServiceState::Running,
5556 starts_automatically: config.start_type == ServiceStartType::AutoStart,
5557 restart_delay,
5558 }))
5559 }
5560
5561 fn start(&self, identity: &ServiceIdentity) -> Result<(), ServiceError> {
5562 let manager = open_manager(ServiceManagerAccess::CONNECT, "start", identity.name())?;
5563 let service = manager
5564 .open_service(identity.name(), ServiceAccess::START)
5565 .map_err(|error| scm_error("start", identity.name(), &error))?;
5566 service
5567 .start::<&OsStr>(&[])
5568 .map_err(|error| scm_error("start", identity.name(), &error))
5569 }
5570
5571 fn stop(&self, identity: &ServiceIdentity) -> Result<bool, ServiceError> {
5572 let manager = open_manager(ServiceManagerAccess::CONNECT, "stop", identity.name())?;
5573 let service = manager
5574 .open_service(
5575 identity.name(),
5576 ServiceAccess::STOP | ServiceAccess::QUERY_STATUS,
5577 )
5578 .map_err(|error| scm_error("stop", identity.name(), &error))?;
5579 let status = service
5580 .query_status()
5581 .map_err(|error| scm_error("stop", identity.name(), &error))?;
5582 if status.current_state == ServiceState::Stopped {
5583 return Ok(false);
5584 }
5585 service
5586 .stop()
5587 .map_err(|error| scm_error("stop", identity.name(), &error))?;
5588 Ok(true)
5589 }
5590 }
5591
5592 fn is_missing(error: &windows_service::Error) -> bool {
5593 matches!(error, windows_service::Error::Winapi(io)
5594 if io.raw_os_error() == Some(SERVICE_DOES_NOT_EXIST))
5595 }
5596
5597 fn is_marked_for_delete(error: &windows_service::Error) -> bool {
5598 matches!(error, windows_service::Error::Winapi(io)
5599 if io.raw_os_error() == Some(SERVICE_MARKED_FOR_DELETE))
5600 }
5601
5602 pub(super) fn wait_until_scm_absent(
5603 timeout: Duration,
5604 poll_interval: Duration,
5605 mut probe_absent: impl FnMut() -> Result<bool, ServiceError>,
5606 ) -> Result<bool, ServiceError> {
5607 let deadline = Instant::now() + timeout;
5608 loop {
5609 if probe_absent()? {
5610 return Ok(true);
5611 }
5612 if Instant::now() >= deadline {
5613 return Ok(false);
5614 }
5615 std::thread::sleep(poll_interval);
5616 }
5617 }
5618
5619 #[derive(Debug)]
5622 struct TaskControl;
5623
5624 fn task_error(operation: &'static str, name: &str, detail: String) -> ServiceError {
5625 if detail.to_ascii_lowercase().contains("access is denied") {
5626 return ServiceError::NeedsElevation {
5627 operation,
5628 name: name.to_string(),
5629 detail,
5630 remedy: ELEVATION_REMEDY,
5631 };
5632 }
5633 ServiceError::Control {
5634 operation,
5635 name: name.to_string(),
5636 manager: "Windows Task Scheduler",
5637 detail,
5638 }
5639 }
5640
5641 fn schtasks(
5642 operation: &'static str,
5643 name: &str,
5644 arguments: &[&OsStr],
5645 ) -> Result<(bool, String), ServiceError> {
5646 match run("schtasks.exe", arguments) {
5647 Ok((ok, stdout, stderr)) => Ok((ok, if ok { stdout } else { stderr })),
5648 Err(error) => Err(task_error(operation, name, error.to_string())),
5649 }
5650 }
5651
5652 fn write_utf16(path: &std::path::Path, text: &str) -> std::io::Result<()> {
5655 let mut bytes = vec![0xFF, 0xFE];
5656 for unit in text.encode_utf16() {
5657 bytes.extend_from_slice(&unit.to_le_bytes());
5658 }
5659 std::fs::write(path, bytes)
5660 }
5661
5662 impl ServiceControl for TaskControl {
5663 fn manager(&self) -> DefinitionKind {
5664 DefinitionKind::WindowsScheduledTask
5665 }
5666
5667 fn install(&self, plan: &InstallPlan) -> Result<ServiceDefinition, ServiceError> {
5668 let name = plan.identity().name().to_string();
5669 let principal = TaskPrincipal::current()?;
5670 let definition = ServiceDefinition::windows_scheduled_task(plan, &principal);
5671 let directory = tempfile::tempdir()
5672 .map_err(|error| task_error("install", &name, error.to_string()))?;
5673 let document = directory.path().join("task.xml");
5674 write_utf16(&document, definition.text())
5675 .map_err(|error| task_error("install", &name, error.to_string()))?;
5676 let (ok, message) = schtasks(
5677 "install",
5678 &name,
5679 &[
5680 OsStr::new("/Create"),
5681 OsStr::new("/TN"),
5682 OsStr::new(&name),
5683 OsStr::new("/XML"),
5684 document.as_os_str(),
5685 ],
5686 )?;
5687 if !ok {
5688 return Err(task_error("install", &name, message));
5689 }
5690 Ok(definition)
5691 }
5692
5693 fn uninstall(&self, identity: &ServiceIdentity) -> Result<bool, ServiceError> {
5694 if self.query(identity)?.is_none() {
5695 return Ok(false);
5696 }
5697 let name = identity.name().to_string();
5698 let (ok, message) = schtasks(
5699 "uninstall",
5700 &name,
5701 &[
5702 OsStr::new("/Delete"),
5703 OsStr::new("/TN"),
5704 OsStr::new(&name),
5705 OsStr::new("/F"),
5706 ],
5707 )?;
5708 if !ok {
5709 return Err(task_error("uninstall", &name, message));
5710 }
5711 Ok(true)
5712 }
5713
5714 fn query(&self, identity: &ServiceIdentity) -> Result<Option<Registration>, ServiceError> {
5715 let name = identity.name().to_string();
5716 let (ok, document) = schtasks(
5717 "inspect",
5718 &name,
5719 &[
5720 OsStr::new("/Query"),
5721 OsStr::new("/TN"),
5722 OsStr::new(&name),
5723 OsStr::new("/XML"),
5724 OsStr::new("ONE"),
5725 ],
5726 )?;
5727 if !ok {
5728 return Ok(None);
5734 }
5735 let command = xml_value(&document, "Command").unwrap_or_default();
5736 let arguments = xml_value(&document, "Arguments").unwrap_or_default();
5737 let command_line = if arguments.is_empty() {
5738 super::quote_argument(&command)
5739 } else {
5740 format!("{} {arguments}", super::quote_argument(&command))
5741 };
5742 Ok(Some(Registration {
5743 manager: DefinitionKind::WindowsScheduledTask,
5744 start_mode: StartMode::Login,
5745 command_line,
5746 account: xml_value(&document, "UserId"),
5747 running: task_is_running(&name),
5748 starts_automatically: super::windows_login_task_starts_automatically(&document),
5749 restart_delay: xml_value(&document, "Interval")
5750 .as_deref()
5751 .and_then(parse_iso8601),
5752 }))
5753 }
5754
5755 fn start(&self, identity: &ServiceIdentity) -> Result<(), ServiceError> {
5756 let name = identity.name().to_string();
5757 let (ok, message) = schtasks(
5758 "start",
5759 &name,
5760 &[OsStr::new("/Run"), OsStr::new("/TN"), OsStr::new(&name)],
5761 )?;
5762 if ok {
5763 Ok(())
5764 } else {
5765 Err(task_error("start", &name, message))
5766 }
5767 }
5768
5769 fn stop(&self, identity: &ServiceIdentity) -> Result<bool, ServiceError> {
5770 let running = self
5771 .query(identity)?
5772 .is_some_and(|registration| registration.running);
5773 if !running {
5774 return Ok(false);
5775 }
5776 let name = identity.name().to_string();
5777 let (ok, message) = schtasks(
5778 "stop",
5779 &name,
5780 &[OsStr::new("/End"), OsStr::new("/TN"), OsStr::new(&name)],
5781 )?;
5782 if ok {
5783 Ok(true)
5784 } else {
5785 Err(task_error("stop", &name, message))
5786 }
5787 }
5788 }
5789
5790 fn task_is_running(name: &str) -> bool {
5804 let Ok((true, stdout, _)) = run(
5805 "schtasks.exe",
5806 &[
5807 OsStr::new("/Query"),
5808 OsStr::new("/TN"),
5809 OsStr::new(name),
5810 OsStr::new("/FO"),
5811 OsStr::new("CSV"),
5812 OsStr::new("/NH"),
5813 ],
5814 ) else {
5815 return false;
5816 };
5817 stdout
5818 .lines()
5819 .filter_map(|line| line.rsplit(',').next())
5820 .any(|status| {
5821 status
5822 .trim()
5823 .trim_matches('"')
5824 .eq_ignore_ascii_case("running")
5825 })
5826 }
5827
5828 fn parse_iso8601(value: &str) -> Option<Duration> {
5832 let rest = value.strip_prefix("PT")?;
5833 if let Some(minutes) = rest.strip_suffix('M') {
5834 return minutes
5835 .parse::<u64>()
5836 .ok()
5837 .map(|minutes| Duration::from_secs(minutes * 60));
5838 }
5839 rest.strip_suffix('S')?
5840 .parse::<u64>()
5841 .ok()
5842 .map(Duration::from_secs)
5843 }
5844}
5845
5846#[cfg(unix)]
5858fn write_definition(
5859 operation: &'static str,
5860 name: &str,
5861 path: &Path,
5862 text: &str,
5863 remedy: &'static str,
5864) -> Result<(), ServiceError> {
5865 if let Some(parent) = path.parent()
5866 && let Err(error) = std::fs::create_dir_all(parent)
5867 && error.kind() != std::io::ErrorKind::AlreadyExists
5868 {
5869 return Err(definition_error(operation, name, error, remedy, parent));
5870 }
5871 std::fs::write(path, text)
5872 .map_err(|error| definition_error(operation, name, error, remedy, path))
5873}
5874
5875#[cfg(unix)]
5876fn definition_error(
5877 operation: &'static str,
5878 name: &str,
5879 error: std::io::Error,
5880 remedy: &'static str,
5881 path: &Path,
5882) -> ServiceError {
5883 let detail = format!("{}: {error}", path.display());
5884 if error.kind() == std::io::ErrorKind::PermissionDenied {
5885 ServiceError::NeedsElevation {
5886 operation,
5887 name: name.to_string(),
5888 detail,
5889 remedy,
5890 }
5891 } else {
5892 ServiceError::Control {
5893 operation,
5894 name: name.to_string(),
5895 manager: "the local service manager",
5896 detail,
5897 }
5898 }
5899}
5900
5901#[cfg(unix)]
5903const SUDO_REMEDY: &str = "A boot-start registration is machine-wide, so it needs root: run the same command with \
5904 `sudo`. `service install --start-at login` needs no elevation at all, at the cost of the \
5905 agent not running until you sign in.";
5906
5907#[cfg(target_os = "macos")]
5912mod sys {
5913 use std::ffi::OsStr;
5926 use std::path::PathBuf;
5927 use std::time::Duration;
5928
5929 use runner_manager_domain::model::StartMode;
5930
5931 use super::{
5932 DefinitionKind, InstallPlan, LAUNCH_AGENTS_SUBDIR, LAUNCH_DAEMONS_DIR, Registration,
5933 SUDO_REMEDY, ServiceControl, ServiceDefinition, ServiceError, ServiceIdentity,
5934 enable_launchd_registration, home_directory, plist_string_value, quote_argument, run,
5935 write_definition, xml_value,
5936 };
5937
5938 pub(super) fn control(mode: StartMode) -> Result<Box<dyn ServiceControl>, ServiceError> {
5939 Ok(Box::new(LaunchdControl { mode }))
5940 }
5941
5942 #[derive(Debug)]
5943 struct LaunchdControl {
5944 mode: StartMode,
5945 }
5946
5947 impl LaunchdControl {
5948 fn domain(&self) -> String {
5950 match self.mode {
5951 StartMode::Boot => "system".to_string(),
5952 StartMode::Login => format!("gui/{}", unsafe { libc::getuid() }),
5955 }
5956 }
5957
5958 fn service_target(&self, identity: &ServiceIdentity) -> String {
5959 format!("{}/{}", self.domain(), identity.launchd_label())
5960 }
5961
5962 fn plist_path(&self, identity: &ServiceIdentity) -> Option<PathBuf> {
5965 let file = format!("{}.plist", identity.launchd_label());
5966 match self.mode {
5967 StartMode::Boot => Some(PathBuf::from(LAUNCH_DAEMONS_DIR).join(file)),
5968 StartMode::Login => {
5969 home_directory().map(|home| home.join(LAUNCH_AGENTS_SUBDIR).join(file))
5970 }
5971 }
5972 }
5973
5974 fn failed(&self, operation: &'static str, name: &str, detail: String) -> ServiceError {
5975 ServiceError::Control {
5976 operation,
5977 name: name.to_string(),
5978 manager: "launchd",
5979 detail,
5980 }
5981 }
5982
5983 fn launchctl(&self, arguments: &[&OsStr]) -> (bool, String) {
5984 match run("launchctl", arguments) {
5985 Ok((ok, stdout, stderr)) => (ok, if ok { stdout } else { stderr }),
5986 Err(error) => (false, error.to_string()),
5987 }
5988 }
5989 }
5990
5991 impl ServiceControl for LaunchdControl {
5992 fn manager(&self) -> DefinitionKind {
5993 DefinitionKind::LaunchdPlist
5994 }
5995
5996 fn install(&self, plan: &InstallPlan) -> Result<ServiceDefinition, ServiceError> {
5997 let name = plan.identity().name().to_string();
5998 let definition = ServiceDefinition::launchd(plan, home_directory().as_deref());
5999 let Some(path) = definition.install_path().map(std::path::Path::to_path_buf) else {
6000 return Err(self.failed(
6001 "install",
6002 &name,
6003 "this account has no home directory, so there is nowhere to put a \
6004 LaunchAgent. Use --start-at boot, which installs a LaunchDaemon under \
6005 /Library/LaunchDaemons."
6006 .to_string(),
6007 ));
6008 };
6009 write_definition("install", &name, &path, definition.text(), SUDO_REMEDY)?;
6010 let target = self.domain();
6011 let (ok, message) = self.launchctl(&[
6012 OsStr::new("bootstrap"),
6013 OsStr::new(&target),
6014 path.as_os_str(),
6015 ]);
6016 if !ok {
6017 let _ = std::fs::remove_file(&path);
6021 if message.to_ascii_lowercase().contains("permission denied") {
6022 return Err(ServiceError::NeedsElevation {
6023 operation: "install",
6024 name,
6025 detail: message,
6026 remedy: SUDO_REMEDY,
6027 });
6028 }
6029 return Err(self.failed("install", &name, message));
6030 }
6031 let service_target = self.service_target(plan.identity());
6034 enable_launchd_registration(
6035 |arguments| self.launchctl(arguments),
6036 &target,
6037 &service_target,
6038 &path,
6039 &name,
6040 SUDO_REMEDY,
6041 )?;
6042 Ok(definition)
6043 }
6044
6045 fn uninstall(&self, identity: &ServiceIdentity) -> Result<bool, ServiceError> {
6046 let Some(path) = self.plist_path(identity) else {
6047 return Ok(false);
6048 };
6049 if !path.exists() {
6050 return Ok(false);
6051 }
6052 let target = self.service_target(identity);
6053 let (ok, message) = self.launchctl(&[OsStr::new("bootout"), OsStr::new(&target)]);
6054 if !ok
6055 && !message.to_ascii_lowercase().contains("no such process")
6056 && !message.contains("113")
6057 {
6058 if message.to_ascii_lowercase().contains("permission denied") {
6059 return Err(ServiceError::NeedsElevation {
6060 operation: "uninstall",
6061 name: identity.name().to_string(),
6062 detail: message,
6063 remedy: SUDO_REMEDY,
6064 });
6065 }
6066 return Err(self.failed("uninstall", identity.name(), message));
6067 }
6068 std::fs::remove_file(&path).map_err(|error| {
6071 super::definition_error(
6072 "uninstall",
6073 identity.name(),
6074 error,
6075 SUDO_REMEDY,
6076 path.as_path(),
6077 )
6078 })?;
6079 Ok(true)
6080 }
6081
6082 fn query(&self, identity: &ServiceIdentity) -> Result<Option<Registration>, ServiceError> {
6083 let Some(path) = self.plist_path(identity) else {
6084 return Ok(None);
6085 };
6086 let Ok(document) = std::fs::read_to_string(&path) else {
6087 return Ok(None);
6088 };
6089 let target = self.service_target(identity);
6090 let (loaded, printed) = self.launchctl(&[OsStr::new("print"), OsStr::new(&target)]);
6091 Ok(Some(Registration {
6092 manager: DefinitionKind::LaunchdPlist,
6093 start_mode: self.mode,
6094 command_line: program_arguments(&document),
6095 account: plist_string_value(&document, "UserName")
6096 .or_else(|| Some("the invoking user".to_string())),
6097 running: loaded && printed.contains("state = running"),
6098 starts_automatically: document.contains("<key>RunAtLoad</key>")
6099 && super::plist_bool_value(&document, "RunAtLoad") == Some(true),
6100 restart_delay: xml_value(
6101 super::plist_value_after_key(&document, "ThrottleInterval").unwrap_or(""),
6102 "integer",
6103 )
6104 .and_then(|value| value.parse::<u64>().ok())
6105 .map(Duration::from_secs),
6106 }))
6107 }
6108
6109 fn start(&self, identity: &ServiceIdentity) -> Result<(), ServiceError> {
6110 let target = self.service_target(identity);
6111 let (ok, message) = self.launchctl(&[
6112 OsStr::new("kickstart"),
6113 OsStr::new("-k"),
6114 OsStr::new(&target),
6115 ]);
6116 if ok {
6117 Ok(())
6118 } else {
6119 Err(self.failed("start", identity.name(), message))
6120 }
6121 }
6122
6123 fn stop(&self, identity: &ServiceIdentity) -> Result<bool, ServiceError> {
6124 let running = self
6125 .query(identity)?
6126 .is_some_and(|registration| registration.running);
6127 if !running {
6128 return Ok(false);
6129 }
6130 let target = self.service_target(identity);
6131 let (ok, message) = self.launchctl(&[
6132 OsStr::new("kill"),
6133 OsStr::new("SIGTERM"),
6134 OsStr::new(&target),
6135 ]);
6136 if ok {
6137 Ok(true)
6138 } else {
6139 Err(self.failed("stop", identity.name(), message))
6140 }
6141 }
6142 }
6143
6144 fn program_arguments(document: &str) -> String {
6146 let Some(rest) = super::plist_value_after_key(document, "ProgramArguments") else {
6147 return String::new();
6148 };
6149 let Some(end) = rest.find("</array>") else {
6150 return String::new();
6151 };
6152 let mut out = Vec::new();
6153 let mut cursor = &rest[..end];
6154 while let Some(open) = cursor.find("<string>") {
6155 let after = &cursor[open + "<string>".len()..];
6156 let Some(close) = after.find("</string>") else {
6157 break;
6158 };
6159 out.push(quote_argument(&super::xml_unescape(&after[..close])));
6160 cursor = &after[close..];
6161 }
6162 out.join(" ")
6163 }
6164}
6165
6166#[cfg(all(unix, not(target_os = "macos")))]
6171mod sys {
6172 use std::ffi::OsStr;
6182 use std::path::PathBuf;
6183 use std::time::Duration;
6184
6185 use runner_manager_domain::model::StartMode;
6186
6187 use super::{
6188 DefinitionKind, InstallPlan, Registration, SUDO_REMEDY, SYSTEMD_SYSTEM_DIR,
6189 SYSTEMD_USER_SUBDIR, ServiceControl, ServiceDefinition, ServiceError, ServiceIdentity,
6190 home_directory, ini_directives, run, write_definition,
6191 };
6192
6193 pub(super) fn control(mode: StartMode) -> Result<Box<dyn ServiceControl>, ServiceError> {
6194 Ok(Box::new(SystemdControl { mode }))
6195 }
6196
6197 #[derive(Debug)]
6198 struct SystemdControl {
6199 mode: StartMode,
6200 }
6201
6202 impl SystemdControl {
6203 fn unit_path(&self, identity: &ServiceIdentity) -> Option<PathBuf> {
6204 let file = identity.systemd_unit();
6205 match self.mode {
6206 StartMode::Boot => Some(PathBuf::from(SYSTEMD_SYSTEM_DIR).join(file)),
6207 StartMode::Login => {
6208 home_directory().map(|home| home.join(SYSTEMD_USER_SUBDIR).join(file))
6209 }
6210 }
6211 }
6212
6213 fn systemctl(&self, arguments: &[&str]) -> (bool, String) {
6215 let mut all: Vec<&OsStr> = Vec::with_capacity(arguments.len() + 1);
6216 if self.mode == StartMode::Login {
6217 all.push(OsStr::new("--user"));
6218 }
6219 all.extend(arguments.iter().map(OsStr::new));
6220 match run("systemctl", &all) {
6221 Ok((ok, stdout, stderr)) => (
6222 ok,
6223 if stdout.trim().is_empty() {
6224 stderr
6225 } else {
6226 stdout
6227 },
6228 ),
6229 Err(error) => (false, error.to_string()),
6230 }
6231 }
6232
6233 fn failed(&self, operation: &'static str, name: &str, detail: String) -> ServiceError {
6234 ServiceError::Control {
6235 operation,
6236 name: name.to_string(),
6237 manager: "systemd",
6238 detail,
6239 }
6240 }
6241 }
6242
6243 impl ServiceControl for SystemdControl {
6244 fn manager(&self) -> DefinitionKind {
6245 DefinitionKind::SystemdUnit
6246 }
6247
6248 fn install(&self, plan: &InstallPlan) -> Result<ServiceDefinition, ServiceError> {
6249 let name = plan.identity().name().to_string();
6250 let definition = ServiceDefinition::systemd(plan, home_directory().as_deref());
6251 let Some(path) = definition.install_path().map(std::path::Path::to_path_buf) else {
6252 return Err(self.failed(
6253 "install",
6254 &name,
6255 "this account has no home directory, so there is nowhere to put a systemd \
6256 user unit. Use --start-at boot, which installs a system unit under \
6257 /etc/systemd/system."
6258 .to_string(),
6259 ));
6260 };
6261 write_definition("install", &name, &path, definition.text(), SUDO_REMEDY)?;
6262 let unit = plan.identity().systemd_unit();
6263 let (reloaded, message) = self.systemctl(&["daemon-reload"]);
6264 if !reloaded {
6265 let _ = std::fs::remove_file(&path);
6266 return Err(self.failed("install", &name, message));
6267 }
6268 let (enabled, message) = self.systemctl(&["enable", &unit]);
6269 if !enabled {
6270 let _ = std::fs::remove_file(&path);
6271 let _ = self.systemctl(&["daemon-reload"]);
6272 return Err(self.failed("install", &name, message));
6273 }
6274 Ok(definition)
6275 }
6276
6277 fn uninstall(&self, identity: &ServiceIdentity) -> Result<bool, ServiceError> {
6278 let Some(path) = self.unit_path(identity) else {
6279 return Ok(false);
6280 };
6281 if !path.exists() {
6282 return Ok(false);
6283 }
6284 let unit = identity.systemd_unit();
6285 let _ = self.systemctl(&["disable", "--now", &unit]);
6290 std::fs::remove_file(&path).map_err(|error| {
6291 super::definition_error(
6292 "uninstall",
6293 identity.name(),
6294 error,
6295 SUDO_REMEDY,
6296 path.as_path(),
6297 )
6298 })?;
6299 let _ = self.systemctl(&["daemon-reload"]);
6300 Ok(true)
6301 }
6302
6303 fn query(&self, identity: &ServiceIdentity) -> Result<Option<Registration>, ServiceError> {
6304 let Some(path) = self.unit_path(identity) else {
6305 return Ok(None);
6306 };
6307 let Ok(document) = std::fs::read_to_string(&path) else {
6308 return Ok(None);
6309 };
6310 let unit = identity.systemd_unit();
6311 let directives = ini_directives(&document, "Service");
6312 let (_, active) = self.systemctl(&["is-active", &unit]);
6313 let (_, enabled) = self.systemctl(&["is-enabled", &unit]);
6314 Ok(Some(Registration {
6315 manager: DefinitionKind::SystemdUnit,
6316 start_mode: self.mode,
6317 command_line: directives.get("ExecStart").cloned().unwrap_or_default(),
6318 account: directives.get("User").cloned().or_else(|| {
6319 Some(match self.mode {
6320 StartMode::Boot => "root".to_string(),
6321 StartMode::Login => "the invoking user".to_string(),
6322 })
6323 }),
6324 running: active.trim() == "active",
6325 starts_automatically: enabled.trim() == "enabled",
6326 restart_delay: directives
6327 .get("RestartSec")
6328 .and_then(|value| value.trim().trim_end_matches('s').parse::<u64>().ok())
6329 .map(Duration::from_secs),
6330 }))
6331 }
6332
6333 fn start(&self, identity: &ServiceIdentity) -> Result<(), ServiceError> {
6334 let unit = identity.systemd_unit();
6335 let (ok, message) = self.systemctl(&["start", &unit]);
6336 if ok {
6337 Ok(())
6338 } else {
6339 Err(self.failed("start", identity.name(), message))
6340 }
6341 }
6342
6343 fn stop(&self, identity: &ServiceIdentity) -> Result<bool, ServiceError> {
6344 let running = self
6345 .query(identity)?
6346 .is_some_and(|registration| registration.running);
6347 if !running {
6348 return Ok(false);
6349 }
6350 let unit = identity.systemd_unit();
6351 let (ok, message) = self.systemctl(&["stop", &unit]);
6352 if ok {
6353 Ok(true)
6354 } else {
6355 Err(self.failed("stop", identity.name(), message))
6356 }
6357 }
6358 }
6359}
6360
6361#[cfg(test)]
6362mod tests {
6363 use super::*;
6364
6365 use std::collections::BTreeMap;
6366
6367 #[test]
6368 fn task_scheduler_omitting_default_enabled_still_means_automatic() {
6369 assert!(windows_login_task_starts_automatically(
6370 "<Task><Triggers><LogonTrigger></LogonTrigger></Triggers></Task>"
6371 ));
6372 assert!(windows_login_task_starts_automatically(
6373 "<Task><LogonTrigger><Enabled>true</Enabled></LogonTrigger></Task>"
6374 ));
6375 assert!(!windows_login_task_starts_automatically(
6376 "<Task><LogonTrigger><Enabled>false</Enabled></LogonTrigger></Task>"
6377 ));
6378 assert!(!windows_login_task_starts_automatically(
6379 "<Task><BootTrigger></BootTrigger></Task>"
6380 ));
6381 }
6382
6383 fn linux_plan(mode: StartMode) -> InstallPlan {
6390 InstallPlan::unchecked(
6391 ServiceIdentity::product(),
6392 mode,
6393 "/opt/runner-manager/bin/runner-manager",
6394 ServiceDirectories {
6395 config: PathBuf::from("/var/lib/runner-manager/config"),
6396 state: PathBuf::from("/var/lib/runner-manager/state"),
6397 runtime: PathBuf::from("/var/lib/runner-manager/runtime"),
6398 logs: PathBuf::from("/var/log/runner-manager"),
6399 },
6400 )
6401 .with_secret_guard("/var/lib/runner-manager/secrets/user-access-token")
6402 }
6403
6404 fn windows_plan(mode: StartMode) -> InstallPlan {
6405 InstallPlan::unchecked(
6406 ServiceIdentity::product(),
6407 mode,
6408 "C:\\Program Files\\runner-manager\\runner-manager.exe",
6409 ServiceDirectories {
6410 config: PathBuf::from("C:\\Users\\op\\AppData\\Local\\rm\\config"),
6411 state: PathBuf::from("C:\\Users\\op\\AppData\\Local\\rm\\state"),
6412 runtime: PathBuf::from("C:\\Users\\op\\AppData\\Local\\rm\\runtime"),
6413 logs: PathBuf::from("C:\\Users\\op\\AppData\\Local\\rm\\logs"),
6414 },
6415 )
6416 }
6417
6418 fn edited(text: &str, from: &str, to: &str) -> String {
6426 assert!(
6427 text.contains(from),
6428 "the rendered definition does not contain `{from}`, so this test would assert \
6429 nothing about a widened one"
6430 );
6431 text.replace(from, to)
6432 }
6433
6434 fn snapshot(roots: &[&Path]) -> BTreeMap<PathBuf, Vec<u8>> {
6436 fn walk(directory: &Path, out: &mut BTreeMap<PathBuf, Vec<u8>>) {
6437 let Ok(entries) = std::fs::read_dir(directory) else {
6438 return;
6439 };
6440 for entry in entries.flatten() {
6441 let path = entry.path();
6442 if path.is_dir() {
6443 walk(&path, out);
6444 } else if let Ok(bytes) = std::fs::read(&path) {
6445 out.insert(path, bytes);
6446 }
6447 }
6448 }
6449 let mut out = BTreeMap::new();
6450 for root in roots {
6451 walk(root, &mut out);
6452 }
6453 out
6454 }
6455
6456 struct Host {
6457 _root: tempfile::TempDir,
6458 paths: AppPaths,
6459 binary: PathBuf,
6460 runner_root: LocalAbsolutePath,
6469 controls: RecordingControls,
6470 }
6471
6472 impl Host {
6473 fn new() -> Self {
6474 let root = tempfile::tempdir().expect("a temporary directory");
6475 let paths = AppPaths::rooted_at(root.path());
6476 paths.create_all().expect("the four directories");
6477 let binary = root.path().join(if cfg!(windows) {
6478 "runner-manager.exe"
6479 } else {
6480 "runner-manager"
6481 });
6482 std::fs::write(&binary, b"not a real binary").expect("a stand-in binary");
6483 let runner_root = LocalAbsolutePath::new(
6484 root.path()
6485 .join("runner-root")
6486 .to_str()
6487 .expect("a unicode temporary path"),
6488 )
6489 .expect("a local absolute path");
6490 Self {
6491 _root: root,
6492 paths,
6493 binary,
6494 runner_root,
6495 controls: RecordingControls::new(),
6496 }
6497 }
6498
6499 fn operations(&self) -> ServiceOperations {
6500 ServiceOperations::with_controls(
6501 self.paths.clone(),
6502 ServiceIdentity::product(),
6503 std::sync::Arc::new(self.controls.clone()),
6504 )
6505 .with_runner_root(self.runner_root.clone())
6506 }
6507
6508 fn request(&self, mode: StartMode) -> InstallRequest {
6509 InstallRequest::new(mode).for_binary(&self.binary)
6510 }
6511 }
6512
6513 #[cfg(windows)]
6514 #[test]
6515 fn windows_uninstall_waits_through_the_marked_for_deletion_window() {
6516 let probes = std::cell::Cell::new(0);
6517 let absent =
6518 super::sys::wait_until_scm_absent(Duration::from_secs(1), Duration::ZERO, || {
6519 let next = probes.get() + 1;
6520 probes.set(next);
6521 Ok(next == 3)
6522 })
6523 .expect("the simulated SCM probe succeeds");
6524
6525 assert!(absent);
6526 assert_eq!(
6527 probes.get(),
6528 3,
6529 "uninstall must recheck after transient presence instead of treating it as a leak"
6530 );
6531 }
6532
6533 #[test]
6534 fn launchd_enable_failure_is_returned_and_removes_the_bootstrapped_registration() {
6535 let root = tempfile::tempdir().expect("a temporary directory");
6536 let plist = root.path().join("fixture.plist");
6537 std::fs::write(&plist, b"fixture").expect("a plist fixture");
6538 let calls = std::cell::RefCell::new(Vec::new());
6539
6540 let error = enable_launchd_registration(
6541 |arguments| {
6542 let call = arguments
6543 .iter()
6544 .map(|argument| argument.to_string_lossy().into_owned())
6545 .collect::<Vec<_>>();
6546 let operation = call[0].clone();
6547 calls.borrow_mut().push(call);
6548 if operation == "enable" {
6549 (false, "label remains disabled".to_string())
6550 } else {
6551 (true, String::new())
6552 }
6553 },
6554 "system",
6555 "system/com.openai.runner-manager-selftest",
6556 &plist,
6557 "runner-manager-selftest",
6558 "rerun with administrative rights",
6559 )
6560 .expect_err("enable failure must fail the install");
6561
6562 assert!(
6563 matches!(
6564 error,
6565 ServiceError::Control {
6566 operation: "enable",
6567 ..
6568 }
6569 ),
6570 "{error}"
6571 );
6572 assert_eq!(calls.borrow().len(), 2);
6573 assert_eq!(calls.borrow()[0][0], "enable");
6574 assert_eq!(calls.borrow()[1][0], "bootout");
6575 assert!(!plist.exists(), "rollback must remove the plist");
6576 }
6577
6578 #[test]
6583 fn a_path_with_spaces_survives_a_round_trip_through_a_command_line() {
6584 let plan = windows_plan(StartMode::Boot);
6585 let command_line = plan.command_line();
6586 assert!(
6587 command_line.starts_with('"'),
6588 "a path with a space must be quoted, got {command_line}"
6589 );
6590 assert_eq!(
6591 executable_from_command_line(&command_line).as_deref(),
6592 Some(plan.binary())
6593 );
6594 }
6595
6596 #[test]
6597 fn a_path_without_spaces_is_not_quoted_and_still_reads_back() {
6598 let plan = linux_plan(StartMode::Boot);
6599 let command_line = plan.command_line();
6600 assert!(!command_line.starts_with('"'), "got {command_line}");
6601 assert_eq!(
6602 executable_from_command_line(&command_line).as_deref(),
6603 Some(plan.binary())
6604 );
6605 }
6606
6607 #[test]
6608 fn a_quoted_path_containing_a_quote_reads_back_verbatim() {
6609 let awkward = r#"C:\odd "name"\rm.exe"#;
6612 let quoted = quote_argument(awkward);
6613 assert_eq!(
6614 executable_from_command_line(&format!("{quoted} daemon run"))
6615 .as_deref()
6616 .map(Path::to_string_lossy)
6617 .as_deref(),
6618 Some(awkward)
6619 );
6620 }
6621
6622 #[test]
6623 fn an_empty_command_line_has_no_executable() {
6624 assert_eq!(executable_from_command_line(" "), None);
6625 assert_eq!(executable_from_command_line(""), None);
6626 }
6627
6628 #[test]
6629 fn xml_escaping_round_trips_the_characters_a_path_or_an_account_may_hold() {
6630 let awkward = r#"DOMAIN\R&D <team> "ops""#;
6631 assert_eq!(
6632 xml_escape(awkward),
6633 "DOMAIN\\R&D <team> "ops""
6634 );
6635 assert_eq!(xml_unescape(&xml_escape(awkward)), awkward);
6636 }
6637
6638 #[test]
6643 fn a_restart_delay_under_the_floor_is_refused() {
6644 let error = RestartPolicy::new(Duration::from_millis(500), Duration::from_secs(60))
6645 .expect_err("half a second is under the one-second floor");
6646 assert!(
6647 matches!(error, ServiceError::RestartDelay { .. }),
6648 "{error}"
6649 );
6650 }
6651
6652 #[test]
6653 fn a_restart_delay_over_the_ceiling_is_refused() {
6654 let error = RestartPolicy::new(Duration::from_secs(3600), Duration::from_secs(7200))
6655 .expect_err("an hour is over the five-minute ceiling");
6656 assert!(
6657 matches!(error, ServiceError::RestartDelay { .. }),
6658 "{error}"
6659 );
6660 }
6661
6662 #[test]
6663 fn a_reset_window_no_longer_than_the_delay_is_refused() {
6664 let error = RestartPolicy::new(Duration::from_secs(15), Duration::from_secs(15))
6665 .expect_err("a window equal to the delay can never elapse between restarts");
6666 assert!(
6667 matches!(error, ServiceError::RestartResetWindow { .. }),
6668 "{error}"
6669 );
6670 }
6671
6672 #[test]
6673 fn a_delay_inside_the_bound_is_accepted() {
6674 let policy = RestartPolicy::new(Duration::from_secs(20), Duration::from_secs(300))
6675 .expect("twenty seconds is inside the bound");
6676 assert_eq!(policy.delay(), Duration::from_secs(20));
6677 assert_eq!(policy.reset_after(), Duration::from_secs(300));
6678 }
6679
6680 #[test]
6685 fn a_fixture_identity_can_never_be_the_product_identity() {
6686 let fixture = ServiceIdentity::fixture("abc123");
6687 assert!(fixture.is_fixture());
6688 assert!(!ServiceIdentity::product().is_fixture());
6689 assert_ne!(fixture.name(), ServiceIdentity::product().name());
6690 assert_ne!(
6691 fixture.launchd_label(),
6692 ServiceIdentity::product().launchd_label()
6693 );
6694 assert_ne!(
6695 fixture.systemd_unit(),
6696 ServiceIdentity::product().systemd_unit()
6697 );
6698 }
6699
6700 #[test]
6701 fn a_fixture_tag_is_reduced_to_characters_every_manager_accepts() {
6702 let fixture = ServiceIdentity::fixture("A b/c:\\d");
6703 assert!(
6704 fixture
6705 .name()
6706 .chars()
6707 .all(|c| c.is_ascii_lowercase() || c.is_ascii_digit() || c == '-'),
6708 "got {}",
6709 fixture.name()
6710 );
6711 }
6712
6713 #[test]
6718 fn the_boot_unit_restarts_on_failure_after_the_bounded_delay() {
6719 let unit = systemd_unit(&linux_plan(StartMode::Boot));
6720 assert!(unit.contains("KillMode=process\n"), "{unit}");
6721 assert!(unit.contains("Restart=on-failure\n"), "{unit}");
6722 assert!(unit.contains("RestartSec=15\n"), "{unit}");
6723 assert!(unit.contains("StartLimitIntervalSec=600\n"), "{unit}");
6724 assert!(unit.contains("StartLimitBurst=5\n"), "{unit}");
6725 assert!(unit.contains("WantedBy=multi-user.target\n"), "{unit}");
6726 }
6727
6728 #[test]
6729 fn the_boot_unit_reads_the_live_store_instead_of_a_frozen_systemd_copy() {
6730 let unit = systemd_unit(&linux_plan(StartMode::Boot));
6731 assert!(
6732 !unit.contains("LoadCredential="),
6733 "a startup snapshot would shadow every rotated credential until restart:\n{unit}"
6734 );
6735 }
6736
6737 #[test]
6738 fn a_login_unit_carries_no_machine_credential_and_wants_the_session_target() {
6739 let unit = systemd_unit(&linux_plan(StartMode::Login));
6740 assert!(
6741 !unit.contains("LoadCredential="),
6742 "a user unit must not name a root-owned credential file, got:\n{unit}"
6743 );
6744 assert!(unit.contains("WantedBy=default.target\n"), "{unit}");
6745 }
6746
6747 #[test]
6748 fn the_unit_can_atomically_replace_the_credential_and_write_only_required_directories() {
6749 let plan = linux_plan(StartMode::Boot);
6750 let unit = systemd_unit(&plan);
6751 let directives = ini_directives(&unit, "Service");
6752 let listed = split_quoted(
6753 directives
6754 .get("ReadWritePaths")
6755 .expect("the unit names its writable paths"),
6756 );
6757 assert_eq!(listed.len(), 5, "{listed:?}");
6758 for path in plan.directories().all() {
6759 assert!(
6760 listed.iter().any(|entry| entry == &path.to_string_lossy()),
6761 "{} is missing from {listed:?}",
6762 path.display()
6763 );
6764 }
6765 assert!(
6766 listed
6767 .iter()
6768 .any(|entry| entry == "/var/lib/runner-manager/secrets"),
6769 "atomic credential replacement needs its parent directory in {listed:?}"
6770 );
6771 }
6772
6773 #[test]
6774 fn the_unit_records_the_absolute_binary_path() {
6775 let plan = linux_plan(StartMode::Boot);
6776 let unit = systemd_unit(&plan);
6777 assert!(
6778 unit.contains("ExecStart=/opt/runner-manager/bin/runner-manager daemon run\n"),
6779 "{unit}"
6780 );
6781 }
6782
6783 #[test]
6788 fn the_daemon_restarts_only_after_an_unsuccessful_exit() {
6789 let plist = launchd_plist(&linux_plan(StartMode::Boot));
6790 assert!(
6793 plist.contains(
6794 "<key>KeepAlive</key>\n <dict>\n <key>SuccessfulExit</key>\n <false/>\n"
6795 ),
6796 "{plist}"
6797 );
6798 assert!(
6799 plist.contains("<key>ThrottleInterval</key>\n <integer>15</integer>"),
6800 "{plist}"
6801 );
6802 }
6803
6804 #[test]
6805 fn a_launch_daemon_names_root_and_a_launch_agent_names_nobody() {
6806 let daemon = launchd_plist(&linux_plan(StartMode::Boot));
6807 assert_eq!(
6808 plist_string_value(&daemon, "UserName").as_deref(),
6809 Some("root")
6810 );
6811 assert_eq!(plist_bool_value(&daemon, "SessionCreate"), Some(false));
6812
6813 let agent = launchd_plist(&linux_plan(StartMode::Login));
6814 assert_eq!(
6815 plist_string_value(&agent, "UserName"),
6816 None,
6817 "a LaunchAgent already runs as the operator:\n{agent}"
6818 );
6819 }
6820
6821 #[test]
6822 fn the_plist_records_the_absolute_binary_path_and_the_daemon_arguments() {
6823 let plist = launchd_plist(&linux_plan(StartMode::Boot));
6824 assert!(
6825 plist.contains("<string>/opt/runner-manager/bin/runner-manager</string>"),
6826 "{plist}"
6827 );
6828 assert!(plist.contains("<string>daemon</string>"), "{plist}");
6829 assert!(plist.contains("<string>run</string>"), "{plist}");
6830 }
6831
6832 #[test]
6833 fn the_launchd_label_is_the_product_identity_in_reverse_domain_form() {
6834 assert_eq!(
6835 ServiceIdentity::product().launchd_label(),
6836 "io.github.IvanMurzak.runner-manager"
6837 );
6838 }
6839
6840 #[test]
6845 fn the_task_runs_at_least_privilege_on_a_logon_trigger() {
6846 let xml = windows_scheduled_task_xml(
6847 &windows_plan(StartMode::Login),
6848 &TaskPrincipal::named("HOST\\operator"),
6849 );
6850 assert!(xml.contains("<LogonTrigger>"), "{xml}");
6851 assert!(xml.contains("<RunLevel>LeastPrivilege</RunLevel>"), "{xml}");
6852 assert!(
6853 xml.contains("<LogonType>InteractiveToken</LogonType>"),
6854 "{xml}"
6855 );
6856 assert!(xml.contains("<UserId>HOST\\operator</UserId>"), "{xml}");
6857 assert!(
6858 xml.contains("<Interval>PT1M</Interval>"),
6859 "Task Scheduler takes whole minutes only, and rejects the registration outright \
6860 for anything finer:\n{xml}"
6861 );
6862 }
6863
6864 #[test]
6865 fn task_schedulers_minute_granularity_only_ever_rounds_the_delay_up() {
6866 for (asked, enforced) in [(1u64, 60u64), (15, 60), (60, 60), (61, 120), (300, 300)] {
6871 let policy =
6872 RestartPolicy::new(Duration::from_secs(asked), Duration::from_secs(asked + 600))
6873 .expect("inside the supported range");
6874 assert_eq!(
6875 policy
6876 .effective_delay(DefinitionKind::WindowsScheduledTask)
6877 .as_secs(),
6878 enforced,
6879 "a {asked}s delay must be enforced as {enforced}s"
6880 );
6881 assert!(
6882 policy.effective_delay(DefinitionKind::WindowsScheduledTask) >= policy.delay(),
6883 "rounding must never shorten the bound"
6884 );
6885 }
6886 }
6887
6888 #[test]
6889 fn every_other_manager_enforces_the_delay_exactly_as_configured() {
6890 let policy = RestartPolicy::default();
6891 for kind in [
6892 DefinitionKind::WindowsService,
6893 DefinitionKind::LaunchdPlist,
6894 DefinitionKind::SystemdUnit,
6895 ] {
6896 assert_eq!(
6897 policy.effective_delay(kind),
6898 policy.delay(),
6899 "{kind:?} takes seconds and enforces exactly what it is given"
6900 );
6901 }
6902 }
6903
6904 #[test]
6905 fn a_task_whose_manager_reports_the_rounded_delay_is_not_a_fault() {
6906 let host = Host::new();
6907 let operations = host.operations();
6908 operations
6909 .install(&host.request(StartMode::Login))
6910 .expect("an install at login");
6911 host.controls.edit("runner-manager", |registration| {
6913 registration.manager = DefinitionKind::WindowsScheduledTask;
6914 registration.restart_delay = Some(Duration::from_secs(60));
6915 });
6916
6917 let status = operations.status().expect("a status");
6918 assert!(
6919 status.is_healthy(),
6920 "minute granularity is the manager's, not a mis-registration: {status}"
6921 );
6922 assert!(
6923 status
6924 .notes()
6925 .iter()
6926 .any(|note| note.contains("whole minutes")),
6927 "but the operator must be told why 15 became 60: {status}"
6928 );
6929
6930 host.controls.edit("runner-manager", |registration| {
6933 registration.restart_delay = Some(Duration::from_secs(1));
6934 });
6935 assert!(
6936 !operations.status().expect("a status").is_healthy(),
6937 "a one-second delay is not what any manager was asked for"
6938 );
6939 }
6940
6941 #[test]
6942 fn the_task_records_the_absolute_binary_path_and_the_daemon_arguments() {
6943 let plan = windows_plan(StartMode::Login);
6944 let xml = windows_scheduled_task_xml(&plan, &TaskPrincipal::named("HOST\\operator"));
6945 assert_eq!(
6946 xml_value(&xml, "Command").as_deref(),
6947 Some("C:\\Program Files\\runner-manager\\runner-manager.exe"),
6948 "{xml}"
6949 );
6950 assert_eq!(xml_value(&xml, "Arguments").as_deref(), Some("daemon run"));
6951 }
6952
6953 #[test]
6954 fn an_account_name_holding_xml_punctuation_is_escaped() {
6955 let xml = windows_scheduled_task_xml(
6956 &windows_plan(StartMode::Login),
6957 &TaskPrincipal::named("R&D\\ops"),
6958 );
6959 assert!(xml.contains("<UserId>R&D\\ops</UserId>"), "{xml}");
6960 assert_eq!(xml_value(&xml, "UserId").as_deref(), Some("R&D\\ops"));
6961 }
6962
6963 #[test]
6968 fn the_service_starts_automatically_under_the_account_the_store_admits() {
6969 let text = windows_service_descriptor(&windows_plan(StartMode::Boot));
6970 let directives = ini_directives(&text, "windows-service");
6971 assert_eq!(
6972 directives.get("StartType").map(String::as_str),
6973 Some("AutoStart")
6974 );
6975 assert_eq!(
6976 directives.get("Account").map(String::as_str),
6977 Some("NT AUTHORITY\\SYSTEM")
6978 );
6979 assert_eq!(
6980 directives.get("ServiceType").map(String::as_str),
6981 Some("OWN_PROCESS")
6982 );
6983 assert_eq!(
6984 directives
6985 .get("FailureActionRestartDelaySecs")
6986 .map(String::as_str),
6987 Some("15")
6988 );
6989 assert_eq!(
6990 directives
6991 .get("FailureActionsOnNonCrashFailures")
6992 .map(String::as_str),
6993 Some("true"),
6994 "without this flag a non-zero exit is not a failure the manager restarts"
6995 );
6996 }
6997
6998 #[test]
6999 fn the_service_spec_leaves_the_account_unnamed_so_the_api_means_local_system() {
7000 let spec = windows_service_spec(&windows_plan(StartMode::Boot));
7001 assert_eq!(spec.account, None);
7002 assert!(spec.automatic_start);
7003 assert!(
7004 spec.command_line.contains("daemon run"),
7005 "{}",
7006 spec.command_line
7007 );
7008 }
7009
7010 #[test]
7015 fn each_definition_goes_where_its_platform_expects_it() {
7016 let home = PathBuf::from("/home/op");
7017 assert_eq!(
7018 ServiceDefinition::launchd(&linux_plan(StartMode::Boot), Some(&home)).install_path(),
7019 Some(Path::new(
7020 "/Library/LaunchDaemons/io.github.IvanMurzak.runner-manager.plist"
7021 ))
7022 );
7023 assert_eq!(
7024 ServiceDefinition::launchd(&linux_plan(StartMode::Login), Some(&home)).install_path(),
7025 Some(Path::new(
7026 "/home/op/Library/LaunchAgents/io.github.IvanMurzak.runner-manager.plist"
7027 ))
7028 );
7029 assert_eq!(
7030 ServiceDefinition::systemd(&linux_plan(StartMode::Boot), Some(&home)).install_path(),
7031 Some(Path::new("/etc/systemd/system/runner-manager.service"))
7032 );
7033 assert_eq!(
7034 ServiceDefinition::systemd(&linux_plan(StartMode::Login), Some(&home)).install_path(),
7035 Some(Path::new(
7036 "/home/op/.config/systemd/user/runner-manager.service"
7037 ))
7038 );
7039 assert_eq!(
7040 ServiceDefinition::windows_service(&windows_plan(StartMode::Boot)).install_path(),
7041 None,
7042 "the Service Control Manager has no file"
7043 );
7044 }
7045
7046 #[test]
7047 fn a_login_definition_without_a_home_directory_has_nowhere_to_go() {
7048 assert_eq!(
7049 ServiceDefinition::systemd(&linux_plan(StartMode::Login), None).install_path(),
7050 None
7051 );
7052 assert_eq!(
7053 ServiceDefinition::launchd(&linux_plan(StartMode::Login), None).install_path(),
7054 None
7055 );
7056 }
7057
7058 #[test]
7063 fn the_rendered_definitions_are_all_least_privilege() {
7064 let linux = linux_plan(StartMode::Boot);
7065 let windows = windows_plan(StartMode::Boot);
7066 for (definition, plan) in [
7067 (ServiceDefinition::systemd(&linux, None), &linux),
7068 (ServiceDefinition::launchd(&linux, None), &linux),
7069 (ServiceDefinition::windows_service(&windows), &windows),
7070 ] {
7071 let review = review_least_privilege(&definition, plan);
7072 assert!(
7073 review.is_least_privilege(),
7074 "{:?} should be least privilege, got:\n{review}",
7075 definition.kind()
7076 );
7077 assert!(
7078 !review.controls().is_empty(),
7079 "a review that confirms nothing proves nothing: {:?}",
7080 definition.kind()
7081 );
7082 }
7083 }
7084
7085 #[test]
7086 fn the_rendered_task_is_least_privilege_and_says_what_it_checked() {
7087 let plan = windows_plan(StartMode::Login);
7088 let definition =
7089 ServiceDefinition::windows_scheduled_task(&plan, &TaskPrincipal::named("HOST\\op"));
7090 let review = review_least_privilege(&definition, &plan);
7091 assert!(review.is_least_privilege(), "{review}");
7092 assert!(
7093 review
7094 .controls()
7095 .iter()
7096 .any(|control| control.contains("LeastPrivilege")),
7097 "{review}"
7098 );
7099 }
7100
7101 #[test]
7102 fn a_unit_that_makes_one_more_directory_writable_is_not_least_privilege() {
7103 let plan = linux_plan(StartMode::Boot);
7104 let rendered = systemd_unit(&plan);
7105 let widened = edited(
7106 &rendered,
7107 "ReadWritePaths=/var/lib/runner-manager/config",
7108 "ReadWritePaths=/etc /var/lib/runner-manager/config",
7109 );
7110 let review = review_least_privilege(
7111 &ServiceDefinition::from_text(DefinitionKind::SystemdUnit, widened),
7112 &plan,
7113 );
7114 assert!(!review.is_least_privilege(), "{review}");
7115 assert!(
7116 review
7117 .excesses()
7118 .iter()
7119 .any(|finding| finding.detail.contains("/etc")),
7120 "the review must name the directory it objects to: {review}"
7121 );
7122 }
7123
7124 #[test]
7125 fn a_unit_that_drops_a_hardening_directive_is_not_least_privilege() {
7126 let plan = linux_plan(StartMode::Boot);
7127 let rendered = systemd_unit(&plan);
7128 let weakened = edited(&rendered, "NoNewPrivileges=yes\n", "");
7129 let review = review_least_privilege(
7130 &ServiceDefinition::from_text(DefinitionKind::SystemdUnit, weakened),
7131 &plan,
7132 );
7133 assert!(!review.is_least_privilege(), "{review}");
7134 assert!(
7135 review
7136 .excesses()
7137 .iter()
7138 .any(|finding| finding.subject == "NoNewPrivileges"),
7139 "{review}"
7140 );
7141 }
7142
7143 #[test]
7144 fn a_unit_that_keeps_capabilities_is_not_least_privilege() {
7145 let plan = linux_plan(StartMode::Boot);
7146 let rendered = systemd_unit(&plan);
7147 let widened = edited(
7148 &rendered,
7149 "CapabilityBoundingSet=\n",
7150 "CapabilityBoundingSet=CAP_NET_ADMIN CAP_SYS_ADMIN\n",
7151 );
7152 let review = review_least_privilege(
7153 &ServiceDefinition::from_text(DefinitionKind::SystemdUnit, widened),
7154 &plan,
7155 );
7156 assert!(!review.is_least_privilege(), "{review}");
7157 assert!(
7158 review
7159 .excesses()
7160 .iter()
7161 .any(|finding| finding.subject == "CapabilityBoundingSet"),
7162 "{review}"
7163 );
7164 }
7165
7166 #[test]
7167 fn a_unit_that_opens_a_listening_socket_is_not_least_privilege() {
7168 let plan = linux_plan(StartMode::Boot);
7169 let rendered = systemd_unit(&plan);
7170 let widened = edited(
7171 &rendered,
7172 "[Install]",
7173 "ListenStream=127.0.0.1:9000\n\n[Install]",
7174 );
7175 let review = review_least_privilege(
7176 &ServiceDefinition::from_text(DefinitionKind::SystemdUnit, widened),
7177 &plan,
7178 );
7179 assert!(
7180 !review.is_least_privilege(),
7181 "07-security.md rule 2 forbids any inbound surface: {review}"
7182 );
7183 }
7184
7185 #[test]
7186 fn a_unit_that_makes_a_directory_unwritable_is_a_shortfall_not_an_excess() {
7187 let plan = linux_plan(StartMode::Boot);
7188 let rendered = systemd_unit(&plan);
7189 let narrowed = edited(&rendered, " /var/lib/runner-manager/runtime", "");
7190 let review = review_least_privilege(
7191 &ServiceDefinition::from_text(DefinitionKind::SystemdUnit, narrowed),
7192 &plan,
7193 );
7194 assert!(
7195 review.is_least_privilege(),
7196 "too little authority is not an excess: {review}"
7197 );
7198 assert!(
7199 review
7200 .findings()
7201 .iter()
7202 .any(|finding| finding.kind == FindingKind::Shortfall
7203 && finding.detail.contains("runtime")),
7204 "{review}"
7205 );
7206 }
7207
7208 #[test]
7209 fn a_launch_agent_that_names_an_account_is_not_least_privilege() {
7210 let plan = linux_plan(StartMode::Login);
7211 let rendered = launchd_plist(&plan);
7212 let widened = edited(
7213 &rendered,
7214 "<key>ProcessType</key>",
7215 "<key>UserName</key>\n <string>root</string>\n <key>ProcessType</key>",
7216 );
7217 let review = review_least_privilege(
7218 &ServiceDefinition::from_text(DefinitionKind::LaunchdPlist, widened),
7219 &plan,
7220 );
7221 assert!(!review.is_least_privilege(), "{review}");
7222 assert!(
7223 review
7224 .excesses()
7225 .iter()
7226 .any(|finding| finding.subject == "UserName"),
7227 "{review}"
7228 );
7229 }
7230
7231 #[test]
7232 fn a_launch_daemon_that_asks_for_a_session_is_not_least_privilege() {
7233 let plan = linux_plan(StartMode::Boot);
7234 let rendered = launchd_plist(&plan);
7235 let widened = edited(
7236 &rendered,
7237 "<key>SessionCreate</key>\n <false/>",
7238 "<key>SessionCreate</key>\n <true/>",
7239 );
7240 let review = review_least_privilege(
7241 &ServiceDefinition::from_text(DefinitionKind::LaunchdPlist, widened),
7242 &plan,
7243 );
7244 assert!(!review.is_least_privilege(), "{review}");
7245 assert!(
7246 review
7247 .excesses()
7248 .iter()
7249 .any(|finding| finding.subject == "SessionCreate"),
7250 "{review}"
7251 );
7252 }
7253
7254 #[test]
7255 fn a_launchd_job_that_publishes_a_mach_service_is_not_least_privilege() {
7256 let plan = linux_plan(StartMode::Boot);
7257 let rendered = launchd_plist(&plan);
7258 let widened = edited(
7259 &rendered,
7260 "<key>ProcessType</key>",
7261 "<key>MachServices</key>\n <dict/>\n <key>ProcessType</key>",
7262 );
7263 let review = review_least_privilege(
7264 &ServiceDefinition::from_text(DefinitionKind::LaunchdPlist, widened),
7265 &plan,
7266 );
7267 assert!(!review.is_least_privilege(), "{review}");
7268 }
7269
7270 #[test]
7271 fn a_task_asking_for_the_highest_available_token_is_not_least_privilege() {
7272 let plan = windows_plan(StartMode::Login);
7273 let rendered = windows_scheduled_task_xml(&plan, &TaskPrincipal::named("HOST\\op"));
7274 let widened = edited(
7275 &rendered,
7276 "<RunLevel>LeastPrivilege</RunLevel>",
7277 "<RunLevel>HighestAvailable</RunLevel>",
7278 );
7279 let review = review_least_privilege(
7280 &ServiceDefinition::from_text(DefinitionKind::WindowsScheduledTask, widened),
7281 &plan,
7282 );
7283 assert!(!review.is_least_privilege(), "{review}");
7284 assert!(
7285 review
7286 .excesses()
7287 .iter()
7288 .any(|finding| finding.subject == "RunLevel"),
7289 "{review}"
7290 );
7291 }
7292
7293 #[test]
7294 fn a_task_that_would_store_a_password_is_not_least_privilege() {
7295 let plan = windows_plan(StartMode::Login);
7296 let rendered = windows_scheduled_task_xml(&plan, &TaskPrincipal::named("HOST\\op"));
7297 let widened = edited(
7298 &rendered,
7299 "<LogonType>InteractiveToken</LogonType>",
7300 "<LogonType>Password</LogonType>",
7301 );
7302 let review = review_least_privilege(
7303 &ServiceDefinition::from_text(DefinitionKind::WindowsScheduledTask, widened),
7304 &plan,
7305 );
7306 assert!(!review.is_least_privilege(), "{review}");
7307 }
7308
7309 #[test]
7310 fn an_interactive_windows_service_is_not_least_privilege() {
7311 let plan = windows_plan(StartMode::Boot);
7312 let rendered = windows_service_descriptor(&plan);
7313 let widened = edited(
7314 &rendered,
7315 "ServiceType=OWN_PROCESS",
7316 "ServiceType=OWN_PROCESS|INTERACTIVE_PROCESS",
7317 );
7318 let review = review_least_privilege(
7319 &ServiceDefinition::from_text(DefinitionKind::WindowsService, widened),
7320 &plan,
7321 );
7322 assert!(!review.is_least_privilege(), "{review}");
7323 assert!(
7324 review
7325 .excesses()
7326 .iter()
7327 .any(|finding| finding.subject == "ServiceType"),
7328 "{review}"
7329 );
7330 }
7331
7332 #[test]
7333 fn a_windows_service_under_an_account_the_store_dacl_does_not_name_is_reported() {
7334 let plan = windows_plan(StartMode::Boot);
7335 let rendered = windows_service_descriptor(&plan);
7336 let changed = edited(
7337 &rendered,
7338 "Account=NT AUTHORITY\\SYSTEM",
7339 "Account=NT AUTHORITY\\LocalService",
7340 );
7341 let review = review_least_privilege(
7342 &ServiceDefinition::from_text(DefinitionKind::WindowsService, changed),
7343 &plan,
7344 );
7345 assert!(review.is_least_privilege(), "{review}");
7351 assert!(
7352 review.findings().iter().any(|finding| {
7353 finding.kind == FindingKind::Shortfall && finding.subject == "Account"
7354 }),
7355 "{review}"
7356 );
7357 }
7358
7359 #[test]
7364 fn a_recorded_path_that_is_still_there_is_current() {
7365 let host = Host::new();
7366 let state = inspect_binary(&host.binary, Some(&host.binary));
7367 assert!(!state.is_error(), "{state}");
7368 assert!(matches!(state, BinaryPath::Current { .. }), "{state}");
7369 }
7370
7371 #[test]
7372 fn the_npm_upgrade_case_reports_a_stale_path_as_an_error() {
7373 let host = Host::new();
7374 let recorded = host.binary.clone();
7378 let healthy = inspect_binary(&recorded, Some(&recorded));
7379 assert!(
7380 !healthy.is_error(),
7381 "the discriminator: before the binary moves, this must be healthy"
7382 );
7383
7384 std::fs::remove_file(&recorded).expect("the binary moves out from under the record");
7385
7386 let state = inspect_binary(&recorded, Some(&recorded));
7387 assert!(state.is_error(), "{state}");
7388 assert!(matches!(state, BinaryPath::Missing { .. }), "{state}");
7389 assert!(
7390 state.to_string().contains("npm"),
7391 "the message must name the cause an operator will not otherwise connect: {state}"
7392 );
7393 }
7394
7395 #[test]
7396 fn a_directory_at_the_recorded_path_is_not_something_the_manager_can_start() {
7397 let root = tempfile::tempdir().expect("a temporary directory");
7398 let state = inspect_binary(root.path(), None);
7399 assert!(state.is_error(), "{state}");
7400 assert!(matches!(state, BinaryPath::NotExecutable { .. }), "{state}");
7401 }
7402
7403 #[test]
7404 fn a_registration_naming_a_different_binary_is_a_divergence() {
7405 let host = Host::new();
7406 let other = host.binary.with_file_name("something-else");
7407 let state = inspect_binary(&host.binary, Some(&other));
7408 assert!(state.is_error(), "{state}");
7409 assert!(matches!(state, BinaryPath::Diverged { .. }), "{state}");
7410 }
7411
7412 #[test]
7413 fn absence_is_reported_before_divergence() {
7414 let host = Host::new();
7415 let recorded = host.binary.clone();
7416 std::fs::remove_file(&recorded).expect("removable");
7417 let other = recorded.with_file_name("something-else");
7418 assert!(matches!(
7422 inspect_binary(&recorded, Some(&other)),
7423 BinaryPath::Missing { .. }
7424 ));
7425 }
7426
7427 #[test]
7432 fn the_record_round_trips_through_toml() {
7433 let host = Host::new();
7434 let plan = InstallPlan::resolve(
7435 ServiceIdentity::product(),
7436 &host.request(StartMode::Boot),
7437 ServiceDirectories::of(&host.paths),
7438 )
7439 .expect("a resolvable plan");
7440 let definition = ServiceDefinition::from_text(DefinitionKind::SystemdUnit, "[Service]\n");
7441 let record = InstallRecord::of(&plan, &definition, Utc::now());
7442 record.write(&host.paths).expect("a writable record");
7443 let read = InstallRecord::read(&host.paths)
7444 .expect("a readable record")
7445 .expect("a record is there");
7446 assert_eq!(read, record);
7447 assert_eq!(read.binary, host.binary);
7448 assert!(read.binary.is_absolute());
7449 }
7450
7451 #[cfg(unix)]
7460 #[test]
7461 fn the_record_is_not_written_readable_only_by_whoever_installed_it() {
7462 use std::os::unix::fs::PermissionsExt as _;
7463
7464 let host = Host::new();
7465 let plan = InstallPlan::resolve(
7466 ServiceIdentity::product(),
7467 &host.request(StartMode::Boot),
7468 ServiceDirectories::of(&host.paths),
7469 )
7470 .expect("a resolvable plan");
7471 let definition = ServiceDefinition::from_text(DefinitionKind::SystemdUnit, "[Service]\n");
7472 InstallRecord::of(&plan, &definition, Utc::now())
7473 .write(&host.paths)
7474 .expect("a writable record");
7475
7476 let mode = std::fs::metadata(InstallRecord::path(&host.paths))
7477 .expect("the record is there")
7478 .permissions()
7479 .mode()
7480 & 0o777;
7481 assert_eq!(
7482 mode, 0o644,
7483 "the record is mode {mode:04o}; at 0600 an operator cannot read a record `sudo \
7484 service install` wrote, and `service status` fails on their own host. It holds no \
7485 credential and sits in a 0700 directory, so 0644 discloses nothing"
7486 );
7487 }
7488
7489 #[test]
7493 fn a_record_without_a_source_binary_still_reads_and_says_it_has_none() {
7494 let host = Host::new();
7495 let path = InstallRecord::path(&host.paths);
7496 std::fs::write(
7497 &path,
7498 format!(
7499 "schema_version = {RECORD_SCHEMA_VERSION}
7500service_name = \"runner-manager\"
7501 manager = \"systemd\"
7502start_mode = \"boot\"
7503account = \"root\"
7504 binary = \"/x\"
7505arguments = []
7506restart_delay_secs = 15
7507 restart_reset_secs = 600
7508log_file = \"/x\"
7509 installed_at = \"2026-01-01T00:00:00Z\"
7510installed_by_version = \"0.1.0\"
7511 [directories]
7512config = \"/a\"
7513state = \"/b\"
7514runtime = \"/c\"
7515logs = \"/d\"
7516"
7517 ),
7518 )
7519 .expect("a writable record");
7520 let read = InstallRecord::read(&host.paths)
7521 .expect("a record missing an optional field is still readable")
7522 .expect("a record is there");
7523 assert_eq!(
7524 read.source_binary, None,
7525 "the legacy layout has no source, and must not invent one"
7526 );
7527 }
7528
7529 #[test]
7531 fn a_registration_remembers_the_file_it_was_copied_from() {
7532 let host = Host::new();
7533 let source = host.binary.with_file_name("npm-installed-runner-manager");
7534 std::fs::copy(&host.binary, &source).expect("a second file to stand in for the package");
7535 let plan = InstallPlan::resolve(
7536 ServiceIdentity::product(),
7537 &host.request(StartMode::Boot).copied_from(&source),
7538 ServiceDirectories::of(&host.paths),
7539 )
7540 .expect("a resolvable plan");
7541 let definition = ServiceDefinition::from_text(
7542 DefinitionKind::SystemdUnit,
7543 "[Service]
7544",
7545 );
7546 let record = InstallRecord::of(&plan, &definition, Utc::now());
7547 record.write(&host.paths).expect("a writable record");
7548
7549 let read = InstallRecord::read(&host.paths)
7550 .expect("a readable record")
7551 .expect("a record is there");
7552 assert_eq!(read.source_binary.as_deref(), Some(source.as_path()));
7553 assert_ne!(
7554 read.source_binary.as_deref(),
7555 Some(read.binary.as_path()),
7556 "the whole point is that the two are different files: one the service holds open, one the package manager is free to replace"
7557 );
7558 }
7559
7560 #[test]
7561 fn a_record_from_a_schema_this_build_cannot_read_is_refused_with_a_remedy() {
7562 let host = Host::new();
7563 let path = InstallRecord::path(&host.paths);
7564 std::fs::write(
7565 &path,
7566 format!(
7567 "schema_version = {}\nservice_name = \"runner-manager\"\nmanager = \"systemd\"\n\
7568 start_mode = \"boot\"\naccount = \"root\"\nbinary = \"/x\"\narguments = []\n\
7569 restart_delay_secs = 15\nrestart_reset_secs = 600\nlog_file = \"/x\"\n\
7570 installed_at = \"2026-01-01T00:00:00Z\"\ninstalled_by_version = \"0.1.0\"\n\
7571 [directories]\nconfig = \"/a\"\nstate = \"/b\"\nruntime = \"/c\"\nlogs = \"/d\"\n",
7572 RECORD_SCHEMA_VERSION + 1
7573 ),
7574 )
7575 .expect("a writable record");
7576 let error = InstallRecord::read(&host.paths).expect_err("a future schema is refused");
7577 assert!(
7578 matches!(error, ServiceError::RecordUnreadable { .. }),
7579 "{error}"
7580 );
7581 assert!(
7582 error.to_string().contains("service uninstall"),
7583 "the message must say how to recover: {error}"
7584 );
7585 }
7586
7587 #[test]
7588 fn no_record_is_not_an_error() {
7589 let host = Host::new();
7590 assert_eq!(InstallRecord::read(&host.paths).expect("no record"), None);
7591 assert!(!InstallRecord::remove(&host.paths).expect("nothing to remove"));
7592 }
7593
7594 #[test]
7599 fn no_heartbeat_reads_as_never_rather_than_as_the_epoch() {
7600 let host = Host::new();
7601 assert_eq!(last_github_contact(&host.paths).expect("readable"), None);
7602 }
7603
7604 #[test]
7605 fn the_heartbeat_round_trips_to_the_second() {
7606 let host = Host::new();
7607 let at = DateTime::parse_from_rfc3339("2026-08-22T10:11:12Z")
7608 .expect("a valid timestamp")
7609 .with_timezone(&Utc);
7610 record_github_contact(&host.paths, at).expect("a writable heartbeat");
7611 assert_eq!(
7612 last_github_contact(&host.paths).expect("readable"),
7613 Some(at)
7614 );
7615 }
7616
7617 #[test]
7618 fn a_malformed_heartbeat_is_an_error_and_not_silently_never() {
7619 let host = Host::new();
7620 std::fs::write(contact_path(&host.paths), b"this is not toml \x00").expect("writable");
7621 let error = last_github_contact(&host.paths)
7622 .expect_err("a heartbeat that cannot be parsed is not the same as no heartbeat");
7623 assert!(matches!(error, ServiceError::Record { .. }), "{error}");
7624 }
7625
7626 #[test]
7632 fn a_runner_root_refusal_round_trips_for_service_status() {
7633 let host = Host::new();
7634 let at = DateTime::from_timestamp(1_760_000_000, 0).expect("a valid instant");
7635
7636 assert!(
7637 runner_root_refusals(&host.paths)
7638 .expect("readable")
7639 .is_empty(),
7640 "no record means every policy is placing runners"
7641 );
7642
7643 record_runner_root_refusal(
7644 &host.paths,
7645 "policy-a",
7646 at,
7647 "denied_by_privacy_policy",
7648 "/Volumes/NVME/runners",
7649 "the runner root /Volumes/NVME/runners cannot be used: ... Grant Full Disk Access",
7650 )
7651 .expect("a writable record");
7652
7653 let refusals = runner_root_refusals(&host.paths).expect("readable");
7654 assert_eq!(refusals.len(), 1);
7655 assert_eq!(refusals[0].policy, "policy-a");
7656 assert_eq!(refusals[0].at, at);
7657 assert_eq!(refusals[0].kind, "denied_by_privacy_policy");
7658 assert_eq!(refusals[0].root, "/Volumes/NVME/runners");
7659 assert!(
7660 refusals[0].detail.contains("/Volumes/NVME/runners")
7661 && refusals[0].detail.contains("Full Disk Access"),
7662 "the path and the remediation are the whole point of this file: {refusals:?}"
7663 );
7664 }
7665
7666 #[test]
7674 fn one_policy_placing_a_runner_does_not_clear_another_policys_refusal() {
7675 let host = Host::new();
7676 let at = DateTime::from_timestamp(1_760_000_000, 0).expect("a valid instant");
7677 record_runner_root_refusal(
7678 &host.paths,
7679 "broken",
7680 at,
7681 "denied_by_privacy_policy",
7682 "/Volumes/NVME/runners",
7683 "detail",
7684 )
7685 .expect("a writable record");
7686 record_runner_root_refusal(
7687 &host.paths,
7688 "also-broken",
7689 at,
7690 "not_writable",
7691 "/srv/other",
7692 "detail",
7693 )
7694 .expect("a writable record");
7695
7696 clear_runner_root_refusal(&host.paths, "healthy")
7698 .expect("clearing an absent policy is fine");
7699 clear_runner_root_refusal(&host.paths, "also-broken").expect("that policy recovered");
7700
7701 let refusals = runner_root_refusals(&host.paths).expect("readable");
7702 assert_eq!(
7703 refusals
7704 .iter()
7705 .map(|r| r.policy.as_str())
7706 .collect::<Vec<_>>(),
7707 vec!["broken"],
7708 "the policy that is still refused must keep its record"
7709 );
7710 }
7711
7712 #[test]
7715 fn clearing_the_last_refusal_removes_the_file_and_is_idempotent() {
7716 let host = Host::new();
7717 record_runner_root_refusal(
7718 &host.paths,
7719 "p",
7720 Utc::now(),
7721 "not_writable",
7722 "/srv/x",
7723 "detail",
7724 )
7725 .expect("a writable record");
7726
7727 clear_runner_root_refusal(&host.paths, "p").expect("the record is removed");
7728 assert!(
7729 !root_refusal_path(&host.paths).exists(),
7730 "a host with nothing refused leaves nothing behind"
7731 );
7732 clear_runner_root_refusal(&host.paths, "p").expect("removing what is gone is not an error");
7733 }
7734
7735 #[test]
7738 fn a_malformed_refusal_is_an_error_and_not_silently_none() {
7739 let host = Host::new();
7740 std::fs::write(root_refusal_path(&host.paths), b"not toml \x00").expect("writable");
7741 let error = runner_root_refusals(&host.paths)
7742 .expect_err("an unparseable record is not the same as no record");
7743 assert!(matches!(error, ServiceError::Record { .. }), "{error}");
7744 }
7745
7746 #[test]
7756 fn service_status_reports_a_refusal_as_a_note_and_stays_healthy() {
7757 let host = Host::new();
7758 record_runner_root_refusal(
7759 &host.paths,
7760 "policy-a",
7761 DateTime::from_timestamp(1_760_000_000, 0).expect("a valid instant"),
7762 "denied_by_privacy_policy",
7763 "/Volumes/NVME/runners",
7764 "Grant Full Disk Access to the program that runs the service",
7765 )
7766 .expect("a writable record");
7767
7768 let status = host.operations().status().expect("a readable status");
7769 let notes = status.notes().join("\n");
7770
7771 assert!(
7772 notes.contains("/Volumes/NVME/runners") && notes.contains("Full Disk Access"),
7773 "the directory and the remediation the log had to scrub must appear here: {notes}"
7774 );
7775 assert!(
7776 notes.contains("policy-a"),
7777 "the operator has to know which target placed no runner: {notes}"
7778 );
7779 assert!(
7780 !status
7781 .problems()
7782 .iter()
7783 .any(|problem| problem.subject == "runner root"),
7784 "a record nothing an operator types can clear must not drive the exit code"
7785 );
7786 }
7787
7788 #[test]
7794 fn install_records_the_absolute_binary_path_and_the_four_directories() {
7795 let host = Host::new();
7796 let installed = host
7797 .operations()
7798 .install(&host.request(StartMode::Boot))
7799 .expect("an install against the recording controls");
7800
7801 assert_eq!(installed.record.binary, host.binary);
7802 assert!(installed.record.binary.is_absolute());
7803 assert_eq!(installed.record.start_mode, StartMode::Boot);
7804 assert_eq!(installed.record.arguments, vec!["daemon", "run"]);
7805 assert_eq!(
7806 installed.record.directories,
7807 ServiceDirectories::of(&host.paths)
7808 );
7809 assert_eq!(
7810 installed.record.log_file,
7811 host.paths.logs_dir().join(LOG_FILE_STEM)
7812 );
7813 assert_eq!(installed.record.restart_delay_secs, 15);
7814
7815 let registrations = host.controls.registrations();
7816 assert_eq!(registrations.len(), 1);
7817 assert_eq!(registrations[0].0, StartMode::Boot);
7818 assert_eq!(registrations[0].1, "runner-manager");
7819 }
7820
7821 #[test]
7822 fn install_is_refused_while_the_single_instance_lock_is_held() {
7823 let host = Host::new();
7824 {
7826 let operations = host.operations();
7827 operations
7828 .install(&host.request(StartMode::Boot))
7829 .expect("an install with the lock free");
7830 operations.uninstall().expect("a clean slate");
7831 }
7832
7833 let _held = HostLock::try_acquire(&host.paths, LockKind::SingleInstance)
7834 .expect("this process takes the lock first");
7835
7836 let error = host
7837 .operations()
7838 .install(&host.request(StartMode::Boot))
7839 .expect_err("a second agent must not be registered while one is running");
7840 assert!(matches!(error, ServiceError::LockHeld { .. }), "{error}");
7841 assert!(
7842 error.to_string().contains("already running"),
7843 "the message must be actionable: {error}"
7844 );
7845 assert!(
7846 host.controls.registrations().is_empty(),
7847 "a refused install must register nothing"
7848 );
7849 assert_eq!(
7850 InstallRecord::read(&host.paths).expect("readable"),
7851 None,
7852 "a refused install must write no record"
7853 );
7854 }
7855
7856 #[test]
7864 fn installing_over_the_same_start_mode_replaces_the_registration() {
7865 let host = Host::new();
7866 let operations = host.operations();
7867 operations
7868 .install(&host.request(StartMode::Boot))
7869 .expect("the first install");
7870
7871 let again = operations
7872 .install(&host.request(StartMode::Boot))
7873 .expect("an install over the same mode replaces rather than refusing");
7874
7875 assert!(
7876 again.replaced_existing,
7877 "the operator is told this replaced something rather than made it"
7878 );
7879 assert_eq!(
7880 host.controls.registrations().len(),
7881 1,
7882 "replacing must not leave two registrations behind"
7883 );
7884 assert_eq!(
7885 InstallRecord::read(&host.paths)
7886 .expect("readable")
7887 .expect("a record")
7888 .start_mode,
7889 StartMode::Boot
7890 );
7891 }
7892
7893 #[test]
7897 fn installing_over_the_other_start_mode_is_refused() {
7898 let host = Host::new();
7899 let operations = host.operations();
7900 operations
7901 .install(&host.request(StartMode::Boot))
7902 .expect("the first install");
7903
7904 let error = operations
7905 .install(&host.request(StartMode::Login))
7906 .expect_err("a mode change is not an install");
7907 assert!(
7908 matches!(
7909 error,
7910 ServiceError::AlreadyInstalled {
7911 existing: StartMode::Boot,
7912 requested: StartMode::Login,
7913 ..
7914 }
7915 ),
7916 "{error}"
7917 );
7918 assert!(
7919 !error
7920 .to_string()
7921 .contains("switch the start mode in place,"),
7922 "the old remedy named a capability no command offers; the terminal UI is where \
7923 the start mode moves: {error}"
7924 );
7925 assert_eq!(host.controls.registrations().len(), 1);
7926 }
7927
7928 #[test]
7929 fn install_rolls_back_the_registration_when_record_persistence_fails() {
7930 let host = Host::new();
7931 let record_path = InstallRecord::path(&host.paths);
7932 std::fs::create_dir(&record_path).expect("a directory blocks the record file");
7933
7934 let error = host
7935 .operations()
7936 .install(&host.request(StartMode::Boot))
7937 .expect_err("record persistence must fail");
7938
7939 assert!(matches!(error, ServiceError::Record { .. }), "{error}");
7940 assert!(
7941 host.controls.registrations().is_empty(),
7942 "a failed install must not leave a live unrecorded registration"
7943 );
7944 assert!(
7945 host.controls
7946 .calls()
7947 .iter()
7948 .any(|call| call == "uninstall runner-manager (boot)"),
7949 "the registration must be explicitly rolled back: {:?}",
7950 host.controls.calls()
7951 );
7952 assert!(
7953 !host.runner_root.as_path().exists(),
7954 "the rollback must take the runner root this install created with it; a directory \
7955 prepared for a registration that does not exist is litter, and on Windows it is \
7956 litter with a security descriptor"
7957 );
7958 }
7959
7960 #[test]
7965 fn the_runner_root_a_boot_registration_needs_admits_only_the_service() {
7966 use crate::runner_root_access::{RootAdmission, default_root_sddl, grants_broad_write};
7967
7968 assert_eq!(
7972 ServiceAccount::for_definition(DefinitionKind::WindowsService, StartMode::Boot),
7973 ServiceAccount::LocalSystem
7974 );
7975 assert_eq!(
7976 ServiceAccount::for_definition(DefinitionKind::WindowsScheduledTask, StartMode::Login),
7977 ServiceAccount::InvokingUser
7978 );
7979
7980 let boot = default_root_sddl(&RootAdmission::LocalSystem);
7981 assert!(!grants_broad_write(&boot), "{boot}");
7982 assert!(
7983 !boot.contains("S-1-5-21"),
7984 "a boot registration runs as LocalSystem, so its root names no operator: {boot}"
7985 );
7986
7987 let login = default_root_sddl(&RootAdmission::Account("S-1-5-21-1-2-3-1001".to_owned()));
7992 assert!(login.contains("S-1-5-21-1-2-3-1001"), "{login}");
7993 assert!(!grants_broad_write(&login), "{login}");
7994 }
7995
7996 #[test]
7997 fn an_install_reports_the_runner_root_it_prepared() {
7998 let host = Host::new();
7999 let installed = host
8000 .operations()
8001 .install(&host.request(StartMode::Boot))
8002 .expect("an install");
8003 let rendered = installed.runner_root.to_string();
8004 assert!(
8005 !rendered.contains("S-1-5-21"),
8006 "the report must add no identity to the output: {rendered}"
8007 );
8008 if cfg!(windows) {
8009 assert_eq!(
8010 installed.runner_root.path(),
8011 Some(host.runner_root.as_path())
8012 );
8013 assert!(
8014 host.runner_root.as_path().is_dir(),
8015 "the directory jobs would run in has to exist once the service is registered"
8016 );
8017 } else {
8018 assert_eq!(
8019 installed.runner_root,
8020 crate::runner_root_access::RootAccessSummary::NotApplicable,
8021 "macOS and Linux keep the runtime directory they have always used"
8022 );
8023 }
8024 }
8025
8026 #[test]
8027 fn switching_start_mode_reconciles_the_runner_root_for_the_new_account() {
8028 let host = Host::new();
8029 let operations = host.operations();
8030 operations
8031 .install(&host.request(StartMode::Boot))
8032 .expect("an install at boot");
8033
8034 let change = operations
8035 .set_start_mode(StartMode::Login)
8036 .expect("a switch to login");
8037
8038 assert!(change.changed);
8039 if cfg!(windows) {
8040 assert_eq!(change.runner_root.path(), Some(host.runner_root.as_path()));
8041 assert!(
8042 host.runner_root.as_path().is_dir(),
8043 "the switch must not remove the directory it reconciled"
8044 );
8045 }
8046 assert!(
8049 change.to_string().contains("runner root"),
8050 "{}",
8051 change.to_string()
8052 );
8053 }
8054
8055 #[test]
8056 fn switching_to_the_mode_already_in_force_touches_no_runner_root() {
8057 let host = Host::new();
8058 let operations = host.operations();
8059 operations
8060 .install(&host.request(StartMode::Boot))
8061 .expect("an install at boot");
8062
8063 let change = operations
8064 .set_start_mode(StartMode::Boot)
8065 .expect("a switch to the mode already in force");
8066
8067 assert!(!change.changed);
8068 assert_eq!(
8069 change.runner_root,
8070 crate::runner_root_access::RootAccessSummary::NotApplicable,
8071 "nothing moves, so nothing about the root's access control has to; reconciling here \
8072 would turn a no-op command into one that can fail on a permission it does not need"
8073 );
8074 }
8075
8076 #[cfg(windows)]
8077 #[test]
8078 fn a_registration_the_manager_refuses_leaves_no_runner_root_behind() {
8079 let host = Host::new();
8080 host.controls
8081 .fail_next_install(StartMode::Boot, "injected registration failure");
8082
8083 let error = host
8084 .operations()
8085 .install(&host.request(StartMode::Boot))
8086 .expect_err("the manager refuses the registration");
8087
8088 assert!(matches!(error, ServiceError::Control { .. }), "{error}");
8089 assert!(
8090 !host.runner_root.as_path().exists(),
8091 "the directory was created for a registration that does not exist"
8092 );
8093 }
8094
8095 #[cfg(windows)]
8096 #[test]
8097 fn an_existing_broad_runner_root_refuses_the_install_before_anything_is_registered() {
8098 let host = Host::new();
8099 crate::runner_root_access::create_with_descriptor_for_tests(
8103 host.runner_root.as_path(),
8104 "D:P(A;OICI;FA;;;SY)(A;OICI;FA;;;WD)",
8105 )
8106 .expect("a deliberately open runner root");
8107 let before = crate::runner_root_access::report(host.runner_root.as_path());
8108
8109 let error = host
8110 .operations()
8111 .install(&host.request(StartMode::Boot))
8112 .expect_err("an open runner root is refused");
8113
8114 assert!(matches!(error, ServiceError::RunnerRoot { .. }), "{error}");
8115 assert!(
8116 error.to_string().contains("nothing was registered"),
8117 "{error}"
8118 );
8119 assert!(
8120 host.controls.registrations().is_empty(),
8121 "the refusal has to come before the platform is asked to register anything: {:?}",
8122 host.controls.calls()
8123 );
8124 assert_eq!(
8125 crate::runner_root_access::report(host.runner_root.as_path()),
8126 before,
8127 "an open directory is refused rather than tightened: its contents cannot be trusted, \
8128 so adopting it would be worse than declining it"
8129 );
8130 }
8131
8132 #[cfg(windows)]
8133 #[test]
8134 fn uninstall_leaves_the_runner_root_exactly_where_it_is() {
8135 let host = Host::new();
8136 let operations = host.operations();
8137 operations
8138 .install(&host.request(StartMode::Boot))
8139 .expect("an install");
8140 assert!(host.runner_root.as_path().is_dir());
8141
8142 operations.uninstall().expect("an uninstall");
8143
8144 assert!(
8145 host.runner_root.as_path().is_dir(),
8146 "`05-infrastructure.md` item 5: uninstall deregisters and deletes nothing else. A \
8147 runner root may hold an operator's retained workspaces."
8148 );
8149 }
8150
8151 #[test]
8152 fn install_reviews_what_it_registered() {
8153 let host = Host::new();
8154 let installed = host
8155 .operations()
8156 .install(&host.request(StartMode::Boot))
8157 .expect("an install");
8158 assert!(
8159 installed.review.is_least_privilege(),
8160 "{}",
8161 installed.review
8162 );
8163 assert!(
8164 !installed.review.controls().is_empty(),
8165 "a review that confirms nothing proves nothing: {}",
8166 installed.review
8167 );
8168 assert_eq!(
8169 installed.review.kind(),
8170 host_definition_kind(StartMode::Boot),
8171 "the review must be of the definition this host's manager was given"
8172 );
8173 assert!(
8174 !installed.review.account().justification().is_empty(),
8175 "a privileged account with no stated reason is an unreviewed one"
8176 );
8177 }
8178
8179 #[test]
8184 fn uninstall_leaves_configuration_sqlite_secrets_and_cache_exactly_as_they_were() {
8185 let host = Host::new();
8186 let operations = host.operations();
8187 operations
8188 .install(&host.request(StartMode::Boot))
8189 .expect("an install");
8190
8191 let config = host.paths.config_dir();
8195 std::fs::write(config.join("runner-manager.db"), b"sqlite fixture").expect("writable");
8196 std::fs::write(config.join("config.toml"), b"host_capacity = 2").expect("writable");
8197 std::fs::create_dir_all(host.paths.state_dir().join("packages/2.330.0")).expect("writable");
8198 std::fs::write(
8199 host.paths
8200 .state_dir()
8201 .join("packages/2.330.0/runner.tar.gz"),
8202 b"cached package",
8203 )
8204 .expect("writable");
8205 std::fs::create_dir_all(host.paths.state_dir().join("secrets")).expect("writable");
8206 std::fs::write(
8207 host.paths.state_dir().join("secrets/user-access-token"),
8208 b"a stand-in for the stored credential",
8209 )
8210 .expect("writable");
8211 std::fs::write(
8212 host.paths.logs_dir().join("runner-manager.log.2026-08-22"),
8213 b"diagnostics",
8214 )
8215 .expect("writable");
8216
8217 let roots: Vec<PathBuf> = host
8218 .paths
8219 .all()
8220 .iter()
8221 .map(|(_, path)| (*path).to_path_buf())
8222 .collect();
8223 let roots: Vec<&Path> = roots.iter().map(PathBuf::as_path).collect();
8224 let before = snapshot(&roots);
8225
8226 assert!(
8229 before.len() >= 6,
8230 "the fixture must actually contain the files this test is about, got {before:#?}"
8231 );
8232 let record_path = InstallRecord::path(&host.paths);
8233 assert!(
8234 before.contains_key(&record_path),
8235 "the install record must be present before uninstall"
8236 );
8237
8238 let uninstalled = operations.uninstall().expect("an uninstall");
8239 assert!(uninstalled.removed_registration);
8240 assert!(uninstalled.removed_record);
8241
8242 let after = snapshot(&roots);
8243
8244 let mut expected = before.clone();
8246 expected.remove(&record_path);
8247 assert_eq!(
8248 after, expected,
8249 "uninstall must remove its own record and nothing else"
8250 );
8251 assert!(
8252 !record_path.exists(),
8253 "the record itself must go, or `uninstall` did nothing at all"
8254 );
8255 assert!(
8256 uninstalled
8257 .preserved
8258 .iter()
8259 .all(|path| roots.contains(&path.as_path())),
8260 "the preserved list must name the four directories: {uninstalled}"
8261 );
8262 }
8263
8264 #[test]
8265 fn uninstall_on_a_host_with_no_registration_is_not_a_failure() {
8266 let host = Host::new();
8267 let uninstalled = host.operations().uninstall().expect("a no-op uninstall");
8268 assert!(!uninstalled.removed_registration);
8269 assert!(!uninstalled.removed_record);
8270 }
8271
8272 #[test]
8273 fn uninstall_removes_a_registration_even_when_the_record_is_gone() {
8274 let host = Host::new();
8275 let operations = host.operations();
8276 operations
8277 .install(&host.request(StartMode::Boot))
8278 .expect("an install");
8279 std::fs::remove_file(InstallRecord::path(&host.paths)).expect("the record is lost");
8280
8281 let uninstalled = operations.uninstall().expect("an uninstall");
8282 assert!(
8283 uninstalled.removed_registration,
8284 "a lost record must not strand a registration"
8285 );
8286 assert!(host.controls.registrations().is_empty());
8287 }
8288
8289 #[test]
8294 fn switching_start_mode_reuses_the_recorded_path_and_re_resolves_nothing() {
8295 let host = Host::new();
8296 let operations = host.operations();
8297 operations
8298 .install(&host.request(StartMode::Boot))
8299 .expect("an install at boot");
8300
8301 std::fs::remove_file(&host.binary).expect("the installed binary goes away");
8305
8306 let change = operations
8307 .set_start_mode(StartMode::Login)
8308 .expect("a switch that does not reinstall the product");
8309 assert!(change.changed);
8310 assert_eq!(change.from, StartMode::Boot);
8311 assert_eq!(change.to, StartMode::Login);
8312 assert_eq!(change.store_scope, crate::secrets::SecretScope::User);
8313
8314 let record = InstallRecord::read(&host.paths)
8315 .expect("readable")
8316 .expect("a record");
8317 assert_eq!(record.start_mode, StartMode::Login);
8318 assert_eq!(
8319 record.binary, host.binary,
8320 "the recorded path must survive the switch untouched"
8321 );
8322
8323 let registrations = host.controls.registrations();
8324 assert_eq!(registrations.len(), 1, "{registrations:?}");
8325 assert_eq!(registrations[0].0, StartMode::Login);
8326 assert!(
8327 registrations[0]
8328 .2
8329 .command_line
8330 .contains(&host.binary.to_string_lossy().into_owned()),
8331 "{:?}",
8332 registrations[0].2
8333 );
8334 }
8335
8336 #[test]
8337 fn switching_start_mode_keeps_the_live_registration_when_target_install_fails() {
8338 let host = Host::new();
8339 let operations = host.operations();
8340 operations
8341 .install(&host.request(StartMode::Boot))
8342 .expect("an install at boot");
8343 let record_before = std::fs::read(InstallRecord::path(&host.paths)).expect("the record");
8344 host.controls
8345 .fail_next_install(StartMode::Login, "injected target failure");
8346
8347 let error = operations
8348 .set_start_mode(StartMode::Login)
8349 .expect_err("the target manager refuses the install");
8350
8351 assert!(matches!(error, ServiceError::Control { .. }), "{error}");
8352 assert_eq!(
8353 std::fs::read(InstallRecord::path(&host.paths)).expect("the old record survives"),
8354 record_before
8355 );
8356 let registrations = host.controls.registrations();
8357 assert_eq!(registrations.len(), 1, "{registrations:?}");
8358 assert_eq!(registrations[0].0, StartMode::Boot);
8359 }
8360
8361 #[test]
8362 fn switching_start_mode_rolls_back_target_when_record_persistence_fails() {
8363 let host = Host::new();
8364 let operations = host.operations();
8365 operations
8366 .install(&host.request(StartMode::Boot))
8367 .expect("an install at boot");
8368 let record_before = std::fs::read(InstallRecord::path(&host.paths)).expect("the record");
8369 let config = host.paths.config_dir().to_path_buf();
8370 let hidden = config.with_file_name("config-hidden-by-fault");
8371 host.controls.hide_directory_after_install(
8372 StartMode::Login,
8373 config.clone(),
8374 hidden.clone(),
8375 );
8376
8377 let error = operations
8378 .set_start_mode(StartMode::Login)
8379 .expect_err("the injected filesystem fault prevents persistence");
8380
8381 std::fs::remove_file(&config).expect("remove the injected blocker");
8382 std::fs::rename(&hidden, &config).expect("restore the record directory");
8383 assert!(matches!(error, ServiceError::Record { .. }), "{error}");
8384 assert_eq!(
8385 std::fs::read(InstallRecord::path(&host.paths)).expect("the old record survives"),
8386 record_before
8387 );
8388 let registrations = host.controls.registrations();
8389 assert_eq!(registrations.len(), 1, "{registrations:?}");
8390 assert_eq!(registrations[0].0, StartMode::Boot);
8391 assert!(
8392 host.controls
8393 .calls()
8394 .iter()
8395 .any(|call| call == "uninstall runner-manager (login)"),
8396 "the target must be rolled back: {:?}",
8397 host.controls.calls()
8398 );
8399 }
8400
8401 #[test]
8402 fn switching_to_the_mode_already_in_force_registers_nothing_again() {
8403 let host = Host::new();
8404 let operations = host.operations();
8405 operations
8406 .install(&host.request(StartMode::Boot))
8407 .expect("an install");
8408 let before = host.controls.calls().len();
8409
8410 let change = operations
8411 .set_start_mode(StartMode::Boot)
8412 .expect("a no-op switch");
8413 assert!(!change.changed);
8414 assert_eq!(
8415 host.controls.calls().len(),
8416 before,
8417 "a no-op switch must not touch the service manager"
8418 );
8419 }
8420
8421 #[test]
8422 fn switching_start_mode_on_a_host_with_no_registration_is_refused() {
8423 let host = Host::new();
8424 let error = host
8425 .operations()
8426 .set_start_mode(StartMode::Login)
8427 .expect_err("there is nothing to switch");
8428 assert!(
8429 matches!(error, ServiceError::NotInstalled { .. }),
8430 "{error}"
8431 );
8432 }
8433
8434 #[test]
8439 fn status_reports_the_four_facts_journey_five_asks_for() {
8440 let host = Host::new();
8441 let operations = host.operations();
8442 operations
8443 .install(&host.request(StartMode::Boot))
8444 .expect("an install");
8445 let at = DateTime::parse_from_rfc3339("2026-08-22T09:00:00Z")
8446 .expect("a valid timestamp")
8447 .with_timezone(&Utc);
8448 record_github_contact(&host.paths, at).expect("a heartbeat");
8449
8450 let status = operations.status().expect("a status");
8451 assert_eq!(status.start_mode(), Some(StartMode::Boot));
8452 assert_eq!(
8453 status.binary().map(BinaryPath::recorded),
8454 Some(host.binary.as_path())
8455 );
8456 assert_eq!(status.log_file(), host.paths.logs_dir().join(LOG_FILE_STEM));
8457 assert_eq!(status.last_github_contact(), Some(at));
8458 assert!(status.is_installed());
8459 assert!(status.is_healthy(), "{status}");
8460
8461 let printed = status.to_string();
8462 for fragment in [
8463 "start mode",
8464 "diagnostic log",
8465 "last GitHub contact",
8466 "binary",
8467 ] {
8468 assert!(printed.contains(fragment), "{printed}");
8469 }
8470 }
8471
8472 #[cfg(unix)]
8484 #[test]
8485 fn status_reports_a_record_it_may_not_read_and_still_reports_the_registration() {
8486 use std::os::unix::fs::PermissionsExt as _;
8487
8488 if unsafe { libc::geteuid() } == 0 {
8492 return;
8493 }
8494
8495 let host = Host::new();
8496 let operations = host.operations();
8497 operations
8498 .install(&host.request(StartMode::Boot))
8499 .expect("an install");
8500 assert!(
8501 operations.status().expect("a status").is_healthy(),
8502 "the discriminator: healthy before the record is made unreadable"
8503 );
8504
8505 let record = InstallRecord::path(&host.paths);
8506 std::fs::set_permissions(&record, std::fs::Permissions::from_mode(0o000))
8507 .expect("the mode is applied");
8508
8509 let status = operations
8510 .status()
8511 .expect("a record this account may not read is reported, not thrown");
8512 assert!(status.is_installed(), "{status}");
8513 assert!(!status.is_healthy(), "{status}");
8514
8515 let printed = status.to_string();
8516 assert!(
8517 printed.contains("this account may not read it"),
8518 "the operator is told which of the two states this is: {printed}"
8519 );
8520 assert!(
8521 !printed.contains("there is no install record"),
8522 "a record that is there and unreadable is not a record that is missing, and the \
8523 missing one's remedy starts with `service uninstall`: {printed}"
8524 );
8525
8526 std::fs::set_permissions(&record, std::fs::Permissions::from_mode(0o644))
8529 .expect("the mode is restored");
8530 }
8531
8532 #[test]
8533 fn status_reports_a_stale_binary_as_an_error_rather_than_appearing_healthy() {
8534 let host = Host::new();
8535 let operations = host.operations();
8536 operations
8537 .install(&host.request(StartMode::Boot))
8538 .expect("an install");
8539
8540 assert!(
8543 operations.status().expect("a status").is_healthy(),
8544 "the freshly installed host must be healthy"
8545 );
8546
8547 std::fs::remove_file(&host.binary).expect("the binary moves out from under the record");
8548
8549 let status = operations.status().expect("a status");
8550 assert!(!status.is_healthy(), "{status}");
8551 assert!(
8552 status
8553 .problems()
8554 .iter()
8555 .any(|problem| problem.subject == "binary"),
8556 "{status}"
8557 );
8558 assert!(status.to_string().contains("STALE"), "{status}");
8559 }
8560
8561 #[test]
8562 fn status_reports_a_registration_that_would_not_start_at_boot() {
8563 let host = Host::new();
8564 let operations = host.operations();
8565 operations
8566 .install(&host.request(StartMode::Boot))
8567 .expect("an install");
8568 assert!(operations.status().expect("a status").is_healthy());
8569
8570 host.controls.edit("runner-manager", |registration| {
8571 registration.starts_automatically = false;
8572 });
8573
8574 let status = operations.status().expect("a status");
8575 assert!(!status.is_healthy(), "{status}");
8576 assert!(
8577 status
8578 .problems()
8579 .iter()
8580 .any(|problem| problem.detail.contains("after a reboot")),
8581 "{status}"
8582 );
8583 }
8584
8585 #[test]
8586 fn install_starts_the_registration_and_a_later_stop_is_unhealthy() {
8587 let host = Host::new();
8588 let operations = host.operations();
8589 operations
8590 .install(&host.request(StartMode::Login))
8591 .expect("install and immediate start");
8592
8593 assert!(
8594 operations.status().expect("running status").is_running(),
8595 "install must not wait for the next login trigger"
8596 );
8597 operations.stop().expect("stop the registration");
8598 let status = operations.status().expect("stopped status");
8599 assert!(!status.is_healthy(), "{status}");
8600 assert!(
8601 status
8602 .problems()
8603 .iter()
8604 .any(|problem| problem.subject == "runtime"),
8605 "{status}"
8606 );
8607 let rendered = status.to_string();
8608 assert!(
8609 rendered.contains("runner-manager service install --start-at login"),
8610 "{rendered}"
8611 );
8612 assert!(
8613 !rendered.contains("runner-manager service start"),
8614 "{rendered}"
8615 );
8616 }
8617
8618 #[test]
8619 fn status_reports_a_restart_policy_something_else_edited() {
8620 let host = Host::new();
8621 let operations = host.operations();
8622 operations
8623 .install(&host.request(StartMode::Boot))
8624 .expect("an install");
8625 assert!(operations.status().expect("a status").is_healthy());
8626
8627 host.controls.edit("runner-manager", |registration| {
8628 registration.restart_delay = Some(Duration::from_secs(1));
8629 });
8630
8631 let status = operations.status().expect("a status");
8632 assert!(!status.is_healthy(), "{status}");
8633 assert!(
8634 status
8635 .problems()
8636 .iter()
8637 .any(|problem| problem.subject == "restart policy"),
8638 "{status}"
8639 );
8640 }
8641
8642 #[test]
8643 fn status_reports_a_registration_naming_a_binary_the_record_does_not() {
8644 let host = Host::new();
8645 let operations = host.operations();
8646 operations
8647 .install(&host.request(StartMode::Boot))
8648 .expect("an install");
8649 let other = host.binary.with_file_name("someone-elses.exe");
8650 std::fs::write(&other, b"x").expect("writable");
8651
8652 host.controls.edit("runner-manager", |registration| {
8653 registration.command_line = quote_argument(&other.to_string_lossy());
8654 });
8655
8656 let status = operations.status().expect("a status");
8657 assert!(!status.is_healthy(), "{status}");
8658 assert!(
8659 matches!(status.binary(), Some(BinaryPath::Diverged { .. })),
8660 "{status}"
8661 );
8662 }
8663
8664 #[test]
8665 fn status_reports_a_record_no_service_manager_knows_about() {
8666 let host = Host::new();
8667 let operations = host.operations();
8668 operations
8669 .install(&host.request(StartMode::Boot))
8670 .expect("an install");
8671 for mode in [StartMode::Boot, StartMode::Login] {
8673 host.controls
8674 .control(mode)
8675 .expect("a control")
8676 .uninstall(&ServiceIdentity::product())
8677 .expect("removed");
8678 }
8679
8680 let status = operations.status().expect("a status");
8681 assert!(!status.is_healthy(), "{status}");
8682 assert!(
8683 status
8684 .problems()
8685 .iter()
8686 .any(|problem| problem.subject == "registration"),
8687 "{status}"
8688 );
8689 }
8690
8691 #[test]
8692 fn status_on_a_host_with_nothing_installed_is_neither_healthy_nor_broken() {
8693 let host = Host::new();
8694 let status = host.operations().status().expect("a status");
8695 assert!(!status.is_installed());
8696 assert!(
8697 status.is_healthy(),
8698 "a host that never installed the service has no fault to report: {status}"
8699 );
8700 assert!(status.to_string().contains("installed"), "{status}");
8701 }
8702
8703 #[test]
8704 fn status_says_a_login_registration_does_not_resume_after_an_unattended_reboot() {
8705 let host = Host::new();
8706 let operations = host.operations();
8707 operations
8708 .install(&host.request(StartMode::Login))
8709 .expect("an install at login");
8710 let status = operations.status().expect("a status");
8711 assert!(
8712 status
8713 .notes()
8714 .iter()
8715 .any(|note| note.contains("does not run until the operator signs in")),
8716 "05-infrastructure.md requires `service status` to say so: {status}"
8717 );
8718 }
8719
8720 #[test]
8725 fn start_and_stop_reach_the_domain_that_holds_the_registration() {
8726 let host = Host::new();
8727 let operations = host.operations();
8728 operations
8729 .install(&host.request(StartMode::Login))
8730 .expect("an install at login");
8731 operations.start().expect("a start");
8732 assert!(operations.status().expect("a status").is_running());
8733 assert!(operations.stop().expect("a stop"));
8734 assert!(!operations.status().expect("a status").is_running());
8735 }
8736
8737 #[test]
8738 fn starting_a_host_with_no_registration_is_refused() {
8739 let host = Host::new();
8740 let error = host.operations().start().expect_err("nothing to start");
8741 assert!(
8742 matches!(error, ServiceError::NotInstalled { .. }),
8743 "{error}"
8744 );
8745 }
8746
8747 #[cfg(windows)]
8763 #[test]
8764 fn the_account_this_installer_registers_is_one_the_stores_own_dacl_admits() {
8765 use crate::secrets::{PlatformSecretStore, SecretScope, SecretStore as _};
8766
8767 let root = tempfile::tempdir().expect("a temporary directory");
8768 let store = PlatformSecretStore::rooted_at(SecretScope::Machine, root.path())
8769 .expect("a rooted machine-scoped store");
8770 store
8771 .store(&secrecy::SecretString::from("a stand-in for the token"))
8772 .expect("the store accepts a value");
8773 let protection = store.protection().expect("the DACL can be read back");
8774
8775 assert!(
8776 protection.description().contains(";;;SY)"),
8777 "the machine-scoped store must admit LocalSystem, or a boot-start service cannot \
8778 read the token. `d2` writes this DACL and it is not this task's to widen. Got: {}",
8779 protection.description()
8780 );
8781 assert_eq!(
8782 ServiceAccount::for_definition(DefinitionKind::WindowsService, StartMode::Boot),
8783 ServiceAccount::LocalSystem,
8784 "and that is the account this installer registers, which is why SY is what matters"
8785 );
8786 assert!(
8787 !protection.readable_by_other_local_users(),
8788 "the same DACL must still exclude ordinary local users: {}",
8789 protection.description()
8790 );
8791
8792 for rejected in [";;;LS)", ";;;NS)"] {
8796 assert!(
8797 !protection.description().contains(rejected),
8798 "if the store ever admitted {rejected}, the least-privilege analysis in \
8799 docs/service-account.md would need redoing: {}",
8800 protection.description()
8801 );
8802 }
8803 }
8804}