// K1355 native EL0 waitset deadline cancellation-race probe. use super::*; use turkos_kernel::runtime::process_table::{Outcome, ProcessId}; use turkos_kernel::runtime::syscall_abi as abi; use super::k1355_deadlines_model::{Completion, DeadlineWait, Error, Phase, Token}; static mut OS_K1355_DEADLINES: Option> = None; static mut OS_K1355_STAGE: u8 = 0; fn set() -> &'static mut DeadlineWait { // Mode 62 runs on one CPU with IRQs masked and no concurrent waitset user. unsafe { (&mut *core::ptr::addr_of_mut!(OS_K1355_DEADLINES)) .as_mut().unwrap_or_else(|| park()) } } fn stage() -> u8 { unsafe { read_volatile(core::ptr::addr_of!(OS_K1355_STAGE)) } } fn set_stage(value: u8) { unsafe { write_volatile(core::ptr::addr_of_mut!(OS_K1355_STAGE), value); } } pub(super) fn begin() { unsafe { write_volatile(core::ptr::addr_of_mut!(OS_K1355_DEADLINES), None); } set_stage(0); } pub(super) fn finish() -> Result<(), &'static str> { let snapshot = set().snapshot(); if stage() != 20 || snapshot.generation != 3 || snapshot.entry.is_some() || snapshot.timers != 0 { return Err("waitset deadline owner or timer leak"); } unsafe { core::ptr::write(core::ptr::addr_of_mut!(OS_K1355_DEADLINES), None); } Ok(()) } pub(super) fn dispatch( episode: &mut OsK1215AEl0Episode, frame: &mut os_k920_exception::Frame, process_table: &mut OsK1105Table<'_>, ) -> u64 { if frame.elr < episode.launch.user_pc || frame.elr >= episode.launch.user_pc + 4096 || !frame.elr.is_multiple_of(4) { park(); } if frame.x[7] == 3 { if episode.case != 1 || episode.calls != 1 || process_table.record(episode.pid).is_err() { park(); } let owner = episode.pid; let arg = frame.x[0]; let t1 = Token::from_raw(1); let t2 = Token::from_raw(2); let t3 = Token::from_raw(3); match frame.x[8] { 80 => { if stage() != 2 || arg != 0 || set().arm(owner, 100) != Ok(t1) { park(); } frame.x[0] = t1.raw(); set_stage(3); write_line("TURKOS/OS/K1355D-TRACE: DEADLINE-ARMED-GEN1"); } 81 => { if stage() != 3 || arg != t1.raw() || set().timer(t1, 100) != Ok(()) || set().snapshot().entry.map(|entry| entry.phase) != Some(Phase::DeadlinePending) { park(); } frame.x[0] = 0; set_stage(4); write_line("TURKOS/OS/K1355D-TRACE: TIMEOUT-PROVISIONAL-AT-DEADLINE"); } 82 => { if stage() != 4 || arg != t1.raw() || set().event(t1, 100) != Ok(()) || set().snapshot().timers != 0 { park(); } frame.x[0] = 0; set_stage(5); write_line("TURKOS/OS/K1355D-TRACE: EVENT-WINS-DEADLINE-TICK"); } 83 => { if stage() != 5 || arg != t1.raw() { park(); } let before = set().snapshot(); if set().timer(t1, 100) != Err(Error::Stale) || set().snapshot() != before { park(); } frame.x[0] = abi::AbiResult::Err(abi::Errno::BadFd).encode(); set_stage(6); write_line("TURKOS/OS/K1355D-TRACE: CANCELED-TIMER-CANNOT-RECOMPLETE"); } 84 => { if stage() != 6 || arg != t1.raw() || set().take(owner, t1) != Ok(Completion::Event) { park(); } frame.x[0] = 1; set_stage(7); write_line("TURKOS/OS/K1355D-TRACE: SINGLE-EVENT-COMPLETION-TAKEN"); } 85 => { if stage() != 7 || arg != 0 || set().arm(owner, 200) != Ok(t2) { park(); } frame.x[0] = t2.raw(); set_stage(8); write_line("TURKOS/OS/K1355D-TRACE: DEADLINE-ARMED-GEN2"); } 86 => { if stage() != 8 || arg != t1.raw() { park(); } let before = set().snapshot(); if set().timer(t1, 200) != Err(Error::Stale) || set().snapshot() != before { park(); } frame.x[0] = abi::AbiResult::Err(abi::Errno::BadFd).encode(); set_stage(9); write_line("TURKOS/OS/K1355D-TRACE: OLD-TIMER-REJECTED-AFTER-REARM"); } 87 => { if stage() != 9 || arg != t2.raw() || set().timer(t2, 200) != Ok(()) { park(); } frame.x[0] = 0; set_stage(10); } 88 => { if stage() != 10 || arg != t2.raw() || set().close_tick(t2, 200) != Ok(()) || set().snapshot().timers != 0 { park(); } frame.x[0] = 0; set_stage(11); write_line("TURKOS/OS/K1355D-TRACE: TIMEOUT-ONLY-TICK-COMPLETED"); } 89 => { if stage() != 11 || arg != t2.raw() || set().take(owner, t2) != Ok(Completion::Timeout) { park(); } frame.x[0] = 2; set_stage(12); write_line("TURKOS/OS/K1355D-TRACE: SINGLE-TIMEOUT-COMPLETION-TAKEN"); } 90 => { if stage() != 12 || arg != 0 || set().arm(owner, 300) != Ok(t3) { park(); } frame.x[0] = t3.raw(); set_stage(13); } 91 => { if stage() != 13 || arg != t3.raw() || set().event(t3, 300) != Ok(()) { park(); } frame.x[0] = 0; set_stage(14); write_line("TURKOS/OS/K1355D-TRACE: EVENT-FIRST-DEADLINE-TICK"); } 92 => { if stage() != 14 || arg != t3.raw() { park(); } let parent = process_table.record(owner) .unwrap_or_else(|_| park()).parent.unwrap_or_else(|| park()); let before = set().snapshot(); if set().take(parent, t3) != Err(Error::Access) || set().snapshot() != before { park(); } frame.x[0] = abi::AbiResult::Err(abi::Errno::Access).encode(); set_stage(15); write_line("TURKOS/OS/K1355D-TRACE: FOREIGN-TAKE-EACCES-NO-MUTATION"); } 93 => { if stage() != 15 || arg != t3.raw() { park(); } let before = set().snapshot(); if set().timer(t3, 300) != Err(Error::Stale) || set().snapshot() != before { park(); } frame.x[0] = abi::AbiResult::Err(abi::Errno::BadFd).encode(); set_stage(16); write_line("TURKOS/OS/K1355D-TRACE: EVENT-FIRST-TIMER-CANCELED"); } 94 => { if stage() != 16 || arg != t3.raw() || set().take(owner, t3) != Ok(Completion::Event) { park(); } frame.x[0] = 1; set_stage(17); write_line("TURKOS/OS/K1355D-TRACE: ALL-REGISTRATIONS-CLEANED"); } _ => { frame.x[0] = abi::AbiResult::Err(abi::Errno::NoSys).encode(); } } 0 } else if frame.x[7] == 2 && frame.x[8] == 3 { let request = turkos_os_user_runtime_abi::decode( frame.x[7], frame.x[8], core::array::from_fn(|i| frame.x[i])) .unwrap_or_else(|_| park()); let turkos_os_user_runtime_abi::Request::Exit(code) = request else { park(); }; if episode.case == 0 { if code != 0 || episode.calls != 2 || stage() != 19 { park(); } } else if code != 42 || episode.calls != 1 || stage() != 18 || set().snapshot().entry.is_some() { park(); } process_table.exit_current(code).unwrap_or_else(|_| park()); let outcome = Outcome::Exited(code); if process_table.record(episode.pid).map(|r| r.outcome) != Ok(Some(outcome)) { park(); } if episode.case == 1 { set_stage(19); } else { set_stage(20); } episode.outcome = Some(outcome); 1 } else { if frame.x[7] != abi::ABI_VERSION { park(); } let mut abi_frame = turkos_os_exceptions::exceptions::Frame::EMPTY; abi_frame.x = frame.x; let call = abi::decode_call(&abi_frame).unwrap_or_else(|_| park()); if call == abi::Call::Role { if episode.calls != 0 { park(); } if episode.case == 0 { if stage() != 0 { park(); } set_stage(1); write_line("TURKOS/OS/K1355D-TRACE: SHELL-READY-FOR-DEADLINES"); } else { if stage() != 1 { park(); } let parent = process_table.record(episode.pid) .unwrap_or_else(|_| park()).parent.unwrap_or_else(|| park()); if unsafe { read_volatile(core::ptr::addr_of!(OS_K1215A_INIT_PID)) } != Some(parent) { park(); } unsafe { core::ptr::write(core::ptr::addr_of_mut!(OS_K1355_DEADLINES), Some(DeadlineWait::new(episode.pid))); } set_stage(2); write_line("TURKOS/OS/K1355D-TRACE: CHILD-OWNS-DEADLINE-WAIT"); } episode.calls = 1; frame.x[0] = episode.case as u64; return 0; } if let abi::Call::RunChild(1) = call { if episode.case != 0 || episode.calls != 1 || stage() != 1 { park(); } return os_k1215a_run_child(episode as *mut _, frame as *mut _, episode.launch.kernel_root); } if let abi::Call::Write(request) = call { if episode.case != 1 || request.fd() != 1 || request.bytes() != 16 || stage() != 17 { park(); } let mut bytes = [0u8; 16]; process_table.copy_current_user_data(episode.pid, request.address(), &mut bytes).unwrap_or_else(|_| park()); if &bytes != b"WAIT:TIMER:PASS\n" { park(); } for byte in bytes { write_byte(byte); } frame.x[0] = 16; set_stage(18); write_line("TURKOS/OS/K1355D-TRACE: USER-WROTE-DEADLINE-RESULT"); return 0; } park() } }