// K1350 native EL0 level-triggered waitset readiness probe. use super::*; use turkos_kernel::runtime::process_table::{Outcome, ProcessId}; use turkos_kernel::runtime::syscall_abi as abi; use super::k1350_readiness_model::{Error, Interest, Kind, ObjectKey, Readiness, RegistrationId}; static mut OS_K1350_READINESS: Option> = None; static mut OS_K1350_STAGE: u8 = 0; const FIRST: u64 = (1u64 << 32) | 1; const SECOND: u64 = (1u64 << 32) | 2; const REUSED: u64 = (2u64 << 32) | 1; const CHANNEL: ObjectKey = ObjectKey {kind: Kind::Channel, id: 52}; const TASK: ObjectKey = ObjectKey {kind: Kind::Task, id: 53}; fn set() -> &'static mut Readiness { // Mode 61 runs on one CPU with IRQs masked and no concurrent waitset user. unsafe { (&mut *core::ptr::addr_of_mut!(OS_K1350_READINESS)) .as_mut().unwrap_or_else(|| park()) } } fn stage() -> u8 { unsafe { read_volatile(core::ptr::addr_of!(OS_K1350_STAGE)) } } fn set_stage(value: u8) { unsafe { write_volatile(core::ptr::addr_of_mut!(OS_K1350_STAGE), value); } } pub(super) fn begin() { unsafe { write_volatile(core::ptr::addr_of_mut!(OS_K1350_READINESS), None); } set_stage(0); } pub(super) fn finish() -> Result<(), &'static str> { let snapshot = set().snapshot(); if stage() != 20 || set().len() != 0 || snapshot.registrations.generations != [2, 1] || snapshot.registrations.entries != [None; 2] || snapshot.levels != [0; 2] { return Err("waitset readiness owner leak"); } unsafe { core::ptr::write(core::ptr::addr_of_mut!(OS_K1350_READINESS), 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() || frame.x[0] == u64::MAX { park(); } let owner = episode.pid; let arg = frame.x[0]; match frame.x[8] { 60 => { if stage() != 2 || arg != 0 { park(); } let id = set().register(owner, CHANNEL, Interest::READ_CLOSED) .unwrap_or_else(|_| park()); if id.raw() != FIRST { park(); } frame.x[0] = id.raw(); set_stage(3); write_line("TURKOS/OS/K1350R-TRACE: CHANNEL-REGISTERED"); } 61 => { if stage() != 3 || arg != 0 { park(); } let id = set().register(owner, TASK, Interest::CLOSED) .unwrap_or_else(|_| park()); if id.raw() != SECOND || set().len() != 2 { park(); } frame.x[0] = id.raw(); set_stage(4); write_line("TURKOS/OS/K1350R-TRACE: TASK-REGISTERED"); } 62 => { if stage() != 4 || arg != FIRST || set().poll(owner, RegistrationId::from_raw(FIRST)) != Ok(None) { park(); } frame.x[0] = 0; set_stage(5); write_line("TURKOS/OS/K1350R-TRACE: INITIAL-POLL-EMPTY"); } 63 => { if stage() != 5 || arg != FIRST || set().replace_level(CHANNEL, Interest::from_bits(3)) != Ok(()) || set().poll(owner, RegistrationId::from_raw(FIRST)) != Ok(Some(Interest::READ)) { park(); } frame.x[0] = 1; set_stage(6); write_line("TURKOS/OS/K1350R-TRACE: READ-FILTERS-UNRELATED-WRITE"); } 64 => { if stage() != 6 || arg != FIRST || set().poll(owner, RegistrationId::from_raw(FIRST)) != Ok(Some(Interest::READ)) { park(); } frame.x[0] = 1; set_stage(7); write_line("TURKOS/OS/K1350R-TRACE: REPEATED-POLL-KEEPS-LEVEL"); } 65 => { if stage() != 7 || arg != FIRST || set().replace_level(CHANNEL, Interest::from_bits(0)) != Ok(()) || set().poll(owner, RegistrationId::from_raw(FIRST)) != Ok(None) { park(); } frame.x[0] = 0; set_stage(8); write_line("TURKOS/OS/K1350R-TRACE: SOURCE-CLEAR-REMOVES-READINESS"); } 66 => { if stage() != 8 || arg != SECOND || set().replace_level(TASK, Interest::CLOSED) != Ok(()) || set().poll(owner, RegistrationId::from_raw(SECOND)) != Ok(Some(Interest::CLOSED)) { park(); } frame.x[0] = 4; set_stage(9); write_line("TURKOS/OS/K1350R-TRACE: TASK-CLOSED-READY"); } 67 => { if stage() != 9 || arg != SECOND { park(); } let parent = process_table.record(owner) .unwrap_or_else(|_| park()).parent.unwrap_or_else(|| park()); let before = set().snapshot(); if set().poll(parent, RegistrationId::from_raw(SECOND)) != Err(Error::Access) || set().snapshot() != before { park(); } frame.x[0] = abi::AbiResult::Err(abi::Errno::Access).encode(); set_stage(10); write_line("TURKOS/OS/K1350R-TRACE: FOREIGN-POLL-EACCES-NO-MUTATION"); } 68 => { if stage() != 10 || arg != 0 { park(); } let before = set().snapshot(); if set().replace_level(CHANNEL, Interest::from_bits(8)) != Err(Error::Invalid) || set().snapshot() != before { park(); } frame.x[0] = abi::AbiResult::Err(abi::Errno::Invalid).encode(); set_stage(11); write_line("TURKOS/OS/K1350R-TRACE: INVALID-LEVEL-EINVAL-NO-MUTATION"); } 69 => { if stage() != 11 || arg != FIRST || set().unregister(owner, RegistrationId::from_raw(FIRST)) != Ok(()) { park(); } frame.x[0] = 0; set_stage(12); } 70 => { if stage() != 12 || arg != FIRST { park(); } let before = set().snapshot(); if set().poll(owner, RegistrationId::from_raw(FIRST)) != Err(Error::BadFd) || set().snapshot() != before { park(); } frame.x[0] = abi::AbiResult::Err(abi::Errno::BadFd).encode(); set_stage(13); write_line("TURKOS/OS/K1350R-TRACE: STALE-POLL-EBADF-NO-MUTATION"); } 71 => { if stage() != 13 || arg != 0 { park(); } let id = set().register(owner, ObjectKey {kind: Kind::File, id: 54}, Interest::READ) .unwrap_or_else(|_| park()); if id.raw() != REUSED || set().poll(owner, id) != Ok(None) { park(); } frame.x[0] = id.raw(); set_stage(14); write_line("TURKOS/OS/K1350R-TRACE: REUSED-SLOT-STARTS-CLEAR"); } 72 | 73 => { let (raw, next) = match (frame.x[8], stage()) { (72, 14) => (SECOND, 15), (73, 15) => (REUSED, 16), _ => park(), }; if arg != raw || set().unregister(owner, RegistrationId::from_raw(raw)) != Ok(()) { park(); } frame.x[0] = 0; set_stage(next); } _ => { 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() != 17 || set().len() != 0 { 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/K1350R-TRACE: SHELL-READY-FOR-READINESS"); } 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_K1350_READINESS), Some(Readiness::new(episode.pid))); } set_stage(2); write_line("TURKOS/OS/K1350R-TRACE: CHILD-OWNS-READINESS"); } 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() != 16 { 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:LEVEL:PASS\n" { park(); } for byte in bytes { write_byte(byte); } frame.x[0] = 16; set_stage(17); write_line("TURKOS/OS/K1350R-TRACE: USER-WROTE-READINESS-RESULT"); return 0; } park() } }