rustsbi/prototyper/src/firmware/payload.rs

use core::arch::naked_asm;
use core::sync::atomic::{AtomicBool, Ordering};
use riscv::register::mstatus;

use super::BootInfo;

/// Determine whether the current hart is boot hart.
///
/// Return true if the current hart is boot hart.
pub fn is_boot_hart(_nonstandard_a2: usize) -> bool {
    static GENESIS: AtomicBool = AtomicBool::new(true);
    GENESIS.swap(false, Ordering::AcqRel)
}

pub fn get_boot_info(_nonstandard_a2: usize) -> BootInfo {
    BootInfo {
        next_address: get_image_address(),
        mpp: mstatus::MPP::Supervisor,
    }
}

#[naked]
#[unsafe(link_section = ".payload")]
pub extern "C" fn payload_image() {
    unsafe { naked_asm!(concat!(".incbin \"", env!("PROTOTYPER_PAYLOAD_PATH"), "\""),) }
}

#[inline]
fn get_image_address() -> usize {
    payload_image as usize
}