artiq/artiq/firmware/runtime/kern_hwreq.rs

use std::io;
use kernel_proto as kern;
use sched::Io;
use session::{kern_acknowledge, kern_send};
#[cfg(has_rtio_core)]
use rtio_mgt;

#[cfg(has_drtio)]
mod drtio_i2c {
    use drtioaux;

    fn basic_reply(nodeno: u8) -> Result<(), ()> {
        match drtioaux::hw::recv_timeout(nodeno, None) {
            Ok(drtioaux::Packet::I2cBasicReply { succeeded }) => {
                if succeeded { Ok(()) } else { Err(()) }
            }
            Ok(_) => {
                error!("received unexpected aux packet");
                Err(())
            }
            Err(e) => {
                error!("aux packet error ({})", e);
                Err(())
            }
        }
    }

    pub fn start(nodeno: u8, busno: u8) -> Result<(), ()> {
        let request = drtioaux::Packet::I2cStartRequest { busno: busno };
        if drtioaux::hw::send(nodeno, &request).is_err() {
            return Err(())
        }
        basic_reply(nodeno)
    }

    pub fn restart(nodeno: u8, busno: u8) -> Result<(), ()> {
        let request = drtioaux::Packet::I2cRestartRequest { busno: busno };
        if drtioaux::hw::send(nodeno, &request).is_err() {
            return Err(())
        }
        basic_reply(nodeno)
    }

    pub fn stop(nodeno: u8, busno: u8) -> Result<(), ()> {
        let request = drtioaux::Packet::I2cStopRequest { busno: busno };
        if drtioaux::hw::send(nodeno, &request).is_err() {
            return Err(())
        }
        basic_reply(nodeno)
    }

    pub fn write(nodeno: u8, busno: u8, data: u8) -> Result<bool, ()> {
        let request = drtioaux::Packet::I2cWriteRequest {
            busno: busno,
            data: data
        };
        if drtioaux::hw::send(nodeno, &request).is_err() {
            return Err(())
        }
        match drtioaux::hw::recv_timeout(nodeno, None) {
            Ok(drtioaux::Packet::I2cWriteReply { succeeded, ack }) => {
                if succeeded { Ok(ack) } else { Err(()) }
            }
            Ok(_) => {
                error!("received unexpected aux packet");
                Err(())
            }
            Err(e) => {
                error!("aux packet error ({})", e);
                Err(())
            }
        }
    }

    pub fn read(nodeno: u8, busno: u8, ack: bool) -> Result<u8, ()> {
        let request = drtioaux::Packet::I2cReadRequest {
            busno: busno,
            ack: ack
        };
        if drtioaux::hw::send(nodeno, &request).is_err() {
            return Err(())
        }
        match drtioaux::hw::recv_timeout(nodeno, None) {
            Ok(drtioaux::Packet::I2cReadReply { succeeded, data }) => {
                if succeeded { Ok(data) } else { Err(()) }
            }
            Ok(_) => {
                error!("received unexpected aux packet");
                Err(())
            }
            Err(e) => {
                error!("aux packet error ({})", e);
                Err(())
            }
        }
    }
}

#[cfg(not(has_drtio))]
mod drtio_i2c {
    pub fn start(_nodeno: u8, _busno: u8) -> Result<(), ()> {
        Err(())
    }

    pub fn restart(_nodeno: u8, _busno: u8) -> Result<(), ()> {
        Err(())
    }

    pub fn stop(_nodeno: u8, _busno: u8) -> Result<(), ()> {
        Err(())
    }

    pub fn write(_nodeno: u8, _busno: u8, _data: u8) -> Result<bool, ()> {
        Err(())
    }

    pub fn read(_nodeno: u8, _busno: u8, _ack: bool) -> Result<u8, ()> {
        Err(())
    }
}

mod i2c {
    use board_artiq::i2c as local_i2c;
    use super::drtio_i2c;

    pub fn start(busno: u32) -> Result<(), ()> {
        let nodeno = (busno >> 16) as u8;
        let node_busno = busno as u8;
        if nodeno == 0 {
            local_i2c::start(node_busno)
        } else {
            drtio_i2c::start(nodeno, node_busno)
        }
    }

    pub fn restart(busno: u32) -> Result<(), ()> {
        let nodeno = (busno >> 16) as u8;
        let node_busno = busno as u8;
        if nodeno == 0 {
            local_i2c::restart(node_busno)
        } else {
            drtio_i2c::restart(nodeno, node_busno)
        }
    }

    pub fn stop(busno: u32) -> Result<(), ()> {
        let nodeno = (busno >> 16) as u8;
        let node_busno = busno as u8;
        if nodeno == 0 {
            local_i2c::stop(node_busno)
        } else {
            drtio_i2c::stop(nodeno, node_busno)
        }
    }

