uboot-shell 0.2.3

A crate for communicating with u-boot
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use std::{
    process::{Child, Command},
    sync::atomic::AtomicU32,
};

use log::{debug, info};
use ntest::timeout;
use tokio::{
    net::TcpStream,
    time::{Duration, sleep},
};
use tokio_util::compat::{TokioAsyncReadCompatExt, TokioAsyncWriteCompatExt};
use uboot_shell::UbootShell;

static PORT: AtomicU32 = AtomicU32::new(10000);

async fn new_uboot() -> (Child, UbootShell) {
    let port = PORT.fetch_add(1, std::sync::atomic::Ordering::SeqCst);

    // qemu-system-aarch64 -machine virt -cpu cortex-a57 -nographic -bios assets/u-boot.bin
    let out = Command::new("qemu-system-aarch64")
        .arg("-serial")
        .arg(format!("tcp::{port},server,nowait"))
        .args([
            "-machine",
            "virt",
            "-cpu",
            "cortex-a57",
            "-nographic",
            "-bios",
            "../assets/u-boot.bin",
        ])
        .spawn()
        .unwrap();

    loop {
        sleep(Duration::from_millis(100)).await;
        match TcpStream::connect(format!("127.0.0.1:{port}")).await {
            Ok(s) => {
                let (rx, tx) = s.into_split();
                info!("connect ok");
                return (
                    out,
                    UbootShell::new(tx.compat_write(), rx.compat())
                        .await
                        .unwrap(),
                );
            }
            Err(e) => {
                debug!("wait for qemu serial port ready: {e}");
            }
        }
    }
}

#[tokio::test]
#[timeout(15000)]
async fn test_shell() {
    let (mut out, _uboot) = new_uboot().await;
    info!("test_shell ok");
    let _ = out.kill();
    out.wait().unwrap();
}

#[tokio::test]
#[timeout(15000)]
async fn test_cmd() {
    let (mut out, mut uboot) = new_uboot().await;
    let res = uboot.cmd("help").await.unwrap();
    println!("{}", res);
    let _ = out.kill();
    out.wait().unwrap();
}

#[tokio::test]
#[timeout(15000)]
async fn test_setenv() {
    let (mut out, mut uboot) = new_uboot().await;
    uboot.set_env("ipaddr", "127.0.0.1").await.unwrap();
    let _ = out.kill();
    out.wait().unwrap();
}

#[tokio::test]
#[timeout(15000)]
async fn test_env() {
    let (mut out, mut uboot) = new_uboot().await;
    uboot.set_env("fdt_addr", "0x40000000").await.unwrap();
    info!("set fdt_addr ok");
    assert_eq!(uboot.env_int("fdt_addr").await.unwrap(), 0x40000000);
    let _ = out.kill();
    out.wait().unwrap();
}