capdl: add FPU support flag

Signed-off-by: Jakub Duchniewicz <j.duchniewicz@unsw.edu.au>

Author: JDuchniewicz

crates/sel4-capdl-initializer/src/initialize.rs

@@ -766,6 +766,8 @@ impl<'a> Initializer<'a> {
                     }
                 }
 
+                let fpu_disabled = obj.extra.fpu_disabled;
+
                 #[allow(unused_variables)]
                 let affinity = obj.extra.affinity.to_sel4();
 
@@ -822,6 +824,14 @@ impl<'a> Initializer<'a> {
                             fault_ep,
                         )?;
 
+                        let tcb_flags = sel4::TcbFlagsBuilder::new()
+                            .fpu_disabled(fpu_disabled)
+                            .build();
+                        tcb.tcb_set_flags(
+                            0,
+                            tcb_flags,
+                        )?;
+
                         tcb.tcb_set_timeout_endpoint(temp_fault_ep)?;
                     } else {
                         let fault_ep = sel4::CPtr::from_bits(obj.extra.master_fault_ep.as_ref().unwrap().to_sel4());
@@ -841,6 +851,14 @@ impl<'a> Initializer<'a> {
                             prio,
                         )?;
 
+                        let tcb_flags = sel4::TcbFlagsBuilder::new()
+                            .fpu_disabled(fpu_disabled)
+                            .build();
+                        tcb.tcb_set_flags(
+                            0,
+                            tcb_flags,
+                        )?;
+
                         sel4::sel4_cfg_if! {
                             if #[sel4_cfg(not(MAX_NUM_NODES = "1"))] {
                                 tcb.tcb_set_affinity(affinity)?;

crates/sel4-capdl-initializer/types/src/spec.rs

@@ -419,6 +419,7 @@ pub mod object {
         pub affinity: Word,
         pub prio: u8,
         pub max_prio: u8,
+        pub fpu_disabled: bool,
         pub resume: bool,
 
         pub domain: Option<u8>,

crates/sel4/src/invocations.rs

@@ -49,6 +49,32 @@ impl<C: InvocationContext> Untyped<C> {
 const USER_CONTEXT_MAX_REG_COUNT: usize =
     mem::size_of::<sys::seL4_UserContext>() / mem::size_of::<Word>();
 
+#[derive(Debug, Clone, PartialEq)]
+pub struct TcbFlagsBuilder(Word);
+
+impl TcbFlagsBuilder {
+    pub fn new() -> Self {
+        Self(sel4_sys::seL4_TCBFlag::seL4_TCBFlag_NoFlag)
+    }
+
+    pub fn build(self) -> Self {
+        self.0
+    }
+
+    pub fn fpu_disabled(mut self, val: bool) -> Self {
+        self.apply_flag_val(sel4_sys::seL4_TCBFlag::seL4_TCBFlag_fpuDisabled, val)
+    }
+
+    pub fn apply_flag_val(mut self, flag: Word, val: bool) -> Self {
+        if val {
+            self.0 |= flag
+        } else {
+            self.0 &= !flag
+        }
+        self
+    }
+}
+
 impl<C: InvocationContext> Tcb<C> {
     /// Corresponds to `seL4_TCB_ReadRegisters`.
     pub fn tcb_read_registers(self, suspend: bool, count: Word) -> Result<UserContext> {
@@ -124,6 +150,16 @@ impl<C: InvocationContext> Tcb<C> {
         }))
     }
 
+    // Corresponds to `seL4_TCB_SetFlags`.
+    pub fn tcb_set_flags(self, clear: Word, set: Word) -> Result<Word> {
+        let ret = self.invoke(|cptr, ipc_buffer| {
+            ipc_buffer
+                .inner_mut()
+                .seL4_TCB_SetFlags(cptr.bits(), clear, set)
+        });
+        Error::or(ret.error, ret.flags)
+    }
+
     sel4_cfg_if! {
         if #[sel4_cfg(KERNEL_MCS)] {
             /// Corresponds to `seL4_TCB_Configure`.

crates/sel4/src/lib.rs

@@ -125,6 +125,7 @@ pub use cptr::{
 pub use error::{Error, Result};
 pub use fault::*;
 pub use invocation_context::{InvocationContext, NoExplicitInvocationContext, NoInvocationContext};
+pub use invocations::TcbFlagsBuilder;
 pub use ipc_buffer::IpcBuffer;
 pub use message_info::{MessageInfo, MessageInfoBuilder};
 pub use object::{