starry-kernel 0.10.0

A Linux-compatible OS kernel built on ArceOS unikernel
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

use ax_runtime::task::{
    runtime::resource::MembarrierRegistration,
    sync::membarrier::{
        MembarrierCommand, MembarrierError, membarrier, register_current_membarrier,
    },
    thread::TaskError,
};
use linux_raw_sys::general::membarrier_cmd;

use crate::StarryError;

/// Memory barrier commands
const MEMBARRIER_CMD_QUERY: i32 = membarrier_cmd::MEMBARRIER_CMD_QUERY as i32;
const MEMBARRIER_CMD_GLOBAL: i32 = membarrier_cmd::MEMBARRIER_CMD_GLOBAL as i32;
const MEMBARRIER_CMD_GLOBAL_EXPEDITED: i32 = membarrier_cmd::MEMBARRIER_CMD_GLOBAL_EXPEDITED as i32;
const MEMBARRIER_CMD_REGISTER_GLOBAL_EXPEDITED: i32 =
    membarrier_cmd::MEMBARRIER_CMD_REGISTER_GLOBAL_EXPEDITED as i32;
const MEMBARRIER_CMD_PRIVATE_EXPEDITED: i32 =
    membarrier_cmd::MEMBARRIER_CMD_PRIVATE_EXPEDITED as i32;
const MEMBARRIER_CMD_REGISTER_PRIVATE_EXPEDITED: i32 =
    membarrier_cmd::MEMBARRIER_CMD_REGISTER_PRIVATE_EXPEDITED as i32;

/// Supported command flags for query
const SUPPORTED_COMMANDS: i32 = MEMBARRIER_CMD_GLOBAL
    | MEMBARRIER_CMD_GLOBAL_EXPEDITED
    | MEMBARRIER_CMD_REGISTER_GLOBAL_EXPEDITED
    | MEMBARRIER_CMD_PRIVATE_EXPEDITED
    | MEMBARRIER_CMD_REGISTER_PRIVATE_EXPEDITED;

fn map_membarrier_error(error: MembarrierError) -> crate::StarryError {
    match error {
        MembarrierError::NotRegistered => crate::StarryError::OperationNotPermitted,
        MembarrierError::Task(TaskError::InvalidConfiguration) => crate::StarryError::InvalidInput,
        MembarrierError::Task(TaskError::UnsafeContext) => {
            crate::StarryError::OperationNotPermitted
        }
        MembarrierError::Task(_) => crate::StarryError::BadState,
    }
}

fn map_membarrier_registration_error(error: TaskError) -> crate::StarryError {
    map_membarrier_error(MembarrierError::Task(error))
}

#[derive(Clone, Copy, Debug, Eq, PartialEq)]
enum MembarrierAction {
    Query,
    Global,
    RegisterGlobalExpedited,
    GlobalExpedited,
    RegisterPrivateExpedited,
    PrivateExpedited,
}

fn decode_membarrier_action(cmd: i32, flags: u32) -> crate::StarryResult<MembarrierAction> {
    if flags != 0 {
        return Err(StarryError::InvalidInput);
    }
    match cmd {
        MEMBARRIER_CMD_QUERY => Ok(MembarrierAction::Query),
        MEMBARRIER_CMD_GLOBAL => Ok(MembarrierAction::Global),
        MEMBARRIER_CMD_REGISTER_GLOBAL_EXPEDITED => Ok(MembarrierAction::RegisterGlobalExpedited),
        MEMBARRIER_CMD_GLOBAL_EXPEDITED => Ok(MembarrierAction::GlobalExpedited),
        MEMBARRIER_CMD_REGISTER_PRIVATE_EXPEDITED => Ok(MembarrierAction::RegisterPrivateExpedited),
        MEMBARRIER_CMD_PRIVATE_EXPEDITED => Ok(MembarrierAction::PrivateExpedited),
        _ => Err(crate::StarryError::InvalidInput),
    }
}

pub fn sys_membarrier(
    _current: &crate::task::UserTaskRef,
    cmd: i32,
    flags: u32,
    _cpu_id: i32,
) -> crate::StarryResult<isize> {
    match decode_membarrier_action(cmd, flags)? {
        MembarrierAction::Query => Ok(SUPPORTED_COMMANDS as isize),
        MembarrierAction::Global => {
            membarrier(MembarrierCommand::Global).map_err(map_membarrier_error)?;
            Ok(0)
        }
        MembarrierAction::RegisterGlobalExpedited => {
            register_current_membarrier(MembarrierRegistration::GlobalExpedited)
                .map_err(map_membarrier_registration_error)?;
            Ok(0)
        }
        MembarrierAction::GlobalExpedited => {
            membarrier(MembarrierCommand::GlobalExpedited).map_err(map_membarrier_error)?;
            Ok(0)
        }
        MembarrierAction::RegisterPrivateExpedited => {
            register_current_membarrier(MembarrierRegistration::PrivateExpedited)
                .map_err(map_membarrier_registration_error)?;
            Ok(0)
        }
        MembarrierAction::PrivateExpedited => {
            membarrier(MembarrierCommand::PrivateExpedited).map_err(map_membarrier_error)?;
            Ok(0)
        }
    }
}

#[cfg(all(test, not(axtest)))]
fn membarrier_query_and_global_rules_hold_for_test() -> bool {
    matches!(
        decode_membarrier_action(MEMBARRIER_CMD_QUERY, 0),
        Ok(MembarrierAction::Query)
    ) && matches!(
        decode_membarrier_action(MEMBARRIER_CMD_QUERY, 1),
        Err(crate::StarryError::InvalidInput)
    ) && matches!(
        decode_membarrier_action(-1, 0),
        Err(crate::StarryError::InvalidInput)
    ) && matches!(
        decode_membarrier_action(MEMBARRIER_CMD_GLOBAL, 0),
        Ok(MembarrierAction::Global)
    )
}

#[cfg(all(test, not(axtest)))]
mod tests {
    #[test]
    fn membarrier_query_and_global_rules_hold() {
        assert!(super::membarrier_query_and_global_rules_hold_for_test());
    }
}