// K1345 native EL0 typed waitset registration probe. use super::*; use turkos_kernel::runtime::process_table::{Outcome, ProcessId}; use turkos_kernel::runtime::syscall_abi as abi; use super::k1345_waitset_model::{Error, Interest, Kind, ObjectKey, RegistrationId, Waitset}; static mut OS_K1345_WAITSET: Option> = None; static mut OS_K1345_STAGE: u8 = 0; const FIRST: u64 = (1u64 << 32) | 1; const SECOND: u64 = (1u64 << 32) | 2; const REUSED: u64 = (2u64 << 32) | 1; fn set() -> &'static mut Waitset { // Mode 60 runs on one CPU with IRQs masked and no concurrent waitset user. unsafe { (&mut *core::ptr::addr_of_mut!(OS_K1345_WAITSET)) .as_mut().unwrap_or_else(|| park()) } } fn stage() -> u8 { unsafe { read_volatile(core::ptr::addr_of!(OS_K1345_STAGE)) } } fn set_stage(value: u8) { unsafe { write_volatile(core::ptr::addr_of_mut!(OS_K1345_STAGE), value); } } pub(super) fn begin() { unsafe { write_volatile(core::ptr::addr_of_mut!(OS_K1345_WAITSET), None); } set_stage(0); } pub(super) fn finish() -> Result<(), &'static str> { let snapshot = set().snapshot(); if stage() != 16 || set().len() != 0 || snapshot.generations != [2, 1] || snapshot.entries != [None; 2] { return Err("waitset registration owner leak"); } unsafe { core::ptr::write(core::ptr::addr_of_mut!(OS_K1345_WAITSET), 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; match frame.x[8] { 50 => { if stage() != 2 || frame.x[0] != 0 { park(); } let id = set().register(owner, ObjectKey {kind: Kind::Channel, id: 52}, Interest::READ_CLOSED).unwrap_or_else(|_| park()); if id.raw() != FIRST { park(); } frame.x[0] = id.raw(); set_stage(3); write_line("TURKOS/OS/K1345W-TRACE: CHANNEL-REGISTERED-GEN1"); 0 } 51 => { if stage() != 3 || frame.x[0] != 0 { park(); } let id = set().register(owner, ObjectKey {kind: Kind::Task, id: 53}, 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/K1345W-TRACE: TASK-REGISTERED-GEN1"); 0 } 52 => { if stage() != 4 || frame.x[0] != 0 { park(); } let before = set().snapshot(); if set().register(owner, ObjectKey {kind: Kind::Channel, id: 52}, Interest::WRITE) != Err(Error::Exists) || set().snapshot() != before { park(); } frame.x[0] = abi::AbiResult::Err(abi::Errno::Exist).encode(); set_stage(5); write_line("TURKOS/OS/K1345W-TRACE: DUPLICATE-EEXIST-NO-MUTATION"); 0 } 59 => { if stage() != 5 || frame.x[0] != 0 { park(); } let parent = process_table.record(owner) .unwrap_or_else(|_| park()).parent.unwrap_or_else(|| park()); let before = set().snapshot(); if set().register(parent, ObjectKey {kind: Kind::File, id: 54}, Interest::READ) != Err(Error::Access) || set().snapshot() != before { park(); } frame.x[0] = abi::AbiResult::Err(abi::Errno::Access).encode(); set_stage(6); write_line("TURKOS/OS/K1345W-TRACE: FOREIGN-OWNER-EACCES-NO-MUTATION"); 0 } 53 => { let (raw, object, next) = match stage() { 6 => (FIRST, ObjectKey {kind: Kind::Channel, id: 52}, 7), 7 => (SECOND, ObjectKey {kind: Kind::Task, id: 53}, 8), _ => park(), }; if frame.x[0] != raw || set().query(owner, RegistrationId::from_raw(raw)) .map(|row| row.0) != Ok(object) { park(); } frame.x[0] = object.id; set_stage(next); write_line("TURKOS/OS/K1345W-TRACE: TYPED-OBJECT-QUERY"); 0 } 54 => { if stage() != 8 || frame.x[0] != FIRST || set().unregister(owner, RegistrationId::from_raw(FIRST)) != Ok(()) { park(); } frame.x[0] = 0; set_stage(9); write_line("TURKOS/OS/K1345W-TRACE: FIRST-UNREGISTERED"); 0 } 55 => { if stage() != 9 || frame.x[0] != FIRST { park(); } let before = set().snapshot(); if set().query(owner, RegistrationId::from_raw(FIRST)) != Err(Error::BadFd) || set().snapshot() != before { park(); } frame.x[0] = abi::AbiResult::Err(abi::Errno::BadFd).encode(); set_stage(10); write_line("TURKOS/OS/K1345W-TRACE: STALE-REGISTRATION-EBADF"); 0 } 56 => { if stage() != 10 || frame.x[0] != 0 { park(); } let id = set().register(owner, ObjectKey {kind: Kind::File, id: 54}, Interest::READ).unwrap_or_else(|_| park()); if id.raw() != REUSED { park(); } frame.x[0] = id.raw(); set_stage(11); write_line("TURKOS/OS/K1345W-TRACE: SLOT-REUSED-GEN2"); 0 } 57 | 58 => { let (raw, next) = match (frame.x[8], stage()) { (57, 11) => (SECOND, 12), (58, 12) => (REUSED, 13), _ => park(), }; if frame.x[0] != raw || set().unregister(owner, RegistrationId::from_raw(raw)) != Ok(()) { park(); } frame.x[0] = 0; set_stage(next); write_line("TURKOS/OS/K1345W-TRACE: OWNER-REGISTRATION-RELEASED"); 0 } _ => { 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() != 15 { park(); } } else if code != 42 || episode.calls != 1 || stage() != 14 || 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(15); } else { set_stage(16); } 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/K1345W-TRACE: SHELL-READY-FOR-WAITSET"); } 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_K1345_WAITSET), Some(Waitset::new(episode.pid))); } set_stage(2); write_line("TURKOS/OS/K1345W-TRACE: CHILD-OWNS-WAITSET"); } 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() != 13 { 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:2,1:EBADF!\n" { park(); } for byte in bytes { write_byte(byte); } frame.x[0] = 16; set_stage(14); write_line("TURKOS/OS/K1345W-TRACE: USER-WROTE-WAITSET-RESULT"); return 0; } park() } }