    pub fn write(busno: u32, data: u8) -> Result<bool, ()> {
        let nodeno = (busno >> 16 )as u8;
        let node_busno = busno as u8;
        if nodeno == 0 {
            local_i2c::write(node_busno, data)
        } else {
            drtio_i2c::write(nodeno, node_busno, data)
        }
    }

    pub fn read(busno: u32, ack: bool) -> Result<u8, ()> {
        let nodeno = (busno >> 16) as u8;
        let node_busno = busno as u8;
        if nodeno == 0 {
            local_i2c::read(node_busno, ack)
        } else {
            drtio_i2c::read(nodeno, node_busno, ack)
        }
    }
}

#[cfg(has_drtio)]
mod drtio_spi {
    use drtioaux;

    fn basic_reply(nodeno: u8) -> Result<(), ()> {
        match drtioaux::hw::recv_timeout(nodeno, None) {
            Ok(drtioaux::Packet::SpiBasicReply { succeeded }) => {
                if succeeded { Ok(()) } else { Err(()) }
            }
            Ok(_) => {
                error!("received unexpected aux packet");
                Err(())
            }
            Err(e) => {
                error!("aux packet error ({})", e);
                Err(())
            }
        }
    }

    pub fn set_config(nodeno: u8, busno: u8, flags: u8, write_div: u8, read_div: u8) -> Result<(), ()> {
        let request = drtioaux::Packet::SpiSetConfigRequest {
            busno: busno,
            flags: flags,
            write_div: write_div,
            read_div: read_div
        };
        if drtioaux::hw::send(nodeno, &request).is_err() {
            return Err(())
        }
        basic_reply(nodeno)
    }

    pub fn set_xfer(nodeno: u8, busno: u8, chip_select: u16, write_length: u8, read_length: u8) -> Result<(), ()> {
        let request = drtioaux::Packet::SpiSetXferRequest {
            busno: busno,
            chip_select: chip_select,
            write_length: write_length,
            read_length: read_length
        };
        if drtioaux::hw::send(nodeno, &request).is_err() {
            return Err(())
        }
        basic_reply(nodeno)
    }

    pub fn write(nodeno: u8, busno: u8, data: u32) -> Result<(), ()> {
        let request = drtioaux::Packet::SpiWriteRequest {
            busno: busno,
            data: data
        };
        if drtioaux::hw::send(nodeno, &request).is_err() {
            return Err(())
        }
        basic_reply(nodeno)
    }

    pub fn read(nodeno: u8, busno: u8) -> Result<u32, ()> {
        let request = drtioaux::Packet::SpiReadRequest { busno: busno };
        if drtioaux::hw::send(nodeno, &request).is_err() {
            return Err(())
        }
        match drtioaux::hw::recv_timeout(nodeno, None) {
            Ok(drtioaux::Packet::SpiReadReply { succeeded, data }) => {
                if succeeded { Ok(data) } else { Err(()) }
            }
            Ok(_) => {
                error!("received unexpected aux packet");
                Err(())
            }
            Err(e) => {
                error!("aux packet error ({})", e);
                Err(())
            }
        }
    }
}

#[cfg(not(has_drtio))]
mod drtio_spi {
    pub fn set_config(_nodeno: u8, _busno: u8, _flags: u8,
                      _write_div: u8, _read_div: u8) -> Result<(), ()> {
        Err(())
    }

    pub fn set_xfer(_nodeno: u8, _busno: u8, _chip_select: u16,
                    _write_length: u8, _read_length: u8) -> Result<(), ()> {
        Err(())
    }

    pub fn write(_nodeno: u8, _busno: u8, _data: u32) -> Result<(), ()> {
        Err(())
    }

    pub fn read(_nodeno: u8, _busno: u8) -> Result<u32, ()> {
        Err(())
    }
}

mod spi {
    use board_artiq::spi as local_spi;
    use super::drtio_spi;

    pub fn set_config(busno: u32, flags: u8, write_div: u8, read_div: u8) -> Result<(), ()> {
        let nodeno = (busno >> 16) as u8;
        let node_busno = busno as u8;
        if nodeno == 0 {
            local_spi::set_config(node_busno, flags, write_div, read_div)
        } else {
            drtio_spi::set_config(nodeno, node_busno, flags, write_div, read_div)
        }
    }

    pub fn set_xfer(busno: u32, chip_select: u16, write_length: u8, read_length: u8) -> Result<(), ()> {
        let nodeno = (busno >> 16) as u8;
        let node_busno = busno as u8;
        if nodeno == 0 {
            local_spi::set_xfer(node_busno, chip_select, write_length, read_length)
        } else {
            drtio_spi::set_xfer(nodeno, node_busno, chip_select, write_length, read_length)
        }
    }

