110 lines
4.1 KiB
Rust
110 lines
4.1 KiB
Rust
use alloc::vec::Vec;
|
|
|
|
use cslice::CSlice;
|
|
|
|
use super::{Message, SubkernelStatus, KERNEL_CHANNEL_0TO1, KERNEL_CHANNEL_1TO0};
|
|
use crate::{artiq_raise, rpc::send_args};
|
|
|
|
pub extern "C" fn load_run(id: u32, run: bool) {
|
|
unsafe {
|
|
KERNEL_CHANNEL_1TO0
|
|
.as_mut()
|
|
.unwrap()
|
|
.send(Message::SubkernelLoadRunRequest { id: id, run: run });
|
|
}
|
|
match unsafe { KERNEL_CHANNEL_0TO1.as_mut().unwrap() }.recv() {
|
|
Message::SubkernelLoadRunReply { succeeded: true } => (),
|
|
Message::SubkernelLoadRunReply { succeeded: false } => {
|
|
artiq_raise!("SubkernelError", "Error loading or running the subkernel")
|
|
}
|
|
_ => panic!("Expected SubkernelLoadRunReply after SubkernelLoadRunRequest!"),
|
|
}
|
|
}
|
|
|
|
pub extern "C" fn await_finish(id: u32, timeout: u64) {
|
|
unsafe {
|
|
KERNEL_CHANNEL_1TO0
|
|
.as_mut()
|
|
.unwrap()
|
|
.send(Message::SubkernelAwaitFinishRequest {
|
|
id: id,
|
|
timeout: timeout,
|
|
});
|
|
}
|
|
match unsafe { KERNEL_CHANNEL_0TO1.as_mut().unwrap() }.recv() {
|
|
Message::SubkernelAwaitFinishReply {
|
|
status: SubkernelStatus::NoError,
|
|
} => (),
|
|
Message::SubkernelAwaitFinishReply {
|
|
status: SubkernelStatus::IncorrectState,
|
|
} => artiq_raise!("SubkernelError", "Subkernel not running"),
|
|
Message::SubkernelAwaitFinishReply {
|
|
status: SubkernelStatus::Timeout,
|
|
} => artiq_raise!("SubkernelError", "Subkernel timed out"),
|
|
Message::SubkernelAwaitFinishReply {
|
|
status: SubkernelStatus::CommLost,
|
|
} => artiq_raise!("SubkernelError", "Lost communication with satellite"),
|
|
Message::SubkernelAwaitFinishReply {
|
|
status: SubkernelStatus::OtherError,
|
|
} => artiq_raise!("SubkernelError", "An error occurred during subkernel operation"),
|
|
_ => panic!("expected SubkernelAwaitFinishReply after SubkernelAwaitFinishRequest"),
|
|
}
|
|
}
|
|
|
|
pub extern "C" fn send_message(id: u32, count: u8, tag: &CSlice<u8>, data: *const *const ()) {
|
|
let mut buffer = Vec::<u8>::new();
|
|
send_args(&mut buffer, 0, tag.as_ref(), data).expect("RPC encoding failed");
|
|
// overwrite service tag, include how many tags are in the message
|
|
buffer[3] = count;
|
|
unsafe {
|
|
KERNEL_CHANNEL_1TO0.as_mut().unwrap().send(Message::SubkernelMsgSend {
|
|
id: id,
|
|
data: buffer[3..].to_vec(),
|
|
});
|
|
}
|
|
match unsafe { KERNEL_CHANNEL_0TO1.as_mut().unwrap() }.recv() {
|
|
Message::SubkernelMsgSent => (),
|
|
_ => panic!("expected SubkernelMsgSent after SubkernelMsgSend"),
|
|
}
|
|
}
|
|
|
|
pub extern "C" fn await_message(id: u32, timeout: u64, min: u8, max: u8) {
|
|
unsafe {
|
|
KERNEL_CHANNEL_1TO0
|
|
.as_mut()
|
|
.unwrap()
|
|
.send(Message::SubkernelMsgRecvRequest {
|
|
id: id,
|
|
timeout: timeout,
|
|
});
|
|
}
|
|
match unsafe { KERNEL_CHANNEL_0TO1.as_mut().unwrap() }.recv() {
|
|
Message::SubkernelMsgRecvReply {
|
|
status: SubkernelStatus::NoError,
|
|
count
|
|
} => {
|
|
if min > count || count > max {
|
|
artiq_raise!("SubkernelError", "Received more or less arguments than required")
|
|
}
|
|
},
|
|
Message::SubkernelMsgRecvReply {
|
|
status: SubkernelStatus::IncorrectState,
|
|
..
|
|
} => artiq_raise!("SubkernelError", "Subkernel not running"),
|
|
Message::SubkernelMsgRecvReply {
|
|
status: SubkernelStatus::Timeout,
|
|
..
|
|
} => artiq_raise!("SubkernelError", "Subkernel timed out"),
|
|
Message::SubkernelMsgRecvReply {
|
|
status: SubkernelStatus::CommLost,
|
|
..
|
|
} => artiq_raise!("SubkernelError", "Lost communication with satellite"),
|
|
Message::SubkernelMsgRecvReply {
|
|
status: SubkernelStatus::OtherError,
|
|
..
|
|
} => artiq_raise!("SubkernelError", "An error occurred during subkernel operation"),
|
|
_ => panic!("expected SubkernelMsgRecvReply after SubkernelMsgRecvRequest"),
|
|
}
|
|
// RpcRecvRequest should be called after this to receive message data
|
|
}
|