ax-fs-ng 0.7.0

ArceOS filesystem module
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

use alloc::{boxed::Box, string::String};
use core::sync::atomic::{AtomicBool, Ordering};

use ax_errno::AxResult;
use ax_kspin::SpinRwLock as RwLock;
use irq_framework::{IrqContext, IrqId};

use crate::block::runtime::BlockIrqAction;

#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum BlockIrqOutcome {
    Handled,
    Wake,
}

pub trait BlockIrqRegistration: Send + Sync {}

pub trait BlockIrqRegistrar: Send + Sync {
    fn register_shared(
        &self,
        name: String,
        irq: IrqId,
        action: Box<dyn FnMut(IrqContext) -> BlockIrqOutcome + Send + 'static>,
    ) -> AxResult<Box<dyn BlockIrqRegistration>>;
}

static IRQ_REGISTRAR: RwLock<Option<&'static dyn BlockIrqRegistrar>> = RwLock::new(None);
static IRQ_READY: AtomicBool = AtomicBool::new(false);

pub fn set_irq_registrar(registrar: &'static dyn BlockIrqRegistrar) {
    *IRQ_REGISTRAR.write() = Some(registrar);
    IRQ_READY.store(true, Ordering::Release);
}

pub fn register_shared_block_irq(
    name: String,
    irq: IrqId,
    action: BlockIrqAction,
) -> AxResult<Box<dyn BlockIrqRegistration>> {
    let registrar = IRQ_REGISTRAR
        .read()
        .as_ref()
        .copied()
        .ok_or(ax_errno::AxError::BadState)?;
    let mut action = action;
    registrar.register_shared(name, irq, Box::new(move |_ctx| action.run()))
}

pub fn has_irq_registrar() -> bool {
    IRQ_READY.load(Ordering::Acquire)
}