    pub fn write(busno: u32, data: u32) -> Result<(), ()> {
        let nodeno = (busno >> 16) as u8;
        let node_busno = busno as u8;
        if nodeno == 0 {
            local_spi::write(node_busno, data)
        } else {
            drtio_spi::write(nodeno, node_busno, data)
        }
    }

    pub fn read(busno: u32) -> Result<u32, ()> {
        let nodeno = (busno >> 16) as u8;
        let node_busno = busno as u8;
        if nodeno == 0 {
            local_spi::read(node_busno)
        } else {
            drtio_spi::read(nodeno, node_busno)
        }
    }
}

pub fn process_kern_hwreq(io: &Io, request: &kern::Message) -> io::Result<bool> {
    match request {
        #[cfg(has_rtio_core)]
        &kern::RtioInitRequest => {
            info!("resetting RTIO");
            rtio_mgt::init_core();
            kern_acknowledge()
        }

        #[cfg(has_rtio_core)]
        &kern::DrtioChannelStateRequest { channel } => {
            let (fifo_space, last_timestamp) = rtio_mgt::drtio_dbg::get_channel_state(channel);
            kern_send(io, &kern::DrtioChannelStateReply { fifo_space: fifo_space,
                                                          last_timestamp: last_timestamp })
        }
        #[cfg(has_rtio_core)]
        &kern::DrtioResetChannelStateRequest { channel } => {
            rtio_mgt::drtio_dbg::reset_channel_state(channel);
            kern_acknowledge()
        }
        #[cfg(has_rtio_core)]
        &kern::DrtioGetFifoSpaceRequest { channel } => {
            rtio_mgt::drtio_dbg::get_fifo_space(channel);
            kern_acknowledge()
        }
        #[cfg(has_rtio_core)]
        &kern::DrtioPacketCountRequest { linkno } => {
            let (tx_cnt, rx_cnt) = rtio_mgt::drtio_dbg::get_packet_counts(linkno);
            kern_send(io, &kern::DrtioPacketCountReply { tx_cnt: tx_cnt, rx_cnt: rx_cnt })
        }
        #[cfg(has_rtio_core)]
        &kern::DrtioFifoSpaceReqCountRequest { linkno } => {
            let cnt = rtio_mgt::drtio_dbg::get_fifo_space_req_count(linkno);
            kern_send(io, &kern::DrtioFifoSpaceReqCountReply { cnt: cnt })
        }

        &kern::I2cStartRequest { busno } => {
            let succeeded = i2c::start(busno).is_ok();
            kern_send(io, &kern::I2cBasicReply { succeeded: succeeded })
        }
        &kern::I2cRestartRequest { busno } => {
            let succeeded = i2c::restart(busno).is_ok();
            kern_send(io, &kern::I2cBasicReply { succeeded: succeeded })
        }
        &kern::I2cStopRequest { busno } => {
            let succeeded = i2c::stop(busno).is_ok();
            kern_send(io, &kern::I2cBasicReply { succeeded: succeeded })
        }
        &kern::I2cWriteRequest { busno, data } => {
            match i2c::write(busno, data) {
                Ok(ack) => kern_send(io, &kern::I2cWriteReply { succeeded: true, ack: ack }),
                Err(_) => kern_send(io, &kern::I2cWriteReply { succeeded: false, ack: false })
            }
        }
        &kern::I2cReadRequest { busno, ack } => {
            match i2c::read(busno, ack) {
                Ok(data) => kern_send(io, &kern::I2cReadReply { succeeded: true, data: data }),
                Err(_) => kern_send(io, &kern::I2cReadReply { succeeded: false, data: 0xff })
            }
        }

        &kern::SpiSetConfigRequest { busno, flags, write_div, read_div } => {
            let succeeded = spi::set_config(busno, flags, write_div, read_div).is_ok();
            kern_send(io, &kern::SpiBasicReply { succeeded: succeeded })
        },
        &kern::SpiSetXferRequest { busno, chip_select, write_length, read_length } => {
            let succeeded = spi::set_xfer(busno, chip_select, write_length, read_length).is_ok();
            kern_send(io, &kern::SpiBasicReply { succeeded: succeeded })
        }
        &kern::SpiWriteRequest { busno, data } => {
            let succeeded = spi::write(busno, data).is_ok();
            kern_send(io, &kern::SpiBasicReply { succeeded: succeeded })
        }
        &kern::SpiReadRequest { busno } => {
            match spi::read(busno) {
                Ok(data) => kern_send(io, &kern::SpiReadReply { succeeded: true, data: data }),
                Err(_) => kern_send(io, &kern::SpiReadReply { succeeded: false, data: 0 })
            }
        }

        _ => return Ok(false)
    }.and(Ok(true))
}