diff --git a/src/runtime/src/comms.rs b/src/runtime/src/comms.rs index 23ba645..d33e618 100644 --- a/src/runtime/src/comms.rs +++ b/src/runtime/src/comms.rs @@ -159,7 +159,7 @@ async fn write_exception_string(stream: &TcpStream, s: CSlice<'static, u8>) -> R Ok(()) } -async fn handle_run_kernel(stream: Option<&TcpStream>, control: &Rc>) -> Result<()> { +async fn handle_run_kernel(stream: Option<&TcpStream>, control: &Rc>, up_destinations: &Rc>) -> Result<()> { control.borrow_mut().tx.async_send(kernel::Message::StartRequest).await; loop { let reply = control.borrow_mut().rx.async_recv().await; @@ -290,6 +290,10 @@ async fn handle_run_kernel(stream: Option<&TcpStream>, control: &Rc { + let result = up_destinations.borrow()[destination as usize]; + control.borrow_mut().tx.async_send(kernel::Message::UpDestinationsReply(result)).await; + } _ => { panic!("unexpected message from core1 while kernel was running: {:?}", reply); } @@ -331,7 +335,7 @@ async fn load_kernel(buffer: &Vec, control: &Rc>, s } } -async fn handle_connection(stream: &mut TcpStream, control: Rc>) -> Result<()> { +async fn handle_connection(stream: &mut TcpStream, control: Rc>, up_destinations: &Rc>) -> Result<()> { stream.set_ack_delay(None); if !expect(stream, b"ARTIQ coredev\n").await? { @@ -354,7 +358,7 @@ async fn handle_connection(stream: &mut TcpStream, control: Rc { - handle_run_kernel(Some(stream), &control).await?; + handle_run_kernel(Some(stream), &control, &up_destinations).await?; }, _ => { error!("unexpected request from host: {:?}", request); @@ -427,7 +431,7 @@ pub fn main(timer: GlobalTimer, cfg: Config) { info!("Loading startup kernel..."); if let Ok(()) = task::block_on(load_kernel(&buffer, &control, None)) { info!("Starting startup kernel..."); - let _ = task::block_on(handle_run_kernel(None, &control)); + let _ = task::block_on(handle_run_kernel(None, &control, &up_destinations)); info!("Startup kernel finished!"); } else { error!("Error loading startup kernel!"); @@ -452,13 +456,14 @@ pub fn main(timer: GlobalTimer, cfg: Config) { let idle_kernel = idle_kernel.clone(); let connection = connection.clone(); let terminate = terminate.clone(); + let up_destinations = up_destinations.clone(); // we make sure the value of terminate is 0 before we start let _ = terminate.try_wait(); task::spawn(async move { select_biased! { _ = (async { - let _ = handle_connection(&mut stream, control.clone()) + let _ = handle_connection(&mut stream, control.clone(), &up_destinations) .await .map_err(|e| warn!("connection terminated: {}", e)); if let Some(buffer) = &*idle_kernel { @@ -466,7 +471,7 @@ pub fn main(timer: GlobalTimer, cfg: Config) { let _ = load_kernel(&buffer, &control, None) .await.map_err(|_| warn!("error loading idle kernel")); info!("Running idle kernel"); - let _ = handle_run_kernel(None, &control) + let _ = handle_run_kernel(None, &control, &up_destinations) .await.map_err(|_| warn!("error running idle kernel")); info!("Idle kernel terminated"); } diff --git a/src/runtime/src/kernel/api.rs b/src/runtime/src/kernel/api.rs index 1c55d3e..7b4c01d 100644 --- a/src/runtime/src/kernel/api.rs +++ b/src/runtime/src/kernel/api.rs @@ -13,6 +13,7 @@ use crate::i2c; use super::rpc::{rpc_send, rpc_send_async, rpc_recv}; use super::dma; use super::cache; +use super::rtio as kernel_rtio; extern "C" { @@ -87,7 +88,7 @@ pub fn resolve(required: &[u8]) -> Option { // rtio api!(rtio_init = rtio::init), - api!(rtio_get_destination_status = rtio::get_destination_status), + api!(rtio_get_destination_status = kernel_rtio::get_destination_status), api!(rtio_get_counter = rtio::get_counter), api!(rtio_output = rtio::output), api!(rtio_output_wide = rtio::output_wide), diff --git a/src/runtime/src/kernel/mod.rs b/src/runtime/src/kernel/mod.rs index 300acc5..c12cdb9 100644 --- a/src/runtime/src/kernel/mod.rs +++ b/src/runtime/src/kernel/mod.rs @@ -7,6 +7,7 @@ use crate::eh_artiq; mod control; pub use control::Control; pub mod core1; +pub mod rtio; mod api; mod rpc; mod dma; @@ -47,6 +48,9 @@ pub enum Message { DmaEraseRequest(String), DmaGetRequest(String), DmaGetReply(Option<(Vec, i64)>), + + UpDestinationsRequest(i32), + UpDestinationsReply(bool), } static CHANNEL_0TO1: Mutex>> = Mutex::new(None); diff --git a/src/runtime/src/kernel/rtio.rs b/src/runtime/src/kernel/rtio.rs new file mode 100644 index 0000000..0e2b679 --- /dev/null +++ b/src/runtime/src/kernel/rtio.rs @@ -0,0 +1,19 @@ +use super::{KERNEL_CHANNEL_0TO1, KERNEL_CHANNEL_1TO0, Message}; + +pub extern fn get_destination_status(destination: i32) -> bool { + #[cfg(has_drtio)] + if destination > 0 && destination < 255 { + let reply = unsafe { + let core1_rx = KERNEL_CHANNEL_0TO1.as_mut().unwrap(); + let core1_tx = KERNEL_CHANNEL_1TO0.as_mut().unwrap(); + core1_tx.send(Message::UpDestinationsRequest(destination)); + core1_rx.recv() + }; + return match reply { + Message::UpDestinationsReply(true) => true, + _ => false + }; + } + + destination == 0 +} \ No newline at end of file diff --git a/src/runtime/src/rtio_acp.rs b/src/runtime/src/rtio_acp.rs index f7987e1..4c1f33b 100644 --- a/src/runtime/src/rtio_acp.rs +++ b/src/runtime/src/rtio_acp.rs @@ -57,11 +57,6 @@ pub extern fn init() { } } -pub extern fn get_destination_status(destination: i32) -> bool { - // TODO - destination == 0 -} - pub extern fn get_counter() -> i64 { unsafe { csr::rtio::counter_update_write(1); diff --git a/src/runtime/src/rtio_csr.rs b/src/runtime/src/rtio_csr.rs index 1edd606..761dcbe 100644 --- a/src/runtime/src/rtio_csr.rs +++ b/src/runtime/src/rtio_csr.rs @@ -25,11 +25,6 @@ pub extern fn init() { } } -pub extern fn get_destination_status(destination: i32) -> bool { - // TODO - destination == 0 -} - pub extern fn get_counter() -> i64 { unsafe { csr::rtio::counter_update_write(1);