Início / user-2902833

Finomnis's questions

Martin Hope
Finomnis
Asked: 2023-12-05 20:08:01 +0800 CST
  • 6

Estou no processo de implementação de um driver SPI.

Agora tenho o seguinte código de driver (mre-ificado):

use tokio::{join, sync::mpsc};

async fn spi_transmit(write_buf: &[u32], read_buf: &mut [u32]) {
    assert_eq!(read_buf.len(), write_buf.len());
    let (write_fifo, mut read_fifo) = mpsc::channel(2);

    let write_task = async {
        // Simulate an SPI bus that respondes with the sent data + 20,
        // just for demo purposes
        for val in write_buf {
            write_fifo.send(*val + 20).await.unwrap();
        }
    };

    let read_task = async {
        for val in read_buf {
            *val = read_fifo.recv().await.unwrap();
        }
    };

    join!(write_task, read_task);
}

#[tokio::main]
async fn main() {
    let buf_out = [1, 2, 3, 4];
    let mut buf_in = [0, 0, 0, 0];

    spi_transmit(&buf_out, &mut buf_in).await;

    println!("{:?}", buf_in);
}

[21, 22, 23, 24]

A API central é async fn spi_transmit(write_buf: &[u32], read_buf: &mut [u32]).

Agora preciso implementar um async fn spi_transmit_in_place(read_write_buf: &mut [u32])que transmita e receba do mesmo buffer, que no final contém os dados lidos.

Eu sei que o acesso de leitura e gravação ao buffer nunca se sobreporá, um byte sempre será lido primeiro e depois, em um momento sem sobreposição, será gravado.

Com esse conhecimento, é considerado correto implementá-lo dessa forma ou isso é considerado um comportamento indefinido?

async fn spi_transmit_in_place(read_write_buf: &mut [u32]) {
    let write_buf = unsafe {
        let data = read_write_buf.as_ptr();
        let len = read_write_buf.len();
        std::slice::from_raw_parts(data, len)
    };

    spi_transmit(write_buf, read_write_buf).await
}

#[tokio::main]
async fn main() {
    let mut buf = [1, 2, 3, 4];
    spi_transmit_in_place(&mut buf).await;
    println!("{:?}", buf);
}

[21, 22, 23, 24]

Funciona, mas miri não está feliz .

Se este for um comportamento indefinido, isso significa que sou forçado a reimplementar spi_transmitusando ponteiros brutos? Ou de que outra forma eu poderia resolver esse problema?

rust