File size: 3,133 Bytes
854994d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 | use super::parse_stat;
use pretty_assertions::assert_eq;
#[test]
fn parses_start_ticks_after_comm_with_spaces_and_parentheses() {
let stat =
b"123 (codex \xff) worker) S 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 987654321 20";
assert_eq!(parse_stat(stat).unwrap(), ("S".to_string(), 987654321));
}
#[test]
fn reads_earlier_linux_identity_field() {
use crate::backend::pid::PidRecord;
let mut expected = serde_json::json!({
"pid": 42,
"processStartTime": "legacy timestamp",
"linuxProcessIdentity": {"bootId": "boot", "startTicks": 123},
});
let record: PidRecord = serde_json::from_value(expected.clone()).unwrap();
expected["processIdentity"] = expected
.as_object_mut()
.unwrap()
.remove("linuxProcessIdentity")
.unwrap();
assert_eq!(serde_json::to_value(record).unwrap(), expected);
}
#[cfg(any(target_os = "linux", target_os = "macos"))]
#[tokio::test]
async fn previous_boot_does_not_require_process_inspection() {
let (_, identity) = super::read_process_details(std::process::id())
.await
.unwrap();
let mut record = serde_json::to_value(identity).unwrap();
record["bootId"] = "previous boot".into();
let identity: super::ProcessIdentity = serde_json::from_value(record).unwrap();
// An invalid PID would make process inspection fail if it preceded the boot check.
assert_eq!(
identity.matches_process(u32::MAX).await.unwrap(),
(false, false)
);
}
#[cfg(target_os = "macos")]
#[tokio::test]
async fn reused_pid_can_belong_to_root() {
let (_, identity) = super::read_process_details(std::process::id())
.await
.unwrap();
// launchd is root-owned: PROC_PIDTBSDINFO returns EPERM for an ordinary caller.
assert_eq!(
identity.matches_process(/*pid*/ 1).await.unwrap(),
(false, false)
);
}
#[cfg(target_os = "macos")]
#[tokio::test]
async fn promotes_timestamp_only_macos_record() {
use crate::backend::pid::PidBackend;
use crate::backend::pid::PidRecord;
let temp = tempfile::TempDir::new().unwrap();
let path = temp.path().join("server.pid");
let (_, identity) = super::read_process_details(std::process::id())
.await
.unwrap();
let mut legacy = serde_json::to_value(&identity).unwrap();
legacy.as_object_mut().unwrap().remove("bootId");
legacy.as_object_mut().unwrap().remove("uniqueId");
let mut record = PidRecord {
pid: std::process::id(),
process_start_time: "unused legacy text".into(),
process_identity: Some(serde_json::from_value(legacy).unwrap()),
executable_identity: None,
};
std::fs::write(&path, serde_json::to_vec(&record).unwrap()).unwrap();
let backend = PidBackend::new(
temp.path().join("codex"),
path.clone(),
/*remote_control_enabled*/ false,
);
backend.promote_legacy_identity().await.unwrap();
record.process_identity = Some(identity);
assert_eq!(
serde_json::from_slice::<PidRecord>(&std::fs::read(path).unwrap()).unwrap(),
record,
);
}
|