From dd7ac81b284668ba0b6983d38ab2f53d0485ff98 Mon Sep 17 00:00:00 2001 From: occheung Date: Thu, 5 Sep 2024 11:59:23 +0800 Subject: [PATCH] runtime mgmt: reorganize uses --- artiq/firmware/runtime/mgmt.rs | 43 +++++++++++++++++----------------- 1 file changed, 21 insertions(+), 22 deletions(-) diff --git a/artiq/firmware/runtime/mgmt.rs b/artiq/firmware/runtime/mgmt.rs index d15a5395e..0244a44f1 100644 --- a/artiq/firmware/runtime/mgmt.rs +++ b/artiq/firmware/runtime/mgmt.rs @@ -1,6 +1,6 @@ use core::cell::RefCell; -use board_artiq::drtio_routing; +use board_artiq::drtio_routing::RoutingTable; use io::{ProtoRead, Write, Error as IoError}; use mgmt_proto::*; use sched::{Io, Mutex, TcpListener, TcpStream, Error as SchedError}; @@ -19,10 +19,10 @@ mod local_coremgmt { use log::LevelFilter; use board_misoc::{config, mem, spiflash}; - use io::{Write, ProtoWrite, Error as IoError}; + use io::ProtoWrite; use logger_artiq::BufferLogger; - use mgmt_proto::{Error, Reply}; - use sched::{Io, TcpStream, Error as SchedError}; + + use super::*; pub fn get_log(io: &Io, stream: &mut TcpStream) -> Result<(), Error> { @@ -177,13 +177,12 @@ mod remote_coremgmt { use alloc::{string::String, vec::Vec}; use log::LevelFilter; - use board_artiq::{drtioaux, drtioaux::Packet, drtio_routing}; - use io::{Cursor, ProtoWrite}; - use mgmt_proto::{Error, Reply}; + use board_artiq::{drtioaux, drtioaux::Packet}; + use io::ProtoWrite; use rtio_mgt::drtio; - use sched::{Io, Mutex, TcpStream, Error as SchedError}; use proto_artiq::drtioaux_proto::MASTER_PAYLOAD_MAX_SIZE; + use super::*; impl From for Error { fn from(_value: drtio::Error) -> Error { @@ -193,7 +192,7 @@ mod remote_coremgmt { pub fn get_log(io: &Io, aux_mutex: &Mutex, ddma_mutex: &Mutex, subkernel_mutex: &Mutex, - routing_table: &drtio_routing::RoutingTable, linkno: u8, + routing_table: &RoutingTable, linkno: u8, destination: u8, stream: &mut TcpStream) -> Result<(), Error> { let mut buffer = String::new(); loop { @@ -226,7 +225,7 @@ mod remote_coremgmt { pub fn clear_log(io: &Io, aux_mutex: &Mutex, ddma_mutex: &Mutex, subkernel_mutex: &Mutex, - routing_table: &drtio_routing::RoutingTable, linkno: u8, + routing_table: &RoutingTable, linkno: u8, destination: u8, stream: &mut TcpStream) -> Result<(), Error> { let reply = drtio::aux_transact(io, aux_mutex, ddma_mutex, subkernel_mutex, routing_table, linkno, &Packet::CoreMgmtClearLogRequest { destination } @@ -252,7 +251,7 @@ mod remote_coremgmt { pub fn pull_log(io: &Io, aux_mutex: &Mutex, ddma_mutex: &Mutex, subkernel_mutex: &Mutex, - routing_table: &drtio_routing::RoutingTable, linkno: u8, + routing_table: &RoutingTable, linkno: u8, destination: u8, stream: &mut TcpStream) -> Result<(), Error> { loop { let reply = drtio::aux_transact(io, aux_mutex, ddma_mutex, subkernel_mutex, routing_table, linkno, @@ -277,7 +276,7 @@ mod remote_coremgmt { pub fn set_log_filter(io: &Io, aux_mutex: &Mutex, ddma_mutex: &Mutex, subkernel_mutex: &Mutex, - routing_table: &drtio_routing::RoutingTable, linkno: u8, + routing_table: &RoutingTable, linkno: u8, destination: u8, stream: &mut TcpStream, level: LevelFilter) -> Result<(), Error> { let reply = drtio::aux_transact(io, aux_mutex, ddma_mutex, subkernel_mutex, routing_table, linkno, &Packet::CoreMgmtSetLogLevelRequest { destination, log_level: level as u8 } @@ -303,7 +302,7 @@ mod remote_coremgmt { pub fn set_uart_log_filter(io: &Io, aux_mutex: &Mutex, ddma_mutex: &Mutex, subkernel_mutex: &Mutex, - routing_table: &drtio_routing::RoutingTable, linkno: u8, + routing_table: &RoutingTable, linkno: u8, destination: u8, stream: &mut TcpStream, level: LevelFilter) -> Result<(), Error> { let reply = drtio::aux_transact(io, aux_mutex, ddma_mutex, subkernel_mutex, routing_table, linkno, &Packet::CoreMgmtSetUartLogLevelRequest { destination, log_level: level as u8 } @@ -329,7 +328,7 @@ mod remote_coremgmt { pub fn config_read(io: &Io, aux_mutex: &Mutex, ddma_mutex: &Mutex, subkernel_mutex: &Mutex, - routing_table: &drtio_routing::RoutingTable, linkno: u8, + routing_table: &RoutingTable, linkno: u8, destination: u8, stream: &mut TcpStream, key: &String) -> Result<(), Error> { let mut config_key: [u8; MASTER_PAYLOAD_MAX_SIZE] = [0; MASTER_PAYLOAD_MAX_SIZE]; let len = key.len(); @@ -376,7 +375,7 @@ mod remote_coremgmt { pub fn config_write(io: &Io, aux_mutex: &Mutex, ddma_mutex: &Mutex, subkernel_mutex: &Mutex, - routing_table: &drtio_routing::RoutingTable, linkno: u8, + routing_table: &RoutingTable, linkno: u8, destination: u8, stream: &mut TcpStream, key: &String, value: &Vec) -> Result<(), Error> { let mut message = Vec::with_capacity(key.len() + value.len() + 4 * 2); message.write_string(key).unwrap(); @@ -411,7 +410,7 @@ mod remote_coremgmt { pub fn config_remove(io: &Io, aux_mutex: &Mutex, ddma_mutex: &Mutex, subkernel_mutex: &Mutex, - routing_table: &drtio_routing::RoutingTable, linkno: u8, + routing_table: &RoutingTable, linkno: u8, destination: u8, stream: &mut TcpStream, key: &String) -> Result<(), Error> { let mut config_key: [u8; MASTER_PAYLOAD_MAX_SIZE] = [0; MASTER_PAYLOAD_MAX_SIZE]; let len = key.len(); @@ -444,7 +443,7 @@ mod remote_coremgmt { pub fn config_erase(io: &Io, aux_mutex: &Mutex, ddma_mutex: &Mutex, subkernel_mutex: &Mutex, - routing_table: &drtio_routing::RoutingTable, linkno: u8, + routing_table: &RoutingTable, linkno: u8, destination: u8, stream: &mut TcpStream) -> Result<(), Error> { let reply = drtio::aux_transact(io, aux_mutex, ddma_mutex, subkernel_mutex, routing_table, linkno, &Packet::CoreMgmtConfigEraseRequest { @@ -471,7 +470,7 @@ mod remote_coremgmt { pub fn reboot(io: &Io, aux_mutex: &Mutex, ddma_mutex: &Mutex, subkernel_mutex: &Mutex, - routing_table: &drtio_routing::RoutingTable, linkno: u8, + routing_table: &RoutingTable, linkno: u8, destination: u8, stream: &mut TcpStream) -> Result<(), Error> { let reply = drtio::aux_transact(io, aux_mutex, ddma_mutex, subkernel_mutex, routing_table, linkno, &Packet::CoreMgmtRebootRequest { @@ -498,7 +497,7 @@ mod remote_coremgmt { pub fn debug_allocator(io: &Io, aux_mutex: &Mutex, ddma_mutex: &Mutex, subkernel_mutex: &Mutex, - routing_table: &drtio_routing::RoutingTable, linkno: u8, + routing_table: &RoutingTable, linkno: u8, destination: u8, _stream: &mut TcpStream) -> Result<(), Error> { let reply = drtio::aux_transact(io, aux_mutex, ddma_mutex, subkernel_mutex, routing_table, linkno, &Packet::CoreMgmtAllocatorDebugRequest { @@ -520,7 +519,7 @@ mod remote_coremgmt { pub fn flash(io: &Io, aux_mutex: &Mutex, ddma_mutex: &Mutex, subkernel_mutex: &Mutex, - routing_table: &drtio_routing::RoutingTable, linkno: u8, + routing_table: &RoutingTable, linkno: u8, destination: u8, stream: &mut TcpStream, image: &[u8]) -> Result<(), Error> { match drtio::partition_data(&image, |slice, status, len: usize| { @@ -582,7 +581,7 @@ macro_rules! process { } fn worker(io: &Io, _aux_mutex: &Mutex, _ddma_mutex: &Mutex, _subkernel_mutex: &Mutex, - _routing_table: &drtio_routing::RoutingTable, stream: &mut TcpStream) -> Result<(), Error> { + _routing_table: &RoutingTable, stream: &mut TcpStream) -> Result<(), Error> { read_magic(stream)?; let _destination = stream.read_u8()?; Write::write_all(stream, "e".as_bytes())?; @@ -606,7 +605,7 @@ fn worker(io: &Io, _aux_mutex: &Mutex, _ddma_mutex: &Mutex, _subkernel_mutex: &M } } -pub fn thread(io: Io, aux_mutex: &Mutex, ddma_mutex: &Mutex, subkernel_mutex: &Mutex, routing_table: &Urc>) { +pub fn thread(io: Io, aux_mutex: &Mutex, ddma_mutex: &Mutex, subkernel_mutex: &Mutex, routing_table: &Urc>) { let listener = TcpListener::new(&io, 8192); listener.listen(1380).expect("mgmt: cannot listen"); info!("management interface active");